Clean up source files:

[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..2c875505c4b4091f2019d4f9af66fa3713599853 100644 (file)
--- a/cλash/CLasH/Translator/Annotations.hs
+++ b/cλash/CLasH/Translator/Annotations.hs
@@ -1,10 +1,10 @@
 {-# 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
+data CLasHAnn = TopEntity | InitState TH.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