summaryrefslogtreecommitdiffstats
path: root/library/cpp/codecs/zstd_dict_codec.cpp
diff options
context:
space:
mode:
authorAlexey Bykov <[email protected]>2022-02-10 16:47:16 +0300
committerDaniil Cherednik <[email protected]>2022-02-10 16:47:16 +0300
commitb50730a77e0c38f2fec0ad5d53fb2034d6470221 (patch)
tree9814fbd1c3effac9b8377c5d604b367b14e2db55 /library/cpp/codecs/zstd_dict_codec.cpp
parent4cadece7a57ab767e762a0bea1995a596aefeb11 (diff)
Restoring authorship annotation for Alexey Bykov <[email protected]>. Commit 2 of 2.
Diffstat (limited to 'library/cpp/codecs/zstd_dict_codec.cpp')
-rw-r--r--library/cpp/codecs/zstd_dict_codec.cpp34
1 files changed, 17 insertions, 17 deletions
diff --git a/library/cpp/codecs/zstd_dict_codec.cpp b/library/cpp/codecs/zstd_dict_codec.cpp
index 3d15e0a823e..c42a2879e6c 100644
--- a/library/cpp/codecs/zstd_dict_codec.cpp
+++ b/library/cpp/codecs/zstd_dict_codec.cpp
@@ -145,7 +145,7 @@ namespace NCodecs {
}
}
- bool Learn(ISequenceReader& in, bool throwOnError) {
+ bool Learn(ISequenceReader& in, bool throwOnError) {
TBuffer data;
TVector<size_t> lens;
@@ -168,19 +168,19 @@ namespace NCodecs {
if (!lens) {
Dict.Reset();
} else {
- size_t trainResult = ZDICT_trainFromBuffer_legacy(
- Dict.data(), Dict.size(), data.Data(), const_cast<const size_t*>(&lens[0]), lens.size(), params);
- if (ZSTD_isError(trainResult)) {
- if (!throwOnError) {
- return false;
- }
- CheckSize(trainResult, __LOCATION__);
- }
- Dict.Resize(trainResult);
+ size_t trainResult = ZDICT_trainFromBuffer_legacy(
+ Dict.data(), Dict.size(), data.Data(), const_cast<const size_t*>(&lens[0]), lens.size(), params);
+ if (ZSTD_isError(trainResult)) {
+ if (!throwOnError) {
+ return false;
+ }
+ CheckSize(trainResult, __LOCATION__);
+ }
+ Dict.Resize(trainResult);
Dict.ShrinkToFit();
}
InitContexts();
- return true;
+ return true;
}
void Save(IOutputStream* out) const {
@@ -247,14 +247,14 @@ namespace NCodecs {
void TZStdDictCodec::DoLearn(ISequenceReader& in) {
Impl = new TImpl(Impl->GetCompressionLevel());
- Impl->Learn(in, true/*throwOnError*/);
+ Impl->Learn(in, true/*throwOnError*/);
+ }
+
+ bool TZStdDictCodec::DoTryToLearn(ISequenceReader& in) {
+ Impl = new TImpl(Impl->GetCompressionLevel());
+ return Impl->Learn(in, false/*throwOnError*/);
}
- bool TZStdDictCodec::DoTryToLearn(ISequenceReader& in) {
- Impl = new TImpl(Impl->GetCompressionLevel());
- return Impl->Learn(in, false/*throwOnError*/);
- }
-
void TZStdDictCodec::Save(IOutputStream* out) const {
Impl->Save(out);
}