#pragma once

//
// This file is distributed under the MIT License. See LICENSE.md for details.
//

#include "revng/ADT/SortedVector.h"
#include "revng/Model/Argument.h"
#include "revng/Model/Type.h"
#include "revng/Model/TypeDefinition.h"

#include "revng/Model/Generated/Early/CABIFunctionDefinition.h"

class model::CABIFunctionDefinition
  : public model::generated::CABIFunctionDefinition {
public:
  using generated::CABIFunctionDefinition::CABIFunctionDefinition;

  Argument &addArgument(UpcastableType &&Type) {
    auto &&[Iterator, Success] = Arguments().emplace(Arguments().size());
    revng_assert(Success);
    Iterator->Type() = std::move(Type);
    return *Iterator;
  }

public:
  llvm::SmallVector<const model::Type *, 4> edges() const {
    llvm::SmallVector<const model::Type *, 4> Result;

    for (const model::Argument &Argument : Arguments())
      if (!Argument.Type().isEmpty())
        Result.push_back(Argument.Type().get());

    if (!ReturnType().isEmpty())
      Result.push_back(ReturnType().get());

    return Result;
  }
  llvm::SmallVector<model::Type *, 4> edges() {
    llvm::SmallVector<model::Type *, 4> Result;

    for (model::Argument &Argument : Arguments())
      if (!Argument.Type().isEmpty())
        Result.push_back(Argument.Type().get());

    if (!ReturnType().isEmpty())
      Result.push_back(ReturnType().get());

    return Result;
  }
};

#include "revng/Model/Generated/Late/CABIFunctionDefinition.h"