From 0456b543855ca32b31d854181dc7b4e75d5fc8f7 Mon Sep 17 00:00:00 2001
From: Daniel Schramm <xschramm@fi.muni.cz>
Date: Wed, 6 Oct 2021 13:25:32 +0200
Subject: [PATCH] Minor refactoring

---
 .../java/cz/fidentis/analyst/distance/DistanceAction.java   | 6 ++----
 .../fidentis/analyst/registration/RegistrationAction.java   | 6 ++----
 2 files changed, 4 insertions(+), 8 deletions(-)

diff --git a/GUI/src/main/java/cz/fidentis/analyst/distance/DistanceAction.java b/GUI/src/main/java/cz/fidentis/analyst/distance/DistanceAction.java
index 4fe3acb0..a2e6be19 100644
--- a/GUI/src/main/java/cz/fidentis/analyst/distance/DistanceAction.java
+++ b/GUI/src/main/java/cz/fidentis/analyst/distance/DistanceAction.java
@@ -151,11 +151,9 @@ public class DistanceAction extends ControlPanelAction {
         
         switch (action) {
             case DistancePanel.ACTION_COMMAND_SHOW_HIDE_HEATMAP:
-                if (((JToggleButton) ae.getSource()).isSelected()) {
+                heatmapRender = ((JToggleButton) ae.getSource()).isSelected();
+                if (heatmapRender) {
                     updateHausdorffDistanceInformation();
-                    heatmapRender = true;
-                } else {
-                    heatmapRender = false;
                 }
                 getSecondaryDrawableFace().setRenderHeatmap(heatmapRender);
                 break;
diff --git a/GUI/src/main/java/cz/fidentis/analyst/registration/RegistrationAction.java b/GUI/src/main/java/cz/fidentis/analyst/registration/RegistrationAction.java
index 360a964a..7cc57136 100644
--- a/GUI/src/main/java/cz/fidentis/analyst/registration/RegistrationAction.java
+++ b/GUI/src/main/java/cz/fidentis/analyst/registration/RegistrationAction.java
@@ -232,12 +232,10 @@ public class RegistrationAction extends ControlPanelAction {
                 }
                 break;
             case RegistrationPanel.ACTION_COMMAND_HD_HEATMAP:
-                if (((JToggleButton) ae.getSource()).isSelected()) {
+                heatmapRender = ((JToggleButton) ae.getSource()).isSelected();
+                if (heatmapRender) {
                     calculateHausdorffDistance();
                     setHeatmap();
-                    heatmapRender = true;
-                } else {
-                    heatmapRender = false;
                 }
                 getSecondaryDrawableFace().setRenderHeatmap(heatmapRender);
                 break;
-- 
GitLab