Changes in src/GenPoly/GenPoly.h [c6b4432:c97b448]

File:

• src/GenPoly/GenPoly.h (modified) (3 diffs)

src/GenPoly/GenPoly.h

@@ -23 +23 @@
 #include "AST/Fwd.hpp"            // for ApplicationExpr, BaseInstType, Func...
 #include "SymTab/Mangler.h"       // for Mangler
+#include "SynTree/Declaration.h"  // for TypeDecl::Data, AggregateDecl, Type...
+#include "SynTree/SynTree.h"      // for Visitor Nodes
 
 namespace ast {
@@ -30 +32 @@
 namespace GenPoly {
 
-        struct TypeVarMap : public ErasableScopedMap<ast::TypeEnvKey, ast::TypeData> {
-                TypeVarMap() : ErasableScopedMap( ast::TypeData() ) {}
-        };
+        typedef ErasableScopedMap< std::string, TypeDecl::Data > TyVarMap;
+        using TypeVarMap = ErasableScopedMap< ast::TypeEnvKey, ast::TypeData >;
 
         /// Replaces a TypeInstType by its referrent in the environment, if applicable
+        Type* replaceTypeInst( Type* type, const TypeSubstitution* env );
+        const Type* replaceTypeInst( const Type* type, const TypeSubstitution* env );
         const ast::Type * replaceTypeInst( const ast::Type *, const ast::TypeSubstitution * );
 
         /// returns polymorphic type if is polymorphic type, NULL otherwise; will look up substitution in env if provided
+        Type *isPolyType( Type *type, const TypeSubstitution *env = 0 );
         const ast::Type * isPolyType(const ast::Type * type, const ast::TypeSubstitution * env = nullptr);
 
         /// returns polymorphic type if is polymorphic type in tyVars, NULL otherwise; will look up substitution in env if provided
+        Type *isPolyType( Type *type, const TyVarMap &tyVars, const TypeSubstitution *env = 0 );
         const ast::Type * isPolyType( const ast::Type * type, const TypeVarMap & typeVars, const ast::TypeSubstitution * subst = nullptr );
 
         /// returns dynamic-layout type if is dynamic-layout type in tyVars, NULL otherwise; will look up substitution in env if provided
+        ReferenceToType *isDynType( Type *type, const TyVarMap &tyVars, const TypeSubstitution *env = 0 );
         const ast::BaseInstType *isDynType( const ast::Type * type, const TypeVarMap & typeVars, const ast::TypeSubstitution * subst = 0 );
 
         /// true iff function has dynamic-layout return type under the given type variable map
+        ReferenceToType *isDynRet( FunctionType *function, const TyVarMap &tyVars );
         const ast::BaseInstType *isDynRet( const ast::FunctionType * type, const TypeVarMap & typeVars );
 
         /// true iff function has dynamic-layout return type under the type variable map generated from its forall-parameters
+        ReferenceToType *isDynRet( FunctionType *function );
         const ast::BaseInstType *isDynRet( const ast::FunctionType * func );
 
         /// A function needs an adapter if it returns a dynamic-layout value or if any of its parameters have dynamic-layout type
+        bool needsAdapter( FunctionType *adaptee, const TyVarMap &tyVarr );
         bool needsAdapter( ast::FunctionType const * adaptee, const TypeVarMap & typeVars );
 
+        /// returns polymorphic type if is pointer to polymorphic type, NULL otherwise; will look up substitution in env if provided
+        Type *isPolyPtr( Type *type, const TypeSubstitution *env = 0 );
+
         /// returns polymorphic type if is pointer to polymorphic type in tyVars, NULL otherwise; will look up substitution in env if provided
+        Type *isPolyPtr( Type *type, const TyVarMap &tyVars, const TypeSubstitution *env = 0 );
         const ast::Type * isPolyPtr( const ast::Type * type, const TypeVarMap & typeVars, const ast::TypeSubstitution * env = 0 );
+
+        /// if the base type (after dereferencing N >= 0 pointers) is a polymorphic type, returns the base type, NULL otherwise;
+        /// N will be stored in levels, if provided, will look up substitution in env if provided
+        Type *hasPolyBase( Type *type, int *levels = 0, const TypeSubstitution *env = 0 );
 
         /// if the base type (after dereferencing N >= 0 pointers) is a polymorphic type in tyVars, returns the base type, NULL otherwise;
         /// N will be stored in levels, if provided, will look up substitution in env if provided
+        Type *hasPolyBase( Type *type, const TyVarMap &tyVars, int *levels = 0, const TypeSubstitution *env = 0 );
         const ast::Type * hasPolyBase( const ast::Type * type, const TypeVarMap & typeVars, int * levels = 0, const ast::TypeSubstitution * env = 0 );
 
+        /// true iff this type or some base of this type after dereferencing pointers is either polymorphic or a generic type with at least one
+        /// polymorphic parameter; will look up substitution in env if provided.
+        bool includesPolyType( Type *type, const TypeSubstitution *env = 0 );
+
+        /// true iff this type or some base of this type after dereferencing pointers is either polymorphic in tyVars, or a generic type with
+        /// at least one polymorphic parameter in tyVars; will look up substitution in env if provided.
+        bool includesPolyType( Type *type, const TyVarMap &tyVars, const TypeSubstitution *env = 0 );
+
         /// Returns a pointer to the base FunctionType if ty is the type of a function (or pointer to one), NULL otherwise
+        FunctionType *getFunctionType( Type *ty );
         const ast::FunctionType * getFunctionType( const ast::Type * ty );
 
+        /// If expr (after dereferencing N >= 0 pointers) is a variable expression, returns the variable expression, NULL otherwise;
+        /// N will be stored in levels, if provided
+        VariableExpr *getBaseVar( Expression *expr, int *levels = 0 );
+
         /// true iff types are structurally identical, where TypeInstType's match any type.
+        bool typesPolyCompatible( Type *aty, Type *bty );
         bool typesPolyCompatible( ast::Type const * lhs, ast::Type const * rhs );
 
         /// true if arg requires boxing given exprTyVars
+        bool needsBoxing( Type * param, Type * arg, const TyVarMap &exprTyVars, const TypeSubstitution * env );
         bool needsBoxing( const ast::Type * param, const ast::Type * arg, const TypeVarMap & typeVars, const ast::TypeSubstitution * subst );
 
         /// true if arg requires boxing in the call to appExpr
+        bool needsBoxing( Type * param, Type * arg, ApplicationExpr * appExpr, const TypeSubstitution * env );
         bool needsBoxing( const ast::Type * param, const ast::Type * arg, const ast::ApplicationExpr * expr, const ast::TypeSubstitution * subst );
 
         /// Adds the type variable `tyVar` to `tyVarMap`
+        void addToTyVarMap( TypeDecl * tyVar, TyVarMap &tyVarMap );
         void addToTypeVarMap( const ast::TypeDecl * type, TypeVarMap & typeVars );
         void addToTypeVarMap( const ast::TypeInstType * type, TypeVarMap & typeVars );
 
         /// Adds the declarations in the forall list of type (and its pointed-to type if it's a pointer type) to `tyVarMap`
+        void makeTyVarMap( Type *type, TyVarMap &tyVarMap );
         void makeTypeVarMap( const ast::Type * type, TypeVarMap & typeVars );
         void makeTypeVarMap( const ast::FunctionDecl * decl, TypeVarMap & typeVars );
+
+        /// Prints type variable map
+        void printTyVarMap( std::ostream &os, const TyVarMap &tyVarMap );
+
+        /// Gets the mangled name of this type; alias for SymTab::Mangler::mangleType().
+        inline std::string mangleType( const Type *ty ) { return SymTab::Mangler::mangleType( ty ); }
 
         /// Gets the name of the sizeof parameter for the type, given its mangled name
@@ -92 +134 @@
 
         /// Gets the name of the layout function for a given aggregate type, given its declaration
+        inline std::string layoutofName( AggregateDecl *decl ) { return std::string( "_layoutof_" ) + decl->get_name(); }
         inline std::string layoutofName( ast::AggregateDecl const * decl ) {
                 return std::string( "_layoutof_" ) + decl->name;