#pragma once

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

#include "revng/Model/EnumEntry.h"
#include "revng/Model/TypeDefinition.h"

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

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

public:
  /// The helper for underlying type unwrapping.
  /// Use this when you need to access/modify the existing type,
  /// and \ref Type() when you need to assign a new one.
  model::PrimitiveType &underlyingType() {
    return UnderlyingType()->toPrimitive();
  }

  /// The helper for segment type unwrapping.
  /// Use this when you need to access/modify the existing struct,
  /// and \ref Type() when you need to assign a new one.
  const model::PrimitiveType &underlyingType() const {
    return UnderlyingType()->toPrimitive();
  }

public:
  llvm::SmallVector<const model::Type *, 4> edges() const {
    if (!UnderlyingType().isEmpty())
      return { UnderlyingType().get() };
    else
      return {};
  }
  llvm::SmallVector<model::Type *, 4> edges() {
    if (!UnderlyingType().isEmpty())
      return { UnderlyingType().get() };
    else
      return {};
  }
};

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