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); 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 135 int main(int argc, char * const argv[]) { 136 return tool_main(argc, (char**) argv); 137 } 138 #endif 139