libclangmm/src/source_location.h

#ifndef SOURCELOCATION_H_
#define SOURCELOCATION_H_
#include <clang-c/Index.h>
#include <ostream>
#include <string>

namespace clangmm {
  class Offset {
  public:
    bool operator==(const Offset &o) const { return line == o.line && index == o.index; }
    bool operator!=(const Offset &o) const { return !(*this == o); }
    bool operator<(const Offset &o) const { return line < o.line || (line == o.line && index < o.index); }
    bool operator>(const Offset &o) const { return o < *this; }
    bool operator<=(const Offset &o) const { return (*this == o) || (*this < o); }
    bool operator>=(const Offset &o) const { return (*this == o) || (*this > o); }
    unsigned line;
    unsigned index; //byte index in line (not char number)
  };

  class SourceLocation {
    friend class TranslationUnit;
    friend class Diagnostic;
    SourceLocation(CXTranslationUnit &tu, const std::string &filepath, unsigned offset);
    SourceLocation(CXTranslationUnit &tu, const std::string &filepath, unsigned line, unsigned column);

  public:
    SourceLocation(const CXSourceLocation &cx_location) : cx_location(cx_location) {}

  public:
    std::string get_path() const;
    clangmm::Offset get_offset() const;

    friend std::ostream &operator<<(std::ostream &os, const SourceLocation &location) {
      auto offset = location.get_offset();
      os << location.get_path() << ':' << offset.line << ':' << offset.index;
      return os;
    }

    CXSourceLocation cx_location;

  private:
    void get_data(std::string *path, unsigned *line, unsigned *column, unsigned *offset) const;
  };

} // namespace clangmm
#endif // SOURCELOCATION_H_