+ repr_tys = map dataConRepArgTys vect_dcs
+
+vectDataConWorker :: Shape -> TyCon -> TyCon -> DataCon
+ -> DataCon -> DataCon -> [[Type]] -> [[Type]]
+ -> VM ()
+vectDataConWorker shape vect_tc arr_tc arr_dc orig_dc vect_dc pre (dc_tys : post)
+ = do
+ clo <- closedV
+ . inBind orig_worker
+ . polyAbstract tvs $ \abstract ->
+ liftM (abstract . vectorised)
+ $ buildClosures tvs [] dc_tys res_ty (liftM2 (,) mk_vect mk_lift)
+
+ worker <- cloneId mkVectOcc orig_worker (exprType clo)
+ hoistBinding worker clo
+ defGlobalVar orig_worker worker
+ return ()
+ where
+ tvs = tyConTyVars vect_tc
+ arg_tys = mkTyVarTys tvs
+ res_ty = mkTyConApp vect_tc arg_tys
+
+ orig_worker = dataConWorkId orig_dc
+
+ mk_vect = return . mkConApp vect_dc $ map Type arg_tys
+ mk_lift = do
+ len <- newLocalVar FSLIT("n") intPrimTy
+ arr_tys <- mapM mkPArrayType dc_tys
+ args <- mapM (newLocalVar FSLIT("xs")) arr_tys
+ shapes <- shapeReplicate shape
+ (Var len)
+ (mkDataConTag vect_dc)
+
+ empty_pre <- mapM emptyPA (concat pre)
+ empty_post <- mapM emptyPA (concat post)
+
+ return . mkLams (len : args)
+ . wrapFamInstBody arr_tc arg_tys
+ . mkConApp arr_dc
+ $ map Type arg_tys ++ shapes
+ ++ empty_pre
+ ++ map Var args
+ ++ empty_post
+
+buildPADict :: Shape -> TyCon -> TyCon -> TyCon -> Var -> VM CoreExpr
+buildPADict shape vect_tc prepr_tc arr_tc dfun
+ = polyAbstract tvs $ \abstract ->
+ do
+ meth_binds <- mapM (mk_method shape) paMethods
+ let meth_exprs = map (Var . fst) meth_binds
+
+ pa_dc <- builtin paDataCon
+ let dict = mkConApp pa_dc (Type (mkTyConApp vect_tc arg_tys) : meth_exprs)
+ body = Let (Rec meth_binds) dict
+ return . mkInlineMe $ abstract body
+ where
+ tvs = tyConTyVars arr_tc
+ arg_tys = mkTyVarTys tvs
+
+ mk_method shape (name, build)
+ = localV
+ $ do
+ body <- build shape vect_tc prepr_tc arr_tc
+ var <- newLocalVar name (exprType body)
+ return (var, mkInlineMe body)
+
+paMethods = [(FSLIT("lengthPA"), buildLengthPA),
+ (FSLIT("replicatePA"), buildReplicatePA),
+ (FSLIT("toPRepr"), buildToPRepr),
+ (FSLIT("fromPRepr"), buildFromPRepr),
+ (FSLIT("dictPRepr"), buildPRDict)]
+
+buildLengthPA :: Shape -> TyCon -> TyCon -> TyCon -> VM CoreExpr
+buildLengthPA shape vect_tc _ arr_tc
+ = do
+ parr_ty <- mkPArrayType (mkTyConApp vect_tc arg_tys)
+ arg <- newLocalVar FSLIT("xs") parr_ty
+ shapes <- mapM (newLocalVar FSLIT("sh")) shape_tys
+ wilds <- mapM newDummyVar repr_tys
+ let scrut = unwrapFamInstScrut arr_tc arg_tys (Var arg)
+ scrut_ty = exprType scrut
+
+ body <- shapeLength shape (map Var shapes)
+
+ return . Lam arg
+ $ Case scrut (mkWildId scrut_ty) intPrimTy
+ [(DataAlt repr_dc, shapes ++ wilds, body)]
+ where
+ arg_tys = mkTyVarTys $ tyConTyVars arr_tc
+ [repr_dc] = tyConDataCons arr_tc
+
+ shape_tys = shapeReprTys shape
+ repr_tys = drop (length shape_tys) (dataConRepArgTys repr_dc)