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 "gm.h"
9
10#include "Resources.h"
11#include "SkCanvas.h"
12#include "SkData.h"
13#include "SkImageGenerator.h"
14#include "SkImageDecoder.h"
15#include "SkOSFile.h"
16#include "SkTemplates.h"
17
18#ifndef SK_IGNORE_ETC1_SUPPORT
19
20#include "etc1.h"
21
22/**
23 *  Remove the last row and column of ETC1 blocks, effectively
24 *  making a texture that started as power of two into a texture
25 *  that is no longer power of two...
26 */
27bool slice_etc1_data(void *data, int* width, int* height) {
28
29    // First, parse the data and get to it...
30    etc1_byte *origData = reinterpret_cast<etc1_byte *>(data);
31    if (!etc1_pkm_is_valid(origData)) {
32        return false;
33    }
34
35    int origW = etc1_pkm_get_width(origData);
36    int origH = etc1_pkm_get_height(origData);
37
38    int blockWidth = (origW + 3) >> 2;
39    int blockHeight = (origH + 3) >> 2;
40
41    // Make sure that we have blocks to trim off..
42    if (blockWidth < 2 || blockHeight < 2) {
43        return false;
44    }
45
46    int newWidth = (blockWidth - 1) << 2;
47    int newHeight = (blockHeight - 1) << 2;
48
49    size_t newDataSz = etc1_get_encoded_data_size(newWidth, newHeight) + ETC_PKM_HEADER_SIZE;
50    SkAutoTMalloc<etc1_byte> am(newDataSz);
51
52    etc1_byte* newData = am.get();
53
54    etc1_pkm_format_header(newData, newWidth, newHeight);
55    newData += ETC_PKM_HEADER_SIZE;
56    origData += ETC_PKM_HEADER_SIZE;
57
58    for (int j = 0; j < blockHeight - 1; ++j) {
59        memcpy(newData, origData, (blockWidth - 1)*ETC1_ENCODED_BLOCK_SIZE);
60        origData += blockWidth*ETC1_ENCODED_BLOCK_SIZE;
61        newData += (blockWidth - 1)*ETC1_ENCODED_BLOCK_SIZE;
62    }
63
64    // Stick the data back whence it came
65    memcpy(data, am.get(), newDataSz);
66    *width = newWidth;
67    *height = newHeight;
68
69    return true;
70}
71#endif  // SK_IGNORE_ETC1_SUPPORT
72
73namespace skiagm {
74
75/**
76 *  Test decoding an image from a PKM or KTX file and then
77 *  from compressed ETC1 data.
78 */
79class ETC1BitmapGM : public GM {
80public:
81    ETC1BitmapGM() { }
82    virtual ~ETC1BitmapGM() { }
83
84protected:
85    SkString onShortName() override {
86        SkString str = SkString("etc1bitmap_");
87        str.append(this->fileExtension());
88        return str;
89    }
90
91    SkISize onISize() override {
92        return SkISize::Make(128, 128);
93    }
94
95    virtual SkString fileExtension() const = 0;
96
97    void onDraw(SkCanvas* canvas) override {
98        SkBitmap bm;
99        SkString filename = GetResourcePath("mandrill_128.");
100        filename.append(this->fileExtension());
101        SkAutoTUnref<SkData> fileData(SkData::NewFromFileName(filename.c_str()));
102        if (nullptr == fileData) {
103            SkDebugf("Could not open the file. Did you forget to set the resourcePath?\n");
104            return;
105        }
106
107        SkAutoTUnref<SkImage> image(SkImage::NewFromEncoded(fileData));
108        if (nullptr == image) {
109            SkDebugf("Could not decode the ETC file. ETC may not be included in this platform.\n");
110            return;
111        }
112        canvas->drawImage(image, 0, 0);
113    }
114
115private:
116    typedef GM INHERITED;
117};
118
119// This class specializes ETC1BitmapGM to load the mandrill_128.pkm file.
120class ETC1Bitmap_PKM_GM : public ETC1BitmapGM {
121public:
122    ETC1Bitmap_PKM_GM() : ETC1BitmapGM() { }
123    virtual ~ETC1Bitmap_PKM_GM() { }
124
125protected:
126
127    SkString fileExtension() const override { return SkString("pkm"); }
128
129private:
130    typedef ETC1BitmapGM INHERITED;
131};
132
133// This class specializes ETC1BitmapGM to load the mandrill_128.ktx file.
134class ETC1Bitmap_KTX_GM : public ETC1BitmapGM {
135public:
136    ETC1Bitmap_KTX_GM() : ETC1BitmapGM() { }
137    virtual ~ETC1Bitmap_KTX_GM() { }
138
139protected:
140
141    SkString fileExtension() const override { return SkString("ktx"); }
142
143private:
144    typedef ETC1BitmapGM INHERITED;
145};
146
147// This class specializes ETC1BitmapGM to load the mandrill_128.r11.ktx file.
148class ETC1Bitmap_R11_KTX_GM : public ETC1BitmapGM {
149public:
150    ETC1Bitmap_R11_KTX_GM() : ETC1BitmapGM() { }
151    virtual ~ETC1Bitmap_R11_KTX_GM() { }
152
153protected:
154
155    SkString fileExtension() const override { return SkString("r11.ktx"); }
156
157private:
158    typedef ETC1BitmapGM INHERITED;
159};
160
161#ifndef SK_IGNORE_ETC1_SUPPORT
162/**
163 *  Test decoding an image from a PKM file and then
164 *  from non-power-of-two compressed ETC1 data. First slice
165 *  off a row and column of blocks in order to make it non-power
166 *  of two.
167 */
168class ETC1Bitmap_NPOT_GM : public GM {
169public:
170    ETC1Bitmap_NPOT_GM() { }
171    virtual ~ETC1Bitmap_NPOT_GM() { }
172
173protected:
174    SkString onShortName() override {
175        return SkString("etc1bitmap_npot");
176    }
177
178    SkISize onISize() override {
179        return SkISize::Make(124, 124);
180    }
181
182    void onDraw(SkCanvas* canvas) override {
183        SkBitmap bm;
184        SkString pkmFilename = GetResourcePath("mandrill_128.pkm");
185        SkAutoDataUnref fileData(SkData::NewFromFileName(pkmFilename.c_str()));
186        if (nullptr == fileData) {
187            SkDebugf("Could not open the file. Did you forget to set the resourcePath?\n");
188            return;
189        }
190
191        SkAutoMalloc am(fileData->size());
192        memcpy(am.get(), fileData->data(), fileData->size());
193
194        int width, height;
195        if (!slice_etc1_data(am.get(), &width, &height)) {
196            SkDebugf("ETC1 Data is poorly formatted.\n");
197            return;
198        }
199
200        SkASSERT(124 == width);
201        SkASSERT(124 == height);
202
203        size_t dataSz = etc1_get_encoded_data_size(width, height) + ETC_PKM_HEADER_SIZE;
204        SkAutoDataUnref nonPOTData(SkData::NewWithCopy(am.get(), dataSz));
205
206        SkAutoTUnref<SkImage> image(SkImage::NewFromEncoded(nonPOTData));
207        canvas->drawImage(image, 0, 0);
208    }
209
210private:
211    typedef GM INHERITED;
212};
213#endif  // SK_IGNORE_ETC1_SUPPORT
214
215}  // namespace skiagm
216
217//////////////////////////////////////////////////////////////////////////////
218
219DEF_GM(return new skiagm::ETC1Bitmap_PKM_GM;)
220DEF_GM(return new skiagm::ETC1Bitmap_KTX_GM;)
221DEF_GM(return new skiagm::ETC1Bitmap_R11_KTX_GM;)
222
223#ifndef SK_IGNORE_ETC1_SUPPORT
224DEF_GM(return new skiagm::ETC1Bitmap_NPOT_GM;)
225#endif  // SK_IGNORE_ETC1_SUPPORT
226