-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
GitOrigin-RevId: 78df1cd44c95380cd4af46f4db809ce28876db1f
- Loading branch information
Showing
28 changed files
with
526 additions
and
105 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Binary file not shown.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,84 @@ | ||
| // | ||
| // Copyright Aliaksei Levin ([email protected]), Arseny Smirnov ([email protected]) 2014-2018 | ||
| // | ||
| // Distributed under the Boost Software License, Version 1.0. (See accompanying | ||
| // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt) | ||
| // | ||
| #include "td/telegram/files/FileBitmask.h" | ||
| #include "td/utils/misc.h" | ||
| namespace td { | ||
| Bitmask::Bitmask(Decode, Slice data) : data_(zero_one_decode(data)) { | ||
| } | ||
| Bitmask::Bitmask(Ones, int64 count) : data_((count + 7) / 8, '\0') { | ||
| for (int64 i = 0; i < count; i++) { | ||
| set(i); | ||
| } | ||
| } | ||
| std::string Bitmask::encode() const { | ||
| // remove zeroes in the end to make encoding deteministic | ||
| td::Slice data(data_); | ||
| while (!data.empty() && data.back() == 0) { | ||
| data.remove_suffix(1); | ||
| } | ||
| return zero_one_encode(data_); | ||
| } | ||
| Bitmask::ReadySize Bitmask::get_ready_size(int64 offset, int64 part_size) const { | ||
| ReadySize res; | ||
| res.offset = offset; | ||
| auto offset_part = offset / part_size; | ||
| auto ones = get_ready_parts(offset_part); | ||
| if (ones == 0) { | ||
| res.ready_size = 0; | ||
| } else { | ||
| res.ready_size = (offset_part + ones) * part_size - offset; | ||
| } | ||
| CHECK(res.ready_size >= 0); | ||
| return res; | ||
| } | ||
| int64 Bitmask::get_total_size(int64 part_size) const { | ||
| int64 res = 0; | ||
| for (int64 i = 0; i < size(); i++) { | ||
| res += get(i); | ||
| } | ||
| return res * part_size; | ||
| } | ||
| bool Bitmask::get(int64 offset) const { | ||
| if (offset < 0) { | ||
| return 0; | ||
| } | ||
| if (offset / 8 >= narrow_cast<int64>(data_.size())) { | ||
| return 0; | ||
| } | ||
| return (data_[offset / 8] & (1 << (offset % 8))) != 0; | ||
| } | ||
|
|
||
| int64 Bitmask::get_ready_parts(int64 offset) const { | ||
| int64 res = 0; | ||
| while (get(offset + res)) { | ||
| res++; | ||
| } | ||
| return res; | ||
| }; | ||
|
|
||
| std::vector<int32> Bitmask::as_vector() const { | ||
| std::vector<int32> res; | ||
| for (int32 i = 0; i < narrow_cast<int32>(data_.size() * 8); i++) { | ||
| if (get(i)) { | ||
| res.push_back(i); | ||
| } | ||
| } | ||
| return res; | ||
| } | ||
| void Bitmask::set(int64 offset) { | ||
| auto need_size = narrow_cast<size_t>(offset / 8 + 1); | ||
| if (need_size > data_.size()) { | ||
| data_.resize(need_size, 0); | ||
| } | ||
| data_[need_size - 1] |= (1 << (offset % 8)); | ||
| } | ||
|
|
||
| int64 Bitmask::size() const { | ||
| return data_.size() * 8; | ||
| } | ||
|
|
||
| } // namespace td |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,50 @@ | ||
| // | ||
| // Copyright Aliaksei Levin ([email protected]), Arseny Smirnov ([email protected]) 2014-2018 | ||
| // | ||
| // Distributed under the Boost Software License, Version 1.0. (See accompanying | ||
| // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt) | ||
| // | ||
| #pragma once | ||
| #include "td/utils/common.h" | ||
| #include "td/utils/Slice.h" | ||
| #include "td/utils/StringBuilder.h" | ||
|
|
||
| namespace td { | ||
| class Bitmask { | ||
| public: | ||
| struct ReadySize { | ||
| int64 offset{-1}; | ||
| int64 ready_size{-1}; | ||
| bool empty() const { | ||
| return offset == -1; | ||
| } | ||
| }; | ||
| struct Decode {}; | ||
| struct Ones {}; | ||
| Bitmask() = default; | ||
| Bitmask(Decode, Slice data); | ||
| Bitmask(Ones, int64 count); | ||
| std::string encode() const; | ||
| ReadySize get_ready_size(int64 offset, int64 part_size) const; | ||
| int64 get_total_size(int64 part_size) const; | ||
| bool get(int64 offset) const; | ||
|
|
||
| int64 get_ready_parts(int64 offset) const; | ||
|
|
||
| std::vector<int32> as_vector() const; | ||
| void set(int64 offset); | ||
| int64 size() const; | ||
|
|
||
| private: | ||
| std::string data_; | ||
| }; | ||
|
|
||
| inline StringBuilder &operator<<(StringBuilder &sb, const Bitmask &mask) { | ||
| std::string res; | ||
| for (int64 i = 0; i < mask.size(); i++) { | ||
| res += mask.get(i) ? "1" : "0"; | ||
| } | ||
| return sb << res; | ||
| } | ||
|
|
||
| } // namespace td |
Oops, something went wrong.