// IR Function：
// - 保存参数列表、基本块列表
// - 记录函数属性/元信息（按需要扩展）
#include "ir/IR.h"

namespace ir {

Function::Function(std::string name, std::shared_ptr<Type> ret_type)
    : Value(std::move(ret_type), std::move(name)) {}

BasicBlock* Function::CreateBlock(const std::string& name) {
  auto block = std::make_unique<BasicBlock>(name);
  auto* ptr = block.get();
  ptr->SetParent(this);
  blocks_.push_back(std::move(block));
  if (!entry_) {
    entry_ = ptr;
  }
  return ptr;
}

BasicBlock* Function::GetEntry() { return entry_; }

const BasicBlock* Function::GetEntry() const { return entry_; }

const std::vector<std::unique_ptr<BasicBlock>>& Function::GetBlocks() const {
  return blocks_;
}

Argument* Function::AddArgument(std::shared_ptr<Type> ty, std::string name) {
  auto arg = std::make_unique<Argument>(std::move(ty), std::move(name), this, args_.size());
  auto* ptr = arg.get();
  args_.push_back(std::move(arg));
  return ptr;
}

const std::vector<std::unique_ptr<Argument>>& Function::GetArgs() const {
  return args_;
}

}  // namespace ir