def updateModel(self):
"""
Refresh the name_comboBox with the current Maya scene state.
"""
self.populateModel(self.model)
index = None
col = lib_state.get_active_collection()
if col is None:
cols = lib_col.get_collections()
if len(cols) > 0:
# If there is no active collection, but there are
# collections already created, we make sure the first
# collection is marked active.
col = cols[0]
lib_state.set_active_collection(col)
else:
index = self.getActiveIndex(self.model, col)
block = self.blockSignals(True)
if index is not None:
self.name_comboBox.setCurrentIndex(index)
self.nameChanged.emit()
self.blockSignals(block)
return
def removeActiveNode(self):
"""
Delete the currently active collection.
"""
col = lib_state.get_active_collection()
if col is None:
LOG.warning('No active collection to delete.')
return
title = 'Remove Collection?'
text = 'Would you like to remove the current Collection?'
text += '\n\nRemove "' + str(col.get_node()) + '"?'
clicked_button = QtWidgets.QMessageBox.question(self, title, text)
if clicked_button != QtWidgets.QMessageBox.Yes:
LOG.warn('User cancelled operation.')
return
cols = lib_col.get_collections()
prev_col = lib_col.get_previous_collection(cols, col)
steps = lib_col.get_solver_steps_from_collection(col)
for step in steps:
lib_col.remove_solver_step_from_collection(col, step)
lib_col.delete_collection(col)
lib_state.set_active_collection(prev_col)
self.itemRemoved.emit()
self.itemChanged.emit()
self.nameChanged.emit()
return
def removeCollectionNode(self):
col = lib_state.get_active_collection()
if col is None:
LOG.warning('No active collection to delete.')
return
cols = lib_col.get_collections()
prev_col = lib_col.get_previous_collection(cols, col)
steps = lib_col.get_solver_steps_from_collection(col)
for step in steps:
lib_col.remove_solver_step_from_collection(col, step)
lib_col.delete_collection(col)
lib_state.set_active_collection(prev_col)
self.updateDynamicWindowTitle()
self.updateStatusWithSolveResult()
self.updateCollectionModel()
self.updateObjectToggleButtons()
self.updateObjectColumnVisibility()
self.updateObjectModel()
self.updateAttributeToggleButtons()
self.updateAttributeColumnVisibility()
self.updateAttributeModel()
self.updateSolverModel()
self.updateSolveValidState()
return
def populateCollectionModel(self, model):
cols = lib_col.get_collections()
string_data_list = []
for col in cols:
node = col.get_node()
string_data_list.append((node, col))
model.setStringDataList(string_data_list)
return
def collectionUpdateClicked(self):
col_list = col_lib.get_collections()
active_col = state_lib.get_active_collection()
self.updateCollectionList(
self.collection_comboBox,
self.collection_model,
active_col,
col_list,
)
return
def populateCollectionModel(self, model):
valid = uiutils.isValidQtObject(model)
if valid is False:
return
cols = lib_col.get_collections()
string_data_list = []
for col in cols:
node = col.get_node()
string_data_list.append((node, col))
model.setStringDataList(string_data_list)
return
def populateModel(self, model):
"""
Set the model based on the current Maya scene state.
"""
valid = uiutils.isValidQtObject(model)
if valid is False:
return
cols = lib_col.get_collections()
string_data_list = []
for col in cols:
node = col.get_node()
string_data_list.append((node, col))
model.setStringDataList(string_data_list)
return
def updateCollectionModel(self):
self.populateCollectionModel(self.collectionName_model)
col = lib_state.get_active_collection()
if col is None:
cols = lib_col.get_collections()
if len(cols) > 0:
# If there is no active collection, but there are
# collections already created, we make sure the first
# collection is marked active.
col = cols[0]
lib_state.set_active_collection(col)
if col is not None:
index = self.getDefaultCollectionIndex(self.collectionName_model, col)
if index is not None:
self.collectionName_comboBox.setCurrentIndex(index)
self.updateDynamicWindowTitle()
return
def removeActiveNode(self):
"""
Delete the currently active collection.
"""
col = lib_state.get_active_collection()
if col is None:
LOG.warning('No active collection to delete.')
return
cols = lib_col.get_collections()
prev_col = lib_col.get_previous_collection(cols, col)
steps = lib_col.get_solver_steps_from_collection(col)
for step in steps:
lib_col.remove_solver_step_from_collection(col, step)
lib_col.delete_collection(col)
lib_state.set_active_collection(prev_col)
self.itemRemoved.emit()
self.itemChanged.emit()
self.nameChanged.emit()
return
def apply(self):
cam = None
mkr_grp = None
col = None
file_path = self.subForm.getFilePath()
load_mode = self.subForm.getLoadModeText()
camera_text = self.subForm.getCameraText()
camera_data = self.subForm.getCameraData()
mkr_grp_text = self.subForm.getMarkerGroupText()
mkr_grp_data = self.subForm.getMarkerGroupData()
add_to_collection = self.subForm.getAddToCollectionValue()
collection_text = self.subForm.getCollectionText()
collection_data = self.subForm.getCollectionData()
load_bnd_pos = self.subForm.getLoadBundlePositions()
undist_mode = self.subForm.getDistortionModeText()
use_overscan = self.subForm.getUseOverscanValue()
undistorted = undist_mode == const.UNDISTORTION_MODE_VALUE
width, height = self.subForm.getImageResolution()
camera_field_of_view = None
if use_overscan is True:
camera_field_of_view = self.subForm.getCameraFieldOfViewValue()
# Temporarily disable adding new Markers to the Active
# Collection.
config = userprefs_lib.get_config()
key = userprefs_const.REG_EVNT_ADD_NEW_MKR_TO_KEY
old_value = userprefs_lib.get_value(config, key)
temp_value = userprefs_const.REG_EVNT_ADD_NEW_MKR_TO_NONE_VALUE
userprefs_lib.set_value(config, key, temp_value)
try:
self.progressBar.setValue(0)
self.progressBar.show()
with undoutils.undo_chunk_context():
_, mkr_data_list = mayareadfile.read(
file_path,
image_width=width,
image_height=height,
undistorted=undistorted,
)
self.progressBar.setValue(50)
if load_mode == const.LOAD_MODE_NEW_VALUE:
# Get Camera and MarkerGroup.
if camera_text == const.NEW_CAMERA_VALUE:
cam = lib.create_new_camera()
mkr_grp = lib.create_new_marker_group(cam)
else:
cam = camera_data
if mkr_grp_text == const.NEW_MARKER_GROUP_VALUE:
mkr_grp = lib.create_new_marker_group(cam)
else:
mkr_grp = mkr_grp_data
self.progressBar.setValue(60)
# Get Collection
col = None
if add_to_collection is True:
if collection_text == const.NEW_COLLECTION_VALUE:
col = col_lib.create_collection()
else:
col = collection_data
self.progressBar.setValue(70)
mayareadfile.create_nodes(
mkr_data_list,
cam=cam,
mkr_grp=mkr_grp,
col=col,
with_bundles=True,
load_bundle_position=load_bnd_pos,
camera_field_of_view=camera_field_of_view,
)
elif load_mode == const.LOAD_MODE_REPLACE_VALUE:
self.progressBar.setValue(60)
mkr_list = lib.get_selected_markers()
# NOTE: camera_field_of_view can only be from one
# camera, because (we assume) only one MarkerData
# file can be loaded at once.
mayareadfile.update_nodes(
mkr_list,
mkr_data_list,
load_bundle_position=load_bnd_pos,
camera_field_of_view=camera_field_of_view)
else:
raise ValueError('Load mode is not valid: %r' % load_mode)
self.progressBar.setValue(99)
lib.trigger_maya_to_refresh()
finally:
self.progressBar.setValue(100)
self.progressBar.hide()
# Restore original config value.
lib.deferred_revert_of_config_value(config, key, old_value)
# Update the camera comboBox with the created camera, or
# the last used camera.
all_camera_nodes = lib.get_cameras()
selected_cameras = [cam]
active_camera = cam
self.subForm.updateCameraList(self.subForm.camera_comboBox,
self.subForm.camera_model,
all_camera_nodes, selected_cameras,
active_camera)
active_camera = cam
active_mkr_grp = mkr_grp
mkr_grp_nodes = lib.get_marker_groups(active_camera)
self.subForm.updateMarkerGroupList(
self.subForm.markerGroup_comboBox,
self.subForm.markerGroup_model, active_mkr_grp, mkr_grp_nodes)
# Update the list of Collections, and pick the last used
# Collection.
active_col = col
col_list = col_lib.get_collections()
self.subForm.updateCollectionList(self.subForm.collection_comboBox,
self.subForm.collection_model,
active_col, col_list)
# Update config file with latest values.
config = get_config()
if config is not None:
config.set_value("data/use_overscan", use_overscan)
config.set_value("data/load_bundle_position", load_bnd_pos)
config.set_value("data/distortion_mode", undist_mode)
config.set_value("data/load_mode", load_mode)
config.write()
return
def populateUi(self):
config = get_config()
self._file_info = None
w, h = lib.get_default_image_resolution()
self.imageRes_label.setEnabled(False)
self.imageResWidth_label.setEnabled(False)
self.imageResWidth_spinBox.setValue(w)
self.imageResWidth_spinBox.setEnabled(False)
self.imageResHeight_label.setEnabled(False)
self.imageResHeight_spinBox.setValue(h)
self.imageResHeight_spinBox.setEnabled(False)
self.fileInfo_plainTextEdit.setReadOnly(True)
value = get_config_value(
config,
'data/load_bundle_position',
const.LOAD_BUNDLE_POS_DEFAULT_VALUE)
self.loadBndPositions_checkBox.setChecked(value)
# Get the file path from the clipboard.
try:
clippy = QtGui.QClipboard()
text = str(clippy.text()).strip()
if fileutils.is_valid_file_path(text):
self.setFilePath(text)
except Exception as e:
msg = 'Could not get file path from clipboard.'
LOG.warning(msg)
LOG.info(str(e))
all_camera_nodes = lib.get_cameras()
selected_cameras = lib.get_selected_cameras()
active_camera = lib.get_active_viewport_camera()
self.updateCameraList(
self.camera_comboBox,
self.camera_model,
all_camera_nodes,
selected_cameras,
active_camera
)
active_camera = self.getCameraData()
mkr_grp_nodes = lib.get_marker_groups(active_camera)
active_mkr_grp = None
self.updateMarkerGroupList(
self.markerGroup_comboBox,
self.markerGroup_model,
active_mkr_grp,
mkr_grp_nodes
)
col_list = col_lib.get_collections()
active_col = state_lib.get_active_collection()
self.updateCollectionList(
self.collection_comboBox,
self.collection_model,
active_col,
col_list
)
# If the user's preferences say to add the marker to the
# collection, then turn on the feature.
auto_add_marker = get_user_prefs_add_marker_to_collection()
self.collection_checkBox.setChecked(auto_add_marker)
self.updateCollectionEnabledState()
value = get_config_value(
config,
"data/load_mode",
const.LOAD_MODE_DEFAULT_VALUE
)
self.populateLoadModeModel(self.loadMode_model)
index = self.loadMode_model.stringList().index(value)
self.loadMode_comboBox.setCurrentIndex(index)
value = get_config_value(
config,
"data/distortion_mode",
const.DISTORTION_MODE_DEFAULT_VALUE
)
self.populateDistortionModeModel(self.distortionMode_model)
index = self.distortionMode_model.stringList().index(value)
self.distortionMode_comboBox.setCurrentIndex(index)
value = get_config_value(
config,
'data/use_overscan',
const.USE_OVERSCAN_DEFAULT_VALUE)
self.overscan_checkBox.setChecked(value)
self.updateOverscanValues()
return
