diff options
-rw-r--r-- | core/fpdfapi/page/cpdf_image.cpp | 26 | ||||
-rw-r--r-- | core/fpdfapi/page/cpdf_image.h | 4 | ||||
-rw-r--r-- | core/fpdfapi/render/cpdf_dibsource.cpp | 2 | ||||
-rw-r--r-- | core/fxcodec/jbig2/JBig2_Image.cpp | 4 | ||||
-rw-r--r-- | core/fxcrt/maybe_owned.h | 22 |
5 files changed, 28 insertions, 30 deletions
diff --git a/core/fpdfapi/page/cpdf_image.cpp b/core/fpdfapi/page/cpdf_image.cpp index 5f82886a9b..65ca78e08e 100644 --- a/core/fpdfapi/page/cpdf_image.cpp +++ b/core/fpdfapi/page/cpdf_image.cpp @@ -35,27 +35,33 @@ CPDF_Image::CPDF_Image(CPDF_Document* pDoc) : m_pDocument(pDoc) {} CPDF_Image::CPDF_Image(CPDF_Document* pDoc, std::unique_ptr<CPDF_Stream> pStream) - : m_bIsInline(true), m_pDocument(pDoc), m_pStream(std::move(pStream)) { + : m_bIsInline(true), + m_pDocument(pDoc), + m_pStream(std::move(pStream)), + m_pDict(ToDictionary(m_pStream->GetDict()->Clone())) { ASSERT(m_pStream.IsOwned()); - FinishInitialization(m_pStream->GetDict()); + ASSERT(m_pDict.IsOwned()); + FinishInitialization(); } CPDF_Image::CPDF_Image(CPDF_Document* pDoc, uint32_t dwStreamObjNum) : m_pDocument(pDoc), - m_pStream(ToStream(pDoc->GetIndirectObject(dwStreamObjNum))) { + m_pStream(ToStream(pDoc->GetIndirectObject(dwStreamObjNum))), + m_pDict(m_pStream->GetDict()) { ASSERT(!m_pStream.IsOwned()); - FinishInitialization(m_pStream->GetDict()); + ASSERT(!m_pDict.IsOwned()); + FinishInitialization(); } CPDF_Image::~CPDF_Image() {} -void CPDF_Image::FinishInitialization(CPDF_Dictionary* pDict) { - m_pOC = pDict->GetDictFor("OC"); +void CPDF_Image::FinishInitialization() { + m_pOC = m_pDict->GetDictFor("OC"); m_bIsMask = - !pDict->KeyExist("ColorSpace") || pDict->GetIntegerFor("ImageMask"); - m_bInterpolate = !!pDict->GetIntegerFor("Interpolate"); - m_Height = pDict->GetIntegerFor("Height"); - m_Width = pDict->GetIntegerFor("Width"); + !m_pDict->KeyExist("ColorSpace") || m_pDict->GetIntegerFor("ImageMask"); + m_bInterpolate = !!m_pDict->GetIntegerFor("Interpolate"); + m_Height = m_pDict->GetIntegerFor("Height"); + m_Width = m_pDict->GetIntegerFor("Width"); } void CPDF_Image::ConvertStreamToIndirectObject() { diff --git a/core/fpdfapi/page/cpdf_image.h b/core/fpdfapi/page/cpdf_image.h index 57cbe94ca2..23864bbf6c 100644 --- a/core/fpdfapi/page/cpdf_image.h +++ b/core/fpdfapi/page/cpdf_image.h @@ -29,6 +29,7 @@ class CPDF_Image : public Retainable { void ConvertStreamToIndirectObject(); + CPDF_Dictionary* GetInlineDict() const { return m_pDict.Get(); } CPDF_Stream* GetStream() const { return m_pStream.Get(); } CPDF_Dictionary* GetDict() const; CPDF_Dictionary* GetOC() const { return m_pOC.Get(); } @@ -67,7 +68,7 @@ class CPDF_Image : public Retainable { CPDF_Image(CPDF_Document* pDoc, uint32_t dwStreamObjNum); ~CPDF_Image() override; - void FinishInitialization(CPDF_Dictionary* pStreamDict); + void FinishInitialization(); std::unique_ptr<CPDF_Dictionary> InitJPEG(uint8_t* pData, uint32_t size); int32_t m_Height = 0; @@ -77,6 +78,7 @@ class CPDF_Image : public Retainable { bool m_bInterpolate = false; UnownedPtr<CPDF_Document> const m_pDocument; MaybeOwned<CPDF_Stream> m_pStream; + MaybeOwned<CPDF_Dictionary> m_pDict; UnownedPtr<CPDF_Dictionary> m_pOC; }; diff --git a/core/fpdfapi/render/cpdf_dibsource.cpp b/core/fpdfapi/render/cpdf_dibsource.cpp index 48715d2ecb..aff63d102d 100644 --- a/core/fpdfapi/render/cpdf_dibsource.cpp +++ b/core/fpdfapi/render/cpdf_dibsource.cpp @@ -341,9 +341,9 @@ int CPDF_DIBSource::ContinueLoadDIBSource(IFX_PauseIndicator* pPause) { } if (iDecodeStatus < 0) { - m_pJbig2Context.reset(); m_pCachedBitmap.Reset(); m_pGlobalStream.Reset(); + m_pJbig2Context.reset(); return 0; } if (iDecodeStatus == FXCODEC_STATUS_DECODE_TOBECONTINUE) diff --git a/core/fxcodec/jbig2/JBig2_Image.cpp b/core/fxcodec/jbig2/JBig2_Image.cpp index d229e0ca01..b0d75d4d96 100644 --- a/core/fxcodec/jbig2/JBig2_Image.cpp +++ b/core/fxcodec/jbig2/JBig2_Image.cpp @@ -234,8 +234,8 @@ void CJBig2_Image::expand(int32_t h, bool v) { return; if (m_pData.IsOwned()) { - m_pData.Reset(std::unique_ptr<uint8_t, FxFreeDeleter>(FX_Realloc( - uint8_t, m_pData.ReleaseAndClear().release(), h * m_nStride))); + m_pData.Reset(std::unique_ptr<uint8_t, FxFreeDeleter>( + FX_Realloc(uint8_t, m_pData.Release().release(), h * m_nStride))); } else { uint8_t* pExternalBuffer = data(); m_pData.Reset(std::unique_ptr<uint8_t, FxFreeDeleter>( diff --git a/core/fxcrt/maybe_owned.h b/core/fxcrt/maybe_owned.h index 130d2bdc3c..11dd68642d 100644 --- a/core/fxcrt/maybe_owned.h +++ b/core/fxcrt/maybe_owned.h @@ -11,7 +11,6 @@ #include "core/fxcrt/fx_memory.h" #include "core/fxcrt/fx_system.h" -#include "core/fxcrt/unowned_ptr.h" namespace fxcrt { @@ -34,34 +33,25 @@ class MaybeOwned { } void Reset(std::unique_ptr<T, D> ptr) { - m_pObj = ptr.get(); m_pOwnedObj = std::move(ptr); + m_pObj = m_pOwnedObj.get(); } void Reset(T* ptr = nullptr) { - m_pObj = ptr; m_pOwnedObj.reset(); + m_pObj = ptr; } - T* Get() const { return m_pObj.Get(); } bool IsOwned() const { return !!m_pOwnedObj; } - - // Downgrades to unowned, caller takes ownership. + T* Get() const { return m_pObj; } std::unique_ptr<T, D> Release() { ASSERT(IsOwned()); return std::move(m_pOwnedObj); } - // Downgrades to empty, caller takes ownership. - std::unique_ptr<T, D> ReleaseAndClear() { - ASSERT(IsOwned()); - m_pObj = nullptr; - return std::move(m_pOwnedObj); - } - MaybeOwned& operator=(const MaybeOwned& that) = delete; MaybeOwned& operator=(MaybeOwned&& that) { - m_pObj = that.m_pObj; m_pOwnedObj = std::move(that.m_pOwnedObj); + m_pObj = that.m_pObj; that.m_pObj = nullptr; return *this; } @@ -88,11 +78,11 @@ class MaybeOwned { explicit operator bool() const { return !!m_pObj; } T& operator*() const { return *m_pObj; } - T* operator->() const { return m_pObj.Get(); } + T* operator->() const { return m_pObj; } private: std::unique_ptr<T, D> m_pOwnedObj; - UnownedPtr<T> m_pObj; + T* m_pObj; }; } // namespace fxcrt |