1/*
2 * Copyright 2013 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 "SkBitmap.h"
9#include "SkColorPriv.h"
10#include "SkCommandLineFlags.h"
11#include "SkData.h"
12#include "SkForceLinking.h"
13#include "SkGraphics.h"
14#include "SkImageDecoder.h"
15#include "SkImageEncoder.h"
16#include "SkOSFile.h"
17#include "SkRandom.h"
18#include "SkStream.h"
19#include "SkTArray.h"
20#include "SkTemplates.h"
21
22__SK_FORCE_IMAGE_DECODER_LINKING;
23
24DEFINE_string2(readPath, r, "", "Folder(s) and files to decode images. Required.");
25
26struct Format {
27    SkImageEncoder::Type    fType;
28    SkImageDecoder::Format  fFormat;
29    const char*             fSuffix;
30};
31
32/*
33static const Format gFormats[] = {
34    { SkImageEncoder::kBMP_Type, SkImageDecoder::kBMP_Format, ".bmp" },
35    { SkImageEncoder::kGIF_Type, SkImageDecoder::kGIF_Format, ".gif" },
36    { SkImageEncoder::kICO_Type, SkImageDecoder::kICO_Format, ".ico" },
37    { SkImageEncoder::kJPEG_Type, SkImageDecoder::kJPEG_Format, ".jpg" },
38    { SkImageEncoder::kPNG_Type, SkImageDecoder::kPNG_Format, ".png" },
39    { SkImageEncoder::kWBMP_Type, SkImageDecoder::kWBMP_Format, ".wbmp" },
40    { SkImageEncoder::kWEBP_Type, SkImageDecoder::kWEBP_Format, ".webp" }
41};
42*/
43
44static SkISize opaqueSize(const SkBitmap& bm) {
45    int width = 1;
46    int height = 1;
47    for (int y = 0 ; y < bm.height(); y++) {
48        for (int x = 0 ; x < bm.width(); x++) {
49            SkColor color = bm.getColor(x, y);
50            if (SkColorGetA(color) != 0) {
51                height = y + 1;
52                width = width > (x + 1) ? width : x + 1;
53            }
54        }
55    }
56
57    return SkISize::Make(width, height);
58}
59
60static void setup_bitmap(SkBitmap* bitmap, int width, int height) {
61    bitmap->allocN32Pixels(width, height);
62}
63
64
65static bool write_bitmap(const char outName[], const SkBitmap& bm) {
66    SkISize size = opaqueSize(bm);
67    SkBitmap dst;
68    setup_bitmap(&dst, size.width(), size.height());
69
70    for (int y = 0 ; y < dst.height(); y++) {
71        for (int x = 0 ; x < dst.width(); x++) {
72            SkColor color = bm.getColor(x, y);
73            if (SkColorGetA(color) != 0xff) {
74                int a = SkColorGetA(color);
75                int r = SkColorGetR(color);
76                int g = SkColorGetG(color);
77                int b = SkColorGetB(color);
78                if (a == 0) {
79                    r = g = b = 0;
80                } else {
81                    r = (r * a) / 255;
82                    g = (g * a) / 255;
83                    b = (b * a) / 255;
84                    a = 255;
85                }
86                color = SkColorSetARGB((U8CPU)a, (U8CPU)r, (U8CPU)g, (U8CPU)b);
87            }
88            *dst.getAddr32(x, y) = color;
89        }
90    }
91
92    return SkImageEncoder::EncodeFile(outName, dst, SkImageEncoder::kPNG_Type, 100);
93}
94
95static void decodeFileAndWrite(const char srcPath[]) {
96    SkBitmap bitmap;
97    SkFILEStream stream(srcPath);
98    if (!stream.isValid()) {
99        return;
100    }
101
102    SkImageDecoder* codec = SkImageDecoder::Factory(&stream);
103    if (NULL == codec) {
104        return;
105    }
106
107    SkAutoTDelete<SkImageDecoder> ad(codec);
108
109    stream.rewind();
110
111    if (!codec->decode(&stream, &bitmap, kN32_SkColorType, SkImageDecoder::kDecodePixels_Mode)) {
112        return;
113    }
114
115    write_bitmap(srcPath, bitmap);
116}
117
118/**
119 *  Return true if the filename represents an image.
120 */
121static bool is_image_file(const char* filename) {
122    const char* gImageExtensions[] = {
123        ".png", ".PNG", ".jpg", ".JPG", ".jpeg", ".JPEG", ".bmp", ".BMP",
124        ".webp", ".WEBP", ".ico", ".ICO", ".wbmp", ".WBMP", ".gif", ".GIF"
125    };
126    for (size_t i = 0; i < SK_ARRAY_COUNT(gImageExtensions); ++i) {
127        if (SkStrEndsWith(filename, gImageExtensions[i])) {
128            return true;
129        }
130    }
131    return false;
132}
133
134int tool_main(int argc, char** argv);
135int tool_main(int argc, char** argv) {
136    SkCommandLineFlags::SetUsage("Decode files, and optionally write the results to files.");
137    SkCommandLineFlags::Parse(argc, argv);
138
139    if (FLAGS_readPath.count() < 1) {
140        SkDebugf("Folder(s) or image(s) to decode are required.\n");
141        return -1;
142    }
143
144
145    SkAutoGraphics ag;
146
147    for (int i = 0; i < FLAGS_readPath.count(); i++) {
148        const char* readPath = FLAGS_readPath[i];
149        if (strlen(readPath) < 1) {
150            break;
151        }
152        if (sk_isdir(readPath)) {
153            const char* dir = readPath;
154            SkOSFile::Iter iter(dir);
155            SkString filename;
156            while (iter.next(&filename)) {
157                if (!is_image_file(filename.c_str())) {
158                    continue;
159                }
160                SkString fullname = SkOSPath::Join(dir, filename.c_str());
161                decodeFileAndWrite(fullname.c_str());
162            }
163        } else if (sk_exists(readPath) && is_image_file(readPath)) {
164            decodeFileAndWrite(readPath);
165        }
166    }
167
168    return 0;
169}
170
171#if !defined SK_BUILD_FOR_IOS
172int main(int argc, char * const argv[]) {
173    return tool_main(argc, (char**) argv);
174}
175#endif
176