From dacce2161e8c59ef0767dd8263470e44340e7782 Mon Sep 17 00:00:00 2001
From: Dan Olson <theoddone33@icculus.org>
Date: Tue, 5 Aug 2008 02:42:17 -0700
Subject: [PATCH] Just playing around... or am I?

---
 dumpwad.py          | 13 +++++++
 pyhexen/__init__.py |  0
 pyhexen/wad.py      | 89 +++++++++++++++++++++++++++++++++++++++++++++
 3 files changed, 102 insertions(+)
 create mode 100644 dumpwad.py
 create mode 100644 pyhexen/__init__.py
 create mode 100644 pyhexen/wad.py

diff --git a/dumpwad.py b/dumpwad.py
new file mode 100644
index 0000000..e3c829f
--- /dev/null
+++ b/dumpwad.py
@@ -0,0 +1,13 @@
+
+import sys
+import pyhexen.wad as wad
+
+myWad = wad.wad()
+
+print ("Loading wads...")
+myWad.InitMultipleFiles (sys.argv[1:])
+
+print ("Dumping wad contents...")
+
+for (handle, pos, len, name) in myWad.lumps:
+	print ("Found '%s' ('%s' offset %d, length %d)" % (name, handle.name, pos, len))
diff --git a/pyhexen/__init__.py b/pyhexen/__init__.py
new file mode 100644
index 0000000..e69de29
diff --git a/pyhexen/wad.py b/pyhexen/wad.py
new file mode 100644
index 0000000..89b3cc5
--- /dev/null
+++ b/pyhexen/wad.py
@@ -0,0 +1,89 @@
+
+import os
+import os.path
+from struct import *
+
+class WadException(Exception):
+	pass
+
+class wad:
+	wadinfo_t = '<4sii'
+	wadinfo_size = calcsize (wadinfo_t)
+	filelump_t = '<ii8s'
+	filelump_size = calcsize (filelump_t)
+
+	def __init__(self):
+		self.lumps = []
+
+	def __del__(self):
+		for (handle, pos, size, name) in self.lumps:
+			if not handle.closed:
+				handle.close()
+
+	def AddFile(self, filename):
+		(base, ext) = os.path.splitext (filename)
+		fileHandle = open (filename, 'rb')
+
+		lumpinfo = []
+		if ext.lower() == '.wad':
+			# yay, a wad file.
+			(id, numLumps, infoTable) = unpack (wad.wadinfo_t, fileHandle.read(wad.wadinfo_size))
+
+			if id != 'IWAD' and id != 'PWAD':
+				raise WadException("Wad file %s doesn't have IWAD or PWAD id" % filename)
+
+			fileHandle.seek (infoTable)
+
+			for i in range (0, numLumps):
+				lumpinfo.append (unpack (wad.filelump_t, fileHandle.read(wad.filelump_size)))
+		else:
+			# single file, boring.
+			size = os.stat (filename).st_size
+			lumpinfo.append ((0, size, base))
+
+		# Dump the info into our table of accumulated crap.
+		for (start, len, name) in lumpinfo:
+			self.lumps.append ((fileHandle, start, len, name))
+
+	def InitMultipleFiles (self, filenames):
+		for f in filenames:
+			self.AddFile (f)
+
+	def InitFile (self, filename):
+		self.AddFile (f)
+
+	def NumLumps (self):
+		return len (self.lumps)
+
+	def CheckNumForName (self, name):
+		# meh sucky in-place crappy way to do
+		# stuff :(
+		searchName = name.upper()[:8]
+		searchLumps = [x for (x, y, z, w) in self.lumps]
+		lumpIndex = -1
+		while True:
+			try:
+				lumpIndex = searchLumps.index(searchName)
+			except ValueError:
+				break
+		return lumpIndex
+
+	def GetNumForName (self, name):
+		return self.CheckNumForName (name)
+
+	def LumpLength (self, lump):
+		(handle, start, len, name) = self.lumps[lump]
+		return size
+
+	def ReadLump (self, lump, dest):
+		(handle, start, len, name) = self.lumps[lump]
+		handle.seek (start)
+		dest.write (handle.read (len))
+
+	def CacheLumpNum (self, lump, tag):
+		pass
+
+	def CacheLumpName (self, name, tag):
+		return self.CacheLumpNum (self.GetNumForName (name), tag)
+
+
-- 
2.39.2