transpiler 0.1.0__py3-none-any.whl

transpiler/backend/c/codegen.py

@@ -0,0 +1,121 @@
+from transpiler.ir.nodes import *
+from transpiler.ir.types import *
+
+class CBackend:
+    def __init__(self):
+        self.indent_level = 0
+        
+    def generate(self, program: Program) -> str:
+        lines = [
+            '#include "runtime/c/transpiler_runtime.h"',
+            ""
+        ]
+        
+        for struct in program.structs:
+            lines.append(f"typedef struct {struct.name} {struct.name};")
+        if program.structs: lines.append("")
+        
+        for struct in program.structs:
+            lines.append(self._generate_struct(struct))
+        
+        for function in program.functions:
+            lines.append(self._generate_function(function))
+            
+        return "\n".join(lines)
+        
+    def _generate_struct(self, node: Struct) -> str:
+        lines = [f"struct {node.name} {{"]
+        self.indent_level += 1
+        for name, typ in node.fields:
+            lines.append(self._indent(f"{self._type_to_str(typ)} {name};"))
+        self.indent_level -= 1
+        lines.append("};\n")
+        return "\n".join(lines)
+        
+    def _generate_function(self, node: Function) -> str:
+        params = []
+        for p in node.params:
+            params.append(f"{self._type_to_str(p.type)} {p.name}")
+            
+        ret_type = self._type_to_str(node.return_type)
+        
+        lines = [f"{ret_type} {node.name}({', '.join(params)}) {{"]
+        self.indent_level += 1
+        body_str = self._generate_block(node.body)
+        if body_str:
+            lines.append(body_str)
+        self.indent_level -= 1
+        lines.append("}\n")
+        return "\n".join(lines)
+        
+    def _generate_block(self, node: Block) -> str:
+        lines = []
+        for stmt in node.statements:
+            stmt_str = self._generate_stmt(stmt)
+            if stmt_str:
+                lines.append(self._indent(stmt_str))
+        return "\n".join(lines)
+        
+    def _generate_stmt(self, node: Stmt) -> str:
+        if isinstance(node, Return):
+            if node.value:
+                return f"return {self._generate_expr(node.value)};"
+            return "return;"
+        elif isinstance(node, Let):
+            return f"{self._type_to_str(node.type)} {node.name} = {self._generate_expr(node.value)};"
+        elif isinstance(node, Assign):
+            return f"{self._generate_expr(node.target)} = {self._generate_expr(node.value)};"
+        elif isinstance(node, ExprStmt):
+            return f"{self._generate_expr(node.expr)};"
+        elif isinstance(node, If):
+            res = f"if ({self._generate_expr(node.condition)}) {{\n"
+            self.indent_level += 1
+            res += self._generate_block(node.then_block)
+            self.indent_level -= 1
+            res += f"\n{self._indent('}')}"
+            if node.else_block:
+                res += " else {\n"
+                self.indent_level += 1
+                res += self._generate_block(node.else_block)
+                self.indent_level -= 1
+                res += f"\n{self._indent('}')}"
+            return res.strip()
+        elif isinstance(node, While):
+            res = f"while ({self._generate_expr(node.condition)}) {{\n"
+            self.indent_level += 1
+            res += self._generate_block(node.body)
+            self.indent_level -= 1
+            res += f"\n{self._indent('}')}"
+            return res.strip()
+        return ""
+        
+    def _generate_expr(self, node: Expr) -> str:
+        if isinstance(node, IntLiteral): return str(node.value)
+        elif isinstance(node, FloatLiteral): return str(node.value)
+        elif isinstance(node, BoolLiteral): return "true" if node.value else "false"
+        elif isinstance(node, StringLiteral): return f'"{node.value}"'
+        elif isinstance(node, NullLiteral): return "NULL"
+        elif isinstance(node, Var): return node.name
+        elif isinstance(node, Binary):
+            return f"({self._generate_expr(node.left)} {node.op.value} {self._generate_expr(node.right)})"
+        elif isinstance(node, Call):
+            func = self._generate_expr(node.function)
+            if func == "print":
+                func = "print_int"
+            args = [self._generate_expr(arg) for arg in node.args]
+            return f"{func}({', '.join(args)})"
+        elif isinstance(node, FieldAccess):
+            return f"{self._generate_expr(node.object)}.{node.field}"
+        return ""
+        
+    def _type_to_str(self, t: Type) -> str:
+        if t.kind == TypeKind.INT: return "int"
+        elif t.kind == TypeKind.FLOAT: return "double"
+        elif t.kind == TypeKind.BOOL: return "bool"
+        elif t.kind == TypeKind.STRING: return "char*"
+        elif t.kind == TypeKind.VOID: return "void"
+        elif t.kind == TypeKind.STRUCT: return f"struct {t.name}"
+        return "void*"
+        
+    def _indent(self, line: str) -> str:
+        return "    " * self.indent_level + line

transpiler/backend/go/codegen.py

@@ -0,0 +1,120 @@
+from transpiler.ir.nodes import *
+from transpiler.ir.types import *
+
+class GoBackend:
+    def __init__(self):
+        self.indent_level = 0
+        
+    def generate(self, program: Program) -> str:
+        lines = [
+            "package main",
+            "",
+            'import "fmt"',
+            ""
+        ]
+        
+        for struct in program.structs:
+            lines.append(self._generate_struct(struct))
+        for function in program.functions:
+            lines.append(self._generate_function(function))
+            
+        return "\n".join(lines)
+        
+    def _generate_struct(self, node: Struct) -> str:
+        lines = [f"type {node.name} struct {{"]
+        self.indent_level += 1
+        for name, typ in node.fields:
+            # Capitalize field names to export them in Go
+            go_name = name.capitalize()
+            lines.append(self._indent(f"{go_name} {self._type_to_str(typ)}"))
+        self.indent_level -= 1
+        lines.append("}\n")
+        return "\n".join(lines)
+        
+    def _generate_function(self, node: Function) -> str:
+        params = []
+        for p in node.params:
+            params.append(f"{p.name} {self._type_to_str(p.type)}")
+            
+        ret_type = self._type_to_str(node.return_type)
+        ret_str = f" {ret_type}" if ret_type and ret_type not in ("void", "interface{}") else ""
+        
+        lines = [f"func {node.name}({', '.join(params)}){ret_str} {{"]
+        self.indent_level += 1
+        body_str = self._generate_block(node.body)
+        if body_str:
+            lines.append(body_str)
+        self.indent_level -= 1
+        lines.append("}\n")
+        return "\n".join(lines)
+        
+    def _generate_block(self, node: Block) -> str:
+        lines = []
+        for stmt in node.statements:
+            stmt_str = self._generate_stmt(stmt)
+            if stmt_str:
+                lines.append(self._indent(stmt_str))
+        return "\n".join(lines)
+        
+    def _generate_stmt(self, node: Stmt) -> str:
+        if isinstance(node, Return):
+            if node.value:
+                return f"return {self._generate_expr(node.value)}"
+            return "return"
+        elif isinstance(node, Let):
+            return f"var {node.name} {self._type_to_str(node.type)} = {self._generate_expr(node.value)}"
+        elif isinstance(node, Assign):
+            return f"{self._generate_expr(node.target)} = {self._generate_expr(node.value)}"
+        elif isinstance(node, ExprStmt):
+            return self._generate_expr(node.expr)
+        elif isinstance(node, If):
+            res = f"if {self._generate_expr(node.condition)} {{\n"
+            self.indent_level += 1
+            res += self._generate_block(node.then_block)
+            self.indent_level -= 1
+            res += f"\n{self._indent('}')}"
+            if node.else_block:
+                res += " else {\n"
+                self.indent_level += 1
+                res += self._generate_block(node.else_block)
+                self.indent_level -= 1
+                res += f"\n{self._indent('}')}"
+            return res.strip()
+        elif isinstance(node, While):
+            res = f"for {self._generate_expr(node.condition)} {{\n"
+            self.indent_level += 1
+            res += self._generate_block(node.body)
+            self.indent_level -= 1
+            res += f"\n{self._indent('}')}"
+            return res.strip()
+        return ""
+        
+    def _generate_expr(self, node: Expr) -> str:
+        if isinstance(node, IntLiteral): return str(node.value)
+        elif isinstance(node, FloatLiteral): return str(node.value)
+        elif isinstance(node, BoolLiteral): return "true" if node.value else "false"
+        elif isinstance(node, StringLiteral): return f'"{node.value}"'
+        elif isinstance(node, NullLiteral): return "nil"
+        elif isinstance(node, Var): return node.name
+        elif isinstance(node, Binary):
+            return f"({self._generate_expr(node.left)} {node.op.value} {self._generate_expr(node.right)})"
+        elif isinstance(node, Call):
+            func = self._generate_expr(node.function)
+            if func == "print":
+                func = "fmt.Println"
+            args = [self._generate_expr(arg) for arg in node.args]
+            return f"{func}({', '.join(args)})"
+        elif isinstance(node, FieldAccess):
+            return f"{self._generate_expr(node.object)}.{node.field.capitalize()}"
+        return ""
+        
+    def _type_to_str(self, t: Type) -> str:
+        if t.kind == TypeKind.INT: return "int"
+        elif t.kind == TypeKind.FLOAT: return "float64"
+        elif t.kind == TypeKind.BOOL: return "bool"
+        elif t.kind == TypeKind.STRING: return "string"
+        elif t.kind == TypeKind.VOID: return "void"
+        return "interface{}"
+        
+    def _indent(self, line: str) -> str:
+        return "\t" * self.indent_level + line

transpiler/backend/java/codegen.py

@@ -0,0 +1,112 @@
+from transpiler.ir.nodes import *
+from transpiler.ir.types import *
+
+class JavaBackend:
+    def __init__(self):
+        self.indent_level = 0
+        
+    def generate(self, program: Program) -> str:
+        lines = [
+            "public class Main {",
+            ""
+        ]
+        self.indent_level += 1
+        
+        for function in program.functions:
+            lines.append(self._generate_function(function))
+            
+        self.indent_level -= 1
+        lines.append("}\n")
+        return "\n".join(lines)
+        
+    def _generate_function(self, node: Function) -> str:
+        params = []
+        for p in node.params:
+            params.append(f"{self._type_to_str(p.type)} {p.name}")
+            
+        ret_type = self._type_to_str(node.return_type)
+        
+        if node.name == "main":
+            lines = [self._indent(f"public static void main(String[] args) {{")]
+        else:
+            lines = [self._indent(f"public static {ret_type} {node.name}({', '.join(params)}) {{")]
+            
+        self.indent_level += 1
+        body_str = self._generate_block(node.body)
+        if body_str:
+            lines.append(body_str)
+        self.indent_level -= 1
+        lines.append(self._indent("}\n"))
+        return "\n".join(lines)
+        
+    def _generate_block(self, node: Block) -> str:
+        lines = []
+        for stmt in node.statements:
+            stmt_str = self._generate_stmt(stmt)
+            if stmt_str:
+                lines.append(self._indent(stmt_str))
+        return "\n".join(lines)
+        
+    def _generate_stmt(self, node: Stmt) -> str:
+        if isinstance(node, Return):
+            if node.value:
+                return f"return {self._generate_expr(node.value)};"
+            return "return;"
+        elif isinstance(node, Let):
+            return f"{self._type_to_str(node.type)} {node.name} = {self._generate_expr(node.value)};"
+        elif isinstance(node, Assign):
+            return f"{self._generate_expr(node.target)} = {self._generate_expr(node.value)};"
+        elif isinstance(node, ExprStmt):
+            return f"{self._generate_expr(node.expr)};"
+        elif isinstance(node, If):
+            res = f"if ({self._generate_expr(node.condition)}) {{\n"
+            self.indent_level += 1
+            res += self._generate_block(node.then_block)
+            self.indent_level -= 1
+            res += f"\n{self._indent('}')}"
+            if node.else_block:
+                res += " else {\n"
+                self.indent_level += 1
+                res += self._generate_block(node.else_block)
+                self.indent_level -= 1
+                res += f"\n{self._indent('}')}"
+            return res.strip()
+        elif isinstance(node, While):
+            res = f"while ({self._generate_expr(node.condition)}) {{\n"
+            self.indent_level += 1
+            res += self._generate_block(node.body)
+            self.indent_level -= 1
+            res += f"\n{self._indent('}')}"
+            return res.strip()
+        return ""
+        
+    def _generate_expr(self, node: Expr) -> str:
+        if isinstance(node, IntLiteral): return str(node.value)
+        elif isinstance(node, FloatLiteral): return str(node.value)
+        elif isinstance(node, BoolLiteral): return "true" if node.value else "false"
+        elif isinstance(node, StringLiteral): return f'"{node.value}"'
+        elif isinstance(node, NullLiteral): return "null"
+        elif isinstance(node, Var): return node.name
+        elif isinstance(node, Binary):
+            return f"({self._generate_expr(node.left)} {node.op.value} {self._generate_expr(node.right)})"
+        elif isinstance(node, Call):
+            func = self._generate_expr(node.function)
+            if func == "print":
+                func = "System.out.println"
+            args = [self._generate_expr(arg) for arg in node.args]
+            return f"{func}({', '.join(args)})"
+        elif isinstance(node, FieldAccess):
+            return f"{self._generate_expr(node.object)}.{node.field}"
+        return ""
+        
+    def _type_to_str(self, t: Type) -> str:
+        if t.kind == TypeKind.INT: return "int"
+        elif t.kind == TypeKind.FLOAT: return "double"
+        elif t.kind == TypeKind.BOOL: return "boolean"
+        elif t.kind == TypeKind.STRING: return "String"
+        elif t.kind == TypeKind.VOID: return "void"
+        elif t.kind == TypeKind.STRUCT: return t.name
+        return "Object"
+        
+    def _indent(self, line: str) -> str:
+        return "    " * self.indent_level + line

transpiler/backend/javascript/codegen.py

@@ -0,0 +1,108 @@
+from transpiler.ir.nodes import *
+from transpiler.ir.types import *
+
+class JSBackend:
+    def __init__(self):
+        self.indent_level = 0
+        
+    def generate(self, program: Program) -> str:
+        lines = []
+        for struct in program.structs:
+            lines.append(self._generate_struct(struct))
+        for function in program.functions:
+            lines.append(self._generate_function(function))
+        return "\n".join(lines)
+        
+    def _generate_struct(self, node: Struct) -> str:
+        lines = [f"class {node.name} {{"]
+        self.indent_level += 1
+        
+        if node.fields:
+            params = [name for name, _ in node.fields]
+            lines.append(self._indent(f"constructor({', '.join(params)}) {{"))
+            self.indent_level += 1
+            for name in params:
+                lines.append(self._indent(f"this.{name} = {name};"))
+            self.indent_level -= 1
+            lines.append(self._indent("}"))
+            
+        self.indent_level -= 1
+        lines.append("}\n")
+        return "\n".join(lines)
+        
+    def _generate_function(self, node: Function) -> str:
+        params = [p.name for p in node.params]
+        lines = [f"function {node.name}({', '.join(params)}) {{"]
+        self.indent_level += 1
+        body_str = self._generate_block(node.body)
+        if body_str:
+            lines.append(body_str)
+        self.indent_level -= 1
+        lines.append("}\n")
+        return "\n".join(lines)
+        
+    def _generate_block(self, node: Block) -> str:
+        lines = []
+        for stmt in node.statements:
+            stmt_str = self._generate_stmt(stmt)
+            if stmt_str:
+                lines.append(self._indent(stmt_str))
+        return "\n".join(lines)
+        
+    def _generate_stmt(self, node: Stmt) -> str:
+        if isinstance(node, Return):
+            if node.value:
+                return f"return {self._generate_expr(node.value)};"
+            return "return;"
+        elif isinstance(node, Let):
+            return f"let {node.name} = {self._generate_expr(node.value)};"
+        elif isinstance(node, Assign):
+            return f"{self._generate_expr(node.target)} = {self._generate_expr(node.value)};"
+        elif isinstance(node, ExprStmt):
+            return f"{self._generate_expr(node.expr)};"
+        elif isinstance(node, If):
+            res = f"if ({self._generate_expr(node.condition)}) {{\n"
+            self.indent_level += 1
+            res += self._generate_block(node.then_block)
+            self.indent_level -= 1
+            res += f"\n{self._indent('}')}"
+            if node.else_block:
+                res += " else {\n"
+                self.indent_level += 1
+                res += self._generate_block(node.else_block)
+                self.indent_level -= 1
+                res += f"\n{self._indent('}')}"
+            return res.strip()
+        elif isinstance(node, While):
+            res = f"while ({self._generate_expr(node.condition)}) {{\n"
+            self.indent_level += 1
+            res += self._generate_block(node.body)
+            self.indent_level -= 1
+            res += f"\n{self._indent('}')}"
+            return res.strip()
+        return ""
+        
+    def _generate_expr(self, node: Expr) -> str:
+        if isinstance(node, IntLiteral): return str(node.value)
+        elif isinstance(node, FloatLiteral): return str(node.value)
+        elif isinstance(node, BoolLiteral): return "true" if node.value else "false"
+        elif isinstance(node, StringLiteral): return f'"{node.value}"'
+        elif isinstance(node, NullLiteral): return "null"
+        elif isinstance(node, Var): return node.name
+        elif isinstance(node, Binary):
+            op = node.op.value
+            if node.op == BinOp.EQ: op = "==="
+            if node.op == BinOp.NEQ: op = "!=="
+            return f"({self._generate_expr(node.left)} {op} {self._generate_expr(node.right)})"
+        elif isinstance(node, Call):
+            func = self._generate_expr(node.function)
+            if func == "print":
+                func = "console.log"
+            args = [self._generate_expr(arg) for arg in node.args]
+            return f"{func}({', '.join(args)})"
+        elif isinstance(node, FieldAccess):
+            return f"{self._generate_expr(node.object)}.{node.field}"
+        return ""
+        
+    def _indent(self, line: str) -> str:
+        return "    " * self.indent_level + line

transpiler/backend/python/codegen.py

@@ -0,0 +1,109 @@
+from transpiler.ir.nodes import *
+from transpiler.ir.types import *
+
+class PythonBackend:
+    def __init__(self):
+        self.indent_level = 0
+        
+    def generate(self, program: Program) -> str:
+        lines = []
+        for struct in program.structs:
+            lines.append(self._generate_struct(struct))
+        for function in program.functions:
+            lines.append(self._generate_function(function))
+        return "\n".join(lines)
+        
+    def _generate_struct(self, node: Struct) -> str:
+        lines = [f"class {node.name}:"]
+        self.indent_level += 1
+        if not node.fields:
+            lines.append(self._indent("pass"))
+        else:
+            for name, typ in node.fields:
+                lines.append(self._indent(f"{name}: {self._type_to_str(typ)}"))
+        self.indent_level -= 1
+        return "\n".join(lines) + "\n"
+        
+    def _generate_function(self, node: Function) -> str:
+        params = []
+        for p in node.params:
+            ptype = self._type_to_str(p.type)
+            params.append(f"{p.name}: {ptype}" if ptype else p.name)
+        
+        ret_type = self._type_to_str(node.return_type)
+        ret_str = f" -> {ret_type}" if ret_type and ret_type != "Any" else ""
+        
+        lines = [f"def {node.name}({', '.join(params)}){ret_str}:"]
+        self.indent_level += 1
+        body_str = self._generate_block(node.body)
+        if not body_str.strip():
+            lines.append(self._indent("pass"))
+        else:
+            lines.append(body_str)
+        self.indent_level -= 1
+        return "\n".join(lines) + "\n"
+        
+    def _generate_block(self, node: Block) -> str:
+        lines = []
+        for stmt in node.statements:
+            stmt_str = self._generate_stmt(stmt)
+            if stmt_str:
+                lines.append(self._indent(stmt_str))
+        return "\n".join(lines)
+        
+    def _generate_stmt(self, node: Stmt) -> str:
+        if isinstance(node, Return):
+            if node.value:
+                return f"return {self._generate_expr(node.value)}"
+            return "return"
+        elif isinstance(node, Let):
+            return f"{node.name}: {self._type_to_str(node.type)} = {self._generate_expr(node.value)}"
+        elif isinstance(node, Assign):
+            return f"{self._generate_expr(node.target)} = {self._generate_expr(node.value)}"
+        elif isinstance(node, ExprStmt):
+            return self._generate_expr(node.expr)
+        elif isinstance(node, If):
+            res = f"if {self._generate_expr(node.condition)}:\n"
+            self.indent_level += 1
+            res += self._generate_block(node.then_block)
+            self.indent_level -= 1
+            if node.else_block:
+                res += f"\n{self._indent('else:')}\n"
+                self.indent_level += 1
+                res += self._generate_block(node.else_block)
+                self.indent_level -= 1
+            return res.strip()
+        elif isinstance(node, While):
+            res = f"while {self._generate_expr(node.condition)}:\n"
+            self.indent_level += 1
+            res += self._generate_block(node.body)
+            self.indent_level -= 1
+            return res.strip()
+        return ""
+        
+    def _generate_expr(self, node: Expr) -> str:
+        if isinstance(node, IntLiteral): return str(node.value)
+        elif isinstance(node, FloatLiteral): return str(node.value)
+        elif isinstance(node, BoolLiteral): return "True" if node.value else "False"
+        elif isinstance(node, StringLiteral): return f'"{node.value}"'
+        elif isinstance(node, NullLiteral): return "None"
+        elif isinstance(node, Var): return node.name
+        elif isinstance(node, Binary):
+            return f"({self._generate_expr(node.left)} {node.op.value} {self._generate_expr(node.right)})"
+        elif isinstance(node, Call):
+            func = self._generate_expr(node.function)
+            args = [self._generate_expr(arg) for arg in node.args]
+            return f"{func}({', '.join(args)})"
+        elif isinstance(node, FieldAccess):
+            return f"{self._generate_expr(node.object)}.{node.field}"
+        return ""
+        
+    def _type_to_str(self, t: Type) -> str:
+        if t.kind == TypeKind.INT: return "int"
+        elif t.kind == TypeKind.FLOAT: return "float"
+        elif t.kind == TypeKind.BOOL: return "bool"
+        elif t.kind == TypeKind.STRING: return "str"
+        return "Any"
+        
+    def _indent(self, line: str) -> str:
+        return "    " * self.indent_level + line