Function.h

Go to the documentation of this file.

// Copyright (c) 2018 the Talvos developers. All rights reserved.
//
// This file is distributed under a three-clause BSD license. For full license
// terms please see the LICENSE file distributed with this source code.


#ifndef TALVOS_FUNCTION_H
#define TALVOS_FUNCTION_H

#include <map>
#include <memory>
#include <vector>

namespace talvos
{

class Block;
class Type;

class Function
{
public:
  Function(uint32_t Id, const Type *FunctionType);

  // Do not allow Function objects to be copied.
  Function(const Function &) = delete;
  Function &operator=(const Function &) = delete;

  void addBlock(std::unique_ptr<Block> B);

  void addParam(uint32_t Id) { Parameters.push_back(Id); }

  const Block *getBlock(uint32_t Id) const { return Blocks.at(Id).get(); }

  const Block *getFirstBlock() const { return Blocks.at(FirstBlockId).get(); }

  uint32_t getFirstBlockId() const { return FirstBlockId; }

  uint32_t getId() const { return Id; }

  uint32_t getParamId(uint32_t I) const { return Parameters[I]; }

  size_t getNumParams() const { return Parameters.size(); }

  void setFirstBlock(uint32_t Id) { FirstBlockId = Id; }

private:
  typedef std::map<uint32_t, std::unique_ptr<Block>> BlockMap;

  uint32_t Id;              
  const Type *FunctionType; 
  uint32_t FirstBlockId;    
  BlockMap Blocks;          

  std::vector<uint32_t> Parameters; 
};

} // namespace talvos

#endif