OoasTools

/**

  *

  *                  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);

};

}