astree.h

// Wai Chun Leung
// wleung11
// Shineng Tang
// stang38
// $Id: astree.h,v 1.10 2016-10-06 16:42:35-07 - - $

#ifndef __ASTREE_H__
#define __ASTREE_H__

#include <string>
#include <vector>
#include <bitset>
#include <unordered_map>

using namespace std;


#include "auxlib.h"
// Attributes for types and properties (2.2)
enum class attr {
    VOID, INT, NULLPTR_T, STRING, STRUCT, ARRAY, FUNCTION, VARIABLE,
    FIELD, TYPEID, PARAM, LOCAL, LVAL, CONST, VREG, VADDR, BITSET_SIZE
};
using attr_bitset = bitset<unsigned(attr::BITSET_SIZE)>;

struct location {
   size_t filenr;
   size_t linenr;
   size_t offset;
};

struct astree {

   // Fields.
   int symbol;               // token code
   location lloc;            // source location
   const string* lexinfo;    // pointer to lexical information
   vector<astree*> children; // children of this n-way node
   const string* struct_id;
   const char* type;
   attr_bitset attributes;//add bitset field to astree 

   // Functions.
   astree (int symbol, const location&, const char* lexinfo);
   ~astree();
   astree* adopt (astree* child1, 
                  astree* child2 = nullptr,astree* child3 = nullptr);
   astree* adopt_sym (astree* child, int symbol);
   astree* change_sym(int symbol_);
   void dump_node (FILE*);
   void dump_tree (FILE*, int depth = 0);

   static void dump (FILE* outfile, astree* tree);
   static void print (FILE* outfile, astree* tree, int depth = 0);
};

void destroy (astree* tree1, astree* tree2 = nullptr ,
               astree* tree3 = nullptr,astree* tree4 = nullptr,
               astree* tree5 = nullptr);
bool is_expr(astree* root);
void set_attribute(astree* root);
void errllocprintf (const location&, const char* format, const char*);
const string to_string (attr attribute);

#endif