From 34d7385e6d47ed11ce12ad8f21235f2ce03aa30f Mon Sep 17 00:00:00 2001
From: Hajime Hoshi <hajimehoshi@gmail.com>
Date: Fri, 6 Nov 2020 01:45:04 +0900
Subject: [PATCH] shareable: Refactoring

---
 internal/shareable/image.go | 9 +++------
 1 file changed, 3 insertions(+), 6 deletions(-)

diff --git a/internal/shareable/image.go b/internal/shareable/image.go
index b70e37624..4b2185dfc 100644
--- a/internal/shareable/image.go
+++ b/internal/shareable/image.go
@@ -188,6 +188,9 @@ func (i *Image) isShared() bool {
 }
 
 func (i *Image) ensureNotShared() {
+	i.nonUpdatedCount = 0
+	delete(imagesToMakeShared, i)
+
 	if i.backend == nil {
 		i.allocate(false)
 		return
@@ -374,9 +377,6 @@ func (i *Image) DrawTriangles(srcs [graphics.ShaderImageNum]*Image, vertices []f
 
 	i.backend.restorable.DrawTriangles(imgs, offsets, vertices, indices, colorm, mode, filter, address, sourceRegion, s, uniforms)
 
-	i.nonUpdatedCount = 0
-	delete(imagesToMakeShared, i)
-
 	for _, src := range srcs {
 		if src == nil {
 			continue
@@ -407,9 +407,6 @@ func (i *Image) Fill(clr color.RGBA) {
 
 	// As *restorable.Image is an independent image, it is fine to fill the entire image.
 	i.backend.restorable.Fill(clr)
-
-	i.nonUpdatedCount = 0
-	delete(imagesToMakeShared, i)
 }
 
 func (i *Image) ReplacePixels(pix []byte) {