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: use the append pattern

Browse Source
This commit is contained in:
Hajime Hoshi 2023-05-04 14:34:09 +09:00
parent e4c9d6705c
commit ceb2bfc89c
3 changed files with 15 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
internal/shaderir/export_test.go
View File

@ -14,6 +14,6 @@
package shaderir
func (p *Program) ReachableUniformVariablesFromBlock(block *Block) []int {
return p.reachableUniformVariablesFromBlock(block)
func (p *Program) AppendReachableUniformVariablesFromBlock(indices []int, block *Block) []int {
return p.appendReachableUniformVariablesFromBlock(indices, block)
}

23
internal/shaderir/program.go
View File

@ -436,7 +436,7 @@ func walkExprsInExpr(f func(expr *Expr), expr *Expr) {
}
}
func (p *Program) reachableUniformVariablesFromBlock(block *Block) []int {
func (p *Program) appendReachableUniformVariablesFromBlock(indices []int, block *Block) []int {
indexToFunc := map[int]*Func{}
for _, f := range p.Funcs {
f := f
@ -444,12 +444,12 @@ func (p *Program) reachableUniformVariablesFromBlock(block *Block) []int {
}
visitedFuncs := map[int]struct{}{}
indices := map[int]struct{}{}
indicesSet := map[int]struct{}{}
var f func(expr *Expr)
f = func(expr *Expr) {
switch expr.Type {
case UniformVariable:
indices[expr.Index] = struct{}{}
indicesSet[expr.Index] = struct{}{}
case FunctionExpr:
if _, ok := visitedFuncs[expr.Index]; ok {
return
@ -460,24 +460,21 @@ func (p *Program) reachableUniformVariablesFromBlock(block *Block) []int {
}
walkExprs(f, block)
is := make([]int, 0, len(indices))
for i := range indices {
is = append(is, i)
for i := range indicesSet {
indices = append(indices, i)
}
sort.Ints(is)
return is
return indices
}
// FilterUniformVariables replaces uniform variables with nil when they are not used.
// By minimizing uniform variables, more commands can be merged in the graphicscommand package.
func (p *Program) FilterUniformVariables(uniforms []uint32) {
if p.uniformMask == nil {
indices := p.appendReachableUniformVariablesFromBlock(nil, p.VertexFunc.Block)
indices = p.appendReachableUniformVariablesFromBlock(indices, p.FragmentFunc.Block)
reachableUniforms := make([]bool, len(p.Uniforms))
for _, i := range p.reachableUniformVariablesFromBlock(p.VertexFunc.Block) {
reachableUniforms[i] = true
}
for _, i := range p.reachableUniformVariablesFromBlock(p.FragmentFunc.Block) {
reachableUniforms[i] = true
for _, idx := range indices {
reachableUniforms[idx] = true
}
for i, typ := range p.Uniforms {
fs := make([]uint32, typ.Uint32Count())

4
internal/shaderir/reachable_test.go
View File

@ -15,6 +15,7 @@
package shaderir_test
import (
"sort"
"testing"
"github.com/hajimehoshi/ebiten/v2/internal/shader"
@ -105,7 +106,8 @@ func neverCalled() float {
if err != nil {
t.Fatal(err)
}
got := ir.ReachableUniformVariablesFromBlock(ir.Funcs[c.index].Block)
got := ir.AppendReachableUniformVariablesFromBlock(nil, ir.Funcs[c.index].Block)
sort.Ints(got)
want := c.expected
if !areIntSlicesEqual(got, want) {
t.Errorf("test: %v, got: %v, want: %v", c, got, want)