ebiten/internal/shareable/shareable.go

382 lines
8.1 KiB
Go
Raw Normal View History

// Copyright 2018 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.
2018-03-10 15:48:10 +01:00
package shareable
import (
"fmt"
"image"
"image/color"
2018-03-10 16:07:32 +01:00
"runtime"
"sync"
"github.com/hajimehoshi/ebiten/internal/affine"
"github.com/hajimehoshi/ebiten/internal/graphics"
"github.com/hajimehoshi/ebiten/internal/graphicsutil"
"github.com/hajimehoshi/ebiten/internal/opengl"
2018-03-05 16:38:56 +01:00
"github.com/hajimehoshi/ebiten/internal/packing"
"github.com/hajimehoshi/ebiten/internal/restorable"
)
var (
quadIndices = []uint16{0, 1, 2, 1, 2, 3}
)
2018-03-10 16:02:23 +01:00
type backend struct {
restorable *restorable.Image
2018-03-15 17:21:33 +01:00
// If page is nil, the backend is not shared.
page *packing.Page
}
func (b *backend) TryAlloc(width, height int) (*packing.Node, bool) {
// If the region is allocated without any extention, it's fine.
if n := b.page.Alloc(width, height); n != nil {
return n, true
}
// Simulate the extending the page and calculate the appropriate page size.
page := b.page.Clone()
nExtended := 0
for {
if !page.Extend() {
// The page can't be extended any more. Return as failure.
return nil, false
}
nExtended++
if n := page.Alloc(width, height); n != nil {
break
}
}
for i := 0; i < nExtended; i++ {
b.page.Extend()
}
s := b.page.Size()
newImg := restorable.NewImage(s, s, false)
oldImg := b.restorable
w, h := oldImg.Size()
2018-06-16 18:48:52 +02:00
vs := graphicsutil.QuadVertices(w, h, 0, 0, w, h, idGeoM{})
newImg.DrawImage(oldImg, vs, quadIndices, nil, opengl.CompositeModeCopy, graphics.FilterNearest)
oldImg.Dispose()
b.restorable = newImg
n := b.page.Alloc(width, height)
if n == nil {
panic("not reached")
}
return n, true
}
var (
// backendsM is a mutex for critical sections of the backend and packing.Node objects.
backendsM sync.Mutex
2018-03-10 16:02:23 +01:00
// theBackends is a set of actually shared images.
theBackends = []*backend{}
)
2018-03-10 16:05:06 +01:00
type Image struct {
width int
height int
disposed bool
2018-03-10 16:02:23 +01:00
backend *backend
// If node is nil, the image is not shared.
node *packing.Node
}
2018-06-16 18:48:52 +02:00
type idGeoM struct{}
func (idGeoM) Apply(x, y float32) (x2, y2 float32) {
2018-06-16 18:48:52 +02:00
return x, y
}
2018-04-08 19:18:46 +02:00
func (i *Image) ensureNotShared() {
if i.backend == nil {
i.allocate(false)
return
}
2018-04-08 19:18:46 +02:00
if i.node == nil {
return
}
2018-04-08 19:18:46 +02:00
x, y, w, h := i.region()
newImg := restorable.NewImage(w, h, false)
vw, vh := i.backend.restorable.Size()
2018-06-16 18:48:52 +02:00
vs := graphicsutil.QuadVertices(vw, vh, x, y, x+w, y+h, idGeoM{})
newImg.DrawImage(i.backend.restorable, vs, quadIndices, nil, opengl.CompositeModeCopy, graphics.FilterNearest)
i.dispose(false)
2018-04-08 19:18:46 +02:00
i.backend = &backend{
restorable: newImg,
}
}
2018-04-08 19:18:46 +02:00
func (i *Image) region() (x, y, width, height int) {
if i.backend == nil {
panic("not reached")
}
2018-04-08 19:18:46 +02:00
if i.node == nil {
w, h := i.backend.restorable.Size()
return 0, 0, w, h
}
2018-04-08 19:18:46 +02:00
return i.node.Region()
}
2018-04-08 19:18:46 +02:00
func (i *Image) Size() (width, height int) {
return i.width, i.height
}
func (i *Image) DrawImage(img *Image, sx0, sy0, sx1, sy1 int, geom graphicsutil.GeoM, colorm *affine.ColorM, mode opengl.CompositeMode, filter graphics.Filter) {
backendsM.Lock()
defer backendsM.Unlock()
if img.disposed {
panic("shareable: the drawing source image must not be disposed")
}
if i.disposed {
panic("shareable: the drawing target image must not be disposed")
}
if img.backend == nil {
img.allocate(true)
}
2018-04-08 19:18:46 +02:00
i.ensureNotShared()
2018-03-10 15:48:10 +01:00
// Compare i and img after ensuring i is not shared, or
// i and img might share the same texture even though i != img.
2018-04-08 19:18:46 +02:00
if i.backend.restorable == img.backend.restorable {
2018-03-10 15:48:10 +01:00
panic("shareable: Image.DrawImage: img must be different from the receiver")
}
dx, dy, _, _ := img.region()
w, h := img.backend.restorable.Size()
2018-06-10 18:02:25 +02:00
vs := graphicsutil.QuadVertices(w, h, sx0+dx, sy0+dy, sx1+dx, sy1+dy, geom)
i.backend.restorable.DrawImage(img.backend.restorable, vs, quadIndices, colorm, mode, filter)
}
2018-04-08 19:18:46 +02:00
func (i *Image) ReplacePixels(p []byte) {
backendsM.Lock()
defer backendsM.Unlock()
if i.disposed {
panic("shareable: the image must not be disposed")
}
if i.backend == nil {
i.allocate(true)
}
2018-04-08 19:18:46 +02:00
x, y, w, h := i.region()
if l := 4 * w * h; len(p) != l {
2018-03-10 15:48:10 +01:00
panic(fmt.Sprintf("shareable: len(p) was %d but must be %d", len(p), l))
}
2018-04-08 19:18:46 +02:00
i.backend.restorable.ReplacePixels(p, x, y, w, h)
}
2018-04-08 19:18:46 +02:00
func (i *Image) At(x, y int) (color.Color, error) {
backendsM.Lock()
2018-04-05 05:02:08 +02:00
defer backendsM.Unlock()
if i.backend == nil {
return color.RGBA{}, nil
}
2018-04-08 19:18:46 +02:00
ox, oy, w, h := i.region()
if x < 0 || y < 0 || x >= w || y >= h {
return color.RGBA{}, nil
}
2018-04-08 19:18:46 +02:00
clr, err := i.backend.restorable.At(x+ox, y+oy)
return clr, err
}
2018-04-08 19:18:46 +02:00
func (i *Image) Dispose() {
backendsM.Lock()
defer backendsM.Unlock()
i.dispose(true)
}
func (i *Image) dispose(markDisposed bool) {
defer func() {
if markDisposed {
i.disposed = true
}
2018-04-08 19:18:46 +02:00
i.backend = nil
i.node = nil
runtime.SetFinalizer(i, nil)
}()
if i.disposed {
return
}
if i.backend == nil {
// Not allocated yet.
return
}
2018-04-08 19:18:46 +02:00
if i.node == nil {
i.backend.restorable.Dispose()
return
}
2018-04-08 19:18:46 +02:00
i.backend.page.Free(i.node)
if !i.backend.page.IsEmpty() {
// As this part can be reused, this should be cleared explicitly.
x0, y0, x1, y1 := i.region()
i.backend.restorable.ReplacePixels(nil, x0, y0, x1, y1)
return
}
2018-04-08 19:18:46 +02:00
i.backend.restorable.Dispose()
index := -1
2018-04-08 19:18:46 +02:00
for idx, sh := range theBackends {
if sh == i.backend {
index = idx
break
}
}
if index == -1 {
panic("not reached")
}
2018-03-10 16:02:23 +01:00
theBackends = append(theBackends[:index], theBackends[index+1:]...)
}
2018-04-08 19:18:46 +02:00
func (i *Image) IsInvalidated() (bool, error) {
backendsM.Lock()
2018-04-05 05:02:08 +02:00
defer backendsM.Unlock()
2018-04-08 19:18:46 +02:00
v, err := i.backend.restorable.IsInvalidated()
return v, err
}
2018-03-10 16:05:06 +01:00
func NewImage(width, height int) *Image {
// Actual allocation is done lazily.
return &Image{
width: width,
height: height,
}
}
func (i *Image) allocate(shareable bool) {
if i.backend != nil {
2018-04-29 12:10:36 +02:00
panic("not reached")
}
const (
initSize = 1024
maxSize = 4096
)
if !shareable || i.width > maxSize || i.height > maxSize {
i.backend = &backend{
restorable: restorable.NewImage(i.width, i.height, false),
}
return
}
2018-03-23 20:27:10 +01:00
for _, b := range theBackends {
if n, ok := b.TryAlloc(i.width, i.height); ok {
i.backend = b
i.node = n
return
}
}
size := initSize
for i.width > size || i.height > size {
if size == maxSize {
panic("not reached")
}
size *= 2
}
2018-03-23 20:27:10 +01:00
b := &backend{
restorable: restorable.NewImage(size, size, false),
page: packing.NewPage(size, maxSize),
2018-03-06 18:18:08 +01:00
}
2018-03-23 20:27:10 +01:00
theBackends = append(theBackends, b)
n := b.page.Alloc(i.width, i.height)
if n == nil {
panic("not reached")
}
i.backend = b
i.node = n
2018-03-10 16:07:32 +01:00
runtime.SetFinalizer(i, (*Image).Dispose)
return
}
2018-03-10 15:48:10 +01:00
2018-03-10 16:05:06 +01:00
func NewVolatileImage(width, height int) *Image {
backendsM.Lock()
defer backendsM.Unlock()
2018-03-10 15:48:10 +01:00
r := restorable.NewImage(width, height, true)
2018-03-10 16:07:32 +01:00
i := &Image{
width: width,
height: height,
2018-03-10 16:02:23 +01:00
backend: &backend{
2018-03-10 15:48:10 +01:00
restorable: r,
},
}
2018-03-10 16:07:32 +01:00
runtime.SetFinalizer(i, (*Image).Dispose)
return i
2018-03-10 15:48:10 +01:00
}
2018-03-10 16:05:06 +01:00
func NewScreenFramebufferImage(width, height int) *Image {
backendsM.Lock()
defer backendsM.Unlock()
2018-03-10 15:48:10 +01:00
r := restorable.NewScreenFramebufferImage(width, height)
2018-03-10 16:07:32 +01:00
i := &Image{
width: width,
height: height,
2018-03-10 16:02:23 +01:00
backend: &backend{
2018-03-10 15:48:10 +01:00
restorable: r,
},
}
2018-03-10 16:07:32 +01:00
runtime.SetFinalizer(i, (*Image).Dispose)
return i
2018-03-10 15:48:10 +01:00
}
func InitializeGLState() error {
backendsM.Lock()
defer backendsM.Unlock()
return restorable.InitializeGLState()
}
func ResolveStaleImages() error {
backendsM.Lock()
defer backendsM.Unlock()
return restorable.ResolveStaleImages()
}
func IsRestoringEnabled() bool {
2018-03-25 17:13:26 +02:00
// As IsRestoringEnabled is an immutable state, no need to lock here.
return restorable.IsRestoringEnabled()
}
func Restore() error {
backendsM.Lock()
defer backendsM.Unlock()
return restorable.Restore()
}
func Images() ([]image.Image, error) {
backendsM.Lock()
defer backendsM.Unlock()
return restorable.Images()
}