2
1
mirror of https://github.com/qpdf/qpdf.git synced 2024-11-01 03:12:29 +00:00
qpdf/libtests/png_filter.cc
Jay Berkenbilt ac4deac187 Call QUtil::safe_fopen in place of fopen
fopen was previuosly called wrapped by QUtil::fopen_wrapper, but
QUtil::safe_fopen does this itself, which is less cumbersome.
2013-03-05 13:35:46 -05:00

76 lines
1.5 KiB
C++

#include <qpdf/Pl_PNGFilter.hh>
#include <qpdf/Pl_StdioFile.hh>
#include <qpdf/QUtil.hh>
#include <iostream>
#include <errno.h>
#include <assert.h>
#include <string.h>
#include <stdlib.h>
void run(char const* filename, bool encode, unsigned int columns)
{
// Decode the file
FILE* in = QUtil::safe_fopen(filename, "rb");
FILE* o1 = QUtil::safe_fopen("out", "wb");
Pipeline* out = new Pl_StdioFile("out", o1);
Pipeline* pl = new Pl_PNGFilter(
"png", out,
encode ? Pl_PNGFilter::a_encode : Pl_PNGFilter::a_decode,
columns, 0 /* not used */);
assert((2 * (columns + 1)) < 1024);
unsigned char buf[1024];
size_t len;
while (true)
{
len = fread(buf, 1, (2 * columns) + 1, in);
if (len == 0)
{
break;
}
pl->write(buf, len);
len = fread(buf, 1, 1, in);
if (len == 0)
{
break;
}
pl->write(buf, len);
len = fread(buf, 1, 1, in);
if (len == 0)
{
break;
}
pl->write(buf, len);
}
pl->finish();
delete pl;
delete out;
fclose(o1);
fclose(in);
std::cout << "done" << std::endl;
}
int main(int argc, char* argv[])
{
if (argc != 4)
{
std::cerr << "Usage: pipeline {en,de}code filename columns" << std::endl;
exit(2);
}
bool encode = (strcmp(argv[1], "encode") == 0);
char* filename = argv[2];
int columns = atoi(argv[3]);
try
{
run(filename, encode, columns);
}
catch (std::exception& e)
{
std::cout << e.what() << std::endl;
}
return 0;
}