halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [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 | |
halcanary | 85e38bc | 2016-05-04 09:34:25 -0700 | [diff] [blame] | 8 | // This sample progam demonstrates how to use Skia and HarfBuzz to |
| 9 | // produce a PDF file from UTF-8 text in stdin. |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 10 | |
| 11 | #include <cassert> |
halcanary | dd05e2a | 2016-05-12 10:09:58 -0700 | [diff] [blame] | 12 | #include <cstdlib> |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 13 | #include <iostream> |
| 14 | #include <map> |
halcanary | dd05e2a | 2016-05-12 10:09:58 -0700 | [diff] [blame] | 15 | #include <sstream> |
halcanary | 8a74f13 | 2016-07-11 14:30:39 -0700 | [diff] [blame] | 16 | #include <string> |
| 17 | #include <vector> |
halcanary | 85e38bc | 2016-05-04 09:34:25 -0700 | [diff] [blame] | 18 | |
| 19 | #include "SkCanvas.h" |
| 20 | #include "SkDocument.h" |
halcanary | 8a74f13 | 2016-07-11 14:30:39 -0700 | [diff] [blame] | 21 | #include "SkShaper.h" |
halcanary | 85e38bc | 2016-05-04 09:34:25 -0700 | [diff] [blame] | 22 | #include "SkStream.h" |
| 23 | #include "SkTextBlob.h" |
| 24 | #include "SkTypeface.h" |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 25 | |
halcanary | 8a74f13 | 2016-07-11 14:30:39 -0700 | [diff] [blame] | 26 | // Options ///////////////////////////////////////////////////////////////////// |
| 27 | |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 28 | struct BaseOption { |
halcanary | 8a74f13 | 2016-07-11 14:30:39 -0700 | [diff] [blame] | 29 | std::string selector; |
| 30 | std::string description; |
| 31 | virtual void set(std::string _value) = 0; |
| 32 | virtual std::string valueToString() = 0; |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 33 | |
halcanary | 8a74f13 | 2016-07-11 14:30:39 -0700 | [diff] [blame] | 34 | BaseOption(std::string _selector, std::string _description) |
| 35 | : selector(_selector), description(_description) {} |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 36 | |
halcanary | 8a74f13 | 2016-07-11 14:30:39 -0700 | [diff] [blame] | 37 | virtual ~BaseOption() {} |
| 38 | |
| 39 | static void Init(const std::vector<BaseOption*> &, int argc, char **argv); |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 40 | }; |
| 41 | |
halcanary | 8a74f13 | 2016-07-11 14:30:39 -0700 | [diff] [blame] | 42 | template <class T> |
| 43 | struct Option : BaseOption { |
| 44 | T value; |
| 45 | Option(std::string selector, std::string description, T defaultValue) |
| 46 | : BaseOption(selector, description), value(defaultValue) {} |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 47 | }; |
| 48 | |
halcanary | 8a74f13 | 2016-07-11 14:30:39 -0700 | [diff] [blame] | 49 | void BaseOption::Init(const std::vector<BaseOption*> &option_list, |
| 50 | int argc, char **argv) { |
| 51 | std::map<std::string, BaseOption *> options; |
| 52 | for (BaseOption *opt : option_list) { |
| 53 | options[opt->selector] = opt; |
| 54 | } |
| 55 | for (int i = 1; i < argc; i++) { |
| 56 | std::string option_selector(argv[i]); |
| 57 | auto it = options.find(option_selector); |
| 58 | if (it != options.end()) { |
| 59 | if (i >= argc) { |
| 60 | break; |
| 61 | } |
| 62 | const char *option_value = argv[i + 1]; |
| 63 | it->second->set(option_value); |
| 64 | i++; |
| 65 | } else { |
| 66 | printf("Ignoring unrecognized option: %s.\n", argv[i]); |
| 67 | printf("Usage: %s {option value}\n", argv[0]); |
| 68 | printf("\tTakes text from stdin and produces pdf file.\n"); |
| 69 | printf("Supported options:\n"); |
| 70 | for (BaseOption *opt : option_list) { |
| 71 | printf("\t%s\t%s (%s)\n", opt->selector.c_str(), |
| 72 | opt->description.c_str(), opt->valueToString().c_str()); |
| 73 | } |
| 74 | exit(-1); |
| 75 | } |
| 76 | } |
| 77 | } |
| 78 | |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 79 | struct DoubleOption : Option<double> { |
halcanary | 8a74f13 | 2016-07-11 14:30:39 -0700 | [diff] [blame] | 80 | virtual void set(std::string _value) { value = atof(_value.c_str()); } |
| 81 | virtual std::string valueToString() { |
| 82 | std::ostringstream stm; |
| 83 | stm << value; |
| 84 | return stm.str(); |
| 85 | } |
| 86 | DoubleOption(std::string selector, |
| 87 | std::string description, |
| 88 | double defaultValue) |
| 89 | : Option<double>(selector, description, defaultValue) {} |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 90 | }; |
| 91 | |
halcanary | 8a74f13 | 2016-07-11 14:30:39 -0700 | [diff] [blame] | 92 | struct StringOption : Option<std::string> { |
| 93 | virtual void set(std::string _value) { value = _value; } |
| 94 | virtual std::string valueToString() { return value; } |
| 95 | StringOption(std::string selector, |
| 96 | std::string description, |
| 97 | std::string defaultValue) |
| 98 | : Option<std::string>(selector, description, defaultValue) {} |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 99 | }; |
| 100 | |
halcanary | 8a74f13 | 2016-07-11 14:30:39 -0700 | [diff] [blame] | 101 | // Config ////////////////////////////////////////////////////////////////////// |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 102 | |
| 103 | struct Config { |
halcanary | 8a74f13 | 2016-07-11 14:30:39 -0700 | [diff] [blame] | 104 | DoubleOption page_width = DoubleOption("-w", "Page width", 600.0f); |
| 105 | DoubleOption page_height = DoubleOption("-h", "Page height", 800.0f); |
| 106 | StringOption title = StringOption("-t", "PDF title", "---"); |
| 107 | StringOption author = StringOption("-a", "PDF author", "---"); |
| 108 | StringOption subject = StringOption("-k", "PDF subject", "---"); |
| 109 | StringOption keywords = StringOption("-c", "PDF keywords", "---"); |
| 110 | StringOption creator = StringOption("-t", "PDF creator", "---"); |
| 111 | StringOption font_file = StringOption("-f", ".ttf font file", ""); |
| 112 | DoubleOption font_size = DoubleOption("-z", "Font size", 8.0f); |
| 113 | DoubleOption left_margin = DoubleOption("-m", "Left margin", 20.0f); |
| 114 | DoubleOption line_spacing_ratio = |
| 115 | DoubleOption("-h", "Line spacing ratio", 1.5f); |
| 116 | StringOption output_file_name = |
| 117 | StringOption("-o", ".pdf output file name", "out-skiahf.pdf"); |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 118 | |
halcanary | 8a74f13 | 2016-07-11 14:30:39 -0700 | [diff] [blame] | 119 | Config(int argc, char **argv) { |
| 120 | BaseOption::Init(std::vector<BaseOption*>{ |
| 121 | &page_width, &page_height, &title, &author, &subject, |
| 122 | &keywords, &creator, &font_file, &font_size, &left_margin, |
| 123 | &line_spacing_ratio, &output_file_name}, argc, argv); |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 124 | } |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 125 | }; |
| 126 | |
halcanary | 8a74f13 | 2016-07-11 14:30:39 -0700 | [diff] [blame] | 127 | // Placement /////////////////////////////////////////////////////////////////// |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 128 | |
| 129 | class Placement { |
halcanary | 8a74f13 | 2016-07-11 14:30:39 -0700 | [diff] [blame] | 130 | public: |
| 131 | Placement(const Config* conf, SkDocument *doc) |
| 132 | : config(conf), document(doc), pageCanvas(nullptr) { |
| 133 | white_paint.setColor(SK_ColorWHITE); |
| 134 | glyph_paint.setColor(SK_ColorBLACK); |
| 135 | glyph_paint.setFlags(SkPaint::kAntiAlias_Flag | |
| 136 | SkPaint::kSubpixelText_Flag); |
| 137 | glyph_paint.setTextSize(config->font_size.value); |
halcanary | 627ad6d | 2016-07-01 08:48:12 -0700 | [diff] [blame] | 138 | } |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 139 | |
halcanary | 8a74f13 | 2016-07-11 14:30:39 -0700 | [diff] [blame] | 140 | void WriteLine(const SkShaper& shaper, const char *text, size_t textBytes) { |
| 141 | if (!pageCanvas || current_y > config->page_height.value) { |
| 142 | if (pageCanvas) { |
| 143 | document->endPage(); |
| 144 | } |
| 145 | pageCanvas = document->beginPage(config->page_width.value, |
| 146 | config->page_height.value); |
| 147 | pageCanvas->drawPaint(white_paint); |
| 148 | current_x = config->left_margin.value; |
| 149 | current_y = config->line_spacing_ratio.value * config->font_size.value; |
| 150 | } |
| 151 | SkTextBlobBuilder textBlobBuilder; |
| 152 | shaper.shape(&textBlobBuilder, glyph_paint, text, textBytes, SkPoint{0, 0}); |
| 153 | sk_sp<const SkTextBlob> blob(textBlobBuilder.build()); |
| 154 | pageCanvas->drawTextBlob(blob.get(), current_x, current_y, glyph_paint); |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 155 | |
halcanary | 8a74f13 | 2016-07-11 14:30:39 -0700 | [diff] [blame] | 156 | // Advance to the next line. |
| 157 | current_y += config->line_spacing_ratio.value * config->font_size.value; |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 158 | } |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 159 | |
| 160 | private: |
halcanary | 8a74f13 | 2016-07-11 14:30:39 -0700 | [diff] [blame] | 161 | const Config* config; |
| 162 | SkDocument *document; |
| 163 | SkCanvas *pageCanvas; |
| 164 | SkPaint white_paint; |
| 165 | SkPaint glyph_paint; |
| 166 | double current_x; |
| 167 | double current_y; |
| 168 | }; |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 169 | |
halcanary | 8a74f13 | 2016-07-11 14:30:39 -0700 | [diff] [blame] | 170 | //////////////////////////////////////////////////////////////////////////////// |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 171 | |
halcanary | 8a74f13 | 2016-07-11 14:30:39 -0700 | [diff] [blame] | 172 | static sk_sp<SkDocument> MakePDFDocument(const Config &config, |
| 173 | SkWStream *wStream) { |
| 174 | SkDocument::PDFMetadata pdf_info; |
| 175 | pdf_info.fTitle = config.title.value.c_str(); |
| 176 | pdf_info.fAuthor = config.author.value.c_str(); |
| 177 | pdf_info.fSubject = config.subject.value.c_str(); |
| 178 | pdf_info.fKeywords = config.keywords.value.c_str(); |
| 179 | pdf_info.fCreator = config.creator.value.c_str(); |
| 180 | bool pdfa = false; |
| 181 | #if 0 |
| 182 | SkTime::DateTime now; |
| 183 | SkTime::GetDateTime(&now); |
| 184 | pdf_info.fCreation.fEnabled = true; |
| 185 | pdf_info.fCreation.fDateTime = now; |
| 186 | pdf_info.fModified.fEnabled = true; |
| 187 | pdf_info.fModified.fDateTime = now; |
| 188 | pdfa = true; |
| 189 | #endif |
| 190 | return SkDocument::MakePDF(wStream, SK_ScalarDefaultRasterDPI, pdf_info, |
| 191 | nullptr, pdfa); |
| 192 | } |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 193 | |
halcanary | 8a74f13 | 2016-07-11 14:30:39 -0700 | [diff] [blame] | 194 | int main(int argc, char **argv) { |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 195 | Config config(argc, argv); |
halcanary | 8a74f13 | 2016-07-11 14:30:39 -0700 | [diff] [blame] | 196 | SkFILEWStream wStream(config.output_file_name.value.c_str()); |
| 197 | sk_sp<SkDocument> doc = MakePDFDocument(config, &wStream); |
| 198 | assert(doc); |
| 199 | Placement placement(&config, doc.get()); |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 200 | |
halcanary | 8a74f13 | 2016-07-11 14:30:39 -0700 | [diff] [blame] | 201 | const std::string &font_file = config.font_file.value; |
| 202 | sk_sp<SkTypeface> typeface; |
| 203 | if (font_file.size() > 0) { |
| 204 | typeface = SkTypeface::MakeFromFile(font_file.c_str(), 0 /* index */); |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 205 | } |
halcanary | 8a74f13 | 2016-07-11 14:30:39 -0700 | [diff] [blame] | 206 | SkShaper shaper(typeface); |
| 207 | assert(shaper.good()); |
| 208 | for (std::string line; std::getline(std::cin, line);) { |
| 209 | placement.WriteLine(shaper, line.c_str(), line.size()); |
| 210 | } |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 211 | |
halcanary | 8a74f13 | 2016-07-11 14:30:39 -0700 | [diff] [blame] | 212 | doc->close(); |
halcanary | 00d44e0 | 2016-05-03 15:09:52 -0700 | [diff] [blame] | 213 | return 0; |
| 214 | } |