1
2 /*
3 * Copyright 2011 Google Inc.
4 *
5 * Use of this source code is governed by a BSD-style license that can be
6 * found in the LICENSE file.
7 */
8
9
10 #include "SkPDFDevice.h"
11
12 #include "SkColor.h"
13 #include "SkClipStack.h"
14 #include "SkData.h"
15 #include "SkDraw.h"
16 #include "SkGlyphCache.h"
17 #include "SkPaint.h"
18 #include "SkPath.h"
19 #include "SkPDFFont.h"
20 #include "SkPDFFormXObject.h"
21 #include "SkPDFGraphicState.h"
22 #include "SkPDFImage.h"
23 #include "SkPDFShader.h"
24 #include "SkPDFStream.h"
25 #include "SkPDFTypes.h"
26 #include "SkPDFUtils.h"
27 #include "SkRect.h"
28 #include "SkString.h"
29 #include "SkTextFormatParams.h"
30 #include "SkTypeface.h"
31 #include "SkTypes.h"
32
33 // Utility functions
34
emit_pdf_color(SkColor color,SkWStream * result)35 static void emit_pdf_color(SkColor color, SkWStream* result) {
36 SkASSERT(SkColorGetA(color) == 0xFF); // We handle alpha elsewhere.
37 SkScalar colorMax = SkIntToScalar(0xFF);
38 SkPDFScalar::Append(
39 SkScalarDiv(SkIntToScalar(SkColorGetR(color)), colorMax), result);
40 result->writeText(" ");
41 SkPDFScalar::Append(
42 SkScalarDiv(SkIntToScalar(SkColorGetG(color)), colorMax), result);
43 result->writeText(" ");
44 SkPDFScalar::Append(
45 SkScalarDiv(SkIntToScalar(SkColorGetB(color)), colorMax), result);
46 result->writeText(" ");
47 }
48
calculate_text_paint(const SkPaint & paint)49 static SkPaint calculate_text_paint(const SkPaint& paint) {
50 SkPaint result = paint;
51 if (result.isFakeBoldText()) {
52 SkScalar fakeBoldScale = SkScalarInterpFunc(result.getTextSize(),
53 kStdFakeBoldInterpKeys,
54 kStdFakeBoldInterpValues,
55 kStdFakeBoldInterpLength);
56 SkScalar width = SkScalarMul(result.getTextSize(), fakeBoldScale);
57 if (result.getStyle() == SkPaint::kFill_Style) {
58 result.setStyle(SkPaint::kStrokeAndFill_Style);
59 } else {
60 width += result.getStrokeWidth();
61 }
62 result.setStrokeWidth(width);
63 }
64 return result;
65 }
66
67 // Stolen from measure_text in SkDraw.cpp and then tweaked.
align_text(SkDrawCacheProc glyphCacheProc,const SkPaint & paint,const uint16_t * glyphs,size_t len,SkScalar * x,SkScalar * y)68 static void align_text(SkDrawCacheProc glyphCacheProc, const SkPaint& paint,
69 const uint16_t* glyphs, size_t len,
70 SkScalar* x, SkScalar* y) {
71 if (paint.getTextAlign() == SkPaint::kLeft_Align) {
72 return;
73 }
74
75 SkMatrix ident;
76 ident.reset();
77 SkAutoGlyphCache autoCache(paint, &ident);
78 SkGlyphCache* cache = autoCache.getCache();
79
80 const char* start = reinterpret_cast<const char*>(glyphs);
81 const char* stop = reinterpret_cast<const char*>(glyphs + len);
82 SkFixed xAdv = 0, yAdv = 0;
83
84 // TODO(vandebo): This probably needs to take kerning into account.
85 while (start < stop) {
86 const SkGlyph& glyph = glyphCacheProc(cache, &start, 0, 0);
87 xAdv += glyph.fAdvanceX;
88 yAdv += glyph.fAdvanceY;
89 };
90 if (paint.getTextAlign() == SkPaint::kLeft_Align) {
91 return;
92 }
93
94 SkScalar xAdj = SkFixedToScalar(xAdv);
95 SkScalar yAdj = SkFixedToScalar(yAdv);
96 if (paint.getTextAlign() == SkPaint::kCenter_Align) {
97 xAdj = SkScalarHalf(xAdj);
98 yAdj = SkScalarHalf(yAdj);
99 }
100 *x = *x - xAdj;
101 *y = *y - yAdj;
102 }
103
set_text_transform(SkScalar x,SkScalar y,SkScalar textSkewX,SkWStream * content)104 static void set_text_transform(SkScalar x, SkScalar y, SkScalar textSkewX,
105 SkWStream* content) {
106 // Flip the text about the x-axis to account for origin swap and include
107 // the passed parameters.
108 content->writeText("1 0 ");
109 SkPDFScalar::Append(0 - textSkewX, content);
110 content->writeText(" -1 ");
111 SkPDFScalar::Append(x, content);
112 content->writeText(" ");
113 SkPDFScalar::Append(y, content);
114 content->writeText(" Tm\n");
115 }
116
117 // It is important to not confuse GraphicStateEntry with SkPDFGraphicState, the
118 // later being our representation of an object in the PDF file.
119 struct GraphicStateEntry {
120 GraphicStateEntry();
121
122 // Compare the fields we care about when setting up a new content entry.
123 bool compareInitialState(const GraphicStateEntry& b);
124
125 SkMatrix fMatrix;
126 // We can't do set operations on Paths, though PDF natively supports
127 // intersect. If the clip stack does anything other than intersect,
128 // we have to fall back to the region. Treat fClipStack as authoritative.
129 // See http://code.google.com/p/skia/issues/detail?id=221
130 SkClipStack fClipStack;
131 SkRegion fClipRegion;
132
133 // When emitting the content entry, we will ensure the graphic state
134 // is set to these values first.
135 SkColor fColor;
136 SkScalar fTextScaleX; // Zero means we don't care what the value is.
137 SkPaint::Style fTextFill; // Only if TextScaleX is non-zero.
138 int fShaderIndex;
139 int fGraphicStateIndex;
140
141 // We may change the font (i.e. for Type1 support) within a
142 // ContentEntry. This is the one currently in effect, or NULL if none.
143 SkPDFFont* fFont;
144 // In PDF, text size has no default value. It is only valid if fFont is
145 // not NULL.
146 SkScalar fTextSize;
147 };
148
GraphicStateEntry()149 GraphicStateEntry::GraphicStateEntry() : fColor(SK_ColorBLACK),
150 fTextScaleX(SK_Scalar1),
151 fTextFill(SkPaint::kFill_Style),
152 fShaderIndex(-1),
153 fGraphicStateIndex(-1),
154 fFont(NULL),
155 fTextSize(SK_ScalarNaN) {
156 fMatrix.reset();
157 }
158
compareInitialState(const GraphicStateEntry & b)159 bool GraphicStateEntry::compareInitialState(const GraphicStateEntry& b) {
160 return fColor == b.fColor &&
161 fShaderIndex == b.fShaderIndex &&
162 fGraphicStateIndex == b.fGraphicStateIndex &&
163 fMatrix == b.fMatrix &&
164 fClipStack == b.fClipStack &&
165 (fTextScaleX == 0 ||
166 b.fTextScaleX == 0 ||
167 (fTextScaleX == b.fTextScaleX && fTextFill == b.fTextFill));
168 }
169
170 class GraphicStackState {
171 public:
GraphicStackState(const SkClipStack & existingClipStack,const SkRegion & existingClipRegion,SkWStream * contentStream)172 GraphicStackState(const SkClipStack& existingClipStack,
173 const SkRegion& existingClipRegion,
174 SkWStream* contentStream)
175 : fStackDepth(0),
176 fContentStream(contentStream) {
177 fEntries[0].fClipStack = existingClipStack;
178 fEntries[0].fClipRegion = existingClipRegion;
179 }
180
181 void updateClip(const SkClipStack& clipStack, const SkRegion& clipRegion,
182 const SkPoint& translation);
183 void updateMatrix(const SkMatrix& matrix);
184 void updateDrawingState(const GraphicStateEntry& state);
185
186 void drainStack();
187
188 private:
189 void push();
190 void pop();
currentEntry()191 GraphicStateEntry* currentEntry() { return &fEntries[fStackDepth]; }
192
193 // Conservative limit on save depth, see impl. notes in PDF 1.4 spec.
194 static const int kMaxStackDepth = 12;
195 GraphicStateEntry fEntries[kMaxStackDepth + 1];
196 int fStackDepth;
197 SkWStream* fContentStream;
198 };
199
drainStack()200 void GraphicStackState::drainStack() {
201 while (fStackDepth) {
202 pop();
203 }
204 }
205
push()206 void GraphicStackState::push() {
207 SkASSERT(fStackDepth < kMaxStackDepth);
208 fContentStream->writeText("q\n");
209 fStackDepth++;
210 fEntries[fStackDepth] = fEntries[fStackDepth - 1];
211 }
212
pop()213 void GraphicStackState::pop() {
214 SkASSERT(fStackDepth > 0);
215 fContentStream->writeText("Q\n");
216 fStackDepth--;
217 }
218
219 // This function initializes iter to be an interator on the "stack" argument
220 // and then skips over the leading entries as specified in prefix. It requires
221 // and asserts that "prefix" will be a prefix to "stack."
skip_clip_stack_prefix(const SkClipStack & prefix,const SkClipStack & stack,SkClipStack::B2FIter * iter)222 static void skip_clip_stack_prefix(const SkClipStack& prefix,
223 const SkClipStack& stack,
224 SkClipStack::B2FIter* iter) {
225 SkClipStack::B2FIter prefixIter(prefix);
226 iter->reset(stack);
227
228 const SkClipStack::B2FIter::Clip* prefixEntry;
229 const SkClipStack::B2FIter::Clip* iterEntry;
230
231 int count = 0;
232 for (prefixEntry = prefixIter.next(); prefixEntry;
233 prefixEntry = prefixIter.next(), count++) {
234 iterEntry = iter->next();
235 SkASSERT(iterEntry);
236 // Because of SkClipStack does internal intersection, the last clip
237 // entry may differ.
238 if (*prefixEntry != *iterEntry) {
239 SkASSERT(prefixEntry->fOp == SkRegion::kIntersect_Op);
240 SkASSERT(iterEntry->fOp == SkRegion::kIntersect_Op);
241 SkASSERT((iterEntry->fRect == NULL) ==
242 (prefixEntry->fRect == NULL));
243 SkASSERT((iterEntry->fPath == NULL) ==
244 (prefixEntry->fPath == NULL));
245 // We need to back up the iterator by one but don't have that
246 // function, so reset and go forward by one less.
247 iter->reset(stack);
248 for (int i = 0; i < count; i++) {
249 iter->next();
250 }
251 prefixEntry = prefixIter.next();
252 break;
253 }
254 }
255
256 SkASSERT(prefixEntry == NULL);
257 }
258
emit_clip(SkPath * clipPath,SkRect * clipRect,SkWStream * contentStream)259 static void emit_clip(SkPath* clipPath, SkRect* clipRect,
260 SkWStream* contentStream) {
261 SkASSERT(clipPath || clipRect);
262
263 SkPath::FillType clipFill;
264 if (clipPath) {
265 SkPDFUtils::EmitPath(*clipPath, contentStream);
266 clipFill = clipPath->getFillType();
267 } else {
268 SkPDFUtils::AppendRectangle(*clipRect, contentStream);
269 clipFill = SkPath::kWinding_FillType;
270 }
271
272 NOT_IMPLEMENTED(clipFill == SkPath::kInverseEvenOdd_FillType, false);
273 NOT_IMPLEMENTED(clipFill == SkPath::kInverseWinding_FillType, false);
274 if (clipFill == SkPath::kEvenOdd_FillType) {
275 contentStream->writeText("W* n\n");
276 } else {
277 contentStream->writeText("W n\n");
278 }
279 }
280
281 // TODO(vandebo): Take advantage of SkClipStack::getSaveCount(), the PDF
282 // graphic state stack, and the fact that we can know all the clips used
283 // on the page to optimize this.
updateClip(const SkClipStack & clipStack,const SkRegion & clipRegion,const SkPoint & translation)284 void GraphicStackState::updateClip(const SkClipStack& clipStack,
285 const SkRegion& clipRegion,
286 const SkPoint& translation) {
287 if (clipStack == currentEntry()->fClipStack) {
288 return;
289 }
290
291 while (fStackDepth > 0) {
292 pop();
293 if (clipStack == currentEntry()->fClipStack) {
294 return;
295 }
296 }
297 push();
298
299 // gsState->initialEntry()->fClipStack/Region specifies the clip that has
300 // already been applied. (If this is a top level device, then it specifies
301 // a clip to the content area. If this is a layer, then it specifies
302 // the clip in effect when the layer was created.) There's no need to
303 // reapply that clip; SKCanvas's SkDrawIter will draw anything outside the
304 // initial clip on the parent layer. (This means there's a bug if the user
305 // expands the clip and then uses any xfer mode that uses dst:
306 // http://code.google.com/p/skia/issues/detail?id=228 )
307 SkClipStack::B2FIter iter;
308 skip_clip_stack_prefix(fEntries[0].fClipStack, clipStack, &iter);
309
310 // If the clip stack does anything other than intersect or if it uses
311 // an inverse fill type, we have to fall back to the clip region.
312 bool needRegion = false;
313 const SkClipStack::B2FIter::Clip* clipEntry;
314 for (clipEntry = iter.next(); clipEntry; clipEntry = iter.next()) {
315 if (clipEntry->fOp != SkRegion::kIntersect_Op ||
316 (clipEntry->fPath && clipEntry->fPath->isInverseFillType())) {
317 needRegion = true;
318 break;
319 }
320 }
321
322 if (needRegion) {
323 SkPath clipPath;
324 SkAssertResult(clipRegion.getBoundaryPath(&clipPath));
325 emit_clip(&clipPath, NULL, fContentStream);
326 } else {
327 skip_clip_stack_prefix(fEntries[0].fClipStack, clipStack, &iter);
328 SkMatrix transform;
329 transform.setTranslate(translation.fX, translation.fY);
330 const SkClipStack::B2FIter::Clip* clipEntry;
331 for (clipEntry = iter.next(); clipEntry; clipEntry = iter.next()) {
332 SkASSERT(clipEntry->fOp == SkRegion::kIntersect_Op);
333 if (clipEntry->fRect) {
334 SkRect translatedClip;
335 transform.mapRect(&translatedClip, *clipEntry->fRect);
336 emit_clip(NULL, &translatedClip, fContentStream);
337 } else if (clipEntry->fPath) {
338 SkPath translatedPath;
339 clipEntry->fPath->transform(transform, &translatedPath);
340 emit_clip(&translatedPath, NULL, fContentStream);
341 } else {
342 SkASSERT(false);
343 }
344 }
345 }
346 currentEntry()->fClipStack = clipStack;
347 currentEntry()->fClipRegion = clipRegion;
348 }
349
updateMatrix(const SkMatrix & matrix)350 void GraphicStackState::updateMatrix(const SkMatrix& matrix) {
351 if (matrix == currentEntry()->fMatrix) {
352 return;
353 }
354
355 if (currentEntry()->fMatrix.getType() != SkMatrix::kIdentity_Mask) {
356 SkASSERT(fStackDepth > 0);
357 SkASSERT(fEntries[fStackDepth].fClipStack ==
358 fEntries[fStackDepth -1].fClipStack);
359 pop();
360
361 SkASSERT(currentEntry()->fMatrix.getType() == SkMatrix::kIdentity_Mask);
362 }
363 if (matrix.getType() == SkMatrix::kIdentity_Mask) {
364 return;
365 }
366
367 push();
368 SkPDFUtils::AppendTransform(matrix, fContentStream);
369 currentEntry()->fMatrix = matrix;
370 }
371
updateDrawingState(const GraphicStateEntry & state)372 void GraphicStackState::updateDrawingState(const GraphicStateEntry& state) {
373 // PDF treats a shader as a color, so we only set one or the other.
374 if (state.fShaderIndex >= 0) {
375 if (state.fShaderIndex != currentEntry()->fShaderIndex) {
376 fContentStream->writeText("/Pattern CS /Pattern cs /P");
377 fContentStream->writeDecAsText(state.fShaderIndex);
378 fContentStream->writeText(" SCN /P");
379 fContentStream->writeDecAsText(state.fShaderIndex);
380 fContentStream->writeText(" scn\n");
381 currentEntry()->fShaderIndex = state.fShaderIndex;
382 }
383 } else {
384 if (state.fColor != currentEntry()->fColor ||
385 currentEntry()->fShaderIndex >= 0) {
386 emit_pdf_color(state.fColor, fContentStream);
387 fContentStream->writeText("RG ");
388 emit_pdf_color(state.fColor, fContentStream);
389 fContentStream->writeText("rg\n");
390 currentEntry()->fColor = state.fColor;
391 currentEntry()->fShaderIndex = -1;
392 }
393 }
394
395 if (state.fGraphicStateIndex != currentEntry()->fGraphicStateIndex) {
396 SkPDFUtils::ApplyGraphicState(state.fGraphicStateIndex, fContentStream);
397 currentEntry()->fGraphicStateIndex = state.fGraphicStateIndex;
398 }
399
400 if (state.fTextScaleX) {
401 if (state.fTextScaleX != currentEntry()->fTextScaleX) {
402 SkScalar pdfScale = SkScalarMul(state.fTextScaleX,
403 SkIntToScalar(100));
404 SkPDFScalar::Append(pdfScale, fContentStream);
405 fContentStream->writeText(" Tz\n");
406 currentEntry()->fTextScaleX = state.fTextScaleX;
407 }
408 if (state.fTextFill != currentEntry()->fTextFill) {
409 SK_COMPILE_ASSERT(SkPaint::kFill_Style == 0, enum_must_match_value);
410 SK_COMPILE_ASSERT(SkPaint::kStroke_Style == 1,
411 enum_must_match_value);
412 SK_COMPILE_ASSERT(SkPaint::kStrokeAndFill_Style == 2,
413 enum_must_match_value);
414 fContentStream->writeDecAsText(state.fTextFill);
415 fContentStream->writeText(" Tr\n");
416 currentEntry()->fTextFill = state.fTextFill;
417 }
418 }
419 }
420
onCreateCompatibleDevice(SkBitmap::Config config,int width,int height,bool isOpaque,Usage usage)421 SkDevice* SkPDFDevice::onCreateCompatibleDevice(SkBitmap::Config config,
422 int width, int height,
423 bool isOpaque,
424 Usage usage) {
425 SkMatrix initialTransform;
426 initialTransform.reset();
427 SkISize size = SkISize::Make(width, height);
428 return SkNEW_ARGS(SkPDFDevice, (size, size, initialTransform));
429 }
430
431
432 struct ContentEntry {
433 GraphicStateEntry fState;
434 SkDynamicMemoryWStream fContent;
435 SkTScopedPtr<ContentEntry> fNext;
436 };
437
438 // A helper class to automatically finish a ContentEntry at the end of a
439 // drawing method and maintain the state needed between set up and finish.
440 class ScopedContentEntry {
441 public:
ScopedContentEntry(SkPDFDevice * device,const SkDraw & draw,const SkPaint & paint,bool hasText=false)442 ScopedContentEntry(SkPDFDevice* device, const SkDraw& draw,
443 const SkPaint& paint, bool hasText = false)
444 : fDevice(device),
445 fContentEntry(NULL),
446 fXfermode(SkXfermode::kSrcOver_Mode) {
447 init(draw.fClipStack, *draw.fClip, *draw.fMatrix, paint, hasText);
448 }
ScopedContentEntry(SkPDFDevice * device,const SkClipStack * clipStack,const SkRegion & clipRegion,const SkMatrix & matrix,const SkPaint & paint,bool hasText=false)449 ScopedContentEntry(SkPDFDevice* device, const SkClipStack* clipStack,
450 const SkRegion& clipRegion, const SkMatrix& matrix,
451 const SkPaint& paint, bool hasText = false)
452 : fDevice(device),
453 fContentEntry(NULL),
454 fXfermode(SkXfermode::kSrcOver_Mode) {
455 init(clipStack, clipRegion, matrix, paint, hasText);
456 }
457
~ScopedContentEntry()458 ~ScopedContentEntry() {
459 if (fContentEntry) {
460 fDevice->finishContentEntry(fXfermode, fDstFormXObject.get());
461 }
462 }
463
entry()464 ContentEntry* entry() { return fContentEntry; }
465 private:
466 SkPDFDevice* fDevice;
467 ContentEntry* fContentEntry;
468 SkXfermode::Mode fXfermode;
469 SkRefPtr<SkPDFFormXObject> fDstFormXObject;
470
init(const SkClipStack * clipStack,const SkRegion & clipRegion,const SkMatrix & matrix,const SkPaint & paint,bool hasText)471 void init(const SkClipStack* clipStack, const SkRegion& clipRegion,
472 const SkMatrix& matrix, const SkPaint& paint, bool hasText) {
473 if (paint.getXfermode()) {
474 paint.getXfermode()->asMode(&fXfermode);
475 }
476 fContentEntry = fDevice->setUpContentEntry(clipStack, clipRegion,
477 matrix, paint, hasText,
478 &fDstFormXObject);
479 }
480 };
481
482 ////////////////////////////////////////////////////////////////////////////////
483
makeContentBitmap(const SkISize & contentSize,const SkMatrix * initialTransform)484 static inline SkBitmap makeContentBitmap(const SkISize& contentSize,
485 const SkMatrix* initialTransform) {
486 SkBitmap bitmap;
487 if (initialTransform) {
488 // Compute the size of the drawing area.
489 SkVector drawingSize;
490 SkMatrix inverse;
491 drawingSize.set(SkIntToScalar(contentSize.fWidth),
492 SkIntToScalar(contentSize.fHeight));
493 initialTransform->invert(&inverse);
494 inverse.mapVectors(&drawingSize, 1);
495 SkISize size = SkSize::Make(drawingSize.fX, drawingSize.fY).toRound();
496 bitmap.setConfig(SkBitmap::kNo_Config, abs(size.fWidth),
497 abs(size.fHeight));
498 } else {
499 bitmap.setConfig(SkBitmap::kNo_Config, abs(contentSize.fWidth),
500 abs(contentSize.fHeight));
501 }
502
503 return bitmap;
504 }
505
506 // TODO(vandebo) change pageSize to SkSize.
SkPDFDevice(const SkISize & pageSize,const SkISize & contentSize,const SkMatrix & initialTransform)507 SkPDFDevice::SkPDFDevice(const SkISize& pageSize, const SkISize& contentSize,
508 const SkMatrix& initialTransform)
509 : SkDevice(makeContentBitmap(contentSize, &initialTransform)),
510 fPageSize(pageSize),
511 fContentSize(contentSize),
512 fLastContentEntry(NULL),
513 fLastMarginContentEntry(NULL) {
514 // Skia generally uses the top left as the origin but PDF natively has the
515 // origin at the bottom left. This matrix corrects for that. But that only
516 // needs to be done once, we don't do it when layering.
517 fInitialTransform.setTranslate(0, SkIntToScalar(pageSize.fHeight));
518 fInitialTransform.preScale(SK_Scalar1, -SK_Scalar1);
519 fInitialTransform.preConcat(initialTransform);
520
521 SkIRect existingClip = SkIRect::MakeWH(this->width(), this->height());
522 fExistingClipRegion.setRect(existingClip);
523
524 this->init();
525 }
526
527 // TODO(vandebo) change layerSize to SkSize.
SkPDFDevice(const SkISize & layerSize,const SkClipStack & existingClipStack,const SkRegion & existingClipRegion)528 SkPDFDevice::SkPDFDevice(const SkISize& layerSize,
529 const SkClipStack& existingClipStack,
530 const SkRegion& existingClipRegion)
531 : SkDevice(makeContentBitmap(layerSize, NULL)),
532 fPageSize(layerSize),
533 fContentSize(layerSize),
534 fExistingClipStack(existingClipStack),
535 fExistingClipRegion(existingClipRegion),
536 fLastContentEntry(NULL),
537 fLastMarginContentEntry(NULL) {
538 fInitialTransform.reset();
539 this->init();
540 }
541
~SkPDFDevice()542 SkPDFDevice::~SkPDFDevice() {
543 this->cleanUp(true);
544 }
545
init()546 void SkPDFDevice::init() {
547 fResourceDict = NULL;
548 fContentEntries.reset();
549 fLastContentEntry = NULL;
550 fMarginContentEntries.reset();
551 fLastMarginContentEntry = NULL;
552 fDrawingArea = kContent_DrawingArea;
553 if (fFontGlyphUsage == NULL) {
554 fFontGlyphUsage.reset(new SkPDFGlyphSetMap());
555 }
556 }
557
cleanUp(bool clearFontUsage)558 void SkPDFDevice::cleanUp(bool clearFontUsage) {
559 fGraphicStateResources.unrefAll();
560 fXObjectResources.unrefAll();
561 fFontResources.unrefAll();
562 fShaderResources.unrefAll();
563 if (clearFontUsage) {
564 fFontGlyphUsage->reset();
565 }
566 }
567
getDeviceCapabilities()568 uint32_t SkPDFDevice::getDeviceCapabilities() {
569 return kVector_Capability;
570 }
571
clear(SkColor color)572 void SkPDFDevice::clear(SkColor color) {
573 this->cleanUp(true);
574 this->init();
575
576 SkPaint paint;
577 paint.setColor(color);
578 paint.setStyle(SkPaint::kFill_Style);
579 SkMatrix identity;
580 identity.reset();
581 ScopedContentEntry content(this, &fExistingClipStack, fExistingClipRegion,
582 identity, paint);
583 internalDrawPaint(paint, content.entry());
584 }
585
drawPaint(const SkDraw & d,const SkPaint & paint)586 void SkPDFDevice::drawPaint(const SkDraw& d, const SkPaint& paint) {
587 SkPaint newPaint = paint;
588 newPaint.setStyle(SkPaint::kFill_Style);
589 ScopedContentEntry content(this, d, newPaint);
590 internalDrawPaint(newPaint, content.entry());
591 }
592
internalDrawPaint(const SkPaint & paint,ContentEntry * contentEntry)593 void SkPDFDevice::internalDrawPaint(const SkPaint& paint,
594 ContentEntry* contentEntry) {
595 if (!contentEntry) {
596 return;
597 }
598 SkRect bbox = SkRect::MakeWH(SkIntToScalar(this->width()),
599 SkIntToScalar(this->height()));
600 SkMatrix totalTransform = fInitialTransform;
601 totalTransform.preConcat(contentEntry->fState.fMatrix);
602 SkMatrix inverse;
603 inverse.reset();
604 totalTransform.invert(&inverse);
605 inverse.mapRect(&bbox);
606
607 SkPDFUtils::AppendRectangle(bbox, &contentEntry->fContent);
608 SkPDFUtils::PaintPath(paint.getStyle(), SkPath::kWinding_FillType,
609 &contentEntry->fContent);
610 }
611
drawPoints(const SkDraw & d,SkCanvas::PointMode mode,size_t count,const SkPoint * points,const SkPaint & passedPaint)612 void SkPDFDevice::drawPoints(const SkDraw& d, SkCanvas::PointMode mode,
613 size_t count, const SkPoint* points,
614 const SkPaint& passedPaint) {
615 if (count == 0) {
616 return;
617 }
618
619 // SkDraw::drawPoints converts to multiple calls to fDevice->drawPath.
620 // We only use this when there's a path effect because of the overhead
621 // of multiple calls to setUpContentEntry it causes.
622 if (passedPaint.getPathEffect()) {
623 if (d.fClip->isEmpty()) {
624 return;
625 }
626 SkDraw pointDraw(d);
627 pointDraw.fDevice = this;
628 pointDraw.drawPoints(mode, count, points, passedPaint, true);
629 return;
630 }
631
632 const SkPaint* paint = &passedPaint;
633 SkPaint modifiedPaint;
634
635 if (mode == SkCanvas::kPoints_PointMode &&
636 paint->getStrokeCap() != SkPaint::kRound_Cap) {
637 modifiedPaint = *paint;
638 paint = &modifiedPaint;
639 if (paint->getStrokeWidth()) {
640 // PDF won't draw a single point with square/butt caps because the
641 // orientation is ambiguous. Draw a rectangle instead.
642 modifiedPaint.setStyle(SkPaint::kFill_Style);
643 SkScalar strokeWidth = paint->getStrokeWidth();
644 SkScalar halfStroke = SkScalarHalf(strokeWidth);
645 for (size_t i = 0; i < count; i++) {
646 SkRect r = SkRect::MakeXYWH(points[i].fX, points[i].fY, 0, 0);
647 r.inset(-halfStroke, -halfStroke);
648 drawRect(d, r, modifiedPaint);
649 }
650 return;
651 } else {
652 modifiedPaint.setStrokeCap(SkPaint::kRound_Cap);
653 }
654 }
655
656 ScopedContentEntry content(this, d, *paint);
657 if (!content.entry()) {
658 return;
659 }
660
661 switch (mode) {
662 case SkCanvas::kPolygon_PointMode:
663 SkPDFUtils::MoveTo(points[0].fX, points[0].fY,
664 &content.entry()->fContent);
665 for (size_t i = 1; i < count; i++) {
666 SkPDFUtils::AppendLine(points[i].fX, points[i].fY,
667 &content.entry()->fContent);
668 }
669 SkPDFUtils::StrokePath(&content.entry()->fContent);
670 break;
671 case SkCanvas::kLines_PointMode:
672 for (size_t i = 0; i < count/2; i++) {
673 SkPDFUtils::MoveTo(points[i * 2].fX, points[i * 2].fY,
674 &content.entry()->fContent);
675 SkPDFUtils::AppendLine(points[i * 2 + 1].fX,
676 points[i * 2 + 1].fY,
677 &content.entry()->fContent);
678 SkPDFUtils::StrokePath(&content.entry()->fContent);
679 }
680 break;
681 case SkCanvas::kPoints_PointMode:
682 SkASSERT(paint->getStrokeCap() == SkPaint::kRound_Cap);
683 for (size_t i = 0; i < count; i++) {
684 SkPDFUtils::MoveTo(points[i].fX, points[i].fY,
685 &content.entry()->fContent);
686 SkPDFUtils::ClosePath(&content.entry()->fContent);
687 SkPDFUtils::StrokePath(&content.entry()->fContent);
688 }
689 break;
690 default:
691 SkASSERT(false);
692 }
693 }
694
drawRect(const SkDraw & d,const SkRect & r,const SkPaint & paint)695 void SkPDFDevice::drawRect(const SkDraw& d, const SkRect& r,
696 const SkPaint& paint) {
697 if (paint.getPathEffect()) {
698 if (d.fClip->isEmpty()) {
699 return;
700 }
701 SkPath path;
702 path.addRect(r);
703 drawPath(d, path, paint, NULL, true);
704 return;
705 }
706
707 ScopedContentEntry content(this, d, paint);
708 if (!content.entry()) {
709 return;
710 }
711 SkPDFUtils::AppendRectangle(r, &content.entry()->fContent);
712 SkPDFUtils::PaintPath(paint.getStyle(), SkPath::kWinding_FillType,
713 &content.entry()->fContent);
714 }
715
drawPath(const SkDraw & d,const SkPath & origPath,const SkPaint & paint,const SkMatrix * prePathMatrix,bool pathIsMutable)716 void SkPDFDevice::drawPath(const SkDraw& d, const SkPath& origPath,
717 const SkPaint& paint, const SkMatrix* prePathMatrix,
718 bool pathIsMutable) {
719 SkPath modifiedPath;
720 SkPath* pathPtr = const_cast<SkPath*>(&origPath);
721
722 SkMatrix matrix = *d.fMatrix;
723 if (prePathMatrix) {
724 if (paint.getPathEffect() || paint.getStyle() != SkPaint::kFill_Style) {
725 if (!pathIsMutable) {
726 pathPtr = &modifiedPath;
727 pathIsMutable = true;
728 }
729 origPath.transform(*prePathMatrix, pathPtr);
730 } else {
731 if (!matrix.preConcat(*prePathMatrix)) {
732 return;
733 }
734 }
735 }
736
737 if (paint.getPathEffect()) {
738 if (d.fClip->isEmpty()) {
739 return;
740 }
741 if (!pathIsMutable) {
742 pathPtr = &modifiedPath;
743 pathIsMutable = true;
744 }
745 bool fill = paint.getFillPath(origPath, pathPtr);
746
747 SkPaint noEffectPaint(paint);
748 noEffectPaint.setPathEffect(NULL);
749 if (fill) {
750 noEffectPaint.setStyle(SkPaint::kFill_Style);
751 } else {
752 noEffectPaint.setStyle(SkPaint::kStroke_Style);
753 noEffectPaint.setStrokeWidth(0);
754 }
755 drawPath(d, *pathPtr, noEffectPaint, NULL, true);
756 return;
757 }
758
759 ScopedContentEntry content(this, d, paint);
760 if (!content.entry()) {
761 return;
762 }
763 SkPDFUtils::EmitPath(*pathPtr, &content.entry()->fContent);
764 SkPDFUtils::PaintPath(paint.getStyle(), pathPtr->getFillType(),
765 &content.entry()->fContent);
766 }
767
drawBitmap(const SkDraw & d,const SkBitmap & bitmap,const SkIRect * srcRect,const SkMatrix & matrix,const SkPaint & paint)768 void SkPDFDevice::drawBitmap(const SkDraw& d, const SkBitmap& bitmap,
769 const SkIRect* srcRect, const SkMatrix& matrix,
770 const SkPaint& paint) {
771 if (d.fClip->isEmpty()) {
772 return;
773 }
774
775 SkMatrix transform = matrix;
776 transform.postConcat(*d.fMatrix);
777 internalDrawBitmap(transform, d.fClipStack, *d.fClip, bitmap, srcRect,
778 paint);
779 }
780
drawSprite(const SkDraw & d,const SkBitmap & bitmap,int x,int y,const SkPaint & paint)781 void SkPDFDevice::drawSprite(const SkDraw& d, const SkBitmap& bitmap,
782 int x, int y, const SkPaint& paint) {
783 if (d.fClip->isEmpty()) {
784 return;
785 }
786
787 SkMatrix matrix;
788 matrix.setTranslate(SkIntToScalar(x), SkIntToScalar(y));
789 internalDrawBitmap(matrix, d.fClipStack, *d.fClip, bitmap, NULL, paint);
790 }
791
drawText(const SkDraw & d,const void * text,size_t len,SkScalar x,SkScalar y,const SkPaint & paint)792 void SkPDFDevice::drawText(const SkDraw& d, const void* text, size_t len,
793 SkScalar x, SkScalar y, const SkPaint& paint) {
794 SkPaint textPaint = calculate_text_paint(paint);
795 ScopedContentEntry content(this, d, textPaint, true);
796 if (!content.entry()) {
797 return;
798 }
799
800 // We want the text in glyph id encoding and a writable buffer, so we end
801 // up making a copy either way.
802 size_t numGlyphs = paint.textToGlyphs(text, len, NULL);
803 uint16_t* glyphIDs = reinterpret_cast<uint16_t*>(
804 sk_malloc_flags(numGlyphs * 2, SK_MALLOC_TEMP | SK_MALLOC_THROW));
805 SkAutoFree autoFreeGlyphIDs(glyphIDs);
806 if (paint.getTextEncoding() != SkPaint::kGlyphID_TextEncoding) {
807 paint.textToGlyphs(text, len, glyphIDs);
808 textPaint.setTextEncoding(SkPaint::kGlyphID_TextEncoding);
809 } else {
810 SkASSERT((len & 1) == 0);
811 SkASSERT(len / 2 == numGlyphs);
812 memcpy(glyphIDs, text, len);
813 }
814
815 SkDrawCacheProc glyphCacheProc = textPaint.getDrawCacheProc();
816 align_text(glyphCacheProc, textPaint, glyphIDs, numGlyphs, &x, &y);
817 content.entry()->fContent.writeText("BT\n");
818 set_text_transform(x, y, textPaint.getTextSkewX(),
819 &content.entry()->fContent);
820 size_t consumedGlyphCount = 0;
821 while (numGlyphs > consumedGlyphCount) {
822 updateFont(textPaint, glyphIDs[consumedGlyphCount], content.entry());
823 SkPDFFont* font = content.entry()->fState.fFont;
824 size_t availableGlyphs =
825 font->glyphsToPDFFontEncoding(glyphIDs + consumedGlyphCount,
826 numGlyphs - consumedGlyphCount);
827 fFontGlyphUsage->noteGlyphUsage(font, glyphIDs + consumedGlyphCount,
828 availableGlyphs);
829 SkString encodedString =
830 SkPDFString::FormatString(glyphIDs + consumedGlyphCount,
831 availableGlyphs, font->multiByteGlyphs());
832 content.entry()->fContent.writeText(encodedString.c_str());
833 consumedGlyphCount += availableGlyphs;
834 content.entry()->fContent.writeText(" Tj\n");
835 }
836 content.entry()->fContent.writeText("ET\n");
837 }
838
drawPosText(const SkDraw & d,const void * text,size_t len,const SkScalar pos[],SkScalar constY,int scalarsPerPos,const SkPaint & paint)839 void SkPDFDevice::drawPosText(const SkDraw& d, const void* text, size_t len,
840 const SkScalar pos[], SkScalar constY,
841 int scalarsPerPos, const SkPaint& paint) {
842 SkASSERT(1 == scalarsPerPos || 2 == scalarsPerPos);
843 SkPaint textPaint = calculate_text_paint(paint);
844 ScopedContentEntry content(this, d, textPaint, true);
845 if (!content.entry()) {
846 return;
847 }
848
849 // Make sure we have a glyph id encoding.
850 SkAutoFree glyphStorage;
851 uint16_t* glyphIDs;
852 size_t numGlyphs;
853 if (paint.getTextEncoding() != SkPaint::kGlyphID_TextEncoding) {
854 numGlyphs = paint.textToGlyphs(text, len, NULL);
855 glyphIDs = reinterpret_cast<uint16_t*>(sk_malloc_flags(
856 numGlyphs * 2, SK_MALLOC_TEMP | SK_MALLOC_THROW));
857 glyphStorage.set(glyphIDs);
858 paint.textToGlyphs(text, len, glyphIDs);
859 textPaint.setTextEncoding(SkPaint::kGlyphID_TextEncoding);
860 } else {
861 SkASSERT((len & 1) == 0);
862 numGlyphs = len / 2;
863 glyphIDs = reinterpret_cast<uint16_t*>(const_cast<void*>((text)));
864 }
865
866 SkDrawCacheProc glyphCacheProc = textPaint.getDrawCacheProc();
867 content.entry()->fContent.writeText("BT\n");
868 updateFont(textPaint, glyphIDs[0], content.entry());
869 for (size_t i = 0; i < numGlyphs; i++) {
870 SkPDFFont* font = content.entry()->fState.fFont;
871 uint16_t encodedValue = glyphIDs[i];
872 if (font->glyphsToPDFFontEncoding(&encodedValue, 1) != 1) {
873 updateFont(textPaint, glyphIDs[i], content.entry());
874 i--;
875 continue;
876 }
877 fFontGlyphUsage->noteGlyphUsage(font, &encodedValue, 1);
878 SkScalar x = pos[i * scalarsPerPos];
879 SkScalar y = scalarsPerPos == 1 ? constY : pos[i * scalarsPerPos + 1];
880 align_text(glyphCacheProc, textPaint, glyphIDs + i, 1, &x, &y);
881 set_text_transform(x, y, textPaint.getTextSkewX(),
882 &content.entry()->fContent);
883 SkString encodedString =
884 SkPDFString::FormatString(&encodedValue, 1,
885 font->multiByteGlyphs());
886 content.entry()->fContent.writeText(encodedString.c_str());
887 content.entry()->fContent.writeText(" Tj\n");
888 }
889 content.entry()->fContent.writeText("ET\n");
890 }
891
drawTextOnPath(const SkDraw & d,const void * text,size_t len,const SkPath & path,const SkMatrix * matrix,const SkPaint & paint)892 void SkPDFDevice::drawTextOnPath(const SkDraw& d, const void* text, size_t len,
893 const SkPath& path, const SkMatrix* matrix,
894 const SkPaint& paint) {
895 if (d.fClip->isEmpty()) {
896 return;
897 }
898 d.drawTextOnPath((const char*)text, len, path, matrix, paint);
899 }
900
drawVertices(const SkDraw & d,SkCanvas::VertexMode,int vertexCount,const SkPoint verts[],const SkPoint texs[],const SkColor colors[],SkXfermode * xmode,const uint16_t indices[],int indexCount,const SkPaint & paint)901 void SkPDFDevice::drawVertices(const SkDraw& d, SkCanvas::VertexMode,
902 int vertexCount, const SkPoint verts[],
903 const SkPoint texs[], const SkColor colors[],
904 SkXfermode* xmode, const uint16_t indices[],
905 int indexCount, const SkPaint& paint) {
906 if (d.fClip->isEmpty()) {
907 return;
908 }
909 NOT_IMPLEMENTED("drawVerticies", true);
910 }
911
drawDevice(const SkDraw & d,SkDevice * device,int x,int y,const SkPaint & paint)912 void SkPDFDevice::drawDevice(const SkDraw& d, SkDevice* device, int x, int y,
913 const SkPaint& paint) {
914 if ((device->getDeviceCapabilities() & kVector_Capability) == 0) {
915 // If we somehow get a raster device, do what our parent would do.
916 SkDevice::drawDevice(d, device, x, y, paint);
917 return;
918 }
919
920 // Assume that a vector capable device means that it's a PDF Device.
921 SkPDFDevice* pdfDevice = static_cast<SkPDFDevice*>(device);
922 if (pdfDevice->isContentEmpty()) {
923 return;
924 }
925
926 SkMatrix matrix;
927 matrix.setTranslate(SkIntToScalar(x), SkIntToScalar(y));
928 ScopedContentEntry content(this, d.fClipStack, *d.fClip, matrix, paint);
929 if (!content.entry()) {
930 return;
931 }
932
933 SkPDFFormXObject* xobject = new SkPDFFormXObject(pdfDevice);
934 fXObjectResources.push(xobject); // Transfer reference.
935 SkPDFUtils::DrawFormXObject(fXObjectResources.count() - 1,
936 &content.entry()->fContent);
937
938 // Merge glyph sets from the drawn device.
939 fFontGlyphUsage->merge(pdfDevice->getFontGlyphUsage());
940 }
941
getLastContentEntry()942 ContentEntry* SkPDFDevice::getLastContentEntry() {
943 if (fDrawingArea == kContent_DrawingArea) {
944 return fLastContentEntry;
945 } else {
946 return fLastMarginContentEntry;
947 }
948 }
949
getContentEntries()950 SkTScopedPtr<ContentEntry>* SkPDFDevice::getContentEntries() {
951 if (fDrawingArea == kContent_DrawingArea) {
952 return &fContentEntries;
953 } else {
954 return &fMarginContentEntries;
955 }
956 }
957
setLastContentEntry(ContentEntry * contentEntry)958 void SkPDFDevice::setLastContentEntry(ContentEntry* contentEntry) {
959 if (fDrawingArea == kContent_DrawingArea) {
960 fLastContentEntry = contentEntry;
961 } else {
962 fLastMarginContentEntry = contentEntry;
963 }
964 }
965
setDrawingArea(DrawingArea drawingArea)966 void SkPDFDevice::setDrawingArea(DrawingArea drawingArea) {
967 // A ScopedContentEntry only exists during the course of a draw call, so
968 // this can't be called while a ScopedContentEntry exists.
969 fDrawingArea = drawingArea;
970 }
971
getResourceDict()972 SkPDFDict* SkPDFDevice::getResourceDict() {
973 if (fResourceDict.get() == NULL) {
974 fResourceDict = new SkPDFDict;
975 fResourceDict->unref(); // SkRefPtr and new both took a reference.
976
977 if (fGraphicStateResources.count()) {
978 SkRefPtr<SkPDFDict> extGState = new SkPDFDict();
979 extGState->unref(); // SkRefPtr and new both took a reference.
980 for (int i = 0; i < fGraphicStateResources.count(); i++) {
981 SkString nameString("G");
982 nameString.appendS32(i);
983 extGState->insert(
984 nameString.c_str(),
985 new SkPDFObjRef(fGraphicStateResources[i]))->unref();
986 }
987 fResourceDict->insert("ExtGState", extGState.get());
988 }
989
990 if (fXObjectResources.count()) {
991 SkRefPtr<SkPDFDict> xObjects = new SkPDFDict();
992 xObjects->unref(); // SkRefPtr and new both took a reference.
993 for (int i = 0; i < fXObjectResources.count(); i++) {
994 SkString nameString("X");
995 nameString.appendS32(i);
996 xObjects->insert(
997 nameString.c_str(),
998 new SkPDFObjRef(fXObjectResources[i]))->unref();
999 }
1000 fResourceDict->insert("XObject", xObjects.get());
1001 }
1002
1003 if (fFontResources.count()) {
1004 SkRefPtr<SkPDFDict> fonts = new SkPDFDict();
1005 fonts->unref(); // SkRefPtr and new both took a reference.
1006 for (int i = 0; i < fFontResources.count(); i++) {
1007 SkString nameString("F");
1008 nameString.appendS32(i);
1009 fonts->insert(nameString.c_str(),
1010 new SkPDFObjRef(fFontResources[i]))->unref();
1011 }
1012 fResourceDict->insert("Font", fonts.get());
1013 }
1014
1015 if (fShaderResources.count()) {
1016 SkRefPtr<SkPDFDict> patterns = new SkPDFDict();
1017 patterns->unref(); // SkRefPtr and new both took a reference.
1018 for (int i = 0; i < fShaderResources.count(); i++) {
1019 SkString nameString("P");
1020 nameString.appendS32(i);
1021 patterns->insert(nameString.c_str(),
1022 new SkPDFObjRef(fShaderResources[i]))->unref();
1023 }
1024 fResourceDict->insert("Pattern", patterns.get());
1025 }
1026
1027 // For compatibility, add all proc sets (only used for output to PS
1028 // devices).
1029 const char procs[][7] = {"PDF", "Text", "ImageB", "ImageC", "ImageI"};
1030 SkRefPtr<SkPDFArray> procSets = new SkPDFArray();
1031 procSets->unref(); // SkRefPtr and new both took a reference.
1032 procSets->reserve(SK_ARRAY_COUNT(procs));
1033 for (size_t i = 0; i < SK_ARRAY_COUNT(procs); i++)
1034 procSets->appendName(procs[i]);
1035 fResourceDict->insert("ProcSet", procSets.get());
1036 }
1037 return fResourceDict.get();
1038 }
1039
getResources(SkTDArray<SkPDFObject * > * resourceList) const1040 void SkPDFDevice::getResources(SkTDArray<SkPDFObject*>* resourceList) const {
1041 resourceList->setReserve(resourceList->count() +
1042 fGraphicStateResources.count() +
1043 fXObjectResources.count() +
1044 fFontResources.count() +
1045 fShaderResources.count());
1046 for (int i = 0; i < fGraphicStateResources.count(); i++) {
1047 resourceList->push(fGraphicStateResources[i]);
1048 fGraphicStateResources[i]->ref();
1049 fGraphicStateResources[i]->getResources(resourceList);
1050 }
1051 for (int i = 0; i < fXObjectResources.count(); i++) {
1052 resourceList->push(fXObjectResources[i]);
1053 fXObjectResources[i]->ref();
1054 fXObjectResources[i]->getResources(resourceList);
1055 }
1056 for (int i = 0; i < fFontResources.count(); i++) {
1057 resourceList->push(fFontResources[i]);
1058 fFontResources[i]->ref();
1059 fFontResources[i]->getResources(resourceList);
1060 }
1061 for (int i = 0; i < fShaderResources.count(); i++) {
1062 resourceList->push(fShaderResources[i]);
1063 fShaderResources[i]->ref();
1064 fShaderResources[i]->getResources(resourceList);
1065 }
1066 }
1067
getFontResources() const1068 const SkTDArray<SkPDFFont*>& SkPDFDevice::getFontResources() const {
1069 return fFontResources;
1070 }
1071
getMediaBox() const1072 SkRefPtr<SkPDFArray> SkPDFDevice::getMediaBox() const {
1073 SkRefPtr<SkPDFInt> zero = new SkPDFInt(0);
1074 zero->unref(); // SkRefPtr and new both took a reference.
1075
1076 SkRefPtr<SkPDFArray> mediaBox = new SkPDFArray();
1077 mediaBox->unref(); // SkRefPtr and new both took a reference.
1078 mediaBox->reserve(4);
1079 mediaBox->append(zero.get());
1080 mediaBox->append(zero.get());
1081 mediaBox->appendInt(fPageSize.fWidth);
1082 mediaBox->appendInt(fPageSize.fHeight);
1083 return mediaBox;
1084 }
1085
content() const1086 SkStream* SkPDFDevice::content() const {
1087 SkMemoryStream* result = new SkMemoryStream;
1088 result->setData(this->copyContentToData())->unref();
1089 return result;
1090 }
1091
copyContentEntriesToData(ContentEntry * entry,SkWStream * data) const1092 void SkPDFDevice::copyContentEntriesToData(ContentEntry* entry,
1093 SkWStream* data) const {
1094 // TODO(ctguil): For margins, I'm not sure fExistingClipStack/Region is the
1095 // right thing to pass here.
1096 GraphicStackState gsState(fExistingClipStack, fExistingClipRegion, data);
1097 while (entry != NULL) {
1098 SkPoint translation;
1099 translation.iset(this->getOrigin());
1100 translation.negate();
1101 gsState.updateClip(entry->fState.fClipStack, entry->fState.fClipRegion,
1102 translation);
1103 gsState.updateMatrix(entry->fState.fMatrix);
1104 gsState.updateDrawingState(entry->fState);
1105
1106 SkAutoDataUnref copy(entry->fContent.copyToData());
1107 data->write(copy.data(), copy.size());
1108 entry = entry->fNext.get();
1109 }
1110 gsState.drainStack();
1111 }
1112
copyContentToData() const1113 SkData* SkPDFDevice::copyContentToData() const {
1114 SkDynamicMemoryWStream data;
1115 if (fInitialTransform.getType() != SkMatrix::kIdentity_Mask) {
1116 SkPDFUtils::AppendTransform(fInitialTransform, &data);
1117 }
1118
1119 // TODO(aayushkumar): Apply clip along the margins. Currently, webkit
1120 // colors the contentArea white before it starts drawing into it and
1121 // that currently acts as our clip.
1122 // Also, think about adding a transform here (or assume that the values
1123 // sent across account for that)
1124 SkPDFDevice::copyContentEntriesToData(fMarginContentEntries.get(), &data);
1125
1126 // If the content area is the entire page, then we don't need to clip
1127 // the content area (PDF area clips to the page size). Otherwise,
1128 // we have to clip to the content area; we've already applied the
1129 // initial transform, so just clip to the device size.
1130 if (fPageSize != fContentSize) {
1131 SkRect r = SkRect::MakeWH(this->width(), this->height());
1132 emit_clip(NULL, &r, &data);
1133 }
1134
1135 SkPDFDevice::copyContentEntriesToData(fContentEntries.get(), &data);
1136
1137 // potentially we could cache this SkData, and only rebuild it if we
1138 // see that our state has changed.
1139 return data.copyToData();
1140 }
1141
createFormXObjectFromDevice(SkRefPtr<SkPDFFormXObject> * xobject)1142 void SkPDFDevice::createFormXObjectFromDevice(
1143 SkRefPtr<SkPDFFormXObject>* xobject) {
1144 *xobject = new SkPDFFormXObject(this);
1145 (*xobject)->unref(); // SkRefPtr and new both took a reference.
1146 // We always draw the form xobjects that we create back into the device, so
1147 // we simply preserve the font usage instead of pulling it out and merging
1148 // it back in later.
1149 cleanUp(false); // Reset this device to have no content.
1150 init();
1151 }
1152
clearClipFromContent(const SkClipStack * clipStack,const SkRegion & clipRegion)1153 void SkPDFDevice::clearClipFromContent(const SkClipStack* clipStack,
1154 const SkRegion& clipRegion) {
1155 if (clipRegion.isEmpty() || isContentEmpty()) {
1156 return;
1157 }
1158 SkRefPtr<SkPDFFormXObject> curContent;
1159 createFormXObjectFromDevice(&curContent);
1160
1161 // Redraw what we already had, but with the clip as a mask.
1162 drawFormXObjectWithClip(curContent.get(), clipStack, clipRegion, true);
1163 }
1164
drawFormXObjectWithClip(SkPDFFormXObject * xobject,const SkClipStack * clipStack,const SkRegion & clipRegion,bool invertClip)1165 void SkPDFDevice::drawFormXObjectWithClip(SkPDFFormXObject* xobject,
1166 const SkClipStack* clipStack,
1167 const SkRegion& clipRegion,
1168 bool invertClip) {
1169 if (clipRegion.isEmpty() && !invertClip) {
1170 return;
1171 }
1172
1173 // Create the mask.
1174 SkMatrix identity;
1175 identity.reset();
1176 SkDraw draw;
1177 draw.fMatrix = &identity;
1178 draw.fClip = &clipRegion;
1179 draw.fClipStack = clipStack;
1180 SkPaint stockPaint;
1181 this->drawPaint(draw, stockPaint);
1182 SkRefPtr<SkPDFFormXObject> maskFormXObject;
1183 createFormXObjectFromDevice(&maskFormXObject);
1184 SkRefPtr<SkPDFGraphicState> sMaskGS =
1185 SkPDFGraphicState::GetSMaskGraphicState(maskFormXObject.get(),
1186 invertClip);
1187 sMaskGS->unref(); // SkRefPtr and getSMaskGraphicState both took a ref.
1188
1189 // Draw the xobject with the clip as a mask.
1190 ScopedContentEntry content(this, &fExistingClipStack, fExistingClipRegion,
1191 identity, stockPaint);
1192 if (!content.entry()) {
1193 return;
1194 }
1195 SkPDFUtils::ApplyGraphicState(addGraphicStateResource(sMaskGS.get()),
1196 &content.entry()->fContent);
1197 SkPDFUtils::DrawFormXObject(fXObjectResources.count(),
1198 &content.entry()->fContent);
1199 fXObjectResources.push(xobject);
1200 xobject->ref();
1201
1202 sMaskGS = SkPDFGraphicState::GetNoSMaskGraphicState();
1203 sMaskGS->unref(); // SkRefPtr and getSMaskGraphicState both took a ref.
1204 SkPDFUtils::ApplyGraphicState(addGraphicStateResource(sMaskGS.get()),
1205 &content.entry()->fContent);
1206 }
1207
setUpContentEntry(const SkClipStack * clipStack,const SkRegion & clipRegion,const SkMatrix & matrix,const SkPaint & paint,bool hasText,SkRefPtr<SkPDFFormXObject> * dst)1208 ContentEntry* SkPDFDevice::setUpContentEntry(const SkClipStack* clipStack,
1209 const SkRegion& clipRegion,
1210 const SkMatrix& matrix,
1211 const SkPaint& paint,
1212 bool hasText,
1213 SkRefPtr<SkPDFFormXObject>* dst) {
1214 if (clipRegion.isEmpty()) {
1215 return NULL;
1216 }
1217
1218 // The clip stack can come from an SkDraw where it is technically optional.
1219 SkClipStack synthesizedClipStack;
1220 if (clipStack == NULL) {
1221 if (clipRegion == fExistingClipRegion) {
1222 clipStack = &fExistingClipStack;
1223 } else {
1224 // GraphicStackState::updateClip expects the clip stack to have
1225 // fExistingClip as a prefix, so start there, then set the clip
1226 // to the passed region.
1227 synthesizedClipStack = fExistingClipStack;
1228 SkPath clipPath;
1229 clipRegion.getBoundaryPath(&clipPath);
1230 synthesizedClipStack.clipDevPath(clipPath, SkRegion::kReplace_Op,
1231 false);
1232 clipStack = &synthesizedClipStack;
1233 }
1234 }
1235
1236 SkXfermode::Mode xfermode = SkXfermode::kSrcOver_Mode;
1237 if (paint.getXfermode()) {
1238 paint.getXfermode()->asMode(&xfermode);
1239 }
1240
1241 if (xfermode == SkXfermode::kClear_Mode ||
1242 xfermode == SkXfermode::kSrc_Mode) {
1243 this->clearClipFromContent(clipStack, clipRegion);
1244 } else if (xfermode == SkXfermode::kSrcIn_Mode ||
1245 xfermode == SkXfermode::kDstIn_Mode ||
1246 xfermode == SkXfermode::kSrcOut_Mode ||
1247 xfermode == SkXfermode::kDstOut_Mode) {
1248 // For the following modes, we use both source and destination, but
1249 // we use one as a smask for the other, so we have to make form xobjects
1250 // out of both of them: SrcIn, DstIn, SrcOut, DstOut.
1251 if (isContentEmpty()) {
1252 return NULL;
1253 } else {
1254 createFormXObjectFromDevice(dst);
1255 }
1256 }
1257 // TODO(vandebo): Figure out how/if we can handle the following modes:
1258 // SrcAtop, DestAtop, Xor, Plus.
1259
1260 // These xfer modes don't draw source at all.
1261 if (xfermode == SkXfermode::kClear_Mode ||
1262 xfermode == SkXfermode::kDst_Mode) {
1263 return NULL;
1264 }
1265
1266 ContentEntry* entry;
1267 SkTScopedPtr<ContentEntry> newEntry;
1268
1269 ContentEntry* lastContentEntry = getLastContentEntry();
1270 if (lastContentEntry && lastContentEntry->fContent.getOffset() == 0) {
1271 entry = lastContentEntry;
1272 } else {
1273 newEntry.reset(new ContentEntry);
1274 entry = newEntry.get();
1275 }
1276
1277 populateGraphicStateEntryFromPaint(matrix, *clipStack, clipRegion, paint,
1278 hasText, &entry->fState);
1279 if (lastContentEntry && xfermode != SkXfermode::kDstOver_Mode &&
1280 entry->fState.compareInitialState(lastContentEntry->fState)) {
1281 return lastContentEntry;
1282 }
1283
1284 SkTScopedPtr<ContentEntry>* contentEntries = getContentEntries();
1285 if (!lastContentEntry) {
1286 contentEntries->reset(entry);
1287 setLastContentEntry(entry);
1288 } else if (xfermode == SkXfermode::kDstOver_Mode) {
1289 entry->fNext.reset(contentEntries->release());
1290 contentEntries->reset(entry);
1291 } else {
1292 lastContentEntry->fNext.reset(entry);
1293 setLastContentEntry(entry);
1294 }
1295 newEntry.release();
1296 return entry;
1297 }
1298
finishContentEntry(const SkXfermode::Mode xfermode,SkPDFFormXObject * dst)1299 void SkPDFDevice::finishContentEntry(const SkXfermode::Mode xfermode,
1300 SkPDFFormXObject* dst) {
1301 if (xfermode != SkXfermode::kSrcIn_Mode &&
1302 xfermode != SkXfermode::kDstIn_Mode &&
1303 xfermode != SkXfermode::kSrcOut_Mode &&
1304 xfermode != SkXfermode::kDstOut_Mode) {
1305 SkASSERT(!dst);
1306 return;
1307 }
1308
1309 ContentEntry* contentEntries = getContentEntries()->get();
1310 SkASSERT(dst);
1311 SkASSERT(!contentEntries->fNext.get());
1312 // We have to make a copy of these here because changing the current
1313 // content into a form xobject will destroy them.
1314 SkClipStack clipStack = contentEntries->fState.fClipStack;
1315 SkRegion clipRegion = contentEntries->fState.fClipRegion;
1316
1317 SkRefPtr<SkPDFFormXObject> srcFormXObject;
1318 if (!isContentEmpty()) {
1319 createFormXObjectFromDevice(&srcFormXObject);
1320 }
1321
1322 drawFormXObjectWithClip(dst, &clipStack, clipRegion, true);
1323
1324 // We've redrawn dst minus the clip area, if there's no src, we're done.
1325 if (!srcFormXObject.get()) {
1326 return;
1327 }
1328
1329 SkMatrix identity;
1330 identity.reset();
1331 SkPaint stockPaint;
1332 ScopedContentEntry inClipContentEntry(this, &fExistingClipStack,
1333 fExistingClipRegion, identity,
1334 stockPaint);
1335 if (!inClipContentEntry.entry()) {
1336 return;
1337 }
1338
1339 SkRefPtr<SkPDFGraphicState> sMaskGS;
1340 if (xfermode == SkXfermode::kSrcIn_Mode ||
1341 xfermode == SkXfermode::kSrcOut_Mode) {
1342 sMaskGS = SkPDFGraphicState::GetSMaskGraphicState(
1343 dst, xfermode == SkXfermode::kSrcOut_Mode);
1344 fXObjectResources.push(srcFormXObject.get());
1345 srcFormXObject->ref();
1346 } else {
1347 sMaskGS = SkPDFGraphicState::GetSMaskGraphicState(
1348 srcFormXObject.get(), xfermode == SkXfermode::kDstOut_Mode);
1349 // dst already added to fXObjectResources in drawFormXObjectWithClip.
1350 }
1351 sMaskGS->unref(); // SkRefPtr and getSMaskGraphicState both took a ref.
1352 SkPDFUtils::ApplyGraphicState(addGraphicStateResource(sMaskGS.get()),
1353 &inClipContentEntry.entry()->fContent);
1354
1355 SkPDFUtils::DrawFormXObject(fXObjectResources.count() - 1,
1356 &inClipContentEntry.entry()->fContent);
1357
1358 sMaskGS = SkPDFGraphicState::GetNoSMaskGraphicState();
1359 sMaskGS->unref(); // SkRefPtr and getSMaskGraphicState both took a ref.
1360 SkPDFUtils::ApplyGraphicState(addGraphicStateResource(sMaskGS.get()),
1361 &inClipContentEntry.entry()->fContent);
1362 }
1363
isContentEmpty()1364 bool SkPDFDevice::isContentEmpty() {
1365 ContentEntry* contentEntries = getContentEntries()->get();
1366 if (!contentEntries || contentEntries->fContent.getOffset() == 0) {
1367 SkASSERT(!contentEntries || !contentEntries->fNext.get());
1368 return true;
1369 }
1370 return false;
1371 }
1372
populateGraphicStateEntryFromPaint(const SkMatrix & matrix,const SkClipStack & clipStack,const SkRegion & clipRegion,const SkPaint & paint,bool hasText,GraphicStateEntry * entry)1373 void SkPDFDevice::populateGraphicStateEntryFromPaint(
1374 const SkMatrix& matrix,
1375 const SkClipStack& clipStack,
1376 const SkRegion& clipRegion,
1377 const SkPaint& paint,
1378 bool hasText,
1379 GraphicStateEntry* entry) {
1380 SkASSERT(paint.getPathEffect() == NULL);
1381
1382 NOT_IMPLEMENTED(paint.getMaskFilter() != NULL, false);
1383 NOT_IMPLEMENTED(paint.getColorFilter() != NULL, false);
1384
1385 entry->fMatrix = matrix;
1386 entry->fClipStack = clipStack;
1387 entry->fClipRegion = clipRegion;
1388
1389 // PDF treats a shader as a color, so we only set one or the other.
1390 SkRefPtr<SkPDFObject> pdfShader;
1391 const SkShader* shader = paint.getShader();
1392 SkColor color = paint.getColor();
1393 if (shader) {
1394 // PDF positions patterns relative to the initial transform, so
1395 // we need to apply the current transform to the shader parameters.
1396 SkMatrix transform = matrix;
1397 transform.postConcat(fInitialTransform);
1398
1399 // PDF doesn't support kClamp_TileMode, so we simulate it by making
1400 // a pattern the size of the current clip.
1401 SkIRect bounds = clipRegion.getBounds();
1402 pdfShader = SkPDFShader::GetPDFShader(*shader, transform, bounds);
1403 SkSafeUnref(pdfShader.get()); // getShader and SkRefPtr both took a ref
1404
1405 if (pdfShader.get()) {
1406 // pdfShader has been canonicalized so we can directly compare
1407 // pointers.
1408 int resourceIndex = fShaderResources.find(pdfShader.get());
1409 if (resourceIndex < 0) {
1410 resourceIndex = fShaderResources.count();
1411 fShaderResources.push(pdfShader.get());
1412 pdfShader->ref();
1413 }
1414 entry->fShaderIndex = resourceIndex;
1415 } else {
1416 // A color shader is treated as an invalid shader so we don't have
1417 // to set a shader just for a color.
1418 entry->fShaderIndex = -1;
1419 entry->fColor = 0;
1420 color = 0;
1421
1422 // Check for a color shader.
1423 SkShader::GradientInfo gradientInfo;
1424 SkColor gradientColor;
1425 gradientInfo.fColors = &gradientColor;
1426 gradientInfo.fColorOffsets = NULL;
1427 gradientInfo.fColorCount = 1;
1428 if (shader->asAGradient(&gradientInfo) ==
1429 SkShader::kColor_GradientType) {
1430 entry->fColor = SkColorSetA(gradientColor, 0xFF);
1431 color = gradientColor;
1432 }
1433 }
1434 } else {
1435 entry->fShaderIndex = -1;
1436 entry->fColor = SkColorSetA(paint.getColor(), 0xFF);
1437 color = paint.getColor();
1438 }
1439
1440 SkRefPtr<SkPDFGraphicState> newGraphicState;
1441 if (color == paint.getColor()) {
1442 newGraphicState = SkPDFGraphicState::GetGraphicStateForPaint(paint);
1443 } else {
1444 SkPaint newPaint = paint;
1445 newPaint.setColor(color);
1446 newGraphicState = SkPDFGraphicState::GetGraphicStateForPaint(newPaint);
1447 }
1448 newGraphicState->unref(); // getGraphicState and SkRefPtr both took a ref.
1449 int resourceIndex = addGraphicStateResource(newGraphicState.get());
1450 entry->fGraphicStateIndex = resourceIndex;
1451
1452 if (hasText) {
1453 entry->fTextScaleX = paint.getTextScaleX();
1454 entry->fTextFill = paint.getStyle();
1455 } else {
1456 entry->fTextScaleX = 0;
1457 }
1458 }
1459
addGraphicStateResource(SkPDFGraphicState * gs)1460 int SkPDFDevice::addGraphicStateResource(SkPDFGraphicState* gs) {
1461 // Assumes that gs has been canonicalized (so we can directly compare
1462 // pointers).
1463 int result = fGraphicStateResources.find(gs);
1464 if (result < 0) {
1465 result = fGraphicStateResources.count();
1466 fGraphicStateResources.push(gs);
1467 gs->ref();
1468 }
1469 return result;
1470 }
1471
updateFont(const SkPaint & paint,uint16_t glyphID,ContentEntry * contentEntry)1472 void SkPDFDevice::updateFont(const SkPaint& paint, uint16_t glyphID,
1473 ContentEntry* contentEntry) {
1474 SkTypeface* typeface = paint.getTypeface();
1475 if (contentEntry->fState.fFont == NULL ||
1476 contentEntry->fState.fTextSize != paint.getTextSize() ||
1477 !contentEntry->fState.fFont->hasGlyph(glyphID)) {
1478 int fontIndex = getFontResourceIndex(typeface, glyphID);
1479 contentEntry->fContent.writeText("/F");
1480 contentEntry->fContent.writeDecAsText(fontIndex);
1481 contentEntry->fContent.writeText(" ");
1482 SkPDFScalar::Append(paint.getTextSize(), &contentEntry->fContent);
1483 contentEntry->fContent.writeText(" Tf\n");
1484 contentEntry->fState.fFont = fFontResources[fontIndex];
1485 }
1486 }
1487
getFontResourceIndex(SkTypeface * typeface,uint16_t glyphID)1488 int SkPDFDevice::getFontResourceIndex(SkTypeface* typeface, uint16_t glyphID) {
1489 SkRefPtr<SkPDFFont> newFont = SkPDFFont::GetFontResource(typeface, glyphID);
1490 newFont->unref(); // getFontResource and SkRefPtr both took a ref.
1491 int resourceIndex = fFontResources.find(newFont.get());
1492 if (resourceIndex < 0) {
1493 resourceIndex = fFontResources.count();
1494 fFontResources.push(newFont.get());
1495 newFont->ref();
1496 }
1497 return resourceIndex;
1498 }
1499
internalDrawBitmap(const SkMatrix & matrix,const SkClipStack * clipStack,const SkRegion & clipRegion,const SkBitmap & bitmap,const SkIRect * srcRect,const SkPaint & paint)1500 void SkPDFDevice::internalDrawBitmap(const SkMatrix& matrix,
1501 const SkClipStack* clipStack,
1502 const SkRegion& clipRegion,
1503 const SkBitmap& bitmap,
1504 const SkIRect* srcRect,
1505 const SkPaint& paint) {
1506 SkMatrix scaled;
1507 // Adjust for origin flip.
1508 scaled.setScale(SK_Scalar1, -SK_Scalar1);
1509 scaled.postTranslate(0, SK_Scalar1);
1510 // Scale the image up from 1x1 to WxH.
1511 SkIRect subset = SkIRect::MakeWH(bitmap.width(), bitmap.height());
1512 scaled.postScale(SkIntToScalar(subset.width()),
1513 SkIntToScalar(subset.height()));
1514 scaled.postConcat(matrix);
1515 ScopedContentEntry content(this, clipStack, clipRegion, scaled, paint);
1516 if (!content.entry()) {
1517 return;
1518 }
1519
1520 if (srcRect && !subset.intersect(*srcRect)) {
1521 return;
1522 }
1523
1524 SkPDFImage* image = SkPDFImage::CreateImage(bitmap, subset, paint);
1525 if (!image) {
1526 return;
1527 }
1528
1529 fXObjectResources.push(image); // Transfer reference.
1530 SkPDFUtils::DrawFormXObject(fXObjectResources.count() - 1,
1531 &content.entry()->fContent);
1532 }
1533
onReadPixels(const SkBitmap & bitmap,int x,int y,SkCanvas::Config8888)1534 bool SkPDFDevice::onReadPixels(const SkBitmap& bitmap, int x, int y,
1535 SkCanvas::Config8888) {
1536 return false;
1537 }
1538
allowImageFilter(SkImageFilter *)1539 bool SkPDFDevice::allowImageFilter(SkImageFilter*) {
1540 return false;
1541 }
1542
1543