| OLD | NEW |
| (Empty) |
| 1 /* | |
| 2 * Copyright 2013 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 "SkPDFResourceDict.h" | |
| 9 #include "SkPostConfig.h" | |
| 10 | |
| 11 SK_DEFINE_INST_COUNT(SkPDFResourceDict) | |
| 12 | |
| 13 // Sanity check that the values of enum SkPDFResourceType correspond to the | |
| 14 // expected values as defined in the arrays below. | |
| 15 // If these are failing, you may need to update the resource_type_prefixes | |
| 16 // and resource_type_names arrays below. | |
| 17 SK_COMPILE_ASSERT(SkPDFResourceDict::kExtGState_ResourceType == 0, | |
| 18 resource_type_mismatch); | |
| 19 SK_COMPILE_ASSERT(SkPDFResourceDict::kPattern_ResourceType == 1, | |
| 20 resource_type_mismatch); | |
| 21 SK_COMPILE_ASSERT(SkPDFResourceDict::kXObject_ResourceType == 2, | |
| 22 resource_type_mismatch); | |
| 23 SK_COMPILE_ASSERT(SkPDFResourceDict::kFont_ResourceType == 3, | |
| 24 resource_type_mismatch); | |
| 25 | |
| 26 static const char resource_type_prefixes[] = { | |
| 27 'G', | |
| 28 'P', | |
| 29 'X', | |
| 30 'F' | |
| 31 }; | |
| 32 | |
| 33 static const char* resource_type_names[] = { | |
| 34 "ExtGState", | |
| 35 "Pattern", | |
| 36 "XObject", | |
| 37 "Font" | |
| 38 }; | |
| 39 | |
| 40 static char get_resource_type_prefix( | |
| 41 SkPDFResourceDict::SkPDFResourceType type) { | |
| 42 SkASSERT(type >= 0); | |
| 43 SkASSERT(type < SkPDFResourceDict::kResourceTypeCount); | |
| 44 | |
| 45 return resource_type_prefixes[type]; | |
| 46 } | |
| 47 | |
| 48 static const char* get_resource_type_name( | |
| 49 SkPDFResourceDict::SkPDFResourceType type) { | |
| 50 SkASSERT(type >= 0); | |
| 51 SkASSERT(type < SkPDFResourceDict::kResourceTypeCount); | |
| 52 | |
| 53 return resource_type_names[type]; | |
| 54 } | |
| 55 | |
| 56 SkPDFResourceDict::SkPDFResourceDict() : SkPDFDict() { | |
| 57 const char procs[][7] = {"PDF", "Text", "ImageB", "ImageC", "ImageI"}; | |
| 58 SkPDFArray* procSets = SkNEW(SkPDFArray()); | |
| 59 | |
| 60 procSets->reserve(SK_ARRAY_COUNT(procs)); | |
| 61 for (size_t i = 0; i < SK_ARRAY_COUNT(procs); i++) { | |
| 62 procSets->appendName(procs[i]); | |
| 63 } | |
| 64 insert("ProcSets", procSets)->unref(); | |
| 65 | |
| 66 // Actual sub-dicts will be lazily added later | |
| 67 fTypes.setCount(kResourceTypeCount); | |
| 68 for (size_t i=0; i < kResourceTypeCount; i++) { | |
| 69 fTypes[i] = NULL; | |
| 70 } | |
| 71 } | |
| 72 | |
| 73 SkPDFObject* SkPDFResourceDict::insertResourceAsRef( | |
| 74 SkPDFResourceType type, int key, SkPDFObject* value) { | |
| 75 SkAutoTUnref<SkPDFObjRef> ref(SkNEW_ARGS(SkPDFObjRef, (value))); | |
| 76 insertResource(type, key, ref); | |
| 77 fResources.add(value); | |
| 78 | |
| 79 return value; | |
| 80 } | |
| 81 | |
| 82 void SkPDFResourceDict::getResources( | |
| 83 const SkTSet<SkPDFObject*>& knownResourceObjects, | |
| 84 SkTSet<SkPDFObject*>* newResourceObjects, | |
| 85 bool recursive) const { | |
| 86 // TODO: reserve not correct if we need to recursively explore. | |
| 87 newResourceObjects->setReserve(newResourceObjects->count() + | |
| 88 fResources.count()); | |
| 89 | |
| 90 for (int i = 0; i < fResources.count(); i++) { | |
| 91 if (!knownResourceObjects.contains(fResources[i]) && | |
| 92 !newResourceObjects->contains(fResources[i])) { | |
| 93 newResourceObjects->add(fResources[i]); | |
| 94 fResources[i]->ref(); | |
| 95 if (recursive) { | |
| 96 fResources[i]->getResources(knownResourceObjects, | |
| 97 newResourceObjects); | |
| 98 } | |
| 99 } | |
| 100 } | |
| 101 } | |
| 102 | |
| 103 SkString SkPDFResourceDict::getResourceName( | |
| 104 SkPDFResourceType type, int key) { | |
| 105 SkString keyString; | |
| 106 keyString.printf("%c%d", get_resource_type_prefix(type), key); | |
| 107 return keyString; | |
| 108 } | |
| 109 | |
| 110 SkPDFObject* SkPDFResourceDict::insertResource( | |
| 111 SkPDFResourceType type, int key, SkPDFObject* value) { | |
| 112 SkPDFDict* typeDict = fTypes[type]; | |
| 113 if (NULL == typeDict) { | |
| 114 SkAutoTUnref<SkPDFDict> newDict(SkNEW(SkPDFDict())); | |
| 115 SkPDFName* typeName = SkNEW_ARGS( | |
| 116 SkPDFName, (get_resource_type_name(type))); | |
| 117 insert(typeName, newDict); // ref counting handled here | |
| 118 fTypes[type] = newDict; | |
| 119 typeDict = newDict.get(); | |
| 120 } | |
| 121 | |
| 122 SkPDFName* keyName = SkNEW_ARGS(SkPDFName, (getResourceName(type, key))); | |
| 123 typeDict->insert(keyName, value); | |
| 124 return value; | |
| 125 } | |
| OLD | NEW |