source: translator/InitTweak/InitModel.h @ 01aeade

//
// Cforall Version 1.0.0 Copyright (C) 2015 University of Waterloo
//
// The contents of this file are covered under the licence agreement in the
// file "LICENCE" distributed with Cforall.
//
// XXX.cc -- 
//
// Author           : Richard C. Bilson
// Created On       : Mon May 18 07:44:20 2015
// Last Modified By : 
// Last Modified On : 
// Update Count     : 0
//
#ifndef _INITTWEAK_MODEL_H_
#define _INITTWEAK_MODEL_H_

#include "Association.h"
#include "SemanticError.h"
#include "SynTree/Visitor.h"

#include "SynTree/Initializer.h"
#include "SynTree/Declaration.h"
#include "SynTree/Expression.h"
#include "SynTree/Type.h"

namespace InitTweak {
    class InitModelBuilder : public AssociationBuilder, public Visitor {
      public:
        InitModelBuilder( Declaration * );
        ~InitModelBuilder();

        virtual Association *grab_assoc() { taken = true; return building; }
        virtual Association *get_assoc() { return building; }
        void set_assoc( Association *newAssoc ) { building = newAssoc; }

        void init();
        static int interpretDimension( Expression *exp ) {
            ConstantFolder folder( exp );
            try {
                return folder.get_constant();
            } catch (...) {
                throw SemanticError("Invalid array dimension");
            }
        }

        // types
        virtual void visit( ArrayType * );
        virtual void visit( StructInstType * );
        virtual void visit( UnionInstType * );
        virtual void visit( EnumInstType * );
        virtual void visit( ContextInstType * ) { throw 0; }
        virtual void visit( TypeInstType * )    { throw 0; }
        // virtual void visit( TupleType *tupleType );
        // declarations
        virtual void visit( StructDecl *);
        virtual void visit( UnionDecl *);
        virtual void visit( EnumDecl *);
      private:
        class ConstantFolder : public Visitor {
          public:
            ConstantFolder( Expression *_expr = 0 ): expr(_expr) {}
            int get_constant() throw() { expr->accept( *this ); return value; }
            void set_constant( Expression *newExp ) { expr = newExp; }
            // Visitor interface
            void visit( Expression * ) { throw 0; }
            void visit( NameExpr * ) { throw 0; }
            void visit( CastExpr * ) { throw 0; }
            void visit( UntypedMemberExpr * ) { throw 0; }
            void visit( VariableExpr * ) { throw 0; }
            void visit( ConstantExpr * );
            void visit( SizeofExpr * ) { throw 0; }
            void visit( AttrExpr * ) { throw 0; }
            void visit( LogicalExpr * ) { throw 0; }
            void visit( ConditionalExpr * ) { throw 0; }
            void visit( CommaExpr * ) { throw 0; }
          private:
            Expression *expr;
            int value;
        };

        bool taken;
        Declaration *decl;  // ?
        Association *building;
    };

    class InitModelFiller : public AssociationFiller, public Visitor {
      public:
        InitModelFiller( Association *, Initializer *, bool _topLevel = false );
        ~InitModelFiller() { /* pointers in here are not owned by object (never created by object either) */ }
        virtual Association *get_assoc() { return model; }
        virtual void set_assoc( Association *newAssoc ) { model = newAssoc; }

        void init();
        // Visitor interface
        virtual void visit( SingleInit *singleInit );
        virtual void visit( ListInit *listInit );
      private:
        Association *model;
        Initializer *orgInit;
        bool topLevel;
        long int next;
    };

    class InitUnspooler : public AssociationVisitor {
      public:
        InitUnspooler() : init(0), taken( false ) {}
        virtual ~InitUnspooler() { if (! taken && (init != 0)) { delete init; init = 0; } }
        Initializer *get_initializer() { return init; }
        Initializer *grab_initializer() { taken = true; return init; }

        virtual void visit( SingleName * );
        virtual void visit( PointAssociation * );
        virtual void visit( RangeAssociation * ) { std::cerr << "InitUnspooler - In a range assoc" << std::endl; return; }
      private:
        Initializer *init;
        bool taken;
    };

} // namespace InitTweak

#endif // _INITTWEAK_MODEL_H_

/*
  Local Variables:
  mode: c++
  End:
*/
// Local Variables: //
// tab-width: 4 //
// mode: c++ //
// compile-command: "make install" //
// End: //