More customizable heatmaps

src/MuTalk-Examples/MTAnalysis.extension.st

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

src/MuTalk-Utilities/MTAnalysis.extension.st

@@ -1,11 +1,27 @@
 Extension { #name : 'MTAnalysis' }
 
 { #category : '*MuTalk-Utilities' }
-MTAnalysis >> generateHeatmap [
+MTAnalysis >> generateHeatmapByClass [
 
 	| matrix |
 	matrix := MTMatrix new analysis: self.
-	matrix generateHeatmap
+	matrix generateHeatmapByClass
+]
+
+{ #category : '*MuTalk-Utilities' }
+MTAnalysis >> generateHeatmapByMethod [
+
+	| matrix |
+	matrix := MTMatrix new analysis: self.
+	matrix generateHeatmapByMethod
+]
+
+{ #category : '*MuTalk-Utilities' }
+MTAnalysis >> generateHeatmapByOperator [
+
+	| matrix |
+	matrix := MTMatrix new analysis: self.
+	matrix generateHeatmapByOperator
 ]
 
 { #category : '*MuTalk-Utilities' }

src/MuTalk-Utilities/MTMatrix.class.st

@@ -213,11 +213,31 @@ MTMatrix >> fillMatrix [
 ]
 
 { #category : 'rendering' }
-MTMatrix >> generateHeatmap [
+MTMatrix >> generateHeatmapByClass [
+
+	self generateHeatmapWithMutationsGroupedBy: [ :mutation |
+		mutation originalClass ]
+]
+
+{ #category : 'rendering' }
+MTMatrix >> generateHeatmapByMethod [
+
+	self generateHeatmapWithMutationsGroupedBy: [ :mutation |
+		mutation originalMethod name ]
+]
+
+{ #category : 'rendering' }
+MTMatrix >> generateHeatmapByOperator [
+
+	self generateHeatmapWithMutationsGroupedBy: [ :mutation |
+		mutation operator species ]
+]
+
+{ #category : 'rendering' }
+MTMatrix >> generateHeatmapWithMutationsGroupedBy: aBlock [
 
 	| heatmap mutationDictionary testDictionary |
-	mutationDictionary := mutations groupedBy: [ :mutation |
-		                      mutation operator species ].
+	mutationDictionary := mutations groupedBy: aBlock.
 	testDictionary := testCases groupedBy: #testCaseClass.
 
 	heatmap := self initializeHeatmap.