//
// 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.
//
// InitModel.cc --
//
// Author           : Rodolfo G. Esteves
// Created On       : Mon May 18 07:44:20 2015
// Last Modified By : Rob Schluntz
// Last Modified On : Thu Jan 07 13:38:46 2016
// Update Count     : 5
//

#include "SynTree/Constant.h"
#include "InitModel.h"

#include <cassert>
#include <cstdlib>
#include <algorithm>

namespace InitTweak {
	InitModelBuilder::InitModelBuilder( Declaration *_decl )
		: taken( false ), decl( 0 ), building(0) {

		ObjectDecl *_odecl = dynamic_cast< ObjectDecl * >( _decl );
		assert( _odecl != 0 );
		Type *objectType = _odecl->get_type();

		/* this to be replaced by dynamic dispatch */
		if ( dynamic_cast< BasicType * >(objectType) != 0 ) {
			if ( building == 0 ) building = new SingleName;
		} else if ( ReferenceToType *rt = dynamic_cast< ReferenceToType * >(objectType) ) {
			rt->accept( *this );
		} else if ( ArrayType *at = dynamic_cast< ArrayType * >(objectType) ) {
			at->accept( *this );
		} else // if (tuples)
			std::cerr << "Got something else" << std::endl;

		if ( decl != 0 ) init();
	}

	InitModelBuilder::~InitModelBuilder() { if ( ! taken ) { delete building; building = 0; } }

	void InitModelBuilder::init() {
		assert( decl != 0 );
		decl->accept( *this );
	}

	// Visitor interface
	void InitModelBuilder::visit( ArrayType *at ) {
		if ( building == 0 ) building = new RangeAssociation(interpretDimension( at->get_dimension() ));
		decl = 0;
		return;
	}

	void InitModelBuilder::visit( StructInstType *st ) {
		if ( building == 0 ) building = new PointAssociation;
		decl = st->get_baseStruct();
		return;
	}

	void InitModelBuilder::visit( UnionInstType *ut ) {
		decl = ut->get_baseUnion();
		return;
	}
	void InitModelBuilder::visit( EnumInstType * ) {}

	void InitModelBuilder::visit( StructDecl *aggregateDecl) {
		PointAssociation *pa = dynamic_cast< PointAssociation * >( building );
		assert( pa != 0 );
		std::list< Declaration * > mem = aggregateDecl->get_members();

		for ( std::list<Declaration *>::iterator i = mem.begin(); i != mem.end(); i++ ) {
			pa->add_member( (*i)->get_name() );
			InitModelBuilder rec(*i);
			pa->set_member( (*i)->get_name(), rec.grab_assoc() );
		} // for

		return;
	}

	void InitModelBuilder::visit( UnionDecl *) {}
	void InitModelBuilder::visit( EnumDecl *) {}

	// InitModelBuilder::ConstantFolder
	void InitModelBuilder::ConstantFolder::visit( ConstantExpr *cnst ) {
		Constant *c = cnst->get_constant();
		assert (c != 0);
		if ( BasicType *bt = dynamic_cast<BasicType *>( c->get_type() ) ) {
			if ( bt->isInteger() ) {
				// need more intelligence here, not necessarily base 10
				value = std::strtol( c->get_value().c_str(), NULL, 10 );
				return;
			} else
				std::cerr << "Basic type but not integer" << std::endl;
		} // if
		throw 0;
	}

	// InitModelFiller
	InitModelFiller::InitModelFiller( Association *_model, Initializer *_init, bool _topLevel )
		: model( _model ), orgInit( _init ), topLevel( _topLevel ), next( 0 ) {
		//next = model.begin();
		if ( orgInit != 0 ) init();
	}

	void InitModelFiller::init() {
		assert( model != 0 ); // change it into a reference
		assert( orgInit != 0 );
		orgInit->accept( *this );
	}

	void InitModelFiller::visit( SingleInit *singleInit ) {
		std::list< Expression *> &des = singleInit->get_designators();

		if ( topLevel ) {
			assert ( des.empty() );
			assert ( dynamic_cast< SingleName * >(model) != 0 );
			try {
				model->add_single( next++, singleInit->get_value() );
			} catch (...) {
				std::cerr << "Illegal initialization" << std::endl;
			}
			return;
		} // if

		if ( des.empty() ) {
			assert( model != 0 );
			try {
				model->add_single( next++, singleInit->get_value() );
			} catch ( AssocException &e ) {
				throw SemanticError( "Illegal initialization: " + e.get_what() );
			} catch ( ... ) {
				std::cerr << "Shouldn't be here" << std::endl;
			} // try
			return;
		} // if

		// not general enough (does not contend well with designated arrays)
		std::list<std::string> desnames;
		std::transform( des.begin(), des.end(), back_inserter(desnames), Initializer::designator_name );

		for ( std::list<std::string>::iterator i = desnames.begin(); i != desnames.end(); i++ ) {
			try {
				next = model->add_single( *i, singleInit->get_value() );
				next++;
			} catch ( AssocException &e ) {
				throw SemanticError( "Illegal initialization: " + e.get_what() );
			} catch ( ... ) {
				std::cerr << "Shouldn't happen, check association" << std::endl;
			} // try
		} // for

		return;
	}

	void InitModelFiller::visit( ListInit *listInit ) {
		assert( listInit != 0 );

		// designators
		std::list< Expression *> &des = listInit->get_designators();
		std::list< Initializer *> &ini = listInit->get_initializers();

		if ( ! des.empty() ) {
			if (topLevel)
				throw SemanticError( "Invalid initializer: designated at top level." );

			std::list<Expression *> des2;
			std::copy (des.begin(), des.end(), back_inserter( des2 ));
			std::list< Expression * > empty;
			listInit->set_designators( empty );
			for ( std::list<Expression *>::iterator i = des2.begin(); i != des2.end(); i++ ) {
				Association * newModel = 0;
				if ( NameExpr *n = dynamic_cast< NameExpr * >( *i ) )
					try {
						newModel = (*model)[ n->get_name() ];
					} catch( AssocException &e ) {
						std::cerr << "Didn't find member: " << e.get_what() << std::endl;
					}
				else // if ( RangeExpr *r = dynamic_cast< RangeExpr * >( *i ) )
					std::cerr << "Invalid designator specification" << std::endl;

				InitModelFiller rec( newModel, listInit, true );
			} // for
		} else
			if (topLevel) {
				topLevel = false;
				for ( std::list<Initializer*>::iterator i = ini.begin(); i != ini.end(); i++ )
					(*i)->accept(*this);
			} else
				// next available uninitialized member
				InitModelFiller rec( (*model)[next++], listInit, true );
	}

	void InitUnspooler::visit( SingleName *single ) {
		assert(init == 0 && single != 0);
		std::list< Expression * > empty;
		init = new SingleInit( single->get_expr(), empty, false ); // cannot be constructed
		return;
	}

	void InitUnspooler::visit( PointAssociation *pa ) {
		assert( pa != 0 );

		std::list< Initializer * > contents;
		for ( int i = 0; i < pa->size(); i++ )
			if ( (*pa)[i] != 0 ) {
				InitUnspooler rec;
				(*pa)[i]->accept( rec );
				assert( rec.get_initializer() != 0 );
				contents.push_back( rec.grab_initializer() );
			} // if

		std::list< Expression * > desig;
		init = new ListInit( contents, desig, false ); // cannot be constructed
		return;
	}
} // namespace InitTweak

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