daScript/src/simulate/simulate_fusion_at_array.cpp at master · Xsankor/daScript · GitHub

History

158 lines (131 loc) · 6.01 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

#include "daScript/misc/platform.h"

#ifdef _MSC_VER

#pragma warning(disable:4505)

#endif

#include "daScript/simulate/simulate_fusion.h"

#if DAS_FUSION

#include "daScript/simulate/runtime_array.h"

#include "daScript/simulate/sim_policy.h"

#include "daScript/ast/ast.h"

#include "daScript/simulate/simulate_visit_op.h"

namespace das {

struct SimNode_Op2ArrayAt : SimNode_Op2Fusion {

virtual SimNode * visit(SimVisitor & vis) override {

V_BEGIN();

string name = op;

name += getSimSourceName(l.type);

name += getSimSourceName(r.type);

if (baseType != Type::none) {

vis.op(name.c_str(), getTypeBaseSize(baseType), das_to_string(baseType));

} else {

vis.op(name.c_str());

}

l.visit(vis);

r.visit(vis);

V_ARG(stride);

V_ARG(offset);

V_END();

}

uint32_t stride, offset;

};

/* ArrayAtR2V SCALAR */

#define IMPLEMENT_OP2_SET_NODE_ANY(INLINE,OPNAME,TYPE,CTYPE,COMPUTEL) \

struct SimNode_##OPNAME##_##COMPUTEL##_Any : SimNode_Op2ArrayAt { \

INLINE auto compute ( Context & context ) { \

DAS_PROFILE_NODE \

auto pl = (Array *) l.compute##COMPUTEL(context); \

auto rr = uint32_t(r.subexpr->evalInt(context)); \

if ( rr >= pl->size ) context.throw_error_at(debugInfo,"array index out of range, %u of %u", rr, pl->size); \

return *((CTYPE *)(pl->data + rr*stride + offset)); \

} \

DAS_NODE(TYPE,CTYPE); \

};

#define IMPLEMENT_OP2_SET_NODE(INLINE,OPNAME,TYPE,CTYPE,COMPUTEL,COMPUTER) \

struct SimNode_##OPNAME##_##COMPUTEL##_##COMPUTER : SimNode_Op2ArrayAt { \

INLINE auto compute ( Context & context ) { \

DAS_PROFILE_NODE \

auto pl = (Array *) l.compute##COMPUTEL(context); \

auto rr = *((uint32_t *)r.compute##COMPUTER(context)); \

if ( rr >= pl->size ) context.throw_error_at(debugInfo,"array index out of range, %u of %u", rr, pl->size); \

return *((CTYPE *)(pl->data + rr*stride + offset)); \

} \

DAS_NODE(TYPE,CTYPE); \

};

#define IMPLEMENT_OP2_SET_SETUP_NODE(result,node) \

auto rn = (SimNode_Op2ArrayAt *)result; \

auto sn = (SimNode_ArrayAt *)node; \

rn->stride = sn->stride; \

rn->offset = sn->offset;

#define FUSION_OP2_SUBEXPR_LEFT(CTYPE,node) ((static_cast<SimNode_ArrayAt *>(node))->l)

#define FUSION_OP2_SUBEXPR_RIGHT(CTYPE,node) ((static_cast<SimNode_ArrayAt *>(node))->r)

#include "daScript/simulate/simulate_fusion_op2_set_impl.h"

#include "daScript/simulate/simulate_fusion_op2_set_perm.h"

IMPLEMENT_SETOP_SCALAR(ArrayAtR2V);

/* ArrayAtR2V VECTOR */

#undef IMPLEMENT_OP2_SET_NODE_ANY

#define IMPLEMENT_OP2_SET_NODE_ANY(INLINE,OPNAME,TYPE,CTYPE,COMPUTEL) \

struct SimNode_##OPNAME##_##COMPUTEL##_Any : SimNode_Op2ArrayAt { \

virtual vec4f eval ( Context & context ) override { \

DAS_PROFILE_NODE \

auto pl = (Array *) l.compute##COMPUTEL(context); \

auto rr = uint32_t(r.subexpr->evalInt(context)); \

if ( rr >= pl->size ) context.throw_error_at(debugInfo,"array index out of range, %u of %u", rr, pl->size); \

return v_ldu((const float *)(pl->data + rr*stride + offset)); \

} \

};

#undef IMPLEMENT_OP2_SET_NODE

#define IMPLEMENT_OP2_SET_NODE(INLINE,OPNAME,TYPE,CTYPE,COMPUTEL,COMPUTER) \

struct SimNode_##OPNAME##_##COMPUTEL##_##COMPUTER : SimNode_Op2ArrayAt { \

virtual vec4f eval ( Context & context ) override { \

DAS_PROFILE_NODE \

auto pl = (Array *) l.compute##COMPUTEL(context); \

auto rr = *((uint32_t *)r.compute##COMPUTER(context)); \

if ( rr >= pl->size ) context.throw_error_at(debugInfo,"array index out of range, %u of %u", rr, pl->size); \

return v_ldu((const float *)(pl->data + rr*stride + offset)); \

} \

};

#include "daScript/simulate/simulate_fusion_op2_set_impl.h"

#include "daScript/simulate/simulate_fusion_op2_set_perm.h"

IMPLEMENT_SETOP_NUMERIC_VEC(ArrayAtR2V);

/* ArrayAt */

#undef IMPLEMENT_OP2_SET_NODE_ANY

#define IMPLEMENT_OP2_SET_NODE_ANY(INLINE,OPNAME,TYPE,CTYPE,COMPUTEL) \

struct SimNode_##OPNAME##_##COMPUTEL##_Any : SimNode_Op2ArrayAt { \

INLINE auto compute ( Context & context ) { \

DAS_PROFILE_NODE \

auto pl = (Array *) l.compute##COMPUTEL(context); \

auto rr = uint32_t(r.subexpr->evalInt(context)); \

if ( rr >= pl->size ) context.throw_error_at(debugInfo,"array index out of range, %u of %u", rr, pl->size); \

return pl->data + rr*stride + offset; \

} \

DAS_PTR_NODE; \

};

#undef IMPLEMENT_OP2_SET_NODE

#define IMPLEMENT_OP2_SET_NODE(INLINE,OPNAME,TYPE,CTYPE,COMPUTEL,COMPUTER) \

struct SimNode_##OPNAME##_##COMPUTEL##_##COMPUTER : SimNode_Op2ArrayAt { \

INLINE auto compute ( Context & context ) { \

DAS_PROFILE_NODE \

auto pl = (Array *) l.compute##COMPUTEL(context); \

auto rr = *((uint32_t *)r.compute##COMPUTER(context)); \

if ( rr >= pl->size ) context.throw_error_at(debugInfo,"array index out of range, %u of %u", rr, pl->size); \

return pl->data + rr*stride + offset; \

} \

DAS_PTR_NODE; \

};

#undef IMPLEMENT_OP2_SET_SETUP_NODE

#define IMPLEMENT_OP2_SET_SETUP_NODE(result,node) \

auto rn = (SimNode_Op2ArrayAt *)result; \

auto sn = (SimNode_ArrayAt *)node; \

rn->stride = sn->stride; \

rn->offset = sn->offset; \

rn->baseType = Type::none;

#include "daScript/simulate/simulate_fusion_op2_set_impl.h"

#include "daScript/simulate/simulate_fusion_op2_set_perm.h"

IMPLEMENT_ANY_SETOP(__forceinline, ArrayAt, Ptr, StringPtr, StringPtr);

void createFusionEngine_at_array() {

REGISTER_SETOP_SCALAR(ArrayAtR2V);

REGISTER_SETOP_NUMERIC_VEC(ArrayAtR2V);

(*g_fusionEngine)["ArrayAt"].emplace_back(new FusionPoint_Set_ArrayAt_StringPtr());

}

#endif

daScript/src/simulate/simulate_fusion_at_array.cpp at master · Xsankor/daScript · GitHub

Expand file tree

Breadcrumbs

simulate_fusion_at_array.cpp

Latest commit

History

Breadcrumbs

simulate_fusion_at_array.cpp

File metadata and controls