Create a matrix and heatmap from existing MTAnalysis

src/MuTalk-Model/MTAnalysis.class.st

@@ -208,7 +208,7 @@ MTAnalysis >> logger: anObject [
 	logger := anObject
 ]
 
-{ #category : 'as yet unclassified' }
+{ #category : 'accessing' }
 MTAnalysis >> methodSize [
 
 	^ (self modelClasses flatCollect: #methods) size

src/MuTalk-Utilities/MTAnalysis.extension.st

@@ -0,0 +1,17 @@
+Extension { #name : 'MTAnalysis' }
+
+{ #category : '*MuTalk-Utilities' }
+MTAnalysis >> generateHeatmap [
+
+	| matrix |
+	matrix := MTMatrix new analysis: self.
+	matrix generateHeatmap
+]
+
+{ #category : '*MuTalk-Utilities' }
+MTAnalysis >> generateMatrix [
+
+	| matrix |
+	matrix := MTMatrix new analysis: self.
+	matrix generateMatrix
+]

src/MuTalk-Utilities/MTMatrix.class.st

@@ -76,6 +76,17 @@ MTMatrix >> analysis [
 	^ analysis
 ]
 
+{ #category : 'accessing' }
+MTMatrix >> analysis: anAnalysis [
+
+	analysis := anAnalysis.
+	testClasses := (anAnalysis testCases groupedBy: [ :testCase |
+		                testCase testCaseClass ]) keys.
+	testCases := analysis testCases.
+	mutations := analysis mutations.
+	self fillMatrix
+]
+
 { #category : 'comparing' }
 MTMatrix >> booleanCollection1: collection1 equalsBooleanCollection2: collection2 [
 
@@ -100,12 +111,7 @@ MTMatrix >> booleanCollection1: collection1 includesBooleanCollection2: collecti
 MTMatrix >> build [
 
 	self runAnalysis.
-	mutations do: [ :mut |
-		self
-			at: mut
-			put:
-			(testCases collect: [ :test |
-				 (self failuresFor: mut) includes: test ]) ]
+	self fillMatrix
 ]
 
 { #category : 'accessing' }
@@ -195,6 +201,17 @@ MTMatrix >> failuresFor: aMutation andTestClass: aTestClass [
 		  testCase testCaseClass = aTestClass ]
 ]
 
+{ #category : 'building' }
+MTMatrix >> fillMatrix [
+
+	mutations do: [ :mut |
+		self
+			at: mut
+			put:
+			(testCases collect: [ :test |
+				 (self failuresFor: mut) includes: test ]) ]
+]
+
 { #category : 'rendering' }
 MTMatrix >> generateHeatmap [