Return number of iterations after decoding

2022-11-25 15:39:41 +01:00
parent 066d9d85d2
commit 5fc8435cb4
2 changed files with 5 additions and 3 deletions
--- a/sw/cpp/src/cpp_decoders.cpp
+++ b/sw/cpp/src/cpp_decoders.cpp
@@ -31,7 +31,7 @@ public:
        Eigen::setNbThreads(8);
    }

-    const std::optional<Eigen::RowVectorXi>
+    std::pair<std::optional<Eigen::RowVectorXi>, int>
    decode(const Eigen::Ref<const Eigen::VectorXd>& y) {
        Eigen::RowVectorXd s = Eigen::RowVectorXd::Zero(mH.cols());
        Eigen::RowVectorXi x_hat;
@@ -49,11 +49,11 @@ public:
                     }).cast<int>();

            if (check_parity(x_hat)) {
-                return x_hat;
+                return {x_hat, i + 1};
            }
        }

-        return std::nullopt;
+        return {std::nullopt, mK};
    }

 private: