michael@0: 
michael@0: /*
michael@0:  * Copyright 2011 Google Inc.
michael@0:  *
michael@0:  * Use of this source code is governed by a BSD-style license that can be
michael@0:  * found in the LICENSE file.
michael@0:  */
michael@0: 
michael@0: 
michael@0: #include "SkData.h"
michael@0: #include "SkGeometry.h"
michael@0: #include "SkPaint.h"
michael@0: #include "SkPath.h"
michael@0: #include "SkPDFResourceDict.h"
michael@0: #include "SkPDFUtils.h"
michael@0: #include "SkStream.h"
michael@0: #include "SkString.h"
michael@0: #include "SkPDFTypes.h"
michael@0: 
michael@0: //static
michael@0: SkPDFArray* SkPDFUtils::RectToArray(const SkRect& rect) {
michael@0:     SkPDFArray* result = new SkPDFArray();
michael@0:     result->reserve(4);
michael@0:     result->appendScalar(rect.fLeft);
michael@0:     result->appendScalar(rect.fTop);
michael@0:     result->appendScalar(rect.fRight);
michael@0:     result->appendScalar(rect.fBottom);
michael@0:     return result;
michael@0: }
michael@0: 
michael@0: // static
michael@0: SkPDFArray* SkPDFUtils::MatrixToArray(const SkMatrix& matrix) {
michael@0:     SkScalar values[6];
michael@0:     if (!matrix.asAffine(values)) {
michael@0:         SkMatrix::SetAffineIdentity(values);
michael@0:     }
michael@0: 
michael@0:     SkPDFArray* result = new SkPDFArray;
michael@0:     result->reserve(6);
michael@0:     for (size_t i = 0; i < SK_ARRAY_COUNT(values); i++) {
michael@0:         result->appendScalar(values[i]);
michael@0:     }
michael@0:     return result;
michael@0: }
michael@0: 
michael@0: // static
michael@0: void SkPDFUtils::AppendTransform(const SkMatrix& matrix, SkWStream* content) {
michael@0:     SkScalar values[6];
michael@0:     if (!matrix.asAffine(values)) {
michael@0:         SkMatrix::SetAffineIdentity(values);
michael@0:     }
michael@0:     for (size_t i = 0; i < SK_ARRAY_COUNT(values); i++) {
michael@0:         SkPDFScalar::Append(values[i], content);
michael@0:         content->writeText(" ");
michael@0:     }
michael@0:     content->writeText("cm\n");
michael@0: }
michael@0: 
michael@0: // static
michael@0: void SkPDFUtils::MoveTo(SkScalar x, SkScalar y, SkWStream* content) {
michael@0:     SkPDFScalar::Append(x, content);
michael@0:     content->writeText(" ");
michael@0:     SkPDFScalar::Append(y, content);
michael@0:     content->writeText(" m\n");
michael@0: }
michael@0: 
michael@0: // static
michael@0: void SkPDFUtils::AppendLine(SkScalar x, SkScalar y, SkWStream* content) {
michael@0:     SkPDFScalar::Append(x, content);
michael@0:     content->writeText(" ");
michael@0:     SkPDFScalar::Append(y, content);
michael@0:     content->writeText(" l\n");
michael@0: }
michael@0: 
michael@0: // static
michael@0: void SkPDFUtils::AppendCubic(SkScalar ctl1X, SkScalar ctl1Y,
michael@0:                              SkScalar ctl2X, SkScalar ctl2Y,
michael@0:                              SkScalar dstX, SkScalar dstY, SkWStream* content) {
michael@0:     SkString cmd("y\n");
michael@0:     SkPDFScalar::Append(ctl1X, content);
michael@0:     content->writeText(" ");
michael@0:     SkPDFScalar::Append(ctl1Y, content);
michael@0:     content->writeText(" ");
michael@0:     if (ctl2X != dstX || ctl2Y != dstY) {
michael@0:         cmd.set("c\n");
michael@0:         SkPDFScalar::Append(ctl2X, content);
michael@0:         content->writeText(" ");
michael@0:         SkPDFScalar::Append(ctl2Y, content);
michael@0:         content->writeText(" ");
michael@0:     }
michael@0:     SkPDFScalar::Append(dstX, content);
michael@0:     content->writeText(" ");
michael@0:     SkPDFScalar::Append(dstY, content);
michael@0:     content->writeText(" ");
michael@0:     content->writeText(cmd.c_str());
michael@0: }
michael@0: 
michael@0: // static
michael@0: void SkPDFUtils::AppendRectangle(const SkRect& rect, SkWStream* content) {
michael@0:     // Skia has 0,0 at top left, pdf at bottom left.  Do the right thing.
michael@0:     SkScalar bottom = SkMinScalar(rect.fBottom, rect.fTop);
michael@0: 
michael@0:     SkPDFScalar::Append(rect.fLeft, content);
michael@0:     content->writeText(" ");
michael@0:     SkPDFScalar::Append(bottom, content);
michael@0:     content->writeText(" ");
michael@0:     SkPDFScalar::Append(rect.width(), content);
michael@0:     content->writeText(" ");
michael@0:     SkPDFScalar::Append(rect.height(), content);
michael@0:     content->writeText(" re\n");
michael@0: }
michael@0: 
michael@0: // static
michael@0: void SkPDFUtils::EmitPath(const SkPath& path, SkPaint::Style paintStyle,
michael@0:                           SkWStream* content) {
michael@0:     // Filling a path with no area results in a drawing in PDF renderers but
michael@0:     // Chrome expects to be able to draw some such entities with no visible
michael@0:     // result, so we detect those cases and discard the drawing for them.
michael@0:     // Specifically: moveTo(X), lineTo(Y) and moveTo(X), lineTo(X), lineTo(Y).
michael@0:     enum SkipFillState {
michael@0:         kEmpty_SkipFillState         = 0,
michael@0:         kSingleLine_SkipFillState    = 1,
michael@0:         kNonSingleLine_SkipFillState = 2,
michael@0:     };
michael@0:     SkipFillState fillState = kEmpty_SkipFillState;
michael@0:     if (paintStyle != SkPaint::kFill_Style) {
michael@0:         fillState = kNonSingleLine_SkipFillState;
michael@0:     }
michael@0:     SkPoint lastMovePt = SkPoint::Make(0,0);
michael@0:     SkDynamicMemoryWStream currentSegment;
michael@0:     SkPoint args[4];
michael@0:     SkPath::Iter iter(path, false);
michael@0:     for (SkPath::Verb verb = iter.next(args);
michael@0:          verb != SkPath::kDone_Verb;
michael@0:          verb = iter.next(args)) {
michael@0:         // args gets all the points, even the implicit first point.
michael@0:         switch (verb) {
michael@0:             case SkPath::kMove_Verb:
michael@0:                 MoveTo(args[0].fX, args[0].fY, &currentSegment);
michael@0:                 lastMovePt = args[0];
michael@0:                 fillState = kEmpty_SkipFillState;
michael@0:                 break;
michael@0:             case SkPath::kLine_Verb:
michael@0:                 AppendLine(args[1].fX, args[1].fY, &currentSegment);
michael@0:                 if (fillState == kEmpty_SkipFillState) {
michael@0:                    if (args[0] != lastMovePt) {
michael@0:                        fillState = kSingleLine_SkipFillState;
michael@0:                    }
michael@0:                 } else if (fillState == kSingleLine_SkipFillState) {
michael@0:                     fillState = kNonSingleLine_SkipFillState;
michael@0:                 }
michael@0:                 break;
michael@0:             case SkPath::kQuad_Verb: {
michael@0:                 SkPoint cubic[4];
michael@0:                 SkConvertQuadToCubic(args, cubic);
michael@0:                 AppendCubic(cubic[1].fX, cubic[1].fY, cubic[2].fX, cubic[2].fY,
michael@0:                             cubic[3].fX, cubic[3].fY, &currentSegment);
michael@0:                 fillState = kNonSingleLine_SkipFillState;
michael@0:                 break;
michael@0:             }
michael@0:             case SkPath::kCubic_Verb:
michael@0:                 AppendCubic(args[1].fX, args[1].fY, args[2].fX, args[2].fY,
michael@0:                             args[3].fX, args[3].fY, &currentSegment);
michael@0:                 fillState = kNonSingleLine_SkipFillState;
michael@0:                 break;
michael@0:             case SkPath::kClose_Verb:
michael@0:                 if (fillState != kSingleLine_SkipFillState) {
michael@0:                     ClosePath(&currentSegment);
michael@0:                     SkData* data = currentSegment.copyToData();
michael@0:                     content->write(data->data(), data->size());
michael@0:                     data->unref();
michael@0:                 }
michael@0:                 currentSegment.reset();
michael@0:                 break;
michael@0:             default:
michael@0:                 SkASSERT(false);
michael@0:                 break;
michael@0:         }
michael@0:     }
michael@0:     if (currentSegment.bytesWritten() > 0) {
michael@0:         SkData* data = currentSegment.copyToData();
michael@0:         content->write(data->data(), data->size());
michael@0:         data->unref();
michael@0:     }
michael@0: }
michael@0: 
michael@0: // static
michael@0: void SkPDFUtils::ClosePath(SkWStream* content) {
michael@0:     content->writeText("h\n");
michael@0: }
michael@0: 
michael@0: // static
michael@0: void SkPDFUtils::PaintPath(SkPaint::Style style, SkPath::FillType fill,
michael@0:                            SkWStream* content) {
michael@0:     if (style == SkPaint::kFill_Style) {
michael@0:         content->writeText("f");
michael@0:     } else if (style == SkPaint::kStrokeAndFill_Style) {
michael@0:         content->writeText("B");
michael@0:     } else if (style == SkPaint::kStroke_Style) {
michael@0:         content->writeText("S");
michael@0:     }
michael@0: 
michael@0:     if (style != SkPaint::kStroke_Style) {
michael@0:         NOT_IMPLEMENTED(fill == SkPath::kInverseEvenOdd_FillType, false);
michael@0:         NOT_IMPLEMENTED(fill == SkPath::kInverseWinding_FillType, false);
michael@0:         if (fill == SkPath::kEvenOdd_FillType) {
michael@0:             content->writeText("*");
michael@0:         }
michael@0:     }
michael@0:     content->writeText("\n");
michael@0: }
michael@0: 
michael@0: // static
michael@0: void SkPDFUtils::StrokePath(SkWStream* content) {
michael@0:     SkPDFUtils::PaintPath(
michael@0:         SkPaint::kStroke_Style, SkPath::kWinding_FillType, content);
michael@0: }
michael@0: 
michael@0: // static
michael@0: void SkPDFUtils::DrawFormXObject(int objectIndex, SkWStream* content) {
michael@0:     content->writeText("/");
michael@0:     content->writeText(SkPDFResourceDict::getResourceName(
michael@0:             SkPDFResourceDict::kXObject_ResourceType,
michael@0:             objectIndex).c_str());
michael@0:     content->writeText(" Do\n");
michael@0: }
michael@0: 
michael@0: // static
michael@0: void SkPDFUtils::ApplyGraphicState(int objectIndex, SkWStream* content) {
michael@0:     content->writeText("/");
michael@0:     content->writeText(SkPDFResourceDict::getResourceName(
michael@0:             SkPDFResourceDict::kExtGState_ResourceType,
michael@0:             objectIndex).c_str());
michael@0:     content->writeText(" gs\n");
michael@0: }
michael@0: 
michael@0: // static
michael@0: void SkPDFUtils::ApplyPattern(int objectIndex, SkWStream* content) {
michael@0:     // Select Pattern color space (CS, cs) and set pattern object as current
michael@0:     // color (SCN, scn)
michael@0:     SkString resourceName = SkPDFResourceDict::getResourceName(
michael@0:             SkPDFResourceDict::kPattern_ResourceType,
michael@0:             objectIndex);
michael@0:     content->writeText("/Pattern CS/Pattern cs/");
michael@0:     content->writeText(resourceName.c_str());
michael@0:     content->writeText(" SCN/");
michael@0:     content->writeText(resourceName.c_str());
michael@0:     content->writeText(" scn\n");
michael@0: }