diff options
author | Justin Maier <Zipp425@gmail.com> | 2022-10-17 23:11:44 +0000 |
---|---|---|
committer | AUTOMATIC1111 <16777216c@gmail.com> | 2022-10-18 11:24:01 +0000 |
commit | 7543787a0a7d8a45c44f2293b242e128da7c5a1d (patch) | |
tree | ebf1dd1eee1e109f9f9caa7e0697c0ff2d02bb5b /javascript | |
parent | d2f459c5cf9f728256775dc1c3380c7e9a7e27fb (diff) | |
download | stable-diffusion-webui-gfx803-7543787a0a7d8a45c44f2293b242e128da7c5a1d.tar.gz stable-diffusion-webui-gfx803-7543787a0a7d8a45c44f2293b242e128da7c5a1d.tar.bz2 stable-diffusion-webui-gfx803-7543787a0a7d8a45c44f2293b242e128da7c5a1d.zip |
Auto select attention block for editing
Diffstat (limited to 'javascript')
-rw-r--r-- | javascript/edit-attention.js | 35 |
1 files changed, 32 insertions, 3 deletions
diff --git a/javascript/edit-attention.js b/javascript/edit-attention.js index 67084e7a..c0d29a74 100644 --- a/javascript/edit-attention.js +++ b/javascript/edit-attention.js @@ -9,9 +9,38 @@ addEventListener('keydown', (event) => { let minus = "ArrowDown"
if (event.key != plus && event.key != minus) return;
- selectionStart = target.selectionStart;
- selectionEnd = target.selectionEnd;
- if(selectionStart == selectionEnd) return;
+ let selectionStart = target.selectionStart;
+ let selectionEnd = target.selectionEnd;
+ // If the user hasn't selected anything, let's select their current parenthesis block
+ if (selectionStart === selectionEnd) {
+ // Find opening parenthesis around current cursor
+ const before = target.value.substring(0, selectionStart);
+ let beforeParen = before.lastIndexOf("(");
+ if (beforeParen == -1) return;
+ let beforeParenClose = before.lastIndexOf(")");
+ while (beforeParenClose !== -1 && beforeParenClose > beforeParen) {
+ beforeParen = before.lastIndexOf("(", beforeParen - 1);
+ beforeParenClose = before.lastIndexOf(")", beforeParenClose - 1);
+ }
+
+ // Find closing parenthesis around current cursor
+ const after = target.value.substring(selectionStart);
+ let afterParen = after.indexOf(")");
+ if (afterParen == -1) return;
+ let afterParenOpen = after.indexOf("(");
+ while (afterParenOpen !== -1 && afterParen > afterParenOpen) {
+ afterParen = after.indexOf(")", afterParen + 1);
+ afterParenOpen = after.indexOf("(", afterParenOpen + 1);
+ }
+ if (beforeParen === -1 || afterParen === -1) return;
+
+ // Set the selection to the text between the parenthesis
+ const parenContent = target.value.substring(beforeParen + 1, selectionStart + afterParen);
+ const lastColon = parenContent.lastIndexOf(":");
+ selectionStart = beforeParen + 1;
+ selectionEnd = selectionStart + lastColon;
+ target.setSelectionRange(selectionStart, selectionEnd);
+ }
event.preventDefault();
|