msarett | 3478f75 | 2016-02-12 14:47:09 -0800 | [diff] [blame] | 1 | /* |
| 2 | * Copyright 2016 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 | |
msarett | 3478f75 | 2016-02-12 14:47:09 -0800 | [diff] [blame] | 8 | #include "SkCodec.h" |
| 9 | #include "SkCommandLineFlags.h" |
| 10 | #include "SkData.h" |
mtklein | 2d225e3 | 2016-02-29 09:05:32 -0800 | [diff] [blame] | 11 | #include "SkMD5.h" |
msarett | 3478f75 | 2016-02-12 14:47:09 -0800 | [diff] [blame] | 12 | #include "SkOSFile.h" |
| 13 | #include "SkPicture.h" |
mtklein | 2d225e3 | 2016-02-29 09:05:32 -0800 | [diff] [blame] | 14 | #include "SkPixelSerializer.h" |
msarett | 3478f75 | 2016-02-12 14:47:09 -0800 | [diff] [blame] | 15 | #include "SkStream.h" |
mtklein | 2d225e3 | 2016-02-29 09:05:32 -0800 | [diff] [blame] | 16 | #include "SkTHash.h" |
msarett | 3478f75 | 2016-02-12 14:47:09 -0800 | [diff] [blame] | 17 | |
mtklein | 2d225e3 | 2016-02-29 09:05:32 -0800 | [diff] [blame] | 18 | DEFINE_string2(skps, s, "skps", "A path to a directory of skps."); |
| 19 | DEFINE_string2(out, o, "img-out", "A path to an output directory."); |
msarett | 3478f75 | 2016-02-12 14:47:09 -0800 | [diff] [blame] | 20 | |
mtklein | 2d225e3 | 2016-02-29 09:05:32 -0800 | [diff] [blame] | 21 | static int gKnown; |
| 22 | static int gUnknown; |
msarett | 3478f75 | 2016-02-12 14:47:09 -0800 | [diff] [blame] | 23 | static const char* gOutputDir; |
| 24 | |
mtklein | 2d225e3 | 2016-02-29 09:05:32 -0800 | [diff] [blame] | 25 | static SkTHashSet<SkMD5::Digest> gSeen; |
msarett | 3478f75 | 2016-02-12 14:47:09 -0800 | [diff] [blame] | 26 | |
mtklein | 2d225e3 | 2016-02-29 09:05:32 -0800 | [diff] [blame] | 27 | struct Sniffer : public SkPixelSerializer { |
msarett | 3478f75 | 2016-02-12 14:47:09 -0800 | [diff] [blame] | 28 | |
mtklein | 2d225e3 | 2016-02-29 09:05:32 -0800 | [diff] [blame] | 29 | void sniff(const void* ptr, size_t len) { |
| 30 | SkMD5 md5; |
| 31 | md5.write(ptr, len); |
| 32 | SkMD5::Digest digest; |
| 33 | md5.finish(digest); |
| 34 | |
| 35 | if (gSeen.contains(digest)) { |
| 36 | return; |
msarett | 3478f75 | 2016-02-12 14:47:09 -0800 | [diff] [blame] | 37 | } |
mtklein | 2d225e3 | 2016-02-29 09:05:32 -0800 | [diff] [blame] | 38 | gSeen.add(digest); |
| 39 | |
| 40 | SkAutoTUnref<SkData> data(SkData::NewWithoutCopy(ptr, len)); |
| 41 | SkAutoTDelete<SkCodec> codec(SkCodec::NewFromData(data)); |
| 42 | if (!codec) { |
| 43 | gUnknown++; |
| 44 | return; |
| 45 | } |
| 46 | SkString ext; |
| 47 | switch (codec->getEncodedFormat()) { |
| 48 | case SkEncodedFormat::kBMP_SkEncodedFormat: ext = "bmp"; break; |
| 49 | case SkEncodedFormat::kGIF_SkEncodedFormat: ext = "gif"; break; |
| 50 | case SkEncodedFormat::kICO_SkEncodedFormat: ext = "ico"; break; |
| 51 | case SkEncodedFormat::kJPEG_SkEncodedFormat: ext = "jpg"; break; |
| 52 | case SkEncodedFormat::kPNG_SkEncodedFormat: ext = "png"; break; |
yujieqin | 7a307df | 2016-03-11 07:32:33 -0800 | [diff] [blame] | 53 | case SkEncodedFormat::kDNG_SkEncodedFormat: ext = "dng"; break; |
mtklein | 2d225e3 | 2016-02-29 09:05:32 -0800 | [diff] [blame] | 54 | case SkEncodedFormat::kWBMP_SkEncodedFormat: ext = "wbmp"; break; |
| 55 | case SkEncodedFormat::kWEBP_SkEncodedFormat: ext = "webp"; break; |
| 56 | default: gUnknown++; return; |
| 57 | } |
| 58 | |
| 59 | SkString path; |
| 60 | path.appendf("%s/%d.%s", gOutputDir, gKnown++, ext.c_str()); |
| 61 | |
| 62 | SkFILEWStream file(path.c_str()); |
| 63 | file.write(ptr, len); |
| 64 | |
| 65 | SkDebugf("%s\n", path.c_str()); |
msarett | 3478f75 | 2016-02-12 14:47:09 -0800 | [diff] [blame] | 66 | } |
| 67 | |
mtklein | 2d225e3 | 2016-02-29 09:05:32 -0800 | [diff] [blame] | 68 | bool onUseEncodedData(const void* ptr, size_t len) override { |
| 69 | this->sniff(ptr, len); |
msarett | 3478f75 | 2016-02-12 14:47:09 -0800 | [diff] [blame] | 70 | return true; |
| 71 | } |
mtklein | 2d225e3 | 2016-02-29 09:05:32 -0800 | [diff] [blame] | 72 | SkData* onEncode(const SkPixmap&) override { return nullptr; } |
| 73 | }; |
msarett | 3478f75 | 2016-02-12 14:47:09 -0800 | [diff] [blame] | 74 | |
msarett | 3478f75 | 2016-02-12 14:47:09 -0800 | [diff] [blame] | 75 | |
| 76 | int main(int argc, char** argv) { |
| 77 | SkCommandLineFlags::SetUsage( |
| 78 | "Usage: get_images_from_skps -s <dir of skps> -o <dir for output images>\n"); |
| 79 | |
| 80 | SkCommandLineFlags::Parse(argc, argv); |
msarett | 3478f75 | 2016-02-12 14:47:09 -0800 | [diff] [blame] | 81 | const char* inputs = FLAGS_skps[0]; |
| 82 | gOutputDir = FLAGS_out[0]; |
mtklein | 2d225e3 | 2016-02-29 09:05:32 -0800 | [diff] [blame] | 83 | |
msarett | 3478f75 | 2016-02-12 14:47:09 -0800 | [diff] [blame] | 84 | if (!sk_isdir(inputs) || !sk_isdir(gOutputDir)) { |
| 85 | SkCommandLineFlags::PrintUsage(); |
| 86 | return 1; |
| 87 | } |
| 88 | |
msarett | 3478f75 | 2016-02-12 14:47:09 -0800 | [diff] [blame] | 89 | SkOSFile::Iter iter(inputs, "skp"); |
| 90 | for (SkString file; iter.next(&file); ) { |
| 91 | SkAutoTDelete<SkStream> stream = |
| 92 | SkStream::NewFromFile(SkOSPath::Join(inputs, file.c_str()).c_str()); |
reed | ca2622b | 2016-03-18 07:25:55 -0700 | [diff] [blame] | 93 | sk_sp<SkPicture> picture(SkPicture::MakeFromStream(stream)); |
msarett | 3478f75 | 2016-02-12 14:47:09 -0800 | [diff] [blame] | 94 | |
mtklein | 2d225e3 | 2016-02-29 09:05:32 -0800 | [diff] [blame] | 95 | SkDynamicMemoryWStream scratch; |
| 96 | Sniffer sniff; |
| 97 | picture->serialize(&scratch, &sniff); |
msarett | 3478f75 | 2016-02-12 14:47:09 -0800 | [diff] [blame] | 98 | } |
mtklein | 2d225e3 | 2016-02-29 09:05:32 -0800 | [diff] [blame] | 99 | SkDebugf("%d known, %d unknown\n", gKnown, gUnknown); |
msarett | 3478f75 | 2016-02-12 14:47:09 -0800 | [diff] [blame] | 100 | |
msarett | 3478f75 | 2016-02-12 14:47:09 -0800 | [diff] [blame] | 101 | return 0; |
| 102 | } |