mirror of
https://github.com/ONLYOFFICE/core.git
synced 2026-07-11 01:52:24 +08:00
466 lines
14 KiB
C++
466 lines
14 KiB
C++
/*
|
|
* (c) Copyright Ascensio System SIA 2010-2016
|
|
*
|
|
* 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 <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <stddef.h>
|
|
#include <ctype.h>
|
|
#include "MemoryUtils.h"
|
|
#include "Constants.h"
|
|
#include "GState.h"
|
|
#include "Object.h"
|
|
#include "Stream.h"
|
|
#include "ExtractImageOutputDev.h"
|
|
|
|
#include "../../DesktopEditor/common/File.h"
|
|
|
|
namespace PdfReader
|
|
{
|
|
#pragma pack(push, 1)
|
|
|
|
struct ColorTableEntry
|
|
{
|
|
unsigned char nRed;
|
|
unsigned char nGreen;
|
|
unsigned char nBlue;
|
|
unsigned char nReserved;
|
|
};
|
|
|
|
#pragma pack(pop)
|
|
|
|
typedef struct
|
|
{
|
|
unsigned short bfType;
|
|
unsigned int bfSize;
|
|
unsigned short bfReserved1;
|
|
unsigned short bfReserved2;
|
|
unsigned int bfOffBits;
|
|
} TBitmapFileHeader;
|
|
|
|
typedef struct
|
|
{
|
|
unsigned int biSize;
|
|
int biWidth;
|
|
int biHeight;
|
|
unsigned short biPlanes;
|
|
unsigned short biBitCount;
|
|
unsigned int biCompression;
|
|
unsigned int biSizeImage;
|
|
int biXPelsPerMeter;
|
|
int biYPelsPerMeter;
|
|
unsigned int biClrUsed;
|
|
unsigned int biClrImportant;
|
|
} TBitmapInfoHeader;
|
|
|
|
typedef struct
|
|
{
|
|
unsigned int bcSize;
|
|
unsigned short bcWidth;
|
|
unsigned short bcHeight;
|
|
unsigned short bcPlanes;
|
|
unsigned short bcBitCount;
|
|
} TBitmapCoreHeader;
|
|
|
|
//-------------------------------------------------------------------------------------------------------------------------------
|
|
// ExtractImageOutputDev
|
|
//-------------------------------------------------------------------------------------------------------------------------------
|
|
|
|
|
|
ExtractImageOutputDev::ExtractImageOutputDev(GlobalParams *pGlobalParams, char *sFilePrefix, bool bDumpJPEG, bool bCountImages)
|
|
{
|
|
m_pGlobalParams = pGlobalParams;
|
|
if (sFilePrefix)
|
|
m_sFilePrefix = CopyString(sFilePrefix);
|
|
else
|
|
m_sFilePrefix = NULL;
|
|
|
|
if (m_sFilePrefix)
|
|
m_sFileName = (char *)MemUtilsMalloc(strlen(m_sFilePrefix) + 20);
|
|
else
|
|
m_sFileName = NULL;
|
|
|
|
m_bDumpJPEG = bDumpJPEG;
|
|
m_nImageCount = 0;
|
|
m_bValid = true;
|
|
|
|
m_bCountImages = bCountImages;
|
|
}
|
|
|
|
ExtractImageOutputDev::~ExtractImageOutputDev()
|
|
{
|
|
MemUtilsFree(m_sFileName);
|
|
MemUtilsFree(m_sFilePrefix);
|
|
}
|
|
|
|
void ExtractImageOutputDev::DrawImageMask(GrState *pGState, Object *pRef, Stream *pStream, int nWidth, int nHeight, bool bInvert, bool bInlineImage)
|
|
{
|
|
FILE *pFile = NULL;
|
|
|
|
// JPEG
|
|
if (m_bDumpJPEG && pStream->GetType() == strDCT && !bInlineImage)
|
|
{
|
|
++m_nImageCount;
|
|
if (m_bCountImages || !m_sFileName || !m_sFilePrefix)
|
|
return;
|
|
|
|
sprintf(m_sFileName, "%s\\%04d.jpg", m_sFilePrefix, m_nImageCount);
|
|
|
|
pFile = NSFile::CFileBinary::OpenFileNative(AStringToWString(m_sFileName), L"wb");
|
|
if (!pFile)
|
|
{
|
|
// TO DO: Error "Couldn't open image file"
|
|
return;
|
|
}
|
|
pStream = ((DCTStream *)pStream)->GetRawStream();
|
|
pStream->Reset();
|
|
|
|
// Копируем поток в файл
|
|
int nChar = 0;
|
|
while ((nChar = pStream->GetChar()) != EOF)
|
|
fputc(nChar, pFile);
|
|
|
|
pStream->Close();
|
|
fclose(pFile);
|
|
}
|
|
else
|
|
{
|
|
int nLeftBytes = 0;
|
|
while (div_t(div(((nWidth + 7) / 8), 4)).rem != 0)
|
|
{
|
|
nWidth -= 8;
|
|
nLeftBytes++;
|
|
if (nWidth <= 0)
|
|
{
|
|
// TO DO: Error "To small picture"
|
|
return;
|
|
}
|
|
}
|
|
|
|
++m_nImageCount;
|
|
if (m_bCountImages || !m_sFileName || !m_sFilePrefix)
|
|
return;
|
|
|
|
sprintf(m_sFileName, "%s\\%04d.bmp", m_sFilePrefix, m_nImageCount);
|
|
|
|
pFile = NSFile::CFileBinary::OpenFileNative(AStringToWString(m_sFileName), L"wb");
|
|
if (!pFile)
|
|
{
|
|
// TO DO: Error "Couldn't open image file"
|
|
return;
|
|
}
|
|
|
|
TBitmapFileHeader oFileHeader;
|
|
oFileHeader.bfType = 'M' * 256 + 'B';
|
|
oFileHeader.bfSize = sizeof(TBitmapFileHeader) + sizeof(TBitmapInfoHeader) + 2 * sizeof(ColorTableEntry) + nHeight * ((nWidth + 7) / 8);
|
|
oFileHeader.bfReserved1 = 0;
|
|
oFileHeader.bfReserved2 = 0;
|
|
oFileHeader.bfOffBits = sizeof(TBitmapFileHeader) + sizeof(TBitmapInfoHeader) + 2 * sizeof(ColorTableEntry);
|
|
|
|
|
|
TBitmapInfoHeader oInfoHeader;
|
|
oInfoHeader.biSize = sizeof(TBitmapInfoHeader);
|
|
oInfoHeader.biWidth = nWidth;
|
|
oInfoHeader.biHeight = nHeight;
|
|
oInfoHeader.biPlanes = 1;
|
|
oInfoHeader.biBitCount = 1;
|
|
oInfoHeader.biCompression = 0;
|
|
oInfoHeader.biSizeImage = nHeight * ((nWidth + 7) / 8);
|
|
oInfoHeader.biXPelsPerMeter = 0;
|
|
oInfoHeader.biYPelsPerMeter = 0;
|
|
oInfoHeader.biClrUsed = 256;
|
|
oInfoHeader.biClrImportant = 0;
|
|
|
|
ColorTableEntry oFirstColor, oSecondColor;
|
|
oFirstColor.nRed = 255;
|
|
oFirstColor.nGreen = 255;
|
|
oFirstColor.nBlue = 255;
|
|
oFirstColor.nReserved = 0;
|
|
|
|
oSecondColor.nRed = 0;
|
|
oSecondColor.nGreen = 0;
|
|
oSecondColor.nBlue = 0;
|
|
oSecondColor.nReserved = 0;
|
|
|
|
::fwrite(&oFileHeader, 1, sizeof(TBitmapFileHeader), pFile);
|
|
::fwrite(&oInfoHeader, 1, sizeof(TBitmapInfoHeader), pFile);
|
|
::fwrite(&oFirstColor, 1, sizeof(ColorTableEntry), pFile);
|
|
::fwrite(&oSecondColor, 1, sizeof(ColorTableEntry), pFile);
|
|
|
|
long lPos = ::ftell(pFile);
|
|
|
|
pStream->Reset();
|
|
|
|
// Копируем поток в файл
|
|
int nSize = nHeight * ((nWidth + 7) / 8);
|
|
|
|
unsigned char *pBuffer = new unsigned char[nSize];
|
|
if (!pBuffer)
|
|
{
|
|
// TO DO: Error "Not enough memory"
|
|
return;
|
|
}
|
|
unsigned char *pBufPointer = pBuffer;
|
|
for (int nY = 0; nY < nHeight; nY++)
|
|
{
|
|
for (int nX = 0; nX < (nWidth + 7) / 8; nX++, pBufPointer++)
|
|
{
|
|
*pBufPointer = pStream->GetChar() ^ 0xff;
|
|
}
|
|
for (int nIndex = 0; nIndex < nLeftBytes; nIndex++)
|
|
{
|
|
pStream->GetChar();
|
|
}
|
|
}
|
|
pBufPointer = pBuffer;
|
|
int nStride = ((nWidth + 7) / 8);
|
|
pBufPointer += (nHeight - 1) * nStride;
|
|
nStride = -nStride;
|
|
|
|
for (int nRow = nHeight; nRow > 0; --nRow, pBufPointer += nStride)
|
|
{
|
|
::fwrite(pBufPointer, 1, abs(nStride), pFile);
|
|
}
|
|
pStream->Close();
|
|
::fclose(pFile);
|
|
delete[]pBuffer;
|
|
}
|
|
}
|
|
|
|
void ExtractImageOutputDev::DrawImage(GrState *pGState, Object *pRef, Stream *pStream, int nWidth, int nHeight, GrImageColorMap *pColorMap, int *pnMaskColors, bool bInlineImage)
|
|
{
|
|
FILE *pFile = NULL;
|
|
// JPEG
|
|
if (m_bDumpJPEG && pStream->GetType() == strDCT && (pColorMap->GetComponentsCount() == 1 || pColorMap->GetComponentsCount() == 3) && !bInlineImage)
|
|
{
|
|
++m_nImageCount;
|
|
if (m_bCountImages || !m_sFileName || !m_sFilePrefix)
|
|
return;
|
|
sprintf(m_sFileName, "%s\\%04d.jpg", m_sFilePrefix, m_nImageCount);
|
|
|
|
pFile = NSFile::CFileBinary::OpenFileNative(AStringToWString(m_sFileName), L"wb");
|
|
if (!pFile)
|
|
{
|
|
// TO DO: Error "Couldn't open image file"
|
|
return;
|
|
}
|
|
|
|
pStream = ((DCTStream *)pStream)->GetRawStream();
|
|
pStream->Reset();
|
|
|
|
// Копируем поток в файл
|
|
int nChar = 0;
|
|
while ((nChar = pStream->GetChar()) != EOF)
|
|
fputc(nChar, pFile);
|
|
|
|
pStream->Close();
|
|
fclose(pFile);
|
|
}
|
|
else if (pColorMap->GetComponentsCount() == 1 && pColorMap->GetBitsPerComponent() == 1)
|
|
{
|
|
int nLeftBytes = 0;
|
|
while (div_t(div(((nWidth + 7) / 8), 4)).rem != 0)
|
|
{
|
|
nWidth -= 8;
|
|
nLeftBytes++;
|
|
if (nWidth <= 0)
|
|
{
|
|
// TO DO: Error "To small picture"
|
|
return;
|
|
}
|
|
}
|
|
|
|
++m_nImageCount;
|
|
if (m_bCountImages || !m_sFileName || !m_sFilePrefix)
|
|
return;
|
|
|
|
sprintf(m_sFileName, "%s\\%04d.bmp", m_sFilePrefix, m_nImageCount);
|
|
|
|
pFile = NSFile::CFileBinary::OpenFileNative(AStringToWString(m_sFileName), L"wb");
|
|
if (!pFile)
|
|
{
|
|
// TO DO: Error "Couldn't open image file"
|
|
return;
|
|
}
|
|
TBitmapFileHeader oFileHeader;
|
|
oFileHeader.bfType = 'M' * 256 + 'B';
|
|
oFileHeader.bfSize = sizeof(TBitmapFileHeader) + sizeof(TBitmapInfoHeader) + 2 * sizeof(ColorTableEntry) + nHeight * ((nWidth + 7) / 8);
|
|
oFileHeader.bfReserved1 = 0;
|
|
oFileHeader.bfReserved2 = 0;
|
|
oFileHeader.bfOffBits = sizeof(TBitmapFileHeader) + sizeof(TBitmapInfoHeader) + 2 * sizeof(ColorTableEntry);
|
|
|
|
|
|
TBitmapInfoHeader oInfoHeader;
|
|
oInfoHeader.biSize = sizeof(TBitmapInfoHeader);
|
|
oInfoHeader.biWidth = nWidth;
|
|
oInfoHeader.biHeight = nHeight;
|
|
oInfoHeader.biPlanes = 1;
|
|
oInfoHeader.biBitCount = 1;
|
|
oInfoHeader.biCompression = 0;
|
|
oInfoHeader.biSizeImage = nHeight * ((nWidth + 7) / 8);
|
|
oInfoHeader.biXPelsPerMeter = 0;
|
|
oInfoHeader.biYPelsPerMeter = 0;
|
|
oInfoHeader.biClrUsed = 0;
|
|
oInfoHeader.biClrImportant = 0;
|
|
|
|
ColorTableEntry oFirstColor, oSecondColor;
|
|
oFirstColor.nRed = 255;
|
|
oFirstColor.nGreen = 255;
|
|
oFirstColor.nBlue = 255;
|
|
oFirstColor.nReserved = 0;
|
|
|
|
oSecondColor.nRed = 0;
|
|
oSecondColor.nGreen = 0;
|
|
oSecondColor.nBlue = 0;
|
|
oSecondColor.nReserved = 0;
|
|
|
|
::fwrite(&oFileHeader, 1, sizeof(TBitmapFileHeader), pFile);
|
|
::fwrite(&oInfoHeader, 1, sizeof(TBitmapInfoHeader), pFile);
|
|
::fwrite(&oFirstColor, 1, sizeof(ColorTableEntry), pFile);
|
|
::fwrite(&oSecondColor, 1, sizeof(ColorTableEntry), pFile);
|
|
|
|
pStream->Reset();
|
|
|
|
// Копируем поток в файл
|
|
int nSize = nHeight * ((nWidth + 7) / 8);
|
|
|
|
unsigned char *pBuffer = new unsigned char[nSize];
|
|
if (!pBuffer)
|
|
{
|
|
// TO DO: Error "Not enough memory"
|
|
return;
|
|
}
|
|
unsigned char *pBufPointer = pBuffer;
|
|
for (int nY = 0; nY < nHeight; nY++)
|
|
{
|
|
for (int nX = 0; nX < (nWidth + 7) / 8; nX++, pBufPointer++)
|
|
{
|
|
*pBufPointer = pStream->GetChar() ^ 0xff;
|
|
}
|
|
for (int nIndex = 0; nIndex < nLeftBytes; nIndex++)
|
|
{
|
|
pStream->GetChar();
|
|
}
|
|
}
|
|
pBufPointer = pBuffer;
|
|
int nStride = ((nWidth + 7) / 8);
|
|
pBufPointer += (nHeight - 1) * nStride;
|
|
nStride = -nStride;
|
|
|
|
for (int nRow = nHeight; nRow > 0; --nRow, pBufPointer += nStride)
|
|
{
|
|
::fwrite(pBufPointer, 1, -nStride, pFile);
|
|
}
|
|
pStream->Close();
|
|
::fclose(pFile);
|
|
delete[]pBuffer;
|
|
}
|
|
else
|
|
{
|
|
++m_nImageCount;
|
|
if (m_bCountImages || !m_sFileName || !m_sFilePrefix)
|
|
return;
|
|
|
|
sprintf(m_sFileName, "%s\\%04d.bmp", m_sFilePrefix, m_nImageCount);
|
|
|
|
pFile = NSFile::CFileBinary::OpenFileNative(AStringToWString(m_sFileName), L"wb");
|
|
if (!pFile)
|
|
{
|
|
// TO DO: Error "Couldn't open image file"
|
|
return;
|
|
}
|
|
TBitmapFileHeader oFileHeader;
|
|
oFileHeader.bfType = 'M' * 256 + 'B';
|
|
oFileHeader.bfSize = sizeof(TBitmapFileHeader) + sizeof(TBitmapCoreHeader) + nWidth * nHeight * 4;
|
|
oFileHeader.bfReserved1 = 0;
|
|
oFileHeader.bfReserved2 = 0;
|
|
oFileHeader.bfOffBits = sizeof(TBitmapFileHeader) + sizeof(TBitmapCoreHeader);
|
|
|
|
TBitmapCoreHeader oCoreHeader;
|
|
oCoreHeader.bcSize = sizeof(TBitmapCoreHeader);
|
|
oCoreHeader.bcWidth = nWidth;
|
|
oCoreHeader.bcHeight = nHeight;
|
|
oCoreHeader.bcPlanes = 1;
|
|
oCoreHeader.bcBitCount = 32;
|
|
|
|
::fwrite(&oFileHeader, 1, sizeof(TBitmapFileHeader), pFile);
|
|
::fwrite(&oCoreHeader, 1, sizeof(TBitmapCoreHeader), pFile);
|
|
|
|
ImageStream *pImageStream = new ImageStream(pStream, nWidth, pColorMap->GetComponentsCount(), pColorMap->GetBitsPerComponent());
|
|
if (!pImageStream)
|
|
{
|
|
// TO DO: Error "Not enough memory"
|
|
return;
|
|
}
|
|
|
|
pImageStream->Reset();
|
|
|
|
unsigned char *pBuffer = new unsigned char[nWidth * nHeight * 4];
|
|
if (!pBuffer)
|
|
{
|
|
// TO DO: Error "Not enough memory"
|
|
delete pImageStream;
|
|
return;
|
|
}
|
|
|
|
unsigned char *pBufPointer = pBuffer;
|
|
for (int nY = 0; nY < nHeight; nY++)
|
|
{
|
|
unsigned char *pImageLine = pImageStream->GetNextLine();
|
|
for (int nX = 0; nX < nWidth; nX++, pBufPointer += 4)
|
|
{
|
|
GrRGB oRGB;
|
|
pColorMap->GetRGB(pImageLine, &oRGB);
|
|
|
|
pBufPointer[0] = ColorToByte(oRGB.b);
|
|
pBufPointer[1] = ColorToByte(oRGB.g);
|
|
pBufPointer[2] = ColorToByte(oRGB.r);
|
|
pBufPointer[3] = 255;
|
|
|
|
pImageLine += pColorMap->GetComponentsCount();
|
|
}
|
|
}
|
|
delete pImageStream;
|
|
|
|
pBufPointer = pBuffer;
|
|
int nStride = nWidth * 4;
|
|
pBufPointer += (nHeight - 1) * nStride;
|
|
nStride = -nStride;
|
|
|
|
for (int nRow = nHeight; nRow > 0; --nRow, pBufPointer += nStride)
|
|
{
|
|
::fwrite(pBufPointer, 1, -nStride, pFile);
|
|
}
|
|
|
|
::fclose(pFile);
|
|
delete[]pBuffer;
|
|
}
|
|
}
|
|
} |