AllocationSiteProfiler: speeding up copyExecutionStack calculcation

src/IllimaniProfiler-Tests/Behavior.extension.st

@@ -0,0 +1,8 @@
+Extension { #name : 'Behavior' }
+
+{ #category : '*IllimaniProfiler-Tests' }
+Behavior >> illNew [
+ "Mocking method for testing "
+
+ ^ self basicNew initialize
+]

src/IllimaniProfiler-Tests/IllAllocationSiteProfiler.extension.st

@@ -0,0 +1,7 @@
+Extension { #name : 'IllAllocationSiteProfiler' }
+
+{ #category : '*IllimaniProfiler-Tests' }
+IllAllocationSiteProfiler >> instrumentator: anInstrumentator [
+
+ instrumentator := anInstrumentator
+]

src/IllimaniProfiler-Tests/IllAllocationSiteProfilerTest.class.st

@@ -12,6 +12,22 @@ IllAllocationSiteProfilerTest >> profilerClass [
  ^ IllAllocationSiteProfiler
 ]
 
+{ #category : 'tests' }
+IllAllocationSiteProfilerTest >> testFilterContextIgnoringClass [
+
+ profiler
+ instrumentator: (IllMockInstrumentator onProfiler: profiler);
+ copyExecutionStack;
+ samplingRate: 1;
+ startProfiling.
+
+ Object illNew.
+
+ self assert: profiler objectAllocations first context receiver class equals: self class.
+ self assert: profiler objectAllocations first context homeMethod
+ equals: self class >> #testFilterContextIgnoringClass
+]
+
 { #category : 'tests' }
 IllAllocationSiteProfilerTest >> testProfileOnCopyExecutionStack [
 

src/IllimaniProfiler-Tests/IllMockInstrumentator.class.st

@@ -0,0 +1,19 @@
+Class {
+ #name : 'IllMockInstrumentator',
+ #superclass : 'IllInstrumentator',
+ #category : 'IllimaniProfiler-Tests-Allocations-Profiler',
+ #package : 'IllimaniProfiler-Tests',
+ #tag : 'Allocations-Profiler'
+}
+
+{ #category : 'initialization' }
+IllMockInstrumentator >> initializeMethodProxies [
+
+ proxyBasicNew := MpMethodProxy onMethod: Behavior >> #illNew handler: handler
+]
+
+{ #category : 'instrumentation' }
+IllMockInstrumentator >> proxyVariables [
+
+ ^ { proxyBasicNew }
+]

src/IllimaniProfiler/IllAllocationSiteProfiler.class.st

@@ -114,9 +114,8 @@ IllAllocationSiteProfiler >> filterContext: aContext ignoringClass: aClass [
  | sender |
  sender := aContext sender.
  [
- sender receiver class = aClass or: [
- sender receiver class = aClass class or: [
- classesToAvoidInTheContextSearch includes: sender receiver class ] ] ]
+ sender receiver class = aClass class or: [
+ classesToAvoidInTheContextSearch includes: sender receiver class ] ]
  whileTrue: [ sender := sender sender ].
  ^ sender
 ]
@@ -128,7 +127,6 @@ IllAllocationSiteProfiler >> initialize [
  objectAllocations := OrderedCollection new.
  copyExecutionStack := false.
  classesToAvoidInTheContextSearch := self defaultClassesToAvoidInTheContextSearch
- , (self defaultClassesToAvoidInTheContextSearch collect: [ :aClass | aClass class ])
 ]
 
 { #category : 'profiling' }