diff options
author | Alexey Bykov <[email protected]> | 2022-02-10 16:47:16 +0300 |
---|---|---|
committer | Daniil Cherednik <[email protected]> | 2022-02-10 16:47:16 +0300 |
commit | 4cadece7a57ab767e762a0bea1995a596aefeb11 (patch) | |
tree | 7649c16cf4b52e994709f6c9e1716c993ca28759 /library/cpp/codecs/zstd_dict_codec.cpp | |
parent | 143876304996506751ade0b80b3c47f188b9834f (diff) |
Restoring authorship annotation for Alexey Bykov <[email protected]>. Commit 1 of 2.
Diffstat (limited to 'library/cpp/codecs/zstd_dict_codec.cpp')
-rw-r--r-- | library/cpp/codecs/zstd_dict_codec.cpp | 34 |
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 c42a2879e6c..3d15e0a823e 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*/); - } - - bool TZStdDictCodec::DoTryToLearn(ISequenceReader& in) { - Impl = new TImpl(Impl->GetCompressionLevel()); - return Impl->Learn(in, false/*throwOnError*/); + Impl->Learn(in, true/*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); } |