X-Git-Url: https://git.stderr.nl/gitweb?a=blobdiff_plain;f=Flatten.hs;h=4bb6e71b268fd140c7f60df47754f12c9dffa6c8;hb=059c20c7b953a21097939a47ecac7f6cad05541a;hp=11738c70f5803509056662013ee711b6fec93c54;hpb=c0fa1614f8bb0126868658fad79b01df447e113a;p=matthijs%2Fmaster-project%2Fc%CE%BBash.git diff --git a/Flatten.hs b/Flatten.hs index 11738c7..4bb6e71 100644 --- a/Flatten.hs +++ b/Flatten.hs @@ -157,13 +157,18 @@ flattenExpr binds var@(Var id) = Left sig_use -> return ([], sig_use) Right _ -> error "Higher order functions not supported." IdInfo.DataConWorkId datacon -> do - lit <- dataConToLiteral datacon - let ty = CoreUtils.exprType var - sig_id <- genSignalId SigInternal ty - -- Add a name hint to the signal - addNameHint (Name.getOccString id) sig_id - addDef (UncondDef (Right $ Literal lit) sig_id) - return ([], Single sig_id) + if DataCon.isTupleCon datacon && (null $ DataCon.dataConAllTyVars datacon) + then do + -- Empty tuple construction + return ([], Tuple []) + else do + lit <- dataConToLiteral datacon + let ty = CoreUtils.exprType var + sig_id <- genSignalId SigInternal ty + -- Add a name hint to the signal + addNameHint (Name.getOccString id) sig_id + addDef (UncondDef (Right $ Literal lit) sig_id) + return ([], Single sig_id) otherwise -> error $ "Ids other than local vars and dataconstructors not supported: " ++ (showSDoc $ ppr id) @@ -326,25 +331,20 @@ flattenExpr binds expr@(Case scrut b _ alts) = do (args', res') <- flattenMultipleAltCaseExpr binds scrut b (a':alts) case a of (DataAlt datacon, bind_vars, expr) -> do - if isDontCare datacon - then do - -- Completely skip the dontcare cases - return (args', res') - else do - lit <- dataConToLiteral datacon - -- The scrutinee must be a single signal - let Single sig = scrut - -- Create a signal that contains a boolean - boolsigid <- genSignalId SigInternal TysWiredIn.boolTy - addNameHint ("s" ++ show sig ++ "_eq_" ++ lit) boolsigid - let expr = EqLit sig lit - addDef (UncondDef (Right expr) boolsigid) - -- Create conditional assignments of either args/res or - -- args'/res based on boolsigid, and return the result. - -- TODO: It seems this adds the name hint twice? - our_args <- Monad.zipWithM (mkConditionals boolsigid) args args' - our_res <- mkConditionals boolsigid res res' - return (our_args, our_res) + lit <- dataConToLiteral datacon + -- The scrutinee must be a single signal + let Single sig = scrut + -- Create a signal that contains a boolean + boolsigid <- genSignalId SigInternal TysWiredIn.boolTy + addNameHint ("s" ++ show sig ++ "_eq_" ++ lit) boolsigid + let expr = EqLit sig lit + addDef (UncondDef (Right expr) boolsigid) + -- Create conditional assignments of either args/res or + -- args'/res based on boolsigid, and return the result. + -- TODO: It seems this adds the name hint twice? + our_args <- Monad.zipWithM (mkConditionals boolsigid) args args' + our_res <- mkConditionals boolsigid res res' + return (our_args, our_res) otherwise -> error $ "Case patterns other than data constructors not supported in case alternative: " ++ (showSDoc $ ppr a) where @@ -369,20 +369,6 @@ flattenExpr binds expr@(Case scrut b _ alts) = do flattenExpr _ expr = do error $ "Unsupported expression: " ++ (showSDoc $ ppr expr) --- | Is the given data constructor a dontcare? -isDontCare :: DataCon.DataCon -> Bool -isDontCare datacon = - case Name.getOccString tyname of - -- TODO: Do something more robust than string matching - "Bit" -> - Name.getOccString dcname == "DontCare" - otherwise -> - False - where - tycon = DataCon.dataConTyCon datacon - tyname = TyCon.tyConName tycon - dcname = DataCon.dataConName datacon - -- | Translates a dataconstructor without arguments to the corresponding -- literal. dataConToLiteral :: DataCon.DataCon -> FlattenState String @@ -393,7 +379,7 @@ dataConToLiteral datacon = do -- TODO: Do something more robust than string matching "Bit" -> do let dcname = DataCon.dataConName datacon - let lit = case Name.getOccString dcname of "High" -> "'1'"; "Low" -> "'0'"; "DontCare" -> "'-'" + let lit = case Name.getOccString dcname of "High" -> "'1'"; "Low" -> "'0'" return lit "Bool" -> do let dcname = DataCon.dataConName datacon @@ -437,28 +423,4 @@ stateList :: stateList uses signals = Maybe.catMaybes $ Foldable.toList $ zipValueMapsWith filterState signals uses --- | Returns pairs of signals that should be mapped to state in this function. -getOwnStates :: - HsFunction -- | The function to look at - -> FlatFunction -- | The function to look at - -> [(StateId, SignalInfo, SignalInfo)] - -- | The state signals. The first is the state number, the second the - -- signal to assign the current state to, the last is the signal - -- that holds the new state. - -getOwnStates hsfunc flatfunc = - [(old_num, old_info, new_info) - | (old_num, old_info) <- args_states - , (new_num, new_info) <- res_states - , old_num == new_num] - where - sigs = flat_sigs flatfunc - -- Translate args and res to lists of (statenum, sigid) - args = concat $ zipWith stateList (hsFuncArgs hsfunc) (flat_args flatfunc) - res = stateList (hsFuncRes hsfunc) (flat_res flatfunc) - -- Replace the second tuple element with the corresponding SignalInfo - args_states = map (Arrow.second $ signalInfo sigs) args - res_states = map (Arrow.second $ signalInfo sigs) res - - -- vim: set ts=8 sw=2 sts=2 expandtab: