/**
  *
  *                  OOAS Compiler - C++ AST
  *
  * Copyright 2015, AIT Austrian Institute of Technology.
  * All rights reserved.
  *
  * SEE THE "LICENSE" FILE FOR THE TERMS UNDER WHICH THIS FILE IS PROVIDED.
  *
  * If you modify the file please update the list of contributors below to in-
  * clude your name. Please also stick to the coding convention of using TABs
  * to do the basic (block-level) indentation and spaces for anything after
  * that. (Enable the display of special chars and it should be pretty obvious
  * what this means.) Also, remove all trailing whitespace.
  *
  * Contributors:
  *               Willibald Krenn (AIT)
  *               Stephan Zimmerer (AIT)
  *               Christoph Czurda (AIT)
  *
  */



#pragma once

#include <CallContext.hpp>
#include <unordered_map>
#include <ast/serialize/gen/Record.pb.h>
#include <ast/serialize/gen/FunctionNames.pb.h>

namespace Ast {

class ProtoBufAstTraversal {
protected:
	CallContext* m_context;
	std::unordered_map<std::uint64_t, void*> m_objects;

	void allocate(const serialize::PBFunctionCall& call);
	void add(const serialize::PBFunctionCall& call);
	void identifier(const serialize::PBFunctionCall& call);
	void type(const serialize::PBFunctionCall& call);
	void statement(const serialize::PBFunctionCall& call);
	void expression(const serialize::PBFunctionCall& call);

	void* doDeserialize(serialize::PBAstTraversal & traversal);

public:
	ProtoBufAstTraversal(CallContext* context);
	~ProtoBufAstTraversal();

	void* deserialize();
	void* deserialize(const std::uint8_t * buffer, std::uint32_t size);
};

}