OLD | NEW |
| (Empty) |
1 /* | |
2 * Copyright 2012 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 "SkCanvas.h" | |
9 #include "SkCommandLineFlags.h" | |
10 #include "SkDocument.h" | |
11 #include "SkForceLinking.h" | |
12 #include "SkGraphics.h" | |
13 #include "SkImageEncoder.h" | |
14 #include "SkOSFile.h" | |
15 #include "SkPicture.h" | |
16 #include "SkStream.h" | |
17 #include "SkTArray.h" | |
18 #include "SkTSort.h" | |
19 #include "ProcStats.h" | |
20 | |
21 __SK_FORCE_IMAGE_DECODER_LINKING; | |
22 | |
23 #ifdef SK_USE_CDB | |
24 #include "win_dbghelp.h" | |
25 #endif | |
26 | |
27 /** | |
28 * render_pdfs | |
29 * | |
30 * Given list of directories and files to use as input, expects to find .skp | |
31 * files and it will convert them to .pdf files writing them in the output | |
32 * directory. | |
33 * | |
34 * Returns zero exit code if all .skp files were converted successfully, | |
35 * otherwise returns error code 1. | |
36 */ | |
37 | |
38 static const char PDF_FILE_EXTENSION[] = "pdf"; | |
39 static const char SKP_FILE_EXTENSION[] = "skp"; | |
40 | |
41 | |
42 DEFINE_string2(inputPaths, r, "", | |
43 "A list of directories and files to use as input. " | |
44 "Files are expected to have the .skp extension."); | |
45 | |
46 DEFINE_string2(outputDir, w, "", | |
47 "Directory to write the rendered pdfs."); | |
48 | |
49 DEFINE_string2(match, m, "", | |
50 "[~][^]substring[$] [...] of filenames to run.\n" | |
51 "Multiple matches may be separated by spaces.\n" | |
52 "~ causes a matching file to always be skipped\n" | |
53 "^ requires the start of the file to match\n" | |
54 "$ requires the end of the file to match\n" | |
55 "^ and $ requires an exact match\n" | |
56 "If a file does not match any list entry,\n" | |
57 "it is skipped unless some list entry starts with ~"); | |
58 | |
59 /** Replaces the extension of a file. | |
60 * @param path File name whose extension will be changed. | |
61 * @param old_extension The old extension. | |
62 * @param new_extension The new extension. | |
63 * @returns false if the file did not has the expected extension. | |
64 * if false is returned, contents of path are undefined. | |
65 */ | |
66 static bool replace_filename_extension(SkString* path, | |
67 const char old_extension[], | |
68 const char new_extension[]) { | |
69 if (path->endsWith(old_extension)) { | |
70 path->remove(path->size() - strlen(old_extension), | |
71 strlen(old_extension)); | |
72 if (!path->endsWith(".")) { | |
73 return false; | |
74 } | |
75 path->append(new_extension); | |
76 return true; | |
77 } | |
78 return false; | |
79 } | |
80 | |
81 /** Builds the output filename. path = dir/name, and it replaces expected | |
82 * .skp extension with .pdf extention. | |
83 * @param path Output filename. | |
84 * @param name The name of the file. | |
85 * @returns false if the file did not has the expected extension. | |
86 * if false is returned, contents of path are undefined. | |
87 */ | |
88 static bool make_output_filepath(SkString* path, const SkString& dir, | |
89 const SkString& name) { | |
90 *path = SkOSPath::Join(dir.c_str(), name.c_str()); | |
91 return replace_filename_extension(path, | |
92 SKP_FILE_EXTENSION, | |
93 PDF_FILE_EXTENSION); | |
94 } | |
95 | |
96 namespace { | |
97 // This is a write-only stream. | |
98 class NullWStream : public SkWStream { | |
99 public: | |
100 NullWStream() : fBytesWritten(0) { } | |
101 bool write(const void*, size_t size) override { | |
102 fBytesWritten += size; | |
103 return true; | |
104 } | |
105 size_t bytesWritten() const override { return fBytesWritten; } | |
106 size_t fBytesWritten; | |
107 }; | |
108 } // namespace | |
109 | |
110 /** Write the output of pdf renderer to a file. | |
111 * @param outputDir Output dir. | |
112 * @param inputFilename The skp file that was read. | |
113 * @param renderer The object responsible to write the pdf file. | |
114 */ | |
115 static SkWStream* open_stream(const SkString& outputDir, | |
116 const SkString& inputFilename) { | |
117 if (outputDir.isEmpty()) { | |
118 return new NullWStream; | |
119 } | |
120 | |
121 SkString outputPath; | |
122 if (!make_output_filepath(&outputPath, outputDir, inputFilename)) { | |
123 return nullptr; | |
124 } | |
125 | |
126 SkAutoTDelete<SkFILEWStream> stream(new SkFILEWStream(outputPath.c_str())); | |
127 if (!stream.get() || !stream->isValid()) { | |
128 SkDebugf("Could not write to file %s\n", outputPath.c_str()); | |
129 return nullptr; | |
130 } | |
131 | |
132 return stream.detach(); | |
133 } | |
134 | |
135 /** | |
136 * Given a SkPicture, write a one-page PDF document to the given | |
137 * output, using the provided encoder. | |
138 */ | |
139 static bool pdf_to_stream(SkPicture* picture, | |
140 SkWStream* output) { | |
141 SkAutoTUnref<SkDocument> pdfDocument( | |
142 SkDocument::CreatePDF(output)); | |
143 SkCanvas* canvas = pdfDocument->beginPage(picture->cullRect().width(), | |
144 picture->cullRect().height()); | |
145 canvas->drawPicture(picture); | |
146 canvas->flush(); | |
147 return pdfDocument->close(); | |
148 } | |
149 | |
150 static bool operator<(const SkString& a, const SkString& b) { | |
151 return strcmp(a.c_str(), b.c_str()) < 0; | |
152 } | |
153 | |
154 /** | |
155 * @param A list of directories or a skp files. | |
156 * @returns an alphabetical list of skp files. | |
157 */ | |
158 static void process_input_files( | |
159 const SkCommandLineFlags::StringArray& inputs, | |
160 SkTArray<SkString>* files) { | |
161 for (int i = 0; i < inputs.count(); i ++) { | |
162 const char* input = inputs[i]; | |
163 if (sk_isdir(input)) { | |
164 SkOSFile::Iter iter(input, SKP_FILE_EXTENSION); | |
165 SkString inputFilename; | |
166 while (iter.next(&inputFilename)) { | |
167 if (!SkCommandLineFlags::ShouldSkip( | |
168 FLAGS_match, inputFilename.c_str())) { | |
169 files->push_back( | |
170 SkOSPath::Join(input, inputFilename.c_str())); | |
171 } | |
172 } | |
173 } else { | |
174 if (!SkCommandLineFlags::ShouldSkip(FLAGS_match, input)) { | |
175 files->push_back(SkString(input)); | |
176 } | |
177 } | |
178 } | |
179 if (files->count() > 0) { | |
180 SkTQSort<SkString>(files->begin(), files->end() - 1); | |
181 } | |
182 } | |
183 | |
184 /** For each input skp file, read it, render it to pdf and write. the | |
185 * output to a pdf file | |
186 */ | |
187 int tool_main_core(int argc, char** argv); | |
188 int tool_main_core(int argc, char** argv) { | |
189 SkCommandLineFlags::Parse(argc, argv); | |
190 | |
191 SkAutoGraphics ag; | |
192 | |
193 SkString outputDir; | |
194 if (FLAGS_outputDir.count() > 0) { | |
195 outputDir = FLAGS_outputDir[0]; | |
196 if (!sk_mkdir(outputDir.c_str())) { | |
197 SkDebugf("Unable to mkdir '%s'\n", outputDir.c_str()); | |
198 return 1; | |
199 } | |
200 } | |
201 | |
202 SkTArray<SkString> files; | |
203 process_input_files(FLAGS_inputPaths, &files); | |
204 | |
205 size_t maximumPathLength = 0; | |
206 for (int i = 0; i < files.count(); i ++) { | |
207 SkString basename = SkOSPath::Basename(files[i].c_str()); | |
208 maximumPathLength = SkTMax(maximumPathLength, basename.size()); | |
209 } | |
210 | |
211 int failures = 0; | |
212 for (int i = 0; i < files.count(); i ++) { | |
213 SkString basename = SkOSPath::Basename(files[i].c_str()); | |
214 | |
215 SkFILEStream inputStream; | |
216 inputStream.setPath(files[i].c_str()); | |
217 if (!inputStream.isValid()) { | |
218 SkDebugf("Could not open file %s\n", files[i].c_str()); | |
219 ++failures; | |
220 continue; | |
221 } | |
222 | |
223 SkAutoTUnref<SkPicture> picture( | |
224 SkPicture::CreateFromStream(&inputStream)); | |
225 if (nullptr == picture.get()) { | |
226 SkDebugf("Could not read an SkPicture from %s\n", | |
227 files[i].c_str()); | |
228 ++failures; | |
229 continue; | |
230 } | |
231 SkDebugf("[%6g %6g %6g %6g] %-*s", | |
232 picture->cullRect().fLeft, picture->cullRect().fTop, | |
233 picture->cullRect().fRight, picture->cullRect().fBottom, | |
234 maximumPathLength, basename.c_str()); | |
235 | |
236 SkAutoTDelete<SkWStream> stream(open_stream(outputDir, files[i])); | |
237 if (!stream.get()) { | |
238 ++failures; | |
239 continue; | |
240 } | |
241 if (!pdf_to_stream(picture, stream.get())) { | |
242 SkDebugf("Error in PDF Serialization."); | |
243 ++failures; | |
244 } | |
245 | |
246 int max_rss_mb = sk_tools::getMaxResidentSetSizeMB(); | |
247 if (max_rss_mb >= 0) { | |
248 SkDebugf(" %4dM peak rss", max_rss_mb); | |
249 } | |
250 | |
251 SkDebugf("\n"); | |
252 } | |
253 if (failures != 0) { | |
254 SkDebugf("Failed to render %i of %i PDFs.\n", failures, files.count()); | |
255 return 1; | |
256 } | |
257 | |
258 return 0; | |
259 } | |
260 | |
261 int tool_main(int argc, char** argv); | |
262 int tool_main(int argc, char** argv) { | |
263 #ifdef SK_USE_CDB | |
264 setUpDebuggingFromArgs(argv[0]); | |
265 __try { | |
266 #endif | |
267 return tool_main_core(argc, argv); | |
268 #ifdef SK_USE_CDB | |
269 } | |
270 __except(GenerateDumpAndPrintCallstack(GetExceptionInformation())) | |
271 { | |
272 return -1; | |
273 } | |
274 #endif | |
275 return 0; | |
276 } | |
277 #if !defined SK_BUILD_FOR_IOS | |
278 int main(int argc, char * const argv[]) { | |
279 return tool_main(argc, (char**) argv); | |
280 } | |
281 #endif | |
OLD | NEW |