//
// 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.
//
// TrackLoc.cc --
//
// Author           : Andrew Beach
// Created On       : Tues May 2 15:46:00 2017
// Last Modified By : Andrew Beach
// Last Modified On : Wed May 3 14:43:00 2017
// Update Count     : 0
//

#include "TrackLoc.h"

#include <cstdlib>

#include <iostream>
#include <sstream>
#include <stack>
#include <string>
#include <typeindex>

#include "Common/utility.h"
#include "Common/PassVisitor.h"
#include "Common/VectorMap.h"
#include "GenPoly/GenPoly.h"
#include "Parser/LinkageSpec.h"
#include "SynTree/Declaration.h"
#include "SynTree/Initializer.h"

namespace CodeTools {

	std::ostream & operator<<(std::ostream & out, CodeLocation const & loc) {
		return out << loc.filename << '[' << loc.linenumber << ']';
	}

	class LocationPrinter {
		size_t printLevel;

		CodeLocation *lastNode;

		std::stack< CodeLocation * > parents;
	public:
		LocationPrinter(size_t printLevel) : 
			printLevel(printLevel), lastNode(nullptr)
		{}

		void print( const std::string& name, BaseSyntaxNode *node) {
			for (size_t i = 0 ; i < parents.size() ; ++i) {
				std::cout << "    ";
			}
			if (2 <= printLevel) {
				std::cout << name << '@';
			}
			std::cout << node->location << std::endl;
		}

		void atNode( BaseSyntaxNode *node ) {
			std::string name = std::type_index(typeid(*node)).name();
			if ( node->location.isUnset() ) {
				if ( !parents.empty() ) {
					node->location = *parents.top();
				} 
				else if (nullptr != lastNode) {
					node->location = *lastNode;
				} 
				else {
					std::cerr << "Top level node has no CodeLocation " << name << std::endl;
					exit(EXIT_FAILURE);
				}
			}

			if (0 < printLevel) {
				print( name, node );
			}
			lastNode = &node->location;
		}

		void previsit(BaseSyntaxNode * node) {
			atNode(node);
			parents.push( &node->location );
		}

		void postvisit( __attribute__((unused)) BaseSyntaxNode * node ) {
			parents.pop();
		}

	}; // LocationPrinter

	void fillLocations( std::list< Declaration * > & translationUnit, size_t printLevel) {
		PassVisitor<LocationPrinter> printer(printLevel);
		acceptAll( translationUnit, printer );
	}

} // namespace CodeTools

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