1 // Copyright 2014 The PDFium Authors
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 <stdint.h>
26
27 #include <memory>
28
29 #include "core/fxcrt/check.h"
30 #include "core/fxcrt/data_vector.h"
31 #include "fxbarcode/BC_TwoDimWriter.h"
32 #include "fxbarcode/BC_Writer.h"
33 #include "fxbarcode/common/BC_CommonByteMatrix.h"
34 #include "fxbarcode/datamatrix/BC_ASCIIEncoder.h"
35 #include "fxbarcode/datamatrix/BC_Base256Encoder.h"
36 #include "fxbarcode/datamatrix/BC_C40Encoder.h"
37 #include "fxbarcode/datamatrix/BC_DataMatrixSymbolInfo144.h"
38 #include "fxbarcode/datamatrix/BC_DefaultPlacement.h"
39 #include "fxbarcode/datamatrix/BC_EdifactEncoder.h"
40 #include "fxbarcode/datamatrix/BC_Encoder.h"
41 #include "fxbarcode/datamatrix/BC_EncoderContext.h"
42 #include "fxbarcode/datamatrix/BC_ErrorCorrection.h"
43 #include "fxbarcode/datamatrix/BC_HighLevelEncoder.h"
44 #include "fxbarcode/datamatrix/BC_SymbolInfo.h"
45 #include "fxbarcode/datamatrix/BC_TextEncoder.h"
46 #include "fxbarcode/datamatrix/BC_X12Encoder.h"
47
48 namespace {
49
EncodeLowLevel(CBC_DefaultPlacement * placement,const CBC_SymbolInfo * symbolInfo)50 std::unique_ptr<CBC_CommonByteMatrix> EncodeLowLevel(
51 CBC_DefaultPlacement* placement,
52 const CBC_SymbolInfo* symbolInfo) {
53 int32_t symbolWidth = symbolInfo->GetSymbolDataWidth();
54 DCHECK(symbolWidth);
55 int32_t symbolHeight = symbolInfo->GetSymbolDataHeight();
56 DCHECK(symbolHeight);
57 int32_t width = symbolInfo->GetSymbolWidth();
58 DCHECK(width);
59 int32_t height = symbolInfo->GetSymbolHeight();
60 DCHECK(height);
61
62 auto matrix = std::make_unique<CBC_CommonByteMatrix>(width, height);
63 int32_t matrixY = 0;
64 for (int32_t y = 0; y < symbolHeight; y++) {
65 int32_t matrixX;
66 if ((y % symbolInfo->matrix_height()) == 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->matrix_width() == 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->matrix_width() == symbolInfo->matrix_width() - 1) {
83 matrix->Set(matrixX, matrixY, y % 2 == 0);
84 matrixX++;
85 }
86 }
87 matrixY++;
88 if (y % symbolInfo->matrix_height() == symbolInfo->matrix_height() - 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() : CBC_TwoDimWriter(true) {}
103
104 CBC_DataMatrixWriter::~CBC_DataMatrixWriter() = default;
105
SetErrorCorrectionLevel(int32_t level)106 bool CBC_DataMatrixWriter::SetErrorCorrectionLevel(int32_t level) {
107 set_error_correction_level(level);
108 return true;
109 }
110
Encode(const WideString & contents,int32_t * pOutWidth,int32_t * pOutHeight)111 DataVector<uint8_t> CBC_DataMatrixWriter::Encode(const WideString& contents,
112 int32_t* pOutWidth,
113 int32_t* pOutHeight) {
114 WideString encoded = CBC_HighLevelEncoder::EncodeHighLevel(contents);
115 if (encoded.IsEmpty())
116 return DataVector<uint8_t>();
117
118 const CBC_SymbolInfo* pSymbolInfo =
119 CBC_SymbolInfo::Lookup(encoded.GetLength(), false);
120 if (!pSymbolInfo)
121 return DataVector<uint8_t>();
122
123 WideString codewords =
124 CBC_ErrorCorrection::EncodeECC200(encoded, pSymbolInfo);
125 if (codewords.IsEmpty())
126 return DataVector<uint8_t>();
127
128 int32_t width = pSymbolInfo->GetSymbolDataWidth();
129 DCHECK(width);
130 int32_t height = pSymbolInfo->GetSymbolDataHeight();
131 DCHECK(height);
132
133 auto placement =
134 std::make_unique<CBC_DefaultPlacement>(codewords, width, height);
135 auto bytematrix = EncodeLowLevel(placement.get(), pSymbolInfo);
136 DCHECK(bytematrix);
137
138 *pOutWidth = bytematrix->GetWidth();
139 *pOutHeight = bytematrix->GetHeight();
140 return bytematrix->TakeArray();
141 }
142