Source file Atom.ml

(*******************************************************************************
 * electrod - a model finder for relational first-order linear temporal logic
 * 
 * Copyright (C) 2016-2019 ONERA
 * Authors: Julien Brunel (ONERA), David Chemouil (ONERA)
 * 
 * This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/.
 * 
 * SPDX-License-Identifier: MPL-2.0
 * License-Filename: LICENSE.md
 ******************************************************************************)

type t =
  { sym : Symbol.t
  ; loc : Location.t option
  }

let compare a1 a2 = Symbol.compare a1.sym a2.sym

let equal a1 a2 = Symbol.equal a1.sym a2.sym

(* |> Fun.tap *)
(*      (fun res -> *)
(*         (Fmt.epr "Atom.equal %a %a = %B" *)
(*            Symbol.pp a1.sym *)
(*            Symbol.pp a2.sym *)
(*            res)) *)

let atom ?loc s = { sym = Symbol.make s; loc }

let of_raw_ident id = atom ~loc:(Raw_ident.location id) (Raw_ident.basename id)

let hash atom = Symbol.hash atom.sym

(** Generic interface implementations *)

let pp out { sym; _ } = Symbol.pp out sym

module P = Intf.Print.Mixin (struct
  type nonrec t = t

  let pp = pp
end)

include P

let pp_list atoms = Fmtc.(braces_ @@ list ~sep:sp pp) atoms