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 ff2bb4bd0c9d5d38465ca78dd56bc64dc674dba4..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 | TestInput | TestCycles
+data CLasHAnn = TopEntity | InitState Name | TestInput | TestCycles
   deriving (Show, Data, Typeable)
   
 isTopEntity :: CLasHAnn -> Bool
@@ -12,8 +12,8 @@ isTopEntity TopEntity = True
 isTopEntity _         = False
 
 isInitState :: CLasHAnn -> Bool
-isInitState InitState = True
-isInitState _         = False
+isInitState (InitState _) = True
+isInitState _             = False
 
 isTestInput :: CLasHAnn -> Bool
 isTestInput TestInput = True