yapcad.dsl.transforms package

Submodules

yapcad.dsl.transforms.base module

AST transformation framework for DSL optimization.

Provides a base class for AST transformations that can be applied before interpretation. This allows for future optimizations like: - Constant folding - Dead code elimination - Pattern unrolling - Common subexpression caching

class yapcad.dsl.transforms.base.AstTransform

Bases: ABC

Base class for AST transformations.

Transforms are applied to a Module and return a (potentially modified) Module. Transforms can be composed in a pipeline.

abstract property name: str

Name of this transform for debugging/logging.

abstractmethod transform(module: Module) → Module

Apply this transform to a module.

Parameters:

module – The input module AST

Returns:

The transformed module (may be the same object or a copy)

class yapcad.dsl.transforms.base.IdentityTransform

Bases: AstTransform

Identity transform - returns module unchanged.

property name: str

Name of this transform for debugging/logging.

transform(module: Module) → Module

Apply this transform to a module.

Parameters:

module – The input module AST

Returns:

The transformed module (may be the same object or a copy)

class yapcad.dsl.transforms.base.TransformPipeline(transforms: List[AstTransform] = None)

Bases: object

A pipeline of AST transforms to apply in sequence.

add(transform: AstTransform) → TransformPipeline

Add a transform to the pipeline.

apply(module: Module) → Module

Apply all transforms in sequence.

class yapcad.dsl.transforms.base.TreeTransform

Bases: AstTransform

A transform that walks the tree and can modify nodes.

Subclasses override visit_* methods to transform specific node types. By default, nodes are copied unchanged.

transform(module: Module) → Module

Transform a module by visiting all nodes.

visit_assignment(node: AssignmentStatement) → Statement

Visit an assignment statement.

visit_binary_op(node: BinaryOp) → Expression

visit_block(node: Block) → Statement

Visit a block.

visit_command(node: FunctionDef) → FunctionDef

Visit a command node.

visit_dict_literal(node: DictLiteral) → Expression

visit_emit(node: EmitStatement) → Statement

Visit an emit statement.

visit_expr_statement(node: ExpressionStatement) → Statement

Visit an expression statement.

visit_expression(node: Expression) → Expression

Visit an expression node.

visit_for(node: ForStatement) → Statement

Visit a for statement.

visit_function_call(node: FunctionCall) → Expression

visit_identifier(node: Identifier) → Expression

visit_if_expr(node: IfExpr) → Expression

visit_index_access(node: IndexAccess) → Expression

visit_lambda(node: LambdaExpr) → Expression

visit_let(node: VarDecl) → Statement

Visit a let statement.

visit_list_comprehension(node: ListComprehension) → Expression

visit_list_literal(node: ListLiteral) → Expression

visit_literal(node: Literal) → Expression

visit_match_expr(node: MatchExpr) → Expression

visit_member_access(node: MemberAccess) → Expression

visit_method_call(node: MethodCall) → Expression

visit_module(node: Module) → Module

Visit a module node.

visit_python_block(node: PythonBlock) → Statement

Visit a Python block (no transformation by default).

visit_python_expr(node: PythonExpr) → Expression

visit_range(node: RangeExpr) → Expression

visit_require(node: AssertStatement) → Statement

Visit a require statement.

visit_return(node: ReturnStatement) → Statement

Visit a return statement.

visit_statement(node: Statement) → Statement

Visit a statement node.

visit_unary_op(node: UnaryOp) → Expression

Module contents

DSL AST Transformation Framework.

Provides a framework for AST transformations that can be applied before interpretation. This allows for optimizations like: - Constant folding - Dead code elimination - Pattern unrolling - Common subexpression caching

Usage:

from yapcad.dsl.transforms import TransformPipeline, ConstantFoldTransform

pipeline = TransformPipeline() pipeline.add(ConstantFoldTransform())

optimized_module = pipeline.apply(module)

class yapcad.dsl.transforms.AstTransform

Bases: ABC

Base class for AST transformations.

Transforms are applied to a Module and return a (potentially modified) Module. Transforms can be composed in a pipeline.

abstract property name: str

Name of this transform for debugging/logging.

abstractmethod transform(module: Module) → Module

Apply this transform to a module.

Parameters:

module – The input module AST

Returns:

The transformed module (may be the same object or a copy)

class yapcad.dsl.transforms.IdentityTransform

Bases: AstTransform

Identity transform - returns module unchanged.

property name: str

Name of this transform for debugging/logging.

transform(module: Module) → Module

Apply this transform to a module.

Parameters:

module – The input module AST

Returns:

The transformed module (may be the same object or a copy)

class yapcad.dsl.transforms.TransformPipeline(transforms: List[AstTransform] = None)

Bases: object

A pipeline of AST transforms to apply in sequence.

add(transform: AstTransform) → TransformPipeline

Add a transform to the pipeline.

apply(module: Module) → Module

Apply all transforms in sequence.

class yapcad.dsl.transforms.TreeTransform

Bases: AstTransform

A transform that walks the tree and can modify nodes.

Subclasses override visit_* methods to transform specific node types. By default, nodes are copied unchanged.

transform(module: Module) → Module

Transform a module by visiting all nodes.

visit_assignment(node: AssignmentStatement) → Statement

Visit an assignment statement.

visit_binary_op(node: BinaryOp) → Expression

visit_block(node: Block) → Statement

Visit a block.

visit_command(node: FunctionDef) → FunctionDef

Visit a command node.

visit_dict_literal(node: DictLiteral) → Expression

visit_emit(node: EmitStatement) → Statement

Visit an emit statement.

visit_expr_statement(node: ExpressionStatement) → Statement

Visit an expression statement.

visit_expression(node: Expression) → Expression

Visit an expression node.

visit_for(node: ForStatement) → Statement

Visit a for statement.

visit_function_call(node: FunctionCall) → Expression

visit_identifier(node: Identifier) → Expression

visit_if_expr(node: IfExpr) → Expression

visit_index_access(node: IndexAccess) → Expression

visit_lambda(node: LambdaExpr) → Expression

visit_let(node: VarDecl) → Statement

Visit a let statement.

visit_list_comprehension(node: ListComprehension) → Expression

visit_list_literal(node: ListLiteral) → Expression

visit_literal(node: Literal) → Expression

visit_match_expr(node: MatchExpr) → Expression

visit_member_access(node: MemberAccess) → Expression

visit_method_call(node: MethodCall) → Expression

visit_module(node: Module) → Module

Visit a module node.

visit_python_block(node: PythonBlock) → Statement

Visit a Python block (no transformation by default).

visit_python_expr(node: PythonExpr) → Expression

visit_range(node: RangeExpr) → Expression

visit_require(node: AssertStatement) → Statement

Visit a require statement.

visit_return(node: ReturnStatement) → Statement

Visit a return statement.

visit_statement(node: Statement) → Statement

Visit a statement node.

visit_unary_op(node: UnaryOp) → Expression