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
14#include "Resources.h"
15#include "Test.h"
16
17// Returned bitmap is lazy.  Only lazy bitmaps hold onto the original data.
18static SkBitmap bitmap_from_data(SkData* data) {
19    SkASSERT(data);
20    SkBitmap bm;
21    SkDEPRECATED_InstallDiscardablePixelRef(data, &bm);
22    return bm;
23}
24
25static bool is_subset_of(SkData* smaller, SkData* larger) {
26    SkASSERT(smaller && larger);
27    if (smaller->size() > larger->size()) {
28        return false;
29    }
30    size_t size = smaller->size();
31    size_t size_diff = larger->size() - size;
32    for (size_t i = 0; i <= size_diff; ++i) {
33        if (0 == memcmp(larger->bytes() + i, smaller->bytes(), size)) {
34            return true;
35        }
36    }
37    return false;
38}
39
40
41static SkData* load_resource(
42        skiatest::Reporter* r, const char* test, const char* filename) {
43    SkString path(GetResourcePath(filename));
44    SkData* data = SkData::NewFromFileName(path.c_str());
45    if (!data) {
46        INFOF(r, "\n%s: Resource '%s' can not be found.\n",
47              test, filename);
48    }
49    return data;  // May return nullptr.
50}
51
52/**
53 *  Test that for Jpeg files that use the JFIF colorspace, they are
54 *  directly embedded into the PDF (without re-encoding) when that
55 *  makes sense.
56 */
57DEF_TEST(PDFJpegEmbedTest, r) {
58    const char test[] = "PDFJpegEmbedTest";
59    SkAutoTUnref<SkData> mandrillData(
60            load_resource(r, test, "mandrill_512_q075.jpg"));
61    SkAutoTUnref<SkData> cmykData(load_resource(r, test, "CMYK.jpg"));
62    if (!mandrillData || !cmykData) {
63        return;
64    }
65    ////////////////////////////////////////////////////////////////////////////
66    SkDynamicMemoryWStream pdf;
67    SkAutoTUnref<SkDocument> document(SkDocument::CreatePDF(&pdf));
68    SkCanvas* canvas = document->beginPage(642, 1028);
69
70    canvas->clear(SK_ColorLTGRAY);
71
72    SkBitmap bm1(bitmap_from_data(mandrillData));
73    canvas->drawBitmap(bm1, 65.0, 0.0, nullptr);
74    SkBitmap bm2(bitmap_from_data(cmykData));
75    canvas->drawBitmap(bm2, 0.0, 512.0, nullptr);
76
77    canvas->flush();
78    document->endPage();
79    document->close();
80    SkAutoTUnref<SkData> pdfData(pdf.copyToData());
81    SkASSERT(pdfData);
82    pdf.reset();
83
84    REPORTER_ASSERT(r, is_subset_of(mandrillData, pdfData));
85
86    // This JPEG uses a nonstandard colorspace - it can not be
87    // embedded into the PDF directly.
88    REPORTER_ASSERT(r, !is_subset_of(cmykData, pdfData));
89    ////////////////////////////////////////////////////////////////////////////
90    pdf.reset();
91    document.reset(SkDocument::CreatePDF(&pdf));
92    canvas = document->beginPage(642, 1028);
93
94    canvas->clear(SK_ColorLTGRAY);
95
96    SkAutoTUnref<SkImage> im1(SkImage::NewFromEncoded(mandrillData));
97    canvas->drawImage(im1, 65.0, 0.0, nullptr);
98    SkAutoTUnref<SkImage> im2(SkImage::NewFromEncoded(cmykData));
99    canvas->drawImage(im2, 0.0, 512.0, nullptr);
100
101    canvas->flush();
102    document->endPage();
103    document->close();
104    pdfData.reset(pdf.copyToData());
105    SkASSERT(pdfData);
106    pdf.reset();
107
108    REPORTER_ASSERT(r, is_subset_of(mandrillData, pdfData));
109
110    // This JPEG uses a nonstandard colorspace - it can not be
111    // embedded into the PDF directly.
112    REPORTER_ASSERT(r, !is_subset_of(cmykData, pdfData));
113}
114
115#include "SkJpegInfo.h"
116
117DEF_TEST(JpegIdentification, r) {
118    static struct {
119        const char* path;
120        bool isJfif;
121        SkJFIFInfo::Type type;
122    } kTests[] = {{"CMYK.jpg", false, SkJFIFInfo::kGrayscale},
123                  {"color_wheel.jpg", true, SkJFIFInfo::kYCbCr},
124                  {"grayscale.jpg", true, SkJFIFInfo::kGrayscale},
125                  {"mandrill_512_q075.jpg", true, SkJFIFInfo::kYCbCr},
126                  {"randPixels.jpg", true, SkJFIFInfo::kYCbCr}};
127    for (size_t i = 0; i < SK_ARRAY_COUNT(kTests); ++i) {
128        SkAutoTUnref<SkData> data(
129                load_resource(r, "JpegIdentification", kTests[i].path));
130        if (!data) {
131            continue;
132        }
133        SkJFIFInfo info;
134        bool isJfif = SkIsJFIF(data, &info);
135        if (isJfif != kTests[i].isJfif) {
136            ERRORF(r, "%s failed isJfif test", kTests[i].path);
137            continue;
138        }
139        if (!isJfif) {
140            continue;  // not applicable
141        }
142        if (kTests[i].type != info.fType) {
143            ERRORF(r, "%s failed jfif type test", kTests[i].path);
144            continue;
145        }
146        INFOF(r, "\nJpegIdentification: %s [%d x %d]\n", kTests[i].path,
147              info.fSize.width(), info.fSize.height());
148    }
149}
150