summaryrefslogtreecommitdiff
path: root/app/IR.hs
blob: fa987e50a2d0d63ed7b4ce3bcf6327f2d8a8350d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
module IR where

import qualified Data.Map as M

data PrlgStr
  = CallS String [PrlgStr]
  | LiteralS String
  deriving (Show)

data Id =
  Id
    { str :: Int
    , arity :: Int
    }
  deriving (Show, Eq, Ord)

data PrlgInt
  = CallI Id [PrlgInt]
  | LiteralI Int
  | VarI Int Int
  | VoidI
  deriving (Show)

data StrTable =
  StrTable Int (M.Map String Int) (M.Map Int String)
  deriving (Show)

emptystrtable = StrTable 1 M.empty M.empty

strtablize t@(StrTable nxt fwd rev) str =
  case fwd M.!? str of
    Just i -> (t, i)
    _ -> (StrTable (nxt + 1) (M.insert str nxt fwd) (M.insert nxt str rev), nxt)