#region ================== Copyright (c) 2007 Pascal vd Heiden /* * Copyright (c) 2007 Pascal vd Heiden, www.codeimp.com * This program is released under GNU General Public License * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * */ #endregion #region ================== Namespaces using System; using System.Collections; using System.Collections.Generic; using System.IO; using System.Text.RegularExpressions; using CodeImp.DoomBuilder.Compilers; using CodeImp.DoomBuilder.Config; using CodeImp.DoomBuilder.Data.Scripting; using CodeImp.DoomBuilder.GZBuilder.Data; using CodeImp.DoomBuilder.IO; using CodeImp.DoomBuilder.ZDoom; #endregion namespace CodeImp.DoomBuilder.Data { internal sealed class WADReader : DataReader { #region ================== Constants //mxd. TEXTUREx flags private const int TX_WORLDPANNING = 0x8000; //mxd. Sprite recognition. Also http://regexr.com/ is a nice site ^.^ private static readonly Regex sprite6 = new Regex(@"(\S{4}[A-Za-z\[\]\\]{1}[0-8]{1})"); private static readonly Regex sprite8 = new Regex(@"(\S{4}[A-Za-z\[\]\\]{1}[0-8]{1}[A-Za-z\[\]\\]{1}[0-8]{1})"); //mxd. Voxel recognition. private static readonly Regex voxel = new Regex(@"^\S{4}(([A-Za-z][0-9]){0,2}|[A-Za-z]{0,1})$"); #endregion #region ================== Structures private struct LumpRange { public int start; public int end; } #endregion #region ================== Variables // Source private WAD file; private bool is_iwad; private bool strictpatches; // Lump ranges private List flatranges; private List invertedflatranges; //mxd private List patchranges; private List spriteranges; private List textureranges; private List hiresranges; //mxd private List colormapranges; private List voxelranges; //mxd #endregion #region ================== Properties public bool IsIWAD { get { return is_iwad; } } internal WAD WadFile { get { return file; } } //mxd internal PK3StructuredReader ParentResource; //mxd #endregion #region ================== Constructor / Disposer // Constructor public WADReader(DataLocation dl, bool asreadonly) : base(dl, asreadonly) { General.WriteLogLine("Opening WAD resource \"" + location.location + "\""); if(!File.Exists(location.location)) throw new FileNotFoundException("Could not find the file \"" + location.location + "\"", location.location); // Initialize file = new WAD(location.location, asreadonly); strictpatches = dl.option1; Initialize(); //mxd // We have no destructor GC.SuppressFinalize(this); } //mxd. Constructor for temporary WAD files internal WADReader(DataLocation dl, bool asreadonly, bool create) : base(dl, asreadonly) { if(!create) { General.WriteLogLine("Opening WAD resource \"" + location.location + "\""); if(!File.Exists(location.location)) throw new FileNotFoundException("Could not find the file \"" + location.location + "\"", location.location); } // Initialize file = new WAD(location.location, asreadonly); strictpatches = dl.option1; Initialize(); // We have no destructor GC.SuppressFinalize(this); } //mxd private void Initialize() { is_iwad = file.IsIWAD; isreadonly |= is_iwad; // Just in case... // Initialize patchranges = new List(); spriteranges = new List(); flatranges = new List(); textureranges = new List(); hiresranges = new List(); //mxd colormapranges = new List(); voxelranges = new List(); //mxd // Find ranges FindRanges(patchranges, General.Map.Config.PatchRanges, "patches", "Patch"); FindRanges(spriteranges, General.Map.Config.SpriteRanges, "sprites", "Sprite"); FindRanges(flatranges, General.Map.Config.FlatRanges, "flats", "Flat"); FindRanges(textureranges, General.Map.Config.TextureRanges, "textures", "Texture"); FindRanges(hiresranges, General.Map.Config.HiResRanges, "hires", "HiRes texture"); FindRanges(colormapranges, General.Map.Config.ColormapRanges, "colormaps", "Colormap"); FindRanges(voxelranges, General.Map.Config.VoxelRanges, "voxels", "Voxel"); //mxd invertedflatranges = new List(); if(flatranges.Count > 0) { //add range before the first flatrange if(flatranges[0].start > 0) { LumpRange range = new LumpRange {start = 0, end = flatranges[0].start - 1}; invertedflatranges.Add(range); } //add ranges between flatranges for(int i = 1; i < flatranges.Count; i++) { LumpRange range = new LumpRange { start = flatranges[i - 1].end + 1, end = flatranges[i].start - 1 }; invertedflatranges.Add(range); } //add range after the last flatrange if(flatranges[flatranges.Count - 1].end < file.Lumps.Count - 1) { LumpRange range = new LumpRange { start = flatranges[flatranges.Count - 1].end + 1, end = file.Lumps.Count - 1 }; invertedflatranges.Add(range); } } else // No flat ranges? Make one giant range then... { LumpRange range = new LumpRange {start = 0, end = file.Lumps.Count - 1}; invertedflatranges.Add(range); } } // Disposer public override void Dispose() { // Not already disposed? if(!isdisposed) { General.WriteLogLine("Closing WAD resource \"" + location.location + "\""); // Clean up file.Dispose(); // Done base.Dispose(); } } #endregion #region ================== Management // Return a short name for this data location public override string GetTitle() { return Path.GetFileName(location.location); } // This suspends use of this resource public override void Suspend() { file.Dispose(); base.Suspend(); } // This resumes use of this resource public override void Resume() { file = new WAD(location.location, true); is_iwad = file.IsIWAD; base.Resume(); } // This fills a ranges list private void FindRanges(List ranges, IDictionary rangeinfos, string rangename, string elementname) { Dictionary> failedranges = new Dictionary>(); //mxd Dictionary successfulrangestarts = new Dictionary(); //mxd Dictionary failedrangestarts = new Dictionary(); //mxd foreach(DictionaryEntry r in rangeinfos) { // Read start and end string rangestart = General.Map.Config.ReadSetting(rangename + "." + r.Key + ".start", ""); string rangeend = General.Map.Config.ReadSetting(rangename + "." + r.Key + ".end", ""); if((rangestart.Length > 0) && (rangeend.Length > 0)) { // Find ranges int startindex = file.FindLumpIndex(rangestart); while(startindex > -1) { LumpRange range = new LumpRange { start = startindex, end = file.FindLumpIndex(rangeend, startindex) }; if(range.end > -1) { if(!successfulrangestarts.ContainsKey(startindex)) successfulrangestarts.Add(startindex, false); //mxd ranges.Add(range); startindex = file.FindLumpIndex(rangestart, range.end); } else { //mxd if(!failedrangestarts.ContainsKey(startindex)) { failedranges.Add(range, new KeyValuePair(rangestart, rangeend)); //mxd failedrangestarts.Add(startindex, false); } startindex = -1; } } } } // Don't check official IWADs if(!file.IsOfficialIWAD) { //mxd. Display warnings for unclosed ranges foreach(KeyValuePair> group in failedranges) { if(successfulrangestarts.ContainsKey(group.Key.start)) continue; General.ErrorLogger.Add(ErrorType.Warning, "\"" + group.Value.Key + "\" range at index " + group.Key.start + " is not closed in resource \"" + location.GetDisplayName() + "\" (\"" + group.Value.Value + "\" marker is missing)."); } //mxd. Check duplicates foreach(LumpRange range in ranges) { HashSet names = new HashSet(StringComparer.OrdinalIgnoreCase); for(int i = range.start + 1; i < range.end; i++) { if(names.Contains(file.Lumps[i].Name)) General.ErrorLogger.Add(ErrorType.Warning, elementname + " \"" + file.Lumps[i].Name + "\", index " + i + " is double defined in resource \"" + location.GetDisplayName() + "\"."); else names.Add(file.Lumps[i].Name); } } } } #endregion #region ================== Palette // This loads the PLAYPAL palette public override Playpal LoadPalette() { // Error when suspended if(issuspended) throw new Exception("Data reader is suspended"); // Look for a lump named PLAYPAL Lump lump = file.FindLump("PLAYPAL"); if(lump != null) return new Playpal(lump.Stream); // Read the PLAYPAL from stream return null; // No palette } #endregion #region ================== Colormaps // This loads the textures public override ICollection LoadColormaps() { List images = new List(); // Error when suspended if(issuspended) throw new Exception("Data reader is suspended"); // Read ranges from configuration foreach(DictionaryEntry r in General.Map.Config.ColormapRanges) { // Read start and end string rangestart = General.Map.Config.ReadSetting("colormaps." + r.Key + ".start", ""); string rangeend = General.Map.Config.ReadSetting("colormaps." + r.Key + ".end", ""); if((rangestart.Length > 0) && (rangeend.Length > 0)) { // Load texture range LoadColormapsRange(rangestart, rangeend, ref images); } } // Add images to the container-specific texture set foreach(ImageData img in images) textureset.AddFlat(img); // Return result return images; } // This loads a range of colormaps private void LoadColormapsRange(string startlump, string endlump, ref List images) { // Continue until no more start can be found int startindex = file.FindLumpIndex(startlump); while(startindex > -1) { // Find end index int endindex = file.FindLumpIndex(endlump, startindex + 1); if(endindex > -1) { // Go for all lumps between start and end exclusive for(int i = startindex + 1; i < endindex; i++) { // Lump not zero-length? if(file.Lumps[i].Length > 0) { // Make the image object ColormapImage image = new ColormapImage(file.Lumps[i].Name); // Add image to collection images.Add(image); } } } // Find the next start startindex = file.FindLumpIndex(startlump, startindex + 1); } } // This finds and returns a colormap stream public override Stream GetColormapData(string pname) { // Error when suspended if(issuspended) throw new Exception("Data reader is suspended"); // Strictly read patches only between C_START and C_END? if(strictpatches) { // Find the lump in ranges foreach(LumpRange range in colormapranges) { Lump lump = file.FindLump(pname, range.start, range.end); if(lump != null) return lump.Stream; } } else { // Find the lump anywhere Lump lump = file.FindLump(pname); if(lump != null) return lump.Stream; } return null; } #endregion #region ================== Textures // This loads the textures public override IEnumerable LoadTextures(PatchNames pnames, Dictionary cachedparsers) { // Error when suspended if(issuspended) throw new Exception("Data reader is suspended"); List images = new List(); float defaultscale = General.Map.Config.DefaultTextureScale; //mxd // Load two sets of textures, if available Lump lump = file.FindLump("TEXTURE1"); if(lump != null) { LoadTextureSet("TEXTURE1", lump.Stream, ref images, pnames); if(images.Count > 0) images.RemoveAt(0); //mxd. The first TEXTURE1 texture cannot be used. Let's remove it here } lump = file.FindLump("TEXTURE2"); if(lump != null) LoadTextureSet("TEXTURE2", lump.Stream, ref images, pnames); // Read ranges from configuration foreach(LumpRange range in textureranges) { // Go for all lumps between start and end exclusive for(int i = range.start + 1; i < range.end; i++) { // Lump not zero length? if(file.Lumps[i].Length > 0) { // Make the image SimpleTextureImage image = new SimpleTextureImage(file.Lumps[i].Name, file.Lumps[i].Name, defaultscale, defaultscale); // Add image to collection images.Add(image); } else { // Can't load image without size General.ErrorLogger.Add(ErrorType.Error, "Can't load texture \"" + file.Lumps[i].Name + "\" from \"" + location.GetDisplayName() + "\" because it doesn't contain any data."); } } } // Load TEXTURES lump file int lumpindex = file.FindLumpIndex("TEXTURES"); while(lumpindex > -1) { //mxd. Added TexturesParser caching string fullpath = Path.Combine(this.location.location, "TEXTURES#" + lumpindex); if(cachedparsers.ContainsKey(fullpath)) { // Make the textures foreach(TextureStructure t in cachedparsers[fullpath].Textures) images.Add(t.MakeImage()); } else { MemoryStream filedata = new MemoryStream(file.Lumps[lumpindex].Stream.ReadAllBytes()); cachedparsers.Add(fullpath, LoadTEXTURESTextures(new TextResourceData(this, filedata, "TEXTURES", lumpindex, true), ref images)); //mxd filedata.Dispose(); } // Find next lumpindex = file.FindLumpIndex("TEXTURES", lumpindex + 1); } // Add images to the container-specific texture set foreach(ImageData img in images) textureset.AddTexture(img); // Return result return images; } //mxd. This loads the HiRes textures public override IEnumerable LoadHiResTextures() { List images = new List(); // Read ranges from configuration foreach(LumpRange range in hiresranges) { // Go for all lumps between start and end exclusive for(int i = range.start + 1; i < range.end; i++) { // Lump not zero length? if(file.Lumps[i].Length > 0) { // Add image to collection images.Add(new HiResImage(file.Lumps[i].Name)); } else { // Can't load image without size General.ErrorLogger.Add(ErrorType.Error, "Can't load HiRes texture \"" + file.Lumps[i].Name + "\" from \"" + location.GetDisplayName() + "\" because it doesn't contain any data."); } } } return images; } // This loads the texture definitions from a TEXTURES lump public static TexturesParser LoadTEXTURESTextures(TextResourceData data, ref List images) { // Parse the data TexturesParser parser = new TexturesParser(); parser.Parse(data, false); if(parser.HasError) parser.LogError(); //mxd // Make the textures foreach(TextureStructure t in parser.Textures) { // Add the texture images.Add(t.MakeImage()); } //mxd. Add to text resources collection if(!General.Map.Data.ScriptResources.ContainsKey(parser.ScriptType)) General.Map.Data.ScriptResources[parser.ScriptType] = new HashSet(); foreach(KeyValuePair group in parser.ScriptResources) General.Map.Data.ScriptResources[parser.ScriptType].Add(group.Value); return parser; //mxd } // This loads a set of textures public static void LoadTextureSet(string sourcename, Stream texturedata, ref List images, PatchNames pnames) { if(texturedata.Length == 0) return; BinaryReader reader = new BinaryReader(texturedata); // Determine default scale float defaultscale = General.Map.Config.DefaultTextureScale; // Get number of textures texturedata.Seek(0, SeekOrigin.Begin); uint numtextures = reader.ReadUInt32(); // Skip offset bytes (we will read all textures sequentially) texturedata.Seek(4 * numtextures, SeekOrigin.Current); // Go for all textures defined in this lump for(uint i = 0; i < numtextures; i++) { // Read texture properties byte[] namebytes = reader.ReadBytes(8); int flags = reader.ReadUInt16(); bool worldpanning = (flags & TX_WORLDPANNING) != 0; //mxd byte scalebytex = reader.ReadByte(); byte scalebytey = reader.ReadByte(); int width = reader.ReadInt16(); int height = reader.ReadInt16(); int patches = reader.ReadInt16(); // Check for doom or strife data format bool strifedata; if(patches == 0) { // Ignore 2 bytes and then read number of patches texturedata.Seek(2, SeekOrigin.Current); patches = reader.ReadInt16(); strifedata = false; } else { // Texture data is in strife format strifedata = true; } // Determine actual scales float scalex = (scalebytex == 0 ? defaultscale : 1f / (scalebytex / 8f)); float scaley = (scalebytey == 0 ? defaultscale : 1f / (scalebytey / 8f)); // Validate data if((width > 0) && (height > 0) && (patches > 0) && (scalex != 0) || (scaley != 0)) { string texname = Lump.MakeNormalName(namebytes, WAD.ENCODING); TextureImage image = null; if(texname.Length > 0) { // Make the image object image = new TextureImage(sourcename, Lump.MakeNormalName(namebytes, WAD.ENCODING), width, height, scalex, scaley, worldpanning); } else { // Can't load image without name General.ErrorLogger.Add(ErrorType.Error, "Can't load an unnamed texture from \"" + sourcename + "\". Please consider giving names to your resources."); } // Go for all patches in texture for(int p = 0; p < patches; p++) { // Read patch properties int px = reader.ReadInt16(); int py = reader.ReadInt16(); int pi = reader.ReadUInt16(); if(!strifedata) texturedata.Seek(4, SeekOrigin.Current); // Validate data if((pi >= 0) && (pi < pnames.Length)) { if(pnames[pi].Length > 0) { // Create patch on image if(image != null) image.AddPatch(new TexturePatch(pnames[pi], px, py)); } else { // Can't load image without name General.ErrorLogger.Add(ErrorType.Error, "Can't use an unnamed patch referenced in \"" + sourcename + "\". Please consider giving names to your resources."); } } } // Add image to collection if(image != null) images.Add(image); } else { // Skip patches data texturedata.Seek(6 * patches, SeekOrigin.Current); if(!strifedata) texturedata.Seek(4 * patches, SeekOrigin.Current); } } } // This returns the patch names from the PNAMES lump public override PatchNames LoadPatchNames() { // Error when suspended if(issuspended) throw new Exception("Data reader is suspended"); // Look for a lump named PNAMES Lump lump = file.FindLump("PNAMES"); if(lump != null) return new PatchNames(lump.Stream); // Read the PNAMES from stream return null; // No patch names found } // This finds and returns a patch stream public override Stream GetPatchData(string pname, bool longname, ref string patchlocation) { // Error when suspended if(issuspended) throw new Exception("Data reader is suspended"); if(longname) return null; //mxd Lump lump; // mxd. First strictly read patches between P_START and P_END foreach(LumpRange range in patchranges) { lump = file.FindLump(pname, range.start, range.end); if(lump != null) return lump.Stream; } if(!strictpatches) { //mxd. Find the lump anywhere EXCEPT flat ranges (the way it's done in ZDoom) foreach(LumpRange range in invertedflatranges) { lump = file.FindLump(pname, range.start, range.end); if(lump != null) { patchlocation = location.GetDisplayName(); return lump.Stream; } } // Find the lump anywhere IN flat ranges foreach(LumpRange range in flatranges) { lump = file.FindLump(pname, range.start, range.end); if(lump != null) { patchlocation = location.GetDisplayName(); return lump.Stream; } } } return null; } // This finds and returns a texture stream public override Stream GetTextureData(string pname, bool longname, ref string texturelocation) { // Error when suspended if(issuspended) throw new Exception("Data reader is suspended"); if(longname) return null; //mxd // Find the lump in ranges foreach(LumpRange range in textureranges) { Lump lump = file.FindLump(pname, range.start, range.end); if(lump != null) { texturelocation = location.GetDisplayName(); //mxd return lump.Stream; } } return null; } //mxd. This finds and returns a HiRes texture stream public override Stream GetHiResTextureData(string name, ref string hireslocation) { // Error when suspended if(issuspended) throw new Exception("Data reader is suspended"); // Find the lump in ranges foreach(LumpRange range in hiresranges) { Lump lump = file.FindLump(name, range.start, range.end); if(lump != null) { hireslocation = location.GetDisplayName(); return lump.Stream; } } return null; } #endregion #region ================== Flats //mxd. This loads the flats public override IEnumerable LoadFlats(Dictionary cachedparsers) { // Error when suspended if(issuspended) throw new Exception("Data reader is suspended"); List images = new List(); foreach(LumpRange range in flatranges) { if(range.end < range.start + 2) continue; for(int i = range.start + 1; i < range.end; i++) { // Lump not zero-length? if(file.Lumps[i].Length > 0) { // Make the image object FlatImage image = new FlatImage(file.Lumps[i].Name); // Add image to collection images.Add(image); } } } // Load TEXTURES lump file int lumpindex = file.FindLumpIndex("TEXTURES"); while(lumpindex > -1) { //mxd. Added TexturesParser caching string fullpath = Path.Combine(this.location.location, "TEXTURES#" + lumpindex); if(cachedparsers.ContainsKey(fullpath)) { // Make the textures foreach(TextureStructure t in cachedparsers[fullpath].Flats) images.Add(t.MakeImage()); } else { MemoryStream filedata = new MemoryStream(file.Lumps[lumpindex].Stream.ReadAllBytes()); cachedparsers.Add(fullpath, LoadTEXTURESFlats(new TextResourceData(this, filedata, "TEXTURES", lumpindex, true), ref images)); //mxd filedata.Dispose(); } // Find next lumpindex = file.FindLumpIndex("TEXTURES", lumpindex + 1); } // Add images to the container-specific texture set foreach(ImageData img in images) textureset.AddFlat(img); // Return result return images; } // This loads the flat definitions from a TEXTURES lump public static TexturesParser LoadTEXTURESFlats(TextResourceData data, ref List images) { // Parse the data TexturesParser parser = new TexturesParser(); parser.Parse(data, false); if(parser.HasError) parser.LogError(); //mxd // Make the flat foreach(TextureStructure t in parser.Flats) { // Add the flat images.Add(t.MakeImage()); } //mxd. Add to text resources collection if(!General.Map.Data.ScriptResources.ContainsKey(parser.ScriptType)) General.Map.Data.ScriptResources[parser.ScriptType] = new HashSet(); foreach(KeyValuePair group in parser.ScriptResources) General.Map.Data.ScriptResources[parser.ScriptType].Add(group.Value); return parser; //mxd } // This finds and returns a patch stream public override Stream GetFlatData(string pname, bool longname, ref string flatlocation) { // Error when suspended if(issuspended) throw new Exception("Data reader is suspended"); if(longname) return null; //mxd // Find the lump in ranges foreach(LumpRange range in flatranges) { Lump lump = file.FindLump(pname, range.start, range.end); if(lump != null) { flatlocation = location.GetDisplayName(); //mxd return lump.Stream; } } return null; } #endregion #region ================== Sprites // This loads the textures public override IEnumerable LoadSprites(Dictionary cachedparsers) { // Error when suspended if(issuspended) throw new Exception("Data reader is suspended"); List images = new List(); // Load TEXTURES lump file int lumpindex = file.FindLumpIndex("TEXTURES"); while(lumpindex > -1) { //mxd. Added TexturesParser caching string fullpath = Path.Combine(this.location.location, "TEXTURES#" + lumpindex); if(cachedparsers.ContainsKey(fullpath)) { // Make the textures foreach(TextureStructure t in cachedparsers[fullpath].Sprites) images.Add(t.MakeImage()); } else { MemoryStream filedata = new MemoryStream(file.Lumps[lumpindex].Stream.ReadAllBytes()); cachedparsers.Add(fullpath, LoadTEXTURESSprites(new TextResourceData(this, filedata, "TEXTURES", lumpindex, true), ref images)); //mxd filedata.Dispose(); } // Find next lumpindex = file.FindLumpIndex("TEXTURES", lumpindex + 1); } // Return result return images; } // This loads the sprites definitions from a TEXTURES lump public static TexturesParser LoadTEXTURESSprites(TextResourceData data, ref List images) { // Parse the data TexturesParser parser = new TexturesParser(); parser.Parse(data, false); if(parser.HasError) parser.LogError(); //mxd // Make the sprites foreach(TextureStructure t in parser.Sprites) { // Add the sprite images.Add(t.MakeImage()); } //mxd. Add to text resources collection if(!General.Map.Data.ScriptResources.ContainsKey(parser.ScriptType)) General.Map.Data.ScriptResources[parser.ScriptType] = new HashSet(); foreach(KeyValuePair group in parser.ScriptResources) General.Map.Data.ScriptResources[parser.ScriptType].Add(group.Value); return parser; //mxd } // This finds and returns a sprite stream public override Stream GetSpriteData(string pname, ref string spritelocation) { // Error when suspended if(issuspended) throw new Exception("Data reader is suspended"); // Find the lump in ranges foreach(LumpRange range in spriteranges) { Lump lump = file.FindLump(pname, range.start, range.end); if(lump != null) { spritelocation = location.GetDisplayName(); //mxd return lump.Stream; } } return null; } // This checks if the given sprite exists public override bool GetSpriteExists(string pname) { // Error when suspended if(issuspended) throw new Exception("Data reader is suspended"); // Find the lump in ranges foreach(LumpRange range in spriteranges) { Lump lump = file.FindLump(pname, range.start, range.end); if(lump != null) return true; } return false; } //mxd. This returns all sprite names in the WAD public override HashSet GetSpriteNames() { // Error when suspended if(issuspended) throw new Exception("Data reader is suspended"); HashSet result = new HashSet(); foreach(LumpRange range in spriteranges) { for(int i = range.start; i < range.end + 1; i++) if(IsValidSpriteName(file.Lumps[i].Name)) result.Add(file.Lumps[i].Name); } return result; } //mxd internal static bool IsValidSpriteName(string name) { return (name.Length == 6 && sprite6.IsMatch(name)) || (name.Length == 8 && sprite8.IsMatch(name)); } #endregion #region ================== Voxels (mxd) //mxd. This returns the list of voxels, which can be used without VOXELDEF definition public override HashSet GetVoxelNames() { // Error when suspended if(issuspended) throw new Exception("Data reader is suspended"); HashSet result = new HashSet(); foreach(LumpRange range in voxelranges) { if(range.start == range.end) continue; for(int i = range.start + 1; i < range.end; i++) { if(IsValidVoxelName(file.Lumps[i].Name)) result.Add(file.Lumps[i].Name); } } return result; } //mxd public override IEnumerable GetVoxeldefData() { if(issuspended) throw new Exception("Data reader is suspended"); return GetAllLumps("VOXELDEF"); } //mxd. This finds and returns a voxel stream or null if no voxel was found public override Stream GetVoxelData(string name, ref string voxellocation) { // Error when suspended if(issuspended) throw new Exception("Data reader is suspended"); foreach(LumpRange range in voxelranges) { if(range.start == range.end) continue; Lump lump = file.FindLump(name, range.start, range.end); if(lump != null) { voxellocation = location.GetDisplayName(); // Copy stream, because model/voxel streams are expected to be disposed lump.Stream.Seek(0, SeekOrigin.Begin); // Rewind before use return new MemoryStream(lump.Stream.ReadAllBytes()); } } return null; } //mxd internal static bool IsValidVoxelName(string name) { return (name.Length > 3 && name.Length < 7) && voxel.IsMatch(name); } #endregion #region ================== Decorate, Gldefs, Mapinfo, etc... // This finds and returns DECORATE streams public override IEnumerable GetDecorateData(string pname) { if(issuspended) throw new Exception("Data reader is suspended"); List result = GetAllLumps(pname); //mxd //mxd. Return ALL DECORATE lumps if(result.Count == 0 || string.Compare(pname, "DECORATE", StringComparison.OrdinalIgnoreCase) == 0) return result; //mxd. Return THE LAST include lump, because that's the way ZDoom seems to operate return new List {result[result.Count - 1]}; } //mxd. Should be only one entry per wad public override IEnumerable GetMapinfoData() { if(issuspended) throw new Exception("Data reader is suspended"); // First look for ZMAPINFO List result = new List(); result.AddRange(GetLastLump("ZMAPINFO")); // Then for MAPINFO if(result.Count == 0) result.AddRange(GetLastLump("MAPINFO")); return result; } //mxd public override IEnumerable GetGldefsData(string basegame) { if(issuspended) throw new Exception("Data reader is suspended"); List result = new List(); // Try to load game specific GLDEFS first if(basegame != GameType.UNKNOWN) { string lumpname = GameType.GldefsLumpsPerGame[basegame]; result.AddRange(GetAllLumps(lumpname)); } // Can be many entries per wad result.AddRange(GetAllLumps("GLDEFS")); return result; } //mxd public override IEnumerable GetModeldefData() { if(issuspended) throw new Exception("Data reader is suspended"); return GetAllLumps("MODELDEF"); } //mxd public override IEnumerable GetReverbsData() { if(issuspended) throw new Exception("Data reader is suspended"); return GetAllLumps("REVERBS"); } //mxd public override IEnumerable GetSndInfoData() { if(issuspended) throw new Exception("Data reader is suspended"); return GetAllLumps("SNDINFO"); } //mxd public override IEnumerable GetSndSeqData() { if(issuspended) throw new Exception("Data reader is suspended"); return GetAllLumps("SNDSEQ"); } //mxd public override IEnumerable GetAnimdefsData() { if(issuspended) throw new Exception("Data reader is suspended"); return GetAllLumps("ANIMDEFS"); } //mxd public override IEnumerable GetTerrainData() { if(issuspended) throw new Exception("Data reader is suspended"); return GetAllLumps("TERRAIN"); } //mxd public override IEnumerable GetX11R6RGBData() { if(issuspended) throw new Exception("Data reader is suspended"); return GetAllLumps("X11R6RGB"); } //mxd public override IEnumerable GetCvarInfoData() { if(issuspended) throw new Exception("Data reader is suspended"); return GetAllLumps("CVARINFO"); } //mxd public override IEnumerable GetLockDefsData() { if(issuspended) throw new Exception("Data reader is suspended"); return GetAllLumps("LOCKDEFS"); } //mxd public override IEnumerable GetMenuDefData() { if(issuspended) throw new Exception("Data reader is suspended"); return GetLastLump("MENUDEF"); } //mxd public override IEnumerable GetSBarInfoData() { if(issuspended) throw new Exception("Data reader is suspended"); return GetLastLump("SBARINFO"); } //mxd private IEnumerable GetLastLump(string name) { List result = new List(); int lumpindex = file.FindLumpIndex(name); if(lumpindex != -1) result.Add(new TextResourceData(this, file.Lumps[lumpindex].Stream, name, lumpindex, true)); return result; } //mxd private List GetAllLumps(string name) { List result = new List(); // Find all lumps with given name int lumpindex = file.FindLumpIndex(name); while(lumpindex > -1) { // Add to collection result.Add(new TextResourceData(this, file.Lumps[lumpindex].Stream, name, lumpindex, true)); // Find next entry lumpindex = file.FindLumpIndex(name, lumpindex + 1); } return result; } #endregion #region ================== IO (mxd) internal override MemoryStream LoadFile(string name) { Lump l = file.FindLump(name); if(l != null) { l.Stream.Seek(0, SeekOrigin.Begin); return new MemoryStream(l.Stream.ReadAllBytes()); } return null; } internal override MemoryStream LoadFile(string name, int lumpindex) { if(lumpindex < 0 || file.Lumps.Count <= lumpindex || file.Lumps[lumpindex].Name != name.ToUpperInvariant()) return null; Lump l = file.Lumps[lumpindex]; l.Stream.Seek(0, SeekOrigin.Begin); return new MemoryStream(l.Stream.ReadAllBytes()); } internal override bool SaveFile(MemoryStream lumpdata, string lumpname) { if(isreadonly) return false; int insertindex = file.Lumps.Count; // Remove the lump if it already exists int li = file.FindLumpIndex(lumpname); if(li > -1) { insertindex = li; file.RemoveAt(li); } // Insert new lump Lump l = file.Insert(lumpname, insertindex, (int)lumpdata.Length); l.Stream.Seek(0, SeekOrigin.Begin); lumpdata.WriteTo(l.Stream); // Update WAD file file.WriteHeaders(); return true; } internal override bool SaveFile(MemoryStream lumpdata, string lumpname, int lumpindex) { if(isreadonly || lumpindex < 0 || file.Lumps.Count <= lumpindex || file.Lumps[lumpindex].Name != lumpname.ToUpperInvariant()) return false; // Remove the lump file.RemoveAt(lumpindex); // Insert new lump Lump l = file.Insert(lumpname, lumpindex, (int)lumpdata.Length); l.Stream.Seek(0, SeekOrigin.Begin); lumpdata.WriteTo(l.Stream); // Update WAD file file.WriteHeaders(); return true; } internal override bool FileExists(string lumpname) { return file.FindLumpIndex(lumpname) != -1; } internal override bool FileExists(string lumpname, int lumpindex) { return file.FindLumpIndex(lumpname) == lumpindex; } #endregion #region ================== Compiling (mxd) // This compiles a script lump and returns any errors that may have occurred // Returns true when our code worked properly (even when the compiler returned errors) internal override bool CompileLump(string lumpname, ScriptConfiguration scriptconfig, List errors) { int index = file.FindLumpIndex(lumpname); if(index == -1) { errors.Add(new CompilerError { description = "Lump \"" + lumpname + "\" does not exist", filename = this.location.GetDisplayName() }); return false; } return CompileLump(lumpname, index, scriptconfig, errors); } // This compiles a script lump and returns any errors that may have occurred // Returns true when our code worked properly (even when the compiler returned errors) internal override bool CompileLump(string lumpname, int lumpindex, ScriptConfiguration scriptconfig, List errors) { // No compiling required if(scriptconfig.Compiler == null) return true; string inputfile; Compiler compiler; string reallumpname = lumpname; // Find the lump if(lumpname == MapManager.CONFIG_MAP_HEADER) reallumpname = MapManager.TEMP_MAP_HEADER; Lump lump = file.FindLump(reallumpname, lumpindex, lumpindex); if(lump == null) throw new Exception("Unable to find lump \"" + reallumpname + "\" to compile in \"" + location.GetDisplayName() + "\"."); // Determine source file string sourcefile = (General.Map.FilePathName.Length > 0 ? General.Map.FilePathName : file.Filename); // Initialize compiler try { compiler = scriptconfig.Compiler.Create(); } catch(Exception e) { // Fail errors.Add(new CompilerError("Unable to initialize compiler. " + e.GetType().Name + ": " + e.Message)); return false; } //mxd. AccCompiler requires some additional settings... if(scriptconfig.ScriptType == ScriptType.ACS) { AccCompiler acccompiler = compiler as AccCompiler; if(acccompiler == null) { // Fail errors.Add(new CompilerError("Unexpected ACS compiler: " + compiler)); return false; } if(lumpname == "SCRIPTS" && this == General.Map.TemporaryMapFile) { acccompiler.SourceIsMapScriptsLump = true; } else { //TODO: implement library compiling... errors.Add(new CompilerError("Compilation of ACS libraries is not supported yet...")); return false; } } try { // Write lump data to temp script file in compiler's temp directory inputfile = General.MakeTempFilename(compiler.Location, "tmp"); lump.Stream.Seek(0, SeekOrigin.Begin); BinaryReader reader = new BinaryReader(lump.Stream); File.WriteAllBytes(inputfile, reader.ReadBytes((int)lump.Stream.Length)); } catch(Exception e) { // Fail compiler.Dispose(); errors.Add(new CompilerError("Unable to write script to working file. " + e.GetType().Name + ": " + e.Message)); return false; } // Make random output filename string outputfile = General.MakeTempFilename(compiler.Location, "tmp"); // Run compiler compiler.Parameters = scriptconfig.Parameters; compiler.InputFile = Path.GetFileName(inputfile); compiler.OutputFile = Path.GetFileName(outputfile); compiler.SourceFile = sourcefile; compiler.WorkingDirectory = Path.GetDirectoryName(inputfile); if(compiler.Run()) { // Process errors foreach(CompilerError e in compiler.Errors) { CompilerError newerror = e; // If the error's filename equals our temporary file, use the lump name instead and prefix it with ? if(string.Compare(e.filename, inputfile, true) == 0) newerror.filename = "?" + reallumpname; errors.Add(newerror); } // No errors? if(compiler.Errors.Length == 0) { // Output file exists? if(File.Exists(outputfile)) { // Copy output file data into a lump? if(!string.IsNullOrEmpty(scriptconfig.ResultLump)) { // Do that now then byte[] filedata; try { filedata = File.ReadAllBytes(outputfile); } catch(Exception e) { // Fail compiler.Dispose(); errors.Add(new CompilerError("Unable to read compiler output file. " + e.GetType().Name + ": " + e.Message)); return false; } // Store data SaveFile(new MemoryStream(filedata), scriptconfig.ResultLump); } } } // Clean up compiler.Dispose(); // Done return true; } // Fail compiler.Dispose(); return false; } #endregion } }