Merge pull request #98 from Xiddoc/feature-classes

Feature classes

Author: Xiddoc

examples/test_class.cpp

@@ -0,0 +1,30 @@
+
+#include <iostream>
+
+void print(auto print_string);
+
+void print(auto print_string) {
+std::cout<<print_string<<std::endl;
+}
+
+
+
+int main() {
+    /* Transpiled with ComPy */
+    class TestClass {
+public:
+int test_field = 0;
+TestClass() {
+int test_var = 123;
+print(test_var);
+};
+void increment_test() {
+test_field += 1;
+};
+};
+TestClass test_instance = TestClass();
+print(test_instance.test_field);
+test_instance.increment_test();
+print(test_instance.test_field);
+    return 0;
+}

examples/test_class.py

@@ -0,0 +1,30 @@
+"""
+Testing a class in Python.
+"""
+
+
+# noinspection PyMissingOrEmptyDocstring
+class TestClass:
+
+    test_field: int = 0
+
+    def __init__(self) -> None:
+        # Initialize new temporary variable
+        test_var: int = 123
+        print(test_var)
+
+    def increment_test(self) -> None:
+        self.test_field += 1
+
+
+# Initialize the class
+test_instance: TestClass = TestClass()
+
+# Get a public field
+print(test_instance.test_field)
+
+# Execute a public method
+test_instance.increment_test()
+
+# The value changed!
+print(test_instance.test_field)

src/compiler/Constants.py

@@ -3,19 +3,21 @@
 """
 from _ast import Constant, BinOp, operator, Add, Sub, Mult, AnnAssign, AST, Expr, Name, Call, FunctionDef, arg, \
     Return, Assign, Module, IfExp, If, cmpop, Eq, Compare, Lt, Gt, BoolOp, NotEq, Or, And, boolop, Div, Mod, FloorDiv, \
-    Pass, GtE, LtE, While, AugAssign, Break, For
+    Pass, GtE, LtE, While, AugAssign, Break, For, ClassDef, Attribute
 from json import dumps
 from typing import Dict, Type, Any, Callable
 
 from src.pyexpressions.abstract.PyExpression import PyExpression
 from src.pyexpressions.concrete.PyAnnAssign import PyAnnAssign
 from src.pyexpressions.concrete.PyArg import PyArg
 from src.pyexpressions.concrete.PyAssign import PyAssign
+from src.pyexpressions.concrete.PyAttribute import PyAttribute
 from src.pyexpressions.concrete.PyAugAssign import PyAugAssign
 from src.pyexpressions.concrete.PyBinOp import PyBinOp
 from src.pyexpressions.concrete.PyBoolOp import PyBoolOp
 from src.pyexpressions.concrete.PyBreak import PyBreak
 from src.pyexpressions.concrete.PyCall import PyCall
+from src.pyexpressions.concrete.PyClassDef import PyClassDef
 from src.pyexpressions.concrete.PyCompare import PyCompare
 from src.pyexpressions.concrete.PyConstant import PyConstant
 from src.pyexpressions.concrete.PyExpr import PyExpr
@@ -32,12 +34,14 @@
 AST_EXPR_TO_PYEXPR: Dict[Type[AST], Type[PyExpression]] = {
     AnnAssign: PyAnnAssign,
     Assign: PyAssign,
+    Attribute: PyAttribute,
     AugAssign: PyAugAssign,
     arg: PyArg,
     BinOp: PyBinOp,
     BoolOp: PyBoolOp,
     Break: PyBreak,
     Call: PyCall,
+    ClassDef: PyClassDef,
     Compare: PyCompare,
     Constant: PyConstant,
     Expr: PyExpr,

src/pybuiltins/PyPortFunction.py

@@ -15,7 +15,7 @@ class PyPortFunction(PyExpression):
 	Port a native function or object to Python.
 	"""
 
-	# __func has a type of PyFunctionDef, but you
+	# __obj has a type of PyFunctionDef, but you
 	# can't specify it here without a circular import
 	# error so we will (overwrite) type hint in the constructor instead.
 	__func: PyFunctionDef
@@ -65,7 +65,7 @@ def get_function_name(self) -> str:
 		"""
 		:return: The native name of the ported function.
 		"""
-		return self.get_interface_function().get_func_name()
+		return self.get_interface_function().get_id()
 
 	def _transpile(self) -> str:
 		"""