validate canons without yatest_common

1 files changed, 60 insertions, 0 deletions

diff --git a/library/cpp/erasure/reed_solomon.h b/library/cpp/erasure/reed_solomon.h
new file mode 100644
index 0000000000..a2e268e81d
--- /dev/null
+++ b/library/cpp/erasure/reed_solomon.h
@@ -0,0 +1,60 @@
+#pragma once
+
+#include "helpers.h"
+
+#include <algorithm>
+#include <optional>
+
+namespace NErasure {
+
+template <int DataPartCount, int ParityPartCount, int WordSize, class TCodecTraits>
+class TReedSolomonBase
+    : public ICodec<typename TCodecTraits::TBlobType>
+{
+public:
+    static constexpr ui64 RequiredDataAlignment = alignof(ui64);
+
+    bool CanRepair(const TPartIndexList& erasedIndices) const final {
+        return erasedIndices.size() <= ParityPartCount;
+    }
+
+    bool CanRepair(const TPartIndexSet& erasedIndices) const final {
+        return erasedIndices.count() <= static_cast<size_t>(ParityPartCount);
+    }
+
+    std::optional<TPartIndexList> GetRepairIndices(const TPartIndexList& erasedIndices) const final {
+        if (erasedIndices.empty()) {
+            return TPartIndexList();
+        }
+
+        TPartIndexList indices = erasedIndices;
+        std::sort(indices.begin(), indices.end());
+        indices.erase(std::unique(indices.begin(), indices.end()), indices.end());
+
+        if (indices.size() > static_cast<size_t>(ParityPartCount)) {
+            return std::nullopt;
+        }
+
+        return Difference(0, DataPartCount + ParityPartCount, indices);
+    }
+
+    int GetDataPartCount() const final {
+        return DataPartCount;
+    }
+
+    int GetParityPartCount() const final {
+        return ParityPartCount;
+    }
+
+    int GetGuaranteedRepairablePartCount() const final {
+        return ParityPartCount;
+    }
+
+    int GetWordSize() const final {
+        return WordSize * sizeof(long);
+    }
+
+    virtual ~TReedSolomonBase() = default;
+};
+
+} // namespace NErasure