1 // Copyright 2017 PDFium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com
6
7 #include "core/fpdfapi/page/cpdf_stitchfunc.h"
8
9 #include <utility>
10
11 #include "core/fpdfapi/parser/cpdf_array.h"
12 #include "core/fpdfapi/parser/cpdf_dictionary.h"
13 #include "core/fxcrt/fx_memory.h"
14
CPDF_StitchFunc()15 CPDF_StitchFunc::CPDF_StitchFunc()
16 : CPDF_Function(Type::kType3Stitching),
17 m_pBounds(nullptr),
18 m_pEncode(nullptr) {}
19
~CPDF_StitchFunc()20 CPDF_StitchFunc::~CPDF_StitchFunc() {
21 FX_Free(m_pBounds);
22 FX_Free(m_pEncode);
23 }
24
v_Init(CPDF_Object * pObj)25 bool CPDF_StitchFunc::v_Init(CPDF_Object* pObj) {
26 CPDF_Dictionary* pDict = pObj->GetDict();
27 if (!pDict) {
28 return false;
29 }
30 if (m_nInputs != kRequiredNumInputs) {
31 return false;
32 }
33 CPDF_Array* pArray = pDict->GetArrayFor("Functions");
34 if (!pArray) {
35 return false;
36 }
37 uint32_t nSubs = pArray->GetCount();
38 if (nSubs == 0)
39 return false;
40 m_nOutputs = 0;
41 for (uint32_t i = 0; i < nSubs; i++) {
42 CPDF_Object* pSub = pArray->GetDirectObjectAt(i);
43 if (pSub == pObj)
44 return false;
45 std::unique_ptr<CPDF_Function> pFunc(CPDF_Function::Load(pSub));
46 if (!pFunc)
47 return false;
48 // Check that the input dimensionality is 1, and that all output
49 // dimensionalities are the same.
50 if (pFunc->CountInputs() != kRequiredNumInputs)
51 return false;
52 if (pFunc->CountOutputs() != m_nOutputs) {
53 if (m_nOutputs)
54 return false;
55
56 m_nOutputs = pFunc->CountOutputs();
57 }
58
59 m_pSubFunctions.push_back(std::move(pFunc));
60 }
61 m_pBounds = FX_Alloc(float, nSubs + 1);
62 m_pBounds[0] = m_pDomains[0];
63 pArray = pDict->GetArrayFor("Bounds");
64 if (!pArray)
65 return false;
66 for (uint32_t i = 0; i < nSubs - 1; i++)
67 m_pBounds[i + 1] = pArray->GetFloatAt(i);
68 m_pBounds[nSubs] = m_pDomains[1];
69 m_pEncode = FX_Alloc2D(float, nSubs, 2);
70 pArray = pDict->GetArrayFor("Encode");
71 if (!pArray)
72 return false;
73
74 for (uint32_t i = 0; i < nSubs * 2; i++)
75 m_pEncode[i] = pArray->GetFloatAt(i);
76 return true;
77 }
78
v_Call(float * inputs,float * outputs) const79 bool CPDF_StitchFunc::v_Call(float* inputs, float* outputs) const {
80 float input = inputs[0];
81 size_t i;
82 for (i = 0; i < m_pSubFunctions.size() - 1; i++) {
83 if (input < m_pBounds[i + 1])
84 break;
85 }
86 input = Interpolate(input, m_pBounds[i], m_pBounds[i + 1], m_pEncode[i * 2],
87 m_pEncode[i * 2 + 1]);
88 int nresults;
89 m_pSubFunctions[i]->Call(&input, kRequiredNumInputs, outputs, &nresults);
90 return true;
91 }
92