mirror of
https://github.com/ONLYOFFICE/core.git
synced 2026-07-10 23:03:14 +08:00
451 lines
13 KiB
C++
451 lines
13 KiB
C++
/*
|
|
* (c) Copyright Ascensio System SIA 2010-2017
|
|
*
|
|
* This program is a free software product. You can redistribute it and/or
|
|
* modify it under the terms of the GNU Affero General Public License (AGPL)
|
|
* version 3 as published by the Free Software Foundation. In accordance with
|
|
* Section 7(a) of the GNU AGPL its Section 15 shall be amended to the effect
|
|
* that Ascensio System SIA expressly excludes the warranty of non-infringement
|
|
* of any third-party rights.
|
|
*
|
|
* This program is distributed WITHOUT ANY WARRANTY; without even the implied
|
|
* warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. For
|
|
* details, see the GNU AGPL at: http://www.gnu.org/licenses/agpl-3.0.html
|
|
*
|
|
* You can contact Ascensio System SIA at Lubanas st. 125a-25, Riga, Latvia,
|
|
* EU, LV-1021.
|
|
*
|
|
* The interactive user interfaces in modified source and object code versions
|
|
* of the Program must display Appropriate Legal Notices, as required under
|
|
* Section 5 of the GNU AGPL version 3.
|
|
*
|
|
* Pursuant to Section 7(b) of the License you must retain the original Product
|
|
* logo when distributing the program. Pursuant to Section 7(e) we decline to
|
|
* grant you any rights under trademark law for use of our trademarks.
|
|
*
|
|
* All the Product's GUI elements, including illustrations and icon sets, as
|
|
* well as technical writing content are licensed under the terms of the
|
|
* Creative Commons Attribution-ShareAlike 4.0 International. See the License
|
|
* terms at http://creativecommons.org/licenses/by-sa/4.0/legalcode
|
|
*
|
|
*/
|
|
#include "Image.h"
|
|
#include "Streams.h"
|
|
#include "Document.h"
|
|
|
|
#include "../../DesktopEditor/raster/JBig2/source/Encoder/jbig2enc.h"
|
|
#include "../../DesktopEditor/raster/JBig2/source/LeptonLib/allheaders.h"
|
|
|
|
namespace PdfWriter
|
|
{
|
|
//----------------------------------------------------------------------------------------
|
|
// CImageDict
|
|
//----------------------------------------------------------------------------------------
|
|
CImageDict::CImageDict(CXref* pXref, CDocument* pDocument)
|
|
{
|
|
m_pXref = pXref;
|
|
m_pDocument = pDocument;
|
|
}
|
|
void CImageDict::LoadJpeg(const wchar_t* wsFilePath, unsigned int unWidth, unsigned int unHeight, bool bGrayScale)
|
|
{
|
|
CImageFileStream* pStream = new CImageFileStream();
|
|
if (!pStream)
|
|
return;
|
|
|
|
pStream->OpenFile(wsFilePath);
|
|
SetStream(m_pXref, pStream);
|
|
Add("Type", "XObject");
|
|
Add("Subtype", "Image");
|
|
Add("Height", unHeight);
|
|
Add("Width", unWidth);
|
|
|
|
if (bGrayScale)
|
|
Add("ColorSpace", "DeviceGray");
|
|
else
|
|
Add("ColorSpace", "DeviceRGB");
|
|
|
|
Add("BitsPerComponent", 8);
|
|
SetFilter(STREAM_FILTER_DCT_DECODE);
|
|
}
|
|
void CImageDict::LoadJpeg(BYTE* pBuffer, int nBufferSize, unsigned int unWidth, unsigned int unHeight, bool bGrayScale)
|
|
{
|
|
CMemoryStream* pStream = new CMemoryStream();
|
|
if (!pStream)
|
|
return;
|
|
|
|
pStream->Write(pBuffer, nBufferSize);
|
|
SetStream(m_pXref, pStream);
|
|
Add("Type", "XObject");
|
|
Add("Subtype", "Image");
|
|
Add("Height", unHeight);
|
|
Add("Width", unWidth);
|
|
|
|
if (bGrayScale)
|
|
Add("ColorSpace", "DeviceGray");
|
|
else
|
|
Add("ColorSpace", "DeviceRGB");
|
|
|
|
Add("BitsPerComponent", 8);
|
|
SetFilter(STREAM_FILTER_DCT_DECODE);
|
|
}
|
|
void CImageDict::LoadJpx(const wchar_t* wsFilePath, unsigned int unWidth, unsigned int unHeight)
|
|
{
|
|
CImageFileStream* pStream = new CImageFileStream();
|
|
if (!pStream)
|
|
return;
|
|
|
|
pStream->OpenFile(wsFilePath);
|
|
SetStream(m_pXref, pStream);
|
|
Add("Type", "XObject");
|
|
Add("Subtype", "Image");
|
|
Add("Height", unHeight);
|
|
Add("Width", unWidth);
|
|
Add("ColorSpace", "DeviceRGB");
|
|
Add("BitsPerComponent", 8);
|
|
SetFilter(STREAM_FILTER_JPX_DECODE);
|
|
}
|
|
void CImageDict::LoadJpx(BYTE* pBuffer, int nBufferSize, unsigned int unWidth, unsigned int unHeight)
|
|
{
|
|
CMemoryStream* pStream = new CMemoryStream();
|
|
if (!pStream)
|
|
return;
|
|
|
|
pStream->Write(pBuffer, nBufferSize);
|
|
SetStream(m_pXref, pStream);
|
|
Add("Type", "XObject");
|
|
Add("Subtype", "Image");
|
|
Add("Height", unHeight);
|
|
Add("Width", unWidth);
|
|
Add("ColorSpace", "DeviceRGB");
|
|
Add("BitsPerComponent", 8);
|
|
SetFilter(STREAM_FILTER_JPX_DECODE);
|
|
}
|
|
void CImageDict::LoadJb2(const wchar_t* wsFilePath, unsigned int unWidth, unsigned int unHeight)
|
|
{
|
|
CImageFileStream* pStream = new CImageFileStream();
|
|
if (!pStream)
|
|
return;
|
|
|
|
pStream->OpenFile(wsFilePath);
|
|
SetStream(m_pXref, pStream);
|
|
Add("Type", "XObject");
|
|
Add("Subtype", "Image");
|
|
Add("Height", unHeight);
|
|
Add("Width", unWidth);
|
|
Add("ColorSpace", "DeviceGray");
|
|
Add("BitsPerComponent", 1);
|
|
SetFilter(STREAM_FILTER_JBIG2_DECODE);
|
|
}
|
|
void CImageDict::LoadCCITT4(const wchar_t* wsTempFile, unsigned int unWidth, unsigned int unHeight)
|
|
{
|
|
CImageFileStream* pStream = new CImageFileStream();
|
|
if (!pStream)
|
|
return;
|
|
|
|
pStream->OpenFile(wsTempFile);
|
|
SetStream(m_pXref, pStream);
|
|
Add("Type", "XObject");
|
|
Add("Subtype", "Image");
|
|
Add("Height", unHeight);
|
|
Add("Width", unWidth);
|
|
Add("ColorSpace", "DeviceGray");
|
|
Add("BitsPerComponent", 1);
|
|
|
|
CArrayObject* pDecodeParams = new CArrayObject();
|
|
CDictObject* pParams = new CDictObject();
|
|
pDecodeParams->Add(pParams);
|
|
pParams->Add("K", 0);
|
|
pParams->Add("Columns", unWidth);
|
|
pParams->Add("EncodedByteAlign", true);
|
|
Add("DecodeParams", pDecodeParams);
|
|
|
|
SetFilter(STREAM_FILTER_CCITT_DECODE);
|
|
|
|
}
|
|
void CImageDict::LoadRaw(const BYTE* pBgra, unsigned int unWidth, unsigned int unHeight)
|
|
{
|
|
CMemoryStream* pStream = new CMemoryStream(3 * unWidth * unHeight);
|
|
if (!pStream)
|
|
return;
|
|
|
|
for (unsigned int unIndex = 0, unSize = 4 * unWidth * unHeight; unIndex < unSize; unIndex += 4)
|
|
{
|
|
pStream->Write(pBgra + unIndex, 3);
|
|
}
|
|
|
|
SetStream(m_pXref, pStream);
|
|
Add("Type", "XObject");
|
|
Add("Subtype", "Image");
|
|
Add("ColorSpace", "DeviceRGB");
|
|
Add("Width", unWidth);
|
|
Add("Height", unHeight);
|
|
Add("BitsPerComponent", 8);
|
|
SetFilter(STREAM_FILTER_FLATE_DECODE);
|
|
}
|
|
void CImageDict::LoadRaw(const BYTE* pBuffer, unsigned int unSize, unsigned int unWidth, unsigned int unHeight)
|
|
{
|
|
CMemoryStream* pStream = new CMemoryStream(unSize);
|
|
if (!pStream)
|
|
return;
|
|
pStream->Write(pBuffer, unSize);
|
|
SetStream(m_pXref, pStream);
|
|
Add("Type", "XObject");
|
|
Add("Subtype", "Image");
|
|
Add("ColorSpace", "DeviceRGB");
|
|
Add("Width", unWidth);
|
|
Add("Height", unHeight);
|
|
Add("BitsPerComponent", 8);
|
|
SetFilter(STREAM_FILTER_FLATE_DECODE);
|
|
}
|
|
void CImageDict::LoadSMask(const BYTE* pBgra, unsigned int unWidth, unsigned int unHeight, unsigned char unAlpha, bool bVerFlip)
|
|
{
|
|
CMemoryStream* pStream = new CMemoryStream(unWidth * unHeight);
|
|
if (!pStream)
|
|
return;
|
|
|
|
if (255 != unAlpha)
|
|
{
|
|
double dKoef = unAlpha / 255.0;
|
|
|
|
if (!bVerFlip)
|
|
{
|
|
for (unsigned int unIndex = 0, unSize = 4 * unWidth * unHeight; unIndex < unSize; unIndex += 4)
|
|
{
|
|
BYTE nChar = *(pBgra + unIndex + 3) * dKoef;
|
|
pStream->Write(&nChar, 1);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
int nWidth = (int)unWidth;
|
|
int nHeight = (int)unHeight;
|
|
for (int nY = nHeight - 1; nY >= 0; nY--)
|
|
{
|
|
for (int nX = 0; nX < nWidth; nX++)
|
|
{
|
|
int unIndex = 4 * (nX + nY * nWidth);
|
|
BYTE nChar = *(pBgra + unIndex + 3) * dKoef;
|
|
pStream->Write(&nChar, 1);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
if (!bVerFlip)
|
|
{
|
|
for (unsigned int unIndex = 0, unSize = 4 * unWidth * unHeight; unIndex < unSize; unIndex += 4)
|
|
{
|
|
pStream->Write(pBgra + unIndex + 3, 1);
|
|
}
|
|
}
|
|
else
|
|
{
|
|
int nWidth = (int)unWidth;
|
|
int nHeight = (int)unHeight;
|
|
for (int nY = nHeight - 1; nY >= 0; nY--)
|
|
{
|
|
for (int nX = 0; nX < nWidth; nX++)
|
|
{
|
|
int unIndex = 4 * (nX + nY * nWidth);
|
|
pStream->Write(pBgra + unIndex + 3, 1);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
CImageDict* pImageSMask = new CImageDict(m_pXref, m_pDocument);
|
|
if (!pImageSMask)
|
|
return;
|
|
|
|
pImageSMask->SetStream(m_pXref, pStream);
|
|
pImageSMask->Add("Type", "XObject");
|
|
pImageSMask->Add("Subtype", "Image");
|
|
pImageSMask->Add("ColorSpace", "DeviceGray");
|
|
pImageSMask->Add("Width", unWidth);
|
|
pImageSMask->Add("Height", unHeight);
|
|
pImageSMask->Add("BitsPerComponent", 8);
|
|
|
|
//#ifndef FILTER_FLATE_DECODE_DISABLED
|
|
// pImageSMask->SetFilter(STREAM_FILTER_LZW_DECODE | STREAM_FILTER_FLATE_DECODE);
|
|
//#else
|
|
pImageSMask->SetFilter(STREAM_FILTER_FLATE_DECODE);
|
|
//#endif
|
|
|
|
Add("SMask", pImageSMask);
|
|
}
|
|
void CImageDict::LoadSMask(const BYTE* pBuffer, unsigned int unSize, unsigned int unWidth, unsigned int unHeight)
|
|
{
|
|
CMemoryStream* pStream = new CMemoryStream(unSize);
|
|
if (!pStream)
|
|
return;
|
|
|
|
pStream->Write(pBuffer, unSize);
|
|
CImageDict* pImageSMask = new CImageDict(m_pXref, m_pDocument);
|
|
if (!pImageSMask)
|
|
return;
|
|
|
|
pImageSMask->SetStream(m_pXref, pStream);
|
|
pImageSMask->Add("Type", "XObject");
|
|
pImageSMask->Add("Subtype", "Image");
|
|
pImageSMask->Add("ColorSpace", "DeviceGray");
|
|
pImageSMask->Add("Width", unWidth);
|
|
pImageSMask->Add("Height", unHeight);
|
|
pImageSMask->Add("BitsPerComponent", 8);
|
|
pImageSMask->SetFilter(STREAM_FILTER_FLATE_DECODE);
|
|
Add("SMask", pImageSMask);
|
|
}
|
|
void CImageDict::LoadBW(const BYTE* pImage, unsigned int unWidth, unsigned int unHeight, unsigned int unStride)
|
|
{
|
|
SetStream(m_pXref, new CMemoryStream());
|
|
CJbig2Global* pJbig2Global = m_pDocument->GetJbig2Global();
|
|
pJbig2Global->AddImage(pImage, unWidth, unHeight, unStride, GetStream());
|
|
|
|
Add("Type", "XObject");
|
|
Add("Subtype", "Image");
|
|
Add("Height", unHeight);
|
|
Add("Width", unWidth);
|
|
Add("ColorSpace", "DeviceGray");
|
|
Add("BitsPerComponent", 1);
|
|
SetFilter(STREAM_FILTER_JBIG2_DECODE);
|
|
|
|
CArrayObject* pDecodeParams = new CArrayObject();
|
|
CDictObject* pParams = new CDictObject();
|
|
pDecodeParams->Add(pParams);
|
|
pParams->Add("JBIG2Globals", pJbig2Global);
|
|
Add("DecodeParms", pDecodeParams);
|
|
}
|
|
void CImageDict::LoadBW(Pix* pPix, unsigned int unWidth, unsigned int unHeight)
|
|
{
|
|
SetStream(m_pXref, new CMemoryStream());
|
|
CJbig2Global* pJbig2Global = m_pDocument->GetJbig2Global();
|
|
pJbig2Global->AddImage(pPix, GetStream());
|
|
|
|
Add("Type", "XObject");
|
|
Add("Subtype", "Image");
|
|
Add("Height", unHeight);
|
|
Add("Width", unWidth);
|
|
Add("ColorSpace", "DeviceGray");
|
|
Add("BitsPerComponent", 1);
|
|
SetFilter(STREAM_FILTER_JBIG2_DECODE);
|
|
|
|
CArrayObject* pDecodeParams = new CArrayObject();
|
|
CDictObject* pParams = new CDictObject();
|
|
pDecodeParams->Add(pParams);
|
|
pParams->Add("JBIG2Globals", pJbig2Global);
|
|
Add("DecodeParms", pDecodeParams);
|
|
}
|
|
void CImageDict::LoadMask(Pix* pPix, unsigned int unWidth, unsigned int unHeight)
|
|
{
|
|
CImageDict* pMask = new CImageDict(m_pXref, m_pDocument);
|
|
if (!pMask)
|
|
return;
|
|
|
|
pMask->SetStream(m_pXref, new CMemoryStream());
|
|
CJbig2Global* pJbig2Global = m_pDocument->GetJbig2Global();
|
|
pJbig2Global->AddImage(pPix, pMask->GetStream());
|
|
|
|
pMask->Add("Type", "XObject");
|
|
pMask->Add("Subtype", "Image");
|
|
pMask->Add("Width", unWidth);
|
|
pMask->Add("Height", unHeight);
|
|
pMask->Add("BitsPerComponent", 1);
|
|
pMask->Add("ImageMask", true);
|
|
pMask->SetFilter(STREAM_FILTER_JBIG2_DECODE);
|
|
|
|
CArrayObject* pDecodeParams = new CArrayObject();
|
|
CDictObject* pParams = new CDictObject();
|
|
pDecodeParams->Add(pParams);
|
|
pParams->Add("JBIG2Globals", pJbig2Global);
|
|
pMask->Add("DecodeParms", pDecodeParams);
|
|
|
|
Add("Mask", pMask);
|
|
}
|
|
//----------------------------------------------------------------------------------------
|
|
// CJbig2Global
|
|
//----------------------------------------------------------------------------------------
|
|
CJbig2Global::CJbig2Global(CXref* pXref) : CDictObject(pXref)
|
|
{
|
|
m_pXref = pXref;
|
|
m_pContext = jbig2_init(0.85, 0.5, -1, -1, false, -1);
|
|
}
|
|
CJbig2Global::~CJbig2Global()
|
|
{
|
|
if (m_pContext)
|
|
jbig2_destroy(m_pContext);
|
|
}
|
|
void CJbig2Global::FlushStreams()
|
|
{
|
|
CStream* pStream = GetStream();
|
|
|
|
int nLen = 0;
|
|
BYTE* pBuffer = jbig2_pages_complete(m_pContext, &nLen);
|
|
|
|
if (pBuffer)
|
|
{
|
|
pStream->Write(pBuffer, nLen);
|
|
free(pBuffer);
|
|
}
|
|
|
|
for (int nIndex = 0, nCount = m_vImages.size(); nIndex < nCount; nIndex++)
|
|
{
|
|
pBuffer = jbig2_produce_page(m_pContext, nIndex, -1, -1, &nLen);
|
|
if (pBuffer)
|
|
{
|
|
pStream = m_vImages.at(nIndex);
|
|
if (pStream)
|
|
{
|
|
pStream->Write(pBuffer, nLen);
|
|
}
|
|
free(pBuffer);
|
|
}
|
|
}
|
|
|
|
jbig2_destroy(m_pContext);
|
|
m_pContext = NULL;
|
|
}
|
|
void CJbig2Global::AddImage(const BYTE* pImage, unsigned int unWidth, unsigned int unHeight, unsigned int unStride, CStream* pImageStream)
|
|
{
|
|
if (!m_pContext)
|
|
return;
|
|
|
|
Pix* pPix = pixCreate(unWidth, unHeight, 1);
|
|
if (!pPix)
|
|
return;
|
|
|
|
BYTE* pLine = (BYTE*)pImage;
|
|
for (unsigned int unY = 0; unY < unHeight; unY++, pLine += unStride)
|
|
{
|
|
char nBit = 0;
|
|
BYTE* pCur = pLine;
|
|
for (unsigned int unX = 0; unX < unWidth; unX++)
|
|
{
|
|
pixSetPixel(pPix, unX, unY, pCur[0] & (1 << nBit));
|
|
nBit++;
|
|
|
|
if (8 == nBit)
|
|
{
|
|
nBit = 0;
|
|
pCur++;
|
|
}
|
|
}
|
|
}
|
|
|
|
jbig2_add_page(m_pContext, pPix);
|
|
pixDestroy(&pPix);
|
|
m_vImages.push_back(pImageStream);
|
|
}
|
|
void CJbig2Global::AddImage(Pix* pPix, CStream* pImageStream)
|
|
{
|
|
if (!m_pContext)
|
|
return;
|
|
|
|
jbig2_add_page(m_pContext, pPix);
|
|
m_vImages.push_back(pImageStream);
|
|
}
|
|
int CJbig2Global::GetImagesCount()
|
|
{
|
|
return m_vImages.size();
|
|
}
|
|
}
|