def map_variable(self, expr):
match_obj = INDEXED_VAR_RE.match(expr.name)
if match_obj is not None:
name = match_obj.group(1)
subscript = int(match_obj.group(2))
if name in self.name_whitelist:
return prim.Variable(name).index(subscript)
else:
return IdentityMapper.map_variable(self, expr)
else:
return IdentityMapper.map_variable(self, expr)
def map_variable(self, expr):
match_obj = INDEXED_VAR_RE.match(expr.name)
if match_obj is not None:
name = match_obj.group(1)
subscript = int(match_obj.group(2))
if name in self.name_whitelist:
return prim.Variable(name).index(subscript)
else:
return IdentityMapper.map_variable(self, expr)
else:
return IdentityMapper.map_variable(self, expr)
def map_variable(self, expr):
if expr.name == "pi":
return prim.Variable("M_PI")
else:
return IdentityMapper.map_variable(self, expr)
def map_variable(self, expr):
if expr.name == "pi":
return prim.Variable("M_PI")
else:
return IdentityMapper.map_variable(self, expr)
