changed heatmaps to be able to group mutations by whichever block is given

Author: DurieuxPol

src/MuTalk-Examples/MTAnalysis.extension.st

@@ -44,7 +44,7 @@ MTAnalysis class >> exampleWithHeatmap [
 	| matrix |
 	matrix := MTMatrix forClasses: { MyVehicle }.
 	matrix build.
-	matrix generateHeatmap
+	matrix generateHeatmapWithMutationsGroupedByOperator
 ]
 
 { #category : '*MuTalk-Examples' }

src/MuTalk-Utilities/MTMatrix.class.st

@@ -196,11 +196,10 @@ MTMatrix >> failuresFor: aMutation andTestClass: aTestClass [
 ]
 
 { #category : 'rendering' }
-MTMatrix >> generateHeatmap [
+MTMatrix >> generateHeatmapWithMutationsGroupedBy: aBlock [
 
 	| heatmap mutationDictionary testDictionary |
-	mutationDictionary := mutations groupedBy: [ :mutation |
-		                      mutation operator species ].
+	mutationDictionary := mutations groupedBy: aBlock.
 	testDictionary := testCases groupedBy: #testCaseClass.
 
 	heatmap := self initializeHeatmap.
@@ -216,6 +215,27 @@ MTMatrix >> generateHeatmap [
 	heatmap open
 ]
 
+{ #category : 'rendering' }
+MTMatrix >> generateHeatmapWithMutationsGroupedByClass [
+
+	self generateHeatmapWithMutationsGroupedBy: [ :mutation |
+		mutation originalClass ]
+]
+
+{ #category : 'rendering' }
+MTMatrix >> generateHeatmapWithMutationsGroupedByMethod [
+
+	self generateHeatmapWithMutationsGroupedBy: [ :mutation |
+		mutation originalMethod name ]
+]
+
+{ #category : 'rendering' }
+MTMatrix >> generateHeatmapWithMutationsGroupedByOperator [
+
+	self generateHeatmapWithMutationsGroupedBy: [ :mutation |
+		mutation operator species ]
+]
+
 { #category : 'rendering' }
 MTMatrix >> generateMatrix [