def __init__(self, parent=None):
super(_RenderGlobalsEditor, self).__init__(parent)
self.setWindowTitle(api.Session["AVALON_LABEL"] + " Render Globals")
self.setAttribute(QtCore.Qt.WA_DeleteOnClose)
pools = QtWidgets.QComboBox()
groups = QtWidgets.QComboBox()
button = QtWidgets.QPushButton("Refresh")
layout = QtWidgets.QVBoxLayout(self)
layout.addWidget(QtWidgets.QLabel("Current Pool"))
layout.addWidget(pools)
layout.addWidget(QtWidgets.QLabel("Current Group"))
layout.addWidget(groups)
layout.addWidget(QtWidgets.QLabel(""))
layout.addWidget(button)
self.pools = pools
self.groups = groups
self.render_globals = None
self.resize(300, 100)
self.setMinimumWidth(200)
self.refresh()
button.clicked.connect(self.refresh)
pools.currentIndexChanged.connect(self.on_pool_changed)
groups.currentIndexChanged.connect(self.on_group_changed)
def __init__(self, parent=None):
super(Window, self).__init__(parent=parent)
self.setWindowIcon(qtawesome.icon("fa.users", color="#F0F3F4"))
self.setWindowTitle("Project Member")
self.setWindowFlags(QtCore.Qt.Window)
widget = {
"refresh": QtWidgets.QPushButton(),
"projects": QtWidgets.QComboBox(),
"btnAdd": QtWidgets.QPushButton(),
"btnDel": QtWidgets.QPushButton(),
}
layout = QtWidgets.QHBoxLayout(self)
layout.addWidget(widget["refresh"])
layout.addWidget(widget["projects"], stretch=True)
layout.addWidget(widget["btnAdd"])
layout.addWidget(widget["btnDel"])
widget["refresh"].setIcon(qtawesome.icon("fa.refresh",
color="#E5E7E9"))
widget["btnAdd"].setIcon(
qtawesome.icon("fa.user-plus", color="#58D68D"))
widget["btnDel"].setIcon(
qtawesome.icon("fa.user-times", color="#A93226"))
widget["projects"].currentTextChanged.connect(self.on_text_changed)
widget["refresh"].clicked.connect(self.on_refresh_clicked)
widget["btnAdd"].clicked.connect(self.on_add_clicked)
widget["btnDel"].clicked.connect(self.on_del_clicked)
self.widget = widget
self.init_projects()
self.resize(480, 40)
def __init__(self, parent=None):
QtWidgets.QWidget.__init__(self, parent)
self._comp = avalon.get_current_comp()
self._comp_name = self._get_comp_name()
self.setWindowTitle("Set Render Mode")
self.setFixedSize(300, 175)
layout = QtWidgets.QVBoxLayout()
# region comp info
comp_info_layout = QtWidgets.QHBoxLayout()
update_btn = QtWidgets.QPushButton(
qtawesome.icon("fa.refresh", color="white"), "")
update_btn.setFixedWidth(25)
update_btn.setFixedHeight(25)
comp_information = QtWidgets.QLineEdit()
comp_information.setEnabled(False)
comp_info_layout.addWidget(comp_information)
comp_info_layout.addWidget(update_btn)
# endregion comp info
# region modes
mode_options = QtWidgets.QComboBox()
mode_options.addItems(_help.keys())
mode_information = QtWidgets.QTextEdit()
mode_information.setReadOnly(True)
# endregion modes
accept_btn = QtWidgets.QPushButton("Accept")
layout.addLayout(comp_info_layout)
layout.addWidget(mode_options)
layout.addWidget(mode_information)
layout.addWidget(accept_btn)
self.setLayout(layout)
self.comp_information = comp_information
self.update_btn = update_btn
self.mode_options = mode_options
self.mode_information = mode_information
self.accept_btn = accept_btn
self.connections()
self.update()
# Force updated render mode help text
self._update_rendermode_info()
def __init__(self, parent=None):
super(HostSelectorWidget, self).__init__(parent=parent)
panel = {
"selection": QtWidgets.QWidget(),
"container": QtWidgets.QWidget(),
}
model = {
"containerModel": models.HostContainerListModel(),
}
widget = {
"selectionChk": QtWidgets.QCheckBox("Use Selection"),
"selectionBtn": QtWidgets.QPushButton("Compare Selection"),
"containerChk": QtWidgets.QCheckBox("Use Container"),
"containerBox": QtWidgets.QComboBox(),
}
widget["containerBox"].setModel(model["containerModel"])
layout = QtWidgets.QVBoxLayout(panel["selection"])
layout.addWidget(widget["selectionChk"])
layout.addWidget(widget["selectionBtn"])
layout = QtWidgets.QVBoxLayout(panel["container"])
layout.addWidget(widget["containerChk"])
layout.addWidget(widget["containerBox"])
layout = QtWidgets.QVBoxLayout(self)
layout.addWidget(panel["selection"])
layout.addWidget(panel["container"])
# Connect
widget["selectionChk"].stateChanged.connect(self.on_use_selection)
widget["containerChk"].stateChanged.connect(self.on_use_container)
widget["selectionBtn"].pressed.connect(self.on_selection_pressed)
widget["containerBox"].currentIndexChanged.connect(
self.on_container_picked)
# Init
self.widget = widget
self.model = model
widget["containerChk"].setCheckState(QtCore.Qt.Checked)
# Confirm host registered
if not has_host():
# Disable all widgets
for widget in self.widget.values():
widget.setEnabled(False)
def __init__(self, parent=None):
super(FrameRangeSetter, self).__init__(parent)
self.setWindowTitle("Set Frame Range")
self.setFocusPolicy(QtCore.Qt.StrongFocus)
self.setAttribute(QtCore.Qt.WA_DeleteOnClose)
self.setWindowFlags(QtCore.Qt.WindowStaysOnTopHint)
asset_menu = QtWidgets.QComboBox()
asset_grp = self.make_group(asset_menu, "Asset Name")
start_box = QtWidgets.QSpinBox()
start_grp = self.make_group(start_box, "Start Frame")
end_box = QtWidgets.QSpinBox()
end_grp = self.make_group(end_box, "End Frame")
handle_box = QtWidgets.QSpinBox()
handle_grp = self.make_group(handle_box, "Handles")
save_btn = QtWidgets.QPushButton("Save")
body = QtWidgets.QVBoxLayout(self)
body.addLayout(asset_grp)
body.addLayout(start_grp)
body.addLayout(end_grp)
body.addLayout(handle_grp)
body.addWidget(save_btn)
asset_menu.currentIndexChanged.connect(self.on_asset_changed)
start_box.valueChanged.connect(self.on_value_changed)
end_box.valueChanged.connect(self.on_value_changed)
handle_box.valueChanged.connect(self.on_value_changed)
save_btn.clicked.connect(self.save_range)
self.assets = asset_menu
self.start = start_box
self.end = end_box
self.handles = handle_box
project = io.find_one({"type": "project"})
self.handles_min = project["data"]["handles"]
self.end.setMaximum(self.MAX)
self.find_assets()
def __init__(self, parent=None):
QtWidgets.QWidget.__init__(self, parent=parent)
self.setWindowFlag(QtCore.Qt.Dialog)
if parent:
self.setStyleSheet(parent.styleSheet())
layout = QtWidgets.QVBoxLayout()
task_icon_layout = QtWidgets.QHBoxLayout()
task_name = QtWidgets.QLineEdit()
icon_name = QtWidgets.QComboBox()
icon_preview_button = QtWidgets.QPushButton()
icon_preview_button.setEnabled(False)
icon_preview_button.setStyleSheet(style.preview_button)
task_icon_layout.addWidget(task_name)
task_icon_layout.addWidget(icon_name)
task_icon_layout.addWidget(icon_preview_button)
buttons_layout = QtWidgets.QHBoxLayout()
accept_button = QtWidgets.QPushButton("Accept")
cancel_button = QtWidgets.QPushButton("Cancel")
buttons_layout.addWidget(accept_button)
buttons_layout.addWidget(cancel_button)
layout.addLayout(task_icon_layout)
layout.addLayout(buttons_layout)
self.setLayout(layout)
self.task_name = task_name
self.icon_name = icon_name
self.icon_preview = icon_preview_button
self.fontlib = {}
self.connect_signals()
self.refresh()
def __init__(self, parent=None):
super(Project_name_getUI, self).__init__(parent)
self.platform = platform.system()
self.new_index = 0
# get projects from ftrack
self.session = ftrack_api.Session()
self.projects_from_ft = self.session.query(
'Project where status is active')
self.disks_from_ft = self.session.query('Disk')
self.schemas_from_ft = self.session.query('ProjectSchema')
self.projects = self._get_projects_ftrack()
# define window geometry
self.setWindowTitle('Set project attributes')
self.setWindowFlags(QtCore.Qt.WindowStaysOnTopHint)
self.resize(550, 340)
self.setStyleSheet(style.load_stylesheet())
# define disk combobox widget
self.disks = self._get_all_disks()
self.disk_combobox_label = QtWidgets.QLabel('Destination storage:')
self.disk_combobox = QtWidgets.QComboBox()
# define schema combobox widget
self.schemas = self._get_all_schemas()
self.schema_combobox_label = QtWidgets.QLabel('Project schema:')
self.schema_combobox = QtWidgets.QComboBox()
# define fps widget
self.fps_label = QtWidgets.QLabel('Fps:')
self.fps_label.setAlignment(QtCore.Qt.AlignRight
| QtCore.Qt.AlignVCenter)
self.fps = QtWidgets.QLineEdit()
# define project dir widget
self.project_dir_label = QtWidgets.QLabel('Project dir:')
self.project_dir_label.setAlignment(QtCore.Qt.AlignRight
| QtCore.Qt.AlignVCenter)
self.project_dir = QtWidgets.QLineEdit()
self.project_path_label = QtWidgets.QLabel(
'Project_path (if not then created):')
self.project_path_label.setAlignment(QtCore.Qt.AlignHCenter
| QtCore.Qt.AlignVCenter)
project_path_font = QtGui.QFont("Helvetica [Cronyx]", 12,
QtGui.QFont.Bold)
self.project_path = QtWidgets.QLabel()
self.project_path.setObjectName('nom_plan_label')
self.project_path.setStyleSheet(
'QtWidgets.QLabel#nom_plan_label {color: red}')
self.project_path.setAlignment(QtCore.Qt.AlignHCenter
| QtCore.Qt.AlignVCenter)
self.project_path.setFont(project_path_font)
# define handles widget
self.handles_label = QtWidgets.QLabel('Handles:')
self.handles_label.setAlignment(QtCore.Qt.AlignRight
| QtCore.Qt.AlignVCenter)
self.handles = QtWidgets.QLineEdit()
# define resolution widget
self.resolution_w_label = QtWidgets.QLabel('W:')
self.resolution_w = QtWidgets.QLineEdit()
self.resolution_h_label = QtWidgets.QLabel('H:')
self.resolution_h = QtWidgets.QLineEdit()
devider = QtWidgets.QFrame()
# devider.Shape(QFrame.HLine)
devider.setFrameShape(QtWidgets.QFrame.HLine)
devider.setFrameShadow(QtWidgets.QFrame.Sunken)
self.generate_lines()
# define push buttons
self.set_pushbutton = QtWidgets.QPushButton('Set project')
self.cancel_pushbutton = QtWidgets.QPushButton('Cancel')
# definition of layouts
############################################
action_layout = QtWidgets.QHBoxLayout()
action_layout.addWidget(self.set_pushbutton)
action_layout.addWidget(self.cancel_pushbutton)
# schema property
schema_layout = QtWidgets.QGridLayout()
schema_layout.addWidget(self.schema_combobox, 0, 1)
schema_layout.addWidget(self.schema_combobox_label, 0, 0)
# storage property
storage_layout = QtWidgets.QGridLayout()
storage_layout.addWidget(self.disk_combobox, 0, 1)
storage_layout.addWidget(self.disk_combobox_label, 0, 0)
# fps property
fps_layout = QtWidgets.QGridLayout()
fps_layout.addWidget(self.fps, 1, 1)
fps_layout.addWidget(self.fps_label, 1, 0)
# project dir property
project_dir_layout = QtWidgets.QGridLayout()
project_dir_layout.addWidget(self.project_dir, 1, 1)
project_dir_layout.addWidget(self.project_dir_label, 1, 0)
# project path property
project_path_layout = QtWidgets.QGridLayout()
spacer_1_item = QtWidgets.QSpacerItem(10, 10)
project_path_layout.addItem(spacer_1_item, 0, 1)
project_path_layout.addWidget(self.project_path_label, 1, 1)
project_path_layout.addWidget(self.project_path, 2, 1)
spacer_2_item = QtWidgets.QSpacerItem(20, 20)
project_path_layout.addItem(spacer_2_item, 3, 1)
# handles property
handles_layout = QtWidgets.QGridLayout()
handles_layout.addWidget(self.handles, 1, 1)
handles_layout.addWidget(self.handles_label, 1, 0)
# resolution property
resolution_layout = QtWidgets.QGridLayout()
resolution_layout.addWidget(self.resolution_w_label, 1, 1)
resolution_layout.addWidget(self.resolution_w, 2, 1)
resolution_layout.addWidget(self.resolution_h_label, 1, 2)
resolution_layout.addWidget(self.resolution_h, 2, 2)
# form project property layout
p_layout = QtWidgets.QGridLayout()
p_layout.addLayout(storage_layout, 1, 0)
p_layout.addLayout(schema_layout, 2, 0)
p_layout.addLayout(project_dir_layout, 3, 0)
p_layout.addLayout(fps_layout, 4, 0)
p_layout.addLayout(handles_layout, 5, 0)
p_layout.addLayout(resolution_layout, 6, 0)
p_layout.addWidget(devider, 7, 0)
spacer_item = QtWidgets.QSpacerItem(150, 40,
QtWidgets.QSizePolicy.Minimum,
QtWidgets.QSizePolicy.Expanding)
p_layout.addItem(spacer_item, 8, 0)
# form with list to one layout with project property
list_layout = QtWidgets.QGridLayout()
list_layout.addLayout(p_layout, 1, 0)
list_layout.addWidget(self.listWidget, 1, 1)
root_layout = QtWidgets.QVBoxLayout()
root_layout.addLayout(project_path_layout)
root_layout.addWidget(devider)
root_layout.addLayout(list_layout)
root_layout.addLayout(action_layout)
self.setLayout(root_layout)
def setup_ui(self):
"""Build the initial UI"""
MULTI_SELECT = QtWidgets.QAbstractItemView.ExtendedSelection
layout = QtWidgets.QVBoxLayout(self)
publish = QtWidgets.QCheckBox("Suspend Publish Job")
extend_frames = QtWidgets.QCheckBox("Extend Frames")
override_frames = QtWidgets.QCheckBox("Override Frames")
override_frames.setEnabled(False)
defaultlayer = QtWidgets.QCheckBox("Include Default Render Layer")
maya_batch_checkbox = QtWidgets.QCheckBox("Use MayaBatch")
maya_batch_checkbox.setChecked(True)
maya_batch_checkbox.setToolTip(
'This uses the MayaBatch plugin. It keeps the scenes loaded '
'between frames, which reduces the overhead of rendering the '
'job.\nIt is however easier to debug a render when turning this '
'off and render using MayaCmd')
# region Priority
priority_grp = QtWidgets.QGroupBox("Priority")
priority_hlayout = QtWidgets.QHBoxLayout()
priority_value = QtWidgets.QSpinBox()
priority_value.setButtonSymbols(QtWidgets.QAbstractSpinBox.NoButtons)
priority_value.setEnabled(False)
priority_slider = QtWidgets.QSlider(QtCore.Qt.Horizontal)
priority_slider.setMinimum(0)
priority_slider.setMaximum(99)
priority_hlayout.addWidget(priority_value)
priority_hlayout.addWidget(priority_slider)
priority_grp.setLayout(priority_hlayout)
# endregion Priority
# region ChunkSize
chunksize_grp = QtWidgets.QGroupBox("Frames Per Task")
chunksize_hlayout = QtWidgets.QHBoxLayout()
chunksize_value = QtWidgets.QSpinBox()
chunksize_value.setButtonSymbols(QtWidgets.QAbstractSpinBox.NoButtons)
chunksize_value.setEnabled(False)
chunksize_value.setValue(1) # default value
chunksize_value.setMinimum(1)
chunksize_value.setMaximum(99)
chunksize_slider = QtWidgets.QSlider(QtCore.Qt.Horizontal)
chunksize_slider.setMinimum(1)
chunksize_slider.setMaximum(99)
chunksize_hlayout.addWidget(chunksize_value)
chunksize_hlayout.addWidget(chunksize_slider)
chunksize_grp.setLayout(chunksize_hlayout)
# endregion
# region pools
pools_grp = QtWidgets.QGroupBox("Pools")
pools_vlayout = QtWidgets.QVBoxLayout()
primary_hlayout = QtWidgets.QHBoxLayout()
primary_label = QtWidgets.QLabel("Primary")
primary_pool = QtWidgets.QComboBox()
primary_hlayout.addWidget(primary_label)
primary_hlayout.addWidget(primary_pool)
secondary_hlayout = QtWidgets.QHBoxLayout()
secondary_label = QtWidgets.QLabel("Secondary")
secondary_pool = QtWidgets.QComboBox()
secondary_hlayout.addWidget(secondary_label)
secondary_hlayout.addWidget(secondary_pool)
pools_vlayout.addLayout(primary_hlayout)
pools_vlayout.addLayout(secondary_hlayout)
pools_grp.setLayout(pools_vlayout)
# endregion pools
# Group box for type of machine list
list_type_grp = QtWidgets.QGroupBox("Machine List Type")
list_type_hlayout = QtWidgets.QHBoxLayout()
black_list = QtWidgets.QRadioButton("Blacklist")
black_list.setChecked(True)
black_list.setToolTip("List machines which the job WILL NOT use")
white_list = QtWidgets.QRadioButton("Whitelist")
white_list.setToolTip("List machines which the job WILL use")
list_type_hlayout.addWidget(black_list)
list_type_hlayout.addWidget(white_list)
list_type_grp.setLayout(list_type_hlayout)
# region Machine selection
machines_hlayout = QtWidgets.QHBoxLayout()
machines_hlayout.setSpacing(2)
machine_list = QtWidgets.QListWidget()
listed_machines = QtWidgets.QListWidget()
# Buttons
button_vlayout = QtWidgets.QVBoxLayout()
button_vlayout.setAlignment(QtCore.Qt.AlignCenter)
button_vlayout.setSpacing(4)
add_machine_btn = QtWidgets.QPushButton(">")
add_machine_btn.setFixedWidth(25)
remove_machine_btn = QtWidgets.QPushButton("<")
remove_machine_btn.setFixedWidth(25)
button_vlayout.addWidget(add_machine_btn)
button_vlayout.addWidget(remove_machine_btn)
machines_hlayout.addWidget(machine_list)
machines_hlayout.addLayout(button_vlayout)
machines_hlayout.addWidget(listed_machines)
# Machine selection widget settings
machine_list.setSelectionMode(MULTI_SELECT)
listed_machines.setSelectionMode(MULTI_SELECT)
# endregion
accept_btn = QtWidgets.QPushButton("Use Settings")
layout.addWidget(defaultlayer)
layout.addWidget(publish)
layout.addWidget(extend_frames)
layout.addWidget(override_frames)
layout.addWidget(maya_batch_checkbox)
layout.addWidget(priority_grp)
layout.addWidget(chunksize_grp)
layout.addWidget(pools_grp)
layout.addWidget(list_type_grp)
layout.addLayout(machines_hlayout)
layout.addWidget(accept_btn)
# Enable access for all methods
self.publish = publish
self.extend_frames = extend_frames
self.override_frames = override_frames
self.defaultlayer = defaultlayer
self.priority_value = priority_value
self.priority_slider = priority_slider
self.chunksize_value = chunksize_value
self.chunksize_slider = chunksize_slider
self.primary_pool = primary_pool
self.secondary_pool = secondary_pool
self.black_list = black_list
self.white_list = white_list
self.machine_list = machine_list
self.listed_machines = listed_machines
self.add_machine_btn = add_machine_btn
self.remove_machine_btn = remove_machine_btn
self.use_maya_batch = maya_batch_checkbox
self.accept = accept_btn
self.setLayout(layout)
def __init__(self, parent=None, context=None):
super(Window, self).__init__(parent)
self.context = context
project_name = io.active_project()
self.setWindowTitle("Asset creator ({0})".format(project_name))
self.setFocusPolicy(QtCore.Qt.StrongFocus)
self.setAttribute(QtCore.Qt.WA_DeleteOnClose)
# Validators
self.valid_parent = False
self.session = None
# assets widget
assets_widget = QtWidgets.QWidget()
assets_widget.setContentsMargins(0, 0, 0, 0)
assets_layout = QtWidgets.QVBoxLayout(assets_widget)
assets = widget.AssetWidget()
assets.view.setSelectionMode(assets.view.ExtendedSelection)
assets_layout.addWidget(assets)
# Outlink
label_outlink = QtWidgets.QLabel("Outlink:")
input_outlink = QtWidgets.QLineEdit()
input_outlink.setReadOnly(True)
input_outlink.setStyleSheet("background-color: #333333;")
checkbox_outlink = QtWidgets.QCheckBox("Use outlink")
# Parent
label_parent = QtWidgets.QLabel("*Parent:")
input_parent = QtWidgets.QLineEdit()
input_parent.setReadOnly(True)
input_parent.setStyleSheet("background-color: #333333;")
# Name
label_name = QtWidgets.QLabel("*Name:")
input_name = QtWidgets.QLineEdit()
input_name.setPlaceholderText("<asset name>")
# Asset Build
label_assetbuild = QtWidgets.QLabel("Asset Build:")
combo_assetbuilt = QtWidgets.QComboBox()
# Task template
label_task_template = QtWidgets.QLabel("Task template:")
combo_task_template = QtWidgets.QComboBox()
# Info widget
info_widget = QtWidgets.QWidget()
info_widget.setContentsMargins(10, 10, 10, 10)
info_layout = QtWidgets.QVBoxLayout(info_widget)
# Inputs widget
inputs_widget = QtWidgets.QWidget()
inputs_widget.setContentsMargins(0, 0, 0, 0)
inputs_layout = QtWidgets.QFormLayout(inputs_widget)
inputs_layout.addRow(label_outlink, input_outlink)
inputs_layout.addRow(None, checkbox_outlink)
inputs_layout.addRow(label_parent, input_parent)
inputs_layout.addRow(label_name, input_name)
inputs_layout.addRow(label_assetbuild, combo_assetbuilt)
inputs_layout.addRow(label_task_template, combo_task_template)
# Add button
btns_widget = QtWidgets.QWidget()
btns_widget.setContentsMargins(0, 0, 0, 0)
btn_layout = QtWidgets.QHBoxLayout(btns_widget)
btn_create_asset = QtWidgets.QPushButton("Create asset")
btn_create_asset.setToolTip(
"Creates all neccessary components for asset"
)
checkbox_app = None
if self.context is not None:
checkbox_app = QtWidgets.QCheckBox("Open {}".format(
self.context.capitalize())
)
btn_layout.addWidget(checkbox_app)
btn_layout.addWidget(btn_create_asset)
task_view = QtWidgets.QTreeView()
task_view.setIndentation(0)
task_model = model.TasksModel()
task_view.setModel(task_model)
info_layout.addWidget(inputs_widget)
info_layout.addWidget(task_view)
info_layout.addWidget(btns_widget)
# Body
body = QtWidgets.QSplitter()
body.setContentsMargins(0, 0, 0, 0)
body.setSizePolicy(QtWidgets.QSizePolicy.Expanding,
QtWidgets.QSizePolicy.Expanding)
body.setOrientation(QtCore.Qt.Horizontal)
body.addWidget(assets_widget)
body.addWidget(info_widget)
body.setStretchFactor(0, 100)
body.setStretchFactor(1, 150)
# statusbar
message = QtWidgets.QLabel()
message.setFixedHeight(20)
statusbar = QtWidgets.QWidget()
layout = QtWidgets.QHBoxLayout(statusbar)
layout.setContentsMargins(0, 0, 0, 0)
layout.addWidget(message)
layout = QtWidgets.QVBoxLayout(self)
layout.addWidget(body)
layout.addWidget(statusbar)
self.data = {
"label": {
"message": message,
},
"view": {
"tasks": task_view
},
"model": {
"assets": assets,
"tasks": task_model
},
"inputs": {
"outlink": input_outlink,
"outlink_cb": checkbox_outlink,
"parent": input_parent,
"name": input_name,
"assetbuild": combo_assetbuilt,
"tasktemplate": combo_task_template,
"open_app": checkbox_app
},
"buttons": {
"create_asset": btn_create_asset
}
}
# signals
btn_create_asset.clicked.connect(self.create_asset)
assets.selection_changed.connect(self.on_asset_changed)
input_name.textChanged.connect(self.on_asset_name_change)
checkbox_outlink.toggled.connect(self.on_outlink_checkbox_change)
combo_task_template.currentTextChanged.connect(
self.on_task_template_changed
)
if self.context is not None:
checkbox_app.toggled.connect(self.on_app_checkbox_change)
# on start
self.on_start()
self.resize(600, 500)
self.echo("Connected to project: {0}".format(project_name))
def setup_ui(self):
"""Build the initial UI"""
layout = QtWidgets.QVBoxLayout(self)
# region executable
executable_grp = QtWidgets.QGroupBox("Executable")
executable_vlayout = QtWidgets.QVBoxLayout()
executable_hlayout = QtWidgets.QHBoxLayout()
executable_filename = QtWidgets.QLineEdit()
executable_button = QtWidgets.QPushButton()
executable_button.setIcon(self.style().standardIcon(
getattr(QtWidgets.QStyle, 'SP_DialogOpenButton')))
executable_hlayout.addWidget(executable_filename)
executable_hlayout.addWidget(executable_button)
executable_vlayout.addLayout(executable_hlayout)
executable_grp.setLayout(executable_vlayout)
# endregion executable
# region options
options_grp = QtWidgets.QGroupBox("Options")
options_vlayout = QtWidgets.QVBoxLayout()
postfix_hlayout = QtWidgets.QHBoxLayout()
postfix_label = QtWidgets.QLabel("Postfix")
postfix_value = QtWidgets.QLineEdit()
postfix_hlayout.addWidget(postfix_label)
postfix_hlayout.addWidget(postfix_value)
compression_hlayout = QtWidgets.QHBoxLayout()
compression_label = QtWidgets.QLabel("Compression")
compression_value = QtWidgets.QComboBox()
compression_hlayout.addWidget(compression_label)
compression_hlayout.addWidget(compression_value)
linear_hlayout = QtWidgets.QHBoxLayout()
linear_label = QtWidgets.QLabel("Linear Conversion")
linear_value = QtWidgets.QComboBox()
linear_hlayout.addWidget(linear_label)
linear_hlayout.addWidget(linear_value)
tilesize_hlayout = QtWidgets.QHBoxLayout()
tilesize_label = QtWidgets.QLabel("Tile Size")
tilesize_value = QtWidgets.QSpinBox()
tilesize_value.setValue(64)
tilesize_hlayout.addWidget(tilesize_label)
tilesize_hlayout.addWidget(tilesize_value)
overwrite_hlayout = QtWidgets.QHBoxLayout()
overwrite_label = QtWidgets.QLabel("Overwrite")
overwrite_value = QtWidgets.QCheckBox()
overwrite_value.setChecked(False)
overwrite_hlayout.addWidget(overwrite_label)
overwrite_hlayout.addWidget(overwrite_value)
preserve_hlayout = QtWidgets.QHBoxLayout()
preserve_label = QtWidgets.QLabel("Preserve Color Space ")
preserve_value = QtWidgets.QCheckBox()
preserve_value.setChecked(True)
preserve_hlayout.addWidget(preserve_label)
preserve_hlayout.addWidget(preserve_value)
preserve_filter_hlayout = QtWidgets.QHBoxLayout()
preserve_filter_label = QtWidgets.QLabel("Preserver Filter")
preserve_filter_value = QtWidgets.QLineEdit()
preserve_filter_hlayout.addWidget(preserve_filter_label)
preserve_filter_hlayout.addWidget(preserve_filter_value)
options_vlayout.addLayout(postfix_hlayout)
options_vlayout.addLayout(compression_hlayout)
options_vlayout.addLayout(linear_hlayout)
options_vlayout.addLayout(tilesize_hlayout)
options_vlayout.addLayout(overwrite_hlayout)
options_vlayout.addLayout(preserve_hlayout)
options_vlayout.addLayout(preserve_filter_hlayout)
options_grp.setLayout(options_vlayout)
# endregion options
# Group box for type of machine list
list_type_grp = QtWidgets.QGroupBox("File Texture Nodes")
list_type_hlayout = QtWidgets.QVBoxLayout()
refresh_button = QtWidgets.QPushButton("Refresh")
refresh_button.setToolTip("Refresh texture lists")
refresh_button.setIcon(self.style().standardIcon(
getattr(QtWidgets.QStyle, 'SP_BrowserReload')))
# region file node list
file_node_hlayout = QtWidgets.QVBoxLayout()
file_node_hlayout.setSpacing(2)
file_node_list = QtWidgets.QListView()
file_node_list.setAlternatingRowColors(True)
file_node_list.setSelectionBehavior(
QtWidgets.QAbstractItemView.SelectRows)
file_node_list.setSelectionMode(
QtWidgets.QAbstractItemView.MultiSelection)
# endregion
# conversion buttons
button_vlayout = QtWidgets.QHBoxLayout()
button_vlayout.setAlignment(QtCore.Qt.AlignCenter)
button_vlayout.setSpacing(4)
convert_button = QtWidgets.QPushButton("Convert")
exr_button = QtWidgets.QPushButton("Show EXR")
source_button = QtWidgets.QPushButton("Show Source")
set_source_button = QtWidgets.QPushButton("Set new source")
set_source_button.setDisabled(True)
button_vlayout.addWidget(convert_button)
button_vlayout.addWidget(exr_button)
button_vlayout.addWidget(source_button)
button_vlayout.addWidget(set_source_button)
file_node_hlayout.addWidget(file_node_list)
file_node_hlayout.addLayout(button_vlayout)
list_type_hlayout.addWidget(refresh_button)
list_type_hlayout.addLayout(file_node_hlayout)
list_type_grp.setLayout(list_type_hlayout)
layout.addWidget(executable_grp)
layout.addWidget(options_grp)
layout.addWidget(list_type_grp)
layout.addLayout(file_node_hlayout)
# Enable access for all methods
self.file_node_list = file_node_list
self.postfix_value = postfix_value
self.compression_value = compression_value
self.linear_value = linear_value
self.tilesize_value = tilesize_value
self.executable_filename = executable_filename
self.exr_button = exr_button
self.source_button = source_button
self.convert_button = convert_button
self.refresh_button = refresh_button
self.overwritevalue = overwrite_value
self.preserve_value = preserve_value
self.preserve_filter_value = preserve_filter_value
self.setLayout(layout)
def __init__(self, parent=None):
QtWidgets.QWidget.__init__(self, parent)
self.setWindowTitle("Project Manager")
self.resize(1200, 800)
self.projects = {}
layout = QtWidgets.QVBoxLayout()
# Main control
ctrl_button_w = 30
ctrl_button_h = 30
# Calculate icon size
icon_size = QtCore.QSize(ctrl_button_w - 4, ctrl_button_h - 4)
main_control_layout = QtWidgets.QHBoxLayout()
database_label = QtWidgets.QLabel()
# Main buttons - create
create_button = QtWidgets.QPushButton()
create_icon = qta.icon("fa.plus-square", color=style.colors.light)
create_button.setIconSize(icon_size)
create_button.setFixedWidth(ctrl_button_w)
create_button.setFixedHeight(ctrl_button_h)
create_button.setIcon(create_icon)
create_button.setStyleSheet(cbstyle.flat_button)
# Main buttons - refresh
refresh_button = QtWidgets.QPushButton()
refresh_icon = qta.icon("fa.refresh", color=style.colors.light)
refresh_button.setIconSize(icon_size)
refresh_button.setFixedWidth(ctrl_button_w)
refresh_button.setFixedHeight(ctrl_button_h)
refresh_button.setIcon(refresh_icon)
refresh_button.setStyleSheet(cbstyle.flat_button)
# Project switch control
projects_label = QtWidgets.QLabel("Project:")
projects = QtWidgets.QComboBox()
projects.insertItem(0, "<None>")
# Add buttons to the main control layout
main_control_layout.addWidget(create_button)
main_control_layout.addStretch()
main_control_layout.addWidget(database_label)
main_control_layout.addWidget(projects_label)
main_control_layout.addWidget(projects)
main_control_layout.addWidget(refresh_button)
# Splitter for tabwidget and preview / details widget
split_widget = QtWidgets.QSplitter()
# Widgets will be stored in a StackedWidget
stacked_widget = QtWidgets.QStackedWidget()
# Control widgets which make the tool
manager_widget = ManageProjectWidget(parent=self)
# Sub manager widgets
overview = OverviewWidget()
manager_widget.add_widget(overview)
stacked_widget.insertWidget(manager_widget.order, manager_widget)
# Navigation panel widget
navigation_panel = Navigation()
# Add buttons to navigation panel
navigation_panel.add_button(manager_widget.label, manager_widget.order)
# Add widgets to the SplitWidget
split_widget.addWidget(navigation_panel)
split_widget.addWidget(stacked_widget)
split_widget.setHandleWidth(4)
split_widget.setSizes([100, 700])
layout.addLayout(main_control_layout)
layout.addWidget(split_widget)
self.setLayout(layout)
# To connect widget store in self attribute
self._navigation_panel = navigation_panel
self._stacked_widget = stacked_widget
self._database_label = database_label
self._create_button = create_button
self._projects = projects
self._refresh_button = refresh_button
self._overview = overview
self.connect_signals()
self.refresh()
manager_widget.setFocus(True)
def __init__(self, parent=None):
super(DatabaseSelectorWidget, self).__init__(parent=parent)
panel = {
"silo": QtWidgets.QWidget(),
"asset": QtWidgets.QWidget(),
"subset": QtWidgets.QWidget(),
"version": QtWidgets.QWidget(),
}
label = {
"silo": QtWidgets.QLabel("Silo"),
"asset": QtWidgets.QLabel("Asset"),
"subset": QtWidgets.QLabel("Subset"),
"version": QtWidgets.QLabel("Version"),
}
widget = {
"silo": QtWidgets.QComboBox(),
"asset": QtWidgets.QComboBox(),
"subset": QtWidgets.QComboBox(),
"version": QtWidgets.QComboBox(),
}
model = {
"silo": models.DatabaseDocumentModel(level="silo"),
"asset": models.DatabaseDocumentModel(level="asset"),
"subset": models.DatabaseDocumentModel(level="subset"),
"version": models.DatabaseDocumentModel(level="version"),
}
view = {
"silo": QtWidgets.QListView(),
"asset": QtWidgets.QListView(),
"subset": QtWidgets.QListView(),
"version": QtWidgets.QListView(),
}
widget["silo"].setModel(model["silo"])
widget["asset"].setModel(model["asset"])
widget["subset"].setModel(model["subset"])
widget["version"].setModel(model["version"])
widget["silo"].setView(view["silo"])
widget["asset"].setView(view["asset"])
widget["subset"].setView(view["subset"])
widget["version"].setView(view["version"])
def build_panel(level):
label[level].setFixedWidth(60)
label[level].setAlignment(QtCore.Qt.AlignVCenter |
QtCore.Qt.AlignRight)
layout = QtWidgets.QHBoxLayout(panel[level])
layout.addWidget(label[level])
layout.addWidget(widget[level])
build_panel("silo")
build_panel("asset")
build_panel("subset")
build_panel("version")
layout = QtWidgets.QVBoxLayout(self)
layout.addWidget(panel["silo"])
layout.addSpacing(-16)
layout.addWidget(panel["asset"])
layout.addSpacing(-16)
layout.addWidget(panel["subset"])
layout.addSpacing(-16)
layout.addWidget(panel["version"])
# Connect
def connect_index_changed(level, callback):
widget[level].currentIndexChanged.connect(callback)
connect_index_changed("silo", self.on_silo_changed)
connect_index_changed("asset", self.on_asset_changed)
connect_index_changed("subset", self.on_subset_changed)
connect_index_changed("version", self.on_version_changed)
# Init
self.widget = widget
self.model = model
self.view = view
silo = api.Session.get("AVALON_SILO")
if silo:
init_index = self.widget["silo"].findText(silo)
self.widget["silo"].setCurrentIndex(init_index)
asset = api.Session.get("AVALON_ASSET")
if silo and asset:
init_index = self.widget["asset"].findText(asset)
self.widget["asset"].setCurrentIndex(init_index)
def __init__(self, parent=None):
super(FocusComparing, self).__init__(parent=parent)
widget = pindict.to_pindict({
"overallDiff": {
"main": QtWidgets.QGroupBox("Compare Features"),
"name": {
"main": QtWidgets.QWidget(),
"icon": QtWidgets.QLabel(),
"label": QtWidgets.QLabel("Hierarchy"),
"status": QtWidgets.QLabel("--"),
},
"id": {
"main": QtWidgets.QWidget(),
"icon": QtWidgets.QLabel(),
"label": QtWidgets.QLabel("Avalon Id"),
"status": QtWidgets.QLabel("--"),
},
"mesh": {
"main": QtWidgets.QWidget(),
"icon": QtWidgets.QLabel(),
"label": QtWidgets.QLabel("Mesh"),
"status": QtWidgets.QLabel("--"),
},
"uv": {
"main": QtWidgets.QWidget(),
"icon": QtWidgets.QLabel(),
"label": QtWidgets.QLabel("UV"),
"status": QtWidgets.QLabel("--"),
},
},
"featureMenu": {
"main": QtWidgets.QWidget(),
"label": QtWidgets.QLabel("Focus On"),
"list": QtWidgets.QComboBox(),
},
"focus": {
"view": QtWidgets.QTreeView(),
"model": models.FocusModel(),
"pathDelegate": delegates.PathTextDelegate(),
}
})
with widget.pin("overallDiff") as diff:
layout = QtWidgets.QVBoxLayout(diff["main"])
for key in ["name", "id", "mesh", "uv"]:
with widget.pin("overallDiff." + key) as feature:
lay = QtWidgets.QHBoxLayout(feature["main"])
lay.addWidget(feature["label"])
lay.addSpacing(8)
lay.addWidget(feature["icon"])
lay.addSpacing(12)
lay.addWidget(feature["status"], stretch=True)
feature["label"].setFixedWidth(60)
feature["label"].setAlignment(QtCore.Qt.AlignRight)
icon = delegates.FEATURE_ICONS[key]
pixmap = lib.icon(icon, models.COLOR_DARK).pixmap(16, 16)
feature["icon"].setPixmap(pixmap)
layout.addWidget(feature["main"])
layout.addSpacing(-16)
layout.addSpacing(16)
layout.setContentsMargins(0, 0, 0, 0)
with widget.pin("featureMenu") as menu:
layout = QtWidgets.QHBoxLayout(menu["main"])
layout.addWidget(menu["label"])
layout.addWidget(menu["list"])
layout.addStretch()
layout = QtWidgets.QVBoxLayout(self)
layout.addSpacing(-16)
layout.addWidget(widget["overallDiff"]["main"])
layout.addSpacing(-8)
layout.addWidget(widget["featureMenu"]["main"])
layout.addSpacing(-8)
layout.addWidget(widget["focus"]["view"])
# Init
with widget.pin("featureMenu") as menu:
menu["list"].addItem(" Hierarchy", "longName")
menu["list"].addItem(" Avalon Id", "avalonId")
menu["list"].addItem(" Full Path", "fullPath")
menu["list"].addItem(" Mesh", "points")
menu["list"].addItem(" UV", "uvmap")
with widget.pin("focus") as focus:
focus["view"].setModel(focus["model"])
focus["view"].setItemDelegateForColumn(1, focus["pathDelegate"])
focus["view"].setHeaderHidden(True)
focus["view"].setUniformRowHeights(True)
focus["view"].setAlternatingRowColors(False)
focus["view"].setIndentation(6)
focus["view"].setStyleSheet("""
QTreeView::item{
padding: 2px 1px;
border: 0px;
}
""")
focus["view"].setSelectionMode(focus["view"].NoSelection)
height = focus["view"].sizeHintForRow(0) * 2 + 4 # MagicNum
focus["view"].setFixedHeight(height)
focus["view"].setColumnWidth(0, 28)
self.widget = widget
self._focusing = False
# Connect
with widget.pin("featureMenu") as menu:
menu["list"].currentIndexChanged.connect(self.on_feature_changed)
self.focus_enabled.connect(self.on_focus_enabled)
def __init__(self, parent=None):
QtWidgets.QWidget.__init__(self, parent=parent)
self.log = logging.getLogger(__name__)
self.setWindowTitle("Create New Project")
self.setWindowFlags(QtCore.Qt.Dialog)
self.setStyleSheet(style.create_widget)
layout = QtWidgets.QVBoxLayout()
layout.setAlignment(QtCore.Qt.AlignCenter)
layout.setContentsMargins(50, 50, 50, 50)
layout.setSpacing(20)
create_vlayout = QtWidgets.QVBoxLayout()
project_label = QtWidgets.QLabel("Project Name")
project_name = QtWidgets.QLineEdit()
project_name.setPlaceholderText("Example: MARVEL_Spiderman_HomeComing")
project_name.setStyleSheet("QLineEdit: {font-size: 50px}")
create_vlayout.addWidget(project_label)
create_vlayout.addWidget(project_name)
# Clone from existing
clone_vlayout = QtWidgets.QVBoxLayout()
clone_label_hlayout = QtWidgets.QHBoxLayout()
clone_toggle = QtWidgets.QCheckBox()
clone_toggle.setFixedWidth(24)
clone_toggle.setFixedHeight(24)
clone_label = QtWidgets.QLabel("Clone")
clone_label_hlayout.addWidget(clone_label)
clone_label_hlayout.addWidget(clone_toggle)
clone_project = QtWidgets.QComboBox()
clone_vlayout.addLayout(clone_label_hlayout)
clone_vlayout.addWidget(clone_project)
create_button = QtWidgets.QPushButton("Create")
create_button.setStyleSheet(style.flat_button)
create_button.setFixedWidth(500)
cancel_button = QtWidgets.QPushButton("Cancel")
cancel_button.setStyleSheet(style.flat_button)
cancel_button.setFixedWidth(500)
layout.addLayout(create_vlayout)
layout.addLayout(clone_vlayout)
layout.addWidget(create_button)
layout.addWidget(cancel_button)
self.setLayout(layout)
self.project_name = project_name
self.clone_toggle = clone_toggle
self.clone_project = clone_project
self.create = create_button
self.cancel = cancel_button
# Build connections
self.connect_signals()
self.refresh()
def createEditor(self, parent, option, index):
editor = QtWidgets.QComboBox(parent)
return editor
def __init__(self, parent=None):
################################################
# |---------------------| |------------------| #
# |Comp | |Asset | #
# |[..][ v]| |[ v]| #
# |---------------------| |------------------| #
# | Update existing comp [ ] | #
# |------------------------------------------| #
# | Switch | #
# |------------------------------------------| #
################################################
QtWidgets.QWidget.__init__(self, parent)
layout = QtWidgets.QVBoxLayout()
# Comp related input
comp_hlayout = QtWidgets.QHBoxLayout()
comp_label = QtWidgets.QLabel("Comp file")
comp_label.setFixedWidth(50)
comp_box = QtWidgets.QComboBox()
button_icon = qta.icon("fa.folder", color="white")
open_from_dir = QtWidgets.QPushButton()
open_from_dir.setIcon(button_icon)
comp_box.setFixedHeight(25)
open_from_dir.setFixedWidth(25)
open_from_dir.setFixedHeight(25)
comp_hlayout.addWidget(comp_label)
comp_hlayout.addWidget(comp_box)
comp_hlayout.addWidget(open_from_dir)
# Asset related input
asset_hlayout = QtWidgets.QHBoxLayout()
asset_label = QtWidgets.QLabel("Shot")
asset_label.setFixedWidth(50)
asset_box = QtWidgets.QComboBox()
asset_box.setLineEdit(QtWidgets.QLineEdit())
asset_box.setFixedHeight(25)
refresh_icon = qta.icon("fa.refresh", color="white")
refresh_btn = QtWidgets.QPushButton()
refresh_btn.setIcon(refresh_icon)
asset_box.setFixedHeight(25)
refresh_btn.setFixedWidth(25)
refresh_btn.setFixedHeight(25)
asset_hlayout.addWidget(asset_label)
asset_hlayout.addWidget(asset_box)
asset_hlayout.addWidget(refresh_btn)
# Options
options = QtWidgets.QHBoxLayout()
options.setAlignment(QtCore.Qt.AlignLeft)
current_comp_check = QtWidgets.QCheckBox()
current_comp_check.setChecked(True)
current_comp_label = QtWidgets.QLabel("Use current comp")
options.addWidget(current_comp_label)
options.addWidget(current_comp_check)
accept_btn = QtWidgets.QPushButton("Switch")
layout.addLayout(options)
layout.addLayout(comp_hlayout)
layout.addLayout(asset_hlayout)
layout.addWidget(accept_btn)
self._open_from_dir = open_from_dir
self._comps = comp_box
self._assets = asset_box
self._use_current = current_comp_check
self._accept_btn = accept_btn
self._refresh_btn = refresh_btn
self.setWindowTitle("Fusion Switch Shot")
self.setLayout(layout)
self.resize(260, 140)
self.setMinimumWidth(260)
self.setFixedHeight(140)
self.connections()
# Update ui to correct state
self._on_use_current_comp()
self._refresh()
