//===--- SourceLoc.cpp - SourceLoc and SourceRange implementations --------===//
//
// This source file is part of the Swift.org open source project
//
// Copyright (c) 2014 - 2015 Apple Inc. and the Swift project authors
// Licensed under Apache License v2.0 with Runtime Library Exception
//
// See http://swift.org/LICENSE.txt for license information
// See http://swift.org/CONTRIBUTORS.txt for the list of Swift project authors
//
//===----------------------------------------------------------------------===//

#include "swift/Basic/SourceLoc.h"
#include "swift/Basic/SourceManager.h"
#include "llvm/Support/MemoryBuffer.h"
#include "llvm/Support/raw_ostream.h"

using namespace swift;

SourceLoc SourceManager::getCodeCompletionLoc() const {
  return getLocForBufferStart(CodeCompletionBufferID)
      .getAdvancedLoc(CodeCompletionOffset);
}

SourceLoc SourceManager::getLocForBufferStart(unsigned BufferID) const {
  auto *Buffer = LLVMSourceMgr.getMemoryBuffer(BufferID);
  return SourceLoc(llvm::SMLoc::getFromPointer(Buffer->getBufferStart()));
}

void SourceLoc::printLineAndColon(raw_ostream &OS,
                                  const SourceManager &SM) const {
  int BufferIndex = SM->FindBufferContainingLoc(Value);
  print(OS, SM, BufferIndex);
}

void SourceLoc::print(raw_ostream &OS, const SourceManager &SM,
                      int &LastBuffer) const {
  if (isInvalid()) {
    OS << "<invalid loc>";
    return;
  }
  int BufferIndex = SM->FindBufferContainingLoc(Value);
  if (BufferIndex == -1) {
    OS << "<malformed loc>";
    return;
  }
  
  if (BufferIndex != LastBuffer) {
    OS << SM->getMemoryBuffer((unsigned)BufferIndex)->getBufferIdentifier();
    LastBuffer = BufferIndex;
  } else {
    OS << "line";
  }

  auto LineAndCol = SM->getLineAndColumn(Value, BufferIndex);

  OS << ':' << LineAndCol.first << ':' << LineAndCol.second;
}

void SourceLoc::dump(const SourceManager &SM) const {
  print(llvm::errs(), SM);
}

void SourceRange::print(raw_ostream &OS, const SourceManager &SM,
                        int &LastBuffer, bool PrintText) const {
  OS << '[';
  Start.print(OS, SM, LastBuffer);
  OS << " - ";
  End.print(OS, SM, LastBuffer);
  OS << ']';
  
  if (Start.isInvalid() || End.isInvalid())
    return;
  
  if (PrintText) {
    OS << " RangeText=\""
       << StringRef(Start.Value.getPointer(),
                    End.Value.getPointer() - Start.Value.getPointer()+1)
       << '"';
  }
}

void SourceRange::dump(const SourceManager &SM) const {
  print(llvm::errs(), SM);
}