1 #include <msp/datafile/parser.h>
2 #include <msp/datafile/statement.h>
3 #include <msp/datafile/writer.h>
4 #include <msp/fs/utils.h>
5 #include <msp/io/memory.h>
6 #include <msp/strings/format.h>
13 Packer::Packer(DataTool &t):
16 tmp_buf(new IO::Buffered(*tmp_file)),
20 IO::File *Packer::tempfile()
22 for(unsigned i=0;; ++i)
26 std::string filename = format("/tmp/mspdatatool.%d", i);
27 /*filename.reserve(25);
28 filename.append("/tmp/mspdatatool.");
29 for(unsigned i=0; i<8; ++i)
30 filename.append(rand());*/
31 IO::File *file = new IO::File(filename, IO::M_RDWR, IO::File::C_NEW);
35 catch(const IO::file_already_exists &)
48 void Packer::pack_file(const string &fn)
51 throw logic_error("Packer::pack_file");
53 unsigned offset = tmp_file->tell();
55 IO::BufferedFile in(fn);
56 DataFile::Parser parser(in, fn);
58 DataFile::Writer *writer = tool.create_writer(*tmp_buf);
60 bool collection = FS::extpart(fn)==".mdc";
61 std::list<Object> objects;
64 DataFile::Statement st = parser.parse(true);
67 bool sys = !st.keyword.compare(0, 2, "__");
68 if(collection && !sys)
70 if(st.get_signature()=="s")
73 obj.name = st.args[0].get<string>();
74 obj.keyword = st.keyword;
75 objects.push_back(obj);
80 if(!sys || st.keyword=="__src")
84 writer->write(DataFile::Statement("__end"));
88 unsigned length = tmp_file->tell()-offset;
90 DataFile::Statement st("file");
91 st.append(FS::basename(fn));
92 st.sub.push_back((DataFile::Statement("slice"), offset, length));
95 for(list<Object>::const_iterator i=objects.begin(); i!=objects.end(); ++i)
96 st.sub.push_back((DataFile::Statement("object"), i->name, i->keyword));
97 dir_alloc += objects.size()*100;
99 directory.push_back(st);
103 void Packer::create_pack(const string &fn)
105 vector<char> dir_buffer(dir_alloc);
106 IO::Memory mem(&dir_buffer[0], dir_buffer.size(), IO::M_WRITE);
108 unsigned base_offset = 0;
111 mem.seek(0, IO::S_BEG);
113 DataFile::Writer *writer = tool.create_writer(mem);
115 for(list<DataFile::Statement>::const_iterator i=directory.begin(); i!=directory.end(); ++i)
118 base_offset = mem.tell();
119 writer->write((DataFile::Statement("base_offset"), base_offset));
120 writer->write(DataFile::Statement("__end"));
123 unsigned dir_size = mem.tell();
124 if(dir_size<=base_offset)
126 base_offset = dir_size;
129 IO::File out(fn, IO::M_WRITE);
130 out.write(&dir_buffer[0], base_offset);
131 tmp_file->seek(0, IO::S_BEG);
132 while(!tmp_file->eof())
135 unsigned len = tmp_file->read(buf, sizeof(buf));