SkPDFCanon

src/pdf/SkPDFCanon.cpp

Index: src/pdf/SkPDFCanon.cpp
diff --git a/src/pdf/SkPDFCanon.cpp b/src/pdf/SkPDFCanon.cpp
new file mode 100644
index 0000000000000000000000000000000000000000..31212f8c91564d5568ea5f56371593435463bc17
--- /dev/null
+++ b/src/pdf/SkPDFCanon.cpp
@@ -0,0 +1,136 @@
+/*
+ * Copyright 2015 Google Inc.
+ *
+ * Use of this source code is governed by a BSD-style license that can be
+ * found in the LICENSE file.
+ */
+
+#include "SkLazyPtr.h"
+#include "SkPDFCanon.h"
+#include "SkPDFFont.h"
+#include "SkPDFGraphicState.h"
+#include "SkPDFShader.h"
+
+////////////////////////////////////////////////////////////////////////////////
+
+SK_DECLARE_STATIC_MUTEX(gSkPDFCanonFontMutex);
+SK_DECLARE_STATIC_MUTEX(gSkPDFCanonShaderMutex);
+SK_DECLARE_STATIC_MUTEX(gSkPDFCanonPaintMutex);
+
+SkBaseMutex& SkPDFCanon::GetFontMutex() { return gSkPDFCanonFontMutex; }
+SkBaseMutex& SkPDFCanon::GetShaderMutex() { return gSkPDFCanonShaderMutex; }
+SkBaseMutex& SkPDFCanon::GetPaintMutex() { return gSkPDFCanonPaintMutex; }
+
+SkPDFCanon::SkPDFCanon() {}
+
+SkPDFCanon::~SkPDFCanon() {}
+
+SK_DECLARE_STATIC_LAZY_PTR(SkPDFCanon, singleton);
+
+SkPDFCanon& SkPDFCanon::GetCanon() { return *singleton.get(); }
+
+////////////////////////////////////////////////////////////////////////////////
+
+static void assert_mutex_held(const SkPDFCanon* canon, SkBaseMutex& mutex) {
+    if (canon == singleton.get()) {
+        mutex.assertHeld();
+    }
+}
+
+SkPDFFont* SkPDFCanon::findFont(uint32_t fontID,
+                                uint16_t glyphID,
+                                SkPDFFont** relatedFontPtr) const {
+    assert_mutex_held(this, gSkPDFCanonFontMutex);
+    SkASSERT(relatedFontPtr);
+
+    SkPDFFont* relatedFont = NULL;
+    for (int i = 0; i < fFontRecords.count(); ++i) {
+        SkPDFFont::Match match = SkPDFFont::IsMatch(
+                fFontRecords[i].fFont, fFontRecords[i].fFontID,
+                fFontRecords[i].fGlyphID, fontID, glyphID);
+        if (SkPDFFont::kExact_Match == match) {
+            return fFontRecords[i].fFont;
+        } else if (!relatedFont && SkPDFFont::kRelated_Match == match) {
+            relatedFont = fFontRecords[i].fFont;
+        }
+    }
+    *relatedFontPtr = relatedFont;  // May still be NULL.
+    return NULL;
+}
+
+void SkPDFCanon::addFont(SkPDFFont* font, uint32_t fontID, uint16_t fGlyphID) {
+    assert_mutex_held(this, gSkPDFCanonFontMutex);
+    SkPDFCanon::FontRec* rec = fFontRecords.push();
+    rec->fFont = font;
+    rec->fFontID = fontID;
+    rec->fGlyphID = fGlyphID;
+}
+
+void SkPDFCanon::removeFont(SkPDFFont* pdfFont) {
+    assert_mutex_held(this, gSkPDFCanonFontMutex);
+    for (int i = 0; i < fFontRecords.count(); i++) {
+        if (fFontRecords[i].fFont == pdfFont) {
+            fFontRecords.removeShuffle(i);
+            return;
+        }
+    }
+    // Not all SkPDFFonts are added to the Canon.
+}
+
+////////////////////////////////////////////////////////////////////////////////
+
+SkPDFShader* SkPDFCanon::findShader(const SkPDFShader::State& state) const {
+    assert_mutex_held(this, gSkPDFCanonShaderMutex);
+    for (int i = 0; i < fShaderRecords.count(); ++i) {
+        if (fShaderRecords[i]->equals(state)) {
+            return fShaderRecords[i];
+        }
+    }
+    return NULL;
+}
+
+void SkPDFCanon::addShader(SkPDFShader* shader) {
+    assert_mutex_held(this, gSkPDFCanonShaderMutex);
+    SkASSERT(shader);
+    fShaderRecords.push(shader);
+}
+
+void SkPDFCanon::removeShader(SkPDFShader* pdfShader) {
+    assert_mutex_held(this, gSkPDFCanonShaderMutex);
+    for (int i = 0; i < fShaderRecords.count(); ++i) {
+        if (fShaderRecords[i] == pdfShader) {
+            fShaderRecords.removeShuffle(i);
+            return;
+        }
+    }
+    SkDEBUGFAIL("pdfShader not found");
+}
+
+////////////////////////////////////////////////////////////////////////////////
+
+SkPDFGraphicState* SkPDFCanon::findGraphicState(const SkPaint& paint) const {
+    assert_mutex_held(this, gSkPDFCanonPaintMutex);
+    for (int i = 0; i < fGraphicStateRecords.count(); ++i) {
+        if (fGraphicStateRecords[i]->equals(paint)) {
+            return fGraphicStateRecords[i];
+        }
+    }
+    return NULL;
+}
+
+void SkPDFCanon::addGraphicState(SkPDFGraphicState* state) {
+    assert_mutex_held(this, gSkPDFCanonPaintMutex);
+    SkASSERT(state);
+    fGraphicStateRecords.push(state);
+}
+
+void SkPDFCanon::removeGraphicState(SkPDFGraphicState* pdfGraphicState) {
+    assert_mutex_held(this, gSkPDFCanonPaintMutex);
+    for (int i = 0; i < fGraphicStateRecords.count(); ++i) {
+        if (fGraphicStateRecords[i] == pdfGraphicState) {
+            fGraphicStateRecords.removeShuffle(i);
+            return;
+        }
+    }
+    SkDEBUGFAIL("pdfGraphicState not found");
+}