virtualx-engine/core/io/file_access_pack.h

229 lines
6.4 KiB
C++
Raw Normal View History

2014-02-10 02:10:30 +01:00
/*************************************************************************/
/* file_access_pack.h */
/*************************************************************************/
/* This file is part of: */
/* GODOT ENGINE */
/* https://godotengine.org */
2014-02-10 02:10:30 +01:00
/*************************************************************************/
/* Copyright (c) 2007-2019 Juan Linietsky, Ariel Manzur. */
/* Copyright (c) 2014-2019 Godot Engine contributors (cf. AUTHORS.md) */
2014-02-10 02:10:30 +01:00
/* */
/* Permission is hereby granted, free of charge, to any person obtaining */
/* a copy of this software and associated documentation files (the */
/* "Software"), to deal in the Software without restriction, including */
/* without limitation the rights to use, copy, modify, merge, publish, */
/* distribute, sublicense, and/or sell copies of the Software, and to */
/* permit persons to whom the Software is furnished to do so, subject to */
/* the following conditions: */
/* */
/* The above copyright notice and this permission notice shall be */
/* included in all copies or substantial portions of the Software. */
/* */
/* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */
/* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */
/* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.*/
/* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */
/* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */
/* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */
/* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */
/*************************************************************************/
2014-02-10 02:10:30 +01:00
#ifndef FILE_ACCESS_PACK_H
#define FILE_ACCESS_PACK_H
#include "core/list.h"
#include "core/map.h"
#include "core/os/dir_access.h"
#include "core/os/file_access.h"
#include "core/print_string.h"
2014-02-10 02:10:30 +01:00
class PackSource;
class PackedData {
friend class FileAccessPack;
friend class DirAccessPack;
friend class PackSource;
2014-02-10 02:10:30 +01:00
public:
struct PackedFile {
String pack;
uint64_t offset; //if offset is ZERO, the file was ERASED
2014-02-10 02:10:30 +01:00
uint64_t size;
uint8_t md5[16];
PackSource *src;
2014-02-10 02:10:30 +01:00
};
private:
struct PackedDir {
PackedDir *parent;
String name;
Map<String, PackedDir *> subdirs;
2014-02-10 02:10:30 +01:00
Set<String> files;
};
struct PathMD5 {
uint64_t a;
uint64_t b;
bool operator<(const PathMD5 &p_md5) const {
if (p_md5.a == a) {
return b < p_md5.b;
} else {
return a < p_md5.a;
}
}
bool operator==(const PathMD5 &p_md5) const {
return a == p_md5.a && b == p_md5.b;
};
PathMD5() {
a = b = 0;
};
PathMD5(const Vector<uint8_t> p_buf) {
a = *((uint64_t *)&p_buf[0]);
b = *((uint64_t *)&p_buf[8]);
};
};
Map<PathMD5, PackedFile> files;
2014-02-10 02:10:30 +01:00
Vector<PackSource *> sources;
2014-02-10 02:10:30 +01:00
PackedDir *root;
//Map<String,PackedDir*> dirs;
static PackedData *singleton;
bool disabled;
void _free_packed_dirs(PackedDir *p_dir);
2014-02-10 02:10:30 +01:00
public:
void add_pack_source(PackSource *p_source);
void add_path(const String &pkg_path, const String &path, uint64_t ofs, uint64_t size, const uint8_t *p_md5, PackSource *p_src); // for PackSource
2014-02-10 02:10:30 +01:00
void set_disabled(bool p_disabled) { disabled = p_disabled; }
2014-02-10 02:10:30 +01:00
_FORCE_INLINE_ bool is_disabled() const { return disabled; }
static PackedData *get_singleton() { return singleton; }
Error add_pack(const String &p_path);
2014-02-10 02:10:30 +01:00
_FORCE_INLINE_ FileAccess *try_open_path(const String &p_path);
_FORCE_INLINE_ bool has_path(const String &p_path);
2014-02-10 02:10:30 +01:00
PackedData();
~PackedData();
2014-02-10 02:10:30 +01:00
};
class PackSource {
public:
virtual bool try_open_pack(const String &p_path) = 0;
virtual FileAccess *get_file(const String &p_path, PackedData::PackedFile *p_file) = 0;
virtual ~PackSource() {}
2014-02-10 02:10:30 +01:00
};
class PackedSourcePCK : public PackSource {
public:
virtual bool try_open_pack(const String &p_path);
virtual FileAccess *get_file(const String &p_path, PackedData::PackedFile *p_file);
2014-02-10 02:10:30 +01:00
};
class FileAccessPack : public FileAccess {
PackedData::PackedFile pf;
mutable size_t pos;
mutable bool eof;
FileAccess *f;
virtual Error _open(const String &p_path, int p_mode_flags);
virtual uint64_t _get_modified_time(const String &p_file) { return 0; }
2014-02-10 02:10:30 +01:00
public:
virtual void close();
virtual bool is_open() const;
virtual void seek(size_t p_position);
virtual void seek_end(int64_t p_position = 0);
virtual size_t get_position() const;
2014-02-10 02:10:30 +01:00
virtual size_t get_len() const;
virtual bool eof_reached() const;
virtual uint8_t get_8() const;
virtual int get_buffer(uint8_t *p_dst, int p_length) const;
2014-02-10 02:10:30 +01:00
virtual void set_endian_swap(bool p_swap);
virtual Error get_error() const;
virtual void flush();
2014-02-10 02:10:30 +01:00
virtual void store_8(uint8_t p_dest);
virtual void store_buffer(const uint8_t *p_src, int p_length);
2014-02-10 02:10:30 +01:00
virtual bool file_exists(const String &p_name);
2014-02-10 02:10:30 +01:00
FileAccessPack(const String &p_path, const PackedData::PackedFile &p_file);
2014-02-10 02:10:30 +01:00
~FileAccessPack();
};
FileAccess *PackedData::try_open_path(const String &p_path) {
2014-02-10 02:10:30 +01:00
PathMD5 pmd5(p_path.md5_buffer());
Map<PathMD5, PackedFile>::Element *E = files.find(pmd5);
if (!E)
return NULL; //not found
if (E->get().offset == 0)
return NULL; //was erased
2014-02-10 02:10:30 +01:00
return E->get().src->get_file(p_path, &E->get());
2014-02-10 02:10:30 +01:00
}
bool PackedData::has_path(const String &p_path) {
2014-02-10 02:10:30 +01:00
return files.has(PathMD5(p_path.md5_buffer()));
2014-02-10 02:10:30 +01:00
}
class DirAccessPack : public DirAccess {
PackedData::PackedDir *current;
List<String> list_dirs;
List<String> list_files;
bool cdir;
public:
virtual Error list_dir_begin();
2014-02-10 02:10:30 +01:00
virtual String get_next();
virtual bool current_is_dir() const;
2015-03-21 18:33:32 +01:00
virtual bool current_is_hidden() const;
2014-02-10 02:10:30 +01:00
virtual void list_dir_end();
virtual int get_drive_count();
virtual String get_drive(int p_drive);
virtual Error change_dir(String p_dir);
virtual String get_current_dir();
virtual bool file_exists(String p_file);
2014-05-25 05:34:51 +02:00
virtual bool dir_exists(String p_dir);
2014-02-10 02:10:30 +01:00
virtual Error make_dir(String p_dir);
virtual Error rename(String p_from, String p_to);
virtual Error remove(String p_name);
size_t get_space_left();
DirAccessPack();
~DirAccessPack();
};
#endif // FILE_ACCESS_PACK_H