mirror of
https://github.com/vcmi/vcmi.git
synced 2024-12-24 22:14:36 +02:00
784 lines
19 KiB
C++
784 lines
19 KiB
C++
#include "stdafx.h"
|
|
#include "CLodHandler.h"
|
|
#include <sstream>
|
|
#include <algorithm>
|
|
#include <cstring>
|
|
|
|
|
|
int readNormalNr (int pos, int bytCon, unsigned char * str)
|
|
{
|
|
int ret=0;
|
|
int amp=1;
|
|
if (str)
|
|
{
|
|
for (int i=0; i<bytCon; i++)
|
|
{
|
|
ret+=str[pos+i]*amp;
|
|
amp*=256;
|
|
}
|
|
}
|
|
else return -1;
|
|
return ret;
|
|
}
|
|
void CPCXConv::openPCX(char * PCX, int len)
|
|
{
|
|
pcx = new unsigned char[len];
|
|
for (int i=0;i<len;i++)
|
|
pcx[i]=PCX[i];
|
|
}
|
|
void CPCXConv::fromFile(std::string path)
|
|
{
|
|
std::ifstream * is = new std::ifstream();
|
|
is -> open(path.c_str(),std::ios::binary);
|
|
is->seekg(0,std::ios::end); // na koniec
|
|
pcxs = is->tellg(); // read length
|
|
is->seekg(0,std::ios::beg); // wracamy na poczatek
|
|
pcx = new unsigned char[pcxs]; // allocate memory
|
|
is->read((char*)pcx, pcxs); // read map file to buffer
|
|
is->close();
|
|
delete is;
|
|
}
|
|
void CPCXConv::saveBMP(std::string path)
|
|
{
|
|
std::ofstream os;
|
|
os.open(path.c_str(), std::ios::binary);
|
|
os.write((char*)bmp,bmps);
|
|
os.close();
|
|
}
|
|
void CPCXConv::convert()
|
|
{
|
|
BMPHeader bh;
|
|
BMPPalette pal[256];
|
|
Epcxformat format;
|
|
int fSize, maxx, maxy,i,y;
|
|
bool check1, check2;
|
|
unsigned char add;
|
|
int it=0;
|
|
|
|
std::stringstream out;
|
|
|
|
fSize = readNormalNr(it,4,pcx);it+=4;
|
|
maxx = readNormalNr(it,4,pcx);it+=4;
|
|
maxy = readNormalNr(it,4,pcx);it+=4;
|
|
if (fSize==maxx*maxy*3)
|
|
check1=true;
|
|
else
|
|
check1=false;
|
|
if (fSize==maxx*maxy)
|
|
check2=true;
|
|
else
|
|
check2=false;
|
|
if (check1)
|
|
format=Epcxformat::PCX24B;
|
|
else if (check2)
|
|
format=Epcxformat::PCX8B;
|
|
else
|
|
return;
|
|
bh.x=maxx;
|
|
bh.y=maxy;
|
|
add=(int)(4*(((float)1)-(((float)maxx/(float)4)-((int)((float)maxx/(float)4)))));
|
|
if (add==4)
|
|
add=0;
|
|
if (format==Epcxformat::PCX8B)
|
|
{
|
|
bh._c1=0x436;
|
|
bh._c2=0x28;
|
|
bh._c3=1;
|
|
bh._c4=8;
|
|
bh.dataSize2=bh.dataSize1=maxx*maxy;
|
|
bh.fullSize = bh.dataSize1+436;
|
|
}
|
|
else
|
|
{
|
|
bh._c1=0x36;
|
|
bh._c2=0x28;
|
|
bh._c3=1;
|
|
bh._c4=0x18;
|
|
bh.dataSize2=bh.dataSize1=0xB12;
|
|
bh.fullSize=(maxx+add)*maxy*3+36;
|
|
}
|
|
if (format==Epcxformat::PCX8B)
|
|
{
|
|
it = pcxs-256*3;
|
|
for (int i=0;i<256;i++)
|
|
{
|
|
pal[i].R=pcx[it++];
|
|
pal[i].G=pcx[it++];
|
|
pal[i].B=pcx[it++];
|
|
pal[i].F='\0';
|
|
}
|
|
}
|
|
out<<"BM";
|
|
bh.print(out);
|
|
if (format==Epcxformat::PCX8B)
|
|
{
|
|
for (int i=0;i<256;i++)
|
|
{
|
|
out<<pal[i].B;
|
|
out<<pal[i].G;
|
|
out<<pal[i].R;
|
|
out<<pal[i].F;
|
|
}
|
|
for (y=maxy;y>0;y--)
|
|
{
|
|
it=0xC+(y-1)*maxx;
|
|
for (int j=0;j<maxx;j++)
|
|
out<<pcx[it+j];
|
|
if (add>0)
|
|
{
|
|
for (int j=0;j<add;j++)
|
|
out<<'\0'; //bylo z buforu, ale onnie byl incjalizowany (?!)
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
for (y=maxy; y>0; y--)
|
|
{
|
|
it=0xC+(y-1)*maxx*3;
|
|
for (int j=0;j<maxx*3;j++)
|
|
out<<pcx[it+j];
|
|
if (add>0)
|
|
{
|
|
for (int j=0;j<add*3;j++)
|
|
out<<'\0'; //bylo z buforu, ale onnie byl incjalizowany (?!)
|
|
}
|
|
}
|
|
}
|
|
std::string temp = out.str();
|
|
bmp = new unsigned char[temp.length()];
|
|
bmps=temp.length();
|
|
for (int a=0;a<temp.length();a++)
|
|
{
|
|
bmp[a]=temp[a];
|
|
}
|
|
}
|
|
|
|
|
|
// if PFileType=pcx24bit then
|
|
// for y:=MaxY downto 1 do
|
|
// begin
|
|
// FPcx.Seek($0C+(y-1)*(MaxX*3),spBegin);
|
|
// Stream2Stream(FTemp,FPcx,MaxX*3);
|
|
// if Add>0 then FTemp.Write(Buffer,Add*3);
|
|
// end;
|
|
// FTemp.Seek(0,spBegin);
|
|
// BufferBitmap.LoadFromStream(FTemp);
|
|
// FTemp.Free;
|
|
// Result:=True;
|
|
//end;
|
|
/*int CLodHandler::decompress (unsigned char * source, int size, int realSize, std::ofstream & dest)
|
|
{
|
|
std::ofstream lb;
|
|
lb.open("lodbuf\\buf.gz", std::ios::out|std::ios::binary);
|
|
for(int i=0; i<size; ++i)
|
|
{
|
|
lb<<source[i];
|
|
}
|
|
lb.close();
|
|
|
|
FILE * inputf = fopen("lodbuf\\buf.gz", "rb+");
|
|
FILE * outputf = fopen("lodbuf\\buf.un", "wb+");
|
|
|
|
int ret = infm(inputf, outputf);
|
|
fclose(inputf);
|
|
fclose(outputf);
|
|
std::stringstream sin;
|
|
//for()
|
|
|
|
int ret = infs(sin, dest);
|
|
return ret;
|
|
} */
|
|
|
|
int CLodHandler::decompress (unsigned char * source, int size, int realSize, std::string & dest)
|
|
{
|
|
std::ofstream lb;
|
|
lb.open("lodbuf\\buf.gz", std::ios::out|std::ios::binary);
|
|
for(int i=0; i<size; ++i)
|
|
{
|
|
lb<<source[i];
|
|
}
|
|
lb.close();
|
|
|
|
FILE * inputf = fopen("lodbuf\\buf.gz", "rb+");
|
|
FILE * outputf = fopen(dest.c_str(), "wb+");
|
|
|
|
int ret = infm(inputf, outputf);
|
|
fclose(inputf);
|
|
fclose(outputf);
|
|
return ret;
|
|
}
|
|
|
|
int CLodHandler::infm(FILE *source, FILE *dest, int wBits)
|
|
{
|
|
int ret;
|
|
unsigned have;
|
|
z_stream strm;
|
|
unsigned char in[NLoadHandlerHelp::fCHUNK];
|
|
unsigned char out[NLoadHandlerHelp::fCHUNK];
|
|
|
|
/* allocate inflate state */
|
|
strm.zalloc = Z_NULL;
|
|
strm.zfree = Z_NULL;
|
|
strm.opaque = Z_NULL;
|
|
strm.avail_in = 0;
|
|
strm.next_in = Z_NULL;
|
|
ret = inflateInit2(&strm, wBits);
|
|
if (ret != Z_OK)
|
|
return ret;
|
|
|
|
/* decompress until deflate stream ends or end of file */
|
|
do
|
|
{
|
|
strm.avail_in = fread(in, 1, NLoadHandlerHelp::fCHUNK, source);
|
|
if (ferror(source)) {
|
|
(void)inflateEnd(&strm);
|
|
return Z_ERRNO;
|
|
}
|
|
if (strm.avail_in == 0)
|
|
break;
|
|
strm.next_in = in;
|
|
|
|
/* run inflate() on input until output buffer not full */
|
|
do
|
|
{
|
|
strm.avail_out = NLoadHandlerHelp::fCHUNK;
|
|
strm.next_out = out;
|
|
ret = inflate(&strm, Z_NO_FLUSH);
|
|
//assert(ret != Z_STREAM_ERROR); /* state not clobbered */
|
|
switch (ret)
|
|
{
|
|
case Z_NEED_DICT:
|
|
ret = Z_DATA_ERROR; /* and fall through */
|
|
case Z_DATA_ERROR:
|
|
case Z_MEM_ERROR:
|
|
(void)inflateEnd(&strm);
|
|
return ret;
|
|
}
|
|
have = NLoadHandlerHelp::fCHUNK - strm.avail_out;
|
|
if (fwrite(out, 1, have, dest) != have || ferror(dest))
|
|
{
|
|
(void)inflateEnd(&strm);
|
|
return Z_ERRNO;
|
|
}
|
|
} while (strm.avail_out == 0);
|
|
|
|
/* done when inflate() says it's done */
|
|
} while (ret != Z_STREAM_END);
|
|
|
|
/* clean up and return */
|
|
(void)inflateEnd(&strm);
|
|
return ret == Z_STREAM_END ? Z_OK : Z_DATA_ERROR;
|
|
}
|
|
|
|
std::vector<CDefHandler *> CLodHandler::extractManyFiles(std::vector<std::string> defNamesIn, std::string lodName)
|
|
{
|
|
std::vector<CDefHandler *> ret(defNamesIn.size());
|
|
for(int hh=0; hh<defNamesIn.size(); ++hh)
|
|
{
|
|
std::transform(defNamesIn[hh].begin(), defNamesIn[hh].end(), defNamesIn[hh].begin(), (int(*)(int))toupper);
|
|
}
|
|
std::ifstream FLOD;
|
|
std::ofstream FOut;
|
|
int i;
|
|
|
|
std::string Ts;
|
|
//std::cout<<"*** Loading FAT ... \n";
|
|
FLOD.open(lodName.c_str(),std::ios::binary);
|
|
//std::cout<<"*** Archive: "+FName+" loaded\n";
|
|
FLOD.seekg(8,std::ios_base::beg);
|
|
unsigned char temp[4];
|
|
FLOD.read((char*)temp,4);
|
|
totalFiles = readNormalNr(temp,4);
|
|
FLOD.seekg(0x5c,std::ios_base::beg);
|
|
entries.reserve(totalFiles);
|
|
//std::cout<<"*** Loading FAT ...\n";
|
|
for (int i=0; i<totalFiles; i++)
|
|
{
|
|
entries.push_back(Entry());
|
|
//FLOD.read((char*)entries[i].name,12);
|
|
char * bufc = new char;
|
|
bool appending = true;
|
|
for(int kk=0; kk<12; ++kk)
|
|
{
|
|
FLOD.read(bufc, 1);
|
|
if(appending)
|
|
{
|
|
entries[i].name[kk] = *bufc;
|
|
}
|
|
else
|
|
{
|
|
entries[i].name[kk] = 0;
|
|
appending = false;
|
|
}
|
|
}
|
|
delete bufc;
|
|
FLOD.read((char*)entries[i].hlam_1,4);
|
|
FLOD.read((char*)temp,4);
|
|
entries[i].offset=readNormalNr(temp,4);
|
|
FLOD.read((char*)temp,4);
|
|
entries[i].realSize=readNormalNr(temp,4);
|
|
FLOD.read((char*)entries[i].hlam_2,4);
|
|
FLOD.read((char*)temp,4);
|
|
entries[i].size=readNormalNr(temp,4);
|
|
}
|
|
//std::cout<<" done\n";
|
|
//std::transform(name.begin(), name.end(), name.begin(), (int(*)(int))toupper);
|
|
std::vector<char> found(defNamesIn.size(), 0);
|
|
for (int i=0;i<totalFiles;i++)
|
|
{
|
|
std::string buf1 = std::string((char*)entries[i].name);
|
|
std::transform(buf1.begin(), buf1.end(), buf1.begin(), (int(*)(int))toupper);
|
|
bool exists = false;
|
|
int curDef;
|
|
for(int hh=0; hh<defNamesIn.size(); ++hh)
|
|
{
|
|
if(buf1==defNamesIn[hh])
|
|
{
|
|
exists = true;
|
|
found[hh] = '\1';
|
|
curDef = hh;
|
|
break;
|
|
}
|
|
}
|
|
if(!exists)
|
|
continue;
|
|
FLOD.seekg(entries[i].offset,std::ios_base::beg);
|
|
std::string bufff = (lodName.substr(0, lodName.size()-4) + "\\" + (char*)entries[i].name);
|
|
unsigned char * outp;
|
|
if (entries[i].size==0) //file is not compressed
|
|
{
|
|
outp = new unsigned char[entries[i].realSize];
|
|
FLOD.read((char*)outp, entries[i].realSize);
|
|
/*std::ofstream out;
|
|
out.open(bufff.c_str(), std::ios::binary);
|
|
if(!out.is_open())
|
|
{
|
|
std::cout<<"Unable to create "<<bufff;
|
|
}
|
|
else
|
|
{
|
|
for(int hh=0; hh<entries[i].realSize; ++hh)
|
|
{
|
|
out<<*(outp+hh);
|
|
}
|
|
out.close();
|
|
}*/
|
|
CDefHandler * nh = new CDefHandler;
|
|
nh->openFromMemory(outp, entries[i].realSize, std::string((char*)entries[i].name));
|
|
ret[curDef] = nh;
|
|
|
|
}
|
|
else //we will decompressing file
|
|
{
|
|
outp = new unsigned char[entries[i].size];
|
|
FLOD.read((char*)outp, entries[i].size);
|
|
FLOD.seekg(0, std::ios_base::beg);
|
|
/*std::ofstream destin;
|
|
destin.open(bufff.c_str(), std::ios::binary);
|
|
//int decRes = decompress(outp, entries[i].size, entries[i].realSize, bufff);
|
|
int decRes = infs(outp, entries[i].size, entries[i].realSize, destin);
|
|
destin.close();
|
|
if(decRes!=0)
|
|
{
|
|
std::cout<<"LOD Extraction error"<<" "<<decRes<<" while extracting to "<<bufff<<std::endl;
|
|
}*/
|
|
unsigned char * decomp = NULL;
|
|
int decRes = infs2(outp, entries[i].size, entries[i].realSize, decomp);
|
|
CDefHandler * nh = new CDefHandler;
|
|
nh->openFromMemory(decomp, entries[i].realSize, std::string((char*)entries[i].name));
|
|
ret[curDef] = nh;
|
|
//delete decomp;
|
|
}
|
|
//for (int j=0; j<entries[i].size; j++)
|
|
// FOut << outp[j];
|
|
//FOut.flush();
|
|
delete outp;
|
|
//FOut.close();
|
|
//std::cout<<"*** done\n";
|
|
}
|
|
//TODO: przypisywanie defhandler�w powtarzaj�cym si� defom
|
|
FLOD.close();
|
|
//std::cout<<"*** Archive: "+FName+" closed\n";
|
|
return ret;
|
|
}
|
|
|
|
int CLodHandler::infs(unsigned char * in, int size, int realSize, std::ofstream & out, int wBits)
|
|
{
|
|
int ret;
|
|
unsigned have;
|
|
z_stream strm;
|
|
unsigned char inx[NLoadHandlerHelp::fCHUNK];
|
|
unsigned char outx[NLoadHandlerHelp::fCHUNK];
|
|
|
|
/* allocate inflate state */
|
|
strm.zalloc = Z_NULL;
|
|
strm.zfree = Z_NULL;
|
|
strm.opaque = Z_NULL;
|
|
strm.avail_in = 0;
|
|
strm.next_in = Z_NULL;
|
|
ret = inflateInit2(&strm, wBits);
|
|
if (ret != Z_OK)
|
|
return ret;
|
|
int chunkNumber = 0;
|
|
do
|
|
{
|
|
int readBytes = 0;
|
|
for(int i=0; i<NLoadHandlerHelp::fCHUNK && (chunkNumber * NLoadHandlerHelp::fCHUNK + i)<size; ++i)
|
|
{
|
|
inx[i] = in[chunkNumber * NLoadHandlerHelp::fCHUNK + i];
|
|
++readBytes;
|
|
}
|
|
++chunkNumber;
|
|
strm.avail_in = readBytes;
|
|
//strm.avail_in = fread(inx, 1, NLoadHandlerHelp::fCHUNK, source);
|
|
/*if (in.bad())
|
|
{
|
|
(void)inflateEnd(&strm);
|
|
return Z_ERRNO;
|
|
}*/
|
|
if (strm.avail_in == 0)
|
|
break;
|
|
strm.next_in = inx;
|
|
|
|
/* run inflate() on input until output buffer not full */
|
|
do
|
|
{
|
|
strm.avail_out = NLoadHandlerHelp::fCHUNK;
|
|
strm.next_out = outx;
|
|
ret = inflate(&strm, Z_NO_FLUSH);
|
|
//assert(ret != Z_STREAM_ERROR); /* state not clobbered */
|
|
switch (ret)
|
|
{
|
|
case Z_NEED_DICT:
|
|
ret = Z_DATA_ERROR; /* and fall through */
|
|
case Z_DATA_ERROR:
|
|
case Z_MEM_ERROR:
|
|
(void)inflateEnd(&strm);
|
|
return ret;
|
|
}
|
|
have = NLoadHandlerHelp::fCHUNK - strm.avail_out;
|
|
/*if (fwrite(out, 1, have, dest) != have || ferror(dest))
|
|
{
|
|
(void)inflateEnd(&strm);
|
|
return Z_ERRNO;
|
|
}*/
|
|
out.write((char*)outx, have);
|
|
if(out.bad())
|
|
{
|
|
(void)inflateEnd(&strm);
|
|
return Z_ERRNO;
|
|
}
|
|
} while (strm.avail_out == 0);
|
|
|
|
/* done when inflate() says it's done */
|
|
} while (ret != Z_STREAM_END);
|
|
|
|
/* clean up and return */
|
|
(void)inflateEnd(&strm);
|
|
return ret == Z_STREAM_END ? Z_OK : Z_DATA_ERROR;
|
|
}
|
|
|
|
int CLodHandler::infs2(unsigned char * in, int size, int realSize, unsigned char *& out, int wBits)
|
|
{
|
|
int ret;
|
|
unsigned have;
|
|
z_stream strm;
|
|
unsigned char inx[NLoadHandlerHelp::fCHUNK];
|
|
unsigned char outx[NLoadHandlerHelp::fCHUNK];
|
|
out = new unsigned char [size];
|
|
int latPosOut = 0;
|
|
|
|
/* allocate inflate state */
|
|
strm.zalloc = Z_NULL;
|
|
strm.zfree = Z_NULL;
|
|
strm.opaque = Z_NULL;
|
|
strm.avail_in = 0;
|
|
strm.next_in = Z_NULL;
|
|
ret = inflateInit2(&strm, wBits);
|
|
if (ret != Z_OK)
|
|
return ret;
|
|
int chunkNumber = 0;
|
|
do
|
|
{
|
|
int readBytes = 0;
|
|
for(int i=0; i<NLoadHandlerHelp::fCHUNK && (chunkNumber * NLoadHandlerHelp::fCHUNK + i)<size; ++i)
|
|
{
|
|
inx[i] = in[chunkNumber * NLoadHandlerHelp::fCHUNK + i];
|
|
++readBytes;
|
|
}
|
|
++chunkNumber;
|
|
strm.avail_in = readBytes;
|
|
//strm.avail_in = fread(inx, 1, NLoadHandlerHelp::fCHUNK, source);
|
|
/*if (in.bad())
|
|
{
|
|
(void)inflateEnd(&strm);
|
|
return Z_ERRNO;
|
|
}*/
|
|
if (strm.avail_in == 0)
|
|
break;
|
|
strm.next_in = inx;
|
|
|
|
/* run inflate() on input until output buffer not full */
|
|
do
|
|
{
|
|
strm.avail_out = NLoadHandlerHelp::fCHUNK;
|
|
strm.next_out = outx;
|
|
ret = inflate(&strm, Z_NO_FLUSH);
|
|
//assert(ret != Z_STREAM_ERROR); /* state not clobbered */
|
|
switch (ret)
|
|
{
|
|
case Z_NEED_DICT:
|
|
ret = Z_DATA_ERROR; /* and fall through */
|
|
case Z_DATA_ERROR:
|
|
case Z_MEM_ERROR:
|
|
(void)inflateEnd(&strm);
|
|
return ret;
|
|
}
|
|
have = NLoadHandlerHelp::fCHUNK - strm.avail_out;
|
|
/*if (fwrite(out, 1, have, dest) != have || ferror(dest))
|
|
{
|
|
(void)inflateEnd(&strm);
|
|
return Z_ERRNO;
|
|
}*/
|
|
//out.write((char*)outx, have);
|
|
for(int oo=0; oo<have; ++oo)
|
|
{
|
|
out[latPosOut] = outx[oo];
|
|
++latPosOut;
|
|
}
|
|
/*if(out.bad())
|
|
{
|
|
(void)inflateEnd(&strm);
|
|
return Z_ERRNO;
|
|
}*/
|
|
} while (strm.avail_out == 0);
|
|
|
|
/* done when inflate() says it's done */
|
|
} while (ret != Z_STREAM_END);
|
|
|
|
/* clean up and return */
|
|
(void)inflateEnd(&strm);
|
|
return ret == Z_STREAM_END ? Z_OK : Z_DATA_ERROR;
|
|
}
|
|
|
|
void CLodHandler::extract(std::string FName)
|
|
{
|
|
std::ifstream FLOD;
|
|
std::ofstream FOut;
|
|
int i;
|
|
|
|
std::string Ts;
|
|
//std::cout<<"*** Loading FAT ... \n";
|
|
FLOD.open(FName.c_str(),std::ios::binary);
|
|
//std::cout<<"*** Archive: "+FName+" loaded\n";
|
|
FLOD.seekg(8,std::ios_base::beg);
|
|
unsigned char temp[4];
|
|
FLOD.read((char*)temp,4);
|
|
totalFiles = readNormalNr(temp,4);
|
|
FLOD.seekg(0x5c,std::ios_base::beg);
|
|
entries.reserve(totalFiles);
|
|
//std::cout<<"*** Loading FAT ...\n";
|
|
for (int i=0; i<totalFiles; i++)
|
|
{
|
|
entries.push_back(Entry());
|
|
//FLOD.read((char*)entries[i].name,12);
|
|
char * bufc = new char;
|
|
bool appending = true;
|
|
for(int kk=0; kk<12; ++kk)
|
|
{
|
|
FLOD.read(bufc, 1);
|
|
if(appending)
|
|
{
|
|
entries[i].name[kk] = *bufc;
|
|
}
|
|
else
|
|
{
|
|
entries[i].name[kk] = 0;
|
|
appending = false;
|
|
}
|
|
}
|
|
delete bufc;
|
|
FLOD.read((char*)entries[i].hlam_1,4);
|
|
FLOD.read((char*)temp,4);
|
|
entries[i].offset=readNormalNr(temp,4);
|
|
FLOD.read((char*)temp,4);
|
|
entries[i].realSize=readNormalNr(temp,4);
|
|
FLOD.read((char*)entries[i].hlam_2,4);
|
|
FLOD.read((char*)temp,4);
|
|
entries[i].size=readNormalNr(temp,4);
|
|
}
|
|
//std::cout<<" done\n";
|
|
for (int i=0;i<totalFiles;i++)
|
|
{
|
|
FLOD.seekg(entries[i].offset,std::ios_base::beg);
|
|
std::string bufff = (FName.substr(0, FName.size()-4) + "\\" + (char*)entries[i].name);
|
|
unsigned char * outp;
|
|
if (entries[i].size==0) //file is not compressed
|
|
{
|
|
outp = new unsigned char[entries[i].realSize];
|
|
FLOD.read((char*)outp, entries[i].realSize);
|
|
std::ofstream out;
|
|
out.open(bufff.c_str(), std::ios::binary);
|
|
if(!out.is_open())
|
|
{
|
|
std::cout<<"Unable to create "<<bufff;
|
|
}
|
|
else
|
|
{
|
|
for(int hh=0; hh<entries[i].realSize; ++hh)
|
|
{
|
|
out<<*(outp+hh);
|
|
}
|
|
out.close();
|
|
}
|
|
}
|
|
else
|
|
{
|
|
outp = new unsigned char[entries[i].size];
|
|
FLOD.read((char*)outp, entries[i].size);
|
|
FLOD.seekg(0, std::ios_base::beg);
|
|
std::ofstream destin;
|
|
destin.open(bufff.c_str(), std::ios::binary);
|
|
//int decRes = decompress(outp, entries[i].size, entries[i].realSize, bufff);
|
|
int decRes = infs(outp, entries[i].size, entries[i].realSize, destin);
|
|
destin.close();
|
|
if(decRes!=0)
|
|
{
|
|
std::cout<<"LOD Extraction error"<<" "<<decRes<<" while extracting to "<<bufff<<std::endl;
|
|
}
|
|
}
|
|
//for (int j=0; j<entries[i].size; j++)
|
|
// FOut << outp[j];
|
|
//FOut.flush();
|
|
delete outp;
|
|
//FOut.close();
|
|
//std::cout<<"*** done\n";
|
|
}
|
|
FLOD.close();
|
|
//std::cout<<"*** Archive: "+FName+" closed\n";
|
|
}
|
|
|
|
void CLodHandler::extractFile(std::string FName, std::string name)
|
|
{
|
|
std::ifstream FLOD;
|
|
std::ofstream FOut;
|
|
int i;
|
|
|
|
std::string Ts;
|
|
//std::cout<<"*** Loading FAT ... \n";
|
|
FLOD.open(FName.c_str(),std::ios::binary);
|
|
//std::cout<<"*** Archive: "+FName+" loaded\n";
|
|
FLOD.seekg(8,std::ios_base::beg);
|
|
unsigned char temp[4];
|
|
FLOD.read((char*)temp,4);
|
|
totalFiles = readNormalNr(temp,4);
|
|
FLOD.seekg(0x5c,std::ios_base::beg);
|
|
entries.reserve(totalFiles);
|
|
//std::cout<<"*** Loading FAT ...\n";
|
|
for (int i=0; i<totalFiles; i++)
|
|
{
|
|
entries.push_back(Entry());
|
|
//FLOD.read((char*)entries[i].name,12);
|
|
char * bufc = new char;
|
|
bool appending = true;
|
|
for(int kk=0; kk<12; ++kk)
|
|
{
|
|
FLOD.read(bufc, 1);
|
|
if(appending)
|
|
{
|
|
entries[i].name[kk] = *bufc;
|
|
}
|
|
else
|
|
{
|
|
entries[i].name[kk] = 0;
|
|
appending = false;
|
|
}
|
|
}
|
|
delete bufc;
|
|
FLOD.read((char*)entries[i].hlam_1,4);
|
|
FLOD.read((char*)temp,4);
|
|
entries[i].offset=readNormalNr(temp,4);
|
|
FLOD.read((char*)temp,4);
|
|
entries[i].realSize=readNormalNr(temp,4);
|
|
FLOD.read((char*)entries[i].hlam_2,4);
|
|
FLOD.read((char*)temp,4);
|
|
entries[i].size=readNormalNr(temp,4);
|
|
}
|
|
//std::cout<<" done\n";
|
|
std::transform(name.begin(), name.end(), name.begin(), (int(*)(int))toupper);
|
|
for (int i=0;i<totalFiles;i++)
|
|
{
|
|
std::string buf1 = std::string((char*)entries[i].name);
|
|
std::transform(buf1.begin(), buf1.end(), buf1.begin(), (int(*)(int))toupper);
|
|
if(buf1!=name)
|
|
continue;
|
|
FLOD.seekg(entries[i].offset,std::ios_base::beg);
|
|
std::string bufff = (FName.substr(0, FName.size()-4) + "\\" + (char*)entries[i].name);
|
|
unsigned char * outp;
|
|
if (entries[i].size==0) //file is not compressed
|
|
{
|
|
outp = new unsigned char[entries[i].realSize];
|
|
FLOD.read((char*)outp, entries[i].realSize);
|
|
std::ofstream out;
|
|
out.open(bufff.c_str(), std::ios::binary);
|
|
if(!out.is_open())
|
|
{
|
|
std::cout<<"Unable to create "<<bufff;
|
|
}
|
|
else
|
|
{
|
|
for(int hh=0; hh<entries[i].realSize; ++hh)
|
|
{
|
|
out<<*(outp+hh);
|
|
}
|
|
out.close();
|
|
}
|
|
}
|
|
else //we will decompressing file
|
|
{
|
|
outp = new unsigned char[entries[i].size];
|
|
FLOD.read((char*)outp, entries[i].size);
|
|
FLOD.seekg(0, std::ios_base::beg);
|
|
std::ofstream destin;
|
|
destin.open(bufff.c_str(), std::ios::binary);
|
|
//int decRes = decompress(outp, entries[i].size, entries[i].realSize, bufff);
|
|
int decRes = infs(outp, entries[i].size, entries[i].realSize, destin);
|
|
destin.close();
|
|
if(decRes!=0)
|
|
{
|
|
std::cout<<"LOD Extraction error"<<" "<<decRes<<" while extracting to "<<bufff<<std::endl;
|
|
}
|
|
}
|
|
//for (int j=0; j<entries[i].size; j++)
|
|
// FOut << outp[j];
|
|
//FOut.flush();
|
|
delete outp;
|
|
//FOut.close();
|
|
//std::cout<<"*** done\n";
|
|
}
|
|
FLOD.close();
|
|
//std::cout<<"*** Archive: "+FName+" closed\n";
|
|
}
|
|
|
|
int CLodHandler::readNormalNr (unsigned char* bufor, int bytCon, bool cyclic)
|
|
{
|
|
int ret=0;
|
|
int amp=1;
|
|
for (int i=0; i<bytCon; i++)
|
|
{
|
|
ret+=bufor[i]*amp;
|
|
amp*=256;
|
|
}
|
|
if(cyclic && bytCon<4 && ret>=amp/2)
|
|
{
|
|
ret = ret-amp;
|
|
}
|
|
return ret;
|
|
}
|
|
|
|
void CLodHandler::init(std::string lodFile)
|
|
{
|
|
extract(lodFile);
|
|
}
|