1 // Copyright 2014 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 // Original code is licensed as follows:
7 /*
8 * Copyright 2008 ZXing authors
9 *
10 * Licensed under the Apache License, Version 2.0 (the "License");
11 * you may not use this file except in compliance with the License.
12 * You may obtain a copy of the License at
13 *
14 * http://www.apache.org/licenses/LICENSE-2.0
15 *
16 * Unless required by applicable law or agreed to in writing, software
17 * distributed under the License is distributed on an "AS IS" BASIS,
18 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
19 * See the License for the specific language governing permissions and
20 * limitations under the License.
21 */
22
23 #include "fxbarcode/datamatrix/BC_DataMatrixWriter.h"
24
25 #include <memory>
26
27 #include "fxbarcode/BC_TwoDimWriter.h"
28 #include "fxbarcode/BC_UtilCodingConvert.h"
29 #include "fxbarcode/BC_Writer.h"
30 #include "fxbarcode/common/BC_CommonBitMatrix.h"
31 #include "fxbarcode/common/BC_CommonByteMatrix.h"
32 #include "fxbarcode/datamatrix/BC_ASCIIEncoder.h"
33 #include "fxbarcode/datamatrix/BC_Base256Encoder.h"
34 #include "fxbarcode/datamatrix/BC_C40Encoder.h"
35 #include "fxbarcode/datamatrix/BC_DataMatrixSymbolInfo144.h"
36 #include "fxbarcode/datamatrix/BC_DefaultPlacement.h"
37 #include "fxbarcode/datamatrix/BC_EdifactEncoder.h"
38 #include "fxbarcode/datamatrix/BC_Encoder.h"
39 #include "fxbarcode/datamatrix/BC_EncoderContext.h"
40 #include "fxbarcode/datamatrix/BC_ErrorCorrection.h"
41 #include "fxbarcode/datamatrix/BC_HighLevelEncoder.h"
42 #include "fxbarcode/datamatrix/BC_SymbolInfo.h"
43 #include "fxbarcode/datamatrix/BC_TextEncoder.h"
44 #include "fxbarcode/datamatrix/BC_X12Encoder.h"
45 #include "third_party/base/ptr_util.h"
46
47 namespace {
48
encodeLowLevel(CBC_DefaultPlacement * placement,CBC_SymbolInfo * symbolInfo)49 std::unique_ptr<CBC_CommonByteMatrix> encodeLowLevel(
50 CBC_DefaultPlacement* placement,
51 CBC_SymbolInfo* symbolInfo) {
52 int32_t symbolWidth = symbolInfo->getSymbolDataWidth();
53 ASSERT(symbolWidth);
54 int32_t symbolHeight = symbolInfo->getSymbolDataHeight();
55 ASSERT(symbolHeight);
56 int32_t width = symbolInfo->getSymbolWidth();
57 ASSERT(width);
58 int32_t height = symbolInfo->getSymbolHeight();
59 ASSERT(height);
60
61 auto matrix = pdfium::MakeUnique<CBC_CommonByteMatrix>(width, height);
62 matrix->Init();
63 int32_t matrixY = 0;
64 for (int32_t y = 0; y < symbolHeight; y++) {
65 int32_t matrixX;
66 if ((y % symbolInfo->matrixHeight()) == 0) {
67 matrixX = 0;
68 for (int32_t x = 0; x < width; x++) {
69 matrix->Set(matrixX, matrixY, x % 2 == 0);
70 matrixX++;
71 }
72 matrixY++;
73 }
74 matrixX = 0;
75 for (int32_t x = 0; x < symbolWidth; x++) {
76 if (x % symbolInfo->matrixWidth() == 0) {
77 matrix->Set(matrixX, matrixY, true);
78 matrixX++;
79 }
80 matrix->Set(matrixX, matrixY, placement->getBit(x, y));
81 matrixX++;
82 if (x % symbolInfo->matrixWidth() == symbolInfo->matrixWidth() - 1) {
83 matrix->Set(matrixX, matrixY, y % 2 == 0);
84 matrixX++;
85 }
86 }
87 matrixY++;
88 if (y % symbolInfo->matrixHeight() == symbolInfo->matrixHeight() - 1) {
89 matrixX = 0;
90 for (int32_t x = 0; x < width; x++) {
91 matrix->Set(matrixX, matrixY, true);
92 matrixX++;
93 }
94 matrixY++;
95 }
96 }
97 return matrix;
98 }
99
100 } // namespace
101
CBC_DataMatrixWriter()102 CBC_DataMatrixWriter::CBC_DataMatrixWriter() {}
103
~CBC_DataMatrixWriter()104 CBC_DataMatrixWriter::~CBC_DataMatrixWriter() {}
105
SetErrorCorrectionLevel(int32_t level)106 bool CBC_DataMatrixWriter::SetErrorCorrectionLevel(int32_t level) {
107 m_iCorrectLevel = level;
108 return true;
109 }
110
Encode(const WideString & contents,int32_t & outWidth,int32_t & outHeight)111 uint8_t* CBC_DataMatrixWriter::Encode(const WideString& contents,
112 int32_t& outWidth,
113 int32_t& outHeight) {
114 if (outWidth < 0 || outHeight < 0)
115 return nullptr;
116
117 WideString ecLevel;
118 int32_t e = BCExceptionNO;
119 WideString encoded =
120 CBC_HighLevelEncoder::encodeHighLevel(contents, ecLevel, false, e);
121 if (e != BCExceptionNO)
122 return nullptr;
123 CBC_SymbolInfo* symbolInfo =
124 CBC_SymbolInfo::lookup(encoded.GetLength(), false, e);
125 if (e != BCExceptionNO)
126 return nullptr;
127 WideString codewords =
128 CBC_ErrorCorrection::encodeECC200(encoded, symbolInfo, e);
129 if (e != BCExceptionNO)
130 return nullptr;
131
132 int32_t width = symbolInfo->getSymbolDataWidth();
133 ASSERT(width);
134 int32_t height = symbolInfo->getSymbolDataHeight();
135 ASSERT(height);
136
137 auto placement =
138 pdfium::MakeUnique<CBC_DefaultPlacement>(codewords, width, height);
139 placement->place();
140 auto bytematrix = encodeLowLevel(placement.get(), symbolInfo);
141 if (!bytematrix)
142 return nullptr;
143
144 outWidth = bytematrix->GetWidth();
145 outHeight = bytematrix->GetHeight();
146 uint8_t* result = FX_Alloc2D(uint8_t, outWidth, outHeight);
147 memcpy(result, bytematrix->GetArray(), outWidth * outHeight);
148 return result;
149 }
150