From ae90d2c485318beb8b4b938d09ffaf5c6f0a5e21 Mon Sep 17 00:00:00 2001
From: Jay Berkenbilt <ejb@ql.org>
Date: Fri, 18 Aug 2017 19:52:53 -0400
Subject: Implement Pl_DCT pipeline

Additional testing is added in later commits to be supported by
additional changes in the library.
---
 libtests/dct_compress.cc | 93 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 93 insertions(+)
 create mode 100644 libtests/dct_compress.cc

(limited to 'libtests/dct_compress.cc')

diff --git a/libtests/dct_compress.cc b/libtests/dct_compress.cc
new file mode 100644
index 00000000..ea5d9c33
--- /dev/null
+++ b/libtests/dct_compress.cc
@@ -0,0 +1,93 @@
+#include <qpdf/Pl_DCT.hh>
+#include <qpdf/Pl_StdioFile.hh>
+#include <qpdf/QUtil.hh>
+
+#include <stdio.h>
+#include <string.h>
+#include <iostream>
+#include <stdlib.h>
+
+static void usage()
+{
+    std::cerr << "Usage: dct_compress infile outfile width height"
+              << " {rgb|cmyk|gray}" << std::endl;
+    exit(2);
+}
+
+class Callback: public Pl_DCT::CompressConfig
+{
+  public:
+    virtual ~Callback()
+    {
+    }
+    virtual void apply(jpeg_compress_struct*);
+    bool called = false;
+};
+
+void Callback::apply(jpeg_compress_struct*)
+{
+    this->called = true;
+}
+
+int main(int argc, char* argv[])
+{
+    if (argc != 6)
+    {
+        usage();
+    }
+
+    char* infilename = argv[1];
+    char* outfilename = argv[2];
+    unsigned int width = atoi(argv[3]);
+    unsigned int height = atoi(argv[4]);
+    char* colorspace = argv[5];
+    J_COLOR_SPACE cs =
+        ((strcmp(colorspace, "rgb") == 0) ? JCS_RGB :
+         (strcmp(colorspace, "cmyk") == 0) ? JCS_CMYK :
+         (strcmp(colorspace, "gray") == 0) ? JCS_GRAYSCALE :
+         JCS_UNKNOWN);
+    int components = 0;
+    switch (cs)
+    {
+      case JCS_RGB:
+        components = 3;
+        break;
+      case JCS_CMYK:
+        components = 4;
+        break;
+      case JCS_GRAYSCALE:
+        components = 1;
+        break;
+      default:
+        usage();
+        break;
+    }
+
+    FILE* infile = QUtil::safe_fopen(infilename, "rb");
+    FILE* outfile = QUtil::safe_fopen(outfilename, "wb");
+    Pl_StdioFile out("stdout", outfile);
+    unsigned char buf[100];
+    bool done = false;
+    Callback callback;
+    Pl_DCT dct("dct", &out, width, height, components, cs, &callback);
+    while (! done)
+    {
+	size_t len = fread(buf, 1, sizeof(buf), infile);
+	if (len <= 0)
+	{
+	    done = true;
+	}
+	else
+	{
+	    dct.write(buf, len);
+	}
+    }
+    dct.finish();
+    if (! callback.called)
+    {
+        std::cout << "Callback was not called" << std::endl;
+    }
+    fclose(infile);
+    fclose(outfile);
+    return 0;
+}
-- 
cgit v1.2.3-54-g00ecf