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

@@ ... / @@ -72,50 +72,56 @@ public partial class MainWindow : Gtk.Wi @@
 			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;
@@ ... / @@ -460,50 +466,52 @@ public partial class MainWindow : Gtk.Wi @@
+	{
 		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]))
+					{
@@ ... / @@ -561,49 +569,49 @@ public partial class MainWindow : Gtk.Wi @@
 									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)

0 comments (0 inline, 0 general)