-vectTyConDecls :: [TyCon] -> VM [TyCon]
-vectTyConDecls tcs = fixV $ \tcs' ->
- do
- mapM_ (uncurry defTyCon) (lazy_zip tcs tcs')
- mapM vectTyConDecl tcs
- where
- lazy_zip [] _ = []
- lazy_zip (x:xs) ~(y:ys) = (x,y) : lazy_zip xs ys
-
-vectTyConDecl :: TyCon -> VM TyCon
-vectTyConDecl tc
- = do
- name' <- cloneName mkVectTyConOcc name
- rhs' <- vectAlgTyConRhs (algTyConRhs tc)
-
- return $ mkAlgTyCon name'
- kind
- tyvars
- [] -- no stupid theta
- rhs'
- [] -- no selector ids
- NoParentTyCon -- FIXME
- rec_flag -- FIXME: is this ok?
- False -- FIXME: no generics
- False -- not GADT syntax
- where
- name = tyConName tc
- kind = tyConKind tc
- tyvars = tyConTyVars tc
- rec_flag = boolToRecFlag (isRecursiveTyCon tc)
-
-vectAlgTyConRhs :: AlgTyConRhs -> VM AlgTyConRhs
-vectAlgTyConRhs (DataTyCon { data_cons = data_cons
- , is_enum = is_enum
- })
- = do
- data_cons' <- mapM vectDataCon data_cons
- zipWithM_ defDataCon data_cons data_cons'
- return $ DataTyCon { data_cons = data_cons'
- , is_enum = is_enum
- }
-
-vectDataCon :: DataCon -> VM DataCon
-vectDataCon dc
- | not . null $ dataConExTyVars dc = pprPanic "vectDataCon: existentials" (ppr dc)
- | not . null $ dataConEqSpec dc = pprPanic "vectDataCon: eq spec" (ppr dc)
- | otherwise
- = do
- name' <- cloneName mkVectDataConOcc name
- tycon' <- vectTyCon tycon
- arg_tys <- mapM vectType rep_arg_tys
- wrk_name <- cloneName mkDataConWorkerOcc name'
-
- let ids = mkDataConIds (panic "vectDataCon: wrapper id")
- wrk_name
- data_con
- data_con = mkDataCon name'
- False -- not infix
- (map (const NotMarkedStrict) arg_tys)
- [] -- no labelled fields
- univ_tvs
- [] -- no existential tvs for now
- [] -- no eq spec for now
- [] -- no theta
- arg_tys
- tycon'
- [] -- no stupid theta
- ids
- return data_con
- where
- name = dataConName dc
- univ_tvs = dataConUnivTyVars dc
- rep_arg_tys = dataConRepArgTys dc
- tycon = dataConTyCon dc
-
-buildPArrayTyCon :: TyCon -> TyCon -> VM TyCon
-buildPArrayTyCon orig_tc vect_tc = fixV $ \repr_tc ->
- do
- name' <- cloneName mkPArrayTyConOcc orig_name
- parent <- buildPArrayParentInfo orig_name vect_tc repr_tc
- rhs <- buildPArrayTyConRhs orig_name vect_tc repr_tc
-
- return $ mkAlgTyCon name'
- kind
- tyvars
- [] -- no stupid theta
- rhs
- [] -- no selector ids
- parent
- rec_flag -- FIXME: is this ok?
- False -- FIXME: no generics
- False -- not GADT syntax
- where
- orig_name = tyConName orig_tc
- name = tyConName vect_tc
- kind = tyConKind vect_tc
- tyvars = tyConTyVars vect_tc
- rec_flag = boolToRecFlag (isRecursiveTyCon vect_tc)
-
-
-buildPArrayParentInfo :: Name -> TyCon -> TyCon -> VM TyConParent
-buildPArrayParentInfo orig_name vect_tc repr_tc
- = do
- parray_tc <- builtin parrayTyCon
- co_name <- cloneName mkInstTyCoOcc (tyConName repr_tc)
-
- let inst_tys = [mkTyConApp vect_tc (map mkTyVarTy tyvars)]
-
- return . FamilyTyCon parray_tc inst_tys
- $ mkFamInstCoercion co_name
- tyvars
- parray_tc
- inst_tys
- repr_tc
- where
- tyvars = tyConTyVars vect_tc
-
-buildPArrayTyConRhs :: Name -> TyCon -> TyCon -> VM AlgTyConRhs
-buildPArrayTyConRhs orig_name vect_tc repr_tc
- = do
- data_con <- buildPArrayDataCon orig_name vect_tc repr_tc
- return $ DataTyCon { data_cons = [data_con], is_enum = False }
-
-buildPArrayDataCon :: Name -> TyCon -> TyCon -> VM DataCon
-buildPArrayDataCon orig_name vect_tc repr_tc
- = do
- dc_name <- cloneName mkPArrayDataConOcc orig_name
- shape_ty <- mkPArrayType intTy -- FIXME: we want to unbox this!
- repr_tys <- mapM mkPArrayType types
- wrk_name <- cloneName mkDataConWorkerOcc dc_name
- wrp_name <- cloneName mkDataConWrapperOcc dc_name
-
- let ids = mkDataConIds wrp_name wrk_name data_con
- data_con = mkDataCon dc_name
- False
- (MarkedStrict : map (const NotMarkedStrict) repr_tys)
- []
- (tyConTyVars vect_tc)
- []
- []
- []
- (shape_ty : repr_tys)
- repr_tc
- []
- ids
-
- return data_con
- where
- types = [ty | dc <- tyConDataCons vect_tc
- , ty <- dataConRepArgTys dc]
-
-buildPAInstance :: TyCon -> TyCon -> VM PAInstance
-buildPAInstance vect_tc arr_tc
- = do
- pa <- builtin paClass
- let inst_ty = mkForAllTys tvs
- . (mkFunTys $ mkPredTys [ClassP pa [ty] | ty <- arg_tys])
- $ mkPredTy (ClassP pa [mkTyConApp vect_tc arg_tys])
-
- dfun <- newExportedVar (mkPADFunOcc $ getOccName vect_tc) inst_ty
-
- return $ PAInstance {
- painstInstance = mkLocalInstance dfun NoOverlap
- , painstVectTyCon = vect_tc
- , painstArrTyCon = arr_tc
- }
- where
- tvs = tyConTyVars arr_tc
- arg_tys = mkTyVarTys tvs
-
-buildPADict :: PAInstance -> VM [(Var, CoreExpr)]
-buildPADict (PAInstance {
- painstInstance = inst
- , painstVectTyCon = vect_tc
- , painstArrTyCon = arr_tc })
- = localV . abstractOverTyVars (tyConTyVars arr_tc) $ \abstract ->
- do
- meth_binds <- mapM mk_method paMethods
- let meth_exprs = map (Var . fst) meth_binds
-
- pa_dc <- builtin paDictDataCon
- let dict = mkConApp pa_dc (Type (mkTyConApp vect_tc arg_tys) : meth_exprs)
- body = Let (Rec meth_binds) dict
- return [(instanceDFunId inst, abstract body)]
- where
- tvs = tyConTyVars arr_tc
- arg_tys = mkTyVarTys tvs
-
- mk_method (name, build)
- = localV
- $ do
- body <- build vect_tc arr_tc
- var <- newLocalVar name (exprType body)
- return (var, mkInlineMe body)
-
-paMethods = [(FSLIT("lengthPA"), buildLengthPA),
- (FSLIT("replicatePA"), buildReplicatePA)]
-
-buildLengthPA :: TyCon -> TyCon -> VM CoreExpr
-buildLengthPA _ arr_tc
- = do
- arg <- newLocalVar FSLIT("xs") arg_ty
- shape <- newLocalVar FSLIT("sel") shape_ty
- body <- lengthPA (Var shape)
- return . Lam arg
- $ Case (Var arg) (mkWildId arg_ty) intPrimTy
- [(DataAlt repr_dc, shape : map mkWildId repr_tys, body)]
- where
- arg_ty = mkTyConApp arr_tc . mkTyVarTys $ tyConTyVars arr_tc
- [repr_dc] = tyConDataCons arr_tc
- shape_ty : repr_tys = dataConRepArgTys repr_dc
-
-
--- data T = C0 t1 ... tm
--- ...
--- Ck u1 ... un
---
--- data [:T:] = A ![:Int:] [:t1:] ... [:un:]
---
--- replicatePA :: Int# -> T -> [:T:]
--- replicatePA n# t
--- = let c = case t of
--- C0 _ ... _ -> 0
--- ...
--- Ck _ ... _ -> k
---
--- xs1 = case t of
--- C0 x1 _ ... _ -> replicatePA @t1 n# x1
--- _ -> emptyPA @t1
---
--- ...
---
--- ysn = case t of
--- Ck _ ... _ yn -> replicatePA @un n# yn
--- _ -> emptyPA @un
--- in
--- A (replicatePA @Int n# c) xs1 ... ysn
---
---
-
-buildReplicatePA :: TyCon -> TyCon -> VM CoreExpr
-buildReplicatePA vect_tc arr_tc
- = do
- len_var <- newLocalVar FSLIT("n") intPrimTy
- val_var <- newLocalVar FSLIT("x") val_ty
-
- let len = Var len_var
- val = Var val_var
-
- shape <- replicatePA len (ctr_num val)
- reprs <- liftM concat $ mapM (mk_comp_arrs len val) vect_dcs
-
- return . mkLams [len_var, val_var]
- $ mkConApp arr_dc (map (Type . TyVarTy) (tyConTyVars arr_tc) ++ (shape : reprs))
- where
- val_ty = mkTyConApp vect_tc . mkTyVarTys $ tyConTyVars arr_tc
- wild = mkWildId val_ty
- vect_dcs = tyConDataCons vect_tc
- [arr_dc] = tyConDataCons arr_tc