Line data    Source code

       1             : // Copyright (c) 2023-2024 The Dash Core developers
       2             : // Distributed under the MIT software license, see the accompanying
       3             : // file COPYING or http://www.opensource.org/licenses/mit-license.php.
       4             : 
       5             : #include <util/ranges_set.h>
       6             : 
       7           0 : CRangesSet::Range::Range() : CRangesSet::Range::Range(0, 0) {}
       8             : 
       9     3114368 : CRangesSet::Range::Range(uint64_t begin, uint64_t end) :
      10     1557184 :     begin(begin),
      11     1557184 :     end(end)
      12     1557184 : {
      13     3114368 : }
      14             : 
      15      163305 : bool CRangesSet::Add(uint64_t value)
      16             : {
      17      163305 :     if (Contains(value)) return false;
      18             : 
      19             :     // If element is not in CRangesSet, we need to add a new range [value, value + 1)
      20             :     // But this operation can cause 2 operation of merge (total 3 cases):
      21             :     // - if there're exist ranges [x, value) and [value + 1, y) than
      22             :     //   all 3 of them should be merged in one range [x, y)
      23             :     // - if there's exist a range [x, value) - we need to replace it to new range [x, value + 1)
      24             :     // - if there's exist a range [value + 1, y) - we need to replace it to new range [value, y)
      25      163305 :     Range new_range{value, value + 1};
      26      163305 :     auto it = ranges.lower_bound({value, value});
      27      163305 :     if (it != ranges.begin()) {
      28      163177 :         auto prev = it;
      29      163177 :         --prev;
      30             : 
      31      163177 :         if (prev->end == value) {
      32       57074 :             new_range.begin = prev->begin;
      33       57074 :             it = ranges.erase(prev);
      34       57074 :         }
      35      163177 :     }
      36      163305 :     const auto next = it;
      37      163305 :     if (next != ranges.end()) {
      38      163184 :         if (next->begin == value + 1) {
      39       57790 :             new_range.end = next->end;
      40       57790 :             ranges.erase(next);
      41       57790 :         }
      42      163184 :     }
      43      163305 :     const auto ret = ranges.insert(new_range);
      44      163305 :     assert(ret.second);
      45      163305 :     return true;
      46      163305 : }
      47             : 
      48       98838 : bool CRangesSet::Remove(uint64_t value)
      49             : {
      50       98838 :     if (!Contains(value)) return false;
      51             : 
      52             :     // If element is in CRangesSet, there's possible 2 cases:
      53             :     // - value in the range that is lower-bound (if begin of that range equals to `value`)
      54             :     // - value in the previous range (otherwise)
      55       98838 :     auto it = ranges.lower_bound({value, value});
      56             : 
      57       98838 :     if (it == ranges.end() || it->begin != value) {
      58       41765 :         assert(it != ranges.begin());
      59       41765 :         --it;
      60       41765 :     }
      61             : 
      62       98838 :     const Range current_range = *it;
      63       98838 :     ranges.erase(it);
      64       98838 :     if (current_range.begin != value) {
      65       41765 :         const auto ret = ranges.insert({current_range.begin, value});
      66       41765 :         assert(ret.second);
      67       41765 :     }
      68       98838 :     if (value + 1 != current_range.end) {
      69       41400 :         const auto ret = ranges.insert({value + 1, current_range.end});
      70       41400 :         assert(ret.second);
      71       41400 :     }
      72       98838 :     return true;
      73       98838 : }
      74             : 
      75           0 : bool CRangesSet::IsEmpty() const noexcept
      76             : {
      77           0 :     return ranges.empty();
      78             : }
      79             : 
      80      262154 : size_t CRangesSet::Size() const noexcept
      81             : {
      82      262154 :     size_t result{0};
      83  2509992705 :     for (auto i : ranges) {
      84  2509730551 :         result += i.end - i.begin;
      85             :     }
      86      262154 :     return result;
      87             : }
      88             : 
      89     1048571 : bool CRangesSet::Contains(uint64_t value) const noexcept
      90             : {
      91     1048571 :     const auto it = ranges.lower_bound({value, value});
      92     1048571 :     if (it != ranges.end() && it->begin == value) return true;
      93      771121 :     if (it == ranges.begin()) return false;
      94      770717 :     auto prev = it;
      95      770717 :     --prev;
      96      770717 :     return prev->begin <= value && prev->end > value;
      97     1048571 : }
      98             :