carbon-lang/explorer/interpreter/builtins.cpp

// Part of the Carbon Language project, under the Apache License v2.0 with LLVM
// Exceptions. See /LICENSE for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

#include "explorer/interpreter/builtins.h"

#include "explorer/common/error_builders.h"

using llvm::dyn_cast;

namespace Carbon {

void Builtins::Register(Nonnull<const Declaration*> decl) {
  if (const auto* interface = dyn_cast<InterfaceDeclaration>(decl)) {
    if (interface->name().is_qualified()) {
      return;
    }

    static std::map<std::string, int, std::less<>>* builtin_indexes = [] {
      std::map<std::string, int, std::less<>> builtin_indexes;
      for (int index = 0; index <= static_cast<int>(Builtin::Last); ++index) {
        builtin_indexes.emplace(BuiltinNames[index], index);
      }
      return new auto(std::move(builtin_indexes));
    }();

    auto it = builtin_indexes->find(interface->name().inner_name());
    if (it != builtin_indexes->end()) {
      builtins_[it->second] = interface;
    }
  }
}

auto Builtins::Get(SourceLocation source_loc, Builtin builtin) const
    -> ErrorOr<Nonnull<const Declaration*>> {
  std::optional<const Declaration*> result =
      builtins_[static_cast<int>(builtin)];
  if (!result.has_value()) {
    return ProgramError(source_loc)
           << "missing declaration for builtin `" << GetName(builtin) << "`";
  }
  return result.value();
}

}  // namespace Carbon