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

namespace ir {

Function::Function(std::string name, std::shared_ptr<Type> ret_type,
                   std::vector<std::shared_ptr<Type>> param_types)
    : Value(std::move(ret_type), std::move(name)) {
  func_type_ = std::static_pointer_cast<FunctionType>(
      Type::GetFunctionType(GetType(), std::move(param_types)));
  entry_ = CreateBlock("entry");
}

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_;
}

void Function::AddParam(Value* param) {
  if (!param) {
    throw std::runtime_error("Function::AddParam cannot add null param");
  }
  params_.push_back(param);
}

std::shared_ptr<FunctionType> Function::GetFunctionType() const {
  return func_type_;
}

}  // namespace ir
