ATITD-Tools/Desert-Paint-Lab Changeset

Changeset - adaf02e991b4

Parent rev.

Child rev.

[Not reviewed]

default

0 1 0

Jason Maltzen (jmaltzen) - 9 years ago 2015-11-19 03:35:37
jason.maltzen@unsanctioned.net

Fix up screen scraping code to work on retina displays (by converting to an Image and then to a Pixbuf)

1 file changed with 11 insertions and 3 deletions:

MainWindow.cs

0 comments (0 inline, 0 general)

MainWindow.cs

➞

Show inline comments

@@ ... / @@ -48,98 +48,104 @@ public partial class MainWindow : Gtk.Wi @@
 	int screenWidth = 0;
 	int screenHeight = 0;
 	int pixelMultiplier = 1;
 	Gdk.Window rootWindow = null;
 	Gdk.Pixbuf screenBuffer = null;
 	Reagent reagent1 = null;
 	Reagent reagent2 = null;
 	Reagent reagent3 = null;
 	public bool ShouldShutDown
+	{
 		get
+		{
 			return shouldShutDown;
+		}
+	}
 	public MainWindow () : base(Gtk.WindowType.Toplevel)
+	{
 		appDataPath = System.IO.Path.Combine(
 			Environment.GetFolderPath(Environment.SpecialFolder.LocalApplicationData),
 		    "DesertPaintLab");
 		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);
+			}
+		}
 		Palette.Load(System.IO.Path.Combine(
 				System.IO.Path.GetDirectoryName(System.Reflection.Assembly.GetExecutingAssembly().Location),
 		    	"colors.txt"));
 		Build();
 		if (unmodifiedSwatch != null)
+		{
 		unmodifiedSwatch.Clear();
+		}
 		if (reactionSwatch != null)
+		{
 		reactionSwatch.Clear();
+		}
 		// get the root window
 		rootWindow = Gdk.Global.DefaultRootWindow;
 		// get its width and height
 		rootWindow.GetSize(out screenWidth, out screenHeight);
 		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();
 		screenBuffer = new Gdk.Pixbuf(Gdk.Colorspace.Rgb, false, 8, screenWidth, screenHeight);
 		swatchHeight    *= pixelMultiplier;
 		colorBarWidth   *= pixelMultiplier;
 		redBarSpacing   *= pixelMultiplier;
 		greenBarSpacing *= pixelMultiplier;
 		blueBarSpacing  *= 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;
+	}
@@ ... / @@ -436,98 +442,100 @@ public partial class MainWindow : Gtk.Wi @@
+							}
+						}
+					}
+				}
 				SetExpectedColor((byte)Math.Round((float)expRedSum / (float)pigmentCount),
 					(byte)Math.Round((float)expGreenSum / (float)pigmentCount),
 					(byte)Math.Round((float)expBlueSum / (float)pigmentCount));
 				if (reactionKnown)
+				{
 					reactedColor.Red = (byte)Math.Min(255, Math.Max(0, expectedColor.Red + reactRedSum));
 					reactedColor.Green = (byte)Math.Min(255, Math.Max(0, expectedColor.Green + reactGreenSum));
 					reactedColor.Blue = (byte)Math.Min(255, Math.Max(0, expectedColor.Blue + reactBlueSum));
 					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.
 		byte r, g, b;
 		int pixelStart, otherPixelStart;
 		bool colorMatch;
 		screenBuffer.GetFromDrawable(rootWindow,
 			rootWindow.Colormap, 0, 0, 0, 0, screenWidth, screenHeight);
 		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;
 		for (int x = 0; x < screenWidth - colorBarWidth; ++x)
+		{
 			for (int y = 0; y < (screenHeight - 53); ++y)
+			{
 				// Look for the color swatch.
 				pixelStart = (y * stride) + (x * 3);
 				r = pixBytes[pixelStart];
 				g = pixBytes[pixelStart + 1];
 				b = pixBytes[pixelStart + 2];
 				// 1.) Check if this is a dark pixel.
 				if ((r < 0x46) && (g < 0x46) && (b < 0x46))
+				{
 					// 2.) Check the pixel above it,
 					// to see if it's from the papy texture.
 					otherPixelStart = pixelStart - stride;
 					if ((otherPixelStart >= 0) &&
 					    IsPapyTexture(pixBytes[otherPixelStart++],
 					                   pixBytes[otherPixelStart++],
 					                   pixBytes[otherPixelStart]))
+					{
 						// 3.) Check the pixel below where the swatch should be,
 						// to see if it's also from the papy texture.
 						otherPixelStart = pixelStart + (stride * swatchHeight);
 						if (IsPapyTexture(pixBytes[otherPixelStart++],
 					                   pixBytes[otherPixelStart++],
 					                   pixBytes[otherPixelStart]))
+						{
 							// pixBytes[pixelStart] = 0xFF;
 							// pixBytes[pixelStart + 1] = 0x00;
 							// pixBytes[pixelStart + 2] = 0xFF;
 							// 4.) Scan the left border of the potential swatch
 							// location.
 							colorMatch = true;
 							for (int i = 1; i < swatchHeight - 2; ++i)
+							{
 								otherPixelStart = pixelStart + (stride * i);
 								if ((Math.Abs(r - pixBytes[otherPixelStart++]) > colorTolerance) ||
 								    (Math.Abs(g - pixBytes[otherPixelStart++]) > colorTolerance) ||
 								    (Math.Abs(b - pixBytes[otherPixelStart]) > colorTolerance))
+								{
 									colorMatch = false;
 									break;
+								}
@@ ... / @@ -537,97 +545,97 @@ public partial class MainWindow : Gtk.Wi @@
+							{
 								// WE FOUND THE SWATCH!
 								// Now we know where the color bars are.
 								otherPixelStart = pixelStart + (redBarSpacing * stride);
 								int pixelCount = 0;
 								while ((pixBytes[otherPixelStart] > 0x9F) &&
 								       (pixBytes[otherPixelStart + 1] < 0x62) &&
 								       (pixBytes[otherPixelStart + 2]  < 0x62))
+								{
 									pixelCount++;
 									// pixBytes[otherPixelStart] = 0x00;
 									// pixBytes[otherPixelStart + 1] = 0xFF;
 									// pixBytes[otherPixelStart + 2] = 0xFF;
 									otherPixelStart += 3;
+								}
 								reactedColor.Red = (byte)Math.Round((float)pixelCount * 255f / (float)colorBarWidth);
 								otherPixelStart = pixelStart + (greenBarSpacing * stride);
 								pixelCount = 0;
 								while ((pixBytes[otherPixelStart] < 0x62) &&
 								       (pixBytes[otherPixelStart + 1] > 0x9F) &&
 								       (pixBytes[otherPixelStart + 2] < 0x62))
+								{
 									pixelCount++;
 									// pixBytes[otherPixelStart] = 0x00;
 									// pixBytes[otherPixelStart + 1] = 0xFF;
 									// pixBytes[otherPixelStart + 2] = 0xFF;
 									otherPixelStart += 3;
+								}
 								reactedColor.Green = (byte)Math.Round((float)pixelCount * 255f / (float)colorBarWidth);
 								otherPixelStart = pixelStart + (blueBarSpacing * stride);
 								pixelCount = 0;
 								while ((pixBytes[otherPixelStart] < 0x62) &&
 									(pixBytes[otherPixelStart + 1] < 0x62) &&
 									(pixBytes[otherPixelStart + 2] > 0x9F))
+								{
 									pixelCount++;
 									// pixBytes[otherPixelStart] = 0x00;
 									// pixBytes[otherPixelStart + 1] = 0xFF;
 									// pixBytes[otherPixelStart + 2] = 0xFF;
 									otherPixelStart += 3;
+								}
 								reactedColor.Blue = (byte)Math.Round((float)pixelCount * 255f / (float)colorBarWidth);
+								// write out the screenshot
 								// screenBuffer.Save("screenshot.png", "png");
 								return true;
+							}
+						}
+					}
+				}
+			}
+		}
 		// screenBuffer.Save("screenshot.png", "png");
 		return false;
+	}
 	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,

0 comments (0 inline, 0 general)