AI fix stack and large refactor + add count

Author: mcDandy

README.md

@@ -178,6 +178,7 @@ You can also get the node from comfy manager under the name of More math.
 - `nan_to_num(x, nan_value, posinf_value, neginf_value)` or `nvl`: Replaces NaN and infinite values in tensor with specified values.
 - `remap(v, i_min, i_max, o_min, o_max)`: Remaps value `v` from input range `[i_min, i_max]` to output range `[o_min, o_max]`.
 - `timestamp()` or `now`: Returns current UNIX timestamp (precision to microseconds, can be different on other systems)
+- `count(x)` or `length(x)` or `cnt(x)`: Returns the length of a list or the size of the first dimension of a tensor.
 
 ### Random Distributions
 

more_math/LatentMathNode.py

@@ -89,7 +89,7 @@ def check_lazy_status(cls, Expression, V, F,batching, length_mismatch="tile",sta
         return needed1
 
     @classmethod
-    def execute(cls, V, F, Expression,batching, length_mismatch="tile",stack=[]) -> io.NodeOutput:
+    def execute(cls, V, F, Expression,batching, length_mismatch="tile",stack=dict()) -> io.NodeOutput:
         # Determine reference latent
         ref_latent = None
         for lat in V.values():

more_math/ModelMathNode.py

@@ -72,7 +72,7 @@ def check_lazy_status(cls, Expression, V, F, length_mismatch="tile",stack=dict()
         return needed1
 
     @classmethod
-    def execute(cls, V, F, Expression, length_mismatch="tile",stack=[]) -> io.NodeOutput:
+    def execute(cls, V, F, Expression, length_mismatch="tile",stack=dict()) -> io.NodeOutput:
         # Determine reference model for cloning
         a = V.get("V0")
         if a is None:

more_math/Parser/MathExpr.g4

@@ -1,7 +1,7 @@
 grammar MathExpr;
 
 // Top-level entry point
-start: (funcDef | varDef | stmt)* expr SEMICOLON? EOF;
+start: (funcDef | varDef | stmt)* expr? SEMICOLON? EOF;
 
 funcDef:
 	VARIABLE LPAREN paramList? RPAREN ARROW (block | expr) SEMICOLON # FunctionDef;
@@ -81,7 +81,9 @@ atom:
 	| PIPE expr PIPE						# AbsExp
 	| LBRACKET expr (COMMA expr)* RBRACKET	# ListExp
 	| VARIABLE LPAREN exprList? RPAREN		# CallExp
-	| NONE									# NoneExp;
+	| NONE									# NoneExp
+	| BREAK									# BreakExp
+	| CONTINUE								# ContinueExp;
 
 exprList: expr (COMMA expr)*;
 
@@ -136,6 +138,7 @@ func1:
 	| MEDIAN LPAREN expr RPAREN				# MedianFunc
 	| MODE LPAREN expr RPAREN				# ModeFunc
 	| CUMSUM LPAREN expr RPAREN				# CumsumFunc
+	| COUNT LPAREN expr RPAREN				# CountFunc
 	| CUMPROD LPAREN expr RPAREN			# CumprodFunc
 	| POP LPAREN expr RPAREN				# PopFunc
 	| CLEAR LPAREN expr RPAREN				# ClearFunc
@@ -307,6 +310,7 @@ COSSIM: 'cossim';
 FLIP: 'flip';
 COV: 'cov';
 SORT: 'sort';
+COUNT: 'count' | 'length' | 'cnt';
 APPEND: 'append';
 GET_VALUE: 'get_value';
 CROP: 'crop';

more_math/Parser/MathExpr.interp

@@ -98,46 +98,47 @@ COSSIM=97
 FLIP=98
 COV=99
 SORT=100
-APPEND=101
-GET_VALUE=102
-CROP=103
-FOR=104
-IN=105
-TIMESTAMP=106
-NONE=107
-BREAK=108
-CONTINUE=109
-PLUS=110
-MINUS=111
-MULT=112
-DIV=113
-MOD=114
-POW=115
-GE=116
-GT=117
-LE=118
-LT=119
-EQ=120
-EQUEALS=121
-NE=122
-PIPE=123
-LPAREN=124
-RPAREN=125
-COMMA=126
-SEMICOLON=127
-ARROW=128
-LBRACKET=129
-RBRACKET=130
-QUESTION=131
-COLON=132
-LBRACE=133
-RBRACE=134
-NUMBER=135
-CONSTANT=136
-VARIABLE=137
-SL_COMMENT=138
-ML_COMMENT=139
-WS=140
+COUNT=101
+APPEND=102
+GET_VALUE=103
+CROP=104
+FOR=105
+IN=106
+TIMESTAMP=107
+NONE=108
+BREAK=109
+CONTINUE=110
+PLUS=111
+MINUS=112
+MULT=113
+DIV=114
+MOD=115
+POW=116
+GE=117
+GT=118
+LE=119
+LT=120
+EQ=121
+EQUEALS=122
+NE=123
+PIPE=124
+LPAREN=125
+RPAREN=126
+COMMA=127
+SEMICOLON=128
+ARROW=129
+LBRACKET=130
+RBRACKET=131
+QUESTION=132
+COLON=133
+LBRACE=134
+RBRACE=135
+NUMBER=136
+CONSTANT=137
+VARIABLE=138
+SL_COMMENT=139
+ML_COMMENT=140
+WS=141
 'sin'=1
 'cos'=2
 'tan'=3
@@ -216,35 +217,35 @@ WS=140
 'flip'=98
 'cov'=99
 'sort'=100
-'append'=101
-'get_value'=102
-'crop'=103
-'for'=104
-'in'=105
-'break'=108
-'continue'=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
+'append'=102
+'get_value'=103
+'crop'=104
+'for'=105
+'in'=106
+'break'=109
+'continue'=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