Add automated testbench generation according to supplied test input

[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..ff2bb4bd0c9d5d38465ca78dd56bc64dc674dba4 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 | TestInput | TestCycles
   deriving (Show, Data, Typeable)
   
 isTopEntity :: CLasHAnn -> Bool
@@ -13,4 +13,12 @@ isTopEntity _         = False
 
 isInitState :: CLasHAnn -> Bool
 isInitState InitState = True
-isInitState _         = False
\ No newline at end of file
+isInitState _         = False
+
+isTestInput :: CLasHAnn -> Bool
+isTestInput TestInput = True
+isTestInput _         = False
+
+isTestCycles :: CLasHAnn -> Bool
+isTestCycles TestCycles = True
+isTestCycles _          = False
\ No newline at end of file