diff --git a/Makefile.am b/Makefile.am index 175d8f7..411bf0d 100644 --- a/Makefile.am +++ b/Makefile.am @@ -68,6 +68,26 @@ vdservice_rc.$(OBJEXT): vdservice/vdservice.rc MAINTAINERCLEANFILES += vdservice_rc.$(OBJEXT) +check_PROGRAMS = imagetest + +imagetest_LDADD = $(LIBPNG_LIBS) $(ZLIB_LIBS) -lwtsapi32 -lgdi32 +imagetest_CXXFLAGS = $(AM_CXXFLAGS) $(LIBPNG_CFLAGS) +imagetest_LDFLAGS = $(AM_LDFLAGS) -Wl,--subsystem,console +imagetest_SOURCES = \ + common/vdcommon.cpp \ + common/vdcommon.h \ + common/vdlog.cpp \ + common/vdlog.h \ + vdagent/imagetest.cpp \ + vdagent/image.cpp \ + vdagent/image.h \ + vdagent/imagepng.cpp \ + vdagent/imagepng.h \ + $(NULL) + +TESTS = test-png +EXTRA_DIST += test-png + deps.txt: $(AM_V_GEN)rpm -qa | grep $(host_os) | sort | unix2dos > $@ diff --git a/test-png b/test-png new file mode 100755 index 0000000..ee9d86e --- /dev/null +++ b/test-png @@ -0,0 +1,58 @@ +#!/bin/bash + +VERBOSE=${VERBOSE:-0} + +IN=in.png +OUT=out.png +OUT_BMP=out.bmp + +error() { + echo "$*" >&2 + exit 1 +} + +verbose() { + if [ $VERBOSE != 0 ]; then + "$@" + fi +} + +compare_images() { + DIFF=$(compare -metric AE $1 $2 - 2>&1 > /dev/null || true) + if [ "$DIFF" != "0" ]; then + error "Images $1 and $2 are too different, diff $DIFF" + fi +} + +do_test() { + echo "Running image $IMAGE with '$*'..." + convert $IMAGE "$@" $IN + wine imagetest.exe $IN $OUT_BMP $OUT + verbose ls -lh $IN + verbose identify $IN + verbose identify $OUT_BMP + compare_images $IN $OUT + compare_images $IN $OUT_BMP + rm -f $IN $OUT $OUT_BMP +} + +do_test_all() { + IMAGE="$1" + do_test + do_test -type Palette + do_test -type Palette -colors 14 + do_test -type Palette -colors 3 + do_test -type TrueColor + do_test -type Grayscale -depth 8 + do_test -type Grayscale -depth 2 + do_test -type Grayscale -depth 4 +} + +set -e +# test with small image +do_test_all rose: +# test with larger image +do_test_all logo: +rm -f $IN $OUT $OUT_BMP + +verbose echo Finish diff --git a/vdagent/imagetest.cpp b/vdagent/imagetest.cpp new file mode 100644 index 0000000..319b188 --- /dev/null +++ b/vdagent/imagetest.cpp @@ -0,0 +1,82 @@ +/* + Copyright (C) 2017 Red Hat, Inc. + + This program is free software; you can redistribute it and/or + modify it under the terms of the GNU General Public License as + published by the Free Software Foundation; either version 2 of + the License, or (at your option) any later version. + + This program is distributed in the hope that it will be useful, + but WITHOUT ANY WARRANTY; without even the implied warranty of + MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + GNU General Public License for more details. + + You should have received a copy of the GNU General Public License + along with this program. If not, see . +*/ + +#undef NDEBUG +#include +#include + +#include "vdcommon.h" +#include "image.h" +#include "imagepng.h" + +int main(int argc, char **argv) +{ + ImageCoder *coder = create_png_coder(); + + assert(coder); + if (argc < 2) { + fprintf(stderr, "Usage: %s [ []]\n", argv[0]); + return 1; + } + + // read all file into memory + FILE *f = fopen(argv[1], "rb"); + assert(f); + assert(fseek(f, 0, SEEK_END) == 0); + long len = ftell(f); + assert(len > 0); + assert(fseek(f, 0, SEEK_SET) == 0); + + std::vector data(len); + assert(fread(&data[0], 1, len, f) == (unsigned long) len); + fclose(f); + + size_t dib_size = coder->get_dib_size(&data[0], len); + assert(dib_size); + std::vector out(dib_size); + memset(&out[0], 0xcc, dib_size); + coder->get_dib_data(&out[0], &data[0], len); + + // looks like many tools wants this header so craft it + BITMAPFILEHEADER head; + memset(&head, 0, sizeof(head)); + head.bfType = 'B'+'M'*256u; + head.bfSize = sizeof(head) + dib_size; + BITMAPINFOHEADER& info(*(BITMAPINFOHEADER*)&out[0]); + head.bfOffBits = sizeof(head) + sizeof(BITMAPINFOHEADER) + 4 * info.biClrUsed; + + f = fopen(argc > 2 ? argv[2] : "out.bmp", "wb"); + assert(f); + assert(fwrite(&head, 1, sizeof(head), f) == sizeof(head)); + assert(fwrite(&out[0], 1, dib_size, f) == dib_size); + fclose(f); + + // convert back to PNG + long png_size = 0; + uint8_t *png = coder->from_bitmap(*((BITMAPINFO*)&out[0]), &out[sizeof(BITMAPINFOHEADER) + 4 * info.biClrUsed], png_size); + assert(png && png_size > 0); + + f = fopen(argc > 3 ? argv[3] : "out.png", "wb"); + assert(f); + assert(fwrite(png, 1, png_size, f) == (unsigned long) png_size); + fclose(f); + free(png); + png = NULL; + + return 0; +} +