Start support on initial state. Substates currently break

[matthijs/master-project/cλash.git] / cλash / CLasH / Translator / Annotations.hs

diff --git a/cλash/CLasH/Translator/Annotations.hs b/cλash/CLasH/Translator/Annotations.hs
index 08e7845086be4027970799f20e2032f51b558875..6176438c1e7349f664324220f127b344d36c293e 100644 (file)
--- a/cλash/CLasH/Translator/Annotations.hs
+++ b/cλash/CLasH/Translator/Annotations.hs
@@ -4,7 +4,7 @@ module CLasH.Translator.Annotations where
 import Language.Haskell.TH
 import Data.Data
 
-data CLasHAnn = TopEntity | InitState
+data CLasHAnn = TopEntity | InitState Name | TestInput | TestCycles
   deriving (Show, Data, Typeable)
   
 isTopEntity :: CLasHAnn -> Bool
@@ -12,5 +12,13 @@ isTopEntity TopEntity = True
 isTopEntity _         = False
 
 isInitState :: CLasHAnn -> Bool
-isInitState InitState = True
-isInitState _         = False
\ No newline at end of file
+isInitState (InitState _) = True
+isInitState _             = False
+
+isTestInput :: CLasHAnn -> Bool
+isTestInput TestInput = True
+isTestInput _         = False
+
+isTestCycles :: CLasHAnn -> Bool
+isTestCycles TestCycles = True
+isTestCycles _          = False
\ No newline at end of file