From f7e10e594b06c9bd270927aa433ed2436ac90b78 Mon Sep 17 00:00:00 2001
From: dragonmacher <48328597+dragonmacher@users.noreply.github.com>
Date: Fri, 18 Nov 2022 18:27:11 -0500
Subject: [PATCH] GP-2861 - Updated Nimbus selected tree row painting to honor
 theme settings

---
 .../generic/theme/laf/GNimbusLookAndFeel.java | 11 +++
 .../theme/laf/nimbus/SelectedTreePainter.java | 73 +++++++++++++++++++
 2 files changed, 84 insertions(+)
 create mode 100644 Ghidra/Framework/Generic/src/main/java/generic/theme/laf/nimbus/SelectedTreePainter.java

diff --git a/Ghidra/Framework/Generic/src/main/java/generic/theme/laf/GNimbusLookAndFeel.java b/Ghidra/Framework/Generic/src/main/java/generic/theme/laf/GNimbusLookAndFeel.java
index e3f4528c02..00867d7e32 100644
--- a/Ghidra/Framework/Generic/src/main/java/generic/theme/laf/GNimbusLookAndFeel.java
+++ b/Ghidra/Framework/Generic/src/main/java/generic/theme/laf/GNimbusLookAndFeel.java
@@ -24,6 +24,7 @@ import javax.swing.plaf.FontUIResource;
 import javax.swing.plaf.nimbus.NimbusLookAndFeel;
 
 import generic.theme.*;
+import generic.theme.laf.nimbus.SelectedTreePainter;
 
 /**
  * Extends the {@link NimbusLookAndFeel} to intercept the {@link #getDefaults()}. Nimbus does 
@@ -49,6 +50,9 @@ public class GNimbusLookAndFeel extends NimbusLookAndFeel {
 	@Override
 	public UIDefaults getDefaults() {
 		UIDefaults defaults = super.getDefaults();
+
+		installCustomPainters(defaults);
+
 		GThemeValueMap javaDefaults = extractJavaDefaults(defaults);
 
 		// replace all colors with GColors
@@ -78,6 +82,13 @@ public class GNimbusLookAndFeel extends NimbusLookAndFeel {
 		return defaults;
 	}
 
+	private void installCustomPainters(UIDefaults defaults) {
+		defaults.put("Tree:TreeCell[Enabled+Selected].backgroundPainter",
+			new SelectedTreePainter());
+		defaults.put("Tree:TreeCell[Focused+Selected].backgroundPainter",
+			new SelectedTreePainter());
+	}
+
 	protected GThemeValueMap extractJavaDefaults(UIDefaults defaults) {
 		GThemeValueMap javaDefaults = new GThemeValueMap();
 
diff --git a/Ghidra/Framework/Generic/src/main/java/generic/theme/laf/nimbus/SelectedTreePainter.java b/Ghidra/Framework/Generic/src/main/java/generic/theme/laf/nimbus/SelectedTreePainter.java
new file mode 100644
index 0000000000..2b5cd731c5
--- /dev/null
+++ b/Ghidra/Framework/Generic/src/main/java/generic/theme/laf/nimbus/SelectedTreePainter.java
@@ -0,0 +1,73 @@
+/* ###
+ * IP: GHIDRA
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ * 
+ *      http://www.apache.org/licenses/LICENSE-2.0
+ * 
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package generic.theme.laf.nimbus;
+
+import java.awt.*;
+import java.awt.geom.Rectangle2D;
+
+import javax.swing.JComponent;
+import javax.swing.plaf.nimbus.AbstractRegionPainter;
+
+import generic.theme.GColor;
+
+/**
+ * Nimbus selected tree row painter
+ */
+public class SelectedTreePainter extends AbstractRegionPainter {
+
+	private PaintContext paintContext = new MyPaintContext();
+	private Rectangle2D shape = new Rectangle2D.Float();
+	private Color selectionColor;
+
+	@Override
+	protected PaintContext getPaintContext() {
+		return paintContext;
+	}
+
+	@Override
+	protected void doPaint(Graphics2D g, JComponent c, int width, int height,
+			Object[] extendedCacheKeys) {
+		lazyLoadColor();
+		updateShape();
+		g.setPaint(selectionColor);
+		g.fill(shape);
+	}
+
+	private void lazyLoadColor() {
+		//
+		// This class gets created before the theme system is full bootstrapped.  By lazy loading,
+		// we ensure the property is available after the theme system is loaded.
+		//
+		if (selectionColor == null) {
+			selectionColor = new GColor("color.bg.tree.selected");
+		}
+	}
+
+	private void updateShape() {
+		float x = 0;
+		float y = 0;
+		float w = decodeX(3.0f);
+		float h = decodeY(3.0f);
+		shape.setRect(x, y, w, h);
+	}
+
+	private class MyPaintContext extends PaintContext {
+		public MyPaintContext() {
+			super(new Insets(5, 5, 5, 5), new Dimension(100, 30),
+				false, CacheMode.NO_CACHING, 1.0, 1.0);
+		}
+	}
+}