Home | History | Annotate | Line # | Download | only in Edit
      1 //===- FileOffset.h - Offset in a file --------------------------*- C++ -*-===//
      2 //
      3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
      4 // See https://llvm.org/LICENSE.txt for license information.
      5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
      6 //
      7 //===----------------------------------------------------------------------===//
      8 
      9 #ifndef LLVM_CLANG_EDIT_FILEOFFSET_H
     10 #define LLVM_CLANG_EDIT_FILEOFFSET_H
     11 
     12 #include "clang/Basic/SourceLocation.h"
     13 #include <tuple>
     14 
     15 namespace clang {
     16 namespace edit {
     17 
     18 class FileOffset {
     19   FileID FID;
     20   unsigned Offs = 0;
     21 
     22 public:
     23   FileOffset() = default;
     24   FileOffset(FileID fid, unsigned offs) : FID(fid), Offs(offs) {}
     25 
     26   bool isInvalid() const { return FID.isInvalid(); }
     27 
     28   FileID getFID() const { return FID; }
     29   unsigned getOffset() const { return Offs; }
     30 
     31   FileOffset getWithOffset(unsigned offset) const {
     32     FileOffset NewOffs = *this;
     33     NewOffs.Offs += offset;
     34     return NewOffs;
     35   }
     36 
     37   friend bool operator==(FileOffset LHS, FileOffset RHS) {
     38     return LHS.FID == RHS.FID && LHS.Offs == RHS.Offs;
     39   }
     40 
     41   friend bool operator!=(FileOffset LHS, FileOffset RHS) {
     42     return !(LHS == RHS);
     43   }
     44 
     45   friend bool operator<(FileOffset LHS, FileOffset RHS) {
     46     return std::tie(LHS.FID, LHS.Offs) < std::tie(RHS.FID, RHS.Offs);
     47   }
     48 
     49   friend bool operator>(FileOffset LHS, FileOffset RHS) {
     50     return RHS < LHS;
     51   }
     52 
     53   friend bool operator>=(FileOffset LHS, FileOffset RHS) {
     54     return !(LHS < RHS);
     55   }
     56 
     57   friend bool operator<=(FileOffset LHS, FileOffset RHS) {
     58     return !(RHS < LHS);
     59   }
     60 };
     61 
     62 } // namespace edit
     63 } // namespace clang
     64 
     65 #endif // LLVM_CLANG_EDIT_FILEOFFSET_H
     66