vex-ast 0.2.0__py3-none-any.whl → 0.2.1__py3-none-any.whl

vex_ast/types/primitives.py

@@ -0,0 +1,69 @@
+from typing import Optional, Union, List, Set, Dict, Any
+from .base import VexType, type_registry
+
+class PrimitiveType(VexType):
+    """Base class for primitive types like int, float, string, etc."""
+    
+    def __init__(self, name: str):
+        self._name = name
+        type_registry.register_type(self)
+    
+    @property
+    def name(self) -> str:
+        return self._name
+    
+    def __str__(self) -> str:
+        return self._name
+
+class NumericType(PrimitiveType):
+    """Base class for numeric types"""
+    
+    def is_compatible_with(self, other: VexType) -> bool:
+        """Numeric types are compatible with other numeric types"""
+        return isinstance(other, NumericType)
+
+class IntegerType(NumericType):
+    """Integer type"""
+    
+    def __init__(self):
+        super().__init__("int")
+    
+    def is_compatible_with(self, other: VexType) -> bool:
+        """Integers are compatible with numeric types"""
+        return isinstance(other, NumericType)
+
+class FloatType(NumericType):
+    """Float type"""
+    
+    def __init__(self):
+        super().__init__("float")
+    
+    def is_compatible_with(self, other: VexType) -> bool:
+        """Floats are compatible with numeric types"""
+        return isinstance(other, NumericType)
+
+class BooleanType(PrimitiveType):
+    """Boolean type"""
+    
+    def __init__(self):
+        super().__init__("bool")
+    
+    def is_compatible_with(self, other: VexType) -> bool:
+        """Booleans are only compatible with booleans"""
+        return isinstance(other, BooleanType)
+
+class StringType(PrimitiveType):
+    """String type"""
+    
+    def __init__(self):
+        super().__init__("string")
+    
+    def is_compatible_with(self, other: VexType) -> bool:
+        """Strings are only compatible with strings"""
+        return isinstance(other, StringType)
+
+# Singleton instances
+INT = IntegerType()
+FLOAT = FloatType()
+BOOL = BooleanType()
+STRING = StringType()

vex_ast/types/type_checker.py

@@ -0,0 +1,32 @@
+from typing import Optional, Any, List, Dict, Union, Tuple
+from .base import VexType, ANY
+from .primitives import INT, FLOAT, BOOL, STRING
+
+class TypeChecker:
+    """Utility for checking type compatibility"""
+    
+    def is_compatible(self, value_type: VexType, expected_type: VexType) -> bool:
+        """Check if value_type is compatible with expected_type"""
+        # Any type is compatible with anything
+        if expected_type == ANY:
+            return True
+            
+        return value_type.is_compatible_with(expected_type)
+    
+    def get_python_type(self, value: Any) -> Optional[VexType]:
+        """Convert a Python value to a VEX type"""
+        if value is None:
+            return None
+        if isinstance(value, bool):
+            return BOOL
+        if isinstance(value, int):
+            return INT
+        if isinstance(value, float):
+            return FLOAT
+        if isinstance(value, str):
+            return STRING
+        # Complex objects would need additional mapping logic
+        return None
+
+# Singleton instance
+type_checker = TypeChecker()

vex_ast/utils/__init__.py

@@ -0,0 +1,38 @@
+"""
+Utilities package for VEX AST.
+
+This package provides utility functions and classes for working with the AST.
+"""
+
+from .errors import (
+    ErrorHandler,
+    ErrorType,
+    VexSyntaxError,
+    VexAstError,
+    Error
+)
+from .source_location import (
+    SourceLocation,
+)
+from .type_definitions import (
+    NodeType,
+    VisitorType,
+    TransformerType
+)
+
+__all__ = [
+    # Error handling
+    "ErrorHandler",
+    "ErrorType",
+    "VexSyntaxError",
+    "VexAstError",
+    "Error",
+    
+    # Source location
+    "SourceLocation",
+    
+    # Type definitions
+    "NodeType",
+    "VisitorType",
+    "TransformerType"
+]

vex_ast/utils/type_definitions.py

@@ -0,0 +1,9 @@
+# vex_ast/utils/type_definitions.py
+"""Type definitions for type hints in the VEX AST."""
+
+from typing import Any, Dict, List, Optional, TypeVar, Union, Type, Protocol
+
+# Type variables for type hints
+NodeType = TypeVar('NodeType')
+VisitorType = TypeVar('VisitorType')
+TransformerType = TypeVar('TransformerType')

vex_ast/visitors/__init__.py

@@ -0,0 +1,28 @@
+"""
+Visitors package for VEX AST.
+
+This package provides visitor pattern implementations for traversing and transforming the AST.
+"""
+
+from .base import (
+    AstVisitor,
+    TypedVisitorMixin
+)
+from .printer import PrintVisitor
+from .analyzer import (
+    NodeCounter,
+    VariableCollector
+)
+
+__all__ = [
+    # Base visitors
+    "AstVisitor",
+    "TypedVisitorMixin",
+    
+    # Concrete visitors
+    "PrintVisitor",
+    
+    # Analysis visitors
+    "NodeCounter",
+    "VariableCollector"
+]

{vex_ast-0.2.0.dist-info → vex_ast-0.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vex_ast
-Version: 0.2.0
+Version: 0.2.1
 Summary: A Python package for generating Abstract Syntax Trees for VEX V5 code.
 Home-page: https://github.com/heartx2/vex_ast
 Author: Chaze

{vex_ast-0.2.0.dist-info → vex_ast-0.2.1.dist-info}/RECORD

@@ -1,19 +1,19 @@
 vex_ast/README.md,sha256=tRaq6n2Ab4IahHSo3utW2imVuTrJu_7zaelgF-lGBYo,2007
 vex_ast/READMEAPI.md,sha256=hGn4IMT9NnI_LW0kd36BdKKXdIhYFLbQcgbVj1EWJps,9107
-vex_ast/__init__.py,sha256=YPsS_1ab9-G3WYmyxQobDA1U8y5KJkj2qxm_p_s9Oow,1724
+vex_ast/__init__.py,sha256=ctZTmMc0ispErWMja0pTwdyst6ELgcfBJ-nwejYg-sc,1604
 vex_ast/ast/README.md,sha256=7IQDHEXmKyJ8zfJNwa3pMGTDZKVPMFD0YeHT06ATRyM,3205
-vex_ast/ast/__init__.py,sha256=3ZtTiNeBPWOYgNWmxThn0h7UQx8Z0bHbK6hRHCUuILQ,2435
+vex_ast/ast/__init__.py,sha256=_KT8R_WG9_DLvPlDhVCNWz-yw0OtdoKxxWZYbtloCzU,2434
 vex_ast/ast/core.py,sha256=q-15q5VfGIVPAmFsSiWeFAZ55MPsXfCRa21rrt2hmlM,2670
 vex_ast/ast/expressions.py,sha256=b0cDs239wvygCIL9s0W8uuyuPeMZz3DwxQ0wmBta_XM,7758
 vex_ast/ast/interfaces.py,sha256=DXKdAzts0EBWjCRgYkz_kVSMcYbNIyPdwYpTr7c6P2A,5928
 vex_ast/ast/literals.py,sha256=PXbOH5Y2fxEngWk_FOiFsx3PC2SEqcx0bcfOGJ3SdbI,2618
-vex_ast/ast/navigator.py,sha256=2ELYIWTZNuaJHLMDiY9bbRkz8XAEDRnPNN8uKRafNYg,7253
+vex_ast/ast/navigator.py,sha256=9DaVXrknBbBr4omAAMHQnZL9Wpj5wjtoCS6_lni8MYM,7529
 vex_ast/ast/operators.py,sha256=I-yWvhsrz-OxmBZs5zIss_GTZF5S-nwcSmIzvAVtddM,3160
 vex_ast/ast/statements.py,sha256=OWRthjYGmTuNozYAHjh_Enp5t-hR1PphtPnFg31FeWw,11841
 vex_ast/ast/validators.py,sha256=lhEm1dt-nzMiaUan_AWEOm0NiwXN7wYNkLCl1c3drqc,4455
 vex_ast/ast/vex_nodes.py,sha256=kVil-ApFIeZ_BoeqYppfVRPSTg1KFUOcvt1vCE6QRgs,10148
 vex_ast/parser/README.md,sha256=P1qq_97skpgluLDpNu9V_pAdkuF9StjkzOEXJJYpEuM,2565
-vex_ast/parser/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+vex_ast/parser/__init__.py,sha256=LGHnFm3UzR4Nw7royGH3c_2RqeY66y8O6DdXHbm9yL4,504
 vex_ast/parser/factory.py,sha256=gaje0jOPu5lpndgX9c3VuQ_q-FJuFz9yC5y2bD69lL4,8770
 vex_ast/parser/interfaces.py,sha256=Ttc0bD_5X420ZCT9MLUf_wE1aZLWLkaJRQqCwBy9wIs,956
 vex_ast/parser/python_parser.py,sha256=e1owk-c6Rb4Bt5SYcgB-urZzqsX5xC9tzcK1sB8gkr4,29687
@@ -27,24 +27,37 @@ vex_ast/registry/registry.py,sha256=gvsQKyeXWp84T2uDB4O_WQgKCeq8W9udl8TD5MnPZzc,
 vex_ast/registry/signature.py,sha256=m8gQKdc5IaORDdfdBX6_qH1OL5LoTzveGoWlIIaOES0,5763
 vex_ast/registry/simulation_behavior.py,sha256=WwkVKae0owtv3WTrfnVH93rpsh9lbqJ_RSAVmpIVA_k,313
 vex_ast/registry/validation.py,sha256=xq5UjSWYAcRxjxsCGg3WU5RtPN-DB-JAqKqDNGvnJGk,1973
+vex_ast/registry/functions/__init__.py,sha256=X9aw0-Y9Q6Gcx01zepP8G_20ybfwC-LDPQfkK2RDOrE,207
+vex_ast/registry/functions/display.py,sha256=icXgTL6Y94ddm-Xjuv1dIOY5k0K2FVnAvmKBvUCHvk0,4854
+vex_ast/registry/functions/drivetrain.py,sha256=F3voRYRfmAKgMiOP77oFTV8VCg-hXCVDUq3D-xGhN7Y,6307
+vex_ast/registry/functions/initialize.py,sha256=r-FoxCMNDtPDayQR1kq0uDDIvObQ0TUK-tyzD2eqvMo,787
+vex_ast/registry/functions/motor.py,sha256=Vby44hV1goNFgFQEFDlv7z_9uGQeuvLMABplOnLqZbo,5218
+vex_ast/registry/functions/sensors.py,sha256=BdhLV5gSWD5HO-o3JRXhbGQF7s1Ybk3b6VLLqVau1i0,6410
+vex_ast/registry/functions/timing.py,sha256=DqwMVPk7VKDeZB4S3NI7wOYgM1R52dcdgaVIb9135Jg,3204
 vex_ast/serialization/__init__.py,sha256=qPTEiMjU8hpVxNH5z4MY7Yj60AxuFurjXZStjhWWf6o,835
 vex_ast/serialization/json_deserializer.py,sha256=m_xiEUEoFXGfFylK3M8BVs7F2EieQJ9wbUa_ZTpkHtQ,10523
 vex_ast/serialization/json_serializer.py,sha256=YvMRUpqXjtbxlIvathEIywUqbH3Ne6GSRODfFB0QCGM,4465
 vex_ast/serialization/schema.py,sha256=zpKujT7u3Qw2kNbwjV09lNzeQAsCxvMx-uAdaRMEstA,14431
 vex_ast/types/README.md,sha256=Wd3jBShiNXNc3iJ69Qps5_0mBq8QVEd_Gz5OachfS34,1029
+vex_ast/types/__init__.py,sha256=naLOT_-qHWxzYj4nwxjCB5dfL6tcIt7TjMEaRaXMWbU,2163
+vex_ast/types/base.py,sha256=hCPCeBNnD5p29Mim-ouRTkG6Lfa8NXrsdYLO8xsbFtM,2258
+vex_ast/types/enums.py,sha256=BNPQrSh0SXXluVzv9wimOEC8B_5qhB6YNEqNWLp7ZYU,2507
+vex_ast/types/objects.py,sha256=3bmfq-tB4uIghFed2XnisRJjUR2woJX46QIUEN_rHdM,2280
+vex_ast/types/primitives.py,sha256=t_4kEVyPSmKRHEIRQcp-X5Yq46JbG1SxzlvHb0vL4IA,1928
+vex_ast/types/type_checker.py,sha256=emzhmc6AlH71w0DrKLlZRMBlJNZAuENihvRXTenqg_Q,1083
 vex_ast/utils/README.md,sha256=Y9RJMQTqQbpjVkvYJmpeRihD1zfW9PhNL_LgoDJ84ak,1945
-vex_ast/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+vex_ast/utils/__init__.py,sha256=azzhhFoMykqOkVVm6X2V7dFW3jGBkOvEgnEP1JTCS3g,621
 vex_ast/utils/errors.py,sha256=uq_jyu-YasifXjKSpPbVxup-XFf0XSHmp7A0Zd87TUY,3850
 vex_ast/utils/source_location.py,sha256=r857ypqUBqex2y_R0RzZo7pz5di9qtEeWFFQ4HwzYQE,1297
-vex_ast/utils/type_definitions.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+vex_ast/utils/type_definitions.py,sha256=2rB85B1vZL1GLWItBuJJpyr-vmp3W346EMMX2TPZ99Q,313
 vex_ast/visitors/README.md,sha256=BKDj8LMuBlCrzgSSLmCkbET7WIqgxe-oCkqQbqXekrE,2725
-vex_ast/visitors/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+vex_ast/visitors/__init__.py,sha256=cndW3yp0Mu9w5vzwqqS2N5EqS0Ioh0V5UCDDrwCO0_Q,492
 vex_ast/visitors/analyzer.py,sha256=cBT5PRbtfQ_Dt1qiHN-wdO5KnC2yl-Ul5RasXg9geHY,3668
 vex_ast/visitors/base.py,sha256=Ph0taTIVMHV_QjXfDd0ipZnCXVJErkyRtPpfwfzo-kY,4859
 vex_ast/visitors/printer.py,sha256=CUY_73hxm7MoC-63JeIXaVYnZ8QqtfMqbek2R2HMEmE,5411
 vex_ast/visitors/transformer.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-vex_ast-0.2.0.dist-info/licenses/LICENSE,sha256=IOSlfCuxGv4OAg421BRDKVi16RZ7-5kCMJ4B16r4kuc,69
-vex_ast-0.2.0.dist-info/METADATA,sha256=xGeh73jQORTXs2-uS-qgiH3aW-Pdsk31J1WW-tKQREE,5295
-vex_ast-0.2.0.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
-vex_ast-0.2.0.dist-info/top_level.txt,sha256=MoZGrpKgNUDiqL9gWp4q3wMw3q93XPEEjmBNPJQcNAs,8
-vex_ast-0.2.0.dist-info/RECORD,,
+vex_ast-0.2.1.dist-info/licenses/LICENSE,sha256=IOSlfCuxGv4OAg421BRDKVi16RZ7-5kCMJ4B16r4kuc,69
+vex_ast-0.2.1.dist-info/METADATA,sha256=Cm7vpEPf4qNlTPKTYcO4_HejvWBeyH-kcPpIq6dRe5I,5295
+vex_ast-0.2.1.dist-info/WHEEL,sha256=pxyMxgL8-pra_rKaQ4drOZAegBVuX-G_4nRHjjgWbmo,91
+vex_ast-0.2.1.dist-info/top_level.txt,sha256=MoZGrpKgNUDiqL9gWp4q3wMw3q93XPEEjmBNPJQcNAs,8
+vex_ast-0.2.1.dist-info/RECORD,,