Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(node-editor): fit view on init

Browse Source
This commit is contained in:
blessedcoolant 2023-07-11 18:22:50 +12:00
parent c3adb301a0
commit acea304348
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
invokeai/frontend/web/src/features/nodes/components/Flow.tsx
View File

@ -7,8 +7,10 @@ import {
OnConnectEnd, OnConnectEnd,
OnConnectStart, OnConnectStart,
OnEdgesChange, OnEdgesChange,
OnInit,
OnNodesChange, OnNodesChange,
ReactFlow, ReactFlow,
ReactFlowInstance,
} from 'reactflow'; } from 'reactflow';
import { import {
connectionEnded, connectionEnded,
@ -67,6 +69,10 @@ export const Flow = () => {
dispatch(connectionEnded()); dispatch(connectionEnded());
}, [dispatch]); }, [dispatch]);
const onInit: OnInit = useCallback((v: ReactFlowInstance) => {
if (v) v.fitView();
}, []);
return ( return (
<ReactFlow <ReactFlow
nodeTypes={nodeTypes} nodeTypes={nodeTypes}
@ -77,6 +83,7 @@ export const Flow = () => {
onConnectStart={onConnectStart} onConnectStart={onConnectStart}
onConnect={onConnect} onConnect={onConnect}
onConnectEnd={onConnectEnd} onConnectEnd={onConnectEnd}
onInit={onInit}
defaultEdgeOptions={{ defaultEdgeOptions={{
style: { strokeWidth: 2 }, style: { strokeWidth: 2 },
}} }}