mirror of
https://github.com/CatalaLang/catala.git
synced 2024-11-08 07:51:43 +03:00
101 lines
3.3 KiB
OCaml
101 lines
3.3 KiB
OCaml
(* This file is part of the Catala compiler, a specification language for tax
|
|
and social benefits computation rules. Copyright (C) 2020 Inria, contributor:
|
|
Denis Merigoux <denis.merigoux@inria.fr>
|
|
|
|
Licensed under the Apache License, Version 2.0 (the "License"); you may not
|
|
use this file except in compliance with the License. You may obtain a copy of
|
|
the License at
|
|
|
|
http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
Unless required by applicable law or agreed to in writing, software
|
|
distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
License for the specific language governing permissions and limitations under
|
|
the License. *)
|
|
|
|
open Utils
|
|
open Shared_ast
|
|
module ScopeMap : Map.S with type key = ScopeName.t = Map.Make (ScopeName)
|
|
|
|
module SubScopeNameSet : Set.S with type elt = SubScopeName.t =
|
|
Set.Make (SubScopeName)
|
|
|
|
module SubScopeMap : Map.S with type key = SubScopeName.t =
|
|
Map.Make (SubScopeName)
|
|
|
|
module StructFieldMapLift = Bindlib.Lift (StructFieldMap)
|
|
module EnumConstructorMapLift = Bindlib.Lift (EnumConstructorMap)
|
|
|
|
type location = scopelang glocation
|
|
|
|
module LocationSet : Set.S with type elt = location Marked.pos =
|
|
Set.Make (struct
|
|
type t = location Marked.pos
|
|
|
|
let compare = Expr.compare_location
|
|
end)
|
|
|
|
type expr = (scopelang, Pos.t) gexpr
|
|
type naked_expr = (scopelang, Pos.t) naked_gexpr
|
|
|
|
module ExprMap = Map.Make (struct
|
|
type t = expr
|
|
|
|
let compare = Expr.compare
|
|
end)
|
|
|
|
let rec locations_used (e : expr) : LocationSet.t =
|
|
match Marked.unmark e with
|
|
| ELocation l -> LocationSet.singleton (l, Marked.get_mark e)
|
|
| EVar _ | ELit _ | EOp _ -> LocationSet.empty
|
|
| EAbs (binder, _) ->
|
|
let _, body = Bindlib.unmbind binder in
|
|
locations_used body
|
|
| EStruct (_, es) ->
|
|
StructFieldMap.fold
|
|
(fun _ e' acc -> LocationSet.union acc (locations_used e'))
|
|
es LocationSet.empty
|
|
| EStructAccess (e1, _, _) -> locations_used e1
|
|
| EEnumInj (e1, _, _) -> locations_used e1
|
|
| EMatchS (e1, _, es) ->
|
|
EnumConstructorMap.fold
|
|
(fun _ e' acc -> LocationSet.union acc (locations_used e'))
|
|
es (locations_used e1)
|
|
| EApp (e1, args) ->
|
|
List.fold_left
|
|
(fun acc arg -> LocationSet.union (locations_used arg) acc)
|
|
(locations_used e1) args
|
|
| EIfThenElse (e1, e2, e3) ->
|
|
LocationSet.union (locations_used e1)
|
|
(LocationSet.union (locations_used e2) (locations_used e3))
|
|
| EDefault (excepts, just, cons) ->
|
|
List.fold_left
|
|
(fun acc except -> LocationSet.union (locations_used except) acc)
|
|
(LocationSet.union (locations_used just) (locations_used cons))
|
|
excepts
|
|
| EArray es ->
|
|
List.fold_left
|
|
(fun acc e' -> LocationSet.union acc (locations_used e'))
|
|
LocationSet.empty es
|
|
| ErrorOnEmpty e' -> locations_used e'
|
|
|
|
type io_input = NoInput | OnlyInput | Reentrant
|
|
type io = { io_output : bool Marked.pos; io_input : io_input Marked.pos }
|
|
|
|
type rule =
|
|
| Definition of location Marked.pos * typ * io * expr
|
|
| Assertion of expr
|
|
| Call of ScopeName.t * SubScopeName.t
|
|
|
|
type scope_decl = {
|
|
scope_decl_name : ScopeName.t;
|
|
scope_sig : (typ * io) ScopeVarMap.t;
|
|
scope_decl_rules : rule list;
|
|
}
|
|
|
|
type program = {
|
|
program_scopes : scope_decl ScopeMap.t;
|
|
program_ctx : decl_ctx;
|
|
}
|