ATITD-Tools/Desert-Paint-Lab Changeset

Changeset - 3154dda31272

Parent rev.

Child rev.

[Not reviewed]

default

0 1 0

Jason Maltzen - 5 years ago 2019-11-02 02:23:43
jason@hiddenachievement.com

Slightly safer parsing of reaction files

1 file changed with 234 insertions and 206 deletions:

PlayerProfile.cs

234

206

0 comments (0 inline, 0 general)

PlayerProfile.cs

➞

Show inline comments

@@ ... / @@ -20,416 +20,444 @@ @@
  THE SOFTWARE.
 */
 using Gtk;
 using System;
 using System.IO;
 using System.IO.Compression;
 using System.Collections.Generic;
 using System.Text.RegularExpressions;
 namespace DesertPaintLab
+{
 	public class PlayerProfile
+	{
         public string Name { get; private set; }
 		string directory;
 		string reactFile;
     public class PlayerProfile
+    {
         public string Name
+        {
             get; private set;
+        }
         string directory;
         string reactFile;
         string reagentFile;
         string settingsFile;
         static Regex recipeHeaderRegex = new Regex(@"^--- Recipe: (?<colorname>(\w*\s)*\w+)\s*");
         static Regex recipeIngredientRegex = new Regex(@"(?<ingredient>(\w+\s)?\w+)\s*\|\s*(?<quantity>\d+)\s*");
         ReactionSet reactions = new ReactionSet();
         // ingredient -> [ingredient, reaction]
 		//SortedDictionary<string, SortedDictionary<string, Reaction>> reactions =
 		//	new SortedDictionary<string, SortedDictionary<string, Reaction>>();
         //SortedDictionary<string, SortedDictionary<string, Reaction>> reactions =
         //	new SortedDictionary<string, SortedDictionary<string, Reaction>>();
         SortedDictionary<string, PaintRecipe> recipes;
         SortedDictionary<string, PaintRecipe> ribbonRecipes;
         Settings settings = new Settings();
         public Settings ProfileSettings {
             get {
         public Settings ProfileSettings
+        {
             get
+            {
                 return settings;
+            }
+        }
         static PlayerProfile current = null;
         static PlayerProfile Current
+        {
             get {
             get
+            {
                 return current;
+            }
+        }
         public string LastError { get; private set; }
 		public PlayerProfile(string name, string directory)
+		{
 			this.Name = name;
 			this.directory = directory;
 			this.reactFile = System.IO.Path.Combine(directory, "dp_reactions.txt");
         public string LastError
+        {
             get; private set;
+        }
         public PlayerProfile(string name, string directory)
+        {
             this.Name = name;
             this.directory = directory;
             this.reactFile = System.IO.Path.Combine(directory, "dp_reactions.txt");
             this.reagentFile = System.IO.Path.Combine(directory, "ingredients.txt");
             this.settingsFile = System.IO.Path.Combine(directory, "settings");
             this.recipes = new SortedDictionary<string, PaintRecipe>();
             this.ribbonRecipes = new SortedDictionary<string, PaintRecipe>();
             foreach (PaintColor color in Palette.Colors)
+            {
                 this.recipes.Add(color.Name, new PaintRecipe());
+            }
             foreach (PaintColor color in Palette.Colors)
+            {
                 this.ribbonRecipes.Add(color.Name, new PaintRecipe());
+            }
+		}
+        }
         public string Directory
+        {
             get {
             get
+            {
                 return this.directory;
+            }
+        }
         public ReactionSet Reactions
+        {
             get {
             get
+            {
                 return this.reactions;
+            }
+        }
         public string ReagentFile
+        {
             get {
             get
+            {
                 return this.reagentFile;
+            }
+        }
         public SortedDictionary<string, PaintRecipe> Recipes
+        {
             get {
             get
+            {
                 return this.recipes;
+            }
+        }
         public SortedDictionary<string, PaintRecipe> RibbonRecipes
+        {
             get {
             get
+            {
                 return this.ribbonRecipes;
+            }
+        }
         public int RecipeCount
+        {
             get {
             get
+            {
                 int count = 0;
                 foreach (PaintRecipe recipe in this.recipes.Values)
+                {
                     if (recipe.IsValidForConcentration(PaintRecipe.PAINT_RECIPE_MIN_CONCENTRATION))
+                    {
                         ++count;
+                    }
+                }
                 return count;
+            }
+        }
         public int RibbonCount
+        {
             get {
             get
+            {
                 int count = 0;
                 foreach (PaintRecipe recipe in this.ribbonRecipes.Values)
+                {
                     if (recipe.IsValidForConcentration(PaintRecipe.RIBBON_RECIPE_MIN_CONCENTRATION))
+                    {
                         ++count;
+                    }
+                }
                 return count;
+            }
+        }
 		public bool Initialize()
+		{
         public bool Initialize()
+        {
             // Copy template files into new directory.
             string templatePath = FileUtils.FindApplicationResourceDirectory("template");
             if (!System.IO.Directory.Exists(templatePath))
+            {
                 LastError = "Failed to find profile template folder.";
                 return false;
+            }
 			// Create new directory.
 			System.IO.Directory.CreateDirectory(directory);
 			DirectoryInfo di = new DirectoryInfo(templatePath);
 			FileInfo[] templateFiles = di.GetFiles();
             // Create new directory.
             System.IO.Directory.CreateDirectory(directory);
 			foreach (FileInfo file in templateFiles)
+			{
             DirectoryInfo di = new DirectoryInfo(templatePath);
             FileInfo[] templateFiles = di.GetFiles();
             foreach (FileInfo file in templateFiles)
+            {
                 string destFile = System.IO.Path.Combine(directory, file.Name);
-				System.IO.File.Copy(file.FullName, destFile, true);
+                System.IO.File.Copy(file.FullName, destFile, true);
                 if (!File.Exists(destFile))
+                {
                     LastError = "Failed to copy template file " + file.Name + ".";
                     return false;
+                }
+			}
+                }
+            }
             return true;
+		}
 		public void ConvertFromPP(string ppFile, string dpFile)
+		{
 			string line;
 			using (StreamReader reader = new StreamReader(ppFile))
+			{
 				using (StreamWriter writer = new StreamWriter(dpFile))
+				{
 					while ((line = reader.ReadLine()) != null)
+                	{
 						string[] tokens = line.Split(null);
 						if ((tokens.Length > 0) && (tokens[0] != "//"))
+						{
 							// Write reaction.
 							writer.Write(tokens[0] + " " + tokens[2] + " ");
 							switch (tokens[4])
+							{
 							case "W":
 								writer.WriteLine(tokens[6] + " " + tokens[6] + " " + tokens[6]);
 								break;
 							case "R":
 								writer.WriteLine(tokens[6] + " 0 0");
 								break;
 							case "G":
 								writer.WriteLine("0 " + tokens[6] + " 0");
 								break;
 							case "B":
 								writer.WriteLine("0 0 " + tokens[6]);
 								break;
+							}
 							// Write reverse reaction.
 							writer.Write(tokens[2] + " " + tokens[0] + " ");
 							switch (tokens[4])
+							{
 							case "W":
 								writer.WriteLine(tokens[8] + " " + tokens[8] + " " + tokens[8]);
 								break;
 							case "R":
 								writer.WriteLine(tokens[8] + " 0 0");
 								break;
 							case "G":
 								writer.WriteLine("0 " + tokens[8] + " 0");
 								break;
 							case "B":
 								writer.WriteLine("0 0 " + tokens[8]);
 								break;
+							}
+						}
+					}
+				}
+			}
+		}
 		public bool SaveToPP(string ppFile)
+		{
 			Reaction reaction1, reaction2;
 			using (StreamWriter writer = new StreamWriter(ppFile))
+			{
+        }
         private void WriteReaction(StreamWriter writer, string reagent1, string reagent2, string r, string g, string b)
+        {
             writer.Write(reagent1);
             writer.Write(" ");
             writer.Write(reagent2);
             writer.Write(" ");
             writer.Write(r);
             writer.Write(" ");
             writer.Write(g);
             writer.Write(" ");
             writer.WriteLine(b);
+        }
         public void ConvertFromPP(string ppFile, string dpFile)
+        {
             string line;
             using (StreamReader reader = new StreamReader(ppFile))
+            {
                 using (StreamWriter writer = new StreamWriter(dpFile))
+                {
                     while ((line = reader.ReadLine()) != null)
+                    {
                         string[] tokens = line.Split('|');
                         //if ((tokens.Length > 0) && (tokens [0] != "//"))
                         if ((tokens.Length != 5) && (tokens[0].Trim() != "//"))
+                        {
                             string reagent1 = tokens[0].Trim();
                             string reagent2 = tokens[1].Trim();
                             string colorCode = tokens[2].Trim();
                             string change1 = tokens[3].Trim();
                             string change2 = tokens[4].Trim();
                             // Write reaction.
                             switch (colorCode)
+                            {
                             case "W":
                                 WriteReaction(writer, reagent1, reagent2, change1, change1, change1);
                                 WriteReaction(writer, reagent2, reagent1, change2, change2, change2);
                                 break;
                             case "R":
                                 WriteReaction(writer, reagent1, reagent2, change1, "0", "0");
                                 WriteReaction(writer, reagent2, reagent1, change2, "0", "0");
                                 break;
                             case "G":
                                 WriteReaction(writer, reagent1, reagent2, "0", change1, "0");
                                 WriteReaction(writer, reagent2, reagent1, "0", change2, "0");
                                 break;
                             case "B":
                                 WriteReaction(writer, reagent1, reagent2, "0", "0", change1);
                                 WriteReaction(writer, reagent2, reagent1, "0", "0", change2);
                                 break;
+                            }
+                        }
+                    }
+                }
+            }
+        }
         public bool SaveToPP(string ppFile)
+        {
             Reaction reaction1, reaction2;
             using (StreamWriter writer = new StreamWriter(ppFile))
+            {
                 foreach (string reagentName1 in ReagentManager.Names)
+                {
                     // TODO: could be more efficient by only iterating over the names after reagent1
                     foreach (string reagentName2 in ReagentManager.Names)
+                    {
                         if (reagentName1.Equals(reagentName2))
+                        {
                             continue;
+                        }
                         Reagent reagent1 = ReagentManager.GetReagent(reagentName1);
                         Reagent reagent2 = ReagentManager.GetReagent(reagentName2);
                         reaction1 = reactions.Find(reagent1, reagent2);
                         if (reaction1 != null && !reaction1.Exported)
+                        {
                             reaction2 = reactions.Find(reagent2, reagent1);
                             if (reaction2 != null)
+                            {
 								writer.Write(reagent1.PracticalPaintName + " | " + reagent2.PracticalPaintName + " | ");
 								if ((Math.Abs(reaction1.Red) > Math.Abs(reaction1.Green)) ||
 								    (Math.Abs(reaction2.Red) > Math.Abs(reaction2.Green)))
+								{
 									writer.WriteLine("R | " + reaction1.Red + " | " + reaction2.Red);
+								}
 								else if ((Math.Abs(reaction1.Green) > Math.Abs(reaction1.Red)) ||
 									(Math.Abs(reaction2.Green) > Math.Abs(reaction2.Red)))
+								{
 									writer.WriteLine("G | " + reaction1.Green + " | " + reaction2.Green);
+								}
 								else if ((Math.Abs(reaction1.Blue) > Math.Abs(reaction1.Red)) ||
 									(Math.Abs(reaction2.Blue) > Math.Abs(reaction2.Red)))
+								{
 									writer.WriteLine("B | " + reaction1.Blue + " | " + reaction2.Blue);
+								}
 								else
+								{
 									writer.WriteLine("W | " + reaction1.Red + " | " + reaction2.Red);
+								}
 								reaction1.Exported = true;
 								reaction2.Exported = true;
+							}
+						}
+					}
+				}
+			}
 			// Clear Exported flags.
                                 writer.Write(reagent1.PracticalPaintName + " | " + reagent2.PracticalPaintName + " | ");
                                 if ((Math.Abs(reaction1.Red) > Math.Abs(reaction1.Green)) ||
                                     (Math.Abs(reaction2.Red) > Math.Abs(reaction2.Green)))
+                                {
                                     writer.WriteLine("R | " + reaction1.Red + " | " + reaction2.Red);
+                                }
                                 else if ((Math.Abs(reaction1.Green) > Math.Abs(reaction1.Red)) ||
                                     (Math.Abs(reaction2.Green) > Math.Abs(reaction2.Red)))
+                                {
                                     writer.WriteLine("G | " + reaction1.Green + " | " + reaction2.Green);
+                                }
                                 else if ((Math.Abs(reaction1.Blue) > Math.Abs(reaction1.Red)) ||
                                     (Math.Abs(reaction2.Blue) > Math.Abs(reaction2.Red)))
+                                {
                                     writer.WriteLine("B | " + reaction1.Blue + " | " + reaction2.Blue);
+                                }
                                 else
+                                {
                                     writer.WriteLine("W | " + reaction1.Red + " | " + reaction2.Red);
+                                }
                                 reaction1.Exported = true;
                                 reaction2.Exported = true;
+                            }
+                        }
+                    }
+                }
+            }
             // Clear Exported flags.
             foreach (string reagentName1 in ReagentManager.Names)
+            {
                 // TODO: could be more efficient by only iterating over the names after reagent1
                 foreach (string reagentName2 in ReagentManager.Names)
+                {
                     if (reagentName1.Equals(reagentName2))
+                    {
                         continue;
+                    }
                     Reagent reagent1 = ReagentManager.GetReagent(reagentName1);
                     Reagent reagent2 = ReagentManager.GetReagent(reagentName2);
                     reaction1 = reactions.Find(reagent1, reagent2);
                     if (reaction1 != null)
+                    {
+                    {
                         reaction1.Exported = false;
+                    }
                     reaction2 = reactions.Find(reagent2, reagent1);
                     if (reaction2 != null)
+                    {
                         reaction2.Exported = false;
+                    }
+				}
+			}
 			return true;
+		}
 		public void ImportFromPP(string importDir)
+		{
 			// Convert old file.
 			ConvertFromPP(
 			    System.IO.Path.Combine(importDir, "reactions.txt"),
 				reactFile);
 			try
+			{
 				// If there is an ingredients file, move it in.
 				System.IO.File.Copy(
 				    System.IO.Path.Combine(importDir, "ingredients.txt"),
 				    System.IO.Path.Combine(directory, "ingredients.txt"),
 				    true);
+			}
 			catch (Exception)
+			{
 				// If there is no ingredients file, we don't really care.
+			}
+		}
+                }
+            }
             return true;
+        }
         public void ImportFromPP(string importDir)
+        {
             // Convert old file.
             ConvertFromPP(
                 System.IO.Path.Combine(importDir, "reactions.txt"),
                 reactFile);
             try
+            {
                 // If there is an ingredients file, move it in.
                 System.IO.File.Copy(
                     System.IO.Path.Combine(importDir, "ingredients.txt"),
                     System.IO.Path.Combine(directory, "ingredients.txt"),
                     true);
+            }
             catch (Exception)
+            {
                 // If there is no ingredients file, we don't really care.
+            }
+        }
         public void Import(string file)
+        {
             ZipFile.ExtractToDirectory(file, directory);
+        }
 		public void Export(string file)
+		{
         public void Export(string file)
+        {
             ZipFile.CreateFromDirectory(directory, file);
+		}
 		public bool Load()
+		{
 			string line;
+        }
         public bool Load()
+        {
             string line;
             settings.Reset();
             settings.Load(settingsFile);
-			reactions.Clear();
+            reactions.Clear();
             if (File.Exists(reagentFile))
+            {
-    			ReagentManager.LoadProfileReagents(reagentFile);
+                ReagentManager.LoadProfileReagents(reagentFile);
+            }
             else
+            {
                 LastError = "Failed to find profile reagents file.";
                 return false;
+            }
-			ReagentManager.InitializeReactions(ref reactions);
+            ReagentManager.InitializeReactions(ref reactions);
             if (!File.Exists(reactFile))
+            {
                 LastError = "Failed to find profile reactions file.";
                 return false;
+            }
 			using (StreamReader reader = new StreamReader(reactFile))
+			{
 				while ((line = reader.ReadLine()) != null)
+               	{
 					string[] tokens = line.Split(null);
                     Reagent reagent1 = ReagentManager.GetReagent(tokens[0]);
                     Reagent reagent2 = ReagentManager.GetReagent(tokens[1]);
 					reactions.Set(reagent1, reagent2, new Reaction(int.Parse(tokens[2]), int.Parse(tokens[3]), int.Parse(tokens[4])));
+				}
+			}
             using (StreamReader reader = new StreamReader(reactFile))
+            {
                 while ((line = reader.ReadLine()) != null)
+                {
                     string[] tokens = line.Split(' ');
                     if (tokens.Length == 5)
+                    {
                         Reagent reagent1 = ReagentManager.GetReagent(tokens[0].Trim());
                         Reagent reagent2 = ReagentManager.GetReagent(tokens[1].Trim());
                         Reaction reaction = new Reaction(
                             int.Parse(tokens[2].Trim()),
                             int.Parse(tokens[3].Trim()),
                             int.Parse(tokens[4].Trim())
                             );
                         reactions.Set(reagent1, reagent2, reaction);
+                    }
+                }
+            }
             return true;
+		}
 		public void Save()
+		{
+        }
         public void Save()
+        {
             settings.Save(settingsFile);
 			Reaction reaction;
 			using (StreamWriter writer = new StreamWriter(reactFile, false))
+			{
             Reaction reaction;
             using (StreamWriter writer = new StreamWriter(reactFile, false))
+            {
                 foreach (string reagentName1 in ReagentManager.Names)
+                {
                     // TODO: could be more efficient by only iterating over the names after reagent1
                     foreach (string reagentName2 in ReagentManager.Names)
+                    {
                         if (reagentName1.Equals(reagentName2))
+                        {
                             continue;
+                        }
                         Reagent reagent1 = ReagentManager.GetReagent(reagentName1);
                         Reagent reagent2 = ReagentManager.GetReagent(reagentName2);
                         reaction = reactions.Find(reagent1, reagent2);
     					if (reaction != null)
+    					{
                                 writer.WriteLine(reagent1.PracticalPaintName + " " + reagent2.PracticalPaintName + " " +
     							reaction.Red + " " + reaction.Green + " " + reaction.Blue);
+    					}
+    				}
+    			}
                         if (reaction != null)
+                        {
                             writer.WriteLine(reagent1.PracticalPaintName + " " + reagent2.PracticalPaintName + " " +
                             reaction.Red + " " + reaction.Green + " " + reaction.Blue);
+                        }
+                    }
+                }
+            }
+		}
+        }
         private void LoadRecipes(SortedDictionary<string, PaintRecipe> recipeDict, string filename, uint concentration)
+        {
             foreach (PaintRecipe recipe in recipeDict.Values)
+            {
                 recipe.Clear();
+            }
             string recipeFile = System.IO.Path.Combine(directory, filename);
             string line;
             Match match;
             bool inRecipe = false;
             PaintRecipe testRecipe = new PaintRecipe();
             testRecipe.Reactions = reactions;
             string currentRecipeColor = null;
             if (File.Exists(recipeFile))
+            {
                 using (StreamReader reader = new StreamReader(recipeFile))
+                {
-                    while ((line = reader.ReadLine()) != null)
+                    while ((line = reader.ReadLine()) != null)
+                    {
-                        match = recipeHeaderRegex.Match(line);
+                        match = recipeHeaderRegex.Match(line);
                         if (match.Success)
+                        {
                             if (testRecipe != null && currentRecipeColor != null)
+                            {
                                 if (testRecipe.IsValidForConcentration(concentration))
+                                {
                                     recipeDict[currentRecipeColor].CopyFrom(testRecipe);
+                                }
+                            }
                             testRecipe.Clear();
                             currentRecipeColor = match.Groups["colorname"].Value;
                             inRecipe = true;
@@ ... / @@ -483,36 +511,36 @@ namespace DesertPaintLab @@
         public void SaveRecipes()
+        {
             SaveRecipes(this.recipes, "dp_recipes.txt");
             SaveRecipes(this.ribbonRecipes, "dp_ribbons.txt");
+        }
         public void ExportWikiRecipes(string file)
+        {
             StreamWriter writer = new StreamWriter(file);
             ExportWikiFormat(writer, this.recipes);
+        }
         public void ExportWikiRibbons(string file)
+        {
             StreamWriter writer = new StreamWriter(file);
             ExportWikiFormat(writer, this.ribbonRecipes);
+        }
         public void ExportWikiRecipes(TextWriter writer)
+        {
             ExportWikiFormat(writer, this.recipes);
+        }
         public void ExportWikiRibbons(TextWriter writer)
+        {
             ExportWikiFormat(writer, this.ribbonRecipes);
+        }
         static public PaintColor RgbToHsv(PaintColor color)
+        {
             float h = 0;
             float s = 0;
             float r = color.Red / 255.0f;
             float g = color.Green / 255.0f;
             float b = color.Blue / 255.0f;
@@ ... / @@ -580,65 +608,65 @@ namespace DesertPaintLab @@
+                    }
                     else
+                    {
                         colorLine += "color: #000000;";
+                    }
                     colorLine += "\" | " + color.Name + " || ";
                     if (recipeDict.TryGetValue(color.Name, out recipe))
+                    {
                         foreach (PaintRecipe.RecipeIngredient ingredient in recipe.Ingredients)
+                        {
                             colorLine += " " + ingredient.ToString();
+                        }
+                    }
+                    }
                     else
+                    {
                         // no recipe
+                    }
                     colorLine += " || ";
                     if (recipe.CheckMissingReactions(ref missing) == true)
+                    {
                         colorLine += "Y";
+                    }
                     else
+                    {
                         colorLine += "N";
+                    }
                     colorLine += " || N";
                     writer.WriteLine(colorLine);
+                }
                 writer.WriteLine("|}");
+            }
+        }
 		public Reaction FindReaction(Reagent reagent1, Reagent reagent2)
+		{
         public Reaction FindReaction(Reagent reagent1, Reagent reagent2)
+        {
             return reactions.Find(reagent1, reagent2);
+		}
 		public void SetReaction(Reagent reagent1, Reagent reagent2, Reaction reaction)
+		{
+        }
         public void SetReaction(Reagent reagent1, Reagent reagent2, Reaction reaction)
+        {
             reactions.Set(reagent1, reagent2, reaction);
+		}
+        }
         public void ClearReaction(Reagent reagent1, Reagent reagent2)
+        {
             reactions.Remove(reagent1, reagent2);
+        }
         public void SetRecipe(PaintRecipe recipe)
+        {
             string colorName = Palette.FindNearest(recipe.ReactedColor);
             recipes[colorName].CopyFrom(recipe);
+        }
         public void SetRibbonRecipe(PaintRecipe recipe)
+        {
             string colorName = Palette.FindNearest(recipe.ReactedColor);
             ribbonRecipes[colorName].CopyFrom(recipe);
+        }
+	}
+    }
+}

0 comments (0 inline, 0 general)