ebiten/imageimpl.go

311 lines
7.4 KiB
Go
Raw Normal View History

// Copyright 2016 Hajime Hoshi
//
// 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 ebiten
import (
"errors"
"fmt"
"image"
"image/color"
"image/draw"
"runtime"
"sync"
"github.com/hajimehoshi/ebiten/internal/graphics"
"github.com/hajimehoshi/ebiten/internal/graphics/opengl"
2016-07-24 19:28:59 +02:00
"github.com/hajimehoshi/ebiten/internal/pixels"
)
type imageImpl struct {
image *graphics.Image
disposed bool
width int
height int
filter Filter
2016-07-27 05:24:30 +02:00
pixels pixels.Pixels
volatile bool
screen bool
m sync.Mutex
}
2016-07-04 20:40:40 +02:00
func newImageImpl(width, height int, filter Filter, volatile bool) (*imageImpl, error) {
2016-07-04 20:15:32 +02:00
img, err := graphics.NewImage(width, height, glFilter(filter))
if err != nil {
return nil, err
}
2016-07-04 17:24:06 +02:00
i := &imageImpl{
2016-07-04 20:40:40 +02:00
image: img,
width: width,
height: height,
filter: filter,
volatile: volatile,
2016-07-04 17:24:06 +02:00
}
runtime.SetFinalizer(i, (*imageImpl).Dispose)
return i, nil
}
func newImageImplFromImage(source image.Image, filter Filter) (*imageImpl, error) {
size := source.Bounds().Size()
w, h := size.X, size.Y
// TODO: Return error when the image is too big!
// Don't lock while manipulating an image.Image interface.
rgbaImg, ok := source.(*image.RGBA)
if !ok || source.Bounds().Min != image.ZP {
origImg := source
newImg := image.NewRGBA(image.Rect(0, 0, w, h))
draw.Draw(newImg, newImg.Bounds(), origImg, origImg.Bounds().Min, draw.Src)
rgbaImg = newImg
}
p := make([]uint8, 4*w*h)
for j := 0; j < h; j++ {
copy(p[j*w*4:(j+1)*w*4], rgbaImg.Pix[j*rgbaImg.Stride:])
}
img, err := graphics.NewImageFromImage(rgbaImg, glFilter(filter))
if err != nil {
// TODO: texture should be removed here?
return nil, err
}
i := &imageImpl{
image: img,
width: w,
height: h,
filter: filter,
}
2016-07-26 03:51:48 +02:00
i.pixels.ReplacePixels(p)
runtime.SetFinalizer(i, (*imageImpl).Dispose)
return i, nil
}
func newScreenImageImpl(width, height int) (*imageImpl, error) {
2016-07-04 20:15:32 +02:00
img, err := graphics.NewScreenFramebufferImage(width, height)
if err != nil {
return nil, err
}
2016-07-04 20:15:32 +02:00
i := &imageImpl{
2016-07-04 20:18:55 +02:00
image: img,
width: width,
height: height,
volatile: true,
screen: true,
}
2016-07-04 20:15:32 +02:00
runtime.SetFinalizer(i, (*imageImpl).Dispose)
return i, nil
}
func (i *imageImpl) Fill(clr color.Color) error {
2016-07-03 15:55:15 +02:00
i.m.Lock()
defer i.m.Unlock()
if i.disposed {
return errors.New("ebiten: image is already disposed")
}
2016-07-24 19:28:59 +02:00
i.pixels.Fill(clr)
return i.image.Fill(clr)
}
func (i *imageImpl) clearIfVolatile() error {
i.m.Lock()
defer i.m.Unlock()
if i.disposed {
return nil
}
if !i.volatile {
return nil
}
2016-07-24 19:28:59 +02:00
i.pixels.Clear()
return i.image.Fill(color.Transparent)
}
func (i *imageImpl) DrawImage(image *Image, options *DrawImageOptions) error {
// Calculate vertices before locking because the user can do anything in
// options.ImageParts interface without deadlock (e.g. Call Image functions).
if options == nil {
options = &DrawImageOptions{}
}
parts := options.ImageParts
if parts == nil {
// Check options.Parts for backward-compatibility.
dparts := options.Parts
if dparts != nil {
parts = imageParts(dparts)
} else {
parts = &wholeImage{image.impl.width, image.impl.height}
}
}
quads := &textureQuads{parts: parts, width: image.impl.width, height: image.impl.height}
// TODO: Reuse one vertices instead of making here, but this would need locking.
vertices := make([]int16, parts.Len()*16)
n := quads.vertices(vertices)
if n == 0 {
return nil
}
if i == image.impl {
return errors.New("ebiten: Image.DrawImage: image should be different from the receiver")
}
2016-07-03 15:55:15 +02:00
i.m.Lock()
defer i.m.Unlock()
if i.disposed {
return errors.New("ebiten: image is already disposed")
}
geom := options.GeoM
colorm := options.ColorM
mode := opengl.CompositeMode(options.CompositeMode)
2016-07-26 19:16:31 +02:00
i.pixels.AppendDrawImageHistory(image.impl.image, vertices, &geom, &colorm, mode)
if err := i.image.DrawImage(image.impl.image, vertices, &geom, &colorm, mode); err != nil {
return err
}
return nil
}
func (i *imageImpl) At(x, y int, context *opengl.Context) color.Color {
if context == nil {
panic("ebiten: At can't be called when the GL context is not initialized (this panic happens as of version 1.4.0-alpha)")
}
2016-07-03 16:31:03 +02:00
i.m.Lock()
defer i.m.Unlock()
if i.disposed {
return color.Transparent
}
2016-07-13 20:00:50 +02:00
idx := 4*x + 4*y*i.width
2016-07-27 05:24:30 +02:00
clr, err := i.pixels.At(idx, i.image, context)
2016-07-13 19:35:20 +02:00
if err != nil {
panic(err)
}
2016-07-13 19:35:20 +02:00
return clr
}
func (i *imageImpl) resolveStalePixels(context *opengl.Context) error {
i.m.Lock()
defer i.m.Unlock()
if i.disposed {
return nil
}
if i.volatile {
return nil
}
2016-07-27 05:24:30 +02:00
if err := i.pixels.ReadPixelsFromVRAMIfStale(i.image, context); err != nil {
return err
}
return nil
}
2016-07-26 05:28:05 +02:00
func (i *imageImpl) resetPixelsIfDependingOn(target *imageImpl, context *opengl.Context) error {
i.m.Lock()
defer i.m.Unlock()
if i == target {
return nil
}
if i.disposed {
return nil
}
if target.isDisposed() {
return errors.New("ebiten: target is already disposed")
}
2016-07-26 05:17:15 +02:00
// target is an image begin tried to mutate.
// If pixels object is related to that image, the pixels must be reset.
2016-07-26 05:28:05 +02:00
if !i.pixels.DependsOn(target.image) {
return nil
}
if context == nil || i.volatile {
2016-07-26 05:17:15 +02:00
// context is nil when this is not initialized yet.
2016-07-26 19:16:31 +02:00
i.pixels.MakeStale()
2016-07-25 01:48:58 +02:00
return nil
}
2016-07-27 05:24:30 +02:00
if err := i.pixels.ReadPixelsFromVRAM(i.image, context); err != nil {
2016-07-13 19:35:20 +02:00
return err
}
return nil
}
2016-07-26 18:28:16 +02:00
func (i *imageImpl) hasDependency() bool {
i.m.Lock()
defer i.m.Unlock()
2016-07-26 18:28:16 +02:00
return i.pixels.HasDependency()
}
func (i *imageImpl) restore(context *opengl.Context) error {
i.m.Lock()
defer i.m.Unlock()
if i.disposed {
return nil
}
if i.screen {
// The screen image should also be recreated because framebuffer might
// be changed.
var err error
i.image, err = graphics.NewScreenFramebufferImage(i.width, i.height)
if err != nil {
return err
}
return nil
}
2016-07-13 05:40:21 +02:00
if i.volatile {
var err error
2016-07-13 05:40:21 +02:00
i.image, err = graphics.NewImage(i.width, i.height, glFilter(i.filter))
if err != nil {
return err
}
2016-07-13 05:40:21 +02:00
return nil
}
var err error
2016-07-26 19:50:53 +02:00
i.image, err = i.pixels.CreateImage(context, i.width, i.height, glFilter(i.filter))
if err != nil {
return err
}
return nil
}
func (i *imageImpl) Dispose() error {
2016-07-03 15:55:15 +02:00
i.m.Lock()
defer i.m.Unlock()
if i.disposed {
return errors.New("ebiten: image is already disposed")
}
if !i.screen {
if err := i.image.Dispose(); err != nil {
return err
}
}
i.image = nil
i.disposed = true
2016-07-27 05:24:30 +02:00
i.pixels.Clear()
runtime.SetFinalizer(i, nil)
return nil
}
func (i *imageImpl) ReplacePixels(p []uint8) error {
if l := 4 * i.width * i.height; len(p) != l {
return fmt.Errorf("ebiten: p's length must be %d", l)
}
2016-07-03 15:55:15 +02:00
i.m.Lock()
defer i.m.Unlock()
2016-07-26 03:51:48 +02:00
i.pixels.ReplacePixels(p)
if i.disposed {
return errors.New("ebiten: image is already disposed")
}
return i.image.ReplacePixels(p)
}
func (i *imageImpl) isDisposed() bool {
i.m.Lock()
defer i.m.Unlock()
return i.disposed
}
func (i *imageImpl) isInvalidated(context *opengl.Context) bool {
i.m.Lock()
defer i.m.Unlock()
return i.image.IsInvalidated(context)
}