Working on the message rewriting

jordanmontt · Dec 20, 2024 · 9e55a79 · 9e55a79
1 parent c8fa82e
commit 9e55a79
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 11 deletions.
diff --git a/src/IllimaniProfiler/IllAnalyzer.class.st b/src/IllimaniProfiler/IllAnalyzer.class.st
@@ -113,13 +113,15 @@ IllAnalyzer >> cleanKernelAllocationSites: aCollection [
 { #category : 'api' }
 IllAnalyzer >> doMagic [
 
-	| toPretenureSites |
+	| toPretenureSites allocationSendersList |
 	self classifyAllocationSites.
 	toPretenureSites := self pruneAllocationSites select: #isLongOrImmortal.
 	toPretenureSites := self cleanKernelAllocationSites: toPretenureSites.
 	toPretenureSites := self removeDoItAllocationSite: toPretenureSites.
-
-	pretenurator doMagic: toPretenureSites.
+
+	allocationSendersList := toPretenureSites values collect: [ :allocSiteBin |
+		                         allocSiteBin allObjects first allocationSenders asMethodWithOffsetObjects ].
+	pretenurator doMagic: allocationSendersList.
 	^ pretenurator
 ]
 

diff --git a/src/IllimaniProfiler/IllPretenurator.class.st b/src/IllimaniProfiler/IllPretenurator.class.st
@@ -12,20 +12,15 @@ Class {
 { #category : 'as yet unclassified' }
 IllPretenurator >> doMagic: toPretenureSites [
 
-	[
-	toPretenureSites valuesDo: [ :allocSiteBin |
+	toPretenureSites do: [ :allocationSenders |
 		| sct allocationSenders initialCaller sendersToTransform |
-		allocationSenders := (allocSiteBin allObjects collect: #allocationSenders as: Set) anyOne
-			                     asMethodWithOffsetObjects.
 		initialCaller := allocationSenders last.
 		sendersToTransform := allocationSenders allButLast.
-
+		
 		sct := SendersChainTransformer new.
 		codeTransformers add: sct.
 		sct backupSenders: sendersToTransform caller: initialCaller.
-		sct transformCallingSenders: sendersToTransform initialCaller: initialCaller ] ]
-
-	ensure: [ codeTransformers do: #restoreOriginalState ]
+		sct transformCallingSenders: sendersToTransform initialCaller: initialCaller ]
 ]
 
 { #category : 'initialization' }
@@ -34,3 +29,9 @@ IllPretenurator >> initialize [
 	super initialize.
 	codeTransformers := OrderedCollection new
 ]
+
+{ #category : 'as yet unclassified' }
+IllPretenurator >> restoreOriginalState [
+
+	codeTransformers do: #restoreOriginalState
+]