Merge branch 'main' of https://github.com/Cr0a3/ygen

src/IR/nodes/alloca.rs

@@ -1,5 +1,5 @@
 use crate::IR::{Function, TypeMetadata, Var};
-use crate::Support::ColorClass;
+use crate::Support::{AsAny, ColorClass};
 
 use super::{Alloca, EvalOptVisitor, Ir};
 
@@ -59,6 +59,21 @@ impl EvalOptVisitor for Alloca<Var, TypeMetadata> {
     }
 }
 
+impl<T, U> Alloca<T, U> where
+    T: Clone + AsAny + 'static,
+    U: Clone + AsAny + 'static
+{
+    /// Returns the output variable
+    pub fn getOut(&self) -> Var {
+        self.inner1.as_any().downcast_ref::<Var>().unwrap().to_owned()
+    }
+
+    /// Returns the type which is allocated
+    pub fn getTypeToAlloc(&self) -> TypeMetadata {
+        self.inner2.as_any().downcast_ref::<TypeMetadata>().unwrap().to_owned()
+    }
+}
+
 impl Function {
     /// Builds an stack allocation (the out var is the pointer to the allocated stack region)
     pub fn BuildAlloca(&mut self, ty: TypeMetadata) -> Var {

src/IR/nodes/assign.rs

@@ -1,3 +1,5 @@
+use std::any::TypeId;
+
 use super::*;
 
 impl Ir for Assign<Var, Type> {
@@ -204,6 +206,40 @@ impl EvalOptVisitor for Assign<Var, Const> {
     }
 }
 
+impl<U> Assign<Var, U> where 
+    U: AsAny + 'static
+{
+    /// Returns the type of the assignment
+    pub fn getType(&self) -> TypeMetadata {
+        self.inner1.ty
+    }
+
+    /// Returns the output variable
+    pub fn getOut(&self) -> Var {
+        self.inner1.to_owned()
+    }
+
+    /// Returns if the operand is an constant
+    pub fn isOpConstVal(&self) -> bool {
+        self.inner2.type_id() == TypeId::of::<Type>()
+    }
+
+    /// Returns if the operand is an constant ptr load
+    pub fn isOpConstAdr(&self) -> bool {
+        self.inner2.type_id() == TypeId::of::<Const>()
+    }
+
+    /// Returns the operand as a constant value
+    pub fn getOpConstVal(&self) -> Type {
+        self.inner2.as_any().downcast_ref::<Type>().unwrap().clone()
+    }
+
+    /// Returns the operand as a constant adr (Const data type)
+    pub fn getOpConstAdr(&self) -> Const {
+        self.inner2.as_any().downcast_ref::<Const>().unwrap().clone()
+    }
+}
+
 /// Trait used for overloading the BuildAssign function
 pub trait BuildAssign<T> {
     /// builds an assignment

src/IR/nodes/br.rs

@@ -1,6 +1,6 @@
 use std::collections::HashMap;
 
-use crate::Support::ColorClass;
+use crate::Support::{AsAny, ColorClass};
 use crate::IR::block::BlockId;
 use crate::IR::{Function, Type, Var};
 
@@ -72,6 +72,15 @@ impl EvalOptVisitor for Br<BlockId> {
     }
 }
 
+impl<T> Br<T> where
+    T: Clone + AsAny + 'static
+{
+    /// Returns the block which the br branches to
+    pub fn getBlockToBranch(&self) -> BlockId {
+        self.inner1.as_any().downcast_ref::<BlockId>().unwrap().to_owned()
+    }
+}
+
 impl Ir for BrCond<Var, BlockId, BlockId> {
     fn dump(&self) -> String {
         format!("br cond {} {}, {}", self.inner1.name, self.inner2.name, self.inner3.name)
@@ -150,6 +159,27 @@ impl EvalOptVisitor for BrCond<Var, BlockId, BlockId> {
     }
 }
 
+impl<T, U, Z> BrCond<T, U, Z> where 
+    T: Clone + AsAny + 'static,
+    U: Clone + AsAny + 'static,
+    Z: Clone + AsAny + 'static
+{
+    /// Returns the condition variable
+    pub fn getCondition(&self) -> Var {
+        self.inner1.as_any().downcast_ref::<Var>().unwrap().to_owned()
+    }
+
+    /// Returns the true branch
+    pub fn getBranchTrue(&self) -> BlockId {
+        self.inner2.as_any().downcast_ref::<BlockId>().unwrap().to_owned()
+    }
+
+    /// Returns the false branch
+    pub fn getBranchFalse(&self) -> BlockId {
+        self.inner2.as_any().downcast_ref::<BlockId>().unwrap().to_owned()
+    }
+}
+
 /// This trait is used for building br nodes
 pub trait BuildBr<T> {
     /// Builds a br node

src/IR/nodes/call.rs

@@ -95,6 +95,27 @@ impl Ir for Call<FuncId, Vec<Var>, Var> {
     }
 }
 
+impl<T, U, Z> Call<T, U, Z> where 
+    T: Clone + AsAny + 'static,
+    U: Clone + AsAny + 'static,
+    Z: Clone + AsAny + 'static
+{
+    /// Returns the call target
+    pub fn getCallTarget(&self) -> FuncId {
+        self.inner1.as_any().downcast_ref::<FuncId>().unwrap().to_owned()
+    }
+
+    /// Returns the arguments
+    pub fn getArgs(&self) -> Vec<Var> {
+        self.inner2.as_any().downcast_ref::<Vec<Var>>().unwrap().to_owned()
+    }
+
+    /// Returns the variable which stores the result of the call
+    pub fn getOutputVar(&self) -> Var {
+        self.inner3.as_any().downcast_ref::<Var>().unwrap().to_owned()
+    }
+}
+
 impl EvalOptVisitor for Call<FuncId, Vec<Var>, Var> {
     fn maybe_inline(&self, _: &HashMap<String, Type>) -> Option<Box<dyn Ir>> {
         None

src/IR/nodes/cast.rs

@@ -61,6 +61,32 @@ impl Ir for Cast<Var, TypeMetadata, Var> {
     }
 }
 
+impl<T, U, Z> Cast<T, U, Z> where 
+    T: Clone + AsAny + 'static,
+    U: Clone + AsAny + 'static,
+    Z: Clone + AsAny + 'static
+{
+    /// Returns the input variable
+    pub fn getInput(&self) -> Var {
+        self.inner1.as_any().downcast_ref::<Var>().unwrap().to_owned()
+    }
+
+    /// Returns the output variable
+    pub fn getOutput(&self) -> Var {
+        self.inner3.as_any().downcast_ref::<Var>().unwrap().to_owned()
+    }
+
+    /// Returns the type to which we cast
+    pub fn getCastType(&self) -> TypeMetadata {
+        self.inner2.as_any().downcast_ref::<TypeMetadata>().unwrap().to_owned()
+    }
+
+    /// Returns the type from which we cast
+    pub fn getFromType(&self) -> TypeMetadata {
+        self.getInput().ty
+    }
+}
+
 impl EvalOptVisitor for Cast<Var, TypeMetadata, Var> {
     fn maybe_inline(&self, vars: &HashMap<String, Type>) -> Option<Box<dyn Ir>> {
         if let Some(var) = vars.get(&self.inner1.name) {

src/IR/nodes/cmp.rs

@@ -123,6 +123,33 @@ impl IsNode for Cmp {
     }
 }
 
+impl Cmp {
+    /// Returns the mode with which the node compares
+    pub fn getCmpMode(&self) -> CmpMode {
+        self.mode
+    }
+
+    /// Returns the left side operand
+    pub fn getLsVar(&self) -> Var {
+        self.ls.to_owned()
+    }
+
+    /// Returns the right side operand
+    pub fn getRsVar(&self) -> Var {
+        self.rs.to_owned()
+    }
+
+    /// Returns the output variable
+    pub fn getOutput(&self) -> Var {
+        self.out.to_owned()
+    }
+
+    /// Returns the type of the node
+    pub fn getType(&self) -> TypeMetadata {
+        self.out.ty
+    }
+}
+
 /// The trait `BuildCmp` is used to build the cmp node
 pub trait BuildCmp {
     /// builds the compare node

src/IR/nodes/ret.rs

@@ -1,3 +1,5 @@
+use std::any::TypeId;
+
 use super::*;
 
 impl Ir for Return<Type> {
@@ -133,6 +135,30 @@ impl EvalOptVisitor for Return<Var> {
     }
 }
 
+impl<T> Return<T> where 
+    T: Clone + AsAny + 'static
+{
+    /// Returns the node a constant type?
+    pub fn isRetConst(&self) -> bool {
+        self.inner1.type_id() == TypeId::of::<Type>()
+    }
+
+    /// Returns the node a variable?
+    pub fn isRetVar(&self) -> bool {
+        self.inner1.type_id() == TypeId::of::<Var>()
+    }
+
+    /// Returns the constant the node returns (else panics)
+    pub fn getRetConst(&self) -> Type {
+        self.inner1.as_any().downcast_ref::<Type>().unwrap().to_owned()
+    }
+
+    /// Returns the variable the node returns (else panics)
+    pub fn getRetVar(&self) -> Var {
+        self.inner1.as_any().downcast_ref::<Var>().unwrap().to_owned()
+    }
+}
+
 /// Trait for the return instruction
 /// Used for overloading the BuildRet function
 pub trait BuildReturn<T> {