1 files changed, 41 insertions, 0 deletions
diff --git a/lib/Language/Elna/Backend/Allocator.hs b/lib/Language/Elna/Backend/Allocator.hs
new file mode 100644
index 0000000..9ad849e
--- /dev/null
+++ b/lib/Language/Elna/Backend/Allocator.hs
@@ -0,0 +1,41 @@
+module Language.Elna.Backend.Allocator
+    ( MachineConfiguration(..)
+    , Store(..)
+    , allocate
+    ) where
+
+import Data.HashMap.Strict (HashMap)
+import Data.Vector (Vector)
+import Language.Elna.Backend.Intermediate (Operand(..), Quadruple(..), Variable(..))
+import Language.Elna.Location (Identifier(..))
+
+newtype Store r = Store r
+
+newtype MachineConfiguration r = MachineConfiguration
+    { temporaryRegister :: r
+    }
+
+allocate
+    :: forall r
+    . MachineConfiguration r
+    -> HashMap Identifier (Vector (Quadruple Variable))
+    -> HashMap Identifier (Vector (Quadruple (Store r)))
+allocate MachineConfiguration{..} = fmap function
+  where
+    function :: Vector (Quadruple Variable) -> Vector (Quadruple (Store r))
+    function = fmap quadruple
+    quadruple :: Quadruple Variable -> Quadruple (Store r)
+    quadruple StartQuadruple = StartQuadruple
+    quadruple StopQuadruple = StopQuadruple
+    quadruple (ParameterQuadruple operand1) =
+        ParameterQuadruple (operand operand1)
+    quadruple (CallQuadruple name count) = CallQuadruple name count
+    quadruple (AddQuadruple operand1 operand2 _) =
+        AddQuadruple (operand operand1) (operand operand2) (Store temporaryRegister)
+    quadruple (SubtractionQuadruple operand1 operand2 _) =
+        SubtractionQuadruple (operand operand1) (operand operand2) (Store temporaryRegister)
+    quadruple (NegationQuadruple operand1 _) =
+        NegationQuadruple (operand operand1) (Store temporaryRegister)
+    operand :: Operand Variable -> Operand (Store r)
+    operand (IntOperand x) = IntOperand x
+    operand (VariableOperand _) = VariableOperand (Store temporaryRegister)