Sets.hs

{-# LANGUAGE FlexibleContexts #-}

-- | Phantom-typed term DSL for the hydra.lib.sets library

module Hydra.Dsl.Meta.Lib.Sets where

import Hydra.Phantoms
import Hydra.Dsl.AsTerm
import Hydra.Dsl.Meta.Phantoms
import qualified Hydra.Dsl.Terms as Terms
import Hydra.Sources.Libraries

import Data.Set


-- | Delete an element from a set.
delete :: AsTerm t a => t -> TTerm (Set a) -> TTerm (Set a)
delete x = primitive2 _sets_delete (asTerm x)

-- | Compute the difference of two sets.
difference :: TTerm (Set a) -> TTerm (Set a) -> TTerm (Set a)
difference = primitive2 _sets_difference

-- | Create an empty set.
empty :: TTerm (Set a)
empty = primitive _sets_empty

-- | Create a set from a list.
fromList :: AsTerm t [a] => t -> TTerm (Set a)
fromList xs = primitive1 _sets_fromList (asTerm xs)

-- | Insert an element into a set.
insert :: TTerm a -> TTerm (Set a) -> TTerm (Set a)
insert = primitive2 _sets_insert

-- | Compute the intersection of two sets.
intersection :: TTerm (Set a) -> TTerm (Set a) -> TTerm (Set a)
intersection = primitive2 _sets_intersection

-- | Map a function over a set.
map :: TTerm (a -> b) -> TTerm (Set a) -> TTerm (Set b)
map = primitive2 _sets_map

-- | Check if an element is in a set.
member :: TTerm a -> TTerm (Set a) -> TTerm Bool
member = primitive2 _sets_member

-- | Check if a set is empty.
null :: TTerm (Set a) -> TTerm Bool
null = primitive1 _sets_null

-- | Create a singleton set.
singleton :: TTerm a -> TTerm (Set a)
singleton = primitive1 _sets_singleton

-- | Get the size of a set.
size :: TTerm (Set a) -> TTerm Int
size = primitive1 _sets_size

-- | Convert a set to a list.
toList :: TTerm (Set a) -> TTerm [a]
toList = primitive1 _sets_toList

-- | Compute the union of two sets.
union :: TTerm (Set a) -> TTerm (Set a) -> TTerm (Set a)
union = primitive2 _sets_union

-- | Compute the union of multiple sets.
unions :: TTerm [Set a] -> TTerm (Set a)
unions = primitive1 _sets_unions