OneTrainer/modules/ui/PySide6CloudTabView.py at 9c05ddd85dd6b7ddafd9f9bfdc81fd1d92bb0b51 · Nerogar/OneTrainer · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
from modules.ui.BaseCloudTabView import BaseCloudTabView
from modules.ui.CloudTabController import CloudTabController
from modules.util.ui import pyside6_components
from modules.util.ui.pyside6_abc import QtABCMeta

from PySide6.QtWidgets import QWidget


class PySide6CloudTabView(BaseCloudTabView, QWidget, metaclass=QtABCMeta):

    def __init__(self, master, controller: CloudTabController, ui_state):
        QWidget.__init__(self, master)
        BaseCloudTabView.__init__(self, pyside6_components, controller)

        self.ui_state = ui_state

        scroll, frame = pyside6_components.scrollable_frame(self)
        pyside6_components._layout(self).addWidget(scroll, 0, 0)
        lo = pyside6_components._layout(frame)
        lo.setColumnStretch(1, 1)
        lo.setColumnStretch(3, 1)
        lo.setColumnStretch(5, 1)
        self.frame = frame

        self.build_content(frame, controller, ui_state)

    def _on_set_gpu_types(self):
        self.gpu_types_menu.clear()
        self.gpu_types_menu.addItems(self.controller.get_gpu_types())

    def _make_reattach_frame(self, frame):
        reattach_frame = QWidget(frame)
        pyside6_components._layout(frame).addWidget(reattach_frame, 9, 3)
        pyside6_components._layout(reattach_frame).setColumnStretch(0, 1)
        return reattach_frame

    def _make_create_frame(self, frame):
        create_frame = QWidget(frame)
        pyside6_components._layout(frame).addWidget(create_frame, 1, 5)
        pyside6_components._layout(create_frame).setColumnStretch(1, 1)
        return create_frame