+ var_tys = mkTyVarTys $ tyConTyVars vect_tc
+ res_ty = mkTyConApp vect_tc var_tys
+
+ cons = map (`mkConApp` map Type var_tys) (tyConDataCons vect_tc)
+ [con] = cons
+
+ from_repr repr@(SumRepr { sum_components = prods
+ , sum_tycon = tycon })
+ expr
+ = do
+ vars <- mapM (newLocalVar (fsLit "x")) (map reprType prods)
+ bodies <- sequence . zipWith3 from_unboxed prods cons
+ $ map Var vars
+ return . Case expr (mkWildId (reprType repr)) res_ty
+ $ zipWith3 sum_alt (tyConDataCons tycon) vars bodies
+ where
+ sum_alt data_con var body = (DataAlt data_con, [var], body)
+
+ from_repr repr@(EnumRepr { enum_data_con = data_con }) expr
+ = do
+ var <- newLocalVar (fsLit "n") intPrimTy
+
+ let res = Case (Var var) (mkWildId intPrimTy) res_ty
+ $ (DEFAULT, [], error_expr)
+ : zipWith mk_alt (tyConDataCons vect_tc) cons
+
+ return $ Case expr (mkWildId (reprType repr)) res_ty
+ [(DataAlt data_con, [var], res)]
+ where
+ mk_alt data_con con = (LitAlt (mkDataConTagLit data_con), [], con)
+
+ error_expr = mkRuntimeErrorApp rUNTIME_ERROR_ID res_ty
+ . showSDoc
+ $ sep [text "Invalid NDP representation of", ppr vect_tc]
+
+ from_repr repr expr = from_unboxed repr con expr
+
+ from_unboxed prod@(ProdRepr { prod_components = tys
+ , prod_data_con = data_con })
+ con
+ expr
+ = do
+ vars <- mapM (newLocalVar (fsLit "y")) tys
+ return $ Case expr (mkWildId (reprType prod)) res_ty
+ [(DataAlt data_con, vars, con `mkVarApps` vars)]
+
+ from_unboxed (IdRepr _) con expr
+ = return $ con `App` expr
+
+ from_unboxed (VoidRepr {}) con _
+ = return con
+
+ from_unboxed _ _ _ = panic "buildFromPRepr/from_unboxed"
+
+buildToArrPRepr :: Repr -> TyCon -> TyCon -> TyCon -> VM CoreExpr
+buildToArrPRepr repr vect_tc prepr_tc arr_tc
+ = do
+ arg_ty <- mkPArrayType el_ty
+ arg <- newLocalVar (fsLit "xs") arg_ty
+
+ res_ty <- mkPArrayType (reprType repr)
+
+ shape_vars <- arrShapeVars repr
+ repr_vars <- arrReprVars repr
+
+ parray_co <- mkBuiltinCo parrayTyCon
+
+ let Just repr_co = tyConFamilyCoercion_maybe prepr_tc
+ co = mkAppCoercion parray_co
+ . mkSymCoercion
+ $ mkTyConApp repr_co var_tys
+
+ scrut = unwrapFamInstScrut arr_tc var_tys (Var arg)
+
+ result <- to_repr shape_vars repr_vars repr
+
+ return . Lam arg
+ . mkCoerce co
+ $ Case scrut (mkWildId (mkTyConApp arr_tc var_tys)) res_ty
+ [(DataAlt arr_dc, shape_vars ++ concat repr_vars, result)]
+ where
+ var_tys = mkTyVarTys $ tyConTyVars vect_tc
+ el_ty = mkTyConApp vect_tc var_tys
+
+ [arr_dc] = tyConDataCons arr_tc
+
+ to_repr shape_vars@(_ : _)
+ repr_vars
+ (SumRepr { sum_components = prods
+ , sum_arr_tycon = tycon
+ , sum_arr_data_con = data_con })
+ = do
+ exprs <- zipWithM to_prod repr_vars prods
+
+ return . wrapFamInstBody tycon tys
+ . mkConApp data_con
+ $ map Type tys ++ map Var shape_vars ++ exprs
+ where
+ tys = map reprType prods
+
+ to_repr [len_var]
+ [repr_vars]
+ (ProdRepr { prod_components = tys
+ , prod_arr_tycon = tycon
+ , prod_arr_data_con = data_con })
+ = return . wrapFamInstBody tycon tys
+ . mkConApp data_con
+ $ map Type tys ++ map Var (len_var : repr_vars)
+
+ to_repr shape_vars
+ _
+ (EnumRepr { enum_arr_tycon = tycon
+ , enum_arr_data_con = data_con })
+ = return . wrapFamInstBody tycon []
+ . mkConApp data_con
+ $ map Var shape_vars
+
+ to_repr _ _ _ = panic "buildToArrPRepr/to_repr"
+
+ to_prod repr_vars@(r : _)
+ (ProdRepr { prod_components = tys@(ty : _)
+ , prod_arr_tycon = tycon
+ , prod_arr_data_con = data_con })
+ = do
+ len <- lengthPA ty (Var r)
+ return . wrapFamInstBody tycon tys
+ . mkConApp data_con
+ $ map Type tys ++ len : map Var repr_vars
+
+ to_prod [var] (IdRepr _) = return (Var var)
+ to_prod [var] (VoidRepr {}) = return (Var var)
+ to_prod _ _ = panic "buildToArrPRepr/to_prod"
+
+
+buildFromArrPRepr :: Repr -> TyCon -> TyCon -> TyCon -> VM CoreExpr
+buildFromArrPRepr repr vect_tc prepr_tc arr_tc
+ = do
+ arg_ty <- mkPArrayType =<< mkPReprType el_ty
+ arg <- newLocalVar (fsLit "xs") arg_ty
+
+ res_ty <- mkPArrayType el_ty
+
+ shape_vars <- arrShapeVars repr
+ repr_vars <- arrReprVars repr
+
+ parray_co <- mkBuiltinCo parrayTyCon
+
+ let Just repr_co = tyConFamilyCoercion_maybe prepr_tc
+ co = mkAppCoercion parray_co
+ $ mkTyConApp repr_co var_tys
+
+ scrut = mkCoerce co (Var arg)
+
+ result = wrapFamInstBody arr_tc var_tys
+ . mkConApp arr_dc
+ $ map Type var_tys ++ map Var (shape_vars ++ concat repr_vars)
+
+ liftM (Lam arg)
+ (from_repr repr scrut shape_vars repr_vars res_ty result)
+ where
+ var_tys = mkTyVarTys $ tyConTyVars vect_tc
+ el_ty = mkTyConApp vect_tc var_tys
+
+ [arr_dc] = tyConDataCons arr_tc
+
+ from_repr (SumRepr { sum_components = prods
+ , sum_arr_tycon = tycon
+ , sum_arr_data_con = data_con })
+ expr
+ shape_vars
+ repr_vars
+ res_ty
+ body
+ = do
+ vars <- mapM (newLocalVar (fsLit "xs")) =<< mapM arrReprType prods
+ result <- go prods repr_vars vars body
+
+ let scrut = unwrapFamInstScrut tycon ty_args expr
+ return . Case scrut (mkWildId scrut_ty) res_ty
+ $ [(DataAlt data_con, shape_vars ++ vars, result)]
+ where
+ ty_args = map reprType prods
+ scrut_ty = mkTyConApp tycon ty_args
+
+ go [] [] [] body = return body
+ go (prod : prods) (repr_vars : rss) (var : vars) body
+ = do
+ shape_vars <- mapM (newLocalVar (fsLit "s")) =<< arrShapeTys prod
+
+ from_prod prod (Var var) shape_vars repr_vars res_ty
+ =<< go prods rss vars body
+ go _ _ _ _ = panic "buildFromArrPRepr/go"
+
+ from_repr repr expr shape_vars [repr_vars] res_ty body
+ = from_prod repr expr shape_vars repr_vars res_ty body
+
+ from_repr _ _ _ _ _ _ = panic "buildFromArrPRepr/from_repr"
+
+ from_prod (ProdRepr { prod_components = tys
+ , prod_arr_tycon = tycon
+ , prod_arr_data_con = data_con })
+ expr
+ shape_vars
+ repr_vars
+ res_ty
+ body
+ = do
+ let scrut = unwrapFamInstScrut tycon tys expr
+ scrut_ty = mkTyConApp tycon tys
+
+ return $ Case scrut (mkWildId scrut_ty) res_ty
+ [(DataAlt data_con, shape_vars ++ repr_vars, body)]
+
+ from_prod (EnumRepr { enum_arr_tycon = tycon
+ , enum_arr_data_con = data_con })
+ expr
+ shape_vars
+ _
+ res_ty
+ body
+ = let scrut = unwrapFamInstScrut tycon [] expr
+ scrut_ty = mkTyConApp tycon []
+ in
+ return $ Case scrut (mkWildId scrut_ty) res_ty
+ [(DataAlt data_con, shape_vars, body)]
+
+ from_prod (IdRepr _)
+ expr
+ _shape_vars
+ [repr_var]
+ _res_ty
+ body
+ = return $ Let (NonRec repr_var expr) body
+
+ from_prod (VoidRepr {})
+ expr
+ _shape_vars
+ [repr_var]
+ _res_ty
+ body
+ = return $ Let (NonRec repr_var expr) body
+
+ from_prod _ _ _ _ _ _ = panic "buildFromArrPRepr/from_prod"
+
+buildPRDictRepr :: Repr -> VM CoreExpr
+buildPRDictRepr (VoidRepr { void_tycon = tycon })
+ = prDFunOfTyCon tycon
+buildPRDictRepr (IdRepr ty) = mkPR ty
+buildPRDictRepr (ProdRepr {
+ prod_components = tys
+ , prod_tycon = tycon
+ })
+ = do
+ prs <- mapM mkPR tys
+ dfun <- prDFunOfTyCon tycon
+ return $ dfun `mkTyApps` tys `mkApps` prs
+
+buildPRDictRepr (SumRepr {
+ sum_components = prods
+ , sum_tycon = tycon })
+ = do
+ prs <- mapM buildPRDictRepr prods
+ dfun <- prDFunOfTyCon tycon
+ return $ dfun `mkTyApps` map reprType prods `mkApps` prs
+
+buildPRDictRepr (EnumRepr { enum_tycon = tycon })
+ = prDFunOfTyCon tycon
+
+buildPRDict :: Repr -> TyCon -> TyCon -> TyCon -> VM CoreExpr
+buildPRDict repr vect_tc prepr_tc _
+ = do
+ dict <- buildPRDictRepr repr
+
+ pr_co <- mkBuiltinCo prTyCon
+ let co = mkAppCoercion pr_co
+ . mkSymCoercion
+ $ mkTyConApp arg_co var_tys
+
+ return $ mkCoerce co dict
+ where
+ var_tys = mkTyVarTys $ tyConTyVars vect_tc
+
+ Just arg_co = tyConFamilyCoercion_maybe prepr_tc
+
+buildPArrayTyCon :: TyCon -> TyCon -> VM TyCon
+buildPArrayTyCon orig_tc vect_tc = fixV $ \repr_tc ->
+ do
+ name' <- cloneName mkPArrayTyConOcc orig_name
+ rhs <- buildPArrayTyConRhs orig_name vect_tc repr_tc
+ parray <- builtin parrayTyCon
+
+ liftDs $ buildAlgTyCon name'
+ tyvars
+ [] -- no stupid theta
+ rhs
+ rec_flag -- FIXME: is this ok?
+ False -- FIXME: no generics
+ False -- not GADT syntax
+ (Just $ mk_fam_inst parray vect_tc)
+ where
+ orig_name = tyConName orig_tc
+ tyvars = tyConTyVars vect_tc
+ rec_flag = boolToRecFlag (isRecursiveTyCon 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
+ repr <- mkRepr vect_tc
+
+ shape_tys <- arrShapeTys repr
+ repr_tys <- arrReprTys repr
+
+ let tys = shape_tys ++ repr_tys
+
+ liftDs $ buildDataCon dc_name
+ False -- not infix
+ (map (const NotMarkedStrict) tys)
+ [] -- no field labels
+ (tyConTyVars vect_tc)
+ [] -- no existentials
+ [] -- no eq spec
+ [] -- no context
+ tys
+ repr_tc
+
+mkPADFun :: TyCon -> VM Var
+mkPADFun vect_tc
+ = newExportedVar (mkPADFunOcc $ getOccName vect_tc) =<< paDFunType vect_tc
+
+buildTyConBindings :: TyCon -> TyCon -> TyCon -> TyCon -> Var
+ -> VM [(Var, CoreExpr)]
+buildTyConBindings orig_tc vect_tc prepr_tc arr_tc dfun
+ = do
+ repr <- mkRepr vect_tc
+ vectDataConWorkers repr orig_tc vect_tc arr_tc
+ dict <- buildPADict repr vect_tc prepr_tc arr_tc dfun
+ binds <- takeHoisted
+ return $ (dfun, dict) : binds
+
+vectDataConWorkers :: Repr -> TyCon -> TyCon -> TyCon
+ -> VM ()
+vectDataConWorkers repr orig_tc vect_tc arr_tc
+ = do
+ bs <- sequence
+ . zipWith3 def_worker (tyConDataCons orig_tc) rep_tys
+ $ zipWith4 mk_data_con (tyConDataCons vect_tc)
+ rep_tys
+ (inits reprs)
+ (tail $ tails reprs)
+ mapM_ (uncurry hoistBinding) bs
+ where
+ tyvars = tyConTyVars vect_tc
+ var_tys = mkTyVarTys tyvars
+ ty_args = map Type var_tys
+
+ res_ty = mkTyConApp vect_tc var_tys
+
+ rep_tys = map dataConRepArgTys $ tyConDataCons vect_tc
+ reprs = splitSumRepr repr
+
+ [arr_dc] = tyConDataCons arr_tc
+
+ mk_data_con con tys pre post
+ = liftM2 (,) (vect_data_con con)
+ (lift_data_con tys pre post (mkDataConTag con))
+
+ vect_data_con con = return $ mkConApp con ty_args
+ lift_data_con tys pre_reprs post_reprs tag
+ = do
+ len <- builtin liftingContext
+ args <- mapM (newLocalVar (fsLit "xs"))
+ =<< mapM mkPArrayType tys
+
+ shape <- replicateShape repr (Var len) tag
+ repr <- mk_arr_repr (Var len) (map Var args)
+
+ pre <- liftM concat $ mapM emptyArrRepr pre_reprs
+ post <- liftM concat $ mapM emptyArrRepr post_reprs
+
+ return . mkLams (len : args)
+ . wrapFamInstBody arr_tc var_tys
+ . mkConApp arr_dc
+ $ ty_args ++ shape ++ pre ++ repr ++ post
+
+ mk_arr_repr len []
+ = do
+ units <- replicatePA len (Var unitDataConId)
+ return [units]
+
+ mk_arr_repr _ arrs = return arrs
+
+ def_worker data_con arg_tys mk_body
+ = do
+ body <- closedV
+ . inBind orig_worker
+ . polyAbstract tyvars $ \abstract ->
+ liftM (abstract . vectorised)
+ $ buildClosures tyvars [] arg_tys res_ty mk_body
+
+ vect_worker <- cloneId mkVectOcc orig_worker (exprType body)
+ defGlobalVar orig_worker vect_worker
+ return (vect_worker, body)
+ where
+ orig_worker = dataConWorkId data_con
+
+buildPADict :: Repr -> TyCon -> TyCon -> TyCon -> Var -> VM CoreExpr
+buildPADict repr vect_tc prepr_tc arr_tc _
+ = polyAbstract tvs $ \abstract ->
+ do
+ meth_binds <- mapM (mk_method repr) 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 repr (name, build)
+ = localV
+ $ do
+ body <- build repr vect_tc prepr_tc arr_tc
+ var <- newLocalVar name (exprType body)
+ return (var, mkInlineMe body)
+
+paMethods :: [(FastString, Repr -> TyCon -> TyCon -> TyCon -> VM CoreExpr)]
+paMethods = [(fsLit "toPRepr", buildToPRepr),
+ (fsLit "fromPRepr", buildFromPRepr),
+ (fsLit "toArrPRepr", buildToArrPRepr),
+ (fsLit "fromArrPRepr", buildFromArrPRepr),
+ (fsLit "dictPRepr", buildPRDict)]