From b53c4fd33559b743daa41a74e8aa457d39d356ce Mon Sep 17 00:00:00 2001
From: Vojtech Moravec <vojtech.moravec.st@vsb.cz>
Date: Fri, 16 Oct 2020 11:38:39 +0200
Subject: [PATCH] Don't try to write non-existent quantizer to compress stream.

---
 src/main/java/cz/it4i/qcmp/compression/VQImageCompressor.java | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/main/java/cz/it4i/qcmp/compression/VQImageCompressor.java b/src/main/java/cz/it4i/qcmp/compression/VQImageCompressor.java
index 848ba15..6d10758 100644
--- a/src/main/java/cz/it4i/qcmp/compression/VQImageCompressor.java
+++ b/src/main/java/cz/it4i/qcmp/compression/VQImageCompressor.java
@@ -166,6 +166,7 @@ public class VQImageCompressor extends CompressorDecompressorBase implements IIm
                 quantizer = loadQuantizerFromCache();
                 huffman = createHuffmanCoder(huffmanSymbols, quantizer.getFrequencies());
                 reportStatusToListeners("Cached quantizer with huffman coder created.");
+                writeQuantizerToCompressStream(quantizer, compressStream);
             } else if (options.getCodebookType() == CompressionOptions.CodebookType.MiddlePlane) {
                 stopwatch.restart();
                 reportStatusToListeners("Training vector quantizer from middle plane.");
@@ -175,8 +176,8 @@ public class VQImageCompressor extends CompressorDecompressorBase implements IIm
                 huffman = createHuffmanCoder(huffmanSymbols, quantizer.getFrequencies());
                 stopwatch.stop();
                 reportStatusToListeners("Middle plane codebook created in: " + stopwatch.getElapsedTimeString());
+                writeQuantizerToCompressStream(quantizer, compressStream);
             }
-            writeQuantizerToCompressStream(quantizer, compressStream);
         }
 
         final int[] planeIndices = getPlaneIndicesForCompression(inputData);
-- 
GitLab