BizHawk/BizHawk.MultiClient/RomGame.cs

using System;
using System.Collections.Generic;
using System.Globalization;

namespace BizHawk.MultiClient
{
    public class RomGame : IGame
    {
        public byte[] RomData;
		public byte[] FileData;
        public string System;

        private string name;
        private List<string> options;
        private const int BankSize = 4096;

        public RomGame(HawkFile file) : this(file, null){}

        public RomGame(HawkFile file, string patch)
        {
			if (!file.Exists)
			    throw new Exception("The file needs to exist, yo.");

            var stream = file.GetStream();

			FileData = Util.ReadAllBytes(stream);

			int header = (int)(stream.Length % BankSize);
			stream.Position = header;
			int length = (int)stream.Length - header;

			RomData = new byte[length];
			stream.Read(RomData, 0, length);

			if (file.Extension == "SMD")
				RomData = DeInterleaveSMD(RomData);

            var info = Database.GetGameInfo(RomData, file.Name);
            name = info.Name;
            System = info.System;
            options = new List<string>(info.GetOptions());
            CheckForPatchOptions();

			if (patch != null)
			{
				using (var patchFile = new HawkFile(patch))
				{
					patchFile.BindFirstOf("IPS");
					if(patchFile.IsBound)
						IPS.Patch(RomData, patchFile.GetStream());
				}    
			}
        }

        public void AddOptions(params string[] options)
        {
            this.options.AddRange(options);
        }

        private byte[] DeInterleaveSMD(byte[] source)
        {
            // SMD files are interleaved in pages of 16k, with the first 8k containing all 
            // odd bytes and the second 8k containing all even bytes.

            int size = source.Length;
            if (size > 0x400000) size = 0x400000;
            int pages = size / 0x4000;
            byte[] output = new byte[size];

            for (int page = 0; page < pages; page++)
            {
                for (int i = 0; i < 0x2000; i++)
                {
                    output[(page * 0x4000) + (i * 2) + 0] = source[(page * 0x4000) + 0x2000 + i];
                    output[(page * 0x4000) + (i * 2) + 1] = source[(page * 0x4000) + 0x0000 + i];
                }
            }
            return output;
        }

        private void CheckForPatchOptions()
        {
            try
            {
                foreach (var opt in options)
                {
                    if (opt.StartsWith("PatchBytes"))
                    {
                        var split1 = opt.Split('=');
                        foreach (var val in split1[1].Split(','))
                        {
                            var split3 = val.Split(':');
                            int offset = int.Parse(split3[0], NumberStyles.HexNumber);
                            byte value = byte.Parse(split3[1], NumberStyles.HexNumber);
                            RomData[offset] = value;
                        }
                    }
                }
            } catch (Exception) { } // No need for errors in patching to propagate.
        }

        public byte[] GetRomData() { return RomData; }
		public byte[] GetFileData() { return FileData; }
        public IList<string> GetOptions() { return options; }
		public string Name { get { return name; } set { name = value; } }

        public string SaveRamPath
        {
            get
            {
                switch (System)
                {
                    case "SMS": return "SMS/SaveRAM/" + Name + ".SaveRAM";
                    case "GG":  return "Game Gear/SaveRAM/" + Name + ".SaveRAM";
                    case "SG":  return "SG-1000/SaveRAM/" + Name + ".SaveRAM";
                    case "SGX": return "TurboGrafx/SaveRAM/" + Name + ".SaveRAM";
                    case "PCE": return "TurboGrafx/SaveRAM/" + Name + ".SaveRAM";
                    case "GB":  return "Gameboy/SaveRAM/" + Name + ".SaveRAM";
                    case "GEN": return "Genesis/SaveRAM/" + Name + ".SaveRAM";
                    case "NES": return "NES/SaveRAM/" + Name + ".SaveRAM";
                    case "TI83": return "TI83/SaveRAM/" + "TI83" + ".SaveRAM";
                    default:    return "";
                }
            }
        }

        public string SaveStatePrefix
        {
            get
            {
                switch (System)
                {
                    case "SMS": return "SMS/State/" + Name;
                    case "GG":  return "Game Gear/State/" + Name;
                    case "SG":  return "SG-1000/State/" + Name;
                    case "PCE": return "TurboGrafx/State/" + Name;
                    case "SGX": return "TurboGrafx/State/" + Name;
                    case "GB":  return "Gameboy/State/" + Name;
                    case "GEN": return "Genesis/State/" + Name;
                    case "NES": return "NES/State/" + Name;
                    case "TI83": return "TI83/State/" + "TI83";
                    default:    return "";
                }
                
            }
        }

        public string MoviePrefix
        {
            get
            {
                switch (System)
                {
                    case "SMS": return "SMS/Movie/" + Name;
                    case "GG":  return "Game Gear/Movie/" + Name;
                    case "SG":  return "SG-1000/Movie/" + Name;
                    case "PCE": return "TurboGrafx/Movie/" + Name;
                    case "SGX": return "TurboGrafx/Movie/" + Name;
                    case "GB":  return "Gameboy/Movie/" + Name;
                    case "GEN": return "Genesis/Movie/" + Name;
                    case "NES": return "NES/Movie/" + Name;
                    case "TI83": return "TI83/Movie/" + Name;
                    default:    return "";
                }
            }
        }

        public string ScreenshotPrefix
        {
            get
            {
                switch (System)
                {
                    case "SMS": return "SMS/Screenshot/" + Name;
                    case "GG":  return "Game Gear/Screenshot/" + Name;
                    case "SG":  return "SG-1000/Screenshot/" + Name;
                    case "PCE": return "TurboGrafx/Screenshot/" + Name;
                    case "SGX": return "TurboGrafx/Screenshot/" + Name;
                    case "GB":  return "Gameboy/Screenshot/" + Name;
                    case "GEN": return "Genesis/Screenshot/" + Name;
                    case "NES": return "NES/Screenshot/" + Name;
                    case "TI83": return "TI83/Screenshot/" + "TI83";
                    default:    return "";
                }
            }
        }
    }
}