Add filtermod for triggers

Author: timholy

src/parcel_snoopi_deep.jl

@@ -644,6 +644,14 @@ Return the MethodInstance `mi` of the caller in the selected stackframe in `itri
 """
 callerinstance(itrig::InferenceTrigger) = itrig.callerframes[end].linfo
 
+function callermodule(itrig::InferenceTrigger)
+    if !isempty(itrig.callerframes)
+        m = callerinstance(itrig).def
+        return isa(m, Module) ? m : m.module
+    end
+    return nothing
+end
+
 # Select the next (caller) frame that's a Julia (as opposed to C) frame; returns the stackframe and its index in bt, or nothing
 function next_julia_frame(bt, idx, Δ=1; methodinstanceonly::Bool=true, methodonly::Bool=true)
     while 1 <= idx+Δ <= length(bt)
@@ -856,6 +864,8 @@ Cthulhu.specTypes(itrig::InferenceTrigger) = Cthulhu.specTypes(Cthulhu.instance(
 Cthulhu.backedges(itrig::InferenceTrigger) = (itrig.callerframes,)
 Cthulhu.nextnode(itrig::InferenceTrigger, edge) = (ret = callingframe(itrig); return isempty(ret.callerframes) ? nothing : ret)
 
+filtermod(mod::Module, itrigs::AbstractVector{InferenceTrigger}) = filter(==(mod) ∘ callermodule, itrigs)
+
 ### inference trigger trees
 # good for organizing into "events"
 
@@ -1010,6 +1020,10 @@ function accumulate_by_source(itrigs::AbstractVector{InferenceTrigger})
     return sort([LocationTrigger(loc, itrigs_loc) for ((loc, _), itrigs_loc) in cs]; by=loctrig->length(loctrig.itrigs))
 end
 
+filtermod(mod::Module, loctrigs::AbstractVector{LocationTrigger}) = filter(loctrigs) do loctrig
+    any(==(mod) ∘ callermodule, loctrig.itrigs)
+end
+
 function linetable_match(linetable::Vector{Core.LineInfoNode}, sffile::String, sffunc::String, sfline::Int)
     idxs = Int[]
     for (idx, line) in enumerate(linetable)

test/snoopi_deep.jl

@@ -124,6 +124,8 @@ fdouble(x) = 2x
     @test occursin("myplus", string(MethodInstance(itrigs[2].node).def.name))
     itrigs = inference_triggers(tinf)
     itrig = only(itrigs)
+    @test filtermod(@__MODULE__, itrigs) == [itrig]
+    @test isempty(filtermod(Base, itrigs))
     io = IOBuffer()
     show(io, itrig)
     str = String(take!(io))
@@ -191,6 +193,8 @@ fdouble(x) = 2x
     loctrigs = accumulate_by_source(itrigs)
     show(io, loctrigs)
     @test any(str->occursin("4 callees from 2 callers", str), split(String(take!(io)), '\n'))
+    @test filtermod(Base, loctrigs) == loctrigs
+    @test isempty(filtermod(@__MODULE__, loctrigs))
 
     # Multiple callees on the same line
     fline(x) = 2*x[]