mirror of
https://github.com/Azornes/Comfyui-LayerForge.git
synced 2026-03-22 05:02:11 -03:00
Compare commits
33 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
835d94a11d | ||
|
|
061e2b7a9a | ||
|
|
b1f29eefdb | ||
|
|
b8fbcee67a | ||
|
|
d44d944f2d | ||
|
|
ab5d71597a | ||
|
|
ce4d332987 | ||
|
|
9b04729561 | ||
|
|
27ad139cd5 | ||
|
|
66cbcb641b | ||
|
|
986e0a23a2 | ||
|
|
068ed9ee59 | ||
|
|
4e5ef18d93 | ||
|
|
be37966b45 | ||
|
|
dd5fc5470f | ||
|
|
1f1d0aeb7d | ||
|
|
da55d741d6 | ||
|
|
959c47c29b | ||
|
|
ab7ab9d1a8 | ||
|
|
d8d33089d2 | ||
|
|
de67252a87 | ||
|
|
4acece1602 | ||
|
|
ffa5b136bf | ||
|
|
7a5ecb3919 | ||
|
|
20ab861315 | ||
|
|
6750141bcc | ||
|
|
5ea2562b32 | ||
|
|
079fb7b362 | ||
|
|
e05e2d8d8a | ||
|
|
ae55c8a827 | ||
|
|
e21fab0061 | ||
|
|
36a80bbb7e | ||
|
|
492e06068a |
171
.github/ISSUE_TEMPLATE/bug_report.yml
vendored
171
.github/ISSUE_TEMPLATE/bug_report.yml
vendored
@@ -1,20 +1,75 @@
|
|||||||
name: 🐞 Bug Report
|
name: 🐞 Bug Report
|
||||||
description: Report an error or unexpected behavior
|
description: 'Report something that is not working correctly'
|
||||||
title: "[BUG] "
|
title: "[BUG] "
|
||||||
labels: [bug]
|
labels: [bug]
|
||||||
body:
|
body:
|
||||||
|
- type: checkboxes
|
||||||
|
attributes:
|
||||||
|
label: Prerequisites
|
||||||
|
options:
|
||||||
|
- label: I am running the latest version of [ComfyUI](https://github.com/comfyanonymous/ComfyUI/releases)
|
||||||
|
required: true
|
||||||
|
- label: I am running the latest version of [ComfyUI_frontend](https://github.com/Comfy-Org/ComfyUI_frontend/releases)
|
||||||
|
required: true
|
||||||
|
- label: I am running the latest version of LayerForge [Github](https://github.com/Azornes/Comfyui-LayerForge/releases) | [Manager](https://registry.comfy.org/publishers/azornes/nodes/layerforge)
|
||||||
|
required: true
|
||||||
|
- label: I have searched existing(open/closed) issues to make sure this isn't a duplicate
|
||||||
|
required: true
|
||||||
|
|
||||||
|
- type: textarea
|
||||||
|
id: description
|
||||||
|
attributes:
|
||||||
|
label: What happened?
|
||||||
|
description: A clear and concise description of the bug. Include screenshots or videos if helpful.
|
||||||
|
placeholder: |
|
||||||
|
Example: "When I connect a image to an Input, the connection line appears but the workflow fails to execute with an error message..."
|
||||||
|
validations:
|
||||||
|
required: true
|
||||||
|
|
||||||
|
- type: textarea
|
||||||
|
id: reproduce
|
||||||
|
attributes:
|
||||||
|
label: Steps to Reproduce
|
||||||
|
description: How can I reproduce this issue? Please attach your workflow (JSON or PNG) if needed.
|
||||||
|
placeholder: |
|
||||||
|
1. Connect Image to Input
|
||||||
|
2. Click Queue Prompt
|
||||||
|
3. See error
|
||||||
|
validations:
|
||||||
|
required: true
|
||||||
|
|
||||||
|
- type: dropdown
|
||||||
|
id: severity
|
||||||
|
attributes:
|
||||||
|
label: How is this affecting you?
|
||||||
|
options:
|
||||||
|
- Crashes ComfyUI completely
|
||||||
|
- Workflow won't execute
|
||||||
|
- Feature doesn't work as expected
|
||||||
|
- Visual/UI issue only
|
||||||
|
- Minor inconvenience
|
||||||
|
validations:
|
||||||
|
required: true
|
||||||
|
|
||||||
|
- type: dropdown
|
||||||
|
id: browser
|
||||||
|
attributes:
|
||||||
|
label: Browser
|
||||||
|
description: Which browser are you using?
|
||||||
|
options:
|
||||||
|
- Chrome/Chromium
|
||||||
|
- Firefox
|
||||||
|
- Safari
|
||||||
|
- Edge
|
||||||
|
- Other
|
||||||
|
validations:
|
||||||
|
required: true
|
||||||
|
|
||||||
- type: markdown
|
- type: markdown
|
||||||
attributes:
|
attributes:
|
||||||
value: |
|
value: |
|
||||||
**Thank you for reporting a bug!**
|
## Additional Information (Optional)
|
||||||
Please follow these steps to capture all necessary information:
|
*The following fields help me debug complex issues but are not required for most bug reports.*
|
||||||
|
|
||||||
### ✅ Before You Report:
|
|
||||||
1. Make sure you have the **latest versions**:
|
|
||||||
- [ComfyUI Github](https://github.com/comfyanonymous/ComfyUI/releases)
|
|
||||||
- [LayerForge Github](https://github.com/Azornes/Comfyui-LayerForge/releases) or via [ComfyUI Node Manager](https://registry.comfy.org/publishers/azornes/nodes/layerforge)
|
|
||||||
2. Gather the required logs:
|
|
||||||
|
|
||||||
### 🔍 Enable Debug Logs (for **full** logs):
|
### 🔍 Enable Debug Logs (for **full** logs):
|
||||||
|
|
||||||
#### 1. Edit `config.js` (Frontend Logs):
|
#### 1. Edit `config.js` (Frontend Logs):
|
||||||
@@ -46,75 +101,13 @@ body:
|
|||||||
```
|
```
|
||||||
|
|
||||||
➡️ **Restart ComfyUI** after applying these changes to activate full logging.
|
➡️ **Restart ComfyUI** after applying these changes to activate full logging.
|
||||||
|
|
||||||
- type: input
|
|
||||||
id: environment
|
|
||||||
attributes:
|
|
||||||
label: Environment (OS, ComfyUI version, LayerForge version)
|
|
||||||
placeholder: e.g. Windows 11, ComfyUI v0.3.43, LayerForge v1.2.4
|
|
||||||
validations:
|
|
||||||
required: true
|
|
||||||
|
|
||||||
- type: input
|
|
||||||
id: browser
|
|
||||||
attributes:
|
|
||||||
label: Browser & Version
|
|
||||||
placeholder: e.g. Chrome 115.0.0, Firefox 120.1.0
|
|
||||||
validations:
|
|
||||||
required: true
|
|
||||||
|
|
||||||
- type: textarea
|
- type: textarea
|
||||||
id: what_happened
|
id: console-errors
|
||||||
attributes:
|
attributes:
|
||||||
label: What Happened?
|
label: Console Errors
|
||||||
placeholder: Describe the issue you encountered
|
|
||||||
validations:
|
|
||||||
required: true
|
|
||||||
|
|
||||||
- type: textarea
|
|
||||||
id: steps
|
|
||||||
attributes:
|
|
||||||
label: Steps to Reproduce
|
|
||||||
placeholder: |
|
|
||||||
1. …
|
|
||||||
2. …
|
|
||||||
3. …
|
|
||||||
validations:
|
|
||||||
required: true
|
|
||||||
|
|
||||||
- type: textarea
|
|
||||||
id: expected
|
|
||||||
attributes:
|
|
||||||
label: Expected Behavior
|
|
||||||
placeholder: Describe what you expected to happen
|
|
||||||
validations:
|
|
||||||
required: true
|
|
||||||
|
|
||||||
- type: textarea
|
|
||||||
id: actual
|
|
||||||
attributes:
|
|
||||||
label: Actual Behavior
|
|
||||||
placeholder: Describe what happened instead
|
|
||||||
validations:
|
|
||||||
required: true
|
|
||||||
|
|
||||||
- type: textarea
|
|
||||||
id: backend_logs
|
|
||||||
attributes:
|
|
||||||
label: ComfyUI (Backend) Logs
|
|
||||||
description: |
|
|
||||||
After enabling DEBUG logs, please:
|
|
||||||
1. Restart ComfyUI.
|
|
||||||
2. Reproduce the issue.
|
|
||||||
3. Copy-paste the newest **TEXT** logs from the terminal/console here.
|
|
||||||
validations:
|
|
||||||
required: true
|
|
||||||
|
|
||||||
- type: textarea
|
|
||||||
id: console_logs
|
|
||||||
attributes:
|
|
||||||
label: Browser Console Logs
|
|
||||||
description: |
|
description: |
|
||||||
|
If you see red error messages in the browser console (F12), paste them here
|
||||||
|
More info:
|
||||||
After enabling DEBUG logs:
|
After enabling DEBUG logs:
|
||||||
1. Open Developer Tools → Console.
|
1. Open Developer Tools → Console.
|
||||||
- Chrome/Edge (Win/Linux): `Ctrl+Shift+J`
|
- Chrome/Edge (Win/Linux): `Ctrl+Shift+J`
|
||||||
@@ -128,11 +121,25 @@ body:
|
|||||||
- Safari: 🗑 icon or `Cmd+K`.
|
- Safari: 🗑 icon or `Cmd+K`.
|
||||||
3. Reproduce the issue.
|
3. Reproduce the issue.
|
||||||
4. Copy-paste the **TEXT** logs here (no screenshots).
|
4. Copy-paste the **TEXT** logs here (no screenshots).
|
||||||
validations:
|
render: javascript
|
||||||
required: true
|
|
||||||
|
|
||||||
- type: markdown
|
- type: textarea
|
||||||
|
id: logs
|
||||||
attributes:
|
attributes:
|
||||||
value: |
|
label: Logs
|
||||||
**Optional:** You can also **attach a screenshot or video** to demonstrate the issue visually.
|
description: |
|
||||||
Simply drag & drop or paste image/video files into this issue form. GitHub supports common image formats and MP4/GIF files.
|
If relevant, paste any terminal/server logs here
|
||||||
|
More info:
|
||||||
|
After enabling DEBUG logs, please:
|
||||||
|
1. Restart ComfyUI.
|
||||||
|
2. Reproduce the issue.
|
||||||
|
3. Copy-paste the newest **TEXT** logs from the terminal/console here.
|
||||||
|
render: shell
|
||||||
|
|
||||||
|
- type: textarea
|
||||||
|
id: additional
|
||||||
|
attributes:
|
||||||
|
label: Additional Context, Environment (OS, ComfyUI versions, Comfyui-LayerForge version)
|
||||||
|
description: Any other information that might help (OS, GPU, specific nodes involved, etc.)
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
14
.github/ISSUE_TEMPLATE/feature-request.yml
vendored
14
.github/ISSUE_TEMPLATE/feature-request.yml
vendored
@@ -6,7 +6,19 @@ body:
|
|||||||
- type: markdown
|
- type: markdown
|
||||||
attributes:
|
attributes:
|
||||||
value: |
|
value: |
|
||||||
**Before suggesting a new feature, please make sure you are using the latest version of the project and that this functionality does not already exist.**
|
## Before suggesting a new feature...
|
||||||
|
Please make sure of the following:
|
||||||
|
|
||||||
|
1. You are using the latest version of the project
|
||||||
|
2. The functionality you want to propose does not already exist
|
||||||
|
|
||||||
|
I also recommend using an AI assistant to check whether the feature is already included.
|
||||||
|
To do this, simply:
|
||||||
|
|
||||||
|
- Copy and paste the entire **README.md** file
|
||||||
|
- Ask if your desired feature is already covered
|
||||||
|
|
||||||
|
This helps to avoid duplicate requests for features that are already available.
|
||||||
- type: markdown
|
- type: markdown
|
||||||
attributes:
|
attributes:
|
||||||
value: |
|
value: |
|
||||||
|
|||||||
2
.github/workflows/ComfyUIdownloads.yml
vendored
2
.github/workflows/ComfyUIdownloads.yml
vendored
@@ -20,7 +20,7 @@ jobs:
|
|||||||
max_downloads=0
|
max_downloads=0
|
||||||
top_node_json="{}"
|
top_node_json="{}"
|
||||||
|
|
||||||
for i in {1..20}; do
|
for i in {1..3}; do
|
||||||
echo "Pobieranie danych z próby $i..."
|
echo "Pobieranie danych z próby $i..."
|
||||||
curl -s https://api.comfy.org/nodes/layerforge > tmp_$i.json
|
curl -s https://api.comfy.org/nodes/layerforge > tmp_$i.json
|
||||||
|
|
||||||
|
|||||||
61
README.md
61
README.md
@@ -19,6 +19,15 @@
|
|||||||
<img alt="JavaScript" src="https://img.shields.io/badge/-JavaScript-000000?logo=javascript&logoColor=F7DF1E&style=for-the-badge&logoWidth=20">
|
<img alt="JavaScript" src="https://img.shields.io/badge/-JavaScript-000000?logo=javascript&logoColor=F7DF1E&style=for-the-badge&logoWidth=20">
|
||||||
</p>
|
</p>
|
||||||
|
|
||||||
|
<p align="center">
|
||||||
|
<strong>🔹 <a href="https://github.com/Azornes/Comfyui-LayerForge?tab=readme-ov-file#-installation">Quick Start</a></strong>
|
||||||
|
|
|
||||||
|
<strong>🧩 <a href="https://github.com/Azornes/Comfyui-LayerForge?tab=readme-ov-file#-workflow-example">Workflow Example</a></strong>
|
||||||
|
|
|
||||||
|
<strong>⚠️ <a href="https://github.com/Azornes/Comfyui-LayerForge?tab=readme-ov-file#%EF%B8%8F-known-issues--compatibility">Known Issues</a></strong>
|
||||||
|
|
||||||
|
</p>
|
||||||
|
|
||||||
### Why LayerForge?
|
### Why LayerForge?
|
||||||
|
|
||||||
- **Full Creative Control:** Move beyond simple image inputs. Composite, mask, and blend multiple elements without
|
- **Full Creative Control:** Move beyond simple image inputs. Composite, mask, and blend multiple elements without
|
||||||
@@ -51,19 +60,27 @@ https://github.com/user-attachments/assets/9c7ce1de-873b-4a3b-8579-0fc67642af3a
|
|||||||
- **AI-Powered Matting:** Optional background removal for any layer using the `BiRefNet` model.
|
- **AI-Powered Matting:** Optional background removal for any layer using the `BiRefNet` model.
|
||||||
- **Efficient Memory Management:** An automatic garbage collection system cleans up unused image data to keep the
|
- **Efficient Memory Management:** An automatic garbage collection system cleans up unused image data to keep the
|
||||||
browser's storage footprint low.
|
browser's storage footprint low.
|
||||||
- **Workflow Integration:** Outputs a final composite **image** and a combined alpha **mask**, ready for any other
|
- **Inputs**
|
||||||
ComfyUI node.
|
- **Image Input (optional):** Accepts a single image.
|
||||||
|
- **Multiple Images:** If you need to feed in more than one image, use the **core ComfyUI Batch Image node**.
|
||||||
|
- This lets you route multiple images into LayerForge.
|
||||||
|
- You can then stack, arrange, and edit them as separate layers inside the canvas.
|
||||||
|
- **Mask Input (optional):** Accepts a single external mask.
|
||||||
|
- When provided, the mask is applied directly to the **output area** of the LayerForge canvas when `Run` is triggered in ComfyUI.
|
||||||
|
- **Outputs**
|
||||||
|
- **Composite Image:** The final flattened result of your layer stack.
|
||||||
|
- **Combined Alpha Mask:** A merged mask representing all visible layers, ready for downstream nodes.
|
||||||
---
|
---
|
||||||
|
|
||||||
## 🚀 Installation
|
## 🚀 Installation
|
||||||
|
|
||||||
### Install via ComfyUI-Manager
|
### Install via ComfyUI-Manager
|
||||||
* Search `Comfyui-LayerForge` in ComfyUI-Manager and click `Install` button.
|
1. Search `Comfyui-LayerForge` in ComfyUI-Manager and click `Install` button.
|
||||||
|
2. Restart ComfyUI.
|
||||||
|
|
||||||
### Manual Install
|
### Manual Install
|
||||||
1. Install [ComfyUi](https://github.com/comfyanonymous/ComfyUI).
|
1. Install [ComfyUi](https://github.com/comfyanonymous/ComfyUI). I use [portable](https://docs.comfy.org/installation/comfyui_portable_windows) version.
|
||||||
2. Clone this repo into `custom_modules`:
|
2. Clone this repo into `custom_nodes`:
|
||||||
```bash
|
```bash
|
||||||
cd ComfyUI/custom_nodes/
|
cd ComfyUI/custom_nodes/
|
||||||
git clone https://github.com/Azornes/Comfyui-LayerForge.git
|
git clone https://github.com/Azornes/Comfyui-LayerForge.git
|
||||||
@@ -223,18 +240,24 @@ optional feature and requires a model.
|
|||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
## 🔧 Troubleshooting
|
## ⚠️ Known Issues / Compatibility
|
||||||
|
|
||||||
### `node_id` not auto-filled → black output
|
#### ○ Incompatibility with Modern Node Design (Vue Nodes)
|
||||||
|
> This node is **not compatible** with the new Vue Nodes display system.
|
||||||
|
>
|
||||||
|
> 🔧 **How to fix:**
|
||||||
|
> Go to **Settings → (search) "Vue Nodes" → Disable "Modern Node Design (Vue Nodes)"**.
|
||||||
|
|
||||||
In some cases, **ComfyUI doesn't auto-fill the `node_id`** when adding a node.
|
---
|
||||||
As a result, the node may produce a **completely black image** or not work at all.
|
|
||||||
|
|
||||||
**Workaround:**
|
#### ○ `node_id` not auto-filled → black output
|
||||||
|
> In some cases, **ComfyUI doesn’t auto-fill the `node_id`** when adding a node.
|
||||||
* Search node ID in ComfyUI settings.
|
> This may cause the node to output a **completely black image** or fail to work.
|
||||||
* In NodesMap check "Enable node ID display"
|
>
|
||||||
* Manually enter the correct `node_id` (match the ID Node "LayerForge" shown above the node, on the right side).
|
> 🛠️ **Workaround:**
|
||||||
|
> - Open **Settings → NodesMap → Enable "Show node IDs"**
|
||||||
|
> - Find the correct ID for your node *(match the ID Node "LayerForge" shown above the node, on the right side)*.
|
||||||
|
> - Manually enter the correct `node_id` in the LayerForge node
|
||||||
|
|
||||||
> [!WARNING]
|
> [!WARNING]
|
||||||
> This is a known issue and not yet fixed.
|
> This is a known issue and not yet fixed.
|
||||||
@@ -248,6 +271,14 @@ This project is licensed under the MIT License. Feel free to use, modify, and di
|
|||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
|
## 💖 Support / Sponsorship
|
||||||
|
• ⭐ Give a star — it means a lot to me!
|
||||||
|
• 🐛 Report a bug or suggest a feature
|
||||||
|
• 💖 If you’d like to support my work:
|
||||||
|
👉 [GitHub Sponsors](https://github.com/sponsors/Azornes)
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
## 🙏 Acknowledgments
|
## 🙏 Acknowledgments
|
||||||
|
|
||||||
Based on the original [**Comfyui-Ycanvas**](https://github.com/yichengup/Comfyui-Ycanvas) by yichengup. This fork
|
Based on the original [**Comfyui-Ycanvas**](https://github.com/yichengup/Comfyui-Ycanvas) by yichengup. This fork
|
||||||
|
|||||||
109
canvas_node.py
109
canvas_node.py
@@ -64,6 +64,8 @@ class BiRefNetConfig(PretrainedConfig):
|
|||||||
|
|
||||||
def __init__(self, bb_pretrained=False, **kwargs):
|
def __init__(self, bb_pretrained=False, **kwargs):
|
||||||
self.bb_pretrained = bb_pretrained
|
self.bb_pretrained = bb_pretrained
|
||||||
|
# Add the missing is_encoder_decoder attribute for compatibility with newer transformers
|
||||||
|
self.is_encoder_decoder = False
|
||||||
super().__init__(**kwargs)
|
super().__init__(**kwargs)
|
||||||
|
|
||||||
|
|
||||||
@@ -755,16 +757,32 @@ class BiRefNetMatting:
|
|||||||
full_model_path = os.path.join(self.base_path, "BiRefNet")
|
full_model_path = os.path.join(self.base_path, "BiRefNet")
|
||||||
log_info(f"Loading BiRefNet model from {full_model_path}...")
|
log_info(f"Loading BiRefNet model from {full_model_path}...")
|
||||||
try:
|
try:
|
||||||
|
# Try loading with additional configuration to handle compatibility issues
|
||||||
self.model = AutoModelForImageSegmentation.from_pretrained(
|
self.model = AutoModelForImageSegmentation.from_pretrained(
|
||||||
"ZhengPeng7/BiRefNet",
|
"ZhengPeng7/BiRefNet",
|
||||||
trust_remote_code=True,
|
trust_remote_code=True,
|
||||||
cache_dir=full_model_path
|
cache_dir=full_model_path,
|
||||||
|
# Add force_download=False to use cached version if available
|
||||||
|
force_download=False,
|
||||||
|
# Add local_files_only=False to allow downloading if needed
|
||||||
|
local_files_only=False
|
||||||
)
|
)
|
||||||
self.model.eval()
|
self.model.eval()
|
||||||
if torch.cuda.is_available():
|
if torch.cuda.is_available():
|
||||||
self.model = self.model.cuda()
|
self.model = self.model.cuda()
|
||||||
self.model_cache[model_path] = self.model
|
self.model_cache[model_path] = self.model
|
||||||
log_info("Model loaded successfully from Hugging Face")
|
log_info("Model loaded successfully from Hugging Face")
|
||||||
|
except AttributeError as e:
|
||||||
|
if "'Config' object has no attribute 'is_encoder_decoder'" in str(e):
|
||||||
|
log_error("Compatibility issue detected with transformers library. This has been fixed in the code.")
|
||||||
|
log_error("If you're still seeing this error, please clear the model cache and try again.")
|
||||||
|
raise RuntimeError(
|
||||||
|
"Model configuration compatibility issue detected. "
|
||||||
|
f"Please delete the model cache directory '{full_model_path}' and restart ComfyUI. "
|
||||||
|
"This will download a fresh copy of the model with the updated configuration."
|
||||||
|
) from e
|
||||||
|
else:
|
||||||
|
raise e
|
||||||
except JSONDecodeError as e:
|
except JSONDecodeError as e:
|
||||||
log_error(f"JSONDecodeError: Failed to load model from {full_model_path}. The model's config.json may be corrupted.")
|
log_error(f"JSONDecodeError: Failed to load model from {full_model_path}. The model's config.json may be corrupted.")
|
||||||
raise RuntimeError(
|
raise RuntimeError(
|
||||||
@@ -894,6 +912,95 @@ class BiRefNetMatting:
|
|||||||
|
|
||||||
_matting_lock = None
|
_matting_lock = None
|
||||||
|
|
||||||
|
@PromptServer.instance.routes.get("/matting/check-model")
|
||||||
|
async def check_matting_model(request):
|
||||||
|
"""Check if the matting model is available and ready to use"""
|
||||||
|
try:
|
||||||
|
if not TRANSFORMERS_AVAILABLE:
|
||||||
|
return web.json_response({
|
||||||
|
"available": False,
|
||||||
|
"reason": "missing_dependency",
|
||||||
|
"message": "The 'transformers' library is required for the matting feature. Please install it by running: pip install transformers"
|
||||||
|
})
|
||||||
|
|
||||||
|
# Check if model exists in cache
|
||||||
|
base_path = os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(os.path.abspath(__file__)))), "models")
|
||||||
|
model_path = os.path.join(base_path, "BiRefNet")
|
||||||
|
|
||||||
|
# Look for the actual BiRefNet model structure
|
||||||
|
model_files_exist = False
|
||||||
|
if os.path.exists(model_path):
|
||||||
|
# BiRefNet model from Hugging Face has a specific structure
|
||||||
|
# Check for subdirectories that indicate the model is downloaded
|
||||||
|
existing_items = os.listdir(model_path) if os.path.isdir(model_path) else []
|
||||||
|
|
||||||
|
# Look for the model subdirectory (usually named with the model ID)
|
||||||
|
model_subdirs = [d for d in existing_items if os.path.isdir(os.path.join(model_path, d)) and
|
||||||
|
(d.startswith("models--") or d == "ZhengPeng7--BiRefNet")]
|
||||||
|
|
||||||
|
if model_subdirs:
|
||||||
|
# Found model subdirectory, check inside for actual model files
|
||||||
|
for subdir in model_subdirs:
|
||||||
|
subdir_path = os.path.join(model_path, subdir)
|
||||||
|
# Navigate through the cache structure
|
||||||
|
if os.path.exists(os.path.join(subdir_path, "snapshots")):
|
||||||
|
snapshots_path = os.path.join(subdir_path, "snapshots")
|
||||||
|
snapshot_dirs = os.listdir(snapshots_path) if os.path.isdir(snapshots_path) else []
|
||||||
|
|
||||||
|
for snapshot in snapshot_dirs:
|
||||||
|
snapshot_path = os.path.join(snapshots_path, snapshot)
|
||||||
|
snapshot_files = os.listdir(snapshot_path) if os.path.isdir(snapshot_path) else []
|
||||||
|
|
||||||
|
# Check for essential files - BiRefNet uses model.safetensors
|
||||||
|
has_config = "config.json" in snapshot_files
|
||||||
|
has_model = "model.safetensors" in snapshot_files or "pytorch_model.bin" in snapshot_files
|
||||||
|
has_backbone = "backbone_swin.pth" in snapshot_files or "swin_base_patch4_window12_384_22kto1k.pth" in snapshot_files
|
||||||
|
has_birefnet = "birefnet.pth" in snapshot_files or any(f.endswith(".pth") for f in snapshot_files)
|
||||||
|
|
||||||
|
# Model is valid if it has config and either model.safetensors or other model files
|
||||||
|
if has_config and (has_model or has_backbone or has_birefnet):
|
||||||
|
model_files_exist = True
|
||||||
|
log_info(f"Found model files in: {snapshot_path} (config: {has_config}, model: {has_model})")
|
||||||
|
break
|
||||||
|
|
||||||
|
if model_files_exist:
|
||||||
|
break
|
||||||
|
|
||||||
|
# Also check if there are .pth files directly in the model_path
|
||||||
|
if not model_files_exist:
|
||||||
|
direct_files = existing_items
|
||||||
|
has_config = "config.json" in direct_files
|
||||||
|
has_model_files = any(f.endswith((".pth", ".bin", ".safetensors")) for f in direct_files)
|
||||||
|
model_files_exist = has_config and has_model_files
|
||||||
|
|
||||||
|
if model_files_exist:
|
||||||
|
log_info(f"Found model files directly in: {model_path}")
|
||||||
|
|
||||||
|
if model_files_exist:
|
||||||
|
# Model files exist, assume it's ready
|
||||||
|
log_info("BiRefNet model files detected")
|
||||||
|
return web.json_response({
|
||||||
|
"available": True,
|
||||||
|
"reason": "ready",
|
||||||
|
"message": "Model is ready to use"
|
||||||
|
})
|
||||||
|
else:
|
||||||
|
log_info(f"BiRefNet model not found in {model_path}")
|
||||||
|
return web.json_response({
|
||||||
|
"available": False,
|
||||||
|
"reason": "not_downloaded",
|
||||||
|
"message": "The matting model needs to be downloaded. This will happen automatically when you first use the matting feature (requires internet connection).",
|
||||||
|
"model_path": model_path
|
||||||
|
})
|
||||||
|
|
||||||
|
except Exception as e:
|
||||||
|
log_error(f"Error checking matting model: {str(e)}")
|
||||||
|
return web.json_response({
|
||||||
|
"available": False,
|
||||||
|
"reason": "error",
|
||||||
|
"message": f"Error checking model status: {str(e)}"
|
||||||
|
}, status=500)
|
||||||
|
|
||||||
@PromptServer.instance.routes.post("/matting")
|
@PromptServer.instance.routes.post("/matting")
|
||||||
async def matting(request):
|
async def matting(request):
|
||||||
global _matting_lock
|
global _matting_lock
|
||||||
|
|||||||
@@ -443,8 +443,8 @@ export class Canvas {
|
|||||||
* Inicjalizuje podstawowe właściwości canvas
|
* Inicjalizuje podstawowe właściwości canvas
|
||||||
*/
|
*/
|
||||||
initCanvas() {
|
initCanvas() {
|
||||||
this.canvas.width = this.width;
|
// Don't set canvas.width/height here - let the render loop handle it based on clientWidth/clientHeight
|
||||||
this.canvas.height = this.height;
|
// this.width and this.height are for the OUTPUT AREA, not the display canvas
|
||||||
this.canvas.style.border = '1px solid black';
|
this.canvas.style.border = '1px solid black';
|
||||||
this.canvas.style.maxWidth = '100%';
|
this.canvas.style.maxWidth = '100%';
|
||||||
this.canvas.style.backgroundColor = '#606060';
|
this.canvas.style.backgroundColor = '#606060';
|
||||||
|
|||||||
@@ -197,6 +197,25 @@ export class CanvasIO {
|
|||||||
}
|
}
|
||||||
async _renderOutputData() {
|
async _renderOutputData() {
|
||||||
log.info("=== RENDERING OUTPUT DATA FOR COMFYUI ===");
|
log.info("=== RENDERING OUTPUT DATA FOR COMFYUI ===");
|
||||||
|
// Check if layers have valid images loaded, with retry logic
|
||||||
|
const maxRetries = 5;
|
||||||
|
const retryDelay = 200;
|
||||||
|
for (let attempt = 0; attempt < maxRetries; attempt++) {
|
||||||
|
const layersWithoutImages = this.canvas.layers.filter(layer => !layer.image || !layer.image.complete);
|
||||||
|
if (layersWithoutImages.length === 0) {
|
||||||
|
break; // All images loaded
|
||||||
|
}
|
||||||
|
if (attempt === 0) {
|
||||||
|
log.warn(`${layersWithoutImages.length} layer(s) have incomplete image data. Waiting for images to load...`);
|
||||||
|
}
|
||||||
|
if (attempt < maxRetries - 1) {
|
||||||
|
await new Promise(resolve => setTimeout(resolve, retryDelay));
|
||||||
|
}
|
||||||
|
else {
|
||||||
|
// Last attempt failed
|
||||||
|
throw new Error(`Canvas not ready after ${maxRetries} attempts: ${layersWithoutImages.length} layer(s) still have incomplete image data. Try waiting a moment and running again.`);
|
||||||
|
}
|
||||||
|
}
|
||||||
// Użyj zunifikowanych funkcji z CanvasLayers
|
// Użyj zunifikowanych funkcji z CanvasLayers
|
||||||
const imageBlob = await this.canvas.canvasLayers.getFlattenedCanvasAsBlob();
|
const imageBlob = await this.canvas.canvasLayers.getFlattenedCanvasAsBlob();
|
||||||
const maskBlob = await this.canvas.canvasLayers.getFlattenedMaskAsBlob();
|
const maskBlob = await this.canvas.canvasLayers.getFlattenedMaskAsBlob();
|
||||||
|
|||||||
@@ -41,6 +41,7 @@ export class CanvasInteractions {
|
|||||||
canvasMoveRect: null,
|
canvasMoveRect: null,
|
||||||
outputAreaTransformHandle: null,
|
outputAreaTransformHandle: null,
|
||||||
outputAreaTransformAnchor: { x: 0, y: 0 },
|
outputAreaTransformAnchor: { x: 0, y: 0 },
|
||||||
|
hoveringGrabIcon: false,
|
||||||
};
|
};
|
||||||
this.originalLayerPositions = new Map();
|
this.originalLayerPositions = new Map();
|
||||||
}
|
}
|
||||||
@@ -103,6 +104,8 @@ export class CanvasInteractions {
|
|||||||
// Add a blur event listener to the window to reset key states
|
// Add a blur event listener to the window to reset key states
|
||||||
window.addEventListener('blur', this.onBlur);
|
window.addEventListener('blur', this.onBlur);
|
||||||
document.addEventListener('paste', this.onPaste);
|
document.addEventListener('paste', this.onPaste);
|
||||||
|
// Intercept Ctrl+V during capture phase to handle layer paste before ComfyUI
|
||||||
|
document.addEventListener('keydown', this.onKeyDown, { capture: true });
|
||||||
this.canvas.canvas.addEventListener('mouseenter', this.onMouseEnter);
|
this.canvas.canvas.addEventListener('mouseenter', this.onMouseEnter);
|
||||||
this.canvas.canvas.addEventListener('mouseleave', this.onMouseLeave);
|
this.canvas.canvas.addEventListener('mouseleave', this.onMouseLeave);
|
||||||
this.canvas.canvas.addEventListener('dragover', this.onDragOver);
|
this.canvas.canvas.addEventListener('dragover', this.onDragOver);
|
||||||
@@ -118,6 +121,8 @@ export class CanvasInteractions {
|
|||||||
this.canvas.canvas.removeEventListener('wheel', this.onWheel);
|
this.canvas.canvas.removeEventListener('wheel', this.onWheel);
|
||||||
this.canvas.canvas.removeEventListener('keydown', this.onKeyDown);
|
this.canvas.canvas.removeEventListener('keydown', this.onKeyDown);
|
||||||
this.canvas.canvas.removeEventListener('keyup', this.onKeyUp);
|
this.canvas.canvas.removeEventListener('keyup', this.onKeyUp);
|
||||||
|
// Remove document-level capture listener
|
||||||
|
document.removeEventListener('keydown', this.onKeyDown, { capture: true });
|
||||||
window.removeEventListener('blur', this.onBlur);
|
window.removeEventListener('blur', this.onBlur);
|
||||||
document.removeEventListener('paste', this.onPaste);
|
document.removeEventListener('paste', this.onPaste);
|
||||||
this.canvas.canvas.removeEventListener('mouseenter', this.onMouseEnter);
|
this.canvas.canvas.removeEventListener('mouseenter', this.onMouseEnter);
|
||||||
@@ -151,6 +156,29 @@ export class CanvasInteractions {
|
|||||||
}
|
}
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
/**
|
||||||
|
* Sprawdza czy punkt znajduje się w obszarze ikony "grab" (środek layera)
|
||||||
|
* Zwraca layer, jeśli kliknięto w ikonę grab
|
||||||
|
*/
|
||||||
|
getGrabIconAtPosition(worldX, worldY) {
|
||||||
|
// Rozmiar ikony grab w pikselach światowych
|
||||||
|
const grabIconRadius = 20 / this.canvas.viewport.zoom;
|
||||||
|
for (const layer of this.canvas.canvasSelection.selectedLayers) {
|
||||||
|
if (!layer.visible)
|
||||||
|
continue;
|
||||||
|
const centerX = layer.x + layer.width / 2;
|
||||||
|
const centerY = layer.y + layer.height / 2;
|
||||||
|
// Sprawdź czy punkt jest w obszarze ikony grab (okrąg wokół środka)
|
||||||
|
const dx = worldX - centerX;
|
||||||
|
const dy = worldY - centerY;
|
||||||
|
const distanceSquared = dx * dx + dy * dy;
|
||||||
|
const radiusSquared = grabIconRadius * grabIconRadius;
|
||||||
|
if (distanceSquared <= radiusSquared) {
|
||||||
|
return layer;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return null;
|
||||||
|
}
|
||||||
resetInteractionState() {
|
resetInteractionState() {
|
||||||
this.interaction.mode = 'none';
|
this.interaction.mode = 'none';
|
||||||
this.interaction.resizeHandle = null;
|
this.interaction.resizeHandle = null;
|
||||||
@@ -164,6 +192,12 @@ export class CanvasInteractions {
|
|||||||
}
|
}
|
||||||
handleMouseDown(e) {
|
handleMouseDown(e) {
|
||||||
this.canvas.canvas.focus();
|
this.canvas.canvas.focus();
|
||||||
|
// Sync modifier states with actual event state to prevent "stuck" modifiers
|
||||||
|
// when focus moves between layers panel and canvas
|
||||||
|
this.interaction.isCtrlPressed = e.ctrlKey;
|
||||||
|
this.interaction.isMetaPressed = e.metaKey;
|
||||||
|
this.interaction.isShiftPressed = e.shiftKey;
|
||||||
|
this.interaction.isAltPressed = e.altKey;
|
||||||
const coords = this.getMouseCoordinates(e);
|
const coords = this.getMouseCoordinates(e);
|
||||||
const mods = this.getModifierState(e);
|
const mods = this.getModifierState(e);
|
||||||
if (this.interaction.mode === 'drawingMask') {
|
if (this.interaction.mode === 'drawingMask') {
|
||||||
@@ -227,6 +261,14 @@ export class CanvasInteractions {
|
|||||||
this.startLayerTransform(transformTarget.layer, transformTarget.handle, coords.world);
|
this.startLayerTransform(transformTarget.layer, transformTarget.handle, coords.world);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
// Check if clicking on grab icon of a selected layer
|
||||||
|
const grabIconLayer = this.getGrabIconAtPosition(coords.world.x, coords.world.y);
|
||||||
|
if (grabIconLayer) {
|
||||||
|
// Start dragging the selected layer(s) without changing selection
|
||||||
|
this.interaction.mode = 'potential-drag';
|
||||||
|
this.interaction.dragStart = { ...coords.world };
|
||||||
|
return;
|
||||||
|
}
|
||||||
const clickedLayerResult = this.canvas.canvasLayers.getLayerAtPosition(coords.world.x, coords.world.y);
|
const clickedLayerResult = this.canvas.canvasLayers.getLayerAtPosition(coords.world.x, coords.world.y);
|
||||||
if (clickedLayerResult) {
|
if (clickedLayerResult) {
|
||||||
this.prepareForDrag(clickedLayerResult.layer, coords.world);
|
this.prepareForDrag(clickedLayerResult.layer, coords.world);
|
||||||
@@ -282,6 +324,13 @@ export class CanvasInteractions {
|
|||||||
}
|
}
|
||||||
break;
|
break;
|
||||||
default:
|
default:
|
||||||
|
// Check if hovering over grab icon
|
||||||
|
const wasHovering = this.interaction.hoveringGrabIcon;
|
||||||
|
this.interaction.hoveringGrabIcon = this.getGrabIconAtPosition(coords.world.x, coords.world.y) !== null;
|
||||||
|
// Re-render if hover state changed to show/hide grab icon
|
||||||
|
if (wasHovering !== this.interaction.hoveringGrabIcon) {
|
||||||
|
this.canvas.render();
|
||||||
|
}
|
||||||
this.updateCursor(coords.world);
|
this.updateCursor(coords.world);
|
||||||
// Update brush cursor on overlay if mask tool is active
|
// Update brush cursor on overlay if mask tool is active
|
||||||
if (this.canvas.maskTool.isActive) {
|
if (this.canvas.maskTool.isActive) {
|
||||||
@@ -480,14 +529,24 @@ export class CanvasInteractions {
|
|||||||
return targetHeight / oldHeight;
|
return targetHeight / oldHeight;
|
||||||
}
|
}
|
||||||
handleKeyDown(e) {
|
handleKeyDown(e) {
|
||||||
|
// Always track modifier keys regardless of focus
|
||||||
if (e.key === 'Control')
|
if (e.key === 'Control')
|
||||||
this.interaction.isCtrlPressed = true;
|
this.interaction.isCtrlPressed = true;
|
||||||
if (e.key === 'Meta')
|
if (e.key === 'Meta')
|
||||||
this.interaction.isMetaPressed = true;
|
this.interaction.isMetaPressed = true;
|
||||||
if (e.key === 'Shift')
|
if (e.key === 'Shift')
|
||||||
this.interaction.isShiftPressed = true;
|
this.interaction.isShiftPressed = true;
|
||||||
if (e.key === 'Alt') {
|
if (e.key === 'Alt')
|
||||||
this.interaction.isAltPressed = true;
|
this.interaction.isAltPressed = true;
|
||||||
|
// Check if canvas is focused before handling any shortcuts
|
||||||
|
const shouldHandle = this.canvas.isMouseOver ||
|
||||||
|
this.canvas.canvas.contains(document.activeElement) ||
|
||||||
|
document.activeElement === this.canvas.canvas;
|
||||||
|
if (!shouldHandle) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
// Canvas-specific key handlers (only when focused)
|
||||||
|
if (e.key === 'Alt') {
|
||||||
e.preventDefault();
|
e.preventDefault();
|
||||||
}
|
}
|
||||||
if (e.key.toLowerCase() === 's') {
|
if (e.key.toLowerCase() === 's') {
|
||||||
@@ -521,6 +580,17 @@ export class CanvasInteractions {
|
|||||||
this.canvas.canvasLayers.copySelectedLayers();
|
this.canvas.canvasLayers.copySelectedLayers();
|
||||||
}
|
}
|
||||||
break;
|
break;
|
||||||
|
case 'v':
|
||||||
|
// Only handle internal clipboard paste here.
|
||||||
|
// If internal clipboard is empty, let the paste event bubble
|
||||||
|
// so handlePasteEvent can access e.clipboardData for system images.
|
||||||
|
if (this.canvas.canvasLayers.internalClipboard.length > 0) {
|
||||||
|
this.canvas.canvasLayers.pasteLayers();
|
||||||
|
} else {
|
||||||
|
// Don't preventDefault - let paste event fire for system clipboard
|
||||||
|
handled = false;
|
||||||
|
}
|
||||||
|
break;
|
||||||
default:
|
default:
|
||||||
handled = false;
|
handled = false;
|
||||||
break;
|
break;
|
||||||
@@ -617,6 +687,11 @@ export class CanvasInteractions {
|
|||||||
this.canvas.canvas.style.cursor = 'grabbing';
|
this.canvas.canvas.style.cursor = 'grabbing';
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
// Check if hovering over grab icon
|
||||||
|
if (this.interaction.hoveringGrabIcon) {
|
||||||
|
this.canvas.canvas.style.cursor = 'grab';
|
||||||
|
return;
|
||||||
|
}
|
||||||
const transformTarget = this.canvas.canvasLayers.getHandleAtPosition(worldCoords.x, worldCoords.y);
|
const transformTarget = this.canvas.canvasLayers.getHandleAtPosition(worldCoords.x, worldCoords.y);
|
||||||
if (transformTarget) {
|
if (transformTarget) {
|
||||||
const handleName = transformTarget.handle;
|
const handleName = transformTarget.handle;
|
||||||
@@ -669,12 +744,11 @@ export class CanvasInteractions {
|
|||||||
if (mods.ctrl || mods.meta) {
|
if (mods.ctrl || mods.meta) {
|
||||||
const index = this.canvas.canvasSelection.selectedLayers.indexOf(layer);
|
const index = this.canvas.canvasSelection.selectedLayers.indexOf(layer);
|
||||||
if (index === -1) {
|
if (index === -1) {
|
||||||
|
// Ctrl-clicking unselected layer: add to selection
|
||||||
this.canvas.canvasSelection.updateSelection([...this.canvas.canvasSelection.selectedLayers, layer]);
|
this.canvas.canvasSelection.updateSelection([...this.canvas.canvasSelection.selectedLayers, layer]);
|
||||||
}
|
}
|
||||||
else {
|
// If already selected, do NOT deselect - allows dragging multiple layers with Ctrl held
|
||||||
const newSelection = this.canvas.canvasSelection.selectedLayers.filter((l) => l !== layer);
|
// User can use right-click in layers panel to deselect individual layers
|
||||||
this.canvas.canvasSelection.updateSelection(newSelection);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
else {
|
else {
|
||||||
if (!this.canvas.canvasSelection.selectedLayers.includes(layer)) {
|
if (!this.canvas.canvasSelection.selectedLayers.includes(layer)) {
|
||||||
@@ -1111,10 +1185,13 @@ export class CanvasInteractions {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
async handlePasteEvent(e) {
|
async handlePasteEvent(e) {
|
||||||
|
// Check if canvas is connected to DOM and visible
|
||||||
|
if (!this.canvas.canvas.isConnected || !document.body.contains(this.canvas.canvas)) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
const shouldHandle = this.canvas.isMouseOver ||
|
const shouldHandle = this.canvas.isMouseOver ||
|
||||||
this.canvas.canvas.contains(document.activeElement) ||
|
this.canvas.canvas.contains(document.activeElement) ||
|
||||||
document.activeElement === this.canvas.canvas ||
|
document.activeElement === this.canvas.canvas;
|
||||||
document.activeElement === document.body;
|
|
||||||
if (!shouldHandle) {
|
if (!shouldHandle) {
|
||||||
log.debug("Paste event ignored - not focused on canvas");
|
log.debug("Paste event ignored - not focused on canvas");
|
||||||
return;
|
return;
|
||||||
|
|||||||
@@ -1100,8 +1100,8 @@ export class CanvasLayers {
|
|||||||
this.canvas.width = width;
|
this.canvas.width = width;
|
||||||
this.canvas.height = height;
|
this.canvas.height = height;
|
||||||
this.canvas.maskTool.resize(width, height);
|
this.canvas.maskTool.resize(width, height);
|
||||||
this.canvas.canvas.width = width;
|
// Don't set canvas.width/height - the render loop will handle display size
|
||||||
this.canvas.canvas.height = height;
|
// this.canvas.width/height are for OUTPUT AREA dimensions, not display canvas
|
||||||
this.canvas.render();
|
this.canvas.render();
|
||||||
if (saveHistory) {
|
if (saveHistory) {
|
||||||
this.canvas.canvasState.saveStateToDB();
|
this.canvas.canvasState.saveStateToDB();
|
||||||
|
|||||||
@@ -123,6 +123,26 @@ export class CanvasLayersPanel {
|
|||||||
e.preventDefault();
|
e.preventDefault();
|
||||||
e.stopPropagation();
|
e.stopPropagation();
|
||||||
this.deleteSelectedLayers();
|
this.deleteSelectedLayers();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
// Handle Ctrl+C/V for layer copy/paste when panel has focus
|
||||||
|
if (e.ctrlKey || e.metaKey) {
|
||||||
|
if (e.key.toLowerCase() === 'c') {
|
||||||
|
if (this.canvas.canvasSelection.selectedLayers.length > 0) {
|
||||||
|
e.preventDefault();
|
||||||
|
e.stopPropagation();
|
||||||
|
this.canvas.canvasLayers.copySelectedLayers();
|
||||||
|
log.info('Layers copied from panel');
|
||||||
|
}
|
||||||
|
}
|
||||||
|
else if (e.key.toLowerCase() === 'v') {
|
||||||
|
e.preventDefault();
|
||||||
|
e.stopPropagation();
|
||||||
|
if (this.canvas.canvasLayers.internalClipboard.length > 0) {
|
||||||
|
this.canvas.canvasLayers.pasteLayers();
|
||||||
|
log.info('Layers pasted from panel');
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
log.debug('Panel structure created');
|
log.debug('Panel structure created');
|
||||||
@@ -329,6 +349,8 @@ export class CanvasLayersPanel {
|
|||||||
// Aktualizuj tylko wygląd (klasy CSS), bez niszczenia DOM
|
// Aktualizuj tylko wygląd (klasy CSS), bez niszczenia DOM
|
||||||
this.updateSelectionAppearance();
|
this.updateSelectionAppearance();
|
||||||
this.updateButtonStates();
|
this.updateButtonStates();
|
||||||
|
// Focus the canvas so keyboard shortcuts (like Ctrl+C/V) work for layer operations
|
||||||
|
this.canvas.canvas.focus();
|
||||||
log.debug(`Layer clicked: ${layer.name}, selection count: ${this.canvas.canvasSelection.selectedLayers.length}`);
|
log.debug(`Layer clicked: ${layer.name}, selection count: ${this.canvas.canvasSelection.selectedLayers.length}`);
|
||||||
}
|
}
|
||||||
startEditingLayerName(nameElement, layer) {
|
startEditingLayerName(nameElement, layer) {
|
||||||
|
|||||||
@@ -141,6 +141,10 @@ export class CanvasRenderer {
|
|||||||
ctx.restore();
|
ctx.restore();
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
// Draw grab icons for selected layers when hovering
|
||||||
|
if (this.canvas.canvasInteractions.interaction.hoveringGrabIcon) {
|
||||||
|
this.drawGrabIcons(ctx);
|
||||||
|
}
|
||||||
this.drawCanvasOutline(ctx);
|
this.drawCanvasOutline(ctx);
|
||||||
this.drawOutputAreaExtensionPreview(ctx); // Draw extension preview
|
this.drawOutputAreaExtensionPreview(ctx); // Draw extension preview
|
||||||
this.drawPendingGenerationAreas(ctx); // Draw snapshot outlines
|
this.drawPendingGenerationAreas(ctx); // Draw snapshot outlines
|
||||||
@@ -833,6 +837,55 @@ export class CanvasRenderer {
|
|||||||
// Just ensure it's the right size
|
// Just ensure it's the right size
|
||||||
this.updateOverlaySize();
|
this.updateOverlaySize();
|
||||||
}
|
}
|
||||||
|
/**
|
||||||
|
* Draw grab icons in the center of selected layers
|
||||||
|
*/
|
||||||
|
drawGrabIcons(ctx) {
|
||||||
|
const selectedLayers = this.canvas.canvasSelection.selectedLayers;
|
||||||
|
if (selectedLayers.length === 0)
|
||||||
|
return;
|
||||||
|
const iconRadius = 20 / this.canvas.viewport.zoom;
|
||||||
|
const innerRadius = 12 / this.canvas.viewport.zoom;
|
||||||
|
selectedLayers.forEach((layer) => {
|
||||||
|
if (!layer.visible)
|
||||||
|
return;
|
||||||
|
const centerX = layer.x + layer.width / 2;
|
||||||
|
const centerY = layer.y + layer.height / 2;
|
||||||
|
ctx.save();
|
||||||
|
// Draw outer circle (background)
|
||||||
|
ctx.beginPath();
|
||||||
|
ctx.arc(centerX, centerY, iconRadius, 0, Math.PI * 2);
|
||||||
|
ctx.fillStyle = 'rgba(0, 150, 255, 0.7)';
|
||||||
|
ctx.fill();
|
||||||
|
ctx.strokeStyle = 'rgba(255, 255, 255, 0.9)';
|
||||||
|
ctx.lineWidth = 2 / this.canvas.viewport.zoom;
|
||||||
|
ctx.stroke();
|
||||||
|
// Draw hand/grab icon (simplified)
|
||||||
|
ctx.fillStyle = 'rgba(255, 255, 255, 0.95)';
|
||||||
|
ctx.strokeStyle = 'rgba(255, 255, 255, 0.95)';
|
||||||
|
ctx.lineWidth = 1.5 / this.canvas.viewport.zoom;
|
||||||
|
// Draw four dots representing grab points
|
||||||
|
const dotRadius = 2 / this.canvas.viewport.zoom;
|
||||||
|
const dotDistance = 6 / this.canvas.viewport.zoom;
|
||||||
|
// Top-left
|
||||||
|
ctx.beginPath();
|
||||||
|
ctx.arc(centerX - dotDistance, centerY - dotDistance, dotRadius, 0, Math.PI * 2);
|
||||||
|
ctx.fill();
|
||||||
|
// Top-right
|
||||||
|
ctx.beginPath();
|
||||||
|
ctx.arc(centerX + dotDistance, centerY - dotDistance, dotRadius, 0, Math.PI * 2);
|
||||||
|
ctx.fill();
|
||||||
|
// Bottom-left
|
||||||
|
ctx.beginPath();
|
||||||
|
ctx.arc(centerX - dotDistance, centerY + dotDistance, dotRadius, 0, Math.PI * 2);
|
||||||
|
ctx.fill();
|
||||||
|
// Bottom-right
|
||||||
|
ctx.beginPath();
|
||||||
|
ctx.arc(centerX + dotDistance, centerY + dotDistance, dotRadius, 0, Math.PI * 2);
|
||||||
|
ctx.fill();
|
||||||
|
ctx.restore();
|
||||||
|
});
|
||||||
|
}
|
||||||
/**
|
/**
|
||||||
* Draw transform handles for output area when in transform mode
|
* Draw transform handles for output area when in transform mode
|
||||||
*/
|
*/
|
||||||
|
|||||||
@@ -88,10 +88,10 @@ export class CanvasState {
|
|||||||
log.debug(`Output Area resized to ${this.canvas.width}x${this.canvas.height} and viewport set.`);
|
log.debug(`Output Area resized to ${this.canvas.width}x${this.canvas.height} and viewport set.`);
|
||||||
const loadedLayers = await this._loadLayers(savedState.layers);
|
const loadedLayers = await this._loadLayers(savedState.layers);
|
||||||
this.canvas.layers = loadedLayers.filter((l) => l !== null);
|
this.canvas.layers = loadedLayers.filter((l) => l !== null);
|
||||||
log.info(`Loaded ${this.canvas.layers.length} layers.`);
|
log.info(`Loaded ${this.canvas.layers.length} layers from ${savedState.layers.length} saved layers.`);
|
||||||
if (this.canvas.layers.length === 0) {
|
if (this.canvas.layers.length === 0 && savedState.layers.length > 0) {
|
||||||
log.warn("No valid layers loaded, state may be corrupted.");
|
log.warn(`Failed to load any layers. Saved state had ${savedState.layers.length} layers but all failed to load. This may indicate corrupted IndexedDB data.`);
|
||||||
return false;
|
// Don't return false - allow empty canvas to be valid
|
||||||
}
|
}
|
||||||
this.canvas.updateSelectionAfterHistory();
|
this.canvas.updateSelectionAfterHistory();
|
||||||
this.canvas.render();
|
this.canvas.render();
|
||||||
|
|||||||
162
js/CanvasView.js
162
js/CanvasView.js
@@ -343,11 +343,38 @@ async function createCanvasWidget(node, widget, app) {
|
|||||||
const button = e.target.closest('.matting-button');
|
const button = e.target.closest('.matting-button');
|
||||||
if (button.classList.contains('loading'))
|
if (button.classList.contains('loading'))
|
||||||
return;
|
return;
|
||||||
const spinner = $el("div.matting-spinner");
|
|
||||||
button.appendChild(spinner);
|
|
||||||
button.classList.add('loading');
|
|
||||||
showInfoNotification("Starting background removal process...", 2000);
|
|
||||||
try {
|
try {
|
||||||
|
// First check if model is available
|
||||||
|
const modelCheckResponse = await fetch("/matting/check-model");
|
||||||
|
const modelStatus = await modelCheckResponse.json();
|
||||||
|
if (!modelStatus.available) {
|
||||||
|
switch (modelStatus.reason) {
|
||||||
|
case 'missing_dependency':
|
||||||
|
showErrorNotification(modelStatus.message, 8000);
|
||||||
|
return;
|
||||||
|
case 'not_downloaded':
|
||||||
|
showWarningNotification("The matting model needs to be downloaded first. This will happen automatically when you proceed (requires internet connection).", 5000);
|
||||||
|
// Ask user if they want to proceed with download
|
||||||
|
if (!confirm("The matting model needs to be downloaded (about 1GB). This is a one-time download. Do you want to proceed?")) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
showInfoNotification("Downloading matting model... This may take a few minutes.", 10000);
|
||||||
|
break;
|
||||||
|
case 'corrupted':
|
||||||
|
showErrorNotification(modelStatus.message, 8000);
|
||||||
|
return;
|
||||||
|
case 'error':
|
||||||
|
showErrorNotification(`Error checking model: ${modelStatus.message}`, 5000);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// Proceed with matting
|
||||||
|
const spinner = $el("div.matting-spinner");
|
||||||
|
button.appendChild(spinner);
|
||||||
|
button.classList.add('loading');
|
||||||
|
if (modelStatus.available) {
|
||||||
|
showInfoNotification("Starting background removal process...", 2000);
|
||||||
|
}
|
||||||
if (canvas.canvasSelection.selectedLayers.length !== 1) {
|
if (canvas.canvasSelection.selectedLayers.length !== 1) {
|
||||||
throw new Error("Please select exactly one image layer for matting.");
|
throw new Error("Please select exactly one image layer for matting.");
|
||||||
}
|
}
|
||||||
@@ -363,7 +390,20 @@ async function createCanvasWidget(node, widget, app) {
|
|||||||
if (!response.ok) {
|
if (!response.ok) {
|
||||||
let errorMsg = `Server error: ${response.status} - ${response.statusText}`;
|
let errorMsg = `Server error: ${response.status} - ${response.statusText}`;
|
||||||
if (result && result.error) {
|
if (result && result.error) {
|
||||||
errorMsg = `Error: ${result.error}. Details: ${result.details || 'Check console'}`;
|
// Handle specific error types
|
||||||
|
if (result.error === "Network Connection Error") {
|
||||||
|
showErrorNotification("Failed to download the matting model. Please check your internet connection and try again.", 8000);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
else if (result.error === "Matting Model Error") {
|
||||||
|
showErrorNotification(result.details || "Model loading error. Please check the console for details.", 8000);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
else if (result.error === "Dependency Not Found") {
|
||||||
|
showErrorNotification(result.details || "Missing required dependencies.", 8000);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
errorMsg = `${result.error}: ${result.details || 'Check console'}`;
|
||||||
}
|
}
|
||||||
throw new Error(errorMsg);
|
throw new Error(errorMsg);
|
||||||
}
|
}
|
||||||
@@ -383,11 +423,16 @@ async function createCanvasWidget(node, widget, app) {
|
|||||||
catch (error) {
|
catch (error) {
|
||||||
log.error("Matting error:", error);
|
log.error("Matting error:", error);
|
||||||
const errorMessage = error.message || "An unknown error occurred.";
|
const errorMessage = error.message || "An unknown error occurred.";
|
||||||
showErrorNotification(`Matting Failed: ${errorMessage}`);
|
if (!errorMessage.includes("Network Connection Error") &&
|
||||||
|
!errorMessage.includes("Matting Model Error") &&
|
||||||
|
!errorMessage.includes("Dependency Not Found")) {
|
||||||
|
showErrorNotification(`Matting Failed: ${errorMessage}`);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
finally {
|
finally {
|
||||||
button.classList.remove('loading');
|
button.classList.remove('loading');
|
||||||
if (button.contains(spinner)) {
|
const spinner = button.querySelector('.matting-spinner');
|
||||||
|
if (spinner && button.contains(spinner)) {
|
||||||
button.removeChild(spinner);
|
button.removeChild(spinner);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -839,6 +884,12 @@ async function createCanvasWidget(node, widget, app) {
|
|||||||
if (controlsElement) {
|
if (controlsElement) {
|
||||||
resizeObserver.observe(controlsElement);
|
resizeObserver.observe(controlsElement);
|
||||||
}
|
}
|
||||||
|
// Watch the canvas container itself to detect size changes and fix canvas dimensions
|
||||||
|
const canvasContainerResizeObserver = new ResizeObserver(() => {
|
||||||
|
// Force re-read of canvas dimensions on next render
|
||||||
|
canvas.render();
|
||||||
|
});
|
||||||
|
canvasContainerResizeObserver.observe(canvasContainer);
|
||||||
canvas.canvas.addEventListener('focus', () => {
|
canvas.canvas.addEventListener('focus', () => {
|
||||||
canvasContainer.classList.add('has-focus');
|
canvasContainer.classList.add('has-focus');
|
||||||
});
|
});
|
||||||
@@ -993,13 +1044,20 @@ app.registerExtension({
|
|||||||
log.info(`Found ${canvasNodeInstances.size} CanvasNode(s). Sending data via WebSocket...`);
|
log.info(`Found ${canvasNodeInstances.size} CanvasNode(s). Sending data via WebSocket...`);
|
||||||
const sendPromises = [];
|
const sendPromises = [];
|
||||||
for (const [nodeId, canvasWidget] of canvasNodeInstances.entries()) {
|
for (const [nodeId, canvasWidget] of canvasNodeInstances.entries()) {
|
||||||
if (app.graph.getNodeById(nodeId) && canvasWidget.canvas && canvasWidget.canvas.canvasIO) {
|
const node = app.graph.getNodeById(nodeId);
|
||||||
log.debug(`Sending data for canvas node ${nodeId}`);
|
if (!node) {
|
||||||
sendPromises.push(canvasWidget.canvas.canvasIO.sendDataViaWebSocket(nodeId));
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
log.warn(`Node ${nodeId} not found in graph, removing from instances map.`);
|
log.warn(`Node ${nodeId} not found in graph, removing from instances map.`);
|
||||||
canvasNodeInstances.delete(nodeId);
|
canvasNodeInstances.delete(nodeId);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
// Skip bypassed nodes
|
||||||
|
if (node.mode === 4) {
|
||||||
|
log.debug(`Node ${nodeId} is bypassed, skipping data send.`);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
if (canvasWidget.canvas && canvasWidget.canvas.canvasIO) {
|
||||||
|
log.debug(`Sending data for canvas node ${nodeId}`);
|
||||||
|
sendPromises.push(canvasWidget.canvas.canvasIO.sendDataViaWebSocket(nodeId));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
try {
|
try {
|
||||||
@@ -1018,6 +1076,8 @@ app.registerExtension({
|
|||||||
},
|
},
|
||||||
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
async beforeRegisterNodeDef(nodeType, nodeData, app) {
|
||||||
if (nodeType.comfyClass === "LayerForgeNode") {
|
if (nodeType.comfyClass === "LayerForgeNode") {
|
||||||
|
// Map to track pending copy sources across node ID changes
|
||||||
|
const pendingCopySources = new Map();
|
||||||
const onNodeCreated = nodeType.prototype.onNodeCreated;
|
const onNodeCreated = nodeType.prototype.onNodeCreated;
|
||||||
nodeType.prototype.onNodeCreated = function () {
|
nodeType.prototype.onNodeCreated = function () {
|
||||||
log.debug("CanvasNode onNodeCreated: Base widget setup.");
|
log.debug("CanvasNode onNodeCreated: Base widget setup.");
|
||||||
@@ -1048,6 +1108,43 @@ app.registerExtension({
|
|||||||
log.info(`Registered CanvasNode instance for ID: ${this.id}`);
|
log.info(`Registered CanvasNode instance for ID: ${this.id}`);
|
||||||
// Store the canvas widget on the node
|
// Store the canvas widget on the node
|
||||||
this.canvasWidget = canvasWidget;
|
this.canvasWidget = canvasWidget;
|
||||||
|
// Check if this node has a pending copy source (from onConfigure)
|
||||||
|
// Check both the current ID and -1 (temporary ID during paste)
|
||||||
|
let sourceNodeId = pendingCopySources.get(this.id);
|
||||||
|
if (!sourceNodeId) {
|
||||||
|
sourceNodeId = pendingCopySources.get(-1);
|
||||||
|
if (sourceNodeId) {
|
||||||
|
// Transfer from -1 to the real ID and clear -1
|
||||||
|
pendingCopySources.delete(-1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if (sourceNodeId && sourceNodeId !== this.id) {
|
||||||
|
log.info(`Node ${this.id} will copy canvas state from node ${sourceNodeId}`);
|
||||||
|
// Clear the flag
|
||||||
|
pendingCopySources.delete(this.id);
|
||||||
|
// Copy the canvas state now that the widget is initialized
|
||||||
|
setTimeout(async () => {
|
||||||
|
try {
|
||||||
|
const { getCanvasState, setCanvasState } = await import('./db.js');
|
||||||
|
let sourceState = await getCanvasState(String(sourceNodeId));
|
||||||
|
// If source node doesn't exist (cross-workflow paste), try clipboard
|
||||||
|
if (!sourceState) {
|
||||||
|
log.debug(`No canvas state found for source node ${sourceNodeId}, checking clipboard`);
|
||||||
|
sourceState = await getCanvasState('__clipboard__');
|
||||||
|
}
|
||||||
|
if (!sourceState) {
|
||||||
|
log.debug(`No canvas state found in clipboard either`);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
await setCanvasState(String(this.id), sourceState);
|
||||||
|
await canvasWidget.canvas.loadInitialState();
|
||||||
|
log.info(`Canvas state copied successfully to node ${this.id}`);
|
||||||
|
}
|
||||||
|
catch (error) {
|
||||||
|
log.error(`Error copying canvas state:`, error);
|
||||||
|
}
|
||||||
|
}, 100);
|
||||||
|
}
|
||||||
// Check if there are already connected inputs
|
// Check if there are already connected inputs
|
||||||
setTimeout(() => {
|
setTimeout(() => {
|
||||||
if (this.inputs && this.inputs.length > 0) {
|
if (this.inputs && this.inputs.length > 0) {
|
||||||
@@ -1213,6 +1310,47 @@ app.registerExtension({
|
|||||||
}
|
}
|
||||||
return onRemoved?.apply(this, arguments);
|
return onRemoved?.apply(this, arguments);
|
||||||
};
|
};
|
||||||
|
// Handle copy/paste - save canvas state when copying
|
||||||
|
const originalSerialize = nodeType.prototype.serialize;
|
||||||
|
nodeType.prototype.serialize = function () {
|
||||||
|
const data = originalSerialize ? originalSerialize.apply(this) : {};
|
||||||
|
// Store a reference to the source node ID so we can copy layer data
|
||||||
|
data.sourceNodeId = this.id;
|
||||||
|
log.debug(`Serializing node ${this.id} for copy`);
|
||||||
|
// Store canvas state in a clipboard entry for cross-workflow paste
|
||||||
|
// This happens async but that's fine since paste happens later
|
||||||
|
(async () => {
|
||||||
|
try {
|
||||||
|
const { getCanvasState, setCanvasState } = await import('./db.js');
|
||||||
|
const sourceState = await getCanvasState(String(this.id));
|
||||||
|
if (sourceState) {
|
||||||
|
// Store in a special "clipboard" entry
|
||||||
|
await setCanvasState('__clipboard__', sourceState);
|
||||||
|
log.debug(`Stored canvas state in clipboard for node ${this.id}`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
catch (error) {
|
||||||
|
log.error('Error storing canvas state to clipboard:', error);
|
||||||
|
}
|
||||||
|
})();
|
||||||
|
return data;
|
||||||
|
};
|
||||||
|
// Handle copy/paste - load canvas state from source node when pasting
|
||||||
|
const originalConfigure = nodeType.prototype.onConfigure;
|
||||||
|
nodeType.prototype.onConfigure = async function (data) {
|
||||||
|
if (originalConfigure) {
|
||||||
|
originalConfigure.apply(this, [data]);
|
||||||
|
}
|
||||||
|
// Store the source node ID in the map (persists across node ID changes)
|
||||||
|
// This will be picked up later in onAdded when the canvas widget is ready
|
||||||
|
if (data.sourceNodeId && data.sourceNodeId !== this.id) {
|
||||||
|
const existingSource = pendingCopySources.get(this.id);
|
||||||
|
if (!existingSource) {
|
||||||
|
pendingCopySources.set(this.id, data.sourceNodeId);
|
||||||
|
log.debug(`Stored pending copy source: ${data.sourceNodeId} for node ${this.id}`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
const originalGetExtraMenuOptions = nodeType.prototype.getExtraMenuOptions;
|
const originalGetExtraMenuOptions = nodeType.prototype.getExtraMenuOptions;
|
||||||
nodeType.prototype.getExtraMenuOptions = function (_, options) {
|
nodeType.prototype.getExtraMenuOptions = function (_, options) {
|
||||||
// FIRST: Call original to let other extensions add their options
|
// FIRST: Call original to let other extensions add their options
|
||||||
|
|||||||
@@ -1,3 +1,4 @@
|
|||||||
|
// @ts-ignore
|
||||||
import { api } from "../../../scripts/api.js";
|
import { api } from "../../../scripts/api.js";
|
||||||
import { createModuleLogger } from "./LoggerUtils.js";
|
import { createModuleLogger } from "./LoggerUtils.js";
|
||||||
import { withErrorHandling, createValidationError, createNetworkError } from "../ErrorHandler.js";
|
import { withErrorHandling, createValidationError, createNetworkError } from "../ErrorHandler.js";
|
||||||
|
|||||||
@@ -1,7 +1,7 @@
|
|||||||
[project]
|
[project]
|
||||||
name = "layerforge"
|
name = "layerforge"
|
||||||
description = "Photoshop-like layered canvas editor to your ComfyUI workflow. This node is perfect for complex compositing, inpainting, and outpainting, featuring multi-layer support, masking, blend modes, and precise transformations. Includes optional AI-powered background removal for streamlined image editing."
|
description = "Photoshop-like layered canvas editor to your ComfyUI workflow. This node is perfect for complex compositing, inpainting, and outpainting, featuring multi-layer support, masking, blend modes, and precise transformations. Includes optional AI-powered background removal for streamlined image editing."
|
||||||
version = "1.5.9"
|
version = "1.5.11"
|
||||||
license = { text = "MIT License" }
|
license = { text = "MIT License" }
|
||||||
dependencies = ["torch", "torchvision", "transformers", "aiohttp", "numpy", "tqdm", "Pillow"]
|
dependencies = ["torch", "torchvision", "transformers", "aiohttp", "numpy", "tqdm", "Pillow"]
|
||||||
|
|
||||||
|
|||||||
@@ -578,8 +578,8 @@ export class Canvas {
|
|||||||
* Inicjalizuje podstawowe właściwości canvas
|
* Inicjalizuje podstawowe właściwości canvas
|
||||||
*/
|
*/
|
||||||
initCanvas() {
|
initCanvas() {
|
||||||
this.canvas.width = this.width;
|
// Don't set canvas.width/height here - let the render loop handle it based on clientWidth/clientHeight
|
||||||
this.canvas.height = this.height;
|
// this.width and this.height are for the OUTPUT AREA, not the display canvas
|
||||||
this.canvas.style.border = '1px solid black';
|
this.canvas.style.border = '1px solid black';
|
||||||
this.canvas.style.maxWidth = '100%';
|
this.canvas.style.maxWidth = '100%';
|
||||||
this.canvas.style.backgroundColor = '#606060';
|
this.canvas.style.backgroundColor = '#606060';
|
||||||
|
|||||||
@@ -218,6 +218,29 @@ export class CanvasIO {
|
|||||||
async _renderOutputData(): Promise<{ image: string, mask: string }> {
|
async _renderOutputData(): Promise<{ image: string, mask: string }> {
|
||||||
log.info("=== RENDERING OUTPUT DATA FOR COMFYUI ===");
|
log.info("=== RENDERING OUTPUT DATA FOR COMFYUI ===");
|
||||||
|
|
||||||
|
// Check if layers have valid images loaded, with retry logic
|
||||||
|
const maxRetries = 5;
|
||||||
|
const retryDelay = 200;
|
||||||
|
|
||||||
|
for (let attempt = 0; attempt < maxRetries; attempt++) {
|
||||||
|
const layersWithoutImages = this.canvas.layers.filter(layer => !layer.image || !layer.image.complete);
|
||||||
|
|
||||||
|
if (layersWithoutImages.length === 0) {
|
||||||
|
break; // All images loaded
|
||||||
|
}
|
||||||
|
|
||||||
|
if (attempt === 0) {
|
||||||
|
log.warn(`${layersWithoutImages.length} layer(s) have incomplete image data. Waiting for images to load...`);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (attempt < maxRetries - 1) {
|
||||||
|
await new Promise(resolve => setTimeout(resolve, retryDelay));
|
||||||
|
} else {
|
||||||
|
// Last attempt failed
|
||||||
|
throw new Error(`Canvas not ready after ${maxRetries} attempts: ${layersWithoutImages.length} layer(s) still have incomplete image data. Try waiting a moment and running again.`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
// Użyj zunifikowanych funkcji z CanvasLayers
|
// Użyj zunifikowanych funkcji z CanvasLayers
|
||||||
const imageBlob = await this.canvas.canvasLayers.getFlattenedCanvasAsBlob();
|
const imageBlob = await this.canvas.canvasLayers.getFlattenedCanvasAsBlob();
|
||||||
const maskBlob = await this.canvas.canvasLayers.getFlattenedMaskAsBlob();
|
const maskBlob = await this.canvas.canvasLayers.getFlattenedMaskAsBlob();
|
||||||
|
|||||||
@@ -51,6 +51,7 @@ interface InteractionState {
|
|||||||
canvasMoveRect: { x: number, y: number, width: number, height: number } | null;
|
canvasMoveRect: { x: number, y: number, width: number, height: number } | null;
|
||||||
outputAreaTransformHandle: string | null;
|
outputAreaTransformHandle: string | null;
|
||||||
outputAreaTransformAnchor: Point;
|
outputAreaTransformAnchor: Point;
|
||||||
|
hoveringGrabIcon: boolean;
|
||||||
}
|
}
|
||||||
|
|
||||||
export class CanvasInteractions {
|
export class CanvasInteractions {
|
||||||
@@ -98,6 +99,7 @@ export class CanvasInteractions {
|
|||||||
canvasMoveRect: null,
|
canvasMoveRect: null,
|
||||||
outputAreaTransformHandle: null,
|
outputAreaTransformHandle: null,
|
||||||
outputAreaTransformAnchor: { x: 0, y: 0 },
|
outputAreaTransformAnchor: { x: 0, y: 0 },
|
||||||
|
hoveringGrabIcon: false,
|
||||||
};
|
};
|
||||||
this.originalLayerPositions = new Map();
|
this.originalLayerPositions = new Map();
|
||||||
}
|
}
|
||||||
@@ -174,6 +176,9 @@ export class CanvasInteractions {
|
|||||||
|
|
||||||
document.addEventListener('paste', this.onPaste as unknown as EventListener);
|
document.addEventListener('paste', this.onPaste as unknown as EventListener);
|
||||||
|
|
||||||
|
// Intercept Ctrl+V during capture phase to handle layer paste before ComfyUI
|
||||||
|
document.addEventListener('keydown', this.onKeyDown as EventListener, { capture: true });
|
||||||
|
|
||||||
this.canvas.canvas.addEventListener('mouseenter', this.onMouseEnter as EventListener);
|
this.canvas.canvas.addEventListener('mouseenter', this.onMouseEnter as EventListener);
|
||||||
this.canvas.canvas.addEventListener('mouseleave', this.onMouseLeave as EventListener);
|
this.canvas.canvas.addEventListener('mouseleave', this.onMouseLeave as EventListener);
|
||||||
|
|
||||||
@@ -193,6 +198,9 @@ export class CanvasInteractions {
|
|||||||
this.canvas.canvas.removeEventListener('keydown', this.onKeyDown as EventListener);
|
this.canvas.canvas.removeEventListener('keydown', this.onKeyDown as EventListener);
|
||||||
this.canvas.canvas.removeEventListener('keyup', this.onKeyUp as EventListener);
|
this.canvas.canvas.removeEventListener('keyup', this.onKeyUp as EventListener);
|
||||||
|
|
||||||
|
// Remove document-level capture listener
|
||||||
|
document.removeEventListener('keydown', this.onKeyDown as EventListener, { capture: true });
|
||||||
|
|
||||||
window.removeEventListener('blur', this.onBlur);
|
window.removeEventListener('blur', this.onBlur);
|
||||||
document.removeEventListener('paste', this.onPaste as unknown as EventListener);
|
document.removeEventListener('paste', this.onPaste as unknown as EventListener);
|
||||||
|
|
||||||
@@ -234,6 +242,33 @@ export class CanvasInteractions {
|
|||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Sprawdza czy punkt znajduje się w obszarze ikony "grab" (środek layera)
|
||||||
|
* Zwraca layer, jeśli kliknięto w ikonę grab
|
||||||
|
*/
|
||||||
|
getGrabIconAtPosition(worldX: number, worldY: number): Layer | null {
|
||||||
|
// Rozmiar ikony grab w pikselach światowych
|
||||||
|
const grabIconRadius = 20 / this.canvas.viewport.zoom;
|
||||||
|
|
||||||
|
for (const layer of this.canvas.canvasSelection.selectedLayers) {
|
||||||
|
if (!layer.visible) continue;
|
||||||
|
|
||||||
|
const centerX = layer.x + layer.width / 2;
|
||||||
|
const centerY = layer.y + layer.height / 2;
|
||||||
|
|
||||||
|
// Sprawdź czy punkt jest w obszarze ikony grab (okrąg wokół środka)
|
||||||
|
const dx = worldX - centerX;
|
||||||
|
const dy = worldY - centerY;
|
||||||
|
const distanceSquared = dx * dx + dy * dy;
|
||||||
|
const radiusSquared = grabIconRadius * grabIconRadius;
|
||||||
|
|
||||||
|
if (distanceSquared <= radiusSquared) {
|
||||||
|
return layer;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return null;
|
||||||
|
}
|
||||||
|
|
||||||
resetInteractionState(): void {
|
resetInteractionState(): void {
|
||||||
this.interaction.mode = 'none';
|
this.interaction.mode = 'none';
|
||||||
this.interaction.resizeHandle = null;
|
this.interaction.resizeHandle = null;
|
||||||
@@ -248,6 +283,14 @@ export class CanvasInteractions {
|
|||||||
|
|
||||||
handleMouseDown(e: MouseEvent): void {
|
handleMouseDown(e: MouseEvent): void {
|
||||||
this.canvas.canvas.focus();
|
this.canvas.canvas.focus();
|
||||||
|
|
||||||
|
// Sync modifier states with actual event state to prevent "stuck" modifiers
|
||||||
|
// when focus moves between layers panel and canvas
|
||||||
|
this.interaction.isCtrlPressed = e.ctrlKey;
|
||||||
|
this.interaction.isMetaPressed = e.metaKey;
|
||||||
|
this.interaction.isShiftPressed = e.shiftKey;
|
||||||
|
this.interaction.isAltPressed = e.altKey;
|
||||||
|
|
||||||
const coords = this.getMouseCoordinates(e);
|
const coords = this.getMouseCoordinates(e);
|
||||||
const mods = this.getModifierState(e);
|
const mods = this.getModifierState(e);
|
||||||
|
|
||||||
@@ -320,6 +363,15 @@ export class CanvasInteractions {
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Check if clicking on grab icon of a selected layer
|
||||||
|
const grabIconLayer = this.getGrabIconAtPosition(coords.world.x, coords.world.y);
|
||||||
|
if (grabIconLayer) {
|
||||||
|
// Start dragging the selected layer(s) without changing selection
|
||||||
|
this.interaction.mode = 'potential-drag';
|
||||||
|
this.interaction.dragStart = { ...coords.world };
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
const clickedLayerResult = this.canvas.canvasLayers.getLayerAtPosition(coords.world.x, coords.world.y);
|
const clickedLayerResult = this.canvas.canvasLayers.getLayerAtPosition(coords.world.x, coords.world.y);
|
||||||
if (clickedLayerResult) {
|
if (clickedLayerResult) {
|
||||||
this.prepareForDrag(clickedLayerResult.layer, coords.world);
|
this.prepareForDrag(clickedLayerResult.layer, coords.world);
|
||||||
@@ -378,6 +430,15 @@ export class CanvasInteractions {
|
|||||||
}
|
}
|
||||||
break;
|
break;
|
||||||
default:
|
default:
|
||||||
|
// Check if hovering over grab icon
|
||||||
|
const wasHovering = this.interaction.hoveringGrabIcon;
|
||||||
|
this.interaction.hoveringGrabIcon = this.getGrabIconAtPosition(coords.world.x, coords.world.y) !== null;
|
||||||
|
|
||||||
|
// Re-render if hover state changed to show/hide grab icon
|
||||||
|
if (wasHovering !== this.interaction.hoveringGrabIcon) {
|
||||||
|
this.canvas.render();
|
||||||
|
}
|
||||||
|
|
||||||
this.updateCursor(coords.world);
|
this.updateCursor(coords.world);
|
||||||
// Update brush cursor on overlay if mask tool is active
|
// Update brush cursor on overlay if mask tool is active
|
||||||
if (this.canvas.maskTool.isActive) {
|
if (this.canvas.maskTool.isActive) {
|
||||||
@@ -599,11 +660,23 @@ export class CanvasInteractions {
|
|||||||
}
|
}
|
||||||
|
|
||||||
handleKeyDown(e: KeyboardEvent): void {
|
handleKeyDown(e: KeyboardEvent): void {
|
||||||
|
// Always track modifier keys regardless of focus
|
||||||
if (e.key === 'Control') this.interaction.isCtrlPressed = true;
|
if (e.key === 'Control') this.interaction.isCtrlPressed = true;
|
||||||
if (e.key === 'Meta') this.interaction.isMetaPressed = true;
|
if (e.key === 'Meta') this.interaction.isMetaPressed = true;
|
||||||
if (e.key === 'Shift') this.interaction.isShiftPressed = true;
|
if (e.key === 'Shift') this.interaction.isShiftPressed = true;
|
||||||
|
if (e.key === 'Alt') this.interaction.isAltPressed = true;
|
||||||
|
|
||||||
|
// Check if canvas is focused before handling any shortcuts
|
||||||
|
const shouldHandle = this.canvas.isMouseOver ||
|
||||||
|
this.canvas.canvas.contains(document.activeElement) ||
|
||||||
|
document.activeElement === this.canvas.canvas;
|
||||||
|
|
||||||
|
if (!shouldHandle) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Canvas-specific key handlers (only when focused)
|
||||||
if (e.key === 'Alt') {
|
if (e.key === 'Alt') {
|
||||||
this.interaction.isAltPressed = true;
|
|
||||||
e.preventDefault();
|
e.preventDefault();
|
||||||
}
|
}
|
||||||
if (e.key.toLowerCase() === 's') {
|
if (e.key.toLowerCase() === 's') {
|
||||||
@@ -638,6 +711,17 @@ export class CanvasInteractions {
|
|||||||
this.canvas.canvasLayers.copySelectedLayers();
|
this.canvas.canvasLayers.copySelectedLayers();
|
||||||
}
|
}
|
||||||
break;
|
break;
|
||||||
|
case 'v':
|
||||||
|
// Only handle internal clipboard paste here.
|
||||||
|
// If internal clipboard is empty, let the paste event bubble
|
||||||
|
// so handlePasteEvent can access e.clipboardData for system images.
|
||||||
|
if (this.canvas.canvasLayers.internalClipboard.length > 0) {
|
||||||
|
this.canvas.canvasLayers.pasteLayers();
|
||||||
|
} else {
|
||||||
|
// Don't preventDefault - let paste event fire for system clipboard
|
||||||
|
handled = false;
|
||||||
|
}
|
||||||
|
break;
|
||||||
default:
|
default:
|
||||||
handled = false;
|
handled = false;
|
||||||
break;
|
break;
|
||||||
@@ -738,6 +822,12 @@ export class CanvasInteractions {
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Check if hovering over grab icon
|
||||||
|
if (this.interaction.hoveringGrabIcon) {
|
||||||
|
this.canvas.canvas.style.cursor = 'grab';
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
const transformTarget = this.canvas.canvasLayers.getHandleAtPosition(worldCoords.x, worldCoords.y);
|
const transformTarget = this.canvas.canvasLayers.getHandleAtPosition(worldCoords.x, worldCoords.y);
|
||||||
|
|
||||||
if (transformTarget) {
|
if (transformTarget) {
|
||||||
@@ -767,7 +857,7 @@ export class CanvasInteractions {
|
|||||||
originalHeight: layer.originalHeight,
|
originalHeight: layer.originalHeight,
|
||||||
cropBounds: layer.cropBounds ? { ...layer.cropBounds } : undefined
|
cropBounds: layer.cropBounds ? { ...layer.cropBounds } : undefined
|
||||||
};
|
};
|
||||||
this.interaction.dragStart = {...worldCoords};
|
this.interaction.dragStart = { ...worldCoords };
|
||||||
|
|
||||||
if (handle === 'rot') {
|
if (handle === 'rot') {
|
||||||
this.interaction.mode = 'rotating';
|
this.interaction.mode = 'rotating';
|
||||||
@@ -791,11 +881,11 @@ export class CanvasInteractions {
|
|||||||
if (mods.ctrl || mods.meta) {
|
if (mods.ctrl || mods.meta) {
|
||||||
const index = this.canvas.canvasSelection.selectedLayers.indexOf(layer);
|
const index = this.canvas.canvasSelection.selectedLayers.indexOf(layer);
|
||||||
if (index === -1) {
|
if (index === -1) {
|
||||||
|
// Ctrl-clicking unselected layer: add to selection
|
||||||
this.canvas.canvasSelection.updateSelection([...this.canvas.canvasSelection.selectedLayers, layer]);
|
this.canvas.canvasSelection.updateSelection([...this.canvas.canvasSelection.selectedLayers, layer]);
|
||||||
} else {
|
|
||||||
const newSelection = this.canvas.canvasSelection.selectedLayers.filter((l: Layer) => l !== layer);
|
|
||||||
this.canvas.canvasSelection.updateSelection(newSelection);
|
|
||||||
}
|
}
|
||||||
|
// If already selected, do NOT deselect - allows dragging multiple layers with Ctrl held
|
||||||
|
// User can use right-click in layers panel to deselect individual layers
|
||||||
} else {
|
} else {
|
||||||
if (!this.canvas.canvasSelection.selectedLayers.includes(layer)) {
|
if (!this.canvas.canvasSelection.selectedLayers.includes(layer)) {
|
||||||
this.canvas.canvasSelection.updateSelection([layer]);
|
this.canvas.canvasSelection.updateSelection([layer]);
|
||||||
@@ -803,7 +893,7 @@ export class CanvasInteractions {
|
|||||||
}
|
}
|
||||||
|
|
||||||
this.interaction.mode = 'potential-drag';
|
this.interaction.mode = 'potential-drag';
|
||||||
this.interaction.dragStart = {...worldCoords};
|
this.interaction.dragStart = { ...worldCoords };
|
||||||
}
|
}
|
||||||
|
|
||||||
startPanning(e: MouseEvent, clearSelection: boolean = true): void {
|
startPanning(e: MouseEvent, clearSelection: boolean = true): void {
|
||||||
@@ -819,8 +909,8 @@ export class CanvasInteractions {
|
|||||||
this.interaction.mode = 'resizingCanvas';
|
this.interaction.mode = 'resizingCanvas';
|
||||||
const startX = snapToGrid(worldCoords.x);
|
const startX = snapToGrid(worldCoords.x);
|
||||||
const startY = snapToGrid(worldCoords.y);
|
const startY = snapToGrid(worldCoords.y);
|
||||||
this.interaction.canvasResizeStart = {x: startX, y: startY};
|
this.interaction.canvasResizeStart = { x: startX, y: startY };
|
||||||
this.interaction.canvasResizeRect = {x: startX, y: startY, width: 0, height: 0};
|
this.interaction.canvasResizeRect = { x: startX, y: startY, width: 0, height: 0 };
|
||||||
this.canvas.render();
|
this.canvas.render();
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -872,7 +962,7 @@ export class CanvasInteractions {
|
|||||||
const dy = e.clientY - this.interaction.panStart.y;
|
const dy = e.clientY - this.interaction.panStart.y;
|
||||||
this.canvas.viewport.x -= dx / this.canvas.viewport.zoom;
|
this.canvas.viewport.x -= dx / this.canvas.viewport.zoom;
|
||||||
this.canvas.viewport.y -= dy / this.canvas.viewport.zoom;
|
this.canvas.viewport.y -= dy / this.canvas.viewport.zoom;
|
||||||
this.interaction.panStart = {x: e.clientX, y: e.clientY};
|
this.interaction.panStart = { x: e.clientX, y: e.clientY };
|
||||||
|
|
||||||
// Update stroke overlay if mask tool is drawing during pan
|
// Update stroke overlay if mask tool is drawing during pan
|
||||||
if (this.canvas.maskTool.isDrawing) {
|
if (this.canvas.maskTool.isDrawing) {
|
||||||
@@ -1029,9 +1119,9 @@ export class CanvasInteractions {
|
|||||||
} else newCropBounds.height += delta_image_y;
|
} else newCropBounds.height += delta_image_y;
|
||||||
}
|
}
|
||||||
|
|
||||||
// Clamp crop bounds to stay within the original image and maintain minimum size
|
// Clamp crop bounds to stay within the original image and maintain minimum size
|
||||||
if (newCropBounds.width < 1) {
|
if (newCropBounds.width < 1) {
|
||||||
if (handle?.includes('w')) newCropBounds.x = o.cropBounds.x + o.cropBounds.width -1;
|
if (handle?.includes('w')) newCropBounds.x = o.cropBounds.x + o.cropBounds.width - 1;
|
||||||
newCropBounds.width = 1;
|
newCropBounds.width = 1;
|
||||||
}
|
}
|
||||||
if (newCropBounds.height < 1) {
|
if (newCropBounds.height < 1) {
|
||||||
@@ -1289,11 +1379,14 @@ export class CanvasInteractions {
|
|||||||
}
|
}
|
||||||
|
|
||||||
async handlePasteEvent(e: ClipboardEvent): Promise<void> {
|
async handlePasteEvent(e: ClipboardEvent): Promise<void> {
|
||||||
|
// Check if canvas is connected to DOM and visible
|
||||||
|
if (!this.canvas.canvas.isConnected || !document.body.contains(this.canvas.canvas)) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
const shouldHandle = this.canvas.isMouseOver ||
|
const shouldHandle = this.canvas.isMouseOver ||
|
||||||
this.canvas.canvas.contains(document.activeElement) ||
|
this.canvas.canvas.contains(document.activeElement) ||
|
||||||
document.activeElement === this.canvas.canvas ||
|
document.activeElement === this.canvas.canvas;
|
||||||
document.activeElement === document.body;
|
|
||||||
|
|
||||||
if (!shouldHandle) {
|
if (!shouldHandle) {
|
||||||
log.debug("Paste event ignored - not focused on canvas");
|
log.debug("Paste event ignored - not focused on canvas");
|
||||||
|
|||||||
@@ -1263,8 +1263,8 @@ export class CanvasLayers {
|
|||||||
this.canvas.height = height;
|
this.canvas.height = height;
|
||||||
this.canvas.maskTool.resize(width, height);
|
this.canvas.maskTool.resize(width, height);
|
||||||
|
|
||||||
this.canvas.canvas.width = width;
|
// Don't set canvas.width/height - the render loop will handle display size
|
||||||
this.canvas.canvas.height = height;
|
// this.canvas.width/height are for OUTPUT AREA dimensions, not display canvas
|
||||||
|
|
||||||
this.canvas.render();
|
this.canvas.render();
|
||||||
|
|
||||||
|
|||||||
@@ -144,6 +144,26 @@ export class CanvasLayersPanel {
|
|||||||
e.preventDefault();
|
e.preventDefault();
|
||||||
e.stopPropagation();
|
e.stopPropagation();
|
||||||
this.deleteSelectedLayers();
|
this.deleteSelectedLayers();
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Handle Ctrl+C/V for layer copy/paste when panel has focus
|
||||||
|
if (e.ctrlKey || e.metaKey) {
|
||||||
|
if (e.key.toLowerCase() === 'c') {
|
||||||
|
if (this.canvas.canvasSelection.selectedLayers.length > 0) {
|
||||||
|
e.preventDefault();
|
||||||
|
e.stopPropagation();
|
||||||
|
this.canvas.canvasLayers.copySelectedLayers();
|
||||||
|
log.info('Layers copied from panel');
|
||||||
|
}
|
||||||
|
} else if (e.key.toLowerCase() === 'v') {
|
||||||
|
e.preventDefault();
|
||||||
|
e.stopPropagation();
|
||||||
|
if (this.canvas.canvasLayers.internalClipboard.length > 0) {
|
||||||
|
this.canvas.canvasLayers.pasteLayers();
|
||||||
|
log.info('Layers pasted from panel');
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
@@ -388,6 +408,9 @@ export class CanvasLayersPanel {
|
|||||||
this.updateSelectionAppearance();
|
this.updateSelectionAppearance();
|
||||||
this.updateButtonStates();
|
this.updateButtonStates();
|
||||||
|
|
||||||
|
// Focus the canvas so keyboard shortcuts (like Ctrl+C/V) work for layer operations
|
||||||
|
this.canvas.canvas.focus();
|
||||||
|
|
||||||
log.debug(`Layer clicked: ${layer.name}, selection count: ${this.canvas.canvasSelection.selectedLayers.length}`);
|
log.debug(`Layer clicked: ${layer.name}, selection count: ${this.canvas.canvasSelection.selectedLayers.length}`);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -188,6 +188,11 @@ export class CanvasRenderer {
|
|||||||
}
|
}
|
||||||
});
|
});
|
||||||
|
|
||||||
|
// Draw grab icons for selected layers when hovering
|
||||||
|
if (this.canvas.canvasInteractions.interaction.hoveringGrabIcon) {
|
||||||
|
this.drawGrabIcons(ctx);
|
||||||
|
}
|
||||||
|
|
||||||
this.drawCanvasOutline(ctx);
|
this.drawCanvasOutline(ctx);
|
||||||
this.drawOutputAreaExtensionPreview(ctx); // Draw extension preview
|
this.drawOutputAreaExtensionPreview(ctx); // Draw extension preview
|
||||||
this.drawPendingGenerationAreas(ctx); // Draw snapshot outlines
|
this.drawPendingGenerationAreas(ctx); // Draw snapshot outlines
|
||||||
@@ -1013,6 +1018,66 @@ export class CanvasRenderer {
|
|||||||
this.updateOverlaySize();
|
this.updateOverlaySize();
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Draw grab icons in the center of selected layers
|
||||||
|
*/
|
||||||
|
drawGrabIcons(ctx: any): void {
|
||||||
|
const selectedLayers = this.canvas.canvasSelection.selectedLayers;
|
||||||
|
if (selectedLayers.length === 0) return;
|
||||||
|
|
||||||
|
const iconRadius = 20 / this.canvas.viewport.zoom;
|
||||||
|
const innerRadius = 12 / this.canvas.viewport.zoom;
|
||||||
|
|
||||||
|
selectedLayers.forEach((layer: any) => {
|
||||||
|
if (!layer.visible) return;
|
||||||
|
|
||||||
|
const centerX = layer.x + layer.width / 2;
|
||||||
|
const centerY = layer.y + layer.height / 2;
|
||||||
|
|
||||||
|
ctx.save();
|
||||||
|
|
||||||
|
// Draw outer circle (background)
|
||||||
|
ctx.beginPath();
|
||||||
|
ctx.arc(centerX, centerY, iconRadius, 0, Math.PI * 2);
|
||||||
|
ctx.fillStyle = 'rgba(0, 150, 255, 0.7)';
|
||||||
|
ctx.fill();
|
||||||
|
ctx.strokeStyle = 'rgba(255, 255, 255, 0.9)';
|
||||||
|
ctx.lineWidth = 2 / this.canvas.viewport.zoom;
|
||||||
|
ctx.stroke();
|
||||||
|
|
||||||
|
// Draw hand/grab icon (simplified)
|
||||||
|
ctx.fillStyle = 'rgba(255, 255, 255, 0.95)';
|
||||||
|
ctx.strokeStyle = 'rgba(255, 255, 255, 0.95)';
|
||||||
|
ctx.lineWidth = 1.5 / this.canvas.viewport.zoom;
|
||||||
|
|
||||||
|
// Draw four dots representing grab points
|
||||||
|
const dotRadius = 2 / this.canvas.viewport.zoom;
|
||||||
|
const dotDistance = 6 / this.canvas.viewport.zoom;
|
||||||
|
|
||||||
|
// Top-left
|
||||||
|
ctx.beginPath();
|
||||||
|
ctx.arc(centerX - dotDistance, centerY - dotDistance, dotRadius, 0, Math.PI * 2);
|
||||||
|
ctx.fill();
|
||||||
|
|
||||||
|
// Top-right
|
||||||
|
ctx.beginPath();
|
||||||
|
ctx.arc(centerX + dotDistance, centerY - dotDistance, dotRadius, 0, Math.PI * 2);
|
||||||
|
ctx.fill();
|
||||||
|
|
||||||
|
// Bottom-left
|
||||||
|
ctx.beginPath();
|
||||||
|
ctx.arc(centerX - dotDistance, centerY + dotDistance, dotRadius, 0, Math.PI * 2);
|
||||||
|
ctx.fill();
|
||||||
|
|
||||||
|
// Bottom-right
|
||||||
|
ctx.beginPath();
|
||||||
|
ctx.arc(centerX + dotDistance, centerY + dotDistance, dotRadius, 0, Math.PI * 2);
|
||||||
|
ctx.fill();
|
||||||
|
|
||||||
|
ctx.restore();
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Draw transform handles for output area when in transform mode
|
* Draw transform handles for output area when in transform mode
|
||||||
*/
|
*/
|
||||||
|
|||||||
@@ -118,11 +118,11 @@ export class CanvasState {
|
|||||||
log.debug(`Output Area resized to ${this.canvas.width}x${this.canvas.height} and viewport set.`);
|
log.debug(`Output Area resized to ${this.canvas.width}x${this.canvas.height} and viewport set.`);
|
||||||
const loadedLayers = await this._loadLayers(savedState.layers);
|
const loadedLayers = await this._loadLayers(savedState.layers);
|
||||||
this.canvas.layers = loadedLayers.filter((l): l is Layer => l !== null);
|
this.canvas.layers = loadedLayers.filter((l): l is Layer => l !== null);
|
||||||
log.info(`Loaded ${this.canvas.layers.length} layers.`);
|
log.info(`Loaded ${this.canvas.layers.length} layers from ${savedState.layers.length} saved layers.`);
|
||||||
|
|
||||||
if (this.canvas.layers.length === 0) {
|
if (this.canvas.layers.length === 0 && savedState.layers.length > 0) {
|
||||||
log.warn("No valid layers loaded, state may be corrupted.");
|
log.warn(`Failed to load any layers. Saved state had ${savedState.layers.length} layers but all failed to load. This may indicate corrupted IndexedDB data.`);
|
||||||
return false;
|
// Don't return false - allow empty canvas to be valid
|
||||||
}
|
}
|
||||||
|
|
||||||
this.canvas.updateSelectionAfterHistory();
|
this.canvas.updateSelectionAfterHistory();
|
||||||
|
|||||||
@@ -418,13 +418,46 @@ async function createCanvasWidget(node: ComfyNode, widget: any, app: ComfyApp):
|
|||||||
const button = (e.target as HTMLElement).closest('.matting-button') as HTMLButtonElement;
|
const button = (e.target as HTMLElement).closest('.matting-button') as HTMLButtonElement;
|
||||||
if (button.classList.contains('loading')) return;
|
if (button.classList.contains('loading')) return;
|
||||||
|
|
||||||
const spinner = $el("div.matting-spinner") as HTMLDivElement;
|
|
||||||
button.appendChild(spinner);
|
|
||||||
button.classList.add('loading');
|
|
||||||
|
|
||||||
showInfoNotification("Starting background removal process...", 2000);
|
|
||||||
|
|
||||||
try {
|
try {
|
||||||
|
// First check if model is available
|
||||||
|
const modelCheckResponse = await fetch("/matting/check-model");
|
||||||
|
const modelStatus = await modelCheckResponse.json();
|
||||||
|
|
||||||
|
if (!modelStatus.available) {
|
||||||
|
switch (modelStatus.reason) {
|
||||||
|
case 'missing_dependency':
|
||||||
|
showErrorNotification(modelStatus.message, 8000);
|
||||||
|
return;
|
||||||
|
|
||||||
|
case 'not_downloaded':
|
||||||
|
showWarningNotification("The matting model needs to be downloaded first. This will happen automatically when you proceed (requires internet connection).", 5000);
|
||||||
|
|
||||||
|
// Ask user if they want to proceed with download
|
||||||
|
if (!confirm("The matting model needs to be downloaded (about 1GB). This is a one-time download. Do you want to proceed?")) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
showInfoNotification("Downloading matting model... This may take a few minutes.", 10000);
|
||||||
|
break;
|
||||||
|
|
||||||
|
case 'corrupted':
|
||||||
|
showErrorNotification(modelStatus.message, 8000);
|
||||||
|
return;
|
||||||
|
|
||||||
|
case 'error':
|
||||||
|
showErrorNotification(`Error checking model: ${modelStatus.message}`, 5000);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Proceed with matting
|
||||||
|
const spinner = $el("div.matting-spinner") as HTMLDivElement;
|
||||||
|
button.appendChild(spinner);
|
||||||
|
button.classList.add('loading');
|
||||||
|
|
||||||
|
if (modelStatus.available) {
|
||||||
|
showInfoNotification("Starting background removal process...", 2000);
|
||||||
|
}
|
||||||
|
|
||||||
if (canvas.canvasSelection.selectedLayers.length !== 1) {
|
if (canvas.canvasSelection.selectedLayers.length !== 1) {
|
||||||
throw new Error("Please select exactly one image layer for matting.");
|
throw new Error("Please select exactly one image layer for matting.");
|
||||||
}
|
}
|
||||||
@@ -443,7 +476,18 @@ async function createCanvasWidget(node: ComfyNode, widget: any, app: ComfyApp):
|
|||||||
if (!response.ok) {
|
if (!response.ok) {
|
||||||
let errorMsg = `Server error: ${response.status} - ${response.statusText}`;
|
let errorMsg = `Server error: ${response.status} - ${response.statusText}`;
|
||||||
if (result && result.error) {
|
if (result && result.error) {
|
||||||
errorMsg = `Error: ${result.error}. Details: ${result.details || 'Check console'}`;
|
// Handle specific error types
|
||||||
|
if (result.error === "Network Connection Error") {
|
||||||
|
showErrorNotification("Failed to download the matting model. Please check your internet connection and try again.", 8000);
|
||||||
|
return;
|
||||||
|
} else if (result.error === "Matting Model Error") {
|
||||||
|
showErrorNotification(result.details || "Model loading error. Please check the console for details.", 8000);
|
||||||
|
return;
|
||||||
|
} else if (result.error === "Dependency Not Found") {
|
||||||
|
showErrorNotification(result.details || "Missing required dependencies.", 8000);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
errorMsg = `${result.error}: ${result.details || 'Check console'}`;
|
||||||
}
|
}
|
||||||
throw new Error(errorMsg);
|
throw new Error(errorMsg);
|
||||||
}
|
}
|
||||||
@@ -468,10 +512,15 @@ async function createCanvasWidget(node: ComfyNode, widget: any, app: ComfyApp):
|
|||||||
} catch (error: any) {
|
} catch (error: any) {
|
||||||
log.error("Matting error:", error);
|
log.error("Matting error:", error);
|
||||||
const errorMessage = error.message || "An unknown error occurred.";
|
const errorMessage = error.message || "An unknown error occurred.";
|
||||||
showErrorNotification(`Matting Failed: ${errorMessage}`);
|
if (!errorMessage.includes("Network Connection Error") &&
|
||||||
|
!errorMessage.includes("Matting Model Error") &&
|
||||||
|
!errorMessage.includes("Dependency Not Found")) {
|
||||||
|
showErrorNotification(`Matting Failed: ${errorMessage}`);
|
||||||
|
}
|
||||||
} finally {
|
} finally {
|
||||||
button.classList.remove('loading');
|
button.classList.remove('loading');
|
||||||
if (button.contains(spinner)) {
|
const spinner = button.querySelector('.matting-spinner');
|
||||||
|
if (spinner && button.contains(spinner)) {
|
||||||
button.removeChild(spinner);
|
button.removeChild(spinner);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -951,6 +1000,13 @@ $el("label.clipboard-switch.mask-switch", {
|
|||||||
resizeObserver.observe(controlsElement);
|
resizeObserver.observe(controlsElement);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Watch the canvas container itself to detect size changes and fix canvas dimensions
|
||||||
|
const canvasContainerResizeObserver = new ResizeObserver(() => {
|
||||||
|
// Force re-read of canvas dimensions on next render
|
||||||
|
canvas.render();
|
||||||
|
});
|
||||||
|
canvasContainerResizeObserver.observe(canvasContainer);
|
||||||
|
|
||||||
canvas.canvas.addEventListener('focus', () => {
|
canvas.canvas.addEventListener('focus', () => {
|
||||||
canvasContainer.classList.add('has-focus');
|
canvasContainer.classList.add('has-focus');
|
||||||
});
|
});
|
||||||
@@ -1146,12 +1202,23 @@ app.registerExtension({
|
|||||||
|
|
||||||
const sendPromises: Promise<any>[] = [];
|
const sendPromises: Promise<any>[] = [];
|
||||||
for (const [nodeId, canvasWidget] of canvasNodeInstances.entries()) {
|
for (const [nodeId, canvasWidget] of canvasNodeInstances.entries()) {
|
||||||
if (app.graph.getNodeById(nodeId) && canvasWidget.canvas && canvasWidget.canvas.canvasIO) {
|
const node = app.graph.getNodeById(nodeId);
|
||||||
log.debug(`Sending data for canvas node ${nodeId}`);
|
|
||||||
sendPromises.push(canvasWidget.canvas.canvasIO.sendDataViaWebSocket(nodeId));
|
if (!node) {
|
||||||
} else {
|
|
||||||
log.warn(`Node ${nodeId} not found in graph, removing from instances map.`);
|
log.warn(`Node ${nodeId} not found in graph, removing from instances map.`);
|
||||||
canvasNodeInstances.delete(nodeId);
|
canvasNodeInstances.delete(nodeId);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
// Skip bypassed nodes
|
||||||
|
if (node.mode === 4) {
|
||||||
|
log.debug(`Node ${nodeId} is bypassed, skipping data send.`);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (canvasWidget.canvas && canvasWidget.canvas.canvasIO) {
|
||||||
|
log.debug(`Sending data for canvas node ${nodeId}`);
|
||||||
|
sendPromises.push(canvasWidget.canvas.canvasIO.sendDataViaWebSocket(nodeId));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -1172,6 +1239,9 @@ app.registerExtension({
|
|||||||
|
|
||||||
async beforeRegisterNodeDef(nodeType: any, nodeData: any, app: ComfyApp) {
|
async beforeRegisterNodeDef(nodeType: any, nodeData: any, app: ComfyApp) {
|
||||||
if (nodeType.comfyClass === "LayerForgeNode") {
|
if (nodeType.comfyClass === "LayerForgeNode") {
|
||||||
|
// Map to track pending copy sources across node ID changes
|
||||||
|
const pendingCopySources = new Map<number, number>();
|
||||||
|
|
||||||
const onNodeCreated = nodeType.prototype.onNodeCreated;
|
const onNodeCreated = nodeType.prototype.onNodeCreated;
|
||||||
nodeType.prototype.onNodeCreated = function (this: ComfyNode) {
|
nodeType.prototype.onNodeCreated = function (this: ComfyNode) {
|
||||||
log.debug("CanvasNode onNodeCreated: Base widget setup.");
|
log.debug("CanvasNode onNodeCreated: Base widget setup.");
|
||||||
@@ -1208,6 +1278,49 @@ app.registerExtension({
|
|||||||
// Store the canvas widget on the node
|
// Store the canvas widget on the node
|
||||||
(this as any).canvasWidget = canvasWidget;
|
(this as any).canvasWidget = canvasWidget;
|
||||||
|
|
||||||
|
// Check if this node has a pending copy source (from onConfigure)
|
||||||
|
// Check both the current ID and -1 (temporary ID during paste)
|
||||||
|
let sourceNodeId = pendingCopySources.get(this.id);
|
||||||
|
if (!sourceNodeId) {
|
||||||
|
sourceNodeId = pendingCopySources.get(-1);
|
||||||
|
if (sourceNodeId) {
|
||||||
|
// Transfer from -1 to the real ID and clear -1
|
||||||
|
pendingCopySources.delete(-1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (sourceNodeId && sourceNodeId !== this.id) {
|
||||||
|
log.info(`Node ${this.id} will copy canvas state from node ${sourceNodeId}`);
|
||||||
|
|
||||||
|
// Clear the flag
|
||||||
|
pendingCopySources.delete(this.id);
|
||||||
|
|
||||||
|
// Copy the canvas state now that the widget is initialized
|
||||||
|
setTimeout(async () => {
|
||||||
|
try {
|
||||||
|
const { getCanvasState, setCanvasState } = await import('./db.js');
|
||||||
|
let sourceState = await getCanvasState(String(sourceNodeId));
|
||||||
|
|
||||||
|
// If source node doesn't exist (cross-workflow paste), try clipboard
|
||||||
|
if (!sourceState) {
|
||||||
|
log.debug(`No canvas state found for source node ${sourceNodeId}, checking clipboard`);
|
||||||
|
sourceState = await getCanvasState('__clipboard__');
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!sourceState) {
|
||||||
|
log.debug(`No canvas state found in clipboard either`);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
await setCanvasState(String(this.id), sourceState);
|
||||||
|
await canvasWidget.canvas.loadInitialState();
|
||||||
|
log.info(`Canvas state copied successfully to node ${this.id}`);
|
||||||
|
} catch (error) {
|
||||||
|
log.error(`Error copying canvas state:`, error);
|
||||||
|
}
|
||||||
|
}, 100);
|
||||||
|
}
|
||||||
|
|
||||||
// Check if there are already connected inputs
|
// Check if there are already connected inputs
|
||||||
setTimeout(() => {
|
setTimeout(() => {
|
||||||
if (this.inputs && this.inputs.length > 0) {
|
if (this.inputs && this.inputs.length > 0) {
|
||||||
@@ -1391,6 +1504,52 @@ app.registerExtension({
|
|||||||
return onRemoved?.apply(this, arguments as any);
|
return onRemoved?.apply(this, arguments as any);
|
||||||
};
|
};
|
||||||
|
|
||||||
|
// Handle copy/paste - save canvas state when copying
|
||||||
|
const originalSerialize = nodeType.prototype.serialize;
|
||||||
|
nodeType.prototype.serialize = function (this: ComfyNode) {
|
||||||
|
const data = originalSerialize ? originalSerialize.apply(this) : {};
|
||||||
|
|
||||||
|
// Store a reference to the source node ID so we can copy layer data
|
||||||
|
data.sourceNodeId = this.id;
|
||||||
|
log.debug(`Serializing node ${this.id} for copy`);
|
||||||
|
|
||||||
|
// Store canvas state in a clipboard entry for cross-workflow paste
|
||||||
|
// This happens async but that's fine since paste happens later
|
||||||
|
(async () => {
|
||||||
|
try {
|
||||||
|
const { getCanvasState, setCanvasState } = await import('./db.js');
|
||||||
|
const sourceState = await getCanvasState(String(this.id));
|
||||||
|
if (sourceState) {
|
||||||
|
// Store in a special "clipboard" entry
|
||||||
|
await setCanvasState('__clipboard__', sourceState);
|
||||||
|
log.debug(`Stored canvas state in clipboard for node ${this.id}`);
|
||||||
|
}
|
||||||
|
} catch (error) {
|
||||||
|
log.error('Error storing canvas state to clipboard:', error);
|
||||||
|
}
|
||||||
|
})();
|
||||||
|
|
||||||
|
return data;
|
||||||
|
};
|
||||||
|
|
||||||
|
// Handle copy/paste - load canvas state from source node when pasting
|
||||||
|
const originalConfigure = nodeType.prototype.onConfigure;
|
||||||
|
nodeType.prototype.onConfigure = async function (this: ComfyNode, data: any) {
|
||||||
|
if (originalConfigure) {
|
||||||
|
originalConfigure.apply(this, [data]);
|
||||||
|
}
|
||||||
|
|
||||||
|
// Store the source node ID in the map (persists across node ID changes)
|
||||||
|
// This will be picked up later in onAdded when the canvas widget is ready
|
||||||
|
if (data.sourceNodeId && data.sourceNodeId !== this.id) {
|
||||||
|
const existingSource = pendingCopySources.get(this.id);
|
||||||
|
if (!existingSource) {
|
||||||
|
pendingCopySources.set(this.id, data.sourceNodeId);
|
||||||
|
log.debug(`Stored pending copy source: ${data.sourceNodeId} for node ${this.id}`);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
const originalGetExtraMenuOptions = nodeType.prototype.getExtraMenuOptions;
|
const originalGetExtraMenuOptions = nodeType.prototype.getExtraMenuOptions;
|
||||||
nodeType.prototype.getExtraMenuOptions = function (this: ComfyNode, _: any, options: any[]) {
|
nodeType.prototype.getExtraMenuOptions = function (this: ComfyNode, _: any, options: any[]) {
|
||||||
// FIRST: Call original to let other extensions add their options
|
// FIRST: Call original to let other extensions add their options
|
||||||
|
|||||||
@@ -1,3 +1,4 @@
|
|||||||
|
// @ts-ignore
|
||||||
import { api } from "../../scripts/api.js";
|
import { api } from "../../scripts/api.js";
|
||||||
// @ts-ignore
|
// @ts-ignore
|
||||||
import { ComfyApp } from "../../scripts/app.js";
|
import { ComfyApp } from "../../scripts/app.js";
|
||||||
|
|||||||
@@ -1,3 +1,4 @@
|
|||||||
|
// @ts-ignore
|
||||||
import { api } from "../../../scripts/api.js";
|
import { api } from "../../../scripts/api.js";
|
||||||
import { createModuleLogger } from "./LoggerUtils.js";
|
import { createModuleLogger } from "./LoggerUtils.js";
|
||||||
import { withErrorHandling, createValidationError, createNetworkError } from "../ErrorHandler.js";
|
import { withErrorHandling, createValidationError, createNetworkError } from "../ErrorHandler.js";
|
||||||
|
|||||||
Reference in New Issue
Block a user