ebiten/internal/testing/shader.go

433 lines
9.8 KiB
Go
Raw Normal View History

// Copyright 2020 The Ebiten 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 testing
import (
2020-06-20 19:26:59 +02:00
"go/constant"
"github.com/hajimehoshi/ebiten/internal/graphics"
"github.com/hajimehoshi/ebiten/internal/shaderir"
)
2020-05-24 15:15:50 +02:00
var (
projectionMatrix = shaderir.Expr{
Type: shaderir.Call,
Exprs: []shaderir.Expr{
{
Type: shaderir.BuiltinFuncExpr,
BuiltinFunc: shaderir.Mat4F,
},
{
Type: shaderir.Binary,
Op: shaderir.Div,
Exprs: []shaderir.Expr{
{
2020-06-20 19:26:59 +02:00
Type: shaderir.NumberExpr,
Const: constant.MakeFloat64(2),
ConstType: shaderir.ConstTypeFloat,
},
{
Type: shaderir.FieldSelector,
Exprs: []shaderir.Expr{
{
Type: shaderir.UniformVariable,
Index: 0,
},
{
Type: shaderir.SwizzlingExpr,
Swizzling: "x",
},
},
},
},
},
{
2020-06-20 19:26:59 +02:00
Type: shaderir.NumberExpr,
Const: constant.MakeFloat64(0),
ConstType: shaderir.ConstTypeFloat,
},
{
2020-06-20 19:26:59 +02:00
Type: shaderir.NumberExpr,
Const: constant.MakeFloat64(0),
ConstType: shaderir.ConstTypeFloat,
},
{
2020-06-20 19:26:59 +02:00
Type: shaderir.NumberExpr,
Const: constant.MakeFloat64(0),
ConstType: shaderir.ConstTypeFloat,
},
{
2020-06-20 19:26:59 +02:00
Type: shaderir.NumberExpr,
Const: constant.MakeFloat64(0),
ConstType: shaderir.ConstTypeFloat,
},
{
Type: shaderir.Binary,
Op: shaderir.Div,
Exprs: []shaderir.Expr{
{
2020-06-20 19:26:59 +02:00
Type: shaderir.NumberExpr,
Const: constant.MakeFloat64(2),
ConstType: shaderir.ConstTypeFloat,
},
{
Type: shaderir.FieldSelector,
Exprs: []shaderir.Expr{
{
Type: shaderir.UniformVariable,
Index: 0,
},
{
Type: shaderir.SwizzlingExpr,
Swizzling: "y",
},
},
},
},
},
{
2020-06-20 19:26:59 +02:00
Type: shaderir.NumberExpr,
Const: constant.MakeFloat64(0),
ConstType: shaderir.ConstTypeFloat,
},
{
2020-06-20 19:26:59 +02:00
Type: shaderir.NumberExpr,
Const: constant.MakeFloat64(0),
ConstType: shaderir.ConstTypeFloat,
},
{
2020-06-20 19:26:59 +02:00
Type: shaderir.NumberExpr,
Const: constant.MakeFloat64(0),
ConstType: shaderir.ConstTypeFloat,
},
{
2020-06-20 19:26:59 +02:00
Type: shaderir.NumberExpr,
Const: constant.MakeFloat64(0),
ConstType: shaderir.ConstTypeFloat,
},
{
2020-06-20 19:26:59 +02:00
Type: shaderir.NumberExpr,
Const: constant.MakeFloat64(1),
ConstType: shaderir.ConstTypeFloat,
},
{
2020-06-20 19:26:59 +02:00
Type: shaderir.NumberExpr,
Const: constant.MakeFloat64(0),
ConstType: shaderir.ConstTypeFloat,
},
{
2020-06-20 19:26:59 +02:00
Type: shaderir.NumberExpr,
Const: constant.MakeFloat64(-1),
ConstType: shaderir.ConstTypeFloat,
},
{
2020-06-20 19:26:59 +02:00
Type: shaderir.NumberExpr,
Const: constant.MakeFloat64(-1),
ConstType: shaderir.ConstTypeFloat,
},
{
2020-06-20 19:26:59 +02:00
Type: shaderir.NumberExpr,
Const: constant.MakeFloat64(0),
ConstType: shaderir.ConstTypeFloat,
},
{
2020-06-20 19:26:59 +02:00
Type: shaderir.NumberExpr,
Const: constant.MakeFloat64(1),
ConstType: shaderir.ConstTypeFloat,
},
},
}
2020-05-24 15:15:50 +02:00
vertexPosition = shaderir.Expr{
Type: shaderir.Call,
Exprs: []shaderir.Expr{
{
Type: shaderir.BuiltinFuncExpr,
BuiltinFunc: shaderir.Vec4F,
},
{
Type: shaderir.LocalVariable,
Index: 0,
},
{
2020-06-20 19:26:59 +02:00
Type: shaderir.NumberExpr,
Const: constant.MakeFloat64(0),
ConstType: shaderir.ConstTypeFloat,
},
{
2020-06-20 19:26:59 +02:00
Type: shaderir.NumberExpr,
Const: constant.MakeFloat64(1),
ConstType: shaderir.ConstTypeFloat,
},
},
}
2020-05-24 15:15:50 +02:00
defaultVertexFunc = shaderir.VertexFunc{
2020-08-09 08:55:59 +02:00
Block: &shaderir.Block{
2020-09-07 20:01:27 +02:00
LocalVarIndexOffset: 4 + 1,
2020-05-24 15:15:50 +02:00
Stmts: []shaderir.Stmt{
{
Type: shaderir.Assign,
Exprs: []shaderir.Expr{
{
Type: shaderir.LocalVariable,
Index: 4, // the varying variable
},
{
Type: shaderir.LocalVariable,
Index: 1, // the 2nd attribute variable
},
},
},
{
Type: shaderir.Assign,
Exprs: []shaderir.Expr{
{
Type: shaderir.LocalVariable,
Index: 3, // gl_Position in GLSL
2020-05-24 15:15:50 +02:00
},
{
Type: shaderir.Binary,
Op: shaderir.Mul,
Exprs: []shaderir.Expr{
projectionMatrix,
vertexPosition,
},
},
},
},
},
},
}
)
func defaultProgram() shaderir.Program {
p := shaderir.Program{
2020-05-24 15:15:50 +02:00
Attributes: []shaderir.Type{
{Main: shaderir.Vec2}, // Local var (0) in the vertex shader
{Main: shaderir.Vec2}, // Local var (1) in the vertex shader
{Main: shaderir.Vec4}, // Local var (2) in the vertex shader
},
Varyings: []shaderir.Type{
2020-08-07 05:26:00 +02:00
{Main: shaderir.Vec2}, // Local var (4) in the vertex shader, (1) in the fragment shader
2020-05-24 15:15:50 +02:00
},
VertexFunc: defaultVertexFunc,
}
p.Uniforms = make([]shaderir.Type, graphics.PreservedUniformVariablesNum)
// Destination texture size
p.Uniforms[0] = shaderir.Type{Main: shaderir.Vec2}
// Source texture sizes
p.Uniforms[1] = shaderir.Type{
Main: shaderir.Array,
Length: graphics.ShaderImageNum,
Sub: []shaderir.Type{{Main: shaderir.Vec2}},
}
// Source texture offsets
p.Uniforms[2] = shaderir.Type{
Main: shaderir.Array,
Length: graphics.ShaderImageNum - 1,
Sub: []shaderir.Type{{Main: shaderir.Vec2}},
}
// Source region origin
p.Uniforms[3] = shaderir.Type{Main: shaderir.Vec2}
// Source region size
p.Uniforms[4] = shaderir.Type{Main: shaderir.Vec2}
return p
}
2020-05-24 15:15:50 +02:00
// ShaderProgramFill returns a shader intermediate representation to fill the frambuffer.
//
// Uniform variable's index and its value are:
2020-05-24 15:15:50 +02:00
//
// 0: the framebuffer size (Vec2)
2020-05-24 15:15:50 +02:00
func ShaderProgramFill(r, g, b, a byte) shaderir.Program {
clr := shaderir.Expr{
Type: shaderir.Call,
Exprs: []shaderir.Expr{
{
Type: shaderir.BuiltinFuncExpr,
BuiltinFunc: shaderir.Vec4F,
},
{
2020-06-20 19:26:59 +02:00
Type: shaderir.NumberExpr,
Const: constant.MakeFloat64(float64(r) / 0xff),
ConstType: shaderir.ConstTypeFloat,
},
{
2020-06-20 19:26:59 +02:00
Type: shaderir.NumberExpr,
Const: constant.MakeFloat64(float64(g) / 0xff),
ConstType: shaderir.ConstTypeFloat,
},
{
2020-06-20 19:26:59 +02:00
Type: shaderir.NumberExpr,
Const: constant.MakeFloat64(float64(b) / 0xff),
ConstType: shaderir.ConstTypeFloat,
},
{
2020-06-20 19:26:59 +02:00
Type: shaderir.NumberExpr,
Const: constant.MakeFloat64(float64(a) / 0xff),
ConstType: shaderir.ConstTypeFloat,
},
},
}
p := defaultProgram()
2020-05-24 15:15:50 +02:00
p.FragmentFunc = shaderir.FragmentFunc{
2020-08-09 08:55:59 +02:00
Block: &shaderir.Block{
2020-09-07 20:01:27 +02:00
LocalVarIndexOffset: 2 + 1,
2020-05-24 15:15:50 +02:00
Stmts: []shaderir.Stmt{
{
Type: shaderir.Assign,
Exprs: []shaderir.Expr{
{
Type: shaderir.LocalVariable,
Index: 2,
},
2020-05-24 15:15:50 +02:00
clr,
},
},
},
},
}
2020-05-24 15:15:50 +02:00
return p
}
// ShaderProgramImages returns a shader intermediate representation to render the frambuffer with the given images.
//
// Uniform variables's indices and their values are:
//
// 0: the framebuffer size (Vec2)
//
// The size and region values are actually not used in this shader so far.
func ShaderProgramImages(imageNum int) shaderir.Program {
if imageNum <= 0 {
panic("testing: imageNum must be >= 1")
}
p := defaultProgram()
p.TextureNum = imageNum
// In the fragment shader, local variables are:
//
// 0: gl_FragCoord
// 1: Varying variables (vec2)
// 2: gl_FragColor
// 3: Actual local variables in the main function
local := shaderir.Expr{
Type: shaderir.LocalVariable,
Index: 3,
}
fragColor := shaderir.Expr{
Type: shaderir.LocalVariable,
Index: 2,
}
texPos := shaderir.Expr{
Type: shaderir.LocalVariable,
Index: 1,
}
var stmts []shaderir.Stmt
for i := 0; i < imageNum; i++ {
var rhs shaderir.Expr
if i == 0 {
rhs = shaderir.Expr{
Type: shaderir.Call,
Exprs: []shaderir.Expr{
{
Type: shaderir.BuiltinFuncExpr,
BuiltinFunc: shaderir.Texture2DF,
},
{
Type: shaderir.TextureVariable,
Index: 0,
},
texPos,
},
}
} else {
texPos2 := shaderir.Expr{
Type: shaderir.Binary,
Op: shaderir.Add,
Exprs: []shaderir.Expr{
texPos,
{
Type: shaderir.Index,
Exprs: []shaderir.Expr{
{
Type: shaderir.UniformVariable,
Index: graphics.TextureSourceOffsetsUniformVariableIndex,
},
{
Type: shaderir.NumberExpr,
Const: constant.MakeInt64(int64(i - 1)),
ConstType: shaderir.ConstTypeInt,
},
},
},
},
}
rhs = shaderir.Expr{
Type: shaderir.Binary,
Op: shaderir.Add,
Exprs: []shaderir.Expr{
local,
{
Type: shaderir.Call,
Exprs: []shaderir.Expr{
{
Type: shaderir.BuiltinFuncExpr,
BuiltinFunc: shaderir.Texture2DF,
},
{
Type: shaderir.TextureVariable,
Index: i,
},
texPos2,
},
},
},
}
}
stmts = append(stmts, shaderir.Stmt{
Type: shaderir.Assign,
Exprs: []shaderir.Expr{
local,
rhs,
},
})
}
stmts = append(stmts, shaderir.Stmt{
Type: shaderir.Assign,
Exprs: []shaderir.Expr{
fragColor,
local,
},
})
p.FragmentFunc = shaderir.FragmentFunc{
2020-08-09 08:55:59 +02:00
Block: &shaderir.Block{
LocalVars: []shaderir.Type{
{Main: shaderir.Vec4},
},
2020-09-07 20:01:27 +02:00
LocalVarIndexOffset: 2 + 1,
Stmts: stmts,
},
}
return p
}