fbitmap.cpp

flash文件的解码程序

#include "FBitmap.h"
#include "zlib/zlib.h"
#include <sstream>
#include <strstream>
#include "windows.h"

std::ostream &operator << (std::ostream &out, FlashTagDefineBitsJPEG1 &data)
{
	out << FlashTagHeader(6, 2+data.len);
	WRITE_UWORD(data.GetID());
	for(UDWORD d=0; d < data.len; d++)
	{
		out.put((unsigned char)data.data[d]);
	}
	return out;
}

std::istream &operator >> (std::istream &in,  FlashTagDefineBitsJPEG1 &data)
{
	
	UWORD id;
	READ_UWORD(id);
	data.SetID(id);
	
	data.data = (unsigned char*)malloc(data.importsize - 2);
	data.datagc.push_back(data.data);

	for(UDWORD d=0; d < data.importsize - 2; d++)
	{
		int i = in.get();		
		data.data[d] = i;
	}
	
	return in;
}

std::ostream &operator << (std::ostream &out, FlashTagJPEGTables &data)
{
	out << FlashTagHeader(8,data.len);
	for(UDWORD d=0; d < data.len; d++)
	{
		out.put((unsigned char)data.data[d]);
	}
	return out;
}

std::istream &operator >> (std::istream &in,  FlashTagJPEGTables &data)
{
	data.data = (unsigned char*)malloc(data.importsize);
	data.datagc.push_back(data.data);

	for(UDWORD d=0; d < data.importsize; d++)
	{
		int i = in.get();		
		data.data[d] = i;
	}
	return in;
}

std::ostream &operator << (std::ostream &out, FlashTagDefineBitsJPEG2 &data)
{
	out << FlashTagHeader(21, 6+2+data.len);
	
	
	WRITE_UWORD(data.GetID());

	out.put((unsigned char)0xff);
	out.put((unsigned char)0xd9);
		
	out.put((unsigned char)0xff);
	out.put((unsigned char)0xd8);
			
	out.write((const char *)data.data, data.len);
	
	out.put((unsigned char)0xff);
	out.put((unsigned char)0xd9);
	
	return out;
}

std::istream &operator >> (std::istream &in,  FlashTagDefineBitsJPEG2 &data)
{
	UWORD id;
	READ_UWORD(id);
	data.SetID(id);	
	{
		data.data = (unsigned char *)malloc(data.importsize-2);
		data.gc.push_back(data.data);
		in.read((char *)data.data, data.importsize-2); //throw if not proper amount read
		data.len = data.importsize-2;
	}
	return in;
}

std::ostream &operator << (std::ostream &out, FlashTagDefineBitsJPEG3 &data)
{
	
	unsigned long csize = 1.01*data.len2+12;
	unsigned char *compression_buffer = (unsigned char*)malloc(csize);

	if( compress2(compression_buffer, &csize, (const unsigned char*)data.data2, data.len2, Z_BEST_SPEED) == Z_OK)
	{
		if(csize > 75)
		{
		out << FlashTagHeader(35, 2+4+data.len+csize);
		
		WRITE_UWORD(data.GetID());
		WRITE_UDWORD(data.len);
		{
			for(UDWORD d=0; d < data.len; d++)
			{
				out.put((unsigned char)data.data[d]);		
			}
		}

		out.write((char*)compression_buffer,csize);
		}
		else
		{
			csize = 1.01 * data.len2+12;
			if(compress2(compression_buffer, &csize, (const unsigned char*)data.data2, data.len2, Z_NO_COMPRESSION) == Z_OK)
			{
				out << FlashTagHeader(35, 2+4+data.len+csize);
				
				WRITE_UWORD(data.GetID());
				WRITE_UDWORD(data.len);
				{
					for(UDWORD d=0; d < data.len; d++)
					{
						out.put((unsigned char)data.data[d]);		
					}
				}

				out.write((char*)compression_buffer,csize);

			}
		}

	}
	else if(compress2(compression_buffer, &csize, (const unsigned char*)data.data2, data.len2, Z_NO_COMPRESSION) == Z_OK)
	{
		out << FlashTagHeader(35, 2+4+data.len+csize);
		
		WRITE_UWORD(data.GetID());
		WRITE_UDWORD(data.len);
		{
			for(UDWORD d=0; d < data.len; d++)
			{
				out.put((unsigned char)data.data[d]);		
			}
		}

		out.write((char*)compression_buffer,csize);

	}
	else
	{		
		//throw
	}
	free(compression_buffer);
	
	return out;
}

std::istream &operator >> (std::istream &in,  FlashTagDefineBitsJPEG3 &data)
{
	UWORD id;
	READ_UWORD(id);
	READ_UDWORD(data.len);
	data.SetID(id);	
	{
		data.data = (unsigned char *)malloc(data.len);
		data.gc.push_back(data.data);
		in.read((char *)data.data, data.len); //throw if not proper amount read
	}

	{
		data.data2 = (unsigned char *)malloc(data.importsize-data.len-6);
		data.gc.push_back(data.data2);
		in.read((char *)data.data2, data.importsize-data.len-6); //throw if not proper amount read
		data.len2 = data.importsize-data.len-6;
	}

	return in;
}

void FlashZLibBitmapData::Write(std::ostream &out, char format)
{
	std::ostringstream tmp;

	for(std::vector<FlashRGB>::iterator i=colors.begin(); i != colors.end(); i++)
	{
		(*i).SetAlphaWriteMode(withalpha);
		tmp << *i;		
	}
	tmp.write((const char *)data,len);

	unsigned long csize = (1.01*tmp.str().size())+12;
	unsigned char *compression_buffer = (unsigned char*)malloc(csize);
	UDWORD size = tmp.str().size();
/*
	FILE *fp = fopen("c:\\compress", "wb");
	fwrite(tmp.str().c_str(), 1, size, fp);
	fclose(fp);

	SHELLEXECUTEINFO sei;

	memset(&sei, 0, sizeof(sei));

	sei.cbSize = sizeof(sei);
	sei.fMask  = SEE_MASK_NOCLOSEPROCESS | SEE_MASK_FLAG_DDEWAIT;
	sei.hwnd   = NULL;
	sei.lpVerb = "open";
	sei.lpFile = "c:\\compress.exe";
	sei.nShow  = SW_SHOWNORMAL;
	sei.lpParameters = "/m=ZLib c:\\compress c:\\compress.gz";

	if ( ShellExecuteEx(&sei) )
	{
	WaitForSingleObject(sei.hProcess, INFINITE);
	CloseHandle(sei.hProcess);
	}


	fp = fopen("c:\\compress.gz", "rb");
	csize = fread(compression_buffer, 1, csize, fp);
	fclose(fp);*/
		
	if(compress2(compression_buffer, &csize, (const unsigned char*)tmp.str().c_str(), size, Z_BEST_SPEED) == Z_OK)
	{
		if(csize > 75)
		{
			out.write((char*)compression_buffer,csize);
		}
		else
		{
			csize = (1.01*tmp.str().size())+12;
			if(compress2(compression_buffer, &csize, (const unsigned char*)tmp.str().c_str(), size, Z_NO_COMPRESSION) == Z_OK)
			{
				out.write((char*)compression_buffer,csize);
			}
		}
	}
	else 
	{
		if(compress2(compression_buffer, &csize, (const unsigned char*)tmp.str().c_str(), size, Z_NO_COMPRESSION) == Z_OK)
		{
			out.write((char*)compression_buffer,csize);
		}
	}
	//else
	{		
		//throw
	}
	free(compression_buffer);
}
void FlashZLibBitmapData::Read(std::istream &in, char format, UDWORD size, UDWORD size_buffer)
{
	int sizec = -1;
	UDWORD len2 = size_buffer;
	if(format==3)
	{
		sizec = in.get();
		sizec++;
		size_buffer +=sizec * 3;
		if(GetAlpha()) size_buffer += sizec;
	}
	unsigned char *datatmp = (unsigned char *)malloc(size);
	in.read((char *)datatmp,len);

	unsigned long csize = size_buffer;
	unsigned char *compression_buffer = (unsigned char*)malloc(csize);
	
	gc.push_back(compression_buffer);

	if(uncompress(compression_buffer, &csize, datatmp, size) == Z_OK)
	{
		if(format == 3)
		{
			for(int it = 0; it < sizec; it++)
			{			
				if(GetAlpha()) colors.push_back(FlashRGB(compression_buffer[it*4],compression_buffer[it*4]+1,compression_buffer[it*4]+2,compression_buffer[it*4]+3));
				else colors.push_back(FlashRGB(compression_buffer[it*3],compression_buffer[it*3]+1,compression_buffer[it*3]+2));
			}
			if(GetAlpha()) data = compression_buffer+sizec*4;
			else		  data = compression_buffer+sizec*3;			
			len = len2;
		}
		else
		{
			data = compression_buffer;
			len = len2;
		}
	}
	else
	{		
		//throw
	}
	free(datatmp);
}

std::ostream &operator << (std::ostream &out, FlashTagDefineBitsLossless &data)
{
	std::ostringstream tmp;
	WRITE_UWORD2(data.GetID(),tmp);
	tmp.put(data.format);
	WRITE_UWORD2(data.width,tmp);
	WRITE_UWORD2(data.height,tmp);
	if(data.format==3) tmp.put((unsigned char)data.d.GetSize()-1);
	data.d.SetAlpha(false);
	data.d.Write(tmp,data.format);
	out << FlashTagHeader(20, tmp.str().size());
	out.write(tmp.str().c_str(), tmp.str().size());
	return out;
}

std::istream &operator >> (std::istream &in,  FlashTagDefineBitsLossless &data)
{
	UWORD id;
	READ_UWORD(id);
	data.SetID(id);
	data.format = in.get();
	// if(data.format == EOF) throw;
	READ_UWORD(data.width);
	READ_UWORD(data.height);
	data.d.SetAlpha(false);
	int datasize = 0;
	if(data.format == 3) datasize = data.width*data.height;
	else if(data.format == 4) datasize = data.width*data.height*2;
	else if(data.format == 5) datasize = data.width*data.height*4;
	//else throw;
	
	data.d.Read(in,data.format,data.importsize - 7, datasize);

	return in;
}
std::ostream &operator << (std::ostream &out, FlashTagDefineBitsLossless2 &data)
{
	std::ostringstream tmp;
	WRITE_UWORD2(data.GetID(),tmp);
	tmp.put(data.format);
	WRITE_UWORD2(data.width,tmp);
	WRITE_UWORD2(data.height,tmp);
	if(data.format==3) tmp.put((unsigned char)data.d.GetSize()-1);
	data.d.SetAlpha(true);
	data.d.Write(tmp,data.format);
	out << FlashTagHeader(36, tmp.str().size());
	out.write(tmp.str().c_str(), tmp.str().size());

	return out;
}

std::istream &operator >> (std::istream &in,  FlashTagDefineBitsLossless2 &data)
{
	UWORD id;
	READ_UWORD(id);
	data.SetID(id);
	data.format = in.get();
	// if(data.format == EOF) throw;
	READ_UWORD(data.width);
	READ_UWORD(data.height);
	data.d.SetAlpha(true);
	int datasize = 0;
	if(data.format == 3) datasize = data.width*data.height;
	else if(data.format == 4) datasize = data.width*data.height*2;
	else if(data.format == 5) datasize = data.width*data.height*4;
	//else throw;
	
	data.d.Read(in,data.format,data.importsize - 7, datasize);

	return in;
}