1 /*
2 * Copyright 2015 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 "SkCommandLineFlags.h"
9 #include "SkImageDecoder.h"
10 #include "SkStream.h"
11
12 DEFINE_bool(header, false, "Print an extra row of the min-max values");
13 DEFINE_string2(label, l, "label", "Label printed as the first value");
14
15 DEFINE_string2(image, i, "", "Input image");
16 DEFINE_int32_2(row, r, -1, "Row to extract");
17 DEFINE_int32_2(column, c, -1, "Column to extract");
18
19 DEFINE_int32_2(min, n, 0, "Minimum row/column to extract - inclusive");
20 DEFINE_int32_2(max, m, 100, "Maximum row/column to extract - inclusive");
21
22 DEFINE_int32(rgb, 0, "Color channel to print (0->b, 1->g, 2->r, 3->a)");
23
24 DEFINE_bool2(quiet, q, false, "Quiet");
25 DEFINE_bool2(reverse, v, false, "Iterate from max to min");
26
27
28 // This tool just loads a single image and prints out a comma-separated row or column
29 // Return codes:
30 static const int kSuccess = 0;
31 static const int kError = 1;
32
33 int tool_main(int argc, char** argv);
tool_main(int argc,char ** argv)34 int tool_main(int argc, char** argv) {
35 SkCommandLineFlags::SetUsage("Print out a row or column of an image.");
36 SkCommandLineFlags::Parse(argc, argv);
37
38 if (FLAGS_rgb > 3 || FLAGS_rgb < 0) {
39 if (!FLAGS_quiet) {
40 SkDebugf("Channel (--rgb) must be between 0 and 3 (inclusive) - value is %d.\n",
41 FLAGS_rgb);
42 }
43 return kError;
44 }
45
46 if (FLAGS_row >= 0 && FLAGS_column >= 0) {
47 if (!FLAGS_quiet) {
48 SkDebugf("Only one of '-c' or '-r' can be specified at at time.\n");
49 }
50 return kError;
51 }
52
53 if (FLAGS_row < 0 && FLAGS_column < 0) {
54 if (!FLAGS_quiet) {
55 SkDebugf("At least one of '-c' or '-r' need to be specified.\n");
56 }
57 return kError;
58 }
59
60 SkFILEStream inputStream(FLAGS_image[0]);
61 if (!inputStream.isValid()) {
62 if (!FLAGS_quiet) {
63 SkDebugf("Couldn't open file: %s\n", FLAGS_image[0]);
64 }
65 return kError;
66 }
67
68 SkAutoTDelete<SkImageDecoder> codec(SkImageDecoder::Factory(&inputStream));
69 if (!codec) {
70 if (!FLAGS_quiet) {
71 SkDebugf("Couldn't create codec for: %s.\n", FLAGS_image[0]);
72 }
73 return kError;
74 }
75
76 SkBitmap bitmap;
77
78 inputStream.rewind();
79 codec->decode(&inputStream, &bitmap, kN32_SkColorType, SkImageDecoder::kDecodePixels_Mode);
80
81 int top, bottom, left, right;
82
83 if (-1 != FLAGS_row) {
84 SkASSERT(-1 == FLAGS_column);
85
86 top = bottom = SkTPin(FLAGS_row, 0, bitmap.height()-1);
87 FLAGS_min = left = SkTPin(FLAGS_min, 0, bitmap.width()-1);
88 FLAGS_max = right = SkTPin(FLAGS_max, left, bitmap.width()-1);
89 } else {
90 SkASSERT(-1 != FLAGS_column);
91 left = right = SkTPin(FLAGS_column, 0, bitmap.width()-1);
92 FLAGS_min = top = SkTPin(FLAGS_min, 0, bitmap.height()-1);
93 FLAGS_max = bottom = SkTPin(FLAGS_max, top, bitmap.height()-1);
94 }
95
96 if (FLAGS_header) {
97 SkDebugf("header");
98 if (FLAGS_reverse) {
99 for (int i = FLAGS_max; i >= FLAGS_min; --i) {
100 SkDebugf(", %d", i);
101 }
102 } else {
103 for (int i = FLAGS_min; i <= FLAGS_max; ++i) {
104 SkDebugf(", %d", i);
105 }
106 }
107 SkDebugf("\n");
108 }
109
110 SkDebugf("%s", FLAGS_label[0]);
111 if (FLAGS_reverse) {
112 for (int y = bottom; y >= top; --y) {
113 for (int x = right; x >= left; --x) {
114 SkColor c = bitmap.getColor(x, y);
115
116 SkDebugf(", %d", ((c) >> (FLAGS_rgb*8)) & 0xFF);
117 }
118 }
119 } else {
120 for (int y = top; y <= bottom; ++y) {
121 for (int x = left; x <= right; ++x) {
122 SkColor c = bitmap.getColor(x, y);
123
124 SkDebugf(", %d", ((c) >> (FLAGS_rgb*8)) & 0xFF);
125 }
126 }
127 }
128
129 SkDebugf("\n");
130
131 return kSuccess;
132 }
133
134 #if !defined SK_BUILD_FOR_IOS
main(int argc,char * const argv[])135 int main(int argc, char * const argv[]) {
136 return tool_main(argc, (char**) argv);
137 }
138 #endif
139