refactor heatmap initialization

DurieuxPol · Feb 15, 2024 · e73f12e · e73f12e
1 parent 4bcfc5e
commit e73f12e
Showing 1 changed file with 14 additions and 10 deletions.
diff --git a/src/MuTalk-Utilities/MTMatrix.class.st b/src/MuTalk-Utilities/MTMatrix.class.st
@@ -193,16 +193,7 @@ MTMatrix >> generateHeatmap [
 		                      mutation operator species ].
 	testDictionary := testCases groupedBy: #testCaseClass.
 
-	heatmap := RSHeatmap new.
-	heatmap colorPalette: (NSScale linear range: {
-				 Color gray.
-				 Color purple muchDarker }).
-	heatmap labelShapeCell labelShape
-		border: (RSBorder new
-				 color: Color black;
-				 width: 0.2);
-		color: Color white.
-	heatmap shape extent: 50 @ 20.
+	heatmap := self initializeHeatmap.
 
 	heatmap objectsX: testClasses.
 	heatmap objectsY: mutationDictionary keys.
@@ -236,6 +227,19 @@ MTMatrix >> includedMutants [
 	^ includedMutants
 ]
 
+{ #category : 'initialization' }
+MTMatrix >> initializeHeatmap [
+
+	| heatmap |
+	heatmap := RSHeatmap new.
+	heatmap colorPalette: (NSScale linear range: {
+				 Color red.
+				 Color red muchLighter }).
+	heatmap labelShapeCell labelShape bold.
+	heatmap shape extent: 50 @ 20.
+	^ heatmap
+]
+
 { #category : 'comparing' }
 MTMatrix >> mutant1: mutant1 equalsMutant2: mutant2 [