{-# LANGUAGE DeriveDataTypeable #-}
module CLasH.Translator.Annotations where
-import Language.Haskell.TH
+import qualified Language.Haskell.TH as TH
import Data.Data
-data CLasHAnn = TopEntity | InitState | TestInput | TestCycles
+data CLasHAnn = TopEntity | InitState TH.Name | TestInput | TestCycles
deriving (Show, Data, Typeable)
isTopEntity :: CLasHAnn -> Bool
isTopEntity _ = False
isInitState :: CLasHAnn -> Bool
-isInitState InitState = True
-isInitState _ = False
+isInitState (InitState _) = True
+isInitState _ = False
isTestInput :: CLasHAnn -> Bool
isTestInput TestInput = True