Support adding packs from IO::Seekable objects

[libs/datafile.git] / source / packsource.cpp

diff --git a/source/packsource.cpp b/source/packsource.cpp
index 00af3ad129079aa71d7d415a39f16a67bf54b5b6..938a627c05fb9b3685cb25cb53678439e39697f7 100644 (file)
--- a/source/packsource.cpp
+++ b/source/packsource.cpp
@@ -24,6 +24,16 @@ void PackSource::add_pack_file(const string &fn)
 }
 
 void PackSource::add_pack_file(const string &fn, const string &filter)
+{
+       add_pack(0, fn, filter);
+}
+
+void PackSource::add_pack_io(IO::Seekable &io, const string &fn)
+{
+       add_pack(&io, fn, string());
+}
+
+void PackSource::add_pack(IO::Seekable *io, const string &fn, const string &filter)
 {
        Pack *pack = 0;
        for(list<Pack>::iterator i=packs.begin(); (!pack && i!=packs.end()); ++i)
@@ -31,9 +41,16 @@ void PackSource::add_pack_file(const string &fn, const string &filter)
                        pack = &*i;
        if(!pack)
        {
-               packs.push_back(Pack(fn));
+               packs.push_back(Pack(io, fn));
                pack = &packs.back();
-               DataFile::load(*pack, fn);
+               if(io)
+               {
+                       DataFile::Parser parser(*io, fn);
+                       Pack::Loader loader(*pack);
+                       loader.load(parser);
+               }
+               else
+                       DataFile::load(*pack, fn);
        }
 
        FileMap pack_files;
@@ -115,8 +132,9 @@ IO::Seekable *PackSource::open(const string &fn) const
 }
 
 
-PackSource::Pack::Pack(const string &fn):
+PackSource::Pack::Pack(IO::Seekable *i, const string &fn):
        filename(fn),
+       io(i),
        base_offset(0)
 { }
 
@@ -148,10 +166,16 @@ PackSource::File::File(const Pack &p, const string &fn):
 
 RefPtr<IO::Seekable> PackSource::File::open() const
 {
-       IO::BufferedFile *io_file = new IO::BufferedFile(pack.get_filename());
-       IO::Slice *io_slice = new IO::Slice(*io_file, pack.get_base_offset()+offset, length);
-       io_slice->signal_deleted.connect(sigc::bind(sigc::ptr_fun(delete_io), io_file));
-       return io_slice;
+       if(pack.get_io())
+               // TODO Performance may be poor without buffering
+               return new IO::Slice(*pack.get_io(), pack.get_base_offset()+offset, length);
+       else
+       {
+               IO::BufferedFile *io_file = new IO::BufferedFile(pack.get_filename());
+               IO::Slice *io_slice = new IO::Slice(*io_file, pack.get_base_offset()+offset, length);
+               io_slice->signal_deleted.connect(sigc::bind(sigc::ptr_fun(delete_io), io_file));
+               return io_slice;
+       }
 }
 
 PackSource::FileInfo PackSource::File::get_info() const