OLD | NEW |
1 // Copyright 2016 PDFium Authors. All rights reserved. | 1 // Copyright 2016 PDFium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com | 5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com |
6 | 6 |
7 #ifndef CORE_FPDFAPI_PARSER_CPDF_STREAM_H_ | 7 #ifndef CORE_FPDFAPI_PARSER_CPDF_STREAM_H_ |
8 #define CORE_FPDFAPI_PARSER_CPDF_STREAM_H_ | 8 #define CORE_FPDFAPI_PARSER_CPDF_STREAM_H_ |
9 | 9 |
10 #include <memory> | 10 #include <memory> |
(...skipping 19 matching lines...) Expand all Loading... |
30 CPDF_Stream* AsStream() override; | 30 CPDF_Stream* AsStream() override; |
31 const CPDF_Stream* AsStream() const override; | 31 const CPDF_Stream* AsStream() const override; |
32 | 32 |
33 uint32_t GetRawSize() const { return m_dwSize; } | 33 uint32_t GetRawSize() const { return m_dwSize; } |
34 uint8_t* GetRawData() const { return m_pDataBuf.get(); } | 34 uint8_t* GetRawData() const { return m_pDataBuf.get(); } |
35 | 35 |
36 // Does not takes onwership of |pData|, copies into internally-owned buffer. | 36 // Does not takes onwership of |pData|, copies into internally-owned buffer. |
37 void SetData(const uint8_t* pData, uint32_t size); | 37 void SetData(const uint8_t* pData, uint32_t size); |
38 | 38 |
39 void InitStream(const uint8_t* pData, uint32_t size, CPDF_Dictionary* pDict); | 39 void InitStream(const uint8_t* pData, uint32_t size, CPDF_Dictionary* pDict); |
40 void InitStreamFromFile(IFX_FileRead* pFile, CPDF_Dictionary* pDict); | 40 void InitStreamFromFile(IFX_SeekableReadStream* pFile, |
| 41 CPDF_Dictionary* pDict); |
41 | 42 |
42 FX_BOOL ReadRawData(FX_FILESIZE start_pos, | 43 FX_BOOL ReadRawData(FX_FILESIZE start_pos, |
43 uint8_t* pBuf, | 44 uint8_t* pBuf, |
44 uint32_t buf_size) const; | 45 uint32_t buf_size) const; |
45 | 46 |
46 bool IsMemoryBased() const { return m_bMemoryBased; } | 47 bool IsMemoryBased() const { return m_bMemoryBased; } |
47 | 48 |
48 protected: | 49 protected: |
49 ~CPDF_Stream() override; | 50 ~CPDF_Stream() override; |
50 CPDF_Object* CloneNonCyclic( | 51 CPDF_Object* CloneNonCyclic( |
51 bool bDirect, | 52 bool bDirect, |
52 std::set<const CPDF_Object*>* pVisited) const override; | 53 std::set<const CPDF_Object*>* pVisited) const override; |
53 | 54 |
54 std::unique_ptr<CPDF_Dictionary, ReleaseDeleter<CPDF_Dictionary>> m_pDict; | 55 std::unique_ptr<CPDF_Dictionary, ReleaseDeleter<CPDF_Dictionary>> m_pDict; |
55 bool m_bMemoryBased = true; | 56 bool m_bMemoryBased = true; |
56 uint32_t m_dwSize = 0; | 57 uint32_t m_dwSize = 0; |
57 std::unique_ptr<uint8_t, FxFreeDeleter> m_pDataBuf; | 58 std::unique_ptr<uint8_t, FxFreeDeleter> m_pDataBuf; |
58 IFX_FileRead* m_pFile = nullptr; | 59 IFX_SeekableReadStream* m_pFile = nullptr; |
59 }; | 60 }; |
60 | 61 |
61 using UniqueStream = std::unique_ptr<CPDF_Stream, ReleaseDeleter<CPDF_Object>>; | 62 using UniqueStream = std::unique_ptr<CPDF_Stream, ReleaseDeleter<CPDF_Object>>; |
62 | 63 |
63 inline CPDF_Stream* ToStream(CPDF_Object* obj) { | 64 inline CPDF_Stream* ToStream(CPDF_Object* obj) { |
64 return obj ? obj->AsStream() : nullptr; | 65 return obj ? obj->AsStream() : nullptr; |
65 } | 66 } |
66 | 67 |
67 inline const CPDF_Stream* ToStream(const CPDF_Object* obj) { | 68 inline const CPDF_Stream* ToStream(const CPDF_Object* obj) { |
68 return obj ? obj->AsStream() : nullptr; | 69 return obj ? obj->AsStream() : nullptr; |
69 } | 70 } |
70 | 71 |
71 inline UniqueStream ToStream(UniqueObject obj) { | 72 inline UniqueStream ToStream(UniqueObject obj) { |
72 CPDF_Stream* pStream = ToStream(obj.get()); | 73 CPDF_Stream* pStream = ToStream(obj.get()); |
73 if (!pStream) | 74 if (!pStream) |
74 return nullptr; | 75 return nullptr; |
75 obj.release(); | 76 obj.release(); |
76 return UniqueStream(pStream); | 77 return UniqueStream(pStream); |
77 } | 78 } |
78 | 79 |
79 #endif // CORE_FPDFAPI_PARSER_CPDF_STREAM_H_ | 80 #endif // CORE_FPDFAPI_PARSER_CPDF_STREAM_H_ |
OLD | NEW |