fix: add loading placeholder on upscale image

scenario-labs · Jun 28, 2024 · 4a553bb · 4a553bb
1 parent e6fe08e
commit 4a553bb
Showing 1 changed file with 19 additions and 2 deletions.
diff --git a/package/Editor/UpscaleEditor/UpscaleEditorUI.cs b/package/Editor/UpscaleEditor/UpscaleEditorUI.cs
@@ -180,6 +180,7 @@ private void DrawRightSection(Rect position)
             EditorStyle.Button("Upscale Image", () =>
             {
                 if (currentImage == null) return;
+                upscaledImages.Add(null);
                 imageDataUrl = CommonUtils.Texture2DToDataURL(currentImage);
                 if (imageData != null)
                 {
@@ -324,12 +325,21 @@ private void FetchUpscaledImage(string imgUrl)
                                         Id = asset.asset.id,
                                         Url = asset.asset.url,
                                     };
-                                    upscaledImages.Insert(0, texture);
+                                    if (upscaledImages[0] == null)
+                                    {
+                                        upscaledImages[0] = texture;
+                                    }
+                                    else
+                                    { 
+                                        upscaledImages.Insert(0, texture);
+                                    }
                                     imageDataList.Insert(0, newImageData);
                                 });
                             });
                         });
                     });
+                }, errorAction => { 
+                    upscaledImages.RemoveAt(0);
                 });
             }
             else
@@ -353,7 +363,14 @@ private void FetchUpscaledImage(string imgUrl)
                                     Id = asset.asset.id,
                                     Url = asset.asset.url,
                                 };
-                                upscaledImages.Insert(0, texture);
+                                if (upscaledImages[0] == null)
+                                {
+                                    upscaledImages[0] = texture;
+                                }
+                                else
+                                {
+                                    upscaledImages.Insert(0, texture);
+                                }
                                 imageDataList.Insert(0, newImageData);
                             });
                         });