From 08038b21c370c8b681046b85badf9ce3e06015f1 Mon Sep 17 00:00:00 2001
From: yohamta <yohamta@gmail.com>
Date: Sat, 26 Nov 2022 13:32:53 +0900
Subject: [PATCH] update view logic to update the layout on draw

---
 view.go | 24 +++++++++++++++++-------
 1 file changed, 17 insertions(+), 7 deletions(-)

diff --git a/view.go b/view.go
index f327a0d..1e24a4e 100644
--- a/view.go
+++ b/view.go
@@ -2,6 +2,7 @@ package furex
 
 import (
 	"image"
+	"sync"
 
 	"github.com/hajimehoshi/ebiten/v2"
 )
@@ -31,17 +32,13 @@ type View struct {
 	containerEmbed
 	flexEmbed
 	hasParent bool
+	lock      sync.Mutex
 }
 
 // Update updates the view
 func (v *View) Update() {
 	if v.isDirty {
-		if !v.hasParent {
-			v.frame = image.Rect(v.Left, v.Top, v.Left+v.Width, v.Top+v.Height)
-		}
-		v.flexEmbed.View = v
-		v.layout(v.frame.Dx(), v.frame.Dy(), &v.containerEmbed)
-		v.isDirty = false
+		v.startLayout()
 	}
 	for _, v := range v.children {
 		v.item.Update()
@@ -55,6 +52,17 @@ func (v *View) Update() {
 	}
 }
 
+func (v *View) startLayout() {
+	v.lock.Lock()
+	defer v.lock.Unlock()
+	if !v.hasParent {
+		v.frame = image.Rect(v.Left, v.Top, v.Left+v.Width, v.Top+v.Height)
+	}
+	v.flexEmbed.View = v
+	v.layout(v.frame.Dx(), v.frame.Dy(), &v.containerEmbed)
+	v.isDirty = false
+}
+
 // UpdateWithSize the view with modified height and width
 func (v *View) UpdateWithSize(width, height int) {
 	if !v.hasParent && (v.Width != width || v.Height != height) {
@@ -67,8 +75,10 @@ func (v *View) UpdateWithSize(width, height int) {
 
 // Draw draws the view
 func (v *View) Draw(screen *ebiten.Image) {
+	if v.isDirty {
+		v.startLayout()
+	}
 	v.containerEmbed.Draw(screen)
-
 	if Debug && !v.hasParent {
 		debugBorders(screen, v.containerEmbed)
 	}