Skip to content
Snippets Groups Projects

Fix symbol counters

Merged Markus Holzer requested to merge holzer/pystencils:Fixcounting into master
2 files
+ 8
5
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -61,8 +61,11 @@ class AssignmentCollection:
@@ -61,8 +61,11 @@ class AssignmentCollection:
self.simplification_hints = simplification_hints
self.simplification_hints = simplification_hints
 
ctrs = [int(n.name[3:])for n in self.rhs_symbols if "xi_" in n.name]
 
max_ctr = max(ctrs) + 1 if len(ctrs) > 0 else 0
 
if subexpression_symbol_generator is None:
if subexpression_symbol_generator is None:
self.subexpression_symbol_generator = SymbolGen()
self.subexpression_symbol_generator = SymbolGen(ctr=max_ctr)
else:
else:
self.subexpression_symbol_generator = subexpression_symbol_generator
self.subexpression_symbol_generator = subexpression_symbol_generator
@@ -453,8 +456,8 @@ class AssignmentCollection:
@@ -453,8 +456,8 @@ class AssignmentCollection:
class SymbolGen:
class SymbolGen:
"""Default symbol generator producing number symbols ζ_0, ζ_1, ..."""
"""Default symbol generator producing number symbols ζ_0, ζ_1, ..."""
def __init__(self, symbol="xi", dtype=None):
def __init__(self, symbol="xi", dtype=None, ctr=0):
self._ctr = 0
self._ctr = ctr
self._symbol = symbol
self._symbol = symbol
self._dtype = dtype
self._dtype = dtype
Loading