74 lines
1.7 KiB
Python
74 lines
1.7 KiB
Python
from krita import DockWidgetFactory, DockWidgetFactoryBase, Krita
|
|
|
|
from .defaults import (
|
|
TAB_CONFIG,
|
|
TAB_IMG2IMG,
|
|
TAB_INPAINT,
|
|
TAB_SDCOMMON,
|
|
TAB_TXT2IMG,
|
|
TAB_UPSCALE,
|
|
)
|
|
from .docker import create_docker
|
|
from .extension import SDPluginExtension
|
|
from .pages import (
|
|
ConfigPage,
|
|
Img2ImgPage,
|
|
InpaintPage,
|
|
SDCommonPage,
|
|
Txt2ImgPage,
|
|
UpscalePage,
|
|
)
|
|
from .script import script
|
|
from .utils import reset_docker_layout
|
|
|
|
instance = Krita.instance()
|
|
instance.addExtension(SDPluginExtension(instance))
|
|
instance.addDockWidgetFactory(
|
|
DockWidgetFactory(
|
|
TAB_SDCOMMON,
|
|
DockWidgetFactoryBase.DockLeft,
|
|
create_docker(SDCommonPage),
|
|
)
|
|
)
|
|
instance.addDockWidgetFactory(
|
|
DockWidgetFactory(
|
|
TAB_TXT2IMG,
|
|
DockWidgetFactoryBase.DockLeft,
|
|
create_docker(Txt2ImgPage),
|
|
)
|
|
)
|
|
instance.addDockWidgetFactory(
|
|
DockWidgetFactory(
|
|
TAB_IMG2IMG,
|
|
DockWidgetFactoryBase.DockLeft,
|
|
create_docker(Img2ImgPage),
|
|
)
|
|
)
|
|
instance.addDockWidgetFactory(
|
|
DockWidgetFactory(
|
|
TAB_INPAINT,
|
|
DockWidgetFactoryBase.DockLeft,
|
|
create_docker(InpaintPage),
|
|
)
|
|
)
|
|
instance.addDockWidgetFactory(
|
|
DockWidgetFactory(
|
|
TAB_UPSCALE,
|
|
DockWidgetFactoryBase.DockLeft,
|
|
create_docker(UpscalePage),
|
|
)
|
|
)
|
|
instance.addDockWidgetFactory(
|
|
DockWidgetFactory(
|
|
TAB_CONFIG,
|
|
DockWidgetFactoryBase.DockLeft,
|
|
create_docker(ConfigPage),
|
|
)
|
|
)
|
|
|
|
|
|
# dumb workaround to ensure its only created once
|
|
if script.cfg("first_setup", bool):
|
|
instance.notifier().windowCreated.connect(reset_docker_layout)
|
|
script.cfg.set("first_setup", False)
|