mal-toolbox 1.2.1__py3-none-any.whl → 2.0.0__py3-none-any.whl

maltoolbox/language/compiler/distributions.py

@@ -0,0 +1,158 @@
+class DistributionsException(Exception):
+    def __init__(self, error_message):
+        self._error_message = error_message
+        super().__init__(self._error_message)
+
+
+class Distributions:
+    @staticmethod
+    def validate(distribution_name: str, params: list) -> None:
+        match distribution_name:
+            case 'Bernoulli':
+                Bernoulli.validate(params)
+            case 'Binomial':
+                Binomial.validate(params)
+            case 'Exponential':
+                Exponential.validate(params)
+            case 'Gamma':
+                Gamma.validate(params)
+            case 'LogNormal':
+                LogNormal.validate(params)
+            case 'Pareto':
+                Pareto.validate(params)
+            case 'TruncatedNormal':
+                TruncatedNormal.validate(params)
+            case 'Uniform':
+                Uniform.validate(params)
+            case (
+                'Enabled'
+                | 'Disabled'
+                | 'Zero'
+                | 'Infinity'
+                | 'EasyAndCertain'
+                | 'EasyAndUncertain'
+                | 'HardAndCertain'
+                | 'HardAndUncertain'
+                | 'VeryHardAndCertain'
+                | 'VeryHardAndUncertain'
+            ):
+                Combination.validate(params)
+            case _:
+                err_msg = f'Distribution {distribution_name} is not supported'
+                raise (DistributionsException(err_msg))
+
+
+class Bernoulli:
+    @staticmethod
+    def validate(params: list) -> None:
+        if not params or len(params) != 1:
+            err_msg = 'Expected exactly one parameter (probability), for Bernoulli distribution'
+            raise (DistributionsException(err_msg))
+        if not 0 <= params[0] <= 1:
+            err_msg = f"{params[0]} is not in valid range '0 <= probability <= 1', for Bernoulli distribution"
+            raise (DistributionsException(err_msg))
+
+
+class Binomial:
+    @staticmethod
+    def validate(params: list) -> None:
+        if not params or len(params) != 2:
+            err_msg = 'Expected exactly two parameters (trials, probability), for Binomial distribution'
+            raise (DistributionsException(err_msg))
+        if not 0 <= params[1] <= 1:
+            err_msg = f"{params[1]} is not in valid range '0 <= probability <= 1', for Binomial distribution"
+            raise (DistributionsException(err_msg))
+
+
+class Exponential:
+    @staticmethod
+    def validate(params: list) -> None:
+        if not params or len(params) != 1:
+            err_msg = (
+                'Expected exactly one parameter (lambda), for Exponential distribution'
+            )
+            raise (DistributionsException(err_msg))
+        if params[0] <= 0:
+            err_msg = f"{params[0]} is not in valid range 'lambda > 0', for Exponential distribution"
+            raise (DistributionsException(err_msg))
+
+
+class Gamma:
+    @staticmethod
+    def validate(params: list) -> None:
+        if not params or len(params) != 2:
+            err_msg = (
+                'Expected exactly two parameters (shape, scale), for Gamma distribution'
+            )
+            raise (DistributionsException(err_msg))
+        if params[0] <= 0:
+            err_msg = (
+                f"{params[0]} is not in valid range 'shape > 0', for Gamma distribution"
+            )
+            raise (DistributionsException(err_msg))
+        if params[1] <= 0:
+            err_msg = (
+                f"{params[1]} is not in valid range 'scale > 0', for Gamma distribution"
+            )
+            raise (DistributionsException(err_msg))
+
+
+class LogNormal:
+    @staticmethod
+    def validate(params: list) -> None:
+        if not params or len(params) != 2:
+            err_msg = 'Expected exactly two parameters (mean, standardDeviation), for LogNormal distribution'
+            raise (DistributionsException(err_msg))
+        if params[1] <= 0:
+            err_msg = f"{params[1]} is not in valid range 'standardDeviation > 0', for LogNormal distribution"
+            raise (DistributionsException(err_msg))
+
+
+class Pareto:
+    @staticmethod
+    def validate(params: list) -> None:
+        if not params or len(params) != 2:
+            err_msg = (
+                'Expected exactly two parameters (min, shape), for Pareto distribution'
+            )
+            raise (DistributionsException(err_msg))
+        if params[0] <= 0:
+            err_msg = (
+                f"{params[0]} is not in valid range 'min > 0', for Pareto distribution"
+            )
+            raise (DistributionsException(err_msg))
+        if params[1] <= 0:
+            err_msg = f"{params[1]} is not in valid range 'shape > 0', for Pareto distribution"
+            raise (DistributionsException(err_msg))
+
+
+class TruncatedNormal:
+    @staticmethod
+    def validate(params: list) -> None:
+        if not params or len(params) != 2:
+            err_msg = 'Expected exactly two parameters (mean, standardDeviation), for TruncatedNormal distribution'
+            raise (DistributionsException(err_msg))
+        if params[1] <= 0:
+            err_msg = f"{params[1]} is not in valid range 'standardDeviation > 0', for TruncatedNormal distribution"
+            raise (DistributionsException(err_msg))
+
+
+class Uniform:
+    @staticmethod
+    def validate(params: list) -> None:
+        if not params or len(params) != 2:
+            err_msg = (
+                'Expected exactly two parameters (min, max), for Uniform distribution'
+            )
+            raise (DistributionsException(err_msg))
+        if params[0] > params[1]:
+            err_msg = f"({params[0]}, {params[1]}) does not meet requirement 'min <= max', for Uniform distribution"
+            raise (DistributionsException(err_msg))
+
+
+class Combination:
+    @staticmethod
+    def validate(params: list) -> None:
+        if params and len(params) != 0:
+            err_msg = 'Expected exactly zero parameters, for combination distributions'
+            raise (DistributionsException(err_msg))

maltoolbox/language/compiler/exceptions.py

@@ -0,0 +1,37 @@
+class MalCompilerError(Exception):
+    """Base exception for MalCompiler errors."""
+
+    pass
+
+
+class MalSyntaxError(MalCompilerError):
+    """Raised when syntax error is encountered during compilation."""
+
+    def __init__(self, message, line=None, column=None):
+        self.line = line
+        self.column = column
+        super().__init__(message)
+
+
+class MalParseError(MalCompilerError):
+    """Raised when parsing fails."""
+
+    pass
+
+
+class MalTypeError(MalCompilerError):
+    """Raised when type checking fails."""
+
+    pass
+
+
+class MalNameError(MalCompilerError):
+    """Raised when an undefined name is referenced."""
+
+    pass
+
+
+class MalCompilationError(MalCompilerError):
+    """Raised when code generation fails."""
+
+    pass

maltoolbox/language/compiler/lang.py

@@ -0,0 +1,5 @@
+from tree_sitter import Language, Parser
+import tree_sitter_mal as ts_mal
+
+MAL_LANGUAGE = Language(ts_mal.language())
+PARSER = Parser(MAL_LANGUAGE)