aboutsummaryrefslogtreecommitdiff
path: root/contrib/extractor/dbcfile.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'contrib/extractor/dbcfile.cpp')
-rw-r--r--contrib/extractor/dbcfile.cpp69
1 files changed, 69 insertions, 0 deletions
diff --git a/contrib/extractor/dbcfile.cpp b/contrib/extractor/dbcfile.cpp
new file mode 100644
index 00000000000..acfb35482ff
--- /dev/null
+++ b/contrib/extractor/dbcfile.cpp
@@ -0,0 +1,69 @@
+#define _CRT_SECURE_NO_DEPRECATE
+
+#include "dbcfile.h"
+#include "mpq_libmpq.h"
+
+DBCFile::DBCFile(const std::string &filename):
+ filename(filename),
+ data(0)
+{
+
+}
+void DBCFile::open()
+{
+ MPQFile f(filename.c_str());
+ char header[4];
+ unsigned int na,nb,es,ss;
+
+ f.read(header,4); // Number of records
+ assert(header[0]=='W' && header[1]=='D' && header[2]=='B' && header[3] == 'C');
+ f.read(&na,4); // Number of records
+ f.read(&nb,4); // Number of fields
+ f.read(&es,4); // Size of a record
+ f.read(&ss,4); // String size
+
+ recordSize = es;
+ recordCount = na;
+ fieldCount = nb;
+ stringSize = ss;
+ assert(fieldCount*4 == recordSize);
+
+ data = new unsigned char[recordSize*recordCount+stringSize];
+ stringTable = data + recordSize*recordCount;
+ f.read(data,recordSize*recordCount+stringSize);
+ f.close();
+}
+DBCFile::~DBCFile()
+{
+ delete [] data;
+}
+
+DBCFile::Record DBCFile::getRecord(size_t id)
+{
+ assert(data);
+ return Record(*this, data + id*recordSize);
+}
+
+size_t DBCFile::getMaxId()
+{
+ assert(data);
+
+ size_t maxId = 0;
+ for(size_t i = 0; i < getRecordCount(); ++i)
+ {
+ if(maxId < getRecord(i).getUInt(0))
+ maxId = getRecord(i).getUInt(0);
+ }
+ return maxId;
+}
+
+DBCFile::Iterator DBCFile::begin()
+{
+ assert(data);
+ return Iterator(*this, data);
+}
+DBCFile::Iterator DBCFile::end()
+{
+ assert(data);
+ return Iterator(*this, stringTable);
+}