{-# LANGUAGE CPP #-}
{-# LANGUAGE LambdaCase #-}
{-# LANGUAGE MultiWayIf #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE RankNTypes #-}
{-# LANGUAGE RecursiveDo #-}
{-# LANGUAGE TemplateHaskell #-}
module Clash.Backend.Verilog
( VerilogState
, include
, uselibs
, encodingNote
, exprLit
, bits
, bit_char
, noEmptyInit
, Range (..)
, continueWithRange
)
where
import qualified Control.Applicative as A
import Control.Lens (Lens',(+=),(-=),(.=),(%=), makeLenses, use)
import Control.Monad (forM)
import Control.Monad.State (State)
import Data.Bits (Bits, testBit)
import Data.HashMap.Strict (HashMap)
import qualified Data.HashMap.Strict as HashMap
import Data.HashSet (HashSet)
import qualified Data.HashSet as HashSet
import Data.Maybe (catMaybes,fromMaybe,mapMaybe)
import Data.List
(mapAccumL, mapAccumR, nubBy, foldl')
import Data.List.Extra ((<:>))
#if !MIN_VERSION_base(4,11,0)
import Data.Monoid hiding (Product, Sum)
#endif
import Data.Semigroup.Monad.Extra
import Data.Text.Lazy (pack)
import qualified Data.Text.Lazy as Text
import qualified Data.Text as TextS
import Data.Text.Prettyprint.Doc.Extra
import qualified System.FilePath
import GHC.Stack (HasCallStack)
import Clash.Annotations.Primitive (HDL (..))
import Clash.Annotations.BitRepresentation.ClashLib
(bitsToBits)
import Clash.Annotations.BitRepresentation.Internal
(ConstrRepr'(..), DataRepr'(..), ConstrRepr'(..))
import Clash.Annotations.BitRepresentation.Util
(BitOrigin(Lit, Field), bitOrigins, bitRanges)
import Clash.Core.Var (Attr'(..))
import Clash.Backend
import Clash.Debug (traceIf)
import Clash.Netlist.BlackBox.Types (HdlSyn)
import Clash.Netlist.BlackBox.Util
(extractLiterals, renderBlackBox, renderFilePath)
import Clash.Netlist.Id (IdType (..), mkBasicId')
import Clash.Netlist.Types hiding (_intWidth, intWidth)
import Clash.Netlist.Util hiding (mkIdentifier, extendIdentifier)
import Clash.Signal.Internal (ActiveEdge (..))
import Clash.Util
(SrcSpan, noSrcSpan, curLoc, on, first, indexNote, makeCached, second)
data VerilogState =
VerilogState
{ _genDepth :: Int
, _idSeen :: HashMap Identifier Word
, _srcSpan :: SrcSpan
, _includes :: [(String,Doc)]
, _imports :: HashSet Text.Text
, _libraries :: HashSet Text.Text
, _dataFiles :: [(String,FilePath)]
, _memoryDataFiles:: [(String,String)]
, _customConstrs :: HashMap Identifier Identifier
, _intWidth :: Int
, _hdlsyn :: HdlSyn
, _escapedIds :: Bool
, _undefValue :: Maybe (Maybe Int)
}
makeLenses ''VerilogState
instance Backend VerilogState where
initBackend = VerilogState
0
HashMap.empty
noSrcSpan
[]
HashSet.empty
HashSet.empty
[]
[]
HashMap.empty
hdlKind = const Verilog
primDirs = const $ do root <- primsRoot
return [ root System.FilePath.</> "common"
, root System.FilePath.</> "commonverilog"
, root System.FilePath.</> "verilog"
]
extractTypes = const HashSet.empty
name = const "verilog"
extension = const ".v"
genHDL = const genVerilog
mkTyPackage _ _ = return []
hdlType _ = verilogType
hdlTypeErrValue = verilogTypeErrValue
hdlTypeMark = verilogTypeMark
hdlRecSel = verilogRecSel
hdlSig t ty = sigDecl (string t) ty
genStmt True = do cnt <- use genDepth
genDepth += 1
if cnt > 0
then emptyDoc
else "generate"
genStmt False = do genDepth -= 1
cnt <- use genDepth
if cnt > 0
then emptyDoc
else "endgenerate"
inst = inst_
expr = expr_
iwWidth = use intWidth
toBV _ = string
fromBV _ = string
hdlSyn = use hdlsyn
mkIdentifier = do
allowEscaped <- use escapedIds
return (go allowEscaped)
where
go _ Basic nm = case (TextS.take 1024 . filterReserved) (mkBasicId' Verilog True nm) of
nm' | TextS.null nm' -> "_clash_internal"
| otherwise -> nm'
go esc Extended (rmSlash -> nm) = case go esc Basic nm of
nm' | esc && nm /= nm' -> TextS.concat ["\\",nm," "]
| otherwise -> nm'
extendIdentifier = do
allowEscaped <- use escapedIds
return (go allowEscaped)
where
go _ Basic nm ext =
case (TextS.take 1024 . filterReserved) (mkBasicId' Verilog True (nm `TextS.append` ext)) of
nm' | TextS.null nm' -> "_clash_internal"
| otherwise -> nm'
go esc Extended (rmSlash . escapeTemplate -> nm) ext =
let nmExt = nm `TextS.append` ext
in case go esc Basic nm ext of
nm' | esc && nm' /= nmExt -> case TextS.isPrefixOf "c$" nmExt of
True -> TextS.concat ["\\",nmExt," "]
_ -> TextS.concat ["\\c$",nmExt," "]
| otherwise -> nm'
setModName _ = id
setSrcSpan = (srcSpan .=)
getSrcSpan = use srcSpan
blockDecl _ ds = do
decs <- decls ds
if isEmpty decs
then insts ds
else
pure decs <> line <>
insts ds
unextend = return rmSlash
addIncludes inc = includes %= (inc ++)
addLibraries libs = libraries %= (\s -> foldl' (flip HashSet.insert) s libs)
addImports inps = imports %= (\s -> foldl' (flip HashSet.insert) s inps)
addAndSetData f = do
fs <- use dataFiles
let (fs',f') = renderFilePath fs f
dataFiles .= fs'
return f'
getDataFiles = use dataFiles
addMemoryDataFile f = memoryDataFiles %= (f:)
getMemoryDataFiles = use memoryDataFiles
seenIdentifiers = idSeen
ifThenElseExpr _ = True
rmSlash :: Identifier -> Identifier
rmSlash nm = fromMaybe nm $ do
nm1 <- TextS.stripPrefix "\\" nm
pure (TextS.filter (not . (== ' ')) nm1)
type VerilogM a = Mon (State VerilogState) a
reservedWords :: [Identifier]
reservedWords = ["always","and","assign","automatic","begin","buf","bufif0"
,"bufif1","case","casex","casez","cell","cmos","config","deassign","default"
,"defparam","design","disable","edge","else","end","endcase","endconfig"
,"endfunction","endgenerate","endmodule","endprimitive","endspecify"
,"endtable","endtask","event","for","force","forever","fork","function"
,"generate","genvar","highz0","highz1","if","ifnone","incdir","include"
,"initial","inout","input","instance","integer","join","large","liblist"
,"library","localparam","macromodule","medium","module","nand","negedge"
,"nmos","nor","noshowcancelled","not","notif0","notif1","or","output"
,"parameter","pmos","posedge","primitive","pull0","pull1","pulldown","pullup"
,"pulsestyle_onevent","pulsestyle_ondetect","rcmos","real","realtime","reg"
,"release","repeat","rnmos","rpmos","rtran","rtranif0","rtranif1","scalared"
,"showcancelled","signed","small","specify","specparam","strong0","strong1"
,"supply0","supply1","table","task","time","tran","tranif0","tranif1","tri"
,"tri0","tri1","triand","trior","trireg","unsigned","use","uwire","vectored"
,"wait","wand","weak0","weak1","while","wire","wor","xnor","xor"]
filterReserved :: Identifier -> Identifier
filterReserved s = if s `elem` reservedWords
then s `TextS.append` "_r"
else s
genVerilog :: SrcSpan -> HashMap Identifier Word -> Component -> VerilogM ((String,Doc),[(String,Doc)])
genVerilog sp seen c = preserveSeen $ do
Mon (idSeen .= seen)
Mon (setSrcSpan sp)
v <- commentHeader <> line <> timescale <> line <> module_ c
incs <- Mon $ use includes
return ((TextS.unpack cName,v),incs)
where
cName = componentName c
commentHeader
= "/* AUTOMATICALLY GENERATED VERILOG-2001 SOURCE CODE."
<> line <> "** GENERATED BY CLASH " <> string (Text.pack clashVer) <> ". DO NOT MODIFY."
<> line <> "*/"
timescale = "`timescale 100fs/100fs"
sigPort :: Maybe WireOrReg
-> TextS.Text
-> HWType
-> Maybe Expr
-> VerilogM Doc
sigPort wor pName hwType iEM =
addAttrs (hwTypeAttrs hwType)
(portType <+> verilogType hwType <+> stringS pName <> iE <> encodingNote hwType)
where
portType = case wor of
Nothing -> if isBiSignalIn hwType then "inout" else "input"
Just Wire -> "output" <+> "wire"
Just Reg -> "output" <+> "reg"
iE = maybe emptyDoc (noEmptyInit . expr_ False) iEM
module_ :: Component -> VerilogM Doc
module_ c =
addSeen c *> modVerilog <* Mon (imports .= HashSet.empty >> libraries .= HashSet.empty)
where
modVerilog = do
body <- modBody
imps <- Mon $ use imports
libs <- Mon $ use libraries
modHeader <> line <> modPorts <> line <>
include (HashSet.toList imps) <>
uselibs (HashSet.toList libs) <>
pure body <> line <> modEnding
modHeader = "module" <+> stringS (componentName c)
modPorts = indent 4 (tupleInputs inPorts <> line <> tupleOutputs outPorts <> semi)
modBody = indent 2 (decls (declarations c)) <> line <> line <> indent 2 (insts (declarations c))
modEnding = "endmodule"
inPorts = sequence [ sigPort Nothing id_ hwType Nothing | (id_, hwType) <- inputs c ]
outPorts = sequence [ sigPort (Just wireOrReg) id_ hwType iEM | (wireOrReg, (id_, hwType), iEM) <- outputs c ]
commafy v = (comma <> space) <> pure v
tupleInputs v = v >>= \case
[] -> lparen <+> string "// No inputs" <> line
(x:xs) -> lparen <+> string "// Inputs"
<> line <> (string " " <> pure x)
<> line <> vcat (forM xs commafy)
<> line
tupleOutputs v = v >>= \case
[] -> string " // No outputs" <> line <> rparen
(x:xs) -> string " // Outputs"
<> line <> (if (length (inputs c)) > 0
then comma <> space <> pure x
else string " " <> pure x)
<> (if null xs then emptyDoc else line <> vcat (forM xs commafy))
<> line <> rparen
include :: Monad m => [Text.Text] -> Mon m Doc
include [] = emptyDoc
include xs = line <>
indent 2 (vcat (mapM (\i -> string "`include" <+> dquotes (string i)) xs))
<> line <> line
uselibs :: Monad m => [Text.Text] -> Mon m Doc
uselibs [] = emptyDoc
uselibs xs = line <>
indent 2 (string "`uselib" <+> (hsep (mapM (\l -> ("lib=" <> string l)) xs)))
<> line <> line
wireRegFileDoc :: WireOrReg -> (Either Identifier HWType) -> VerilogM Doc
wireRegFileDoc _ (Right FileType) = "integer"
wireRegFileDoc Wire _ = "wire"
wireRegFileDoc Reg _ = "reg"
addSeen :: Component -> VerilogM ()
addSeen c = do
let iport = [iName | (iName, _) <- inputs c]
oport = [oName | (_, (oName, _), _) <- outputs c]
nets = mapMaybe (\case {NetDecl' _ _ i _ _ -> Just i; _ -> Nothing}) $ declarations c
Mon $ idSeen %= (HashMap.unionWith max (HashMap.fromList (concatMap (map (,0)) [iport,oport,nets])))
verilogType :: HWType -> VerilogM Doc
verilogType t = case t of
Signed n -> "signed" <+> brackets (int (n-1) <> colon <> int 0)
Clock {} -> emptyDoc
Reset {} -> emptyDoc
Bit -> emptyDoc
Bool -> emptyDoc
FileType -> emptyDoc
_ -> brackets (int (typeSize t -1) <> colon <> int 0)
sigDecl :: VerilogM Doc -> HWType -> VerilogM Doc
sigDecl d t = verilogType t <+> d
verilogTypeMark :: HWType -> VerilogM Doc
verilogTypeMark = const emptyDoc
verilogTypeErrValue :: HWType -> VerilogM Doc
verilogTypeErrValue ty = do
udf <- Mon (use undefValue)
case udf of
Nothing -> braces (int (typeSize ty) <+> braces "1'bx")
Just Nothing -> int (typeSize ty) <> "'d0 /* undefined */"
Just (Just x) -> braces (int (typeSize ty) <+> braces ("1'b" <> int x)) <+> "/* undefined */"
verilogRecSel
:: HWType
-> Int
-> VerilogM Doc
verilogRecSel ty i = case modifier (Contiguous 0 0) (Indexed (ty,0,i)) of
Just (Contiguous start end,_resTy) -> brackets (int start <> colon <> int end)
_ -> error "Can't make a record selector"
decls :: [Declaration] -> VerilogM Doc
decls [] = emptyDoc
decls ds = do
dsDoc <- catMaybes <$> (mapM decl ds)
case dsDoc of
[] -> emptyDoc
_ -> punctuate' semi (A.pure dsDoc)
addAttrs
:: [Attr']
-> VerilogM Doc
-> VerilogM Doc
addAttrs [] t = t
addAttrs attrs' t =
"(*" <+> attrs'' <+> "*)" <+> t
where
attrs'' = string $ Text.intercalate ", " (map renderAttr attrs')
renderAttr :: Attr' -> Text.Text
renderAttr (StringAttr' key value) = pack $ concat [key, " = ", show value]
renderAttr (IntegerAttr' key value) = pack $ concat [key, " = ", show value]
renderAttr (BoolAttr' key True ) = pack $ concat [key, " = ", "1"]
renderAttr (BoolAttr' key False) = pack $ concat [key, " = ", "0"]
renderAttr (Attr' key ) = pack $ key
decl :: Declaration -> VerilogM (Maybe Doc)
decl (NetDecl' noteM wr id_ tyE iEM) =
Just A.<$> maybe id addNote noteM (addAttrs attrs (wireRegFileDoc wr tyE <+> tyDec tyE))
where
tyDec (Left ty) = stringS ty <+> stringS id_ <> iE
tyDec (Right ty) = sigDecl (stringS id_) ty <> iE
addNote n = mappend ("//" <+> stringS n <> line)
attrs = fromMaybe [] (hwTypeAttrs A.<$> either (const Nothing) Just tyE)
iE = maybe emptyDoc (noEmptyInit . expr_ False) iEM
decl _ = return Nothing
noEmptyInit :: (Monad m, Semigroup (m Doc)) => m Doc -> m Doc
noEmptyInit d = do
d1 <- d
if isEmpty d1
then emptyDoc
else (space <> string "=" <+> d)
insts :: [Declaration] -> VerilogM Doc
insts [] = emptyDoc
insts (TickDecl id_:ds) = comment "//" id_ <> line <> insts ds
insts (d:ds) = do
docM <- inst_ d
case docM of
Nothing -> insts ds
Just doc -> pure doc <> line <> line <> insts ds
stdMatch
:: Bits a
=> Int
-> a
-> a
-> String
stdMatch 0 _mask _value = []
stdMatch size mask value =
symbol : stdMatch (size - 1) mask value
where
symbol =
if testBit mask (size - 1) then
if testBit value (size - 1) then
'1'
else
'0'
else
'?'
patLitCustom'
:: Int
-> ConstrRepr'
-> VerilogM Doc
patLitCustom' size (ConstrRepr' _name _n mask value _anns) =
int size <> squote <> "b" <> (string $ Text.pack $ stdMatch size mask value)
patLitCustom
:: HWType
-> Literal
-> VerilogM Doc
patLitCustom (CustomSum _name _dataRepr size reprs) (NumLit (fromIntegral -> i)) =
patLitCustom' size (fst $ reprs !! i)
patLitCustom (CustomSP _name _dataRepr size reprs) (NumLit (fromIntegral -> i)) =
let (cRepr, _id, _tys) = reprs !! i in
patLitCustom' size cRepr
patLitCustom hwTy _
| CustomProduct _name dataRepr size _maybeFieldNames _reprs <- hwTy
, DataRepr' _typ _size [cRepr] <- dataRepr =
patLitCustom' size cRepr
patLitCustom x y = error $ $(curLoc) ++ unwords
[ "You can only pass CustomSP / CustomSum / CustomProduct and a NumLit to "
, "this function, not", show x, "and", show y ]
patMod :: HWType -> Literal -> Literal
patMod hwTy (NumLit i) = NumLit (i `mod` (2 ^ typeSize hwTy))
patMod _ l = l
inst_'
:: TextS.Text
-> Expr
-> HWType
-> [(Maybe Literal, Expr)]
-> VerilogM (Maybe Doc)
inst_' id_ scrut scrutTy es = fmap Just $
"always @(*) begin" <> line <>
indent 2 casez <> line <>
"end"
where
casez =
"casez" <+> parens var <> line <>
indent 2 (conds esNub) <> line <>
"endcase"
esMod = map (first (fmap (patMod scrutTy))) es
esNub = nubBy ((==) `on` fst) esMod
var = expr_ True scrut
conds :: [(Maybe Literal,Expr)] -> VerilogM Doc
conds [] = error $ $(curLoc) ++ "Empty list of conditions invalid."
conds [(_,e)] = "default" <+> ":" <+> stringS id_ <+> "=" <+> expr_ False e <> ";"
conds ((Nothing,e):_) = "default" <+> ":" <+> stringS id_ <+> "=" <+> expr_ False e <> ";"
conds ((Just c ,e):es') =
mask' <+> ":" <+> stringS id_ <+> "=" <+> expr_ False e <> ";" <> line <> conds es'
where
mask' = patLitCustom scrutTy c
inst_ :: Declaration -> VerilogM (Maybe Doc)
inst_ (TickDecl {}) = return Nothing
inst_ (Assignment id_ e) = fmap Just $
"assign" <+> stringS id_ <+> equals <+> expr_ False e <> semi
inst_ (CondAssignment id_ _ scrut _ [(Just (BoolLit b), l),(_,r)]) = fmap Just $
"always @(*) begin" <> line <>
indent 2 ("if" <> parens (expr_ True scrut) <> line <>
(indent 2 $ stringS id_ <+> equals <+> expr_ False t <> semi) <> line <>
"else" <> line <>
(indent 2 $ stringS id_ <+> equals <+> expr_ False f <> semi)) <> line <>
"end"
where
(t,f) = if b then (l,r) else (r,l)
inst_ (CondAssignment id_ _ scrut scrutTy@(CustomSP {}) es) =
inst_' id_ scrut scrutTy es
inst_ (CondAssignment id_ _ scrut scrutTy@(CustomSum {}) es) =
inst_' id_ scrut scrutTy es
inst_ (CondAssignment id_ _ scrut scrutTy@(CustomProduct {}) es) =
inst_' id_ scrut scrutTy es
inst_ (CondAssignment id_ _ scrut scrutTy es) = fmap Just $
"always @(*) begin" <> line <>
indent 2 ("case" <> parens (expr_ True scrut) <> line <>
(indent 2 $ vcat $ punctuate semi (conds id_ es)) <> semi <> line <>
"endcase") <> line <>
"end"
where
conds :: Identifier -> [(Maybe Literal,Expr)] -> VerilogM [Doc]
conds _ [] = return []
conds i [(_,e)] = ("default" <+> colon <+> stringS i <+> equals <+> expr_ False e) <:> return []
conds i ((Nothing,e):_) = ("default" <+> colon <+> stringS i <+> equals <+> expr_ False e) <:> return []
conds i ((Just c ,e):es') = (exprLitV (Just (scrutTy,conSize scrutTy)) c <+> colon <+> stringS i <+> equals <+> expr_ False e) <:> conds i es'
inst_ (InstDecl _ _ nm lbl ps pms) = fmap Just $
nest 2 (stringS nm <> params <> stringS lbl <> line <> pms' <> semi)
where
pms' = tupled $ sequence [dot <> expr_ False i <+> parens (expr_ False e) | (i,_,_,e) <- pms]
params
| null ps = space
| otherwise = line <> "#" <> tupled (sequence [dot <> expr_ False i <+> parens (expr_ False e) | (i,_,e) <- ps]) <> line
inst_ (BlackBoxD _ libs imps inc bs bbCtx) =
fmap Just (Mon (column (renderBlackBox libs imps inc bs bbCtx)))
inst_ (Seq ds) = Just <$> seqs ds
inst_ (NetDecl' {}) = return Nothing
seq_ :: Seq -> VerilogM Doc
seq_ (AlwaysClocked edge clk ds) =
"always @" <>
parens (case edge of {Rising -> "posedge"; _ -> "negedge"} <+>
expr_ False clk) <+> "begin" <> line <>
indent 2 (seqs ds) <> line <>
"end"
seq_ (Initial ds) =
"initial begin" <> line <>
indent 2 (seqs ds) <> line <>
"end"
seq_ (AlwaysComb ds) =
"always @* begin" <> line <>
indent 2 (seqs ds) <> line <>
"end"
seq_ (Branch scrut scrutTy es) =
"case" <> parens (expr_ True scrut) <> line <>
(indent 2 $ vcat $ conds es) <> line <>
"endcase"
where
conds :: [(Maybe Literal,[Seq])] -> VerilogM [Doc]
conds [] =
return []
conds [(_,sq)] =
("default" <+> colon <+> "begin" <> line <>
indent 2 (seqs sq) <> line <>
"end") <:> return []
conds ((Nothing,sq):_) =
("default" <+> colon <+> "begin" <> line <>
indent 2 (seqs sq) <> line <>
"end") <:> return []
conds ((Just c ,sq):es') =
(exprLitV (Just (scrutTy,conSize scrutTy)) c <+> colon <+> "begin" <> line <>
indent 2 (seqs sq) <> line <>
"end") <:> conds es'
seq_ (SeqDecl sd) = case sd of
Assignment id_ e ->
stringS id_ <+> equals <+> expr_ False e <> semi
BlackBoxD {} ->
fromMaybe <$> emptyDoc <*> inst_ sd
Seq ds ->
seqs ds
_ -> error ("seq_: " ++ show sd)
seqs :: [Seq] -> VerilogM Doc
seqs [] = emptyDoc
seqs (SeqDecl (TickDecl id_):ds) = "//" <+> stringS id_ <> line <> seqs ds
seqs (d:ds) = seq_ d <> line <> line <> seqs ds
data Range
= Contiguous Int Int
| Split [(Int,Int,Provenance)]
data Provenance
= Provenance Int Int
inRange
:: [(Int,Int)]
-> (Int,Int,Provenance)
-> ([(Int,Int)],[(Int,Int,Provenance)])
inRange [] _ = ([],[])
inRange ((start,end):ses) orig@(_,endRange,Provenance _ endProvenance) =
let startOffset = start - endProvenance
endOffset = end - endProvenance
in
if startOffset >= 0 then
let startRangeNew = endRange + startOffset
endRangeNew =
if endOffset >= 0 then
endRange + endOffset
else
endRange
startProvenanceNew = start - end
endProvenanceNew =
if endOffset >= 0 then
0
else
startProvenanceNew - startOffset
newSplitRange =
( startRangeNew
, endRangeNew
, Provenance startProvenanceNew endProvenanceNew)
in
if endOffset >= 0 then
second (newSplitRange:) (inRange ses orig)
else
((endProvenance-1,end):ses,[newSplitRange])
else
((start,end):ses,[])
buildSplitRange
:: Int
-> Int
-> (Int,Int)
-> (Int,(Int,Int,Provenance))
buildSplitRange offset eP (s,e) =
let d = s-e in
(eP+d+1,(s + offset, e + offset, Provenance (eP+d) eP))
continueWithRange
:: [(Int,Int)]
-> HWType
-> Range
-> (Range, HWType)
continueWithRange ses hty r = case r of
Contiguous _ offset -> case ses of
[(start,end)] ->
(Contiguous (start+offset) (end+offset), hty)
ses1 ->
let ses2 = snd (mapAccumR (buildSplitRange offset) 0 ses1) in
(Split ses2, hty)
Split rs -> case concat (snd (mapAccumL inRange ses rs)) of
[] -> error "internal error"
[(s1,e1,_)] -> (Contiguous s1 e1,hty)
rs1 -> (Split rs1,hty)
modifier
:: HasCallStack
=> Range
-> Modifier
-> Maybe (Range,HWType)
modifier r (Sliced (BitVector _,start,end)) =
Just (continueWithRange [(start,end)] hty r)
where
hty = BitVector (start-end-1)
modifier r (Indexed (ty@(SP _ args),dcI,fI)) =
Just (continueWithRange [(start,end)] argTy r)
where
argTys = snd $ args !! dcI
argTy = argTys !! fI
argSize = typeSize argTy
other = otherSize argTys (fI-1)
start = typeSize ty - 1 - conSize ty - other
end = start - argSize + 1
modifier r (Indexed (ty@(Product _ _ argTys),_,fI)) =
Just (continueWithRange [(start,end)] argTy r)
where
argTy = argTys !! fI
argSize = typeSize argTy
otherSz = otherSize argTys (fI - 1)
start = typeSize ty - 1 - otherSz
end = start - argSize + 1
modifier r (Indexed (ty@(Vector _ argTy),1,0)) =
Just (continueWithRange [(start,end)] argTy r)
where
argSize = typeSize argTy
start = typeSize ty - 1
end = start - argSize + 1
modifier r (Indexed (ty@(Vector n argTy),1,1)) =
Just (continueWithRange [(start,0)] hty r)
where
argSize = typeSize argTy
start = typeSize ty - argSize - 1
hty = Vector (n-1) argTy
modifier r (Indexed (ty@(RTree 0 argTy),0,0)) =
Just (continueWithRange [(start,0)] argTy r)
where
start = typeSize ty - 1
modifier r (Indexed (ty@(RTree d argTy),1,0)) =
Just (continueWithRange [(start,end)] hty r)
where
start = typeSize ty - 1
end = typeSize ty `div` 2
hty = RTree (d-1) argTy
modifier r (Indexed (ty@(RTree d argTy),1,1)) =
Just (continueWithRange [(start,0)] hty r)
where
start = (typeSize ty `div` 2) - 1
hty = RTree (d-1) argTy
modifier r (Indexed (ty@(Vector _ argTy),10,fI)) =
Just (continueWithRange [(start,end)] argTy r)
where
argSize = typeSize argTy
start = typeSize ty - (fI * argSize) - 1
end = start - argSize + 1
modifier r (Indexed (ty@(RTree _ argTy),10,fI)) =
Just (continueWithRange [(start,end)] argTy r)
where
argSize = typeSize argTy
start = typeSize ty - (fI * argSize) - 1
end = start - argSize + 1
modifier r (Indexed (CustomSP _typName _dataRepr _size args,dcI,fI)) =
Just (continueWithRange ses argTy r)
where
ses = bitRanges (anns !! fI)
(ConstrRepr' _name _n _mask _value anns, _, argTys) = args !! dcI
argTy = argTys !! fI
modifier r (Indexed (CustomProduct _typName dataRepr _size _maybeFieldNames args,_,fI))
| DataRepr' _typ _size [cRepr] <- dataRepr
, ConstrRepr' _cName _pos _mask _val fieldAnns <- cRepr
= let ses = bitRanges (fieldAnns !! fI) in Just (continueWithRange ses argTy r)
where
argTy = map snd args !! fI
modifier r (DC (ty@(SP _ _),_)) =
Just (continueWithRange [(start,end)] ty r)
where
start = typeSize ty - 1
end = typeSize ty - conSize ty
modifier r (Nested m1 m2) = do
case modifier r m1 of
Nothing -> modifier r m2
Just (r1,argTy) -> case modifier r1 m2 of
Nothing -> Just (r1,argTy)
m -> m
modifier _ _ = Nothing
customReprDataCon
:: DataRepr'
-> ConstrRepr'
-> [(HWType, Expr)]
-> VerilogM Doc
customReprDataCon dataRepr constrRepr [] =
let origins = bitOrigins dataRepr constrRepr :: [BitOrigin] in
case origins of
[Lit (bitsToBits -> ns)] ->
int (length ns) <> squote <> "b" <> hcat (mapM (bit_char undefValue) ns)
_ -> error "internal error"
customReprDataCon dataRepr constrRepr args = do
funId <- mkConstrFunction
Mon (imports %= HashSet.insert (Text.pack (TextS.unpack funId ++ ".inc")))
stringS funId <> tupled (mapM (expr_ False . snd) nzArgs)
where
nzArgs = filter ((/=0) . typeSize . fst) args
mkConstrFunction = makeCached (crName constrRepr) customConstrs $ do
let size = drSize dataRepr
aTys = map fst args
origins = bitOrigins dataRepr constrRepr :: [BitOrigin]
mkId <- Mon mkIdentifier
let ids = [ mkId Basic (TextS.pack ('v':show n)) | n <- [1..length args] ]
fId = mkId Basic (crName constrRepr)
let fInps =
[ case typeSize t of
0 -> emptyDoc
1 -> "input" <+> stringS i <> semi <> line
n -> "input" <+> brackets (int (n-1) <> colon <> int 0) <+>
stringS i <> semi <> line
| (i,t) <- zip ids aTys
]
let range' (Lit (bitsToBits -> ns)) =
int (length ns) <> squote <> "b" <> hcat (mapM (bit_char undefValue) ns)
range' (Field n start end) =
let v = ids !! n
aTy = aTys !! n
in case typeSize aTy of
0 -> error "internal error"
1 -> if start == 0 && end == 0 then
stringS v
else
error "internal error"
_ -> stringS v <> brackets (int start <> colon <> int end)
let val = case origins of
[] -> error "internal error"
[r] -> range' r
rs -> listBraces (mapM range' rs)
let oSz = case size of
0 -> error "internal error"
1 -> emptyDoc
n -> brackets (int (n-1) <> colon <> int 0)
funDoc <-
"function" <+> oSz <+> stringS fId <> semi <> line <>
hcat (sequence fInps) <>
"begin" <> line <>
indent 2 (stringS fId <+> "=" <+> val <> semi) <> line <>
"end" <> line <>
"endfunction"
Mon (includes %= ((TextS.unpack fId ++ ".inc",funDoc):))
return fId
expr_ :: Bool
-> Expr
-> VerilogM Doc
expr_ _ (Literal sizeM lit) = exprLitV sizeM lit
expr_ _ (Identifier id_ Nothing) = stringS id_
expr_ _ (Identifier id_ (Just (Indexed (CustomSP _id dataRepr _size args,dcI,fI)))) =
case fieldTy of
Void {} -> error (unexpectedProjectionErrorMsg dataRepr dcI fI)
_ -> braces $ hcat $ punctuate ", " $ sequence ranges
where
(ConstrRepr' _name _n _mask _value anns, _, fieldTypes) = args !! dcI
ranges = map range' $ bitRanges (anns !! fI)
range' (start, end) = stringS id_ <> brackets (int start <> ":" <> int end)
fieldTy = indexNote ($(curLoc) ++ "panic") fieldTypes fI
expr_ _ (Identifier d_ (Just (Indexed (CustomProduct _id dataRepr _size _maybeFieldNames tys, dcI, fI))))
| DataRepr' _typ _size [cRepr] <- dataRepr
, ConstrRepr' _cName _pos _mask _val anns <- cRepr =
let ranges = map range' (bitRanges (anns !! fI)) in
case fieldTy of
Void {} -> error (unexpectedProjectionErrorMsg dataRepr dcI fI)
_ -> braces $ hcat $ punctuate ", " $ sequence ranges
where
(_fieldAnn, fieldTy) = indexNote ($(curLoc) ++ "panic") tys fI
range' (start, end) = stringS d_ <> brackets (int start <> ":" <> int end)
expr_ _ (Identifier id_ (Just (Indexed ((Signed w),_,_)))) = do
iw <- Mon $ use intWidth
traceIf (iw < w) ($(curLoc) ++ "WARNING: result smaller than argument") $
stringS id_
expr_ _ (Identifier id_ (Just (Indexed ((Unsigned w),_,_)))) = do
iw <- Mon $ use intWidth
traceIf (iw < w) ($(curLoc) ++ "WARNING: result smaller than argument") $
stringS id_
expr_ _ (Identifier _ (Just (Indexed ((BitVector _),_,0)))) = do
iw <- Mon $ use intWidth
traceIf True ($(curLoc) ++ "WARNING: synthesizing bitvector mask to dontcare") $
verilogTypeErrValue (Signed iw)
expr_ _ (Identifier id_ (Just (Indexed ((BitVector w),_,1)))) = do
iw <- Mon $ use intWidth
traceIf (iw < w) ($(curLoc) ++ "WARNING: result smaller than argument") $
stringS id_
expr_ _ (Identifier id_ (Just m)) = case modifier (Contiguous 0 0) m of
Nothing -> stringS id_
Just (Contiguous start end,resTy) -> case resTy of
Signed _ -> "$signed" <> parens (slice start end)
_ -> slice start end
Just (Split rs,resTy) ->
let rs1 = listBraces (mapM (\(start,end,_) -> slice start end) rs) in
case resTy of
Signed _ -> "$signed" <> parens rs1
_ -> rs1
where
slice s e = stringS id_ <> brackets (int s <> colon <> int e)
expr_ b (DataCon _ (DC (Void {}, -1)) [e]) = expr_ b e
expr_ _ (DataCon ty@(Vector 0 _) _ _) = verilogTypeErrValue ty
expr_ _ (DataCon (Vector 1 _) _ [e]) = expr_ False e
expr_ _ e@(DataCon (Vector _ _) _ es@[_,_]) =
case vectorChain e of
Just es' -> listBraces (mapM (expr_ False) es')
Nothing -> listBraces (mapM (expr_ False) es)
expr_ _ (DataCon (RTree 0 _) _ [e]) = expr_ False e
expr_ _ e@(DataCon (RTree _ _) _ es@[_,_]) =
case rtreeChain e of
Just es' -> listBraces (mapM (expr_ False) es')
Nothing -> listBraces (mapM (expr_ False) es)
expr_ _ (DataCon (SP {}) (DC (BitVector _,_)) es) = assignExpr
where
argExprs = map (expr_ False) es
assignExpr = braces (hcat $ punctuate comma $ sequence argExprs)
expr_ _ (DataCon ty@(SP _ args) (DC (_,i)) es) = assignExpr
where
argTys = snd $ args !! i
dcSize = conSize ty + sum (map typeSize argTys)
dcExpr = expr_ False (dcToExpr ty i)
argExprs = map (expr_ False) es
extraArg = case typeSize ty - dcSize of
0 -> []
n -> [int n <> "'b" <> bits undefValue (replicate n U)]
assignExpr = braces (hcat $ punctuate comma $ sequence (dcExpr:argExprs ++ extraArg))
expr_ _ (DataCon ty@(Sum _ _) (DC (_,i)) []) = int (typeSize ty) <> "'d" <> int i
expr_ _ (DataCon ty@(CustomSum _ _ _ tys) (DC (_,i)) []) =
let (ConstrRepr' _ _ _ value _) = fst $ tys !! i in
int (typeSize ty) <> squote <> "d" <> int (fromIntegral value)
expr_ _ (DataCon (CustomSP _name dataRepr _size constrs) (DC (_,constrNr)) es) =
let (cRepr, _, argTys) = constrs !! constrNr in
customReprDataCon dataRepr cRepr (zip argTys es)
expr_ _ (DataCon (CustomProduct _ dataRepr _size _labels tys) _ es) |
DataRepr' _typ _size [cRepr] <- dataRepr =
customReprDataCon dataRepr cRepr (zip (map snd tys) es)
expr_ _ (DataCon (Product {}) _ es) = listBraces (mapM (expr_ False) es)
expr_ _ (BlackBoxE pNm _ _ _ _ bbCtx _)
| pNm == "Clash.Sized.Internal.Signed.fromInteger#"
, [Literal _ (NumLit n), Literal _ i] <- extractLiterals bbCtx
= exprLitV (Just (Signed (fromInteger n),fromInteger n)) i
expr_ _ (BlackBoxE pNm _ _ _ _ bbCtx _)
| pNm == "Clash.Sized.Internal.Unsigned.fromInteger#"
, [Literal _ (NumLit n), Literal _ i] <- extractLiterals bbCtx
= exprLitV (Just (Unsigned (fromInteger n),fromInteger n)) i
expr_ _ (BlackBoxE pNm _ _ _ _ bbCtx _)
| pNm == "Clash.Sized.Internal.BitVector.fromInteger#"
, [Literal _ (NumLit n), Literal _ m, Literal _ i] <- extractLiterals bbCtx
= let NumLit m' = m
NumLit i' = i
in exprLitV (Just (BitVector (fromInteger n),fromInteger n)) (BitVecLit m' i')
expr_ _ (BlackBoxE pNm _ _ _ _ bbCtx _)
| pNm == "Clash.Sized.Internal.BitVector.fromInteger##"
, [Literal _ m, Literal _ i] <- extractLiterals bbCtx
= let NumLit m' = m
NumLit i' = i
in exprLitV (Just (Bit,1)) (BitLit $ toBit m' i')
expr_ _ (BlackBoxE pNm _ _ _ _ bbCtx _)
| pNm == "Clash.Sized.Internal.Index.fromInteger#"
, [Literal _ (NumLit n), Literal _ i] <- extractLiterals bbCtx
= exprLit undefValue (Just (Index (fromInteger n),fromInteger n)) i
expr_ b (BlackBoxE _ libs imps inc bs bbCtx b') = do
parenIf (b || b') (Mon (renderBlackBox libs imps inc bs bbCtx <*> pure 0))
expr_ _ (DataTag Bool (Left id_)) = stringS id_ <> brackets (int 0)
expr_ _ (DataTag Bool (Right id_)) = do
iw <- Mon (use intWidth)
"$unsigned" <> parens (listBraces (sequence [braces (int (iw-1) <+> braces "1'b0"),stringS id_]))
expr_ _ (DataTag (Sum _ _) (Left id_)) = "$unsigned" <> parens (stringS id_)
expr_ _ (DataTag (Sum _ _) (Right id_)) = "$unsigned" <> parens (stringS id_)
expr_ _ (DataTag (Product {}) (Right _)) = do
iw <- Mon (use intWidth)
int iw <> "'sd0"
expr_ _ (DataTag hty@(SP _ _) (Right id_)) = "$unsigned" <> parens
(stringS id_ <> brackets
(int start <> colon <> int end))
where
start = typeSize hty - 1
end = typeSize hty - conSize hty
expr_ _ (DataTag (Vector 0 _) (Right _)) = do
iw <- Mon $ use intWidth
int iw <> "'sd0"
expr_ _ (DataTag (Vector _ _) (Right _)) = do
iw <- Mon $ use intWidth
int iw <> "'sd1"
expr_ _ (DataTag (RTree 0 _) (Right _)) = do
iw <- Mon $ use intWidth
int iw <> "'sd0"
expr_ _ (DataTag (RTree _ _) (Right _)) = do
iw <- Mon $ use intWidth
int iw <> "'sd1"
expr_ b (ConvBV _ _ _ e) = expr_ b e
expr_ b (IfThenElse c t e) =
parenIf b (expr_ True c <+> "?" <+> expr_ True t <+> ":" <+> expr_ True e)
expr_ _ e = error $ $(curLoc) ++ (show e)
otherSize :: [HWType] -> Int -> Int
otherSize _ n | n < 0 = 0
otherSize [] _ = 0
otherSize (a:as) n = typeSize a + otherSize as (n-1)
vectorChain :: Expr -> Maybe [Expr]
vectorChain (DataCon (Vector 0 _) _ _) = Just []
vectorChain (DataCon (Vector 1 _) _ [e]) = Just [e]
vectorChain (DataCon (Vector _ _) _ [e1,e2]) = Just e1 <:> vectorChain e2
vectorChain _ = Nothing
rtreeChain :: Expr -> Maybe [Expr]
rtreeChain (DataCon (RTree 0 _) _ [e]) = Just [e]
rtreeChain (DataCon (RTree _ _) _ [e1,e2]) = Just e1 <:> rtreeChain e2
rtreeChain _ = Nothing
exprLitV :: Maybe (HWType,Size) -> Literal -> VerilogM Doc
exprLitV = exprLit undefValue
exprLit :: Lens' s (Maybe (Maybe Int)) -> Maybe (HWType,Size) -> Literal -> Mon (State s) Doc
exprLit _ Nothing (NumLit i) = integer i
exprLit k (Just (hty,sz)) (NumLit i) = case hty of
Unsigned _
| i < 0 -> string "-" <> int sz <> string "'d" <> integer (abs i)
| otherwise -> int sz <> string "'d" <> integer i
Index _ -> int (typeSize hty) <> string "'d" <> integer i
Signed _
| i < 0 -> string "-" <> int sz <> string "'sd" <> integer (abs i)
| otherwise -> int sz <> string "'sd" <> integer i
_ -> int sz <> string "'b" <> blit
where
blit = bits k (toBits sz i)
exprLit k (Just (_,sz)) (BitVecLit m i) = int sz <> string "'b" <> bvlit
where
bvlit = bits k (toBits' sz m i)
exprLit _ _ (BoolLit t) = string $ if t then "1'b1" else "1'b0"
exprLit k _ (BitLit b) = string "1'b" <> bit_char k b
exprLit _ _ (StringLit s) = string . pack $ show s
exprLit _ _ l = error $ $(curLoc) ++ "exprLit: " ++ show l
toBits :: Integral a => Int -> a -> [Bit]
toBits size val = map (\x -> if odd x then H else L)
$ reverse
$ take size
$ map (`mod` 2)
$ iterate (`div` 2) val
toBits' :: Integral a => Int -> a -> a -> [Bit]
toBits' size msk val = map (\(m,i) -> if odd m then U else (if odd i then H else L))
$
( reverse . take size)
$ zip
( map (`mod` 2) $ iterate (`div` 2) msk)
( map (`mod` 2) $ iterate (`div` 2) val)
bits :: Lens' s (Maybe (Maybe Int)) -> [Bit] -> Mon (State s) Doc
bits k = hcat . traverse (bit_char k)
bit_char' :: Bit -> Char
bit_char' H = '1'
bit_char' L = '0'
bit_char' U = 'x'
bit_char' Z = 'z'
bit_char :: Lens' s (Maybe (Maybe Int)) -> Bit -> Mon (State s) Doc
bit_char k b = do
udf <- Mon (use k)
case (udf,b) of
(Just Nothing,U) -> char '0'
(Just (Just i),U) -> "'" <> int i <> "'"
_ -> char (bit_char' b)
dcToExpr :: HWType -> Int -> Expr
dcToExpr ty i = Literal (Just (ty,conSize ty)) (NumLit (toInteger i))
listBraces :: Monad m => m [Doc] -> m Doc
listBraces = align . encloseSep lbrace rbrace comma
parenIf :: Monad m => Bool -> m Doc -> m Doc
parenIf True = parens
parenIf False = id
punctuate' :: Monad m => Mon m Doc -> Mon m [Doc] -> Mon m Doc
punctuate' s d = vcat (punctuate s d) <> s
encodingNote :: Applicative m => HWType -> m Doc
encodingNote (Clock _) = string " // clock"
encodingNote (Reset _) = string " // reset"
encodingNote _ = emptyDoc