mth/ebiten
1
0
Fork 0
mirror of https://github.com/hajimehoshi/ebiten.git synced 2024-09-20 05:12:18 +02:00
Code Issues Packages Projects Releases Wiki Activity

internal/shaderir: refactoring

Browse Source
This commit is contained in:
Hajime Hoshi 2022-11-03 18:16:54 +09:00
parent 7dbb078a9e
commit 0762db3b3e
1 changed files with 31 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
internal/shaderir/program.go
View File

@ -372,15 +372,31 @@ func (p *Program) ReachableFuncsFromFragmentShader() []*Func {
return p.reachableFuncsFromBlockEntryPoint(p.FragmentFunc.Block) return p.reachableFuncsFromBlockEntryPoint(p.FragmentFunc.Block)
} }
func (p *Program) reachableFuncsFromBlockEntryPoint(b *Block) []*Func { func (p *Program) reachableFuncsFromBlockEntryPoint(block *Block) []*Func {
indexToFunc := map[int]*Func{} indexToFunc := map[int]*Func{}
for _, f := range p.Funcs { for _, f := range p.Funcs {
f := f f := f
indexToFunc[f.Index] = &f indexToFunc[f.Index] = &f
} }
visited := map[int]struct{}{} visited := map[int]struct{}{}
indices := reachableFuncIndicesFromBlock(b, indexToFunc, visited) var indices []int
var f func(expr *Expr)
f = func(expr *Expr) {
if expr.Type != FunctionExpr {
return
}
if _, ok := visited[expr.Index]; ok {
return
}
indices = append(indices, expr.Index)
visited[expr.Index] = struct{}{}
walkExprs(f, indexToFunc[expr.Index].Block)
}
walkExprs(f, block)
sort.Ints(indices) sort.Ints(indices)
funcs := make([]*Func, 0, len(indices)) funcs := make([]*Func, 0, len(indices))
for _, i := range indices { for _, i := range indices {
funcs = append(funcs, indexToFunc[i]) funcs = append(funcs, indexToFunc[i])
@ -388,36 +404,26 @@ func (p *Program) reachableFuncsFromBlockEntryPoint(b *Block) []*Func {
return funcs return funcs
} }
func reachableFuncIndicesFromBlock(b *Block, indexToFunc map[int]*Func, visited map[int]struct{}) []int { func walkExprs(f func(expr *Expr), block *Block) {
if b == nil { if block == nil {
return nil return
} }
for _, s := range block.Stmts {
var fs []int
for _, s := range b.Stmts {
for _, e := range s.Exprs { for _, e := range s.Exprs {
fs = append(fs, reachableFuncIndicesFromExpr(&e, indexToFunc, visited)...) walkExprsInExpr(f, &e)
} }
for _, bb := range s.Blocks { for _, b := range s.Blocks {
fs = append(fs, reachableFuncIndicesFromBlock(bb, indexToFunc, visited)...) walkExprs(f, b)
} }
} }
return fs
} }
func reachableFuncIndicesFromExpr(e *Expr, indexToFunc map[int]*Func, visited map[int]struct{}) []int { func walkExprsInExpr(f func(expr *Expr), expr *Expr) {
var fs []int if expr == nil {
return
if e.Type == FunctionExpr {
if _, ok := visited[e.Index]; !ok {
fs = append(fs, e.Index)
visited[e.Index] = struct{}{}
fs = append(fs, reachableFuncIndicesFromBlock(indexToFunc[e.Index].Block, indexToFunc, visited)...)
}
} }
for _, ee := range e.Exprs { f(expr)
fs = append(fs, reachableFuncIndicesFromExpr(&ee, indexToFunc, visited)...) for _, e := range expr.Exprs {
walkExprsInExpr(f, &e)
} }
return fs
} }