/* * Copyright (c) 2010, Tess Snider Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions: The above copyright notice and this permission notice shall be included in all copies or substantial portions of the Software. THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */ using System; using System.IO; using System.Collections.Generic; using System.Text.RegularExpressions; using Gtk; using DesertPaintLab; public partial class MainWindow : Gtk.Window { const string APP_VERSION = "1.7.11"; bool unsavedData = false; bool shouldShutDown = false; List profileList = new List(); PlayerProfile profile = null; PaintColor expectedColor = new PaintColor(); PaintColor reactedColor = new PaintColor(); Gdk.Window rootWindow = null; Gdk.Pixbuf screenBuffer = null; Reagent[] reagents = new Reagent[3]; PaintRecipe recipe = new PaintRecipe(); public bool ShouldShutDown { get { return shouldShutDown; } } public MainWindow () : base(Gtk.WindowType.Toplevel) { string appDataPath = FileUtils.AppDataPath; if (!System.IO.Directory.Exists(appDataPath)) { System.IO.Directory.CreateDirectory(appDataPath); } DirectoryInfo di = new DirectoryInfo(appDataPath); DirectoryInfo[] dirs = di.GetDirectories(); foreach (DirectoryInfo dir in dirs) { if (dir.Name != "template") { profileList.Add(dir.Name); } } reagents[0] = null; reagents[1] = null; reagents[2] = null; string colorsPath = FileUtils.FindApplicationResourceFile("colors.txt"); if (colorsPath == null) { // failed to find colors.txt file MessageDialog md = new MessageDialog(this, DialogFlags.DestroyWithParent, MessageType.Error, ButtonsType.Close, "Failed to find colors.txt file. Please check your installation."); md.Run(); md.Destroy(); Application.Quit(); } Palette.Load(colorsPath); string ingredientsPath = FileUtils.FindApplicationResourceFile("ingredients.txt"); if (ingredientsPath == null) { // failed to find ingredients.txt file MessageDialog md = new MessageDialog(this, DialogFlags.DestroyWithParent, MessageType.Error, ButtonsType.Close, "Failed to find ingredients.txt file. Please check your installation."); md.Run(); md.Destroy(); Application.Quit(); } ReagentManager.Load(ingredientsPath); Build(); if (unmodifiedSwatch != null) { unmodifiedSwatch.Clear(); } if (reactionSwatch != null) { reactionSwatch.Clear(); } // get the root window rootWindow = Gdk.Global.DefaultRootWindow; // get its width and height int screenWidth; int screenHeight; rootWindow.GetSize(out screenWidth, out screenHeight); int pixelMultiplier = 1; if ( DesertPaintLab.Settings.Load() == true ) { DesertPaintLab.Settings.Get("ScreenWidth", out screenWidth); DesertPaintLab.Settings.Get("ScreenHeight", out screenHeight); DesertPaintLab.Settings.Get("PixelMultiplier", out pixelMultiplier); } bool enableDebugMenu; DesertPaintLab.Settings.Get("EnableDebugMenu", out enableDebugMenu); this.DebugAction.Visible = enableDebugMenu; ScreenCheckDialog screenCheckDialog = new ScreenCheckDialog(); screenCheckDialog.ScreenWidth = screenWidth; screenCheckDialog.ScreenHeight = screenHeight; screenCheckDialog.GamePixelWidth = pixelMultiplier; ResponseType resp = (ResponseType)screenCheckDialog.Run(); screenWidth = screenCheckDialog.ScreenWidth; screenHeight = screenCheckDialog.ScreenHeight; pixelMultiplier = screenCheckDialog.GamePixelWidth; screenCheckDialog.Destroy(); DesertPaintLab.Settings.Set("ScreenWidth", screenWidth); DesertPaintLab.Settings.Set("ScreenHeight", screenHeight); DesertPaintLab.Settings.Set("PixelMultiplier", pixelMultiplier); DesertPaintLab.Settings.Save(); screenBuffer = new Gdk.Pixbuf(Gdk.Colorspace.Rgb, false, 8, screenWidth, screenHeight); ReactionRecorder.SetPixelMultiplier(pixelMultiplier); if (!OpenProfile()) { shouldShutDown = true; } } bool ConfirmedExit() { if (unsavedData) { MessageDialog md = new MessageDialog(this, DialogFlags.DestroyWithParent, MessageType.Warning, ButtonsType.OkCancel, "Your last reaction was unsaved." + "Are you sure you want to quit?"); ResponseType resp = (ResponseType)md.Run(); md.Destroy(); return (resp == ResponseType.Ok); } return true; } void SetProfileName(string name) { profile = new PlayerProfile(name, System.IO.Path.Combine(FileUtils.AppDataPath, name)); statusBar.Push(0, name); } bool NewProfile() { bool newProfileCreated = false; bool duplicateName = false; NewProfileDialog newProfileDialog = new NewProfileDialog(); ResponseType resp = (ResponseType)newProfileDialog.Run(); if (resp == ResponseType.Ok) { // Make sure profile doesn't already exist. foreach (string profileName in profileList) { if (profileName == newProfileDialog.ProfileName) { MessageDialog md = new MessageDialog(this, DialogFlags.DestroyWithParent, MessageType.Error, ButtonsType.Ok, "That profile name already exists."); resp = (ResponseType)md.Run(); md.Destroy(); duplicateName = true; break; } } if (!duplicateName) { // Set profile name. SetProfileName(newProfileDialog.ProfileName); newProfileCreated = profile.Initialize(); if (!newProfileCreated) { MessageDialog md = new MessageDialog(this, DialogFlags.DestroyWithParent, MessageType.Error, ButtonsType.Ok, "Failed to initialize profile: " + profile.LastError); resp = (ResponseType)md.Run(); md.Destroy(); duplicateName = false; } } } newProfileDialog.Destroy(); return newProfileCreated; } bool OpenProfile() { bool profileSelected = false; if (profileList.Count > 0) { SelectProfileDialog selectProfileDialog = new SelectProfileDialog(); selectProfileDialog.ProfileList = profileList; ResponseType resp = (ResponseType)selectProfileDialog.Run(); selectProfileDialog.Destroy(); string selectedProfile = selectProfileDialog.SelectedProfile; if ((resp == ResponseType.Ok) && (selectedProfile.Length > 0)) // Selected a profile. { SetProfileName(selectedProfile); profileSelected = true; } else if (resp == ResponseType.Accept) // New profile. { profileSelected = NewProfile(); } } else { FirstRunDialog firstRunDialog = new FirstRunDialog(); ResponseType resp = (ResponseType)firstRunDialog.Run(); firstRunDialog.Destroy(); if (resp == ResponseType.Ok) // New profile { profileSelected = NewProfile(); } else if (resp == ResponseType.Accept) // Import { FileChooserDialog fileDialog = new FileChooserDialog("Select reactions.txt file.", this, FileChooserAction.Open, Gtk.Stock.Cancel, ResponseType.Cancel, Gtk.Stock.Open, ResponseType.Accept); resp = (ResponseType)fileDialog.Run(); if (resp == ResponseType.Accept) { string fileName = fileDialog.Filename; string directory = fileDialog.CurrentFolder; if (fileName == "reactions.txt") { profileSelected = NewProfile(); if (profileSelected) { profile.ImportFromPP(directory); } } } } } if (profileSelected) { bool ok = profile.Load(); if (ok) { PopulateDropDowns(); recipe.Reactions = profile.Reactions; } else { MessageDialog md = new MessageDialog(this, DialogFlags.DestroyWithParent, MessageType.Error, ButtonsType.Ok, "Error loading profile: " + profile.LastError); md.Run(); md.Destroy(); profileSelected = false; } } return profileSelected; } void PopulateDropDowns() { ReagentManager.PopulateReagents(ref ingredient1ComboBox); ReagentManager.PopulateReagents(ref ingredient2ComboBox); ReagentManager.PopulateReagents(ref ingredient3ComboBox); ingredient2ComboBox.Sensitive = false; ingredient3ComboBox.Sensitive = false; Gtk.TreeIter iter; ingredient1ComboBox.Model.IterNthChild(out iter, 0); ingredient1ComboBox.SetActiveIter(iter); ingredient2ComboBox.Model.IterNthChild(out iter, 0); ingredient2ComboBox.SetActiveIter(iter); ingredient3ComboBox.Model.IterNthChild(out iter, 0); ingredient3ComboBox.SetActiveIter(iter); } protected void SetExpectedColor(byte red, byte green, byte blue) { expectedColor.Red = red; expectedColor.Green = green; expectedColor.Blue = blue; unmodifiedSwatch.Color = expectedColor; } protected void SetExpectedColor(PaintColor color) { SetExpectedColor(color.Red, color.Green, color.Blue); } protected void UpdateIngredients() { Reaction reaction1, reaction2; TreeIter selectIter; string reagentName; reagents[0] = null; reagents[1] = null; reagents[2] = null; bool reactionKnown = true; saveButton.Sensitive = false; recipe.Clear(); if (ingredient1ComboBox.GetActiveIter(out selectIter)) { reagentName = (string)ingredient1ComboBox.Model.GetValue(selectIter, 0); if ((reagentName == null) || (reagentName.Length == 0)) { // Nothing selected as reagent 1 ingredient2ComboBox.Sensitive = false; ingredient3ComboBox.Sensitive = false; unmodifiedSwatch.Clear(); captureButton.Sensitive = false; } else { recipe.AddReagent(reagentName); reagents[0] = ReagentManager.GetReagent(reagentName); ingredient2ComboBox.Sensitive = true; if (ingredient2ComboBox.GetActiveIter(out selectIter)) { reagentName = (string)ingredient2ComboBox.Model.GetValue(selectIter, 0); if ((reagentName == null) || (reagentName.Length == 0)) { ingredient3ComboBox.Sensitive = false; saveButton.Sensitive = false; reactionKnown = false; } else { recipe.AddReagent(reagentName); reagents[1] = ReagentManager.GetReagent(reagentName); ingredient3ComboBox.Sensitive = true; captureButton.Sensitive = true; reaction1 = profile.FindReaction(reagents[0], reagents[1]); if ((reaction1 != null) || (reagents[0] == reagents[1])) { ingredient3ComboBox.Sensitive = true; } else { reactionKnown = false; ingredient3ComboBox.Sensitive = false; } if (ingredient3ComboBox.GetActiveIter(out selectIter)) { reagentName = (string)ingredient3ComboBox.Model.GetValue(selectIter, 0); if ((reagentName != null) && (reagentName.Length != 0)) { recipe.AddReagent(reagentName); reagents[2] = ReagentManager.GetReagent(reagentName); if (!reactionKnown) { MessageDialog md = new MessageDialog(this, DialogFlags.DestroyWithParent, MessageType.Error, ButtonsType.Ok, "To do a three-ingredient reaction test, " + "you must first recored the reaction of " + "the first two ingredients."); md.Run(); md.Destroy(); captureButton.Sensitive = false; } reaction1 = profile.FindReaction(reagents[0], reagents[2]); reaction2 = profile.FindReaction(reagents[1], reagents[2]); if (reactionKnown && (reaction1 == null) && (reaction2 == null)) { MessageDialog md = new MessageDialog(this, DialogFlags.DestroyWithParent, MessageType.Error, ButtonsType.Ok, "To do a three-ingredient reaction test, " + "you must first record the reaction of " + "either the first or second ingredient " + "with the third ingredient."); md.Run(); md.Destroy(); captureButton.Sensitive = false; } if ((reaction1 == null) && (reagents[0] != reagents[2])) { reactionKnown = false; } if ((reaction2 == null) && (reagents[1] != reagents[2])) { reactionKnown = false; } } } } } expectedColor.Set(recipe.BaseColor); unmodifiedSwatch.Color = expectedColor; //SetExpectedColor(recipeColor.Red, recipeColor.Green, recipeColor.Blue); if (reactionKnown) { reactedColor.Set(recipe.ReactedColor); reactionSwatch.Color = reactedColor; } else { reactionSwatch.Clear(); } } } } protected void OnDeleteEvent(object sender, DeleteEventArgs a) { if (ConfirmedExit()) { a.RetVal = true; Application.Quit(); } else { a.RetVal = false; } } bool IsPapyTexture(byte r, byte g, byte b) { return ((r > 0xD0) && (g > 0xC8) && (b > 0xA0)) && ((r < 0xF4) && (g < 0xE0) && (b < 0xC4)); } unsafe bool CaptureReactionColor() { // Take a screenshot. int screenWidth, screenHeight; bool debugScreenshot = false; bool enableDebugMenu = false; DesertPaintLab.Settings.Get("ScreenWidth", out screenWidth); DesertPaintLab.Settings.Get("ScreenHeight", out screenHeight); DesertPaintLab.Settings.Get("EnableDebugMenu", out enableDebugMenu); DesertPaintLab.Settings.Get("DebugScreenshot", out debugScreenshot); Gdk.Image rootImage = rootWindow.GetImage(0, 0, screenWidth, screenHeight); screenBuffer.GetFromImage(rootImage, rootImage.Colormap, 0, 0, 0, 0, screenWidth, screenHeight); //screenBuffer.GetFromDrawable(rootWindow, // rootWindow.Colormap, 0, 0, 0, 0, screenWidth, screenHeight); int stride = screenBuffer.Rowstride; byte* pixBytes = (byte*)screenBuffer.Pixels; int redPixelStart = -1; bool wasCaptured = ReactionRecorder.CaptureReaction(pixBytes, screenWidth, screenHeight, stride, ref reactedColor, ref redPixelStart); if (!wasCaptured && enableDebugMenu && debugScreenshot) { // write out the screenshot string screenshotDir = Environment.GetFolderPath(Environment.SpecialFolder.MyDocuments); string filename; int i = 0; do { ++i; filename = System.IO.Path.Combine(screenshotDir, String.Format("DesertPaintLab_Colormatch{0}.png", i)); } while (System.IO.File.Exists(filename)); screenBuffer.Save(filename, "png"); } else { // convert to pixel offset instead of byte int redPixelStartX = (redPixelStart % stride) / 3; int redPixelStartY = (redPixelStart / stride); // write out the screenshot string screenshotDir = Environment.GetFolderPath(Environment.SpecialFolder.MyDocuments); string filename; int i = 0; do { ++i; filename = System.IO.Path.Combine(screenshotDir, String.Format("DesertPaintLab_Colormatch{0}.png", i)); } while (System.IO.File.Exists(filename)); int captureAreaWidth = Math.Min(64, screenWidth - redPixelStartX + 64); int captureAreaHeight = Math.Min(64, screenHeight - redPixelStartY + 64); Gdk.Pixbuf outPixBuf = new Gdk.Pixbuf(screenBuffer, Math.Max(0, redPixelStartX - 16), Math.Max(0, redPixelStartY - 16), captureAreaWidth, captureAreaHeight); //screenBuffer.Save(filename, "png"); outPixBuf.Save(filename, "png"); } //screenBuffer.Save("screenshot.png", "png"); return wasCaptured; } protected virtual void OnDebugScreenshot(object sender, System.EventArgs e) { int screenWidth, screenHeight; DesertPaintLab.Settings.Get("ScreenWidth", out screenWidth); DesertPaintLab.Settings.Get("ScreenHeight", out screenHeight); Gdk.Image rootImage = rootWindow.GetImage(0, 0, screenWidth, screenHeight); screenBuffer.GetFromImage(rootImage, rootImage.Colormap, 0, 0, 0, 0, screenWidth, screenHeight); string screenshotDir = Environment.GetFolderPath(Environment.SpecialFolder.MyDocuments); string filename; int i = 0; do { ++i; filename = System.IO.Path.Combine(screenshotDir, String.Format("DesertPaintLab_{0}.png", i)); } while (System.IO.File.Exists(filename)); screenBuffer.Save(filename, "png"); } protected virtual void OnCaptureButton(object sender, System.EventArgs e) { if (CaptureReactionColor()) { string warning = ""; if (reactedColor.Red == 0) { warning = warning + "\nRed is too low."; } if (reactedColor.Green == 0) { warning = warning + "\nGreen is too low."; } if (reactedColor.Blue == 0) { warning = warning + "\nBlue is too low."; } if (reactedColor.Red == 255) { warning = warning + "\nRed is too high."; } if (reactedColor.Green == 255) { warning = warning + "\nGreen is too high."; } if (reactedColor.Blue == 255) { warning = warning + "\nBlue is too high."; } if (warning.Length != 0) { MessageDialog md = new MessageDialog(this, DialogFlags.DestroyWithParent, MessageType.Error, ButtonsType.Ok, "Reaction clipped. You will need to do a " + "3-way reaction to test this pair. Details: " + warning); md.Run(); md.Destroy(); } else { this.reactionSwatch.Color = reactedColor; saveButton.Sensitive = true; } } else { MessageDialog md = new MessageDialog(this, DialogFlags.DestroyWithParent, MessageType.Error, ButtonsType.Ok, "Pigment Lab dialog box NOT FOUND. Please ensure " + "that there is an unobstructed view of the dialog " + "and that your interface size is set to 'small' " + "when you press the Capture button."); md.Run(); md.Destroy(); } } protected virtual void OnSaveButton(object sender, System.EventArgs e) { if (ReactionRecorder.RecordReaction(profile, expectedColor, reactedColor, reagents)) { saveButton.Sensitive = false; } } protected virtual void OnChangedIngredient1(object sender, System.EventArgs e) { UpdateIngredients(); } protected virtual void OnChangedIngredient2(object sender, System.EventArgs e) { UpdateIngredients(); } protected virtual void OnChangedIngredient3(object sender, System.EventArgs e) { UpdateIngredients(); } protected virtual void OnNewProfile(object sender, System.EventArgs e) { if (unsavedData) { MessageDialog md = new MessageDialog(this, DialogFlags.DestroyWithParent, MessageType.Warning, ButtonsType.OkCancel, "Your last reaction was unsaved." + "Are you sure you want to lose your changes?"); ResponseType resp = (ResponseType)md.Run(); md.Destroy(); if (resp != ResponseType.Ok) { return; } } if (NewProfile()) { bool ok = profile.Load(); if (ok) { PopulateDropDowns(); recipe.Reactions = profile.Reactions; } else { MessageDialog md = new MessageDialog(this, DialogFlags.DestroyWithParent, MessageType.Warning, ButtonsType.OkCancel, "Failed to load profile: " + profile.LastError); md.Run(); md.Destroy(); } } } protected virtual void OnOpenProfile(object sender, System.EventArgs e) { bool profileSelected = false; SelectProfileDialog selectProfileDialog = new SelectProfileDialog(); selectProfileDialog.ProfileList = profileList; ResponseType resp = (ResponseType)selectProfileDialog.Run(); selectProfileDialog.Destroy(); if (resp == ResponseType.Ok) // Selected a profile. { SetProfileName(selectProfileDialog.SelectedProfile); profileSelected = true; } else if (resp == ResponseType.Accept) // New profile. { profileSelected = NewProfile(); } if (profileSelected) { bool ok = profile.Load(); if (ok) { PopulateDropDowns(); } else { MessageDialog md = new MessageDialog(this, DialogFlags.DestroyWithParent, MessageType.Warning, ButtonsType.OkCancel, "Failed to load profile: " + profile.LastError); md.Run(); md.Destroy(); } } } protected virtual void OnAbout(object sender, System.EventArgs e) { AboutDialog aboutDialog = new AboutDialog(); aboutDialog.ProgramName = "Desert Paint Lab"; aboutDialog.Version = APP_VERSION ; aboutDialog.Comments = "Desert Paint Lab paint reaction recorder for A Tale in the Desert"; aboutDialog.Authors = new string [] {"Tess Snider", "Jason Maltzen"}; //aboutDialog.Website = "http://www.google.com/"; aboutDialog.Run(); aboutDialog.Destroy(); } protected virtual void OnMenuExit (object sender, System.EventArgs e) { if (ConfirmedExit()) { Application.Quit(); } } protected virtual void OnExport(object sender, System.EventArgs e) { FileChooserDialog fileDialog = new FileChooserDialog("Select destination file.", this, FileChooserAction.Save, Gtk.Stock.Cancel, ResponseType.Cancel, Gtk.Stock.Save, ResponseType.Accept); ResponseType resp = (ResponseType)fileDialog.Run(); if (resp == ResponseType.Accept) { string fileName = fileDialog.Filename; string directory = fileDialog.CurrentFolder; profile.SaveToPP(System.IO.Path.Combine(directory, fileName)); } fileDialog.Destroy(); } protected virtual void RunSimulator(object sender, System.EventArgs e) { SimulatorWindow win = new SimulatorWindow(profile); win.Show(); } protected void OnOpenRecipeGenerator(object sender, EventArgs e) { RecipeGeneratorWindow win = new RecipeGeneratorWindow(profile); win.Show(); //RecipeGenerator gen = new RecipeGenerator(); //gen.BeginRecipeGeneration(profile.Reactions, 15, 7); //MessageDialog md = new MessageDialog(this, // DialogFlags.DestroyWithParent, // MessageType.Info, ButtonsType.Close, // "Coming Soon!"); //md.Run(); //md.Destroy(); } protected void OnShowReactionStatus(object sender, EventArgs e) { ReactionStatusWindow win = new ReactionStatusWindow(profile); win.Show(); } protected void OnShowIngredients(object sender, EventArgs e) { ReagentWindow win = new ReagentWindow(profile); win.Show(); } protected void OnExportProfile(object sender, EventArgs e) { FileChooserDialog fileDialog = new FileChooserDialog("Select destination file.", this, FileChooserAction.Save, Gtk.Stock.Cancel, ResponseType.Cancel, Gtk.Stock.Save, ResponseType.Accept); ResponseType resp = (ResponseType)fileDialog.Run(); if (resp == ResponseType.Accept) { string fileName = fileDialog.Filename; string directory = fileDialog.CurrentFolder; string targetFile = System.IO.Path.Combine(directory, fileName); if (File.Exists(targetFile)) { // prompt to overwrite MessageDialog md = new MessageDialog(this, DialogFlags.DestroyWithParent, MessageType.Warning, ButtonsType.OkCancel, "Overwrite profile at" + targetFile + "?"); resp = (ResponseType)md.Run(); md.Destroy(); if (resp == ResponseType.Ok) { File.Delete(targetFile); profile.Export(targetFile); } } else { profile.Export(targetFile); } } fileDialog.Destroy(); } protected void OnImportProfile(object sender, EventArgs e) { FileChooserDialog fileDialog = new FileChooserDialog("Select file to import.", this, FileChooserAction.Open, Gtk.Stock.Cancel, ResponseType.Cancel, Gtk.Stock.Open, ResponseType.Accept); ResponseType resp = (ResponseType)fileDialog.Run(); if (resp == ResponseType.Accept) { string fileName = fileDialog.Filename; string directory = fileDialog.CurrentFolder; string targetFile = fileName; if (directory != null) { targetFile = System.IO.Path.Combine(directory, fileName); } if (Directory.Exists(profile.Directory)) { // prompt to overwrite MessageDialog md = new MessageDialog(this, DialogFlags.DestroyWithParent, MessageType.Warning, ButtonsType.OkCancel, "Overwrite profile at" + directory + "?"); resp = (ResponseType)md.Run(); md.Destroy(); if (resp == ResponseType.Ok) { Directory.Delete(profile.Directory, true); profile.Import(targetFile); } } else { profile.Import(targetFile); } bool ok = profile.Load(); if (ok) { PopulateDropDowns(); recipe.Reactions = profile.Reactions; } else { MessageDialog md = new MessageDialog(this, DialogFlags.DestroyWithParent, MessageType.Warning, ButtonsType.OkCancel, "Failed to load imported profile: " + profile.LastError); md.Run(); md.Destroy(); } } fileDialog.Destroy(); } }