package pixelgl import ( "image/color" "github.com/faiface/glhf" "github.com/faiface/mainthread" "github.com/faiface/pixel" "github.com/go-gl/mathgl/mgl32" "github.com/pkg/errors" ) // Canvas is basically a Picture that you can draw onto. // // Canvas supports TrianglesPosition, TrianglesColor and TrianglesTexture. type Canvas struct { f *glhf.Frame s *glhf.Shader copyVs *glhf.VertexSlice smooth bool drawTd pixel.TrianglesDrawer pic *pixel.Picture mat mgl32.Mat3 col mgl32.Vec4 bnd mgl32.Vec4 } // NewCanvas creates a new fully transparent Canvas with specified dimensions in pixels. func NewCanvas(width, height float64, smooth bool) *Canvas { c := &Canvas{smooth: smooth} mainthread.Call(func() { var err error c.f = glhf.NewFrame(int(width), int(height), smooth) c.s, err = glhf.NewShader( canvasVertexFormat, canvasUniformFormat, canvasVertexShader, canvasFragmentShader, ) if err != nil { panic(errors.Wrap(err, "failed to create canvas")) } c.copyVs = glhf.MakeVertexSlice(c.s, 6, 6) c.copyVs.Begin() c.copyVs.SetVertexData([]float32{ -1, -1, 1, 1, 1, 1, 0, 0, 1, -1, 1, 1, 1, 1, 1, 0, 1, 1, 1, 1, 1, 1, 1, 1, -1, -1, 1, 1, 1, 1, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, -1, 1, 1, 1, 1, 1, 0, 1, }) c.copyVs.End() }) white := pixel.NRGBA{R: 1, G: 1, B: 1, A: 1} c.drawTd = pixel.TrianglesDrawer{Triangles: &pixel.TrianglesData{ {Position: pixel.V(-1, -1), Color: white, Texture: pixel.V(0, 0)}, {Position: pixel.V(1, -1), Color: white, Texture: pixel.V(1, 0)}, {Position: pixel.V(1, 1), Color: white, Texture: pixel.V(1, 1)}, {Position: pixel.V(-1, -1), Color: white, Texture: pixel.V(0, 0)}, {Position: pixel.V(1, 1), Color: white, Texture: pixel.V(1, 1)}, {Position: pixel.V(-1, 1), Color: white, Texture: pixel.V(0, 1)}, }} c.pic = nil c.mat = mgl32.Ident3() c.col = mgl32.Vec4{1, 1, 1, 1} c.bnd = mgl32.Vec4{0, 0, 1, 1} return c } // SetSize resizes the Canvas. The original content will be stretched to fit the new size. func (c *Canvas) SetSize(width, height float64) { if pixel.V(width, height) == pixel.V(c.Size()) { return } mainthread.Call(func() { oldF := c.f c.f = glhf.NewFrame(int(width), int(height), c.smooth) c.f.Begin() c.s.Begin() c.s.SetUniformAttr(canvasTransformMat3, mgl32.Ident3()) c.s.SetUniformAttr(canvasMaskColorVec4, mgl32.Vec4{1, 1, 1, 1}) c.s.SetUniformAttr(canvasBoundsVec4, mgl32.Vec4{0, 0, 1, 1}) oldF.Texture().Begin() c.copyVs.Begin() c.copyVs.Draw() c.copyVs.End() oldF.Texture().End() c.s.End() c.f.End() }) } // Size returns the width and the height of the Canvas in pixels. func (c *Canvas) Size() (width, height float64) { return float64(c.f.Width()), float64(c.f.Height()) } // Content returns a Picture that contains the content of this Canvas. The returned Picture changes // as you draw onto the Canvas, so there is no real need to call this method more than once (but it // might be beneficial to your code to do so). func (c *Canvas) Content() *pixel.Picture { return pixel.PictureFromTexture(c.f.Texture()) } // Clear fills the whole Canvas with one specified color. func (c *Canvas) Clear(col color.Color) { mainthread.CallNonBlock(func() { c.f.Begin() col := pixel.NRGBAModel.Convert(col).(pixel.NRGBA) glhf.Clear(float32(col.R), float32(col.G), float32(col.B), float32(col.A)) c.f.End() }) } // Draw draws the content of the Canvas onto another Target. If no transform is applied, the content // is fully stretched to fit the Target. func (c *Canvas) Draw(t pixel.Target) { t.SetPicture(c.Content()) c.drawTd.Draw(t) } // MakeTriangles returns Triangles that draw onto this Canvas. func (c *Canvas) MakeTriangles(t pixel.Triangles) pixel.TargetTriangles { gt := NewGLTriangles(c.s, t).(*glTriangles) return &canvasTriangles{ c: c, glTriangles: gt, } } // SetPicture sets a Picture that will be used in further draw operations. // // This does not set the Picture that this Canvas draws onto, don't confuse it. func (c *Canvas) SetPicture(p *pixel.Picture) { if p != nil { min := pictureBounds(p, pixel.V(0, 0)) max := pictureBounds(p, pixel.V(1, 1)) c.bnd = mgl32.Vec4{ float32(min.X()), float32(min.Y()), float32(max.X()), float32(max.Y()), } } c.pic = p } // SetTransform sets the transformations used in further draw operations. func (c *Canvas) SetTransform(t ...pixel.Transform) { c.mat = transformToMat(t...) } // SetMaskColor sets the mask color used in further draw operations. func (c *Canvas) SetMaskColor(col color.Color) { if col == nil { col = pixel.NRGBA{R: 1, G: 1, B: 1, A: 1} } nrgba := pixel.NRGBAModel.Convert(col).(pixel.NRGBA) r := float32(nrgba.R) g := float32(nrgba.G) b := float32(nrgba.B) a := float32(nrgba.A) c.col = mgl32.Vec4{r, g, b, a} } type canvasTriangles struct { c *Canvas *glTriangles } func (ct *canvasTriangles) Draw() { // avoid possible race condition pic := ct.c.pic mat := ct.c.mat col := ct.c.col bnd := ct.c.bnd mainthread.CallNonBlock(func() { ct.c.f.Begin() ct.c.s.Begin() ct.c.s.SetUniformAttr(canvasTransformMat3, mat) ct.c.s.SetUniformAttr(canvasMaskColorVec4, col) ct.c.s.SetUniformAttr(canvasBoundsVec4, bnd) if pic != nil { pic.Texture().Begin() ct.glTriangles.Draw() pic.Texture().End() } else { ct.glTriangles.Draw() } ct.c.s.End() ct.c.f.End() }) } const ( canvasPositionVec2 int = iota canvasColorVec4 canvasTextureVec2 ) var canvasVertexFormat = glhf.AttrFormat{ canvasPositionVec2: {Name: "position", Type: glhf.Vec2}, canvasColorVec4: {Name: "color", Type: glhf.Vec4}, canvasTextureVec2: {Name: "texture", Type: glhf.Vec2}, } const ( canvasMaskColorVec4 int = iota canvasTransformMat3 canvasBoundsVec4 ) var canvasUniformFormat = glhf.AttrFormat{ {Name: "maskColor", Type: glhf.Vec4}, {Name: "transform", Type: glhf.Mat3}, {Name: "bounds", Type: glhf.Vec4}, } var canvasVertexShader = ` #version 330 core in vec2 position; in vec4 color; in vec2 texture; out vec4 Color; out vec2 Texture; uniform mat3 transform; void main() { gl_Position = vec4((transform * vec3(position.x, position.y, 1.0)).xy, 0.0, 1.0); Color = color; Texture = texture; } ` var canvasFragmentShader = ` #version 330 core in vec4 Color; in vec2 Texture; out vec4 color; uniform vec4 maskColor; uniform vec4 bounds; uniform sampler2D tex; void main() { vec2 boundsMin = bounds.xy; vec2 boundsMax = bounds.zw; if (Texture == vec2(-1, -1)) { color = maskColor * Color; } else { float tx = boundsMin.x * (1 - Texture.x) + boundsMax.x * Texture.x; float ty = boundsMin.y * (1 - Texture.y) + boundsMax.y * Texture.y; color = maskColor * Color * texture(tex, vec2(tx, ty)); } } `