1/*
2 * Copyright 2014 Google Inc.
3 *
4 * Use of this source code is governed by a BSD-style license that can be
5 * found in the LICENSE file.
6 */
7
8#include "SkDocument.h"
9#include "SkCanvas.h"
10#include "SkImageGenerator.h"
11#include "SkData.h"
12#include "SkStream.h"
13#include "SkDecodingImageGenerator.h"
14
15#include "Resources.h"
16#include "Test.h"
17
18// Returned bitmap is lazy.  Only lazy bitmaps hold onto the original data.
19static SkBitmap bitmap_from_data(SkData* data) {
20    SkASSERT(data);
21    SkBitmap bm;
22    SkInstallDiscardablePixelRef(
23            SkDecodingImageGenerator::Create(
24                    data, SkDecodingImageGenerator::Options()), &bm);
25    return bm;
26}
27
28static bool is_subset_of(SkData* smaller, SkData* larger) {
29    SkASSERT(smaller && larger);
30    if (smaller->size() > larger->size()) {
31        return false;
32    }
33    size_t size = smaller->size();
34    size_t size_diff = larger->size() - size;
35    for (size_t i = 0; i <= size_diff; ++i) {
36        if (0 == memcmp(larger->bytes() + i, smaller->bytes(), size)) {
37            return true;
38        }
39    }
40    return false;
41}
42
43
44static SkData* load_resource(
45        skiatest::Reporter* r, const char* test, const char* filename) {
46    SkString path(GetResourcePath(filename));
47    SkData* data = SkData::NewFromFileName(path.c_str());
48    if (!data && r->verbose()) {
49        SkDebugf("\n%s: Resource '%s' can not be found.\n",
50                 test, filename);
51    }
52    return data;  // May return NULL.
53}
54
55/**
56 *  Test that for Jpeg files that use the JFIF colorspace, they are
57 *  directly embedded into the PDF (without re-encoding) when that
58 *  makes sense.
59 */
60DEF_TEST(PDFJpegEmbedTest, r) {
61    const char test[] = "PDFJpegEmbedTest";
62    SkAutoTUnref<SkData> mandrillData(
63            load_resource(r, test, "mandrill_512_q075.jpg"));
64    SkAutoTUnref<SkData> cmykData(load_resource(r, test, "CMYK.jpg"));
65    if (!mandrillData || !cmykData) {
66        return;
67    }
68
69    SkDynamicMemoryWStream pdf;
70    SkAutoTUnref<SkDocument> document(SkDocument::CreatePDF(&pdf));
71    SkCanvas* canvas = document->beginPage(642, 1028);
72
73    canvas->clear(SK_ColorLTGRAY);
74
75    SkBitmap bm1(bitmap_from_data(mandrillData));
76    canvas->drawBitmap(bm1, 65.0, 0.0, NULL);
77    SkBitmap bm2(bitmap_from_data(cmykData));
78    canvas->drawBitmap(bm2, 0.0, 512.0, NULL);
79
80    canvas->flush();
81    document->endPage();
82    document->close();
83    SkAutoTUnref<SkData> pdfData(pdf.copyToData());
84    SkASSERT(pdfData);
85    pdf.reset();
86
87    REPORTER_ASSERT(r, is_subset_of(mandrillData, pdfData));
88
89    // This JPEG uses a nonstandard colorspace - it can not be
90    // embedded into the PDF directly.
91    REPORTER_ASSERT(r, !is_subset_of(cmykData, pdfData));
92
93    // The following is for debugging purposes only.
94    const char* outputPath = getenv("SKIA_TESTS_PDF_JPEG_EMBED_OUTPUT_PATH");
95    if (outputPath) {
96        SkFILEWStream output(outputPath);
97        if (output.isValid()) {
98            output.write(pdfData->data(), pdfData->size());
99        }
100    }
101}
102