Method implementations 1: Variable references

examples/class_method.st

@@ -0,0 +1,12 @@
+CLASS MyClass
+    VAR
+        x, y : INT;
+    END_VAR
+
+    METHOD testMethod
+        VAR_INPUT myMethodArg : INT; END_VAR
+        VAR myMethodLocalVar : INT; END_VAR
+
+        x := myMethodArg;
+    END_METHOD
+END_CLASS

examples/hello_world.st

@@ -0,0 +1,17 @@
+@EXTERNAL FUNCTION puts : DINT
+VAR_INPUT
+    text : STRING;
+END_VAR
+END_FUNCTION
+
+@EXTERNAL FUNCTION exit : __VOID
+    VAR_INPUT status : DINT; END_VAR
+END_FUNCTION
+
+FUNCTION main : DINT
+	puts('hello, world');
+END_FUNCTION
+
+FUNCTION _start : DINT
+    exit(main(0));
+END_FUNCTION 

examples/simple_program.st

@@ -1,8 +1,8 @@
-        TYPE the_struct : STRUCT END_STRUCT END_TYPE
-            
-        PROGRAM prg
-            VAR
-                my_struct : STRUCT
-                END_STRUCT
-            END_VAR
-        END_PROGRAM
+TYPE the_struct : STRUCT END_STRUCT END_TYPE
+
+PROGRAM prg
+    VAR
+        my_struct : STRUCT
+        END_STRUCT
+    END_VAR
+END_PROGRAM

src/ast.rs

@@ -60,6 +60,7 @@ pub struct Implementation {
     pub location: SourceRange,
     pub overriding: bool,
     pub access: Option<AccessModifier>,
+    pub associated_class: Option<String>,
 }
 
 #[derive(Debug, Copy, PartialEq, Clone)]

src/codegen.rs

@@ -71,21 +71,33 @@ impl<'ink> CodeGen<'ink> {
     fn generate_llvm_index(
         &self,
         module: &Module<'ink>,
+        annotations: &AnnotationMap,
         global_index: &Index,
     ) -> Result<LlvmTypedIndex<'ink>, CompileError> {
         let llvm = Llvm::new(self.context, self.context.create_builder());
         let mut index = LlvmTypedIndex::new();
         //Generate types index, and any global variables associated with them.
-        let llvm_type_index = data_type_generator::generate_data_types(&llvm, global_index)?;
+        let llvm_type_index =
+            data_type_generator::generate_data_types(&llvm, global_index, annotations)?;
         index.merge(llvm_type_index);
         //Generate global variables
-        let llvm_gv_index =
-            variable_generator::generate_global_variables(module, &llvm, global_index, &index)?;
+        let llvm_gv_index = variable_generator::generate_global_variables(
+            module,
+            &llvm,
+            global_index,
+            annotations,
+            &index,
+        )?;
         index.merge(llvm_gv_index);
         //Generate opaque functions for implementations and associate them with their types
         let llvm = Llvm::new(self.context, self.context.create_builder());
-        let llvm_impl_index =
-            pou_generator::generate_implementation_stubs(module, llvm, global_index, &index)?;
+        let llvm_impl_index = pou_generator::generate_implementation_stubs(
+            module,
+            llvm,
+            global_index,
+            annotations,
+            &index,
+        )?;
         index.merge(llvm_impl_index);
         Ok(index)
     }
@@ -94,15 +106,15 @@ impl<'ink> CodeGen<'ink> {
     pub fn generate(
         &self,
         unit: &CompilationUnit,
-        _annotations: &AnnotationMap,
+        annotations: &AnnotationMap,
         global_index: &Index,
     ) -> Result<String, CompileError> {
         //Associate the index type with LLVM types
-        let llvm_index = self.generate_llvm_index(&self.module, global_index)?;
+        let llvm_index = self.generate_llvm_index(&self.module, annotations, global_index)?;
 
         //generate all pous
         let llvm = Llvm::new(self.context, self.context.create_builder());
-        let pou_generator = PouGenerator::new(llvm, global_index, &llvm_index);
+        let pou_generator = PouGenerator::new(llvm, global_index, annotations, &llvm_index);
         //Generate the POU stubs in the first go to make sure they can be referenced.
         for implementation in &unit.implementations {
             //Don't generate external functions