| // Copyright 2016 PDFium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com |
| |
| #ifndef CORE_FPDFAPI_PARSER_CPDF_DICTIONARY_H_ |
| #define CORE_FPDFAPI_PARSER_CPDF_DICTIONARY_H_ |
| |
| #include <map> |
| #include <set> |
| |
| #include "core/fpdfapi/parser/cpdf_object.h" |
| #include "core/fxcrt/cfx_string_pool_template.h" |
| #include "core/fxcrt/cfx_weak_ptr.h" |
| #include "core/fxcrt/fx_coordinates.h" |
| #include "core/fxcrt/fx_string.h" |
| #include "third_party/base/ptr_util.h" |
| |
| class CPDF_IndirectObjectHolder; |
| |
| class CPDF_Dictionary : public CPDF_Object { |
| public: |
| using iterator = std::map<CFX_ByteString, CPDF_Object*>::iterator; |
| using const_iterator = std::map<CFX_ByteString, CPDF_Object*>::const_iterator; |
| |
| CPDF_Dictionary(); |
| explicit CPDF_Dictionary(const CFX_WeakPtr<CFX_ByteStringPool>& pPool); |
| |
| // CPDF_Object. |
| Type GetType() const override; |
| CPDF_Object* Clone() const override; |
| CPDF_Dictionary* GetDict() const override; |
| bool IsDictionary() const override; |
| CPDF_Dictionary* AsDictionary() override; |
| const CPDF_Dictionary* AsDictionary() const override; |
| |
| size_t GetCount() const { return m_Map.size(); } |
| CPDF_Object* GetObjectFor(const CFX_ByteString& key) const; |
| CPDF_Object* GetDirectObjectFor(const CFX_ByteString& key) const; |
| CFX_ByteString GetStringFor(const CFX_ByteString& key) const; |
| CFX_ByteString GetStringFor(const CFX_ByteString& key, |
| const CFX_ByteString& default_str) const; |
| CFX_WideString GetUnicodeTextFor(const CFX_ByteString& key) const; |
| int GetIntegerFor(const CFX_ByteString& key) const; |
| int GetIntegerFor(const CFX_ByteString& key, int default_int) const; |
| bool GetBooleanFor(const CFX_ByteString& key, bool bDefault = false) const; |
| FX_FLOAT GetNumberFor(const CFX_ByteString& key) const; |
| CPDF_Dictionary* GetDictFor(const CFX_ByteString& key) const; |
| CPDF_Stream* GetStreamFor(const CFX_ByteString& key) const; |
| CPDF_Array* GetArrayFor(const CFX_ByteString& key) const; |
| CFX_FloatRect GetRectFor(const CFX_ByteString& key) const; |
| CFX_Matrix GetMatrixFor(const CFX_ByteString& key) const; |
| FX_FLOAT GetFloatFor(const CFX_ByteString& key) const { |
| return GetNumberFor(key); |
| } |
| |
| bool KeyExist(const CFX_ByteString& key) const; |
| bool IsSignatureDict() const; |
| |
| // Set* functions invalidate iterators for the element with the key |key|. |
| void SetFor(const CFX_ByteString& key, CPDF_Object* pObj); |
| void SetNameFor(const CFX_ByteString& key, const CFX_ByteString& name); |
| void SetStringFor(const CFX_ByteString& key, const CFX_ByteString& str); |
| void SetIntegerFor(const CFX_ByteString& key, int i); |
| void SetNumberFor(const CFX_ByteString& key, FX_FLOAT f); |
| void SetReferenceFor(const CFX_ByteString& key, |
| CPDF_IndirectObjectHolder* pDoc, |
| uint32_t objnum); |
| void SetRectFor(const CFX_ByteString& key, const CFX_FloatRect& rect); |
| void SetMatrixFor(const CFX_ByteString& key, const CFX_Matrix& matrix); |
| void SetBooleanFor(const CFX_ByteString& key, bool bValue); |
| |
| void ConvertToIndirectObjectFor(const CFX_ByteString& key, |
| CPDF_IndirectObjectHolder* pHolder); |
| |
| // Invalidates iterators for the element with the key |key|. |
| void RemoveFor(const CFX_ByteString& key); |
| |
| // Invalidates iterators for the element with the key |oldkey|. |
| void ReplaceKey(const CFX_ByteString& oldkey, const CFX_ByteString& newkey); |
| |
| iterator begin() { return m_Map.begin(); } |
| iterator end() { return m_Map.end(); } |
| const_iterator begin() const { return m_Map.begin(); } |
| const_iterator end() const { return m_Map.end(); } |
| |
| CFX_WeakPtr<CFX_ByteStringPool> GetByteStringPool() const { return m_pPool; } |
| |
| protected: |
| ~CPDF_Dictionary() override; |
| |
| CFX_ByteString MaybeIntern(const CFX_ByteString& str); |
| CPDF_Object* CloneNonCyclic( |
| bool bDirect, |
| std::set<const CPDF_Object*>* visited) const override; |
| |
| CFX_WeakPtr<CFX_ByteStringPool> m_pPool; |
| std::map<CFX_ByteString, CPDF_Object*> m_Map; |
| }; |
| |
| using UniqueDictionary = |
| std::unique_ptr<CPDF_Dictionary, ReleaseDeleter<CPDF_Object>>; |
| |
| inline CPDF_Dictionary* ToDictionary(CPDF_Object* obj) { |
| return obj ? obj->AsDictionary() : nullptr; |
| } |
| |
| inline const CPDF_Dictionary* ToDictionary(const CPDF_Object* obj) { |
| return obj ? obj->AsDictionary() : nullptr; |
| } |
| |
| inline UniqueDictionary ToDictionary(UniqueObject obj) { |
| CPDF_Dictionary* pDict = ToDictionary(obj.get()); |
| if (!pDict) |
| return nullptr; |
| obj.release(); |
| return UniqueDictionary(pDict); |
| } |
| |
| #endif // CORE_FPDFAPI_PARSER_CPDF_DICTIONARY_H_ |