summaryrefslogtreecommitdiff
path: root/Jennisys/Jennisys/TypeChecker.fs
blob: 8227b6934d8079317d6f01d5239d9e96eeece4a9 (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
34
35
36
37
38
39
40
41
42
43
module TypeChecker

open Ast
open Printer
open System.Collections.Generic

let GetClass name decls =
  match decls |> List.tryFind (function Class(n,_,_) when n = name -> true | _ -> false) with
  | Some(cl) -> cl
  | None -> Class(name,[],[])

let GetModel name decls =
  match decls |> List.tryFind (function Model(n,_,_,_,_) when n = name -> true | _ -> false) with
  | Some(m) -> m
  | None -> Model(name,[],[],[],BoolLiteral(true))

let GetCode name decls =
  match decls |> List.tryFind (function Code(n,_) when n = name -> true | _ -> false) with
  | Some(c) -> c
  | None -> Code(name,[])

let IsUserType prog tpo =
  match tpo with
  | Some(tp) ->
      let tpname = match tp with
                   | NamedType(tname,_) -> tname
                   | InstantiatedType(tname, _) -> tname
                   | _ -> ""
      match prog with
      | Program(components) -> components |> List.filter (function Component(Class(name,_,_),_,_) when name = tpname -> true
                                                                   | _                                               -> false) |> List.isEmpty |> not
  | None -> false

let TypeCheck prog =
  match prog with
  | SProgram(decls) ->
      let componentNames = decls |> List.choose (function Class(name,_,_) -> Some(name) | _ -> None)
      let clist = componentNames |> List.map (fun name -> Component(GetClass name decls, GetModel name decls, GetCode name decls))
      Some(Program(clist))

// TODO: implement this
let InferType prog expr = 
  NamedType("unknown", [])