def parse_terminal(self, pstate):
scope = self.tree_walker.scope_stack[-1]
from pymbolic.primitives import Subscript, Call, Variable
from pymbolic.parser import (
_identifier, _openpar, _closepar, _float)
next_tag = pstate.next_tag()
if next_tag is _float:
value = pstate.next_str_and_advance().lower()
if "d" in value:
dtype = np.float64
else:
dtype = np.float32
value = value.replace("d", "e")
if value.startswith("."):
prev_value = value
value = "0"+value
print value, prev_value
elif value.startswith("-."):
prev_value = value
value = "-0"+value[1:]
print value, prev_value
return TypedLiteral(value, dtype)
elif next_tag is _identifier:
name = pstate.next_str_and_advance()
if pstate.is_at_end() or pstate.next_tag() is not _openpar:
# not a subscript
scope.use_name(name)
return Variable(name)
left_exp = Variable(name)
pstate.advance()
pstate.expect_not_end()
if scope.is_known(name):
cls = Subscript
else:
cls = Call
if pstate.next_tag is _closepar:
pstate.advance()
left_exp = cls(left_exp, ())
else:
args = self.parse_expression(pstate, self._PREC_FUNC_ARGS)
if not isinstance(args, tuple):
args = (args,)
left_exp = cls(left_exp, args)
pstate.expect(_closepar)
pstate.advance()
return left_exp
else:
return ExpressionParserBase.parse_terminal(
self, pstate)
def parse_postfix(self, pstate, min_precedence, left_exp):
from pymbolic.parser import (
_PREC_CALL, _PREC_COMPARISON, _openpar,
_PREC_LOGICAL_OR, _PREC_LOGICAL_AND)
from pymbolic.primitives import (
Comparison, LogicalAnd, LogicalOr)
next_tag = pstate.next_tag()
if next_tag is _openpar and _PREC_CALL > min_precedence:
raise TranslationError("parenthesis operator only works on names")
elif next_tag in self.COMP_MAP and _PREC_COMPARISON > min_precedence:
pstate.advance()
left_exp = Comparison(
left_exp,
self.COMP_MAP[next_tag],
self.parse_expression(pstate, _PREC_COMPARISON))
did_something = True
elif next_tag is _and and _PREC_LOGICAL_AND > min_precedence:
pstate.advance()
left_exp = LogicalAnd((left_exp,
self.parse_expression(pstate, _PREC_LOGICAL_AND)))
did_something = True
elif next_tag is _or and _PREC_LOGICAL_OR > min_precedence:
pstate.advance()
left_exp = LogicalOr((left_exp,
self.parse_expression(pstate, _PREC_LOGICAL_OR)))
did_something = True
else:
left_exp, did_something = ExpressionParserBase.parse_postfix(
self, pstate, min_precedence, left_exp)
return left_exp, did_something
def parse_terminal(self, pstate):
scope = self.tree_walker.scope_stack[-1]
from pymbolic.primitives import Subscript, Call, Variable
from pymbolic.parser import (_identifier, _openpar, _closepar, _float)
next_tag = pstate.next_tag()
if next_tag is _float:
value = pstate.next_str_and_advance().lower()
if "d" in value:
dtype = np.float64
else:
dtype = np.float32
value = value.replace("d", "e")
if value.startswith("."):
prev_value = value
value = "0" + value
print value, prev_value
elif value.startswith("-."):
prev_value = value
value = "-0" + value[1:]
print value, prev_value
return TypedLiteral(value, dtype)
elif next_tag is _identifier:
name = pstate.next_str_and_advance()
if pstate.is_at_end() or pstate.next_tag() is not _openpar:
# not a subscript
scope.use_name(name)
return Variable(name)
left_exp = Variable(name)
pstate.advance()
pstate.expect_not_end()
if scope.is_known(name):
cls = Subscript
else:
cls = Call
if pstate.next_tag is _closepar:
pstate.advance()
left_exp = cls(left_exp, ())
else:
args = self.parse_expression(pstate, self._PREC_FUNC_ARGS)
if not isinstance(args, tuple):
args = (args, )
left_exp = cls(left_exp, args)
pstate.expect(_closepar)
pstate.advance()
return left_exp
else:
return ExpressionParserBase.parse_terminal(self, pstate)
def parse_prefix(self, pstate, min_precedence=0):
from pymbolic.parser import _PREC_UNARY
import pymbolic.primitives as primitives
pstate.expect_not_end()
if pstate.is_next(_not):
pstate.advance()
return primitives.LogicalNot(self.parse_expression(pstate, _PREC_UNARY))
else:
return ExpressionParserBase.parse_prefix(self, pstate)
def parse_prefix(self, pstate, min_precedence=0):
from pymbolic.parser import _PREC_UNARY
import pymbolic.primitives as primitives
pstate.expect_not_end()
if pstate.is_next(_not):
pstate.advance()
return primitives.LogicalNot(
self.parse_expression(pstate, _PREC_UNARY))
else:
return ExpressionParserBase.parse_prefix(self, pstate)
def parse_postfix(self, pstate, min_precedence, left_exp):
from pymbolic.parser import _PREC_CALL, _closebracket
if pstate.next_tag() is _open_dbl_bracket and _PREC_CALL > min_precedence:
pstate.advance()
pstate.expect_not_end()
left_exp = LinearSubscript(left_exp, self.parse_expression(pstate))
pstate.expect(_closebracket)
pstate.advance()
pstate.expect(_closebracket)
pstate.advance()
return left_exp, True
return ParserBase.parse_postfix(self, pstate, min_precedence, left_exp)
def parse_postfix(self, pstate, min_precedence, left_exp):
from pymbolic.parser import _PREC_CALL, _closebracket
if pstate.next_tag() is _open_dbl_bracket and _PREC_CALL > min_precedence:
pstate.advance()
pstate.expect_not_end()
left_exp = LinearSubscript(left_exp, self.parse_expression(pstate))
pstate.expect(_closebracket)
pstate.advance()
pstate.expect(_closebracket)
pstate.advance()
return left_exp, True
return ParserBase.parse_postfix(self, pstate, min_precedence, left_exp)
def parse_postfix(self, pstate, min_precedence, left_exp):
from pymbolic.parser import (
_PREC_CALL, _PREC_COMPARISON, _openpar,
_PREC_LOGICAL_OR, _PREC_LOGICAL_AND)
from pymbolic.primitives import (
Comparison, LogicalAnd, LogicalOr)
next_tag = pstate.next_tag()
if next_tag is _openpar and _PREC_CALL > min_precedence:
raise TranslationError("parenthesis operator only works on names")
elif next_tag in self.COMP_MAP and _PREC_COMPARISON > min_precedence:
pstate.advance()
left_exp = Comparison(
left_exp,
self.COMP_MAP[next_tag],
self.parse_expression(pstate, _PREC_COMPARISON))
did_something = True
elif next_tag is _and and _PREC_LOGICAL_AND > min_precedence:
pstate.advance()
left_exp = LogicalAnd((left_exp,
self.parse_expression(pstate, _PREC_LOGICAL_AND)))
did_something = True
elif next_tag is _or and _PREC_LOGICAL_OR > min_precedence:
pstate.advance()
left_exp = LogicalOr((left_exp,
self.parse_expression(pstate, _PREC_LOGICAL_OR)))
did_something = True
else:
left_exp, did_something = ExpressionParserBase.parse_postfix(
self, pstate, min_precedence, left_exp)
if isinstance(left_exp, tuple) and min_precedence < self._PREC_FUNC_ARGS:
# this must be a complex literal
if len(left_exp) != 2:
raise TranslationError("complex literals must have "
"two entries")
r, i = left_exp
dtype = (r.dtype.type(0) + i.dtype.type(0))
if dtype == np.float32:
dtype = np.complex64
else:
dtype = np.complex128
left_exp = dtype(float(r) + float(i)*1j)
return left_exp, did_something
def parse_postfix(self, pstate, min_precedence, left_exp):
from pymbolic.parser import (_PREC_CALL, _PREC_COMPARISON, _openpar,
_PREC_LOGICAL_OR, _PREC_LOGICAL_AND)
from pymbolic.primitives import (Comparison, LogicalAnd, LogicalOr)
next_tag = pstate.next_tag()
if next_tag is _openpar and _PREC_CALL > min_precedence:
raise TranslationError("parenthesis operator only works on names")
elif next_tag in self.COMP_MAP and _PREC_COMPARISON > min_precedence:
pstate.advance()
left_exp = Comparison(
left_exp, self.COMP_MAP[next_tag],
self.parse_expression(pstate, _PREC_COMPARISON))
did_something = True
elif next_tag is _and and _PREC_LOGICAL_AND > min_precedence:
pstate.advance()
left_exp = LogicalAnd(
(left_exp, self.parse_expression(pstate, _PREC_LOGICAL_AND)))
did_something = True
elif next_tag is _or and _PREC_LOGICAL_OR > min_precedence:
pstate.advance()
left_exp = LogicalOr(
(left_exp, self.parse_expression(pstate, _PREC_LOGICAL_OR)))
did_something = True
else:
left_exp, did_something = ExpressionParserBase.parse_postfix(
self, pstate, min_precedence, left_exp)
if isinstance(left_exp,
tuple) and min_precedence < self._PREC_FUNC_ARGS:
# this must be a complex literal
if len(left_exp) != 2:
raise TranslationError("complex literals must have "
"two entries")
r, i = left_exp
dtype = (r.dtype.type(0) + i.dtype.type(0))
if dtype == np.float32:
dtype = np.complex64
else:
dtype = np.complex128
left_exp = dtype(float(r) + float(i) * 1j)
return left_exp, did_something
def parse(s: str) -> ScalarExpression:
from pymbolic.parser import Parser
return Parser()(s)