summaryrefslogtreecommitdiff
path: root/core/io
diff options
context:
space:
mode:
Diffstat (limited to 'core/io')
-rw-r--r--core/io/file_access_compressed.cpp786
1 files changed, 393 insertions, 393 deletions
diff --git a/core/io/file_access_compressed.cpp b/core/io/file_access_compressed.cpp
index ec9dce28e2..b90a41df54 100644
--- a/core/io/file_access_compressed.cpp
+++ b/core/io/file_access_compressed.cpp
@@ -26,396 +26,396 @@
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
/*************************************************************************/
-#include "file_access_compressed.h"
-#include "print_string.h"
-void FileAccessCompressed::configure(const String& p_magic, Compression::Mode p_mode, int p_block_size) {
-
- magic=p_magic.ascii().get_data();
- if (magic.length()>4)
- magic=magic.substr(0,4);
- else {
- while(magic.length()<4)
- magic+=" ";
- }
-
- cmode=p_mode;
- block_size=p_block_size;
-
-}
-
-
-#define WRITE_FIT(m_bytes) \
-{\
-if (write_pos+(m_bytes) > write_max) {\
- write_max=write_pos+(m_bytes);\
-}\
-if (write_max > write_buffer_size) {\
- write_buffer_size = nearest_power_of_2( write_max );\
- buffer.resize(write_buffer_size);\
- write_ptr=buffer.ptr();\
-}\
-}
-
-
-Error FileAccessCompressed::open_after_magic(FileAccess *p_base) {
-
-
- f=p_base;
- cmode=(Compression::Mode)f->get_32();
- block_size=f->get_32();
- read_total=f->get_32();
- int bc = (read_total/block_size)+1;
- int acc_ofs=f->get_pos()+bc*4;
- int max_bs=0;
- for(int i=0;i<bc;i++) {
-
- ReadBlock rb;
- rb.offset=acc_ofs;
- rb.csize=f->get_32();
- acc_ofs+=rb.csize;
- max_bs=MAX(max_bs,rb.csize);
- read_blocks.push_back(rb);
-
-
- }
-
- comp_buffer.resize(max_bs);
- buffer.resize(block_size);
- read_ptr=buffer.ptr();
- f->get_buffer(comp_buffer.ptr(),read_blocks[0].csize);
- at_end=false;
- read_eof=false;
- read_block_count=bc;
- read_block_size=read_blocks.size()==1?read_total:block_size;
-
- Compression::decompress(buffer.ptr(),read_block_size,comp_buffer.ptr(),read_blocks[0].csize,cmode);
- read_block=0;
- read_pos=0;
-
- return OK;
-
-}
-
-Error FileAccessCompressed::_open(const String& p_path, int p_mode_flags){
-
- ERR_FAIL_COND_V(p_mode_flags==READ_WRITE,ERR_UNAVAILABLE);
-
- if (f)
- close();
-
-
- Error err;
- f = FileAccess::open(p_path,p_mode_flags,&err);
- if (err!=OK) {
- //not openable
-
- f=NULL;
- return err;
- }
-
- if (p_mode_flags&WRITE) {
-
- buffer.clear();
- writing=true;
- write_pos=0;
- write_buffer_size=256;
- buffer.resize(256);
- write_max=0;
- write_ptr=buffer.ptr();
-
-
-
- //don't store anything else unless it's done saving!
- } else {
-
- char rmagic[5];
- f->get_buffer((uint8_t*)rmagic,4);
- rmagic[4]=0;
- if (magic!=rmagic) {
- memdelete(f);
- f=NULL;
- return ERR_FILE_UNRECOGNIZED;
- }
-
- open_after_magic(f);
-
- }
-
- return OK;
-
-}
-void FileAccessCompressed::close(){
-
- if (!f)
- return;
-
-
- if (writing) {
- //save block table and all compressed blocks
-
- CharString mgc = magic.utf8();
- f->store_buffer((const uint8_t*)mgc.get_data(),mgc.length()); //write header 4
- f->store_32(cmode); //write compression mode 4
- f->store_32(block_size); //write block size 4
- f->store_32(write_max); //max amount of data written 4
- int bc=(write_max/block_size)+1;
-
- for(int i=0;i<bc;i++) {
- f->store_32(0); //compressed sizes, will update later
- }
-
-
- Vector<int> block_sizes;
- for(int i=0;i<bc;i++) {
-
- int bl = i==(bc-1) ? write_max % block_size : block_size;
- uint8_t *bp = &write_ptr[i*block_size];
-
- Vector<uint8_t> cblock;
- cblock.resize(Compression::get_max_compressed_buffer_size(bl,cmode));
- int s = Compression::compress(cblock.ptr(),bp,bl,cmode);
-
- f->store_buffer(cblock.ptr(),s);
- block_sizes.push_back(s);
- }
-
- f->seek(16); //ok write block sizes
- for(int i=0;i<bc;i++)
- f->store_32(block_sizes[i]);
- f->seek_end();
- f->store_buffer((const uint8_t*)mgc.get_data(),mgc.length()); //magic at the end too
-
- buffer.clear();
-
- } else {
-
-
- comp_buffer.clear();
- buffer.clear();
- read_blocks.clear();
- }
-
- memdelete(f);
- f=NULL;
-
-}
-
-bool FileAccessCompressed::is_open() const{
-
- return f!=NULL;
-}
-
-void FileAccessCompressed::seek(size_t p_position){
-
- ERR_FAIL_COND(!f);
- if (writing) {
-
- ERR_FAIL_COND(p_position>write_max);
-
- write_pos=p_position;
-
- } else {
-
- ERR_FAIL_COND(p_position>read_total);
- if (p_position==read_total) {
- at_end=true;
- } else {
-
- int block_idx = p_position/block_size;
- if (block_idx!=read_block) {
-
- read_block=block_idx;
- f->seek(read_blocks[read_block].offset);
- f->get_buffer(comp_buffer.ptr(),read_blocks[read_block].csize);
- Compression::decompress(buffer.ptr(),read_blocks.size()==1?read_total:block_size,comp_buffer.ptr(),read_blocks[read_block].csize,cmode);
- read_block_size=read_block==read_block_count-1?read_total%block_size:block_size;
- }
-
- read_pos=p_position%block_size;
- }
- }
-}
-
-
-void FileAccessCompressed::seek_end(int64_t p_position){
-
- ERR_FAIL_COND(!f);
- if (writing) {
-
- seek(write_max+p_position);
- } else {
-
- seek(read_total+p_position);
-
- }
-
-
-}
-size_t FileAccessCompressed::get_pos() const{
-
- ERR_FAIL_COND_V(!f,0);
- if (writing) {
-
- return write_pos;
- } else {
-
- return read_block*block_size+read_pos;
- }
-
-}
-size_t FileAccessCompressed::get_len() const{
-
- ERR_FAIL_COND_V(!f,0);
- if (writing) {
-
- return write_max;
- } else {
- return read_total;
- }
-}
-
-bool FileAccessCompressed::eof_reached() const{
-
- ERR_FAIL_COND_V(!f,false);
- if (writing) {
- return false;
- } else {
- return read_eof;
- }
-}
-
-uint8_t FileAccessCompressed::get_8() const{
-
- ERR_FAIL_COND_V(writing,0);
- ERR_FAIL_COND_V(!f,0);
-
- if (at_end) {
- read_eof=true;
- return 0;
- }
-
- uint8_t ret = read_ptr[read_pos];
-
- read_pos++;
- if (read_pos>=read_block_size) {
- read_block++;
-
- if (read_block<read_block_count) {
- //read another block of compressed data
- f->get_buffer(comp_buffer.ptr(),read_blocks[read_block].csize);
- Compression::decompress(buffer.ptr(),read_blocks.size()==1?read_total:block_size,comp_buffer.ptr(),read_blocks[read_block].csize,cmode);
- read_block_size=read_block==read_block_count-1?read_total%block_size:block_size;
- read_pos=0;
-
- } else {
- read_block--;
- at_end=true;
- ret =0;
- }
- }
-
- return ret;
-
-}
-int FileAccessCompressed::get_buffer(uint8_t *p_dst, int p_length) const{
-
- ERR_FAIL_COND_V(writing,0);
- ERR_FAIL_COND_V(!f,0);
-
- if (at_end) {
- read_eof=true;
- return 0;
- }
-
-
- for(int i=0;i<p_length;i++) {
-
-
- p_dst[i]=read_ptr[read_pos];
- read_pos++;
- if (read_pos>=read_block_size) {
- read_block++;
-
- if (read_block<read_block_count) {
- //read another block of compressed data
- f->get_buffer(comp_buffer.ptr(),read_blocks[read_block].csize);
- Compression::decompress(buffer.ptr(),read_blocks.size()==1?read_total:block_size,comp_buffer.ptr(),read_blocks[read_block].csize,cmode);
- read_block_size=read_block==read_block_count-1?read_total%block_size:block_size;
- read_pos=0;
-
- } else {
- read_block--;
- at_end=true;
- if (i<p_length-1)
- read_eof=true;
- return i;
-
- }
- }
-
- }
-
- return p_length;
-
-}
-
-Error FileAccessCompressed::get_error() const{
-
- return read_eof?ERR_FILE_EOF:OK;
-}
-
-void FileAccessCompressed::store_8(uint8_t p_dest){
-
- ERR_FAIL_COND(!f);
- ERR_FAIL_COND(!writing);
-
- WRITE_FIT(1);
- write_ptr[write_pos++]=p_dest;
-
-}
-
-bool FileAccessCompressed::file_exists(const String& p_name){
-
- FileAccess *fa = FileAccess::open(p_name,FileAccess::READ);
- if (!fa)
- return false;
- memdelete(fa);
- return true;
-}
-
-uint64_t FileAccessCompressed::_get_modified_time(const String& p_file) {
-
- if (f)
- return f->get_modified_time(p_file);
- else
- return 0;
-}
-
-FileAccessCompressed::FileAccessCompressed() {
-
- f=NULL;
- magic="GCMP";
- block_size=4096;
- cmode=Compression::MODE_FASTLZ;
- writing=false;
- write_ptr=0;
- write_buffer_size=0;
- write_max=0;
- block_size=0;
- read_eof=false;
- at_end=false;
- read_total=0;
- read_ptr=NULL;
- read_block=0;
- read_block_count=0;
- read_block_size=0;
- read_pos=0;
-
-}
-
-FileAccessCompressed::~FileAccessCompressed(){
-
- if (f)
- close();
-
-}
+#include "file_access_compressed.h"
+#include "print_string.h"
+void FileAccessCompressed::configure(const String& p_magic, Compression::Mode p_mode, int p_block_size) {
+
+ magic=p_magic.ascii().get_data();
+ if (magic.length()>4)
+ magic=magic.substr(0,4);
+ else {
+ while(magic.length()<4)
+ magic+=" ";
+ }
+
+ cmode=p_mode;
+ block_size=p_block_size;
+
+}
+
+
+#define WRITE_FIT(m_bytes) \
+{\
+if (write_pos+(m_bytes) > write_max) {\
+ write_max=write_pos+(m_bytes);\
+}\
+if (write_max > write_buffer_size) {\
+ write_buffer_size = nearest_power_of_2( write_max );\
+ buffer.resize(write_buffer_size);\
+ write_ptr=buffer.ptr();\
+}\
+}
+
+
+Error FileAccessCompressed::open_after_magic(FileAccess *p_base) {
+
+
+ f=p_base;
+ cmode=(Compression::Mode)f->get_32();
+ block_size=f->get_32();
+ read_total=f->get_32();
+ int bc = (read_total/block_size)+1;
+ int acc_ofs=f->get_pos()+bc*4;
+ int max_bs=0;
+ for(int i=0;i<bc;i++) {
+
+ ReadBlock rb;
+ rb.offset=acc_ofs;
+ rb.csize=f->get_32();
+ acc_ofs+=rb.csize;
+ max_bs=MAX(max_bs,rb.csize);
+ read_blocks.push_back(rb);
+
+
+ }
+
+ comp_buffer.resize(max_bs);
+ buffer.resize(block_size);
+ read_ptr=buffer.ptr();
+ f->get_buffer(comp_buffer.ptr(),read_blocks[0].csize);
+ at_end=false;
+ read_eof=false;
+ read_block_count=bc;
+ read_block_size=read_blocks.size()==1?read_total:block_size;
+
+ Compression::decompress(buffer.ptr(),read_block_size,comp_buffer.ptr(),read_blocks[0].csize,cmode);
+ read_block=0;
+ read_pos=0;
+
+ return OK;
+
+}
+
+Error FileAccessCompressed::_open(const String& p_path, int p_mode_flags){
+
+ ERR_FAIL_COND_V(p_mode_flags==READ_WRITE,ERR_UNAVAILABLE);
+
+ if (f)
+ close();
+
+
+ Error err;
+ f = FileAccess::open(p_path,p_mode_flags,&err);
+ if (err!=OK) {
+ //not openable
+
+ f=NULL;
+ return err;
+ }
+
+ if (p_mode_flags&WRITE) {
+
+ buffer.clear();
+ writing=true;
+ write_pos=0;
+ write_buffer_size=256;
+ buffer.resize(256);
+ write_max=0;
+ write_ptr=buffer.ptr();
+
+
+
+ //don't store anything else unless it's done saving!
+ } else {
+
+ char rmagic[5];
+ f->get_buffer((uint8_t*)rmagic,4);
+ rmagic[4]=0;
+ if (magic!=rmagic) {
+ memdelete(f);
+ f=NULL;
+ return ERR_FILE_UNRECOGNIZED;
+ }
+
+ open_after_magic(f);
+
+ }
+
+ return OK;
+
+}
+void FileAccessCompressed::close(){
+
+ if (!f)
+ return;
+
+
+ if (writing) {
+ //save block table and all compressed blocks
+
+ CharString mgc = magic.utf8();
+ f->store_buffer((const uint8_t*)mgc.get_data(),mgc.length()); //write header 4
+ f->store_32(cmode); //write compression mode 4
+ f->store_32(block_size); //write block size 4
+ f->store_32(write_max); //max amount of data written 4
+ int bc=(write_max/block_size)+1;
+
+ for(int i=0;i<bc;i++) {
+ f->store_32(0); //compressed sizes, will update later
+ }
+
+
+ Vector<int> block_sizes;
+ for(int i=0;i<bc;i++) {
+
+ int bl = i==(bc-1) ? write_max % block_size : block_size;
+ uint8_t *bp = &write_ptr[i*block_size];
+
+ Vector<uint8_t> cblock;
+ cblock.resize(Compression::get_max_compressed_buffer_size(bl,cmode));
+ int s = Compression::compress(cblock.ptr(),bp,bl,cmode);
+
+ f->store_buffer(cblock.ptr(),s);
+ block_sizes.push_back(s);
+ }
+
+ f->seek(16); //ok write block sizes
+ for(int i=0;i<bc;i++)
+ f->store_32(block_sizes[i]);
+ f->seek_end();
+ f->store_buffer((const uint8_t*)mgc.get_data(),mgc.length()); //magic at the end too
+
+ buffer.clear();
+
+ } else {
+
+
+ comp_buffer.clear();
+ buffer.clear();
+ read_blocks.clear();
+ }
+
+ memdelete(f);
+ f=NULL;
+
+}
+
+bool FileAccessCompressed::is_open() const{
+
+ return f!=NULL;
+}
+
+void FileAccessCompressed::seek(size_t p_position){
+
+ ERR_FAIL_COND(!f);
+ if (writing) {
+
+ ERR_FAIL_COND(p_position>write_max);
+
+ write_pos=p_position;
+
+ } else {
+
+ ERR_FAIL_COND(p_position>read_total);
+ if (p_position==read_total) {
+ at_end=true;
+ } else {
+
+ int block_idx = p_position/block_size;
+ if (block_idx!=read_block) {
+
+ read_block=block_idx;
+ f->seek(read_blocks[read_block].offset);
+ f->get_buffer(comp_buffer.ptr(),read_blocks[read_block].csize);
+ Compression::decompress(buffer.ptr(),read_blocks.size()==1?read_total:block_size,comp_buffer.ptr(),read_blocks[read_block].csize,cmode);
+ read_block_size=read_block==read_block_count-1?read_total%block_size:block_size;
+ }
+
+ read_pos=p_position%block_size;
+ }
+ }
+}
+
+
+void FileAccessCompressed::seek_end(int64_t p_position){
+
+ ERR_FAIL_COND(!f);
+ if (writing) {
+
+ seek(write_max+p_position);
+ } else {
+
+ seek(read_total+p_position);
+
+ }
+
+
+}
+size_t FileAccessCompressed::get_pos() const{
+
+ ERR_FAIL_COND_V(!f,0);
+ if (writing) {
+
+ return write_pos;
+ } else {
+
+ return read_block*block_size+read_pos;
+ }
+
+}
+size_t FileAccessCompressed::get_len() const{
+
+ ERR_FAIL_COND_V(!f,0);
+ if (writing) {
+
+ return write_max;
+ } else {
+ return read_total;
+ }
+}
+
+bool FileAccessCompressed::eof_reached() const{
+
+ ERR_FAIL_COND_V(!f,false);
+ if (writing) {
+ return false;
+ } else {
+ return read_eof;
+ }
+}
+
+uint8_t FileAccessCompressed::get_8() const{
+
+ ERR_FAIL_COND_V(writing,0);
+ ERR_FAIL_COND_V(!f,0);
+
+ if (at_end) {
+ read_eof=true;
+ return 0;
+ }
+
+ uint8_t ret = read_ptr[read_pos];
+
+ read_pos++;
+ if (read_pos>=read_block_size) {
+ read_block++;
+
+ if (read_block<read_block_count) {
+ //read another block of compressed data
+ f->get_buffer(comp_buffer.ptr(),read_blocks[read_block].csize);
+ Compression::decompress(buffer.ptr(),read_blocks.size()==1?read_total:block_size,comp_buffer.ptr(),read_blocks[read_block].csize,cmode);
+ read_block_size=read_block==read_block_count-1?read_total%block_size:block_size;
+ read_pos=0;
+
+ } else {
+ read_block--;
+ at_end=true;
+ ret =0;
+ }
+ }
+
+ return ret;
+
+}
+int FileAccessCompressed::get_buffer(uint8_t *p_dst, int p_length) const{
+
+ ERR_FAIL_COND_V(writing,0);
+ ERR_FAIL_COND_V(!f,0);
+
+ if (at_end) {
+ read_eof=true;
+ return 0;
+ }
+
+
+ for(int i=0;i<p_length;i++) {
+
+
+ p_dst[i]=read_ptr[read_pos];
+ read_pos++;
+ if (read_pos>=read_block_size) {
+ read_block++;
+
+ if (read_block<read_block_count) {
+ //read another block of compressed data
+ f->get_buffer(comp_buffer.ptr(),read_blocks[read_block].csize);
+ Compression::decompress(buffer.ptr(),read_blocks.size()==1?read_total:block_size,comp_buffer.ptr(),read_blocks[read_block].csize,cmode);
+ read_block_size=read_block==read_block_count-1?read_total%block_size:block_size;
+ read_pos=0;
+
+ } else {
+ read_block--;
+ at_end=true;
+ if (i<p_length-1)
+ read_eof=true;
+ return i;
+
+ }
+ }
+
+ }
+
+ return p_length;
+
+}
+
+Error FileAccessCompressed::get_error() const{
+
+ return read_eof?ERR_FILE_EOF:OK;
+}
+
+void FileAccessCompressed::store_8(uint8_t p_dest){
+
+ ERR_FAIL_COND(!f);
+ ERR_FAIL_COND(!writing);
+
+ WRITE_FIT(1);
+ write_ptr[write_pos++]=p_dest;
+
+}
+
+bool FileAccessCompressed::file_exists(const String& p_name){
+
+ FileAccess *fa = FileAccess::open(p_name,FileAccess::READ);
+ if (!fa)
+ return false;
+ memdelete(fa);
+ return true;
+}
+
+uint64_t FileAccessCompressed::_get_modified_time(const String& p_file) {
+
+ if (f)
+ return f->get_modified_time(p_file);
+ else
+ return 0;
+}
+
+FileAccessCompressed::FileAccessCompressed() {
+
+ f=NULL;
+ magic="GCMP";
+ block_size=16384;
+ cmode=Compression::MODE_DEFLATE;
+ writing=false;
+ write_ptr=0;
+ write_buffer_size=0;
+ write_max=0;
+ block_size=0;
+ read_eof=false;
+ at_end=false;
+ read_total=0;
+ read_ptr=NULL;
+ read_block=0;
+ read_block_count=0;
+ read_block_size=0;
+ read_pos=0;
+
+}
+
+FileAccessCompressed::~FileAccessCompressed(){
+
+ if (f)
+ close();
+
+}
generated by cgit v1.2.3 (git 2.25.1) at 2024-11-02 21:41:00 +0000