mirror of
https://github.com/hajimehoshi/ebiten.git
synced 2024-11-10 13:07:26 +01:00
128 lines
2.2 KiB
Go
128 lines
2.2 KiB
Go
// Copyright 2022 The Ebitengine Authors
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package shaderir_test
|
|
|
|
import (
|
|
"go/parser"
|
|
"go/token"
|
|
"testing"
|
|
|
|
"github.com/hajimehoshi/ebiten/v2/internal/shader"
|
|
"github.com/hajimehoshi/ebiten/v2/internal/shaderir"
|
|
)
|
|
|
|
func compileToIR(src []byte) (*shaderir.Program, error) {
|
|
fset := token.NewFileSet()
|
|
f, err := parser.ParseFile(fset, "", src, parser.AllErrors)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
ir, err := shader.Compile(fset, f, "Vertex", "Fragment", 0)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return ir, nil
|
|
}
|
|
|
|
func areIntSlicesEqual(a, b []int) bool {
|
|
if len(a) != len(b) {
|
|
return false
|
|
}
|
|
for i := range a {
|
|
if a[i] != b[i] {
|
|
return false
|
|
}
|
|
}
|
|
return true
|
|
}
|
|
|
|
func TestReachableUniformVariablesFromBlock(t *testing.T) {
|
|
src0 := `package main
|
|
|
|
var U0 float
|
|
var U1 float
|
|
|
|
func F0() float {
|
|
return U0
|
|
}
|
|
|
|
func F1() {
|
|
a := U0
|
|
_ = a
|
|
}
|
|
|
|
func F2() {
|
|
}
|
|
|
|
func F3() float {
|
|
return F0()
|
|
}
|
|
|
|
func F4() float {
|
|
return F0() + U1
|
|
}
|
|
|
|
func neverCalled() float {
|
|
return U0 + U1
|
|
}
|
|
`
|
|
|
|
cases := []struct {
|
|
source string
|
|
index int
|
|
expected []int
|
|
}{
|
|
{
|
|
source: src0,
|
|
index: 0,
|
|
expected: []int{0},
|
|
},
|
|
{
|
|
source: src0,
|
|
index: 1,
|
|
expected: []int{0},
|
|
},
|
|
{
|
|
source: src0,
|
|
index: 2,
|
|
expected: []int{},
|
|
},
|
|
{
|
|
source: src0,
|
|
index: 3,
|
|
expected: []int{0},
|
|
},
|
|
{
|
|
source: src0,
|
|
index: 4,
|
|
expected: []int{0, 1},
|
|
},
|
|
}
|
|
|
|
for _, c := range cases {
|
|
ir, err := compileToIR([]byte(c.source))
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
got := ir.ReachableUniformVariablesFromBlock(ir.Funcs[c.index].Block)
|
|
want := c.expected
|
|
if !areIntSlicesEqual(got, want) {
|
|
t.Errorf("test: %v, got: %v, want: %v", c, got, want)
|
|
}
|
|
}
|
|
}
|