fix sirius operator

ornladios · Oct 29, 2021 · 54ba768 · 54ba768
1 parent d6f80ff
commit 54ba768
Show file tree

Hide file tree

Showing 3 changed files with 19 additions and 0 deletions.
diff --git a/source/adios2/operator/compress/CompressSirius.cpp b/source/adios2/operator/compress/CompressSirius.cpp
@@ -34,6 +34,13 @@ CompressSirius::CompressSirius(const Params &parameters)
     m_TierBuffers.resize(m_Tiers);
 }
 
+CompressSirius::CompressSirius() : Operator("sirius")
+{
+    m_Tiers = 4;
+    m_TierBuffersMap.resize(m_Tiers);
+    m_TierBuffers.resize(m_Tiers);
+}
+
 size_t CompressSirius::Operate(const char *dataIn, const Dims &blockStart,
                                const Dims &blockCount, const DataType varType,
                                char *bufferOut, const Params &params)

diff --git a/source/adios2/operator/compress/CompressSirius.h b/source/adios2/operator/compress/CompressSirius.h
@@ -27,6 +27,8 @@ class CompressSirius : public Operator
 public:
     CompressSirius(const Params &parameters);
 
+    CompressSirius();
+
     ~CompressSirius() = default;
 
     size_t Operate(const char *dataIn, const Dims &blockStart,

diff --git a/source/adios2/operator/compress/CompressorFactory.cpp b/source/adios2/operator/compress/CompressorFactory.cpp
@@ -158,6 +158,16 @@ size_t CompressorFactory::Decompress(const char *bufferIn, const size_t sizeIn,
 #else
         throw std::runtime_error("ERROR: current ADIOS2 library didn't compile "
                                  "with PNG, can't use compressor\n");
+#endif
+    }
+    else if (compressorType == Operator::OperatorType::SIRIUS)
+    {
+#ifdef ADIOS2_HAVE_MHS
+        compress::CompressSirius op;
+        return op.InverseOperate(bufferIn, sizeIn, dataOut);
+#else
+        throw std::runtime_error("ERROR: current ADIOS2 library didn't compile "
+                                 "with MHS, can't use Sirius compressor\n");
 #endif
     }
     else if (compressorType == Operator::OperatorType::Sz)