Skip to content
Snippets Groups Projects
Commit 145155f7 authored by Rafael Ravedutti's avatar Rafael Ravedutti
Browse files

Update data for DEM simulation and fix BoolOp mapping

parent c7c04d1e
No related branches found
No related tags found
1 merge request!1Implement DEM and many other features
This diff is collapsed.
This diff is collapsed.
......@@ -131,11 +131,19 @@ class BuildParticleIR(ast.NodeVisitor):
def visit_BoolOp(self, node):
#print(ast.dump(node))
lhs = self.visit(node.values[0])
assert not isinstance(lhs, UndefinedSymbol), f"Undefined lhs used in BoolOp: {lhs.symbol_id}"
rhs = self.visit(node.values[1])
assert not isinstance(rhs, UndefinedSymbol), f"Undefined rhs used in BoolOp: {rhs.symbol_id}"
return ScalarOp(self.sim, lhs, rhs, BuildParticleIR.get_binary_op(node.op))
op = BuildParticleIR.get_binary_op(node.op)
first = self.visit(node.values[0])
assert not isinstance(first, UndefinedSymbol), \
f"Undefined operator used in BoolOp: {first.symbol_id}"
expr = first
for value in node.values[1:]:
voper = self.visit(value)
assert not isinstance(voper, UndefinedSymbol), \
f"Undefined operator used in BoolOp: {voper.symbol_id}"
expr = ScalarOp(self.sim, expr, voper, op)
return expr
def visit_Call(self, node):
func = self.visit(node.func).symbol_id
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment