blob: ac3e24cfd9de0d4f92a295a0b4f8b952984c0cdc [file] [log] [blame] [edit]
// 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
// Original code is licensed as follows:
/*
* Copyright 2008 ZXing authors
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "fxbarcode/datamatrix/BC_DataMatrixWriter.h"
#include <memory>
#include "fxbarcode/BC_TwoDimWriter.h"
#include "fxbarcode/BC_Writer.h"
#include "fxbarcode/common/BC_CommonBitMatrix.h"
#include "fxbarcode/common/BC_CommonByteMatrix.h"
#include "fxbarcode/datamatrix/BC_ASCIIEncoder.h"
#include "fxbarcode/datamatrix/BC_Base256Encoder.h"
#include "fxbarcode/datamatrix/BC_C40Encoder.h"
#include "fxbarcode/datamatrix/BC_DataMatrixSymbolInfo144.h"
#include "fxbarcode/datamatrix/BC_DefaultPlacement.h"
#include "fxbarcode/datamatrix/BC_EdifactEncoder.h"
#include "fxbarcode/datamatrix/BC_Encoder.h"
#include "fxbarcode/datamatrix/BC_EncoderContext.h"
#include "fxbarcode/datamatrix/BC_ErrorCorrection.h"
#include "fxbarcode/datamatrix/BC_HighLevelEncoder.h"
#include "fxbarcode/datamatrix/BC_SymbolInfo.h"
#include "fxbarcode/datamatrix/BC_TextEncoder.h"
#include "fxbarcode/datamatrix/BC_X12Encoder.h"
#include "third_party/base/ptr_util.h"
#include "third_party/base/stl_util.h"
namespace {
std::unique_ptr<CBC_CommonByteMatrix> encodeLowLevel(
CBC_DefaultPlacement* placement,
const CBC_SymbolInfo* symbolInfo) {
int32_t symbolWidth = symbolInfo->GetSymbolDataWidth();
ASSERT(symbolWidth);
int32_t symbolHeight = symbolInfo->GetSymbolDataHeight();
ASSERT(symbolHeight);
int32_t width = symbolInfo->GetSymbolWidth();
ASSERT(width);
int32_t height = symbolInfo->GetSymbolHeight();
ASSERT(height);
auto matrix = pdfium::MakeUnique<CBC_CommonByteMatrix>(width, height);
int32_t matrixY = 0;
for (int32_t y = 0; y < symbolHeight; y++) {
int32_t matrixX;
if ((y % symbolInfo->matrix_height()) == 0) {
matrixX = 0;
for (int32_t x = 0; x < width; x++) {
matrix->Set(matrixX, matrixY, x % 2 == 0);
matrixX++;
}
matrixY++;
}
matrixX = 0;
for (int32_t x = 0; x < symbolWidth; x++) {
if (x % symbolInfo->matrix_width() == 0) {
matrix->Set(matrixX, matrixY, true);
matrixX++;
}
matrix->Set(matrixX, matrixY, placement->getBit(x, y));
matrixX++;
if (x % symbolInfo->matrix_width() == symbolInfo->matrix_width() - 1) {
matrix->Set(matrixX, matrixY, y % 2 == 0);
matrixX++;
}
}
matrixY++;
if (y % symbolInfo->matrix_height() == symbolInfo->matrix_height() - 1) {
matrixX = 0;
for (int32_t x = 0; x < width; x++) {
matrix->Set(matrixX, matrixY, true);
matrixX++;
}
matrixY++;
}
}
return matrix;
}
} // namespace
CBC_DataMatrixWriter::CBC_DataMatrixWriter() : CBC_TwoDimWriter(true) {}
CBC_DataMatrixWriter::~CBC_DataMatrixWriter() = default;
bool CBC_DataMatrixWriter::SetErrorCorrectionLevel(int32_t level) {
set_error_correction_level(level);
return true;
}
std::vector<uint8_t, FxAllocAllocator<uint8_t>> CBC_DataMatrixWriter::Encode(
const WideString& contents,
int32_t* pOutWidth,
int32_t* pOutHeight) {
std::vector<uint8_t, FxAllocAllocator<uint8_t>> results;
WideString encoded = CBC_HighLevelEncoder::EncodeHighLevel(contents);
if (encoded.IsEmpty())
return results;
const CBC_SymbolInfo* pSymbolInfo =
CBC_SymbolInfo::Lookup(encoded.GetLength(), false);
if (!pSymbolInfo)
return results;
WideString codewords =
CBC_ErrorCorrection::EncodeECC200(encoded, pSymbolInfo);
if (codewords.IsEmpty())
return results;
int32_t width = pSymbolInfo->GetSymbolDataWidth();
ASSERT(width);
int32_t height = pSymbolInfo->GetSymbolDataHeight();
ASSERT(height);
auto placement =
pdfium::MakeUnique<CBC_DefaultPlacement>(codewords, width, height);
placement->place();
auto bytematrix = encodeLowLevel(placement.get(), pSymbolInfo);
if (!bytematrix)
return results;
*pOutWidth = bytematrix->GetWidth();
*pOutHeight = bytematrix->GetHeight();
results = pdfium::Vector2D<uint8_t, FxAllocAllocator<uint8_t>>(*pOutWidth,
*pOutHeight);
memcpy(results.data(), bytematrix->GetArray().data(),
*pOutWidth * *pOutHeight);
return results;
}