{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE UndecidableInstances #-}
{-|
   Module for the definition of 'EClass'.
-}
module Data.Equality.Graph.Classes
    ( module Data.Equality.Graph.Classes
    , module Data.Equality.Graph.Classes.Id
    ) where

import qualified Data.Set as S

import Data.Functor.Classes

import Data.Equality.Graph.Classes.Id
import Data.Equality.Graph.Nodes

import Data.Equality.Analysis

-- | An e-class (an equivalence class of terms) of a language @l@.
--
-- Intuitively, an e-graph is a set of equivalence classes (e-classes). Each
-- e-class is a set of e-nodes representing equivalent terms from a given
-- language, and an e-node is a function symbol paired with a list of children
-- e-classes.
data EClass l = EClass
    { forall (l :: * -> *). EClass l -> ClassId
eClassId      :: {-# UNPACK #-} !ClassId -- ^ E-class identifier
    , forall (l :: * -> *). EClass l -> Set (ENode l)
eClassNodes   :: !(S.Set (ENode l))      -- ^ E-nodes in this class
    , forall (l :: * -> *). EClass l -> Domain l
eClassData    :: Domain l                -- ^ The analysis data associated with this eclass.
    , forall (l :: * -> *). EClass l -> NodeMap l ClassId
eClassParents :: !(NodeMap l ClassId)    -- ^ E-nodes which are parents of this e-class and their corresponding e-class ids. We found a mapping from nodes to e-class ids a better representation than @[(ENode l, ClassId)]@, and we get de-duplication built-in.
    }

instance (Show (Domain l), Show1 l) => Show (EClass l) where
    show :: EClass l -> String
show (EClass ClassId
a Set (ENode l)
b Domain l
d NodeMap l ClassId
c) = String
"Id: " String -> ShowS
forall a. Semigroup a => a -> a -> a
<> ClassId -> String
forall a. Show a => a -> String
show ClassId
a String -> ShowS
forall a. Semigroup a => a -> a -> a
<> String
"\nNodes: " String -> ShowS
forall a. Semigroup a => a -> a -> a
<> Set (ENode l) -> String
forall a. Show a => a -> String
show Set (ENode l)
b String -> ShowS
forall a. Semigroup a => a -> a -> a
<> String
"\nParents: " String -> ShowS
forall a. Semigroup a => a -> a -> a
<> NodeMap l ClassId -> String
forall a. Show a => a -> String
show NodeMap l ClassId
c String -> ShowS
forall a. Semigroup a => a -> a -> a
<> String
"\nData: " String -> ShowS
forall a. Semigroup a => a -> a -> a
<> Domain l -> String
forall a. Show a => a -> String
show Domain l
d