0

zlib を直接使用して PNG を書き込もうとしています。ただし、出力ファイルが正しくありません。これを frhed で見ると、IDAT チャンクは周期的なパターンとゼロの大きなブロックを示しています。これは、zlib の呼び出し時に何か問題があることを示しています。コードは次のとおりです。

#include "FrameGrabber.h"
#include "Image.h"
#include "crc.h"
#include <zlib.h>
#include <stdint.h>

#include <cstdio>
#include <cassert>



template<class T>
void assign(void* addr,T x)
{
T* p_out= (T*)addr ;
*p_out=x;
}

inline int32_t swap_bytes(int32_t x)
{
asm("bswap %1":"=r"(x):"r"(x):);
return x;
}

//Input is BGRx (convert to BGR (should be swapped to RGB later) and add a byte in the beginning)
void filterApply(const unsigned char* const* scanlines,char* buffer_png,int width,int height)
{
for(int k=0;k<height;++k)
    {
    *buffer_png=0;
    ++buffer_png;
    for(int l=0;l<width;++l)
        {
        assign(buffer_png,( (int32_t*)(scanlines[k])  )[l] );
        buffer_png+=3;
        }
    }
}

size_t compress(const char* buffer_uncompressed,char* buffer_compressed,size_t length_in)
{
z_stream stream;
memset(&stream,0,sizeof(stream));
//  15 bits=32K?
deflateInit2(&stream,6,Z_DEFLATED,15,9,Z_FILTERED);

stream.avail_in=length_in;
stream.next_in=(unsigned char*)buffer_uncompressed;
stream.avail_out=length_in;
stream.next_out =(unsigned char*)buffer_compressed;
do
    {       
    deflate(&stream, Z_FINISH);
    }
while (stream.avail_out == 0);
deflateEnd(&stream);
return stream.total_out;
}



const size_t CHUNK_BASE_SIZE=12;

void Chunk_sizeSet(char* chunk,uint32_t size)
{assign(chunk,swap_bytes(size));}

void Chunk_IDSet(char* chunk,uint32_t id)
{assign(chunk+4,swap_bytes(id));}

void Chunk_CRCSet(char* chunk,const PngCRC& crc,uint32_t size_chunk_data)
{assign(chunk+8+size_chunk_data, swap_bytes(crc(chunk+4,size_chunk_data+4)) );}



const size_t IHDR_SIZE=13;
const int IHDR_COLORTYPE_RGB=2;

void IHDR_widthSet(char* chunk,int32_t width)
{assign(chunk+8,swap_bytes(width));}

void IHDR_heightSet(char* chunk,int32_t height)
{assign(chunk+12,swap_bytes(height));}

void IHDR_bitDepthSet(char* chunk,char bd)
{chunk[16]=bd;}

void IHDR_colorTypeSet(char* chunk,char ct)
{chunk[17]=ct;}

void IHDR_compressionMethodSet(char* chunk,char cmprm)
{chunk[18]=cmprm;}

void IHDR_filterMethodSet(char* chunk,char filter)
{chunk[19]=filter;}

void IHDR_interlaceMethodSet(char* chunk,char interlace)
{chunk[20]=interlace;}



int main()
{
PngCRC crc;                              //The CRC code works
char signature[8]={137,80,78,71,13,10,26,10};

char IEND[CHUNK_BASE_SIZE];
Chunk_sizeSet(IEND,0);
Chunk_IDSet(IEND,0x49454e44);
Chunk_CRCSet(IEND,crc,0);

FrameGrabber grabber(GetDesktopWindow()); //Grab the desktop (works)
Image img(grabber);
grabber.grab();

char IHDR[CHUNK_BASE_SIZE+IHDR_SIZE];
Chunk_sizeSet(IHDR,CHUNK_BASE_SIZE+IHDR_SIZE);
Chunk_IDSet(IHDR,0x49484452);
IHDR_widthSet(IHDR,grabber.widthGet());
IHDR_heightSet(IHDR,grabber.heightGet());
IHDR_bitDepthSet(IHDR,8);
IHDR_colorTypeSet(IHDR,IHDR_COLORTYPE_RGB);
IHDR_compressionMethodSet(IHDR,0);
IHDR_filterMethodSet(IHDR,0);
IHDR_interlaceMethodSet(IHDR,0);
Chunk_CRCSet(IHDR,crc,IHDR_SIZE);   


size_t size_uncompressed=(1+3*grabber.widthGet())*grabber.heightGet();
char* img_png_uncompressed=(char*)malloc(size_uncompressed);
filterApply(img.rowsGet(),img_png_uncompressed,grabber.widthGet(),grabber.heightGet());

    //The compressed chunk should not be larger than the uncompressed one.
char* IDAT=(char*)malloc(size_uncompressed+CHUNK_BASE_SIZE);

int32_t size_idat=compress(img_png_uncompressed,IDAT+CHUNK_BASE_SIZE,size_uncompressed);
Chunk_sizeSet(IDAT,size_idat);
Chunk_IDSet(IDAT,0x49444154);
Chunk_CRCSet(IDAT,crc,size_idat);


FILE* file_out=fopen("test.png","wb");

fwrite(signature,1,sizeof(signature),file_out);
fwrite(IHDR,1,sizeof(IHDR),file_out);
fwrite(IDAT,1,size_idat,file_out);
fwrite(IEND,1,sizeof(IEND),file_out);
free(IDAT);
fclose(file_out);

return 0;
}

編集: filterApply にバグが見つかりました。大きな空のブロックは取得されませんが、ファイルはまだ無効です。入力データに影響するだけなので正しいです。

4

1 に答える 1

0

まず、libpngを使用するだけで、これについて心配する必要はありません。

チャンクサイズを正しく設定していません。チャンクサイズには、長さ、チャンクID、またはcrcは含まれません。また、計算対象に関して、crcを誤って計算しています。そして、圧縮されたデータをIDATチャンクの間違った場所に書き込んでいます。ここでやめます。PNG仕様をもっと注意深く読む必要があります。

他のコメント：

何であるかはわかりませんcrc.hが、zlibはすでにcrc32()pngと互換性のある関数を提供しています。crc()使用しているルーチンがcrc.h別のcrcを計算している可能性があります。同じcrcであり、crc32()すでにzlibによってリンクされている場合でも、それを使用することをお勧めします。

compress()その関数はすでにzlibによって提供されているため、この名前は使用しないでください。zlibを使用しているため、compress()異なる関数を使用すると読者が混乱します。

while (stream.avail_out == 0);意味がありません。の最初の呼び出しは、でdeflate()完了するかavail_out != 0、deflate（）を再度呼び出してもそれが変更されないため、無限ループでスタックします。

を使用する場合は、を使用して出力バッファのサイズを設定し、それが完了することを保証するZ_FINISH必要があります。deflateBound()ソースコードのコメント「圧縮されたチャンクは、圧縮されていないチャンクより大きくてはいけません。」保証されません。非圧縮性データは、非圧縮性入力よりもわずかに大きい圧縮データになります。

関数からの戻りコードをチェックしていませんdeflate！戻りコードを確認してください。

于 2012-07-25T16:37:05.683 に答える