Merge to XFA: Add type cast definitions for CPDF_Stream.

core/src/fpdfapi/fpdf_page/fpdf_page_func.cpp

 // Copyright 2014 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
 
 #include <limits.h>
 
 #include "../../../../third_party/base/numerics/safe_conversions_impl.h"
 #include "../../../include/fpdfapi/fpdf_module.h"
@@ skipping 473 matching lines @@
   m_pSampleStream = NULL;
   m_pEncodeInfo = NULL;
   m_pDecodeInfo = NULL;
 }
 CPDF_SampledFunc::~CPDF_SampledFunc() {
   delete m_pSampleStream;
   FX_Free(m_pEncodeInfo);
   FX_Free(m_pDecodeInfo);
 }
 FX_BOOL CPDF_SampledFunc::v_Init(CPDF_Object* pObj) {
-  if (pObj->GetType() != PDFOBJ_STREAM) {
-    return FALSE;
-  }
-  CPDF_Stream* pStream = (CPDF_Stream*)pObj;
+  CPDF_Stream* pStream = pObj->AsStream();
+  if (!pStream)
+    return false;
+
   CPDF_Dictionary* pDict = pStream->GetDict();
   CPDF_Array* pSize = pDict->GetArray(FX_BSTRC("Size"));
   CPDF_Array* pEncode = pDict->GetArray(FX_BSTRC("Encode"));
   CPDF_Array* pDecode = pDict->GetArray(FX_BSTRC("Decode"));
   m_nBitsPerSample = pDict->GetInteger(FX_BSTRC("BitsPerSample"));
   if (m_nBitsPerSample > 32) {
     return FALSE;
   }
   m_SampleMax = 0xffffffff >> (32 - m_nBitsPerSample);
   m_pSampleStream = new CPDF_StreamAcc;
@@ skipping 117 matching lines @@
 class CPDF_PSFunc : public CPDF_Function {
  public:
   // CPDF_Function
   FX_BOOL v_Init(CPDF_Object* pObj) override;
   FX_BOOL v_Call(FX_FLOAT* inputs, FX_FLOAT* results) const override;
 
   CPDF_PSEngine m_PS;
 };
 
 FX_BOOL CPDF_PSFunc::v_Init(CPDF_Object* pObj) {
-  CPDF_Stream* pStream = (CPDF_Stream*)pObj;
+  CPDF_Stream* pStream = pObj->AsStream();
   CPDF_StreamAcc acc;
   acc.LoadAllData(pStream, FALSE);
   return m_PS.Parse((const FX_CHAR*)acc.GetData(), acc.GetSize());
 }
 FX_BOOL CPDF_PSFunc::v_Call(FX_FLOAT* inputs, FX_FLOAT* results) const {
   CPDF_PSEngine& PS = (CPDF_PSEngine&)m_PS;
   PS.Reset();
   int i;
   for (i = 0; i < m_nInputs; i++) {
     PS.Push(inputs[i]);
@@ skipping 161 matching lines @@
   int nresults;
   m_pSubFunctions[i]->Call(&input, m_nInputs, outputs, nresults);
   return TRUE;
 }
 CPDF_Function* CPDF_Function::Load(CPDF_Object* pFuncObj) {
   if (pFuncObj == NULL) {
     return NULL;
   }
   CPDF_Function* pFunc = NULL;
   int type;
-  if (pFuncObj->GetType() == PDFOBJ_STREAM) {
-    type = ((CPDF_Stream*)pFuncObj)
-               ->GetDict()
-               ->GetInteger(FX_BSTRC("FunctionType"));
+  if (CPDF_Stream* pStream = pFuncObj->AsStream()) {
+    type = pStream->GetDict()->GetInteger(FX_BSTRC("FunctionType"));
   } else if (CPDF_Dictionary* pDict = pFuncObj->AsDictionary()) {
     type = pDict->GetInteger(FX_BSTRC("FunctionType"));
   } else {
     return NULL;
   }
   if (type == 0) {
     pFunc = new CPDF_SampledFunc;
   } else if (type == 2) {
     pFunc = new CPDF_ExpIntFunc;
   } else if (type == 3) {
@@ skipping 11 matching lines @@
 }
 CPDF_Function::CPDF_Function() {
   m_pDomains = NULL;
   m_pRanges = NULL;
 }
 CPDF_Function::~CPDF_Function() {
   FX_Free(m_pDomains);
   FX_Free(m_pRanges);
 }
 FX_BOOL CPDF_Function::Init(CPDF_Object* pObj) {
-  CPDF_Dictionary* pDict;
-  if (pObj->GetType() == PDFOBJ_STREAM) {
-    pDict = ((CPDF_Stream*)pObj)->GetDict();
-  } else {
-    pDict = pObj->AsDictionary();
-  }
+  CPDF_Stream* pStream = pObj->AsStream();
+  CPDF_Dictionary* pDict = pStream ? pStream->GetDict() : pObj->AsDictionary();
+
   CPDF_Array* pDomains = pDict->GetArray(FX_BSTRC("Domain"));
-  if (pDomains == NULL) {
+  if (!pDomains)
     return FALSE;
-  }
+
   m_nInputs = pDomains->GetCount() / 2;
-  if (m_nInputs == 0) {
+  if (m_nInputs == 0)
     return FALSE;
-  }
+
   m_pDomains = FX_Alloc2D(FX_FLOAT, m_nInputs, 2);
   for (int i = 0; i < m_nInputs * 2; i++) {
     m_pDomains[i] = pDomains->GetFloat(i);
   }
   CPDF_Array* pRanges = pDict->GetArray(FX_BSTRC("Range"));
   m_nOutputs = 0;
   if (pRanges) {
     m_nOutputs = pRanges->GetCount() / 2;
     m_pRanges = FX_Alloc2D(FX_FLOAT, m_nOutputs, 2);
     for (int i = 0; i < m_nOutputs * 2; i++) {
@@ skipping 33 matching lines @@
     for (int i = 0; i < m_nOutputs; i++) {
       if (results[i] < m_pRanges[i * 2]) {
         results[i] = m_pRanges[i * 2];
       } else if (results[i] > m_pRanges[i * 2 + 1]) {
         results[i] = m_pRanges[i * 2 + 1];
       }
     }
   }
   return TRUE;
 }