1 /*
2  * Copyright 2011 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 "SkData.h"
11 #include "SkDocument.h"
12 #include "SkGraphics.h"
13 #include "SkSurface.h"
14 #include "SkImage.h"
15 #include "SkStream.h"
16 #include "SkString.h"
17 
18 DEFINE_string2(outFile, o, "skhello", "The filename to write the image.");
19 DEFINE_string2(text, t, "Hello", "The string to write.");
20 
21 static void doDraw(SkCanvas* canvas, const SkPaint& paint, const char text[]) {
22     SkRect bounds = canvas->getLocalClipBounds();
23 
24     canvas->drawColor(SK_ColorWHITE);
25     canvas->drawText(text, strlen(text),
26                      bounds.centerX(), bounds.centerY(),
27                      paint);
28 }
29 
30 static bool do_surface(int w, int h, const char path[], const char text[],
31                        const SkPaint& paint) {
32     SkSurfaceProps props(0, kUnknown_SkPixelGeometry);
33     sk_sp<SkSurface> surface(SkSurface::MakeRasterN32Premul(w, h, &props));
34     doDraw(surface->getCanvas(), paint, text);
35 
36     sk_sp<SkImage> image(surface->makeImageSnapshot());
37     sk_sp<SkData> data(image->encode());
38     if (!data) {
39         return false;
40     }
41     SkFILEWStream stream(path);
42     return stream.write(data->data(), data->size());
43 }
44 
45 static bool do_document(int w, int h, const char path[], const char text[],
46                         const SkPaint& paint) {
47     sk_sp<SkDocument> doc(SkDocument::MakePDF(path));
48     if (doc.get()) {
49         SkScalar width = SkIntToScalar(w);
50         SkScalar height = SkIntToScalar(h);
51         doDraw(doc->beginPage(width, height, nullptr), paint, text);
52         return true;
53     }
54     return false;
55 }
56 
57 int main(int argc, char** argv) {
58     SkCommandLineFlags::SetUsage("");
59     SkCommandLineFlags::Parse(argc, argv);
60 
61     SkAutoGraphics ag;
62     SkString path("skhello");
63     SkString text("Hello");
64 
65     if (!FLAGS_outFile.isEmpty()) {
66         path.set(FLAGS_outFile[0]);
67     }
68     if (!FLAGS_text.isEmpty()) {
69         text.set(FLAGS_text[0]);
70     }
71 
72     SkPaint paint;
73     paint.setAntiAlias(true);
74     paint.setTextSize(SkIntToScalar(30));
75     paint.setTextAlign(SkPaint::kCenter_Align);
76 
77     SkScalar width = paint.measureText(text.c_str(), text.size());
78     SkScalar spacing = paint.getFontSpacing();
79 
80     int w = SkScalarRoundToInt(width) + 30;
81     int h = SkScalarRoundToInt(spacing) + 30;
82 
83     static const struct {
84         bool (*fProc)(int w, int h, const char path[], const char text[],
85                       const SkPaint&);
86         const char* fSuffix;
87     } gRec[] = {
88         { do_surface, ".png" },
89         { do_document, ".pdf" },
90     };
91 
92     for (size_t i = 0; i < SK_ARRAY_COUNT(gRec); ++i) {
93         SkString file;
94         file.printf("%s%s", path.c_str(), gRec[i].fSuffix);
95         if (!gRec[i].fProc(w, h, file.c_str(), text.c_str(), paint)) {
96             return -1;
97         }
98     }
99     return 0;
100 }
101