//===----------------------------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#include "llvm/AsmParser/AsmParserContext.h"

namespace llvm {

std::optional<FileLocRange>
AsmParserContext::getFunctionLocation(const Function *F) const {
  if (auto FIt = Functions.find(F); FIt != Functions.end())
    return FIt->second;
  return std::nullopt;
}

std::optional<FileLocRange>
AsmParserContext::getBlockLocation(const BasicBlock *BB) const {
  if (auto BBIt = Blocks.find(BB); BBIt != Blocks.end())
    return BBIt->second;
  return std::nullopt;
}

std::optional<FileLocRange>
AsmParserContext::getInstructionOrArgumentLocation(const Value *IA) const {
  assert(isa<Instruction>(IA) || isa<Argument>(IA));
  if (auto IIt = InstructionsAndArguments.find(IA);
      IIt != InstructionsAndArguments.end())
    return IIt->second;
  return std::nullopt;
}

Function *
AsmParserContext::getFunctionAtLocation(const FileLocRange &Query) const {
  auto It = FunctionsInverse.find(Query.Start);
  if (It.stop() <= Query.End)
    return *It;
  return nullptr;
}

Function *AsmParserContext::getFunctionAtLocation(const FileLoc &Query) const {
  return FunctionsInverse.lookup(Query, nullptr);
}

BasicBlock *
AsmParserContext::getBlockAtLocation(const FileLocRange &Query) const {
  auto It = BlocksInverse.find(Query.Start);
  if (It.stop() <= Query.End)
    return *It;
  return nullptr;
}

BasicBlock *AsmParserContext::getBlockAtLocation(const FileLoc &Query) const {
  return BlocksInverse.lookup(Query, nullptr);
}

Value *AsmParserContext::getInstructionOrArgumentAtLocation(
    const FileLocRange &Query) const {
  auto It = InstructionsAndArgumentsInverse.find(Query.Start);
  if (It.stop() <= Query.End)
    return *It;
  return nullptr;
}

Value *AsmParserContext::getInstructionOrArgumentAtLocation(
    const FileLoc &Query) const {
  return InstructionsAndArgumentsInverse.lookup(Query, nullptr);
}

Value *AsmParserContext::getValueReferencedAtLocation(
    const FileLocRange &Query) const {
  auto It = ReferencedValues.find(Query.Start);
  if (It.stop() <= Query.End)
    return *It;
  return nullptr;
}

Value *
AsmParserContext::getValueReferencedAtLocation(const FileLoc &Query) const {
  return ReferencedValues.lookup(Query, nullptr);
}

bool AsmParserContext::addFunctionLocation(Function *F,
                                           const FileLocRange &Loc) {
  bool Inserted = Functions.insert({F, Loc}).second;
  if (Inserted)
    FunctionsInverse.insert(Loc.Start, Loc.End, F);
  return Inserted;
}

bool AsmParserContext::addBlockLocation(BasicBlock *BB,
                                        const FileLocRange &Loc) {
  bool Inserted = Blocks.insert({BB, Loc}).second;
  if (Inserted)
    BlocksInverse.insert(Loc.Start, Loc.End, BB);
  return Inserted;
}

bool AsmParserContext::addInstructionOrArgumentLocation(
    Value *IA, const FileLocRange &Loc) {
  assert(isa<Instruction>(IA) || isa<Argument>(IA));
  bool Inserted = InstructionsAndArguments.insert({IA, Loc}).second;
  if (Inserted)
    InstructionsAndArgumentsInverse.insert(Loc.Start, Loc.End, IA);
  return Inserted;
}

bool AsmParserContext::addValueReferenceAtLocation(Value *V,
                                                   const FileLocRange &Loc) {
  ReferencedValues.insert(Loc.Start, Loc.End, V);
  return true;
}

} // namespace llvm