diff options
| author | 2017-12-30 17:54:27 -0800 | |
|---|---|---|
| committer | 2018-01-02 18:23:43 -0800 | |
| commit | ee6143cde25cbe41df754337de89d608ca890bb1 (patch) | |
| tree | 7e972f8184630590cc38542c9cda9758cd7609f1 /android/mutator.go | |
| parent | b4330e222b8df54727d0054075d63f2dcdf0a959 (diff) | |
Add VisitDirectDepsWithTag
Add a method on ModuleContext and TopDownMutatorContext to visit
direct dependencies that have a given dependency tag.
Test: m checkbuild
Change-Id: Ib875563091dcae6b7282b3e3427d0eb07d8c8af5
Diffstat (limited to 'android/mutator.go')
| -rw-r--r-- | android/mutator.go | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/android/mutator.go b/android/mutator.go index 8de57e1b6..2f13f6caf 100644 --- a/android/mutator.go +++ b/android/mutator.go @@ -127,6 +127,7 @@ type TopDownMutatorContext interface { GetDirectDep(name string) (blueprint.Module, blueprint.DependencyTag) VisitDirectDeps(visit func(Module)) + VisitDirectDepsWithTag(tag blueprint.DependencyTag, visit func(Module)) VisitDirectDepsIf(pred func(Module) bool, visit func(Module)) VisitDepsDepthFirst(visit func(Module)) VisitDepsDepthFirstIf(pred func(Module) bool, visit func(Module)) @@ -230,6 +231,16 @@ func (a *androidTopDownMutatorContext) VisitDirectDeps(visit func(Module)) { }) } +func (a *androidTopDownMutatorContext) VisitDirectDepsWithTag(tag blueprint.DependencyTag, visit func(Module)) { + a.TopDownMutatorContext.VisitDirectDeps(func(module blueprint.Module) { + if aModule, _ := module.(Module); aModule != nil { + if a.TopDownMutatorContext.OtherModuleDependencyTag(aModule) == tag { + visit(aModule) + } + } + }) +} + func (a *androidTopDownMutatorContext) VisitDirectDepsIf(pred func(Module) bool, visit func(Module)) { a.TopDownMutatorContext.VisitDirectDepsIf( // pred |