The openHarmony library enables people to create scripts for Harmony using less code and a simpler synthax.The complete documentation is available at this address:https://cfourney.github.io/OpenHarmony/
Install this library to be able to use the scripts that require it.
\",\n \"repository\": \"https://github.com/cfourney/OpenHarmony/\",\n \"isPackage\": false,\n \"files\": [\n \"openHarmony.js\",\n \"openHarmony/\"\n ],\n \"keywords\": [\n \"openHarmony\",\n \"library\"\n ],\n \"author\": \"OpenHarmony\",\n \"license\": \"MPL-2.0\",\n \"website\": \"https://github.com/cfourney/OpenHarmony/\",\n \"localFiles\": \"\"\n },\n {\n \"name\": \"oH Anim tools - Smartkey\",\n \"version\": \"1.0.0\",\n \"compatibility\": \"Harmony Premium 15\",\n \"description\": \"This script creates a key frame on the current timeline layer, and set the stop motion or interpolated mode of the key according to the surrounding keyframes. A Keyframe placed on an interpolation will remain interpolated, and a key placed between stop motion keyframes will also be set to stop motion.\\nThis script uses the OpenHarmony library. Install it first to be able to use it.
\\n\\nAssign this script to a shortcut with the script ScriptShortcuts.
\",\n \"repository\": \"https://github.com/cfourney/OpenHarmony/\",\n \"isPackage\": false,\n \"files\": [\n \"tools/OpenHarmony_basic/openHarmony_anim_tools.js\"\n ],\n \"keywords\": [\n \"openHarmony\",\n \"animation\"\n ],\n \"author\": \"Chris F\",\n \"license\": \"MPL-v2.0\",\n \"website\": \"https://github.com/cfourney/OpenHarmony/\",\n \"localFiles\": \"\"\n },\n {\n \"name\": \"oH Rigging tools\",\n \"version\": \"1.0.0\",\n \"compatibility\": \"Harmony Premium 15\",\n \"description\": \"OpenHarmony Rigging Tools\\nThose scripts require the openHarmony lib to work. Install it first for the scripts to work.
\\nAdd Centered Weighted Peg
Adds a peg with a pivot at the center of the selected drawing.
\\n\\nPlace Pivot with Click
Place the pivot with a simple click.
\\n\\nClean Unused Palettes
\\nFinds and removes all unnecessary palettes files from the filesystem. Doesn't support Element Palettes yet!
\\n\\nCreate Backdrop on Selection
Set up backdrops easily on the selection with this script.
\\n\",\n \"repository\": \"https://github.com/cfourney/OpenHarmony/\",\n \"isPackage\": false,\n \"files\": [\n \"tools/OpenHarmony_basic/openHarmony_rigging_tools.js\",\n \"tools/OpenHarmony_basic/openHarmony_basic_backdropPicker.ui\"\n ],\n \"keywords\": [\n \"openHarmony\",\n \"palettes\"\n ],\n \"author\": \"Chris F\",\n \"license\": \"MPL-v2.0\",\n \"website\": \"https://github.com/cfourney/OpenHarmony/\",\n \"localFiles\": \"\"\n }\n ]\n}"
},
{
"path": "openpype/hosts/harmony/vendor/OpenHarmony/tools/OpenHarmony_basic/INSTALL",
"content": "scripts"
},
{
"path": "openpype/hosts/harmony/vendor/OpenHarmony/tools/OpenHarmony_basic/README",
"content": "Basic Helper Scripts, used for rigging, animation and compositing."
},
{
"path": "openpype/hosts/harmony/vendor/OpenHarmony/tools/OpenHarmony_basic/openHarmony_anim_tools.js",
"content": "/**\n * Load the Open Harmony Library as needed.\n */\ntry{\n var oh_incl = preferences.getString( 'openHarmonyIncludeDebug', false );\n if( oh_incl==\"false\" ) oh_incl = preferences.getString( 'openHarmonyInclude', false );\n if( oh_incl==\"false\" ) oh_incl = \"openHarmony.js\";\n include(oh_incl);\n}catch(err){\n MessageBox.information (\"OpenHarmony is not installed. Get it here: \\nhttps://github.com/cfourney/OpenHarmony\")\n}\n\n\n/**\n * Smart key button. Only adds a key if a column already exists. Maintains sections that are tweens and other sections that are stop-motion/holds.\n */\nfunction oh_anim_smartKey(){\n System.println(\"\");\n \n scene.beginUndoRedoAccum( \"oh_anim_smartKey\" );\n \n var timeline = $.scene.getTimeline();\n var layers = timeline.selectedLayers;\n \n //--------------------------------------------------\n //--- The key function, used \n \n var smart_key_item = function( attr ){\n var cfrm = $.scene.currentFrame; \n \n var frame = attr.frames[ cfrm ];\n\n \n if( attr.node.type == \"READ\" ){\n \n if( attr.column.type == \"DRAWING\" ){\n frame.isKeyFrame = true;\n return;\n }\n \n //DONT KEY THE OFFSETS IF ITS NOT ANIMATEABLE.\n var check_pos = {\n \"offset.x\" : true,\n \"offset.y\" : true,\n \"offset.z\" : true,\n \"skew\" : true,\n \"scale.x\" : true,\n \"scale.y\" : true,\n \"scale.z\" : true,\n \"rotation.anglez\" : true\n }\n if( !attr.node[\"can_animate\"] ){\n return;\n }\n }\n \n if( !frame.isKeyFrame ){\n var lk = frame.keyframeLeft;\n frame.isKeyFrame = true;\n \n if(!lk){\n //Consider this as static.\n frame.constant = true;\n }else{\n if( lk.constant ){\n frame.constant = true;\n // lk.constant = true; //UNNECESSARY, DEFAULT APPRAOCH TO isKeyFrame = true;\n }else{\n var rk = frame.keyframeRight;\n \n if( rk ){\n //Something is to the right, keep the tween.\n frame.constant = false;\n // lk.constant = false; //UNNECESSARY, DEFAULT APPRAOCH TO isKeyFrame = true;\n }else{\n frame.constant = true;\n lk.constant = true;\n }\n }\n }\n }\n }\n \n \n if( layers.length == 0 ){\n var layers = timeline.compositionLayers;\n \n var items = [];\n for( var n=0;n\n\n Dialog\n \n \n \n 0\n 0\n 471\n 223\n \n \n \n Backdrop Details\n \n \n true\n \n \n - \n \n
- \n \n \n \n 75\n 0\n \n \n \n Name\n \n \n
\n - \n \n
\n \n \n - \n \n
- \n \n \n \n 75\n 0\n \n \n \n Text\n \n \n Qt::AlignLeading|Qt::AlignLeft|Qt::AlignTop\n \n \n
\n - \n \n
\n \n \n - \n \n
- \n \n \n \n 74\n 0\n \n \n \n Color\n \n \n
\n - \n \n \n \n 1\n 0\n \n \n \n \n \n \n
\n - \n \n \n Change Color\n \n \n
\n \n \n - \n \n \n Qt::Horizontal\n \n \n QDialogButtonBox::Cancel|QDialogButtonBox::Ok\n \n \n
\n \n \n \n \n \n buttonBox\n accepted()\n Dialog\n accept()\n \n \n 248\n 254\n \n \n 157\n 274\n \n \n \n \n buttonBox\n rejected()\n Dialog\n reject()\n \n \n 316\n 260\n \n \n 286\n 274\n \n \n \n \n\n"
},
{
"path": "openpype/hosts/harmony/vendor/OpenHarmony/tools/OpenHarmony_basic/openHarmony_rigging_tools.js",
"content": "/**\n * Load the Open Harmony Library as needed.\n */\n\ntry{\n var oh_incl = preferences.getString( 'openHarmonyIncludeDebug', false );\n if( oh_incl==\"false\" ) oh_incl = preferences.getString( 'openHarmonyInclude', false );\n if( oh_incl==\"false\" ) oh_incl = \"openHarmony.js\";\n include(oh_incl);\n}catch(err){\n MessageBox.information (\"OpenHarmony is not installed. Get it here: \\nhttps://github.com/cfourney/OpenHarmony\")\n}\n\n/**\n * Finds and removes all unnecessary asset files from the filesystem.\n */\nfunction oh_rigging_removeUnnecesaryPaletteFiles(){\n var palette_list = $.scene.palettes;\n var registered_palette_files = {};\n \n //Find the path of all registered palettes. Add it to a group for easy lookup.\n for( var n=0;n3 ){\n labelText += '\\n and '+(unreferenced_palettes.length-3)+' more . . .';\n }\n \n var confirmation = $.dialog.confirm( \"Remove Palettes\", labelText );\n \n if( confirmation ){\n //Delete all palettes from disk.\n var prog = new $.dialog.Progress( \"Removing Palettes\", unreferenced_palettes.length, true );\n \n for( var n=0;n2 ){\n var clean_lcs = lcs.sequence;\n \n if( clean_lcs.slice( clean_lcs.length-1 ) == (\"_\") ){\n clean_lcs = clean_lcs.slice( 0, clean_lcs.length-1 );\n }\n \n if( clean_lcs.slice( 0,1 ) == (\"_\") ){\n clean_lcs = clean_lcs.slice( 1 );\n }\n \n if(!common_substrings[clean_lcs]){\n common_substrings[clean_lcs] = 0;\n }\n \n common_substrings[clean_lcs]++;\n }\n }\n \n laststring = bnm;\n }\n\n var names = [];\n for( var n in common_substrings ){\n names.push( n );\n }\n \n //Now compare cleaned LCS and accumulate them as votes as well.\n for( var n=0;n2 ){\n var clean_lcs = lcs.sequence;\n \n if( clean_lcs.slice( clean_lcs.length-1 ) == (\"_\") ){\n clean_lcs = clean_lcs.slice( 0, clean_lcs.length-1 );\n } \n if( clean_lcs.slice( 0,1 ) == (\"_\") ){\n clean_lcs = clean_lcs.slice( 1 );\n }\n \n if(!common_substrings[clean_lcs]){\n common_substrings[clean_lcs] = 0;\n }\n common_substrings[clean_lcs]++;\n \n if( equivalent[clean_lcs] ){\n var clean_lcs2 = equivalent[clean_lcs];\n if(!common_substrings[clean_lcs2]){\n common_substrings[clean_lcs2] = 0;\n }\n common_substrings[clean_lcs2]++;\n }\n }\n }\n }\n \n //Find the highest voted LCS.\n var highest = 0;\n var common_name = 'Backdrop';\n for( var n in common_substrings ){\n if( common_substrings[n] > highest ){\n if( n.toUpperCase()==\"DRAWING\" ){\n continue;\n }\n \n highest = common_substrings[n];\n common_name = n;\n }\n }\n \n var res = color_selector.exec( common_name, \"\", \"\" );\n if( !res ){ //A cancel option was selected.\n continue;\n }\n \n //Add that beautiful backdrop.\n $.scene.addBackdropToNodes( grp, grp_items, res.name, res.text, res.color, 0, 0, 35, 35 );\n }\n \n scene.endUndoRedoAccum( );\n}\n\n\n/**\n * Sets the peg's pivot based on a clicked position in the interface.\n */\nfunction oh_rigging_setSelectedPegPivotWithClick(){\n var nodes = $.scene.nodeSearch( \"#PEG(SELECTED)\" );\n \n if( nodes.length == 0 ){\n $.dialog.alert( \"No peg selected.\" );\n return;\n }\n \n Action.perform( \"onActionChoosePencilTool()\" );\n var context = this;\n \n var setPiv = function( res ){\n var $ = context.$;\n var m = context;\n \n $.beginUndo();\n try{\n for( var n=0;n 1:\n if b in [bin.name() for bin in root_bin.bins()]:\n bin = [bin for bin in root_bin.bins() if b in bin.name()][0]\n done_bin_lst.append(bin)\n else:\n create_bin = hiero.core.Bin(b)\n root_bin.addItem(create_bin)\n done_bin_lst.append(create_bin)\n\n elif i >= 1 and i < len(path) - 1:\n if b in [bin.name() for bin in done_bin_lst[i - 1].bins()]:\n bin = [\n bin for bin in done_bin_lst[i - 1].bins()\n if b in bin.name()\n ][0]\n done_bin_lst.append(bin)\n else:\n create_bin = hiero.core.Bin(b)\n done_bin_lst[i - 1].addItem(create_bin)\n done_bin_lst.append(create_bin)\n\n elif i == len(path) - 1:\n if b in [bin.name() for bin in done_bin_lst[i - 1].bins()]:\n bin = [\n bin for bin in done_bin_lst[i - 1].bins()\n if b in bin.name()\n ][0]\n done_bin_lst.append(bin)\n else:\n create_bin = hiero.core.Bin(b)\n done_bin_lst[i - 1].addItem(create_bin)\n done_bin_lst.append(create_bin)\n\n return done_bin_lst[-1]\n\n\ndef split_by_client_version(string):\n regex = r\"[/_.]v\\d+\"\n try:\n matches = re.findall(regex, string, re.IGNORECASE)\n return string.split(matches[0])\n except Exception as error:\n log.error(error)\n return None\n\n\ndef get_selected_track_items(sequence=None):\n _sequence = sequence or get_current_sequence()\n\n # Getting selection\n timeline_editor = hiero.ui.getTimelineEditor(_sequence)\n return timeline_editor.selection()\n\n\ndef set_selected_track_items(track_items_list, sequence=None):\n _sequence = sequence or get_current_sequence()\n\n # make sure only trackItems are in list selection\n only_track_items = [\n i for i in track_items_list\n if isinstance(i, hiero.core.TrackItem)]\n\n # Getting selection\n timeline_editor = hiero.ui.getTimelineEditor(_sequence)\n return timeline_editor.setSelection(only_track_items)\n\n\ndef _read_doc_from_path(path):\n # reading QtXml.QDomDocument from HROX path\n hrox_file = QtCore.QFile(path)\n if not hrox_file.open(QtCore.QFile.ReadOnly):\n raise RuntimeError(\"Failed to open file for reading\")\n doc = QtXml.QDomDocument()\n doc.setContent(hrox_file)\n hrox_file.close()\n return doc\n\n\ndef _write_doc_to_path(doc, path):\n # write QtXml.QDomDocument to path as HROX\n hrox_file = QtCore.QFile(path)\n if not hrox_file.open(QtCore.QFile.WriteOnly):\n raise RuntimeError(\"Failed to open file for writing\")\n stream = QtCore.QTextStream(hrox_file)\n doc.save(stream, 1)\n hrox_file.close()\n\n\ndef _set_hrox_project_knobs(doc, **knobs):\n # set attributes to Project Tag\n proj_elem = doc.documentElement().firstChildElement(\"Project\")\n for k, v in knobs.items():\n if \"ocioconfigpath\" in k:\n paths_to_format = v[platform.system().lower()]\n for _path in paths_to_format:\n v = _path.format(**os.environ)\n if not os.path.exists(v):\n continue\n log.debug(\"Project colorspace knob `{}` was set to `{}`\".format(k, v))\n if isinstance(v, dict):\n continue\n proj_elem.setAttribute(str(k), v)\n\n\ndef apply_colorspace_project():\n \"\"\"Apply colorspaces from settings.\n\n Due to not being able to set the project settings through the Python API,\n we need to do use some dubious code to find the widgets and set them. It is\n possible to set the project settings without traversing through the widgets\n but it involves reading the hrox files from disk with XML, so no in-memory\n support. See https://community.foundry.com/discuss/topic/137771/change-a-project-s-default-color-transform-with-python # noqa\n for more details.\n \"\"\"\n # get presets for hiero\n project_name = get_current_project_name()\n imageio = get_project_settings(project_name)[\"hiero\"][\"imageio\"]\n presets = imageio.get(\"workfile\")\n\n # Open Project Settings UI.\n for act in hiero.ui.registeredActions():\n if act.objectName() == \"foundry.project.settings\":\n act.trigger()\n\n # Find widgets from their sibling label.\n labels = {\n \"Working Space:\": \"workingSpace\",\n \"Viewer:\": \"viewerLut\",\n \"Thumbnails:\": \"thumbnailLut\",\n \"Monitor Out:\": \"monitorOutLut\",\n \"8 Bit Files:\": \"eightBitLut\",\n \"16 Bit Files:\": \"sixteenBitLut\",\n \"Log Files:\": \"logLut\",\n \"Floating Point Files:\": \"floatLut\"\n }\n widgets = {x: None for x in labels.values()}\n\n def _recursive_children(widget, labels, widgets):\n children = widget.children()\n for count, child in enumerate(children):\n if isinstance(child, QtWidgets.QLabel):\n if child.text() in labels.keys():\n widgets[labels[child.text()]] = children[count + 1]\n _recursive_children(child, labels, widgets)\n\n app = QtWidgets.QApplication.instance()\n title = \"Project Settings\"\n for widget in app.topLevelWidgets():\n if isinstance(widget, QtWidgets.QMainWindow):\n if widget.windowTitle() != title:\n continue\n _recursive_children(widget, labels, widgets)\n widget.close()\n\n msg = \"Setting value \\\"{}\\\" is not a valid option for \\\"{}\\\"\"\n for key, widget in widgets.items():\n options = [widget.itemText(i) for i in range(widget.count())]\n setting_value = presets[key]\n assert setting_value in options, msg.format(setting_value, key)\n widget.setCurrentText(presets[key])\n\n # This code block is for setting up project colorspaces for files on disk.\n # Due to not having Python API access to set the project settings, the\n # Foundry recommended way is to modify the hrox files on disk with XML. See\n # this forum thread for more details;\n # https://community.foundry.com/discuss/topic/137771/change-a-project-s-default-color-transform-with-python # noqa\n '''\n # backward compatibility layer\n # TODO: remove this after some time\n config_data = get_imageio_config(\n project_name=get_current_project_name(),\n host_name=\"hiero\"\n )\n\n if config_data:\n presets.update({\n \"ocioConfigName\": \"custom\"\n })\n\n # get path the the active projects\n project = get_current_project()\n current_file = project.path()\n\n msg = \"The project needs to be saved to disk to apply colorspace settings.\"\n assert current_file, msg\n\n # save the workfile as subversion \"comment:_colorspaceChange\"\n split_current_file = os.path.splitext(current_file)\n copy_current_file = current_file\n\n if \"_colorspaceChange\" not in current_file:\n copy_current_file = (\n split_current_file[0]\n + \"_colorspaceChange\"\n + split_current_file[1]\n )\n\n try:\n # duplicate the file so the changes are applied only to the copy\n shutil.copyfile(current_file, copy_current_file)\n except shutil.Error:\n # in case the file already exists and it want to copy to the\n # same filewe need to do this trick\n # TEMP file name change\n copy_current_file_tmp = copy_current_file + \"_tmp\"\n # create TEMP file\n shutil.copyfile(current_file, copy_current_file_tmp)\n # remove original file\n os.remove(current_file)\n # copy TEMP back to original name\n shutil.copyfile(copy_current_file_tmp, copy_current_file)\n # remove the TEMP file as we dont need it\n os.remove(copy_current_file_tmp)\n\n # use the code from below for changing xml hrox Attributes\n presets.update({\"name\": os.path.basename(copy_current_file)})\n\n # read HROX in as QDomSocument\n doc = _read_doc_from_path(copy_current_file)\n\n # apply project colorspace properties\n _set_hrox_project_knobs(doc, **presets)\n\n # write QDomSocument back as HROX\n _write_doc_to_path(doc, copy_current_file)\n\n # open the file as current project\n hiero.core.openProject(copy_current_file)\n '''\n\n\ndef apply_colorspace_clips():\n project_name = get_current_project_name()\n project = get_current_project(remove_untitled=True)\n clips = project.clips()\n\n # get presets for hiero\n imageio = get_project_settings(project_name)[\"hiero\"][\"imageio\"]\n\n presets = imageio.get(\"regexInputs\", {}).get(\"inputs\", {})\n for clip in clips:\n clip_media_source_path = clip.mediaSource().firstpath()\n clip_name = clip.name()\n clip_colorspace = clip.sourceMediaColourTransform()\n\n if \"default\" in clip_colorspace:\n continue\n\n # check if any colorspace presets for read is matching\n preset_clrsp = None\n for k in presets:\n if not bool(re.search(k[\"regex\"], clip_media_source_path)):\n continue\n preset_clrsp = k[\"colorspace\"]\n\n if preset_clrsp:\n log.debug(\"Changing clip.path: {}\".format(clip_media_source_path))\n log.info(\"Changing clip `{}` colorspace {} to {}\".format(\n clip_name, clip_colorspace, preset_clrsp))\n # set the found preset to the clip\n clip.setSourceMediaColourTransform(preset_clrsp)\n\n # save project after all is changed\n project.save()\n\n\ndef is_overlapping(ti_test, ti_original, strict=False):\n covering_exp = (\n (ti_test.timelineIn() <= ti_original.timelineIn())\n and (ti_test.timelineOut() >= ti_original.timelineOut())\n )\n\n if strict:\n return covering_exp\n\n inside_exp = (\n (ti_test.timelineIn() >= ti_original.timelineIn())\n and (ti_test.timelineOut() <= ti_original.timelineOut())\n )\n overlaying_right_exp = (\n (ti_test.timelineIn() < ti_original.timelineOut())\n and (ti_test.timelineOut() >= ti_original.timelineOut())\n )\n overlaying_left_exp = (\n (ti_test.timelineOut() > ti_original.timelineIn())\n and (ti_test.timelineIn() <= ti_original.timelineIn())\n )\n\n return any((\n covering_exp,\n inside_exp,\n overlaying_right_exp,\n overlaying_left_exp\n ))\n\n\ndef get_sequence_pattern_and_padding(file):\n \"\"\" Return sequence pattern and padding from file\n\n Attributes:\n file (string): basename form path\n\n Example:\n Can find file.0001.ext, file.%02d.ext, file.####.ext\n\n Return:\n string: any matching sequence pattern\n int: padding of sequnce numbering\n \"\"\"\n foundall = re.findall(\n r\"(#+)|(%\\d+d)|(?<=[^a-zA-Z0-9])(\\d+)(?=\\.\\w+$)\", file)\n if not foundall:\n return None, None\n found = sorted(list(set(foundall[0])))[-1]\n\n padding = int(\n re.findall(r\"\\d+\", found)[-1]) if \"%\" in found else len(found)\n return found, padding\n\n\ndef sync_clip_name_to_data_asset(track_items_list):\n # loop through all selected clips\n for track_item in track_items_list:\n # ignore if parent track is locked or disabled\n if track_item.parent().isLocked():\n continue\n if not track_item.parent().isEnabled():\n continue\n # ignore if the track item is disabled\n if not track_item.isEnabled():\n continue\n\n # get name and data\n ti_name = track_item.name()\n data = get_trackitem_openpype_data(track_item)\n\n # ignore if no data on the clip or not publish instance\n if not data:\n continue\n if data.get(\"id\") != \"pyblish.avalon.instance\":\n continue\n\n # fix data if wrong name\n if data[\"asset\"] != ti_name:\n data[\"asset\"] = ti_name\n # remove the original tag\n tag = get_trackitem_openpype_tag(track_item)\n track_item.removeTag(tag)\n # create new tag with updated data\n set_trackitem_openpype_tag(track_item, data)\n print(\"asset was changed in clip: {}\".format(ti_name))\n\n\ndef set_track_color(track_item, color):\n track_item.source().binItem().setColor(color)\n\n\ndef check_inventory_versions(track_items=None):\n \"\"\"\n Actual version color identifier of Loaded containers\n\n Check all track items and filter only\n Loader nodes for its version. It will get all versions from database\n and check if the node is having actual version. If not then it will color\n it to red.\n \"\"\"\n from . import parse_container\n\n track_items = track_items or get_track_items()\n # presets\n clip_color_last = \"green\"\n clip_color = \"red\"\n\n containers = []\n # Find all containers and collect it's node and representation ids\n for track_item in track_items:\n container = parse_container(track_item)\n if container:\n containers.append(container)\n\n # Skip if nothing was found\n if not containers:\n return\n\n project_name = get_current_project_name()\n filter_result = filter_containers(containers, project_name)\n for container in filter_result.latest:\n set_track_color(container[\"_item\"], clip_color_last)\n\n for container in filter_result.outdated:\n set_track_color(container[\"_item\"], clip_color)\n\n\ndef selection_changed_timeline(event):\n \"\"\"Callback on timeline to check if asset in data is the same as clip name.\n\n Args:\n event (hiero.core.Event): timeline event\n \"\"\"\n timeline_editor = event.sender\n selection = timeline_editor.selection()\n\n track_items = get_track_items(\n selection=selection,\n track_type=\"video\",\n check_enabled=True,\n check_locked=True,\n check_tagged=True\n )\n\n # run checking function\n sync_clip_name_to_data_asset(track_items)\n\n\ndef before_project_save(event):\n track_items = get_track_items(\n track_type=\"video\",\n check_enabled=True,\n check_locked=True,\n check_tagged=True\n )\n\n # run checking function\n sync_clip_name_to_data_asset(track_items)\n\n # also mark old versions of loaded containers\n check_inventory_versions(track_items)\n\n\ndef get_main_window():\n \"\"\"Acquire Nuke's main window\"\"\"\n if _CTX.parent_gui is None:\n top_widgets = QtWidgets.QApplication.topLevelWidgets()\n name = \"Foundry::UI::DockMainWindow\"\n main_window = next(widget for widget in top_widgets if\n widget.inherits(\"QMainWindow\") and\n widget.metaObject().className() == name)\n _CTX.parent_gui = main_window\n return _CTX.parent_gui\n"
},
{
"path": "openpype/hosts/hiero/api/menu.py",
"content": "import os\nimport sys\n\nimport hiero.core\nfrom hiero.ui import findMenuAction\n\nfrom qtpy import QtGui\n\nfrom openpype.lib import Logger\nfrom openpype.tools.utils import host_tools\nfrom openpype.settings import get_project_settings\nfrom openpype.pipeline import (\n get_current_project_name,\n get_current_asset_name,\n get_current_task_name\n)\n\nfrom . import tags\n\nlog = Logger.get_logger(__name__)\n\nself = sys.modules[__name__]\nself._change_context_menu = None\n\n\ndef get_context_label():\n return \"{}, {}\".format(\n get_current_asset_name(),\n get_current_task_name()\n )\n\n\ndef update_menu_task_label():\n \"\"\"Update the task label in Avalon menu to current session\"\"\"\n\n object_name = self._change_context_menu\n found_menu = findMenuAction(object_name)\n\n if not found_menu:\n log.warning(\"Can't find menuItem: {}\".format(object_name))\n return\n\n label = get_context_label()\n\n menu = found_menu.menu()\n self._change_context_menu = label\n menu.setTitle(label)\n\n\ndef menu_install():\n \"\"\"\n Installing menu into Hiero\n\n \"\"\"\n\n from . import (\n publish, launch_workfiles_app, reload_config,\n apply_colorspace_project, apply_colorspace_clips\n )\n from .lib import get_main_window\n\n main_window = get_main_window()\n\n # here is the best place to add menu\n\n menu_name = os.environ['AVALON_LABEL']\n\n context_label = get_context_label()\n\n self._change_context_menu = context_label\n\n try:\n check_made_menu = findMenuAction(menu_name)\n except Exception:\n check_made_menu = None\n\n if not check_made_menu:\n # Grab Hiero's MenuBar\n menu = hiero.ui.menuBar().addMenu(menu_name)\n else:\n menu = check_made_menu.menu()\n\n context_label_action = menu.addAction(context_label)\n context_label_action.setEnabled(False)\n\n menu.addSeparator()\n\n workfiles_action = menu.addAction(\"Work Files...\")\n workfiles_action.setIcon(QtGui.QIcon(\"icons:Position.png\"))\n workfiles_action.triggered.connect(launch_workfiles_app)\n\n default_tags_action = menu.addAction(\"Create Default Tags\")\n default_tags_action.setIcon(QtGui.QIcon(\"icons:Position.png\"))\n default_tags_action.triggered.connect(tags.add_tags_to_workfile)\n\n menu.addSeparator()\n\n creator_action = menu.addAction(\"Create...\")\n creator_action.setIcon(QtGui.QIcon(\"icons:CopyRectangle.png\"))\n creator_action.triggered.connect(\n lambda: host_tools.show_creator(parent=main_window)\n )\n\n publish_action = menu.addAction(\"Publish...\")\n publish_action.setIcon(QtGui.QIcon(\"icons:Output.png\"))\n publish_action.triggered.connect(\n lambda *args: publish(hiero.ui.mainWindow())\n )\n\n loader_action = menu.addAction(\"Load...\")\n loader_action.setIcon(QtGui.QIcon(\"icons:CopyRectangle.png\"))\n loader_action.triggered.connect(\n lambda: host_tools.show_loader(parent=main_window)\n )\n\n sceneinventory_action = menu.addAction(\"Manage...\")\n sceneinventory_action.setIcon(QtGui.QIcon(\"icons:CopyRectangle.png\"))\n sceneinventory_action.triggered.connect(\n lambda: host_tools.show_scene_inventory(parent=main_window)\n )\n\n library_action = menu.addAction(\"Library...\")\n library_action.setIcon(QtGui.QIcon(\"icons:CopyRectangle.png\"))\n library_action.triggered.connect(\n lambda: host_tools.show_library_loader(parent=main_window)\n )\n\n if os.getenv(\"OPENPYPE_DEVELOP\"):\n menu.addSeparator()\n reload_action = menu.addAction(\"Reload pipeline\")\n reload_action.setIcon(QtGui.QIcon(\"icons:ColorAdd.png\"))\n reload_action.triggered.connect(reload_config)\n\n menu.addSeparator()\n apply_colorspace_p_action = menu.addAction(\"Apply Colorspace Project\")\n apply_colorspace_p_action.setIcon(QtGui.QIcon(\"icons:ColorAdd.png\"))\n apply_colorspace_p_action.triggered.connect(apply_colorspace_project)\n\n apply_colorspace_c_action = menu.addAction(\"Apply Colorspace Clips\")\n apply_colorspace_c_action.setIcon(QtGui.QIcon(\"icons:ColorAdd.png\"))\n apply_colorspace_c_action.triggered.connect(apply_colorspace_clips)\n\n menu.addSeparator()\n\n exeprimental_action = menu.addAction(\"Experimental tools...\")\n exeprimental_action.triggered.connect(\n lambda: host_tools.show_experimental_tools_dialog(parent=main_window)\n )\n\n\ndef add_scripts_menu():\n try:\n from . import launchforhiero\n except ImportError:\n\n log.warning(\n \"Skipping studio.menu install, because \"\n \"'scriptsmenu' module seems unavailable.\"\n )\n return\n\n # load configuration of custom menu\n project_settings = get_project_settings(get_current_project_name())\n config = project_settings[\"hiero\"][\"scriptsmenu\"][\"definition\"]\n _menu = project_settings[\"hiero\"][\"scriptsmenu\"][\"name\"]\n\n if not config:\n log.warning(\"Skipping studio menu, no definition found.\")\n return\n\n # run the launcher for Hiero menu\n studio_menu = launchforhiero.main(title=_menu.title())\n\n # apply configuration\n studio_menu.build_from_configuration(studio_menu, config)\n"
},
{
"path": "openpype/hosts/hiero/api/otio/__init__.py",
"content": ""
},
{
"path": "openpype/hosts/hiero/api/otio/hiero_export.py",
"content": "\"\"\" compatibility OpenTimelineIO 0.12.0 and newer\n\"\"\"\n\nimport os\nimport re\nimport ast\nimport opentimelineio as otio\nfrom . import utils\nimport hiero.core\nimport hiero.ui\n\n\nTRACK_TYPE_MAP = {\n hiero.core.VideoTrack: otio.schema.TrackKind.Video,\n hiero.core.AudioTrack: otio.schema.TrackKind.Audio\n}\nMARKER_COLOR_MAP = {\n \"magenta\": otio.schema.MarkerColor.MAGENTA,\n \"red\": otio.schema.MarkerColor.RED,\n \"yellow\": otio.schema.MarkerColor.YELLOW,\n \"green\": otio.schema.MarkerColor.GREEN,\n \"cyan\": otio.schema.MarkerColor.CYAN,\n \"blue\": otio.schema.MarkerColor.BLUE,\n}\n\n\nclass CTX:\n project_fps = None\n timeline = None\n include_tags = True\n\n\ndef flatten(list_):\n for item_ in list_:\n if isinstance(item_, (list, tuple)):\n for sub_item in flatten(item_):\n yield sub_item\n else:\n yield item_\n\n\ndef create_otio_rational_time(frame, fps):\n return otio.opentime.RationalTime(\n float(frame),\n float(fps)\n )\n\n\ndef create_otio_time_range(start_frame, frame_duration, fps):\n return otio.opentime.TimeRange(\n start_time=create_otio_rational_time(start_frame, fps),\n duration=create_otio_rational_time(frame_duration, fps)\n )\n\n\ndef _get_metadata(item):\n if hasattr(item, 'metadata'):\n return {key: value for key, value in dict(item.metadata()).items()}\n return {}\n\n\ndef create_time_effects(otio_clip, track_item):\n # get all subtrack items\n subTrackItems = flatten(track_item.parent().subTrackItems())\n speed = track_item.playbackSpeed()\n\n otio_effect = None\n # retime on track item\n if speed != 1.:\n # make effect\n otio_effect = otio.schema.LinearTimeWarp()\n otio_effect.name = \"Speed\"\n otio_effect.time_scalar = speed\n\n # freeze frame effect\n if speed == 0.:\n otio_effect = otio.schema.FreezeFrame()\n otio_effect.name = \"FreezeFrame\"\n\n if otio_effect:\n # add otio effect to clip effects\n otio_clip.effects.append(otio_effect)\n\n # loop through and get all Timewarps\n for effect in subTrackItems:\n if ((track_item not in effect.linkedItems())\n and (len(effect.linkedItems()) > 0)):\n continue\n # avoid all effect which are not TimeWarp and disabled\n if \"TimeWarp\" not in effect.name():\n continue\n\n if not effect.isEnabled():\n continue\n\n node = effect.node()\n name = node[\"name\"].value()\n\n # solve effect class as effect name\n _name = effect.name()\n if \"_\" in _name:\n effect_name = re.sub(r\"(?:_)[_0-9]+\", \"\", _name) # more numbers\n else:\n effect_name = re.sub(r\"\\d+\", \"\", _name) # one number\n\n metadata = {}\n # add knob to metadata\n for knob in [\"lookup\", \"length\"]:\n value = node[knob].value()\n animated = node[knob].isAnimated()\n if animated:\n value = [\n ((node[knob].getValueAt(i)) - i)\n for i in range(\n track_item.timelineIn(), track_item.timelineOut() + 1)\n ]\n\n metadata[knob] = value\n\n # make effect\n otio_effect = otio.schema.TimeEffect()\n otio_effect.name = name\n otio_effect.effect_name = effect_name\n otio_effect.metadata.update(metadata)\n\n # add otio effect to clip effects\n otio_clip.effects.append(otio_effect)\n\n\ndef create_otio_reference(clip):\n metadata = _get_metadata(clip)\n media_source = clip.mediaSource()\n\n # get file info for path and start frame\n file_info = media_source.fileinfos().pop()\n frame_start = file_info.startFrame()\n path = file_info.filename()\n\n # get padding and other file infos\n padding = media_source.filenamePadding()\n file_head = media_source.filenameHead()\n is_sequence = not media_source.singleFile()\n frame_duration = media_source.duration()\n fps = utils.get_rate(clip) or CTX.project_fps\n extension = os.path.splitext(path)[-1]\n\n if is_sequence:\n metadata.update({\n \"isSequence\": True,\n \"padding\": padding\n })\n\n # add resolution metadata\n metadata.update({\n \"openpype.source.colourtransform\": clip.sourceMediaColourTransform(),\n \"openpype.source.width\": int(media_source.width()),\n \"openpype.source.height\": int(media_source.height()),\n \"openpype.source.pixelAspect\": float(media_source.pixelAspect())\n })\n\n otio_ex_ref_item = None\n\n if is_sequence:\n # if it is file sequence try to create `ImageSequenceReference`\n # the OTIO might not be compatible so return nothing and do it old way\n try:\n dirname = os.path.dirname(path)\n otio_ex_ref_item = otio.schema.ImageSequenceReference(\n target_url_base=dirname + os.sep,\n name_prefix=file_head,\n name_suffix=extension,\n start_frame=frame_start,\n frame_zero_padding=padding,\n rate=fps,\n available_range=create_otio_time_range(\n frame_start,\n frame_duration,\n fps\n )\n )\n except AttributeError:\n pass\n\n if not otio_ex_ref_item:\n reformat_path = utils.get_reformated_path(path, padded=False)\n # in case old OTIO or video file create `ExternalReference`\n otio_ex_ref_item = otio.schema.ExternalReference(\n target_url=reformat_path,\n available_range=create_otio_time_range(\n frame_start,\n frame_duration,\n fps\n )\n )\n\n # add metadata to otio item\n add_otio_metadata(otio_ex_ref_item, media_source, **metadata)\n\n return otio_ex_ref_item\n\n\ndef get_marker_color(tag):\n icon = tag.icon()\n pat = r'icons:Tag(?P\\w+)\\.\\w+'\n\n res = re.search(pat, icon)\n if res:\n color = res.groupdict().get('color')\n if color.lower() in MARKER_COLOR_MAP:\n return MARKER_COLOR_MAP[color.lower()]\n\n return otio.schema.MarkerColor.RED\n\n\ndef create_otio_markers(otio_item, item):\n for tag in item.tags():\n if not tag.visible():\n continue\n\n if tag.name() == 'Copy':\n # Hiero adds this tag to a lot of clips\n continue\n\n frame_rate = utils.get_rate(item) or CTX.project_fps\n\n marked_range = otio.opentime.TimeRange(\n start_time=otio.opentime.RationalTime(\n tag.inTime(),\n frame_rate\n ),\n duration=otio.opentime.RationalTime(\n int(tag.metadata().dict().get('tag.length', '0')),\n frame_rate\n )\n )\n # add tag metadata but remove \"tag.\" string\n metadata = {}\n\n for key, value in tag.metadata().dict().items():\n _key = key.replace(\"tag.\", \"\")\n\n try:\n # capture exceptions which are related to strings only\n _value = ast.literal_eval(value)\n except (ValueError, SyntaxError):\n _value = value\n\n metadata.update({_key: _value})\n\n # Store the source item for future import assignment\n metadata['hiero_source_type'] = item.__class__.__name__\n\n marker = otio.schema.Marker(\n name=tag.name(),\n color=get_marker_color(tag),\n marked_range=marked_range,\n metadata=metadata\n )\n\n otio_item.markers.append(marker)\n\n\ndef create_otio_clip(track_item):\n clip = track_item.source()\n speed = track_item.playbackSpeed()\n # flip if speed is in minus\n source_in = track_item.sourceIn() if speed > 0 else track_item.sourceOut()\n\n duration = int(track_item.duration())\n\n fps = utils.get_rate(track_item) or CTX.project_fps\n name = track_item.name()\n\n media_reference = create_otio_reference(clip)\n source_range = create_otio_time_range(\n int(source_in),\n int(duration),\n fps\n )\n\n otio_clip = otio.schema.Clip(\n name=name,\n source_range=source_range,\n media_reference=media_reference\n )\n\n # Add tags as markers\n if CTX.include_tags:\n create_otio_markers(otio_clip, track_item)\n create_otio_markers(otio_clip, track_item.source())\n\n # only if video\n if not clip.mediaSource().hasAudio():\n # Add effects to clips\n create_time_effects(otio_clip, track_item)\n\n return otio_clip\n\n\ndef create_otio_gap(gap_start, clip_start, tl_start_frame, fps):\n return otio.schema.Gap(\n source_range=create_otio_time_range(\n gap_start,\n (clip_start - tl_start_frame) - gap_start,\n fps\n )\n )\n\n\ndef _create_otio_timeline():\n project = CTX.timeline.project()\n metadata = _get_metadata(CTX.timeline)\n\n metadata.update({\n \"openpype.timeline.width\": int(CTX.timeline.format().width()),\n \"openpype.timeline.height\": int(CTX.timeline.format().height()),\n \"openpype.timeline.pixelAspect\": int(CTX.timeline.format().pixelAspect()), # noqa\n \"openpype.project.useOCIOEnvironmentOverride\": project.useOCIOEnvironmentOverride(), # noqa\n \"openpype.project.lutSetting16Bit\": project.lutSetting16Bit(),\n \"openpype.project.lutSetting8Bit\": project.lutSetting8Bit(),\n \"openpype.project.lutSettingFloat\": project.lutSettingFloat(),\n \"openpype.project.lutSettingLog\": project.lutSettingLog(),\n \"openpype.project.lutSettingViewer\": project.lutSettingViewer(),\n \"openpype.project.lutSettingWorkingSpace\": project.lutSettingWorkingSpace(), # noqa\n \"openpype.project.lutUseOCIOForExport\": project.lutUseOCIOForExport(),\n \"openpype.project.ocioConfigName\": project.ocioConfigName(),\n \"openpype.project.ocioConfigPath\": project.ocioConfigPath()\n })\n\n start_time = create_otio_rational_time(\n CTX.timeline.timecodeStart(), CTX.project_fps)\n\n return otio.schema.Timeline(\n name=CTX.timeline.name(),\n global_start_time=start_time,\n metadata=metadata\n )\n\n\ndef create_otio_track(track_type, track_name):\n return otio.schema.Track(\n name=track_name,\n kind=TRACK_TYPE_MAP[track_type]\n )\n\n\ndef add_otio_gap(track_item, otio_track, prev_out):\n gap_length = track_item.timelineIn() - prev_out\n if prev_out != 0:\n gap_length -= 1\n\n gap = otio.opentime.TimeRange(\n duration=otio.opentime.RationalTime(\n gap_length,\n CTX.project_fps\n )\n )\n otio_gap = otio.schema.Gap(source_range=gap)\n otio_track.append(otio_gap)\n\n\ndef add_otio_metadata(otio_item, media_source, **kwargs):\n metadata = _get_metadata(media_source)\n\n # add additional metadata from kwargs\n if kwargs:\n metadata.update(kwargs)\n\n # add metadata to otio item metadata\n for key, value in metadata.items():\n otio_item.metadata.update({key: value})\n\n\ndef create_otio_timeline():\n\n def set_prev_item(itemindex, track_item):\n # Add Gap if needed\n if itemindex == 0:\n # if it is first track item at track then add\n # it to previous item\n return track_item\n\n else:\n # get previous item\n return track_item.parent().items()[itemindex - 1]\n\n # get current timeline\n CTX.timeline = hiero.ui.activeSequence()\n CTX.project_fps = CTX.timeline.framerate().toFloat()\n\n # convert timeline to otio\n otio_timeline = _create_otio_timeline()\n\n # loop all defined track types\n for track in CTX.timeline.items():\n # skip if track is disabled\n if not track.isEnabled():\n continue\n\n # convert track to otio\n otio_track = create_otio_track(\n type(track), track.name())\n\n for itemindex, track_item in enumerate(track):\n # Add Gap if needed\n if itemindex == 0:\n # if it is first track item at track then add\n # it to previous item\n prev_item = track_item\n\n else:\n # get previous item\n prev_item = track_item.parent().items()[itemindex - 1]\n\n # calculate clip frame range difference from each other\n clip_diff = track_item.timelineIn() - prev_item.timelineOut()\n\n # add gap if first track item is not starting\n # at first timeline frame\n if itemindex == 0 and track_item.timelineIn() > 0:\n add_otio_gap(track_item, otio_track, 0)\n\n # or add gap if following track items are having\n # frame range differences from each other\n elif itemindex and clip_diff != 1:\n add_otio_gap(track_item, otio_track, prev_item.timelineOut())\n\n # create otio clip and add it to track\n otio_clip = create_otio_clip(track_item)\n otio_track.append(otio_clip)\n\n # Add tags as markers\n if CTX.include_tags:\n create_otio_markers(otio_track, track)\n\n # add track to otio timeline\n otio_timeline.tracks.append(otio_track)\n\n return otio_timeline\n\n\ndef write_to_file(otio_timeline, path):\n otio.adapters.write_to_file(otio_timeline, path)\n"
},
{
"path": "openpype/hosts/hiero/api/otio/hiero_import.py",
"content": "#!/usr/bin/env python\n# -*- coding: utf-8 -*-\n\n__author__ = \"Daniel Flehner Heen\"\n__credits__ = [\"Jakub Jezek\", \"Daniel Flehner Heen\"]\n\n\nimport os\nimport hiero.core\nimport hiero.ui\n\nimport PySide2.QtWidgets as qw\n\ntry:\n from urllib import unquote\n\nexcept ImportError:\n from urllib.parse import unquote # lint:ok\n\nimport opentimelineio as otio\n\n_otio_old = False\n\n\ndef inform(messages):\n if isinstance(messages, type('')):\n messages = [messages]\n\n qw.QMessageBox.information(\n hiero.ui.mainWindow(),\n 'OTIO Import',\n '\\n'.join(messages),\n qw.QMessageBox.StandardButton.Ok\n )\n\n\ndef get_transition_type(otio_item, otio_track):\n _in, _out = otio_track.neighbors_of(otio_item)\n\n if isinstance(_in, otio.schema.Gap):\n _in = None\n\n if isinstance(_out, otio.schema.Gap):\n _out = None\n\n if _in and _out:\n return 'dissolve'\n\n elif _in and not _out:\n return 'fade_out'\n\n elif not _in and _out:\n return 'fade_in'\n\n else:\n return 'unknown'\n\n\ndef find_trackitem(otio_clip, hiero_track):\n for item in hiero_track.items():\n if item.timelineIn() == otio_clip.range_in_parent().start_time.value:\n if item.name() == otio_clip.name:\n return item\n\n return None\n\n\ndef get_neighboring_trackitems(otio_item, otio_track, hiero_track):\n _in, _out = otio_track.neighbors_of(otio_item)\n trackitem_in = None\n trackitem_out = None\n\n if _in:\n trackitem_in = find_trackitem(_in, hiero_track)\n\n if _out:\n trackitem_out = find_trackitem(_out, hiero_track)\n\n return trackitem_in, trackitem_out\n\n\ndef apply_transition(otio_track, otio_item, track):\n warning = None\n\n # Figure out type of transition\n transition_type = get_transition_type(otio_item, otio_track)\n\n # Figure out track kind for getattr below\n kind = ''\n if isinstance(track, hiero.core.AudioTrack):\n kind = 'Audio'\n\n # Gather TrackItems involved in trasition\n item_in, item_out = get_neighboring_trackitems(\n otio_item,\n otio_track,\n track\n )\n\n # Create transition object\n if transition_type == 'dissolve':\n transition_func = getattr(\n hiero.core.Transition,\n 'create{kind}DissolveTransition'.format(kind=kind)\n )\n\n try:\n transition = transition_func(\n item_in,\n item_out,\n otio_item.in_offset.value,\n otio_item.out_offset.value\n )\n\n # Catch error raised if transition is bigger than TrackItem source\n except RuntimeError as e:\n transition = None\n warning = (\n \"Unable to apply transition \\\"{t.name}\\\": {e} \"\n \"Ignoring the transition.\").format(t=otio_item, e=str(e))\n\n elif transition_type == 'fade_in':\n transition_func = getattr(\n hiero.core.Transition,\n 'create{kind}FadeInTransition'.format(kind=kind)\n )\n\n # Warn user if part of fade is outside of clip\n if otio_item.in_offset.value:\n warning = \\\n 'Fist half of transition \"{t.name}\" is outside of clip and ' \\\n 'not valid in Hiero. Only applied second half.' \\\n .format(t=otio_item)\n\n transition = transition_func(\n item_out,\n otio_item.out_offset.value\n )\n\n elif transition_type == 'fade_out':\n transition_func = getattr(\n hiero.core.Transition,\n 'create{kind}FadeOutTransition'.format(kind=kind)\n )\n transition = transition_func(\n item_in,\n otio_item.in_offset.value\n )\n\n # Warn user if part of fade is outside of clip\n if otio_item.out_offset.value:\n warning = \\\n 'Second half of transition \"{t.name}\" is outside of clip ' \\\n 'and not valid in Hiero. Only applied first half.' \\\n .format(t=otio_item)\n\n else:\n # Unknown transition\n return\n\n # Apply transition to track\n if transition:\n track.addTransition(transition)\n\n # Inform user about missing or adjusted transitions\n return warning\n\n\ndef prep_url(url_in):\n url = unquote(url_in)\n\n if url.startswith('file://localhost/'):\n return url\n\n url = 'file://localhost{sep}{url}'.format(\n sep=url.startswith(os.sep) and '' or os.sep,\n url=url.startswith(os.sep) and url[1:] or url\n )\n\n return url\n\n\ndef create_offline_mediasource(otio_clip, path=None):\n global _otio_old\n\n hiero_rate = hiero.core.TimeBase(\n otio_clip.source_range.start_time.rate\n )\n\n try:\n legal_media_refs = (\n otio.schema.ExternalReference,\n otio.schema.ImageSequenceReference\n )\n except AttributeError:\n _otio_old = True\n legal_media_refs = (\n otio.schema.ExternalReference\n )\n\n if isinstance(otio_clip.media_reference, legal_media_refs):\n source_range = otio_clip.available_range()\n\n else:\n source_range = otio_clip.source_range\n\n if path is None:\n path = otio_clip.name\n\n media = hiero.core.MediaSource.createOfflineVideoMediaSource(\n prep_url(path),\n source_range.start_time.value,\n source_range.duration.value,\n hiero_rate,\n source_range.start_time.value\n )\n\n return media\n\n\ndef load_otio(otio_file, project=None, sequence=None):\n otio_timeline = otio.adapters.read_from_file(otio_file)\n build_sequence(otio_timeline, project=project, sequence=sequence)\n\n\nmarker_color_map = {\n \"PINK\": \"Magenta\",\n \"RED\": \"Red\",\n \"ORANGE\": \"Yellow\",\n \"YELLOW\": \"Yellow\",\n \"GREEN\": \"Green\",\n \"CYAN\": \"Cyan\",\n \"BLUE\": \"Blue\",\n \"PURPLE\": \"Magenta\",\n \"MAGENTA\": \"Magenta\",\n \"BLACK\": \"Blue\",\n \"WHITE\": \"Green\"\n}\n\n\ndef get_tag(tagname, tagsbin):\n for tag in tagsbin.items():\n if tag.name() == tagname:\n return tag\n\n if isinstance(tag, hiero.core.Bin):\n tag = get_tag(tagname, tag)\n\n if tag is not None:\n return tag\n\n return None\n\n\ndef add_metadata(metadata, hiero_item):\n for key, value in metadata.get('Hiero', dict()).items():\n if key == 'source_type':\n # Only used internally to reassign tag to correct Hiero item\n continue\n\n if isinstance(value, dict):\n add_metadata(value, hiero_item)\n continue\n\n if value is not None:\n if not key.startswith('tag.'):\n key = 'tag.' + key\n\n hiero_item.metadata().setValue(key, str(value))\n\n\ndef add_markers(otio_item, hiero_item, tagsbin):\n if isinstance(otio_item, (otio.schema.Stack, otio.schema.Clip)):\n markers = otio_item.markers\n\n elif isinstance(otio_item, otio.schema.Timeline):\n markers = otio_item.tracks.markers\n\n else:\n markers = []\n\n for marker in markers:\n meta = marker.metadata.get('Hiero', dict())\n if 'source_type' in meta:\n if hiero_item.__class__.__name__ != meta.get('source_type'):\n continue\n\n marker_color = marker.color\n\n _tag = get_tag(marker.name, tagsbin)\n if _tag is None:\n _tag = get_tag(marker_color_map[marker_color], tagsbin)\n\n if _tag is None:\n _tag = hiero.core.Tag(marker_color_map[marker.color])\n\n start = marker.marked_range.start_time.value\n end = (\n marker.marked_range.start_time.value +\n marker.marked_range.duration.value\n )\n\n if hasattr(hiero_item, 'addTagToRange'):\n tag = hiero_item.addTagToRange(_tag, start, end)\n\n else:\n tag = hiero_item.addTag(_tag)\n\n tag.setName(marker.name or marker_color_map[marker_color])\n # tag.setNote(meta.get('tag.note', ''))\n\n # Add metadata\n add_metadata(marker.metadata, tag)\n\n\ndef create_track(otio_track, tracknum, track_kind):\n if track_kind is None and hasattr(otio_track, 'kind'):\n track_kind = otio_track.kind\n\n # Create a Track\n if track_kind == otio.schema.TrackKind.Video:\n track = hiero.core.VideoTrack(\n otio_track.name or 'Video{n}'.format(n=tracknum)\n )\n\n else:\n track = hiero.core.AudioTrack(\n otio_track.name or 'Audio{n}'.format(n=tracknum)\n )\n\n return track\n\n\ndef create_clip(otio_clip, tagsbin, sequencebin):\n # Create MediaSource\n url = None\n media = None\n otio_media = otio_clip.media_reference\n\n if isinstance(otio_media, otio.schema.ExternalReference):\n url = prep_url(otio_media.target_url)\n media = hiero.core.MediaSource(url)\n\n elif not _otio_old:\n if isinstance(otio_media, otio.schema.ImageSequenceReference):\n url = prep_url(otio_media.abstract_target_url('#'))\n media = hiero.core.MediaSource(url)\n\n if media is None or media.isOffline():\n media = create_offline_mediasource(otio_clip, url)\n\n # Reuse previous clip if possible\n clip = None\n for item in sequencebin.clips():\n if item.activeItem().mediaSource() == media:\n clip = item.activeItem()\n break\n\n if not clip:\n # Create new Clip\n clip = hiero.core.Clip(media)\n\n # Add Clip to a Bin\n sequencebin.addItem(hiero.core.BinItem(clip))\n\n # Add markers\n add_markers(otio_clip, clip, tagsbin)\n\n return clip\n\n\ndef create_trackitem(playhead, track, otio_clip, clip):\n source_range = otio_clip.source_range\n\n trackitem = track.createTrackItem(otio_clip.name)\n trackitem.setPlaybackSpeed(source_range.start_time.rate)\n trackitem.setSource(clip)\n\n time_scalar = 1.\n\n # Check for speed effects and adjust playback speed accordingly\n for effect in otio_clip.effects:\n if isinstance(effect, otio.schema.LinearTimeWarp):\n time_scalar = effect.time_scalar\n # Only reverse effect can be applied here\n if abs(time_scalar) == 1.:\n trackitem.setPlaybackSpeed(\n trackitem.playbackSpeed() * time_scalar)\n\n elif isinstance(effect, otio.schema.FreezeFrame):\n # For freeze frame, playback speed must be set after range\n time_scalar = 0.\n\n # If reverse playback speed swap source in and out\n if trackitem.playbackSpeed() < 0:\n source_out = source_range.start_time.value\n source_in = source_range.end_time_inclusive().value\n\n timeline_in = playhead + source_out\n timeline_out = (\n timeline_in +\n source_range.duration.value\n ) - 1\n else:\n # Normal playback speed\n source_in = source_range.start_time.value\n source_out = source_range.end_time_inclusive().value\n\n timeline_in = playhead\n timeline_out = (\n timeline_in +\n source_range.duration.value\n ) - 1\n\n # Set source and timeline in/out points\n trackitem.setTimes(\n timeline_in,\n timeline_out,\n source_in,\n source_out\n\n )\n\n # Apply playback speed for freeze frames\n if abs(time_scalar) != 1.:\n trackitem.setPlaybackSpeed(trackitem.playbackSpeed() * time_scalar)\n\n # Link audio to video when possible\n if isinstance(track, hiero.core.AudioTrack):\n for other in track.parent().trackItemsAt(playhead):\n if other.source() == clip:\n trackitem.link(other)\n\n return trackitem\n\n\ndef build_sequence(\n otio_timeline, project=None, sequence=None, track_kind=None):\n if project is None:\n if sequence:\n project = sequence.project()\n\n else:\n # Per version 12.1v2 there is no way of getting active project\n project = hiero.core.projects(hiero.core.Project.kUserProjects)[-1]\n\n projectbin = project.clipsBin()\n\n if not sequence:\n # Create a Sequence\n sequence = hiero.core.Sequence(otio_timeline.name or 'OTIOSequence')\n\n # Set sequence settings from otio timeline if available\n if (\n hasattr(otio_timeline, 'global_start_time')\n and otio_timeline.global_start_time\n ):\n start_time = otio_timeline.global_start_time\n sequence.setFramerate(start_time.rate)\n sequence.setTimecodeStart(start_time.value)\n\n # Create a Bin to hold clips\n projectbin.addItem(hiero.core.BinItem(sequence))\n\n sequencebin = hiero.core.Bin(sequence.name())\n projectbin.addItem(sequencebin)\n\n else:\n sequencebin = projectbin\n\n # Get tagsBin\n tagsbin = hiero.core.project(\"Tag Presets\").tagsBin()\n\n # Add timeline markers\n add_markers(otio_timeline, sequence, tagsbin)\n\n if isinstance(otio_timeline, otio.schema.Timeline):\n tracks = otio_timeline.tracks\n\n else:\n tracks = [otio_timeline]\n\n for tracknum, otio_track in enumerate(tracks):\n playhead = 0\n _transitions = []\n\n # Add track to sequence\n track = create_track(otio_track, tracknum, track_kind)\n sequence.addTrack(track)\n\n # iterate over items in track\n for _itemnum, otio_clip in enumerate(otio_track):\n if isinstance(otio_clip, (otio.schema.Track, otio.schema.Stack)):\n inform('Nested sequences/tracks are created separately.')\n\n # Add gap where the nested sequence would have been\n playhead += otio_clip.source_range.duration.value\n\n # Process nested sequence\n build_sequence(\n otio_clip,\n project=project,\n track_kind=otio_track.kind\n )\n\n elif isinstance(otio_clip, otio.schema.Clip):\n # Create a Clip\n clip = create_clip(otio_clip, tagsbin, sequencebin)\n\n # Create TrackItem\n trackitem = create_trackitem(\n playhead,\n track,\n otio_clip,\n clip\n )\n\n # Add markers\n add_markers(otio_clip, trackitem, tagsbin)\n\n # Add trackitem to track\n track.addTrackItem(trackitem)\n\n # Update playhead\n playhead = trackitem.timelineOut() + 1\n\n elif isinstance(otio_clip, otio.schema.Transition):\n # Store transitions for when all clips in the track are created\n _transitions.append((otio_track, otio_clip))\n\n elif isinstance(otio_clip, otio.schema.Gap):\n # Hiero has no fillers, slugs or blanks at the moment\n playhead += otio_clip.source_range.duration.value\n\n # Apply transitions we stored earlier now that all clips are present\n warnings = []\n for otio_track, otio_item in _transitions:\n # Catch warnings form transitions in case\n # of unsupported transitions\n warning = apply_transition(otio_track, otio_item, track)\n if warning:\n warnings.append(warning)\n\n if warnings:\n inform(warnings)\n"
},
{
"path": "openpype/hosts/hiero/api/otio/utils.py",
"content": "import re\nimport opentimelineio as otio\n\n\ndef timecode_to_frames(timecode, framerate):\n rt = otio.opentime.from_timecode(timecode, 24)\n return int(otio.opentime.to_frames(rt))\n\n\ndef frames_to_timecode(frames, framerate):\n rt = otio.opentime.from_frames(frames, framerate)\n return otio.opentime.to_timecode(rt)\n\n\ndef frames_to_secons(frames, framerate):\n rt = otio.opentime.from_frames(frames, framerate)\n return otio.opentime.to_seconds(rt)\n\n\ndef get_reformated_path(path, padded=True):\n \"\"\"\n Return fixed python expression path\n\n Args:\n path (str): path url or simple file name\n\n Returns:\n type: string with reformated path\n\n Example:\n get_reformated_path(\"plate.[0001-1008].exr\") > plate.%04d.exr\n\n \"\"\"\n if \"%\" in path:\n padding_pattern = r\"(\\d+)\"\n padding = int(re.findall(padding_pattern, path).pop())\n num_pattern = r\"(%\\d+d)\"\n if padded:\n path = re.sub(num_pattern, \"%0{}d\".format(padding), path)\n else:\n path = re.sub(num_pattern, \"%d\", path)\n return path\n\n\ndef get_padding_from_path(path):\n \"\"\"\n Return padding number from DaVinci Resolve sequence path style\n\n Args:\n path (str): path url or simple file name\n\n Returns:\n int: padding number\n\n Example:\n get_padding_from_path(\"plate.[0001-1008].exr\") > 4\n\n \"\"\"\n padding_pattern = \"(\\\\d+)(?=-)\"\n if \"[\" in path:\n return len(re.findall(padding_pattern, path).pop())\n\n return None\n\n\ndef get_rate(item):\n if not hasattr(item, 'framerate'):\n return None\n\n num, den = item.framerate().toRational()\n\n try:\n rate = float(num) / float(den)\n except ZeroDivisionError:\n return None\n\n if rate.is_integer():\n return rate\n\n return round(rate, 4)\n"
},
{
"path": "openpype/hosts/hiero/api/pipeline.py",
"content": "\"\"\"\nBasic avalon integration\n\"\"\"\nfrom copy import deepcopy\nimport os\nimport contextlib\nfrom collections import OrderedDict\n\nfrom pyblish import api as pyblish\nfrom openpype.lib import Logger\nfrom openpype.pipeline import (\n schema,\n register_creator_plugin_path,\n register_loader_plugin_path,\n deregister_creator_plugin_path,\n deregister_loader_plugin_path,\n AVALON_CONTAINER_ID,\n)\nfrom openpype.tools.utils import host_tools\nfrom . import lib, menu, events\nimport hiero\n\nlog = Logger.get_logger(__name__)\n\n# plugin paths\nAPI_DIR = os.path.dirname(os.path.abspath(__file__))\nHOST_DIR = os.path.dirname(API_DIR)\nPLUGINS_DIR = os.path.join(HOST_DIR, \"plugins\")\nPUBLISH_PATH = os.path.join(PLUGINS_DIR, \"publish\").replace(\"\\\\\", \"/\")\nLOAD_PATH = os.path.join(PLUGINS_DIR, \"load\").replace(\"\\\\\", \"/\")\nCREATE_PATH = os.path.join(PLUGINS_DIR, \"create\").replace(\"\\\\\", \"/\")\n\nAVALON_CONTAINERS = \":AVALON_CONTAINERS\"\n\n\ndef install():\n \"\"\"Installing Hiero integration.\"\"\"\n\n # adding all events\n events.register_events()\n\n log.info(\"Registering Hiero plug-ins..\")\n pyblish.register_host(\"hiero\")\n pyblish.register_plugin_path(PUBLISH_PATH)\n register_loader_plugin_path(LOAD_PATH)\n register_creator_plugin_path(CREATE_PATH)\n\n # register callback for switching publishable\n pyblish.register_callback(\"instanceToggled\", on_pyblish_instance_toggled)\n\n # install menu\n menu.menu_install()\n menu.add_scripts_menu()\n\n # register hiero events\n events.register_hiero_events()\n\n\ndef uninstall():\n \"\"\"\n Uninstalling Hiero integration for avalon\n\n \"\"\"\n log.info(\"Deregistering Hiero plug-ins..\")\n pyblish.deregister_host(\"hiero\")\n pyblish.deregister_plugin_path(PUBLISH_PATH)\n deregister_loader_plugin_path(LOAD_PATH)\n deregister_creator_plugin_path(CREATE_PATH)\n\n # register callback for switching publishable\n pyblish.deregister_callback(\"instanceToggled\", on_pyblish_instance_toggled)\n\n\ndef containerise(track_item,\n name,\n namespace,\n context,\n loader=None,\n data=None):\n \"\"\"Bundle Hiero's object into an assembly and imprint it with metadata\n\n Containerisation enables a tracking of version, author and origin\n for loaded assets.\n\n Arguments:\n track_item (hiero.core.TrackItem): object to imprint as container\n name (str): Name of resulting assembly\n namespace (str): Namespace under which to host container\n context (dict): Asset information\n loader (str, optional): Name of node used to produce this container.\n\n Returns:\n track_item (hiero.core.TrackItem): containerised object\n\n \"\"\"\n\n data_imprint = OrderedDict({\n \"schema\": \"openpype:container-2.0\",\n \"id\": AVALON_CONTAINER_ID,\n \"name\": str(name),\n \"namespace\": str(namespace),\n \"loader\": str(loader),\n \"representation\": str(context[\"representation\"][\"_id\"]),\n })\n\n if data:\n for k, v in data.items():\n data_imprint.update({k: v})\n\n log.debug(\"_ data_imprint: {}\".format(data_imprint))\n lib.set_trackitem_openpype_tag(track_item, data_imprint)\n\n return track_item\n\n\ndef ls():\n \"\"\"List available containers.\n\n This function is used by the Container Manager in Nuke. You'll\n need to implement a for-loop that then *yields* one Container at\n a time.\n\n See the `container.json` schema for details on how it should look,\n and the Maya equivalent, which is in `avalon.maya.pipeline`\n \"\"\"\n\n # get all track items from current timeline\n all_items = lib.get_track_items()\n\n # append all video tracks\n for track in lib.get_current_sequence():\n if type(track) != hiero.core.VideoTrack:\n continue\n all_items.append(track)\n\n for item in all_items:\n container_data = parse_container(item)\n\n if isinstance(container_data, list):\n for _c in container_data:\n yield _c\n elif container_data:\n yield container_data\n\n\ndef parse_container(item, validate=True):\n \"\"\"Return container data from track_item's pype tag.\n\n Args:\n item (hiero.core.TrackItem or hiero.core.VideoTrack):\n A containerised track item.\n validate (bool)[optional]: validating with avalon scheme\n\n Returns:\n dict: The container schema data for input containerized track item.\n\n \"\"\"\n def data_to_container(item, data):\n if (\n not data\n or data.get(\"id\") != \"pyblish.avalon.container\"\n ):\n return\n\n if validate and data and data.get(\"schema\"):\n schema.validate(data)\n\n if not isinstance(data, dict):\n return\n\n # If not all required data return the empty container\n required = ['schema', 'id', 'name',\n 'namespace', 'loader', 'representation']\n\n if any(key not in data for key in required):\n return\n\n container = {key: data[key] for key in required}\n\n container[\"objectName\"] = item.name()\n\n # Store reference to the node object\n container[\"_item\"] = item\n\n return container\n\n # convert tag metadata to normal keys names\n if type(item) == hiero.core.VideoTrack:\n return_list = []\n _data = lib.get_track_openpype_data(item)\n\n if not _data:\n return\n # convert the data to list and validate them\n for _, obj_data in _data.items():\n container = data_to_container(item, obj_data)\n return_list.append(container)\n return return_list\n else:\n _data = lib.get_trackitem_openpype_data(item)\n return data_to_container(item, _data)\n\n\ndef _update_container_data(container, data):\n for key in container:\n try:\n container[key] = data[key]\n except KeyError:\n pass\n return container\n\n\ndef update_container(item, data=None):\n \"\"\"Update container data to input track_item or track's\n openpype tag.\n\n Args:\n item (hiero.core.TrackItem or hiero.core.VideoTrack):\n A containerised track item.\n data (dict)[optional]: dictionery with data to be updated\n\n Returns:\n bool: True if container was updated correctly\n\n \"\"\"\n\n data = data or {}\n data = deepcopy(data)\n\n if type(item) == hiero.core.VideoTrack:\n # form object data for test\n object_name = data[\"objectName\"]\n\n # get all available containers\n containers = lib.get_track_openpype_data(item)\n container = lib.get_track_openpype_data(item, object_name)\n\n containers = deepcopy(containers)\n container = deepcopy(container)\n\n # update data in container\n updated_container = _update_container_data(container, data)\n # merge updated container back to containers\n containers.update({object_name: updated_container})\n\n return bool(lib.set_track_openpype_tag(item, containers))\n else:\n container = lib.get_trackitem_openpype_data(item)\n updated_container = _update_container_data(container, data)\n\n log.info(\"Updating container: `{}`\".format(item.name()))\n return bool(lib.set_trackitem_openpype_tag(item, updated_container))\n\n\ndef launch_workfiles_app(*args):\n ''' Wrapping function for workfiles launcher '''\n from .lib import get_main_window\n\n main_window = get_main_window()\n # show workfile gui\n host_tools.show_workfiles(parent=main_window)\n\n\ndef publish(parent):\n \"\"\"Shorthand to publish from within host\"\"\"\n return host_tools.show_publish(parent)\n\n\n@contextlib.contextmanager\ndef maintained_selection():\n \"\"\"Maintain selection during context\n\n Example:\n >>> with maintained_selection():\n ... for track_item in track_items:\n ... < do some stuff >\n \"\"\"\n from .lib import (\n set_selected_track_items,\n get_selected_track_items\n )\n previous_selection = get_selected_track_items()\n reset_selection()\n try:\n # do the operation\n yield\n finally:\n reset_selection()\n set_selected_track_items(previous_selection)\n\n\ndef reset_selection():\n \"\"\"Deselect all selected nodes\n \"\"\"\n from .lib import set_selected_track_items\n set_selected_track_items([])\n\n\ndef reload_config():\n \"\"\"Attempt to reload pipeline at run-time.\n\n CAUTION: This is primarily for development and debugging purposes.\n\n \"\"\"\n import importlib\n\n for module in (\n \"openpype.hosts.hiero.lib\",\n \"openpype.hosts.hiero.menu\",\n \"openpype.hosts.hiero.tags\"\n ):\n log.info(\"Reloading module: {}...\".format(module))\n try:\n module = importlib.import_module(module)\n import imp\n imp.reload(module)\n except Exception as e:\n log.warning(\"Cannot reload module: {}\".format(e))\n importlib.reload(module)\n\n\ndef on_pyblish_instance_toggled(instance, old_value, new_value):\n \"\"\"Toggle node passthrough states on instance toggles.\"\"\"\n\n log.info(\"instance toggle: {}, old_value: {}, new_value:{} \".format(\n instance, old_value, new_value))\n\n from openpype.hosts.hiero.api import (\n get_trackitem_openpype_tag,\n set_publish_attribute\n )\n\n # Whether instances should be passthrough based on new value\n track_item = instance.data[\"item\"]\n tag = get_trackitem_openpype_tag(track_item)\n set_publish_attribute(tag, new_value)\n"
},
{
"path": "openpype/hosts/hiero/api/plugin.py",
"content": "import os\nfrom pprint import pformat\nimport re\nfrom copy import deepcopy\n\nimport hiero\n\nfrom qtpy import QtWidgets, QtCore\nimport qargparse\n\nfrom openpype.settings import get_current_project_settings\nfrom openpype.lib import Logger\nfrom openpype.pipeline import LoaderPlugin, LegacyCreator\nfrom openpype.pipeline.load import get_representation_path_from_context\nfrom . import lib\n\nlog = Logger.get_logger(__name__)\n\n\ndef load_stylesheet():\n path = os.path.join(os.path.dirname(__file__), \"style.css\")\n if not os.path.exists(path):\n log.warning(\"Unable to load stylesheet, file not found in resources\")\n return \"\"\n\n with open(path, \"r\") as file_stream:\n stylesheet = file_stream.read()\n return stylesheet\n\n\nclass CreatorWidget(QtWidgets.QDialog):\n\n # output items\n items = {}\n\n def __init__(self, name, info, ui_inputs, parent=None):\n super(CreatorWidget, self).__init__(parent)\n\n self.setObjectName(name)\n\n self.setWindowFlags(\n QtCore.Qt.Window\n | QtCore.Qt.CustomizeWindowHint\n | QtCore.Qt.WindowTitleHint\n | QtCore.Qt.WindowCloseButtonHint\n | QtCore.Qt.WindowStaysOnTopHint\n )\n self.setWindowTitle(name or \"Pype Creator Input\")\n self.resize(500, 700)\n\n # Where inputs and labels are set\n self.content_widget = [QtWidgets.QWidget(self)]\n top_layout = QtWidgets.QFormLayout(self.content_widget[0])\n top_layout.setObjectName(\"ContentLayout\")\n top_layout.addWidget(Spacer(5, self))\n\n # first add widget tag line\n top_layout.addWidget(QtWidgets.QLabel(info))\n\n # main dynamic layout\n self.scroll_area = QtWidgets.QScrollArea(self, widgetResizable=True)\n self.scroll_area.setVerticalScrollBarPolicy(\n QtCore.Qt.ScrollBarAsNeeded)\n self.scroll_area.setVerticalScrollBarPolicy(\n QtCore.Qt.ScrollBarAlwaysOn)\n self.scroll_area.setHorizontalScrollBarPolicy(\n QtCore.Qt.ScrollBarAlwaysOff)\n self.scroll_area.setWidgetResizable(True)\n\n self.content_widget.append(self.scroll_area)\n\n scroll_widget = QtWidgets.QWidget(self)\n in_scroll_area = QtWidgets.QVBoxLayout(scroll_widget)\n self.content_layout = [in_scroll_area]\n\n # add preset data into input widget layout\n self.items = self.populate_widgets(ui_inputs)\n self.scroll_area.setWidget(scroll_widget)\n\n # Confirmation buttons\n btns_widget = QtWidgets.QWidget(self)\n btns_layout = QtWidgets.QHBoxLayout(btns_widget)\n\n cancel_btn = QtWidgets.QPushButton(\"Cancel\")\n btns_layout.addWidget(cancel_btn)\n\n ok_btn = QtWidgets.QPushButton(\"Ok\")\n btns_layout.addWidget(ok_btn)\n\n # Main layout of the dialog\n main_layout = QtWidgets.QVBoxLayout(self)\n main_layout.setContentsMargins(10, 10, 10, 10)\n main_layout.setSpacing(0)\n\n # adding content widget\n for w in self.content_widget:\n main_layout.addWidget(w)\n\n main_layout.addWidget(btns_widget)\n\n ok_btn.clicked.connect(self._on_ok_clicked)\n cancel_btn.clicked.connect(self._on_cancel_clicked)\n\n stylesheet = load_stylesheet()\n self.setStyleSheet(stylesheet)\n\n def _on_ok_clicked(self):\n self.result = self.value(self.items)\n self.close()\n\n def _on_cancel_clicked(self):\n self.result = None\n self.close()\n\n def value(self, data, new_data=None):\n new_data = new_data or dict()\n for k, v in data.items():\n new_data[k] = {\n \"target\": None,\n \"value\": None\n }\n if v[\"type\"] == \"dict\":\n new_data[k][\"target\"] = v[\"target\"]\n new_data[k][\"value\"] = self.value(v[\"value\"])\n if v[\"type\"] == \"section\":\n new_data.pop(k)\n new_data = self.value(v[\"value\"], new_data)\n elif getattr(v[\"value\"], \"currentText\", None):\n new_data[k][\"target\"] = v[\"target\"]\n new_data[k][\"value\"] = v[\"value\"].currentText()\n elif getattr(v[\"value\"], \"isChecked\", None):\n new_data[k][\"target\"] = v[\"target\"]\n new_data[k][\"value\"] = v[\"value\"].isChecked()\n elif getattr(v[\"value\"], \"value\", None):\n new_data[k][\"target\"] = v[\"target\"]\n new_data[k][\"value\"] = v[\"value\"].value()\n elif getattr(v[\"value\"], \"text\", None):\n new_data[k][\"target\"] = v[\"target\"]\n new_data[k][\"value\"] = v[\"value\"].text()\n\n return new_data\n\n def camel_case_split(self, text):\n matches = re.finditer(\n '.+?(?:(?<=[a-z])(?=[A-Z])|(?<=[A-Z])(?=[A-Z][a-z])|$)', text)\n return \" \".join([str(m.group(0)).capitalize() for m in matches])\n\n def create_row(self, layout, type, text, **kwargs):\n value_keys = [\"setText\", \"setCheckState\", \"setValue\", \"setChecked\"]\n\n # get type attribute from qwidgets\n attr = getattr(QtWidgets, type)\n\n # convert label text to normal capitalized text with spaces\n label_text = self.camel_case_split(text)\n\n # assign the new text to label widget\n label = QtWidgets.QLabel(label_text)\n label.setObjectName(\"LineLabel\")\n\n # create attribute name text strip of spaces\n attr_name = text.replace(\" \", \"\")\n\n # create attribute and assign default values\n setattr(\n self,\n attr_name,\n attr(parent=self))\n\n # assign the created attribute to variable\n item = getattr(self, attr_name)\n\n # set attributes to item which are not values\n for func, val in kwargs.items():\n if func in value_keys:\n continue\n\n if getattr(item, func):\n log.debug(\"Setting {} to {}\".format(func, val))\n func_attr = getattr(item, func)\n if isinstance(val, tuple):\n func_attr(*val)\n else:\n func_attr(val)\n\n # set values to item\n for value_item in value_keys:\n if value_item not in kwargs:\n continue\n if getattr(item, value_item):\n getattr(item, value_item)(kwargs[value_item])\n\n # add to layout\n layout.addRow(label, item)\n\n return item\n\n def populate_widgets(self, data, content_layout=None):\n \"\"\"\n Populate widget from input dict.\n\n Each plugin has its own set of widget rows defined in dictionary\n each row values should have following keys: `type`, `target`,\n `label`, `order`, `value` and optionally also `toolTip`.\n\n Args:\n data (dict): widget rows or organized groups defined\n by types `dict` or `section`\n content_layout (QtWidgets.QFormLayout)[optional]: used when nesting\n\n Returns:\n dict: redefined data dict updated with created widgets\n\n \"\"\"\n\n content_layout = content_layout or self.content_layout[-1]\n # fix order of process by defined order value\n ordered_keys = list(data.keys())\n for k, v in data.items():\n try:\n # try removing a key from index which should\n # be filled with new\n ordered_keys.pop(v[\"order\"])\n except IndexError:\n pass\n # add key into correct order\n ordered_keys.insert(v[\"order\"], k)\n\n # process ordered\n for k in ordered_keys:\n v = data[k]\n tool_tip = v.get(\"toolTip\", \"\")\n if v[\"type\"] == \"dict\":\n # adding spacer between sections\n self.content_layout.append(QtWidgets.QWidget(self))\n content_layout.addWidget(self.content_layout[-1])\n self.content_layout[-1].setObjectName(\"sectionHeadline\")\n\n headline = QtWidgets.QVBoxLayout(self.content_layout[-1])\n headline.addWidget(Spacer(20, self))\n headline.addWidget(QtWidgets.QLabel(v[\"label\"]))\n\n # adding nested layout with label\n self.content_layout.append(QtWidgets.QWidget(self))\n self.content_layout[-1].setObjectName(\"sectionContent\")\n\n nested_content_layout = QtWidgets.QFormLayout(\n self.content_layout[-1])\n nested_content_layout.setObjectName(\"NestedContentLayout\")\n content_layout.addWidget(self.content_layout[-1])\n\n # add nested key as label\n data[k][\"value\"] = self.populate_widgets(\n v[\"value\"], nested_content_layout)\n\n if v[\"type\"] == \"section\":\n # adding spacer between sections\n self.content_layout.append(QtWidgets.QWidget(self))\n content_layout.addWidget(self.content_layout[-1])\n self.content_layout[-1].setObjectName(\"sectionHeadline\")\n\n headline = QtWidgets.QVBoxLayout(self.content_layout[-1])\n headline.addWidget(Spacer(20, self))\n headline.addWidget(QtWidgets.QLabel(v[\"label\"]))\n\n # adding nested layout with label\n self.content_layout.append(QtWidgets.QWidget(self))\n self.content_layout[-1].setObjectName(\"sectionContent\")\n\n nested_content_layout = QtWidgets.QFormLayout(\n self.content_layout[-1])\n nested_content_layout.setObjectName(\"NestedContentLayout\")\n content_layout.addWidget(self.content_layout[-1])\n\n # add nested key as label\n data[k][\"value\"] = self.populate_widgets(\n v[\"value\"], nested_content_layout)\n\n elif v[\"type\"] == \"QLineEdit\":\n data[k][\"value\"] = self.create_row(\n content_layout, \"QLineEdit\", v[\"label\"],\n setText=v[\"value\"], setToolTip=tool_tip)\n elif v[\"type\"] == \"QComboBox\":\n data[k][\"value\"] = self.create_row(\n content_layout, \"QComboBox\", v[\"label\"],\n addItems=v[\"value\"], setToolTip=tool_tip)\n elif v[\"type\"] == \"QCheckBox\":\n data[k][\"value\"] = self.create_row(\n content_layout, \"QCheckBox\", v[\"label\"],\n setChecked=v[\"value\"], setToolTip=tool_tip)\n elif v[\"type\"] == \"QSpinBox\":\n data[k][\"value\"] = self.create_row(\n content_layout, \"QSpinBox\", v[\"label\"],\n setValue=v[\"value\"],\n setDisplayIntegerBase=10000,\n setRange=(0, 99999), setMinimum=0,\n setMaximum=100000, setToolTip=tool_tip)\n\n return data\n\n\nclass Spacer(QtWidgets.QWidget):\n def __init__(self, height, *args, **kwargs):\n super(self.__class__, self).__init__(*args, **kwargs)\n\n self.setFixedHeight(height)\n\n real_spacer = QtWidgets.QWidget(self)\n real_spacer.setObjectName(\"Spacer\")\n real_spacer.setFixedHeight(height)\n\n layout = QtWidgets.QVBoxLayout(self)\n layout.setContentsMargins(0, 0, 0, 0)\n layout.addWidget(real_spacer)\n\n self.setLayout(layout)\n\n\nclass SequenceLoader(LoaderPlugin):\n \"\"\"A basic SequenceLoader for Resolve\n\n This will implement the basic behavior for a loader to inherit from that\n will containerize the reference and will implement the `remove` and\n `update` logic.\n\n \"\"\"\n\n options = [\n qargparse.Boolean(\n \"handles\",\n label=\"Include handles\",\n default=0,\n help=\"Load with handles or without?\"\n ),\n qargparse.Choice(\n \"load_to\",\n label=\"Where to load clips\",\n items=[\n \"Current timeline\",\n \"New timeline\"\n ],\n default=\"Current timeline\",\n help=\"Where do you want clips to be loaded?\"\n ),\n qargparse.Choice(\n \"load_how\",\n label=\"How to load clips\",\n items=[\n \"Original timing\",\n \"Sequentially in order\"\n ],\n default=\"Original timing\",\n help=\"Would you like to place it at original timing?\"\n )\n ]\n\n def load(\n self,\n context,\n name=None,\n namespace=None,\n options=None\n ):\n pass\n\n def update(self, container, representation):\n \"\"\"Update an existing `container`\n \"\"\"\n pass\n\n def remove(self, container):\n \"\"\"Remove an existing `container`\n \"\"\"\n pass\n\n\nclass ClipLoader:\n\n active_bin = None\n data = dict()\n\n def __init__(self, cls, context, path, **options):\n \"\"\" Initialize object\n\n Arguments:\n cls (avalon.api.Loader): plugin object\n context (dict): loader plugin context\n options (dict)[optional]: possible keys:\n projectBinPath: \"path/to/binItem\"\n\n \"\"\"\n self.__dict__.update(cls.__dict__)\n self.context = context\n self.active_project = lib.get_current_project()\n self.fname = path\n\n # try to get value from options or evaluate key value for `handles`\n self.with_handles = options.get(\"handles\") or bool(\n options.get(\"handles\") is True)\n # try to get value from options or evaluate key value for `load_how`\n self.sequencial_load = options.get(\"sequentially\") or bool(\n \"Sequentially in order\" in options.get(\"load_how\", \"\"))\n # try to get value from options or evaluate key value for `load_to`\n self.new_sequence = options.get(\"newSequence\") or bool(\n \"New timeline\" in options.get(\"load_to\", \"\"))\n self.clip_name_template = options.get(\n \"clipNameTemplate\") or \"{asset}_{subset}_{representation}\"\n assert self._populate_data(), str(\n \"Cannot Load selected data, look into database \"\n \"or call your supervisor\")\n\n # inject asset data to representation dict\n self._get_asset_data()\n log.info(\"__init__ self.data: `{}`\".format(pformat(self.data)))\n log.info(\"__init__ options: `{}`\".format(pformat(options)))\n\n # add active components to class\n if self.new_sequence:\n if options.get(\"sequence\"):\n # if multiselection is set then use options sequence\n self.active_sequence = options[\"sequence\"]\n else:\n # create new sequence\n self.active_sequence = lib.get_current_sequence(new=True)\n self.active_sequence.setFramerate(\n hiero.core.TimeBase.fromString(\n str(self.data[\"assetData\"][\"fps\"])))\n else:\n self.active_sequence = lib.get_current_sequence()\n\n if options.get(\"track\"):\n # if multiselection is set then use options track\n self.active_track = options[\"track\"]\n else:\n self.active_track = lib.get_current_track(\n self.active_sequence, self.data[\"track_name\"])\n\n def _populate_data(self):\n \"\"\" Gets context and convert it to self.data\n data structure:\n {\n \"name\": \"assetName_subsetName_representationName\"\n \"path\": \"path/to/file/created/by/get_repr..\",\n \"binPath\": \"projectBinPath\",\n }\n \"\"\"\n # create name\n repr = self.context[\"representation\"]\n repr_cntx = repr[\"context\"]\n asset = str(repr_cntx[\"asset\"])\n subset = str(repr_cntx[\"subset\"])\n representation = str(repr_cntx[\"representation\"])\n self.data[\"clip_name\"] = self.clip_name_template.format(**repr_cntx)\n self.data[\"track_name\"] = \"_\".join([subset, representation])\n self.data[\"versionData\"] = self.context[\"version\"][\"data\"]\n # gets file path\n file = get_representation_path_from_context(self.context)\n if not file:\n repr_id = repr[\"_id\"]\n log.warning(\n \"Representation id `{}` is failing to load\".format(repr_id))\n return None\n self.data[\"path\"] = file.replace(\"\\\\\", \"/\")\n\n # convert to hashed path\n if repr_cntx.get(\"frame\"):\n self._fix_path_hashes()\n\n # solve project bin structure path\n hierarchy = str(\"/\".join((\n \"Loader\",\n repr_cntx[\"hierarchy\"].replace(\"\\\\\", \"/\"),\n asset\n )))\n\n self.data[\"binPath\"] = hierarchy\n\n return True\n\n def _fix_path_hashes(self):\n \"\"\" Convert file path where it is needed padding with hashes\n \"\"\"\n file = self.data[\"path\"]\n if \"#\" not in file:\n frame = self.context[\"representation\"][\"context\"].get(\"frame\")\n padding = len(frame)\n file = file.replace(frame, \"#\" * padding)\n self.data[\"path\"] = file\n\n def _get_asset_data(self):\n \"\"\" Get all available asset data\n\n joint `data` key with asset.data dict into the representation\n\n \"\"\"\n\n asset_doc = self.context[\"asset\"]\n self.data[\"assetData\"] = asset_doc[\"data\"]\n\n def _make_track_item(self, source_bin_item, audio=False):\n \"\"\" Create track item with \"\"\"\n\n clip = source_bin_item.activeItem()\n\n # add to track as clip item\n if not audio:\n track_item = hiero.core.TrackItem(\n self.data[\"clip_name\"], hiero.core.TrackItem.kVideo)\n else:\n track_item = hiero.core.TrackItem(\n self.data[\"clip_name\"], hiero.core.TrackItem.kAudio)\n\n track_item.setSource(clip)\n track_item.setSourceIn(self.handle_start)\n track_item.setTimelineIn(self.timeline_in)\n track_item.setSourceOut((self.media_duration) - self.handle_end)\n track_item.setTimelineOut(self.timeline_out)\n track_item.setPlaybackSpeed(1)\n self.active_track.addTrackItem(track_item)\n\n return track_item\n\n def load(self):\n # create project bin for the media to be imported into\n self.active_bin = lib.create_bin(self.data[\"binPath\"])\n\n # create mediaItem in active project bin\n # create clip media\n self.media = hiero.core.MediaSource(self.data[\"path\"])\n self.media_duration = int(self.media.duration())\n\n # get handles\n self.handle_start = self.data[\"versionData\"].get(\"handleStart\")\n self.handle_end = self.data[\"versionData\"].get(\"handleEnd\")\n if self.handle_start is None:\n self.handle_start = self.data[\"assetData\"][\"handleStart\"]\n if self.handle_end is None:\n self.handle_end = self.data[\"assetData\"][\"handleEnd\"]\n\n self.handle_start = int(self.handle_start)\n self.handle_end = int(self.handle_end)\n\n if self.sequencial_load:\n last_track_item = lib.get_track_items(\n sequence_name=self.active_sequence.name(),\n track_name=self.active_track.name()\n )\n if len(last_track_item) == 0:\n last_timeline_out = 0\n else:\n last_track_item = last_track_item[-1]\n last_timeline_out = int(last_track_item.timelineOut()) + 1\n self.timeline_in = last_timeline_out\n self.timeline_out = last_timeline_out + int(\n self.data[\"assetData\"][\"clipOut\"]\n - self.data[\"assetData\"][\"clipIn\"])\n else:\n self.timeline_in = int(self.data[\"assetData\"][\"clipIn\"])\n self.timeline_out = int(self.data[\"assetData\"][\"clipOut\"])\n\n log.debug(\"__ self.timeline_in: {}\".format(self.timeline_in))\n log.debug(\"__ self.timeline_out: {}\".format(self.timeline_out))\n\n # check if slate is included\n slate_on = \"slate\" in self.context[\"version\"][\"data\"][\"families\"]\n log.debug(\"__ slate_on: {}\".format(slate_on))\n\n # if slate is on then remove the slate frame from beginning\n if slate_on:\n self.media_duration -= 1\n self.handle_start += 1\n\n # create Clip from Media\n clip = hiero.core.Clip(self.media)\n clip.setName(self.data[\"clip_name\"])\n\n # add Clip to bin if not there yet\n if self.data[\"clip_name\"] not in [\n b.name() for b in self.active_bin.items()]:\n bin_item = hiero.core.BinItem(clip)\n self.active_bin.addItem(bin_item)\n\n # just make sure the clip is created\n # there were some cases were hiero was not creating it\n source_bin_item = None\n for item in self.active_bin.items():\n if self.data[\"clip_name\"] == item.name():\n source_bin_item = item\n if not source_bin_item:\n log.warning(\"Problem with created Source clip: `{}`\".format(\n self.data[\"clip_name\"]))\n\n # include handles\n if self.with_handles:\n self.timeline_in -= self.handle_start\n self.timeline_out += self.handle_end\n self.handle_start = 0\n self.handle_end = 0\n\n # make track item from source in bin as item\n track_item = self._make_track_item(source_bin_item)\n\n log.info(\"Loading clips: `{}`\".format(self.data[\"clip_name\"]))\n return track_item\n\n\nclass Creator(LegacyCreator):\n \"\"\"Creator class wrapper\n \"\"\"\n clip_color = \"Purple\"\n rename_index = None\n\n def __init__(self, *args, **kwargs):\n super(Creator, self).__init__(*args, **kwargs)\n import openpype.hosts.hiero.api as phiero\n self.presets = get_current_project_settings()[\n \"hiero\"][\"create\"].get(self.__class__.__name__, {})\n\n # adding basic current context resolve objects\n self.project = phiero.get_current_project()\n self.sequence = phiero.get_current_sequence()\n\n if (self.options or {}).get(\"useSelection\"):\n timeline_selection = phiero.get_timeline_selection()\n self.selected = phiero.get_track_items(\n selection=timeline_selection\n )\n else:\n self.selected = phiero.get_track_items()\n\n self.widget = CreatorWidget\n\n\nclass PublishClip:\n \"\"\"\n Convert a track item to publishable instance\n\n Args:\n track_item (hiero.core.TrackItem): hiero track item object\n kwargs (optional): additional data needed for rename=True (presets)\n\n Returns:\n hiero.core.TrackItem: hiero track item object with pype tag\n \"\"\"\n vertical_clip_match = {}\n tag_data = {}\n types = {\n \"shot\": \"shot\",\n \"folder\": \"folder\",\n \"episode\": \"episode\",\n \"sequence\": \"sequence\",\n \"track\": \"sequence\",\n }\n\n # parents search pattern\n parents_search_pattern = r\"\\{([a-z]*?)\\}\"\n\n # default templates for non-ui use\n rename_default = False\n hierarchy_default = \"{_folder_}/{_sequence_}/{_track_}\"\n clip_name_default = \"shot_{_trackIndex_:0>3}_{_clipIndex_:0>4}\"\n subset_name_default = \"\"\n review_track_default = \"< none >\"\n subset_family_default = \"plate\"\n count_from_default = 10\n count_steps_default = 10\n vertical_sync_default = False\n driving_layer_default = \"\"\n\n def __init__(self, cls, track_item, **kwargs):\n # populate input cls attribute onto self.[attr]\n self.__dict__.update(cls.__dict__)\n\n # get main parent objects\n self.track_item = track_item\n sequence_name = lib.get_current_sequence().name()\n self.sequence_name = str(sequence_name).replace(\" \", \"_\")\n\n # track item (clip) main attributes\n self.ti_name = track_item.name()\n self.ti_index = int(track_item.eventNumber())\n\n # get track name and index\n track_name = track_item.parent().name()\n self.track_name = str(track_name).replace(\" \", \"_\")\n self.track_index = int(track_item.parent().trackIndex())\n\n # adding tag.family into tag\n if kwargs.get(\"avalon\"):\n self.tag_data.update(kwargs[\"avalon\"])\n\n # add publish attribute to tag data\n self.tag_data.update({\"publish\": True})\n\n # adding ui inputs if any\n self.ui_inputs = kwargs.get(\"ui_inputs\", {})\n\n # populate default data before we get other attributes\n self._populate_track_item_default_data()\n\n # use all populated default data to create all important attributes\n self._populate_attributes()\n\n # create parents with correct types\n self._create_parents()\n\n def convert(self):\n # solve track item data and add them to tag data\n tag_hierarchy_data = self._convert_to_tag_data()\n\n self.tag_data.update(tag_hierarchy_data)\n\n # if track name is in review track name and also if driving track name\n # is not in review track name: skip tag creation\n if (self.track_name in self.review_layer) and (\n self.driving_layer not in self.review_layer):\n return\n\n # deal with clip name\n new_name = self.tag_data.pop(\"newClipName\")\n\n if self.rename:\n # rename track item\n self.track_item.setName(new_name)\n self.tag_data[\"asset_name\"] = new_name\n else:\n self.tag_data[\"asset_name\"] = self.ti_name\n self.tag_data[\"hierarchyData\"][\"shot\"] = self.ti_name\n\n # AYON unique identifier\n folder_path = \"/{}/{}\".format(\n tag_hierarchy_data[\"hierarchy\"],\n self.tag_data[\"asset_name\"]\n )\n self.tag_data[\"folderPath\"] = folder_path\n if self.tag_data[\"heroTrack\"] and self.review_layer:\n self.tag_data.update({\"reviewTrack\": self.review_layer})\n else:\n self.tag_data.update({\"reviewTrack\": None})\n\n # TODO: remove debug print\n log.debug(\"___ self.tag_data: {}\".format(\n pformat(self.tag_data)\n ))\n\n # create pype tag on track_item and add data\n lib.imprint(self.track_item, self.tag_data)\n\n return self.track_item\n\n def _populate_track_item_default_data(self):\n \"\"\" Populate default formatting data from track item. \"\"\"\n\n self.track_item_default_data = {\n \"_folder_\": \"shots\",\n \"_sequence_\": self.sequence_name,\n \"_track_\": self.track_name,\n \"_clip_\": self.ti_name,\n \"_trackIndex_\": self.track_index,\n \"_clipIndex_\": self.ti_index\n }\n\n def _populate_attributes(self):\n \"\"\" Populate main object attributes. \"\"\"\n # track item frame range and parent track name for vertical sync check\n self.clip_in = int(self.track_item.timelineIn())\n self.clip_out = int(self.track_item.timelineOut())\n\n # define ui inputs if non gui mode was used\n self.shot_num = self.ti_index\n log.debug(\n \"____ self.shot_num: {}\".format(self.shot_num))\n\n # ui_inputs data or default values if gui was not used\n self.rename = self.ui_inputs.get(\n \"clipRename\", {}).get(\"value\") or self.rename_default\n self.clip_name = self.ui_inputs.get(\n \"clipName\", {}).get(\"value\") or self.clip_name_default\n self.hierarchy = self.ui_inputs.get(\n \"hierarchy\", {}).get(\"value\") or self.hierarchy_default\n self.hierarchy_data = self.ui_inputs.get(\n \"hierarchyData\", {}).get(\"value\") or \\\n self.track_item_default_data.copy()\n self.count_from = self.ui_inputs.get(\n \"countFrom\", {}).get(\"value\") or self.count_from_default\n self.count_steps = self.ui_inputs.get(\n \"countSteps\", {}).get(\"value\") or self.count_steps_default\n self.subset_name = self.ui_inputs.get(\n \"subsetName\", {}).get(\"value\") or self.subset_name_default\n self.subset_family = self.ui_inputs.get(\n \"subsetFamily\", {}).get(\"value\") or self.subset_family_default\n self.vertical_sync = self.ui_inputs.get(\n \"vSyncOn\", {}).get(\"value\") or self.vertical_sync_default\n self.driving_layer = self.ui_inputs.get(\n \"vSyncTrack\", {}).get(\"value\") or self.driving_layer_default\n self.review_track = self.ui_inputs.get(\n \"reviewTrack\", {}).get(\"value\") or self.review_track_default\n self.audio = self.ui_inputs.get(\n \"audio\", {}).get(\"value\") or False\n\n # build subset name from layer name\n if self.subset_name == \"\":\n self.subset_name = self.track_name\n\n # create subset for publishing\n self.subset = self.subset_family + self.subset_name.capitalize()\n\n def _replace_hash_to_expression(self, name, text):\n \"\"\" Replace hash with number in correct padding. \"\"\"\n _spl = text.split(\"#\")\n _len = (len(_spl) - 1)\n _repl = \"{{{0}:0>{1}}}\".format(name, _len)\n return text.replace((\"#\" * _len), _repl)\n\n\n def _convert_to_tag_data(self):\n \"\"\" Convert internal data to tag data.\n\n Populating the tag data into internal variable self.tag_data\n \"\"\"\n # define vertical sync attributes\n hero_track = True\n self.review_layer = \"\"\n if self.vertical_sync:\n # check if track name is not in driving layer\n if self.track_name not in self.driving_layer:\n # if it is not then define vertical sync as None\n hero_track = False\n\n # increasing steps by index of rename iteration\n self.count_steps *= self.rename_index\n\n hierarchy_formatting_data = {}\n hierarchy_data = deepcopy(self.hierarchy_data)\n _data = self.track_item_default_data.copy()\n if self.ui_inputs:\n # adding tag metadata from ui\n for _k, _v in self.ui_inputs.items():\n if _v[\"target\"] == \"tag\":\n self.tag_data[_k] = _v[\"value\"]\n\n # driving layer is set as positive match\n if hero_track or self.vertical_sync:\n # mark review layer\n if self.review_track and (\n self.review_track not in self.review_track_default):\n # if review layer is defined and not the same as default\n self.review_layer = self.review_track\n # shot num calculate\n if self.rename_index == 0:\n self.shot_num = self.count_from\n else:\n self.shot_num = self.count_from + self.count_steps\n\n # clip name sequence number\n _data.update({\"shot\": self.shot_num})\n\n # solve # in test to pythonic expression\n for _k, _v in hierarchy_data.items():\n if \"#\" not in _v[\"value\"]:\n continue\n hierarchy_data[\n _k][\"value\"] = self._replace_hash_to_expression(\n _k, _v[\"value\"])\n\n # fill up pythonic expresisons in hierarchy data\n for k, _v in hierarchy_data.items():\n hierarchy_formatting_data[k] = _v[\"value\"].format(**_data)\n else:\n # if no gui mode then just pass default data\n hierarchy_formatting_data = hierarchy_data\n\n tag_hierarchy_data = self._solve_tag_hierarchy_data(\n hierarchy_formatting_data\n )\n\n tag_hierarchy_data.update({\"heroTrack\": True})\n if hero_track and self.vertical_sync:\n self.vertical_clip_match.update({\n (self.clip_in, self.clip_out): tag_hierarchy_data\n })\n\n if not hero_track and self.vertical_sync:\n # driving layer is set as negative match\n for (_in, _out), hero_data in self.vertical_clip_match.items():\n hero_data.update({\"heroTrack\": False})\n if _in == self.clip_in and _out == self.clip_out:\n data_subset = hero_data[\"subset\"]\n # add track index in case duplicity of names in hero data\n if self.subset in data_subset:\n hero_data[\"subset\"] = self.subset + str(\n self.track_index)\n # in case track name and subset name is the same then add\n if self.subset_name == self.track_name:\n hero_data[\"subset\"] = self.subset\n # assign data to return hierarchy data to tag\n tag_hierarchy_data = hero_data\n\n # add data to return data dict\n return tag_hierarchy_data\n\n def _solve_tag_hierarchy_data(self, hierarchy_formatting_data):\n \"\"\" Solve tag data from hierarchy data and templates. \"\"\"\n # fill up clip name and hierarchy keys\n hierarchy_filled = self.hierarchy.format(**hierarchy_formatting_data)\n clip_name_filled = self.clip_name.format(**hierarchy_formatting_data)\n\n # remove shot from hierarchy data: is not needed anymore\n hierarchy_formatting_data.pop(\"shot\")\n\n return {\n \"newClipName\": clip_name_filled,\n \"hierarchy\": hierarchy_filled,\n \"parents\": self.parents,\n \"hierarchyData\": hierarchy_formatting_data,\n \"subset\": self.subset,\n \"family\": self.subset_family,\n \"families\": [self.data[\"family\"]]\n }\n\n def _convert_to_entity(self, type, template):\n \"\"\" Converting input key to key with type. \"\"\"\n # convert to entity type\n entity_type = self.types.get(type, None)\n\n assert entity_type, \"Missing entity type for `{}`\".format(\n type\n )\n\n # first collect formatting data to use for formatting template\n formatting_data = {}\n for _k, _v in self.hierarchy_data.items():\n value = _v[\"value\"].format(\n **self.track_item_default_data)\n formatting_data[_k] = value\n\n return {\n \"entity_type\": entity_type,\n \"entity_name\": template.format(\n **formatting_data\n )\n }\n\n def _create_parents(self):\n \"\"\" Create parents and return it in list. \"\"\"\n self.parents = []\n\n pattern = re.compile(self.parents_search_pattern)\n\n par_split = [(pattern.findall(t).pop(), t)\n for t in self.hierarchy.split(\"/\")]\n\n for type, template in par_split:\n parent = self._convert_to_entity(type, template)\n self.parents.append(parent)\n"
},
{
"path": "openpype/hosts/hiero/api/startup/HieroPlayer/PlayerPresets.hrox",
"content": "\n\n\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n 2\n 70\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n 2\n 70\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n 2\n 70\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n 2\n 70\n \n \n 2\n 70\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n 2\n 70\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n 2\n 70\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n 2\n 70\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n 2\n 70\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n 2\n 50\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n 2\n 70\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n 2\n 70\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n 2\n 70\n \n \n 2\n 70\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n 2\n 70\n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n \n 2\n 70\n \n \n 2\n 70\n \n \n 2\n 70\n 17\n \n \n 126935040\n 70\n -1\n \n \n 2\n 70\n 2\n \n \n \n \n \n\n"
},
{
"path": "openpype/hosts/hiero/api/startup/Python/Startup/SpreadsheetExport.py",
"content": "# This action adds itself to the Spreadsheet View context menu allowing the contents of the Spreadsheet be exported as a CSV file.\n# Usage: Right-click in Spreadsheet > \"Export as .CSV\"\n# Note: This only prints the text data that is visible in the active Spreadsheet View.\n# If you've filtered text, only the visible text will be printed to the CSV file\n# Usage: Copy to ~/.hiero/Python/StartupUI\nimport hiero.core.events\nimport hiero.ui\nimport os, csv\ntry:\n from PySide.QtGui import *\n from PySide.QtCore import *\nexcept:\n from PySide2.QtGui import *\n from PySide2.QtWidgets import *\n from PySide2.QtCore import *\n\n\n### Magic Widget Finding Methods - This stuff crawls all the PySide widgets, looking for an answer\ndef findWidget(w):\n global foundryWidgets\n if \"Foundry\" in w.metaObject().className():\n foundryWidgets += [w]\n\n for c in w.children():\n findWidget(c)\n return foundryWidgets\n\n\ndef getFoundryWidgetsWithClassName(filter=None):\n global foundryWidgets\n foundryWidgets = []\n widgets = []\n app = QApplication.instance()\n for w in app.topLevelWidgets():\n findWidget(w)\n\n filteredWidgets = foundryWidgets\n if filter:\n filteredWidgets = []\n for widget in foundryWidgets:\n if filter in widget.metaObject().className():\n filteredWidgets += [widget]\n return filteredWidgets\n\n\n# When right click, get the Sequence Name\ndef activeSpreadsheetTreeView():\n \"\"\"\n Does some PySide widget Magic to detect the Active Spreadsheet TreeView.\n \"\"\"\n spreadsheetViews = getFoundryWidgetsWithClassName(\n filter=\"SpreadsheetTreeView\")\n for spreadSheet in spreadsheetViews:\n if spreadSheet.hasFocus():\n activeSpreadSheet = spreadSheet\n return activeSpreadSheet\n return None\n\n\n#### Adds \"Export .CSV\" action to the Spreadsheet Context menu ####\nclass SpreadsheetExportCSVAction(QAction):\n def __init__(self):\n QAction.__init__(self, \"Export as .CSV\", None)\n self.triggered.connect(self.exportCSVFromActiveSpreadsheetView)\n hiero.core.events.registerInterest(\"kShowContextMenu/kSpreadsheet\",\n self.eventHandler)\n self.setIcon(QIcon(\"icons:FBGridView.png\"))\n\n def eventHandler(self, event):\n # Insert the action to the Export CSV menu\n event.menu.addAction(self)\n\n #### The guts!.. Writes a CSV file from a Sequence Object ####\n def exportCSVFromActiveSpreadsheetView(self):\n\n # Get the active QTreeView from the active Spreadsheet\n spreadsheetTreeView = activeSpreadsheetTreeView()\n\n if not spreadsheetTreeView:\n return \"Unable to detect the active TreeView.\"\n seq = hiero.ui.activeView().sequence()\n if not seq:\n print(\"Unable to detect the active Sequence from the activeView.\")\n return\n\n # The data model of the QTreeView\n model = spreadsheetTreeView.model()\n\n csvSavePath = os.path.join(QDir.homePath(), \"Desktop\",\n seq.name() + \".csv\")\n savePath, filter = QFileDialog.getSaveFileName(\n None,\n caption=\"Export Spreadsheet to .CSV as...\",\n dir=csvSavePath,\n filter=\"*.csv\")\n print(\"Saving To: {}\".format(savePath))\n\n # Saving was cancelled...\n if len(savePath) == 0:\n return\n\n # Get the Visible Header Columns from the QTreeView\n\n #csvHeader = [\"Event\", \"Status\", \"Shot Name\", \"Reel\", \"Track\", \"Speed\", \"Src In\", \"Src Out\",\"Src Duration\", \"Dst In\", \"Dst Out\", \"Dst Duration\", \"Clip\", \"Clip Media\"]\n\n # Get a CSV writer object\n f = open(savePath, \"w\")\n csvWriter = csv.writer(\n f, delimiter=',', quotechar=\"|\", quoting=csv.QUOTE_MINIMAL)\n\n # This is a list of the Column titles\n csvHeader = []\n\n for col in range(0, model.columnCount()):\n if not spreadsheetTreeView.isColumnHidden(col):\n csvHeader += [model.headerData(col, Qt.Horizontal)]\n\n # Write the Header row to the CSV file\n csvWriter.writerow(csvHeader)\n\n # Go through each row/column and print\n for row in range(model.rowCount()):\n row_data = []\n for col in range(model.columnCount()):\n if not spreadsheetTreeView.isColumnHidden(col):\n row_data.append(\n model.index(row, col, QModelIndex()).data(\n Qt.DisplayRole))\n\n # Write row to CSV file...\n csvWriter.writerow(row_data)\n\n f.close()\n # Conveniently show the CSV file in the native file browser...\n QDesktopServices.openUrl(\n QUrl('file:///%s' % (os.path.dirname(savePath))))\n\n\n# Add the action...\ncsvActions = SpreadsheetExportCSVAction()\n"
},
{
"path": "openpype/hosts/hiero/api/startup/Python/Startup/Startup.py",
"content": "import traceback\n\n# activate hiero from pype\nfrom openpype.pipeline import install_host\nimport openpype.hosts.hiero.api as phiero\ninstall_host(phiero)\n\ntry:\n __import__(\"openpype.hosts.hiero.api\")\n __import__(\"pyblish\")\n\nexcept ImportError as e:\n print(traceback.format_exc())\n print(\"pyblish: Could not load integration: %s \" % e)\n\nelse:\n # Setup integration\n import openpype.hosts.hiero.api as phiero\n phiero.lib.setup()\n"
},
{
"path": "openpype/hosts/hiero/api/startup/Python/Startup/otioexporter/OTIOExportTask.py",
"content": "# -*- coding: utf-8 -*-\n\n__author__ = \"Daniel Flehner Heen\"\n__credits__ = [\"Jakub Jezek\", \"Daniel Flehner Heen\"]\n\nimport os\nimport hiero.core\nfrom hiero.core import util\n\nimport opentimelineio as otio\nfrom openpype.hosts.hiero.api.otio import hiero_export\n\nclass OTIOExportTask(hiero.core.TaskBase):\n\n def __init__(self, initDict):\n \"\"\"Initialize\"\"\"\n hiero.core.TaskBase.__init__(self, initDict)\n self.otio_timeline = None\n\n def name(self):\n return str(type(self))\n\n def startTask(self):\n self.otio_timeline = hiero_export.create_otio_timeline()\n\n def taskStep(self):\n return False\n\n def finishTask(self):\n try:\n exportPath = self.resolvedExportPath()\n\n # Check file extension\n if not exportPath.lower().endswith(\".otio\"):\n exportPath += \".otio\"\n\n # check export root exists\n dirname = os.path.dirname(exportPath)\n util.filesystem.makeDirs(dirname)\n\n # write otio file\n hiero_export.write_to_file(self.otio_timeline, exportPath)\n\n # Catch all exceptions and log error\n except Exception as e:\n self.setError(\"failed to write file {f}\\n{e}\".format(\n f=exportPath,\n e=e)\n )\n\n hiero.core.TaskBase.finishTask(self)\n\n def forcedAbort(self):\n pass\n\n\nclass OTIOExportPreset(hiero.core.TaskPresetBase):\n def __init__(self, name, properties):\n \"\"\"Initialise presets to default values\"\"\"\n hiero.core.TaskPresetBase.__init__(self, OTIOExportTask, name)\n\n self.properties()[\"includeTags\"] = hiero_export.include_tags = True\n self.properties().update(properties)\n\n def supportedItems(self):\n return hiero.core.TaskPresetBase.kSequence\n\n def addCustomResolveEntries(self, resolver):\n resolver.addResolver(\n \"{ext}\",\n \"Extension of the file to be output\",\n lambda keyword, task: \"otio\"\n )\n\n def supportsAudio(self):\n return True\n\n\nhiero.core.taskRegistry.registerTask(OTIOExportPreset, OTIOExportTask)\n"
},
{
"path": "openpype/hosts/hiero/api/startup/Python/Startup/otioexporter/OTIOExportUI.py",
"content": "# -*- coding: utf-8 -*-\n\n__author__ = \"Daniel Flehner Heen\"\n__credits__ = [\"Jakub Jezek\", \"Daniel Flehner Heen\"]\n\nimport hiero.ui\nfrom .OTIOExportTask import (\n OTIOExportTask,\n OTIOExportPreset\n)\n\ntry:\n # Hiero >= 11.x\n from PySide2 import QtCore\n from PySide2.QtWidgets import QCheckBox\n from hiero.ui.FnTaskUIFormLayout import TaskUIFormLayout as FormLayout\n\nexcept ImportError:\n # Hiero <= 10.x\n from PySide import QtCore # lint:ok\n from PySide.QtGui import QCheckBox, QFormLayout # lint:ok\n\n FormLayout = QFormLayout # lint:ok\n\nfrom openpype.hosts.hiero.api.otio import hiero_export\n\nclass OTIOExportUI(hiero.ui.TaskUIBase):\n def __init__(self, preset):\n \"\"\"Initialize\"\"\"\n hiero.ui.TaskUIBase.__init__(\n self,\n OTIOExportTask,\n preset,\n \"OTIO Exporter\"\n )\n\n def includeMarkersCheckboxChanged(self, state):\n # Slot to handle change of checkbox state\n hiero_export.include_tags = state == QtCore.Qt.Checked\n\n def populateUI(self, widget, exportTemplate):\n layout = widget.layout()\n formLayout = FormLayout()\n\n # Hiero ~= 10.0v4\n if layout is None:\n layout = formLayout\n widget.setLayout(layout)\n\n else:\n layout.addLayout(formLayout)\n\n # Checkboxes for whether the OTIO should contain markers or not\n self.includeMarkersCheckbox = QCheckBox()\n self.includeMarkersCheckbox.setToolTip(\n \"Enable to include Tags as markers in the exported OTIO file.\"\n )\n self.includeMarkersCheckbox.setCheckState(QtCore.Qt.Unchecked)\n\n if self._preset.properties()[\"includeTags\"]:\n self.includeMarkersCheckbox.setCheckState(QtCore.Qt.Checked)\n\n self.includeMarkersCheckbox.stateChanged.connect(\n self.includeMarkersCheckboxChanged\n )\n\n # Add Checkbox to layout\n formLayout.addRow(\"Include Tags:\", self.includeMarkersCheckbox)\n\n\nhiero.ui.taskUIRegistry.registerTaskUI(\n OTIOExportPreset,\n OTIOExportUI\n)\n"
},
{
"path": "openpype/hosts/hiero/api/startup/Python/Startup/otioexporter/__init__.py",
"content": "from .OTIOExportTask import OTIOExportTask\nfrom .OTIOExportUI import OTIOExportUI\n\n__all__ = [\n \"OTIOExportTask\",\n \"OTIOExportUI\"\n]\n"
},
{
"path": "openpype/hosts/hiero/api/startup/Python/Startup/project_helpers.py",
"content": "try:\n from PySide.QtGui import *\n from PySide.QtCore import *\nexcept:\n from PySide2.QtGui import *\n from PySide2.QtWidgets import *\n from PySide2.QtCore import *\n\nfrom hiero.core.util import uniquify, version_get, version_set\nimport hiero.core\nimport hiero.ui\nimport nuke\n\n# A globally variable for storing the current Project\ngTrackedActiveProject = None\n\n# This selection handler will track changes in items selected/deselected in the Bin/Timeline/Spreadsheet Views\n\n\ndef __trackActiveProjectHandler(event):\n global gTrackedActiveProject\n selection = event.sender.selection()\n binSelection = selection\n if len(binSelection) > 0 and hasattr(binSelection[0], \"project\"):\n proj = binSelection[0].project()\n\n # We only store this if its a valid, active User Project\n if proj in hiero.core.projects(hiero.core.Project.kUserProjects):\n gTrackedActiveProject = proj\n\n\nhiero.core.events.registerInterest(\n \"kSelectionChanged/kBin\", __trackActiveProjectHandler)\nhiero.core.events.registerInterest(\n \"kSelectionChanged/kTimeline\", __trackActiveProjectHandler)\nhiero.core.events.registerInterest(\n \"kSelectionChanged/Spreadsheet\", __trackActiveProjectHandler)\n\n\ndef activeProject():\n \"\"\"hiero.ui.activeProject() -> returns the current Project\n\n Note: There is not technically a notion of a \"active\" Project in Hiero/NukeStudio, as it is a multi-project App.\n This method determines what is \"active\" by going down the following rules...\n\n # 1 - If the current Viewer (hiero.ui.currentViewer) contains a Clip or Sequence, this item is assumed to give the active Project\n # 2 - If nothing is currently in the Viewer, look to the active View, determine project from active selection\n # 3 - If no current selection can be determined, fall back to a globally tracked last selection from trackActiveProjectHandler\n # 4 - If all those rules fail, fall back to the last project in the list of hiero.core.projects()\n\n @return: hiero.core.Project\"\"\"\n global gTrackedActiveProject\n activeProject = None\n\n # Case 1 : Look for what the current Viewr tells us - this might not be what we want, and relies on hiero.ui.currentViewer() being robust.\n cv = hiero.ui.currentViewer().player().sequence()\n if hasattr(cv, \"project\"):\n activeProject = cv.project()\n else:\n # Case 2: We can't determine a project from the current Viewer, so try seeing what's selected in the activeView\n # Note that currently, if you run activeProject from the Script Editor, the activeView is always None, so this will rarely get used!\n activeView = hiero.ui.activeView()\n if activeView:\n # We can determine an active View.. see what's being worked with\n selection = activeView.selection()\n\n # Handle the case where nothing is selected in the active view\n if len(selection) == 0:\n # It's possible that there is no selection in a Timeline/Spreadsheet, but these views have \"sequence\" method, so try that...\n if isinstance(hiero.ui.activeView(), (hiero.ui.TimelineEditor, hiero.ui.SpreadsheetView)):\n activeSequence = activeView.sequence()\n if hasattr(currentItem, \"project\"):\n activeProject = activeSequence.project()\n\n # The active view has a selection... assume that the first item in the selection has the active Project\n else:\n currentItem = selection[0]\n if hasattr(currentItem, \"project\"):\n activeProject = currentItem.project()\n\n # Finally, Cases 3 and 4...\n if not activeProject:\n activeProjects = hiero.core.projects(hiero.core.Project.kUserProjects)\n if gTrackedActiveProject in activeProjects:\n activeProject = gTrackedActiveProject\n else:\n activeProject = activeProjects[-1]\n\n return activeProject\n\n# Method to get all recent projects\n\n\ndef recentProjects():\n \"\"\"hiero.core.recentProjects() -> Returns a list of paths to recently opened projects\n\n Hiero stores up to 5 recent projects in uistate.ini with the [recentFile]/# key.\n\n @return: list of paths to .hrox Projects\"\"\"\n\n appSettings = hiero.core.ApplicationSettings()\n recentProjects = []\n for i in range(0, 5):\n proj = appSettings.value('recentFile/%i' % i)\n if len(proj) > 0:\n recentProjects.append(proj)\n return recentProjects\n\n# Method to get recent project by index\n\n\ndef recentProject(k=0):\n \"\"\"hiero.core.recentProject(k) -> Returns the recent project path, specified by integer k (0-4)\n\n @param: k (optional, default = 0) - an integer from 0-4, relating to the index of recent projects.\n\n @return: hiero.core.Project\"\"\"\n\n appSettings = hiero.core.ApplicationSettings()\n proj = appSettings.value('recentFile/%i' % int(k), None)\n return proj\n\n# Method to get open project by index\n\n\ndef openRecentProject(k=0):\n \"\"\"hiero.core.openRecentProject(k) -> Opens the most the recent project as listed in the Open Recent list.\n\n @param: k (optional, default = 0) - an integer from 0-4, relating to the index of recent projects.\n @return: hiero.core.Project\"\"\"\n\n appSettings = hiero.core.ApplicationSettings()\n proj = appSettings.value('recentFile/%i' % int(k), None)\n proj = hiero.core.openProject(proj)\n return proj\n\n\n# Duck punch these methods into the relevant ui/core namespaces\nhiero.ui.activeProject = activeProject\nhiero.core.recentProjects = recentProjects\nhiero.core.recentProject = recentProject\nhiero.core.openRecentProject = openRecentProject\n\n\n# Method to Save a new Version of the activeHrox Project\nclass SaveAllProjects(QAction):\n\n def __init__(self):\n QAction.__init__(self, \"Save All Projects\", None)\n self.triggered.connect(self.projectSaveAll)\n hiero.core.events.registerInterest(\n \"kShowContextMenu/kBin\", self.eventHandler)\n\n def projectSaveAll(self):\n allProjects = hiero.core.projects()\n for proj in allProjects:\n try:\n proj.save()\n print(\"Saved Project: {} to: {} \".format(\n proj.name(), proj.path()\n ))\n except:\n print((\n \"Unable to save Project: {} to: {}. \"\n \"Check file permissions.\").format(\n proj.name(), proj.path()))\n\n def eventHandler(self, event):\n event.menu.addAction(self)\n\n# For projects with v# in the path name, saves out a new Project with v#+1\n\n\nclass SaveNewProjectVersion(QAction):\n\n def __init__(self):\n QAction.__init__(self, \"Save New Version...\", None)\n self.triggered.connect(self.saveNewVersion)\n hiero.core.events.registerInterest(\n \"kShowContextMenu/kBin\", self.eventHandler)\n self.selectedProjects = []\n\n def saveNewVersion(self):\n if len(self.selectedProjects) > 0:\n projects = self.selectedProjects\n else:\n projects = [hiero.ui.activeProject()]\n\n if len(projects) < 1:\n return\n\n for proj in projects:\n oldName = proj.name()\n path = proj.path()\n v = None\n prefix = None\n try:\n (prefix, v) = version_get(path, \"v\")\n except ValueError as msg:\n print(msg)\n\n if (prefix is not None) and (v is not None):\n v = int(v)\n newPath = version_set(path, prefix, v, v + 1)\n try:\n proj.saveAs(newPath)\n print(\"Saved new project version: {} to: {} \".format(\n oldName, newPath))\n except:\n print((\n \"Unable to save Project: {}. Check file permissions.\"\n ).format(oldName))\n else:\n newPath = path.replace(\".hrox\", \"_v01.hrox\")\n answer = nuke.ask(\n \"%s does not contain a version number.\\nDo you want to save as %s?\" % (proj, newPath))\n if answer:\n try:\n proj.saveAs(newPath)\n print(\"Saved new project version: {} to: {} \".format(\n oldName, newPath))\n except:\n print((\n \"Unable to save Project: {}. Check file \"\n \"permissions.\").format(oldName))\n\n def eventHandler(self, event):\n self.selectedProjects = []\n if hasattr(event.sender, \"selection\") and event.sender.selection() is not None and len(event.sender.selection()) != 0:\n selection = event.sender.selection()\n self.selectedProjects = uniquify(\n [item.project() for item in selection])\n event.menu.addAction(self)\n\n\n# Instantiate the actions\nsaveAllAct = SaveAllProjects()\nsaveNewAct = SaveNewProjectVersion()\n\nfileMenu = hiero.ui.findMenuAction(\"foundry.menu.file\")\nimportAct = hiero.ui.findMenuAction(\"foundry.project.importFiles\")\nhiero.ui.insertMenuAction(saveNewAct, fileMenu.menu(),\n before=\"Import File(s)...\")\nhiero.ui.insertMenuAction(saveAllAct, fileMenu.menu(),\n before=\"Import File(s)...\")\nfileMenu.menu().insertSeparator(importAct)\n"
},
{
"path": "openpype/hosts/hiero/api/startup/Python/Startup/selection_tracker.py",
"content": "\"\"\"Puts the selection project into \"hiero.selection\"\"\"\n\nimport hiero\n\n\ndef selectionChanged(event):\n hiero.selection = event.sender.selection()\n\nhiero.core.events.registerInterest(\"kSelectionChanged\", selectionChanged)\n"
},
{
"path": "openpype/hosts/hiero/api/startup/Python/Startup/setFrameRate.py",
"content": "# setFrameRate - adds a Right-click menu to the Project Bin view, allowing multiple BinItems (Clips/Sequences) to have their frame rates set.\n# Install in: ~/.hiero/Python/StartupUI\n# Requires 1.5v1 or later\n\nimport hiero.core\nimport hiero.ui\ntry:\n from PySide.QtGui import *\n from PySide.QtCore import *\nexcept:\n from PySide2.QtGui import *\n from PySide2.QtCore import *\n from PySide2.QtWidgets import *\n\n# Dialog for setting a Custom frame rate.\nclass SetFrameRateDialog(QDialog):\n\n def __init__(self,itemSelection=None,parent=None):\n super(SetFrameRateDialog, self).__init__(parent)\n self.setWindowTitle(\"Set Custom Frame Rate\")\n self.setSizePolicy( QSizePolicy.Expanding, QSizePolicy.Fixed )\n layout = QFormLayout()\n self._itemSelection = itemSelection\n\n self._frameRateField = QLineEdit()\n self._frameRateField.setToolTip(\"Enter custom frame rate here.\")\n self._frameRateField.setValidator(QDoubleValidator(1, 99, 3, self))\n self._frameRateField.textChanged.connect(self._textChanged)\n layout.addRow(\"Enter fps: \",self._frameRateField)\n\n # Standard buttons for Add/Cancel\n self._buttonbox = QDialogButtonBox(QDialogButtonBox.Ok | QDialogButtonBox.Cancel)\n self._buttonbox.accepted.connect(self.accept)\n self._buttonbox.rejected.connect(self.reject)\n self._buttonbox.button(QDialogButtonBox.Ok).setEnabled(False)\n layout.addRow(\"\",self._buttonbox)\n self.setLayout(layout)\n\n def _updateOkButtonState(self):\n # Cancel is always an option but only enable Ok if there is some text.\n currentFramerate = float(self.currentFramerateString())\n enableOk = False\n enableOk = ((currentFramerate > 0.0) and (currentFramerate <= 250.0))\n print(\"enabledOk\", enableOk)\n self._buttonbox.button(QDialogButtonBox.Ok).setEnabled(enableOk)\n\n def _textChanged(self, newText):\n self._updateOkButtonState()\n\n # Returns the current frame rate as a string\n def currentFramerateString(self):\n return str(self._frameRateField.text())\n\n # Presents the Dialog and sets the Frame rate from a selection\n def showDialogAndSetFrameRateFromSelection(self):\n\n if self._itemSelection is not None:\n if self.exec_():\n # For the Undo loop...\n\n # Construct an TimeBase object for setting the Frame Rate (fps)\n fps = hiero.core.TimeBase().fromString(self.currentFramerateString())\n\n\n # Set the frame rate for the selected BinItmes\n for item in self._itemSelection:\n item.setFramerate(fps)\n return\n\n# This is just a convenience method for returning QActions with a title, triggered method and icon.\ndef makeAction(title, method, icon = None):\n action = QAction(title,None)\n action.setIcon(QIcon(icon))\n\n # We do this magic, so that the title string from the action is used to set the frame rate!\n def methodWrapper():\n method(title)\n\n action.triggered.connect( methodWrapper )\n return action\n\n# Menu which adds a Set Frame Rate Menu to Project Bin view\nclass SetFrameRateMenu:\n\n def __init__(self):\n self._frameRateMenu = None\n self._frameRatesDialog = None\n\n\n # ant: Could use hiero.core.defaultFrameRates() here but messes up with string matching because we seem to mix decimal points\n self.frameRates = [\"8\",\"12\",\"12.50\",\"15\",\"23.98\",\"24\",\"25\",\"29.97\",\"30\",\"48\",\"50\",\"59.94\",\"60\"]\n hiero.core.events.registerInterest(\"kShowContextMenu/kBin\", self.binViewEventHandler)\n\n self.menuActions = []\n\n def createFrameRateMenus(self,selection):\n selectedClipFPS = [str(bi.activeItem().framerate()) for bi in selection if (isinstance(bi,hiero.core.BinItem) and hasattr(bi,\"activeItem\"))]\n selectedClipFPS = hiero.core.util.uniquify(selectedClipFPS)\n sameFrameRate = len(selectedClipFPS)==1\n self.menuActions = []\n for fps in self.frameRates:\n if fps in selectedClipFPS:\n if sameFrameRate:\n self.menuActions+=[makeAction(fps,self.setFrameRateFromMenuSelection, icon=\"icons:Ticked.png\")]\n else:\n self.menuActions+=[makeAction(fps,self.setFrameRateFromMenuSelection, icon=\"icons:remove active.png\")]\n else:\n self.menuActions+=[makeAction(fps,self.setFrameRateFromMenuSelection, icon=None)]\n\n # Now add Custom... menu\n self.menuActions += [makeAction(\n \"Custom...\", self.setFrameRateFromMenuSelection, icon=None)\n ]\n\n frameRateMenu = QMenu(\"Set Frame Rate\")\n for a in self.menuActions:\n frameRateMenu.addAction(a)\n\n return frameRateMenu\n\n def setFrameRateFromMenuSelection(self, menuSelectionFPS):\n\n selectedBinItems = [bi.activeItem() for bi in self._selection if (isinstance(bi,hiero.core.BinItem) and hasattr(bi,\"activeItem\"))]\n currentProject = selectedBinItems[0].project()\n\n with currentProject.beginUndo(\"Set Frame Rate\"):\n if menuSelectionFPS == \"Custom...\":\n self._frameRatesDialog = SetFrameRateDialog(itemSelection = selectedBinItems )\n self._frameRatesDialog.showDialogAndSetFrameRateFromSelection()\n\n else:\n for b in selectedBinItems:\n b.setFramerate(hiero.core.TimeBase().fromString(menuSelectionFPS))\n\n return\n\n # This handles events from the Project Bin View\n def binViewEventHandler(self,event):\n if not hasattr(event.sender, \"selection\"):\n # Something has gone wrong, we should only be here if raised\n # by the Bin view which gives a selection.\n return\n\n # Reset the selection to None...\n self._selection = None\n s = event.sender.selection()\n\n # Return if there's no Selection. We won't add the Menu.\n if s == None:\n return\n # Filter the selection to BinItems\n self._selection = [item for item in s if isinstance(item, hiero.core.BinItem)]\n if len(self._selection)==0:\n return\n # Creating the menu based on items selected, to highlight which frame rates are contained\n\n self._frameRateMenu = self.createFrameRateMenus(self._selection)\n\n # Insert the Set Frame Rate Button before the Set Media Colour Transform Action\n for action in event.menu.actions():\n if str(action.text()) == \"Set Media Colour Transform\":\n event.menu.insertMenu(action, self._frameRateMenu)\n break\n\n# Instantiate the Menu to get it to register itself.\nSetFrameRateMenu = SetFrameRateMenu()"
},
{
"path": "openpype/hosts/hiero/api/startup/Python/StartupUI/PimpMySpreadsheet.py",
"content": "# PimpMySpreadsheet 1.0, Antony Nasce, 23/05/13.\n# Adds custom spreadsheet columns and right-click menu for setting the Shot Status, and Artist Shot Assignment.\n# gStatusTags is a global dictionary of key(status)-value(icon) pairs, which can be overridden with custom icons if required\n# Requires Hiero 1.7v2 or later.\n# Install Instructions: Copy to ~/.hiero/Python/StartupUI\n\nimport hiero.core\nimport hiero.ui\n\ntry:\n from PySide.QtGui import *\n from PySide.QtCore import *\nexcept:\n from PySide2.QtGui import *\n from PySide2.QtWidgets import *\n from PySide2.QtCore import *\n\n# Set to True, if you wat \"Set Status\" right-click menu, False if not\nkAddStatusMenu = True\n\n# Set to True, if you wat \"Assign Artist\" right-click menu, False if not\nkAssignArtistMenu = True\n\n# Global list of Artist Name Dictionaries\n# Note: Override this to add different names, icons, department, IDs.\ngArtistList = [{\n \"artistName\": \"John Smith\",\n \"artistIcon\": \"icons:TagActor.png\",\n \"artistDepartment\": \"3D\",\n \"artistID\": 0\n}, {\n \"artistName\": \"Savlvador Dali\",\n \"artistIcon\": \"icons:TagActor.png\",\n \"artistDepartment\": \"Roto\",\n \"artistID\": 1\n}, {\n \"artistName\": \"Leonardo Da Vinci\",\n \"artistIcon\": \"icons:TagActor.png\",\n \"artistDepartment\": \"Paint\",\n \"artistID\": 2\n}, {\n \"artistName\": \"Claude Monet\",\n \"artistIcon\": \"icons:TagActor.png\",\n \"artistDepartment\": \"Comp\",\n \"artistID\": 3\n}, {\n \"artistName\": \"Pablo Picasso\",\n \"artistIcon\": \"icons:TagActor.png\",\n \"artistDepartment\": \"Animation\",\n \"artistID\": 4\n}]\n\n# Global Dictionary of Status Tags.\n# Note: This can be overwritten if you want to add a new status cellType or custom icon\n# Override the gStatusTags dictionary by adding your own \"Status\":\"Icon.png\" key-value pairs.\n# Add new custom keys like so: gStatusTags[\"For Client\"] = \"forClient.png\"\ngStatusTags = {\n \"Approved\": \"icons:status/TagApproved.png\",\n \"Unapproved\": \"icons:status/TagUnapproved.png\",\n \"Ready To Start\": \"icons:status/TagReadyToStart.png\",\n \"Blocked\": \"icons:status/TagBlocked.png\",\n \"On Hold\": \"icons:status/TagOnHold.png\",\n \"In Progress\": \"icons:status/TagInProgress.png\",\n \"Awaiting Approval\": \"icons:status/TagAwaitingApproval.png\",\n \"Omitted\": \"icons:status/TagOmitted.png\",\n \"Final\": \"icons:status/TagFinal.png\"\n}\n\n\n# The Custom Spreadsheet Columns\nclass CustomSpreadsheetColumns(QObject):\n \"\"\"\n A class defining custom columns for Hiero's spreadsheet view. This has a similar, but\n slightly simplified, interface to the QAbstractItemModel and QItemDelegate classes.\n \"\"\"\n global gStatusTags\n global gArtistList\n\n # Ideally, we'd set this list on a Per Item basis, but this is expensive for a large mixed selection\n standardColourSpaces = [\n \"linear\", \"sRGB\", \"rec709\", \"Cineon\", \"Gamma1.8\", \"Gamma2.2\",\n \"Panalog\", \"REDLog\", \"ViperLog\"\n ]\n arriColourSpaces = [\n \"Video - Rec709\", \"LogC - Camera Native\", \"Video - P3\", \"ACES\",\n \"LogC - Film\", \"LogC - Wide Gamut\"\n ]\n r3dColourSpaces = [\n \"Linear\", \"Rec709\", \"REDspace\", \"REDlog\", \"PDlog685\", \"PDlog985\",\n \"CustomPDlog\", \"REDgamma\", \"SRGB\", \"REDlogFilm\", \"REDgamma2\",\n \"REDgamma3\"\n ]\n gColourSpaces = standardColourSpaces + arriColourSpaces + r3dColourSpaces\n\n currentView = hiero.ui.activeView()\n\n # This is the list of Columns available\n gCustomColumnList = [\n {\n \"name\": \"Tags\",\n \"cellType\": \"readonly\"\n },\n {\n \"name\": \"Colourspace\",\n \"cellType\": \"dropdown\"\n },\n {\n \"name\": \"Notes\",\n \"cellType\": \"readonly\"\n },\n {\n \"name\": \"FileType\",\n \"cellType\": \"readonly\"\n },\n {\n \"name\": \"Shot Status\",\n \"cellType\": \"dropdown\"\n },\n {\n \"name\": \"Thumbnail\",\n \"cellType\": \"readonly\"\n },\n {\n \"name\": \"MediaType\",\n \"cellType\": \"readonly\"\n },\n {\n \"name\": \"Width\",\n \"cellType\": \"readonly\"\n },\n {\n \"name\": \"Height\",\n \"cellType\": \"readonly\"\n },\n {\n \"name\": \"Pixel Aspect\",\n \"cellType\": \"readonly\"\n },\n {\n \"name\": \"Artist\",\n \"cellType\": \"dropdown\"\n },\n {\n \"name\": \"Department\",\n \"cellType\": \"readonly\"\n },\n ]\n\n def numColumns(self):\n \"\"\"\n Return the number of custom columns in the spreadsheet view\n \"\"\"\n return len(self.gCustomColumnList)\n\n def columnName(self, column):\n \"\"\"\n Return the name of a custom column\n \"\"\"\n return self.gCustomColumnList[column][\"name\"]\n\n def getTagsString(self, item):\n \"\"\"\n Convenience method for returning all the Notes in a Tag as a string\n \"\"\"\n tagNames = []\n tags = item.tags()\n for tag in tags:\n tagNames += [tag.name()]\n tagNameString = ','.join(tagNames)\n return tagNameString\n\n def getNotes(self, item):\n \"\"\"\n Convenience method for returning all the Notes in a Tag as a string\n \"\"\"\n notes = \"\"\n tags = item.tags()\n for tag in tags:\n note = tag.note()\n if len(note) > 0:\n notes += tag.note() + ', '\n return notes[:-2]\n\n def getData(self, row, column, item):\n \"\"\"\n Return the data in a cell\n \"\"\"\n currentColumn = self.gCustomColumnList[column]\n if currentColumn[\"name\"] == \"Tags\":\n return self.getTagsString(item)\n\n if currentColumn[\"name\"] == \"Colourspace\":\n try:\n colTransform = item.sourceMediaColourTransform()\n except:\n colTransform = \"--\"\n return colTransform\n\n if currentColumn[\"name\"] == \"Notes\":\n try:\n note = self.getNotes(item)\n except:\n note = \"\"\n return note\n\n if currentColumn[\"name\"] == \"FileType\":\n fileType = \"--\"\n M = item.source().mediaSource().metadata()\n if M.hasKey(\"foundry.source.type\"):\n fileType = M.value(\"foundry.source.type\")\n elif M.hasKey(\"media.input.filereader\"):\n fileType = M.value(\"media.input.filereader\")\n return fileType\n\n if currentColumn[\"name\"] == \"Shot Status\":\n status = item.status()\n if not status:\n status = \"--\"\n return str(status)\n\n if currentColumn[\"name\"] == \"MediaType\":\n M = item.mediaType()\n return str(M).split(\"MediaType\")[-1].replace(\".k\", \"\")\n\n if currentColumn[\"name\"] == \"Thumbnail\":\n return str(item.eventNumber())\n\n if currentColumn[\"name\"] == \"Width\":\n return str(item.source().format().width())\n\n if currentColumn[\"name\"] == \"Height\":\n return str(item.source().format().height())\n\n if currentColumn[\"name\"] == \"Pixel Aspect\":\n return str(item.source().format().pixelAspect())\n\n if currentColumn[\"name\"] == \"Artist\":\n if item.artist():\n name = item.artist()[\"artistName\"]\n return name\n else:\n return \"--\"\n\n if currentColumn[\"name\"] == \"Department\":\n if item.artist():\n dep = item.artist()[\"artistDepartment\"]\n return dep\n else:\n return \"--\"\n\n return \"\"\n\n def setData(self, row, column, item, data):\n \"\"\"\n Set the data in a cell - unused in this example\n \"\"\"\n\n return None\n\n def getTooltip(self, row, column, item):\n \"\"\"\n Return the tooltip for a cell\n \"\"\"\n currentColumn = self.gCustomColumnList[column]\n if currentColumn[\"name\"] == \"Tags\":\n return str([item.name() for item in item.tags()])\n\n if currentColumn[\"name\"] == \"Notes\":\n return str(self.getNotes(item))\n return \"\"\n\n def getFont(self, row, column, item):\n \"\"\"\n Return the tooltip for a cell\n \"\"\"\n return None\n\n def getBackground(self, row, column, item):\n \"\"\"\n Return the background colour for a cell\n \"\"\"\n if not item.source().mediaSource().isMediaPresent():\n return QColor(80, 20, 20)\n return None\n\n def getForeground(self, row, column, item):\n \"\"\"\n Return the text colour for a cell\n \"\"\"\n #if column == 1:\n # return QColor(255, 64, 64)\n return None\n\n def getIcon(self, row, column, item):\n \"\"\"\n Return the icon for a cell\n \"\"\"\n currentColumn = self.gCustomColumnList[column]\n if currentColumn[\"name\"] == \"Colourspace\":\n return QIcon(\"icons:LUT.png\")\n\n if currentColumn[\"name\"] == \"Shot Status\":\n status = item.status()\n if status:\n return QIcon(gStatusTags[status])\n\n if currentColumn[\"name\"] == \"MediaType\":\n mediaType = item.mediaType()\n if mediaType == hiero.core.TrackItem.kVideo:\n return QIcon(\"icons:VideoOnly.png\")\n elif mediaType == hiero.core.TrackItem.kAudio:\n return QIcon(\"icons:AudioOnly.png\")\n\n if currentColumn[\"name\"] == \"Artist\":\n try:\n return QIcon(item.artist()[\"artistIcon\"])\n except:\n return None\n return None\n\n def getSizeHint(self, row, column, item):\n \"\"\"\n Return the size hint for a cell\n \"\"\"\n currentColumnName = self.gCustomColumnList[column][\"name\"]\n\n if currentColumnName == \"Thumbnail\":\n return QSize(90, 50)\n\n return QSize(50, 50)\n\n def paintCell(self, row, column, item, painter, option):\n \"\"\"\n Paint a custom cell. Return True if the cell was painted, or False to continue\n with the default cell painting.\n \"\"\"\n currentColumn = self.gCustomColumnList[column]\n if currentColumn[\"name\"] == \"Tags\":\n if option.state & QStyle.State_Selected:\n painter.fillRect(option.rect, option.palette.highlight())\n iconSize = 20\n r = QRect(option.rect.x(),\n option.rect.y() + (option.rect.height() - iconSize) / 2,\n iconSize, iconSize)\n tags = item.tags()\n if len(tags) > 0:\n painter.save()\n painter.setClipRect(option.rect)\n for tag in item.tags():\n M = tag.metadata()\n if not (M.hasKey(\"tag.status\")\n or M.hasKey(\"tag.artistID\")):\n QIcon(tag.icon()).paint(painter, r, Qt.AlignLeft)\n r.translate(r.width() + 2, 0)\n painter.restore()\n return True\n\n if currentColumn[\"name\"] == \"Thumbnail\":\n imageView = None\n pen = QPen()\n r = QRect(option.rect.x() + 2, (option.rect.y() +\n (option.rect.height() - 46) / 2),\n 85, 46)\n if not item.source().mediaSource().isMediaPresent():\n imageView = QImage(\"icons:Offline.png\")\n pen.setColor(QColor(Qt.red))\n\n if item.mediaType() == hiero.core.TrackItem.MediaType.kAudio:\n imageView = QImage(\"icons:AudioOnly.png\")\n #pen.setColor(QColor(Qt.green))\n painter.fillRect(r, QColor(45, 59, 45))\n\n if option.state & QStyle.State_Selected:\n painter.fillRect(option.rect, option.palette.highlight())\n\n tags = item.tags()\n painter.save()\n painter.setClipRect(option.rect)\n\n if not imageView:\n try:\n imageView = item.thumbnail(item.sourceIn())\n pen.setColor(QColor(20, 20, 20))\n # If we're here, we probably have a TC error, no thumbnail, so get it from the source Clip...\n except:\n pen.setColor(QColor(Qt.red))\n\n if not imageView:\n try:\n imageView = item.source().thumbnail()\n pen.setColor(QColor(Qt.yellow))\n except:\n imageView = QImage(\"icons:Offline.png\")\n pen.setColor(QColor(Qt.red))\n\n QIcon(QPixmap.fromImage(imageView)).paint(painter, r,\n Qt.AlignCenter)\n painter.setPen(pen)\n painter.drawRoundedRect(r, 1, 1)\n painter.restore()\n return True\n\n return False\n\n def createEditor(self, row, column, item, view):\n \"\"\"\n Create an editing widget for a custom cell\n \"\"\"\n self.currentView = view\n\n currentColumn = self.gCustomColumnList[column]\n if currentColumn[\"cellType\"] == \"readonly\":\n cle = QLabel()\n cle.setEnabled(False)\n cle.setVisible(False)\n return cle\n\n if currentColumn[\"name\"] == \"Colourspace\":\n cb = QComboBox()\n for colourspace in self.gColourSpaces:\n cb.addItem(colourspace)\n cb.currentIndexChanged.connect(self.colourspaceChanged)\n return cb\n\n if currentColumn[\"name\"] == \"Shot Status\":\n cb = QComboBox()\n cb.addItem(\"\")\n for key in gStatusTags.keys():\n cb.addItem(QIcon(gStatusTags[key]), key)\n cb.addItem(\"--\")\n cb.currentIndexChanged.connect(self.statusChanged)\n\n return cb\n\n if currentColumn[\"name\"] == \"Artist\":\n cb = QComboBox()\n cb.addItem(\"\")\n for artist in gArtistList:\n cb.addItem(artist[\"artistName\"])\n cb.addItem(\"--\")\n cb.currentIndexChanged.connect(self.artistNameChanged)\n return cb\n return None\n\n def setModelData(self, row, column, item, editor):\n return False\n\n def dropMimeData(self, row, column, item, data, items):\n \"\"\"\n Handle a drag and drop operation - adds a Dragged Tag to the shot\n \"\"\"\n for thing in items:\n if isinstance(thing, hiero.core.Tag):\n item.addTag(thing)\n return None\n\n def colourspaceChanged(self, index):\n \"\"\"\n This method is called when Colourspace widget changes index.\n \"\"\"\n index = self.sender().currentIndex()\n colourspace = self.gColourSpaces[index]\n selection = self.currentView.selection()\n project = selection[0].project()\n with project.beginUndo(\"Set Colourspace\"):\n items = [\n item for item in selection\n if (item.mediaType() == hiero.core.TrackItem.MediaType.kVideo)\n ]\n for trackItem in items:\n trackItem.setSourceMediaColourTransform(colourspace)\n\n def statusChanged(self, arg):\n \"\"\"\n This method is called when Shot Status widget changes index.\n \"\"\"\n view = hiero.ui.activeView()\n selection = view.selection()\n status = self.sender().currentText()\n project = selection[0].project()\n with project.beginUndo(\"Set Status\"):\n # A string of \"--\" characters denotes clear the status\n if status != \"--\":\n for trackItem in selection:\n trackItem.setStatus(status)\n else:\n for trackItem in selection:\n tTags = trackItem.tags()\n for tag in tTags:\n if tag.metadata().hasKey(\"tag.status\"):\n trackItem.removeTag(tag)\n break\n\n def artistNameChanged(self, arg):\n \"\"\"\n This method is called when Artist widget changes index.\n \"\"\"\n view = hiero.ui.activeView()\n selection = view.selection()\n name = self.sender().currentText()\n project = selection[0].project()\n with project.beginUndo(\"Assign Artist\"):\n # A string of \"--\" denotes clear the assignee...\n if name != \"--\":\n for trackItem in selection:\n trackItem.setArtistByName(name)\n else:\n for trackItem in selection:\n tTags = trackItem.tags()\n for tag in tTags:\n if tag.metadata().hasKey(\"tag.artistID\"):\n trackItem.removeTag(tag)\n break\n\n\ndef _getArtistFromID(self, artistID):\n \"\"\" getArtistFromID -> returns an artist dictionary, by their given ID\"\"\"\n global gArtistList\n artist = [\n element for element in gArtistList\n if element[\"artistID\"] == int(artistID)\n ]\n if not artist:\n return None\n return artist[0]\n\n\ndef _getArtistFromName(self, artistName):\n \"\"\" getArtistFromID -> returns an artist dictionary, by their given ID \"\"\"\n global gArtistList\n artist = [\n element for element in gArtistList\n if element[\"artistName\"] == artistName\n ]\n if not artist:\n return None\n return artist[0]\n\n\ndef _artist(self):\n \"\"\"_artist -> Returns the artist dictionary assigned to this shot\"\"\"\n artist = None\n tags = self.tags()\n for tag in tags:\n if tag.metadata().hasKey(\"tag.artistID\"):\n artistID = tag.metadata().value(\"tag.artistID\")\n artist = self.getArtistFromID(artistID)\n return artist\n\n\ndef _updateArtistTag(self, artistDict):\n # A shot will only have one artist assigned. Check if one exists and set accordingly\n\n artistTag = None\n tags = self.tags()\n for tag in tags:\n if tag.metadata().hasKey(\"tag.artistID\"):\n artistTag = tag\n break\n\n if not artistTag:\n artistTag = hiero.core.Tag(\"Artist\")\n artistTag.setIcon(artistDict[\"artistIcon\"])\n artistTag.metadata().setValue(\"tag.artistID\",\n str(artistDict[\"artistID\"]))\n artistTag.metadata().setValue(\"tag.artistName\",\n str(artistDict[\"artistName\"]))\n artistTag.metadata().setValue(\"tag.artistDepartment\",\n str(artistDict[\"artistDepartment\"]))\n self.sequence().editFinished()\n self.addTag(artistTag)\n self.sequence().editFinished()\n return\n\n artistTag.setIcon(artistDict[\"artistIcon\"])\n artistTag.metadata().setValue(\"tag.artistID\", str(artistDict[\"artistID\"]))\n artistTag.metadata().setValue(\"tag.artistName\",\n str(artistDict[\"artistName\"]))\n artistTag.metadata().setValue(\"tag.artistDepartment\",\n str(artistDict[\"artistDepartment\"]))\n self.sequence().editFinished()\n return\n\n\ndef _setArtistByName(self, artistName):\n \"\"\" setArtistByName(artistName) -> sets the artist tag on a TrackItem by a given artistName string\"\"\"\n global gArtistList\n\n artist = self.getArtistFromName(artistName)\n if not artist:\n print((\n \"Artist name: {} was not found in \"\n \"the gArtistList.\").format(artistName))\n return\n\n # Do the update.\n self.updateArtistTag(artist)\n\n\ndef _setArtistByID(self, artistID):\n \"\"\" setArtistByID(artistID) -> sets the artist tag on a TrackItem by a given artistID integer\"\"\"\n global gArtistList\n\n artist = self.getArtistFromID(artistID)\n if not artist:\n print(\"Artist name: {} was not found in the gArtistList.\".format(\n artistID))\n return\n\n # Do the update.\n self.updateArtistTag(artist)\n\n\n# Inject status getter and setter methods into hiero.core.TrackItem\nhiero.core.TrackItem.artist = _artist\nhiero.core.TrackItem.setArtistByName = _setArtistByName\nhiero.core.TrackItem.setArtistByID = _setArtistByID\nhiero.core.TrackItem.getArtistFromName = _getArtistFromName\nhiero.core.TrackItem.getArtistFromID = _getArtistFromID\nhiero.core.TrackItem.updateArtistTag = _updateArtistTag\n\n\ndef _status(self):\n \"\"\"status -> Returns the Shot status. None if no Status is set.\"\"\"\n\n status = None\n tags = self.tags()\n for tag in tags:\n if tag.metadata().hasKey(\"tag.status\"):\n status = tag.metadata().value(\"tag.status\")\n return status\n\n\ndef _setStatus(self, status):\n \"\"\"setShotStatus(status) -> Method to set the Status of a Shot.\n Adds a special kind of status Tag to a TrackItem\n Example: myTrackItem.setStatus(\"Final\")\n\n @param status - a string, corresponding to the Status name\n \"\"\"\n global gStatusTags\n\n # Get a valid Tag object from the Global list of statuses\n if not status in gStatusTags.keys():\n print(\"Status requested was not a valid Status string.\")\n return\n\n # A shot should only have one status. Check if one exists and set accordingly\n statusTag = None\n tags = self.tags()\n for tag in tags:\n if tag.metadata().hasKey(\"tag.status\"):\n statusTag = tag\n break\n\n if not statusTag:\n statusTag = hiero.core.Tag(\"Status\")\n statusTag.setIcon(gStatusTags[status])\n statusTag.metadata().setValue(\"tag.status\", status)\n self.addTag(statusTag)\n\n statusTag.setIcon(gStatusTags[status])\n statusTag.metadata().setValue(\"tag.status\", status)\n\n self.sequence().editFinished()\n return\n\n\n# Inject status getter and setter methods into hiero.core.TrackItem\nhiero.core.TrackItem.setStatus = _setStatus\nhiero.core.TrackItem.status = _status\n\n\n# This is a convenience method for returning QActions with a triggered method based on the title string\ndef titleStringTriggeredAction(title, method, icon=None):\n action = QAction(title, None)\n action.setIcon(QIcon(icon))\n\n # We do this magic, so that the title string from the action is used to set the status\n def methodWrapper():\n method(title)\n\n action.triggered.connect(methodWrapper)\n return action\n\n\n# Menu which adds a Set Status Menu to Timeline and Spreadsheet Views\nclass SetStatusMenu(QMenu):\n def __init__(self):\n QMenu.__init__(self, \"Set Status\", None)\n\n global gStatusTags\n self.statuses = gStatusTags\n self._statusActions = self.createStatusMenuActions()\n\n # Add the Actions to the Menu.\n for act in self.menuActions:\n self.addAction(act)\n\n hiero.core.events.registerInterest(\"kShowContextMenu/kTimeline\",\n self.eventHandler)\n hiero.core.events.registerInterest(\"kShowContextMenu/kSpreadsheet\",\n self.eventHandler)\n\n def createStatusMenuActions(self):\n self.menuActions = []\n for status in self.statuses:\n self.menuActions += [\n titleStringTriggeredAction(\n status,\n self.setStatusFromMenuSelection,\n icon=gStatusTags[status])\n ]\n\n def setStatusFromMenuSelection(self, menuSelectionStatus):\n selectedShots = [\n item for item in self._selection\n if (isinstance(item, hiero.core.TrackItem))\n ]\n selectedTracks = [\n item for item in self._selection\n if (isinstance(item, (hiero.core.VideoTrack,\n hiero.core.AudioTrack)))\n ]\n\n # If we have a Track Header Selection, no shots could be selected, so create shotSelection list\n if len(selectedTracks) >= 1:\n for track in selectedTracks:\n selectedShots += [\n item for item in track.items()\n if (isinstance(item, hiero.core.TrackItem))\n ]\n\n # It's possible no shots exist on the Track, in which case nothing is required\n if len(selectedShots) == 0:\n return\n\n currentProject = selectedShots[0].project()\n\n with currentProject.beginUndo(\"Set Status\"):\n # Shots selected\n for shot in selectedShots:\n shot.setStatus(menuSelectionStatus)\n\n # This handles events from the Project Bin View\n def eventHandler(self, event):\n if not hasattr(event.sender, \"selection\"):\n # Something has gone wrong, we should only be here if raised\n # by the Timeline/Spreadsheet view which gives a selection.\n return\n\n # Set the current selection\n self._selection = event.sender.selection()\n\n # Return if there's no Selection. We won't add the Menu.\n if len(self._selection) == 0:\n return\n\n event.menu.addMenu(self)\n\n\n# Menu which adds a Set Status Menu to Timeline and Spreadsheet Views\nclass AssignArtistMenu(QMenu):\n def __init__(self):\n QMenu.__init__(self, \"Assign Artist\", None)\n\n global gArtistList\n self.artists = gArtistList\n self._artistsActions = self.createAssignArtistMenuActions()\n\n # Add the Actions to the Menu.\n for act in self.menuActions:\n self.addAction(act)\n\n hiero.core.events.registerInterest(\"kShowContextMenu/kTimeline\",\n self.eventHandler)\n hiero.core.events.registerInterest(\"kShowContextMenu/kSpreadsheet\",\n self.eventHandler)\n\n def createAssignArtistMenuActions(self):\n self.menuActions = []\n for artist in self.artists:\n self.menuActions += [\n titleStringTriggeredAction(\n artist[\"artistName\"],\n self.setArtistFromMenuSelection,\n icon=artist[\"artistIcon\"])\n ]\n\n def setArtistFromMenuSelection(self, menuSelectionArtist):\n selectedShots = [\n item for item in self._selection\n if (isinstance(item, hiero.core.TrackItem))\n ]\n selectedTracks = [\n item for item in self._selection\n if (isinstance(item, (hiero.core.VideoTrack,\n hiero.core.AudioTrack)))\n ]\n\n # If we have a Track Header Selection, no shots could be selected, so create shotSelection list\n if len(selectedTracks) >= 1:\n for track in selectedTracks:\n selectedShots += [\n item for item in track.items()\n if (isinstance(item, hiero.core.TrackItem))\n ]\n\n # It's possible no shots exist on the Track, in which case nothing is required\n if len(selectedShots) == 0:\n return\n\n currentProject = selectedShots[0].project()\n\n with currentProject.beginUndo(\"Assign Artist\"):\n # Shots selected\n for shot in selectedShots:\n shot.setArtistByName(menuSelectionArtist)\n\n # This handles events from the Project Bin View\n def eventHandler(self, event):\n if not hasattr(event.sender, \"selection\"):\n # Something has gone wrong, we should only be here if raised\n # by the Timeline/Spreadsheet view which gives a selection.\n return\n\n # Set the current selection\n self._selection = event.sender.selection()\n\n # Return if there's no Selection. We won't add the Menu.\n if len(self._selection) == 0:\n return\n\n event.menu.addMenu(self)\n\n\n# Add the \"Set Status\" context menu to Timeline and Spreadsheet\nif kAddStatusMenu:\n setStatusMenu = SetStatusMenu()\n\nif kAssignArtistMenu:\n assignArtistMenu = AssignArtistMenu()\n\n# Register our custom columns\nhiero.ui.customColumn = CustomSpreadsheetColumns()\n"
},
{
"path": "openpype/hosts/hiero/api/startup/Python/StartupUI/Purge.py",
"content": "# Purge Unused Clips - Removes any unused Clips from a Project\n# Usage: Copy to ~/.hiero/Python/StartupUI\n# Demonstrates the use of hiero.core.find_items module.\n# Usage: Right-click on an item in the Bin View > \"Purge Unused Clips\"\n# Result: Any Clips not used in a Sequence in the active project will be removed\n# Requires Hiero 1.5v1 or later.\n# Version 1.1\n\nimport hiero\nimport hiero.core.find_items\ntry:\n from PySide.QtGui import *\n from PySide.QtCore import *\nexcept:\n from PySide2.QtGui import *\n from PySide2.QtWidgets import *\n from PySide2.QtCore import *\n\n\nclass PurgeUnusedAction(QAction):\n def __init__(self):\n QAction.__init__(self, \"Purge Unused Clips\", None)\n self.triggered.connect(self.PurgeUnused)\n hiero.core.events.registerInterest(\"kShowContextMenu/kBin\",\n self.eventHandler)\n self.setIcon(QIcon(\"icons:TagDelete.png\"))\n\n # Method to return whether a Bin is empty...\n def binIsEmpty(self, b):\n numBinItems = 0\n bItems = b.items()\n empty = False\n\n if len(bItems) == 0:\n empty = True\n return empty\n else:\n for b in bItems:\n if isinstance(b, hiero.core.BinItem) or isinstance(\n b, hiero.core.Bin):\n numBinItems += 1\n if numBinItems == 0:\n empty = True\n\n return empty\n\n def PurgeUnused(self):\n\n #Get selected items\n item = self.selectedItem\n proj = item.project()\n\n # Build a list of Projects\n SEQS = hiero.core.findItems(proj, \"Sequences\")\n\n # Build a list of Clips\n CLIPSTOREMOVE = hiero.core.findItems(proj, \"Clips\")\n\n if len(SEQS) == 0:\n # Present Dialog Asking if User wants to remove Clips\n msgBox = QMessageBox()\n msgBox.setText(\"Purge Unused Clips\")\n msgBox.setInformativeText(\n \"You have no Sequences in this Project. Do you want to remove all Clips (%i) from Project: %s?\"\n % (len(CLIPSTOREMOVE), proj.name()))\n msgBox.setStandardButtons(QMessageBox.Ok | QMessageBox.Cancel)\n msgBox.setDefaultButton(QMessageBox.Ok)\n ret = msgBox.exec_()\n if ret == QMessageBox.Cancel:\n print(\"Not purging anything.\")\n elif ret == QMessageBox.Ok:\n with proj.beginUndo(\"Purge Unused Clips\"):\n BINS = []\n for clip in CLIPSTOREMOVE:\n BI = clip.binItem()\n B = BI.parentBin()\n BINS += [B]\n print(\"Removing: {}\".format(BI))\n try:\n B.removeItem(BI)\n except:\n print(\"Unable to remove: {}\".format(BI))\n return\n\n # For each sequence, iterate through each track Item, see if the Clip is in the CLIPS list.\n # Remaining items in CLIPS will be removed\n\n for seq in SEQS:\n\n #Loop through selected and make folders\n for track in seq:\n for trackitem in track:\n\n if trackitem.source() in CLIPSTOREMOVE:\n CLIPSTOREMOVE.remove(trackitem.source())\n\n # Present Dialog Asking if User wants to remove Clips\n msgBox = QMessageBox()\n msgBox.setText(\"Purge Unused Clips\")\n msgBox.setInformativeText(\"Remove %i unused Clips from Project %s?\" %\n (len(CLIPSTOREMOVE), proj.name()))\n msgBox.setStandardButtons(QMessageBox.Ok | QMessageBox.Cancel)\n msgBox.setDefaultButton(QMessageBox.Ok)\n ret = msgBox.exec_()\n\n if ret == QMessageBox.Cancel:\n print(\"Cancel\")\n return\n elif ret == QMessageBox.Ok:\n BINS = []\n with proj.beginUndo(\"Purge Unused Clips\"):\n # Delete the rest of the Clips\n for clip in CLIPSTOREMOVE:\n BI = clip.binItem()\n B = BI.parentBin()\n BINS += [B]\n print(\"Removing: {}\".format(BI))\n try:\n B.removeItem(BI)\n except:\n print(\"Unable to remove: {}\".format(BI))\n\n def eventHandler(self, event):\n if not hasattr(event.sender, \"selection\"):\n # Something has gone wrong, we shouldn't only be here if raised\n # by the Bin view which will give a selection.\n return\n\n self.selectedItem = None\n s = event.sender.selection()\n\n if len(s) >= 1:\n self.selectedItem = s[0]\n title = \"Purge Unused Clips\"\n self.setText(title)\n event.menu.addAction(self)\n\n return\n\n\n# Instantiate the action to get it to register itself.\nPurgeUnusedAction = PurgeUnusedAction()\n"
},
{
"path": "openpype/hosts/hiero/api/startup/Python/StartupUI/nukeStyleKeyboardShortcuts.py",
"content": "# nukeStyleKeyboardShortcuts, v1, 30/07/2012, Ant Nasce.\n# A few Nuke-Style File menu shortcuts for those whose muscle memory has set in...\n# Usage: Copy this file to ~/.hiero/Python/StartupUI/\n\nimport hiero.ui\ntry:\n from PySide.QtGui import *\n from PySide.QtCore import *\nexcept:\n from PySide2.QtGui import *\n from PySide2.QtWidgets import *\n from PySide2.QtCore import *\n\n#----------------------------------------------\na = hiero.ui.findMenuAction('Import File(s)...')\n# Note: You probably best to make this 'Ctrl+R' - currently conflicts with \"Red\" in the Viewer!\na.setShortcut(QKeySequence(\"R\"))\n#----------------------------------------------\na = hiero.ui.findMenuAction('Import Folder(s)...')\na.setShortcut(QKeySequence('Shift+R'))\n#----------------------------------------------\na = hiero.ui.findMenuAction(\"Import EDL/XML/AAF...\")\na.setShortcut(QKeySequence('Ctrl+Shift+O'))\n#----------------------------------------------\na = hiero.ui.findMenuAction(\"Metadata View\")\na.setShortcut(QKeySequence(\"I\"))\n#----------------------------------------------\na = hiero.ui.findMenuAction(\"Edit Settings\")\na.setShortcut(QKeySequence(\"S\"))\n#----------------------------------------------\na = hiero.ui.findMenuAction(\"Monitor Output\")\nif a:\n a.setShortcut(QKeySequence('Ctrl+U'))\n#----------------------------------------------\n"
},
{
"path": "openpype/hosts/hiero/api/startup/Python/StartupUI/otioimporter/OTIOImport.py",
"content": "# MIT License\n#\n# Copyright (c) 2018 Daniel Flehner Heen (Storm Studios)\n#\n# Permission is hereby granted, free of charge, to any person obtaining a copy\n# of this software and associated documentation files (the \"Software\"), to deal\n# in the Software without restriction, including without limitation the rights\n# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell\n# copies of the Software, and to permit persons to whom the Software is\n# furnished to do so, subject to the following conditions:\n#\n# The above copyright notice and this permission notice shall be included in\n# all copies or substantial portions of the Software.\n#\n# THE SOFTWARE IS PROVIDED \"AS IS\", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR\n# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,\n# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE\n# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER\n# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,\n# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE\n# SOFTWARE.\n\nimport os\nimport sys\nimport hiero.core\nimport hiero.ui\n\ntry:\n from urllib import unquote\n\nexcept ImportError:\n from urllib.parse import unquote # lint:ok\n\nimport opentimelineio as otio\n\n\ndef get_transition_type(otio_item, otio_track):\n _in, _out = otio_track.neighbors_of(otio_item)\n\n if isinstance(_in, otio.schema.Gap):\n _in = None\n\n if isinstance(_out, otio.schema.Gap):\n _out = None\n\n if _in and _out:\n return \"dissolve\"\n\n elif _in and not _out:\n return \"fade_out\"\n\n elif not _in and _out:\n return \"fade_in\"\n\n else:\n return \"unknown\"\n\n\ndef find_trackitem(name, hiero_track):\n for item in hiero_track.items():\n if item.name() == name:\n return item\n\n return None\n\n\ndef get_neighboring_trackitems(otio_item, otio_track, hiero_track):\n _in, _out = otio_track.neighbors_of(otio_item)\n trackitem_in = None\n trackitem_out = None\n\n if _in:\n trackitem_in = find_trackitem(_in.name, hiero_track)\n\n if _out:\n trackitem_out = find_trackitem(_out.name, hiero_track)\n\n return trackitem_in, trackitem_out\n\n\ndef apply_transition(otio_track, otio_item, track):\n # Figure out type of transition\n transition_type = get_transition_type(otio_item, otio_track)\n\n # Figure out track kind for getattr below\n if isinstance(track, hiero.core.VideoTrack):\n kind = \"\"\n\n else:\n kind = \"Audio\"\n\n try:\n # Gather TrackItems involved in trasition\n item_in, item_out = get_neighboring_trackitems(\n otio_item,\n otio_track,\n track\n )\n\n # Create transition object\n if transition_type == \"dissolve\":\n transition_func = getattr(\n hiero.core.Transition,\n 'create{kind}DissolveTransition'.format(kind=kind)\n )\n\n transition = transition_func(\n item_in,\n item_out,\n otio_item.in_offset.value,\n otio_item.out_offset.value\n )\n\n elif transition_type == \"fade_in\":\n transition_func = getattr(\n hiero.core.Transition,\n 'create{kind}FadeInTransition'.format(kind=kind)\n )\n transition = transition_func(\n item_out,\n otio_item.out_offset.value\n )\n\n elif transition_type == \"fade_out\":\n transition_func = getattr(\n hiero.core.Transition,\n 'create{kind}FadeOutTransition'.format(kind=kind)\n )\n transition = transition_func(\n item_in,\n otio_item.in_offset.value\n )\n\n else:\n # Unknown transition\n return\n\n # Apply transition to track\n track.addTransition(transition)\n\n except Exception, e:\n sys.stderr.write(\n 'Unable to apply transition \"{t}\": \"{e}\"\\n'.format(\n t=otio_item,\n e=e\n )\n )\n\n\ndef prep_url(url_in):\n url = unquote(url_in)\n\n if url.startswith(\"file://localhost/\"):\n return url.replace(\"file://localhost/\", \"\")\n\n url = '{url}'.format(\n sep=url.startswith(os.sep) and \"\" or os.sep,\n url=url.startswith(os.sep) and url[1:] or url\n )\n\n return url\n\n\ndef create_offline_mediasource(otio_clip, path=None):\n hiero_rate = hiero.core.TimeBase(\n otio_clip.source_range.start_time.rate\n )\n\n if isinstance(otio_clip.media_reference, otio.schema.ExternalReference):\n source_range = otio_clip.available_range()\n\n else:\n source_range = otio_clip.source_range\n\n if path is None:\n path = otio_clip.name\n\n media = hiero.core.MediaSource.createOfflineVideoMediaSource(\n prep_url(path),\n source_range.start_time.value,\n source_range.duration.value,\n hiero_rate,\n source_range.start_time.value\n )\n\n return media\n\n\ndef load_otio(otio_file):\n otio_timeline = otio.adapters.read_from_file(otio_file)\n build_sequence(otio_timeline)\n\n\nmarker_color_map = {\n \"PINK\": \"Magenta\",\n \"RED\": \"Red\",\n \"ORANGE\": \"Yellow\",\n \"YELLOW\": \"Yellow\",\n \"GREEN\": \"Green\",\n \"CYAN\": \"Cyan\",\n \"BLUE\": \"Blue\",\n \"PURPLE\": \"Magenta\",\n \"MAGENTA\": \"Magenta\",\n \"BLACK\": \"Blue\",\n \"WHITE\": \"Green\",\n \"MINT\": \"Cyan\"\n}\n\n\ndef get_tag(tagname, tagsbin):\n for tag in tagsbin.items():\n if tag.name() == tagname:\n return tag\n\n if isinstance(tag, hiero.core.Bin):\n tag = get_tag(tagname, tag)\n\n if tag is not None:\n return tag\n\n return None\n\n\ndef add_metadata(metadata, hiero_item):\n for key, value in metadata.items():\n if isinstance(value, dict):\n add_metadata(value, hiero_item)\n continue\n\n if value is not None:\n if not key.startswith(\"tag.\"):\n key = \"tag.\" + key\n\n hiero_item.metadata().setValue(key, str(value))\n\n\ndef add_markers(otio_item, hiero_item, tagsbin):\n if isinstance(otio_item, (otio.schema.Stack, otio.schema.Clip)):\n markers = otio_item.markers\n\n elif isinstance(otio_item, otio.schema.Timeline):\n markers = otio_item.tracks.markers\n\n else:\n markers = []\n\n for marker in markers:\n marker_color = marker.color\n\n _tag = get_tag(marker.name, tagsbin)\n if _tag is None:\n _tag = get_tag(marker_color_map[marker_color], tagsbin)\n\n if _tag is None:\n _tag = hiero.core.Tag(marker_color_map[marker.color])\n\n start = marker.marked_range.start_time.value\n end = (\n marker.marked_range.start_time.value +\n marker.marked_range.duration.value\n )\n\n tag = hiero_item.addTag(_tag)\n tag.setName(marker.name or marker_color_map[marker_color])\n\n # Add metadata\n add_metadata(marker.metadata, tag)\n\n\ndef create_track(otio_track, tracknum, track_kind):\n # Add track kind when dealing with nested stacks\n if isinstance(otio_track, otio.schema.Stack):\n otio_track.kind = track_kind\n\n # Create a Track\n if otio_track.kind == otio.schema.TrackKind.Video:\n track = hiero.core.VideoTrack(\n otio_track.name or 'Video{n}'.format(n=tracknum)\n )\n\n else:\n track = hiero.core.AudioTrack(\n otio_track.name or 'Audio{n}'.format(n=tracknum)\n )\n\n return track\n\n\ndef create_clip(otio_clip):\n # Create MediaSource\n otio_media = otio_clip.media_reference\n if isinstance(otio_media, otio.schema.ExternalReference):\n url = prep_url(otio_media.target_url)\n media = hiero.core.MediaSource(url)\n if media.isOffline():\n media = create_offline_mediasource(otio_clip, url)\n\n else:\n media = create_offline_mediasource(otio_clip)\n\n # Create Clip\n clip = hiero.core.Clip(media)\n\n return clip\n\n\ndef create_trackitem(playhead, track, otio_clip, clip, tagsbin):\n source_range = otio_clip.source_range\n\n trackitem = track.createTrackItem(otio_clip.name)\n trackitem.setPlaybackSpeed(source_range.start_time.rate)\n trackitem.setSource(clip)\n\n # Check for speed effects and adjust playback speed accordingly\n for effect in otio_clip.effects:\n if isinstance(effect, otio.schema.LinearTimeWarp):\n trackitem.setPlaybackSpeed(\n trackitem.playbackSpeed() *\n effect.time_scalar\n )\n\n # If reverse playback speed swap source in and out\n if trackitem.playbackSpeed() < 0:\n source_out = source_range.start_time.value\n source_in = (\n source_range.start_time.value +\n source_range.duration.value\n ) - 1\n timeline_in = playhead + source_out\n timeline_out = (\n timeline_in +\n source_range.duration.value\n ) - 1\n else:\n # Normal playback speed\n source_in = source_range.start_time.value\n source_out = (\n source_range.start_time.value +\n source_range.duration.value\n ) - 1\n timeline_in = playhead\n timeline_out = (\n timeline_in +\n source_range.duration.value\n ) - 1\n\n # Set source and timeline in/out points\n trackitem.setSourceIn(source_in)\n trackitem.setSourceOut(source_out)\n trackitem.setTimelineIn(timeline_in)\n trackitem.setTimelineOut(timeline_out)\n\n # Add markers\n add_markers(otio_clip, trackitem, tagsbin)\n\n return trackitem\n\n\ndef build_sequence(\n otio_timeline, project=None, sequence=None, track_kind=None):\n\n if project is None:\n if sequence:\n project = sequence.project()\n\n else:\n # Per version 12.1v2 there is no way of getting active project\n project = hiero.core.projects(hiero.core.Project.kUserProjects)[-1]\n\n projectbin = project.clipsBin()\n\n if not sequence:\n # Create a Sequence\n sequence = hiero.core.Sequence(otio_timeline.name or \"OTIOSequence\")\n\n # Set sequence settings from otio timeline if available\n if hasattr(otio_timeline, \"global_start_time\"):\n if otio_timeline.global_start_time:\n start_time = otio_timeline.global_start_time\n sequence.setFramerate(start_time.rate)\n sequence.setTimecodeStart(start_time.value)\n\n # Create a Bin to hold clips\n projectbin.addItem(hiero.core.BinItem(sequence))\n\n sequencebin = hiero.core.Bin(sequence.name())\n projectbin.addItem(sequencebin)\n\n else:\n sequencebin = projectbin\n\n # Get tagsBin\n tagsbin = hiero.core.project(\"Tag Presets\").tagsBin()\n\n # Add timeline markers\n add_markers(otio_timeline, sequence, tagsbin)\n\n if isinstance(otio_timeline, otio.schema.Timeline):\n tracks = otio_timeline.tracks\n\n else:\n tracks = [otio_timeline]\n\n for tracknum, otio_track in enumerate(tracks):\n playhead = 0\n _transitions = []\n\n # Add track to sequence\n track = create_track(otio_track, tracknum, track_kind)\n sequence.addTrack(track)\n\n # iterate over items in track\n for itemnum, otio_clip in enumerate(otio_track):\n if isinstance(otio_clip, otio.schema.Stack):\n bar = hiero.ui.mainWindow().statusBar()\n bar.showMessage(\n \"Nested sequences are created separately.\",\n timeout=3000\n )\n build_sequence(otio_clip, project, otio_track.kind)\n\n elif isinstance(otio_clip, otio.schema.Clip):\n # Create a Clip\n clip = create_clip(otio_clip)\n\n # Add Clip to a Bin\n sequencebin.addItem(hiero.core.BinItem(clip))\n\n # Create TrackItem\n trackitem = create_trackitem(\n playhead,\n track,\n otio_clip,\n clip,\n tagsbin\n )\n\n # Add trackitem to track\n track.addTrackItem(trackitem)\n\n # Update playhead\n playhead = trackitem.timelineOut() + 1\n\n elif isinstance(otio_clip, otio.schema.Transition):\n # Store transitions for when all clips in the track are created\n _transitions.append((otio_track, otio_clip))\n\n elif isinstance(otio_clip, otio.schema.Gap):\n # Hiero has no fillers, slugs or blanks at the moment\n playhead += otio_clip.source_range.duration.value\n\n # Apply transitions we stored earlier now that all clips are present\n for otio_track, otio_item in _transitions:\n apply_transition(otio_track, otio_item, track)\n"
},
{
"path": "openpype/hosts/hiero/api/startup/Python/StartupUI/otioimporter/__init__.py",
"content": "#!/usr/bin/env python\n# -*- coding: utf-8 -*-\n\n__author__ = \"Daniel Flehner Heen\"\n__credits__ = [\"Jakub Jezek\", \"Daniel Flehner Heen\"]\n\nimport hiero.ui\nimport hiero.core\n\nimport PySide2.QtWidgets as qw\n\nfrom openpype.hosts.hiero.api.otio.hiero_import import load_otio\n\n\nclass OTIOProjectSelect(qw.QDialog):\n\n def __init__(self, projects, *args, **kwargs):\n super(OTIOProjectSelect, self).__init__(*args, **kwargs)\n self.setWindowTitle(\"Please select active project\")\n self.layout = qw.QVBoxLayout()\n\n self.label = qw.QLabel(\n \"Unable to determine which project to import sequence to.\\n\"\n \"Please select one.\"\n )\n self.layout.addWidget(self.label)\n\n self.projects = qw.QComboBox()\n self.projects.addItems(map(lambda p: p.name(), projects))\n self.layout.addWidget(self.projects)\n\n QBtn = qw.QDialogButtonBox.Ok | qw.QDialogButtonBox.Cancel\n self.buttonBox = qw.QDialogButtonBox(QBtn)\n self.buttonBox.accepted.connect(self.accept)\n self.buttonBox.rejected.connect(self.reject)\n\n self.layout.addWidget(self.buttonBox)\n self.setLayout(self.layout)\n\n\ndef get_sequence(view):\n sequence = None\n if isinstance(view, hiero.ui.TimelineEditor):\n sequence = view.sequence()\n\n elif isinstance(view, hiero.ui.BinView):\n for item in view.selection():\n if not hasattr(item, \"acitveItem\"):\n continue\n\n if isinstance(item.activeItem(), hiero.core.Sequence):\n sequence = item.activeItem()\n\n return sequence\n\n\ndef OTIO_menu_action(event):\n # Menu actions\n otio_import_action = hiero.ui.createMenuAction(\n \"Import OTIO...\",\n open_otio_file,\n icon=None\n )\n\n otio_add_track_action = hiero.ui.createMenuAction(\n \"New Track(s) from OTIO...\",\n open_otio_file,\n icon=None\n )\n otio_add_track_action.setEnabled(False)\n\n hiero.ui.registerAction(otio_import_action)\n hiero.ui.registerAction(otio_add_track_action)\n\n view = hiero.ui.currentContextMenuView()\n\n if view:\n sequence = get_sequence(view)\n if sequence:\n otio_add_track_action.setEnabled(True)\n\n for action in event.menu.actions():\n if action.text() == \"Import\":\n action.menu().addAction(otio_import_action)\n action.menu().addAction(otio_add_track_action)\n\n elif action.text() == \"New Track\":\n action.menu().addAction(otio_add_track_action)\n\n\ndef open_otio_file():\n files = hiero.ui.openFileBrowser(\n caption=\"Please select an OTIO file of choice\",\n pattern=\"*.otio\",\n requiredExtension=\".otio\"\n )\n\n selection = None\n sequence = None\n\n view = hiero.ui.currentContextMenuView()\n if view:\n sequence = get_sequence(view)\n selection = view.selection()\n\n if sequence:\n project = sequence.project()\n\n elif selection:\n project = selection[0].project()\n\n elif len(hiero.core.projects()) > 1:\n dialog = OTIOProjectSelect(hiero.core.projects())\n if dialog.exec_():\n project = hiero.core.projects()[dialog.projects.currentIndex()]\n\n else:\n bar = hiero.ui.mainWindow().statusBar()\n bar.showMessage(\n \"OTIO Import aborted by user\",\n timeout=3000\n )\n return\n\n else:\n project = hiero.core.projects()[-1]\n\n for otio_file in files:\n load_otio(otio_file, project, sequence)\n\n\n# HieroPlayer is quite limited and can't create transitions etc.\nif not hiero.core.isHieroPlayer():\n hiero.core.events.registerInterest(\n \"kShowContextMenu/kBin\",\n OTIO_menu_action\n )\n hiero.core.events.registerInterest(\n \"kShowContextMenu/kTimeline\",\n OTIO_menu_action\n )\n"
},
{
"path": "openpype/hosts/hiero/api/startup/Python/StartupUI/setPosterFrame.py",
"content": "import hiero.core\nimport hiero.ui\ntry:\n from PySide.QtGui import *\n from PySide.QtCore import *\nexcept:\n from PySide2.QtGui import *\n from PySide2.QtWidgets import *\n from PySide2.QtCore import *\n\n\ndef setPosterFrame(posterFrame=.5):\n \"\"\"\n Update the poster frame of the given clipItmes\n posterFrame = .5 uses the centre frame, a value of 0 uses the first frame, a value of 1 uses the last frame\n \"\"\"\n view = hiero.ui.activeView()\n\n selectedBinItems = view.selection()\n selectedClipItems = [(item.activeItem()\n if hasattr(item, \"activeItem\") else item)\n for item in selectedBinItems]\n\n for clip in selectedClipItems:\n centreFrame = int(clip.duration() * posterFrame)\n clip.setPosterFrame(centreFrame)\n\n\nclass SetPosterFrameAction(QAction):\n def __init__(self):\n QAction.__init__(self, \"Set Poster Frame (centre)\", None)\n self._selection = None\n\n self.triggered.connect(lambda: setPosterFrame(.5))\n hiero.core.events.registerInterest(\"kShowContextMenu/kBin\",\n self.eventHandler)\n\n def eventHandler(self, event):\n view = event.sender\n # Add the Menu to the right-click menu\n event.menu.addAction(self)\n\n\n# The act of initialising the action adds it to the right-click menu...\nSetPosterFrameAction()\n"
},
{
"path": "openpype/hosts/hiero/api/startup/TaskPresets/10.5/Processors/hiero.exporters.FnShotProcessor.ShotProcessor/pipeline.xml",
"content": "\n 991\n //10.11.0.184/171001_ftrack/tgbvfx/editorial/hiero/workspace/\n 1\n True\n 3\n \n \n {shot}/editorial_raw.%04d.{fileext}\n \n \n default\n exr\n False\n all\n False\n False\n False\n False\n True\n \n \n 8 bit\n (auto detect)\n True\n False\n \n False\n \n None\n None\n None\n None\n None\n None\n None\n None\n None\n \n \n Zip (16 scanline)\n 32 bit float\n False\n False\n False\n channels, layers and views\n 45.0\n False\n all metadata\n \n Write_{ext}\n \n Cubic\n None\n 1.0\n True\n width\n \n False\n Blend\n \n \n \n \n {shot}/editorial.%04d.{ext}\n \n \n default\n exr\n False\n all\n False\n False\n False\n False\n True\n \n \n 8 bit\n (auto detect)\n True\n False\n \n True\n \n None\n None\n None\n None\n None\n None\n None\n None\n None\n \n \n Zip (16 scanline)\n 16 bit half\n False\n False\n False\n channels, layers and views\n 45.0\n False\n all metadata\n \n Write_{ext}\n \n Cubic\n To Sequence Resolution\n 1.0\n True\n width\n \n False\n Blend\n \n \n \n \n {shot}/editorial.nk\n \n \n True\n default\n mov\n \n rgb\n False\n \n False\n False\n False\n \n True\n True\n \n {shot}/editorial_raw.%04d.{fileext}\n \n \n Cubic\n None\n 1.0\n True\n width\n \n False\n Blend\n False\n True\n True\n \n 0\n 40000000\n 12\n 31\n 2\n avc1 H.264\n Auto\n mov32\n 20000\n \n False\n True\n True\n False\n False\n {shot}/editorial_raw.%04d.{fileext}\n \n None\n None\n None\n None\n None\n None\n None\n None\n None\n \n \n 8 bit\n (auto detect)\n True\n False\n \n Write_{ext}\n False\n \n \n \n \n \n \n False\n Custom\n True\n 10\n\n"
},
{
"path": "openpype/hosts/hiero/api/startup/TaskPresets/11.1/Processors/hiero.exporters.FnShotProcessor.ShotProcessor/pipeline.xml",
"content": "\n 991\n //10.11.0.184/171001_ftrack/tgbvfx/editorial/hiero/workspace/\n 1\n True\n 3\n \n \n {shot}/editorial_raw.%04d.{fileext}\n \n \n default\n exr\n False\n all\n False\n False\n False\n False\n True\n \n \n 8 bit\n (auto detect)\n True\n False\n \n False\n \n None\n None\n None\n None\n None\n None\n None\n None\n None\n \n \n Zip (16 scanline)\n 32 bit float\n False\n False\n False\n channels, layers and views\n 45.0\n False\n all metadata\n \n Write_{ext}\n \n Cubic\n None\n 1.0\n True\n width\n \n False\n Blend\n \n \n \n \n {shot}/editorial.%04d.{ext}\n \n \n default\n exr\n False\n all\n False\n False\n False\n False\n True\n \n \n 8 bit\n (auto detect)\n True\n False\n \n True\n \n None\n None\n None\n None\n None\n None\n None\n None\n None\n \n \n Zip (16 scanline)\n 16 bit half\n False\n False\n False\n channels, layers and views\n 45.0\n False\n all metadata\n \n Write_{ext}\n \n Cubic\n To Sequence Resolution\n 1.0\n True\n width\n \n False\n Blend\n \n \n \n \n {shot}/editorial.nk\n \n \n True\n default\n mov\n \n rgb\n False\n \n False\n False\n False\n \n True\n True\n \n {shot}/editorial_raw.%04d.{fileext}\n \n \n Cubic\n None\n 1.0\n True\n width\n \n False\n Blend\n False\n True\n True\n \n 0\n 40000000\n 12\n 31\n 2\n avc1 H.264\n Auto\n mov32\n 20000\n \n False\n True\n True\n False\n False\n {shot}/editorial_raw.%04d.{fileext}\n \n None\n None\n None\n None\n None\n None\n None\n None\n None\n \n \n 8 bit\n (auto detect)\n True\n False\n \n Write_{ext}\n False\n \n \n \n \n \n \n False\n Custom\n True\n 10\n\n"
},
{
"path": "openpype/hosts/hiero/api/startup/TaskPresets/11.2/hiero.exporters.FnShotProcessor.ShotProcessor/pipeline.xml",
"content": "\n 991\n //10.11.0.184/171001_ftrack/tgbvfx/editorial/hiero/workspace/\n 1\n True\n 3\n \n \n {shot}/editorial_raw.%04d.{fileext}\n \n \n default\n exr\n False\n all\n False\n False\n False\n False\n True\n \n \n 8 bit\n (auto detect)\n True\n False\n \n False\n \n None\n None\n None\n None\n None\n None\n None\n None\n None\n \n \n Zip (16 scanline)\n 32 bit float\n False\n False\n False\n channels, layers and views\n 45.0\n False\n all metadata\n \n Write_{ext}\n \n Cubic\n None\n 1.0\n True\n width\n \n False\n Blend\n \n \n \n \n {shot}/editorial.%04d.{ext}\n \n \n default\n exr\n False\n all\n False\n False\n False\n False\n True\n \n \n 8 bit\n (auto detect)\n True\n False\n \n True\n \n None\n None\n None\n None\n None\n None\n None\n None\n None\n \n \n Zip (16 scanline)\n 16 bit half\n False\n False\n False\n channels, layers and views\n 45.0\n False\n all metadata\n \n Write_{ext}\n \n Cubic\n To Sequence Resolution\n 1.0\n True\n width\n \n False\n Blend\n \n \n \n \n {shot}/editorial.nk\n \n \n True\n default\n mov\n \n rgb\n False\n \n False\n False\n False\n \n True\n True\n \n {shot}/editorial_raw.%04d.{fileext}\n \n \n Cubic\n None\n 1.0\n True\n width\n \n False\n Blend\n False\n True\n True\n \n 0\n 40000000\n 12\n 31\n 2\n avc1 H.264\n Auto\n mov32\n 20000\n \n False\n True\n True\n False\n False\n {shot}/editorial_raw.%04d.{fileext}\n \n None\n None\n None\n None\n None\n None\n None\n None\n None\n \n \n 8 bit\n (auto detect)\n True\n False\n \n Write_{ext}\n False\n \n \n \n \n \n \n False\n Custom\n True\n 10\n\n"
},
{
"path": "openpype/hosts/hiero/api/style.css",
"content": "QWidget {\n font-size: 13px;\n}\n\nQSpinBox {\n padding: 2;\n max-width: 8em;\n}\n\nQLineEdit {\n padding: 2;\n min-width: 15em;\n}\n\nQVBoxLayout {\n min-width: 15em;\n background-color: #201f1f;\n}\n\nQComboBox {\n min-width: 8em;\n}\n\n#sectionContent {\n background-color: #2E2D2D;\n}"
},
{
"path": "openpype/hosts/hiero/api/tags.py",
"content": "import json\nimport re\nimport os\nimport hiero\n\nfrom openpype.client import get_project, get_assets\nfrom openpype.lib import Logger\nfrom openpype.pipeline import get_current_project_name\n\nlog = Logger.get_logger(__name__)\n\n\ndef tag_data():\n return {\n \"[Lenses]\": {\n \"Set lense here\": {\n \"editable\": \"1\",\n \"note\": \"Adjust parameters of your lense and then drop to clip. Remember! You can always overwrite on clip\", # noqa\n \"icon\": \"lense.png\",\n \"metadata\": {\n \"focalLengthMm\": 57\n\n }\n }\n },\n # \"NukeScript\": {\n # \"editable\": \"1\",\n # \"note\": \"Collecting track items to Nuke scripts.\",\n # \"icon\": \"icons:TagNuke.png\",\n # \"metadata\": {\n # \"family\": \"nukescript\",\n # \"subset\": \"main\"\n # }\n # },\n \"Comment\": {\n \"editable\": \"1\",\n \"note\": \"Comment on a shot.\",\n \"icon\": \"icons:TagComment.png\",\n \"metadata\": {\n \"family\": \"comment\",\n \"subset\": \"main\"\n }\n },\n \"FrameMain\": {\n \"editable\": \"1\",\n \"note\": \"Publishing a frame subset.\",\n \"icon\": \"z_layer_main.png\",\n \"metadata\": {\n \"family\": \"frame\",\n \"subset\": \"main\",\n \"format\": \"png\"\n }\n }\n }\n\n\ndef create_tag(key, data):\n \"\"\"\n Creating Tag object.\n\n Args:\n key (str): name of tag\n data (dict): parameters of tag\n\n Returns:\n object: Tag object\n \"\"\"\n tag = hiero.core.Tag(str(key))\n return update_tag(tag, data)\n\n\ndef update_tag(tag, data):\n \"\"\"\n Fixing Tag object.\n\n Args:\n tag (obj): Tag object\n data (dict): parameters of tag\n \"\"\"\n # set icon if any available in input data\n if data.get(\"icon\"):\n tag.setIcon(str(data[\"icon\"]))\n\n # get metadata of tag\n mtd = tag.metadata()\n # get metadata key from data\n data_mtd = data.get(\"metadata\", {})\n\n # set all data metadata to tag metadata\n for _k, _v in data_mtd.items():\n value = str(_v)\n if type(_v) == dict:\n value = json.dumps(_v)\n\n # set the value\n mtd.setValue(\n \"tag.{}\".format(str(_k)),\n value\n )\n\n # set note description of tag\n tag.setNote(str(data[\"note\"]))\n return tag\n\n\ndef add_tags_to_workfile():\n \"\"\"\n Will create default tags from presets.\n \"\"\"\n from .lib import get_current_project\n\n def add_tag_to_bin(root_bin, name, data):\n # for Tags to be created in root level Bin\n # at first check if any of input data tag is not already created\n done_tag = next((t for t in root_bin.items()\n if str(name) in t.name()), None)\n\n if not done_tag:\n # create Tag\n tag = create_tag(name, data)\n tag.setName(str(name))\n\n log.debug(\"__ creating tag: {}\".format(tag))\n # adding Tag to Root Bin\n root_bin.addItem(tag)\n else:\n # update only non hierarchy tags\n update_tag(done_tag, data)\n done_tag.setName(str(name))\n log.debug(\"__ updating tag: {}\".format(done_tag))\n\n # get project and root bin object\n project = get_current_project()\n root_bin = project.tagsBin()\n\n if \"Tag Presets\" in project.name():\n return\n\n log.debug(\"Setting default tags on project: {}\".format(project.name()))\n\n # get hiero tags.json\n nks_pres_tags = tag_data()\n\n # Get project task types.\n project_name = get_current_project_name()\n project_doc = get_project(project_name)\n tasks = project_doc[\"config\"][\"tasks\"]\n nks_pres_tags[\"[Tasks]\"] = {}\n log.debug(\"__ tasks: {}\".format(tasks))\n for task_type in tasks.keys():\n nks_pres_tags[\"[Tasks]\"][task_type.lower()] = {\n \"editable\": \"1\",\n \"note\": task_type,\n \"icon\": \"icons:TagGood.png\",\n \"metadata\": {\n \"family\": \"task\",\n \"type\": task_type\n }\n }\n\n # Get project assets. Currently Ftrack specific to differentiate between\n # asset builds and shots.\n if int(os.getenv(\"TAG_ASSETBUILD_STARTUP\", 0)) == 1:\n nks_pres_tags[\"[AssetBuilds]\"] = {}\n for asset in get_assets(\n project_name, fields=[\"name\", \"data.entityType\"]\n ):\n if asset[\"data\"][\"entityType\"] == \"AssetBuild\":\n nks_pres_tags[\"[AssetBuilds]\"][asset[\"name\"]] = {\n \"editable\": \"1\",\n \"note\": \"\",\n \"icon\": {\n \"path\": \"icons:TagActor.png\"\n },\n \"metadata\": {\n \"family\": \"assetbuild\"\n }\n }\n\n # loop through tag data dict and create deep tag structure\n for _k, _val in nks_pres_tags.items():\n # check if key is not decorated with [] so it is defined as bin\n bin_find = None\n pattern = re.compile(r\"\\[(.*)\\]\")\n _bin_finds = pattern.findall(_k)\n # if there is available any then pop it to string\n if _bin_finds:\n bin_find = _bin_finds.pop()\n\n # if bin was found then create or update\n if bin_find:\n root_add = False\n # first check if in root lever is not already created bins\n bins = [b for b in root_bin.items()\n if b.name() in str(bin_find)]\n\n if bins:\n bin = bins.pop()\n else:\n root_add = True\n # create Bin object for processing\n bin = hiero.core.Bin(str(bin_find))\n\n # update or create tags in the bin\n for __k, __v in _val.items():\n add_tag_to_bin(bin, __k, __v)\n\n # finally add the Bin object to the root level Bin\n if root_add:\n # adding Tag to Root Bin\n root_bin.addItem(bin)\n else:\n add_tag_to_bin(root_bin, _k, _val)\n\n log.info(\"Default Tags were set...\")\n"
},
{
"path": "openpype/hosts/hiero/api/workio.py",
"content": "import os\nimport hiero\n\nfrom openpype.lib import Logger\n\nlog = Logger.get_logger(__name__)\n\n\ndef file_extensions():\n return [\".hrox\"]\n\n\ndef has_unsaved_changes():\n # There are no methods for querying unsaved changes to a project, so\n # enforcing to always save.\n # but we could at least check if a current open script has a path\n project = hiero.core.projects()[-1]\n if project.path():\n return True\n else:\n return False\n\n\ndef save_file(filepath):\n file = os.path.basename(filepath)\n project = hiero.core.projects()[-1]\n\n if project:\n log.info(\"Saving project: `{}` as '{}'\".format(project.name(), file))\n project.saveAs(filepath)\n else:\n log.info(\"Creating new project...\")\n project = hiero.core.newProject()\n project.saveAs(filepath)\n\n\ndef open_file(filepath):\n \"\"\"Manually fire the kBeforeProjectLoad event in order to work around a bug in Hiero.\n The Foundry has logged this bug as:\n Bug 40413 - Python API - kBeforeProjectLoad event type is not triggered\n when calling hiero.core.openProject() (only triggered through UI)\n It exists in all versions of Hiero through (at least) v1.9v1b12.\n\n Once this bug is fixed, a version check will need to be added here in order to\n prevent accidentally firing this event twice. The following commented-out code\n is just an example, and will need to be updated when the bug is fixed to catch the\n correct versions.\"\"\"\n # if (hiero.core.env['VersionMajor'] < 1 or\n # hiero.core.env['VersionMajor'] == 1 and hiero.core.env['VersionMinor'] < 10:\n hiero.core.events.sendEvent(\"kBeforeProjectLoad\", None)\n\n project = hiero.core.projects()[-1]\n\n # open project file\n hiero.core.openProject(filepath.replace(os.path.sep, \"/\"))\n\n # close previous project\n project.close()\n\n\n\n return True\n\n\ndef current_file():\n current_file = hiero.core.projects()[-1].path()\n if not current_file:\n return None\n return os.path.normpath(current_file)\n\n\ndef work_root(session):\n return os.path.normpath(session[\"AVALON_WORKDIR\"]).replace(\"\\\\\", \"/\")\n"
},
{
"path": "openpype/hosts/hiero/plugins/create/create_shot_clip.py",
"content": "from copy import deepcopy\nimport openpype.hosts.hiero.api as phiero\n# from openpype.hosts.hiero.api import plugin, lib\n# reload(lib)\n# reload(plugin)\n# reload(phiero)\n\n\nclass CreateShotClip(phiero.Creator):\n \"\"\"Publishable clip\"\"\"\n\n label = \"Create Publishable Clip\"\n family = \"clip\"\n icon = \"film\"\n defaults = [\"Main\"]\n\n gui_tracks = [track.name()\n for track in phiero.get_current_sequence().videoTracks()]\n gui_name = \"Pype publish attributes creator\"\n gui_info = \"Define sequential rename and fill hierarchy data.\"\n gui_inputs = {\n \"renameHierarchy\": {\n \"type\": \"section\",\n \"label\": \"Shot Hierarchy And Rename Settings\",\n \"target\": \"ui\",\n \"order\": 0,\n \"value\": {\n \"hierarchy\": {\n \"value\": \"{folder}/{sequence}\",\n \"type\": \"QLineEdit\",\n \"label\": \"Shot Parent Hierarchy\",\n \"target\": \"tag\",\n \"toolTip\": \"Parents folder for shot root folder, Template filled with `Hierarchy Data` section\", # noqa\n \"order\": 0},\n \"clipRename\": {\n \"value\": False,\n \"type\": \"QCheckBox\",\n \"label\": \"Rename clips\",\n \"target\": \"ui\",\n \"toolTip\": \"Renaming selected clips on fly\", # noqa\n \"order\": 1},\n \"clipName\": {\n \"value\": \"{sequence}{shot}\",\n \"type\": \"QLineEdit\",\n \"label\": \"Clip Name Template\",\n \"target\": \"ui\",\n \"toolTip\": \"template for creating shot namespaused for renaming (use rename: on)\", # noqa\n \"order\": 2},\n \"countFrom\": {\n \"value\": 10,\n \"type\": \"QSpinBox\",\n \"label\": \"Count sequence from\",\n \"target\": \"ui\",\n \"toolTip\": \"Set when the sequence number stafrom\", # noqa\n \"order\": 3},\n \"countSteps\": {\n \"value\": 10,\n \"type\": \"QSpinBox\",\n \"label\": \"Stepping number\",\n \"target\": \"ui\",\n \"toolTip\": \"What number is adding every new step\", # noqa\n \"order\": 4},\n }\n },\n \"hierarchyData\": {\n \"type\": \"dict\",\n \"label\": \"Shot Template Keywords\",\n \"target\": \"tag\",\n \"order\": 1,\n \"value\": {\n \"folder\": {\n \"value\": \"shots\",\n \"type\": \"QLineEdit\",\n \"label\": \"{folder}\",\n \"target\": \"tag\",\n \"toolTip\": \"Name of folder used for root of generated shots.\\nUsable tokens:\\n\\t{_clip_}: name of used clip\\n\\t{_track_}: name of parent track layer\\n\\t{_sequence_}: name of parent sequence (timeline)\", # noqa\n \"order\": 0},\n \"episode\": {\n \"value\": \"ep01\",\n \"type\": \"QLineEdit\",\n \"label\": \"{episode}\",\n \"target\": \"tag\",\n \"toolTip\": \"Name of episode.\\nUsable tokens:\\n\\t{_clip_}: name of used clip\\n\\t{_track_}: name of parent track layer\\n\\t{_sequence_}: name of parent sequence (timeline)\", # noqa\n \"order\": 1},\n \"sequence\": {\n \"value\": \"sq01\",\n \"type\": \"QLineEdit\",\n \"label\": \"{sequence}\",\n \"target\": \"tag\",\n \"toolTip\": \"Name of sequence of shots.\\nUsable tokens:\\n\\t{_clip_}: name of used clip\\n\\t{_track_}: name of parent track layer\\n\\t{_sequence_}: name of parent sequence (timeline)\", # noqa\n \"order\": 2},\n \"track\": {\n \"value\": \"{_track_}\",\n \"type\": \"QLineEdit\",\n \"label\": \"{track}\",\n \"target\": \"tag\",\n \"toolTip\": \"Name of sequence of shots.\\nUsable tokens:\\n\\t{_clip_}: name of used clip\\n\\t{_track_}: name of parent track layer\\n\\t{_sequence_}: name of parent sequence (timeline)\", # noqa\n \"order\": 3},\n \"shot\": {\n \"value\": \"sh###\",\n \"type\": \"QLineEdit\",\n \"label\": \"{shot}\",\n \"target\": \"tag\",\n \"toolTip\": \"Name of shot. `#` is converted to paded number. \\nAlso could be used with usable tokens:\\n\\t{_clip_}: name of used clip\\n\\t{_track_}: name of parent track layer\\n\\t{_sequence_}: name of parent sequence (timeline)\", # noqa\n \"order\": 4}\n }\n },\n \"verticalSync\": {\n \"type\": \"section\",\n \"label\": \"Vertical Synchronization Of Attributes\",\n \"target\": \"ui\",\n \"order\": 2,\n \"value\": {\n \"vSyncOn\": {\n \"value\": True,\n \"type\": \"QCheckBox\",\n \"label\": \"Enable Vertical Sync\",\n \"target\": \"ui\",\n \"toolTip\": \"Switch on if you want clips above each other to share its attributes\", # noqa\n \"order\": 0},\n \"vSyncTrack\": {\n \"value\": gui_tracks, # noqa\n \"type\": \"QComboBox\",\n \"label\": \"Hero track\",\n \"target\": \"ui\",\n \"toolTip\": \"Select driving track name which should be hero for all others\", # noqa\n \"order\": 1}\n }\n },\n \"publishSettings\": {\n \"type\": \"section\",\n \"label\": \"Publish Settings\",\n \"target\": \"ui\",\n \"order\": 3,\n \"value\": {\n \"subsetName\": {\n \"value\": [\"\", \"main\", \"bg\", \"fg\", \"bg\",\n \"animatic\"],\n \"type\": \"QComboBox\",\n \"label\": \"Subset Name\",\n \"target\": \"ui\",\n \"toolTip\": \"chose subset name pattern, if is selected, name of track layer will be used\", # noqa\n \"order\": 0},\n \"subsetFamily\": {\n \"value\": [\"plate\", \"take\"],\n \"type\": \"QComboBox\",\n \"label\": \"Subset Family\",\n \"target\": \"ui\", \"toolTip\": \"What use of this subset is for\", # noqa\n \"order\": 1},\n \"reviewTrack\": {\n \"value\": [\"< none >\"] + gui_tracks,\n \"type\": \"QComboBox\",\n \"label\": \"Use Review Track\",\n \"target\": \"ui\",\n \"toolTip\": \"Generate preview videos on fly, if `< none >` is defined nothing will be generated.\", # noqa\n \"order\": 2},\n \"audio\": {\n \"value\": False,\n \"type\": \"QCheckBox\",\n \"label\": \"Include audio\",\n \"target\": \"tag\",\n \"toolTip\": \"Process subsets with corresponding audio\", # noqa\n \"order\": 3},\n \"sourceResolution\": {\n \"value\": False,\n \"type\": \"QCheckBox\",\n \"label\": \"Source resolution\",\n \"target\": \"tag\",\n \"toolTip\": \"Is resloution taken from timeline or source?\", # noqa\n \"order\": 4},\n }\n },\n \"frameRangeAttr\": {\n \"type\": \"section\",\n \"label\": \"Shot Attributes\",\n \"target\": \"ui\",\n \"order\": 4,\n \"value\": {\n \"workfileFrameStart\": {\n \"value\": 1001,\n \"type\": \"QSpinBox\",\n \"label\": \"Workfiles Start Frame\",\n \"target\": \"tag\",\n \"toolTip\": \"Set workfile starting frame number\", # noqa\n \"order\": 0\n },\n \"handleStart\": {\n \"value\": 0,\n \"type\": \"QSpinBox\",\n \"label\": \"Handle Start\",\n \"target\": \"tag\",\n \"toolTip\": \"Handle at start of clip\", # noqa\n \"order\": 1\n },\n \"handleEnd\": {\n \"value\": 0,\n \"type\": \"QSpinBox\",\n \"label\": \"Handle End\",\n \"target\": \"tag\",\n \"toolTip\": \"Handle at end of clip\", # noqa\n \"order\": 2\n }\n }\n }\n }\n\n presets = None\n\n def process(self):\n # Creator copy of object attributes that are modified during `process`\n presets = deepcopy(self.presets)\n gui_inputs = deepcopy(self.gui_inputs)\n\n # get key pares from presets and match it on ui inputs\n for k, v in gui_inputs.items():\n if v[\"type\"] in (\"dict\", \"section\"):\n # nested dictionary (only one level allowed\n # for sections and dict)\n for _k, _v in v[\"value\"].items():\n if presets.get(_k):\n gui_inputs[k][\n \"value\"][_k][\"value\"] = presets[_k]\n if presets.get(k):\n gui_inputs[k][\"value\"] = presets[k]\n\n # open widget for plugins inputs\n widget = self.widget(self.gui_name, self.gui_info, gui_inputs)\n widget.exec_()\n\n if len(self.selected) < 1:\n return\n\n if not widget.result:\n print(\"Operation aborted\")\n return\n\n self.rename_add = 0\n\n # get ui output for track name for vertical sync\n v_sync_track = widget.result[\"vSyncTrack\"][\"value\"]\n\n # sort selected trackItems by\n sorted_selected_track_items = list()\n unsorted_selected_track_items = list()\n for _ti in self.selected:\n if _ti.parent().name() in v_sync_track:\n sorted_selected_track_items.append(_ti)\n else:\n unsorted_selected_track_items.append(_ti)\n\n sorted_selected_track_items.extend(unsorted_selected_track_items)\n\n kwargs = {\n \"ui_inputs\": widget.result,\n \"avalon\": self.data\n }\n\n for i, track_item in enumerate(sorted_selected_track_items):\n self.rename_index = i\n\n # convert track item to timeline media pool item\n phiero.PublishClip(self, track_item, **kwargs).convert()\n"
},
{
"path": "openpype/hosts/hiero/plugins/load/load_clip.py",
"content": "from openpype.client import (\n get_version_by_id,\n get_last_version_by_subset_id\n)\nfrom openpype.pipeline import (\n get_representation_path,\n get_current_project_name,\n)\nfrom openpype.lib.transcoding import (\n VIDEO_EXTENSIONS,\n IMAGE_EXTENSIONS\n)\nimport openpype.hosts.hiero.api as phiero\n\n\nclass LoadClip(phiero.SequenceLoader):\n \"\"\"Load a subset to timeline as clip\n\n Place clip to timeline on its asset origin timings collected\n during conforming to project\n \"\"\"\n\n families = [\"render2d\", \"source\", \"plate\", \"render\", \"review\"]\n representations = [\"*\"]\n extensions = set(\n ext.lstrip(\".\") for ext in IMAGE_EXTENSIONS.union(VIDEO_EXTENSIONS)\n )\n\n label = \"Load as clip\"\n order = -10\n icon = \"code-fork\"\n color = \"orange\"\n\n # for loader multiselection\n sequence = None\n track = None\n\n # presets\n clip_color_last = \"green\"\n clip_color = \"red\"\n\n clip_name_template = \"{asset}_{subset}_{representation}\"\n\n @classmethod\n def apply_settings(cls, project_settings, system_settings):\n plugin_type_settings = (\n project_settings\n .get(\"hiero\", {})\n .get(\"load\", {})\n )\n\n if not plugin_type_settings:\n return\n\n plugin_name = cls.__name__\n\n plugin_settings = None\n # Look for plugin settings in host specific settings\n if plugin_name in plugin_type_settings:\n plugin_settings = plugin_type_settings[plugin_name]\n\n if not plugin_settings:\n return\n\n print(\">>> We have preset for {}\".format(plugin_name))\n for option, value in plugin_settings.items():\n if option == \"enabled\" and value is False:\n print(\" - is disabled by preset\")\n elif option == \"representations\":\n continue\n else:\n print(\" - setting `{}`: `{}`\".format(option, value))\n setattr(cls, option, value)\n\n\n def load(self, context, name, namespace, options):\n # add clip name template to options\n options.update({\n \"clipNameTemplate\": self.clip_name_template\n })\n # in case loader uses multiselection\n if self.track and self.sequence:\n options.update({\n \"sequence\": self.sequence,\n \"track\": self.track,\n \"clipNameTemplate\": self.clip_name_template\n })\n\n # load clip to timeline and get main variables\n path = self.filepath_from_context(context)\n track_item = phiero.ClipLoader(self, context, path, **options).load()\n namespace = namespace or track_item.name()\n version = context['version']\n version_data = version.get(\"data\", {})\n version_name = version.get(\"name\", None)\n colorspace = version_data.get(\"colorspace\", None)\n object_name = self.clip_name_template.format(\n **context[\"representation\"][\"context\"])\n\n # set colorspace\n if colorspace:\n track_item.source().setSourceMediaColourTransform(colorspace)\n\n # add additional metadata from the version to imprint Avalon knob\n add_keys = [\n \"frameStart\", \"frameEnd\", \"source\", \"author\",\n \"fps\", \"handleStart\", \"handleEnd\"\n ]\n\n # move all version data keys to tag data\n data_imprint = {}\n for key in add_keys:\n data_imprint.update({\n key: version_data.get(key, str(None))\n })\n\n # add variables related to version context\n data_imprint.update({\n \"version\": version_name,\n \"colorspace\": colorspace,\n \"objectName\": object_name\n })\n\n # update color of clip regarding the version order\n self.set_item_color(track_item, version)\n\n # deal with multiselection\n self.multiselection(track_item)\n\n self.log.info(\"Loader done: `{}`\".format(name))\n\n return phiero.containerise(\n track_item,\n name, namespace, context,\n self.__class__.__name__,\n data_imprint)\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def update(self, container, representation):\n \"\"\" Updating previously loaded clips\n \"\"\"\n\n # load clip to timeline and get main variables\n name = container['name']\n namespace = container['namespace']\n track_item = phiero.get_track_items(\n track_item_name=namespace).pop()\n\n project_name = get_current_project_name()\n version_doc = get_version_by_id(project_name, representation[\"parent\"])\n\n version_data = version_doc.get(\"data\", {})\n version_name = version_doc.get(\"name\", None)\n colorspace = version_data.get(\"colorspace\", None)\n object_name = \"{}_{}\".format(name, namespace)\n file = get_representation_path(representation).replace(\"\\\\\", \"/\")\n clip = track_item.source()\n\n # reconnect media to new path\n clip.reconnectMedia(file)\n\n # set colorspace\n if colorspace:\n clip.setSourceMediaColourTransform(colorspace)\n\n # add additional metadata from the version to imprint Avalon knob\n add_keys = [\n \"frameStart\", \"frameEnd\", \"source\", \"author\",\n \"fps\", \"handleStart\", \"handleEnd\"\n ]\n\n # move all version data keys to tag data\n data_imprint = {}\n for key in add_keys:\n data_imprint.update({\n key: version_data.get(key, str(None))\n })\n\n # add variables related to version context\n data_imprint.update({\n \"representation\": str(representation[\"_id\"]),\n \"version\": version_name,\n \"colorspace\": colorspace,\n \"objectName\": object_name\n })\n\n # update color of clip regarding the version order\n self.set_item_color(track_item, version_doc)\n\n return phiero.update_container(track_item, data_imprint)\n\n def remove(self, container):\n \"\"\" Removing previously loaded clips\n \"\"\"\n # load clip to timeline and get main variables\n namespace = container['namespace']\n track_item = phiero.get_track_items(\n track_item_name=namespace).pop()\n track = track_item.parent()\n\n # remove track item from track\n track.removeItem(track_item)\n\n @classmethod\n def multiselection(cls, track_item):\n if not cls.track:\n cls.track = track_item.parent()\n cls.sequence = cls.track.parent()\n\n @classmethod\n def set_item_color(cls, track_item, version_doc):\n project_name = get_current_project_name()\n last_version_doc = get_last_version_by_subset_id(\n project_name, version_doc[\"parent\"], fields=[\"_id\"]\n )\n clip = track_item.source()\n # set clip colour\n if version_doc[\"_id\"] == last_version_doc[\"_id\"]:\n clip.binItem().setColor(cls.clip_color_last)\n else:\n clip.binItem().setColor(cls.clip_color)\n"
},
{
"path": "openpype/hosts/hiero/plugins/load/load_effects.py",
"content": "import json\nfrom collections import OrderedDict\nimport six\n\nfrom openpype.client import (\n get_version_by_id\n)\n\nfrom openpype.pipeline import (\n AVALON_CONTAINER_ID,\n load,\n get_representation_path,\n get_current_project_name\n)\nfrom openpype.hosts.hiero import api as phiero\nfrom openpype.lib import Logger\n\n\nclass LoadEffects(load.LoaderPlugin):\n \"\"\"Loading colorspace soft effect exported from nukestudio\"\"\"\n\n families = [\"effect\"]\n representations = [\"*\"]\n extension = {\"json\"}\n\n label = \"Load Effects\"\n order = 0\n icon = \"cc\"\n color = \"white\"\n\n log = Logger.get_logger(__name__)\n\n def load(self, context, name, namespace, data):\n \"\"\"\n Loading function to get the soft effects to particular read node\n\n Arguments:\n context (dict): context of version\n name (str): name of the version\n namespace (str): asset name\n data (dict): compulsory attribute > not used\n\n Returns:\n nuke node: containerised nuke node object\n \"\"\"\n active_sequence = phiero.get_current_sequence()\n active_track = phiero.get_current_track(\n active_sequence, \"Loaded_{}\".format(name))\n\n # get main variables\n namespace = namespace or context[\"asset\"][\"name\"]\n object_name = \"{}_{}\".format(name, namespace)\n clip_in = context[\"asset\"][\"data\"][\"clipIn\"]\n clip_out = context[\"asset\"][\"data\"][\"clipOut\"]\n\n data_imprint = {\n \"objectName\": object_name,\n \"children_names\": []\n }\n\n # getting file path\n file = self.filepath_from_context(context)\n file = file.replace(\"\\\\\", \"/\")\n\n if self._shared_loading(\n file,\n active_track,\n clip_in,\n clip_out,\n data_imprint\n ):\n self.containerise(\n active_track,\n name=name,\n namespace=namespace,\n object_name=object_name,\n context=context,\n loader=self.__class__.__name__,\n data=data_imprint)\n\n def _shared_loading(\n self,\n file,\n active_track,\n clip_in,\n clip_out,\n data_imprint,\n update=False\n ):\n # getting data from json file with unicode conversion\n with open(file, \"r\") as f:\n json_f = {self.byteify(key): self.byteify(value)\n for key, value in json.load(f).items()}\n\n # get correct order of nodes by positions on track and subtrack\n nodes_order = self.reorder_nodes(json_f)\n\n used_subtracks = {\n stitem.name(): stitem\n for stitem in phiero.flatten(active_track.subTrackItems())\n }\n\n loaded = False\n for index_order, (ef_name, ef_val) in enumerate(nodes_order.items()):\n new_name = \"{}_loaded\".format(ef_name)\n if new_name not in used_subtracks:\n effect_track_item = active_track.createEffect(\n effectType=ef_val[\"class\"],\n timelineIn=clip_in,\n timelineOut=clip_out,\n subTrackIndex=index_order\n\n )\n effect_track_item.setName(new_name)\n else:\n effect_track_item = used_subtracks[new_name]\n\n node = effect_track_item.node()\n for knob_name, knob_value in ef_val[\"node\"].items():\n if (\n not knob_value\n or knob_name == \"name\"\n ):\n continue\n\n try:\n # assume list means animation\n # except 4 values could be RGBA or vector\n if isinstance(knob_value, list) and len(knob_value) > 4:\n node[knob_name].setAnimated()\n for i, value in enumerate(knob_value):\n if isinstance(value, list):\n # list can have vector animation\n for ci, cv in enumerate(value):\n node[knob_name].setValueAt(\n cv,\n (clip_in + i),\n ci\n )\n else:\n # list is single values\n node[knob_name].setValueAt(\n value,\n (clip_in + i)\n )\n else:\n node[knob_name].setValue(knob_value)\n except NameError:\n self.log.warning(\"Knob: {} cannot be set\".format(\n knob_name))\n\n # register all loaded children\n data_imprint[\"children_names\"].append(new_name)\n\n # make sure containerisation will happen\n loaded = True\n\n return loaded\n\n def update(self, container, representation):\n \"\"\" Updating previously loaded effects\n \"\"\"\n active_track = container[\"_item\"]\n file = get_representation_path(representation).replace(\"\\\\\", \"/\")\n\n # get main variables\n name = container['name']\n namespace = container['namespace']\n\n # get timeline in out data\n project_name = get_current_project_name()\n version_doc = get_version_by_id(project_name, representation[\"parent\"])\n version_data = version_doc[\"data\"]\n clip_in = version_data[\"clipIn\"]\n clip_out = version_data[\"clipOut\"]\n\n object_name = \"{}_{}\".format(name, namespace)\n\n # Disable previously created nodes\n used_subtracks = {\n stitem.name(): stitem\n for stitem in phiero.flatten(active_track.subTrackItems())\n }\n container = phiero.get_track_openpype_data(\n active_track, object_name\n )\n\n loaded_subtrack_items = container[\"children_names\"]\n for loaded_stitem in loaded_subtrack_items:\n if loaded_stitem not in used_subtracks:\n continue\n item_to_remove = used_subtracks.pop(loaded_stitem)\n # TODO: find a way to erase nodes\n self.log.debug(\n \"This node needs to be removed: {}\".format(item_to_remove))\n\n data_imprint = {\n \"objectName\": object_name,\n \"name\": name,\n \"representation\": str(representation[\"_id\"]),\n \"children_names\": []\n }\n\n if self._shared_loading(\n file,\n active_track,\n clip_in,\n clip_out,\n data_imprint,\n update=True\n ):\n return phiero.update_container(active_track, data_imprint)\n\n def reorder_nodes(self, data):\n new_order = OrderedDict()\n trackNums = [v[\"trackIndex\"] for k, v in data.items()\n if isinstance(v, dict)]\n subTrackNums = [v[\"subTrackIndex\"] for k, v in data.items()\n if isinstance(v, dict)]\n\n for trackIndex in range(\n min(trackNums), max(trackNums) + 1):\n for subTrackIndex in range(\n min(subTrackNums), max(subTrackNums) + 1):\n item = self.get_item(data, trackIndex, subTrackIndex)\n if item is not {}:\n new_order.update(item)\n return new_order\n\n def get_item(self, data, trackIndex, subTrackIndex):\n return {key: val for key, val in data.items()\n if isinstance(val, dict)\n if subTrackIndex == val[\"subTrackIndex\"]\n if trackIndex == val[\"trackIndex\"]}\n\n def byteify(self, input):\n \"\"\"\n Converts unicode strings to strings\n It goes through all dictionary\n\n Arguments:\n input (dict/str): input\n\n Returns:\n dict: with fixed values and keys\n\n \"\"\"\n\n if isinstance(input, dict):\n return {self.byteify(key): self.byteify(value)\n for key, value in input.items()}\n elif isinstance(input, list):\n return [self.byteify(element) for element in input]\n elif isinstance(input, six.text_type):\n return str(input)\n else:\n return input\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def remove(self, container):\n pass\n\n def containerise(\n self,\n track,\n name,\n namespace,\n object_name,\n context,\n loader=None,\n data=None\n ):\n \"\"\"Bundle Hiero's object into an assembly and imprint it with metadata\n\n Containerisation enables a tracking of version, author and origin\n for loaded assets.\n\n Arguments:\n track (hiero.core.VideoTrack): object to imprint as container\n name (str): Name of resulting assembly\n namespace (str): Namespace under which to host container\n object_name (str): name of container\n context (dict): Asset information\n loader (str, optional): Name of node used to produce this\n container.\n\n Returns:\n track_item (hiero.core.TrackItem): containerised object\n\n \"\"\"\n\n data_imprint = {\n object_name: {\n \"schema\": \"openpype:container-2.0\",\n \"id\": AVALON_CONTAINER_ID,\n \"name\": str(name),\n \"namespace\": str(namespace),\n \"loader\": str(loader),\n \"representation\": str(context[\"representation\"][\"_id\"]),\n }\n }\n\n if data:\n for k, v in data.items():\n data_imprint[object_name].update({k: v})\n\n self.log.debug(\"_ data_imprint: {}\".format(data_imprint))\n phiero.set_track_openpype_tag(track, data_imprint)\n"
},
{
"path": "openpype/hosts/hiero/plugins/publish/collect_clip_effects.py",
"content": "import re\nimport pyblish.api\n\n\nclass CollectClipEffects(pyblish.api.InstancePlugin):\n \"\"\"Collect soft effects instances.\"\"\"\n\n order = pyblish.api.CollectorOrder - 0.078\n label = \"Collect Clip Effects Instances\"\n families = [\"clip\"]\n\n effect_categories = []\n\n def process(self, instance):\n family = \"effect\"\n effects = {}\n review = instance.data.get(\"review\")\n review_track_index = instance.context.data.get(\"reviewTrackIndex\")\n item = instance.data[\"item\"]\n\n if \"audio\" in instance.data[\"family\"]:\n return\n\n # frame range\n self.handle_start = instance.data[\"handleStart\"]\n self.handle_end = instance.data[\"handleEnd\"]\n self.clip_in = int(item.timelineIn())\n self.clip_out = int(item.timelineOut())\n self.clip_in_h = self.clip_in - self.handle_start\n self.clip_out_h = self.clip_out + self.handle_end\n\n track_item = instance.data[\"item\"]\n track = track_item.parent()\n track_index = track.trackIndex()\n tracks_effect_items = instance.context.data.get(\"tracksEffectItems\")\n clip_effect_items = instance.data.get(\"clipEffectItems\")\n\n # add clips effects to track's:\n if clip_effect_items:\n tracks_effect_items[track_index] = clip_effect_items\n\n # process all effects and divide them to instance\n for _track_index, sub_track_items in tracks_effect_items.items():\n # skip if track index is the same as review track index\n if review and review_track_index == _track_index:\n continue\n for sitem in sub_track_items:\n # make sure this subtrack item is relative of track item\n if ((track_item not in sitem.linkedItems())\n and (len(sitem.linkedItems()) > 0)):\n continue\n\n if not (track_index <= _track_index):\n continue\n\n effect = self.add_effect(_track_index, sitem)\n if effect:\n effects.update(effect)\n\n # skip any without effects\n if not effects:\n return\n\n subset = instance.data.get(\"subset\")\n effects.update({\"assignTo\": subset})\n\n subset_split = re.findall(r'[A-Z][^A-Z]*', subset)\n\n if len(subset_split) > 0:\n root_name = subset.replace(subset_split[0], \"\")\n subset_split.insert(0, root_name.capitalize())\n\n subset_split.insert(0, \"effect\")\n\n # Need to convert to dict for AYON settings. This isinstance check can\n # be removed in the future when OpenPype is no longer.\n effect_categories = self.effect_categories\n if isinstance(self.effect_categories, list):\n effect_categories = {\n x[\"name\"]: x[\"effect_classes\"] for x in self.effect_categories\n }\n\n category_by_effect = {\"\": \"\"}\n for key, values in effect_categories.items():\n for cls in values:\n category_by_effect[cls] = key\n\n effects_categorized = {k: {} for k in effect_categories.keys()}\n effects_categorized[\"\"] = {}\n for key, value in effects.items():\n if key == \"assignTo\":\n continue\n\n # Some classes can have a number in them. Like Text2.\n found_cls = \"\"\n for cls in category_by_effect.keys():\n if cls in value[\"class\"]:\n found_cls = cls\n\n effects_categorized[category_by_effect[found_cls]][key] = value\n\n categories = list(effects_categorized.keys())\n for category in categories:\n if not effects_categorized[category]:\n effects_categorized.pop(category)\n continue\n\n effects_categorized[category][\"assignTo\"] = effects[\"assignTo\"]\n\n for category, effects in effects_categorized.items():\n name = \"\".join(subset_split)\n name += category.capitalize()\n\n # create new instance and inherit data\n data = {}\n for key, value in instance.data.items():\n if \"clipEffectItems\" in key:\n continue\n data[key] = value\n\n # change names\n data[\"subset\"] = name\n data[\"family\"] = family\n data[\"families\"] = [family]\n data[\"name\"] = data[\"subset\"] + \"_\" + data[\"asset\"]\n data[\"label\"] = \"{} - {}\".format(\n data['asset'], data[\"subset\"]\n )\n data[\"effects\"] = effects\n\n # create new instance\n _instance = instance.context.create_instance(**data)\n self.log.info(\"Created instance `{}`\".format(_instance))\n self.log.debug(\"instance.data `{}`\".format(_instance.data))\n\n def test_overlap(self, effect_t_in, effect_t_out):\n covering_exp = bool(\n (effect_t_in <= self.clip_in)\n and (effect_t_out >= self.clip_out)\n )\n overlaying_right_exp = bool(\n (effect_t_in < self.clip_out)\n and (effect_t_out >= self.clip_out)\n )\n overlaying_left_exp = bool(\n (effect_t_out > self.clip_in)\n and (effect_t_in <= self.clip_in)\n )\n\n return any((\n covering_exp,\n overlaying_right_exp,\n overlaying_left_exp\n ))\n\n def add_effect(self, track_index, sitem):\n track = sitem.parentTrack().name()\n # node serialization\n node = sitem.node()\n node_serialized = self.node_serialization(node)\n node_name = sitem.name()\n node_class = node.Class()\n\n # collect timelineIn/Out\n effect_t_in = int(sitem.timelineIn())\n effect_t_out = int(sitem.timelineOut())\n\n if not self.test_overlap(effect_t_in, effect_t_out):\n return\n\n self.log.debug(\"node_name: `{}`\".format(node_name))\n self.log.debug(\"node_class: `{}`\".format(node_class))\n\n return {node_name: {\n \"class\": node_class,\n \"timelineIn\": effect_t_in,\n \"timelineOut\": effect_t_out,\n \"subTrackIndex\": sitem.subTrackIndex(),\n \"trackIndex\": track_index,\n \"track\": track,\n \"node\": node_serialized\n }}\n\n def node_serialization(self, node):\n node_serialized = {}\n\n # adding ignoring knob keys\n _ignoring_keys = ['invert_mask', 'help', 'mask',\n 'xpos', 'ypos', 'layer', 'process_mask', 'channel',\n 'channels', 'maskChannelMask', 'maskChannelInput',\n 'note_font', 'note_font_size', 'unpremult',\n 'postage_stamp_frame', 'maskChannel', 'export_cc',\n 'select_cccid', 'mix', 'version', 'matrix']\n\n # loop through all knobs and collect not ignored\n # and any with any value\n for knob in node.knobs().keys():\n # skip nodes in ignore keys\n if knob in _ignoring_keys:\n continue\n\n # get animation if node is animated\n if node[knob].isAnimated():\n # grab animation including handles\n knob_anim = [node[knob].getValueAt(i)\n for i in range(\n self.clip_in_h, self.clip_out_h + 1)]\n\n node_serialized[knob] = knob_anim\n else:\n node_serialized[knob] = node[knob].value()\n\n return node_serialized\n"
},
{
"path": "openpype/hosts/hiero/plugins/publish/collect_frame_tag_instances.py",
"content": "from pprint import pformat\nimport re\nimport ast\nimport json\n\nimport pyblish.api\n\nfrom openpype.client import get_asset_name_identifier\n\n\nclass CollectFrameTagInstances(pyblish.api.ContextPlugin):\n \"\"\"Collect frames from tags.\n\n Tag is expected to have metadata:\n {\n \"family\": \"frame\"\n \"subset\": \"main\"\n }\n \"\"\"\n\n order = pyblish.api.CollectorOrder\n label = \"Collect Frames\"\n hosts = [\"hiero\"]\n\n def process(self, context):\n self._context = context\n\n # collect all sequence tags\n subset_data = self._create_frame_subset_data_sequence(context)\n\n self.log.debug(\"__ subset_data: {}\".format(\n pformat(subset_data)\n ))\n\n # create instances\n self._create_instances(subset_data)\n\n def _get_tag_data(self, tag):\n data = {}\n\n # get tag metadata attribute\n tag_data = tag.metadata()\n\n # convert tag metadata to normal keys names and values to correct types\n for k, v in dict(tag_data).items():\n key = k.replace(\"tag.\", \"\")\n\n try:\n # capture exceptions which are related to strings only\n if re.match(r\"^[\\d]+$\", v):\n value = int(v)\n elif re.match(r\"^True$\", v):\n value = True\n elif re.match(r\"^False$\", v):\n value = False\n elif re.match(r\"^None$\", v):\n value = None\n elif re.match(r\"^[\\w\\d_]+$\", v):\n value = v\n else:\n value = ast.literal_eval(v)\n except (ValueError, SyntaxError):\n value = v\n\n data[key] = value\n\n return data\n\n def _create_frame_subset_data_sequence(self, context):\n\n sequence_tags = []\n sequence = context.data[\"activeTimeline\"]\n\n # get all publishable sequence frames\n publish_frames = range(int(sequence.duration() + 1))\n\n self.log.debug(\"__ publish_frames: {}\".format(\n pformat(publish_frames)\n ))\n\n # get all sequence tags\n for tag in sequence.tags():\n tag_data = self._get_tag_data(tag)\n self.log.debug(\"__ tag_data: {}\".format(\n pformat(tag_data)\n ))\n if not tag_data:\n continue\n\n if \"family\" not in tag_data:\n continue\n\n if tag_data[\"family\"] != \"frame\":\n continue\n\n sequence_tags.append(tag_data)\n\n self.log.debug(\"__ sequence_tags: {}\".format(\n pformat(sequence_tags)\n ))\n\n # first collect all available subset tag frames\n subset_data = {}\n context_asset_doc = context.data[\"assetEntity\"]\n context_asset_name = get_asset_name_identifier(context_asset_doc)\n\n for tag_data in sequence_tags:\n frame = int(tag_data[\"start\"])\n\n if frame not in publish_frames:\n continue\n\n subset = tag_data[\"subset\"]\n\n if subset in subset_data:\n # update existing subset key\n subset_data[subset][\"frames\"].append(frame)\n else:\n # create new subset key\n subset_data[subset] = {\n \"frames\": [frame],\n \"format\": tag_data[\"format\"],\n \"asset\": context_asset_name\n }\n return subset_data\n\n def _create_instances(self, subset_data):\n # create instance per subset\n for subset_name, subset_data in subset_data.items():\n name = \"frame\" + subset_name.title()\n data = {\n \"name\": name,\n \"label\": \"{} {}\".format(name, subset_data[\"frames\"]),\n \"family\": \"image\",\n \"families\": [\"frame\"],\n \"asset\": subset_data[\"asset\"],\n \"subset\": name,\n \"format\": subset_data[\"format\"],\n \"frames\": subset_data[\"frames\"]\n }\n self._context.create_instance(**data)\n\n self.log.info(\n \"Created instance: {}\".format(\n json.dumps(data, sort_keys=True, indent=4)\n )\n )\n"
},
{
"path": "openpype/hosts/hiero/plugins/publish/collect_tag_tasks.py",
"content": "from pyblish import api\n\n\nclass CollectClipTagTasks(api.InstancePlugin):\n \"\"\"Collect Tags from selected track items.\"\"\"\n\n order = api.CollectorOrder - 0.077\n label = \"Collect Tag Tasks\"\n hosts = [\"hiero\"]\n families = [\"shot\"]\n\n def process(self, instance):\n # gets tags\n tags = instance.data[\"tags\"]\n\n tasks = {}\n for tag in tags:\n t_metadata = dict(tag.metadata())\n t_family = t_metadata.get(\"tag.family\", \"\")\n\n # gets only task family tags and collect labels\n if \"task\" in t_family:\n t_task_name = t_metadata.get(\"tag.label\", \"\")\n t_task_type = t_metadata.get(\"tag.type\", \"\")\n tasks[t_task_name] = {\"type\": t_task_type}\n\n instance.data[\"tasks\"] = tasks\n\n self.log.info(\"Collected Tasks from Tags: `{}`\".format(\n instance.data[\"tasks\"]))\n return\n"
},
{
"path": "openpype/hosts/hiero/plugins/publish/extract_clip_effects.py",
"content": "# from openpype import plugins\nimport os\nimport json\nimport pyblish.api\n\nfrom openpype.pipeline import publish\n\n\nclass ExtractClipEffects(publish.Extractor):\n \"\"\"Extract clip effects instances.\"\"\"\n\n order = pyblish.api.ExtractorOrder\n label = \"Export Clip Effects\"\n families = [\"effect\"]\n\n def process(self, instance):\n item = instance.data[\"item\"]\n effects = instance.data.get(\"effects\")\n\n # skip any without effects\n if not effects:\n return\n\n subset = instance.data.get(\"subset\")\n family = instance.data[\"family\"]\n\n self.log.debug(\"creating staging dir\")\n staging_dir = self.staging_dir(instance)\n\n transfers = list()\n if \"transfers\" not in instance.data:\n instance.data[\"transfers\"] = list()\n\n ext = \"json\"\n file = subset + \".\" + ext\n\n # when instance is created during collection part\n resources_dir = instance.data[\"resourcesDir\"]\n\n # change paths in effects to files\n for k, effect in effects.items():\n if \"assignTo\" in k:\n continue\n trn = self.copy_linked_files(effect, resources_dir)\n if trn:\n transfers.append((trn[0], trn[1]))\n\n instance.data[\"transfers\"].extend(transfers)\n self.log.debug(\"_ transfers: `{}`\".format(\n instance.data[\"transfers\"]))\n\n # create representations\n instance.data[\"representations\"] = list()\n\n transfer_data = [\n \"handleStart\", \"handleEnd\",\n \"sourceStart\", \"sourceStartH\", \"sourceEnd\", \"sourceEndH\",\n \"frameStart\", \"frameEnd\",\n \"clipIn\", \"clipOut\", \"clipInH\", \"clipOutH\",\n \"asset\", \"version\"\n ]\n\n # pass data to version\n version_data = dict()\n version_data.update({k: instance.data[k] for k in transfer_data})\n\n # add to data of representation\n version_data.update({\n \"colorspace\": item.sourceMediaColourTransform(),\n \"colorspaceScript\": instance.context.data[\"colorspace\"],\n \"families\": [family, \"plate\"],\n \"subset\": subset,\n \"fps\": instance.context.data[\"fps\"]\n })\n instance.data[\"versionData\"] = version_data\n\n representation = {\n 'files': file,\n 'stagingDir': staging_dir,\n 'name': family + ext.title(),\n 'ext': ext\n }\n instance.data[\"representations\"].append(representation)\n\n self.log.debug(\"_ representations: `{}`\".format(\n instance.data[\"representations\"]))\n\n self.log.debug(\"_ version_data: `{}`\".format(\n instance.data[\"versionData\"]))\n\n with open(os.path.join(staging_dir, file), \"w\") as outfile:\n outfile.write(json.dumps(effects, indent=4, sort_keys=True))\n\n def copy_linked_files(self, effect, dst_dir):\n for k, v in effect[\"node\"].items():\n if k in \"file\" and v != '':\n base_name = os.path.basename(v)\n dst = os.path.join(dst_dir, base_name).replace(\"\\\\\", \"/\")\n\n # add it to the json\n effect[\"node\"][k] = dst\n return (v, dst)\n"
},
{
"path": "openpype/hosts/hiero/plugins/publish/extract_frames.py",
"content": "import os\nimport pyblish.api\n\nfrom openpype.lib import (\n get_oiio_tool_args,\n run_subprocess,\n)\nfrom openpype.pipeline import publish\n\n\nclass ExtractFrames(publish.Extractor):\n \"\"\"Extracts frames\"\"\"\n\n order = pyblish.api.ExtractorOrder\n label = \"Extract Frames\"\n hosts = [\"hiero\"]\n families = [\"frame\"]\n movie_extensions = [\"mov\", \"mp4\"]\n\n def process(self, instance):\n oiio_tool_args = get_oiio_tool_args(\"oiiotool\")\n staging_dir = self.staging_dir(instance)\n output_template = os.path.join(staging_dir, instance.data[\"name\"])\n sequence = instance.context.data[\"activeTimeline\"]\n\n files = []\n for frame in instance.data[\"frames\"]:\n track_item = sequence.trackItemAt(frame)\n media_source = track_item.source().mediaSource()\n input_path = media_source.fileinfos()[0].filename()\n input_frame = (\n track_item.mapTimelineToSource(frame) +\n track_item.source().mediaSource().startTime()\n )\n output_ext = instance.data[\"format\"]\n output_path = output_template\n output_path += \".{:04d}.{}\".format(int(frame), output_ext)\n\n args = list(oiio_tool_args)\n\n ext = os.path.splitext(input_path)[1][1:]\n if ext in self.movie_extensions:\n args.extend([\"--subimage\", str(int(input_frame))])\n else:\n args.extend([\"--frames\", str(int(input_frame))])\n\n if ext == \"exr\":\n args.extend([\"--powc\", \"0.45,0.45,0.45,1.0\"])\n\n args.extend([input_path, \"-o\", output_path])\n output = run_subprocess(args)\n\n failed_output = \"oiiotool produced no output.\"\n if failed_output in output:\n raise ValueError(\n \"oiiotool processing failed. Args: {}\".format(args)\n )\n\n files.append(output_path)\n\n # Feedback to user because \"oiiotool\" can make the publishing\n # appear unresponsive.\n self.log.info(\n \"Processed {} of {} frames\".format(\n instance.data[\"frames\"].index(frame) + 1,\n len(instance.data[\"frames\"])\n )\n )\n\n if len(files) == 1:\n instance.data[\"representations\"] = [\n {\n \"name\": output_ext,\n \"ext\": output_ext,\n \"files\": os.path.basename(files[0]),\n \"stagingDir\": staging_dir\n }\n ]\n else:\n instance.data[\"representations\"] = [\n {\n \"name\": output_ext,\n \"ext\": output_ext,\n \"files\": [os.path.basename(x) for x in files],\n \"stagingDir\": staging_dir\n }\n ]\n"
},
{
"path": "openpype/hosts/hiero/plugins/publish/extract_thumbnail.py",
"content": "import os\nimport pyblish.api\n\nfrom openpype.pipeline import publish\n\n\nclass ExtractThumnail(publish.Extractor):\n \"\"\"\n Extractor for track item's tumnails\n \"\"\"\n\n label = \"Extract Thumnail\"\n order = pyblish.api.ExtractorOrder\n families = [\"plate\", \"take\"]\n hosts = [\"hiero\"]\n\n def process(self, instance):\n # create representation data\n if \"representations\" not in instance.data:\n instance.data[\"representations\"] = []\n\n staging_dir = self.staging_dir(instance)\n\n self.create_thumbnail(staging_dir, instance)\n\n def create_thumbnail(self, staging_dir, instance):\n track_item = instance.data[\"item\"]\n track_item_name = track_item.name()\n\n # frames\n duration = track_item.sourceDuration()\n frame_start = track_item.sourceIn()\n self.log.debug(\n \"__ frame_start: `{}`, duration: `{}`\".format(\n frame_start, duration))\n\n # get thumbnail frame from the middle\n thumb_frame = int(frame_start + (duration / 2))\n\n thumb_file = \"{}thumbnail{}{}\".format(\n track_item_name, thumb_frame, \".png\")\n thumb_path = os.path.join(staging_dir, thumb_file)\n\n thumbnail = track_item.thumbnail(thumb_frame, \"colour\").save(\n thumb_path,\n format='png'\n )\n self.log.debug(\n \"__ thumb_path: `{}`, frame: `{}`\".format(thumbnail, thumb_frame))\n\n self.log.info(\"Thumnail was generated to: {}\".format(thumb_path))\n thumb_representation = {\n 'files': thumb_file,\n 'stagingDir': staging_dir,\n 'name': \"thumbnail\",\n 'thumbnail': True,\n 'ext': \"png\"\n }\n instance.data[\"representations\"].append(\n thumb_representation)\n"
},
{
"path": "openpype/hosts/hiero/plugins/publish/integrate_version_up_workfile.py",
"content": "from pyblish import api\n\nfrom openpype.lib import version_up\n\n\nclass IntegrateVersionUpWorkfile(api.ContextPlugin):\n \"\"\"Save as new workfile version\"\"\"\n\n order = api.IntegratorOrder + 10.1\n label = \"Version-up Workfile\"\n hosts = [\"hiero\"]\n\n optional = True\n active = True\n\n def process(self, context):\n project = context.data[\"activeProject\"]\n path = context.data.get(\"currentFile\")\n new_path = version_up(path)\n\n if project:\n project.saveAs(new_path)\n\n self.log.info(\"Project workfile was versioned up\")\n"
},
{
"path": "openpype/hosts/hiero/plugins/publish/precollect_instances.py",
"content": "import pyblish\n\nfrom openpype import AYON_SERVER_ENABLED\nfrom openpype.pipeline.editorial import is_overlapping_otio_ranges\n\nfrom openpype.hosts.hiero import api as phiero\nfrom openpype.hosts.hiero.api.otio import hiero_export\n\nimport hiero\n# # developer reload modules\nfrom pprint import pformat\n\n\nclass PrecollectInstances(pyblish.api.ContextPlugin):\n \"\"\"Collect all Track items selection.\"\"\"\n\n order = pyblish.api.CollectorOrder - 0.49\n label = \"Precollect Instances\"\n hosts = [\"hiero\"]\n\n audio_track_items = []\n\n def process(self, context):\n self.otio_timeline = context.data[\"otioTimeline\"]\n timeline_selection = phiero.get_timeline_selection()\n selected_timeline_items = phiero.get_track_items(\n selection=timeline_selection,\n check_tagged=True,\n check_enabled=True\n )\n\n # only return enabled track items\n if not selected_timeline_items:\n selected_timeline_items = phiero.get_track_items(\n check_enabled=True, check_tagged=True)\n\n self.log.info(\n \"Processing enabled track items: {}\".format(\n selected_timeline_items))\n\n # add all tracks subtreck effect items to context\n all_tracks = hiero.ui.activeSequence().videoTracks()\n tracks_effect_items = self.collect_sub_track_items(all_tracks)\n context.data[\"tracksEffectItems\"] = tracks_effect_items\n\n # process all sellected timeline track items\n for track_item in selected_timeline_items:\n data = {}\n clip_name = track_item.name()\n source_clip = track_item.source()\n self.log.debug(\"clip_name: {}\".format(clip_name))\n\n # get openpype tag data\n tag_data = phiero.get_trackitem_openpype_data(track_item)\n self.log.debug(\"__ tag_data: {}\".format(pformat(tag_data)))\n\n if not tag_data:\n continue\n\n if tag_data.get(\"id\") != \"pyblish.avalon.instance\":\n continue\n\n # get clips subtracks and anotations\n annotations = self.clip_annotations(source_clip)\n subtracks = self.clip_subtrack(track_item)\n self.log.debug(\"Annotations: {}\".format(annotations))\n self.log.debug(\">> Subtracks: {}\".format(subtracks))\n\n # solve handles length\n tag_data[\"handleStart\"] = min(\n tag_data[\"handleStart\"], int(track_item.handleInLength()))\n tag_data[\"handleEnd\"] = min(\n tag_data[\"handleEnd\"], int(track_item.handleOutLength()))\n\n # add audio to families\n with_audio = False\n if tag_data.pop(\"audio\"):\n with_audio = True\n\n # add tag data to instance data\n data.update({\n k: v for k, v in tag_data.items()\n if k not in (\"id\", \"applieswhole\", \"label\")\n })\n\n asset, asset_name = self._get_asset_data(tag_data)\n\n subset = tag_data[\"subset\"]\n\n # insert family into families\n families = [str(f) for f in tag_data[\"families\"]]\n\n # form label\n label = \"{} -\".format(asset)\n if asset_name != clip_name:\n label += \" ({})\".format(clip_name)\n label += \" {}\".format(subset)\n\n data.update({\n \"name\": \"{}_{}\".format(asset, subset),\n \"label\": label,\n \"asset\": asset,\n \"asset_name\": asset_name,\n \"item\": track_item,\n \"families\": families,\n \"publish\": tag_data[\"publish\"],\n \"fps\": context.data[\"fps\"],\n\n # clip's effect\n \"clipEffectItems\": subtracks,\n \"clipAnnotations\": annotations,\n\n # add all additional tags\n \"tags\": phiero.get_track_item_tags(track_item),\n \"newAssetPublishing\": True\n })\n\n # otio clip data\n otio_data = self.get_otio_clip_instance_data(track_item) or {}\n self.log.debug(\"__ otio_data: {}\".format(pformat(otio_data)))\n data.update(otio_data)\n self.log.debug(\"__ data: {}\".format(pformat(data)))\n\n # add resolution\n self.get_resolution_to_data(data, context)\n\n # create instance\n instance = context.create_instance(**data)\n\n # add colorspace data\n instance.data.update({\n \"versionData\": {\n \"colorspace\": track_item.sourceMediaColourTransform(),\n }\n })\n\n # create shot instance for shot attributes create/update\n self.create_shot_instance(context, **data)\n\n self.log.info(\"Creating instance: {}\".format(instance))\n self.log.info(\n \"_ instance.data: {}\".format(pformat(instance.data)))\n\n if not with_audio:\n continue\n\n # create audio subset instance\n self.create_audio_instance(context, **data)\n\n # add audioReview attribute to plate instance data\n # if reviewTrack is on\n if tag_data.get(\"reviewTrack\") is not None:\n instance.data[\"reviewAudio\"] = True\n\n def get_resolution_to_data(self, data, context):\n assert data.get(\"otioClip\"), \"Missing `otioClip` data\"\n\n # solve source resolution option\n if data.get(\"sourceResolution\", None):\n otio_clip_metadata = data[\n \"otioClip\"].media_reference.metadata\n data.update({\n \"resolutionWidth\": otio_clip_metadata[\n \"openpype.source.width\"],\n \"resolutionHeight\": otio_clip_metadata[\n \"openpype.source.height\"],\n \"pixelAspect\": otio_clip_metadata[\n \"openpype.source.pixelAspect\"]\n })\n else:\n otio_tl_metadata = context.data[\"otioTimeline\"].metadata\n data.update({\n \"resolutionWidth\": otio_tl_metadata[\"openpype.timeline.width\"],\n \"resolutionHeight\": otio_tl_metadata[\n \"openpype.timeline.height\"],\n \"pixelAspect\": otio_tl_metadata[\n \"openpype.timeline.pixelAspect\"]\n })\n\n def create_shot_instance(self, context, **data):\n subset = \"shotMain\"\n master_layer = data.get(\"heroTrack\")\n hierarchy_data = data.get(\"hierarchyData\")\n item = data.get(\"item\")\n clip_name = item.name()\n\n if not master_layer:\n return\n\n if not hierarchy_data:\n return\n\n asset = data[\"asset\"]\n asset_name = data[\"asset_name\"]\n\n # insert family into families\n family = \"shot\"\n\n # form label\n label = \"{} -\".format(asset)\n if asset_name != clip_name:\n label += \" ({}) \".format(clip_name)\n label += \" {}\".format(subset)\n\n data.update({\n \"name\": \"{}_{}\".format(asset, subset),\n \"label\": label,\n \"subset\": subset,\n \"family\": family,\n \"families\": []\n })\n\n instance = context.create_instance(**data)\n self.log.info(\"Creating instance: {}\".format(instance))\n self.log.debug(\n \"_ instance.data: {}\".format(pformat(instance.data)))\n\n def _get_asset_data(self, data):\n folder_path = data.pop(\"folderPath\", None)\n\n if data.get(\"asset_name\"):\n asset_name = data[\"asset_name\"]\n else:\n asset_name = data[\"asset\"]\n\n # backward compatibility for clip tags\n # which are missing folderPath key\n # TODO remove this in future versions\n if not folder_path:\n hierarchy_path = data[\"hierarchy\"]\n folder_path = \"/{}/{}\".format(\n hierarchy_path,\n asset_name\n )\n\n if AYON_SERVER_ENABLED:\n asset = folder_path\n else:\n asset = asset_name\n\n return asset, asset_name\n\n def create_audio_instance(self, context, **data):\n subset = \"audioMain\"\n master_layer = data.get(\"heroTrack\")\n\n if not master_layer:\n return\n\n asset = data.get(\"asset\")\n item = data.get(\"item\")\n clip_name = item.name()\n\n # test if any audio clips\n if not self.test_any_audio(item):\n return\n\n asset = data[\"asset\"]\n asset_name = data[\"asset_name\"]\n\n # insert family into families\n family = \"audio\"\n\n # form label\n label = \"{} -\".format(asset)\n if asset_name != clip_name:\n label += \" ({}) \".format(clip_name)\n label += \" {}\".format(subset)\n\n data.update({\n \"name\": \"{}_{}\".format(asset, subset),\n \"label\": label,\n \"subset\": subset,\n \"family\": family,\n \"families\": [\"clip\"]\n })\n # remove review track attr if any\n data.pop(\"reviewTrack\")\n\n # create instance\n instance = context.create_instance(**data)\n self.log.info(\"Creating instance: {}\".format(instance))\n self.log.debug(\n \"_ instance.data: {}\".format(pformat(instance.data)))\n\n def test_any_audio(self, track_item):\n # collect all audio tracks to class variable\n if not self.audio_track_items:\n for otio_clip in self.otio_timeline.each_clip():\n if otio_clip.parent().kind != \"Audio\":\n continue\n self.audio_track_items.append(otio_clip)\n\n # get track item timeline range\n timeline_range = self.create_otio_time_range_from_timeline_item_data(\n track_item)\n\n # loop through audio track items and search for overlapping clip\n for otio_audio in self.audio_track_items:\n parent_range = otio_audio.range_in_parent()\n\n # if any overaling clip found then return True\n if is_overlapping_otio_ranges(\n parent_range, timeline_range, strict=False):\n return True\n\n def get_otio_clip_instance_data(self, track_item):\n \"\"\"\n Return otio objects for timeline, track and clip\n\n Args:\n timeline_item_data (dict): timeline_item_data from list returned by\n resolve.get_current_timeline_items()\n otio_timeline (otio.schema.Timeline): otio object\n\n Returns:\n dict: otio clip object\n\n \"\"\"\n ti_track_name = track_item.parent().name()\n timeline_range = self.create_otio_time_range_from_timeline_item_data(\n track_item)\n for otio_clip in self.otio_timeline.each_clip():\n track_name = otio_clip.parent().name\n parent_range = otio_clip.range_in_parent()\n if ti_track_name != track_name:\n continue\n if otio_clip.name != track_item.name():\n continue\n self.log.debug(\"__ parent_range: {}\".format(parent_range))\n self.log.debug(\"__ timeline_range: {}\".format(timeline_range))\n if is_overlapping_otio_ranges(\n parent_range, timeline_range, strict=True):\n\n # add pypedata marker to otio_clip metadata\n for marker in otio_clip.markers:\n if phiero.OPENPYPE_TAG_NAME in marker.name:\n otio_clip.metadata.update(marker.metadata)\n return {\"otioClip\": otio_clip}\n\n return None\n\n @staticmethod\n def create_otio_time_range_from_timeline_item_data(track_item):\n timeline = phiero.get_current_sequence()\n frame_start = int(track_item.timelineIn())\n frame_duration = int(track_item.duration())\n fps = timeline.framerate().toFloat()\n\n return hiero_export.create_otio_time_range(\n frame_start, frame_duration, fps)\n\n def collect_sub_track_items(self, tracks):\n \"\"\"\n Returns dictionary with track index as key and list of subtracks\n \"\"\"\n # collect all subtrack items\n sub_track_items = {}\n for track in tracks:\n items = track.items()\n\n effet_items = track.subTrackItems()\n\n # skip if no clips on track > need track with effect only\n if not effet_items:\n continue\n\n # skip all disabled tracks\n if not track.isEnabled():\n continue\n\n track_index = track.trackIndex()\n _sub_track_items = phiero.flatten(effet_items)\n\n _sub_track_items = list(_sub_track_items)\n # continue only if any subtrack items are collected\n if not _sub_track_items:\n continue\n\n enabled_sti = []\n # loop all found subtrack items and check if they are enabled\n for _sti in _sub_track_items:\n # checking if not enabled\n if not _sti.isEnabled():\n continue\n if isinstance(_sti, hiero.core.Annotation):\n continue\n # collect the subtrack item\n enabled_sti.append(_sti)\n\n # continue only if any subtrack items are collected\n if not enabled_sti:\n continue\n\n # add collection of subtrackitems to dict\n sub_track_items[track_index] = enabled_sti\n\n return sub_track_items\n\n @staticmethod\n def clip_annotations(clip):\n \"\"\"\n Returns list of Clip's hiero.core.Annotation\n \"\"\"\n annotations = []\n subTrackItems = phiero.flatten(clip.subTrackItems())\n annotations += [item for item in subTrackItems if isinstance(\n item, hiero.core.Annotation)]\n return annotations\n\n @staticmethod\n def clip_subtrack(clip):\n \"\"\"\n Returns list of Clip's hiero.core.SubTrackItem\n \"\"\"\n subtracks = []\n subTrackItems = phiero.flatten(clip.parent().subTrackItems())\n for item in subTrackItems:\n if \"TimeWarp\" in item.name():\n continue\n # avoid all anotation\n if isinstance(item, hiero.core.Annotation):\n continue\n # # avoid all not anaibled\n if not item.isEnabled():\n continue\n subtracks.append(item)\n return subtracks\n"
},
{
"path": "openpype/hosts/hiero/plugins/publish/precollect_workfile.py",
"content": "import os\nimport tempfile\nfrom pprint import pformat\n\nimport pyblish.api\nfrom qtpy.QtGui import QPixmap\n\nimport hiero.ui\n\nfrom openpype import AYON_SERVER_ENABLED\nfrom openpype.hosts.hiero.api.otio import hiero_export\n\n\nclass PrecollectWorkfile(pyblish.api.ContextPlugin):\n \"\"\"Inject the current working file into context\"\"\"\n\n label = \"Precollect Workfile\"\n order = pyblish.api.CollectorOrder - 0.491\n\n def process(self, context):\n asset = context.data[\"asset\"]\n asset_name = asset\n if AYON_SERVER_ENABLED:\n asset_name = asset_name.split(\"/\")[-1]\n\n active_timeline = hiero.ui.activeSequence()\n project = active_timeline.project()\n fps = active_timeline.framerate().toFloat()\n\n # adding otio timeline to context\n otio_timeline = hiero_export.create_otio_timeline()\n\n # get workfile thumbnail paths\n tmp_staging = tempfile.mkdtemp(prefix=\"pyblish_tmp_\")\n thumbnail_name = \"workfile_thumbnail.png\"\n thumbnail_path = os.path.join(tmp_staging, thumbnail_name)\n\n # search for all windows with name of actual sequence\n _windows = [w for w in hiero.ui.windowManager().windows()\n if active_timeline.name() in w.windowTitle()]\n\n # export window to thumb path\n QPixmap.grabWidget(_windows[-1]).save(thumbnail_path, 'png')\n\n # thumbnail\n thumb_representation = {\n 'files': thumbnail_name,\n 'stagingDir': tmp_staging,\n 'name': \"thumbnail\",\n 'thumbnail': True,\n 'ext': \"png\"\n }\n\n # get workfile paths\n current_file = project.path()\n staging_dir, base_name = os.path.split(current_file)\n\n # creating workfile representation\n workfile_representation = {\n 'name': 'hrox',\n 'ext': 'hrox',\n 'files': base_name,\n \"stagingDir\": staging_dir,\n }\n family = \"workfile\"\n instance_data = {\n \"label\": \"{} - {}Main\".format(\n asset, family),\n \"name\": \"{}_{}\".format(asset_name, family),\n \"asset\": context.data[\"asset\"],\n # TODO use 'get_subset_name'\n \"subset\": \"{}{}Main\".format(asset_name, family.capitalize()),\n \"item\": project,\n \"family\": family,\n \"families\": [],\n \"representations\": [workfile_representation, thumb_representation]\n }\n\n # create instance with workfile\n instance = context.create_instance(**instance_data)\n\n # update context with main project attributes\n context_data = {\n \"activeProject\": project,\n \"activeTimeline\": active_timeline,\n \"otioTimeline\": otio_timeline,\n \"currentFile\": current_file,\n \"colorspace\": self.get_colorspace(project),\n \"fps\": fps\n }\n self.log.debug(\"__ context_data: {}\".format(pformat(context_data)))\n context.data.update(context_data)\n\n self.log.info(\"Creating instance: {}\".format(instance))\n self.log.debug(\"__ instance.data: {}\".format(pformat(instance.data)))\n self.log.debug(\"__ context_data: {}\".format(pformat(context_data)))\n\n def get_colorspace(self, project):\n # get workfile's colorspace properties\n return {\n \"useOCIOEnvironmentOverride\": project.useOCIOEnvironmentOverride(),\n \"lutSetting16Bit\": project.lutSetting16Bit(),\n \"lutSetting8Bit\": project.lutSetting8Bit(),\n \"lutSettingFloat\": project.lutSettingFloat(),\n \"lutSettingLog\": project.lutSettingLog(),\n \"lutSettingViewer\": project.lutSettingViewer(),\n \"lutSettingWorkingSpace\": project.lutSettingWorkingSpace(),\n \"lutUseOCIOForExport\": project.lutUseOCIOForExport(),\n \"ocioConfigName\": project.ocioConfigName(),\n \"ocioConfigPath\": project.ocioConfigPath()\n }\n"
},
{
"path": "openpype/hosts/hiero/plugins/publish_old_workflow/collect_assetbuilds.py",
"content": "from pyblish import api\n\nfrom openpype.client import get_assets, get_asset_name_identifier\n\n\nclass CollectAssetBuilds(api.ContextPlugin):\n \"\"\"Collect asset from tags.\n\n Tag is expected to have name of the asset and metadata:\n {\n \"family\": \"assetbuild\"\n }\n \"\"\"\n\n # Run just after CollectClip\n order = api.CollectorOrder + 0.02\n label = \"Collect AssetBuilds\"\n hosts = [\"hiero\"]\n\n def process(self, context):\n project_name = context.data[\"projectName\"]\n asset_builds = {}\n for asset_doc in get_assets(project_name):\n if asset_doc[\"data\"].get(\"entityType\") != \"AssetBuild\":\n continue\n\n asset_name = get_asset_name_identifier(asset_doc)\n self.log.debug(\"Found \\\"{}\\\" in database.\".format(asset_doc))\n asset_builds[asset_name] = asset_doc\n\n for instance in context:\n if instance.data[\"family\"] != \"clip\":\n continue\n\n # Exclude non-tagged instances.\n tagged = False\n asset_names = []\n\n for tag in instance.data[\"tags\"]:\n t_metadata = dict(tag.metadata())\n t_family = t_metadata.get(\"tag.family\", \"\")\n\n if t_family.lower() == \"assetbuild\":\n asset_names.append(tag[\"name\"])\n tagged = True\n\n if not tagged:\n self.log.debug(\n \"Skipping \\\"{}\\\" because its not tagged with \"\n \"\\\"assetbuild\\\"\".format(instance)\n )\n continue\n\n # Collect asset builds.\n data = {\"assetbuilds\": []}\n for name in asset_names:\n data[\"assetbuilds\"].append(asset_builds[name])\n self.log.debug(\n \"Found asset builds: {}\".format(data[\"assetbuilds\"])\n )\n\n instance.data.update(data)\n"
},
{
"path": "openpype/hosts/hiero/plugins/publish_old_workflow/collect_tag_comments.py",
"content": "from pyblish import api\n\n\nclass CollectClipTagComments(api.InstancePlugin):\n \"\"\"Collect comments from tags on selected track items and their sources.\"\"\"\n\n order = api.CollectorOrder + 0.013\n label = \"Collect Comments\"\n hosts = [\"hiero\"]\n families = [\"clip\"]\n\n def process(self, instance):\n # Collect comments.\n instance.data[\"comments\"] = []\n\n # Exclude non-tagged instances.\n for tag in instance.data[\"tags\"]:\n if tag[\"name\"].lower() == \"comment\":\n instance.data[\"comments\"].append(\n tag[\"metadata\"][\"tag.note\"]\n )\n\n # Find tags on the source clip.\n tags = instance.data[\"item\"].source().tags()\n for tag in tags:\n if tag.name().lower() == \"comment\":\n instance.data[\"comments\"].append(\n tag.metadata().dict()[\"tag.note\"]\n )\n\n # Update label with comments counter.\n instance.data[\"label\"] = \"{} - comments:{}\".format(\n instance.data[\"label\"],\n len(instance.data[\"comments\"])\n )\n"
},
{
"path": "openpype/hosts/hiero/plugins/publish_old_workflow/precollect_retime.py",
"content": "from pyblish import api\nimport hiero\nimport math\nfrom openpype.hosts.hiero.api.otio.hiero_export import create_otio_time_range\n\nclass PrecollectRetime(api.InstancePlugin):\n \"\"\"Calculate Retiming of selected track items.\"\"\"\n\n order = api.CollectorOrder - 0.578\n label = \"Precollect Retime\"\n hosts = [\"hiero\"]\n families = ['retime_']\n\n def process(self, instance):\n if not instance.data.get(\"versionData\"):\n instance.data[\"versionData\"] = {}\n\n # get basic variables\n otio_clip = instance.data[\"otioClip\"]\n\n source_range = otio_clip.source_range\n oc_source_fps = source_range.start_time.rate\n oc_source_in = source_range.start_time.value\n\n handle_start = instance.data[\"handleStart\"]\n handle_end = instance.data[\"handleEnd\"]\n frame_start = instance.data[\"frameStart\"]\n\n track_item = instance.data[\"item\"]\n\n # define basic clip frame range variables\n timeline_in = int(track_item.timelineIn())\n timeline_out = int(track_item.timelineOut())\n source_in = int(track_item.sourceIn())\n source_out = int(track_item.sourceOut())\n speed = track_item.playbackSpeed()\n\n # calculate available material before retime\n available_in = int(track_item.handleInLength() * speed)\n available_out = int(track_item.handleOutLength() * speed)\n\n self.log.debug((\n \"_BEFORE: \\n timeline_in: `{0}`,\\n timeline_out: `{1}`, \\n \"\n \"source_in: `{2}`,\\n source_out: `{3}`,\\n speed: `{4}`,\\n \"\n \"handle_start: `{5}`,\\n handle_end: `{6}`\").format(\n timeline_in,\n timeline_out,\n source_in,\n source_out,\n speed,\n handle_start,\n handle_end\n ))\n\n # loop within subtrack items\n time_warp_nodes = []\n source_in_change = 0\n source_out_change = 0\n for s_track_item in track_item.linkedItems():\n if isinstance(s_track_item, hiero.core.EffectTrackItem) \\\n and \"TimeWarp\" in s_track_item.node().Class():\n\n # adding timewarp attribute to instance\n time_warp_nodes = []\n\n # ignore item if not enabled\n if s_track_item.isEnabled():\n node = s_track_item.node()\n name = node[\"name\"].value()\n look_up = node[\"lookup\"].value()\n animated = node[\"lookup\"].isAnimated()\n if animated:\n look_up = [\n ((node[\"lookup\"].getValueAt(i)) - i)\n for i in range(\n (timeline_in - handle_start),\n (timeline_out + handle_end) + 1)\n ]\n # calculate difference\n diff_in = (node[\"lookup\"].getValueAt(\n timeline_in)) - timeline_in\n diff_out = (node[\"lookup\"].getValueAt(\n timeline_out)) - timeline_out\n\n # calculate source\n source_in_change += diff_in\n source_out_change += diff_out\n\n # calculate speed\n speed_in = (node[\"lookup\"].getValueAt(timeline_in) / (\n float(timeline_in) * .01)) * .01\n speed_out = (node[\"lookup\"].getValueAt(timeline_out) / (\n float(timeline_out) * .01)) * .01\n\n # calculate handles\n handle_start = int(\n math.ceil(\n (handle_start * speed_in * 1000) / 1000.0)\n )\n\n handle_end = int(\n math.ceil(\n (handle_end * speed_out * 1000) / 1000.0)\n )\n self.log.debug(\n (\"diff_in, diff_out\", diff_in, diff_out))\n self.log.debug(\n (\"source_in_change, source_out_change\",\n source_in_change, source_out_change))\n\n time_warp_nodes.append({\n \"Class\": \"TimeWarp\",\n \"name\": name,\n \"lookup\": look_up\n })\n\n self.log.debug(\n \"timewarp source in changes: in {}, out {}\".format(\n source_in_change, source_out_change))\n\n # recalculate handles by the speed\n handle_start *= speed\n handle_end *= speed\n self.log.debug(\"speed: handle_start: '{0}', handle_end: '{1}'\".format(\n handle_start, handle_end))\n\n # recalculate source with timewarp and by the speed\n source_in += int(source_in_change)\n source_out += int(source_out_change * speed)\n\n source_in_h = int(source_in - math.ceil(\n (handle_start * 1000) / 1000.0))\n source_out_h = int(source_out + math.ceil(\n (handle_end * 1000) / 1000.0))\n\n self.log.debug(\n \"retimed: source_in_h: '{0}', source_out_h: '{1}'\".format(\n source_in_h, source_out_h))\n\n # add all data to Instance\n instance.data[\"handleStart\"] = handle_start\n instance.data[\"handleEnd\"] = handle_end\n instance.data[\"sourceIn\"] = source_in\n instance.data[\"sourceOut\"] = source_out\n instance.data[\"sourceInH\"] = source_in_h\n instance.data[\"sourceOutH\"] = source_out_h\n instance.data[\"speed\"] = speed\n\n source_handle_start = source_in_h - source_in\n # frame_start = instance.data[\"frameStart\"] + source_handle_start\n duration = source_out_h - source_in_h\n frame_end = int(frame_start + duration - (handle_start + handle_end))\n\n instance.data[\"versionData\"].update({\n \"retime\": True,\n \"speed\": speed,\n \"timewarps\": time_warp_nodes,\n \"frameStart\": frame_start,\n \"frameEnd\": frame_end,\n \"handleStart\": abs(source_handle_start),\n \"handleEnd\": source_out_h - source_out\n })\n self.log.debug(\"versionData: {}\".format(instance.data[\"versionData\"]))\n self.log.debug(\"sourceIn: {}\".format(instance.data[\"sourceIn\"]))\n self.log.debug(\"sourceOut: {}\".format(instance.data[\"sourceOut\"]))\n self.log.debug(\"speed: {}\".format(instance.data[\"speed\"]))\n\n # change otio clip data\n instance.data[\"otioClip\"].source_range = create_otio_time_range(\n oc_source_in, (source_out - source_in + 1), oc_source_fps)\n self.log.debug(\"otioClip: {}\".format(instance.data[\"otioClip\"]))\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/__init__.py",
"content": "# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n# Copyright 2007 Google Inc. All Rights Reserved.\n\n__version__ = '3.20.1'\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/any_pb2.py",
"content": "# -*- coding: utf-8 -*-\n# Generated by the protocol buffer compiler. DO NOT EDIT!\n# source: google/protobuf/any.proto\n\"\"\"Generated protocol buffer code.\"\"\"\nfrom google.protobuf.internal import builder as _builder\nfrom google.protobuf import descriptor as _descriptor\nfrom google.protobuf import descriptor_pool as _descriptor_pool\nfrom google.protobuf import symbol_database as _symbol_database\n# @@protoc_insertion_point(imports)\n\n_sym_db = _symbol_database.Default()\n\n\n\n\nDESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\\n\\x19google/protobuf/any.proto\\x12\\x0fgoogle.protobuf\\\"&\\n\\x03\\x41ny\\x12\\x10\\n\\x08type_url\\x18\\x01 \\x01(\\t\\x12\\r\\n\\x05value\\x18\\x02 \\x01(\\x0c\\x42v\\n\\x13\\x63om.google.protobufB\\x08\\x41nyProtoP\\x01Z,google.golang.org/protobuf/types/known/anypb\\xa2\\x02\\x03GPB\\xaa\\x02\\x1eGoogle.Protobuf.WellKnownTypesb\\x06proto3')\n\n_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())\n_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.any_pb2', globals())\nif _descriptor._USE_C_DESCRIPTORS == False:\n\n DESCRIPTOR._options = None\n DESCRIPTOR._serialized_options = b'\\n\\023com.google.protobufB\\010AnyProtoP\\001Z,google.golang.org/protobuf/types/known/anypb\\242\\002\\003GPB\\252\\002\\036Google.Protobuf.WellKnownTypes'\n _ANY._serialized_start=46\n _ANY._serialized_end=84\n# @@protoc_insertion_point(module_scope)\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/api_pb2.py",
"content": "# -*- coding: utf-8 -*-\n# Generated by the protocol buffer compiler. DO NOT EDIT!\n# source: google/protobuf/api.proto\n\"\"\"Generated protocol buffer code.\"\"\"\nfrom google.protobuf.internal import builder as _builder\nfrom google.protobuf import descriptor as _descriptor\nfrom google.protobuf import descriptor_pool as _descriptor_pool\nfrom google.protobuf import symbol_database as _symbol_database\n# @@protoc_insertion_point(imports)\n\n_sym_db = _symbol_database.Default()\n\n\nfrom google.protobuf import source_context_pb2 as google_dot_protobuf_dot_source__context__pb2\nfrom google.protobuf import type_pb2 as google_dot_protobuf_dot_type__pb2\n\n\nDESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\\n\\x19google/protobuf/api.proto\\x12\\x0fgoogle.protobuf\\x1a$google/protobuf/source_context.proto\\x1a\\x1agoogle/protobuf/type.proto\\\"\\x81\\x02\\n\\x03\\x41pi\\x12\\x0c\\n\\x04name\\x18\\x01 \\x01(\\t\\x12(\\n\\x07methods\\x18\\x02 \\x03(\\x0b\\x32\\x17.google.protobuf.Method\\x12(\\n\\x07options\\x18\\x03 \\x03(\\x0b\\x32\\x17.google.protobuf.Option\\x12\\x0f\\n\\x07version\\x18\\x04 \\x01(\\t\\x12\\x36\\n\\x0esource_context\\x18\\x05 \\x01(\\x0b\\x32\\x1e.google.protobuf.SourceContext\\x12&\\n\\x06mixins\\x18\\x06 \\x03(\\x0b\\x32\\x16.google.protobuf.Mixin\\x12\\'\\n\\x06syntax\\x18\\x07 \\x01(\\x0e\\x32\\x17.google.protobuf.Syntax\\\"\\xd5\\x01\\n\\x06Method\\x12\\x0c\\n\\x04name\\x18\\x01 \\x01(\\t\\x12\\x18\\n\\x10request_type_url\\x18\\x02 \\x01(\\t\\x12\\x19\\n\\x11request_streaming\\x18\\x03 \\x01(\\x08\\x12\\x19\\n\\x11response_type_url\\x18\\x04 \\x01(\\t\\x12\\x1a\\n\\x12response_streaming\\x18\\x05 \\x01(\\x08\\x12(\\n\\x07options\\x18\\x06 \\x03(\\x0b\\x32\\x17.google.protobuf.Option\\x12\\'\\n\\x06syntax\\x18\\x07 \\x01(\\x0e\\x32\\x17.google.protobuf.Syntax\\\"#\\n\\x05Mixin\\x12\\x0c\\n\\x04name\\x18\\x01 \\x01(\\t\\x12\\x0c\\n\\x04root\\x18\\x02 \\x01(\\tBv\\n\\x13\\x63om.google.protobufB\\x08\\x41piProtoP\\x01Z,google.golang.org/protobuf/types/known/apipb\\xa2\\x02\\x03GPB\\xaa\\x02\\x1eGoogle.Protobuf.WellKnownTypesb\\x06proto3')\n\n_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())\n_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.api_pb2', globals())\nif _descriptor._USE_C_DESCRIPTORS == False:\n\n DESCRIPTOR._options = None\n DESCRIPTOR._serialized_options = b'\\n\\023com.google.protobufB\\010ApiProtoP\\001Z,google.golang.org/protobuf/types/known/apipb\\242\\002\\003GPB\\252\\002\\036Google.Protobuf.WellKnownTypes'\n _API._serialized_start=113\n _API._serialized_end=370\n _METHOD._serialized_start=373\n _METHOD._serialized_end=586\n _MIXIN._serialized_start=588\n _MIXIN._serialized_end=623\n# @@protoc_insertion_point(module_scope)\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/compiler/__init__.py",
"content": ""
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/compiler/plugin_pb2.py",
"content": "# -*- coding: utf-8 -*-\n# Generated by the protocol buffer compiler. DO NOT EDIT!\n# source: google/protobuf/compiler/plugin.proto\n\"\"\"Generated protocol buffer code.\"\"\"\nfrom google.protobuf.internal import builder as _builder\nfrom google.protobuf import descriptor as _descriptor\nfrom google.protobuf import descriptor_pool as _descriptor_pool\nfrom google.protobuf import symbol_database as _symbol_database\n# @@protoc_insertion_point(imports)\n\n_sym_db = _symbol_database.Default()\n\n\nfrom google.protobuf import descriptor_pb2 as google_dot_protobuf_dot_descriptor__pb2\n\n\nDESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\\n%google/protobuf/compiler/plugin.proto\\x12\\x18google.protobuf.compiler\\x1a google/protobuf/descriptor.proto\\\"F\\n\\x07Version\\x12\\r\\n\\x05major\\x18\\x01 \\x01(\\x05\\x12\\r\\n\\x05minor\\x18\\x02 \\x01(\\x05\\x12\\r\\n\\x05patch\\x18\\x03 \\x01(\\x05\\x12\\x0e\\n\\x06suffix\\x18\\x04 \\x01(\\t\\\"\\xba\\x01\\n\\x14\\x43odeGeneratorRequest\\x12\\x18\\n\\x10\\x66ile_to_generate\\x18\\x01 \\x03(\\t\\x12\\x11\\n\\tparameter\\x18\\x02 \\x01(\\t\\x12\\x38\\n\\nproto_file\\x18\\x0f \\x03(\\x0b\\x32$.google.protobuf.FileDescriptorProto\\x12;\\n\\x10\\x63ompiler_version\\x18\\x03 \\x01(\\x0b\\x32!.google.protobuf.compiler.Version\\\"\\xc1\\x02\\n\\x15\\x43odeGeneratorResponse\\x12\\r\\n\\x05\\x65rror\\x18\\x01 \\x01(\\t\\x12\\x1a\\n\\x12supported_features\\x18\\x02 \\x01(\\x04\\x12\\x42\\n\\x04\\x66ile\\x18\\x0f \\x03(\\x0b\\x32\\x34.google.protobuf.compiler.CodeGeneratorResponse.File\\x1a\\x7f\\n\\x04\\x46ile\\x12\\x0c\\n\\x04name\\x18\\x01 \\x01(\\t\\x12\\x17\\n\\x0finsertion_point\\x18\\x02 \\x01(\\t\\x12\\x0f\\n\\x07\\x63ontent\\x18\\x0f \\x01(\\t\\x12?\\n\\x13generated_code_info\\x18\\x10 \\x01(\\x0b\\x32\\\".google.protobuf.GeneratedCodeInfo\\\"8\\n\\x07\\x46\\x65\\x61ture\\x12\\x10\\n\\x0c\\x46\\x45\\x41TURE_NONE\\x10\\x00\\x12\\x1b\\n\\x17\\x46\\x45\\x41TURE_PROTO3_OPTIONAL\\x10\\x01\\x42W\\n\\x1c\\x63om.google.protobuf.compilerB\\x0cPluginProtosZ)google.golang.org/protobuf/types/pluginpb')\n\n_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())\n_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.compiler.plugin_pb2', globals())\nif _descriptor._USE_C_DESCRIPTORS == False:\n\n DESCRIPTOR._options = None\n DESCRIPTOR._serialized_options = b'\\n\\034com.google.protobuf.compilerB\\014PluginProtosZ)google.golang.org/protobuf/types/pluginpb'\n _VERSION._serialized_start=101\n _VERSION._serialized_end=171\n _CODEGENERATORREQUEST._serialized_start=174\n _CODEGENERATORREQUEST._serialized_end=360\n _CODEGENERATORRESPONSE._serialized_start=363\n _CODEGENERATORRESPONSE._serialized_end=684\n _CODEGENERATORRESPONSE_FILE._serialized_start=499\n _CODEGENERATORRESPONSE_FILE._serialized_end=626\n _CODEGENERATORRESPONSE_FEATURE._serialized_start=628\n _CODEGENERATORRESPONSE_FEATURE._serialized_end=684\n# @@protoc_insertion_point(module_scope)\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/descriptor.py",
"content": "# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n\"\"\"Descriptors essentially contain exactly the information found in a .proto\nfile, in types that make this information accessible in Python.\n\"\"\"\n\n__author__ = 'robinson@google.com (Will Robinson)'\n\nimport threading\nimport warnings\n\nfrom google.protobuf.internal import api_implementation\n\n_USE_C_DESCRIPTORS = False\nif api_implementation.Type() == 'cpp':\n # Used by MakeDescriptor in cpp mode\n import binascii\n import os\n from google.protobuf.pyext import _message\n _USE_C_DESCRIPTORS = True\n\n\nclass Error(Exception):\n \"\"\"Base error for this module.\"\"\"\n\n\nclass TypeTransformationError(Error):\n \"\"\"Error transforming between python proto type and corresponding C++ type.\"\"\"\n\n\nif _USE_C_DESCRIPTORS:\n # This metaclass allows to override the behavior of code like\n # isinstance(my_descriptor, FieldDescriptor)\n # and make it return True when the descriptor is an instance of the extension\n # type written in C++.\n class DescriptorMetaclass(type):\n def __instancecheck__(cls, obj):\n if super(DescriptorMetaclass, cls).__instancecheck__(obj):\n return True\n if isinstance(obj, cls._C_DESCRIPTOR_CLASS):\n return True\n return False\nelse:\n # The standard metaclass; nothing changes.\n DescriptorMetaclass = type\n\n\nclass _Lock(object):\n \"\"\"Wrapper class of threading.Lock(), which is allowed by 'with'.\"\"\"\n\n def __new__(cls):\n self = object.__new__(cls)\n self._lock = threading.Lock() # pylint: disable=protected-access\n return self\n\n def __enter__(self):\n self._lock.acquire()\n\n def __exit__(self, exc_type, exc_value, exc_tb):\n self._lock.release()\n\n\n_lock = threading.Lock()\n\n\ndef _Deprecated(name):\n if _Deprecated.count > 0:\n _Deprecated.count -= 1\n warnings.warn(\n 'Call to deprecated create function %s(). Note: Create unlinked '\n 'descriptors is going to go away. Please use get/find descriptors from '\n 'generated code or query the descriptor_pool.'\n % name,\n category=DeprecationWarning, stacklevel=3)\n\n\n# Deprecated warnings will print 100 times at most which should be enough for\n# users to notice and do not cause timeout.\n_Deprecated.count = 100\n\n\n_internal_create_key = object()\n\n\nclass DescriptorBase(metaclass=DescriptorMetaclass):\n\n \"\"\"Descriptors base class.\n\n This class is the base of all descriptor classes. It provides common options\n related functionality.\n\n Attributes:\n has_options: True if the descriptor has non-default options. Usually it\n is not necessary to read this -- just call GetOptions() which will\n happily return the default instance. However, it's sometimes useful\n for efficiency, and also useful inside the protobuf implementation to\n avoid some bootstrapping issues.\n \"\"\"\n\n if _USE_C_DESCRIPTORS:\n # The class, or tuple of classes, that are considered as \"virtual\n # subclasses\" of this descriptor class.\n _C_DESCRIPTOR_CLASS = ()\n\n def __init__(self, options, serialized_options, options_class_name):\n \"\"\"Initialize the descriptor given its options message and the name of the\n class of the options message. The name of the class is required in case\n the options message is None and has to be created.\n \"\"\"\n self._options = options\n self._options_class_name = options_class_name\n self._serialized_options = serialized_options\n\n # Does this descriptor have non-default options?\n self.has_options = (options is not None) or (serialized_options is not None)\n\n def _SetOptions(self, options, options_class_name):\n \"\"\"Sets the descriptor's options\n\n This function is used in generated proto2 files to update descriptor\n options. It must not be used outside proto2.\n \"\"\"\n self._options = options\n self._options_class_name = options_class_name\n\n # Does this descriptor have non-default options?\n self.has_options = options is not None\n\n def GetOptions(self):\n \"\"\"Retrieves descriptor options.\n\n This method returns the options set or creates the default options for the\n descriptor.\n \"\"\"\n if self._options:\n return self._options\n\n from google.protobuf import descriptor_pb2\n try:\n options_class = getattr(descriptor_pb2,\n self._options_class_name)\n except AttributeError:\n raise RuntimeError('Unknown options class name %s!' %\n (self._options_class_name))\n\n with _lock:\n if self._serialized_options is None:\n self._options = options_class()\n else:\n self._options = _ParseOptions(options_class(),\n self._serialized_options)\n\n return self._options\n\n\nclass _NestedDescriptorBase(DescriptorBase):\n \"\"\"Common class for descriptors that can be nested.\"\"\"\n\n def __init__(self, options, options_class_name, name, full_name,\n file, containing_type, serialized_start=None,\n serialized_end=None, serialized_options=None):\n \"\"\"Constructor.\n\n Args:\n options: Protocol message options or None\n to use default message options.\n options_class_name (str): The class name of the above options.\n name (str): Name of this protocol message type.\n full_name (str): Fully-qualified name of this protocol message type,\n which will include protocol \"package\" name and the name of any\n enclosing types.\n file (FileDescriptor): Reference to file info.\n containing_type: if provided, this is a nested descriptor, with this\n descriptor as parent, otherwise None.\n serialized_start: The start index (inclusive) in block in the\n file.serialized_pb that describes this descriptor.\n serialized_end: The end index (exclusive) in block in the\n file.serialized_pb that describes this descriptor.\n serialized_options: Protocol message serialized options or None.\n \"\"\"\n super(_NestedDescriptorBase, self).__init__(\n options, serialized_options, options_class_name)\n\n self.name = name\n # TODO(falk): Add function to calculate full_name instead of having it in\n # memory?\n self.full_name = full_name\n self.file = file\n self.containing_type = containing_type\n\n self._serialized_start = serialized_start\n self._serialized_end = serialized_end\n\n def CopyToProto(self, proto):\n \"\"\"Copies this to the matching proto in descriptor_pb2.\n\n Args:\n proto: An empty proto instance from descriptor_pb2.\n\n Raises:\n Error: If self couldn't be serialized, due to to few constructor\n arguments.\n \"\"\"\n if (self.file is not None and\n self._serialized_start is not None and\n self._serialized_end is not None):\n proto.ParseFromString(self.file.serialized_pb[\n self._serialized_start:self._serialized_end])\n else:\n raise Error('Descriptor does not contain serialization.')\n\n\nclass Descriptor(_NestedDescriptorBase):\n\n \"\"\"Descriptor for a protocol message type.\n\n Attributes:\n name (str): Name of this protocol message type.\n full_name (str): Fully-qualified name of this protocol message type,\n which will include protocol \"package\" name and the name of any\n enclosing types.\n containing_type (Descriptor): Reference to the descriptor of the type\n containing us, or None if this is top-level.\n fields (list[FieldDescriptor]): Field descriptors for all fields in\n this type.\n fields_by_number (dict(int, FieldDescriptor)): Same\n :class:`FieldDescriptor` objects as in :attr:`fields`, but indexed\n by \"number\" attribute in each FieldDescriptor.\n fields_by_name (dict(str, FieldDescriptor)): Same\n :class:`FieldDescriptor` objects as in :attr:`fields`, but indexed by\n \"name\" attribute in each :class:`FieldDescriptor`.\n nested_types (list[Descriptor]): Descriptor references\n for all protocol message types nested within this one.\n nested_types_by_name (dict(str, Descriptor)): Same Descriptor\n objects as in :attr:`nested_types`, but indexed by \"name\" attribute\n in each Descriptor.\n enum_types (list[EnumDescriptor]): :class:`EnumDescriptor` references\n for all enums contained within this type.\n enum_types_by_name (dict(str, EnumDescriptor)): Same\n :class:`EnumDescriptor` objects as in :attr:`enum_types`, but\n indexed by \"name\" attribute in each EnumDescriptor.\n enum_values_by_name (dict(str, EnumValueDescriptor)): Dict mapping\n from enum value name to :class:`EnumValueDescriptor` for that value.\n extensions (list[FieldDescriptor]): All extensions defined directly\n within this message type (NOT within a nested type).\n extensions_by_name (dict(str, FieldDescriptor)): Same FieldDescriptor\n objects as :attr:`extensions`, but indexed by \"name\" attribute of each\n FieldDescriptor.\n is_extendable (bool): Does this type define any extension ranges?\n oneofs (list[OneofDescriptor]): The list of descriptors for oneof fields\n in this message.\n oneofs_by_name (dict(str, OneofDescriptor)): Same objects as in\n :attr:`oneofs`, but indexed by \"name\" attribute.\n file (FileDescriptor): Reference to file descriptor.\n\n \"\"\"\n\n if _USE_C_DESCRIPTORS:\n _C_DESCRIPTOR_CLASS = _message.Descriptor\n\n def __new__(\n cls,\n name=None,\n full_name=None,\n filename=None,\n containing_type=None,\n fields=None,\n nested_types=None,\n enum_types=None,\n extensions=None,\n options=None,\n serialized_options=None,\n is_extendable=True,\n extension_ranges=None,\n oneofs=None,\n file=None, # pylint: disable=redefined-builtin\n serialized_start=None,\n serialized_end=None,\n syntax=None,\n create_key=None):\n _message.Message._CheckCalledFromGeneratedFile()\n return _message.default_pool.FindMessageTypeByName(full_name)\n\n # NOTE(tmarek): The file argument redefining a builtin is nothing we can\n # fix right now since we don't know how many clients already rely on the\n # name of the argument.\n def __init__(self, name, full_name, filename, containing_type, fields,\n nested_types, enum_types, extensions, options=None,\n serialized_options=None,\n is_extendable=True, extension_ranges=None, oneofs=None,\n file=None, serialized_start=None, serialized_end=None, # pylint: disable=redefined-builtin\n syntax=None, create_key=None):\n \"\"\"Arguments to __init__() are as described in the description\n of Descriptor fields above.\n\n Note that filename is an obsolete argument, that is not used anymore.\n Please use file.name to access this as an attribute.\n \"\"\"\n if create_key is not _internal_create_key:\n _Deprecated('Descriptor')\n\n super(Descriptor, self).__init__(\n options, 'MessageOptions', name, full_name, file,\n containing_type, serialized_start=serialized_start,\n serialized_end=serialized_end, serialized_options=serialized_options)\n\n # We have fields in addition to fields_by_name and fields_by_number,\n # so that:\n # 1. Clients can index fields by \"order in which they're listed.\"\n # 2. Clients can easily iterate over all fields with the terse\n # syntax: for f in descriptor.fields: ...\n self.fields = fields\n for field in self.fields:\n field.containing_type = self\n self.fields_by_number = dict((f.number, f) for f in fields)\n self.fields_by_name = dict((f.name, f) for f in fields)\n self._fields_by_camelcase_name = None\n\n self.nested_types = nested_types\n for nested_type in nested_types:\n nested_type.containing_type = self\n self.nested_types_by_name = dict((t.name, t) for t in nested_types)\n\n self.enum_types = enum_types\n for enum_type in self.enum_types:\n enum_type.containing_type = self\n self.enum_types_by_name = dict((t.name, t) for t in enum_types)\n self.enum_values_by_name = dict(\n (v.name, v) for t in enum_types for v in t.values)\n\n self.extensions = extensions\n for extension in self.extensions:\n extension.extension_scope = self\n self.extensions_by_name = dict((f.name, f) for f in extensions)\n self.is_extendable = is_extendable\n self.extension_ranges = extension_ranges\n self.oneofs = oneofs if oneofs is not None else []\n self.oneofs_by_name = dict((o.name, o) for o in self.oneofs)\n for oneof in self.oneofs:\n oneof.containing_type = self\n self.syntax = syntax or \"proto2\"\n\n @property\n def fields_by_camelcase_name(self):\n \"\"\"Same FieldDescriptor objects as in :attr:`fields`, but indexed by\n :attr:`FieldDescriptor.camelcase_name`.\n \"\"\"\n if self._fields_by_camelcase_name is None:\n self._fields_by_camelcase_name = dict(\n (f.camelcase_name, f) for f in self.fields)\n return self._fields_by_camelcase_name\n\n def EnumValueName(self, enum, value):\n \"\"\"Returns the string name of an enum value.\n\n This is just a small helper method to simplify a common operation.\n\n Args:\n enum: string name of the Enum.\n value: int, value of the enum.\n\n Returns:\n string name of the enum value.\n\n Raises:\n KeyError if either the Enum doesn't exist or the value is not a valid\n value for the enum.\n \"\"\"\n return self.enum_types_by_name[enum].values_by_number[value].name\n\n def CopyToProto(self, proto):\n \"\"\"Copies this to a descriptor_pb2.DescriptorProto.\n\n Args:\n proto: An empty descriptor_pb2.DescriptorProto.\n \"\"\"\n # This function is overridden to give a better doc comment.\n super(Descriptor, self).CopyToProto(proto)\n\n\n# TODO(robinson): We should have aggressive checking here,\n# for example:\n# * If you specify a repeated field, you should not be allowed\n# to specify a default value.\n# * [Other examples here as needed].\n#\n# TODO(robinson): for this and other *Descriptor classes, we\n# might also want to lock things down aggressively (e.g.,\n# prevent clients from setting the attributes). Having\n# stronger invariants here in general will reduce the number\n# of runtime checks we must do in reflection.py...\nclass FieldDescriptor(DescriptorBase):\n\n \"\"\"Descriptor for a single field in a .proto file.\n\n Attributes:\n name (str): Name of this field, exactly as it appears in .proto.\n full_name (str): Name of this field, including containing scope. This is\n particularly relevant for extensions.\n index (int): Dense, 0-indexed index giving the order that this\n field textually appears within its message in the .proto file.\n number (int): Tag number declared for this field in the .proto file.\n\n type (int): (One of the TYPE_* constants below) Declared type.\n cpp_type (int): (One of the CPPTYPE_* constants below) C++ type used to\n represent this field.\n\n label (int): (One of the LABEL_* constants below) Tells whether this\n field is optional, required, or repeated.\n has_default_value (bool): True if this field has a default value defined,\n otherwise false.\n default_value (Varies): Default value of this field. Only\n meaningful for non-repeated scalar fields. Repeated fields\n should always set this to [], and non-repeated composite\n fields should always set this to None.\n\n containing_type (Descriptor): Descriptor of the protocol message\n type that contains this field. Set by the Descriptor constructor\n if we're passed into one.\n Somewhat confusingly, for extension fields, this is the\n descriptor of the EXTENDED message, not the descriptor\n of the message containing this field. (See is_extension and\n extension_scope below).\n message_type (Descriptor): If a composite field, a descriptor\n of the message type contained in this field. Otherwise, this is None.\n enum_type (EnumDescriptor): If this field contains an enum, a\n descriptor of that enum. Otherwise, this is None.\n\n is_extension: True iff this describes an extension field.\n extension_scope (Descriptor): Only meaningful if is_extension is True.\n Gives the message that immediately contains this extension field.\n Will be None iff we're a top-level (file-level) extension field.\n\n options (descriptor_pb2.FieldOptions): Protocol message field options or\n None to use default field options.\n\n containing_oneof (OneofDescriptor): If the field is a member of a oneof\n union, contains its descriptor. Otherwise, None.\n\n file (FileDescriptor): Reference to file descriptor.\n \"\"\"\n\n # Must be consistent with C++ FieldDescriptor::Type enum in\n # descriptor.h.\n #\n # TODO(robinson): Find a way to eliminate this repetition.\n TYPE_DOUBLE = 1\n TYPE_FLOAT = 2\n TYPE_INT64 = 3\n TYPE_UINT64 = 4\n TYPE_INT32 = 5\n TYPE_FIXED64 = 6\n TYPE_FIXED32 = 7\n TYPE_BOOL = 8\n TYPE_STRING = 9\n TYPE_GROUP = 10\n TYPE_MESSAGE = 11\n TYPE_BYTES = 12\n TYPE_UINT32 = 13\n TYPE_ENUM = 14\n TYPE_SFIXED32 = 15\n TYPE_SFIXED64 = 16\n TYPE_SINT32 = 17\n TYPE_SINT64 = 18\n MAX_TYPE = 18\n\n # Must be consistent with C++ FieldDescriptor::CppType enum in\n # descriptor.h.\n #\n # TODO(robinson): Find a way to eliminate this repetition.\n CPPTYPE_INT32 = 1\n CPPTYPE_INT64 = 2\n CPPTYPE_UINT32 = 3\n CPPTYPE_UINT64 = 4\n CPPTYPE_DOUBLE = 5\n CPPTYPE_FLOAT = 6\n CPPTYPE_BOOL = 7\n CPPTYPE_ENUM = 8\n CPPTYPE_STRING = 9\n CPPTYPE_MESSAGE = 10\n MAX_CPPTYPE = 10\n\n _PYTHON_TO_CPP_PROTO_TYPE_MAP = {\n TYPE_DOUBLE: CPPTYPE_DOUBLE,\n TYPE_FLOAT: CPPTYPE_FLOAT,\n TYPE_ENUM: CPPTYPE_ENUM,\n TYPE_INT64: CPPTYPE_INT64,\n TYPE_SINT64: CPPTYPE_INT64,\n TYPE_SFIXED64: CPPTYPE_INT64,\n TYPE_UINT64: CPPTYPE_UINT64,\n TYPE_FIXED64: CPPTYPE_UINT64,\n TYPE_INT32: CPPTYPE_INT32,\n TYPE_SFIXED32: CPPTYPE_INT32,\n TYPE_SINT32: CPPTYPE_INT32,\n TYPE_UINT32: CPPTYPE_UINT32,\n TYPE_FIXED32: CPPTYPE_UINT32,\n TYPE_BYTES: CPPTYPE_STRING,\n TYPE_STRING: CPPTYPE_STRING,\n TYPE_BOOL: CPPTYPE_BOOL,\n TYPE_MESSAGE: CPPTYPE_MESSAGE,\n TYPE_GROUP: CPPTYPE_MESSAGE\n }\n\n # Must be consistent with C++ FieldDescriptor::Label enum in\n # descriptor.h.\n #\n # TODO(robinson): Find a way to eliminate this repetition.\n LABEL_OPTIONAL = 1\n LABEL_REQUIRED = 2\n LABEL_REPEATED = 3\n MAX_LABEL = 3\n\n # Must be consistent with C++ constants kMaxNumber, kFirstReservedNumber,\n # and kLastReservedNumber in descriptor.h\n MAX_FIELD_NUMBER = (1 << 29) - 1\n FIRST_RESERVED_FIELD_NUMBER = 19000\n LAST_RESERVED_FIELD_NUMBER = 19999\n\n if _USE_C_DESCRIPTORS:\n _C_DESCRIPTOR_CLASS = _message.FieldDescriptor\n\n def __new__(cls, name, full_name, index, number, type, cpp_type, label,\n default_value, message_type, enum_type, containing_type,\n is_extension, extension_scope, options=None,\n serialized_options=None,\n has_default_value=True, containing_oneof=None, json_name=None,\n file=None, create_key=None): # pylint: disable=redefined-builtin\n _message.Message._CheckCalledFromGeneratedFile()\n if is_extension:\n return _message.default_pool.FindExtensionByName(full_name)\n else:\n return _message.default_pool.FindFieldByName(full_name)\n\n def __init__(self, name, full_name, index, number, type, cpp_type, label,\n default_value, message_type, enum_type, containing_type,\n is_extension, extension_scope, options=None,\n serialized_options=None,\n has_default_value=True, containing_oneof=None, json_name=None,\n file=None, create_key=None): # pylint: disable=redefined-builtin\n \"\"\"The arguments are as described in the description of FieldDescriptor\n attributes above.\n\n Note that containing_type may be None, and may be set later if necessary\n (to deal with circular references between message types, for example).\n Likewise for extension_scope.\n \"\"\"\n if create_key is not _internal_create_key:\n _Deprecated('FieldDescriptor')\n\n super(FieldDescriptor, self).__init__(\n options, serialized_options, 'FieldOptions')\n self.name = name\n self.full_name = full_name\n self.file = file\n self._camelcase_name = None\n if json_name is None:\n self.json_name = _ToJsonName(name)\n else:\n self.json_name = json_name\n self.index = index\n self.number = number\n self.type = type\n self.cpp_type = cpp_type\n self.label = label\n self.has_default_value = has_default_value\n self.default_value = default_value\n self.containing_type = containing_type\n self.message_type = message_type\n self.enum_type = enum_type\n self.is_extension = is_extension\n self.extension_scope = extension_scope\n self.containing_oneof = containing_oneof\n if api_implementation.Type() == 'cpp':\n if is_extension:\n self._cdescriptor = _message.default_pool.FindExtensionByName(full_name)\n else:\n self._cdescriptor = _message.default_pool.FindFieldByName(full_name)\n else:\n self._cdescriptor = None\n\n @property\n def camelcase_name(self):\n \"\"\"Camelcase name of this field.\n\n Returns:\n str: the name in CamelCase.\n \"\"\"\n if self._camelcase_name is None:\n self._camelcase_name = _ToCamelCase(self.name)\n return self._camelcase_name\n\n @property\n def has_presence(self):\n \"\"\"Whether the field distinguishes between unpopulated and default values.\n\n Raises:\n RuntimeError: singular field that is not linked with message nor file.\n \"\"\"\n if self.label == FieldDescriptor.LABEL_REPEATED:\n return False\n if (self.cpp_type == FieldDescriptor.CPPTYPE_MESSAGE or\n self.containing_oneof):\n return True\n if hasattr(self.file, 'syntax'):\n return self.file.syntax == 'proto2'\n if hasattr(self.message_type, 'syntax'):\n return self.message_type.syntax == 'proto2'\n raise RuntimeError(\n 'has_presence is not ready to use because field %s is not'\n ' linked with message type nor file' % self.full_name)\n\n @staticmethod\n def ProtoTypeToCppProtoType(proto_type):\n \"\"\"Converts from a Python proto type to a C++ Proto Type.\n\n The Python ProtocolBuffer classes specify both the 'Python' datatype and the\n 'C++' datatype - and they're not the same. This helper method should\n translate from one to another.\n\n Args:\n proto_type: the Python proto type (descriptor.FieldDescriptor.TYPE_*)\n Returns:\n int: descriptor.FieldDescriptor.CPPTYPE_*, the C++ type.\n Raises:\n TypeTransformationError: when the Python proto type isn't known.\n \"\"\"\n try:\n return FieldDescriptor._PYTHON_TO_CPP_PROTO_TYPE_MAP[proto_type]\n except KeyError:\n raise TypeTransformationError('Unknown proto_type: %s' % proto_type)\n\n\nclass EnumDescriptor(_NestedDescriptorBase):\n\n \"\"\"Descriptor for an enum defined in a .proto file.\n\n Attributes:\n name (str): Name of the enum type.\n full_name (str): Full name of the type, including package name\n and any enclosing type(s).\n\n values (list[EnumValueDescriptor]): List of the values\n in this enum.\n values_by_name (dict(str, EnumValueDescriptor)): Same as :attr:`values`,\n but indexed by the \"name\" field of each EnumValueDescriptor.\n values_by_number (dict(int, EnumValueDescriptor)): Same as :attr:`values`,\n but indexed by the \"number\" field of each EnumValueDescriptor.\n containing_type (Descriptor): Descriptor of the immediate containing\n type of this enum, or None if this is an enum defined at the\n top level in a .proto file. Set by Descriptor's constructor\n if we're passed into one.\n file (FileDescriptor): Reference to file descriptor.\n options (descriptor_pb2.EnumOptions): Enum options message or\n None to use default enum options.\n \"\"\"\n\n if _USE_C_DESCRIPTORS:\n _C_DESCRIPTOR_CLASS = _message.EnumDescriptor\n\n def __new__(cls, name, full_name, filename, values,\n containing_type=None, options=None,\n serialized_options=None, file=None, # pylint: disable=redefined-builtin\n serialized_start=None, serialized_end=None, create_key=None):\n _message.Message._CheckCalledFromGeneratedFile()\n return _message.default_pool.FindEnumTypeByName(full_name)\n\n def __init__(self, name, full_name, filename, values,\n containing_type=None, options=None,\n serialized_options=None, file=None, # pylint: disable=redefined-builtin\n serialized_start=None, serialized_end=None, create_key=None):\n \"\"\"Arguments are as described in the attribute description above.\n\n Note that filename is an obsolete argument, that is not used anymore.\n Please use file.name to access this as an attribute.\n \"\"\"\n if create_key is not _internal_create_key:\n _Deprecated('EnumDescriptor')\n\n super(EnumDescriptor, self).__init__(\n options, 'EnumOptions', name, full_name, file,\n containing_type, serialized_start=serialized_start,\n serialized_end=serialized_end, serialized_options=serialized_options)\n\n self.values = values\n for value in self.values:\n value.type = self\n self.values_by_name = dict((v.name, v) for v in values)\n # Values are reversed to ensure that the first alias is retained.\n self.values_by_number = dict((v.number, v) for v in reversed(values))\n\n def CopyToProto(self, proto):\n \"\"\"Copies this to a descriptor_pb2.EnumDescriptorProto.\n\n Args:\n proto (descriptor_pb2.EnumDescriptorProto): An empty descriptor proto.\n \"\"\"\n # This function is overridden to give a better doc comment.\n super(EnumDescriptor, self).CopyToProto(proto)\n\n\nclass EnumValueDescriptor(DescriptorBase):\n\n \"\"\"Descriptor for a single value within an enum.\n\n Attributes:\n name (str): Name of this value.\n index (int): Dense, 0-indexed index giving the order that this\n value appears textually within its enum in the .proto file.\n number (int): Actual number assigned to this enum value.\n type (EnumDescriptor): :class:`EnumDescriptor` to which this value\n belongs. Set by :class:`EnumDescriptor`'s constructor if we're\n passed into one.\n options (descriptor_pb2.EnumValueOptions): Enum value options message or\n None to use default enum value options options.\n \"\"\"\n\n if _USE_C_DESCRIPTORS:\n _C_DESCRIPTOR_CLASS = _message.EnumValueDescriptor\n\n def __new__(cls, name, index, number,\n type=None, # pylint: disable=redefined-builtin\n options=None, serialized_options=None, create_key=None):\n _message.Message._CheckCalledFromGeneratedFile()\n # There is no way we can build a complete EnumValueDescriptor with the\n # given parameters (the name of the Enum is not known, for example).\n # Fortunately generated files just pass it to the EnumDescriptor()\n # constructor, which will ignore it, so returning None is good enough.\n return None\n\n def __init__(self, name, index, number,\n type=None, # pylint: disable=redefined-builtin\n options=None, serialized_options=None, create_key=None):\n \"\"\"Arguments are as described in the attribute description above.\"\"\"\n if create_key is not _internal_create_key:\n _Deprecated('EnumValueDescriptor')\n\n super(EnumValueDescriptor, self).__init__(\n options, serialized_options, 'EnumValueOptions')\n self.name = name\n self.index = index\n self.number = number\n self.type = type\n\n\nclass OneofDescriptor(DescriptorBase):\n \"\"\"Descriptor for a oneof field.\n\n Attributes:\n name (str): Name of the oneof field.\n full_name (str): Full name of the oneof field, including package name.\n index (int): 0-based index giving the order of the oneof field inside\n its containing type.\n containing_type (Descriptor): :class:`Descriptor` of the protocol message\n type that contains this field. Set by the :class:`Descriptor` constructor\n if we're passed into one.\n fields (list[FieldDescriptor]): The list of field descriptors this\n oneof can contain.\n \"\"\"\n\n if _USE_C_DESCRIPTORS:\n _C_DESCRIPTOR_CLASS = _message.OneofDescriptor\n\n def __new__(\n cls, name, full_name, index, containing_type, fields, options=None,\n serialized_options=None, create_key=None):\n _message.Message._CheckCalledFromGeneratedFile()\n return _message.default_pool.FindOneofByName(full_name)\n\n def __init__(\n self, name, full_name, index, containing_type, fields, options=None,\n serialized_options=None, create_key=None):\n \"\"\"Arguments are as described in the attribute description above.\"\"\"\n if create_key is not _internal_create_key:\n _Deprecated('OneofDescriptor')\n\n super(OneofDescriptor, self).__init__(\n options, serialized_options, 'OneofOptions')\n self.name = name\n self.full_name = full_name\n self.index = index\n self.containing_type = containing_type\n self.fields = fields\n\n\nclass ServiceDescriptor(_NestedDescriptorBase):\n\n \"\"\"Descriptor for a service.\n\n Attributes:\n name (str): Name of the service.\n full_name (str): Full name of the service, including package name.\n index (int): 0-indexed index giving the order that this services\n definition appears within the .proto file.\n methods (list[MethodDescriptor]): List of methods provided by this\n service.\n methods_by_name (dict(str, MethodDescriptor)): Same\n :class:`MethodDescriptor` objects as in :attr:`methods_by_name`, but\n indexed by \"name\" attribute in each :class:`MethodDescriptor`.\n options (descriptor_pb2.ServiceOptions): Service options message or\n None to use default service options.\n file (FileDescriptor): Reference to file info.\n \"\"\"\n\n if _USE_C_DESCRIPTORS:\n _C_DESCRIPTOR_CLASS = _message.ServiceDescriptor\n\n def __new__(\n cls,\n name=None,\n full_name=None,\n index=None,\n methods=None,\n options=None,\n serialized_options=None,\n file=None, # pylint: disable=redefined-builtin\n serialized_start=None,\n serialized_end=None,\n create_key=None):\n _message.Message._CheckCalledFromGeneratedFile() # pylint: disable=protected-access\n return _message.default_pool.FindServiceByName(full_name)\n\n def __init__(self, name, full_name, index, methods, options=None,\n serialized_options=None, file=None, # pylint: disable=redefined-builtin\n serialized_start=None, serialized_end=None, create_key=None):\n if create_key is not _internal_create_key:\n _Deprecated('ServiceDescriptor')\n\n super(ServiceDescriptor, self).__init__(\n options, 'ServiceOptions', name, full_name, file,\n None, serialized_start=serialized_start,\n serialized_end=serialized_end, serialized_options=serialized_options)\n self.index = index\n self.methods = methods\n self.methods_by_name = dict((m.name, m) for m in methods)\n # Set the containing service for each method in this service.\n for method in self.methods:\n method.containing_service = self\n\n def FindMethodByName(self, name):\n \"\"\"Searches for the specified method, and returns its descriptor.\n\n Args:\n name (str): Name of the method.\n Returns:\n MethodDescriptor or None: the descriptor for the requested method, if\n found.\n \"\"\"\n return self.methods_by_name.get(name, None)\n\n def CopyToProto(self, proto):\n \"\"\"Copies this to a descriptor_pb2.ServiceDescriptorProto.\n\n Args:\n proto (descriptor_pb2.ServiceDescriptorProto): An empty descriptor proto.\n \"\"\"\n # This function is overridden to give a better doc comment.\n super(ServiceDescriptor, self).CopyToProto(proto)\n\n\nclass MethodDescriptor(DescriptorBase):\n\n \"\"\"Descriptor for a method in a service.\n\n Attributes:\n name (str): Name of the method within the service.\n full_name (str): Full name of method.\n index (int): 0-indexed index of the method inside the service.\n containing_service (ServiceDescriptor): The service that contains this\n method.\n input_type (Descriptor): The descriptor of the message that this method\n accepts.\n output_type (Descriptor): The descriptor of the message that this method\n returns.\n client_streaming (bool): Whether this method uses client streaming.\n server_streaming (bool): Whether this method uses server streaming.\n options (descriptor_pb2.MethodOptions or None): Method options message, or\n None to use default method options.\n \"\"\"\n\n if _USE_C_DESCRIPTORS:\n _C_DESCRIPTOR_CLASS = _message.MethodDescriptor\n\n def __new__(cls,\n name,\n full_name,\n index,\n containing_service,\n input_type,\n output_type,\n client_streaming=False,\n server_streaming=False,\n options=None,\n serialized_options=None,\n create_key=None):\n _message.Message._CheckCalledFromGeneratedFile() # pylint: disable=protected-access\n return _message.default_pool.FindMethodByName(full_name)\n\n def __init__(self,\n name,\n full_name,\n index,\n containing_service,\n input_type,\n output_type,\n client_streaming=False,\n server_streaming=False,\n options=None,\n serialized_options=None,\n create_key=None):\n \"\"\"The arguments are as described in the description of MethodDescriptor\n attributes above.\n\n Note that containing_service may be None, and may be set later if necessary.\n \"\"\"\n if create_key is not _internal_create_key:\n _Deprecated('MethodDescriptor')\n\n super(MethodDescriptor, self).__init__(\n options, serialized_options, 'MethodOptions')\n self.name = name\n self.full_name = full_name\n self.index = index\n self.containing_service = containing_service\n self.input_type = input_type\n self.output_type = output_type\n self.client_streaming = client_streaming\n self.server_streaming = server_streaming\n\n def CopyToProto(self, proto):\n \"\"\"Copies this to a descriptor_pb2.MethodDescriptorProto.\n\n Args:\n proto (descriptor_pb2.MethodDescriptorProto): An empty descriptor proto.\n\n Raises:\n Error: If self couldn't be serialized, due to too few constructor\n arguments.\n \"\"\"\n if self.containing_service is not None:\n from google.protobuf import descriptor_pb2\n service_proto = descriptor_pb2.ServiceDescriptorProto()\n self.containing_service.CopyToProto(service_proto)\n proto.CopyFrom(service_proto.method[self.index])\n else:\n raise Error('Descriptor does not contain a service.')\n\n\nclass FileDescriptor(DescriptorBase):\n \"\"\"Descriptor for a file. Mimics the descriptor_pb2.FileDescriptorProto.\n\n Note that :attr:`enum_types_by_name`, :attr:`extensions_by_name`, and\n :attr:`dependencies` fields are only set by the\n :py:mod:`google.protobuf.message_factory` module, and not by the generated\n proto code.\n\n Attributes:\n name (str): Name of file, relative to root of source tree.\n package (str): Name of the package\n syntax (str): string indicating syntax of the file (can be \"proto2\" or\n \"proto3\")\n serialized_pb (bytes): Byte string of serialized\n :class:`descriptor_pb2.FileDescriptorProto`.\n dependencies (list[FileDescriptor]): List of other :class:`FileDescriptor`\n objects this :class:`FileDescriptor` depends on.\n public_dependencies (list[FileDescriptor]): A subset of\n :attr:`dependencies`, which were declared as \"public\".\n message_types_by_name (dict(str, Descriptor)): Mapping from message names\n to their :class:`Descriptor`.\n enum_types_by_name (dict(str, EnumDescriptor)): Mapping from enum names to\n their :class:`EnumDescriptor`.\n extensions_by_name (dict(str, FieldDescriptor)): Mapping from extension\n names declared at file scope to their :class:`FieldDescriptor`.\n services_by_name (dict(str, ServiceDescriptor)): Mapping from services'\n names to their :class:`ServiceDescriptor`.\n pool (DescriptorPool): The pool this descriptor belongs to. When not\n passed to the constructor, the global default pool is used.\n \"\"\"\n\n if _USE_C_DESCRIPTORS:\n _C_DESCRIPTOR_CLASS = _message.FileDescriptor\n\n def __new__(cls, name, package, options=None,\n serialized_options=None, serialized_pb=None,\n dependencies=None, public_dependencies=None,\n syntax=None, pool=None, create_key=None):\n # FileDescriptor() is called from various places, not only from generated\n # files, to register dynamic proto files and messages.\n # pylint: disable=g-explicit-bool-comparison\n if serialized_pb == b'':\n # Cpp generated code must be linked in if serialized_pb is ''\n try:\n return _message.default_pool.FindFileByName(name)\n except KeyError:\n raise RuntimeError('Please link in cpp generated lib for %s' % (name))\n elif serialized_pb:\n return _message.default_pool.AddSerializedFile(serialized_pb)\n else:\n return super(FileDescriptor, cls).__new__(cls)\n\n def __init__(self, name, package, options=None,\n serialized_options=None, serialized_pb=None,\n dependencies=None, public_dependencies=None,\n syntax=None, pool=None, create_key=None):\n \"\"\"Constructor.\"\"\"\n if create_key is not _internal_create_key:\n _Deprecated('FileDescriptor')\n\n super(FileDescriptor, self).__init__(\n options, serialized_options, 'FileOptions')\n\n if pool is None:\n from google.protobuf import descriptor_pool\n pool = descriptor_pool.Default()\n self.pool = pool\n self.message_types_by_name = {}\n self.name = name\n self.package = package\n self.syntax = syntax or \"proto2\"\n self.serialized_pb = serialized_pb\n\n self.enum_types_by_name = {}\n self.extensions_by_name = {}\n self.services_by_name = {}\n self.dependencies = (dependencies or [])\n self.public_dependencies = (public_dependencies or [])\n\n def CopyToProto(self, proto):\n \"\"\"Copies this to a descriptor_pb2.FileDescriptorProto.\n\n Args:\n proto: An empty descriptor_pb2.FileDescriptorProto.\n \"\"\"\n proto.ParseFromString(self.serialized_pb)\n\n\ndef _ParseOptions(message, string):\n \"\"\"Parses serialized options.\n\n This helper function is used to parse serialized options in generated\n proto2 files. It must not be used outside proto2.\n \"\"\"\n message.ParseFromString(string)\n return message\n\n\ndef _ToCamelCase(name):\n \"\"\"Converts name to camel-case and returns it.\"\"\"\n capitalize_next = False\n result = []\n\n for c in name:\n if c == '_':\n if result:\n capitalize_next = True\n elif capitalize_next:\n result.append(c.upper())\n capitalize_next = False\n else:\n result += c\n\n # Lower-case the first letter.\n if result and result[0].isupper():\n result[0] = result[0].lower()\n return ''.join(result)\n\n\ndef _OptionsOrNone(descriptor_proto):\n \"\"\"Returns the value of the field `options`, or None if it is not set.\"\"\"\n if descriptor_proto.HasField('options'):\n return descriptor_proto.options\n else:\n return None\n\n\ndef _ToJsonName(name):\n \"\"\"Converts name to Json name and returns it.\"\"\"\n capitalize_next = False\n result = []\n\n for c in name:\n if c == '_':\n capitalize_next = True\n elif capitalize_next:\n result.append(c.upper())\n capitalize_next = False\n else:\n result += c\n\n return ''.join(result)\n\n\ndef MakeDescriptor(desc_proto, package='', build_file_if_cpp=True,\n syntax=None):\n \"\"\"Make a protobuf Descriptor given a DescriptorProto protobuf.\n\n Handles nested descriptors. Note that this is limited to the scope of defining\n a message inside of another message. Composite fields can currently only be\n resolved if the message is defined in the same scope as the field.\n\n Args:\n desc_proto: The descriptor_pb2.DescriptorProto protobuf message.\n package: Optional package name for the new message Descriptor (string).\n build_file_if_cpp: Update the C++ descriptor pool if api matches.\n Set to False on recursion, so no duplicates are created.\n syntax: The syntax/semantics that should be used. Set to \"proto3\" to get\n proto3 field presence semantics.\n Returns:\n A Descriptor for protobuf messages.\n \"\"\"\n if api_implementation.Type() == 'cpp' and build_file_if_cpp:\n # The C++ implementation requires all descriptors to be backed by the same\n # definition in the C++ descriptor pool. To do this, we build a\n # FileDescriptorProto with the same definition as this descriptor and build\n # it into the pool.\n from google.protobuf import descriptor_pb2\n file_descriptor_proto = descriptor_pb2.FileDescriptorProto()\n file_descriptor_proto.message_type.add().MergeFrom(desc_proto)\n\n # Generate a random name for this proto file to prevent conflicts with any\n # imported ones. We need to specify a file name so the descriptor pool\n # accepts our FileDescriptorProto, but it is not important what that file\n # name is actually set to.\n proto_name = binascii.hexlify(os.urandom(16)).decode('ascii')\n\n if package:\n file_descriptor_proto.name = os.path.join(package.replace('.', '/'),\n proto_name + '.proto')\n file_descriptor_proto.package = package\n else:\n file_descriptor_proto.name = proto_name + '.proto'\n\n _message.default_pool.Add(file_descriptor_proto)\n result = _message.default_pool.FindFileByName(file_descriptor_proto.name)\n\n if _USE_C_DESCRIPTORS:\n return result.message_types_by_name[desc_proto.name]\n\n full_message_name = [desc_proto.name]\n if package: full_message_name.insert(0, package)\n\n # Create Descriptors for enum types\n enum_types = {}\n for enum_proto in desc_proto.enum_type:\n full_name = '.'.join(full_message_name + [enum_proto.name])\n enum_desc = EnumDescriptor(\n enum_proto.name, full_name, None, [\n EnumValueDescriptor(enum_val.name, ii, enum_val.number,\n create_key=_internal_create_key)\n for ii, enum_val in enumerate(enum_proto.value)],\n create_key=_internal_create_key)\n enum_types[full_name] = enum_desc\n\n # Create Descriptors for nested types\n nested_types = {}\n for nested_proto in desc_proto.nested_type:\n full_name = '.'.join(full_message_name + [nested_proto.name])\n # Nested types are just those defined inside of the message, not all types\n # used by fields in the message, so no loops are possible here.\n nested_desc = MakeDescriptor(nested_proto,\n package='.'.join(full_message_name),\n build_file_if_cpp=False,\n syntax=syntax)\n nested_types[full_name] = nested_desc\n\n fields = []\n for field_proto in desc_proto.field:\n full_name = '.'.join(full_message_name + [field_proto.name])\n enum_desc = None\n nested_desc = None\n if field_proto.json_name:\n json_name = field_proto.json_name\n else:\n json_name = None\n if field_proto.HasField('type_name'):\n type_name = field_proto.type_name\n full_type_name = '.'.join(full_message_name +\n [type_name[type_name.rfind('.')+1:]])\n if full_type_name in nested_types:\n nested_desc = nested_types[full_type_name]\n elif full_type_name in enum_types:\n enum_desc = enum_types[full_type_name]\n # Else type_name references a non-local type, which isn't implemented\n field = FieldDescriptor(\n field_proto.name, full_name, field_proto.number - 1,\n field_proto.number, field_proto.type,\n FieldDescriptor.ProtoTypeToCppProtoType(field_proto.type),\n field_proto.label, None, nested_desc, enum_desc, None, False, None,\n options=_OptionsOrNone(field_proto), has_default_value=False,\n json_name=json_name, create_key=_internal_create_key)\n fields.append(field)\n\n desc_name = '.'.join(full_message_name)\n return Descriptor(desc_proto.name, desc_name, None, None, fields,\n list(nested_types.values()), list(enum_types.values()), [],\n options=_OptionsOrNone(desc_proto),\n create_key=_internal_create_key)\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/descriptor_database.py",
"content": "# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n\"\"\"Provides a container for DescriptorProtos.\"\"\"\n\n__author__ = 'matthewtoia@google.com (Matt Toia)'\n\nimport warnings\n\n\nclass Error(Exception):\n pass\n\n\nclass DescriptorDatabaseConflictingDefinitionError(Error):\n \"\"\"Raised when a proto is added with the same name & different descriptor.\"\"\"\n\n\nclass DescriptorDatabase(object):\n \"\"\"A container accepting FileDescriptorProtos and maps DescriptorProtos.\"\"\"\n\n def __init__(self):\n self._file_desc_protos_by_file = {}\n self._file_desc_protos_by_symbol = {}\n\n def Add(self, file_desc_proto):\n \"\"\"Adds the FileDescriptorProto and its types to this database.\n\n Args:\n file_desc_proto: The FileDescriptorProto to add.\n Raises:\n DescriptorDatabaseConflictingDefinitionError: if an attempt is made to\n add a proto with the same name but different definition than an\n existing proto in the database.\n \"\"\"\n proto_name = file_desc_proto.name\n if proto_name not in self._file_desc_protos_by_file:\n self._file_desc_protos_by_file[proto_name] = file_desc_proto\n elif self._file_desc_protos_by_file[proto_name] != file_desc_proto:\n raise DescriptorDatabaseConflictingDefinitionError(\n '%s already added, but with different descriptor.' % proto_name)\n else:\n return\n\n # Add all the top-level descriptors to the index.\n package = file_desc_proto.package\n for message in file_desc_proto.message_type:\n for name in _ExtractSymbols(message, package):\n self._AddSymbol(name, file_desc_proto)\n for enum in file_desc_proto.enum_type:\n self._AddSymbol(('.'.join((package, enum.name))), file_desc_proto)\n for enum_value in enum.value:\n self._file_desc_protos_by_symbol[\n '.'.join((package, enum_value.name))] = file_desc_proto\n for extension in file_desc_proto.extension:\n self._AddSymbol(('.'.join((package, extension.name))), file_desc_proto)\n for service in file_desc_proto.service:\n self._AddSymbol(('.'.join((package, service.name))), file_desc_proto)\n\n def FindFileByName(self, name):\n \"\"\"Finds the file descriptor proto by file name.\n\n Typically the file name is a relative path ending to a .proto file. The\n proto with the given name will have to have been added to this database\n using the Add method or else an error will be raised.\n\n Args:\n name: The file name to find.\n\n Returns:\n The file descriptor proto matching the name.\n\n Raises:\n KeyError if no file by the given name was added.\n \"\"\"\n\n return self._file_desc_protos_by_file[name]\n\n def FindFileContainingSymbol(self, symbol):\n \"\"\"Finds the file descriptor proto containing the specified symbol.\n\n The symbol should be a fully qualified name including the file descriptor's\n package and any containing messages. Some examples:\n\n 'some.package.name.Message'\n 'some.package.name.Message.NestedEnum'\n 'some.package.name.Message.some_field'\n\n The file descriptor proto containing the specified symbol must be added to\n this database using the Add method or else an error will be raised.\n\n Args:\n symbol: The fully qualified symbol name.\n\n Returns:\n The file descriptor proto containing the symbol.\n\n Raises:\n KeyError if no file contains the specified symbol.\n \"\"\"\n try:\n return self._file_desc_protos_by_symbol[symbol]\n except KeyError:\n # Fields, enum values, and nested extensions are not in\n # _file_desc_protos_by_symbol. Try to find the top level\n # descriptor. Non-existent nested symbol under a valid top level\n # descriptor can also be found. The behavior is the same with\n # protobuf C++.\n top_level, _, _ = symbol.rpartition('.')\n try:\n return self._file_desc_protos_by_symbol[top_level]\n except KeyError:\n # Raise the original symbol as a KeyError for better diagnostics.\n raise KeyError(symbol)\n\n def FindFileContainingExtension(self, extendee_name, extension_number):\n # TODO(jieluo): implement this API.\n return None\n\n def FindAllExtensionNumbers(self, extendee_name):\n # TODO(jieluo): implement this API.\n return []\n\n def _AddSymbol(self, name, file_desc_proto):\n if name in self._file_desc_protos_by_symbol:\n warn_msg = ('Conflict register for file \"' + file_desc_proto.name +\n '\": ' + name +\n ' is already defined in file \"' +\n self._file_desc_protos_by_symbol[name].name + '\"')\n warnings.warn(warn_msg, RuntimeWarning)\n self._file_desc_protos_by_symbol[name] = file_desc_proto\n\n\ndef _ExtractSymbols(desc_proto, package):\n \"\"\"Pulls out all the symbols from a descriptor proto.\n\n Args:\n desc_proto: The proto to extract symbols from.\n package: The package containing the descriptor type.\n\n Yields:\n The fully qualified name found in the descriptor.\n \"\"\"\n message_name = package + '.' + desc_proto.name if package else desc_proto.name\n yield message_name\n for nested_type in desc_proto.nested_type:\n for symbol in _ExtractSymbols(nested_type, message_name):\n yield symbol\n for enum_type in desc_proto.enum_type:\n yield '.'.join((message_name, enum_type.name))\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/descriptor_pb2.py",
"content": "# -*- coding: utf-8 -*-\n# Generated by the protocol buffer compiler. DO NOT EDIT!\n# source: google/protobuf/descriptor.proto\n\"\"\"Generated protocol buffer code.\"\"\"\nfrom google.protobuf.internal import builder as _builder\nfrom google.protobuf import descriptor as _descriptor\nfrom google.protobuf import descriptor_pool as _descriptor_pool\nfrom google.protobuf import symbol_database as _symbol_database\n# @@protoc_insertion_point(imports)\n\n_sym_db = _symbol_database.Default()\n\n\n\n\nif _descriptor._USE_C_DESCRIPTORS == False:\n DESCRIPTOR = _descriptor.FileDescriptor(\n name='google/protobuf/descriptor.proto',\n package='google.protobuf',\n syntax='proto2',\n serialized_options=None,\n create_key=_descriptor._internal_create_key,\n serialized_pb=b'\\n google/protobuf/descriptor.proto\\x12\\x0fgoogle.protobuf\\\"G\\n\\x11\\x46ileDescriptorSet\\x12\\x32\\n\\x04\\x66ile\\x18\\x01 \\x03(\\x0b\\x32$.google.protobuf.FileDescriptorProto\\\"\\xdb\\x03\\n\\x13\\x46ileDescriptorProto\\x12\\x0c\\n\\x04name\\x18\\x01 \\x01(\\t\\x12\\x0f\\n\\x07package\\x18\\x02 \\x01(\\t\\x12\\x12\\n\\ndependency\\x18\\x03 \\x03(\\t\\x12\\x19\\n\\x11public_dependency\\x18\\n \\x03(\\x05\\x12\\x17\\n\\x0fweak_dependency\\x18\\x0b \\x03(\\x05\\x12\\x36\\n\\x0cmessage_type\\x18\\x04 \\x03(\\x0b\\x32 .google.protobuf.DescriptorProto\\x12\\x37\\n\\tenum_type\\x18\\x05 \\x03(\\x0b\\x32$.google.protobuf.EnumDescriptorProto\\x12\\x38\\n\\x07service\\x18\\x06 \\x03(\\x0b\\x32\\'.google.protobuf.ServiceDescriptorProto\\x12\\x38\\n\\textension\\x18\\x07 \\x03(\\x0b\\x32%.google.protobuf.FieldDescriptorProto\\x12-\\n\\x07options\\x18\\x08 \\x01(\\x0b\\x32\\x1c.google.protobuf.FileOptions\\x12\\x39\\n\\x10source_code_info\\x18\\t \\x01(\\x0b\\x32\\x1f.google.protobuf.SourceCodeInfo\\x12\\x0e\\n\\x06syntax\\x18\\x0c \\x01(\\t\\\"\\xa9\\x05\\n\\x0f\\x44\\x65scriptorProto\\x12\\x0c\\n\\x04name\\x18\\x01 \\x01(\\t\\x12\\x34\\n\\x05\\x66ield\\x18\\x02 \\x03(\\x0b\\x32%.google.protobuf.FieldDescriptorProto\\x12\\x38\\n\\textension\\x18\\x06 \\x03(\\x0b\\x32%.google.protobuf.FieldDescriptorProto\\x12\\x35\\n\\x0bnested_type\\x18\\x03 \\x03(\\x0b\\x32 .google.protobuf.DescriptorProto\\x12\\x37\\n\\tenum_type\\x18\\x04 \\x03(\\x0b\\x32$.google.protobuf.EnumDescriptorProto\\x12H\\n\\x0f\\x65xtension_range\\x18\\x05 \\x03(\\x0b\\x32/.google.protobuf.DescriptorProto.ExtensionRange\\x12\\x39\\n\\noneof_decl\\x18\\x08 \\x03(\\x0b\\x32%.google.protobuf.OneofDescriptorProto\\x12\\x30\\n\\x07options\\x18\\x07 \\x01(\\x0b\\x32\\x1f.google.protobuf.MessageOptions\\x12\\x46\\n\\x0ereserved_range\\x18\\t \\x03(\\x0b\\x32..google.protobuf.DescriptorProto.ReservedRange\\x12\\x15\\n\\rreserved_name\\x18\\n \\x03(\\t\\x1a\\x65\\n\\x0e\\x45xtensionRange\\x12\\r\\n\\x05start\\x18\\x01 \\x01(\\x05\\x12\\x0b\\n\\x03\\x65nd\\x18\\x02 \\x01(\\x05\\x12\\x37\\n\\x07options\\x18\\x03 \\x01(\\x0b\\x32&.google.protobuf.ExtensionRangeOptions\\x1a+\\n\\rReservedRange\\x12\\r\\n\\x05start\\x18\\x01 \\x01(\\x05\\x12\\x0b\\n\\x03\\x65nd\\x18\\x02 \\x01(\\x05\\\"g\\n\\x15\\x45xtensionRangeOptions\\x12\\x43\\n\\x14uninterpreted_option\\x18\\xe7\\x07 \\x03(\\x0b\\x32$.google.protobuf.UninterpretedOption*\\t\\x08\\xe8\\x07\\x10\\x80\\x80\\x80\\x80\\x02\\\"\\xd5\\x05\\n\\x14\\x46ieldDescriptorProto\\x12\\x0c\\n\\x04name\\x18\\x01 \\x01(\\t\\x12\\x0e\\n\\x06number\\x18\\x03 \\x01(\\x05\\x12:\\n\\x05label\\x18\\x04 \\x01(\\x0e\\x32+.google.protobuf.FieldDescriptorProto.Label\\x12\\x38\\n\\x04type\\x18\\x05 \\x01(\\x0e\\x32*.google.protobuf.FieldDescriptorProto.Type\\x12\\x11\\n\\ttype_name\\x18\\x06 \\x01(\\t\\x12\\x10\\n\\x08\\x65xtendee\\x18\\x02 \\x01(\\t\\x12\\x15\\n\\rdefault_value\\x18\\x07 \\x01(\\t\\x12\\x13\\n\\x0boneof_index\\x18\\t \\x01(\\x05\\x12\\x11\\n\\tjson_name\\x18\\n \\x01(\\t\\x12.\\n\\x07options\\x18\\x08 \\x01(\\x0b\\x32\\x1d.google.protobuf.FieldOptions\\x12\\x17\\n\\x0fproto3_optional\\x18\\x11 \\x01(\\x08\\\"\\xb6\\x02\\n\\x04Type\\x12\\x0f\\n\\x0bTYPE_DOUBLE\\x10\\x01\\x12\\x0e\\n\\nTYPE_FLOAT\\x10\\x02\\x12\\x0e\\n\\nTYPE_INT64\\x10\\x03\\x12\\x0f\\n\\x0bTYPE_UINT64\\x10\\x04\\x12\\x0e\\n\\nTYPE_INT32\\x10\\x05\\x12\\x10\\n\\x0cTYPE_FIXED64\\x10\\x06\\x12\\x10\\n\\x0cTYPE_FIXED32\\x10\\x07\\x12\\r\\n\\tTYPE_BOOL\\x10\\x08\\x12\\x0f\\n\\x0bTYPE_STRING\\x10\\t\\x12\\x0e\\n\\nTYPE_GROUP\\x10\\n\\x12\\x10\\n\\x0cTYPE_MESSAGE\\x10\\x0b\\x12\\x0e\\n\\nTYPE_BYTES\\x10\\x0c\\x12\\x0f\\n\\x0bTYPE_UINT32\\x10\\r\\x12\\r\\n\\tTYPE_ENUM\\x10\\x0e\\x12\\x11\\n\\rTYPE_SFIXED32\\x10\\x0f\\x12\\x11\\n\\rTYPE_SFIXED64\\x10\\x10\\x12\\x0f\\n\\x0bTYPE_SINT32\\x10\\x11\\x12\\x0f\\n\\x0bTYPE_SINT64\\x10\\x12\\\"C\\n\\x05Label\\x12\\x12\\n\\x0eLABEL_OPTIONAL\\x10\\x01\\x12\\x12\\n\\x0eLABEL_REQUIRED\\x10\\x02\\x12\\x12\\n\\x0eLABEL_REPEATED\\x10\\x03\\\"T\\n\\x14OneofDescriptorProto\\x12\\x0c\\n\\x04name\\x18\\x01 \\x01(\\t\\x12.\\n\\x07options\\x18\\x02 \\x01(\\x0b\\x32\\x1d.google.protobuf.OneofOptions\\\"\\xa4\\x02\\n\\x13\\x45numDescriptorProto\\x12\\x0c\\n\\x04name\\x18\\x01 \\x01(\\t\\x12\\x38\\n\\x05value\\x18\\x02 \\x03(\\x0b\\x32).google.protobuf.EnumValueDescriptorProto\\x12-\\n\\x07options\\x18\\x03 \\x01(\\x0b\\x32\\x1c.google.protobuf.EnumOptions\\x12N\\n\\x0ereserved_range\\x18\\x04 \\x03(\\x0b\\x32\\x36.google.protobuf.EnumDescriptorProto.EnumReservedRange\\x12\\x15\\n\\rreserved_name\\x18\\x05 \\x03(\\t\\x1a/\\n\\x11\\x45numReservedRange\\x12\\r\\n\\x05start\\x18\\x01 \\x01(\\x05\\x12\\x0b\\n\\x03\\x65nd\\x18\\x02 \\x01(\\x05\\\"l\\n\\x18\\x45numValueDescriptorProto\\x12\\x0c\\n\\x04name\\x18\\x01 \\x01(\\t\\x12\\x0e\\n\\x06number\\x18\\x02 \\x01(\\x05\\x12\\x32\\n\\x07options\\x18\\x03 \\x01(\\x0b\\x32!.google.protobuf.EnumValueOptions\\\"\\x90\\x01\\n\\x16ServiceDescriptorProto\\x12\\x0c\\n\\x04name\\x18\\x01 \\x01(\\t\\x12\\x36\\n\\x06method\\x18\\x02 \\x03(\\x0b\\x32&.google.protobuf.MethodDescriptorProto\\x12\\x30\\n\\x07options\\x18\\x03 \\x01(\\x0b\\x32\\x1f.google.protobuf.ServiceOptions\\\"\\xc1\\x01\\n\\x15MethodDescriptorProto\\x12\\x0c\\n\\x04name\\x18\\x01 \\x01(\\t\\x12\\x12\\n\\ninput_type\\x18\\x02 \\x01(\\t\\x12\\x13\\n\\x0boutput_type\\x18\\x03 \\x01(\\t\\x12/\\n\\x07options\\x18\\x04 \\x01(\\x0b\\x32\\x1e.google.protobuf.MethodOptions\\x12\\x1f\\n\\x10\\x63lient_streaming\\x18\\x05 \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x1f\\n\\x10server_streaming\\x18\\x06 \\x01(\\x08:\\x05\\x66\\x61lse\\\"\\xa5\\x06\\n\\x0b\\x46ileOptions\\x12\\x14\\n\\x0cjava_package\\x18\\x01 \\x01(\\t\\x12\\x1c\\n\\x14java_outer_classname\\x18\\x08 \\x01(\\t\\x12\\\"\\n\\x13java_multiple_files\\x18\\n \\x01(\\x08:\\x05\\x66\\x61lse\\x12)\\n\\x1djava_generate_equals_and_hash\\x18\\x14 \\x01(\\x08\\x42\\x02\\x18\\x01\\x12%\\n\\x16java_string_check_utf8\\x18\\x1b \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x46\\n\\x0coptimize_for\\x18\\t \\x01(\\x0e\\x32).google.protobuf.FileOptions.OptimizeMode:\\x05SPEED\\x12\\x12\\n\\ngo_package\\x18\\x0b \\x01(\\t\\x12\\\"\\n\\x13\\x63\\x63_generic_services\\x18\\x10 \\x01(\\x08:\\x05\\x66\\x61lse\\x12$\\n\\x15java_generic_services\\x18\\x11 \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\\"\\n\\x13py_generic_services\\x18\\x12 \\x01(\\x08:\\x05\\x66\\x61lse\\x12#\\n\\x14php_generic_services\\x18* \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x19\\n\\ndeprecated\\x18\\x17 \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x1e\\n\\x10\\x63\\x63_enable_arenas\\x18\\x1f \\x01(\\x08:\\x04true\\x12\\x19\\n\\x11objc_class_prefix\\x18$ \\x01(\\t\\x12\\x18\\n\\x10\\x63sharp_namespace\\x18% \\x01(\\t\\x12\\x14\\n\\x0cswift_prefix\\x18\\' \\x01(\\t\\x12\\x18\\n\\x10php_class_prefix\\x18( \\x01(\\t\\x12\\x15\\n\\rphp_namespace\\x18) \\x01(\\t\\x12\\x1e\\n\\x16php_metadata_namespace\\x18, \\x01(\\t\\x12\\x14\\n\\x0cruby_package\\x18- \\x01(\\t\\x12\\x43\\n\\x14uninterpreted_option\\x18\\xe7\\x07 \\x03(\\x0b\\x32$.google.protobuf.UninterpretedOption\\\":\\n\\x0cOptimizeMode\\x12\\t\\n\\x05SPEED\\x10\\x01\\x12\\r\\n\\tCODE_SIZE\\x10\\x02\\x12\\x10\\n\\x0cLITE_RUNTIME\\x10\\x03*\\t\\x08\\xe8\\x07\\x10\\x80\\x80\\x80\\x80\\x02J\\x04\\x08&\\x10\\'\\\"\\x84\\x02\\n\\x0eMessageOptions\\x12&\\n\\x17message_set_wire_format\\x18\\x01 \\x01(\\x08:\\x05\\x66\\x61lse\\x12.\\n\\x1fno_standard_descriptor_accessor\\x18\\x02 \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x19\\n\\ndeprecated\\x18\\x03 \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x11\\n\\tmap_entry\\x18\\x07 \\x01(\\x08\\x12\\x43\\n\\x14uninterpreted_option\\x18\\xe7\\x07 \\x03(\\x0b\\x32$.google.protobuf.UninterpretedOption*\\t\\x08\\xe8\\x07\\x10\\x80\\x80\\x80\\x80\\x02J\\x04\\x08\\x04\\x10\\x05J\\x04\\x08\\x05\\x10\\x06J\\x04\\x08\\x06\\x10\\x07J\\x04\\x08\\x08\\x10\\tJ\\x04\\x08\\t\\x10\\n\\\"\\xbe\\x03\\n\\x0c\\x46ieldOptions\\x12:\\n\\x05\\x63type\\x18\\x01 \\x01(\\x0e\\x32#.google.protobuf.FieldOptions.CType:\\x06STRING\\x12\\x0e\\n\\x06packed\\x18\\x02 \\x01(\\x08\\x12?\\n\\x06jstype\\x18\\x06 \\x01(\\x0e\\x32$.google.protobuf.FieldOptions.JSType:\\tJS_NORMAL\\x12\\x13\\n\\x04lazy\\x18\\x05 \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x1e\\n\\x0funverified_lazy\\x18\\x0f \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x19\\n\\ndeprecated\\x18\\x03 \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x13\\n\\x04weak\\x18\\n \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x43\\n\\x14uninterpreted_option\\x18\\xe7\\x07 \\x03(\\x0b\\x32$.google.protobuf.UninterpretedOption\\\"/\\n\\x05\\x43Type\\x12\\n\\n\\x06STRING\\x10\\x00\\x12\\x08\\n\\x04\\x43ORD\\x10\\x01\\x12\\x10\\n\\x0cSTRING_PIECE\\x10\\x02\\\"5\\n\\x06JSType\\x12\\r\\n\\tJS_NORMAL\\x10\\x00\\x12\\r\\n\\tJS_STRING\\x10\\x01\\x12\\r\\n\\tJS_NUMBER\\x10\\x02*\\t\\x08\\xe8\\x07\\x10\\x80\\x80\\x80\\x80\\x02J\\x04\\x08\\x04\\x10\\x05\\\"^\\n\\x0cOneofOptions\\x12\\x43\\n\\x14uninterpreted_option\\x18\\xe7\\x07 \\x03(\\x0b\\x32$.google.protobuf.UninterpretedOption*\\t\\x08\\xe8\\x07\\x10\\x80\\x80\\x80\\x80\\x02\\\"\\x93\\x01\\n\\x0b\\x45numOptions\\x12\\x13\\n\\x0b\\x61llow_alias\\x18\\x02 \\x01(\\x08\\x12\\x19\\n\\ndeprecated\\x18\\x03 \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x43\\n\\x14uninterpreted_option\\x18\\xe7\\x07 \\x03(\\x0b\\x32$.google.protobuf.UninterpretedOption*\\t\\x08\\xe8\\x07\\x10\\x80\\x80\\x80\\x80\\x02J\\x04\\x08\\x05\\x10\\x06\\\"}\\n\\x10\\x45numValueOptions\\x12\\x19\\n\\ndeprecated\\x18\\x01 \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x43\\n\\x14uninterpreted_option\\x18\\xe7\\x07 \\x03(\\x0b\\x32$.google.protobuf.UninterpretedOption*\\t\\x08\\xe8\\x07\\x10\\x80\\x80\\x80\\x80\\x02\\\"{\\n\\x0eServiceOptions\\x12\\x19\\n\\ndeprecated\\x18! \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x43\\n\\x14uninterpreted_option\\x18\\xe7\\x07 \\x03(\\x0b\\x32$.google.protobuf.UninterpretedOption*\\t\\x08\\xe8\\x07\\x10\\x80\\x80\\x80\\x80\\x02\\\"\\xad\\x02\\n\\rMethodOptions\\x12\\x19\\n\\ndeprecated\\x18! \\x01(\\x08:\\x05\\x66\\x61lse\\x12_\\n\\x11idempotency_level\\x18\\\" \\x01(\\x0e\\x32/.google.protobuf.MethodOptions.IdempotencyLevel:\\x13IDEMPOTENCY_UNKNOWN\\x12\\x43\\n\\x14uninterpreted_option\\x18\\xe7\\x07 \\x03(\\x0b\\x32$.google.protobuf.UninterpretedOption\\\"P\\n\\x10IdempotencyLevel\\x12\\x17\\n\\x13IDEMPOTENCY_UNKNOWN\\x10\\x00\\x12\\x13\\n\\x0fNO_SIDE_EFFECTS\\x10\\x01\\x12\\x0e\\n\\nIDEMPOTENT\\x10\\x02*\\t\\x08\\xe8\\x07\\x10\\x80\\x80\\x80\\x80\\x02\\\"\\x9e\\x02\\n\\x13UninterpretedOption\\x12;\\n\\x04name\\x18\\x02 \\x03(\\x0b\\x32-.google.protobuf.UninterpretedOption.NamePart\\x12\\x18\\n\\x10identifier_value\\x18\\x03 \\x01(\\t\\x12\\x1a\\n\\x12positive_int_value\\x18\\x04 \\x01(\\x04\\x12\\x1a\\n\\x12negative_int_value\\x18\\x05 \\x01(\\x03\\x12\\x14\\n\\x0c\\x64ouble_value\\x18\\x06 \\x01(\\x01\\x12\\x14\\n\\x0cstring_value\\x18\\x07 \\x01(\\x0c\\x12\\x17\\n\\x0f\\x61ggregate_value\\x18\\x08 \\x01(\\t\\x1a\\x33\\n\\x08NamePart\\x12\\x11\\n\\tname_part\\x18\\x01 \\x02(\\t\\x12\\x14\\n\\x0cis_extension\\x18\\x02 \\x02(\\x08\\\"\\xd5\\x01\\n\\x0eSourceCodeInfo\\x12:\\n\\x08location\\x18\\x01 \\x03(\\x0b\\x32(.google.protobuf.SourceCodeInfo.Location\\x1a\\x86\\x01\\n\\x08Location\\x12\\x10\\n\\x04path\\x18\\x01 \\x03(\\x05\\x42\\x02\\x10\\x01\\x12\\x10\\n\\x04span\\x18\\x02 \\x03(\\x05\\x42\\x02\\x10\\x01\\x12\\x18\\n\\x10leading_comments\\x18\\x03 \\x01(\\t\\x12\\x19\\n\\x11trailing_comments\\x18\\x04 \\x01(\\t\\x12!\\n\\x19leading_detached_comments\\x18\\x06 \\x03(\\t\\\"\\xa7\\x01\\n\\x11GeneratedCodeInfo\\x12\\x41\\n\\nannotation\\x18\\x01 \\x03(\\x0b\\x32-.google.protobuf.GeneratedCodeInfo.Annotation\\x1aO\\n\\nAnnotation\\x12\\x10\\n\\x04path\\x18\\x01 \\x03(\\x05\\x42\\x02\\x10\\x01\\x12\\x13\\n\\x0bsource_file\\x18\\x02 \\x01(\\t\\x12\\r\\n\\x05\\x62\\x65gin\\x18\\x03 \\x01(\\x05\\x12\\x0b\\n\\x03\\x65nd\\x18\\x04 \\x01(\\x05\\x42~\\n\\x13\\x63om.google.protobufB\\x10\\x44\\x65scriptorProtosH\\x01Z-google.golang.org/protobuf/types/descriptorpb\\xf8\\x01\\x01\\xa2\\x02\\x03GPB\\xaa\\x02\\x1aGoogle.Protobuf.Reflection'\n )\nelse:\n DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\\n google/protobuf/descriptor.proto\\x12\\x0fgoogle.protobuf\\\"G\\n\\x11\\x46ileDescriptorSet\\x12\\x32\\n\\x04\\x66ile\\x18\\x01 \\x03(\\x0b\\x32$.google.protobuf.FileDescriptorProto\\\"\\xdb\\x03\\n\\x13\\x46ileDescriptorProto\\x12\\x0c\\n\\x04name\\x18\\x01 \\x01(\\t\\x12\\x0f\\n\\x07package\\x18\\x02 \\x01(\\t\\x12\\x12\\n\\ndependency\\x18\\x03 \\x03(\\t\\x12\\x19\\n\\x11public_dependency\\x18\\n \\x03(\\x05\\x12\\x17\\n\\x0fweak_dependency\\x18\\x0b \\x03(\\x05\\x12\\x36\\n\\x0cmessage_type\\x18\\x04 \\x03(\\x0b\\x32 .google.protobuf.DescriptorProto\\x12\\x37\\n\\tenum_type\\x18\\x05 \\x03(\\x0b\\x32$.google.protobuf.EnumDescriptorProto\\x12\\x38\\n\\x07service\\x18\\x06 \\x03(\\x0b\\x32\\'.google.protobuf.ServiceDescriptorProto\\x12\\x38\\n\\textension\\x18\\x07 \\x03(\\x0b\\x32%.google.protobuf.FieldDescriptorProto\\x12-\\n\\x07options\\x18\\x08 \\x01(\\x0b\\x32\\x1c.google.protobuf.FileOptions\\x12\\x39\\n\\x10source_code_info\\x18\\t \\x01(\\x0b\\x32\\x1f.google.protobuf.SourceCodeInfo\\x12\\x0e\\n\\x06syntax\\x18\\x0c \\x01(\\t\\\"\\xa9\\x05\\n\\x0f\\x44\\x65scriptorProto\\x12\\x0c\\n\\x04name\\x18\\x01 \\x01(\\t\\x12\\x34\\n\\x05\\x66ield\\x18\\x02 \\x03(\\x0b\\x32%.google.protobuf.FieldDescriptorProto\\x12\\x38\\n\\textension\\x18\\x06 \\x03(\\x0b\\x32%.google.protobuf.FieldDescriptorProto\\x12\\x35\\n\\x0bnested_type\\x18\\x03 \\x03(\\x0b\\x32 .google.protobuf.DescriptorProto\\x12\\x37\\n\\tenum_type\\x18\\x04 \\x03(\\x0b\\x32$.google.protobuf.EnumDescriptorProto\\x12H\\n\\x0f\\x65xtension_range\\x18\\x05 \\x03(\\x0b\\x32/.google.protobuf.DescriptorProto.ExtensionRange\\x12\\x39\\n\\noneof_decl\\x18\\x08 \\x03(\\x0b\\x32%.google.protobuf.OneofDescriptorProto\\x12\\x30\\n\\x07options\\x18\\x07 \\x01(\\x0b\\x32\\x1f.google.protobuf.MessageOptions\\x12\\x46\\n\\x0ereserved_range\\x18\\t \\x03(\\x0b\\x32..google.protobuf.DescriptorProto.ReservedRange\\x12\\x15\\n\\rreserved_name\\x18\\n \\x03(\\t\\x1a\\x65\\n\\x0e\\x45xtensionRange\\x12\\r\\n\\x05start\\x18\\x01 \\x01(\\x05\\x12\\x0b\\n\\x03\\x65nd\\x18\\x02 \\x01(\\x05\\x12\\x37\\n\\x07options\\x18\\x03 \\x01(\\x0b\\x32&.google.protobuf.ExtensionRangeOptions\\x1a+\\n\\rReservedRange\\x12\\r\\n\\x05start\\x18\\x01 \\x01(\\x05\\x12\\x0b\\n\\x03\\x65nd\\x18\\x02 \\x01(\\x05\\\"g\\n\\x15\\x45xtensionRangeOptions\\x12\\x43\\n\\x14uninterpreted_option\\x18\\xe7\\x07 \\x03(\\x0b\\x32$.google.protobuf.UninterpretedOption*\\t\\x08\\xe8\\x07\\x10\\x80\\x80\\x80\\x80\\x02\\\"\\xd5\\x05\\n\\x14\\x46ieldDescriptorProto\\x12\\x0c\\n\\x04name\\x18\\x01 \\x01(\\t\\x12\\x0e\\n\\x06number\\x18\\x03 \\x01(\\x05\\x12:\\n\\x05label\\x18\\x04 \\x01(\\x0e\\x32+.google.protobuf.FieldDescriptorProto.Label\\x12\\x38\\n\\x04type\\x18\\x05 \\x01(\\x0e\\x32*.google.protobuf.FieldDescriptorProto.Type\\x12\\x11\\n\\ttype_name\\x18\\x06 \\x01(\\t\\x12\\x10\\n\\x08\\x65xtendee\\x18\\x02 \\x01(\\t\\x12\\x15\\n\\rdefault_value\\x18\\x07 \\x01(\\t\\x12\\x13\\n\\x0boneof_index\\x18\\t \\x01(\\x05\\x12\\x11\\n\\tjson_name\\x18\\n \\x01(\\t\\x12.\\n\\x07options\\x18\\x08 \\x01(\\x0b\\x32\\x1d.google.protobuf.FieldOptions\\x12\\x17\\n\\x0fproto3_optional\\x18\\x11 \\x01(\\x08\\\"\\xb6\\x02\\n\\x04Type\\x12\\x0f\\n\\x0bTYPE_DOUBLE\\x10\\x01\\x12\\x0e\\n\\nTYPE_FLOAT\\x10\\x02\\x12\\x0e\\n\\nTYPE_INT64\\x10\\x03\\x12\\x0f\\n\\x0bTYPE_UINT64\\x10\\x04\\x12\\x0e\\n\\nTYPE_INT32\\x10\\x05\\x12\\x10\\n\\x0cTYPE_FIXED64\\x10\\x06\\x12\\x10\\n\\x0cTYPE_FIXED32\\x10\\x07\\x12\\r\\n\\tTYPE_BOOL\\x10\\x08\\x12\\x0f\\n\\x0bTYPE_STRING\\x10\\t\\x12\\x0e\\n\\nTYPE_GROUP\\x10\\n\\x12\\x10\\n\\x0cTYPE_MESSAGE\\x10\\x0b\\x12\\x0e\\n\\nTYPE_BYTES\\x10\\x0c\\x12\\x0f\\n\\x0bTYPE_UINT32\\x10\\r\\x12\\r\\n\\tTYPE_ENUM\\x10\\x0e\\x12\\x11\\n\\rTYPE_SFIXED32\\x10\\x0f\\x12\\x11\\n\\rTYPE_SFIXED64\\x10\\x10\\x12\\x0f\\n\\x0bTYPE_SINT32\\x10\\x11\\x12\\x0f\\n\\x0bTYPE_SINT64\\x10\\x12\\\"C\\n\\x05Label\\x12\\x12\\n\\x0eLABEL_OPTIONAL\\x10\\x01\\x12\\x12\\n\\x0eLABEL_REQUIRED\\x10\\x02\\x12\\x12\\n\\x0eLABEL_REPEATED\\x10\\x03\\\"T\\n\\x14OneofDescriptorProto\\x12\\x0c\\n\\x04name\\x18\\x01 \\x01(\\t\\x12.\\n\\x07options\\x18\\x02 \\x01(\\x0b\\x32\\x1d.google.protobuf.OneofOptions\\\"\\xa4\\x02\\n\\x13\\x45numDescriptorProto\\x12\\x0c\\n\\x04name\\x18\\x01 \\x01(\\t\\x12\\x38\\n\\x05value\\x18\\x02 \\x03(\\x0b\\x32).google.protobuf.EnumValueDescriptorProto\\x12-\\n\\x07options\\x18\\x03 \\x01(\\x0b\\x32\\x1c.google.protobuf.EnumOptions\\x12N\\n\\x0ereserved_range\\x18\\x04 \\x03(\\x0b\\x32\\x36.google.protobuf.EnumDescriptorProto.EnumReservedRange\\x12\\x15\\n\\rreserved_name\\x18\\x05 \\x03(\\t\\x1a/\\n\\x11\\x45numReservedRange\\x12\\r\\n\\x05start\\x18\\x01 \\x01(\\x05\\x12\\x0b\\n\\x03\\x65nd\\x18\\x02 \\x01(\\x05\\\"l\\n\\x18\\x45numValueDescriptorProto\\x12\\x0c\\n\\x04name\\x18\\x01 \\x01(\\t\\x12\\x0e\\n\\x06number\\x18\\x02 \\x01(\\x05\\x12\\x32\\n\\x07options\\x18\\x03 \\x01(\\x0b\\x32!.google.protobuf.EnumValueOptions\\\"\\x90\\x01\\n\\x16ServiceDescriptorProto\\x12\\x0c\\n\\x04name\\x18\\x01 \\x01(\\t\\x12\\x36\\n\\x06method\\x18\\x02 \\x03(\\x0b\\x32&.google.protobuf.MethodDescriptorProto\\x12\\x30\\n\\x07options\\x18\\x03 \\x01(\\x0b\\x32\\x1f.google.protobuf.ServiceOptions\\\"\\xc1\\x01\\n\\x15MethodDescriptorProto\\x12\\x0c\\n\\x04name\\x18\\x01 \\x01(\\t\\x12\\x12\\n\\ninput_type\\x18\\x02 \\x01(\\t\\x12\\x13\\n\\x0boutput_type\\x18\\x03 \\x01(\\t\\x12/\\n\\x07options\\x18\\x04 \\x01(\\x0b\\x32\\x1e.google.protobuf.MethodOptions\\x12\\x1f\\n\\x10\\x63lient_streaming\\x18\\x05 \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x1f\\n\\x10server_streaming\\x18\\x06 \\x01(\\x08:\\x05\\x66\\x61lse\\\"\\xa5\\x06\\n\\x0b\\x46ileOptions\\x12\\x14\\n\\x0cjava_package\\x18\\x01 \\x01(\\t\\x12\\x1c\\n\\x14java_outer_classname\\x18\\x08 \\x01(\\t\\x12\\\"\\n\\x13java_multiple_files\\x18\\n \\x01(\\x08:\\x05\\x66\\x61lse\\x12)\\n\\x1djava_generate_equals_and_hash\\x18\\x14 \\x01(\\x08\\x42\\x02\\x18\\x01\\x12%\\n\\x16java_string_check_utf8\\x18\\x1b \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x46\\n\\x0coptimize_for\\x18\\t \\x01(\\x0e\\x32).google.protobuf.FileOptions.OptimizeMode:\\x05SPEED\\x12\\x12\\n\\ngo_package\\x18\\x0b \\x01(\\t\\x12\\\"\\n\\x13\\x63\\x63_generic_services\\x18\\x10 \\x01(\\x08:\\x05\\x66\\x61lse\\x12$\\n\\x15java_generic_services\\x18\\x11 \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\\"\\n\\x13py_generic_services\\x18\\x12 \\x01(\\x08:\\x05\\x66\\x61lse\\x12#\\n\\x14php_generic_services\\x18* \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x19\\n\\ndeprecated\\x18\\x17 \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x1e\\n\\x10\\x63\\x63_enable_arenas\\x18\\x1f \\x01(\\x08:\\x04true\\x12\\x19\\n\\x11objc_class_prefix\\x18$ \\x01(\\t\\x12\\x18\\n\\x10\\x63sharp_namespace\\x18% \\x01(\\t\\x12\\x14\\n\\x0cswift_prefix\\x18\\' \\x01(\\t\\x12\\x18\\n\\x10php_class_prefix\\x18( \\x01(\\t\\x12\\x15\\n\\rphp_namespace\\x18) \\x01(\\t\\x12\\x1e\\n\\x16php_metadata_namespace\\x18, \\x01(\\t\\x12\\x14\\n\\x0cruby_package\\x18- \\x01(\\t\\x12\\x43\\n\\x14uninterpreted_option\\x18\\xe7\\x07 \\x03(\\x0b\\x32$.google.protobuf.UninterpretedOption\\\":\\n\\x0cOptimizeMode\\x12\\t\\n\\x05SPEED\\x10\\x01\\x12\\r\\n\\tCODE_SIZE\\x10\\x02\\x12\\x10\\n\\x0cLITE_RUNTIME\\x10\\x03*\\t\\x08\\xe8\\x07\\x10\\x80\\x80\\x80\\x80\\x02J\\x04\\x08&\\x10\\'\\\"\\x84\\x02\\n\\x0eMessageOptions\\x12&\\n\\x17message_set_wire_format\\x18\\x01 \\x01(\\x08:\\x05\\x66\\x61lse\\x12.\\n\\x1fno_standard_descriptor_accessor\\x18\\x02 \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x19\\n\\ndeprecated\\x18\\x03 \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x11\\n\\tmap_entry\\x18\\x07 \\x01(\\x08\\x12\\x43\\n\\x14uninterpreted_option\\x18\\xe7\\x07 \\x03(\\x0b\\x32$.google.protobuf.UninterpretedOption*\\t\\x08\\xe8\\x07\\x10\\x80\\x80\\x80\\x80\\x02J\\x04\\x08\\x04\\x10\\x05J\\x04\\x08\\x05\\x10\\x06J\\x04\\x08\\x06\\x10\\x07J\\x04\\x08\\x08\\x10\\tJ\\x04\\x08\\t\\x10\\n\\\"\\xbe\\x03\\n\\x0c\\x46ieldOptions\\x12:\\n\\x05\\x63type\\x18\\x01 \\x01(\\x0e\\x32#.google.protobuf.FieldOptions.CType:\\x06STRING\\x12\\x0e\\n\\x06packed\\x18\\x02 \\x01(\\x08\\x12?\\n\\x06jstype\\x18\\x06 \\x01(\\x0e\\x32$.google.protobuf.FieldOptions.JSType:\\tJS_NORMAL\\x12\\x13\\n\\x04lazy\\x18\\x05 \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x1e\\n\\x0funverified_lazy\\x18\\x0f \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x19\\n\\ndeprecated\\x18\\x03 \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x13\\n\\x04weak\\x18\\n \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x43\\n\\x14uninterpreted_option\\x18\\xe7\\x07 \\x03(\\x0b\\x32$.google.protobuf.UninterpretedOption\\\"/\\n\\x05\\x43Type\\x12\\n\\n\\x06STRING\\x10\\x00\\x12\\x08\\n\\x04\\x43ORD\\x10\\x01\\x12\\x10\\n\\x0cSTRING_PIECE\\x10\\x02\\\"5\\n\\x06JSType\\x12\\r\\n\\tJS_NORMAL\\x10\\x00\\x12\\r\\n\\tJS_STRING\\x10\\x01\\x12\\r\\n\\tJS_NUMBER\\x10\\x02*\\t\\x08\\xe8\\x07\\x10\\x80\\x80\\x80\\x80\\x02J\\x04\\x08\\x04\\x10\\x05\\\"^\\n\\x0cOneofOptions\\x12\\x43\\n\\x14uninterpreted_option\\x18\\xe7\\x07 \\x03(\\x0b\\x32$.google.protobuf.UninterpretedOption*\\t\\x08\\xe8\\x07\\x10\\x80\\x80\\x80\\x80\\x02\\\"\\x93\\x01\\n\\x0b\\x45numOptions\\x12\\x13\\n\\x0b\\x61llow_alias\\x18\\x02 \\x01(\\x08\\x12\\x19\\n\\ndeprecated\\x18\\x03 \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x43\\n\\x14uninterpreted_option\\x18\\xe7\\x07 \\x03(\\x0b\\x32$.google.protobuf.UninterpretedOption*\\t\\x08\\xe8\\x07\\x10\\x80\\x80\\x80\\x80\\x02J\\x04\\x08\\x05\\x10\\x06\\\"}\\n\\x10\\x45numValueOptions\\x12\\x19\\n\\ndeprecated\\x18\\x01 \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x43\\n\\x14uninterpreted_option\\x18\\xe7\\x07 \\x03(\\x0b\\x32$.google.protobuf.UninterpretedOption*\\t\\x08\\xe8\\x07\\x10\\x80\\x80\\x80\\x80\\x02\\\"{\\n\\x0eServiceOptions\\x12\\x19\\n\\ndeprecated\\x18! \\x01(\\x08:\\x05\\x66\\x61lse\\x12\\x43\\n\\x14uninterpreted_option\\x18\\xe7\\x07 \\x03(\\x0b\\x32$.google.protobuf.UninterpretedOption*\\t\\x08\\xe8\\x07\\x10\\x80\\x80\\x80\\x80\\x02\\\"\\xad\\x02\\n\\rMethodOptions\\x12\\x19\\n\\ndeprecated\\x18! \\x01(\\x08:\\x05\\x66\\x61lse\\x12_\\n\\x11idempotency_level\\x18\\\" \\x01(\\x0e\\x32/.google.protobuf.MethodOptions.IdempotencyLevel:\\x13IDEMPOTENCY_UNKNOWN\\x12\\x43\\n\\x14uninterpreted_option\\x18\\xe7\\x07 \\x03(\\x0b\\x32$.google.protobuf.UninterpretedOption\\\"P\\n\\x10IdempotencyLevel\\x12\\x17\\n\\x13IDEMPOTENCY_UNKNOWN\\x10\\x00\\x12\\x13\\n\\x0fNO_SIDE_EFFECTS\\x10\\x01\\x12\\x0e\\n\\nIDEMPOTENT\\x10\\x02*\\t\\x08\\xe8\\x07\\x10\\x80\\x80\\x80\\x80\\x02\\\"\\x9e\\x02\\n\\x13UninterpretedOption\\x12;\\n\\x04name\\x18\\x02 \\x03(\\x0b\\x32-.google.protobuf.UninterpretedOption.NamePart\\x12\\x18\\n\\x10identifier_value\\x18\\x03 \\x01(\\t\\x12\\x1a\\n\\x12positive_int_value\\x18\\x04 \\x01(\\x04\\x12\\x1a\\n\\x12negative_int_value\\x18\\x05 \\x01(\\x03\\x12\\x14\\n\\x0c\\x64ouble_value\\x18\\x06 \\x01(\\x01\\x12\\x14\\n\\x0cstring_value\\x18\\x07 \\x01(\\x0c\\x12\\x17\\n\\x0f\\x61ggregate_value\\x18\\x08 \\x01(\\t\\x1a\\x33\\n\\x08NamePart\\x12\\x11\\n\\tname_part\\x18\\x01 \\x02(\\t\\x12\\x14\\n\\x0cis_extension\\x18\\x02 \\x02(\\x08\\\"\\xd5\\x01\\n\\x0eSourceCodeInfo\\x12:\\n\\x08location\\x18\\x01 \\x03(\\x0b\\x32(.google.protobuf.SourceCodeInfo.Location\\x1a\\x86\\x01\\n\\x08Location\\x12\\x10\\n\\x04path\\x18\\x01 \\x03(\\x05\\x42\\x02\\x10\\x01\\x12\\x10\\n\\x04span\\x18\\x02 \\x03(\\x05\\x42\\x02\\x10\\x01\\x12\\x18\\n\\x10leading_comments\\x18\\x03 \\x01(\\t\\x12\\x19\\n\\x11trailing_comments\\x18\\x04 \\x01(\\t\\x12!\\n\\x19leading_detached_comments\\x18\\x06 \\x03(\\t\\\"\\xa7\\x01\\n\\x11GeneratedCodeInfo\\x12\\x41\\n\\nannotation\\x18\\x01 \\x03(\\x0b\\x32-.google.protobuf.GeneratedCodeInfo.Annotation\\x1aO\\n\\nAnnotation\\x12\\x10\\n\\x04path\\x18\\x01 \\x03(\\x05\\x42\\x02\\x10\\x01\\x12\\x13\\n\\x0bsource_file\\x18\\x02 \\x01(\\t\\x12\\r\\n\\x05\\x62\\x65gin\\x18\\x03 \\x01(\\x05\\x12\\x0b\\n\\x03\\x65nd\\x18\\x04 \\x01(\\x05\\x42~\\n\\x13\\x63om.google.protobufB\\x10\\x44\\x65scriptorProtosH\\x01Z-google.golang.org/protobuf/types/descriptorpb\\xf8\\x01\\x01\\xa2\\x02\\x03GPB\\xaa\\x02\\x1aGoogle.Protobuf.Reflection')\n\nif _descriptor._USE_C_DESCRIPTORS == False:\n _FIELDDESCRIPTORPROTO_TYPE = _descriptor.EnumDescriptor(\n name='Type',\n full_name='google.protobuf.FieldDescriptorProto.Type',\n filename=None,\n file=DESCRIPTOR,\n create_key=_descriptor._internal_create_key,\n values=[\n _descriptor.EnumValueDescriptor(\n name='TYPE_DOUBLE', index=0, number=1,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='TYPE_FLOAT', index=1, number=2,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='TYPE_INT64', index=2, number=3,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='TYPE_UINT64', index=3, number=4,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='TYPE_INT32', index=4, number=5,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='TYPE_FIXED64', index=5, number=6,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='TYPE_FIXED32', index=6, number=7,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='TYPE_BOOL', index=7, number=8,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='TYPE_STRING', index=8, number=9,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='TYPE_GROUP', index=9, number=10,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='TYPE_MESSAGE', index=10, number=11,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='TYPE_BYTES', index=11, number=12,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='TYPE_UINT32', index=12, number=13,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='TYPE_ENUM', index=13, number=14,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='TYPE_SFIXED32', index=14, number=15,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='TYPE_SFIXED64', index=15, number=16,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='TYPE_SINT32', index=16, number=17,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='TYPE_SINT64', index=17, number=18,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n ],\n containing_type=None,\n serialized_options=None,\n )\n _sym_db.RegisterEnumDescriptor(_FIELDDESCRIPTORPROTO_TYPE)\n\n _FIELDDESCRIPTORPROTO_LABEL = _descriptor.EnumDescriptor(\n name='Label',\n full_name='google.protobuf.FieldDescriptorProto.Label',\n filename=None,\n file=DESCRIPTOR,\n create_key=_descriptor._internal_create_key,\n values=[\n _descriptor.EnumValueDescriptor(\n name='LABEL_OPTIONAL', index=0, number=1,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='LABEL_REQUIRED', index=1, number=2,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='LABEL_REPEATED', index=2, number=3,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n ],\n containing_type=None,\n serialized_options=None,\n )\n _sym_db.RegisterEnumDescriptor(_FIELDDESCRIPTORPROTO_LABEL)\n\n _FILEOPTIONS_OPTIMIZEMODE = _descriptor.EnumDescriptor(\n name='OptimizeMode',\n full_name='google.protobuf.FileOptions.OptimizeMode',\n filename=None,\n file=DESCRIPTOR,\n create_key=_descriptor._internal_create_key,\n values=[\n _descriptor.EnumValueDescriptor(\n name='SPEED', index=0, number=1,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='CODE_SIZE', index=1, number=2,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='LITE_RUNTIME', index=2, number=3,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n ],\n containing_type=None,\n serialized_options=None,\n )\n _sym_db.RegisterEnumDescriptor(_FILEOPTIONS_OPTIMIZEMODE)\n\n _FIELDOPTIONS_CTYPE = _descriptor.EnumDescriptor(\n name='CType',\n full_name='google.protobuf.FieldOptions.CType',\n filename=None,\n file=DESCRIPTOR,\n create_key=_descriptor._internal_create_key,\n values=[\n _descriptor.EnumValueDescriptor(\n name='STRING', index=0, number=0,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='CORD', index=1, number=1,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='STRING_PIECE', index=2, number=2,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n ],\n containing_type=None,\n serialized_options=None,\n )\n _sym_db.RegisterEnumDescriptor(_FIELDOPTIONS_CTYPE)\n\n _FIELDOPTIONS_JSTYPE = _descriptor.EnumDescriptor(\n name='JSType',\n full_name='google.protobuf.FieldOptions.JSType',\n filename=None,\n file=DESCRIPTOR,\n create_key=_descriptor._internal_create_key,\n values=[\n _descriptor.EnumValueDescriptor(\n name='JS_NORMAL', index=0, number=0,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='JS_STRING', index=1, number=1,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='JS_NUMBER', index=2, number=2,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n ],\n containing_type=None,\n serialized_options=None,\n )\n _sym_db.RegisterEnumDescriptor(_FIELDOPTIONS_JSTYPE)\n\n _METHODOPTIONS_IDEMPOTENCYLEVEL = _descriptor.EnumDescriptor(\n name='IdempotencyLevel',\n full_name='google.protobuf.MethodOptions.IdempotencyLevel',\n filename=None,\n file=DESCRIPTOR,\n create_key=_descriptor._internal_create_key,\n values=[\n _descriptor.EnumValueDescriptor(\n name='IDEMPOTENCY_UNKNOWN', index=0, number=0,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='NO_SIDE_EFFECTS', index=1, number=1,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n _descriptor.EnumValueDescriptor(\n name='IDEMPOTENT', index=2, number=2,\n serialized_options=None,\n type=None,\n create_key=_descriptor._internal_create_key),\n ],\n containing_type=None,\n serialized_options=None,\n )\n _sym_db.RegisterEnumDescriptor(_METHODOPTIONS_IDEMPOTENCYLEVEL)\n\n\n _FILEDESCRIPTORSET = _descriptor.Descriptor(\n name='FileDescriptorSet',\n full_name='google.protobuf.FileDescriptorSet',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='file', full_name='google.protobuf.FileDescriptorSet.file', index=0,\n number=1, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[],\n enum_types=[\n ],\n serialized_options=None,\n is_extendable=False,\n syntax='proto2',\n extension_ranges=[],\n oneofs=[\n ],\n )\n\n\n _FILEDESCRIPTORPROTO = _descriptor.Descriptor(\n name='FileDescriptorProto',\n full_name='google.protobuf.FileDescriptorProto',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='name', full_name='google.protobuf.FileDescriptorProto.name', index=0,\n number=1, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='package', full_name='google.protobuf.FileDescriptorProto.package', index=1,\n number=2, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='dependency', full_name='google.protobuf.FileDescriptorProto.dependency', index=2,\n number=3, type=9, cpp_type=9, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='public_dependency', full_name='google.protobuf.FileDescriptorProto.public_dependency', index=3,\n number=10, type=5, cpp_type=1, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='weak_dependency', full_name='google.protobuf.FileDescriptorProto.weak_dependency', index=4,\n number=11, type=5, cpp_type=1, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='message_type', full_name='google.protobuf.FileDescriptorProto.message_type', index=5,\n number=4, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='enum_type', full_name='google.protobuf.FileDescriptorProto.enum_type', index=6,\n number=5, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='service', full_name='google.protobuf.FileDescriptorProto.service', index=7,\n number=6, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='extension', full_name='google.protobuf.FileDescriptorProto.extension', index=8,\n number=7, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='options', full_name='google.protobuf.FileDescriptorProto.options', index=9,\n number=8, type=11, cpp_type=10, label=1,\n has_default_value=False, default_value=None,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='source_code_info', full_name='google.protobuf.FileDescriptorProto.source_code_info', index=10,\n number=9, type=11, cpp_type=10, label=1,\n has_default_value=False, default_value=None,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='syntax', full_name='google.protobuf.FileDescriptorProto.syntax', index=11,\n number=12, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[],\n enum_types=[\n ],\n serialized_options=None,\n is_extendable=False,\n syntax='proto2',\n extension_ranges=[],\n oneofs=[\n ],\n )\n\n\n _DESCRIPTORPROTO_EXTENSIONRANGE = _descriptor.Descriptor(\n name='ExtensionRange',\n full_name='google.protobuf.DescriptorProto.ExtensionRange',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='start', full_name='google.protobuf.DescriptorProto.ExtensionRange.start', index=0,\n number=1, type=5, cpp_type=1, label=1,\n has_default_value=False, default_value=0,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='end', full_name='google.protobuf.DescriptorProto.ExtensionRange.end', index=1,\n number=2, type=5, cpp_type=1, label=1,\n has_default_value=False, default_value=0,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='options', full_name='google.protobuf.DescriptorProto.ExtensionRange.options', index=2,\n number=3, type=11, cpp_type=10, label=1,\n has_default_value=False, default_value=None,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[],\n enum_types=[\n ],\n serialized_options=None,\n is_extendable=False,\n syntax='proto2',\n extension_ranges=[],\n oneofs=[\n ],\n )\n\n _DESCRIPTORPROTO_RESERVEDRANGE = _descriptor.Descriptor(\n name='ReservedRange',\n full_name='google.protobuf.DescriptorProto.ReservedRange',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='start', full_name='google.protobuf.DescriptorProto.ReservedRange.start', index=0,\n number=1, type=5, cpp_type=1, label=1,\n has_default_value=False, default_value=0,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='end', full_name='google.protobuf.DescriptorProto.ReservedRange.end', index=1,\n number=2, type=5, cpp_type=1, label=1,\n has_default_value=False, default_value=0,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[],\n enum_types=[\n ],\n serialized_options=None,\n is_extendable=False,\n syntax='proto2',\n extension_ranges=[],\n oneofs=[\n ],\n )\n\n _DESCRIPTORPROTO = _descriptor.Descriptor(\n name='DescriptorProto',\n full_name='google.protobuf.DescriptorProto',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='name', full_name='google.protobuf.DescriptorProto.name', index=0,\n number=1, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='field', full_name='google.protobuf.DescriptorProto.field', index=1,\n number=2, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='extension', full_name='google.protobuf.DescriptorProto.extension', index=2,\n number=6, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='nested_type', full_name='google.protobuf.DescriptorProto.nested_type', index=3,\n number=3, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='enum_type', full_name='google.protobuf.DescriptorProto.enum_type', index=4,\n number=4, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='extension_range', full_name='google.protobuf.DescriptorProto.extension_range', index=5,\n number=5, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='oneof_decl', full_name='google.protobuf.DescriptorProto.oneof_decl', index=6,\n number=8, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='options', full_name='google.protobuf.DescriptorProto.options', index=7,\n number=7, type=11, cpp_type=10, label=1,\n has_default_value=False, default_value=None,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='reserved_range', full_name='google.protobuf.DescriptorProto.reserved_range', index=8,\n number=9, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='reserved_name', full_name='google.protobuf.DescriptorProto.reserved_name', index=9,\n number=10, type=9, cpp_type=9, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[_DESCRIPTORPROTO_EXTENSIONRANGE, _DESCRIPTORPROTO_RESERVEDRANGE, ],\n enum_types=[\n ],\n serialized_options=None,\n is_extendable=False,\n syntax='proto2',\n extension_ranges=[],\n oneofs=[\n ],\n )\n\n\n _EXTENSIONRANGEOPTIONS = _descriptor.Descriptor(\n name='ExtensionRangeOptions',\n full_name='google.protobuf.ExtensionRangeOptions',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='uninterpreted_option', full_name='google.protobuf.ExtensionRangeOptions.uninterpreted_option', index=0,\n number=999, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[],\n enum_types=[\n ],\n serialized_options=None,\n is_extendable=True,\n syntax='proto2',\n extension_ranges=[(1000, 536870912), ],\n oneofs=[\n ],\n )\n\n\n _FIELDDESCRIPTORPROTO = _descriptor.Descriptor(\n name='FieldDescriptorProto',\n full_name='google.protobuf.FieldDescriptorProto',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='name', full_name='google.protobuf.FieldDescriptorProto.name', index=0,\n number=1, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='number', full_name='google.protobuf.FieldDescriptorProto.number', index=1,\n number=3, type=5, cpp_type=1, label=1,\n has_default_value=False, default_value=0,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='label', full_name='google.protobuf.FieldDescriptorProto.label', index=2,\n number=4, type=14, cpp_type=8, label=1,\n has_default_value=False, default_value=1,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='type', full_name='google.protobuf.FieldDescriptorProto.type', index=3,\n number=5, type=14, cpp_type=8, label=1,\n has_default_value=False, default_value=1,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='type_name', full_name='google.protobuf.FieldDescriptorProto.type_name', index=4,\n number=6, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='extendee', full_name='google.protobuf.FieldDescriptorProto.extendee', index=5,\n number=2, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='default_value', full_name='google.protobuf.FieldDescriptorProto.default_value', index=6,\n number=7, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='oneof_index', full_name='google.protobuf.FieldDescriptorProto.oneof_index', index=7,\n number=9, type=5, cpp_type=1, label=1,\n has_default_value=False, default_value=0,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='json_name', full_name='google.protobuf.FieldDescriptorProto.json_name', index=8,\n number=10, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='options', full_name='google.protobuf.FieldDescriptorProto.options', index=9,\n number=8, type=11, cpp_type=10, label=1,\n has_default_value=False, default_value=None,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='proto3_optional', full_name='google.protobuf.FieldDescriptorProto.proto3_optional', index=10,\n number=17, type=8, cpp_type=7, label=1,\n has_default_value=False, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[],\n enum_types=[\n _FIELDDESCRIPTORPROTO_TYPE,\n _FIELDDESCRIPTORPROTO_LABEL,\n ],\n serialized_options=None,\n is_extendable=False,\n syntax='proto2',\n extension_ranges=[],\n oneofs=[\n ],\n )\n\n\n _ONEOFDESCRIPTORPROTO = _descriptor.Descriptor(\n name='OneofDescriptorProto',\n full_name='google.protobuf.OneofDescriptorProto',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='name', full_name='google.protobuf.OneofDescriptorProto.name', index=0,\n number=1, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='options', full_name='google.protobuf.OneofDescriptorProto.options', index=1,\n number=2, type=11, cpp_type=10, label=1,\n has_default_value=False, default_value=None,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[],\n enum_types=[\n ],\n serialized_options=None,\n is_extendable=False,\n syntax='proto2',\n extension_ranges=[],\n oneofs=[\n ],\n )\n\n\n _ENUMDESCRIPTORPROTO_ENUMRESERVEDRANGE = _descriptor.Descriptor(\n name='EnumReservedRange',\n full_name='google.protobuf.EnumDescriptorProto.EnumReservedRange',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='start', full_name='google.protobuf.EnumDescriptorProto.EnumReservedRange.start', index=0,\n number=1, type=5, cpp_type=1, label=1,\n has_default_value=False, default_value=0,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='end', full_name='google.protobuf.EnumDescriptorProto.EnumReservedRange.end', index=1,\n number=2, type=5, cpp_type=1, label=1,\n has_default_value=False, default_value=0,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[],\n enum_types=[\n ],\n serialized_options=None,\n is_extendable=False,\n syntax='proto2',\n extension_ranges=[],\n oneofs=[\n ],\n )\n\n _ENUMDESCRIPTORPROTO = _descriptor.Descriptor(\n name='EnumDescriptorProto',\n full_name='google.protobuf.EnumDescriptorProto',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='name', full_name='google.protobuf.EnumDescriptorProto.name', index=0,\n number=1, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='value', full_name='google.protobuf.EnumDescriptorProto.value', index=1,\n number=2, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='options', full_name='google.protobuf.EnumDescriptorProto.options', index=2,\n number=3, type=11, cpp_type=10, label=1,\n has_default_value=False, default_value=None,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='reserved_range', full_name='google.protobuf.EnumDescriptorProto.reserved_range', index=3,\n number=4, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='reserved_name', full_name='google.protobuf.EnumDescriptorProto.reserved_name', index=4,\n number=5, type=9, cpp_type=9, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[_ENUMDESCRIPTORPROTO_ENUMRESERVEDRANGE, ],\n enum_types=[\n ],\n serialized_options=None,\n is_extendable=False,\n syntax='proto2',\n extension_ranges=[],\n oneofs=[\n ],\n )\n\n\n _ENUMVALUEDESCRIPTORPROTO = _descriptor.Descriptor(\n name='EnumValueDescriptorProto',\n full_name='google.protobuf.EnumValueDescriptorProto',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='name', full_name='google.protobuf.EnumValueDescriptorProto.name', index=0,\n number=1, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='number', full_name='google.protobuf.EnumValueDescriptorProto.number', index=1,\n number=2, type=5, cpp_type=1, label=1,\n has_default_value=False, default_value=0,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='options', full_name='google.protobuf.EnumValueDescriptorProto.options', index=2,\n number=3, type=11, cpp_type=10, label=1,\n has_default_value=False, default_value=None,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[],\n enum_types=[\n ],\n serialized_options=None,\n is_extendable=False,\n syntax='proto2',\n extension_ranges=[],\n oneofs=[\n ],\n )\n\n\n _SERVICEDESCRIPTORPROTO = _descriptor.Descriptor(\n name='ServiceDescriptorProto',\n full_name='google.protobuf.ServiceDescriptorProto',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='name', full_name='google.protobuf.ServiceDescriptorProto.name', index=0,\n number=1, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='method', full_name='google.protobuf.ServiceDescriptorProto.method', index=1,\n number=2, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='options', full_name='google.protobuf.ServiceDescriptorProto.options', index=2,\n number=3, type=11, cpp_type=10, label=1,\n has_default_value=False, default_value=None,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[],\n enum_types=[\n ],\n serialized_options=None,\n is_extendable=False,\n syntax='proto2',\n extension_ranges=[],\n oneofs=[\n ],\n )\n\n\n _METHODDESCRIPTORPROTO = _descriptor.Descriptor(\n name='MethodDescriptorProto',\n full_name='google.protobuf.MethodDescriptorProto',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='name', full_name='google.protobuf.MethodDescriptorProto.name', index=0,\n number=1, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='input_type', full_name='google.protobuf.MethodDescriptorProto.input_type', index=1,\n number=2, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='output_type', full_name='google.protobuf.MethodDescriptorProto.output_type', index=2,\n number=3, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='options', full_name='google.protobuf.MethodDescriptorProto.options', index=3,\n number=4, type=11, cpp_type=10, label=1,\n has_default_value=False, default_value=None,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='client_streaming', full_name='google.protobuf.MethodDescriptorProto.client_streaming', index=4,\n number=5, type=8, cpp_type=7, label=1,\n has_default_value=True, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='server_streaming', full_name='google.protobuf.MethodDescriptorProto.server_streaming', index=5,\n number=6, type=8, cpp_type=7, label=1,\n has_default_value=True, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[],\n enum_types=[\n ],\n serialized_options=None,\n is_extendable=False,\n syntax='proto2',\n extension_ranges=[],\n oneofs=[\n ],\n )\n\n\n _FILEOPTIONS = _descriptor.Descriptor(\n name='FileOptions',\n full_name='google.protobuf.FileOptions',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='java_package', full_name='google.protobuf.FileOptions.java_package', index=0,\n number=1, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='java_outer_classname', full_name='google.protobuf.FileOptions.java_outer_classname', index=1,\n number=8, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='java_multiple_files', full_name='google.protobuf.FileOptions.java_multiple_files', index=2,\n number=10, type=8, cpp_type=7, label=1,\n has_default_value=True, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='java_generate_equals_and_hash', full_name='google.protobuf.FileOptions.java_generate_equals_and_hash', index=3,\n number=20, type=8, cpp_type=7, label=1,\n has_default_value=False, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='java_string_check_utf8', full_name='google.protobuf.FileOptions.java_string_check_utf8', index=4,\n number=27, type=8, cpp_type=7, label=1,\n has_default_value=True, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='optimize_for', full_name='google.protobuf.FileOptions.optimize_for', index=5,\n number=9, type=14, cpp_type=8, label=1,\n has_default_value=True, default_value=1,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='go_package', full_name='google.protobuf.FileOptions.go_package', index=6,\n number=11, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='cc_generic_services', full_name='google.protobuf.FileOptions.cc_generic_services', index=7,\n number=16, type=8, cpp_type=7, label=1,\n has_default_value=True, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='java_generic_services', full_name='google.protobuf.FileOptions.java_generic_services', index=8,\n number=17, type=8, cpp_type=7, label=1,\n has_default_value=True, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='py_generic_services', full_name='google.protobuf.FileOptions.py_generic_services', index=9,\n number=18, type=8, cpp_type=7, label=1,\n has_default_value=True, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='php_generic_services', full_name='google.protobuf.FileOptions.php_generic_services', index=10,\n number=42, type=8, cpp_type=7, label=1,\n has_default_value=True, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='deprecated', full_name='google.protobuf.FileOptions.deprecated', index=11,\n number=23, type=8, cpp_type=7, label=1,\n has_default_value=True, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='cc_enable_arenas', full_name='google.protobuf.FileOptions.cc_enable_arenas', index=12,\n number=31, type=8, cpp_type=7, label=1,\n has_default_value=True, default_value=True,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='objc_class_prefix', full_name='google.protobuf.FileOptions.objc_class_prefix', index=13,\n number=36, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='csharp_namespace', full_name='google.protobuf.FileOptions.csharp_namespace', index=14,\n number=37, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='swift_prefix', full_name='google.protobuf.FileOptions.swift_prefix', index=15,\n number=39, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='php_class_prefix', full_name='google.protobuf.FileOptions.php_class_prefix', index=16,\n number=40, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='php_namespace', full_name='google.protobuf.FileOptions.php_namespace', index=17,\n number=41, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='php_metadata_namespace', full_name='google.protobuf.FileOptions.php_metadata_namespace', index=18,\n number=44, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='ruby_package', full_name='google.protobuf.FileOptions.ruby_package', index=19,\n number=45, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='uninterpreted_option', full_name='google.protobuf.FileOptions.uninterpreted_option', index=20,\n number=999, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[],\n enum_types=[\n _FILEOPTIONS_OPTIMIZEMODE,\n ],\n serialized_options=None,\n is_extendable=True,\n syntax='proto2',\n extension_ranges=[(1000, 536870912), ],\n oneofs=[\n ],\n )\n\n\n _MESSAGEOPTIONS = _descriptor.Descriptor(\n name='MessageOptions',\n full_name='google.protobuf.MessageOptions',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='message_set_wire_format', full_name='google.protobuf.MessageOptions.message_set_wire_format', index=0,\n number=1, type=8, cpp_type=7, label=1,\n has_default_value=True, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='no_standard_descriptor_accessor', full_name='google.protobuf.MessageOptions.no_standard_descriptor_accessor', index=1,\n number=2, type=8, cpp_type=7, label=1,\n has_default_value=True, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='deprecated', full_name='google.protobuf.MessageOptions.deprecated', index=2,\n number=3, type=8, cpp_type=7, label=1,\n has_default_value=True, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='map_entry', full_name='google.protobuf.MessageOptions.map_entry', index=3,\n number=7, type=8, cpp_type=7, label=1,\n has_default_value=False, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='uninterpreted_option', full_name='google.protobuf.MessageOptions.uninterpreted_option', index=4,\n number=999, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[],\n enum_types=[\n ],\n serialized_options=None,\n is_extendable=True,\n syntax='proto2',\n extension_ranges=[(1000, 536870912), ],\n oneofs=[\n ],\n )\n\n\n _FIELDOPTIONS = _descriptor.Descriptor(\n name='FieldOptions',\n full_name='google.protobuf.FieldOptions',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='ctype', full_name='google.protobuf.FieldOptions.ctype', index=0,\n number=1, type=14, cpp_type=8, label=1,\n has_default_value=True, default_value=0,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='packed', full_name='google.protobuf.FieldOptions.packed', index=1,\n number=2, type=8, cpp_type=7, label=1,\n has_default_value=False, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='jstype', full_name='google.protobuf.FieldOptions.jstype', index=2,\n number=6, type=14, cpp_type=8, label=1,\n has_default_value=True, default_value=0,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='lazy', full_name='google.protobuf.FieldOptions.lazy', index=3,\n number=5, type=8, cpp_type=7, label=1,\n has_default_value=True, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='unverified_lazy', full_name='google.protobuf.FieldOptions.unverified_lazy', index=4,\n number=15, type=8, cpp_type=7, label=1,\n has_default_value=True, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='deprecated', full_name='google.protobuf.FieldOptions.deprecated', index=5,\n number=3, type=8, cpp_type=7, label=1,\n has_default_value=True, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='weak', full_name='google.protobuf.FieldOptions.weak', index=6,\n number=10, type=8, cpp_type=7, label=1,\n has_default_value=True, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='uninterpreted_option', full_name='google.protobuf.FieldOptions.uninterpreted_option', index=7,\n number=999, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[],\n enum_types=[\n _FIELDOPTIONS_CTYPE,\n _FIELDOPTIONS_JSTYPE,\n ],\n serialized_options=None,\n is_extendable=True,\n syntax='proto2',\n extension_ranges=[(1000, 536870912), ],\n oneofs=[\n ],\n )\n\n\n _ONEOFOPTIONS = _descriptor.Descriptor(\n name='OneofOptions',\n full_name='google.protobuf.OneofOptions',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='uninterpreted_option', full_name='google.protobuf.OneofOptions.uninterpreted_option', index=0,\n number=999, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[],\n enum_types=[\n ],\n serialized_options=None,\n is_extendable=True,\n syntax='proto2',\n extension_ranges=[(1000, 536870912), ],\n oneofs=[\n ],\n )\n\n\n _ENUMOPTIONS = _descriptor.Descriptor(\n name='EnumOptions',\n full_name='google.protobuf.EnumOptions',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='allow_alias', full_name='google.protobuf.EnumOptions.allow_alias', index=0,\n number=2, type=8, cpp_type=7, label=1,\n has_default_value=False, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='deprecated', full_name='google.protobuf.EnumOptions.deprecated', index=1,\n number=3, type=8, cpp_type=7, label=1,\n has_default_value=True, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='uninterpreted_option', full_name='google.protobuf.EnumOptions.uninterpreted_option', index=2,\n number=999, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[],\n enum_types=[\n ],\n serialized_options=None,\n is_extendable=True,\n syntax='proto2',\n extension_ranges=[(1000, 536870912), ],\n oneofs=[\n ],\n )\n\n\n _ENUMVALUEOPTIONS = _descriptor.Descriptor(\n name='EnumValueOptions',\n full_name='google.protobuf.EnumValueOptions',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='deprecated', full_name='google.protobuf.EnumValueOptions.deprecated', index=0,\n number=1, type=8, cpp_type=7, label=1,\n has_default_value=True, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='uninterpreted_option', full_name='google.protobuf.EnumValueOptions.uninterpreted_option', index=1,\n number=999, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[],\n enum_types=[\n ],\n serialized_options=None,\n is_extendable=True,\n syntax='proto2',\n extension_ranges=[(1000, 536870912), ],\n oneofs=[\n ],\n )\n\n\n _SERVICEOPTIONS = _descriptor.Descriptor(\n name='ServiceOptions',\n full_name='google.protobuf.ServiceOptions',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='deprecated', full_name='google.protobuf.ServiceOptions.deprecated', index=0,\n number=33, type=8, cpp_type=7, label=1,\n has_default_value=True, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='uninterpreted_option', full_name='google.protobuf.ServiceOptions.uninterpreted_option', index=1,\n number=999, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[],\n enum_types=[\n ],\n serialized_options=None,\n is_extendable=True,\n syntax='proto2',\n extension_ranges=[(1000, 536870912), ],\n oneofs=[\n ],\n )\n\n\n _METHODOPTIONS = _descriptor.Descriptor(\n name='MethodOptions',\n full_name='google.protobuf.MethodOptions',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='deprecated', full_name='google.protobuf.MethodOptions.deprecated', index=0,\n number=33, type=8, cpp_type=7, label=1,\n has_default_value=True, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='idempotency_level', full_name='google.protobuf.MethodOptions.idempotency_level', index=1,\n number=34, type=14, cpp_type=8, label=1,\n has_default_value=True, default_value=0,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='uninterpreted_option', full_name='google.protobuf.MethodOptions.uninterpreted_option', index=2,\n number=999, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[],\n enum_types=[\n _METHODOPTIONS_IDEMPOTENCYLEVEL,\n ],\n serialized_options=None,\n is_extendable=True,\n syntax='proto2',\n extension_ranges=[(1000, 536870912), ],\n oneofs=[\n ],\n )\n\n\n _UNINTERPRETEDOPTION_NAMEPART = _descriptor.Descriptor(\n name='NamePart',\n full_name='google.protobuf.UninterpretedOption.NamePart',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='name_part', full_name='google.protobuf.UninterpretedOption.NamePart.name_part', index=0,\n number=1, type=9, cpp_type=9, label=2,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='is_extension', full_name='google.protobuf.UninterpretedOption.NamePart.is_extension', index=1,\n number=2, type=8, cpp_type=7, label=2,\n has_default_value=False, default_value=False,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[],\n enum_types=[\n ],\n serialized_options=None,\n is_extendable=False,\n syntax='proto2',\n extension_ranges=[],\n oneofs=[\n ],\n )\n\n _UNINTERPRETEDOPTION = _descriptor.Descriptor(\n name='UninterpretedOption',\n full_name='google.protobuf.UninterpretedOption',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='name', full_name='google.protobuf.UninterpretedOption.name', index=0,\n number=2, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='identifier_value', full_name='google.protobuf.UninterpretedOption.identifier_value', index=1,\n number=3, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='positive_int_value', full_name='google.protobuf.UninterpretedOption.positive_int_value', index=2,\n number=4, type=4, cpp_type=4, label=1,\n has_default_value=False, default_value=0,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='negative_int_value', full_name='google.protobuf.UninterpretedOption.negative_int_value', index=3,\n number=5, type=3, cpp_type=2, label=1,\n has_default_value=False, default_value=0,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='double_value', full_name='google.protobuf.UninterpretedOption.double_value', index=4,\n number=6, type=1, cpp_type=5, label=1,\n has_default_value=False, default_value=float(0),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='string_value', full_name='google.protobuf.UninterpretedOption.string_value', index=5,\n number=7, type=12, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\",\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='aggregate_value', full_name='google.protobuf.UninterpretedOption.aggregate_value', index=6,\n number=8, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[_UNINTERPRETEDOPTION_NAMEPART, ],\n enum_types=[\n ],\n serialized_options=None,\n is_extendable=False,\n syntax='proto2',\n extension_ranges=[],\n oneofs=[\n ],\n )\n\n\n _SOURCECODEINFO_LOCATION = _descriptor.Descriptor(\n name='Location',\n full_name='google.protobuf.SourceCodeInfo.Location',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='path', full_name='google.protobuf.SourceCodeInfo.Location.path', index=0,\n number=1, type=5, cpp_type=1, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='span', full_name='google.protobuf.SourceCodeInfo.Location.span', index=1,\n number=2, type=5, cpp_type=1, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='leading_comments', full_name='google.protobuf.SourceCodeInfo.Location.leading_comments', index=2,\n number=3, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='trailing_comments', full_name='google.protobuf.SourceCodeInfo.Location.trailing_comments', index=3,\n number=4, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='leading_detached_comments', full_name='google.protobuf.SourceCodeInfo.Location.leading_detached_comments', index=4,\n number=6, type=9, cpp_type=9, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[],\n enum_types=[\n ],\n serialized_options=None,\n is_extendable=False,\n syntax='proto2',\n extension_ranges=[],\n oneofs=[\n ],\n )\n\n _SOURCECODEINFO = _descriptor.Descriptor(\n name='SourceCodeInfo',\n full_name='google.protobuf.SourceCodeInfo',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='location', full_name='google.protobuf.SourceCodeInfo.location', index=0,\n number=1, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[_SOURCECODEINFO_LOCATION, ],\n enum_types=[\n ],\n serialized_options=None,\n is_extendable=False,\n syntax='proto2',\n extension_ranges=[],\n oneofs=[\n ],\n )\n\n\n _GENERATEDCODEINFO_ANNOTATION = _descriptor.Descriptor(\n name='Annotation',\n full_name='google.protobuf.GeneratedCodeInfo.Annotation',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='path', full_name='google.protobuf.GeneratedCodeInfo.Annotation.path', index=0,\n number=1, type=5, cpp_type=1, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='source_file', full_name='google.protobuf.GeneratedCodeInfo.Annotation.source_file', index=1,\n number=2, type=9, cpp_type=9, label=1,\n has_default_value=False, default_value=b\"\".decode('utf-8'),\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='begin', full_name='google.protobuf.GeneratedCodeInfo.Annotation.begin', index=2,\n number=3, type=5, cpp_type=1, label=1,\n has_default_value=False, default_value=0,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n _descriptor.FieldDescriptor(\n name='end', full_name='google.protobuf.GeneratedCodeInfo.Annotation.end', index=3,\n number=4, type=5, cpp_type=1, label=1,\n has_default_value=False, default_value=0,\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[],\n enum_types=[\n ],\n serialized_options=None,\n is_extendable=False,\n syntax='proto2',\n extension_ranges=[],\n oneofs=[\n ],\n )\n\n _GENERATEDCODEINFO = _descriptor.Descriptor(\n name='GeneratedCodeInfo',\n full_name='google.protobuf.GeneratedCodeInfo',\n filename=None,\n file=DESCRIPTOR,\n containing_type=None,\n create_key=_descriptor._internal_create_key,\n fields=[\n _descriptor.FieldDescriptor(\n name='annotation', full_name='google.protobuf.GeneratedCodeInfo.annotation', index=0,\n number=1, type=11, cpp_type=10, label=3,\n has_default_value=False, default_value=[],\n message_type=None, enum_type=None, containing_type=None,\n is_extension=False, extension_scope=None,\n serialized_options=None, file=DESCRIPTOR, create_key=_descriptor._internal_create_key),\n ],\n extensions=[\n ],\n nested_types=[_GENERATEDCODEINFO_ANNOTATION, ],\n enum_types=[\n ],\n serialized_options=None,\n is_extendable=False,\n syntax='proto2',\n extension_ranges=[],\n oneofs=[\n ],\n )\n\n _FILEDESCRIPTORSET.fields_by_name['file'].message_type = _FILEDESCRIPTORPROTO\n _FILEDESCRIPTORPROTO.fields_by_name['message_type'].message_type = _DESCRIPTORPROTO\n _FILEDESCRIPTORPROTO.fields_by_name['enum_type'].message_type = _ENUMDESCRIPTORPROTO\n _FILEDESCRIPTORPROTO.fields_by_name['service'].message_type = _SERVICEDESCRIPTORPROTO\n _FILEDESCRIPTORPROTO.fields_by_name['extension'].message_type = _FIELDDESCRIPTORPROTO\n _FILEDESCRIPTORPROTO.fields_by_name['options'].message_type = _FILEOPTIONS\n _FILEDESCRIPTORPROTO.fields_by_name['source_code_info'].message_type = _SOURCECODEINFO\n _DESCRIPTORPROTO_EXTENSIONRANGE.fields_by_name['options'].message_type = _EXTENSIONRANGEOPTIONS\n _DESCRIPTORPROTO_EXTENSIONRANGE.containing_type = _DESCRIPTORPROTO\n _DESCRIPTORPROTO_RESERVEDRANGE.containing_type = _DESCRIPTORPROTO\n _DESCRIPTORPROTO.fields_by_name['field'].message_type = _FIELDDESCRIPTORPROTO\n _DESCRIPTORPROTO.fields_by_name['extension'].message_type = _FIELDDESCRIPTORPROTO\n _DESCRIPTORPROTO.fields_by_name['nested_type'].message_type = _DESCRIPTORPROTO\n _DESCRIPTORPROTO.fields_by_name['enum_type'].message_type = _ENUMDESCRIPTORPROTO\n _DESCRIPTORPROTO.fields_by_name['extension_range'].message_type = _DESCRIPTORPROTO_EXTENSIONRANGE\n _DESCRIPTORPROTO.fields_by_name['oneof_decl'].message_type = _ONEOFDESCRIPTORPROTO\n _DESCRIPTORPROTO.fields_by_name['options'].message_type = _MESSAGEOPTIONS\n _DESCRIPTORPROTO.fields_by_name['reserved_range'].message_type = _DESCRIPTORPROTO_RESERVEDRANGE\n _EXTENSIONRANGEOPTIONS.fields_by_name['uninterpreted_option'].message_type = _UNINTERPRETEDOPTION\n _FIELDDESCRIPTORPROTO.fields_by_name['label'].enum_type = _FIELDDESCRIPTORPROTO_LABEL\n _FIELDDESCRIPTORPROTO.fields_by_name['type'].enum_type = _FIELDDESCRIPTORPROTO_TYPE\n _FIELDDESCRIPTORPROTO.fields_by_name['options'].message_type = _FIELDOPTIONS\n _FIELDDESCRIPTORPROTO_TYPE.containing_type = _FIELDDESCRIPTORPROTO\n _FIELDDESCRIPTORPROTO_LABEL.containing_type = _FIELDDESCRIPTORPROTO\n _ONEOFDESCRIPTORPROTO.fields_by_name['options'].message_type = _ONEOFOPTIONS\n _ENUMDESCRIPTORPROTO_ENUMRESERVEDRANGE.containing_type = _ENUMDESCRIPTORPROTO\n _ENUMDESCRIPTORPROTO.fields_by_name['value'].message_type = _ENUMVALUEDESCRIPTORPROTO\n _ENUMDESCRIPTORPROTO.fields_by_name['options'].message_type = _ENUMOPTIONS\n _ENUMDESCRIPTORPROTO.fields_by_name['reserved_range'].message_type = _ENUMDESCRIPTORPROTO_ENUMRESERVEDRANGE\n _ENUMVALUEDESCRIPTORPROTO.fields_by_name['options'].message_type = _ENUMVALUEOPTIONS\n _SERVICEDESCRIPTORPROTO.fields_by_name['method'].message_type = _METHODDESCRIPTORPROTO\n _SERVICEDESCRIPTORPROTO.fields_by_name['options'].message_type = _SERVICEOPTIONS\n _METHODDESCRIPTORPROTO.fields_by_name['options'].message_type = _METHODOPTIONS\n _FILEOPTIONS.fields_by_name['optimize_for'].enum_type = _FILEOPTIONS_OPTIMIZEMODE\n _FILEOPTIONS.fields_by_name['uninterpreted_option'].message_type = _UNINTERPRETEDOPTION\n _FILEOPTIONS_OPTIMIZEMODE.containing_type = _FILEOPTIONS\n _MESSAGEOPTIONS.fields_by_name['uninterpreted_option'].message_type = _UNINTERPRETEDOPTION\n _FIELDOPTIONS.fields_by_name['ctype'].enum_type = _FIELDOPTIONS_CTYPE\n _FIELDOPTIONS.fields_by_name['jstype'].enum_type = _FIELDOPTIONS_JSTYPE\n _FIELDOPTIONS.fields_by_name['uninterpreted_option'].message_type = _UNINTERPRETEDOPTION\n _FIELDOPTIONS_CTYPE.containing_type = _FIELDOPTIONS\n _FIELDOPTIONS_JSTYPE.containing_type = _FIELDOPTIONS\n _ONEOFOPTIONS.fields_by_name['uninterpreted_option'].message_type = _UNINTERPRETEDOPTION\n _ENUMOPTIONS.fields_by_name['uninterpreted_option'].message_type = _UNINTERPRETEDOPTION\n _ENUMVALUEOPTIONS.fields_by_name['uninterpreted_option'].message_type = _UNINTERPRETEDOPTION\n _SERVICEOPTIONS.fields_by_name['uninterpreted_option'].message_type = _UNINTERPRETEDOPTION\n _METHODOPTIONS.fields_by_name['idempotency_level'].enum_type = _METHODOPTIONS_IDEMPOTENCYLEVEL\n _METHODOPTIONS.fields_by_name['uninterpreted_option'].message_type = _UNINTERPRETEDOPTION\n _METHODOPTIONS_IDEMPOTENCYLEVEL.containing_type = _METHODOPTIONS\n _UNINTERPRETEDOPTION_NAMEPART.containing_type = _UNINTERPRETEDOPTION\n _UNINTERPRETEDOPTION.fields_by_name['name'].message_type = _UNINTERPRETEDOPTION_NAMEPART\n _SOURCECODEINFO_LOCATION.containing_type = _SOURCECODEINFO\n _SOURCECODEINFO.fields_by_name['location'].message_type = _SOURCECODEINFO_LOCATION\n _GENERATEDCODEINFO_ANNOTATION.containing_type = _GENERATEDCODEINFO\n _GENERATEDCODEINFO.fields_by_name['annotation'].message_type = _GENERATEDCODEINFO_ANNOTATION\n DESCRIPTOR.message_types_by_name['FileDescriptorSet'] = _FILEDESCRIPTORSET\n DESCRIPTOR.message_types_by_name['FileDescriptorProto'] = _FILEDESCRIPTORPROTO\n DESCRIPTOR.message_types_by_name['DescriptorProto'] = _DESCRIPTORPROTO\n DESCRIPTOR.message_types_by_name['ExtensionRangeOptions'] = _EXTENSIONRANGEOPTIONS\n DESCRIPTOR.message_types_by_name['FieldDescriptorProto'] = _FIELDDESCRIPTORPROTO\n DESCRIPTOR.message_types_by_name['OneofDescriptorProto'] = _ONEOFDESCRIPTORPROTO\n DESCRIPTOR.message_types_by_name['EnumDescriptorProto'] = _ENUMDESCRIPTORPROTO\n DESCRIPTOR.message_types_by_name['EnumValueDescriptorProto'] = _ENUMVALUEDESCRIPTORPROTO\n DESCRIPTOR.message_types_by_name['ServiceDescriptorProto'] = _SERVICEDESCRIPTORPROTO\n DESCRIPTOR.message_types_by_name['MethodDescriptorProto'] = _METHODDESCRIPTORPROTO\n DESCRIPTOR.message_types_by_name['FileOptions'] = _FILEOPTIONS\n DESCRIPTOR.message_types_by_name['MessageOptions'] = _MESSAGEOPTIONS\n DESCRIPTOR.message_types_by_name['FieldOptions'] = _FIELDOPTIONS\n DESCRIPTOR.message_types_by_name['OneofOptions'] = _ONEOFOPTIONS\n DESCRIPTOR.message_types_by_name['EnumOptions'] = _ENUMOPTIONS\n DESCRIPTOR.message_types_by_name['EnumValueOptions'] = _ENUMVALUEOPTIONS\n DESCRIPTOR.message_types_by_name['ServiceOptions'] = _SERVICEOPTIONS\n DESCRIPTOR.message_types_by_name['MethodOptions'] = _METHODOPTIONS\n DESCRIPTOR.message_types_by_name['UninterpretedOption'] = _UNINTERPRETEDOPTION\n DESCRIPTOR.message_types_by_name['SourceCodeInfo'] = _SOURCECODEINFO\n DESCRIPTOR.message_types_by_name['GeneratedCodeInfo'] = _GENERATEDCODEINFO\n _sym_db.RegisterFileDescriptor(DESCRIPTOR)\n\nelse:\n _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())\n_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.descriptor_pb2', globals())\nif _descriptor._USE_C_DESCRIPTORS == False:\n\n DESCRIPTOR._options = None\n _FILEDESCRIPTORSET._serialized_start=53\n _FILEDESCRIPTORSET._serialized_end=124\n _FILEDESCRIPTORPROTO._serialized_start=127\n _FILEDESCRIPTORPROTO._serialized_end=602\n _DESCRIPTORPROTO._serialized_start=605\n _DESCRIPTORPROTO._serialized_end=1286\n _DESCRIPTORPROTO_EXTENSIONRANGE._serialized_start=1140\n _DESCRIPTORPROTO_EXTENSIONRANGE._serialized_end=1241\n _DESCRIPTORPROTO_RESERVEDRANGE._serialized_start=1243\n _DESCRIPTORPROTO_RESERVEDRANGE._serialized_end=1286\n _EXTENSIONRANGEOPTIONS._serialized_start=1288\n _EXTENSIONRANGEOPTIONS._serialized_end=1391\n _FIELDDESCRIPTORPROTO._serialized_start=1394\n _FIELDDESCRIPTORPROTO._serialized_end=2119\n _FIELDDESCRIPTORPROTO_TYPE._serialized_start=1740\n _FIELDDESCRIPTORPROTO_TYPE._serialized_end=2050\n _FIELDDESCRIPTORPROTO_LABEL._serialized_start=2052\n _FIELDDESCRIPTORPROTO_LABEL._serialized_end=2119\n _ONEOFDESCRIPTORPROTO._serialized_start=2121\n _ONEOFDESCRIPTORPROTO._serialized_end=2205\n _ENUMDESCRIPTORPROTO._serialized_start=2208\n _ENUMDESCRIPTORPROTO._serialized_end=2500\n _ENUMDESCRIPTORPROTO_ENUMRESERVEDRANGE._serialized_start=2453\n _ENUMDESCRIPTORPROTO_ENUMRESERVEDRANGE._serialized_end=2500\n _ENUMVALUEDESCRIPTORPROTO._serialized_start=2502\n _ENUMVALUEDESCRIPTORPROTO._serialized_end=2610\n _SERVICEDESCRIPTORPROTO._serialized_start=2613\n _SERVICEDESCRIPTORPROTO._serialized_end=2757\n _METHODDESCRIPTORPROTO._serialized_start=2760\n _METHODDESCRIPTORPROTO._serialized_end=2953\n _FILEOPTIONS._serialized_start=2956\n _FILEOPTIONS._serialized_end=3761\n _FILEOPTIONS_OPTIMIZEMODE._serialized_start=3686\n _FILEOPTIONS_OPTIMIZEMODE._serialized_end=3744\n _MESSAGEOPTIONS._serialized_start=3764\n _MESSAGEOPTIONS._serialized_end=4024\n _FIELDOPTIONS._serialized_start=4027\n _FIELDOPTIONS._serialized_end=4473\n _FIELDOPTIONS_CTYPE._serialized_start=4354\n _FIELDOPTIONS_CTYPE._serialized_end=4401\n _FIELDOPTIONS_JSTYPE._serialized_start=4403\n _FIELDOPTIONS_JSTYPE._serialized_end=4456\n _ONEOFOPTIONS._serialized_start=4475\n _ONEOFOPTIONS._serialized_end=4569\n _ENUMOPTIONS._serialized_start=4572\n _ENUMOPTIONS._serialized_end=4719\n _ENUMVALUEOPTIONS._serialized_start=4721\n _ENUMVALUEOPTIONS._serialized_end=4846\n _SERVICEOPTIONS._serialized_start=4848\n _SERVICEOPTIONS._serialized_end=4971\n _METHODOPTIONS._serialized_start=4974\n _METHODOPTIONS._serialized_end=5275\n _METHODOPTIONS_IDEMPOTENCYLEVEL._serialized_start=5184\n _METHODOPTIONS_IDEMPOTENCYLEVEL._serialized_end=5264\n _UNINTERPRETEDOPTION._serialized_start=5278\n _UNINTERPRETEDOPTION._serialized_end=5564\n _UNINTERPRETEDOPTION_NAMEPART._serialized_start=5513\n _UNINTERPRETEDOPTION_NAMEPART._serialized_end=5564\n _SOURCECODEINFO._serialized_start=5567\n _SOURCECODEINFO._serialized_end=5780\n _SOURCECODEINFO_LOCATION._serialized_start=5646\n _SOURCECODEINFO_LOCATION._serialized_end=5780\n _GENERATEDCODEINFO._serialized_start=5783\n _GENERATEDCODEINFO._serialized_end=5950\n _GENERATEDCODEINFO_ANNOTATION._serialized_start=5871\n _GENERATEDCODEINFO_ANNOTATION._serialized_end=5950\n# @@protoc_insertion_point(module_scope)\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/descriptor_pool.py",
"content": "# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n\"\"\"Provides DescriptorPool to use as a container for proto2 descriptors.\n\nThe DescriptorPool is used in conjection with a DescriptorDatabase to maintain\na collection of protocol buffer descriptors for use when dynamically creating\nmessage types at runtime.\n\nFor most applications protocol buffers should be used via modules generated by\nthe protocol buffer compiler tool. This should only be used when the type of\nprotocol buffers used in an application or library cannot be predetermined.\n\nBelow is a straightforward example on how to use this class::\n\n pool = DescriptorPool()\n file_descriptor_protos = [ ... ]\n for file_descriptor_proto in file_descriptor_protos:\n pool.Add(file_descriptor_proto)\n my_message_descriptor = pool.FindMessageTypeByName('some.package.MessageType')\n\nThe message descriptor can be used in conjunction with the message_factory\nmodule in order to create a protocol buffer class that can be encoded and\ndecoded.\n\nIf you want to get a Python class for the specified proto, use the\nhelper functions inside google.protobuf.message_factory\ndirectly instead of this class.\n\"\"\"\n\n__author__ = 'matthewtoia@google.com (Matt Toia)'\n\nimport collections\nimport warnings\n\nfrom google.protobuf import descriptor\nfrom google.protobuf import descriptor_database\nfrom google.protobuf import text_encoding\n\n\n_USE_C_DESCRIPTORS = descriptor._USE_C_DESCRIPTORS # pylint: disable=protected-access\n\n\ndef _Deprecated(func):\n \"\"\"Mark functions as deprecated.\"\"\"\n\n def NewFunc(*args, **kwargs):\n warnings.warn(\n 'Call to deprecated function %s(). Note: Do add unlinked descriptors '\n 'to descriptor_pool is wrong. Use Add() or AddSerializedFile() '\n 'instead.' % func.__name__,\n category=DeprecationWarning)\n return func(*args, **kwargs)\n NewFunc.__name__ = func.__name__\n NewFunc.__doc__ = func.__doc__\n NewFunc.__dict__.update(func.__dict__)\n return NewFunc\n\n\ndef _NormalizeFullyQualifiedName(name):\n \"\"\"Remove leading period from fully-qualified type name.\n\n Due to b/13860351 in descriptor_database.py, types in the root namespace are\n generated with a leading period. This function removes that prefix.\n\n Args:\n name (str): The fully-qualified symbol name.\n\n Returns:\n str: The normalized fully-qualified symbol name.\n \"\"\"\n return name.lstrip('.')\n\n\ndef _OptionsOrNone(descriptor_proto):\n \"\"\"Returns the value of the field `options`, or None if it is not set.\"\"\"\n if descriptor_proto.HasField('options'):\n return descriptor_proto.options\n else:\n return None\n\n\ndef _IsMessageSetExtension(field):\n return (field.is_extension and\n field.containing_type.has_options and\n field.containing_type.GetOptions().message_set_wire_format and\n field.type == descriptor.FieldDescriptor.TYPE_MESSAGE and\n field.label == descriptor.FieldDescriptor.LABEL_OPTIONAL)\n\n\nclass DescriptorPool(object):\n \"\"\"A collection of protobufs dynamically constructed by descriptor protos.\"\"\"\n\n if _USE_C_DESCRIPTORS:\n\n def __new__(cls, descriptor_db=None):\n # pylint: disable=protected-access\n return descriptor._message.DescriptorPool(descriptor_db)\n\n def __init__(self, descriptor_db=None):\n \"\"\"Initializes a Pool of proto buffs.\n\n The descriptor_db argument to the constructor is provided to allow\n specialized file descriptor proto lookup code to be triggered on demand. An\n example would be an implementation which will read and compile a file\n specified in a call to FindFileByName() and not require the call to Add()\n at all. Results from this database will be cached internally here as well.\n\n Args:\n descriptor_db: A secondary source of file descriptors.\n \"\"\"\n\n self._internal_db = descriptor_database.DescriptorDatabase()\n self._descriptor_db = descriptor_db\n self._descriptors = {}\n self._enum_descriptors = {}\n self._service_descriptors = {}\n self._file_descriptors = {}\n self._toplevel_extensions = {}\n # TODO(jieluo): Remove _file_desc_by_toplevel_extension after\n # maybe year 2020 for compatibility issue (with 3.4.1 only).\n self._file_desc_by_toplevel_extension = {}\n self._top_enum_values = {}\n # We store extensions in two two-level mappings: The first key is the\n # descriptor of the message being extended, the second key is the extension\n # full name or its tag number.\n self._extensions_by_name = collections.defaultdict(dict)\n self._extensions_by_number = collections.defaultdict(dict)\n\n def _CheckConflictRegister(self, desc, desc_name, file_name):\n \"\"\"Check if the descriptor name conflicts with another of the same name.\n\n Args:\n desc: Descriptor of a message, enum, service, extension or enum value.\n desc_name (str): the full name of desc.\n file_name (str): The file name of descriptor.\n \"\"\"\n for register, descriptor_type in [\n (self._descriptors, descriptor.Descriptor),\n (self._enum_descriptors, descriptor.EnumDescriptor),\n (self._service_descriptors, descriptor.ServiceDescriptor),\n (self._toplevel_extensions, descriptor.FieldDescriptor),\n (self._top_enum_values, descriptor.EnumValueDescriptor)]:\n if desc_name in register:\n old_desc = register[desc_name]\n if isinstance(old_desc, descriptor.EnumValueDescriptor):\n old_file = old_desc.type.file.name\n else:\n old_file = old_desc.file.name\n\n if not isinstance(desc, descriptor_type) or (\n old_file != file_name):\n error_msg = ('Conflict register for file \"' + file_name +\n '\": ' + desc_name +\n ' is already defined in file \"' +\n old_file + '\". Please fix the conflict by adding '\n 'package name on the proto file, or use different '\n 'name for the duplication.')\n if isinstance(desc, descriptor.EnumValueDescriptor):\n error_msg += ('\\nNote: enum values appear as '\n 'siblings of the enum type instead of '\n 'children of it.')\n\n raise TypeError(error_msg)\n\n return\n\n def Add(self, file_desc_proto):\n \"\"\"Adds the FileDescriptorProto and its types to this pool.\n\n Args:\n file_desc_proto (FileDescriptorProto): The file descriptor to add.\n \"\"\"\n\n self._internal_db.Add(file_desc_proto)\n\n def AddSerializedFile(self, serialized_file_desc_proto):\n \"\"\"Adds the FileDescriptorProto and its types to this pool.\n\n Args:\n serialized_file_desc_proto (bytes): A bytes string, serialization of the\n :class:`FileDescriptorProto` to add.\n\n Returns:\n FileDescriptor: Descriptor for the added file.\n \"\"\"\n\n # pylint: disable=g-import-not-at-top\n from google.protobuf import descriptor_pb2\n file_desc_proto = descriptor_pb2.FileDescriptorProto.FromString(\n serialized_file_desc_proto)\n file_desc = self._ConvertFileProtoToFileDescriptor(file_desc_proto)\n file_desc.serialized_pb = serialized_file_desc_proto\n return file_desc\n\n # Add Descriptor to descriptor pool is dreprecated. Please use Add()\n # or AddSerializedFile() to add a FileDescriptorProto instead.\n @_Deprecated\n def AddDescriptor(self, desc):\n self._AddDescriptor(desc)\n\n # Never call this method. It is for internal usage only.\n def _AddDescriptor(self, desc):\n \"\"\"Adds a Descriptor to the pool, non-recursively.\n\n If the Descriptor contains nested messages or enums, the caller must\n explicitly register them. This method also registers the FileDescriptor\n associated with the message.\n\n Args:\n desc: A Descriptor.\n \"\"\"\n if not isinstance(desc, descriptor.Descriptor):\n raise TypeError('Expected instance of descriptor.Descriptor.')\n\n self._CheckConflictRegister(desc, desc.full_name, desc.file.name)\n\n self._descriptors[desc.full_name] = desc\n self._AddFileDescriptor(desc.file)\n\n # Add EnumDescriptor to descriptor pool is dreprecated. Please use Add()\n # or AddSerializedFile() to add a FileDescriptorProto instead.\n @_Deprecated\n def AddEnumDescriptor(self, enum_desc):\n self._AddEnumDescriptor(enum_desc)\n\n # Never call this method. It is for internal usage only.\n def _AddEnumDescriptor(self, enum_desc):\n \"\"\"Adds an EnumDescriptor to the pool.\n\n This method also registers the FileDescriptor associated with the enum.\n\n Args:\n enum_desc: An EnumDescriptor.\n \"\"\"\n\n if not isinstance(enum_desc, descriptor.EnumDescriptor):\n raise TypeError('Expected instance of descriptor.EnumDescriptor.')\n\n file_name = enum_desc.file.name\n self._CheckConflictRegister(enum_desc, enum_desc.full_name, file_name)\n self._enum_descriptors[enum_desc.full_name] = enum_desc\n\n # Top enum values need to be indexed.\n # Count the number of dots to see whether the enum is toplevel or nested\n # in a message. We cannot use enum_desc.containing_type at this stage.\n if enum_desc.file.package:\n top_level = (enum_desc.full_name.count('.')\n - enum_desc.file.package.count('.') == 1)\n else:\n top_level = enum_desc.full_name.count('.') == 0\n if top_level:\n file_name = enum_desc.file.name\n package = enum_desc.file.package\n for enum_value in enum_desc.values:\n full_name = _NormalizeFullyQualifiedName(\n '.'.join((package, enum_value.name)))\n self._CheckConflictRegister(enum_value, full_name, file_name)\n self._top_enum_values[full_name] = enum_value\n self._AddFileDescriptor(enum_desc.file)\n\n # Add ServiceDescriptor to descriptor pool is dreprecated. Please use Add()\n # or AddSerializedFile() to add a FileDescriptorProto instead.\n @_Deprecated\n def AddServiceDescriptor(self, service_desc):\n self._AddServiceDescriptor(service_desc)\n\n # Never call this method. It is for internal usage only.\n def _AddServiceDescriptor(self, service_desc):\n \"\"\"Adds a ServiceDescriptor to the pool.\n\n Args:\n service_desc: A ServiceDescriptor.\n \"\"\"\n\n if not isinstance(service_desc, descriptor.ServiceDescriptor):\n raise TypeError('Expected instance of descriptor.ServiceDescriptor.')\n\n self._CheckConflictRegister(service_desc, service_desc.full_name,\n service_desc.file.name)\n self._service_descriptors[service_desc.full_name] = service_desc\n\n # Add ExtensionDescriptor to descriptor pool is dreprecated. Please use Add()\n # or AddSerializedFile() to add a FileDescriptorProto instead.\n @_Deprecated\n def AddExtensionDescriptor(self, extension):\n self._AddExtensionDescriptor(extension)\n\n # Never call this method. It is for internal usage only.\n def _AddExtensionDescriptor(self, extension):\n \"\"\"Adds a FieldDescriptor describing an extension to the pool.\n\n Args:\n extension: A FieldDescriptor.\n\n Raises:\n AssertionError: when another extension with the same number extends the\n same message.\n TypeError: when the specified extension is not a\n descriptor.FieldDescriptor.\n \"\"\"\n if not (isinstance(extension, descriptor.FieldDescriptor) and\n extension.is_extension):\n raise TypeError('Expected an extension descriptor.')\n\n if extension.extension_scope is None:\n self._toplevel_extensions[extension.full_name] = extension\n\n try:\n existing_desc = self._extensions_by_number[\n extension.containing_type][extension.number]\n except KeyError:\n pass\n else:\n if extension is not existing_desc:\n raise AssertionError(\n 'Extensions \"%s\" and \"%s\" both try to extend message type \"%s\" '\n 'with field number %d.' %\n (extension.full_name, existing_desc.full_name,\n extension.containing_type.full_name, extension.number))\n\n self._extensions_by_number[extension.containing_type][\n extension.number] = extension\n self._extensions_by_name[extension.containing_type][\n extension.full_name] = extension\n\n # Also register MessageSet extensions with the type name.\n if _IsMessageSetExtension(extension):\n self._extensions_by_name[extension.containing_type][\n extension.message_type.full_name] = extension\n\n @_Deprecated\n def AddFileDescriptor(self, file_desc):\n self._InternalAddFileDescriptor(file_desc)\n\n # Never call this method. It is for internal usage only.\n def _InternalAddFileDescriptor(self, file_desc):\n \"\"\"Adds a FileDescriptor to the pool, non-recursively.\n\n If the FileDescriptor contains messages or enums, the caller must explicitly\n register them.\n\n Args:\n file_desc: A FileDescriptor.\n \"\"\"\n\n self._AddFileDescriptor(file_desc)\n # TODO(jieluo): This is a temporary solution for FieldDescriptor.file.\n # FieldDescriptor.file is added in code gen. Remove this solution after\n # maybe 2020 for compatibility reason (with 3.4.1 only).\n for extension in file_desc.extensions_by_name.values():\n self._file_desc_by_toplevel_extension[\n extension.full_name] = file_desc\n\n def _AddFileDescriptor(self, file_desc):\n \"\"\"Adds a FileDescriptor to the pool, non-recursively.\n\n If the FileDescriptor contains messages or enums, the caller must explicitly\n register them.\n\n Args:\n file_desc: A FileDescriptor.\n \"\"\"\n\n if not isinstance(file_desc, descriptor.FileDescriptor):\n raise TypeError('Expected instance of descriptor.FileDescriptor.')\n self._file_descriptors[file_desc.name] = file_desc\n\n def FindFileByName(self, file_name):\n \"\"\"Gets a FileDescriptor by file name.\n\n Args:\n file_name (str): The path to the file to get a descriptor for.\n\n Returns:\n FileDescriptor: The descriptor for the named file.\n\n Raises:\n KeyError: if the file cannot be found in the pool.\n \"\"\"\n\n try:\n return self._file_descriptors[file_name]\n except KeyError:\n pass\n\n try:\n file_proto = self._internal_db.FindFileByName(file_name)\n except KeyError as error:\n if self._descriptor_db:\n file_proto = self._descriptor_db.FindFileByName(file_name)\n else:\n raise error\n if not file_proto:\n raise KeyError('Cannot find a file named %s' % file_name)\n return self._ConvertFileProtoToFileDescriptor(file_proto)\n\n def FindFileContainingSymbol(self, symbol):\n \"\"\"Gets the FileDescriptor for the file containing the specified symbol.\n\n Args:\n symbol (str): The name of the symbol to search for.\n\n Returns:\n FileDescriptor: Descriptor for the file that contains the specified\n symbol.\n\n Raises:\n KeyError: if the file cannot be found in the pool.\n \"\"\"\n\n symbol = _NormalizeFullyQualifiedName(symbol)\n try:\n return self._InternalFindFileContainingSymbol(symbol)\n except KeyError:\n pass\n\n try:\n # Try fallback database. Build and find again if possible.\n self._FindFileContainingSymbolInDb(symbol)\n return self._InternalFindFileContainingSymbol(symbol)\n except KeyError:\n raise KeyError('Cannot find a file containing %s' % symbol)\n\n def _InternalFindFileContainingSymbol(self, symbol):\n \"\"\"Gets the already built FileDescriptor containing the specified symbol.\n\n Args:\n symbol (str): The name of the symbol to search for.\n\n Returns:\n FileDescriptor: Descriptor for the file that contains the specified\n symbol.\n\n Raises:\n KeyError: if the file cannot be found in the pool.\n \"\"\"\n try:\n return self._descriptors[symbol].file\n except KeyError:\n pass\n\n try:\n return self._enum_descriptors[symbol].file\n except KeyError:\n pass\n\n try:\n return self._service_descriptors[symbol].file\n except KeyError:\n pass\n\n try:\n return self._top_enum_values[symbol].type.file\n except KeyError:\n pass\n\n try:\n return self._file_desc_by_toplevel_extension[symbol]\n except KeyError:\n pass\n\n # Try fields, enum values and nested extensions inside a message.\n top_name, _, sub_name = symbol.rpartition('.')\n try:\n message = self.FindMessageTypeByName(top_name)\n assert (sub_name in message.extensions_by_name or\n sub_name in message.fields_by_name or\n sub_name in message.enum_values_by_name)\n return message.file\n except (KeyError, AssertionError):\n raise KeyError('Cannot find a file containing %s' % symbol)\n\n def FindMessageTypeByName(self, full_name):\n \"\"\"Loads the named descriptor from the pool.\n\n Args:\n full_name (str): The full name of the descriptor to load.\n\n Returns:\n Descriptor: The descriptor for the named type.\n\n Raises:\n KeyError: if the message cannot be found in the pool.\n \"\"\"\n\n full_name = _NormalizeFullyQualifiedName(full_name)\n if full_name not in self._descriptors:\n self._FindFileContainingSymbolInDb(full_name)\n return self._descriptors[full_name]\n\n def FindEnumTypeByName(self, full_name):\n \"\"\"Loads the named enum descriptor from the pool.\n\n Args:\n full_name (str): The full name of the enum descriptor to load.\n\n Returns:\n EnumDescriptor: The enum descriptor for the named type.\n\n Raises:\n KeyError: if the enum cannot be found in the pool.\n \"\"\"\n\n full_name = _NormalizeFullyQualifiedName(full_name)\n if full_name not in self._enum_descriptors:\n self._FindFileContainingSymbolInDb(full_name)\n return self._enum_descriptors[full_name]\n\n def FindFieldByName(self, full_name):\n \"\"\"Loads the named field descriptor from the pool.\n\n Args:\n full_name (str): The full name of the field descriptor to load.\n\n Returns:\n FieldDescriptor: The field descriptor for the named field.\n\n Raises:\n KeyError: if the field cannot be found in the pool.\n \"\"\"\n full_name = _NormalizeFullyQualifiedName(full_name)\n message_name, _, field_name = full_name.rpartition('.')\n message_descriptor = self.FindMessageTypeByName(message_name)\n return message_descriptor.fields_by_name[field_name]\n\n def FindOneofByName(self, full_name):\n \"\"\"Loads the named oneof descriptor from the pool.\n\n Args:\n full_name (str): The full name of the oneof descriptor to load.\n\n Returns:\n OneofDescriptor: The oneof descriptor for the named oneof.\n\n Raises:\n KeyError: if the oneof cannot be found in the pool.\n \"\"\"\n full_name = _NormalizeFullyQualifiedName(full_name)\n message_name, _, oneof_name = full_name.rpartition('.')\n message_descriptor = self.FindMessageTypeByName(message_name)\n return message_descriptor.oneofs_by_name[oneof_name]\n\n def FindExtensionByName(self, full_name):\n \"\"\"Loads the named extension descriptor from the pool.\n\n Args:\n full_name (str): The full name of the extension descriptor to load.\n\n Returns:\n FieldDescriptor: The field descriptor for the named extension.\n\n Raises:\n KeyError: if the extension cannot be found in the pool.\n \"\"\"\n full_name = _NormalizeFullyQualifiedName(full_name)\n try:\n # The proto compiler does not give any link between the FileDescriptor\n # and top-level extensions unless the FileDescriptorProto is added to\n # the DescriptorDatabase, but this can impact memory usage.\n # So we registered these extensions by name explicitly.\n return self._toplevel_extensions[full_name]\n except KeyError:\n pass\n message_name, _, extension_name = full_name.rpartition('.')\n try:\n # Most extensions are nested inside a message.\n scope = self.FindMessageTypeByName(message_name)\n except KeyError:\n # Some extensions are defined at file scope.\n scope = self._FindFileContainingSymbolInDb(full_name)\n return scope.extensions_by_name[extension_name]\n\n def FindExtensionByNumber(self, message_descriptor, number):\n \"\"\"Gets the extension of the specified message with the specified number.\n\n Extensions have to be registered to this pool by calling :func:`Add` or\n :func:`AddExtensionDescriptor`.\n\n Args:\n message_descriptor (Descriptor): descriptor of the extended message.\n number (int): Number of the extension field.\n\n Returns:\n FieldDescriptor: The descriptor for the extension.\n\n Raises:\n KeyError: when no extension with the given number is known for the\n specified message.\n \"\"\"\n try:\n return self._extensions_by_number[message_descriptor][number]\n except KeyError:\n self._TryLoadExtensionFromDB(message_descriptor, number)\n return self._extensions_by_number[message_descriptor][number]\n\n def FindAllExtensions(self, message_descriptor):\n \"\"\"Gets all the known extensions of a given message.\n\n Extensions have to be registered to this pool by build related\n :func:`Add` or :func:`AddExtensionDescriptor`.\n\n Args:\n message_descriptor (Descriptor): Descriptor of the extended message.\n\n Returns:\n list[FieldDescriptor]: Field descriptors describing the extensions.\n \"\"\"\n # Fallback to descriptor db if FindAllExtensionNumbers is provided.\n if self._descriptor_db and hasattr(\n self._descriptor_db, 'FindAllExtensionNumbers'):\n full_name = message_descriptor.full_name\n all_numbers = self._descriptor_db.FindAllExtensionNumbers(full_name)\n for number in all_numbers:\n if number in self._extensions_by_number[message_descriptor]:\n continue\n self._TryLoadExtensionFromDB(message_descriptor, number)\n\n return list(self._extensions_by_number[message_descriptor].values())\n\n def _TryLoadExtensionFromDB(self, message_descriptor, number):\n \"\"\"Try to Load extensions from descriptor db.\n\n Args:\n message_descriptor: descriptor of the extended message.\n number: the extension number that needs to be loaded.\n \"\"\"\n if not self._descriptor_db:\n return\n # Only supported when FindFileContainingExtension is provided.\n if not hasattr(\n self._descriptor_db, 'FindFileContainingExtension'):\n return\n\n full_name = message_descriptor.full_name\n file_proto = self._descriptor_db.FindFileContainingExtension(\n full_name, number)\n\n if file_proto is None:\n return\n\n try:\n self._ConvertFileProtoToFileDescriptor(file_proto)\n except:\n warn_msg = ('Unable to load proto file %s for extension number %d.' %\n (file_proto.name, number))\n warnings.warn(warn_msg, RuntimeWarning)\n\n def FindServiceByName(self, full_name):\n \"\"\"Loads the named service descriptor from the pool.\n\n Args:\n full_name (str): The full name of the service descriptor to load.\n\n Returns:\n ServiceDescriptor: The service descriptor for the named service.\n\n Raises:\n KeyError: if the service cannot be found in the pool.\n \"\"\"\n full_name = _NormalizeFullyQualifiedName(full_name)\n if full_name not in self._service_descriptors:\n self._FindFileContainingSymbolInDb(full_name)\n return self._service_descriptors[full_name]\n\n def FindMethodByName(self, full_name):\n \"\"\"Loads the named service method descriptor from the pool.\n\n Args:\n full_name (str): The full name of the method descriptor to load.\n\n Returns:\n MethodDescriptor: The method descriptor for the service method.\n\n Raises:\n KeyError: if the method cannot be found in the pool.\n \"\"\"\n full_name = _NormalizeFullyQualifiedName(full_name)\n service_name, _, method_name = full_name.rpartition('.')\n service_descriptor = self.FindServiceByName(service_name)\n return service_descriptor.methods_by_name[method_name]\n\n def _FindFileContainingSymbolInDb(self, symbol):\n \"\"\"Finds the file in descriptor DB containing the specified symbol.\n\n Args:\n symbol (str): The name of the symbol to search for.\n\n Returns:\n FileDescriptor: The file that contains the specified symbol.\n\n Raises:\n KeyError: if the file cannot be found in the descriptor database.\n \"\"\"\n try:\n file_proto = self._internal_db.FindFileContainingSymbol(symbol)\n except KeyError as error:\n if self._descriptor_db:\n file_proto = self._descriptor_db.FindFileContainingSymbol(symbol)\n else:\n raise error\n if not file_proto:\n raise KeyError('Cannot find a file containing %s' % symbol)\n return self._ConvertFileProtoToFileDescriptor(file_proto)\n\n def _ConvertFileProtoToFileDescriptor(self, file_proto):\n \"\"\"Creates a FileDescriptor from a proto or returns a cached copy.\n\n This method also has the side effect of loading all the symbols found in\n the file into the appropriate dictionaries in the pool.\n\n Args:\n file_proto: The proto to convert.\n\n Returns:\n A FileDescriptor matching the passed in proto.\n \"\"\"\n if file_proto.name not in self._file_descriptors:\n built_deps = list(self._GetDeps(file_proto.dependency))\n direct_deps = [self.FindFileByName(n) for n in file_proto.dependency]\n public_deps = [direct_deps[i] for i in file_proto.public_dependency]\n\n file_descriptor = descriptor.FileDescriptor(\n pool=self,\n name=file_proto.name,\n package=file_proto.package,\n syntax=file_proto.syntax,\n options=_OptionsOrNone(file_proto),\n serialized_pb=file_proto.SerializeToString(),\n dependencies=direct_deps,\n public_dependencies=public_deps,\n # pylint: disable=protected-access\n create_key=descriptor._internal_create_key)\n scope = {}\n\n # This loop extracts all the message and enum types from all the\n # dependencies of the file_proto. This is necessary to create the\n # scope of available message types when defining the passed in\n # file proto.\n for dependency in built_deps:\n scope.update(self._ExtractSymbols(\n dependency.message_types_by_name.values()))\n scope.update((_PrefixWithDot(enum.full_name), enum)\n for enum in dependency.enum_types_by_name.values())\n\n for message_type in file_proto.message_type:\n message_desc = self._ConvertMessageDescriptor(\n message_type, file_proto.package, file_descriptor, scope,\n file_proto.syntax)\n file_descriptor.message_types_by_name[message_desc.name] = (\n message_desc)\n\n for enum_type in file_proto.enum_type:\n file_descriptor.enum_types_by_name[enum_type.name] = (\n self._ConvertEnumDescriptor(enum_type, file_proto.package,\n file_descriptor, None, scope, True))\n\n for index, extension_proto in enumerate(file_proto.extension):\n extension_desc = self._MakeFieldDescriptor(\n extension_proto, file_proto.package, index, file_descriptor,\n is_extension=True)\n extension_desc.containing_type = self._GetTypeFromScope(\n file_descriptor.package, extension_proto.extendee, scope)\n self._SetFieldType(extension_proto, extension_desc,\n file_descriptor.package, scope)\n file_descriptor.extensions_by_name[extension_desc.name] = (\n extension_desc)\n self._file_desc_by_toplevel_extension[extension_desc.full_name] = (\n file_descriptor)\n\n for desc_proto in file_proto.message_type:\n self._SetAllFieldTypes(file_proto.package, desc_proto, scope)\n\n if file_proto.package:\n desc_proto_prefix = _PrefixWithDot(file_proto.package)\n else:\n desc_proto_prefix = ''\n\n for desc_proto in file_proto.message_type:\n desc = self._GetTypeFromScope(\n desc_proto_prefix, desc_proto.name, scope)\n file_descriptor.message_types_by_name[desc_proto.name] = desc\n\n for index, service_proto in enumerate(file_proto.service):\n file_descriptor.services_by_name[service_proto.name] = (\n self._MakeServiceDescriptor(service_proto, index, scope,\n file_proto.package, file_descriptor))\n\n self._file_descriptors[file_proto.name] = file_descriptor\n\n # Add extensions to the pool\n file_desc = self._file_descriptors[file_proto.name]\n for extension in file_desc.extensions_by_name.values():\n self._AddExtensionDescriptor(extension)\n for message_type in file_desc.message_types_by_name.values():\n for extension in message_type.extensions:\n self._AddExtensionDescriptor(extension)\n\n return file_desc\n\n def _ConvertMessageDescriptor(self, desc_proto, package=None, file_desc=None,\n scope=None, syntax=None):\n \"\"\"Adds the proto to the pool in the specified package.\n\n Args:\n desc_proto: The descriptor_pb2.DescriptorProto protobuf message.\n package: The package the proto should be located in.\n file_desc: The file containing this message.\n scope: Dict mapping short and full symbols to message and enum types.\n syntax: string indicating syntax of the file (\"proto2\" or \"proto3\")\n\n Returns:\n The added descriptor.\n \"\"\"\n\n if package:\n desc_name = '.'.join((package, desc_proto.name))\n else:\n desc_name = desc_proto.name\n\n if file_desc is None:\n file_name = None\n else:\n file_name = file_desc.name\n\n if scope is None:\n scope = {}\n\n nested = [\n self._ConvertMessageDescriptor(\n nested, desc_name, file_desc, scope, syntax)\n for nested in desc_proto.nested_type]\n enums = [\n self._ConvertEnumDescriptor(enum, desc_name, file_desc, None,\n scope, False)\n for enum in desc_proto.enum_type]\n fields = [self._MakeFieldDescriptor(field, desc_name, index, file_desc)\n for index, field in enumerate(desc_proto.field)]\n extensions = [\n self._MakeFieldDescriptor(extension, desc_name, index, file_desc,\n is_extension=True)\n for index, extension in enumerate(desc_proto.extension)]\n oneofs = [\n # pylint: disable=g-complex-comprehension\n descriptor.OneofDescriptor(\n desc.name,\n '.'.join((desc_name, desc.name)),\n index,\n None,\n [],\n _OptionsOrNone(desc),\n # pylint: disable=protected-access\n create_key=descriptor._internal_create_key)\n for index, desc in enumerate(desc_proto.oneof_decl)\n ]\n extension_ranges = [(r.start, r.end) for r in desc_proto.extension_range]\n if extension_ranges:\n is_extendable = True\n else:\n is_extendable = False\n desc = descriptor.Descriptor(\n name=desc_proto.name,\n full_name=desc_name,\n filename=file_name,\n containing_type=None,\n fields=fields,\n oneofs=oneofs,\n nested_types=nested,\n enum_types=enums,\n extensions=extensions,\n options=_OptionsOrNone(desc_proto),\n is_extendable=is_extendable,\n extension_ranges=extension_ranges,\n file=file_desc,\n serialized_start=None,\n serialized_end=None,\n syntax=syntax,\n # pylint: disable=protected-access\n create_key=descriptor._internal_create_key)\n for nested in desc.nested_types:\n nested.containing_type = desc\n for enum in desc.enum_types:\n enum.containing_type = desc\n for field_index, field_desc in enumerate(desc_proto.field):\n if field_desc.HasField('oneof_index'):\n oneof_index = field_desc.oneof_index\n oneofs[oneof_index].fields.append(fields[field_index])\n fields[field_index].containing_oneof = oneofs[oneof_index]\n\n scope[_PrefixWithDot(desc_name)] = desc\n self._CheckConflictRegister(desc, desc.full_name, desc.file.name)\n self._descriptors[desc_name] = desc\n return desc\n\n def _ConvertEnumDescriptor(self, enum_proto, package=None, file_desc=None,\n containing_type=None, scope=None, top_level=False):\n \"\"\"Make a protobuf EnumDescriptor given an EnumDescriptorProto protobuf.\n\n Args:\n enum_proto: The descriptor_pb2.EnumDescriptorProto protobuf message.\n package: Optional package name for the new message EnumDescriptor.\n file_desc: The file containing the enum descriptor.\n containing_type: The type containing this enum.\n scope: Scope containing available types.\n top_level: If True, the enum is a top level symbol. If False, the enum\n is defined inside a message.\n\n Returns:\n The added descriptor\n \"\"\"\n\n if package:\n enum_name = '.'.join((package, enum_proto.name))\n else:\n enum_name = enum_proto.name\n\n if file_desc is None:\n file_name = None\n else:\n file_name = file_desc.name\n\n values = [self._MakeEnumValueDescriptor(value, index)\n for index, value in enumerate(enum_proto.value)]\n desc = descriptor.EnumDescriptor(name=enum_proto.name,\n full_name=enum_name,\n filename=file_name,\n file=file_desc,\n values=values,\n containing_type=containing_type,\n options=_OptionsOrNone(enum_proto),\n # pylint: disable=protected-access\n create_key=descriptor._internal_create_key)\n scope['.%s' % enum_name] = desc\n self._CheckConflictRegister(desc, desc.full_name, desc.file.name)\n self._enum_descriptors[enum_name] = desc\n\n # Add top level enum values.\n if top_level:\n for value in values:\n full_name = _NormalizeFullyQualifiedName(\n '.'.join((package, value.name)))\n self._CheckConflictRegister(value, full_name, file_name)\n self._top_enum_values[full_name] = value\n\n return desc\n\n def _MakeFieldDescriptor(self, field_proto, message_name, index,\n file_desc, is_extension=False):\n \"\"\"Creates a field descriptor from a FieldDescriptorProto.\n\n For message and enum type fields, this method will do a look up\n in the pool for the appropriate descriptor for that type. If it\n is unavailable, it will fall back to the _source function to\n create it. If this type is still unavailable, construction will\n fail.\n\n Args:\n field_proto: The proto describing the field.\n message_name: The name of the containing message.\n index: Index of the field\n file_desc: The file containing the field descriptor.\n is_extension: Indication that this field is for an extension.\n\n Returns:\n An initialized FieldDescriptor object\n \"\"\"\n\n if message_name:\n full_name = '.'.join((message_name, field_proto.name))\n else:\n full_name = field_proto.name\n\n if field_proto.json_name:\n json_name = field_proto.json_name\n else:\n json_name = None\n\n return descriptor.FieldDescriptor(\n name=field_proto.name,\n full_name=full_name,\n index=index,\n number=field_proto.number,\n type=field_proto.type,\n cpp_type=None,\n message_type=None,\n enum_type=None,\n containing_type=None,\n label=field_proto.label,\n has_default_value=False,\n default_value=None,\n is_extension=is_extension,\n extension_scope=None,\n options=_OptionsOrNone(field_proto),\n json_name=json_name,\n file=file_desc,\n # pylint: disable=protected-access\n create_key=descriptor._internal_create_key)\n\n def _SetAllFieldTypes(self, package, desc_proto, scope):\n \"\"\"Sets all the descriptor's fields's types.\n\n This method also sets the containing types on any extensions.\n\n Args:\n package: The current package of desc_proto.\n desc_proto: The message descriptor to update.\n scope: Enclosing scope of available types.\n \"\"\"\n\n package = _PrefixWithDot(package)\n\n main_desc = self._GetTypeFromScope(package, desc_proto.name, scope)\n\n if package == '.':\n nested_package = _PrefixWithDot(desc_proto.name)\n else:\n nested_package = '.'.join([package, desc_proto.name])\n\n for field_proto, field_desc in zip(desc_proto.field, main_desc.fields):\n self._SetFieldType(field_proto, field_desc, nested_package, scope)\n\n for extension_proto, extension_desc in (\n zip(desc_proto.extension, main_desc.extensions)):\n extension_desc.containing_type = self._GetTypeFromScope(\n nested_package, extension_proto.extendee, scope)\n self._SetFieldType(extension_proto, extension_desc, nested_package, scope)\n\n for nested_type in desc_proto.nested_type:\n self._SetAllFieldTypes(nested_package, nested_type, scope)\n\n def _SetFieldType(self, field_proto, field_desc, package, scope):\n \"\"\"Sets the field's type, cpp_type, message_type and enum_type.\n\n Args:\n field_proto: Data about the field in proto format.\n field_desc: The descriptor to modify.\n package: The package the field's container is in.\n scope: Enclosing scope of available types.\n \"\"\"\n if field_proto.type_name:\n desc = self._GetTypeFromScope(package, field_proto.type_name, scope)\n else:\n desc = None\n\n if not field_proto.HasField('type'):\n if isinstance(desc, descriptor.Descriptor):\n field_proto.type = descriptor.FieldDescriptor.TYPE_MESSAGE\n else:\n field_proto.type = descriptor.FieldDescriptor.TYPE_ENUM\n\n field_desc.cpp_type = descriptor.FieldDescriptor.ProtoTypeToCppProtoType(\n field_proto.type)\n\n if (field_proto.type == descriptor.FieldDescriptor.TYPE_MESSAGE\n or field_proto.type == descriptor.FieldDescriptor.TYPE_GROUP):\n field_desc.message_type = desc\n\n if field_proto.type == descriptor.FieldDescriptor.TYPE_ENUM:\n field_desc.enum_type = desc\n\n if field_proto.label == descriptor.FieldDescriptor.LABEL_REPEATED:\n field_desc.has_default_value = False\n field_desc.default_value = []\n elif field_proto.HasField('default_value'):\n field_desc.has_default_value = True\n if (field_proto.type == descriptor.FieldDescriptor.TYPE_DOUBLE or\n field_proto.type == descriptor.FieldDescriptor.TYPE_FLOAT):\n field_desc.default_value = float(field_proto.default_value)\n elif field_proto.type == descriptor.FieldDescriptor.TYPE_STRING:\n field_desc.default_value = field_proto.default_value\n elif field_proto.type == descriptor.FieldDescriptor.TYPE_BOOL:\n field_desc.default_value = field_proto.default_value.lower() == 'true'\n elif field_proto.type == descriptor.FieldDescriptor.TYPE_ENUM:\n field_desc.default_value = field_desc.enum_type.values_by_name[\n field_proto.default_value].number\n elif field_proto.type == descriptor.FieldDescriptor.TYPE_BYTES:\n field_desc.default_value = text_encoding.CUnescape(\n field_proto.default_value)\n elif field_proto.type == descriptor.FieldDescriptor.TYPE_MESSAGE:\n field_desc.default_value = None\n else:\n # All other types are of the \"int\" type.\n field_desc.default_value = int(field_proto.default_value)\n else:\n field_desc.has_default_value = False\n if (field_proto.type == descriptor.FieldDescriptor.TYPE_DOUBLE or\n field_proto.type == descriptor.FieldDescriptor.TYPE_FLOAT):\n field_desc.default_value = 0.0\n elif field_proto.type == descriptor.FieldDescriptor.TYPE_STRING:\n field_desc.default_value = u''\n elif field_proto.type == descriptor.FieldDescriptor.TYPE_BOOL:\n field_desc.default_value = False\n elif field_proto.type == descriptor.FieldDescriptor.TYPE_ENUM:\n field_desc.default_value = field_desc.enum_type.values[0].number\n elif field_proto.type == descriptor.FieldDescriptor.TYPE_BYTES:\n field_desc.default_value = b''\n elif field_proto.type == descriptor.FieldDescriptor.TYPE_MESSAGE:\n field_desc.default_value = None\n elif field_proto.type == descriptor.FieldDescriptor.TYPE_GROUP:\n field_desc.default_value = None\n else:\n # All other types are of the \"int\" type.\n field_desc.default_value = 0\n\n field_desc.type = field_proto.type\n\n def _MakeEnumValueDescriptor(self, value_proto, index):\n \"\"\"Creates a enum value descriptor object from a enum value proto.\n\n Args:\n value_proto: The proto describing the enum value.\n index: The index of the enum value.\n\n Returns:\n An initialized EnumValueDescriptor object.\n \"\"\"\n\n return descriptor.EnumValueDescriptor(\n name=value_proto.name,\n index=index,\n number=value_proto.number,\n options=_OptionsOrNone(value_proto),\n type=None,\n # pylint: disable=protected-access\n create_key=descriptor._internal_create_key)\n\n def _MakeServiceDescriptor(self, service_proto, service_index, scope,\n package, file_desc):\n \"\"\"Make a protobuf ServiceDescriptor given a ServiceDescriptorProto.\n\n Args:\n service_proto: The descriptor_pb2.ServiceDescriptorProto protobuf message.\n service_index: The index of the service in the File.\n scope: Dict mapping short and full symbols to message and enum types.\n package: Optional package name for the new message EnumDescriptor.\n file_desc: The file containing the service descriptor.\n\n Returns:\n The added descriptor.\n \"\"\"\n\n if package:\n service_name = '.'.join((package, service_proto.name))\n else:\n service_name = service_proto.name\n\n methods = [self._MakeMethodDescriptor(method_proto, service_name, package,\n scope, index)\n for index, method_proto in enumerate(service_proto.method)]\n desc = descriptor.ServiceDescriptor(\n name=service_proto.name,\n full_name=service_name,\n index=service_index,\n methods=methods,\n options=_OptionsOrNone(service_proto),\n file=file_desc,\n # pylint: disable=protected-access\n create_key=descriptor._internal_create_key)\n self._CheckConflictRegister(desc, desc.full_name, desc.file.name)\n self._service_descriptors[service_name] = desc\n return desc\n\n def _MakeMethodDescriptor(self, method_proto, service_name, package, scope,\n index):\n \"\"\"Creates a method descriptor from a MethodDescriptorProto.\n\n Args:\n method_proto: The proto describing the method.\n service_name: The name of the containing service.\n package: Optional package name to look up for types.\n scope: Scope containing available types.\n index: Index of the method in the service.\n\n Returns:\n An initialized MethodDescriptor object.\n \"\"\"\n full_name = '.'.join((service_name, method_proto.name))\n input_type = self._GetTypeFromScope(\n package, method_proto.input_type, scope)\n output_type = self._GetTypeFromScope(\n package, method_proto.output_type, scope)\n return descriptor.MethodDescriptor(\n name=method_proto.name,\n full_name=full_name,\n index=index,\n containing_service=None,\n input_type=input_type,\n output_type=output_type,\n client_streaming=method_proto.client_streaming,\n server_streaming=method_proto.server_streaming,\n options=_OptionsOrNone(method_proto),\n # pylint: disable=protected-access\n create_key=descriptor._internal_create_key)\n\n def _ExtractSymbols(self, descriptors):\n \"\"\"Pulls out all the symbols from descriptor protos.\n\n Args:\n descriptors: The messages to extract descriptors from.\n Yields:\n A two element tuple of the type name and descriptor object.\n \"\"\"\n\n for desc in descriptors:\n yield (_PrefixWithDot(desc.full_name), desc)\n for symbol in self._ExtractSymbols(desc.nested_types):\n yield symbol\n for enum in desc.enum_types:\n yield (_PrefixWithDot(enum.full_name), enum)\n\n def _GetDeps(self, dependencies, visited=None):\n \"\"\"Recursively finds dependencies for file protos.\n\n Args:\n dependencies: The names of the files being depended on.\n visited: The names of files already found.\n\n Yields:\n Each direct and indirect dependency.\n \"\"\"\n\n visited = visited or set()\n for dependency in dependencies:\n if dependency not in visited:\n visited.add(dependency)\n dep_desc = self.FindFileByName(dependency)\n yield dep_desc\n public_files = [d.name for d in dep_desc.public_dependencies]\n yield from self._GetDeps(public_files, visited)\n\n def _GetTypeFromScope(self, package, type_name, scope):\n \"\"\"Finds a given type name in the current scope.\n\n Args:\n package: The package the proto should be located in.\n type_name: The name of the type to be found in the scope.\n scope: Dict mapping short and full symbols to message and enum types.\n\n Returns:\n The descriptor for the requested type.\n \"\"\"\n if type_name not in scope:\n components = _PrefixWithDot(package).split('.')\n while components:\n possible_match = '.'.join(components + [type_name])\n if possible_match in scope:\n type_name = possible_match\n break\n else:\n components.pop(-1)\n return scope[type_name]\n\n\ndef _PrefixWithDot(name):\n return name if name.startswith('.') else '.%s' % name\n\n\nif _USE_C_DESCRIPTORS:\n # TODO(amauryfa): This pool could be constructed from Python code, when we\n # support a flag like 'use_cpp_generated_pool=True'.\n # pylint: disable=protected-access\n _DEFAULT = descriptor._message.default_pool\nelse:\n _DEFAULT = DescriptorPool()\n\n\ndef Default():\n return _DEFAULT\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/duration_pb2.py",
"content": "# -*- coding: utf-8 -*-\n# Generated by the protocol buffer compiler. DO NOT EDIT!\n# source: google/protobuf/duration.proto\n\"\"\"Generated protocol buffer code.\"\"\"\nfrom google.protobuf.internal import builder as _builder\nfrom google.protobuf import descriptor as _descriptor\nfrom google.protobuf import descriptor_pool as _descriptor_pool\nfrom google.protobuf import symbol_database as _symbol_database\n# @@protoc_insertion_point(imports)\n\n_sym_db = _symbol_database.Default()\n\n\n\n\nDESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\\n\\x1egoogle/protobuf/duration.proto\\x12\\x0fgoogle.protobuf\\\"*\\n\\x08\\x44uration\\x12\\x0f\\n\\x07seconds\\x18\\x01 \\x01(\\x03\\x12\\r\\n\\x05nanos\\x18\\x02 \\x01(\\x05\\x42\\x83\\x01\\n\\x13\\x63om.google.protobufB\\rDurationProtoP\\x01Z1google.golang.org/protobuf/types/known/durationpb\\xf8\\x01\\x01\\xa2\\x02\\x03GPB\\xaa\\x02\\x1eGoogle.Protobuf.WellKnownTypesb\\x06proto3')\n\n_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())\n_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.duration_pb2', globals())\nif _descriptor._USE_C_DESCRIPTORS == False:\n\n DESCRIPTOR._options = None\n DESCRIPTOR._serialized_options = b'\\n\\023com.google.protobufB\\rDurationProtoP\\001Z1google.golang.org/protobuf/types/known/durationpb\\370\\001\\001\\242\\002\\003GPB\\252\\002\\036Google.Protobuf.WellKnownTypes'\n _DURATION._serialized_start=51\n _DURATION._serialized_end=93\n# @@protoc_insertion_point(module_scope)\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/empty_pb2.py",
"content": "# -*- coding: utf-8 -*-\n# Generated by the protocol buffer compiler. DO NOT EDIT!\n# source: google/protobuf/empty.proto\n\"\"\"Generated protocol buffer code.\"\"\"\nfrom google.protobuf.internal import builder as _builder\nfrom google.protobuf import descriptor as _descriptor\nfrom google.protobuf import descriptor_pool as _descriptor_pool\nfrom google.protobuf import symbol_database as _symbol_database\n# @@protoc_insertion_point(imports)\n\n_sym_db = _symbol_database.Default()\n\n\n\n\nDESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\\n\\x1bgoogle/protobuf/empty.proto\\x12\\x0fgoogle.protobuf\\\"\\x07\\n\\x05\\x45mptyB}\\n\\x13\\x63om.google.protobufB\\nEmptyProtoP\\x01Z.google.golang.org/protobuf/types/known/emptypb\\xf8\\x01\\x01\\xa2\\x02\\x03GPB\\xaa\\x02\\x1eGoogle.Protobuf.WellKnownTypesb\\x06proto3')\n\n_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())\n_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.empty_pb2', globals())\nif _descriptor._USE_C_DESCRIPTORS == False:\n\n DESCRIPTOR._options = None\n DESCRIPTOR._serialized_options = b'\\n\\023com.google.protobufB\\nEmptyProtoP\\001Z.google.golang.org/protobuf/types/known/emptypb\\370\\001\\001\\242\\002\\003GPB\\252\\002\\036Google.Protobuf.WellKnownTypes'\n _EMPTY._serialized_start=48\n _EMPTY._serialized_end=55\n# @@protoc_insertion_point(module_scope)\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/field_mask_pb2.py",
"content": "# -*- coding: utf-8 -*-\n# Generated by the protocol buffer compiler. DO NOT EDIT!\n# source: google/protobuf/field_mask.proto\n\"\"\"Generated protocol buffer code.\"\"\"\nfrom google.protobuf.internal import builder as _builder\nfrom google.protobuf import descriptor as _descriptor\nfrom google.protobuf import descriptor_pool as _descriptor_pool\nfrom google.protobuf import symbol_database as _symbol_database\n# @@protoc_insertion_point(imports)\n\n_sym_db = _symbol_database.Default()\n\n\n\n\nDESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\\n google/protobuf/field_mask.proto\\x12\\x0fgoogle.protobuf\\\"\\x1a\\n\\tFieldMask\\x12\\r\\n\\x05paths\\x18\\x01 \\x03(\\tB\\x85\\x01\\n\\x13\\x63om.google.protobufB\\x0e\\x46ieldMaskProtoP\\x01Z2google.golang.org/protobuf/types/known/fieldmaskpb\\xf8\\x01\\x01\\xa2\\x02\\x03GPB\\xaa\\x02\\x1eGoogle.Protobuf.WellKnownTypesb\\x06proto3')\n\n_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())\n_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.field_mask_pb2', globals())\nif _descriptor._USE_C_DESCRIPTORS == False:\n\n DESCRIPTOR._options = None\n DESCRIPTOR._serialized_options = b'\\n\\023com.google.protobufB\\016FieldMaskProtoP\\001Z2google.golang.org/protobuf/types/known/fieldmaskpb\\370\\001\\001\\242\\002\\003GPB\\252\\002\\036Google.Protobuf.WellKnownTypes'\n _FIELDMASK._serialized_start=53\n _FIELDMASK._serialized_end=79\n# @@protoc_insertion_point(module_scope)\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/internal/__init__.py",
"content": ""
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/internal/_parameterized.py",
"content": "#! /usr/bin/env python\n#\n# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n\"\"\"Adds support for parameterized tests to Python's unittest TestCase class.\n\nA parameterized test is a method in a test case that is invoked with different\nargument tuples.\n\nA simple example:\n\n class AdditionExample(parameterized.TestCase):\n @parameterized.parameters(\n (1, 2, 3),\n (4, 5, 9),\n (1, 1, 3))\n def testAddition(self, op1, op2, result):\n self.assertEqual(result, op1 + op2)\n\n\nEach invocation is a separate test case and properly isolated just\nlike a normal test method, with its own setUp/tearDown cycle. In the\nexample above, there are three separate testcases, one of which will\nfail due to an assertion error (1 + 1 != 3).\n\nParameters for individual test cases can be tuples (with positional parameters)\nor dictionaries (with named parameters):\n\n class AdditionExample(parameterized.TestCase):\n @parameterized.parameters(\n {'op1': 1, 'op2': 2, 'result': 3},\n {'op1': 4, 'op2': 5, 'result': 9},\n )\n def testAddition(self, op1, op2, result):\n self.assertEqual(result, op1 + op2)\n\nIf a parameterized test fails, the error message will show the\noriginal test name (which is modified internally) and the arguments\nfor the specific invocation, which are part of the string returned by\nthe shortDescription() method on test cases.\n\nThe id method of the test, used internally by the unittest framework,\nis also modified to show the arguments. To make sure that test names\nstay the same across several invocations, object representations like\n\n >>> class Foo(object):\n ... pass\n >>> repr(Foo())\n '<__main__.Foo object at 0x23d8610>'\n\nare turned into '<__main__.Foo>'. For even more descriptive names,\nespecially in test logs, you can use the named_parameters decorator. In\nthis case, only tuples are supported, and the first parameters has to\nbe a string (or an object that returns an apt name when converted via\nstr()):\n\n class NamedExample(parameterized.TestCase):\n @parameterized.named_parameters(\n ('Normal', 'aa', 'aaa', True),\n ('EmptyPrefix', '', 'abc', True),\n ('BothEmpty', '', '', True))\n def testStartsWith(self, prefix, string, result):\n self.assertEqual(result, strings.startswith(prefix))\n\nNamed tests also have the benefit that they can be run individually\nfrom the command line:\n\n $ testmodule.py NamedExample.testStartsWithNormal\n .\n --------------------------------------------------------------------\n Ran 1 test in 0.000s\n\n OK\n\nParameterized Classes\n=====================\nIf invocation arguments are shared across test methods in a single\nTestCase class, instead of decorating all test methods\nindividually, the class itself can be decorated:\n\n @parameterized.parameters(\n (1, 2, 3)\n (4, 5, 9))\n class ArithmeticTest(parameterized.TestCase):\n def testAdd(self, arg1, arg2, result):\n self.assertEqual(arg1 + arg2, result)\n\n def testSubtract(self, arg2, arg2, result):\n self.assertEqual(result - arg1, arg2)\n\nInputs from Iterables\n=====================\nIf parameters should be shared across several test cases, or are dynamically\ncreated from other sources, a single non-tuple iterable can be passed into\nthe decorator. This iterable will be used to obtain the test cases:\n\n class AdditionExample(parameterized.TestCase):\n @parameterized.parameters(\n c.op1, c.op2, c.result for c in testcases\n )\n def testAddition(self, op1, op2, result):\n self.assertEqual(result, op1 + op2)\n\n\nSingle-Argument Test Methods\n============================\nIf a test method takes only one argument, the single argument does not need to\nbe wrapped into a tuple:\n\n class NegativeNumberExample(parameterized.TestCase):\n @parameterized.parameters(\n -1, -3, -4, -5\n )\n def testIsNegative(self, arg):\n self.assertTrue(IsNegative(arg))\n\"\"\"\n\n__author__ = 'tmarek@google.com (Torsten Marek)'\n\nimport functools\nimport re\nimport types\nimport unittest\nimport uuid\n\ntry:\n # Since python 3\n import collections.abc as collections_abc\nexcept ImportError:\n # Won't work after python 3.8\n import collections as collections_abc\n\nADDR_RE = re.compile(r'\\<([a-zA-Z0-9_\\-\\.]+) object at 0x[a-fA-F0-9]+\\>')\n_SEPARATOR = uuid.uuid1().hex\n_FIRST_ARG = object()\n_ARGUMENT_REPR = object()\n\n\ndef _CleanRepr(obj):\n return ADDR_RE.sub(r'<\\1>', repr(obj))\n\n\n# Helper function formerly from the unittest module, removed from it in\n# Python 2.7.\ndef _StrClass(cls):\n return '%s.%s' % (cls.__module__, cls.__name__)\n\n\ndef _NonStringIterable(obj):\n return (isinstance(obj, collections_abc.Iterable) and\n not isinstance(obj, str))\n\n\ndef _FormatParameterList(testcase_params):\n if isinstance(testcase_params, collections_abc.Mapping):\n return ', '.join('%s=%s' % (argname, _CleanRepr(value))\n for argname, value in testcase_params.items())\n elif _NonStringIterable(testcase_params):\n return ', '.join(map(_CleanRepr, testcase_params))\n else:\n return _FormatParameterList((testcase_params,))\n\n\nclass _ParameterizedTestIter(object):\n \"\"\"Callable and iterable class for producing new test cases.\"\"\"\n\n def __init__(self, test_method, testcases, naming_type):\n \"\"\"Returns concrete test functions for a test and a list of parameters.\n\n The naming_type is used to determine the name of the concrete\n functions as reported by the unittest framework. If naming_type is\n _FIRST_ARG, the testcases must be tuples, and the first element must\n have a string representation that is a valid Python identifier.\n\n Args:\n test_method: The decorated test method.\n testcases: (list of tuple/dict) A list of parameter\n tuples/dicts for individual test invocations.\n naming_type: The test naming type, either _NAMED or _ARGUMENT_REPR.\n \"\"\"\n self._test_method = test_method\n self.testcases = testcases\n self._naming_type = naming_type\n\n def __call__(self, *args, **kwargs):\n raise RuntimeError('You appear to be running a parameterized test case '\n 'without having inherited from parameterized.'\n 'TestCase. This is bad because none of '\n 'your test cases are actually being run.')\n\n def __iter__(self):\n test_method = self._test_method\n naming_type = self._naming_type\n\n def MakeBoundParamTest(testcase_params):\n @functools.wraps(test_method)\n def BoundParamTest(self):\n if isinstance(testcase_params, collections_abc.Mapping):\n test_method(self, **testcase_params)\n elif _NonStringIterable(testcase_params):\n test_method(self, *testcase_params)\n else:\n test_method(self, testcase_params)\n\n if naming_type is _FIRST_ARG:\n # Signal the metaclass that the name of the test function is unique\n # and descriptive.\n BoundParamTest.__x_use_name__ = True\n BoundParamTest.__name__ += str(testcase_params[0])\n testcase_params = testcase_params[1:]\n elif naming_type is _ARGUMENT_REPR:\n # __x_extra_id__ is used to pass naming information to the __new__\n # method of TestGeneratorMetaclass.\n # The metaclass will make sure to create a unique, but nondescriptive\n # name for this test.\n BoundParamTest.__x_extra_id__ = '(%s)' % (\n _FormatParameterList(testcase_params),)\n else:\n raise RuntimeError('%s is not a valid naming type.' % (naming_type,))\n\n BoundParamTest.__doc__ = '%s(%s)' % (\n BoundParamTest.__name__, _FormatParameterList(testcase_params))\n if test_method.__doc__:\n BoundParamTest.__doc__ += '\\n%s' % (test_method.__doc__,)\n return BoundParamTest\n return (MakeBoundParamTest(c) for c in self.testcases)\n\n\ndef _IsSingletonList(testcases):\n \"\"\"True iff testcases contains only a single non-tuple element.\"\"\"\n return len(testcases) == 1 and not isinstance(testcases[0], tuple)\n\n\ndef _ModifyClass(class_object, testcases, naming_type):\n assert not getattr(class_object, '_id_suffix', None), (\n 'Cannot add parameters to %s,'\n ' which already has parameterized methods.' % (class_object,))\n class_object._id_suffix = id_suffix = {}\n # We change the size of __dict__ while we iterate over it,\n # which Python 3.x will complain about, so use copy().\n for name, obj in class_object.__dict__.copy().items():\n if (name.startswith(unittest.TestLoader.testMethodPrefix)\n and isinstance(obj, types.FunctionType)):\n delattr(class_object, name)\n methods = {}\n _UpdateClassDictForParamTestCase(\n methods, id_suffix, name,\n _ParameterizedTestIter(obj, testcases, naming_type))\n for name, meth in methods.items():\n setattr(class_object, name, meth)\n\n\ndef _ParameterDecorator(naming_type, testcases):\n \"\"\"Implementation of the parameterization decorators.\n\n Args:\n naming_type: The naming type.\n testcases: Testcase parameters.\n\n Returns:\n A function for modifying the decorated object.\n \"\"\"\n def _Apply(obj):\n if isinstance(obj, type):\n _ModifyClass(\n obj,\n list(testcases) if not isinstance(testcases, collections_abc.Sequence)\n else testcases,\n naming_type)\n return obj\n else:\n return _ParameterizedTestIter(obj, testcases, naming_type)\n\n if _IsSingletonList(testcases):\n assert _NonStringIterable(testcases[0]), (\n 'Single parameter argument must be a non-string iterable')\n testcases = testcases[0]\n\n return _Apply\n\n\ndef parameters(*testcases): # pylint: disable=invalid-name\n \"\"\"A decorator for creating parameterized tests.\n\n See the module docstring for a usage example.\n Args:\n *testcases: Parameters for the decorated method, either a single\n iterable, or a list of tuples/dicts/objects (for tests\n with only one argument).\n\n Returns:\n A test generator to be handled by TestGeneratorMetaclass.\n \"\"\"\n return _ParameterDecorator(_ARGUMENT_REPR, testcases)\n\n\ndef named_parameters(*testcases): # pylint: disable=invalid-name\n \"\"\"A decorator for creating parameterized tests.\n\n See the module docstring for a usage example. The first element of\n each parameter tuple should be a string and will be appended to the\n name of the test method.\n\n Args:\n *testcases: Parameters for the decorated method, either a single\n iterable, or a list of tuples.\n\n Returns:\n A test generator to be handled by TestGeneratorMetaclass.\n \"\"\"\n return _ParameterDecorator(_FIRST_ARG, testcases)\n\n\nclass TestGeneratorMetaclass(type):\n \"\"\"Metaclass for test cases with test generators.\n\n A test generator is an iterable in a testcase that produces callables. These\n callables must be single-argument methods. These methods are injected into\n the class namespace and the original iterable is removed. If the name of the\n iterable conforms to the test pattern, the injected methods will be picked\n up as tests by the unittest framework.\n\n In general, it is supposed to be used in conjunction with the\n parameters decorator.\n \"\"\"\n\n def __new__(mcs, class_name, bases, dct):\n dct['_id_suffix'] = id_suffix = {}\n for name, obj in dct.copy().items():\n if (name.startswith(unittest.TestLoader.testMethodPrefix) and\n _NonStringIterable(obj)):\n iterator = iter(obj)\n dct.pop(name)\n _UpdateClassDictForParamTestCase(dct, id_suffix, name, iterator)\n\n return type.__new__(mcs, class_name, bases, dct)\n\n\ndef _UpdateClassDictForParamTestCase(dct, id_suffix, name, iterator):\n \"\"\"Adds individual test cases to a dictionary.\n\n Args:\n dct: The target dictionary.\n id_suffix: The dictionary for mapping names to test IDs.\n name: The original name of the test case.\n iterator: The iterator generating the individual test cases.\n \"\"\"\n for idx, func in enumerate(iterator):\n assert callable(func), 'Test generators must yield callables, got %r' % (\n func,)\n if getattr(func, '__x_use_name__', False):\n new_name = func.__name__\n else:\n new_name = '%s%s%d' % (name, _SEPARATOR, idx)\n assert new_name not in dct, (\n 'Name of parameterized test case \"%s\" not unique' % (new_name,))\n dct[new_name] = func\n id_suffix[new_name] = getattr(func, '__x_extra_id__', '')\n\n\nclass TestCase(unittest.TestCase, metaclass=TestGeneratorMetaclass):\n \"\"\"Base class for test cases using the parameters decorator.\"\"\"\n\n def _OriginalName(self):\n return self._testMethodName.split(_SEPARATOR)[0]\n\n def __str__(self):\n return '%s (%s)' % (self._OriginalName(), _StrClass(self.__class__))\n\n def id(self): # pylint: disable=invalid-name\n \"\"\"Returns the descriptive ID of the test.\n\n This is used internally by the unittesting framework to get a name\n for the test to be used in reports.\n\n Returns:\n The test id.\n \"\"\"\n return '%s.%s%s' % (_StrClass(self.__class__),\n self._OriginalName(),\n self._id_suffix.get(self._testMethodName, ''))\n\n\ndef CoopTestCase(other_base_class):\n \"\"\"Returns a new base class with a cooperative metaclass base.\n\n This enables the TestCase to be used in combination\n with other base classes that have custom metaclasses, such as\n mox.MoxTestBase.\n\n Only works with metaclasses that do not override type.__new__.\n\n Example:\n\n import google3\n import mox\n\n from google3.testing.pybase import parameterized\n\n class ExampleTest(parameterized.CoopTestCase(mox.MoxTestBase)):\n ...\n\n Args:\n other_base_class: (class) A test case base class.\n\n Returns:\n A new class object.\n \"\"\"\n metaclass = type(\n 'CoopMetaclass',\n (other_base_class.__metaclass__,\n TestGeneratorMetaclass), {})\n return metaclass(\n 'CoopTestCase',\n (other_base_class, TestCase), {})\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/internal/api_implementation.py",
"content": "# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n\"\"\"Determine which implementation of the protobuf API is used in this process.\n\"\"\"\n\nimport os\nimport sys\nimport warnings\n\ntry:\n # pylint: disable=g-import-not-at-top\n from google.protobuf.internal import _api_implementation\n # The compile-time constants in the _api_implementation module can be used to\n # switch to a certain implementation of the Python API at build time.\n _api_version = _api_implementation.api_version\nexcept ImportError:\n _api_version = -1 # Unspecified by compiler flags.\n\nif _api_version == 1:\n raise ValueError('api_version=1 is no longer supported.')\n\n\n_default_implementation_type = ('cpp' if _api_version > 0 else 'python')\n\n\n# This environment variable can be used to switch to a certain implementation\n# of the Python API, overriding the compile-time constants in the\n# _api_implementation module. Right now only 'python' and 'cpp' are valid\n# values. Any other value will be ignored.\n_implementation_type = os.getenv('PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION',\n _default_implementation_type)\n\nif _implementation_type != 'python':\n _implementation_type = 'cpp'\n\nif 'PyPy' in sys.version and _implementation_type == 'cpp':\n warnings.warn('PyPy does not work yet with cpp protocol buffers. '\n 'Falling back to the python implementation.')\n _implementation_type = 'python'\n\n\n# Detect if serialization should be deterministic by default\ntry:\n # The presence of this module in a build allows the proto implementation to\n # be upgraded merely via build deps.\n #\n # NOTE: Merely importing this automatically enables deterministic proto\n # serialization for C++ code, but we still need to export it as a boolean so\n # that we can do the same for `_implementation_type == 'python'`.\n #\n # NOTE2: It is possible for C++ code to enable deterministic serialization by\n # default _without_ affecting Python code, if the C++ implementation is not in\n # use by this module. That is intended behavior, so we don't actually expose\n # this boolean outside of this module.\n #\n # pylint: disable=g-import-not-at-top,unused-import\n from google.protobuf import enable_deterministic_proto_serialization\n _python_deterministic_proto_serialization = True\nexcept ImportError:\n _python_deterministic_proto_serialization = False\n\n\n# Usage of this function is discouraged. Clients shouldn't care which\n# implementation of the API is in use. Note that there is no guarantee\n# that differences between APIs will be maintained.\n# Please don't use this function if possible.\ndef Type():\n return _implementation_type\n\n\ndef _SetType(implementation_type):\n \"\"\"Never use! Only for protobuf benchmark.\"\"\"\n global _implementation_type\n _implementation_type = implementation_type\n\n\n# See comment on 'Type' above.\ndef Version():\n return 2\n\n\n# For internal use only\ndef IsPythonDefaultSerializationDeterministic():\n return _python_deterministic_proto_serialization\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/internal/builder.py",
"content": "# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n\"\"\"Builds descriptors, message classes and services for generated _pb2.py.\n\nThis file is only called in python generated _pb2.py files. It builds\ndescriptors, message classes and services that users can directly use\nin generated code.\n\"\"\"\n\n__author__ = 'jieluo@google.com (Jie Luo)'\n\nfrom google.protobuf.internal import enum_type_wrapper\nfrom google.protobuf import message as _message\nfrom google.protobuf import reflection as _reflection\nfrom google.protobuf import symbol_database as _symbol_database\n\n_sym_db = _symbol_database.Default()\n\n\ndef BuildMessageAndEnumDescriptors(file_des, module):\n \"\"\"Builds message and enum descriptors.\n\n Args:\n file_des: FileDescriptor of the .proto file\n module: Generated _pb2 module\n \"\"\"\n\n def BuildNestedDescriptors(msg_des, prefix):\n for (name, nested_msg) in msg_des.nested_types_by_name.items():\n module_name = prefix + name.upper()\n module[module_name] = nested_msg\n BuildNestedDescriptors(nested_msg, module_name + '_')\n for enum_des in msg_des.enum_types:\n module[prefix + enum_des.name.upper()] = enum_des\n\n for (name, msg_des) in file_des.message_types_by_name.items():\n module_name = '_' + name.upper()\n module[module_name] = msg_des\n BuildNestedDescriptors(msg_des, module_name + '_')\n\n\ndef BuildTopDescriptorsAndMessages(file_des, module_name, module):\n \"\"\"Builds top level descriptors and message classes.\n\n Args:\n file_des: FileDescriptor of the .proto file\n module_name: str, the name of generated _pb2 module\n module: Generated _pb2 module\n \"\"\"\n\n def BuildMessage(msg_des):\n create_dict = {}\n for (name, nested_msg) in msg_des.nested_types_by_name.items():\n create_dict[name] = BuildMessage(nested_msg)\n create_dict['DESCRIPTOR'] = msg_des\n create_dict['__module__'] = module_name\n message_class = _reflection.GeneratedProtocolMessageType(\n msg_des.name, (_message.Message,), create_dict)\n _sym_db.RegisterMessage(message_class)\n return message_class\n\n # top level enums\n for (name, enum_des) in file_des.enum_types_by_name.items():\n module['_' + name.upper()] = enum_des\n module[name] = enum_type_wrapper.EnumTypeWrapper(enum_des)\n for enum_value in enum_des.values:\n module[enum_value.name] = enum_value.number\n\n # top level extensions\n for (name, extension_des) in file_des.extensions_by_name.items():\n module[name.upper() + '_FIELD_NUMBER'] = extension_des.number\n module[name] = extension_des\n\n # services\n for (name, service) in file_des.services_by_name.items():\n module['_' + name.upper()] = service\n\n # Build messages.\n for (name, msg_des) in file_des.message_types_by_name.items():\n module[name] = BuildMessage(msg_des)\n\n\ndef BuildServices(file_des, module_name, module):\n \"\"\"Builds services classes and services stub class.\n\n Args:\n file_des: FileDescriptor of the .proto file\n module_name: str, the name of generated _pb2 module\n module: Generated _pb2 module\n \"\"\"\n # pylint: disable=g-import-not-at-top\n from google.protobuf import service as _service\n from google.protobuf import service_reflection\n # pylint: enable=g-import-not-at-top\n for (name, service) in file_des.services_by_name.items():\n module[name] = service_reflection.GeneratedServiceType(\n name, (_service.Service,),\n dict(DESCRIPTOR=service, __module__=module_name))\n stub_name = name + '_Stub'\n module[stub_name] = service_reflection.GeneratedServiceStubType(\n stub_name, (module[name],),\n dict(DESCRIPTOR=service, __module__=module_name))\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/internal/containers.py",
"content": "# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n\"\"\"Contains container classes to represent different protocol buffer types.\n\nThis file defines container classes which represent categories of protocol\nbuffer field types which need extra maintenance. Currently these categories\nare:\n\n- Repeated scalar fields - These are all repeated fields which aren't\n composite (e.g. they are of simple types like int32, string, etc).\n- Repeated composite fields - Repeated fields which are composite. This\n includes groups and nested messages.\n\"\"\"\n\nimport collections.abc\nimport copy\nimport pickle\nfrom typing import (\n Any,\n Iterable,\n Iterator,\n List,\n MutableMapping,\n MutableSequence,\n NoReturn,\n Optional,\n Sequence,\n TypeVar,\n Union,\n overload,\n)\n\n\n_T = TypeVar('_T')\n_K = TypeVar('_K')\n_V = TypeVar('_V')\n\n\nclass BaseContainer(Sequence[_T]):\n \"\"\"Base container class.\"\"\"\n\n # Minimizes memory usage and disallows assignment to other attributes.\n __slots__ = ['_message_listener', '_values']\n\n def __init__(self, message_listener: Any) -> None:\n \"\"\"\n Args:\n message_listener: A MessageListener implementation.\n The RepeatedScalarFieldContainer will call this object's\n Modified() method when it is modified.\n \"\"\"\n self._message_listener = message_listener\n self._values = []\n\n @overload\n def __getitem__(self, key: int) -> _T:\n ...\n\n @overload\n def __getitem__(self, key: slice) -> List[_T]:\n ...\n\n def __getitem__(self, key):\n \"\"\"Retrieves item by the specified key.\"\"\"\n return self._values[key]\n\n def __len__(self) -> int:\n \"\"\"Returns the number of elements in the container.\"\"\"\n return len(self._values)\n\n def __ne__(self, other: Any) -> bool:\n \"\"\"Checks if another instance isn't equal to this one.\"\"\"\n # The concrete classes should define __eq__.\n return not self == other\n\n __hash__ = None\n\n def __repr__(self) -> str:\n return repr(self._values)\n\n def sort(self, *args, **kwargs) -> None:\n # Continue to support the old sort_function keyword argument.\n # This is expected to be a rare occurrence, so use LBYL to avoid\n # the overhead of actually catching KeyError.\n if 'sort_function' in kwargs:\n kwargs['cmp'] = kwargs.pop('sort_function')\n self._values.sort(*args, **kwargs)\n\n def reverse(self) -> None:\n self._values.reverse()\n\n\n# TODO(slebedev): Remove this. BaseContainer does *not* conform to\n# MutableSequence, only its subclasses do.\ncollections.abc.MutableSequence.register(BaseContainer)\n\n\nclass RepeatedScalarFieldContainer(BaseContainer[_T], MutableSequence[_T]):\n \"\"\"Simple, type-checked, list-like container for holding repeated scalars.\"\"\"\n\n # Disallows assignment to other attributes.\n __slots__ = ['_type_checker']\n\n def __init__(\n self,\n message_listener: Any,\n type_checker: Any,\n ) -> None:\n \"\"\"Args:\n\n message_listener: A MessageListener implementation. The\n RepeatedScalarFieldContainer will call this object's Modified() method\n when it is modified.\n type_checker: A type_checkers.ValueChecker instance to run on elements\n inserted into this container.\n \"\"\"\n super().__init__(message_listener)\n self._type_checker = type_checker\n\n def append(self, value: _T) -> None:\n \"\"\"Appends an item to the list. Similar to list.append().\"\"\"\n self._values.append(self._type_checker.CheckValue(value))\n if not self._message_listener.dirty:\n self._message_listener.Modified()\n\n def insert(self, key: int, value: _T) -> None:\n \"\"\"Inserts the item at the specified position. Similar to list.insert().\"\"\"\n self._values.insert(key, self._type_checker.CheckValue(value))\n if not self._message_listener.dirty:\n self._message_listener.Modified()\n\n def extend(self, elem_seq: Iterable[_T]) -> None:\n \"\"\"Extends by appending the given iterable. Similar to list.extend().\"\"\"\n if elem_seq is None:\n return\n try:\n elem_seq_iter = iter(elem_seq)\n except TypeError:\n if not elem_seq:\n # silently ignore falsy inputs :-/.\n # TODO(ptucker): Deprecate this behavior. b/18413862\n return\n raise\n\n new_values = [self._type_checker.CheckValue(elem) for elem in elem_seq_iter]\n if new_values:\n self._values.extend(new_values)\n self._message_listener.Modified()\n\n def MergeFrom(\n self,\n other: Union['RepeatedScalarFieldContainer[_T]', Iterable[_T]],\n ) -> None:\n \"\"\"Appends the contents of another repeated field of the same type to this\n one. We do not check the types of the individual fields.\n \"\"\"\n self._values.extend(other)\n self._message_listener.Modified()\n\n def remove(self, elem: _T):\n \"\"\"Removes an item from the list. Similar to list.remove().\"\"\"\n self._values.remove(elem)\n self._message_listener.Modified()\n\n def pop(self, key: Optional[int] = -1) -> _T:\n \"\"\"Removes and returns an item at a given index. Similar to list.pop().\"\"\"\n value = self._values[key]\n self.__delitem__(key)\n return value\n\n @overload\n def __setitem__(self, key: int, value: _T) -> None:\n ...\n\n @overload\n def __setitem__(self, key: slice, value: Iterable[_T]) -> None:\n ...\n\n def __setitem__(self, key, value) -> None:\n \"\"\"Sets the item on the specified position.\"\"\"\n if isinstance(key, slice):\n if key.step is not None:\n raise ValueError('Extended slices not supported')\n self._values[key] = map(self._type_checker.CheckValue, value)\n self._message_listener.Modified()\n else:\n self._values[key] = self._type_checker.CheckValue(value)\n self._message_listener.Modified()\n\n def __delitem__(self, key: Union[int, slice]) -> None:\n \"\"\"Deletes the item at the specified position.\"\"\"\n del self._values[key]\n self._message_listener.Modified()\n\n def __eq__(self, other: Any) -> bool:\n \"\"\"Compares the current instance with another one.\"\"\"\n if self is other:\n return True\n # Special case for the same type which should be common and fast.\n if isinstance(other, self.__class__):\n return other._values == self._values\n # We are presumably comparing against some other sequence type.\n return other == self._values\n\n def __deepcopy__(\n self,\n unused_memo: Any = None,\n ) -> 'RepeatedScalarFieldContainer[_T]':\n clone = RepeatedScalarFieldContainer(\n copy.deepcopy(self._message_listener), self._type_checker)\n clone.MergeFrom(self)\n return clone\n\n def __reduce__(self, **kwargs) -> NoReturn:\n raise pickle.PickleError(\n \"Can't pickle repeated scalar fields, convert to list first\")\n\n\n# TODO(slebedev): Constrain T to be a subtype of Message.\nclass RepeatedCompositeFieldContainer(BaseContainer[_T], MutableSequence[_T]):\n \"\"\"Simple, list-like container for holding repeated composite fields.\"\"\"\n\n # Disallows assignment to other attributes.\n __slots__ = ['_message_descriptor']\n\n def __init__(self, message_listener: Any, message_descriptor: Any) -> None:\n \"\"\"\n Note that we pass in a descriptor instead of the generated directly,\n since at the time we construct a _RepeatedCompositeFieldContainer we\n haven't yet necessarily initialized the type that will be contained in the\n container.\n\n Args:\n message_listener: A MessageListener implementation.\n The RepeatedCompositeFieldContainer will call this object's\n Modified() method when it is modified.\n message_descriptor: A Descriptor instance describing the protocol type\n that should be present in this container. We'll use the\n _concrete_class field of this descriptor when the client calls add().\n \"\"\"\n super().__init__(message_listener)\n self._message_descriptor = message_descriptor\n\n def add(self, **kwargs: Any) -> _T:\n \"\"\"Adds a new element at the end of the list and returns it. Keyword\n arguments may be used to initialize the element.\n \"\"\"\n new_element = self._message_descriptor._concrete_class(**kwargs)\n new_element._SetListener(self._message_listener)\n self._values.append(new_element)\n if not self._message_listener.dirty:\n self._message_listener.Modified()\n return new_element\n\n def append(self, value: _T) -> None:\n \"\"\"Appends one element by copying the message.\"\"\"\n new_element = self._message_descriptor._concrete_class()\n new_element._SetListener(self._message_listener)\n new_element.CopyFrom(value)\n self._values.append(new_element)\n if not self._message_listener.dirty:\n self._message_listener.Modified()\n\n def insert(self, key: int, value: _T) -> None:\n \"\"\"Inserts the item at the specified position by copying.\"\"\"\n new_element = self._message_descriptor._concrete_class()\n new_element._SetListener(self._message_listener)\n new_element.CopyFrom(value)\n self._values.insert(key, new_element)\n if not self._message_listener.dirty:\n self._message_listener.Modified()\n\n def extend(self, elem_seq: Iterable[_T]) -> None:\n \"\"\"Extends by appending the given sequence of elements of the same type\n\n as this one, copying each individual message.\n \"\"\"\n message_class = self._message_descriptor._concrete_class\n listener = self._message_listener\n values = self._values\n for message in elem_seq:\n new_element = message_class()\n new_element._SetListener(listener)\n new_element.MergeFrom(message)\n values.append(new_element)\n listener.Modified()\n\n def MergeFrom(\n self,\n other: Union['RepeatedCompositeFieldContainer[_T]', Iterable[_T]],\n ) -> None:\n \"\"\"Appends the contents of another repeated field of the same type to this\n one, copying each individual message.\n \"\"\"\n self.extend(other)\n\n def remove(self, elem: _T) -> None:\n \"\"\"Removes an item from the list. Similar to list.remove().\"\"\"\n self._values.remove(elem)\n self._message_listener.Modified()\n\n def pop(self, key: Optional[int] = -1) -> _T:\n \"\"\"Removes and returns an item at a given index. Similar to list.pop().\"\"\"\n value = self._values[key]\n self.__delitem__(key)\n return value\n\n @overload\n def __setitem__(self, key: int, value: _T) -> None:\n ...\n\n @overload\n def __setitem__(self, key: slice, value: Iterable[_T]) -> None:\n ...\n\n def __setitem__(self, key, value):\n # This method is implemented to make RepeatedCompositeFieldContainer\n # structurally compatible with typing.MutableSequence. It is\n # otherwise unsupported and will always raise an error.\n raise TypeError(\n f'{self.__class__.__name__} object does not support item assignment')\n\n def __delitem__(self, key: Union[int, slice]) -> None:\n \"\"\"Deletes the item at the specified position.\"\"\"\n del self._values[key]\n self._message_listener.Modified()\n\n def __eq__(self, other: Any) -> bool:\n \"\"\"Compares the current instance with another one.\"\"\"\n if self is other:\n return True\n if not isinstance(other, self.__class__):\n raise TypeError('Can only compare repeated composite fields against '\n 'other repeated composite fields.')\n return self._values == other._values\n\n\nclass ScalarMap(MutableMapping[_K, _V]):\n \"\"\"Simple, type-checked, dict-like container for holding repeated scalars.\"\"\"\n\n # Disallows assignment to other attributes.\n __slots__ = ['_key_checker', '_value_checker', '_values', '_message_listener',\n '_entry_descriptor']\n\n def __init__(\n self,\n message_listener: Any,\n key_checker: Any,\n value_checker: Any,\n entry_descriptor: Any,\n ) -> None:\n \"\"\"\n Args:\n message_listener: A MessageListener implementation.\n The ScalarMap will call this object's Modified() method when it\n is modified.\n key_checker: A type_checkers.ValueChecker instance to run on keys\n inserted into this container.\n value_checker: A type_checkers.ValueChecker instance to run on values\n inserted into this container.\n entry_descriptor: The MessageDescriptor of a map entry: key and value.\n \"\"\"\n self._message_listener = message_listener\n self._key_checker = key_checker\n self._value_checker = value_checker\n self._entry_descriptor = entry_descriptor\n self._values = {}\n\n def __getitem__(self, key: _K) -> _V:\n try:\n return self._values[key]\n except KeyError:\n key = self._key_checker.CheckValue(key)\n val = self._value_checker.DefaultValue()\n self._values[key] = val\n return val\n\n def __contains__(self, item: _K) -> bool:\n # We check the key's type to match the strong-typing flavor of the API.\n # Also this makes it easier to match the behavior of the C++ implementation.\n self._key_checker.CheckValue(item)\n return item in self._values\n\n @overload\n def get(self, key: _K) -> Optional[_V]:\n ...\n\n @overload\n def get(self, key: _K, default: _T) -> Union[_V, _T]:\n ...\n\n # We need to override this explicitly, because our defaultdict-like behavior\n # will make the default implementation (from our base class) always insert\n # the key.\n def get(self, key, default=None):\n if key in self:\n return self[key]\n else:\n return default\n\n def __setitem__(self, key: _K, value: _V) -> _T:\n checked_key = self._key_checker.CheckValue(key)\n checked_value = self._value_checker.CheckValue(value)\n self._values[checked_key] = checked_value\n self._message_listener.Modified()\n\n def __delitem__(self, key: _K) -> None:\n del self._values[key]\n self._message_listener.Modified()\n\n def __len__(self) -> int:\n return len(self._values)\n\n def __iter__(self) -> Iterator[_K]:\n return iter(self._values)\n\n def __repr__(self) -> str:\n return repr(self._values)\n\n def MergeFrom(self, other: 'ScalarMap[_K, _V]') -> None:\n self._values.update(other._values)\n self._message_listener.Modified()\n\n def InvalidateIterators(self) -> None:\n # It appears that the only way to reliably invalidate iterators to\n # self._values is to ensure that its size changes.\n original = self._values\n self._values = original.copy()\n original[None] = None\n\n # This is defined in the abstract base, but we can do it much more cheaply.\n def clear(self) -> None:\n self._values.clear()\n self._message_listener.Modified()\n\n def GetEntryClass(self) -> Any:\n return self._entry_descriptor._concrete_class\n\n\nclass MessageMap(MutableMapping[_K, _V]):\n \"\"\"Simple, type-checked, dict-like container for with submessage values.\"\"\"\n\n # Disallows assignment to other attributes.\n __slots__ = ['_key_checker', '_values', '_message_listener',\n '_message_descriptor', '_entry_descriptor']\n\n def __init__(\n self,\n message_listener: Any,\n message_descriptor: Any,\n key_checker: Any,\n entry_descriptor: Any,\n ) -> None:\n \"\"\"\n Args:\n message_listener: A MessageListener implementation.\n The ScalarMap will call this object's Modified() method when it\n is modified.\n key_checker: A type_checkers.ValueChecker instance to run on keys\n inserted into this container.\n value_checker: A type_checkers.ValueChecker instance to run on values\n inserted into this container.\n entry_descriptor: The MessageDescriptor of a map entry: key and value.\n \"\"\"\n self._message_listener = message_listener\n self._message_descriptor = message_descriptor\n self._key_checker = key_checker\n self._entry_descriptor = entry_descriptor\n self._values = {}\n\n def __getitem__(self, key: _K) -> _V:\n key = self._key_checker.CheckValue(key)\n try:\n return self._values[key]\n except KeyError:\n new_element = self._message_descriptor._concrete_class()\n new_element._SetListener(self._message_listener)\n self._values[key] = new_element\n self._message_listener.Modified()\n return new_element\n\n def get_or_create(self, key: _K) -> _V:\n \"\"\"get_or_create() is an alias for getitem (ie. map[key]).\n\n Args:\n key: The key to get or create in the map.\n\n This is useful in cases where you want to be explicit that the call is\n mutating the map. This can avoid lint errors for statements like this\n that otherwise would appear to be pointless statements:\n\n msg.my_map[key]\n \"\"\"\n return self[key]\n\n @overload\n def get(self, key: _K) -> Optional[_V]:\n ...\n\n @overload\n def get(self, key: _K, default: _T) -> Union[_V, _T]:\n ...\n\n # We need to override this explicitly, because our defaultdict-like behavior\n # will make the default implementation (from our base class) always insert\n # the key.\n def get(self, key, default=None):\n if key in self:\n return self[key]\n else:\n return default\n\n def __contains__(self, item: _K) -> bool:\n item = self._key_checker.CheckValue(item)\n return item in self._values\n\n def __setitem__(self, key: _K, value: _V) -> NoReturn:\n raise ValueError('May not set values directly, call my_map[key].foo = 5')\n\n def __delitem__(self, key: _K) -> None:\n key = self._key_checker.CheckValue(key)\n del self._values[key]\n self._message_listener.Modified()\n\n def __len__(self) -> int:\n return len(self._values)\n\n def __iter__(self) -> Iterator[_K]:\n return iter(self._values)\n\n def __repr__(self) -> str:\n return repr(self._values)\n\n def MergeFrom(self, other: 'MessageMap[_K, _V]') -> None:\n # pylint: disable=protected-access\n for key in other._values:\n # According to documentation: \"When parsing from the wire or when merging,\n # if there are duplicate map keys the last key seen is used\".\n if key in self:\n del self[key]\n self[key].CopyFrom(other[key])\n # self._message_listener.Modified() not required here, because\n # mutations to submessages already propagate.\n\n def InvalidateIterators(self) -> None:\n # It appears that the only way to reliably invalidate iterators to\n # self._values is to ensure that its size changes.\n original = self._values\n self._values = original.copy()\n original[None] = None\n\n # This is defined in the abstract base, but we can do it much more cheaply.\n def clear(self) -> None:\n self._values.clear()\n self._message_listener.Modified()\n\n def GetEntryClass(self) -> Any:\n return self._entry_descriptor._concrete_class\n\n\nclass _UnknownField:\n \"\"\"A parsed unknown field.\"\"\"\n\n # Disallows assignment to other attributes.\n __slots__ = ['_field_number', '_wire_type', '_data']\n\n def __init__(self, field_number, wire_type, data):\n self._field_number = field_number\n self._wire_type = wire_type\n self._data = data\n return\n\n def __lt__(self, other):\n # pylint: disable=protected-access\n return self._field_number < other._field_number\n\n def __eq__(self, other):\n if self is other:\n return True\n # pylint: disable=protected-access\n return (self._field_number == other._field_number and\n self._wire_type == other._wire_type and\n self._data == other._data)\n\n\nclass UnknownFieldRef: # pylint: disable=missing-class-docstring\n\n def __init__(self, parent, index):\n self._parent = parent\n self._index = index\n\n def _check_valid(self):\n if not self._parent:\n raise ValueError('UnknownField does not exist. '\n 'The parent message might be cleared.')\n if self._index >= len(self._parent):\n raise ValueError('UnknownField does not exist. '\n 'The parent message might be cleared.')\n\n @property\n def field_number(self):\n self._check_valid()\n # pylint: disable=protected-access\n return self._parent._internal_get(self._index)._field_number\n\n @property\n def wire_type(self):\n self._check_valid()\n # pylint: disable=protected-access\n return self._parent._internal_get(self._index)._wire_type\n\n @property\n def data(self):\n self._check_valid()\n # pylint: disable=protected-access\n return self._parent._internal_get(self._index)._data\n\n\nclass UnknownFieldSet:\n \"\"\"UnknownField container\"\"\"\n\n # Disallows assignment to other attributes.\n __slots__ = ['_values']\n\n def __init__(self):\n self._values = []\n\n def __getitem__(self, index):\n if self._values is None:\n raise ValueError('UnknownFields does not exist. '\n 'The parent message might be cleared.')\n size = len(self._values)\n if index < 0:\n index += size\n if index < 0 or index >= size:\n raise IndexError('index %d out of range'.index)\n\n return UnknownFieldRef(self, index)\n\n def _internal_get(self, index):\n return self._values[index]\n\n def __len__(self):\n if self._values is None:\n raise ValueError('UnknownFields does not exist. '\n 'The parent message might be cleared.')\n return len(self._values)\n\n def _add(self, field_number, wire_type, data):\n unknown_field = _UnknownField(field_number, wire_type, data)\n self._values.append(unknown_field)\n return unknown_field\n\n def __iter__(self):\n for i in range(len(self)):\n yield UnknownFieldRef(self, i)\n\n def _extend(self, other):\n if other is None:\n return\n # pylint: disable=protected-access\n self._values.extend(other._values)\n\n def __eq__(self, other):\n if self is other:\n return True\n # Sort unknown fields because their order shouldn't\n # affect equality test.\n values = list(self._values)\n if other is None:\n return not values\n values.sort()\n # pylint: disable=protected-access\n other_values = sorted(other._values)\n return values == other_values\n\n def _clear(self):\n for value in self._values:\n # pylint: disable=protected-access\n if isinstance(value._data, UnknownFieldSet):\n value._data._clear() # pylint: disable=protected-access\n self._values = None\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/internal/decoder.py",
"content": "# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n\"\"\"Code for decoding protocol buffer primitives.\n\nThis code is very similar to encoder.py -- read the docs for that module first.\n\nA \"decoder\" is a function with the signature:\n Decode(buffer, pos, end, message, field_dict)\nThe arguments are:\n buffer: The string containing the encoded message.\n pos: The current position in the string.\n end: The position in the string where the current message ends. May be\n less than len(buffer) if we're reading a sub-message.\n message: The message object into which we're parsing.\n field_dict: message._fields (avoids a hashtable lookup).\nThe decoder reads the field and stores it into field_dict, returning the new\nbuffer position. A decoder for a repeated field may proactively decode all of\nthe elements of that field, if they appear consecutively.\n\nNote that decoders may throw any of the following:\n IndexError: Indicates a truncated message.\n struct.error: Unpacking of a fixed-width field failed.\n message.DecodeError: Other errors.\n\nDecoders are expected to raise an exception if they are called with pos > end.\nThis allows callers to be lax about bounds checking: it's fineto read past\n\"end\" as long as you are sure that someone else will notice and throw an\nexception later on.\n\nSomething up the call stack is expected to catch IndexError and struct.error\nand convert them to message.DecodeError.\n\nDecoders are constructed using decoder constructors with the signature:\n MakeDecoder(field_number, is_repeated, is_packed, key, new_default)\nThe arguments are:\n field_number: The field number of the field we want to decode.\n is_repeated: Is the field a repeated field? (bool)\n is_packed: Is the field a packed field? (bool)\n key: The key to use when looking up the field within field_dict.\n (This is actually the FieldDescriptor but nothing in this\n file should depend on that.)\n new_default: A function which takes a message object as a parameter and\n returns a new instance of the default value for this field.\n (This is called for repeated fields and sub-messages, when an\n instance does not already exist.)\n\nAs with encoders, we define a decoder constructor for every type of field.\nThen, for every field of every message class we construct an actual decoder.\nThat decoder goes into a dict indexed by tag, so when we decode a message\nwe repeatedly read a tag, look up the corresponding decoder, and invoke it.\n\"\"\"\n\n__author__ = 'kenton@google.com (Kenton Varda)'\n\nimport math\nimport struct\n\nfrom google.protobuf.internal import containers\nfrom google.protobuf.internal import encoder\nfrom google.protobuf.internal import wire_format\nfrom google.protobuf import message\n\n\n# This is not for optimization, but rather to avoid conflicts with local\n# variables named \"message\".\n_DecodeError = message.DecodeError\n\n\ndef _VarintDecoder(mask, result_type):\n \"\"\"Return an encoder for a basic varint value (does not include tag).\n\n Decoded values will be bitwise-anded with the given mask before being\n returned, e.g. to limit them to 32 bits. The returned decoder does not\n take the usual \"end\" parameter -- the caller is expected to do bounds checking\n after the fact (often the caller can defer such checking until later). The\n decoder returns a (value, new_pos) pair.\n \"\"\"\n\n def DecodeVarint(buffer, pos):\n result = 0\n shift = 0\n while 1:\n b = buffer[pos]\n result |= ((b & 0x7f) << shift)\n pos += 1\n if not (b & 0x80):\n result &= mask\n result = result_type(result)\n return (result, pos)\n shift += 7\n if shift >= 64:\n raise _DecodeError('Too many bytes when decoding varint.')\n return DecodeVarint\n\n\ndef _SignedVarintDecoder(bits, result_type):\n \"\"\"Like _VarintDecoder() but decodes signed values.\"\"\"\n\n signbit = 1 << (bits - 1)\n mask = (1 << bits) - 1\n\n def DecodeVarint(buffer, pos):\n result = 0\n shift = 0\n while 1:\n b = buffer[pos]\n result |= ((b & 0x7f) << shift)\n pos += 1\n if not (b & 0x80):\n result &= mask\n result = (result ^ signbit) - signbit\n result = result_type(result)\n return (result, pos)\n shift += 7\n if shift >= 64:\n raise _DecodeError('Too many bytes when decoding varint.')\n return DecodeVarint\n\n# All 32-bit and 64-bit values are represented as int.\n_DecodeVarint = _VarintDecoder((1 << 64) - 1, int)\n_DecodeSignedVarint = _SignedVarintDecoder(64, int)\n\n# Use these versions for values which must be limited to 32 bits.\n_DecodeVarint32 = _VarintDecoder((1 << 32) - 1, int)\n_DecodeSignedVarint32 = _SignedVarintDecoder(32, int)\n\n\ndef ReadTag(buffer, pos):\n \"\"\"Read a tag from the memoryview, and return a (tag_bytes, new_pos) tuple.\n\n We return the raw bytes of the tag rather than decoding them. The raw\n bytes can then be used to look up the proper decoder. This effectively allows\n us to trade some work that would be done in pure-python (decoding a varint)\n for work that is done in C (searching for a byte string in a hash table).\n In a low-level language it would be much cheaper to decode the varint and\n use that, but not in Python.\n\n Args:\n buffer: memoryview object of the encoded bytes\n pos: int of the current position to start from\n\n Returns:\n Tuple[bytes, int] of the tag data and new position.\n \"\"\"\n start = pos\n while buffer[pos] & 0x80:\n pos += 1\n pos += 1\n\n tag_bytes = buffer[start:pos].tobytes()\n return tag_bytes, pos\n\n\n# --------------------------------------------------------------------\n\n\ndef _SimpleDecoder(wire_type, decode_value):\n \"\"\"Return a constructor for a decoder for fields of a particular type.\n\n Args:\n wire_type: The field's wire type.\n decode_value: A function which decodes an individual value, e.g.\n _DecodeVarint()\n \"\"\"\n\n def SpecificDecoder(field_number, is_repeated, is_packed, key, new_default,\n clear_if_default=False):\n if is_packed:\n local_DecodeVarint = _DecodeVarint\n def DecodePackedField(buffer, pos, end, message, field_dict):\n value = field_dict.get(key)\n if value is None:\n value = field_dict.setdefault(key, new_default(message))\n (endpoint, pos) = local_DecodeVarint(buffer, pos)\n endpoint += pos\n if endpoint > end:\n raise _DecodeError('Truncated message.')\n while pos < endpoint:\n (element, pos) = decode_value(buffer, pos)\n value.append(element)\n if pos > endpoint:\n del value[-1] # Discard corrupt value.\n raise _DecodeError('Packed element was truncated.')\n return pos\n return DecodePackedField\n elif is_repeated:\n tag_bytes = encoder.TagBytes(field_number, wire_type)\n tag_len = len(tag_bytes)\n def DecodeRepeatedField(buffer, pos, end, message, field_dict):\n value = field_dict.get(key)\n if value is None:\n value = field_dict.setdefault(key, new_default(message))\n while 1:\n (element, new_pos) = decode_value(buffer, pos)\n value.append(element)\n # Predict that the next tag is another copy of the same repeated\n # field.\n pos = new_pos + tag_len\n if buffer[new_pos:pos] != tag_bytes or new_pos >= end:\n # Prediction failed. Return.\n if new_pos > end:\n raise _DecodeError('Truncated message.')\n return new_pos\n return DecodeRepeatedField\n else:\n def DecodeField(buffer, pos, end, message, field_dict):\n (new_value, pos) = decode_value(buffer, pos)\n if pos > end:\n raise _DecodeError('Truncated message.')\n if clear_if_default and not new_value:\n field_dict.pop(key, None)\n else:\n field_dict[key] = new_value\n return pos\n return DecodeField\n\n return SpecificDecoder\n\n\ndef _ModifiedDecoder(wire_type, decode_value, modify_value):\n \"\"\"Like SimpleDecoder but additionally invokes modify_value on every value\n before storing it. Usually modify_value is ZigZagDecode.\n \"\"\"\n\n # Reusing _SimpleDecoder is slightly slower than copying a bunch of code, but\n # not enough to make a significant difference.\n\n def InnerDecode(buffer, pos):\n (result, new_pos) = decode_value(buffer, pos)\n return (modify_value(result), new_pos)\n return _SimpleDecoder(wire_type, InnerDecode)\n\n\ndef _StructPackDecoder(wire_type, format):\n \"\"\"Return a constructor for a decoder for a fixed-width field.\n\n Args:\n wire_type: The field's wire type.\n format: The format string to pass to struct.unpack().\n \"\"\"\n\n value_size = struct.calcsize(format)\n local_unpack = struct.unpack\n\n # Reusing _SimpleDecoder is slightly slower than copying a bunch of code, but\n # not enough to make a significant difference.\n\n # Note that we expect someone up-stack to catch struct.error and convert\n # it to _DecodeError -- this way we don't have to set up exception-\n # handling blocks every time we parse one value.\n\n def InnerDecode(buffer, pos):\n new_pos = pos + value_size\n result = local_unpack(format, buffer[pos:new_pos])[0]\n return (result, new_pos)\n return _SimpleDecoder(wire_type, InnerDecode)\n\n\ndef _FloatDecoder():\n \"\"\"Returns a decoder for a float field.\n\n This code works around a bug in struct.unpack for non-finite 32-bit\n floating-point values.\n \"\"\"\n\n local_unpack = struct.unpack\n\n def InnerDecode(buffer, pos):\n \"\"\"Decode serialized float to a float and new position.\n\n Args:\n buffer: memoryview of the serialized bytes\n pos: int, position in the memory view to start at.\n\n Returns:\n Tuple[float, int] of the deserialized float value and new position\n in the serialized data.\n \"\"\"\n # We expect a 32-bit value in little-endian byte order. Bit 1 is the sign\n # bit, bits 2-9 represent the exponent, and bits 10-32 are the significand.\n new_pos = pos + 4\n float_bytes = buffer[pos:new_pos].tobytes()\n\n # If this value has all its exponent bits set, then it's non-finite.\n # In Python 2.4, struct.unpack will convert it to a finite 64-bit value.\n # To avoid that, we parse it specially.\n if (float_bytes[3:4] in b'\\x7F\\xFF' and float_bytes[2:3] >= b'\\x80'):\n # If at least one significand bit is set...\n if float_bytes[0:3] != b'\\x00\\x00\\x80':\n return (math.nan, new_pos)\n # If sign bit is set...\n if float_bytes[3:4] == b'\\xFF':\n return (-math.inf, new_pos)\n return (math.inf, new_pos)\n\n # Note that we expect someone up-stack to catch struct.error and convert\n # it to _DecodeError -- this way we don't have to set up exception-\n # handling blocks every time we parse one value.\n result = local_unpack('= b'\\xF0')\n and (double_bytes[0:7] != b'\\x00\\x00\\x00\\x00\\x00\\x00\\xF0')):\n return (math.nan, new_pos)\n\n # Note that we expect someone up-stack to catch struct.error and convert\n # it to _DecodeError -- this way we don't have to set up exception-\n # handling blocks every time we parse one value.\n result = local_unpack(' end:\n raise _DecodeError('Truncated message.')\n while pos < endpoint:\n value_start_pos = pos\n (element, pos) = _DecodeSignedVarint32(buffer, pos)\n # pylint: disable=protected-access\n if element in enum_type.values_by_number:\n value.append(element)\n else:\n if not message._unknown_fields:\n message._unknown_fields = []\n tag_bytes = encoder.TagBytes(field_number,\n wire_format.WIRETYPE_VARINT)\n\n message._unknown_fields.append(\n (tag_bytes, buffer[value_start_pos:pos].tobytes()))\n if message._unknown_field_set is None:\n message._unknown_field_set = containers.UnknownFieldSet()\n message._unknown_field_set._add(\n field_number, wire_format.WIRETYPE_VARINT, element)\n # pylint: enable=protected-access\n if pos > endpoint:\n if element in enum_type.values_by_number:\n del value[-1] # Discard corrupt value.\n else:\n del message._unknown_fields[-1]\n # pylint: disable=protected-access\n del message._unknown_field_set._values[-1]\n # pylint: enable=protected-access\n raise _DecodeError('Packed element was truncated.')\n return pos\n return DecodePackedField\n elif is_repeated:\n tag_bytes = encoder.TagBytes(field_number, wire_format.WIRETYPE_VARINT)\n tag_len = len(tag_bytes)\n def DecodeRepeatedField(buffer, pos, end, message, field_dict):\n \"\"\"Decode serialized repeated enum to its value and a new position.\n\n Args:\n buffer: memoryview of the serialized bytes.\n pos: int, position in the memory view to start at.\n end: int, end position of serialized data\n message: Message object to store unknown fields in\n field_dict: Map[Descriptor, Any] to store decoded values in.\n\n Returns:\n int, new position in serialized data.\n \"\"\"\n value = field_dict.get(key)\n if value is None:\n value = field_dict.setdefault(key, new_default(message))\n while 1:\n (element, new_pos) = _DecodeSignedVarint32(buffer, pos)\n # pylint: disable=protected-access\n if element in enum_type.values_by_number:\n value.append(element)\n else:\n if not message._unknown_fields:\n message._unknown_fields = []\n message._unknown_fields.append(\n (tag_bytes, buffer[pos:new_pos].tobytes()))\n if message._unknown_field_set is None:\n message._unknown_field_set = containers.UnknownFieldSet()\n message._unknown_field_set._add(\n field_number, wire_format.WIRETYPE_VARINT, element)\n # pylint: enable=protected-access\n # Predict that the next tag is another copy of the same repeated\n # field.\n pos = new_pos + tag_len\n if buffer[new_pos:pos] != tag_bytes or new_pos >= end:\n # Prediction failed. Return.\n if new_pos > end:\n raise _DecodeError('Truncated message.')\n return new_pos\n return DecodeRepeatedField\n else:\n def DecodeField(buffer, pos, end, message, field_dict):\n \"\"\"Decode serialized repeated enum to its value and a new position.\n\n Args:\n buffer: memoryview of the serialized bytes.\n pos: int, position in the memory view to start at.\n end: int, end position of serialized data\n message: Message object to store unknown fields in\n field_dict: Map[Descriptor, Any] to store decoded values in.\n\n Returns:\n int, new position in serialized data.\n \"\"\"\n value_start_pos = pos\n (enum_value, pos) = _DecodeSignedVarint32(buffer, pos)\n if pos > end:\n raise _DecodeError('Truncated message.')\n if clear_if_default and not enum_value:\n field_dict.pop(key, None)\n return pos\n # pylint: disable=protected-access\n if enum_value in enum_type.values_by_number:\n field_dict[key] = enum_value\n else:\n if not message._unknown_fields:\n message._unknown_fields = []\n tag_bytes = encoder.TagBytes(field_number,\n wire_format.WIRETYPE_VARINT)\n message._unknown_fields.append(\n (tag_bytes, buffer[value_start_pos:pos].tobytes()))\n if message._unknown_field_set is None:\n message._unknown_field_set = containers.UnknownFieldSet()\n message._unknown_field_set._add(\n field_number, wire_format.WIRETYPE_VARINT, enum_value)\n # pylint: enable=protected-access\n return pos\n return DecodeField\n\n\n# --------------------------------------------------------------------\n\n\nInt32Decoder = _SimpleDecoder(\n wire_format.WIRETYPE_VARINT, _DecodeSignedVarint32)\n\nInt64Decoder = _SimpleDecoder(\n wire_format.WIRETYPE_VARINT, _DecodeSignedVarint)\n\nUInt32Decoder = _SimpleDecoder(wire_format.WIRETYPE_VARINT, _DecodeVarint32)\nUInt64Decoder = _SimpleDecoder(wire_format.WIRETYPE_VARINT, _DecodeVarint)\n\nSInt32Decoder = _ModifiedDecoder(\n wire_format.WIRETYPE_VARINT, _DecodeVarint32, wire_format.ZigZagDecode)\nSInt64Decoder = _ModifiedDecoder(\n wire_format.WIRETYPE_VARINT, _DecodeVarint, wire_format.ZigZagDecode)\n\n# Note that Python conveniently guarantees that when using the '<' prefix on\n# formats, they will also have the same size across all platforms (as opposed\n# to without the prefix, where their sizes depend on the C compiler's basic\n# type sizes).\nFixed32Decoder = _StructPackDecoder(wire_format.WIRETYPE_FIXED32, ' end:\n raise _DecodeError('Truncated string.')\n value.append(_ConvertToUnicode(buffer[pos:new_pos]))\n # Predict that the next tag is another copy of the same repeated field.\n pos = new_pos + tag_len\n if buffer[new_pos:pos] != tag_bytes or new_pos == end:\n # Prediction failed. Return.\n return new_pos\n return DecodeRepeatedField\n else:\n def DecodeField(buffer, pos, end, message, field_dict):\n (size, pos) = local_DecodeVarint(buffer, pos)\n new_pos = pos + size\n if new_pos > end:\n raise _DecodeError('Truncated string.')\n if clear_if_default and not size:\n field_dict.pop(key, None)\n else:\n field_dict[key] = _ConvertToUnicode(buffer[pos:new_pos])\n return new_pos\n return DecodeField\n\n\ndef BytesDecoder(field_number, is_repeated, is_packed, key, new_default,\n clear_if_default=False):\n \"\"\"Returns a decoder for a bytes field.\"\"\"\n\n local_DecodeVarint = _DecodeVarint\n\n assert not is_packed\n if is_repeated:\n tag_bytes = encoder.TagBytes(field_number,\n wire_format.WIRETYPE_LENGTH_DELIMITED)\n tag_len = len(tag_bytes)\n def DecodeRepeatedField(buffer, pos, end, message, field_dict):\n value = field_dict.get(key)\n if value is None:\n value = field_dict.setdefault(key, new_default(message))\n while 1:\n (size, pos) = local_DecodeVarint(buffer, pos)\n new_pos = pos + size\n if new_pos > end:\n raise _DecodeError('Truncated string.')\n value.append(buffer[pos:new_pos].tobytes())\n # Predict that the next tag is another copy of the same repeated field.\n pos = new_pos + tag_len\n if buffer[new_pos:pos] != tag_bytes or new_pos == end:\n # Prediction failed. Return.\n return new_pos\n return DecodeRepeatedField\n else:\n def DecodeField(buffer, pos, end, message, field_dict):\n (size, pos) = local_DecodeVarint(buffer, pos)\n new_pos = pos + size\n if new_pos > end:\n raise _DecodeError('Truncated string.')\n if clear_if_default and not size:\n field_dict.pop(key, None)\n else:\n field_dict[key] = buffer[pos:new_pos].tobytes()\n return new_pos\n return DecodeField\n\n\ndef GroupDecoder(field_number, is_repeated, is_packed, key, new_default):\n \"\"\"Returns a decoder for a group field.\"\"\"\n\n end_tag_bytes = encoder.TagBytes(field_number,\n wire_format.WIRETYPE_END_GROUP)\n end_tag_len = len(end_tag_bytes)\n\n assert not is_packed\n if is_repeated:\n tag_bytes = encoder.TagBytes(field_number,\n wire_format.WIRETYPE_START_GROUP)\n tag_len = len(tag_bytes)\n def DecodeRepeatedField(buffer, pos, end, message, field_dict):\n value = field_dict.get(key)\n if value is None:\n value = field_dict.setdefault(key, new_default(message))\n while 1:\n value = field_dict.get(key)\n if value is None:\n value = field_dict.setdefault(key, new_default(message))\n # Read sub-message.\n pos = value.add()._InternalParse(buffer, pos, end)\n # Read end tag.\n new_pos = pos+end_tag_len\n if buffer[pos:new_pos] != end_tag_bytes or new_pos > end:\n raise _DecodeError('Missing group end tag.')\n # Predict that the next tag is another copy of the same repeated field.\n pos = new_pos + tag_len\n if buffer[new_pos:pos] != tag_bytes or new_pos == end:\n # Prediction failed. Return.\n return new_pos\n return DecodeRepeatedField\n else:\n def DecodeField(buffer, pos, end, message, field_dict):\n value = field_dict.get(key)\n if value is None:\n value = field_dict.setdefault(key, new_default(message))\n # Read sub-message.\n pos = value._InternalParse(buffer, pos, end)\n # Read end tag.\n new_pos = pos+end_tag_len\n if buffer[pos:new_pos] != end_tag_bytes or new_pos > end:\n raise _DecodeError('Missing group end tag.')\n return new_pos\n return DecodeField\n\n\ndef MessageDecoder(field_number, is_repeated, is_packed, key, new_default):\n \"\"\"Returns a decoder for a message field.\"\"\"\n\n local_DecodeVarint = _DecodeVarint\n\n assert not is_packed\n if is_repeated:\n tag_bytes = encoder.TagBytes(field_number,\n wire_format.WIRETYPE_LENGTH_DELIMITED)\n tag_len = len(tag_bytes)\n def DecodeRepeatedField(buffer, pos, end, message, field_dict):\n value = field_dict.get(key)\n if value is None:\n value = field_dict.setdefault(key, new_default(message))\n while 1:\n # Read length.\n (size, pos) = local_DecodeVarint(buffer, pos)\n new_pos = pos + size\n if new_pos > end:\n raise _DecodeError('Truncated message.')\n # Read sub-message.\n if value.add()._InternalParse(buffer, pos, new_pos) != new_pos:\n # The only reason _InternalParse would return early is if it\n # encountered an end-group tag.\n raise _DecodeError('Unexpected end-group tag.')\n # Predict that the next tag is another copy of the same repeated field.\n pos = new_pos + tag_len\n if buffer[new_pos:pos] != tag_bytes or new_pos == end:\n # Prediction failed. Return.\n return new_pos\n return DecodeRepeatedField\n else:\n def DecodeField(buffer, pos, end, message, field_dict):\n value = field_dict.get(key)\n if value is None:\n value = field_dict.setdefault(key, new_default(message))\n # Read length.\n (size, pos) = local_DecodeVarint(buffer, pos)\n new_pos = pos + size\n if new_pos > end:\n raise _DecodeError('Truncated message.')\n # Read sub-message.\n if value._InternalParse(buffer, pos, new_pos) != new_pos:\n # The only reason _InternalParse would return early is if it encountered\n # an end-group tag.\n raise _DecodeError('Unexpected end-group tag.')\n return new_pos\n return DecodeField\n\n\n# --------------------------------------------------------------------\n\nMESSAGE_SET_ITEM_TAG = encoder.TagBytes(1, wire_format.WIRETYPE_START_GROUP)\n\ndef MessageSetItemDecoder(descriptor):\n \"\"\"Returns a decoder for a MessageSet item.\n\n The parameter is the message Descriptor.\n\n The message set message looks like this:\n message MessageSet {\n repeated group Item = 1 {\n required int32 type_id = 2;\n required string message = 3;\n }\n }\n \"\"\"\n\n type_id_tag_bytes = encoder.TagBytes(2, wire_format.WIRETYPE_VARINT)\n message_tag_bytes = encoder.TagBytes(3, wire_format.WIRETYPE_LENGTH_DELIMITED)\n item_end_tag_bytes = encoder.TagBytes(1, wire_format.WIRETYPE_END_GROUP)\n\n local_ReadTag = ReadTag\n local_DecodeVarint = _DecodeVarint\n local_SkipField = SkipField\n\n def DecodeItem(buffer, pos, end, message, field_dict):\n \"\"\"Decode serialized message set to its value and new position.\n\n Args:\n buffer: memoryview of the serialized bytes.\n pos: int, position in the memory view to start at.\n end: int, end position of serialized data\n message: Message object to store unknown fields in\n field_dict: Map[Descriptor, Any] to store decoded values in.\n\n Returns:\n int, new position in serialized data.\n \"\"\"\n message_set_item_start = pos\n type_id = -1\n message_start = -1\n message_end = -1\n\n # Technically, type_id and message can appear in any order, so we need\n # a little loop here.\n while 1:\n (tag_bytes, pos) = local_ReadTag(buffer, pos)\n if tag_bytes == type_id_tag_bytes:\n (type_id, pos) = local_DecodeVarint(buffer, pos)\n elif tag_bytes == message_tag_bytes:\n (size, message_start) = local_DecodeVarint(buffer, pos)\n pos = message_end = message_start + size\n elif tag_bytes == item_end_tag_bytes:\n break\n else:\n pos = SkipField(buffer, pos, end, tag_bytes)\n if pos == -1:\n raise _DecodeError('Missing group end tag.')\n\n if pos > end:\n raise _DecodeError('Truncated message.')\n\n if type_id == -1:\n raise _DecodeError('MessageSet item missing type_id.')\n if message_start == -1:\n raise _DecodeError('MessageSet item missing message.')\n\n extension = message.Extensions._FindExtensionByNumber(type_id)\n # pylint: disable=protected-access\n if extension is not None:\n value = field_dict.get(extension)\n if value is None:\n message_type = extension.message_type\n if not hasattr(message_type, '_concrete_class'):\n # pylint: disable=protected-access\n message._FACTORY.GetPrototype(message_type)\n value = field_dict.setdefault(\n extension, message_type._concrete_class())\n if value._InternalParse(buffer, message_start,message_end) != message_end:\n # The only reason _InternalParse would return early is if it encountered\n # an end-group tag.\n raise _DecodeError('Unexpected end-group tag.')\n else:\n if not message._unknown_fields:\n message._unknown_fields = []\n message._unknown_fields.append(\n (MESSAGE_SET_ITEM_TAG, buffer[message_set_item_start:pos].tobytes()))\n if message._unknown_field_set is None:\n message._unknown_field_set = containers.UnknownFieldSet()\n message._unknown_field_set._add(\n type_id,\n wire_format.WIRETYPE_LENGTH_DELIMITED,\n buffer[message_start:message_end].tobytes())\n # pylint: enable=protected-access\n\n return pos\n\n return DecodeItem\n\n# --------------------------------------------------------------------\n\ndef MapDecoder(field_descriptor, new_default, is_message_map):\n \"\"\"Returns a decoder for a map field.\"\"\"\n\n key = field_descriptor\n tag_bytes = encoder.TagBytes(field_descriptor.number,\n wire_format.WIRETYPE_LENGTH_DELIMITED)\n tag_len = len(tag_bytes)\n local_DecodeVarint = _DecodeVarint\n # Can't read _concrete_class yet; might not be initialized.\n message_type = field_descriptor.message_type\n\n def DecodeMap(buffer, pos, end, message, field_dict):\n submsg = message_type._concrete_class()\n value = field_dict.get(key)\n if value is None:\n value = field_dict.setdefault(key, new_default(message))\n while 1:\n # Read length.\n (size, pos) = local_DecodeVarint(buffer, pos)\n new_pos = pos + size\n if new_pos > end:\n raise _DecodeError('Truncated message.')\n # Read sub-message.\n submsg.Clear()\n if submsg._InternalParse(buffer, pos, new_pos) != new_pos:\n # The only reason _InternalParse would return early is if it\n # encountered an end-group tag.\n raise _DecodeError('Unexpected end-group tag.')\n\n if is_message_map:\n value[submsg.key].CopyFrom(submsg.value)\n else:\n value[submsg.key] = submsg.value\n\n # Predict that the next tag is another copy of the same repeated field.\n pos = new_pos + tag_len\n if buffer[new_pos:pos] != tag_bytes or new_pos == end:\n # Prediction failed. Return.\n return new_pos\n\n return DecodeMap\n\n# --------------------------------------------------------------------\n# Optimization is not as heavy here because calls to SkipField() are rare,\n# except for handling end-group tags.\n\ndef _SkipVarint(buffer, pos, end):\n \"\"\"Skip a varint value. Returns the new position.\"\"\"\n # Previously ord(buffer[pos]) raised IndexError when pos is out of range.\n # With this code, ord(b'') raises TypeError. Both are handled in\n # python_message.py to generate a 'Truncated message' error.\n while ord(buffer[pos:pos+1].tobytes()) & 0x80:\n pos += 1\n pos += 1\n if pos > end:\n raise _DecodeError('Truncated message.')\n return pos\n\ndef _SkipFixed64(buffer, pos, end):\n \"\"\"Skip a fixed64 value. Returns the new position.\"\"\"\n\n pos += 8\n if pos > end:\n raise _DecodeError('Truncated message.')\n return pos\n\n\ndef _DecodeFixed64(buffer, pos):\n \"\"\"Decode a fixed64.\"\"\"\n new_pos = pos + 8\n return (struct.unpack(' end:\n raise _DecodeError('Truncated message.')\n return pos\n\n\ndef _SkipGroup(buffer, pos, end):\n \"\"\"Skip sub-group. Returns the new position.\"\"\"\n\n while 1:\n (tag_bytes, pos) = ReadTag(buffer, pos)\n new_pos = SkipField(buffer, pos, end, tag_bytes)\n if new_pos == -1:\n return pos\n pos = new_pos\n\n\ndef _DecodeUnknownFieldSet(buffer, pos, end_pos=None):\n \"\"\"Decode UnknownFieldSet. Returns the UnknownFieldSet and new position.\"\"\"\n\n unknown_field_set = containers.UnknownFieldSet()\n while end_pos is None or pos < end_pos:\n (tag_bytes, pos) = ReadTag(buffer, pos)\n (tag, _) = _DecodeVarint(tag_bytes, 0)\n field_number, wire_type = wire_format.UnpackTag(tag)\n if wire_type == wire_format.WIRETYPE_END_GROUP:\n break\n (data, pos) = _DecodeUnknownField(buffer, pos, wire_type)\n # pylint: disable=protected-access\n unknown_field_set._add(field_number, wire_type, data)\n\n return (unknown_field_set, pos)\n\n\ndef _DecodeUnknownField(buffer, pos, wire_type):\n \"\"\"Decode a unknown field. Returns the UnknownField and new position.\"\"\"\n\n if wire_type == wire_format.WIRETYPE_VARINT:\n (data, pos) = _DecodeVarint(buffer, pos)\n elif wire_type == wire_format.WIRETYPE_FIXED64:\n (data, pos) = _DecodeFixed64(buffer, pos)\n elif wire_type == wire_format.WIRETYPE_FIXED32:\n (data, pos) = _DecodeFixed32(buffer, pos)\n elif wire_type == wire_format.WIRETYPE_LENGTH_DELIMITED:\n (size, pos) = _DecodeVarint(buffer, pos)\n data = buffer[pos:pos+size].tobytes()\n pos += size\n elif wire_type == wire_format.WIRETYPE_START_GROUP:\n (data, pos) = _DecodeUnknownFieldSet(buffer, pos)\n elif wire_type == wire_format.WIRETYPE_END_GROUP:\n return (0, -1)\n else:\n raise _DecodeError('Wrong wire type in tag.')\n\n return (data, pos)\n\n\ndef _EndGroup(buffer, pos, end):\n \"\"\"Skipping an END_GROUP tag returns -1 to tell the parent loop to break.\"\"\"\n\n return -1\n\n\ndef _SkipFixed32(buffer, pos, end):\n \"\"\"Skip a fixed32 value. Returns the new position.\"\"\"\n\n pos += 4\n if pos > end:\n raise _DecodeError('Truncated message.')\n return pos\n\n\ndef _DecodeFixed32(buffer, pos):\n \"\"\"Decode a fixed32.\"\"\"\n\n new_pos = pos + 4\n return (struct.unpack('B').pack\n\n def EncodeVarint(write, value, unused_deterministic=None):\n bits = value & 0x7f\n value >>= 7\n while value:\n write(local_int2byte(0x80|bits))\n bits = value & 0x7f\n value >>= 7\n return write(local_int2byte(bits))\n\n return EncodeVarint\n\n\ndef _SignedVarintEncoder():\n \"\"\"Return an encoder for a basic signed varint value (does not include\n tag).\"\"\"\n\n local_int2byte = struct.Struct('>B').pack\n\n def EncodeSignedVarint(write, value, unused_deterministic=None):\n if value < 0:\n value += (1 << 64)\n bits = value & 0x7f\n value >>= 7\n while value:\n write(local_int2byte(0x80|bits))\n bits = value & 0x7f\n value >>= 7\n return write(local_int2byte(bits))\n\n return EncodeSignedVarint\n\n\n_EncodeVarint = _VarintEncoder()\n_EncodeSignedVarint = _SignedVarintEncoder()\n\n\ndef _VarintBytes(value):\n \"\"\"Encode the given integer as a varint and return the bytes. This is only\n called at startup time so it doesn't need to be fast.\"\"\"\n\n pieces = []\n _EncodeVarint(pieces.append, value, True)\n return b\"\".join(pieces)\n\n\ndef TagBytes(field_number, wire_type):\n \"\"\"Encode the given tag and return the bytes. Only called at startup.\"\"\"\n\n return bytes(_VarintBytes(wire_format.PackTag(field_number, wire_type)))\n\n# --------------------------------------------------------------------\n# As with sizers (see above), we have a number of common encoder\n# implementations.\n\n\ndef _SimpleEncoder(wire_type, encode_value, compute_value_size):\n \"\"\"Return a constructor for an encoder for fields of a particular type.\n\n Args:\n wire_type: The field's wire type, for encoding tags.\n encode_value: A function which encodes an individual value, e.g.\n _EncodeVarint().\n compute_value_size: A function which computes the size of an individual\n value, e.g. _VarintSize().\n \"\"\"\n\n def SpecificEncoder(field_number, is_repeated, is_packed):\n if is_packed:\n tag_bytes = TagBytes(field_number, wire_format.WIRETYPE_LENGTH_DELIMITED)\n local_EncodeVarint = _EncodeVarint\n def EncodePackedField(write, value, deterministic):\n write(tag_bytes)\n size = 0\n for element in value:\n size += compute_value_size(element)\n local_EncodeVarint(write, size, deterministic)\n for element in value:\n encode_value(write, element, deterministic)\n return EncodePackedField\n elif is_repeated:\n tag_bytes = TagBytes(field_number, wire_type)\n def EncodeRepeatedField(write, value, deterministic):\n for element in value:\n write(tag_bytes)\n encode_value(write, element, deterministic)\n return EncodeRepeatedField\n else:\n tag_bytes = TagBytes(field_number, wire_type)\n def EncodeField(write, value, deterministic):\n write(tag_bytes)\n return encode_value(write, value, deterministic)\n return EncodeField\n\n return SpecificEncoder\n\n\ndef _ModifiedEncoder(wire_type, encode_value, compute_value_size, modify_value):\n \"\"\"Like SimpleEncoder but additionally invokes modify_value on every value\n before passing it to encode_value. Usually modify_value is ZigZagEncode.\"\"\"\n\n def SpecificEncoder(field_number, is_repeated, is_packed):\n if is_packed:\n tag_bytes = TagBytes(field_number, wire_format.WIRETYPE_LENGTH_DELIMITED)\n local_EncodeVarint = _EncodeVarint\n def EncodePackedField(write, value, deterministic):\n write(tag_bytes)\n size = 0\n for element in value:\n size += compute_value_size(modify_value(element))\n local_EncodeVarint(write, size, deterministic)\n for element in value:\n encode_value(write, modify_value(element), deterministic)\n return EncodePackedField\n elif is_repeated:\n tag_bytes = TagBytes(field_number, wire_type)\n def EncodeRepeatedField(write, value, deterministic):\n for element in value:\n write(tag_bytes)\n encode_value(write, modify_value(element), deterministic)\n return EncodeRepeatedField\n else:\n tag_bytes = TagBytes(field_number, wire_type)\n def EncodeField(write, value, deterministic):\n write(tag_bytes)\n return encode_value(write, modify_value(value), deterministic)\n return EncodeField\n\n return SpecificEncoder\n\n\ndef _StructPackEncoder(wire_type, format):\n \"\"\"Return a constructor for an encoder for a fixed-width field.\n\n Args:\n wire_type: The field's wire type, for encoding tags.\n format: The format string to pass to struct.pack().\n \"\"\"\n\n value_size = struct.calcsize(format)\n\n def SpecificEncoder(field_number, is_repeated, is_packed):\n local_struct_pack = struct.pack\n if is_packed:\n tag_bytes = TagBytes(field_number, wire_format.WIRETYPE_LENGTH_DELIMITED)\n local_EncodeVarint = _EncodeVarint\n def EncodePackedField(write, value, deterministic):\n write(tag_bytes)\n local_EncodeVarint(write, len(value) * value_size, deterministic)\n for element in value:\n write(local_struct_pack(format, element))\n return EncodePackedField\n elif is_repeated:\n tag_bytes = TagBytes(field_number, wire_type)\n def EncodeRepeatedField(write, value, unused_deterministic=None):\n for element in value:\n write(tag_bytes)\n write(local_struct_pack(format, element))\n return EncodeRepeatedField\n else:\n tag_bytes = TagBytes(field_number, wire_type)\n def EncodeField(write, value, unused_deterministic=None):\n write(tag_bytes)\n return write(local_struct_pack(format, value))\n return EncodeField\n\n return SpecificEncoder\n\n\ndef _FloatingPointEncoder(wire_type, format):\n \"\"\"Return a constructor for an encoder for float fields.\n\n This is like StructPackEncoder, but catches errors that may be due to\n passing non-finite floating-point values to struct.pack, and makes a\n second attempt to encode those values.\n\n Args:\n wire_type: The field's wire type, for encoding tags.\n format: The format string to pass to struct.pack().\n \"\"\"\n\n value_size = struct.calcsize(format)\n if value_size == 4:\n def EncodeNonFiniteOrRaise(write, value):\n # Remember that the serialized form uses little-endian byte order.\n if value == _POS_INF:\n write(b'\\x00\\x00\\x80\\x7F')\n elif value == _NEG_INF:\n write(b'\\x00\\x00\\x80\\xFF')\n elif value != value: # NaN\n write(b'\\x00\\x00\\xC0\\x7F')\n else:\n raise\n elif value_size == 8:\n def EncodeNonFiniteOrRaise(write, value):\n if value == _POS_INF:\n write(b'\\x00\\x00\\x00\\x00\\x00\\x00\\xF0\\x7F')\n elif value == _NEG_INF:\n write(b'\\x00\\x00\\x00\\x00\\x00\\x00\\xF0\\xFF')\n elif value != value: # NaN\n write(b'\\x00\\x00\\x00\\x00\\x00\\x00\\xF8\\x7F')\n else:\n raise\n else:\n raise ValueError('Can\\'t encode floating-point values that are '\n '%d bytes long (only 4 or 8)' % value_size)\n\n def SpecificEncoder(field_number, is_repeated, is_packed):\n local_struct_pack = struct.pack\n if is_packed:\n tag_bytes = TagBytes(field_number, wire_format.WIRETYPE_LENGTH_DELIMITED)\n local_EncodeVarint = _EncodeVarint\n def EncodePackedField(write, value, deterministic):\n write(tag_bytes)\n local_EncodeVarint(write, len(value) * value_size, deterministic)\n for element in value:\n # This try/except block is going to be faster than any code that\n # we could write to check whether element is finite.\n try:\n write(local_struct_pack(format, element))\n except SystemError:\n EncodeNonFiniteOrRaise(write, element)\n return EncodePackedField\n elif is_repeated:\n tag_bytes = TagBytes(field_number, wire_type)\n def EncodeRepeatedField(write, value, unused_deterministic=None):\n for element in value:\n write(tag_bytes)\n try:\n write(local_struct_pack(format, element))\n except SystemError:\n EncodeNonFiniteOrRaise(write, element)\n return EncodeRepeatedField\n else:\n tag_bytes = TagBytes(field_number, wire_type)\n def EncodeField(write, value, unused_deterministic=None):\n write(tag_bytes)\n try:\n write(local_struct_pack(format, value))\n except SystemError:\n EncodeNonFiniteOrRaise(write, value)\n return EncodeField\n\n return SpecificEncoder\n\n\n# ====================================================================\n# Here we declare an encoder constructor for each field type. These work\n# very similarly to sizer constructors, described earlier.\n\n\nInt32Encoder = Int64Encoder = EnumEncoder = _SimpleEncoder(\n wire_format.WIRETYPE_VARINT, _EncodeSignedVarint, _SignedVarintSize)\n\nUInt32Encoder = UInt64Encoder = _SimpleEncoder(\n wire_format.WIRETYPE_VARINT, _EncodeVarint, _VarintSize)\n\nSInt32Encoder = SInt64Encoder = _ModifiedEncoder(\n wire_format.WIRETYPE_VARINT, _EncodeVarint, _VarintSize,\n wire_format.ZigZagEncode)\n\n# Note that Python conveniently guarantees that when using the '<' prefix on\n# formats, they will also have the same size across all platforms (as opposed\n# to without the prefix, where their sizes depend on the C compiler's basic\n# type sizes).\nFixed32Encoder = _StructPackEncoder(wire_format.WIRETYPE_FIXED32, ' str\n ValueType = int\n\n def __init__(self, enum_type):\n \"\"\"Inits EnumTypeWrapper with an EnumDescriptor.\"\"\"\n self._enum_type = enum_type\n self.DESCRIPTOR = enum_type # pylint: disable=invalid-name\n\n def Name(self, number): # pylint: disable=invalid-name\n \"\"\"Returns a string containing the name of an enum value.\"\"\"\n try:\n return self._enum_type.values_by_number[number].name\n except KeyError:\n pass # fall out to break exception chaining\n\n if not isinstance(number, int):\n raise TypeError(\n 'Enum value for {} must be an int, but got {} {!r}.'.format(\n self._enum_type.name, type(number), number))\n else:\n # repr here to handle the odd case when you pass in a boolean.\n raise ValueError('Enum {} has no name defined for value {!r}'.format(\n self._enum_type.name, number))\n\n def Value(self, name): # pylint: disable=invalid-name\n \"\"\"Returns the value corresponding to the given enum name.\"\"\"\n try:\n return self._enum_type.values_by_name[name].number\n except KeyError:\n pass # fall out to break exception chaining\n raise ValueError('Enum {} has no value defined for name {!r}'.format(\n self._enum_type.name, name))\n\n def keys(self):\n \"\"\"Return a list of the string names in the enum.\n\n Returns:\n A list of strs, in the order they were defined in the .proto file.\n \"\"\"\n\n return [value_descriptor.name\n for value_descriptor in self._enum_type.values]\n\n def values(self):\n \"\"\"Return a list of the integer values in the enum.\n\n Returns:\n A list of ints, in the order they were defined in the .proto file.\n \"\"\"\n\n return [value_descriptor.number\n for value_descriptor in self._enum_type.values]\n\n def items(self):\n \"\"\"Return a list of the (name, value) pairs of the enum.\n\n Returns:\n A list of (str, int) pairs, in the order they were defined\n in the .proto file.\n \"\"\"\n return [(value_descriptor.name, value_descriptor.number)\n for value_descriptor in self._enum_type.values]\n\n def __getattr__(self, name):\n \"\"\"Returns the value corresponding to the given enum name.\"\"\"\n try:\n return super(\n EnumTypeWrapper,\n self).__getattribute__('_enum_type').values_by_name[name].number\n except KeyError:\n pass # fall out to break exception chaining\n raise AttributeError('Enum {} has no value defined for name {!r}'.format(\n self._enum_type.name, name))\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/internal/extension_dict.py",
"content": "# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n\"\"\"Contains _ExtensionDict class to represent extensions.\n\"\"\"\n\nfrom google.protobuf.internal import type_checkers\nfrom google.protobuf.descriptor import FieldDescriptor\n\n\ndef _VerifyExtensionHandle(message, extension_handle):\n \"\"\"Verify that the given extension handle is valid.\"\"\"\n\n if not isinstance(extension_handle, FieldDescriptor):\n raise KeyError('HasExtension() expects an extension handle, got: %s' %\n extension_handle)\n\n if not extension_handle.is_extension:\n raise KeyError('\"%s\" is not an extension.' % extension_handle.full_name)\n\n if not extension_handle.containing_type:\n raise KeyError('\"%s\" is missing a containing_type.'\n % extension_handle.full_name)\n\n if extension_handle.containing_type is not message.DESCRIPTOR:\n raise KeyError('Extension \"%s\" extends message type \"%s\", but this '\n 'message is of type \"%s\".' %\n (extension_handle.full_name,\n extension_handle.containing_type.full_name,\n message.DESCRIPTOR.full_name))\n\n\n# TODO(robinson): Unify error handling of \"unknown extension\" crap.\n# TODO(robinson): Support iteritems()-style iteration over all\n# extensions with the \"has\" bits turned on?\nclass _ExtensionDict(object):\n\n \"\"\"Dict-like container for Extension fields on proto instances.\n\n Note that in all cases we expect extension handles to be\n FieldDescriptors.\n \"\"\"\n\n def __init__(self, extended_message):\n \"\"\"\n Args:\n extended_message: Message instance for which we are the Extensions dict.\n \"\"\"\n self._extended_message = extended_message\n\n def __getitem__(self, extension_handle):\n \"\"\"Returns the current value of the given extension handle.\"\"\"\n\n _VerifyExtensionHandle(self._extended_message, extension_handle)\n\n result = self._extended_message._fields.get(extension_handle)\n if result is not None:\n return result\n\n if extension_handle.label == FieldDescriptor.LABEL_REPEATED:\n result = extension_handle._default_constructor(self._extended_message)\n elif extension_handle.cpp_type == FieldDescriptor.CPPTYPE_MESSAGE:\n message_type = extension_handle.message_type\n if not hasattr(message_type, '_concrete_class'):\n # pylint: disable=protected-access\n self._extended_message._FACTORY.GetPrototype(message_type)\n assert getattr(extension_handle.message_type, '_concrete_class', None), (\n 'Uninitialized concrete class found for field %r (message type %r)'\n % (extension_handle.full_name,\n extension_handle.message_type.full_name))\n result = extension_handle.message_type._concrete_class()\n try:\n result._SetListener(self._extended_message._listener_for_children)\n except ReferenceError:\n pass\n else:\n # Singular scalar -- just return the default without inserting into the\n # dict.\n return extension_handle.default_value\n\n # Atomically check if another thread has preempted us and, if not, swap\n # in the new object we just created. If someone has preempted us, we\n # take that object and discard ours.\n # WARNING: We are relying on setdefault() being atomic. This is true\n # in CPython but we haven't investigated others. This warning appears\n # in several other locations in this file.\n result = self._extended_message._fields.setdefault(\n extension_handle, result)\n\n return result\n\n def __eq__(self, other):\n if not isinstance(other, self.__class__):\n return False\n\n my_fields = self._extended_message.ListFields()\n other_fields = other._extended_message.ListFields()\n\n # Get rid of non-extension fields.\n my_fields = [field for field in my_fields if field.is_extension]\n other_fields = [field for field in other_fields if field.is_extension]\n\n return my_fields == other_fields\n\n def __ne__(self, other):\n return not self == other\n\n def __len__(self):\n fields = self._extended_message.ListFields()\n # Get rid of non-extension fields.\n extension_fields = [field for field in fields if field[0].is_extension]\n return len(extension_fields)\n\n def __hash__(self):\n raise TypeError('unhashable object')\n\n # Note that this is only meaningful for non-repeated, scalar extension\n # fields. Note also that we may have to call _Modified() when we do\n # successfully set a field this way, to set any necessary \"has\" bits in the\n # ancestors of the extended message.\n def __setitem__(self, extension_handle, value):\n \"\"\"If extension_handle specifies a non-repeated, scalar extension\n field, sets the value of that field.\n \"\"\"\n\n _VerifyExtensionHandle(self._extended_message, extension_handle)\n\n if (extension_handle.label == FieldDescriptor.LABEL_REPEATED or\n extension_handle.cpp_type == FieldDescriptor.CPPTYPE_MESSAGE):\n raise TypeError(\n 'Cannot assign to extension \"%s\" because it is a repeated or '\n 'composite type.' % extension_handle.full_name)\n\n # It's slightly wasteful to lookup the type checker each time,\n # but we expect this to be a vanishingly uncommon case anyway.\n type_checker = type_checkers.GetTypeChecker(extension_handle)\n # pylint: disable=protected-access\n self._extended_message._fields[extension_handle] = (\n type_checker.CheckValue(value))\n self._extended_message._Modified()\n\n def __delitem__(self, extension_handle):\n self._extended_message.ClearExtension(extension_handle)\n\n def _FindExtensionByName(self, name):\n \"\"\"Tries to find a known extension with the specified name.\n\n Args:\n name: Extension full name.\n\n Returns:\n Extension field descriptor.\n \"\"\"\n return self._extended_message._extensions_by_name.get(name, None)\n\n def _FindExtensionByNumber(self, number):\n \"\"\"Tries to find a known extension with the field number.\n\n Args:\n number: Extension field number.\n\n Returns:\n Extension field descriptor.\n \"\"\"\n return self._extended_message._extensions_by_number.get(number, None)\n\n def __iter__(self):\n # Return a generator over the populated extension fields\n return (f[0] for f in self._extended_message.ListFields()\n if f[0].is_extension)\n\n def __contains__(self, extension_handle):\n _VerifyExtensionHandle(self._extended_message, extension_handle)\n\n if extension_handle not in self._extended_message._fields:\n return False\n\n if extension_handle.label == FieldDescriptor.LABEL_REPEATED:\n return bool(self._extended_message._fields.get(extension_handle))\n\n if extension_handle.cpp_type == FieldDescriptor.CPPTYPE_MESSAGE:\n value = self._extended_message._fields.get(extension_handle)\n # pylint: disable=protected-access\n return value is not None and value._is_present_in_parent\n\n return True\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/internal/message_listener.py",
"content": "# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n\"\"\"Defines a listener interface for observing certain\nstate transitions on Message objects.\n\nAlso defines a null implementation of this interface.\n\"\"\"\n\n__author__ = 'robinson@google.com (Will Robinson)'\n\n\nclass MessageListener(object):\n\n \"\"\"Listens for modifications made to a message. Meant to be registered via\n Message._SetListener().\n\n Attributes:\n dirty: If True, then calling Modified() would be a no-op. This can be\n used to avoid these calls entirely in the common case.\n \"\"\"\n\n def Modified(self):\n \"\"\"Called every time the message is modified in such a way that the parent\n message may need to be updated. This currently means either:\n (a) The message was modified for the first time, so the parent message\n should henceforth mark the message as present.\n (b) The message's cached byte size became dirty -- i.e. the message was\n modified for the first time after a previous call to ByteSize().\n Therefore the parent should also mark its byte size as dirty.\n Note that (a) implies (b), since new objects start out with a client cached\n size (zero). However, we document (a) explicitly because it is important.\n\n Modified() will *only* be called in response to one of these two events --\n not every time the sub-message is modified.\n\n Note that if the listener's |dirty| attribute is true, then calling\n Modified at the moment would be a no-op, so it can be skipped. Performance-\n sensitive callers should check this attribute directly before calling since\n it will be true most of the time.\n \"\"\"\n\n raise NotImplementedError\n\n\nclass NullMessageListener(object):\n\n \"\"\"No-op MessageListener implementation.\"\"\"\n\n def Modified(self):\n pass\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/internal/message_set_extensions_pb2.py",
"content": "# -*- coding: utf-8 -*-\n# Generated by the protocol buffer compiler. DO NOT EDIT!\n# source: google/protobuf/internal/message_set_extensions.proto\n\"\"\"Generated protocol buffer code.\"\"\"\nfrom google.protobuf.internal import builder as _builder\nfrom google.protobuf import descriptor as _descriptor\nfrom google.protobuf import descriptor_pool as _descriptor_pool\nfrom google.protobuf import symbol_database as _symbol_database\n# @@protoc_insertion_point(imports)\n\n_sym_db = _symbol_database.Default()\n\n\n\n\nDESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\\n5google/protobuf/internal/message_set_extensions.proto\\x12\\x18google.protobuf.internal\\\"\\x1e\\n\\x0eTestMessageSet*\\x08\\x08\\x04\\x10\\xff\\xff\\xff\\xff\\x07:\\x02\\x08\\x01\\\"\\xa5\\x01\\n\\x18TestMessageSetExtension1\\x12\\t\\n\\x01i\\x18\\x0f \\x01(\\x05\\x32~\\n\\x15message_set_extension\\x12(.google.protobuf.internal.TestMessageSet\\x18\\xab\\xff\\xf6. \\x01(\\x0b\\x32\\x32.google.protobuf.internal.TestMessageSetExtension1\\\"\\xa7\\x01\\n\\x18TestMessageSetExtension2\\x12\\x0b\\n\\x03str\\x18\\x19 \\x01(\\t2~\\n\\x15message_set_extension\\x12(.google.protobuf.internal.TestMessageSet\\x18\\xca\\xff\\xf6. \\x01(\\x0b\\x32\\x32.google.protobuf.internal.TestMessageSetExtension2\\\"(\\n\\x18TestMessageSetExtension3\\x12\\x0c\\n\\x04text\\x18# \\x01(\\t:\\x7f\\n\\x16message_set_extension3\\x12(.google.protobuf.internal.TestMessageSet\\x18\\xdf\\xff\\xf6. \\x01(\\x0b\\x32\\x32.google.protobuf.internal.TestMessageSetExtension3')\n\n_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())\n_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.internal.message_set_extensions_pb2', globals())\nif _descriptor._USE_C_DESCRIPTORS == False:\n TestMessageSet.RegisterExtension(message_set_extension3)\n TestMessageSet.RegisterExtension(_TESTMESSAGESETEXTENSION1.extensions_by_name['message_set_extension'])\n TestMessageSet.RegisterExtension(_TESTMESSAGESETEXTENSION2.extensions_by_name['message_set_extension'])\n\n DESCRIPTOR._options = None\n _TESTMESSAGESET._options = None\n _TESTMESSAGESET._serialized_options = b'\\010\\001'\n _TESTMESSAGESET._serialized_start=83\n _TESTMESSAGESET._serialized_end=113\n _TESTMESSAGESETEXTENSION1._serialized_start=116\n _TESTMESSAGESETEXTENSION1._serialized_end=281\n _TESTMESSAGESETEXTENSION2._serialized_start=284\n _TESTMESSAGESETEXTENSION2._serialized_end=451\n _TESTMESSAGESETEXTENSION3._serialized_start=453\n _TESTMESSAGESETEXTENSION3._serialized_end=493\n# @@protoc_insertion_point(module_scope)\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/internal/missing_enum_values_pb2.py",
"content": "# -*- coding: utf-8 -*-\n# Generated by the protocol buffer compiler. DO NOT EDIT!\n# source: google/protobuf/internal/missing_enum_values.proto\n\"\"\"Generated protocol buffer code.\"\"\"\nfrom google.protobuf.internal import builder as _builder\nfrom google.protobuf import descriptor as _descriptor\nfrom google.protobuf import descriptor_pool as _descriptor_pool\nfrom google.protobuf import symbol_database as _symbol_database\n# @@protoc_insertion_point(imports)\n\n_sym_db = _symbol_database.Default()\n\n\n\n\nDESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\\n2google/protobuf/internal/missing_enum_values.proto\\x12\\x1fgoogle.protobuf.python.internal\\\"\\xc1\\x02\\n\\x0eTestEnumValues\\x12X\\n\\x14optional_nested_enum\\x18\\x01 \\x01(\\x0e\\x32:.google.protobuf.python.internal.TestEnumValues.NestedEnum\\x12X\\n\\x14repeated_nested_enum\\x18\\x02 \\x03(\\x0e\\x32:.google.protobuf.python.internal.TestEnumValues.NestedEnum\\x12Z\\n\\x12packed_nested_enum\\x18\\x03 \\x03(\\x0e\\x32:.google.protobuf.python.internal.TestEnumValues.NestedEnumB\\x02\\x10\\x01\\\"\\x1f\\n\\nNestedEnum\\x12\\x08\\n\\x04ZERO\\x10\\x00\\x12\\x07\\n\\x03ONE\\x10\\x01\\\"\\xd3\\x02\\n\\x15TestMissingEnumValues\\x12_\\n\\x14optional_nested_enum\\x18\\x01 \\x01(\\x0e\\x32\\x41.google.protobuf.python.internal.TestMissingEnumValues.NestedEnum\\x12_\\n\\x14repeated_nested_enum\\x18\\x02 \\x03(\\x0e\\x32\\x41.google.protobuf.python.internal.TestMissingEnumValues.NestedEnum\\x12\\x61\\n\\x12packed_nested_enum\\x18\\x03 \\x03(\\x0e\\x32\\x41.google.protobuf.python.internal.TestMissingEnumValues.NestedEnumB\\x02\\x10\\x01\\\"\\x15\\n\\nNestedEnum\\x12\\x07\\n\\x03TWO\\x10\\x02\\\"\\x1b\\n\\nJustString\\x12\\r\\n\\x05\\x64ummy\\x18\\x01 \\x02(\\t')\n\n_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())\n_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.internal.missing_enum_values_pb2', globals())\nif _descriptor._USE_C_DESCRIPTORS == False:\n\n DESCRIPTOR._options = None\n _TESTENUMVALUES.fields_by_name['packed_nested_enum']._options = None\n _TESTENUMVALUES.fields_by_name['packed_nested_enum']._serialized_options = b'\\020\\001'\n _TESTMISSINGENUMVALUES.fields_by_name['packed_nested_enum']._options = None\n _TESTMISSINGENUMVALUES.fields_by_name['packed_nested_enum']._serialized_options = b'\\020\\001'\n _TESTENUMVALUES._serialized_start=88\n _TESTENUMVALUES._serialized_end=409\n _TESTENUMVALUES_NESTEDENUM._serialized_start=378\n _TESTENUMVALUES_NESTEDENUM._serialized_end=409\n _TESTMISSINGENUMVALUES._serialized_start=412\n _TESTMISSINGENUMVALUES._serialized_end=751\n _TESTMISSINGENUMVALUES_NESTEDENUM._serialized_start=730\n _TESTMISSINGENUMVALUES_NESTEDENUM._serialized_end=751\n _JUSTSTRING._serialized_start=753\n _JUSTSTRING._serialized_end=780\n# @@protoc_insertion_point(module_scope)\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/internal/more_extensions_dynamic_pb2.py",
"content": "# -*- coding: utf-8 -*-\n# Generated by the protocol buffer compiler. DO NOT EDIT!\n# source: google/protobuf/internal/more_extensions_dynamic.proto\n\"\"\"Generated protocol buffer code.\"\"\"\nfrom google.protobuf.internal import builder as _builder\nfrom google.protobuf import descriptor as _descriptor\nfrom google.protobuf import descriptor_pool as _descriptor_pool\nfrom google.protobuf import symbol_database as _symbol_database\n# @@protoc_insertion_point(imports)\n\n_sym_db = _symbol_database.Default()\n\n\nfrom google.protobuf.internal import more_extensions_pb2 as google_dot_protobuf_dot_internal_dot_more__extensions__pb2\n\n\nDESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\\n6google/protobuf/internal/more_extensions_dynamic.proto\\x12\\x18google.protobuf.internal\\x1a.google/protobuf/internal/more_extensions.proto\\\"\\x1f\\n\\x12\\x44ynamicMessageType\\x12\\t\\n\\x01\\x61\\x18\\x01 \\x01(\\x05:J\\n\\x17\\x64ynamic_int32_extension\\x12).google.protobuf.internal.ExtendedMessage\\x18\\x64 \\x01(\\x05:z\\n\\x19\\x64ynamic_message_extension\\x12).google.protobuf.internal.ExtendedMessage\\x18\\x65 \\x01(\\x0b\\x32,.google.protobuf.internal.DynamicMessageType:\\x83\\x01\\n\\\"repeated_dynamic_message_extension\\x12).google.protobuf.internal.ExtendedMessage\\x18\\x66 \\x03(\\x0b\\x32,.google.protobuf.internal.DynamicMessageType')\n\n_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())\n_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.internal.more_extensions_dynamic_pb2', globals())\nif _descriptor._USE_C_DESCRIPTORS == False:\n google_dot_protobuf_dot_internal_dot_more__extensions__pb2.ExtendedMessage.RegisterExtension(dynamic_int32_extension)\n google_dot_protobuf_dot_internal_dot_more__extensions__pb2.ExtendedMessage.RegisterExtension(dynamic_message_extension)\n google_dot_protobuf_dot_internal_dot_more__extensions__pb2.ExtendedMessage.RegisterExtension(repeated_dynamic_message_extension)\n\n DESCRIPTOR._options = None\n _DYNAMICMESSAGETYPE._serialized_start=132\n _DYNAMICMESSAGETYPE._serialized_end=163\n# @@protoc_insertion_point(module_scope)\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/internal/more_extensions_pb2.py",
"content": "# -*- coding: utf-8 -*-\n# Generated by the protocol buffer compiler. DO NOT EDIT!\n# source: google/protobuf/internal/more_extensions.proto\n\"\"\"Generated protocol buffer code.\"\"\"\nfrom google.protobuf.internal import builder as _builder\nfrom google.protobuf import descriptor as _descriptor\nfrom google.protobuf import descriptor_pool as _descriptor_pool\nfrom google.protobuf import symbol_database as _symbol_database\n# @@protoc_insertion_point(imports)\n\n_sym_db = _symbol_database.Default()\n\n\n\n\nDESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\\n.google/protobuf/internal/more_extensions.proto\\x12\\x18google.protobuf.internal\\\"\\x99\\x01\\n\\x0fTopLevelMessage\\x12\\x41\\n\\nsubmessage\\x18\\x01 \\x01(\\x0b\\x32).google.protobuf.internal.ExtendedMessageB\\x02(\\x01\\x12\\x43\\n\\x0enested_message\\x18\\x02 \\x01(\\x0b\\x32\\'.google.protobuf.internal.NestedMessageB\\x02(\\x01\\\"R\\n\\rNestedMessage\\x12\\x41\\n\\nsubmessage\\x18\\x01 \\x01(\\x0b\\x32).google.protobuf.internal.ExtendedMessageB\\x02(\\x01\\\"K\\n\\x0f\\x45xtendedMessage\\x12\\x17\\n\\x0eoptional_int32\\x18\\xe9\\x07 \\x01(\\x05\\x12\\x18\\n\\x0frepeated_string\\x18\\xea\\x07 \\x03(\\t*\\x05\\x08\\x01\\x10\\xe8\\x07\\\"-\\n\\x0e\\x46oreignMessage\\x12\\x1b\\n\\x13\\x66oreign_message_int\\x18\\x01 \\x01(\\x05:I\\n\\x16optional_int_extension\\x12).google.protobuf.internal.ExtendedMessage\\x18\\x01 \\x01(\\x05:w\\n\\x1aoptional_message_extension\\x12).google.protobuf.internal.ExtendedMessage\\x18\\x02 \\x01(\\x0b\\x32(.google.protobuf.internal.ForeignMessage:I\\n\\x16repeated_int_extension\\x12).google.protobuf.internal.ExtendedMessage\\x18\\x03 \\x03(\\x05:w\\n\\x1arepeated_message_extension\\x12).google.protobuf.internal.ExtendedMessage\\x18\\x04 \\x03(\\x0b\\x32(.google.protobuf.internal.ForeignMessage')\n\n_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())\n_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.internal.more_extensions_pb2', globals())\nif _descriptor._USE_C_DESCRIPTORS == False:\n ExtendedMessage.RegisterExtension(optional_int_extension)\n ExtendedMessage.RegisterExtension(optional_message_extension)\n ExtendedMessage.RegisterExtension(repeated_int_extension)\n ExtendedMessage.RegisterExtension(repeated_message_extension)\n\n DESCRIPTOR._options = None\n _TOPLEVELMESSAGE.fields_by_name['submessage']._options = None\n _TOPLEVELMESSAGE.fields_by_name['submessage']._serialized_options = b'(\\001'\n _TOPLEVELMESSAGE.fields_by_name['nested_message']._options = None\n _TOPLEVELMESSAGE.fields_by_name['nested_message']._serialized_options = b'(\\001'\n _NESTEDMESSAGE.fields_by_name['submessage']._options = None\n _NESTEDMESSAGE.fields_by_name['submessage']._serialized_options = b'(\\001'\n _TOPLEVELMESSAGE._serialized_start=77\n _TOPLEVELMESSAGE._serialized_end=230\n _NESTEDMESSAGE._serialized_start=232\n _NESTEDMESSAGE._serialized_end=314\n _EXTENDEDMESSAGE._serialized_start=316\n _EXTENDEDMESSAGE._serialized_end=391\n _FOREIGNMESSAGE._serialized_start=393\n _FOREIGNMESSAGE._serialized_end=438\n# @@protoc_insertion_point(module_scope)\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/internal/more_messages_pb2.py",
"content": "# -*- coding: utf-8 -*-\n# Generated by the protocol buffer compiler. DO NOT EDIT!\n# source: google/protobuf/internal/more_messages.proto\n\"\"\"Generated protocol buffer code.\"\"\"\nfrom google.protobuf.internal import builder as _builder\nfrom google.protobuf import descriptor as _descriptor\nfrom google.protobuf import descriptor_pool as _descriptor_pool\nfrom google.protobuf import symbol_database as _symbol_database\n# @@protoc_insertion_point(imports)\n\n_sym_db = _symbol_database.Default()\n\n\n\n\nDESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\\n,google/protobuf/internal/more_messages.proto\\x12\\x18google.protobuf.internal\\\"h\\n\\x10OutOfOrderFields\\x12\\x17\\n\\x0foptional_sint32\\x18\\x05 \\x01(\\x11\\x12\\x17\\n\\x0foptional_uint32\\x18\\x03 \\x01(\\r\\x12\\x16\\n\\x0eoptional_int32\\x18\\x01 \\x01(\\x05*\\x04\\x08\\x04\\x10\\x05*\\x04\\x08\\x02\\x10\\x03\\\"\\xcd\\x02\\n\\x05\\x63lass\\x12\\x1b\\n\\tint_field\\x18\\x01 \\x01(\\x05R\\x08json_int\\x12\\n\\n\\x02if\\x18\\x02 \\x01(\\x05\\x12(\\n\\x02\\x61s\\x18\\x03 \\x01(\\x0e\\x32\\x1c.google.protobuf.internal.is\\x12\\x30\\n\\nenum_field\\x18\\x04 \\x01(\\x0e\\x32\\x1c.google.protobuf.internal.is\\x12>\\n\\x11nested_enum_field\\x18\\x05 \\x01(\\x0e\\x32#.google.protobuf.internal.class.for\\x12;\\n\\x0enested_message\\x18\\x06 \\x01(\\x0b\\x32#.google.protobuf.internal.class.try\\x1a\\x1c\\n\\x03try\\x12\\r\\n\\x05\\x66ield\\x18\\x01 \\x01(\\x05*\\x06\\x08\\xe7\\x07\\x10\\x90N\\\"\\x1c\\n\\x03\\x66or\\x12\\x0b\\n\\x07\\x64\\x65\\x66\\x61ult\\x10\\x00\\x12\\x08\\n\\x04True\\x10\\x01*\\x06\\x08\\xe7\\x07\\x10\\x90N\\\"?\\n\\x0b\\x45xtendClass20\\n\\x06return\\x12\\x1f.google.protobuf.internal.class\\x18\\xea\\x07 \\x01(\\x05\\\"~\\n\\x0fTestFullKeyword\\x12:\\n\\x06\\x66ield1\\x18\\x01 \\x01(\\x0b\\x32*.google.protobuf.internal.OutOfOrderFields\\x12/\\n\\x06\\x66ield2\\x18\\x02 \\x01(\\x0b\\x32\\x1f.google.protobuf.internal.class\\\"\\xa5\\x0f\\n\\x11LotsNestedMessage\\x1a\\x04\\n\\x02\\x42\\x30\\x1a\\x04\\n\\x02\\x42\\x31\\x1a\\x04\\n\\x02\\x42\\x32\\x1a\\x04\\n\\x02\\x42\\x33\\x1a\\x04\\n\\x02\\x42\\x34\\x1a\\x04\\n\\x02\\x42\\x35\\x1a\\x04\\n\\x02\\x42\\x36\\x1a\\x04\\n\\x02\\x42\\x37\\x1a\\x04\\n\\x02\\x42\\x38\\x1a\\x04\\n\\x02\\x42\\x39\\x1a\\x05\\n\\x03\\x42\\x31\\x30\\x1a\\x05\\n\\x03\\x42\\x31\\x31\\x1a\\x05\\n\\x03\\x42\\x31\\x32\\x1a\\x05\\n\\x03\\x42\\x31\\x33\\x1a\\x05\\n\\x03\\x42\\x31\\x34\\x1a\\x05\\n\\x03\\x42\\x31\\x35\\x1a\\x05\\n\\x03\\x42\\x31\\x36\\x1a\\x05\\n\\x03\\x42\\x31\\x37\\x1a\\x05\\n\\x03\\x42\\x31\\x38\\x1a\\x05\\n\\x03\\x42\\x31\\x39\\x1a\\x05\\n\\x03\\x42\\x32\\x30\\x1a\\x05\\n\\x03\\x42\\x32\\x31\\x1a\\x05\\n\\x03\\x42\\x32\\x32\\x1a\\x05\\n\\x03\\x42\\x32\\x33\\x1a\\x05\\n\\x03\\x42\\x32\\x34\\x1a\\x05\\n\\x03\\x42\\x32\\x35\\x1a\\x05\\n\\x03\\x42\\x32\\x36\\x1a\\x05\\n\\x03\\x42\\x32\\x37\\x1a\\x05\\n\\x03\\x42\\x32\\x38\\x1a\\x05\\n\\x03\\x42\\x32\\x39\\x1a\\x05\\n\\x03\\x42\\x33\\x30\\x1a\\x05\\n\\x03\\x42\\x33\\x31\\x1a\\x05\\n\\x03\\x42\\x33\\x32\\x1a\\x05\\n\\x03\\x42\\x33\\x33\\x1a\\x05\\n\\x03\\x42\\x33\\x34\\x1a\\x05\\n\\x03\\x42\\x33\\x35\\x1a\\x05\\n\\x03\\x42\\x33\\x36\\x1a\\x05\\n\\x03\\x42\\x33\\x37\\x1a\\x05\\n\\x03\\x42\\x33\\x38\\x1a\\x05\\n\\x03\\x42\\x33\\x39\\x1a\\x05\\n\\x03\\x42\\x34\\x30\\x1a\\x05\\n\\x03\\x42\\x34\\x31\\x1a\\x05\\n\\x03\\x42\\x34\\x32\\x1a\\x05\\n\\x03\\x42\\x34\\x33\\x1a\\x05\\n\\x03\\x42\\x34\\x34\\x1a\\x05\\n\\x03\\x42\\x34\\x35\\x1a\\x05\\n\\x03\\x42\\x34\\x36\\x1a\\x05\\n\\x03\\x42\\x34\\x37\\x1a\\x05\\n\\x03\\x42\\x34\\x38\\x1a\\x05\\n\\x03\\x42\\x34\\x39\\x1a\\x05\\n\\x03\\x42\\x35\\x30\\x1a\\x05\\n\\x03\\x42\\x35\\x31\\x1a\\x05\\n\\x03\\x42\\x35\\x32\\x1a\\x05\\n\\x03\\x42\\x35\\x33\\x1a\\x05\\n\\x03\\x42\\x35\\x34\\x1a\\x05\\n\\x03\\x42\\x35\\x35\\x1a\\x05\\n\\x03\\x42\\x35\\x36\\x1a\\x05\\n\\x03\\x42\\x35\\x37\\x1a\\x05\\n\\x03\\x42\\x35\\x38\\x1a\\x05\\n\\x03\\x42\\x35\\x39\\x1a\\x05\\n\\x03\\x42\\x36\\x30\\x1a\\x05\\n\\x03\\x42\\x36\\x31\\x1a\\x05\\n\\x03\\x42\\x36\\x32\\x1a\\x05\\n\\x03\\x42\\x36\\x33\\x1a\\x05\\n\\x03\\x42\\x36\\x34\\x1a\\x05\\n\\x03\\x42\\x36\\x35\\x1a\\x05\\n\\x03\\x42\\x36\\x36\\x1a\\x05\\n\\x03\\x42\\x36\\x37\\x1a\\x05\\n\\x03\\x42\\x36\\x38\\x1a\\x05\\n\\x03\\x42\\x36\\x39\\x1a\\x05\\n\\x03\\x42\\x37\\x30\\x1a\\x05\\n\\x03\\x42\\x37\\x31\\x1a\\x05\\n\\x03\\x42\\x37\\x32\\x1a\\x05\\n\\x03\\x42\\x37\\x33\\x1a\\x05\\n\\x03\\x42\\x37\\x34\\x1a\\x05\\n\\x03\\x42\\x37\\x35\\x1a\\x05\\n\\x03\\x42\\x37\\x36\\x1a\\x05\\n\\x03\\x42\\x37\\x37\\x1a\\x05\\n\\x03\\x42\\x37\\x38\\x1a\\x05\\n\\x03\\x42\\x37\\x39\\x1a\\x05\\n\\x03\\x42\\x38\\x30\\x1a\\x05\\n\\x03\\x42\\x38\\x31\\x1a\\x05\\n\\x03\\x42\\x38\\x32\\x1a\\x05\\n\\x03\\x42\\x38\\x33\\x1a\\x05\\n\\x03\\x42\\x38\\x34\\x1a\\x05\\n\\x03\\x42\\x38\\x35\\x1a\\x05\\n\\x03\\x42\\x38\\x36\\x1a\\x05\\n\\x03\\x42\\x38\\x37\\x1a\\x05\\n\\x03\\x42\\x38\\x38\\x1a\\x05\\n\\x03\\x42\\x38\\x39\\x1a\\x05\\n\\x03\\x42\\x39\\x30\\x1a\\x05\\n\\x03\\x42\\x39\\x31\\x1a\\x05\\n\\x03\\x42\\x39\\x32\\x1a\\x05\\n\\x03\\x42\\x39\\x33\\x1a\\x05\\n\\x03\\x42\\x39\\x34\\x1a\\x05\\n\\x03\\x42\\x39\\x35\\x1a\\x05\\n\\x03\\x42\\x39\\x36\\x1a\\x05\\n\\x03\\x42\\x39\\x37\\x1a\\x05\\n\\x03\\x42\\x39\\x38\\x1a\\x05\\n\\x03\\x42\\x39\\x39\\x1a\\x06\\n\\x04\\x42\\x31\\x30\\x30\\x1a\\x06\\n\\x04\\x42\\x31\\x30\\x31\\x1a\\x06\\n\\x04\\x42\\x31\\x30\\x32\\x1a\\x06\\n\\x04\\x42\\x31\\x30\\x33\\x1a\\x06\\n\\x04\\x42\\x31\\x30\\x34\\x1a\\x06\\n\\x04\\x42\\x31\\x30\\x35\\x1a\\x06\\n\\x04\\x42\\x31\\x30\\x36\\x1a\\x06\\n\\x04\\x42\\x31\\x30\\x37\\x1a\\x06\\n\\x04\\x42\\x31\\x30\\x38\\x1a\\x06\\n\\x04\\x42\\x31\\x30\\x39\\x1a\\x06\\n\\x04\\x42\\x31\\x31\\x30\\x1a\\x06\\n\\x04\\x42\\x31\\x31\\x31\\x1a\\x06\\n\\x04\\x42\\x31\\x31\\x32\\x1a\\x06\\n\\x04\\x42\\x31\\x31\\x33\\x1a\\x06\\n\\x04\\x42\\x31\\x31\\x34\\x1a\\x06\\n\\x04\\x42\\x31\\x31\\x35\\x1a\\x06\\n\\x04\\x42\\x31\\x31\\x36\\x1a\\x06\\n\\x04\\x42\\x31\\x31\\x37\\x1a\\x06\\n\\x04\\x42\\x31\\x31\\x38\\x1a\\x06\\n\\x04\\x42\\x31\\x31\\x39\\x1a\\x06\\n\\x04\\x42\\x31\\x32\\x30\\x1a\\x06\\n\\x04\\x42\\x31\\x32\\x31\\x1a\\x06\\n\\x04\\x42\\x31\\x32\\x32\\x1a\\x06\\n\\x04\\x42\\x31\\x32\\x33\\x1a\\x06\\n\\x04\\x42\\x31\\x32\\x34\\x1a\\x06\\n\\x04\\x42\\x31\\x32\\x35\\x1a\\x06\\n\\x04\\x42\\x31\\x32\\x36\\x1a\\x06\\n\\x04\\x42\\x31\\x32\\x37\\x1a\\x06\\n\\x04\\x42\\x31\\x32\\x38\\x1a\\x06\\n\\x04\\x42\\x31\\x32\\x39\\x1a\\x06\\n\\x04\\x42\\x31\\x33\\x30\\x1a\\x06\\n\\x04\\x42\\x31\\x33\\x31\\x1a\\x06\\n\\x04\\x42\\x31\\x33\\x32\\x1a\\x06\\n\\x04\\x42\\x31\\x33\\x33\\x1a\\x06\\n\\x04\\x42\\x31\\x33\\x34\\x1a\\x06\\n\\x04\\x42\\x31\\x33\\x35\\x1a\\x06\\n\\x04\\x42\\x31\\x33\\x36\\x1a\\x06\\n\\x04\\x42\\x31\\x33\\x37\\x1a\\x06\\n\\x04\\x42\\x31\\x33\\x38\\x1a\\x06\\n\\x04\\x42\\x31\\x33\\x39\\x1a\\x06\\n\\x04\\x42\\x31\\x34\\x30\\x1a\\x06\\n\\x04\\x42\\x31\\x34\\x31\\x1a\\x06\\n\\x04\\x42\\x31\\x34\\x32\\x1a\\x06\\n\\x04\\x42\\x31\\x34\\x33\\x1a\\x06\\n\\x04\\x42\\x31\\x34\\x34\\x1a\\x06\\n\\x04\\x42\\x31\\x34\\x35\\x1a\\x06\\n\\x04\\x42\\x31\\x34\\x36\\x1a\\x06\\n\\x04\\x42\\x31\\x34\\x37\\x1a\\x06\\n\\x04\\x42\\x31\\x34\\x38\\x1a\\x06\\n\\x04\\x42\\x31\\x34\\x39\\x1a\\x06\\n\\x04\\x42\\x31\\x35\\x30\\x1a\\x06\\n\\x04\\x42\\x31\\x35\\x31\\x1a\\x06\\n\\x04\\x42\\x31\\x35\\x32\\x1a\\x06\\n\\x04\\x42\\x31\\x35\\x33\\x1a\\x06\\n\\x04\\x42\\x31\\x35\\x34\\x1a\\x06\\n\\x04\\x42\\x31\\x35\\x35\\x1a\\x06\\n\\x04\\x42\\x31\\x35\\x36\\x1a\\x06\\n\\x04\\x42\\x31\\x35\\x37\\x1a\\x06\\n\\x04\\x42\\x31\\x35\\x38\\x1a\\x06\\n\\x04\\x42\\x31\\x35\\x39\\x1a\\x06\\n\\x04\\x42\\x31\\x36\\x30\\x1a\\x06\\n\\x04\\x42\\x31\\x36\\x31\\x1a\\x06\\n\\x04\\x42\\x31\\x36\\x32\\x1a\\x06\\n\\x04\\x42\\x31\\x36\\x33\\x1a\\x06\\n\\x04\\x42\\x31\\x36\\x34\\x1a\\x06\\n\\x04\\x42\\x31\\x36\\x35\\x1a\\x06\\n\\x04\\x42\\x31\\x36\\x36\\x1a\\x06\\n\\x04\\x42\\x31\\x36\\x37\\x1a\\x06\\n\\x04\\x42\\x31\\x36\\x38\\x1a\\x06\\n\\x04\\x42\\x31\\x36\\x39\\x1a\\x06\\n\\x04\\x42\\x31\\x37\\x30\\x1a\\x06\\n\\x04\\x42\\x31\\x37\\x31\\x1a\\x06\\n\\x04\\x42\\x31\\x37\\x32\\x1a\\x06\\n\\x04\\x42\\x31\\x37\\x33\\x1a\\x06\\n\\x04\\x42\\x31\\x37\\x34\\x1a\\x06\\n\\x04\\x42\\x31\\x37\\x35\\x1a\\x06\\n\\x04\\x42\\x31\\x37\\x36\\x1a\\x06\\n\\x04\\x42\\x31\\x37\\x37\\x1a\\x06\\n\\x04\\x42\\x31\\x37\\x38\\x1a\\x06\\n\\x04\\x42\\x31\\x37\\x39\\x1a\\x06\\n\\x04\\x42\\x31\\x38\\x30\\x1a\\x06\\n\\x04\\x42\\x31\\x38\\x31\\x1a\\x06\\n\\x04\\x42\\x31\\x38\\x32\\x1a\\x06\\n\\x04\\x42\\x31\\x38\\x33\\x1a\\x06\\n\\x04\\x42\\x31\\x38\\x34\\x1a\\x06\\n\\x04\\x42\\x31\\x38\\x35\\x1a\\x06\\n\\x04\\x42\\x31\\x38\\x36\\x1a\\x06\\n\\x04\\x42\\x31\\x38\\x37\\x1a\\x06\\n\\x04\\x42\\x31\\x38\\x38\\x1a\\x06\\n\\x04\\x42\\x31\\x38\\x39\\x1a\\x06\\n\\x04\\x42\\x31\\x39\\x30\\x1a\\x06\\n\\x04\\x42\\x31\\x39\\x31\\x1a\\x06\\n\\x04\\x42\\x31\\x39\\x32\\x1a\\x06\\n\\x04\\x42\\x31\\x39\\x33\\x1a\\x06\\n\\x04\\x42\\x31\\x39\\x34\\x1a\\x06\\n\\x04\\x42\\x31\\x39\\x35\\x1a\\x06\\n\\x04\\x42\\x31\\x39\\x36\\x1a\\x06\\n\\x04\\x42\\x31\\x39\\x37\\x1a\\x06\\n\\x04\\x42\\x31\\x39\\x38\\x1a\\x06\\n\\x04\\x42\\x31\\x39\\x39\\x1a\\x06\\n\\x04\\x42\\x32\\x30\\x30\\x1a\\x06\\n\\x04\\x42\\x32\\x30\\x31\\x1a\\x06\\n\\x04\\x42\\x32\\x30\\x32\\x1a\\x06\\n\\x04\\x42\\x32\\x30\\x33\\x1a\\x06\\n\\x04\\x42\\x32\\x30\\x34\\x1a\\x06\\n\\x04\\x42\\x32\\x30\\x35\\x1a\\x06\\n\\x04\\x42\\x32\\x30\\x36\\x1a\\x06\\n\\x04\\x42\\x32\\x30\\x37\\x1a\\x06\\n\\x04\\x42\\x32\\x30\\x38\\x1a\\x06\\n\\x04\\x42\\x32\\x30\\x39\\x1a\\x06\\n\\x04\\x42\\x32\\x31\\x30\\x1a\\x06\\n\\x04\\x42\\x32\\x31\\x31\\x1a\\x06\\n\\x04\\x42\\x32\\x31\\x32\\x1a\\x06\\n\\x04\\x42\\x32\\x31\\x33\\x1a\\x06\\n\\x04\\x42\\x32\\x31\\x34\\x1a\\x06\\n\\x04\\x42\\x32\\x31\\x35\\x1a\\x06\\n\\x04\\x42\\x32\\x31\\x36\\x1a\\x06\\n\\x04\\x42\\x32\\x31\\x37\\x1a\\x06\\n\\x04\\x42\\x32\\x31\\x38\\x1a\\x06\\n\\x04\\x42\\x32\\x31\\x39\\x1a\\x06\\n\\x04\\x42\\x32\\x32\\x30\\x1a\\x06\\n\\x04\\x42\\x32\\x32\\x31\\x1a\\x06\\n\\x04\\x42\\x32\\x32\\x32\\x1a\\x06\\n\\x04\\x42\\x32\\x32\\x33\\x1a\\x06\\n\\x04\\x42\\x32\\x32\\x34\\x1a\\x06\\n\\x04\\x42\\x32\\x32\\x35\\x1a\\x06\\n\\x04\\x42\\x32\\x32\\x36\\x1a\\x06\\n\\x04\\x42\\x32\\x32\\x37\\x1a\\x06\\n\\x04\\x42\\x32\\x32\\x38\\x1a\\x06\\n\\x04\\x42\\x32\\x32\\x39\\x1a\\x06\\n\\x04\\x42\\x32\\x33\\x30\\x1a\\x06\\n\\x04\\x42\\x32\\x33\\x31\\x1a\\x06\\n\\x04\\x42\\x32\\x33\\x32\\x1a\\x06\\n\\x04\\x42\\x32\\x33\\x33\\x1a\\x06\\n\\x04\\x42\\x32\\x33\\x34\\x1a\\x06\\n\\x04\\x42\\x32\\x33\\x35\\x1a\\x06\\n\\x04\\x42\\x32\\x33\\x36\\x1a\\x06\\n\\x04\\x42\\x32\\x33\\x37\\x1a\\x06\\n\\x04\\x42\\x32\\x33\\x38\\x1a\\x06\\n\\x04\\x42\\x32\\x33\\x39\\x1a\\x06\\n\\x04\\x42\\x32\\x34\\x30\\x1a\\x06\\n\\x04\\x42\\x32\\x34\\x31\\x1a\\x06\\n\\x04\\x42\\x32\\x34\\x32\\x1a\\x06\\n\\x04\\x42\\x32\\x34\\x33\\x1a\\x06\\n\\x04\\x42\\x32\\x34\\x34\\x1a\\x06\\n\\x04\\x42\\x32\\x34\\x35\\x1a\\x06\\n\\x04\\x42\\x32\\x34\\x36\\x1a\\x06\\n\\x04\\x42\\x32\\x34\\x37\\x1a\\x06\\n\\x04\\x42\\x32\\x34\\x38\\x1a\\x06\\n\\x04\\x42\\x32\\x34\\x39\\x1a\\x06\\n\\x04\\x42\\x32\\x35\\x30\\x1a\\x06\\n\\x04\\x42\\x32\\x35\\x31\\x1a\\x06\\n\\x04\\x42\\x32\\x35\\x32\\x1a\\x06\\n\\x04\\x42\\x32\\x35\\x33\\x1a\\x06\\n\\x04\\x42\\x32\\x35\\x34\\x1a\\x06\\n\\x04\\x42\\x32\\x35\\x35*\\x1b\\n\\x02is\\x12\\x0b\\n\\x07\\x64\\x65\\x66\\x61ult\\x10\\x00\\x12\\x08\\n\\x04\\x65lse\\x10\\x01:C\\n\\x0foptional_uint64\\x12*.google.protobuf.internal.OutOfOrderFields\\x18\\x04 \\x01(\\x04:B\\n\\x0eoptional_int64\\x12*.google.protobuf.internal.OutOfOrderFields\\x18\\x02 \\x01(\\x03:2\\n\\x08\\x63ontinue\\x12\\x1f.google.protobuf.internal.class\\x18\\xe9\\x07 \\x01(\\x05:2\\n\\x04with\\x12#.google.protobuf.internal.class.try\\x18\\xe9\\x07 \\x01(\\x05')\n\n_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())\n_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.internal.more_messages_pb2', globals())\nif _descriptor._USE_C_DESCRIPTORS == False:\n OutOfOrderFields.RegisterExtension(optional_uint64)\n OutOfOrderFields.RegisterExtension(optional_int64)\n globals()['class'].RegisterExtension(globals()['continue'])\n getattr(globals()['class'], 'try').RegisterExtension(globals()['with'])\n globals()['class'].RegisterExtension(_EXTENDCLASS.extensions_by_name['return'])\n\n DESCRIPTOR._options = None\n _IS._serialized_start=2669\n _IS._serialized_end=2696\n _OUTOFORDERFIELDS._serialized_start=74\n _OUTOFORDERFIELDS._serialized_end=178\n _CLASS._serialized_start=181\n _CLASS._serialized_end=514\n _CLASS_TRY._serialized_start=448\n _CLASS_TRY._serialized_end=476\n _CLASS_FOR._serialized_start=478\n _CLASS_FOR._serialized_end=506\n _EXTENDCLASS._serialized_start=516\n _EXTENDCLASS._serialized_end=579\n _TESTFULLKEYWORD._serialized_start=581\n _TESTFULLKEYWORD._serialized_end=707\n _LOTSNESTEDMESSAGE._serialized_start=710\n _LOTSNESTEDMESSAGE._serialized_end=2667\n _LOTSNESTEDMESSAGE_B0._serialized_start=731\n _LOTSNESTEDMESSAGE_B0._serialized_end=735\n _LOTSNESTEDMESSAGE_B1._serialized_start=737\n _LOTSNESTEDMESSAGE_B1._serialized_end=741\n _LOTSNESTEDMESSAGE_B2._serialized_start=743\n _LOTSNESTEDMESSAGE_B2._serialized_end=747\n _LOTSNESTEDMESSAGE_B3._serialized_start=749\n _LOTSNESTEDMESSAGE_B3._serialized_end=753\n _LOTSNESTEDMESSAGE_B4._serialized_start=755\n _LOTSNESTEDMESSAGE_B4._serialized_end=759\n _LOTSNESTEDMESSAGE_B5._serialized_start=761\n _LOTSNESTEDMESSAGE_B5._serialized_end=765\n _LOTSNESTEDMESSAGE_B6._serialized_start=767\n _LOTSNESTEDMESSAGE_B6._serialized_end=771\n _LOTSNESTEDMESSAGE_B7._serialized_start=773\n _LOTSNESTEDMESSAGE_B7._serialized_end=777\n _LOTSNESTEDMESSAGE_B8._serialized_start=779\n _LOTSNESTEDMESSAGE_B8._serialized_end=783\n _LOTSNESTEDMESSAGE_B9._serialized_start=785\n _LOTSNESTEDMESSAGE_B9._serialized_end=789\n _LOTSNESTEDMESSAGE_B10._serialized_start=791\n _LOTSNESTEDMESSAGE_B10._serialized_end=796\n _LOTSNESTEDMESSAGE_B11._serialized_start=798\n _LOTSNESTEDMESSAGE_B11._serialized_end=803\n _LOTSNESTEDMESSAGE_B12._serialized_start=805\n _LOTSNESTEDMESSAGE_B12._serialized_end=810\n _LOTSNESTEDMESSAGE_B13._serialized_start=812\n _LOTSNESTEDMESSAGE_B13._serialized_end=817\n _LOTSNESTEDMESSAGE_B14._serialized_start=819\n _LOTSNESTEDMESSAGE_B14._serialized_end=824\n _LOTSNESTEDMESSAGE_B15._serialized_start=826\n _LOTSNESTEDMESSAGE_B15._serialized_end=831\n _LOTSNESTEDMESSAGE_B16._serialized_start=833\n _LOTSNESTEDMESSAGE_B16._serialized_end=838\n _LOTSNESTEDMESSAGE_B17._serialized_start=840\n _LOTSNESTEDMESSAGE_B17._serialized_end=845\n _LOTSNESTEDMESSAGE_B18._serialized_start=847\n _LOTSNESTEDMESSAGE_B18._serialized_end=852\n _LOTSNESTEDMESSAGE_B19._serialized_start=854\n _LOTSNESTEDMESSAGE_B19._serialized_end=859\n _LOTSNESTEDMESSAGE_B20._serialized_start=861\n _LOTSNESTEDMESSAGE_B20._serialized_end=866\n _LOTSNESTEDMESSAGE_B21._serialized_start=868\n _LOTSNESTEDMESSAGE_B21._serialized_end=873\n _LOTSNESTEDMESSAGE_B22._serialized_start=875\n _LOTSNESTEDMESSAGE_B22._serialized_end=880\n _LOTSNESTEDMESSAGE_B23._serialized_start=882\n _LOTSNESTEDMESSAGE_B23._serialized_end=887\n _LOTSNESTEDMESSAGE_B24._serialized_start=889\n _LOTSNESTEDMESSAGE_B24._serialized_end=894\n _LOTSNESTEDMESSAGE_B25._serialized_start=896\n _LOTSNESTEDMESSAGE_B25._serialized_end=901\n _LOTSNESTEDMESSAGE_B26._serialized_start=903\n _LOTSNESTEDMESSAGE_B26._serialized_end=908\n _LOTSNESTEDMESSAGE_B27._serialized_start=910\n _LOTSNESTEDMESSAGE_B27._serialized_end=915\n _LOTSNESTEDMESSAGE_B28._serialized_start=917\n _LOTSNESTEDMESSAGE_B28._serialized_end=922\n _LOTSNESTEDMESSAGE_B29._serialized_start=924\n _LOTSNESTEDMESSAGE_B29._serialized_end=929\n _LOTSNESTEDMESSAGE_B30._serialized_start=931\n _LOTSNESTEDMESSAGE_B30._serialized_end=936\n _LOTSNESTEDMESSAGE_B31._serialized_start=938\n _LOTSNESTEDMESSAGE_B31._serialized_end=943\n _LOTSNESTEDMESSAGE_B32._serialized_start=945\n _LOTSNESTEDMESSAGE_B32._serialized_end=950\n _LOTSNESTEDMESSAGE_B33._serialized_start=952\n _LOTSNESTEDMESSAGE_B33._serialized_end=957\n _LOTSNESTEDMESSAGE_B34._serialized_start=959\n _LOTSNESTEDMESSAGE_B34._serialized_end=964\n _LOTSNESTEDMESSAGE_B35._serialized_start=966\n _LOTSNESTEDMESSAGE_B35._serialized_end=971\n _LOTSNESTEDMESSAGE_B36._serialized_start=973\n _LOTSNESTEDMESSAGE_B36._serialized_end=978\n _LOTSNESTEDMESSAGE_B37._serialized_start=980\n _LOTSNESTEDMESSAGE_B37._serialized_end=985\n _LOTSNESTEDMESSAGE_B38._serialized_start=987\n _LOTSNESTEDMESSAGE_B38._serialized_end=992\n _LOTSNESTEDMESSAGE_B39._serialized_start=994\n _LOTSNESTEDMESSAGE_B39._serialized_end=999\n _LOTSNESTEDMESSAGE_B40._serialized_start=1001\n _LOTSNESTEDMESSAGE_B40._serialized_end=1006\n _LOTSNESTEDMESSAGE_B41._serialized_start=1008\n _LOTSNESTEDMESSAGE_B41._serialized_end=1013\n _LOTSNESTEDMESSAGE_B42._serialized_start=1015\n _LOTSNESTEDMESSAGE_B42._serialized_end=1020\n _LOTSNESTEDMESSAGE_B43._serialized_start=1022\n _LOTSNESTEDMESSAGE_B43._serialized_end=1027\n _LOTSNESTEDMESSAGE_B44._serialized_start=1029\n _LOTSNESTEDMESSAGE_B44._serialized_end=1034\n _LOTSNESTEDMESSAGE_B45._serialized_start=1036\n _LOTSNESTEDMESSAGE_B45._serialized_end=1041\n _LOTSNESTEDMESSAGE_B46._serialized_start=1043\n _LOTSNESTEDMESSAGE_B46._serialized_end=1048\n _LOTSNESTEDMESSAGE_B47._serialized_start=1050\n _LOTSNESTEDMESSAGE_B47._serialized_end=1055\n _LOTSNESTEDMESSAGE_B48._serialized_start=1057\n _LOTSNESTEDMESSAGE_B48._serialized_end=1062\n _LOTSNESTEDMESSAGE_B49._serialized_start=1064\n _LOTSNESTEDMESSAGE_B49._serialized_end=1069\n _LOTSNESTEDMESSAGE_B50._serialized_start=1071\n _LOTSNESTEDMESSAGE_B50._serialized_end=1076\n _LOTSNESTEDMESSAGE_B51._serialized_start=1078\n _LOTSNESTEDMESSAGE_B51._serialized_end=1083\n _LOTSNESTEDMESSAGE_B52._serialized_start=1085\n _LOTSNESTEDMESSAGE_B52._serialized_end=1090\n _LOTSNESTEDMESSAGE_B53._serialized_start=1092\n _LOTSNESTEDMESSAGE_B53._serialized_end=1097\n _LOTSNESTEDMESSAGE_B54._serialized_start=1099\n _LOTSNESTEDMESSAGE_B54._serialized_end=1104\n _LOTSNESTEDMESSAGE_B55._serialized_start=1106\n _LOTSNESTEDMESSAGE_B55._serialized_end=1111\n _LOTSNESTEDMESSAGE_B56._serialized_start=1113\n _LOTSNESTEDMESSAGE_B56._serialized_end=1118\n _LOTSNESTEDMESSAGE_B57._serialized_start=1120\n _LOTSNESTEDMESSAGE_B57._serialized_end=1125\n _LOTSNESTEDMESSAGE_B58._serialized_start=1127\n _LOTSNESTEDMESSAGE_B58._serialized_end=1132\n _LOTSNESTEDMESSAGE_B59._serialized_start=1134\n _LOTSNESTEDMESSAGE_B59._serialized_end=1139\n _LOTSNESTEDMESSAGE_B60._serialized_start=1141\n _LOTSNESTEDMESSAGE_B60._serialized_end=1146\n _LOTSNESTEDMESSAGE_B61._serialized_start=1148\n _LOTSNESTEDMESSAGE_B61._serialized_end=1153\n _LOTSNESTEDMESSAGE_B62._serialized_start=1155\n _LOTSNESTEDMESSAGE_B62._serialized_end=1160\n _LOTSNESTEDMESSAGE_B63._serialized_start=1162\n _LOTSNESTEDMESSAGE_B63._serialized_end=1167\n _LOTSNESTEDMESSAGE_B64._serialized_start=1169\n _LOTSNESTEDMESSAGE_B64._serialized_end=1174\n _LOTSNESTEDMESSAGE_B65._serialized_start=1176\n _LOTSNESTEDMESSAGE_B65._serialized_end=1181\n _LOTSNESTEDMESSAGE_B66._serialized_start=1183\n _LOTSNESTEDMESSAGE_B66._serialized_end=1188\n _LOTSNESTEDMESSAGE_B67._serialized_start=1190\n _LOTSNESTEDMESSAGE_B67._serialized_end=1195\n _LOTSNESTEDMESSAGE_B68._serialized_start=1197\n _LOTSNESTEDMESSAGE_B68._serialized_end=1202\n _LOTSNESTEDMESSAGE_B69._serialized_start=1204\n _LOTSNESTEDMESSAGE_B69._serialized_end=1209\n _LOTSNESTEDMESSAGE_B70._serialized_start=1211\n _LOTSNESTEDMESSAGE_B70._serialized_end=1216\n _LOTSNESTEDMESSAGE_B71._serialized_start=1218\n _LOTSNESTEDMESSAGE_B71._serialized_end=1223\n _LOTSNESTEDMESSAGE_B72._serialized_start=1225\n _LOTSNESTEDMESSAGE_B72._serialized_end=1230\n _LOTSNESTEDMESSAGE_B73._serialized_start=1232\n _LOTSNESTEDMESSAGE_B73._serialized_end=1237\n _LOTSNESTEDMESSAGE_B74._serialized_start=1239\n _LOTSNESTEDMESSAGE_B74._serialized_end=1244\n _LOTSNESTEDMESSAGE_B75._serialized_start=1246\n _LOTSNESTEDMESSAGE_B75._serialized_end=1251\n _LOTSNESTEDMESSAGE_B76._serialized_start=1253\n _LOTSNESTEDMESSAGE_B76._serialized_end=1258\n _LOTSNESTEDMESSAGE_B77._serialized_start=1260\n _LOTSNESTEDMESSAGE_B77._serialized_end=1265\n _LOTSNESTEDMESSAGE_B78._serialized_start=1267\n _LOTSNESTEDMESSAGE_B78._serialized_end=1272\n _LOTSNESTEDMESSAGE_B79._serialized_start=1274\n _LOTSNESTEDMESSAGE_B79._serialized_end=1279\n _LOTSNESTEDMESSAGE_B80._serialized_start=1281\n _LOTSNESTEDMESSAGE_B80._serialized_end=1286\n _LOTSNESTEDMESSAGE_B81._serialized_start=1288\n _LOTSNESTEDMESSAGE_B81._serialized_end=1293\n _LOTSNESTEDMESSAGE_B82._serialized_start=1295\n _LOTSNESTEDMESSAGE_B82._serialized_end=1300\n _LOTSNESTEDMESSAGE_B83._serialized_start=1302\n _LOTSNESTEDMESSAGE_B83._serialized_end=1307\n _LOTSNESTEDMESSAGE_B84._serialized_start=1309\n _LOTSNESTEDMESSAGE_B84._serialized_end=1314\n _LOTSNESTEDMESSAGE_B85._serialized_start=1316\n _LOTSNESTEDMESSAGE_B85._serialized_end=1321\n _LOTSNESTEDMESSAGE_B86._serialized_start=1323\n _LOTSNESTEDMESSAGE_B86._serialized_end=1328\n _LOTSNESTEDMESSAGE_B87._serialized_start=1330\n _LOTSNESTEDMESSAGE_B87._serialized_end=1335\n _LOTSNESTEDMESSAGE_B88._serialized_start=1337\n _LOTSNESTEDMESSAGE_B88._serialized_end=1342\n _LOTSNESTEDMESSAGE_B89._serialized_start=1344\n _LOTSNESTEDMESSAGE_B89._serialized_end=1349\n _LOTSNESTEDMESSAGE_B90._serialized_start=1351\n _LOTSNESTEDMESSAGE_B90._serialized_end=1356\n _LOTSNESTEDMESSAGE_B91._serialized_start=1358\n _LOTSNESTEDMESSAGE_B91._serialized_end=1363\n _LOTSNESTEDMESSAGE_B92._serialized_start=1365\n _LOTSNESTEDMESSAGE_B92._serialized_end=1370\n _LOTSNESTEDMESSAGE_B93._serialized_start=1372\n _LOTSNESTEDMESSAGE_B93._serialized_end=1377\n _LOTSNESTEDMESSAGE_B94._serialized_start=1379\n _LOTSNESTEDMESSAGE_B94._serialized_end=1384\n _LOTSNESTEDMESSAGE_B95._serialized_start=1386\n _LOTSNESTEDMESSAGE_B95._serialized_end=1391\n _LOTSNESTEDMESSAGE_B96._serialized_start=1393\n _LOTSNESTEDMESSAGE_B96._serialized_end=1398\n _LOTSNESTEDMESSAGE_B97._serialized_start=1400\n _LOTSNESTEDMESSAGE_B97._serialized_end=1405\n _LOTSNESTEDMESSAGE_B98._serialized_start=1407\n _LOTSNESTEDMESSAGE_B98._serialized_end=1412\n _LOTSNESTEDMESSAGE_B99._serialized_start=1414\n _LOTSNESTEDMESSAGE_B99._serialized_end=1419\n _LOTSNESTEDMESSAGE_B100._serialized_start=1421\n _LOTSNESTEDMESSAGE_B100._serialized_end=1427\n _LOTSNESTEDMESSAGE_B101._serialized_start=1429\n _LOTSNESTEDMESSAGE_B101._serialized_end=1435\n _LOTSNESTEDMESSAGE_B102._serialized_start=1437\n _LOTSNESTEDMESSAGE_B102._serialized_end=1443\n _LOTSNESTEDMESSAGE_B103._serialized_start=1445\n _LOTSNESTEDMESSAGE_B103._serialized_end=1451\n _LOTSNESTEDMESSAGE_B104._serialized_start=1453\n _LOTSNESTEDMESSAGE_B104._serialized_end=1459\n _LOTSNESTEDMESSAGE_B105._serialized_start=1461\n _LOTSNESTEDMESSAGE_B105._serialized_end=1467\n _LOTSNESTEDMESSAGE_B106._serialized_start=1469\n _LOTSNESTEDMESSAGE_B106._serialized_end=1475\n _LOTSNESTEDMESSAGE_B107._serialized_start=1477\n _LOTSNESTEDMESSAGE_B107._serialized_end=1483\n _LOTSNESTEDMESSAGE_B108._serialized_start=1485\n _LOTSNESTEDMESSAGE_B108._serialized_end=1491\n _LOTSNESTEDMESSAGE_B109._serialized_start=1493\n _LOTSNESTEDMESSAGE_B109._serialized_end=1499\n _LOTSNESTEDMESSAGE_B110._serialized_start=1501\n _LOTSNESTEDMESSAGE_B110._serialized_end=1507\n _LOTSNESTEDMESSAGE_B111._serialized_start=1509\n _LOTSNESTEDMESSAGE_B111._serialized_end=1515\n _LOTSNESTEDMESSAGE_B112._serialized_start=1517\n _LOTSNESTEDMESSAGE_B112._serialized_end=1523\n _LOTSNESTEDMESSAGE_B113._serialized_start=1525\n _LOTSNESTEDMESSAGE_B113._serialized_end=1531\n _LOTSNESTEDMESSAGE_B114._serialized_start=1533\n _LOTSNESTEDMESSAGE_B114._serialized_end=1539\n _LOTSNESTEDMESSAGE_B115._serialized_start=1541\n _LOTSNESTEDMESSAGE_B115._serialized_end=1547\n _LOTSNESTEDMESSAGE_B116._serialized_start=1549\n _LOTSNESTEDMESSAGE_B116._serialized_end=1555\n _LOTSNESTEDMESSAGE_B117._serialized_start=1557\n _LOTSNESTEDMESSAGE_B117._serialized_end=1563\n _LOTSNESTEDMESSAGE_B118._serialized_start=1565\n _LOTSNESTEDMESSAGE_B118._serialized_end=1571\n _LOTSNESTEDMESSAGE_B119._serialized_start=1573\n _LOTSNESTEDMESSAGE_B119._serialized_end=1579\n _LOTSNESTEDMESSAGE_B120._serialized_start=1581\n _LOTSNESTEDMESSAGE_B120._serialized_end=1587\n _LOTSNESTEDMESSAGE_B121._serialized_start=1589\n _LOTSNESTEDMESSAGE_B121._serialized_end=1595\n _LOTSNESTEDMESSAGE_B122._serialized_start=1597\n _LOTSNESTEDMESSAGE_B122._serialized_end=1603\n _LOTSNESTEDMESSAGE_B123._serialized_start=1605\n _LOTSNESTEDMESSAGE_B123._serialized_end=1611\n _LOTSNESTEDMESSAGE_B124._serialized_start=1613\n _LOTSNESTEDMESSAGE_B124._serialized_end=1619\n _LOTSNESTEDMESSAGE_B125._serialized_start=1621\n _LOTSNESTEDMESSAGE_B125._serialized_end=1627\n _LOTSNESTEDMESSAGE_B126._serialized_start=1629\n _LOTSNESTEDMESSAGE_B126._serialized_end=1635\n _LOTSNESTEDMESSAGE_B127._serialized_start=1637\n _LOTSNESTEDMESSAGE_B127._serialized_end=1643\n _LOTSNESTEDMESSAGE_B128._serialized_start=1645\n _LOTSNESTEDMESSAGE_B128._serialized_end=1651\n _LOTSNESTEDMESSAGE_B129._serialized_start=1653\n _LOTSNESTEDMESSAGE_B129._serialized_end=1659\n _LOTSNESTEDMESSAGE_B130._serialized_start=1661\n _LOTSNESTEDMESSAGE_B130._serialized_end=1667\n _LOTSNESTEDMESSAGE_B131._serialized_start=1669\n _LOTSNESTEDMESSAGE_B131._serialized_end=1675\n _LOTSNESTEDMESSAGE_B132._serialized_start=1677\n _LOTSNESTEDMESSAGE_B132._serialized_end=1683\n _LOTSNESTEDMESSAGE_B133._serialized_start=1685\n _LOTSNESTEDMESSAGE_B133._serialized_end=1691\n _LOTSNESTEDMESSAGE_B134._serialized_start=1693\n _LOTSNESTEDMESSAGE_B134._serialized_end=1699\n _LOTSNESTEDMESSAGE_B135._serialized_start=1701\n _LOTSNESTEDMESSAGE_B135._serialized_end=1707\n _LOTSNESTEDMESSAGE_B136._serialized_start=1709\n _LOTSNESTEDMESSAGE_B136._serialized_end=1715\n _LOTSNESTEDMESSAGE_B137._serialized_start=1717\n _LOTSNESTEDMESSAGE_B137._serialized_end=1723\n _LOTSNESTEDMESSAGE_B138._serialized_start=1725\n _LOTSNESTEDMESSAGE_B138._serialized_end=1731\n _LOTSNESTEDMESSAGE_B139._serialized_start=1733\n _LOTSNESTEDMESSAGE_B139._serialized_end=1739\n _LOTSNESTEDMESSAGE_B140._serialized_start=1741\n _LOTSNESTEDMESSAGE_B140._serialized_end=1747\n _LOTSNESTEDMESSAGE_B141._serialized_start=1749\n _LOTSNESTEDMESSAGE_B141._serialized_end=1755\n _LOTSNESTEDMESSAGE_B142._serialized_start=1757\n _LOTSNESTEDMESSAGE_B142._serialized_end=1763\n _LOTSNESTEDMESSAGE_B143._serialized_start=1765\n _LOTSNESTEDMESSAGE_B143._serialized_end=1771\n _LOTSNESTEDMESSAGE_B144._serialized_start=1773\n _LOTSNESTEDMESSAGE_B144._serialized_end=1779\n _LOTSNESTEDMESSAGE_B145._serialized_start=1781\n _LOTSNESTEDMESSAGE_B145._serialized_end=1787\n _LOTSNESTEDMESSAGE_B146._serialized_start=1789\n _LOTSNESTEDMESSAGE_B146._serialized_end=1795\n _LOTSNESTEDMESSAGE_B147._serialized_start=1797\n _LOTSNESTEDMESSAGE_B147._serialized_end=1803\n _LOTSNESTEDMESSAGE_B148._serialized_start=1805\n _LOTSNESTEDMESSAGE_B148._serialized_end=1811\n _LOTSNESTEDMESSAGE_B149._serialized_start=1813\n _LOTSNESTEDMESSAGE_B149._serialized_end=1819\n _LOTSNESTEDMESSAGE_B150._serialized_start=1821\n _LOTSNESTEDMESSAGE_B150._serialized_end=1827\n _LOTSNESTEDMESSAGE_B151._serialized_start=1829\n _LOTSNESTEDMESSAGE_B151._serialized_end=1835\n _LOTSNESTEDMESSAGE_B152._serialized_start=1837\n _LOTSNESTEDMESSAGE_B152._serialized_end=1843\n _LOTSNESTEDMESSAGE_B153._serialized_start=1845\n _LOTSNESTEDMESSAGE_B153._serialized_end=1851\n _LOTSNESTEDMESSAGE_B154._serialized_start=1853\n _LOTSNESTEDMESSAGE_B154._serialized_end=1859\n _LOTSNESTEDMESSAGE_B155._serialized_start=1861\n _LOTSNESTEDMESSAGE_B155._serialized_end=1867\n _LOTSNESTEDMESSAGE_B156._serialized_start=1869\n _LOTSNESTEDMESSAGE_B156._serialized_end=1875\n _LOTSNESTEDMESSAGE_B157._serialized_start=1877\n _LOTSNESTEDMESSAGE_B157._serialized_end=1883\n _LOTSNESTEDMESSAGE_B158._serialized_start=1885\n _LOTSNESTEDMESSAGE_B158._serialized_end=1891\n _LOTSNESTEDMESSAGE_B159._serialized_start=1893\n _LOTSNESTEDMESSAGE_B159._serialized_end=1899\n _LOTSNESTEDMESSAGE_B160._serialized_start=1901\n _LOTSNESTEDMESSAGE_B160._serialized_end=1907\n _LOTSNESTEDMESSAGE_B161._serialized_start=1909\n _LOTSNESTEDMESSAGE_B161._serialized_end=1915\n _LOTSNESTEDMESSAGE_B162._serialized_start=1917\n _LOTSNESTEDMESSAGE_B162._serialized_end=1923\n _LOTSNESTEDMESSAGE_B163._serialized_start=1925\n _LOTSNESTEDMESSAGE_B163._serialized_end=1931\n _LOTSNESTEDMESSAGE_B164._serialized_start=1933\n _LOTSNESTEDMESSAGE_B164._serialized_end=1939\n _LOTSNESTEDMESSAGE_B165._serialized_start=1941\n _LOTSNESTEDMESSAGE_B165._serialized_end=1947\n _LOTSNESTEDMESSAGE_B166._serialized_start=1949\n _LOTSNESTEDMESSAGE_B166._serialized_end=1955\n _LOTSNESTEDMESSAGE_B167._serialized_start=1957\n _LOTSNESTEDMESSAGE_B167._serialized_end=1963\n _LOTSNESTEDMESSAGE_B168._serialized_start=1965\n _LOTSNESTEDMESSAGE_B168._serialized_end=1971\n _LOTSNESTEDMESSAGE_B169._serialized_start=1973\n _LOTSNESTEDMESSAGE_B169._serialized_end=1979\n _LOTSNESTEDMESSAGE_B170._serialized_start=1981\n _LOTSNESTEDMESSAGE_B170._serialized_end=1987\n _LOTSNESTEDMESSAGE_B171._serialized_start=1989\n _LOTSNESTEDMESSAGE_B171._serialized_end=1995\n _LOTSNESTEDMESSAGE_B172._serialized_start=1997\n _LOTSNESTEDMESSAGE_B172._serialized_end=2003\n _LOTSNESTEDMESSAGE_B173._serialized_start=2005\n _LOTSNESTEDMESSAGE_B173._serialized_end=2011\n _LOTSNESTEDMESSAGE_B174._serialized_start=2013\n _LOTSNESTEDMESSAGE_B174._serialized_end=2019\n _LOTSNESTEDMESSAGE_B175._serialized_start=2021\n _LOTSNESTEDMESSAGE_B175._serialized_end=2027\n _LOTSNESTEDMESSAGE_B176._serialized_start=2029\n _LOTSNESTEDMESSAGE_B176._serialized_end=2035\n _LOTSNESTEDMESSAGE_B177._serialized_start=2037\n _LOTSNESTEDMESSAGE_B177._serialized_end=2043\n _LOTSNESTEDMESSAGE_B178._serialized_start=2045\n _LOTSNESTEDMESSAGE_B178._serialized_end=2051\n _LOTSNESTEDMESSAGE_B179._serialized_start=2053\n _LOTSNESTEDMESSAGE_B179._serialized_end=2059\n _LOTSNESTEDMESSAGE_B180._serialized_start=2061\n _LOTSNESTEDMESSAGE_B180._serialized_end=2067\n _LOTSNESTEDMESSAGE_B181._serialized_start=2069\n _LOTSNESTEDMESSAGE_B181._serialized_end=2075\n _LOTSNESTEDMESSAGE_B182._serialized_start=2077\n _LOTSNESTEDMESSAGE_B182._serialized_end=2083\n _LOTSNESTEDMESSAGE_B183._serialized_start=2085\n _LOTSNESTEDMESSAGE_B183._serialized_end=2091\n _LOTSNESTEDMESSAGE_B184._serialized_start=2093\n _LOTSNESTEDMESSAGE_B184._serialized_end=2099\n _LOTSNESTEDMESSAGE_B185._serialized_start=2101\n _LOTSNESTEDMESSAGE_B185._serialized_end=2107\n _LOTSNESTEDMESSAGE_B186._serialized_start=2109\n _LOTSNESTEDMESSAGE_B186._serialized_end=2115\n _LOTSNESTEDMESSAGE_B187._serialized_start=2117\n _LOTSNESTEDMESSAGE_B187._serialized_end=2123\n _LOTSNESTEDMESSAGE_B188._serialized_start=2125\n _LOTSNESTEDMESSAGE_B188._serialized_end=2131\n _LOTSNESTEDMESSAGE_B189._serialized_start=2133\n _LOTSNESTEDMESSAGE_B189._serialized_end=2139\n _LOTSNESTEDMESSAGE_B190._serialized_start=2141\n _LOTSNESTEDMESSAGE_B190._serialized_end=2147\n _LOTSNESTEDMESSAGE_B191._serialized_start=2149\n _LOTSNESTEDMESSAGE_B191._serialized_end=2155\n _LOTSNESTEDMESSAGE_B192._serialized_start=2157\n _LOTSNESTEDMESSAGE_B192._serialized_end=2163\n _LOTSNESTEDMESSAGE_B193._serialized_start=2165\n _LOTSNESTEDMESSAGE_B193._serialized_end=2171\n _LOTSNESTEDMESSAGE_B194._serialized_start=2173\n _LOTSNESTEDMESSAGE_B194._serialized_end=2179\n _LOTSNESTEDMESSAGE_B195._serialized_start=2181\n _LOTSNESTEDMESSAGE_B195._serialized_end=2187\n _LOTSNESTEDMESSAGE_B196._serialized_start=2189\n _LOTSNESTEDMESSAGE_B196._serialized_end=2195\n _LOTSNESTEDMESSAGE_B197._serialized_start=2197\n _LOTSNESTEDMESSAGE_B197._serialized_end=2203\n _LOTSNESTEDMESSAGE_B198._serialized_start=2205\n _LOTSNESTEDMESSAGE_B198._serialized_end=2211\n _LOTSNESTEDMESSAGE_B199._serialized_start=2213\n _LOTSNESTEDMESSAGE_B199._serialized_end=2219\n _LOTSNESTEDMESSAGE_B200._serialized_start=2221\n _LOTSNESTEDMESSAGE_B200._serialized_end=2227\n _LOTSNESTEDMESSAGE_B201._serialized_start=2229\n _LOTSNESTEDMESSAGE_B201._serialized_end=2235\n _LOTSNESTEDMESSAGE_B202._serialized_start=2237\n _LOTSNESTEDMESSAGE_B202._serialized_end=2243\n _LOTSNESTEDMESSAGE_B203._serialized_start=2245\n _LOTSNESTEDMESSAGE_B203._serialized_end=2251\n _LOTSNESTEDMESSAGE_B204._serialized_start=2253\n _LOTSNESTEDMESSAGE_B204._serialized_end=2259\n _LOTSNESTEDMESSAGE_B205._serialized_start=2261\n _LOTSNESTEDMESSAGE_B205._serialized_end=2267\n _LOTSNESTEDMESSAGE_B206._serialized_start=2269\n _LOTSNESTEDMESSAGE_B206._serialized_end=2275\n _LOTSNESTEDMESSAGE_B207._serialized_start=2277\n _LOTSNESTEDMESSAGE_B207._serialized_end=2283\n _LOTSNESTEDMESSAGE_B208._serialized_start=2285\n _LOTSNESTEDMESSAGE_B208._serialized_end=2291\n _LOTSNESTEDMESSAGE_B209._serialized_start=2293\n _LOTSNESTEDMESSAGE_B209._serialized_end=2299\n _LOTSNESTEDMESSAGE_B210._serialized_start=2301\n _LOTSNESTEDMESSAGE_B210._serialized_end=2307\n _LOTSNESTEDMESSAGE_B211._serialized_start=2309\n _LOTSNESTEDMESSAGE_B211._serialized_end=2315\n _LOTSNESTEDMESSAGE_B212._serialized_start=2317\n _LOTSNESTEDMESSAGE_B212._serialized_end=2323\n _LOTSNESTEDMESSAGE_B213._serialized_start=2325\n _LOTSNESTEDMESSAGE_B213._serialized_end=2331\n _LOTSNESTEDMESSAGE_B214._serialized_start=2333\n _LOTSNESTEDMESSAGE_B214._serialized_end=2339\n _LOTSNESTEDMESSAGE_B215._serialized_start=2341\n _LOTSNESTEDMESSAGE_B215._serialized_end=2347\n _LOTSNESTEDMESSAGE_B216._serialized_start=2349\n _LOTSNESTEDMESSAGE_B216._serialized_end=2355\n _LOTSNESTEDMESSAGE_B217._serialized_start=2357\n _LOTSNESTEDMESSAGE_B217._serialized_end=2363\n _LOTSNESTEDMESSAGE_B218._serialized_start=2365\n _LOTSNESTEDMESSAGE_B218._serialized_end=2371\n _LOTSNESTEDMESSAGE_B219._serialized_start=2373\n _LOTSNESTEDMESSAGE_B219._serialized_end=2379\n _LOTSNESTEDMESSAGE_B220._serialized_start=2381\n _LOTSNESTEDMESSAGE_B220._serialized_end=2387\n _LOTSNESTEDMESSAGE_B221._serialized_start=2389\n _LOTSNESTEDMESSAGE_B221._serialized_end=2395\n _LOTSNESTEDMESSAGE_B222._serialized_start=2397\n _LOTSNESTEDMESSAGE_B222._serialized_end=2403\n _LOTSNESTEDMESSAGE_B223._serialized_start=2405\n _LOTSNESTEDMESSAGE_B223._serialized_end=2411\n _LOTSNESTEDMESSAGE_B224._serialized_start=2413\n _LOTSNESTEDMESSAGE_B224._serialized_end=2419\n _LOTSNESTEDMESSAGE_B225._serialized_start=2421\n _LOTSNESTEDMESSAGE_B225._serialized_end=2427\n _LOTSNESTEDMESSAGE_B226._serialized_start=2429\n _LOTSNESTEDMESSAGE_B226._serialized_end=2435\n _LOTSNESTEDMESSAGE_B227._serialized_start=2437\n _LOTSNESTEDMESSAGE_B227._serialized_end=2443\n _LOTSNESTEDMESSAGE_B228._serialized_start=2445\n _LOTSNESTEDMESSAGE_B228._serialized_end=2451\n _LOTSNESTEDMESSAGE_B229._serialized_start=2453\n _LOTSNESTEDMESSAGE_B229._serialized_end=2459\n _LOTSNESTEDMESSAGE_B230._serialized_start=2461\n _LOTSNESTEDMESSAGE_B230._serialized_end=2467\n _LOTSNESTEDMESSAGE_B231._serialized_start=2469\n _LOTSNESTEDMESSAGE_B231._serialized_end=2475\n _LOTSNESTEDMESSAGE_B232._serialized_start=2477\n _LOTSNESTEDMESSAGE_B232._serialized_end=2483\n _LOTSNESTEDMESSAGE_B233._serialized_start=2485\n _LOTSNESTEDMESSAGE_B233._serialized_end=2491\n _LOTSNESTEDMESSAGE_B234._serialized_start=2493\n _LOTSNESTEDMESSAGE_B234._serialized_end=2499\n _LOTSNESTEDMESSAGE_B235._serialized_start=2501\n _LOTSNESTEDMESSAGE_B235._serialized_end=2507\n _LOTSNESTEDMESSAGE_B236._serialized_start=2509\n _LOTSNESTEDMESSAGE_B236._serialized_end=2515\n _LOTSNESTEDMESSAGE_B237._serialized_start=2517\n _LOTSNESTEDMESSAGE_B237._serialized_end=2523\n _LOTSNESTEDMESSAGE_B238._serialized_start=2525\n _LOTSNESTEDMESSAGE_B238._serialized_end=2531\n _LOTSNESTEDMESSAGE_B239._serialized_start=2533\n _LOTSNESTEDMESSAGE_B239._serialized_end=2539\n _LOTSNESTEDMESSAGE_B240._serialized_start=2541\n _LOTSNESTEDMESSAGE_B240._serialized_end=2547\n _LOTSNESTEDMESSAGE_B241._serialized_start=2549\n _LOTSNESTEDMESSAGE_B241._serialized_end=2555\n _LOTSNESTEDMESSAGE_B242._serialized_start=2557\n _LOTSNESTEDMESSAGE_B242._serialized_end=2563\n _LOTSNESTEDMESSAGE_B243._serialized_start=2565\n _LOTSNESTEDMESSAGE_B243._serialized_end=2571\n _LOTSNESTEDMESSAGE_B244._serialized_start=2573\n _LOTSNESTEDMESSAGE_B244._serialized_end=2579\n _LOTSNESTEDMESSAGE_B245._serialized_start=2581\n _LOTSNESTEDMESSAGE_B245._serialized_end=2587\n _LOTSNESTEDMESSAGE_B246._serialized_start=2589\n _LOTSNESTEDMESSAGE_B246._serialized_end=2595\n _LOTSNESTEDMESSAGE_B247._serialized_start=2597\n _LOTSNESTEDMESSAGE_B247._serialized_end=2603\n _LOTSNESTEDMESSAGE_B248._serialized_start=2605\n _LOTSNESTEDMESSAGE_B248._serialized_end=2611\n _LOTSNESTEDMESSAGE_B249._serialized_start=2613\n _LOTSNESTEDMESSAGE_B249._serialized_end=2619\n _LOTSNESTEDMESSAGE_B250._serialized_start=2621\n _LOTSNESTEDMESSAGE_B250._serialized_end=2627\n _LOTSNESTEDMESSAGE_B251._serialized_start=2629\n _LOTSNESTEDMESSAGE_B251._serialized_end=2635\n _LOTSNESTEDMESSAGE_B252._serialized_start=2637\n _LOTSNESTEDMESSAGE_B252._serialized_end=2643\n _LOTSNESTEDMESSAGE_B253._serialized_start=2645\n _LOTSNESTEDMESSAGE_B253._serialized_end=2651\n _LOTSNESTEDMESSAGE_B254._serialized_start=2653\n _LOTSNESTEDMESSAGE_B254._serialized_end=2659\n _LOTSNESTEDMESSAGE_B255._serialized_start=2661\n _LOTSNESTEDMESSAGE_B255._serialized_end=2667\n# @@protoc_insertion_point(module_scope)\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/internal/no_package_pb2.py",
"content": "# -*- coding: utf-8 -*-\n# Generated by the protocol buffer compiler. DO NOT EDIT!\n# source: google/protobuf/internal/no_package.proto\n\"\"\"Generated protocol buffer code.\"\"\"\nfrom google.protobuf.internal import builder as _builder\nfrom google.protobuf import descriptor as _descriptor\nfrom google.protobuf import descriptor_pool as _descriptor_pool\nfrom google.protobuf import symbol_database as _symbol_database\n# @@protoc_insertion_point(imports)\n\n_sym_db = _symbol_database.Default()\n\n\n\n\nDESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\\n)google/protobuf/internal/no_package.proto\\\";\\n\\x10NoPackageMessage\\x12\\'\\n\\x0fno_package_enum\\x18\\x01 \\x01(\\x0e\\x32\\x0e.NoPackageEnum*?\\n\\rNoPackageEnum\\x12\\x16\\n\\x12NO_PACKAGE_VALUE_0\\x10\\x00\\x12\\x16\\n\\x12NO_PACKAGE_VALUE_1\\x10\\x01')\n\n_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())\n_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.internal.no_package_pb2', globals())\nif _descriptor._USE_C_DESCRIPTORS == False:\n\n DESCRIPTOR._options = None\n _NOPACKAGEENUM._serialized_start=106\n _NOPACKAGEENUM._serialized_end=169\n _NOPACKAGEMESSAGE._serialized_start=45\n _NOPACKAGEMESSAGE._serialized_end=104\n# @@protoc_insertion_point(module_scope)\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/internal/python_message.py",
"content": "# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n# This code is meant to work on Python 2.4 and above only.\n#\n# TODO(robinson): Helpers for verbose, common checks like seeing if a\n# descriptor's cpp_type is CPPTYPE_MESSAGE.\n\n\"\"\"Contains a metaclass and helper functions used to create\nprotocol message classes from Descriptor objects at runtime.\n\nRecall that a metaclass is the \"type\" of a class.\n(A class is to a metaclass what an instance is to a class.)\n\nIn this case, we use the GeneratedProtocolMessageType metaclass\nto inject all the useful functionality into the classes\noutput by the protocol compiler at compile-time.\n\nThe upshot of all this is that the real implementation\ndetails for ALL pure-Python protocol buffers are *here in\nthis file*.\n\"\"\"\n\n__author__ = 'robinson@google.com (Will Robinson)'\n\nfrom io import BytesIO\nimport struct\nimport sys\nimport weakref\n\n# We use \"as\" to avoid name collisions with variables.\nfrom google.protobuf.internal import api_implementation\nfrom google.protobuf.internal import containers\nfrom google.protobuf.internal import decoder\nfrom google.protobuf.internal import encoder\nfrom google.protobuf.internal import enum_type_wrapper\nfrom google.protobuf.internal import extension_dict\nfrom google.protobuf.internal import message_listener as message_listener_mod\nfrom google.protobuf.internal import type_checkers\nfrom google.protobuf.internal import well_known_types\nfrom google.protobuf.internal import wire_format\nfrom google.protobuf import descriptor as descriptor_mod\nfrom google.protobuf import message as message_mod\nfrom google.protobuf import text_format\n\n_FieldDescriptor = descriptor_mod.FieldDescriptor\n_AnyFullTypeName = 'google.protobuf.Any'\n_ExtensionDict = extension_dict._ExtensionDict\n\nclass GeneratedProtocolMessageType(type):\n\n \"\"\"Metaclass for protocol message classes created at runtime from Descriptors.\n\n We add implementations for all methods described in the Message class. We\n also create properties to allow getting/setting all fields in the protocol\n message. Finally, we create slots to prevent users from accidentally\n \"setting\" nonexistent fields in the protocol message, which then wouldn't get\n serialized / deserialized properly.\n\n The protocol compiler currently uses this metaclass to create protocol\n message classes at runtime. Clients can also manually create their own\n classes at runtime, as in this example:\n\n mydescriptor = Descriptor(.....)\n factory = symbol_database.Default()\n factory.pool.AddDescriptor(mydescriptor)\n MyProtoClass = factory.GetPrototype(mydescriptor)\n myproto_instance = MyProtoClass()\n myproto.foo_field = 23\n ...\n \"\"\"\n\n # Must be consistent with the protocol-compiler code in\n # proto2/compiler/internal/generator.*.\n _DESCRIPTOR_KEY = 'DESCRIPTOR'\n\n def __new__(cls, name, bases, dictionary):\n \"\"\"Custom allocation for runtime-generated class types.\n\n We override __new__ because this is apparently the only place\n where we can meaningfully set __slots__ on the class we're creating(?).\n (The interplay between metaclasses and slots is not very well-documented).\n\n Args:\n name: Name of the class (ignored, but required by the\n metaclass protocol).\n bases: Base classes of the class we're constructing.\n (Should be message.Message). We ignore this field, but\n it's required by the metaclass protocol\n dictionary: The class dictionary of the class we're\n constructing. dictionary[_DESCRIPTOR_KEY] must contain\n a Descriptor object describing this protocol message\n type.\n\n Returns:\n Newly-allocated class.\n\n Raises:\n RuntimeError: Generated code only work with python cpp extension.\n \"\"\"\n descriptor = dictionary[GeneratedProtocolMessageType._DESCRIPTOR_KEY]\n\n if isinstance(descriptor, str):\n raise RuntimeError('The generated code only work with python cpp '\n 'extension, but it is using pure python runtime.')\n\n # If a concrete class already exists for this descriptor, don't try to\n # create another. Doing so will break any messages that already exist with\n # the existing class.\n #\n # The C++ implementation appears to have its own internal `PyMessageFactory`\n # to achieve similar results.\n #\n # This most commonly happens in `text_format.py` when using descriptors from\n # a custom pool; it calls symbol_database.Global().getPrototype() on a\n # descriptor which already has an existing concrete class.\n new_class = getattr(descriptor, '_concrete_class', None)\n if new_class:\n return new_class\n\n if descriptor.full_name in well_known_types.WKTBASES:\n bases += (well_known_types.WKTBASES[descriptor.full_name],)\n _AddClassAttributesForNestedExtensions(descriptor, dictionary)\n _AddSlots(descriptor, dictionary)\n\n superclass = super(GeneratedProtocolMessageType, cls)\n new_class = superclass.__new__(cls, name, bases, dictionary)\n return new_class\n\n def __init__(cls, name, bases, dictionary):\n \"\"\"Here we perform the majority of our work on the class.\n We add enum getters, an __init__ method, implementations\n of all Message methods, and properties for all fields\n in the protocol type.\n\n Args:\n name: Name of the class (ignored, but required by the\n metaclass protocol).\n bases: Base classes of the class we're constructing.\n (Should be message.Message). We ignore this field, but\n it's required by the metaclass protocol\n dictionary: The class dictionary of the class we're\n constructing. dictionary[_DESCRIPTOR_KEY] must contain\n a Descriptor object describing this protocol message\n type.\n \"\"\"\n descriptor = dictionary[GeneratedProtocolMessageType._DESCRIPTOR_KEY]\n\n # If this is an _existing_ class looked up via `_concrete_class` in the\n # __new__ method above, then we don't need to re-initialize anything.\n existing_class = getattr(descriptor, '_concrete_class', None)\n if existing_class:\n assert existing_class is cls, (\n 'Duplicate `GeneratedProtocolMessageType` created for descriptor %r'\n % (descriptor.full_name))\n return\n\n cls._decoders_by_tag = {}\n if (descriptor.has_options and\n descriptor.GetOptions().message_set_wire_format):\n cls._decoders_by_tag[decoder.MESSAGE_SET_ITEM_TAG] = (\n decoder.MessageSetItemDecoder(descriptor), None)\n\n # Attach stuff to each FieldDescriptor for quick lookup later on.\n for field in descriptor.fields:\n _AttachFieldHelpers(cls, field)\n\n descriptor._concrete_class = cls # pylint: disable=protected-access\n _AddEnumValues(descriptor, cls)\n _AddInitMethod(descriptor, cls)\n _AddPropertiesForFields(descriptor, cls)\n _AddPropertiesForExtensions(descriptor, cls)\n _AddStaticMethods(cls)\n _AddMessageMethods(descriptor, cls)\n _AddPrivateHelperMethods(descriptor, cls)\n\n superclass = super(GeneratedProtocolMessageType, cls)\n superclass.__init__(name, bases, dictionary)\n\n\n# Stateless helpers for GeneratedProtocolMessageType below.\n# Outside clients should not access these directly.\n#\n# I opted not to make any of these methods on the metaclass, to make it more\n# clear that I'm not really using any state there and to keep clients from\n# thinking that they have direct access to these construction helpers.\n\n\ndef _PropertyName(proto_field_name):\n \"\"\"Returns the name of the public property attribute which\n clients can use to get and (in some cases) set the value\n of a protocol message field.\n\n Args:\n proto_field_name: The protocol message field name, exactly\n as it appears (or would appear) in a .proto file.\n \"\"\"\n # TODO(robinson): Escape Python keywords (e.g., yield), and test this support.\n # nnorwitz makes my day by writing:\n # \"\"\"\n # FYI. See the keyword module in the stdlib. This could be as simple as:\n #\n # if keyword.iskeyword(proto_field_name):\n # return proto_field_name + \"_\"\n # return proto_field_name\n # \"\"\"\n # Kenton says: The above is a BAD IDEA. People rely on being able to use\n # getattr() and setattr() to reflectively manipulate field values. If we\n # rename the properties, then every such user has to also make sure to apply\n # the same transformation. Note that currently if you name a field \"yield\",\n # you can still access it just fine using getattr/setattr -- it's not even\n # that cumbersome to do so.\n # TODO(kenton): Remove this method entirely if/when everyone agrees with my\n # position.\n return proto_field_name\n\n\ndef _AddSlots(message_descriptor, dictionary):\n \"\"\"Adds a __slots__ entry to dictionary, containing the names of all valid\n attributes for this message type.\n\n Args:\n message_descriptor: A Descriptor instance describing this message type.\n dictionary: Class dictionary to which we'll add a '__slots__' entry.\n \"\"\"\n dictionary['__slots__'] = ['_cached_byte_size',\n '_cached_byte_size_dirty',\n '_fields',\n '_unknown_fields',\n '_unknown_field_set',\n '_is_present_in_parent',\n '_listener',\n '_listener_for_children',\n '__weakref__',\n '_oneofs']\n\n\ndef _IsMessageSetExtension(field):\n return (field.is_extension and\n field.containing_type.has_options and\n field.containing_type.GetOptions().message_set_wire_format and\n field.type == _FieldDescriptor.TYPE_MESSAGE and\n field.label == _FieldDescriptor.LABEL_OPTIONAL)\n\n\ndef _IsMapField(field):\n return (field.type == _FieldDescriptor.TYPE_MESSAGE and\n field.message_type.has_options and\n field.message_type.GetOptions().map_entry)\n\n\ndef _IsMessageMapField(field):\n value_type = field.message_type.fields_by_name['value']\n return value_type.cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE\n\n\ndef _AttachFieldHelpers(cls, field_descriptor):\n is_repeated = (field_descriptor.label == _FieldDescriptor.LABEL_REPEATED)\n is_packable = (is_repeated and\n wire_format.IsTypePackable(field_descriptor.type))\n is_proto3 = field_descriptor.containing_type.syntax == 'proto3'\n if not is_packable:\n is_packed = False\n elif field_descriptor.containing_type.syntax == 'proto2':\n is_packed = (field_descriptor.has_options and\n field_descriptor.GetOptions().packed)\n else:\n has_packed_false = (field_descriptor.has_options and\n field_descriptor.GetOptions().HasField('packed') and\n field_descriptor.GetOptions().packed == False)\n is_packed = not has_packed_false\n is_map_entry = _IsMapField(field_descriptor)\n\n if is_map_entry:\n field_encoder = encoder.MapEncoder(field_descriptor)\n sizer = encoder.MapSizer(field_descriptor,\n _IsMessageMapField(field_descriptor))\n elif _IsMessageSetExtension(field_descriptor):\n field_encoder = encoder.MessageSetItemEncoder(field_descriptor.number)\n sizer = encoder.MessageSetItemSizer(field_descriptor.number)\n else:\n field_encoder = type_checkers.TYPE_TO_ENCODER[field_descriptor.type](\n field_descriptor.number, is_repeated, is_packed)\n sizer = type_checkers.TYPE_TO_SIZER[field_descriptor.type](\n field_descriptor.number, is_repeated, is_packed)\n\n field_descriptor._encoder = field_encoder\n field_descriptor._sizer = sizer\n field_descriptor._default_constructor = _DefaultValueConstructorForField(\n field_descriptor)\n\n def AddDecoder(wiretype, is_packed):\n tag_bytes = encoder.TagBytes(field_descriptor.number, wiretype)\n decode_type = field_descriptor.type\n if (decode_type == _FieldDescriptor.TYPE_ENUM and\n type_checkers.SupportsOpenEnums(field_descriptor)):\n decode_type = _FieldDescriptor.TYPE_INT32\n\n oneof_descriptor = None\n clear_if_default = False\n if field_descriptor.containing_oneof is not None:\n oneof_descriptor = field_descriptor\n elif (is_proto3 and not is_repeated and\n field_descriptor.cpp_type != _FieldDescriptor.CPPTYPE_MESSAGE):\n clear_if_default = True\n\n if is_map_entry:\n is_message_map = _IsMessageMapField(field_descriptor)\n\n field_decoder = decoder.MapDecoder(\n field_descriptor, _GetInitializeDefaultForMap(field_descriptor),\n is_message_map)\n elif decode_type == _FieldDescriptor.TYPE_STRING:\n field_decoder = decoder.StringDecoder(\n field_descriptor.number, is_repeated, is_packed,\n field_descriptor, field_descriptor._default_constructor,\n clear_if_default)\n elif field_descriptor.cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE:\n field_decoder = type_checkers.TYPE_TO_DECODER[decode_type](\n field_descriptor.number, is_repeated, is_packed,\n field_descriptor, field_descriptor._default_constructor)\n else:\n field_decoder = type_checkers.TYPE_TO_DECODER[decode_type](\n field_descriptor.number, is_repeated, is_packed,\n # pylint: disable=protected-access\n field_descriptor, field_descriptor._default_constructor,\n clear_if_default)\n\n cls._decoders_by_tag[tag_bytes] = (field_decoder, oneof_descriptor)\n\n AddDecoder(type_checkers.FIELD_TYPE_TO_WIRE_TYPE[field_descriptor.type],\n False)\n\n if is_repeated and wire_format.IsTypePackable(field_descriptor.type):\n # To support wire compatibility of adding packed = true, add a decoder for\n # packed values regardless of the field's options.\n AddDecoder(wire_format.WIRETYPE_LENGTH_DELIMITED, True)\n\n\ndef _AddClassAttributesForNestedExtensions(descriptor, dictionary):\n extensions = descriptor.extensions_by_name\n for extension_name, extension_field in extensions.items():\n assert extension_name not in dictionary\n dictionary[extension_name] = extension_field\n\n\ndef _AddEnumValues(descriptor, cls):\n \"\"\"Sets class-level attributes for all enum fields defined in this message.\n\n Also exporting a class-level object that can name enum values.\n\n Args:\n descriptor: Descriptor object for this message type.\n cls: Class we're constructing for this message type.\n \"\"\"\n for enum_type in descriptor.enum_types:\n setattr(cls, enum_type.name, enum_type_wrapper.EnumTypeWrapper(enum_type))\n for enum_value in enum_type.values:\n setattr(cls, enum_value.name, enum_value.number)\n\n\ndef _GetInitializeDefaultForMap(field):\n if field.label != _FieldDescriptor.LABEL_REPEATED:\n raise ValueError('map_entry set on non-repeated field %s' % (\n field.name))\n fields_by_name = field.message_type.fields_by_name\n key_checker = type_checkers.GetTypeChecker(fields_by_name['key'])\n\n value_field = fields_by_name['value']\n if _IsMessageMapField(field):\n def MakeMessageMapDefault(message):\n return containers.MessageMap(\n message._listener_for_children, value_field.message_type, key_checker,\n field.message_type)\n return MakeMessageMapDefault\n else:\n value_checker = type_checkers.GetTypeChecker(value_field)\n def MakePrimitiveMapDefault(message):\n return containers.ScalarMap(\n message._listener_for_children, key_checker, value_checker,\n field.message_type)\n return MakePrimitiveMapDefault\n\ndef _DefaultValueConstructorForField(field):\n \"\"\"Returns a function which returns a default value for a field.\n\n Args:\n field: FieldDescriptor object for this field.\n\n The returned function has one argument:\n message: Message instance containing this field, or a weakref proxy\n of same.\n\n That function in turn returns a default value for this field. The default\n value may refer back to |message| via a weak reference.\n \"\"\"\n\n if _IsMapField(field):\n return _GetInitializeDefaultForMap(field)\n\n if field.label == _FieldDescriptor.LABEL_REPEATED:\n if field.has_default_value and field.default_value != []:\n raise ValueError('Repeated field default value not empty list: %s' % (\n field.default_value))\n if field.cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE:\n # We can't look at _concrete_class yet since it might not have\n # been set. (Depends on order in which we initialize the classes).\n message_type = field.message_type\n def MakeRepeatedMessageDefault(message):\n return containers.RepeatedCompositeFieldContainer(\n message._listener_for_children, field.message_type)\n return MakeRepeatedMessageDefault\n else:\n type_checker = type_checkers.GetTypeChecker(field)\n def MakeRepeatedScalarDefault(message):\n return containers.RepeatedScalarFieldContainer(\n message._listener_for_children, type_checker)\n return MakeRepeatedScalarDefault\n\n if field.cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE:\n # _concrete_class may not yet be initialized.\n message_type = field.message_type\n def MakeSubMessageDefault(message):\n assert getattr(message_type, '_concrete_class', None), (\n 'Uninitialized concrete class found for field %r (message type %r)'\n % (field.full_name, message_type.full_name))\n result = message_type._concrete_class()\n result._SetListener(\n _OneofListener(message, field)\n if field.containing_oneof is not None\n else message._listener_for_children)\n return result\n return MakeSubMessageDefault\n\n def MakeScalarDefault(message):\n # TODO(protobuf-team): This may be broken since there may not be\n # default_value. Combine with has_default_value somehow.\n return field.default_value\n return MakeScalarDefault\n\n\ndef _ReraiseTypeErrorWithFieldName(message_name, field_name):\n \"\"\"Re-raise the currently-handled TypeError with the field name added.\"\"\"\n exc = sys.exc_info()[1]\n if len(exc.args) == 1 and type(exc) is TypeError:\n # simple TypeError; add field name to exception message\n exc = TypeError('%s for field %s.%s' % (str(exc), message_name, field_name))\n\n # re-raise possibly-amended exception with original traceback:\n raise exc.with_traceback(sys.exc_info()[2])\n\n\ndef _AddInitMethod(message_descriptor, cls):\n \"\"\"Adds an __init__ method to cls.\"\"\"\n\n def _GetIntegerEnumValue(enum_type, value):\n \"\"\"Convert a string or integer enum value to an integer.\n\n If the value is a string, it is converted to the enum value in\n enum_type with the same name. If the value is not a string, it's\n returned as-is. (No conversion or bounds-checking is done.)\n \"\"\"\n if isinstance(value, str):\n try:\n return enum_type.values_by_name[value].number\n except KeyError:\n raise ValueError('Enum type %s: unknown label \"%s\"' % (\n enum_type.full_name, value))\n return value\n\n def init(self, **kwargs):\n self._cached_byte_size = 0\n self._cached_byte_size_dirty = len(kwargs) > 0\n self._fields = {}\n # Contains a mapping from oneof field descriptors to the descriptor\n # of the currently set field in that oneof field.\n self._oneofs = {}\n\n # _unknown_fields is () when empty for efficiency, and will be turned into\n # a list if fields are added.\n self._unknown_fields = ()\n # _unknown_field_set is None when empty for efficiency, and will be\n # turned into UnknownFieldSet struct if fields are added.\n self._unknown_field_set = None # pylint: disable=protected-access\n self._is_present_in_parent = False\n self._listener = message_listener_mod.NullMessageListener()\n self._listener_for_children = _Listener(self)\n for field_name, field_value in kwargs.items():\n field = _GetFieldByName(message_descriptor, field_name)\n if field is None:\n raise TypeError('%s() got an unexpected keyword argument \"%s\"' %\n (message_descriptor.name, field_name))\n if field_value is None:\n # field=None is the same as no field at all.\n continue\n if field.label == _FieldDescriptor.LABEL_REPEATED:\n copy = field._default_constructor(self)\n if field.cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE: # Composite\n if _IsMapField(field):\n if _IsMessageMapField(field):\n for key in field_value:\n copy[key].MergeFrom(field_value[key])\n else:\n copy.update(field_value)\n else:\n for val in field_value:\n if isinstance(val, dict):\n copy.add(**val)\n else:\n copy.add().MergeFrom(val)\n else: # Scalar\n if field.cpp_type == _FieldDescriptor.CPPTYPE_ENUM:\n field_value = [_GetIntegerEnumValue(field.enum_type, val)\n for val in field_value]\n copy.extend(field_value)\n self._fields[field] = copy\n elif field.cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE:\n copy = field._default_constructor(self)\n new_val = field_value\n if isinstance(field_value, dict):\n new_val = field.message_type._concrete_class(**field_value)\n try:\n copy.MergeFrom(new_val)\n except TypeError:\n _ReraiseTypeErrorWithFieldName(message_descriptor.name, field_name)\n self._fields[field] = copy\n else:\n if field.cpp_type == _FieldDescriptor.CPPTYPE_ENUM:\n field_value = _GetIntegerEnumValue(field.enum_type, field_value)\n try:\n setattr(self, field_name, field_value)\n except TypeError:\n _ReraiseTypeErrorWithFieldName(message_descriptor.name, field_name)\n\n init.__module__ = None\n init.__doc__ = None\n cls.__init__ = init\n\n\ndef _GetFieldByName(message_descriptor, field_name):\n \"\"\"Returns a field descriptor by field name.\n\n Args:\n message_descriptor: A Descriptor describing all fields in message.\n field_name: The name of the field to retrieve.\n Returns:\n The field descriptor associated with the field name.\n \"\"\"\n try:\n return message_descriptor.fields_by_name[field_name]\n except KeyError:\n raise ValueError('Protocol message %s has no \"%s\" field.' %\n (message_descriptor.name, field_name))\n\n\ndef _AddPropertiesForFields(descriptor, cls):\n \"\"\"Adds properties for all fields in this protocol message type.\"\"\"\n for field in descriptor.fields:\n _AddPropertiesForField(field, cls)\n\n if descriptor.is_extendable:\n # _ExtensionDict is just an adaptor with no state so we allocate a new one\n # every time it is accessed.\n cls.Extensions = property(lambda self: _ExtensionDict(self))\n\n\ndef _AddPropertiesForField(field, cls):\n \"\"\"Adds a public property for a protocol message field.\n Clients can use this property to get and (in the case\n of non-repeated scalar fields) directly set the value\n of a protocol message field.\n\n Args:\n field: A FieldDescriptor for this field.\n cls: The class we're constructing.\n \"\"\"\n # Catch it if we add other types that we should\n # handle specially here.\n assert _FieldDescriptor.MAX_CPPTYPE == 10\n\n constant_name = field.name.upper() + '_FIELD_NUMBER'\n setattr(cls, constant_name, field.number)\n\n if field.label == _FieldDescriptor.LABEL_REPEATED:\n _AddPropertiesForRepeatedField(field, cls)\n elif field.cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE:\n _AddPropertiesForNonRepeatedCompositeField(field, cls)\n else:\n _AddPropertiesForNonRepeatedScalarField(field, cls)\n\n\nclass _FieldProperty(property):\n __slots__ = ('DESCRIPTOR',)\n\n def __init__(self, descriptor, getter, setter, doc):\n property.__init__(self, getter, setter, doc=doc)\n self.DESCRIPTOR = descriptor\n\n\ndef _AddPropertiesForRepeatedField(field, cls):\n \"\"\"Adds a public property for a \"repeated\" protocol message field. Clients\n can use this property to get the value of the field, which will be either a\n RepeatedScalarFieldContainer or RepeatedCompositeFieldContainer (see\n below).\n\n Note that when clients add values to these containers, we perform\n type-checking in the case of repeated scalar fields, and we also set any\n necessary \"has\" bits as a side-effect.\n\n Args:\n field: A FieldDescriptor for this field.\n cls: The class we're constructing.\n \"\"\"\n proto_field_name = field.name\n property_name = _PropertyName(proto_field_name)\n\n def getter(self):\n field_value = self._fields.get(field)\n if field_value is None:\n # Construct a new object to represent this field.\n field_value = field._default_constructor(self)\n\n # Atomically check if another thread has preempted us and, if not, swap\n # in the new object we just created. If someone has preempted us, we\n # take that object and discard ours.\n # WARNING: We are relying on setdefault() being atomic. This is true\n # in CPython but we haven't investigated others. This warning appears\n # in several other locations in this file.\n field_value = self._fields.setdefault(field, field_value)\n return field_value\n getter.__module__ = None\n getter.__doc__ = 'Getter for %s.' % proto_field_name\n\n # We define a setter just so we can throw an exception with a more\n # helpful error message.\n def setter(self, new_value):\n raise AttributeError('Assignment not allowed to repeated field '\n '\"%s\" in protocol message object.' % proto_field_name)\n\n doc = 'Magic attribute generated for \"%s\" proto field.' % proto_field_name\n setattr(cls, property_name, _FieldProperty(field, getter, setter, doc=doc))\n\n\ndef _AddPropertiesForNonRepeatedScalarField(field, cls):\n \"\"\"Adds a public property for a nonrepeated, scalar protocol message field.\n Clients can use this property to get and directly set the value of the field.\n Note that when the client sets the value of a field by using this property,\n all necessary \"has\" bits are set as a side-effect, and we also perform\n type-checking.\n\n Args:\n field: A FieldDescriptor for this field.\n cls: The class we're constructing.\n \"\"\"\n proto_field_name = field.name\n property_name = _PropertyName(proto_field_name)\n type_checker = type_checkers.GetTypeChecker(field)\n default_value = field.default_value\n is_proto3 = field.containing_type.syntax == 'proto3'\n\n def getter(self):\n # TODO(protobuf-team): This may be broken since there may not be\n # default_value. Combine with has_default_value somehow.\n return self._fields.get(field, default_value)\n getter.__module__ = None\n getter.__doc__ = 'Getter for %s.' % proto_field_name\n\n clear_when_set_to_default = is_proto3 and not field.containing_oneof\n\n def field_setter(self, new_value):\n # pylint: disable=protected-access\n # Testing the value for truthiness captures all of the proto3 defaults\n # (0, 0.0, enum 0, and False).\n try:\n new_value = type_checker.CheckValue(new_value)\n except TypeError as e:\n raise TypeError(\n 'Cannot set %s to %.1024r: %s' % (field.full_name, new_value, e))\n if clear_when_set_to_default and not new_value:\n self._fields.pop(field, None)\n else:\n self._fields[field] = new_value\n # Check _cached_byte_size_dirty inline to improve performance, since scalar\n # setters are called frequently.\n if not self._cached_byte_size_dirty:\n self._Modified()\n\n if field.containing_oneof:\n def setter(self, new_value):\n field_setter(self, new_value)\n self._UpdateOneofState(field)\n else:\n setter = field_setter\n\n setter.__module__ = None\n setter.__doc__ = 'Setter for %s.' % proto_field_name\n\n # Add a property to encapsulate the getter/setter.\n doc = 'Magic attribute generated for \"%s\" proto field.' % proto_field_name\n setattr(cls, property_name, _FieldProperty(field, getter, setter, doc=doc))\n\n\ndef _AddPropertiesForNonRepeatedCompositeField(field, cls):\n \"\"\"Adds a public property for a nonrepeated, composite protocol message field.\n A composite field is a \"group\" or \"message\" field.\n\n Clients can use this property to get the value of the field, but cannot\n assign to the property directly.\n\n Args:\n field: A FieldDescriptor for this field.\n cls: The class we're constructing.\n \"\"\"\n # TODO(robinson): Remove duplication with similar method\n # for non-repeated scalars.\n proto_field_name = field.name\n property_name = _PropertyName(proto_field_name)\n\n def getter(self):\n field_value = self._fields.get(field)\n if field_value is None:\n # Construct a new object to represent this field.\n field_value = field._default_constructor(self)\n\n # Atomically check if another thread has preempted us and, if not, swap\n # in the new object we just created. If someone has preempted us, we\n # take that object and discard ours.\n # WARNING: We are relying on setdefault() being atomic. This is true\n # in CPython but we haven't investigated others. This warning appears\n # in several other locations in this file.\n field_value = self._fields.setdefault(field, field_value)\n return field_value\n getter.__module__ = None\n getter.__doc__ = 'Getter for %s.' % proto_field_name\n\n # We define a setter just so we can throw an exception with a more\n # helpful error message.\n def setter(self, new_value):\n raise AttributeError('Assignment not allowed to composite field '\n '\"%s\" in protocol message object.' % proto_field_name)\n\n # Add a property to encapsulate the getter.\n doc = 'Magic attribute generated for \"%s\" proto field.' % proto_field_name\n setattr(cls, property_name, _FieldProperty(field, getter, setter, doc=doc))\n\n\ndef _AddPropertiesForExtensions(descriptor, cls):\n \"\"\"Adds properties for all fields in this protocol message type.\"\"\"\n extensions = descriptor.extensions_by_name\n for extension_name, extension_field in extensions.items():\n constant_name = extension_name.upper() + '_FIELD_NUMBER'\n setattr(cls, constant_name, extension_field.number)\n\n # TODO(amauryfa): Migrate all users of these attributes to functions like\n # pool.FindExtensionByNumber(descriptor).\n if descriptor.file is not None:\n # TODO(amauryfa): Use cls.MESSAGE_FACTORY.pool when available.\n pool = descriptor.file.pool\n cls._extensions_by_number = pool._extensions_by_number[descriptor]\n cls._extensions_by_name = pool._extensions_by_name[descriptor]\n\ndef _AddStaticMethods(cls):\n # TODO(robinson): This probably needs to be thread-safe(?)\n def RegisterExtension(extension_handle):\n extension_handle.containing_type = cls.DESCRIPTOR\n # TODO(amauryfa): Use cls.MESSAGE_FACTORY.pool when available.\n # pylint: disable=protected-access\n cls.DESCRIPTOR.file.pool._AddExtensionDescriptor(extension_handle)\n _AttachFieldHelpers(cls, extension_handle)\n cls.RegisterExtension = staticmethod(RegisterExtension)\n\n def FromString(s):\n message = cls()\n message.MergeFromString(s)\n return message\n cls.FromString = staticmethod(FromString)\n\n\ndef _IsPresent(item):\n \"\"\"Given a (FieldDescriptor, value) tuple from _fields, return true if the\n value should be included in the list returned by ListFields().\"\"\"\n\n if item[0].label == _FieldDescriptor.LABEL_REPEATED:\n return bool(item[1])\n elif item[0].cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE:\n return item[1]._is_present_in_parent\n else:\n return True\n\n\ndef _AddListFieldsMethod(message_descriptor, cls):\n \"\"\"Helper for _AddMessageMethods().\"\"\"\n\n def ListFields(self):\n all_fields = [item for item in self._fields.items() if _IsPresent(item)]\n all_fields.sort(key = lambda item: item[0].number)\n return all_fields\n\n cls.ListFields = ListFields\n\n_PROTO3_ERROR_TEMPLATE = \\\n ('Protocol message %s has no non-repeated submessage field \"%s\" '\n 'nor marked as optional')\n_PROTO2_ERROR_TEMPLATE = 'Protocol message %s has no non-repeated field \"%s\"'\n\ndef _AddHasFieldMethod(message_descriptor, cls):\n \"\"\"Helper for _AddMessageMethods().\"\"\"\n\n is_proto3 = (message_descriptor.syntax == \"proto3\")\n error_msg = _PROTO3_ERROR_TEMPLATE if is_proto3 else _PROTO2_ERROR_TEMPLATE\n\n hassable_fields = {}\n for field in message_descriptor.fields:\n if field.label == _FieldDescriptor.LABEL_REPEATED:\n continue\n # For proto3, only submessages and fields inside a oneof have presence.\n if (is_proto3 and field.cpp_type != _FieldDescriptor.CPPTYPE_MESSAGE and\n not field.containing_oneof):\n continue\n hassable_fields[field.name] = field\n\n # Has methods are supported for oneof descriptors.\n for oneof in message_descriptor.oneofs:\n hassable_fields[oneof.name] = oneof\n\n def HasField(self, field_name):\n try:\n field = hassable_fields[field_name]\n except KeyError:\n raise ValueError(error_msg % (message_descriptor.full_name, field_name))\n\n if isinstance(field, descriptor_mod.OneofDescriptor):\n try:\n return HasField(self, self._oneofs[field].name)\n except KeyError:\n return False\n else:\n if field.cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE:\n value = self._fields.get(field)\n return value is not None and value._is_present_in_parent\n else:\n return field in self._fields\n\n cls.HasField = HasField\n\n\ndef _AddClearFieldMethod(message_descriptor, cls):\n \"\"\"Helper for _AddMessageMethods().\"\"\"\n def ClearField(self, field_name):\n try:\n field = message_descriptor.fields_by_name[field_name]\n except KeyError:\n try:\n field = message_descriptor.oneofs_by_name[field_name]\n if field in self._oneofs:\n field = self._oneofs[field]\n else:\n return\n except KeyError:\n raise ValueError('Protocol message %s has no \"%s\" field.' %\n (message_descriptor.name, field_name))\n\n if field in self._fields:\n # To match the C++ implementation, we need to invalidate iterators\n # for map fields when ClearField() happens.\n if hasattr(self._fields[field], 'InvalidateIterators'):\n self._fields[field].InvalidateIterators()\n\n # Note: If the field is a sub-message, its listener will still point\n # at us. That's fine, because the worst than can happen is that it\n # will call _Modified() and invalidate our byte size. Big deal.\n del self._fields[field]\n\n if self._oneofs.get(field.containing_oneof, None) is field:\n del self._oneofs[field.containing_oneof]\n\n # Always call _Modified() -- even if nothing was changed, this is\n # a mutating method, and thus calling it should cause the field to become\n # present in the parent message.\n self._Modified()\n\n cls.ClearField = ClearField\n\n\ndef _AddClearExtensionMethod(cls):\n \"\"\"Helper for _AddMessageMethods().\"\"\"\n def ClearExtension(self, extension_handle):\n extension_dict._VerifyExtensionHandle(self, extension_handle)\n\n # Similar to ClearField(), above.\n if extension_handle in self._fields:\n del self._fields[extension_handle]\n self._Modified()\n cls.ClearExtension = ClearExtension\n\n\ndef _AddHasExtensionMethod(cls):\n \"\"\"Helper for _AddMessageMethods().\"\"\"\n def HasExtension(self, extension_handle):\n extension_dict._VerifyExtensionHandle(self, extension_handle)\n if extension_handle.label == _FieldDescriptor.LABEL_REPEATED:\n raise KeyError('\"%s\" is repeated.' % extension_handle.full_name)\n\n if extension_handle.cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE:\n value = self._fields.get(extension_handle)\n return value is not None and value._is_present_in_parent\n else:\n return extension_handle in self._fields\n cls.HasExtension = HasExtension\n\ndef _InternalUnpackAny(msg):\n \"\"\"Unpacks Any message and returns the unpacked message.\n\n This internal method is different from public Any Unpack method which takes\n the target message as argument. _InternalUnpackAny method does not have\n target message type and need to find the message type in descriptor pool.\n\n Args:\n msg: An Any message to be unpacked.\n\n Returns:\n The unpacked message.\n \"\"\"\n # TODO(amauryfa): Don't use the factory of generated messages.\n # To make Any work with custom factories, use the message factory of the\n # parent message.\n # pylint: disable=g-import-not-at-top\n from google.protobuf import symbol_database\n factory = symbol_database.Default()\n\n type_url = msg.type_url\n\n if not type_url:\n return None\n\n # TODO(haberman): For now we just strip the hostname. Better logic will be\n # required.\n type_name = type_url.split('/')[-1]\n descriptor = factory.pool.FindMessageTypeByName(type_name)\n\n if descriptor is None:\n return None\n\n message_class = factory.GetPrototype(descriptor)\n message = message_class()\n\n message.ParseFromString(msg.value)\n return message\n\n\ndef _AddEqualsMethod(message_descriptor, cls):\n \"\"\"Helper for _AddMessageMethods().\"\"\"\n def __eq__(self, other):\n if (not isinstance(other, message_mod.Message) or\n other.DESCRIPTOR != self.DESCRIPTOR):\n return False\n\n if self is other:\n return True\n\n if self.DESCRIPTOR.full_name == _AnyFullTypeName:\n any_a = _InternalUnpackAny(self)\n any_b = _InternalUnpackAny(other)\n if any_a and any_b:\n return any_a == any_b\n\n if not self.ListFields() == other.ListFields():\n return False\n\n # TODO(jieluo): Fix UnknownFieldSet to consider MessageSet extensions,\n # then use it for the comparison.\n unknown_fields = list(self._unknown_fields)\n unknown_fields.sort()\n other_unknown_fields = list(other._unknown_fields)\n other_unknown_fields.sort()\n return unknown_fields == other_unknown_fields\n\n cls.__eq__ = __eq__\n\n\ndef _AddStrMethod(message_descriptor, cls):\n \"\"\"Helper for _AddMessageMethods().\"\"\"\n def __str__(self):\n return text_format.MessageToString(self)\n cls.__str__ = __str__\n\n\ndef _AddReprMethod(message_descriptor, cls):\n \"\"\"Helper for _AddMessageMethods().\"\"\"\n def __repr__(self):\n return text_format.MessageToString(self)\n cls.__repr__ = __repr__\n\n\ndef _AddUnicodeMethod(unused_message_descriptor, cls):\n \"\"\"Helper for _AddMessageMethods().\"\"\"\n\n def __unicode__(self):\n return text_format.MessageToString(self, as_utf8=True).decode('utf-8')\n cls.__unicode__ = __unicode__\n\n\ndef _BytesForNonRepeatedElement(value, field_number, field_type):\n \"\"\"Returns the number of bytes needed to serialize a non-repeated element.\n The returned byte count includes space for tag information and any\n other additional space associated with serializing value.\n\n Args:\n value: Value we're serializing.\n field_number: Field number of this value. (Since the field number\n is stored as part of a varint-encoded tag, this has an impact\n on the total bytes required to serialize the value).\n field_type: The type of the field. One of the TYPE_* constants\n within FieldDescriptor.\n \"\"\"\n try:\n fn = type_checkers.TYPE_TO_BYTE_SIZE_FN[field_type]\n return fn(field_number, value)\n except KeyError:\n raise message_mod.EncodeError('Unrecognized field type: %d' % field_type)\n\n\ndef _AddByteSizeMethod(message_descriptor, cls):\n \"\"\"Helper for _AddMessageMethods().\"\"\"\n\n def ByteSize(self):\n if not self._cached_byte_size_dirty:\n return self._cached_byte_size\n\n size = 0\n descriptor = self.DESCRIPTOR\n if descriptor.GetOptions().map_entry:\n # Fields of map entry should always be serialized.\n size = descriptor.fields_by_name['key']._sizer(self.key)\n size += descriptor.fields_by_name['value']._sizer(self.value)\n else:\n for field_descriptor, field_value in self.ListFields():\n size += field_descriptor._sizer(field_value)\n for tag_bytes, value_bytes in self._unknown_fields:\n size += len(tag_bytes) + len(value_bytes)\n\n self._cached_byte_size = size\n self._cached_byte_size_dirty = False\n self._listener_for_children.dirty = False\n return size\n\n cls.ByteSize = ByteSize\n\n\ndef _AddSerializeToStringMethod(message_descriptor, cls):\n \"\"\"Helper for _AddMessageMethods().\"\"\"\n\n def SerializeToString(self, **kwargs):\n # Check if the message has all of its required fields set.\n if not self.IsInitialized():\n raise message_mod.EncodeError(\n 'Message %s is missing required fields: %s' % (\n self.DESCRIPTOR.full_name, ','.join(self.FindInitializationErrors())))\n return self.SerializePartialToString(**kwargs)\n cls.SerializeToString = SerializeToString\n\n\ndef _AddSerializePartialToStringMethod(message_descriptor, cls):\n \"\"\"Helper for _AddMessageMethods().\"\"\"\n\n def SerializePartialToString(self, **kwargs):\n out = BytesIO()\n self._InternalSerialize(out.write, **kwargs)\n return out.getvalue()\n cls.SerializePartialToString = SerializePartialToString\n\n def InternalSerialize(self, write_bytes, deterministic=None):\n if deterministic is None:\n deterministic = (\n api_implementation.IsPythonDefaultSerializationDeterministic())\n else:\n deterministic = bool(deterministic)\n\n descriptor = self.DESCRIPTOR\n if descriptor.GetOptions().map_entry:\n # Fields of map entry should always be serialized.\n descriptor.fields_by_name['key']._encoder(\n write_bytes, self.key, deterministic)\n descriptor.fields_by_name['value']._encoder(\n write_bytes, self.value, deterministic)\n else:\n for field_descriptor, field_value in self.ListFields():\n field_descriptor._encoder(write_bytes, field_value, deterministic)\n for tag_bytes, value_bytes in self._unknown_fields:\n write_bytes(tag_bytes)\n write_bytes(value_bytes)\n cls._InternalSerialize = InternalSerialize\n\n\ndef _AddMergeFromStringMethod(message_descriptor, cls):\n \"\"\"Helper for _AddMessageMethods().\"\"\"\n def MergeFromString(self, serialized):\n serialized = memoryview(serialized)\n length = len(serialized)\n try:\n if self._InternalParse(serialized, 0, length) != length:\n # The only reason _InternalParse would return early is if it\n # encountered an end-group tag.\n raise message_mod.DecodeError('Unexpected end-group tag.')\n except (IndexError, TypeError):\n # Now ord(buf[p:p+1]) == ord('') gets TypeError.\n raise message_mod.DecodeError('Truncated message.')\n except struct.error as e:\n raise message_mod.DecodeError(e)\n return length # Return this for legacy reasons.\n cls.MergeFromString = MergeFromString\n\n local_ReadTag = decoder.ReadTag\n local_SkipField = decoder.SkipField\n decoders_by_tag = cls._decoders_by_tag\n\n def InternalParse(self, buffer, pos, end):\n \"\"\"Create a message from serialized bytes.\n\n Args:\n self: Message, instance of the proto message object.\n buffer: memoryview of the serialized data.\n pos: int, position to start in the serialized data.\n end: int, end position of the serialized data.\n\n Returns:\n Message object.\n \"\"\"\n # Guard against internal misuse, since this function is called internally\n # quite extensively, and its easy to accidentally pass bytes.\n assert isinstance(buffer, memoryview)\n self._Modified()\n field_dict = self._fields\n # pylint: disable=protected-access\n unknown_field_set = self._unknown_field_set\n while pos != end:\n (tag_bytes, new_pos) = local_ReadTag(buffer, pos)\n field_decoder, field_desc = decoders_by_tag.get(tag_bytes, (None, None))\n if field_decoder is None:\n if not self._unknown_fields: # pylint: disable=protected-access\n self._unknown_fields = [] # pylint: disable=protected-access\n if unknown_field_set is None:\n # pylint: disable=protected-access\n self._unknown_field_set = containers.UnknownFieldSet()\n # pylint: disable=protected-access\n unknown_field_set = self._unknown_field_set\n # pylint: disable=protected-access\n (tag, _) = decoder._DecodeVarint(tag_bytes, 0)\n field_number, wire_type = wire_format.UnpackTag(tag)\n if field_number == 0:\n raise message_mod.DecodeError('Field number 0 is illegal.')\n # TODO(jieluo): remove old_pos.\n old_pos = new_pos\n (data, new_pos) = decoder._DecodeUnknownField(\n buffer, new_pos, wire_type) # pylint: disable=protected-access\n if new_pos == -1:\n return pos\n # pylint: disable=protected-access\n unknown_field_set._add(field_number, wire_type, data)\n # TODO(jieluo): remove _unknown_fields.\n new_pos = local_SkipField(buffer, old_pos, end, tag_bytes)\n if new_pos == -1:\n return pos\n self._unknown_fields.append(\n (tag_bytes, buffer[old_pos:new_pos].tobytes()))\n pos = new_pos\n else:\n pos = field_decoder(buffer, new_pos, end, self, field_dict)\n if field_desc:\n self._UpdateOneofState(field_desc)\n return pos\n cls._InternalParse = InternalParse\n\n\ndef _AddIsInitializedMethod(message_descriptor, cls):\n \"\"\"Adds the IsInitialized and FindInitializationError methods to the\n protocol message class.\"\"\"\n\n required_fields = [field for field in message_descriptor.fields\n if field.label == _FieldDescriptor.LABEL_REQUIRED]\n\n def IsInitialized(self, errors=None):\n \"\"\"Checks if all required fields of a message are set.\n\n Args:\n errors: A list which, if provided, will be populated with the field\n paths of all missing required fields.\n\n Returns:\n True iff the specified message has all required fields set.\n \"\"\"\n\n # Performance is critical so we avoid HasField() and ListFields().\n\n for field in required_fields:\n if (field not in self._fields or\n (field.cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE and\n not self._fields[field]._is_present_in_parent)):\n if errors is not None:\n errors.extend(self.FindInitializationErrors())\n return False\n\n for field, value in list(self._fields.items()): # dict can change size!\n if field.cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE:\n if field.label == _FieldDescriptor.LABEL_REPEATED:\n if (field.message_type.has_options and\n field.message_type.GetOptions().map_entry):\n continue\n for element in value:\n if not element.IsInitialized():\n if errors is not None:\n errors.extend(self.FindInitializationErrors())\n return False\n elif value._is_present_in_parent and not value.IsInitialized():\n if errors is not None:\n errors.extend(self.FindInitializationErrors())\n return False\n\n return True\n\n cls.IsInitialized = IsInitialized\n\n def FindInitializationErrors(self):\n \"\"\"Finds required fields which are not initialized.\n\n Returns:\n A list of strings. Each string is a path to an uninitialized field from\n the top-level message, e.g. \"foo.bar[5].baz\".\n \"\"\"\n\n errors = [] # simplify things\n\n for field in required_fields:\n if not self.HasField(field.name):\n errors.append(field.name)\n\n for field, value in self.ListFields():\n if field.cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE:\n if field.is_extension:\n name = '(%s)' % field.full_name\n else:\n name = field.name\n\n if _IsMapField(field):\n if _IsMessageMapField(field):\n for key in value:\n element = value[key]\n prefix = '%s[%s].' % (name, key)\n sub_errors = element.FindInitializationErrors()\n errors += [prefix + error for error in sub_errors]\n else:\n # ScalarMaps can't have any initialization errors.\n pass\n elif field.label == _FieldDescriptor.LABEL_REPEATED:\n for i in range(len(value)):\n element = value[i]\n prefix = '%s[%d].' % (name, i)\n sub_errors = element.FindInitializationErrors()\n errors += [prefix + error for error in sub_errors]\n else:\n prefix = name + '.'\n sub_errors = value.FindInitializationErrors()\n errors += [prefix + error for error in sub_errors]\n\n return errors\n\n cls.FindInitializationErrors = FindInitializationErrors\n\n\ndef _FullyQualifiedClassName(klass):\n module = klass.__module__\n name = getattr(klass, '__qualname__', klass.__name__)\n if module in (None, 'builtins', '__builtin__'):\n return name\n return module + '.' + name\n\n\ndef _AddMergeFromMethod(cls):\n LABEL_REPEATED = _FieldDescriptor.LABEL_REPEATED\n CPPTYPE_MESSAGE = _FieldDescriptor.CPPTYPE_MESSAGE\n\n def MergeFrom(self, msg):\n if not isinstance(msg, cls):\n raise TypeError(\n 'Parameter to MergeFrom() must be instance of same class: '\n 'expected %s got %s.' % (_FullyQualifiedClassName(cls),\n _FullyQualifiedClassName(msg.__class__)))\n\n assert msg is not self\n self._Modified()\n\n fields = self._fields\n\n for field, value in msg._fields.items():\n if field.label == LABEL_REPEATED:\n field_value = fields.get(field)\n if field_value is None:\n # Construct a new object to represent this field.\n field_value = field._default_constructor(self)\n fields[field] = field_value\n field_value.MergeFrom(value)\n elif field.cpp_type == CPPTYPE_MESSAGE:\n if value._is_present_in_parent:\n field_value = fields.get(field)\n if field_value is None:\n # Construct a new object to represent this field.\n field_value = field._default_constructor(self)\n fields[field] = field_value\n field_value.MergeFrom(value)\n else:\n self._fields[field] = value\n if field.containing_oneof:\n self._UpdateOneofState(field)\n\n if msg._unknown_fields:\n if not self._unknown_fields:\n self._unknown_fields = []\n self._unknown_fields.extend(msg._unknown_fields)\n # pylint: disable=protected-access\n if self._unknown_field_set is None:\n self._unknown_field_set = containers.UnknownFieldSet()\n self._unknown_field_set._extend(msg._unknown_field_set)\n\n cls.MergeFrom = MergeFrom\n\n\ndef _AddWhichOneofMethod(message_descriptor, cls):\n def WhichOneof(self, oneof_name):\n \"\"\"Returns the name of the currently set field inside a oneof, or None.\"\"\"\n try:\n field = message_descriptor.oneofs_by_name[oneof_name]\n except KeyError:\n raise ValueError(\n 'Protocol message has no oneof \"%s\" field.' % oneof_name)\n\n nested_field = self._oneofs.get(field, None)\n if nested_field is not None and self.HasField(nested_field.name):\n return nested_field.name\n else:\n return None\n\n cls.WhichOneof = WhichOneof\n\n\ndef _Clear(self):\n # Clear fields.\n self._fields = {}\n self._unknown_fields = ()\n # pylint: disable=protected-access\n if self._unknown_field_set is not None:\n self._unknown_field_set._clear()\n self._unknown_field_set = None\n\n self._oneofs = {}\n self._Modified()\n\n\ndef _UnknownFields(self):\n if self._unknown_field_set is None: # pylint: disable=protected-access\n # pylint: disable=protected-access\n self._unknown_field_set = containers.UnknownFieldSet()\n return self._unknown_field_set # pylint: disable=protected-access\n\n\ndef _DiscardUnknownFields(self):\n self._unknown_fields = []\n self._unknown_field_set = None # pylint: disable=protected-access\n for field, value in self.ListFields():\n if field.cpp_type == _FieldDescriptor.CPPTYPE_MESSAGE:\n if _IsMapField(field):\n if _IsMessageMapField(field):\n for key in value:\n value[key].DiscardUnknownFields()\n elif field.label == _FieldDescriptor.LABEL_REPEATED:\n for sub_message in value:\n sub_message.DiscardUnknownFields()\n else:\n value.DiscardUnknownFields()\n\n\ndef _SetListener(self, listener):\n if listener is None:\n self._listener = message_listener_mod.NullMessageListener()\n else:\n self._listener = listener\n\n\ndef _AddMessageMethods(message_descriptor, cls):\n \"\"\"Adds implementations of all Message methods to cls.\"\"\"\n _AddListFieldsMethod(message_descriptor, cls)\n _AddHasFieldMethod(message_descriptor, cls)\n _AddClearFieldMethod(message_descriptor, cls)\n if message_descriptor.is_extendable:\n _AddClearExtensionMethod(cls)\n _AddHasExtensionMethod(cls)\n _AddEqualsMethod(message_descriptor, cls)\n _AddStrMethod(message_descriptor, cls)\n _AddReprMethod(message_descriptor, cls)\n _AddUnicodeMethod(message_descriptor, cls)\n _AddByteSizeMethod(message_descriptor, cls)\n _AddSerializeToStringMethod(message_descriptor, cls)\n _AddSerializePartialToStringMethod(message_descriptor, cls)\n _AddMergeFromStringMethod(message_descriptor, cls)\n _AddIsInitializedMethod(message_descriptor, cls)\n _AddMergeFromMethod(cls)\n _AddWhichOneofMethod(message_descriptor, cls)\n # Adds methods which do not depend on cls.\n cls.Clear = _Clear\n cls.UnknownFields = _UnknownFields\n cls.DiscardUnknownFields = _DiscardUnknownFields\n cls._SetListener = _SetListener\n\n\ndef _AddPrivateHelperMethods(message_descriptor, cls):\n \"\"\"Adds implementation of private helper methods to cls.\"\"\"\n\n def Modified(self):\n \"\"\"Sets the _cached_byte_size_dirty bit to true,\n and propagates this to our listener iff this was a state change.\n \"\"\"\n\n # Note: Some callers check _cached_byte_size_dirty before calling\n # _Modified() as an extra optimization. So, if this method is ever\n # changed such that it does stuff even when _cached_byte_size_dirty is\n # already true, the callers need to be updated.\n if not self._cached_byte_size_dirty:\n self._cached_byte_size_dirty = True\n self._listener_for_children.dirty = True\n self._is_present_in_parent = True\n self._listener.Modified()\n\n def _UpdateOneofState(self, field):\n \"\"\"Sets field as the active field in its containing oneof.\n\n Will also delete currently active field in the oneof, if it is different\n from the argument. Does not mark the message as modified.\n \"\"\"\n other_field = self._oneofs.setdefault(field.containing_oneof, field)\n if other_field is not field:\n del self._fields[other_field]\n self._oneofs[field.containing_oneof] = field\n\n cls._Modified = Modified\n cls.SetInParent = Modified\n cls._UpdateOneofState = _UpdateOneofState\n\n\nclass _Listener(object):\n\n \"\"\"MessageListener implementation that a parent message registers with its\n child message.\n\n In order to support semantics like:\n\n foo.bar.baz.qux = 23\n assert foo.HasField('bar')\n\n ...child objects must have back references to their parents.\n This helper class is at the heart of this support.\n \"\"\"\n\n def __init__(self, parent_message):\n \"\"\"Args:\n parent_message: The message whose _Modified() method we should call when\n we receive Modified() messages.\n \"\"\"\n # This listener establishes a back reference from a child (contained) object\n # to its parent (containing) object. We make this a weak reference to avoid\n # creating cyclic garbage when the client finishes with the 'parent' object\n # in the tree.\n if isinstance(parent_message, weakref.ProxyType):\n self._parent_message_weakref = parent_message\n else:\n self._parent_message_weakref = weakref.proxy(parent_message)\n\n # As an optimization, we also indicate directly on the listener whether\n # or not the parent message is dirty. This way we can avoid traversing\n # up the tree in the common case.\n self.dirty = False\n\n def Modified(self):\n if self.dirty:\n return\n try:\n # Propagate the signal to our parents iff this is the first field set.\n self._parent_message_weakref._Modified()\n except ReferenceError:\n # We can get here if a client has kept a reference to a child object,\n # and is now setting a field on it, but the child's parent has been\n # garbage-collected. This is not an error.\n pass\n\n\nclass _OneofListener(_Listener):\n \"\"\"Special listener implementation for setting composite oneof fields.\"\"\"\n\n def __init__(self, parent_message, field):\n \"\"\"Args:\n parent_message: The message whose _Modified() method we should call when\n we receive Modified() messages.\n field: The descriptor of the field being set in the parent message.\n \"\"\"\n super(_OneofListener, self).__init__(parent_message)\n self._field = field\n\n def Modified(self):\n \"\"\"Also updates the state of the containing oneof in the parent message.\"\"\"\n try:\n self._parent_message_weakref._UpdateOneofState(self._field)\n super(_OneofListener, self).Modified()\n except ReferenceError:\n pass\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/internal/type_checkers.py",
"content": "# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n\"\"\"Provides type checking routines.\n\nThis module defines type checking utilities in the forms of dictionaries:\n\nVALUE_CHECKERS: A dictionary of field types and a value validation object.\nTYPE_TO_BYTE_SIZE_FN: A dictionary with field types and a size computing\n function.\nTYPE_TO_SERIALIZE_METHOD: A dictionary with field types and serialization\n function.\nFIELD_TYPE_TO_WIRE_TYPE: A dictionary with field typed and their\n corresponding wire types.\nTYPE_TO_DESERIALIZE_METHOD: A dictionary with field types and deserialization\n function.\n\"\"\"\n\n__author__ = 'robinson@google.com (Will Robinson)'\n\nimport ctypes\nimport numbers\n\nfrom google.protobuf.internal import decoder\nfrom google.protobuf.internal import encoder\nfrom google.protobuf.internal import wire_format\nfrom google.protobuf import descriptor\n\n_FieldDescriptor = descriptor.FieldDescriptor\n\n\ndef TruncateToFourByteFloat(original):\n return ctypes.c_float(original).value\n\n\ndef ToShortestFloat(original):\n \"\"\"Returns the shortest float that has same value in wire.\"\"\"\n # All 4 byte floats have between 6 and 9 significant digits, so we\n # start with 6 as the lower bound.\n # It has to be iterative because use '.9g' directly can not get rid\n # of the noises for most values. For example if set a float_field=0.9\n # use '.9g' will print 0.899999976.\n precision = 6\n rounded = float('{0:.{1}g}'.format(original, precision))\n while TruncateToFourByteFloat(rounded) != original:\n precision += 1\n rounded = float('{0:.{1}g}'.format(original, precision))\n return rounded\n\n\ndef SupportsOpenEnums(field_descriptor):\n return field_descriptor.containing_type.syntax == 'proto3'\n\n\ndef GetTypeChecker(field):\n \"\"\"Returns a type checker for a message field of the specified types.\n\n Args:\n field: FieldDescriptor object for this field.\n\n Returns:\n An instance of TypeChecker which can be used to verify the types\n of values assigned to a field of the specified type.\n \"\"\"\n if (field.cpp_type == _FieldDescriptor.CPPTYPE_STRING and\n field.type == _FieldDescriptor.TYPE_STRING):\n return UnicodeValueChecker()\n if field.cpp_type == _FieldDescriptor.CPPTYPE_ENUM:\n if SupportsOpenEnums(field):\n # When open enums are supported, any int32 can be assigned.\n return _VALUE_CHECKERS[_FieldDescriptor.CPPTYPE_INT32]\n else:\n return EnumValueChecker(field.enum_type)\n return _VALUE_CHECKERS[field.cpp_type]\n\n\n# None of the typecheckers below make any attempt to guard against people\n# subclassing builtin types and doing weird things. We're not trying to\n# protect against malicious clients here, just people accidentally shooting\n# themselves in the foot in obvious ways.\nclass TypeChecker(object):\n\n \"\"\"Type checker used to catch type errors as early as possible\n when the client is setting scalar fields in protocol messages.\n \"\"\"\n\n def __init__(self, *acceptable_types):\n self._acceptable_types = acceptable_types\n\n def CheckValue(self, proposed_value):\n \"\"\"Type check the provided value and return it.\n\n The returned value might have been normalized to another type.\n \"\"\"\n if not isinstance(proposed_value, self._acceptable_types):\n message = ('%.1024r has type %s, but expected one of: %s' %\n (proposed_value, type(proposed_value), self._acceptable_types))\n raise TypeError(message)\n return proposed_value\n\n\nclass TypeCheckerWithDefault(TypeChecker):\n\n def __init__(self, default_value, *acceptable_types):\n TypeChecker.__init__(self, *acceptable_types)\n self._default_value = default_value\n\n def DefaultValue(self):\n return self._default_value\n\n\nclass BoolValueChecker(object):\n \"\"\"Type checker used for bool fields.\"\"\"\n\n def CheckValue(self, proposed_value):\n if not hasattr(proposed_value, '__index__') or (\n type(proposed_value).__module__ == 'numpy' and\n type(proposed_value).__name__ == 'ndarray'):\n message = ('%.1024r has type %s, but expected one of: %s' %\n (proposed_value, type(proposed_value), (bool, int)))\n raise TypeError(message)\n return bool(proposed_value)\n\n def DefaultValue(self):\n return False\n\n\n# IntValueChecker and its subclasses perform integer type-checks\n# and bounds-checks.\nclass IntValueChecker(object):\n\n \"\"\"Checker used for integer fields. Performs type-check and range check.\"\"\"\n\n def CheckValue(self, proposed_value):\n if not hasattr(proposed_value, '__index__') or (\n type(proposed_value).__module__ == 'numpy' and\n type(proposed_value).__name__ == 'ndarray'):\n message = ('%.1024r has type %s, but expected one of: %s' %\n (proposed_value, type(proposed_value), (int,)))\n raise TypeError(message)\n\n if not self._MIN <= int(proposed_value) <= self._MAX:\n raise ValueError('Value out of range: %d' % proposed_value)\n # We force all values to int to make alternate implementations where the\n # distinction is more significant (e.g. the C++ implementation) simpler.\n proposed_value = int(proposed_value)\n return proposed_value\n\n def DefaultValue(self):\n return 0\n\n\nclass EnumValueChecker(object):\n\n \"\"\"Checker used for enum fields. Performs type-check and range check.\"\"\"\n\n def __init__(self, enum_type):\n self._enum_type = enum_type\n\n def CheckValue(self, proposed_value):\n if not isinstance(proposed_value, numbers.Integral):\n message = ('%.1024r has type %s, but expected one of: %s' %\n (proposed_value, type(proposed_value), (int,)))\n raise TypeError(message)\n if int(proposed_value) not in self._enum_type.values_by_number:\n raise ValueError('Unknown enum value: %d' % proposed_value)\n return proposed_value\n\n def DefaultValue(self):\n return self._enum_type.values[0].number\n\n\nclass UnicodeValueChecker(object):\n\n \"\"\"Checker used for string fields.\n\n Always returns a unicode value, even if the input is of type str.\n \"\"\"\n\n def CheckValue(self, proposed_value):\n if not isinstance(proposed_value, (bytes, str)):\n message = ('%.1024r has type %s, but expected one of: %s' %\n (proposed_value, type(proposed_value), (bytes, str)))\n raise TypeError(message)\n\n # If the value is of type 'bytes' make sure that it is valid UTF-8 data.\n if isinstance(proposed_value, bytes):\n try:\n proposed_value = proposed_value.decode('utf-8')\n except UnicodeDecodeError:\n raise ValueError('%.1024r has type bytes, but isn\\'t valid UTF-8 '\n 'encoding. Non-UTF-8 strings must be converted to '\n 'unicode objects before being added.' %\n (proposed_value))\n else:\n try:\n proposed_value.encode('utf8')\n except UnicodeEncodeError:\n raise ValueError('%.1024r isn\\'t a valid unicode string and '\n 'can\\'t be encoded in UTF-8.'%\n (proposed_value))\n\n return proposed_value\n\n def DefaultValue(self):\n return u\"\"\n\n\nclass Int32ValueChecker(IntValueChecker):\n # We're sure to use ints instead of longs here since comparison may be more\n # efficient.\n _MIN = -2147483648\n _MAX = 2147483647\n\n\nclass Uint32ValueChecker(IntValueChecker):\n _MIN = 0\n _MAX = (1 << 32) - 1\n\n\nclass Int64ValueChecker(IntValueChecker):\n _MIN = -(1 << 63)\n _MAX = (1 << 63) - 1\n\n\nclass Uint64ValueChecker(IntValueChecker):\n _MIN = 0\n _MAX = (1 << 64) - 1\n\n\n# The max 4 bytes float is about 3.4028234663852886e+38\n_FLOAT_MAX = float.fromhex('0x1.fffffep+127')\n_FLOAT_MIN = -_FLOAT_MAX\n_INF = float('inf')\n_NEG_INF = float('-inf')\n\n\nclass DoubleValueChecker(object):\n \"\"\"Checker used for double fields.\n\n Performs type-check and range check.\n \"\"\"\n\n def CheckValue(self, proposed_value):\n \"\"\"Check and convert proposed_value to float.\"\"\"\n if (not hasattr(proposed_value, '__float__') and\n not hasattr(proposed_value, '__index__')) or (\n type(proposed_value).__module__ == 'numpy' and\n type(proposed_value).__name__ == 'ndarray'):\n message = ('%.1024r has type %s, but expected one of: int, float' %\n (proposed_value, type(proposed_value)))\n raise TypeError(message)\n return float(proposed_value)\n\n def DefaultValue(self):\n return 0.0\n\n\nclass FloatValueChecker(DoubleValueChecker):\n \"\"\"Checker used for float fields.\n\n Performs type-check and range check.\n\n Values exceeding a 32-bit float will be converted to inf/-inf.\n \"\"\"\n\n def CheckValue(self, proposed_value):\n \"\"\"Check and convert proposed_value to float.\"\"\"\n converted_value = super().CheckValue(proposed_value)\n # This inf rounding matches the C++ proto SafeDoubleToFloat logic.\n if converted_value > _FLOAT_MAX:\n return _INF\n if converted_value < _FLOAT_MIN:\n return _NEG_INF\n\n return TruncateToFourByteFloat(converted_value)\n\n# Type-checkers for all scalar CPPTYPEs.\n_VALUE_CHECKERS = {\n _FieldDescriptor.CPPTYPE_INT32: Int32ValueChecker(),\n _FieldDescriptor.CPPTYPE_INT64: Int64ValueChecker(),\n _FieldDescriptor.CPPTYPE_UINT32: Uint32ValueChecker(),\n _FieldDescriptor.CPPTYPE_UINT64: Uint64ValueChecker(),\n _FieldDescriptor.CPPTYPE_DOUBLE: DoubleValueChecker(),\n _FieldDescriptor.CPPTYPE_FLOAT: FloatValueChecker(),\n _FieldDescriptor.CPPTYPE_BOOL: BoolValueChecker(),\n _FieldDescriptor.CPPTYPE_STRING: TypeCheckerWithDefault(b'', bytes),\n}\n\n\n# Map from field type to a function F, such that F(field_num, value)\n# gives the total byte size for a value of the given type. This\n# byte size includes tag information and any other additional space\n# associated with serializing \"value\".\nTYPE_TO_BYTE_SIZE_FN = {\n _FieldDescriptor.TYPE_DOUBLE: wire_format.DoubleByteSize,\n _FieldDescriptor.TYPE_FLOAT: wire_format.FloatByteSize,\n _FieldDescriptor.TYPE_INT64: wire_format.Int64ByteSize,\n _FieldDescriptor.TYPE_UINT64: wire_format.UInt64ByteSize,\n _FieldDescriptor.TYPE_INT32: wire_format.Int32ByteSize,\n _FieldDescriptor.TYPE_FIXED64: wire_format.Fixed64ByteSize,\n _FieldDescriptor.TYPE_FIXED32: wire_format.Fixed32ByteSize,\n _FieldDescriptor.TYPE_BOOL: wire_format.BoolByteSize,\n _FieldDescriptor.TYPE_STRING: wire_format.StringByteSize,\n _FieldDescriptor.TYPE_GROUP: wire_format.GroupByteSize,\n _FieldDescriptor.TYPE_MESSAGE: wire_format.MessageByteSize,\n _FieldDescriptor.TYPE_BYTES: wire_format.BytesByteSize,\n _FieldDescriptor.TYPE_UINT32: wire_format.UInt32ByteSize,\n _FieldDescriptor.TYPE_ENUM: wire_format.EnumByteSize,\n _FieldDescriptor.TYPE_SFIXED32: wire_format.SFixed32ByteSize,\n _FieldDescriptor.TYPE_SFIXED64: wire_format.SFixed64ByteSize,\n _FieldDescriptor.TYPE_SINT32: wire_format.SInt32ByteSize,\n _FieldDescriptor.TYPE_SINT64: wire_format.SInt64ByteSize\n }\n\n\n# Maps from field types to encoder constructors.\nTYPE_TO_ENCODER = {\n _FieldDescriptor.TYPE_DOUBLE: encoder.DoubleEncoder,\n _FieldDescriptor.TYPE_FLOAT: encoder.FloatEncoder,\n _FieldDescriptor.TYPE_INT64: encoder.Int64Encoder,\n _FieldDescriptor.TYPE_UINT64: encoder.UInt64Encoder,\n _FieldDescriptor.TYPE_INT32: encoder.Int32Encoder,\n _FieldDescriptor.TYPE_FIXED64: encoder.Fixed64Encoder,\n _FieldDescriptor.TYPE_FIXED32: encoder.Fixed32Encoder,\n _FieldDescriptor.TYPE_BOOL: encoder.BoolEncoder,\n _FieldDescriptor.TYPE_STRING: encoder.StringEncoder,\n _FieldDescriptor.TYPE_GROUP: encoder.GroupEncoder,\n _FieldDescriptor.TYPE_MESSAGE: encoder.MessageEncoder,\n _FieldDescriptor.TYPE_BYTES: encoder.BytesEncoder,\n _FieldDescriptor.TYPE_UINT32: encoder.UInt32Encoder,\n _FieldDescriptor.TYPE_ENUM: encoder.EnumEncoder,\n _FieldDescriptor.TYPE_SFIXED32: encoder.SFixed32Encoder,\n _FieldDescriptor.TYPE_SFIXED64: encoder.SFixed64Encoder,\n _FieldDescriptor.TYPE_SINT32: encoder.SInt32Encoder,\n _FieldDescriptor.TYPE_SINT64: encoder.SInt64Encoder,\n }\n\n\n# Maps from field types to sizer constructors.\nTYPE_TO_SIZER = {\n _FieldDescriptor.TYPE_DOUBLE: encoder.DoubleSizer,\n _FieldDescriptor.TYPE_FLOAT: encoder.FloatSizer,\n _FieldDescriptor.TYPE_INT64: encoder.Int64Sizer,\n _FieldDescriptor.TYPE_UINT64: encoder.UInt64Sizer,\n _FieldDescriptor.TYPE_INT32: encoder.Int32Sizer,\n _FieldDescriptor.TYPE_FIXED64: encoder.Fixed64Sizer,\n _FieldDescriptor.TYPE_FIXED32: encoder.Fixed32Sizer,\n _FieldDescriptor.TYPE_BOOL: encoder.BoolSizer,\n _FieldDescriptor.TYPE_STRING: encoder.StringSizer,\n _FieldDescriptor.TYPE_GROUP: encoder.GroupSizer,\n _FieldDescriptor.TYPE_MESSAGE: encoder.MessageSizer,\n _FieldDescriptor.TYPE_BYTES: encoder.BytesSizer,\n _FieldDescriptor.TYPE_UINT32: encoder.UInt32Sizer,\n _FieldDescriptor.TYPE_ENUM: encoder.EnumSizer,\n _FieldDescriptor.TYPE_SFIXED32: encoder.SFixed32Sizer,\n _FieldDescriptor.TYPE_SFIXED64: encoder.SFixed64Sizer,\n _FieldDescriptor.TYPE_SINT32: encoder.SInt32Sizer,\n _FieldDescriptor.TYPE_SINT64: encoder.SInt64Sizer,\n }\n\n\n# Maps from field type to a decoder constructor.\nTYPE_TO_DECODER = {\n _FieldDescriptor.TYPE_DOUBLE: decoder.DoubleDecoder,\n _FieldDescriptor.TYPE_FLOAT: decoder.FloatDecoder,\n _FieldDescriptor.TYPE_INT64: decoder.Int64Decoder,\n _FieldDescriptor.TYPE_UINT64: decoder.UInt64Decoder,\n _FieldDescriptor.TYPE_INT32: decoder.Int32Decoder,\n _FieldDescriptor.TYPE_FIXED64: decoder.Fixed64Decoder,\n _FieldDescriptor.TYPE_FIXED32: decoder.Fixed32Decoder,\n _FieldDescriptor.TYPE_BOOL: decoder.BoolDecoder,\n _FieldDescriptor.TYPE_STRING: decoder.StringDecoder,\n _FieldDescriptor.TYPE_GROUP: decoder.GroupDecoder,\n _FieldDescriptor.TYPE_MESSAGE: decoder.MessageDecoder,\n _FieldDescriptor.TYPE_BYTES: decoder.BytesDecoder,\n _FieldDescriptor.TYPE_UINT32: decoder.UInt32Decoder,\n _FieldDescriptor.TYPE_ENUM: decoder.EnumDecoder,\n _FieldDescriptor.TYPE_SFIXED32: decoder.SFixed32Decoder,\n _FieldDescriptor.TYPE_SFIXED64: decoder.SFixed64Decoder,\n _FieldDescriptor.TYPE_SINT32: decoder.SInt32Decoder,\n _FieldDescriptor.TYPE_SINT64: decoder.SInt64Decoder,\n }\n\n# Maps from field type to expected wiretype.\nFIELD_TYPE_TO_WIRE_TYPE = {\n _FieldDescriptor.TYPE_DOUBLE: wire_format.WIRETYPE_FIXED64,\n _FieldDescriptor.TYPE_FLOAT: wire_format.WIRETYPE_FIXED32,\n _FieldDescriptor.TYPE_INT64: wire_format.WIRETYPE_VARINT,\n _FieldDescriptor.TYPE_UINT64: wire_format.WIRETYPE_VARINT,\n _FieldDescriptor.TYPE_INT32: wire_format.WIRETYPE_VARINT,\n _FieldDescriptor.TYPE_FIXED64: wire_format.WIRETYPE_FIXED64,\n _FieldDescriptor.TYPE_FIXED32: wire_format.WIRETYPE_FIXED32,\n _FieldDescriptor.TYPE_BOOL: wire_format.WIRETYPE_VARINT,\n _FieldDescriptor.TYPE_STRING:\n wire_format.WIRETYPE_LENGTH_DELIMITED,\n _FieldDescriptor.TYPE_GROUP: wire_format.WIRETYPE_START_GROUP,\n _FieldDescriptor.TYPE_MESSAGE:\n wire_format.WIRETYPE_LENGTH_DELIMITED,\n _FieldDescriptor.TYPE_BYTES:\n wire_format.WIRETYPE_LENGTH_DELIMITED,\n _FieldDescriptor.TYPE_UINT32: wire_format.WIRETYPE_VARINT,\n _FieldDescriptor.TYPE_ENUM: wire_format.WIRETYPE_VARINT,\n _FieldDescriptor.TYPE_SFIXED32: wire_format.WIRETYPE_FIXED32,\n _FieldDescriptor.TYPE_SFIXED64: wire_format.WIRETYPE_FIXED64,\n _FieldDescriptor.TYPE_SINT32: wire_format.WIRETYPE_VARINT,\n _FieldDescriptor.TYPE_SINT64: wire_format.WIRETYPE_VARINT,\n }\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/internal/well_known_types.py",
"content": "# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n\"\"\"Contains well known classes.\n\nThis files defines well known classes which need extra maintenance including:\n - Any\n - Duration\n - FieldMask\n - Struct\n - Timestamp\n\"\"\"\n\n__author__ = 'jieluo@google.com (Jie Luo)'\n\nimport calendar\nimport collections.abc\nimport datetime\n\nfrom google.protobuf.descriptor import FieldDescriptor\n\n_TIMESTAMPFOMAT = '%Y-%m-%dT%H:%M:%S'\n_NANOS_PER_SECOND = 1000000000\n_NANOS_PER_MILLISECOND = 1000000\n_NANOS_PER_MICROSECOND = 1000\n_MILLIS_PER_SECOND = 1000\n_MICROS_PER_SECOND = 1000000\n_SECONDS_PER_DAY = 24 * 3600\n_DURATION_SECONDS_MAX = 315576000000\n\n\nclass Any(object):\n \"\"\"Class for Any Message type.\"\"\"\n\n __slots__ = ()\n\n def Pack(self, msg, type_url_prefix='type.googleapis.com/',\n deterministic=None):\n \"\"\"Packs the specified message into current Any message.\"\"\"\n if len(type_url_prefix) < 1 or type_url_prefix[-1] != '/':\n self.type_url = '%s/%s' % (type_url_prefix, msg.DESCRIPTOR.full_name)\n else:\n self.type_url = '%s%s' % (type_url_prefix, msg.DESCRIPTOR.full_name)\n self.value = msg.SerializeToString(deterministic=deterministic)\n\n def Unpack(self, msg):\n \"\"\"Unpacks the current Any message into specified message.\"\"\"\n descriptor = msg.DESCRIPTOR\n if not self.Is(descriptor):\n return False\n msg.ParseFromString(self.value)\n return True\n\n def TypeName(self):\n \"\"\"Returns the protobuf type name of the inner message.\"\"\"\n # Only last part is to be used: b/25630112\n return self.type_url.split('/')[-1]\n\n def Is(self, descriptor):\n \"\"\"Checks if this Any represents the given protobuf type.\"\"\"\n return '/' in self.type_url and self.TypeName() == descriptor.full_name\n\n\n_EPOCH_DATETIME_NAIVE = datetime.datetime.utcfromtimestamp(0)\n_EPOCH_DATETIME_AWARE = datetime.datetime.fromtimestamp(\n 0, tz=datetime.timezone.utc)\n\n\nclass Timestamp(object):\n \"\"\"Class for Timestamp message type.\"\"\"\n\n __slots__ = ()\n\n def ToJsonString(self):\n \"\"\"Converts Timestamp to RFC 3339 date string format.\n\n Returns:\n A string converted from timestamp. The string is always Z-normalized\n and uses 3, 6 or 9 fractional digits as required to represent the\n exact time. Example of the return format: '1972-01-01T10:00:20.021Z'\n \"\"\"\n nanos = self.nanos % _NANOS_PER_SECOND\n total_sec = self.seconds + (self.nanos - nanos) // _NANOS_PER_SECOND\n seconds = total_sec % _SECONDS_PER_DAY\n days = (total_sec - seconds) // _SECONDS_PER_DAY\n dt = datetime.datetime(1970, 1, 1) + datetime.timedelta(days, seconds)\n\n result = dt.isoformat()\n if (nanos % 1e9) == 0:\n # If there are 0 fractional digits, the fractional\n # point '.' should be omitted when serializing.\n return result + 'Z'\n if (nanos % 1e6) == 0:\n # Serialize 3 fractional digits.\n return result + '.%03dZ' % (nanos / 1e6)\n if (nanos % 1e3) == 0:\n # Serialize 6 fractional digits.\n return result + '.%06dZ' % (nanos / 1e3)\n # Serialize 9 fractional digits.\n return result + '.%09dZ' % nanos\n\n def FromJsonString(self, value):\n \"\"\"Parse a RFC 3339 date string format to Timestamp.\n\n Args:\n value: A date string. Any fractional digits (or none) and any offset are\n accepted as long as they fit into nano-seconds precision.\n Example of accepted format: '1972-01-01T10:00:20.021-05:00'\n\n Raises:\n ValueError: On parsing problems.\n \"\"\"\n if not isinstance(value, str):\n raise ValueError('Timestamp JSON value not a string: {!r}'.format(value))\n timezone_offset = value.find('Z')\n if timezone_offset == -1:\n timezone_offset = value.find('+')\n if timezone_offset == -1:\n timezone_offset = value.rfind('-')\n if timezone_offset == -1:\n raise ValueError(\n 'Failed to parse timestamp: missing valid timezone offset.')\n time_value = value[0:timezone_offset]\n # Parse datetime and nanos.\n point_position = time_value.find('.')\n if point_position == -1:\n second_value = time_value\n nano_value = ''\n else:\n second_value = time_value[:point_position]\n nano_value = time_value[point_position + 1:]\n if 't' in second_value:\n raise ValueError(\n 'time data \\'{0}\\' does not match format \\'%Y-%m-%dT%H:%M:%S\\', '\n 'lowercase \\'t\\' is not accepted'.format(second_value))\n date_object = datetime.datetime.strptime(second_value, _TIMESTAMPFOMAT)\n td = date_object - datetime.datetime(1970, 1, 1)\n seconds = td.seconds + td.days * _SECONDS_PER_DAY\n if len(nano_value) > 9:\n raise ValueError(\n 'Failed to parse Timestamp: nanos {0} more than '\n '9 fractional digits.'.format(nano_value))\n if nano_value:\n nanos = round(float('0.' + nano_value) * 1e9)\n else:\n nanos = 0\n # Parse timezone offsets.\n if value[timezone_offset] == 'Z':\n if len(value) != timezone_offset + 1:\n raise ValueError('Failed to parse timestamp: invalid trailing'\n ' data {0}.'.format(value))\n else:\n timezone = value[timezone_offset:]\n pos = timezone.find(':')\n if pos == -1:\n raise ValueError(\n 'Invalid timezone offset value: {0}.'.format(timezone))\n if timezone[0] == '+':\n seconds -= (int(timezone[1:pos])*60+int(timezone[pos+1:]))*60\n else:\n seconds += (int(timezone[1:pos])*60+int(timezone[pos+1:]))*60\n # Set seconds and nanos\n self.seconds = int(seconds)\n self.nanos = int(nanos)\n\n def GetCurrentTime(self):\n \"\"\"Get the current UTC into Timestamp.\"\"\"\n self.FromDatetime(datetime.datetime.utcnow())\n\n def ToNanoseconds(self):\n \"\"\"Converts Timestamp to nanoseconds since epoch.\"\"\"\n return self.seconds * _NANOS_PER_SECOND + self.nanos\n\n def ToMicroseconds(self):\n \"\"\"Converts Timestamp to microseconds since epoch.\"\"\"\n return (self.seconds * _MICROS_PER_SECOND +\n self.nanos // _NANOS_PER_MICROSECOND)\n\n def ToMilliseconds(self):\n \"\"\"Converts Timestamp to milliseconds since epoch.\"\"\"\n return (self.seconds * _MILLIS_PER_SECOND +\n self.nanos // _NANOS_PER_MILLISECOND)\n\n def ToSeconds(self):\n \"\"\"Converts Timestamp to seconds since epoch.\"\"\"\n return self.seconds\n\n def FromNanoseconds(self, nanos):\n \"\"\"Converts nanoseconds since epoch to Timestamp.\"\"\"\n self.seconds = nanos // _NANOS_PER_SECOND\n self.nanos = nanos % _NANOS_PER_SECOND\n\n def FromMicroseconds(self, micros):\n \"\"\"Converts microseconds since epoch to Timestamp.\"\"\"\n self.seconds = micros // _MICROS_PER_SECOND\n self.nanos = (micros % _MICROS_PER_SECOND) * _NANOS_PER_MICROSECOND\n\n def FromMilliseconds(self, millis):\n \"\"\"Converts milliseconds since epoch to Timestamp.\"\"\"\n self.seconds = millis // _MILLIS_PER_SECOND\n self.nanos = (millis % _MILLIS_PER_SECOND) * _NANOS_PER_MILLISECOND\n\n def FromSeconds(self, seconds):\n \"\"\"Converts seconds since epoch to Timestamp.\"\"\"\n self.seconds = seconds\n self.nanos = 0\n\n def ToDatetime(self, tzinfo=None):\n \"\"\"Converts Timestamp to a datetime.\n\n Args:\n tzinfo: A datetime.tzinfo subclass; defaults to None.\n\n Returns:\n If tzinfo is None, returns a timezone-naive UTC datetime (with no timezone\n information, i.e. not aware that it's UTC).\n\n Otherwise, returns a timezone-aware datetime in the input timezone.\n \"\"\"\n delta = datetime.timedelta(\n seconds=self.seconds,\n microseconds=_RoundTowardZero(self.nanos, _NANOS_PER_MICROSECOND))\n if tzinfo is None:\n return _EPOCH_DATETIME_NAIVE + delta\n else:\n return _EPOCH_DATETIME_AWARE.astimezone(tzinfo) + delta\n\n def FromDatetime(self, dt):\n \"\"\"Converts datetime to Timestamp.\n\n Args:\n dt: A datetime. If it's timezone-naive, it's assumed to be in UTC.\n \"\"\"\n # Using this guide: http://wiki.python.org/moin/WorkingWithTime\n # And this conversion guide: http://docs.python.org/library/time.html\n\n # Turn the date parameter into a tuple (struct_time) that can then be\n # manipulated into a long value of seconds. During the conversion from\n # struct_time to long, the source date in UTC, and so it follows that the\n # correct transformation is calendar.timegm()\n self.seconds = calendar.timegm(dt.utctimetuple())\n self.nanos = dt.microsecond * _NANOS_PER_MICROSECOND\n\n\nclass Duration(object):\n \"\"\"Class for Duration message type.\"\"\"\n\n __slots__ = ()\n\n def ToJsonString(self):\n \"\"\"Converts Duration to string format.\n\n Returns:\n A string converted from self. The string format will contains\n 3, 6, or 9 fractional digits depending on the precision required to\n represent the exact Duration value. For example: \"1s\", \"1.010s\",\n \"1.000000100s\", \"-3.100s\"\n \"\"\"\n _CheckDurationValid(self.seconds, self.nanos)\n if self.seconds < 0 or self.nanos < 0:\n result = '-'\n seconds = - self.seconds + int((0 - self.nanos) // 1e9)\n nanos = (0 - self.nanos) % 1e9\n else:\n result = ''\n seconds = self.seconds + int(self.nanos // 1e9)\n nanos = self.nanos % 1e9\n result += '%d' % seconds\n if (nanos % 1e9) == 0:\n # If there are 0 fractional digits, the fractional\n # point '.' should be omitted when serializing.\n return result + 's'\n if (nanos % 1e6) == 0:\n # Serialize 3 fractional digits.\n return result + '.%03ds' % (nanos / 1e6)\n if (nanos % 1e3) == 0:\n # Serialize 6 fractional digits.\n return result + '.%06ds' % (nanos / 1e3)\n # Serialize 9 fractional digits.\n return result + '.%09ds' % nanos\n\n def FromJsonString(self, value):\n \"\"\"Converts a string to Duration.\n\n Args:\n value: A string to be converted. The string must end with 's'. Any\n fractional digits (or none) are accepted as long as they fit into\n precision. For example: \"1s\", \"1.01s\", \"1.0000001s\", \"-3.100s\n\n Raises:\n ValueError: On parsing problems.\n \"\"\"\n if not isinstance(value, str):\n raise ValueError('Duration JSON value not a string: {!r}'.format(value))\n if len(value) < 1 or value[-1] != 's':\n raise ValueError(\n 'Duration must end with letter \"s\": {0}.'.format(value))\n try:\n pos = value.find('.')\n if pos == -1:\n seconds = int(value[:-1])\n nanos = 0\n else:\n seconds = int(value[:pos])\n if value[0] == '-':\n nanos = int(round(float('-0{0}'.format(value[pos: -1])) *1e9))\n else:\n nanos = int(round(float('0{0}'.format(value[pos: -1])) *1e9))\n _CheckDurationValid(seconds, nanos)\n self.seconds = seconds\n self.nanos = nanos\n except ValueError as e:\n raise ValueError(\n 'Couldn\\'t parse duration: {0} : {1}.'.format(value, e))\n\n def ToNanoseconds(self):\n \"\"\"Converts a Duration to nanoseconds.\"\"\"\n return self.seconds * _NANOS_PER_SECOND + self.nanos\n\n def ToMicroseconds(self):\n \"\"\"Converts a Duration to microseconds.\"\"\"\n micros = _RoundTowardZero(self.nanos, _NANOS_PER_MICROSECOND)\n return self.seconds * _MICROS_PER_SECOND + micros\n\n def ToMilliseconds(self):\n \"\"\"Converts a Duration to milliseconds.\"\"\"\n millis = _RoundTowardZero(self.nanos, _NANOS_PER_MILLISECOND)\n return self.seconds * _MILLIS_PER_SECOND + millis\n\n def ToSeconds(self):\n \"\"\"Converts a Duration to seconds.\"\"\"\n return self.seconds\n\n def FromNanoseconds(self, nanos):\n \"\"\"Converts nanoseconds to Duration.\"\"\"\n self._NormalizeDuration(nanos // _NANOS_PER_SECOND,\n nanos % _NANOS_PER_SECOND)\n\n def FromMicroseconds(self, micros):\n \"\"\"Converts microseconds to Duration.\"\"\"\n self._NormalizeDuration(\n micros // _MICROS_PER_SECOND,\n (micros % _MICROS_PER_SECOND) * _NANOS_PER_MICROSECOND)\n\n def FromMilliseconds(self, millis):\n \"\"\"Converts milliseconds to Duration.\"\"\"\n self._NormalizeDuration(\n millis // _MILLIS_PER_SECOND,\n (millis % _MILLIS_PER_SECOND) * _NANOS_PER_MILLISECOND)\n\n def FromSeconds(self, seconds):\n \"\"\"Converts seconds to Duration.\"\"\"\n self.seconds = seconds\n self.nanos = 0\n\n def ToTimedelta(self):\n \"\"\"Converts Duration to timedelta.\"\"\"\n return datetime.timedelta(\n seconds=self.seconds, microseconds=_RoundTowardZero(\n self.nanos, _NANOS_PER_MICROSECOND))\n\n def FromTimedelta(self, td):\n \"\"\"Converts timedelta to Duration.\"\"\"\n self._NormalizeDuration(td.seconds + td.days * _SECONDS_PER_DAY,\n td.microseconds * _NANOS_PER_MICROSECOND)\n\n def _NormalizeDuration(self, seconds, nanos):\n \"\"\"Set Duration by seconds and nanos.\"\"\"\n # Force nanos to be negative if the duration is negative.\n if seconds < 0 and nanos > 0:\n seconds += 1\n nanos -= _NANOS_PER_SECOND\n self.seconds = seconds\n self.nanos = nanos\n\n\ndef _CheckDurationValid(seconds, nanos):\n if seconds < -_DURATION_SECONDS_MAX or seconds > _DURATION_SECONDS_MAX:\n raise ValueError(\n 'Duration is not valid: Seconds {0} must be in range '\n '[-315576000000, 315576000000].'.format(seconds))\n if nanos <= -_NANOS_PER_SECOND or nanos >= _NANOS_PER_SECOND:\n raise ValueError(\n 'Duration is not valid: Nanos {0} must be in range '\n '[-999999999, 999999999].'.format(nanos))\n if (nanos < 0 and seconds > 0) or (nanos > 0 and seconds < 0):\n raise ValueError(\n 'Duration is not valid: Sign mismatch.')\n\n\ndef _RoundTowardZero(value, divider):\n \"\"\"Truncates the remainder part after division.\"\"\"\n # For some languages, the sign of the remainder is implementation\n # dependent if any of the operands is negative. Here we enforce\n # \"rounded toward zero\" semantics. For example, for (-5) / 2 an\n # implementation may give -3 as the result with the remainder being\n # 1. This function ensures we always return -2 (closer to zero).\n result = value // divider\n remainder = value % divider\n if result < 0 and remainder > 0:\n return result + 1\n else:\n return result\n\n\nclass FieldMask(object):\n \"\"\"Class for FieldMask message type.\"\"\"\n\n __slots__ = ()\n\n def ToJsonString(self):\n \"\"\"Converts FieldMask to string according to proto3 JSON spec.\"\"\"\n camelcase_paths = []\n for path in self.paths:\n camelcase_paths.append(_SnakeCaseToCamelCase(path))\n return ','.join(camelcase_paths)\n\n def FromJsonString(self, value):\n \"\"\"Converts string to FieldMask according to proto3 JSON spec.\"\"\"\n if not isinstance(value, str):\n raise ValueError('FieldMask JSON value not a string: {!r}'.format(value))\n self.Clear()\n if value:\n for path in value.split(','):\n self.paths.append(_CamelCaseToSnakeCase(path))\n\n def IsValidForDescriptor(self, message_descriptor):\n \"\"\"Checks whether the FieldMask is valid for Message Descriptor.\"\"\"\n for path in self.paths:\n if not _IsValidPath(message_descriptor, path):\n return False\n return True\n\n def AllFieldsFromDescriptor(self, message_descriptor):\n \"\"\"Gets all direct fields of Message Descriptor to FieldMask.\"\"\"\n self.Clear()\n for field in message_descriptor.fields:\n self.paths.append(field.name)\n\n def CanonicalFormFromMask(self, mask):\n \"\"\"Converts a FieldMask to the canonical form.\n\n Removes paths that are covered by another path. For example,\n \"foo.bar\" is covered by \"foo\" and will be removed if \"foo\"\n is also in the FieldMask. Then sorts all paths in alphabetical order.\n\n Args:\n mask: The original FieldMask to be converted.\n \"\"\"\n tree = _FieldMaskTree(mask)\n tree.ToFieldMask(self)\n\n def Union(self, mask1, mask2):\n \"\"\"Merges mask1 and mask2 into this FieldMask.\"\"\"\n _CheckFieldMaskMessage(mask1)\n _CheckFieldMaskMessage(mask2)\n tree = _FieldMaskTree(mask1)\n tree.MergeFromFieldMask(mask2)\n tree.ToFieldMask(self)\n\n def Intersect(self, mask1, mask2):\n \"\"\"Intersects mask1 and mask2 into this FieldMask.\"\"\"\n _CheckFieldMaskMessage(mask1)\n _CheckFieldMaskMessage(mask2)\n tree = _FieldMaskTree(mask1)\n intersection = _FieldMaskTree()\n for path in mask2.paths:\n tree.IntersectPath(path, intersection)\n intersection.ToFieldMask(self)\n\n def MergeMessage(\n self, source, destination,\n replace_message_field=False, replace_repeated_field=False):\n \"\"\"Merges fields specified in FieldMask from source to destination.\n\n Args:\n source: Source message.\n destination: The destination message to be merged into.\n replace_message_field: Replace message field if True. Merge message\n field if False.\n replace_repeated_field: Replace repeated field if True. Append\n elements of repeated field if False.\n \"\"\"\n tree = _FieldMaskTree(self)\n tree.MergeMessage(\n source, destination, replace_message_field, replace_repeated_field)\n\n\ndef _IsValidPath(message_descriptor, path):\n \"\"\"Checks whether the path is valid for Message Descriptor.\"\"\"\n parts = path.split('.')\n last = parts.pop()\n for name in parts:\n field = message_descriptor.fields_by_name.get(name)\n if (field is None or\n field.label == FieldDescriptor.LABEL_REPEATED or\n field.type != FieldDescriptor.TYPE_MESSAGE):\n return False\n message_descriptor = field.message_type\n return last in message_descriptor.fields_by_name\n\n\ndef _CheckFieldMaskMessage(message):\n \"\"\"Raises ValueError if message is not a FieldMask.\"\"\"\n message_descriptor = message.DESCRIPTOR\n if (message_descriptor.name != 'FieldMask' or\n message_descriptor.file.name != 'google/protobuf/field_mask.proto'):\n raise ValueError('Message {0} is not a FieldMask.'.format(\n message_descriptor.full_name))\n\n\ndef _SnakeCaseToCamelCase(path_name):\n \"\"\"Converts a path name from snake_case to camelCase.\"\"\"\n result = []\n after_underscore = False\n for c in path_name:\n if c.isupper():\n raise ValueError(\n 'Fail to print FieldMask to Json string: Path name '\n '{0} must not contain uppercase letters.'.format(path_name))\n if after_underscore:\n if c.islower():\n result.append(c.upper())\n after_underscore = False\n else:\n raise ValueError(\n 'Fail to print FieldMask to Json string: The '\n 'character after a \"_\" must be a lowercase letter '\n 'in path name {0}.'.format(path_name))\n elif c == '_':\n after_underscore = True\n else:\n result += c\n\n if after_underscore:\n raise ValueError('Fail to print FieldMask to Json string: Trailing \"_\" '\n 'in path name {0}.'.format(path_name))\n return ''.join(result)\n\n\ndef _CamelCaseToSnakeCase(path_name):\n \"\"\"Converts a field name from camelCase to snake_case.\"\"\"\n result = []\n for c in path_name:\n if c == '_':\n raise ValueError('Fail to parse FieldMask: Path name '\n '{0} must not contain \"_\"s.'.format(path_name))\n if c.isupper():\n result += '_'\n result += c.lower()\n else:\n result += c\n return ''.join(result)\n\n\nclass _FieldMaskTree(object):\n \"\"\"Represents a FieldMask in a tree structure.\n\n For example, given a FieldMask \"foo.bar,foo.baz,bar.baz\",\n the FieldMaskTree will be:\n [_root] -+- foo -+- bar\n | |\n | +- baz\n |\n +- bar --- baz\n In the tree, each leaf node represents a field path.\n \"\"\"\n\n __slots__ = ('_root',)\n\n def __init__(self, field_mask=None):\n \"\"\"Initializes the tree by FieldMask.\"\"\"\n self._root = {}\n if field_mask:\n self.MergeFromFieldMask(field_mask)\n\n def MergeFromFieldMask(self, field_mask):\n \"\"\"Merges a FieldMask to the tree.\"\"\"\n for path in field_mask.paths:\n self.AddPath(path)\n\n def AddPath(self, path):\n \"\"\"Adds a field path into the tree.\n\n If the field path to add is a sub-path of an existing field path\n in the tree (i.e., a leaf node), it means the tree already matches\n the given path so nothing will be added to the tree. If the path\n matches an existing non-leaf node in the tree, that non-leaf node\n will be turned into a leaf node with all its children removed because\n the path matches all the node's children. Otherwise, a new path will\n be added.\n\n Args:\n path: The field path to add.\n \"\"\"\n node = self._root\n for name in path.split('.'):\n if name not in node:\n node[name] = {}\n elif not node[name]:\n # Pre-existing empty node implies we already have this entire tree.\n return\n node = node[name]\n # Remove any sub-trees we might have had.\n node.clear()\n\n def ToFieldMask(self, field_mask):\n \"\"\"Converts the tree to a FieldMask.\"\"\"\n field_mask.Clear()\n _AddFieldPaths(self._root, '', field_mask)\n\n def IntersectPath(self, path, intersection):\n \"\"\"Calculates the intersection part of a field path with this tree.\n\n Args:\n path: The field path to calculates.\n intersection: The out tree to record the intersection part.\n \"\"\"\n node = self._root\n for name in path.split('.'):\n if name not in node:\n return\n elif not node[name]:\n intersection.AddPath(path)\n return\n node = node[name]\n intersection.AddLeafNodes(path, node)\n\n def AddLeafNodes(self, prefix, node):\n \"\"\"Adds leaf nodes begin with prefix to this tree.\"\"\"\n if not node:\n self.AddPath(prefix)\n for name in node:\n child_path = prefix + '.' + name\n self.AddLeafNodes(child_path, node[name])\n\n def MergeMessage(\n self, source, destination,\n replace_message, replace_repeated):\n \"\"\"Merge all fields specified by this tree from source to destination.\"\"\"\n _MergeMessage(\n self._root, source, destination, replace_message, replace_repeated)\n\n\ndef _StrConvert(value):\n \"\"\"Converts value to str if it is not.\"\"\"\n # This file is imported by c extension and some methods like ClearField\n # requires string for the field name. py2/py3 has different text\n # type and may use unicode.\n if not isinstance(value, str):\n return value.encode('utf-8')\n return value\n\n\ndef _MergeMessage(\n node, source, destination, replace_message, replace_repeated):\n \"\"\"Merge all fields specified by a sub-tree from source to destination.\"\"\"\n source_descriptor = source.DESCRIPTOR\n for name in node:\n child = node[name]\n field = source_descriptor.fields_by_name[name]\n if field is None:\n raise ValueError('Error: Can\\'t find field {0} in message {1}.'.format(\n name, source_descriptor.full_name))\n if child:\n # Sub-paths are only allowed for singular message fields.\n if (field.label == FieldDescriptor.LABEL_REPEATED or\n field.cpp_type != FieldDescriptor.CPPTYPE_MESSAGE):\n raise ValueError('Error: Field {0} in message {1} is not a singular '\n 'message field and cannot have sub-fields.'.format(\n name, source_descriptor.full_name))\n if source.HasField(name):\n _MergeMessage(\n child, getattr(source, name), getattr(destination, name),\n replace_message, replace_repeated)\n continue\n if field.label == FieldDescriptor.LABEL_REPEATED:\n if replace_repeated:\n destination.ClearField(_StrConvert(name))\n repeated_source = getattr(source, name)\n repeated_destination = getattr(destination, name)\n repeated_destination.MergeFrom(repeated_source)\n else:\n if field.cpp_type == FieldDescriptor.CPPTYPE_MESSAGE:\n if replace_message:\n destination.ClearField(_StrConvert(name))\n if source.HasField(name):\n getattr(destination, name).MergeFrom(getattr(source, name))\n else:\n setattr(destination, name, getattr(source, name))\n\n\ndef _AddFieldPaths(node, prefix, field_mask):\n \"\"\"Adds the field paths descended from node to field_mask.\"\"\"\n if not node and prefix:\n field_mask.paths.append(prefix)\n return\n for name in sorted(node):\n if prefix:\n child_path = prefix + '.' + name\n else:\n child_path = name\n _AddFieldPaths(node[name], child_path, field_mask)\n\n\ndef _SetStructValue(struct_value, value):\n if value is None:\n struct_value.null_value = 0\n elif isinstance(value, bool):\n # Note: this check must come before the number check because in Python\n # True and False are also considered numbers.\n struct_value.bool_value = value\n elif isinstance(value, str):\n struct_value.string_value = value\n elif isinstance(value, (int, float)):\n struct_value.number_value = value\n elif isinstance(value, (dict, Struct)):\n struct_value.struct_value.Clear()\n struct_value.struct_value.update(value)\n elif isinstance(value, (list, ListValue)):\n struct_value.list_value.Clear()\n struct_value.list_value.extend(value)\n else:\n raise ValueError('Unexpected type')\n\n\ndef _GetStructValue(struct_value):\n which = struct_value.WhichOneof('kind')\n if which == 'struct_value':\n return struct_value.struct_value\n elif which == 'null_value':\n return None\n elif which == 'number_value':\n return struct_value.number_value\n elif which == 'string_value':\n return struct_value.string_value\n elif which == 'bool_value':\n return struct_value.bool_value\n elif which == 'list_value':\n return struct_value.list_value\n elif which is None:\n raise ValueError('Value not set')\n\n\nclass Struct(object):\n \"\"\"Class for Struct message type.\"\"\"\n\n __slots__ = ()\n\n def __getitem__(self, key):\n return _GetStructValue(self.fields[key])\n\n def __contains__(self, item):\n return item in self.fields\n\n def __setitem__(self, key, value):\n _SetStructValue(self.fields[key], value)\n\n def __delitem__(self, key):\n del self.fields[key]\n\n def __len__(self):\n return len(self.fields)\n\n def __iter__(self):\n return iter(self.fields)\n\n def keys(self): # pylint: disable=invalid-name\n return self.fields.keys()\n\n def values(self): # pylint: disable=invalid-name\n return [self[key] for key in self]\n\n def items(self): # pylint: disable=invalid-name\n return [(key, self[key]) for key in self]\n\n def get_or_create_list(self, key):\n \"\"\"Returns a list for this key, creating if it didn't exist already.\"\"\"\n if not self.fields[key].HasField('list_value'):\n # Clear will mark list_value modified which will indeed create a list.\n self.fields[key].list_value.Clear()\n return self.fields[key].list_value\n\n def get_or_create_struct(self, key):\n \"\"\"Returns a struct for this key, creating if it didn't exist already.\"\"\"\n if not self.fields[key].HasField('struct_value'):\n # Clear will mark struct_value modified which will indeed create a struct.\n self.fields[key].struct_value.Clear()\n return self.fields[key].struct_value\n\n def update(self, dictionary): # pylint: disable=invalid-name\n for key, value in dictionary.items():\n _SetStructValue(self.fields[key], value)\n\ncollections.abc.MutableMapping.register(Struct)\n\n\nclass ListValue(object):\n \"\"\"Class for ListValue message type.\"\"\"\n\n __slots__ = ()\n\n def __len__(self):\n return len(self.values)\n\n def append(self, value):\n _SetStructValue(self.values.add(), value)\n\n def extend(self, elem_seq):\n for value in elem_seq:\n self.append(value)\n\n def __getitem__(self, index):\n \"\"\"Retrieves item by the specified index.\"\"\"\n return _GetStructValue(self.values.__getitem__(index))\n\n def __setitem__(self, index, value):\n _SetStructValue(self.values.__getitem__(index), value)\n\n def __delitem__(self, key):\n del self.values[key]\n\n def items(self):\n for i in range(len(self)):\n yield self[i]\n\n def add_struct(self):\n \"\"\"Appends and returns a struct value as the next value in the list.\"\"\"\n struct_value = self.values.add().struct_value\n # Clear will mark struct_value modified which will indeed create a struct.\n struct_value.Clear()\n return struct_value\n\n def add_list(self):\n \"\"\"Appends and returns a list value as the next value in the list.\"\"\"\n list_value = self.values.add().list_value\n # Clear will mark list_value modified which will indeed create a list.\n list_value.Clear()\n return list_value\n\ncollections.abc.MutableSequence.register(ListValue)\n\n\nWKTBASES = {\n 'google.protobuf.Any': Any,\n 'google.protobuf.Duration': Duration,\n 'google.protobuf.FieldMask': FieldMask,\n 'google.protobuf.ListValue': ListValue,\n 'google.protobuf.Struct': Struct,\n 'google.protobuf.Timestamp': Timestamp,\n}\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/internal/wire_format.py",
"content": "# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n\"\"\"Constants and static functions to support protocol buffer wire format.\"\"\"\n\n__author__ = 'robinson@google.com (Will Robinson)'\n\nimport struct\nfrom google.protobuf import descriptor\nfrom google.protobuf import message\n\n\nTAG_TYPE_BITS = 3 # Number of bits used to hold type info in a proto tag.\nTAG_TYPE_MASK = (1 << TAG_TYPE_BITS) - 1 # 0x7\n\n# These numbers identify the wire type of a protocol buffer value.\n# We use the least-significant TAG_TYPE_BITS bits of the varint-encoded\n# tag-and-type to store one of these WIRETYPE_* constants.\n# These values must match WireType enum in google/protobuf/wire_format.h.\nWIRETYPE_VARINT = 0\nWIRETYPE_FIXED64 = 1\nWIRETYPE_LENGTH_DELIMITED = 2\nWIRETYPE_START_GROUP = 3\nWIRETYPE_END_GROUP = 4\nWIRETYPE_FIXED32 = 5\n_WIRETYPE_MAX = 5\n\n\n# Bounds for various integer types.\nINT32_MAX = int((1 << 31) - 1)\nINT32_MIN = int(-(1 << 31))\nUINT32_MAX = (1 << 32) - 1\n\nINT64_MAX = (1 << 63) - 1\nINT64_MIN = -(1 << 63)\nUINT64_MAX = (1 << 64) - 1\n\n# \"struct\" format strings that will encode/decode the specified formats.\nFORMAT_UINT32_LITTLE_ENDIAN = '> TAG_TYPE_BITS), (tag & TAG_TYPE_MASK)\n\n\ndef ZigZagEncode(value):\n \"\"\"ZigZag Transform: Encodes signed integers so that they can be\n effectively used with varint encoding. See wire_format.h for\n more details.\n \"\"\"\n if value >= 0:\n return value << 1\n return (value << 1) ^ (~0)\n\n\ndef ZigZagDecode(value):\n \"\"\"Inverse of ZigZagEncode().\"\"\"\n if not value & 0x1:\n return value >> 1\n return (value >> 1) ^ (~0)\n\n\n\n# The *ByteSize() functions below return the number of bytes required to\n# serialize \"field number + type\" information and then serialize the value.\n\n\ndef Int32ByteSize(field_number, int32):\n return Int64ByteSize(field_number, int32)\n\n\ndef Int32ByteSizeNoTag(int32):\n return _VarUInt64ByteSizeNoTag(0xffffffffffffffff & int32)\n\n\ndef Int64ByteSize(field_number, int64):\n # Have to convert to uint before calling UInt64ByteSize().\n return UInt64ByteSize(field_number, 0xffffffffffffffff & int64)\n\n\ndef UInt32ByteSize(field_number, uint32):\n return UInt64ByteSize(field_number, uint32)\n\n\ndef UInt64ByteSize(field_number, uint64):\n return TagByteSize(field_number) + _VarUInt64ByteSizeNoTag(uint64)\n\n\ndef SInt32ByteSize(field_number, int32):\n return UInt32ByteSize(field_number, ZigZagEncode(int32))\n\n\ndef SInt64ByteSize(field_number, int64):\n return UInt64ByteSize(field_number, ZigZagEncode(int64))\n\n\ndef Fixed32ByteSize(field_number, fixed32):\n return TagByteSize(field_number) + 4\n\n\ndef Fixed64ByteSize(field_number, fixed64):\n return TagByteSize(field_number) + 8\n\n\ndef SFixed32ByteSize(field_number, sfixed32):\n return TagByteSize(field_number) + 4\n\n\ndef SFixed64ByteSize(field_number, sfixed64):\n return TagByteSize(field_number) + 8\n\n\ndef FloatByteSize(field_number, flt):\n return TagByteSize(field_number) + 4\n\n\ndef DoubleByteSize(field_number, double):\n return TagByteSize(field_number) + 8\n\n\ndef BoolByteSize(field_number, b):\n return TagByteSize(field_number) + 1\n\n\ndef EnumByteSize(field_number, enum):\n return UInt32ByteSize(field_number, enum)\n\n\ndef StringByteSize(field_number, string):\n return BytesByteSize(field_number, string.encode('utf-8'))\n\n\ndef BytesByteSize(field_number, b):\n return (TagByteSize(field_number)\n + _VarUInt64ByteSizeNoTag(len(b))\n + len(b))\n\n\ndef GroupByteSize(field_number, message):\n return (2 * TagByteSize(field_number) # START and END group.\n + message.ByteSize())\n\n\ndef MessageByteSize(field_number, message):\n return (TagByteSize(field_number)\n + _VarUInt64ByteSizeNoTag(message.ByteSize())\n + message.ByteSize())\n\n\ndef MessageSetItemByteSize(field_number, msg):\n # First compute the sizes of the tags.\n # There are 2 tags for the beginning and ending of the repeated group, that\n # is field number 1, one with field number 2 (type_id) and one with field\n # number 3 (message).\n total_size = (2 * TagByteSize(1) + TagByteSize(2) + TagByteSize(3))\n\n # Add the number of bytes for type_id.\n total_size += _VarUInt64ByteSizeNoTag(field_number)\n\n message_size = msg.ByteSize()\n\n # The number of bytes for encoding the length of the message.\n total_size += _VarUInt64ByteSizeNoTag(message_size)\n\n # The size of the message.\n total_size += message_size\n return total_size\n\n\ndef TagByteSize(field_number):\n \"\"\"Returns the bytes required to serialize a tag with this field number.\"\"\"\n # Just pass in type 0, since the type won't affect the tag+type size.\n return _VarUInt64ByteSizeNoTag(PackTag(field_number, 0))\n\n\n# Private helper function for the *ByteSize() functions above.\n\ndef _VarUInt64ByteSizeNoTag(uint64):\n \"\"\"Returns the number of bytes required to serialize a single varint\n using boundary value comparisons. (unrolled loop optimization -WPierce)\n uint64 must be unsigned.\n \"\"\"\n if uint64 <= 0x7f: return 1\n if uint64 <= 0x3fff: return 2\n if uint64 <= 0x1fffff: return 3\n if uint64 <= 0xfffffff: return 4\n if uint64 <= 0x7ffffffff: return 5\n if uint64 <= 0x3ffffffffff: return 6\n if uint64 <= 0x1ffffffffffff: return 7\n if uint64 <= 0xffffffffffffff: return 8\n if uint64 <= 0x7fffffffffffffff: return 9\n if uint64 > UINT64_MAX:\n raise message.EncodeError('Value out of range: %d' % uint64)\n return 10\n\n\nNON_PACKABLE_TYPES = (\n descriptor.FieldDescriptor.TYPE_STRING,\n descriptor.FieldDescriptor.TYPE_GROUP,\n descriptor.FieldDescriptor.TYPE_MESSAGE,\n descriptor.FieldDescriptor.TYPE_BYTES\n)\n\n\ndef IsTypePackable(field_type):\n \"\"\"Return true iff packable = true is valid for fields of this type.\n\n Args:\n field_type: a FieldDescriptor::Type value.\n\n Returns:\n True iff fields of this type are packable.\n \"\"\"\n return field_type not in NON_PACKABLE_TYPES\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/json_format.py",
"content": "# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n\"\"\"Contains routines for printing protocol messages in JSON format.\n\nSimple usage example:\n\n # Create a proto object and serialize it to a json format string.\n message = my_proto_pb2.MyMessage(foo='bar')\n json_string = json_format.MessageToJson(message)\n\n # Parse a json format string to proto object.\n message = json_format.Parse(json_string, my_proto_pb2.MyMessage())\n\"\"\"\n\n__author__ = 'jieluo@google.com (Jie Luo)'\n\n\nimport base64\nfrom collections import OrderedDict\nimport json\nimport math\nfrom operator import methodcaller\nimport re\nimport sys\n\nfrom google.protobuf.internal import type_checkers\nfrom google.protobuf import descriptor\nfrom google.protobuf import symbol_database\n\n\n_TIMESTAMPFOMAT = '%Y-%m-%dT%H:%M:%S'\n_INT_TYPES = frozenset([descriptor.FieldDescriptor.CPPTYPE_INT32,\n descriptor.FieldDescriptor.CPPTYPE_UINT32,\n descriptor.FieldDescriptor.CPPTYPE_INT64,\n descriptor.FieldDescriptor.CPPTYPE_UINT64])\n_INT64_TYPES = frozenset([descriptor.FieldDescriptor.CPPTYPE_INT64,\n descriptor.FieldDescriptor.CPPTYPE_UINT64])\n_FLOAT_TYPES = frozenset([descriptor.FieldDescriptor.CPPTYPE_FLOAT,\n descriptor.FieldDescriptor.CPPTYPE_DOUBLE])\n_INFINITY = 'Infinity'\n_NEG_INFINITY = '-Infinity'\n_NAN = 'NaN'\n\n_UNPAIRED_SURROGATE_PATTERN = re.compile(\n u'[\\ud800-\\udbff](?![\\udc00-\\udfff])|(? self.max_recursion_depth:\n raise ParseError('Message too deep. Max recursion depth is {0}'.format(\n self.max_recursion_depth))\n message_descriptor = message.DESCRIPTOR\n full_name = message_descriptor.full_name\n if not path:\n path = message_descriptor.name\n if _IsWrapperMessage(message_descriptor):\n self._ConvertWrapperMessage(value, message, path)\n elif full_name in _WKTJSONMETHODS:\n methodcaller(_WKTJSONMETHODS[full_name][1], value, message, path)(self)\n else:\n self._ConvertFieldValuePair(value, message, path)\n self.recursion_depth -= 1\n\n def _ConvertFieldValuePair(self, js, message, path):\n \"\"\"Convert field value pairs into regular message.\n\n Args:\n js: A JSON object to convert the field value pairs.\n message: A regular protocol message to record the data.\n path: parent path to log parse error info.\n\n Raises:\n ParseError: In case of problems converting.\n \"\"\"\n names = []\n message_descriptor = message.DESCRIPTOR\n fields_by_json_name = dict((f.json_name, f)\n for f in message_descriptor.fields)\n for name in js:\n try:\n field = fields_by_json_name.get(name, None)\n if not field:\n field = message_descriptor.fields_by_name.get(name, None)\n if not field and _VALID_EXTENSION_NAME.match(name):\n if not message_descriptor.is_extendable:\n raise ParseError(\n 'Message type {0} does not have extensions at {1}'.format(\n message_descriptor.full_name, path))\n identifier = name[1:-1] # strip [] brackets\n # pylint: disable=protected-access\n field = message.Extensions._FindExtensionByName(identifier)\n # pylint: enable=protected-access\n if not field:\n # Try looking for extension by the message type name, dropping the\n # field name following the final . separator in full_name.\n identifier = '.'.join(identifier.split('.')[:-1])\n # pylint: disable=protected-access\n field = message.Extensions._FindExtensionByName(identifier)\n # pylint: enable=protected-access\n if not field:\n if self.ignore_unknown_fields:\n continue\n raise ParseError(\n ('Message type \"{0}\" has no field named \"{1}\" at \"{2}\".\\n'\n ' Available Fields(except extensions): \"{3}\"').format(\n message_descriptor.full_name, name, path,\n [f.json_name for f in message_descriptor.fields]))\n if name in names:\n raise ParseError('Message type \"{0}\" should not have multiple '\n '\"{1}\" fields at \"{2}\".'.format(\n message.DESCRIPTOR.full_name, name, path))\n names.append(name)\n value = js[name]\n # Check no other oneof field is parsed.\n if field.containing_oneof is not None and value is not None:\n oneof_name = field.containing_oneof.name\n if oneof_name in names:\n raise ParseError('Message type \"{0}\" should not have multiple '\n '\"{1}\" oneof fields at \"{2}\".'.format(\n message.DESCRIPTOR.full_name, oneof_name,\n path))\n names.append(oneof_name)\n\n if value is None:\n if (field.cpp_type == descriptor.FieldDescriptor.CPPTYPE_MESSAGE\n and field.message_type.full_name == 'google.protobuf.Value'):\n sub_message = getattr(message, field.name)\n sub_message.null_value = 0\n elif (field.cpp_type == descriptor.FieldDescriptor.CPPTYPE_ENUM\n and field.enum_type.full_name == 'google.protobuf.NullValue'):\n setattr(message, field.name, 0)\n else:\n message.ClearField(field.name)\n continue\n\n # Parse field value.\n if _IsMapEntry(field):\n message.ClearField(field.name)\n self._ConvertMapFieldValue(value, message, field,\n '{0}.{1}'.format(path, name))\n elif field.label == descriptor.FieldDescriptor.LABEL_REPEATED:\n message.ClearField(field.name)\n if not isinstance(value, list):\n raise ParseError('repeated field {0} must be in [] which is '\n '{1} at {2}'.format(name, value, path))\n if field.cpp_type == descriptor.FieldDescriptor.CPPTYPE_MESSAGE:\n # Repeated message field.\n for index, item in enumerate(value):\n sub_message = getattr(message, field.name).add()\n # None is a null_value in Value.\n if (item is None and\n sub_message.DESCRIPTOR.full_name != 'google.protobuf.Value'):\n raise ParseError('null is not allowed to be used as an element'\n ' in a repeated field at {0}.{1}[{2}]'.format(\n path, name, index))\n self.ConvertMessage(item, sub_message,\n '{0}.{1}[{2}]'.format(path, name, index))\n else:\n # Repeated scalar field.\n for index, item in enumerate(value):\n if item is None:\n raise ParseError('null is not allowed to be used as an element'\n ' in a repeated field at {0}.{1}[{2}]'.format(\n path, name, index))\n getattr(message, field.name).append(\n _ConvertScalarFieldValue(\n item, field, '{0}.{1}[{2}]'.format(path, name, index)))\n elif field.cpp_type == descriptor.FieldDescriptor.CPPTYPE_MESSAGE:\n if field.is_extension:\n sub_message = message.Extensions[field]\n else:\n sub_message = getattr(message, field.name)\n sub_message.SetInParent()\n self.ConvertMessage(value, sub_message, '{0}.{1}'.format(path, name))\n else:\n if field.is_extension:\n message.Extensions[field] = _ConvertScalarFieldValue(\n value, field, '{0}.{1}'.format(path, name))\n else:\n setattr(\n message, field.name,\n _ConvertScalarFieldValue(value, field,\n '{0}.{1}'.format(path, name)))\n except ParseError as e:\n if field and field.containing_oneof is None:\n raise ParseError('Failed to parse {0} field: {1}.'.format(name, e))\n else:\n raise ParseError(str(e))\n except ValueError as e:\n raise ParseError('Failed to parse {0} field: {1}.'.format(name, e))\n except TypeError as e:\n raise ParseError('Failed to parse {0} field: {1}.'.format(name, e))\n\n def _ConvertAnyMessage(self, value, message, path):\n \"\"\"Convert a JSON representation into Any message.\"\"\"\n if isinstance(value, dict) and not value:\n return\n try:\n type_url = value['@type']\n except KeyError:\n raise ParseError(\n '@type is missing when parsing any message at {0}'.format(path))\n\n try:\n sub_message = _CreateMessageFromTypeUrl(type_url, self.descriptor_pool)\n except TypeError as e:\n raise ParseError('{0} at {1}'.format(e, path))\n message_descriptor = sub_message.DESCRIPTOR\n full_name = message_descriptor.full_name\n if _IsWrapperMessage(message_descriptor):\n self._ConvertWrapperMessage(value['value'], sub_message,\n '{0}.value'.format(path))\n elif full_name in _WKTJSONMETHODS:\n methodcaller(_WKTJSONMETHODS[full_name][1], value['value'], sub_message,\n '{0}.value'.format(path))(\n self)\n else:\n del value['@type']\n self._ConvertFieldValuePair(value, sub_message, path)\n value['@type'] = type_url\n # Sets Any message\n message.value = sub_message.SerializeToString()\n message.type_url = type_url\n\n def _ConvertGenericMessage(self, value, message, path):\n \"\"\"Convert a JSON representation into message with FromJsonString.\"\"\"\n # Duration, Timestamp, FieldMask have a FromJsonString method to do the\n # conversion. Users can also call the method directly.\n try:\n message.FromJsonString(value)\n except ValueError as e:\n raise ParseError('{0} at {1}'.format(e, path))\n\n def _ConvertValueMessage(self, value, message, path):\n \"\"\"Convert a JSON representation into Value message.\"\"\"\n if isinstance(value, dict):\n self._ConvertStructMessage(value, message.struct_value, path)\n elif isinstance(value, list):\n self._ConvertListValueMessage(value, message.list_value, path)\n elif value is None:\n message.null_value = 0\n elif isinstance(value, bool):\n message.bool_value = value\n elif isinstance(value, str):\n message.string_value = value\n elif isinstance(value, _INT_OR_FLOAT):\n message.number_value = value\n else:\n raise ParseError('Value {0} has unexpected type {1} at {2}'.format(\n value, type(value), path))\n\n def _ConvertListValueMessage(self, value, message, path):\n \"\"\"Convert a JSON representation into ListValue message.\"\"\"\n if not isinstance(value, list):\n raise ParseError('ListValue must be in [] which is {0} at {1}'.format(\n value, path))\n message.ClearField('values')\n for index, item in enumerate(value):\n self._ConvertValueMessage(item, message.values.add(),\n '{0}[{1}]'.format(path, index))\n\n def _ConvertStructMessage(self, value, message, path):\n \"\"\"Convert a JSON representation into Struct message.\"\"\"\n if not isinstance(value, dict):\n raise ParseError('Struct must be in a dict which is {0} at {1}'.format(\n value, path))\n # Clear will mark the struct as modified so it will be created even if\n # there are no values.\n message.Clear()\n for key in value:\n self._ConvertValueMessage(value[key], message.fields[key],\n '{0}.{1}'.format(path, key))\n return\n\n def _ConvertWrapperMessage(self, value, message, path):\n \"\"\"Convert a JSON representation into Wrapper message.\"\"\"\n field = message.DESCRIPTOR.fields_by_name['value']\n setattr(\n message, 'value',\n _ConvertScalarFieldValue(value, field, path='{0}.value'.format(path)))\n\n def _ConvertMapFieldValue(self, value, message, field, path):\n \"\"\"Convert map field value for a message map field.\n\n Args:\n value: A JSON object to convert the map field value.\n message: A protocol message to record the converted data.\n field: The descriptor of the map field to be converted.\n path: parent path to log parse error info.\n\n Raises:\n ParseError: In case of convert problems.\n \"\"\"\n if not isinstance(value, dict):\n raise ParseError(\n 'Map field {0} must be in a dict which is {1} at {2}'.format(\n field.name, value, path))\n key_field = field.message_type.fields_by_name['key']\n value_field = field.message_type.fields_by_name['value']\n for key in value:\n key_value = _ConvertScalarFieldValue(key, key_field,\n '{0}.key'.format(path), True)\n if value_field.cpp_type == descriptor.FieldDescriptor.CPPTYPE_MESSAGE:\n self.ConvertMessage(value[key],\n getattr(message, field.name)[key_value],\n '{0}[{1}]'.format(path, key_value))\n else:\n getattr(message, field.name)[key_value] = _ConvertScalarFieldValue(\n value[key], value_field, path='{0}[{1}]'.format(path, key_value))\n\n\ndef _ConvertScalarFieldValue(value, field, path, require_str=False):\n \"\"\"Convert a single scalar field value.\n\n Args:\n value: A scalar value to convert the scalar field value.\n field: The descriptor of the field to convert.\n path: parent path to log parse error info.\n require_str: If True, the field value must be a str.\n\n Returns:\n The converted scalar field value\n\n Raises:\n ParseError: In case of convert problems.\n \"\"\"\n try:\n if field.cpp_type in _INT_TYPES:\n return _ConvertInteger(value)\n elif field.cpp_type in _FLOAT_TYPES:\n return _ConvertFloat(value, field)\n elif field.cpp_type == descriptor.FieldDescriptor.CPPTYPE_BOOL:\n return _ConvertBool(value, require_str)\n elif field.cpp_type == descriptor.FieldDescriptor.CPPTYPE_STRING:\n if field.type == descriptor.FieldDescriptor.TYPE_BYTES:\n if isinstance(value, str):\n encoded = value.encode('utf-8')\n else:\n encoded = value\n # Add extra padding '='\n padded_value = encoded + b'=' * (4 - len(encoded) % 4)\n return base64.urlsafe_b64decode(padded_value)\n else:\n # Checking for unpaired surrogates appears to be unreliable,\n # depending on the specific Python version, so we check manually.\n if _UNPAIRED_SURROGATE_PATTERN.search(value):\n raise ParseError('Unpaired surrogate')\n return value\n elif field.cpp_type == descriptor.FieldDescriptor.CPPTYPE_ENUM:\n # Convert an enum value.\n enum_value = field.enum_type.values_by_name.get(value, None)\n if enum_value is None:\n try:\n number = int(value)\n enum_value = field.enum_type.values_by_number.get(number, None)\n except ValueError:\n raise ParseError('Invalid enum value {0} for enum type {1}'.format(\n value, field.enum_type.full_name))\n if enum_value is None:\n if field.file.syntax == 'proto3':\n # Proto3 accepts unknown enums.\n return number\n raise ParseError('Invalid enum value {0} for enum type {1}'.format(\n value, field.enum_type.full_name))\n return enum_value.number\n except ParseError as e:\n raise ParseError('{0} at {1}'.format(e, path))\n\n\ndef _ConvertInteger(value):\n \"\"\"Convert an integer.\n\n Args:\n value: A scalar value to convert.\n\n Returns:\n The integer value.\n\n Raises:\n ParseError: If an integer couldn't be consumed.\n \"\"\"\n if isinstance(value, float) and not value.is_integer():\n raise ParseError('Couldn\\'t parse integer: {0}'.format(value))\n\n if isinstance(value, str) and value.find(' ') != -1:\n raise ParseError('Couldn\\'t parse integer: \"{0}\"'.format(value))\n\n if isinstance(value, bool):\n raise ParseError('Bool value {0} is not acceptable for '\n 'integer field'.format(value))\n\n return int(value)\n\n\ndef _ConvertFloat(value, field):\n \"\"\"Convert an floating point number.\"\"\"\n if isinstance(value, float):\n if math.isnan(value):\n raise ParseError('Couldn\\'t parse NaN, use quoted \"NaN\" instead')\n if math.isinf(value):\n if value > 0:\n raise ParseError('Couldn\\'t parse Infinity or value too large, '\n 'use quoted \"Infinity\" instead')\n else:\n raise ParseError('Couldn\\'t parse -Infinity or value too small, '\n 'use quoted \"-Infinity\" instead')\n if field.cpp_type == descriptor.FieldDescriptor.CPPTYPE_FLOAT:\n # pylint: disable=protected-access\n if value > type_checkers._FLOAT_MAX:\n raise ParseError('Float value too large')\n # pylint: disable=protected-access\n if value < type_checkers._FLOAT_MIN:\n raise ParseError('Float value too small')\n if value == 'nan':\n raise ParseError('Couldn\\'t parse float \"nan\", use \"NaN\" instead')\n try:\n # Assume Python compatible syntax.\n return float(value)\n except ValueError:\n # Check alternative spellings.\n if value == _NEG_INFINITY:\n return float('-inf')\n elif value == _INFINITY:\n return float('inf')\n elif value == _NAN:\n return float('nan')\n else:\n raise ParseError('Couldn\\'t parse float: {0}'.format(value))\n\n\ndef _ConvertBool(value, require_str):\n \"\"\"Convert a boolean value.\n\n Args:\n value: A scalar value to convert.\n require_str: If True, value must be a str.\n\n Returns:\n The bool parsed.\n\n Raises:\n ParseError: If a boolean value couldn't be consumed.\n \"\"\"\n if require_str:\n if value == 'true':\n return True\n elif value == 'false':\n return False\n else:\n raise ParseError('Expected \"true\" or \"false\", not {0}'.format(value))\n\n if not isinstance(value, bool):\n raise ParseError('Expected true or false without quotes')\n return value\n\n_WKTJSONMETHODS = {\n 'google.protobuf.Any': ['_AnyMessageToJsonObject',\n '_ConvertAnyMessage'],\n 'google.protobuf.Duration': ['_GenericMessageToJsonObject',\n '_ConvertGenericMessage'],\n 'google.protobuf.FieldMask': ['_GenericMessageToJsonObject',\n '_ConvertGenericMessage'],\n 'google.protobuf.ListValue': ['_ListValueMessageToJsonObject',\n '_ConvertListValueMessage'],\n 'google.protobuf.Struct': ['_StructMessageToJsonObject',\n '_ConvertStructMessage'],\n 'google.protobuf.Timestamp': ['_GenericMessageToJsonObject',\n '_ConvertGenericMessage'],\n 'google.protobuf.Value': ['_ValueMessageToJsonObject',\n '_ConvertValueMessage']\n}\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/message.py",
"content": "# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n# TODO(robinson): We should just make these methods all \"pure-virtual\" and move\n# all implementation out, into reflection.py for now.\n\n\n\"\"\"Contains an abstract base class for protocol messages.\"\"\"\n\n__author__ = 'robinson@google.com (Will Robinson)'\n\nclass Error(Exception):\n \"\"\"Base error type for this module.\"\"\"\n pass\n\n\nclass DecodeError(Error):\n \"\"\"Exception raised when deserializing messages.\"\"\"\n pass\n\n\nclass EncodeError(Error):\n \"\"\"Exception raised when serializing messages.\"\"\"\n pass\n\n\nclass Message(object):\n\n \"\"\"Abstract base class for protocol messages.\n\n Protocol message classes are almost always generated by the protocol\n compiler. These generated types subclass Message and implement the methods\n shown below.\n \"\"\"\n\n # TODO(robinson): Link to an HTML document here.\n\n # TODO(robinson): Document that instances of this class will also\n # have an Extensions attribute with __getitem__ and __setitem__.\n # Again, not sure how to best convey this.\n\n # TODO(robinson): Document that the class must also have a static\n # RegisterExtension(extension_field) method.\n # Not sure how to best express at this point.\n\n # TODO(robinson): Document these fields and methods.\n\n __slots__ = []\n\n #: The :class:`google.protobuf.descriptor.Descriptor` for this message type.\n DESCRIPTOR = None\n\n def __deepcopy__(self, memo=None):\n clone = type(self)()\n clone.MergeFrom(self)\n return clone\n\n def __eq__(self, other_msg):\n \"\"\"Recursively compares two messages by value and structure.\"\"\"\n raise NotImplementedError\n\n def __ne__(self, other_msg):\n # Can't just say self != other_msg, since that would infinitely recurse. :)\n return not self == other_msg\n\n def __hash__(self):\n raise TypeError('unhashable object')\n\n def __str__(self):\n \"\"\"Outputs a human-readable representation of the message.\"\"\"\n raise NotImplementedError\n\n def __unicode__(self):\n \"\"\"Outputs a human-readable representation of the message.\"\"\"\n raise NotImplementedError\n\n def MergeFrom(self, other_msg):\n \"\"\"Merges the contents of the specified message into current message.\n\n This method merges the contents of the specified message into the current\n message. Singular fields that are set in the specified message overwrite\n the corresponding fields in the current message. Repeated fields are\n appended. Singular sub-messages and groups are recursively merged.\n\n Args:\n other_msg (Message): A message to merge into the current message.\n \"\"\"\n raise NotImplementedError\n\n def CopyFrom(self, other_msg):\n \"\"\"Copies the content of the specified message into the current message.\n\n The method clears the current message and then merges the specified\n message using MergeFrom.\n\n Args:\n other_msg (Message): A message to copy into the current one.\n \"\"\"\n if self is other_msg:\n return\n self.Clear()\n self.MergeFrom(other_msg)\n\n def Clear(self):\n \"\"\"Clears all data that was set in the message.\"\"\"\n raise NotImplementedError\n\n def SetInParent(self):\n \"\"\"Mark this as present in the parent.\n\n This normally happens automatically when you assign a field of a\n sub-message, but sometimes you want to make the sub-message\n present while keeping it empty. If you find yourself using this,\n you may want to reconsider your design.\n \"\"\"\n raise NotImplementedError\n\n def IsInitialized(self):\n \"\"\"Checks if the message is initialized.\n\n Returns:\n bool: The method returns True if the message is initialized (i.e. all of\n its required fields are set).\n \"\"\"\n raise NotImplementedError\n\n # TODO(robinson): MergeFromString() should probably return None and be\n # implemented in terms of a helper that returns the # of bytes read. Our\n # deserialization routines would use the helper when recursively\n # deserializing, but the end user would almost always just want the no-return\n # MergeFromString().\n\n def MergeFromString(self, serialized):\n \"\"\"Merges serialized protocol buffer data into this message.\n\n When we find a field in `serialized` that is already present\n in this message:\n\n - If it's a \"repeated\" field, we append to the end of our list.\n - Else, if it's a scalar, we overwrite our field.\n - Else, (it's a nonrepeated composite), we recursively merge\n into the existing composite.\n\n Args:\n serialized (bytes): Any object that allows us to call\n ``memoryview(serialized)`` to access a string of bytes using the\n buffer interface.\n\n Returns:\n int: The number of bytes read from `serialized`.\n For non-group messages, this will always be `len(serialized)`,\n but for messages which are actually groups, this will\n generally be less than `len(serialized)`, since we must\n stop when we reach an ``END_GROUP`` tag. Note that if\n we *do* stop because of an ``END_GROUP`` tag, the number\n of bytes returned does not include the bytes\n for the ``END_GROUP`` tag information.\n\n Raises:\n DecodeError: if the input cannot be parsed.\n \"\"\"\n # TODO(robinson): Document handling of unknown fields.\n # TODO(robinson): When we switch to a helper, this will return None.\n raise NotImplementedError\n\n def ParseFromString(self, serialized):\n \"\"\"Parse serialized protocol buffer data into this message.\n\n Like :func:`MergeFromString()`, except we clear the object first.\n\n Raises:\n message.DecodeError if the input cannot be parsed.\n \"\"\"\n self.Clear()\n return self.MergeFromString(serialized)\n\n def SerializeToString(self, **kwargs):\n \"\"\"Serializes the protocol message to a binary string.\n\n Keyword Args:\n deterministic (bool): If true, requests deterministic serialization\n of the protobuf, with predictable ordering of map keys.\n\n Returns:\n A binary string representation of the message if all of the required\n fields in the message are set (i.e. the message is initialized).\n\n Raises:\n EncodeError: if the message isn't initialized (see :func:`IsInitialized`).\n \"\"\"\n raise NotImplementedError\n\n def SerializePartialToString(self, **kwargs):\n \"\"\"Serializes the protocol message to a binary string.\n\n This method is similar to SerializeToString but doesn't check if the\n message is initialized.\n\n Keyword Args:\n deterministic (bool): If true, requests deterministic serialization\n of the protobuf, with predictable ordering of map keys.\n\n Returns:\n bytes: A serialized representation of the partial message.\n \"\"\"\n raise NotImplementedError\n\n # TODO(robinson): Decide whether we like these better\n # than auto-generated has_foo() and clear_foo() methods\n # on the instances themselves. This way is less consistent\n # with C++, but it makes reflection-type access easier and\n # reduces the number of magically autogenerated things.\n #\n # TODO(robinson): Be sure to document (and test) exactly\n # which field names are accepted here. Are we case-sensitive?\n # What do we do with fields that share names with Python keywords\n # like 'lambda' and 'yield'?\n #\n # nnorwitz says:\n # \"\"\"\n # Typically (in python), an underscore is appended to names that are\n # keywords. So they would become lambda_ or yield_.\n # \"\"\"\n def ListFields(self):\n \"\"\"Returns a list of (FieldDescriptor, value) tuples for present fields.\n\n A message field is non-empty if HasField() would return true. A singular\n primitive field is non-empty if HasField() would return true in proto2 or it\n is non zero in proto3. A repeated field is non-empty if it contains at least\n one element. The fields are ordered by field number.\n\n Returns:\n list[tuple(FieldDescriptor, value)]: field descriptors and values\n for all fields in the message which are not empty. The values vary by\n field type.\n \"\"\"\n raise NotImplementedError\n\n def HasField(self, field_name):\n \"\"\"Checks if a certain field is set for the message.\n\n For a oneof group, checks if any field inside is set. Note that if the\n field_name is not defined in the message descriptor, :exc:`ValueError` will\n be raised.\n\n Args:\n field_name (str): The name of the field to check for presence.\n\n Returns:\n bool: Whether a value has been set for the named field.\n\n Raises:\n ValueError: if the `field_name` is not a member of this message.\n \"\"\"\n raise NotImplementedError\n\n def ClearField(self, field_name):\n \"\"\"Clears the contents of a given field.\n\n Inside a oneof group, clears the field set. If the name neither refers to a\n defined field or oneof group, :exc:`ValueError` is raised.\n\n Args:\n field_name (str): The name of the field to check for presence.\n\n Raises:\n ValueError: if the `field_name` is not a member of this message.\n \"\"\"\n raise NotImplementedError\n\n def WhichOneof(self, oneof_group):\n \"\"\"Returns the name of the field that is set inside a oneof group.\n\n If no field is set, returns None.\n\n Args:\n oneof_group (str): the name of the oneof group to check.\n\n Returns:\n str or None: The name of the group that is set, or None.\n\n Raises:\n ValueError: no group with the given name exists\n \"\"\"\n raise NotImplementedError\n\n def HasExtension(self, extension_handle):\n \"\"\"Checks if a certain extension is present for this message.\n\n Extensions are retrieved using the :attr:`Extensions` mapping (if present).\n\n Args:\n extension_handle: The handle for the extension to check.\n\n Returns:\n bool: Whether the extension is present for this message.\n\n Raises:\n KeyError: if the extension is repeated. Similar to repeated fields,\n there is no separate notion of presence: a \"not present\" repeated\n extension is an empty list.\n \"\"\"\n raise NotImplementedError\n\n def ClearExtension(self, extension_handle):\n \"\"\"Clears the contents of a given extension.\n\n Args:\n extension_handle: The handle for the extension to clear.\n \"\"\"\n raise NotImplementedError\n\n def UnknownFields(self):\n \"\"\"Returns the UnknownFieldSet.\n\n Returns:\n UnknownFieldSet: The unknown fields stored in this message.\n \"\"\"\n raise NotImplementedError\n\n def DiscardUnknownFields(self):\n \"\"\"Clears all fields in the :class:`UnknownFieldSet`.\n\n This operation is recursive for nested message.\n \"\"\"\n raise NotImplementedError\n\n def ByteSize(self):\n \"\"\"Returns the serialized size of this message.\n\n Recursively calls ByteSize() on all contained messages.\n\n Returns:\n int: The number of bytes required to serialize this message.\n \"\"\"\n raise NotImplementedError\n\n @classmethod\n def FromString(cls, s):\n raise NotImplementedError\n\n @staticmethod\n def RegisterExtension(extension_handle):\n raise NotImplementedError\n\n def _SetListener(self, message_listener):\n \"\"\"Internal method used by the protocol message implementation.\n Clients should not call this directly.\n\n Sets a listener that this message will call on certain state transitions.\n\n The purpose of this method is to register back-edges from children to\n parents at runtime, for the purpose of setting \"has\" bits and\n byte-size-dirty bits in the parent and ancestor objects whenever a child or\n descendant object is modified.\n\n If the client wants to disconnect this Message from the object tree, she\n explicitly sets callback to None.\n\n If message_listener is None, unregisters any existing listener. Otherwise,\n message_listener must implement the MessageListener interface in\n internal/message_listener.py, and we discard any listener registered\n via a previous _SetListener() call.\n \"\"\"\n raise NotImplementedError\n\n def __getstate__(self):\n \"\"\"Support the pickle protocol.\"\"\"\n return dict(serialized=self.SerializePartialToString())\n\n def __setstate__(self, state):\n \"\"\"Support the pickle protocol.\"\"\"\n self.__init__()\n serialized = state['serialized']\n # On Python 3, using encoding='latin1' is required for unpickling\n # protos pickled by Python 2.\n if not isinstance(serialized, bytes):\n serialized = serialized.encode('latin1')\n self.ParseFromString(serialized)\n\n def __reduce__(self):\n message_descriptor = self.DESCRIPTOR\n if message_descriptor.containing_type is None:\n return type(self), (), self.__getstate__()\n # the message type must be nested.\n # Python does not pickle nested classes; use the symbol_database on the\n # receiving end.\n container = message_descriptor\n return (_InternalConstructMessage, (container.full_name,),\n self.__getstate__())\n\n\ndef _InternalConstructMessage(full_name):\n \"\"\"Constructs a nested message.\"\"\"\n from google.protobuf import symbol_database # pylint:disable=g-import-not-at-top\n\n return symbol_database.Default().GetSymbol(full_name)()\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/message_factory.py",
"content": "# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n\"\"\"Provides a factory class for generating dynamic messages.\n\nThe easiest way to use this class is if you have access to the FileDescriptor\nprotos containing the messages you want to create you can just do the following:\n\nmessage_classes = message_factory.GetMessages(iterable_of_file_descriptors)\nmy_proto_instance = message_classes['some.proto.package.MessageName']()\n\"\"\"\n\n__author__ = 'matthewtoia@google.com (Matt Toia)'\n\nfrom google.protobuf.internal import api_implementation\nfrom google.protobuf import descriptor_pool\nfrom google.protobuf import message\n\nif api_implementation.Type() == 'cpp':\n from google.protobuf.pyext import cpp_message as message_impl\nelse:\n from google.protobuf.internal import python_message as message_impl\n\n\n# The type of all Message classes.\n_GENERATED_PROTOCOL_MESSAGE_TYPE = message_impl.GeneratedProtocolMessageType\n\n\nclass MessageFactory(object):\n \"\"\"Factory for creating Proto2 messages from descriptors in a pool.\"\"\"\n\n def __init__(self, pool=None):\n \"\"\"Initializes a new factory.\"\"\"\n self.pool = pool or descriptor_pool.DescriptorPool()\n\n # local cache of all classes built from protobuf descriptors\n self._classes = {}\n\n def GetPrototype(self, descriptor):\n \"\"\"Obtains a proto2 message class based on the passed in descriptor.\n\n Passing a descriptor with a fully qualified name matching a previous\n invocation will cause the same class to be returned.\n\n Args:\n descriptor: The descriptor to build from.\n\n Returns:\n A class describing the passed in descriptor.\n \"\"\"\n if descriptor not in self._classes:\n result_class = self.CreatePrototype(descriptor)\n # The assignment to _classes is redundant for the base implementation, but\n # might avoid confusion in cases where CreatePrototype gets overridden and\n # does not call the base implementation.\n self._classes[descriptor] = result_class\n return result_class\n return self._classes[descriptor]\n\n def CreatePrototype(self, descriptor):\n \"\"\"Builds a proto2 message class based on the passed in descriptor.\n\n Don't call this function directly, it always creates a new class. Call\n GetPrototype() instead. This method is meant to be overridden in subblasses\n to perform additional operations on the newly constructed class.\n\n Args:\n descriptor: The descriptor to build from.\n\n Returns:\n A class describing the passed in descriptor.\n \"\"\"\n descriptor_name = descriptor.name\n result_class = _GENERATED_PROTOCOL_MESSAGE_TYPE(\n descriptor_name,\n (message.Message,),\n {\n 'DESCRIPTOR': descriptor,\n # If module not set, it wrongly points to message_factory module.\n '__module__': None,\n })\n result_class._FACTORY = self # pylint: disable=protected-access\n # Assign in _classes before doing recursive calls to avoid infinite\n # recursion.\n self._classes[descriptor] = result_class\n for field in descriptor.fields:\n if field.message_type:\n self.GetPrototype(field.message_type)\n for extension in result_class.DESCRIPTOR.extensions:\n if extension.containing_type not in self._classes:\n self.GetPrototype(extension.containing_type)\n extended_class = self._classes[extension.containing_type]\n extended_class.RegisterExtension(extension)\n return result_class\n\n def GetMessages(self, files):\n \"\"\"Gets all the messages from a specified file.\n\n This will find and resolve dependencies, failing if the descriptor\n pool cannot satisfy them.\n\n Args:\n files: The file names to extract messages from.\n\n Returns:\n A dictionary mapping proto names to the message classes. This will include\n any dependent messages as well as any messages defined in the same file as\n a specified message.\n \"\"\"\n result = {}\n for file_name in files:\n file_desc = self.pool.FindFileByName(file_name)\n for desc in file_desc.message_types_by_name.values():\n result[desc.full_name] = self.GetPrototype(desc)\n\n # While the extension FieldDescriptors are created by the descriptor pool,\n # the python classes created in the factory need them to be registered\n # explicitly, which is done below.\n #\n # The call to RegisterExtension will specifically check if the\n # extension was already registered on the object and either\n # ignore the registration if the original was the same, or raise\n # an error if they were different.\n\n for extension in file_desc.extensions_by_name.values():\n if extension.containing_type not in self._classes:\n self.GetPrototype(extension.containing_type)\n extended_class = self._classes[extension.containing_type]\n extended_class.RegisterExtension(extension)\n return result\n\n\n_FACTORY = MessageFactory()\n\n\ndef GetMessages(file_protos):\n \"\"\"Builds a dictionary of all the messages available in a set of files.\n\n Args:\n file_protos: Iterable of FileDescriptorProto to build messages out of.\n\n Returns:\n A dictionary mapping proto names to the message classes. This will include\n any dependent messages as well as any messages defined in the same file as\n a specified message.\n \"\"\"\n # The cpp implementation of the protocol buffer library requires to add the\n # message in topological order of the dependency graph.\n file_by_name = {file_proto.name: file_proto for file_proto in file_protos}\n def _AddFile(file_proto):\n for dependency in file_proto.dependency:\n if dependency in file_by_name:\n # Remove from elements to be visited, in order to cut cycles.\n _AddFile(file_by_name.pop(dependency))\n _FACTORY.pool.Add(file_proto)\n while file_by_name:\n _AddFile(file_by_name.popitem()[1])\n return _FACTORY.GetMessages([file_proto.name for file_proto in file_protos])\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/proto_builder.py",
"content": "# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n\"\"\"Dynamic Protobuf class creator.\"\"\"\n\nfrom collections import OrderedDict\nimport hashlib\nimport os\n\nfrom google.protobuf import descriptor_pb2\nfrom google.protobuf import descriptor\nfrom google.protobuf import message_factory\n\n\ndef _GetMessageFromFactory(factory, full_name):\n \"\"\"Get a proto class from the MessageFactory by name.\n\n Args:\n factory: a MessageFactory instance.\n full_name: str, the fully qualified name of the proto type.\n Returns:\n A class, for the type identified by full_name.\n Raises:\n KeyError, if the proto is not found in the factory's descriptor pool.\n \"\"\"\n proto_descriptor = factory.pool.FindMessageTypeByName(full_name)\n proto_cls = factory.GetPrototype(proto_descriptor)\n return proto_cls\n\n\ndef MakeSimpleProtoClass(fields, full_name=None, pool=None):\n \"\"\"Create a Protobuf class whose fields are basic types.\n\n Note: this doesn't validate field names!\n\n Args:\n fields: dict of {name: field_type} mappings for each field in the proto. If\n this is an OrderedDict the order will be maintained, otherwise the\n fields will be sorted by name.\n full_name: optional str, the fully-qualified name of the proto type.\n pool: optional DescriptorPool instance.\n Returns:\n a class, the new protobuf class with a FileDescriptor.\n \"\"\"\n factory = message_factory.MessageFactory(pool=pool)\n\n if full_name is not None:\n try:\n proto_cls = _GetMessageFromFactory(factory, full_name)\n return proto_cls\n except KeyError:\n # The factory's DescriptorPool doesn't know about this class yet.\n pass\n\n # Get a list of (name, field_type) tuples from the fields dict. If fields was\n # an OrderedDict we keep the order, but otherwise we sort the field to ensure\n # consistent ordering.\n field_items = fields.items()\n if not isinstance(fields, OrderedDict):\n field_items = sorted(field_items)\n\n # Use a consistent file name that is unlikely to conflict with any imported\n # proto files.\n fields_hash = hashlib.sha1()\n for f_name, f_type in field_items:\n fields_hash.update(f_name.encode('utf-8'))\n fields_hash.update(str(f_type).encode('utf-8'))\n proto_file_name = fields_hash.hexdigest() + '.proto'\n\n # If the proto is anonymous, use the same hash to name it.\n if full_name is None:\n full_name = ('net.proto2.python.public.proto_builder.AnonymousProto_' +\n fields_hash.hexdigest())\n try:\n proto_cls = _GetMessageFromFactory(factory, full_name)\n return proto_cls\n except KeyError:\n # The factory's DescriptorPool doesn't know about this class yet.\n pass\n\n # This is the first time we see this proto: add a new descriptor to the pool.\n factory.pool.Add(\n _MakeFileDescriptorProto(proto_file_name, full_name, field_items))\n return _GetMessageFromFactory(factory, full_name)\n\n\ndef _MakeFileDescriptorProto(proto_file_name, full_name, field_items):\n \"\"\"Populate FileDescriptorProto for MessageFactory's DescriptorPool.\"\"\"\n package, name = full_name.rsplit('.', 1)\n file_proto = descriptor_pb2.FileDescriptorProto()\n file_proto.name = os.path.join(package.replace('.', '/'), proto_file_name)\n file_proto.package = package\n desc_proto = file_proto.message_type.add()\n desc_proto.name = name\n for f_number, (f_name, f_type) in enumerate(field_items, 1):\n field_proto = desc_proto.field.add()\n field_proto.name = f_name\n # # If the number falls in the reserved range, reassign it to the correct\n # # number after the range.\n if f_number >= descriptor.FieldDescriptor.FIRST_RESERVED_FIELD_NUMBER:\n f_number += (\n descriptor.FieldDescriptor.LAST_RESERVED_FIELD_NUMBER -\n descriptor.FieldDescriptor.FIRST_RESERVED_FIELD_NUMBER + 1)\n field_proto.number = f_number\n field_proto.label = descriptor_pb2.FieldDescriptorProto.LABEL_OPTIONAL\n field_proto.type = f_type\n return file_proto\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/pyext/__init__.py",
"content": ""
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/pyext/cpp_message.py",
"content": "# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n\"\"\"Protocol message implementation hooks for C++ implementation.\n\nContains helper functions used to create protocol message classes from\nDescriptor objects at runtime backed by the protocol buffer C++ API.\n\"\"\"\n\n__author__ = 'tibell@google.com (Johan Tibell)'\n\nfrom google.protobuf.pyext import _message\n\n\nclass GeneratedProtocolMessageType(_message.MessageMeta):\n\n \"\"\"Metaclass for protocol message classes created at runtime from Descriptors.\n\n The protocol compiler currently uses this metaclass to create protocol\n message classes at runtime. Clients can also manually create their own\n classes at runtime, as in this example:\n\n mydescriptor = Descriptor(.....)\n factory = symbol_database.Default()\n factory.pool.AddDescriptor(mydescriptor)\n MyProtoClass = factory.GetPrototype(mydescriptor)\n myproto_instance = MyProtoClass()\n myproto.foo_field = 23\n ...\n\n The above example will not work for nested types. If you wish to include them,\n use reflection.MakeClass() instead of manually instantiating the class in\n order to create the appropriate class structure.\n \"\"\"\n\n # Must be consistent with the protocol-compiler code in\n # proto2/compiler/internal/generator.*.\n _DESCRIPTOR_KEY = 'DESCRIPTOR'\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/pyext/python_pb2.py",
"content": "# -*- coding: utf-8 -*-\n# Generated by the protocol buffer compiler. DO NOT EDIT!\n# source: google/protobuf/pyext/python.proto\n\"\"\"Generated protocol buffer code.\"\"\"\nfrom google.protobuf.internal import builder as _builder\nfrom google.protobuf import descriptor as _descriptor\nfrom google.protobuf import descriptor_pool as _descriptor_pool\nfrom google.protobuf import symbol_database as _symbol_database\n# @@protoc_insertion_point(imports)\n\n_sym_db = _symbol_database.Default()\n\n\n\n\nDESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\\n\\\"google/protobuf/pyext/python.proto\\x12\\x1fgoogle.protobuf.python.internal\\\"\\xbc\\x02\\n\\x0cTestAllTypes\\x12\\\\\\n\\x17repeated_nested_message\\x18\\x01 \\x03(\\x0b\\x32;.google.protobuf.python.internal.TestAllTypes.NestedMessage\\x12\\\\\\n\\x17optional_nested_message\\x18\\x02 \\x01(\\x0b\\x32;.google.protobuf.python.internal.TestAllTypes.NestedMessage\\x12\\x16\\n\\x0eoptional_int32\\x18\\x03 \\x01(\\x05\\x1aX\\n\\rNestedMessage\\x12\\n\\n\\x02\\x62\\x62\\x18\\x01 \\x01(\\x05\\x12;\\n\\x02\\x63\\x63\\x18\\x02 \\x01(\\x0b\\x32/.google.protobuf.python.internal.ForeignMessage\\\"&\\n\\x0e\\x46oreignMessage\\x12\\t\\n\\x01\\x63\\x18\\x01 \\x01(\\x05\\x12\\t\\n\\x01\\x64\\x18\\x02 \\x03(\\x05\\\"\\x1d\\n\\x11TestAllExtensions*\\x08\\x08\\x01\\x10\\x80\\x80\\x80\\x80\\x02:\\x9a\\x01\\n!optional_nested_message_extension\\x12\\x32.google.protobuf.python.internal.TestAllExtensions\\x18\\x01 \\x01(\\x0b\\x32;.google.protobuf.python.internal.TestAllTypes.NestedMessage:\\x9a\\x01\\n!repeated_nested_message_extension\\x12\\x32.google.protobuf.python.internal.TestAllExtensions\\x18\\x02 \\x03(\\x0b\\x32;.google.protobuf.python.internal.TestAllTypes.NestedMessageB\\x02H\\x01')\n\n_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())\n_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.pyext.python_pb2', globals())\nif _descriptor._USE_C_DESCRIPTORS == False:\n TestAllExtensions.RegisterExtension(optional_nested_message_extension)\n TestAllExtensions.RegisterExtension(repeated_nested_message_extension)\n\n DESCRIPTOR._options = None\n DESCRIPTOR._serialized_options = b'H\\001'\n _TESTALLTYPES._serialized_start=72\n _TESTALLTYPES._serialized_end=388\n _TESTALLTYPES_NESTEDMESSAGE._serialized_start=300\n _TESTALLTYPES_NESTEDMESSAGE._serialized_end=388\n _FOREIGNMESSAGE._serialized_start=390\n _FOREIGNMESSAGE._serialized_end=428\n _TESTALLEXTENSIONS._serialized_start=430\n _TESTALLEXTENSIONS._serialized_end=459\n# @@protoc_insertion_point(module_scope)\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/reflection.py",
"content": "# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n# This code is meant to work on Python 2.4 and above only.\n\n\"\"\"Contains a metaclass and helper functions used to create\nprotocol message classes from Descriptor objects at runtime.\n\nRecall that a metaclass is the \"type\" of a class.\n(A class is to a metaclass what an instance is to a class.)\n\nIn this case, we use the GeneratedProtocolMessageType metaclass\nto inject all the useful functionality into the classes\noutput by the protocol compiler at compile-time.\n\nThe upshot of all this is that the real implementation\ndetails for ALL pure-Python protocol buffers are *here in\nthis file*.\n\"\"\"\n\n__author__ = 'robinson@google.com (Will Robinson)'\n\n\nfrom google.protobuf import message_factory\nfrom google.protobuf import symbol_database\n\n# The type of all Message classes.\n# Part of the public interface, but normally only used by message factories.\nGeneratedProtocolMessageType = message_factory._GENERATED_PROTOCOL_MESSAGE_TYPE\n\nMESSAGE_CLASS_CACHE = {}\n\n\n# Deprecated. Please NEVER use reflection.ParseMessage().\ndef ParseMessage(descriptor, byte_str):\n \"\"\"Generate a new Message instance from this Descriptor and a byte string.\n\n DEPRECATED: ParseMessage is deprecated because it is using MakeClass().\n Please use MessageFactory.GetPrototype() instead.\n\n Args:\n descriptor: Protobuf Descriptor object\n byte_str: Serialized protocol buffer byte string\n\n Returns:\n Newly created protobuf Message object.\n \"\"\"\n result_class = MakeClass(descriptor)\n new_msg = result_class()\n new_msg.ParseFromString(byte_str)\n return new_msg\n\n\n# Deprecated. Please NEVER use reflection.MakeClass().\ndef MakeClass(descriptor):\n \"\"\"Construct a class object for a protobuf described by descriptor.\n\n DEPRECATED: use MessageFactory.GetPrototype() instead.\n\n Args:\n descriptor: A descriptor.Descriptor object describing the protobuf.\n Returns:\n The Message class object described by the descriptor.\n \"\"\"\n # Original implementation leads to duplicate message classes, which won't play\n # well with extensions. Message factory info is also missing.\n # Redirect to message_factory.\n return symbol_database.Default().GetPrototype(descriptor)\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/service.py",
"content": "# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n\"\"\"DEPRECATED: Declares the RPC service interfaces.\n\nThis module declares the abstract interfaces underlying proto2 RPC\nservices. These are intended to be independent of any particular RPC\nimplementation, so that proto2 services can be used on top of a variety\nof implementations. Starting with version 2.3.0, RPC implementations should\nnot try to build on these, but should instead provide code generator plugins\nwhich generate code specific to the particular RPC implementation. This way\nthe generated code can be more appropriate for the implementation in use\nand can avoid unnecessary layers of indirection.\n\"\"\"\n\n__author__ = 'petar@google.com (Petar Petrov)'\n\n\nclass RpcException(Exception):\n \"\"\"Exception raised on failed blocking RPC method call.\"\"\"\n pass\n\n\nclass Service(object):\n\n \"\"\"Abstract base interface for protocol-buffer-based RPC services.\n\n Services themselves are abstract classes (implemented either by servers or as\n stubs), but they subclass this base interface. The methods of this\n interface can be used to call the methods of the service without knowing\n its exact type at compile time (analogous to the Message interface).\n \"\"\"\n\n def GetDescriptor():\n \"\"\"Retrieves this service's descriptor.\"\"\"\n raise NotImplementedError\n\n def CallMethod(self, method_descriptor, rpc_controller,\n request, done):\n \"\"\"Calls a method of the service specified by method_descriptor.\n\n If \"done\" is None then the call is blocking and the response\n message will be returned directly. Otherwise the call is asynchronous\n and \"done\" will later be called with the response value.\n\n In the blocking case, RpcException will be raised on error.\n\n Preconditions:\n\n * method_descriptor.service == GetDescriptor\n * request is of the exact same classes as returned by\n GetRequestClass(method).\n * After the call has started, the request must not be modified.\n * \"rpc_controller\" is of the correct type for the RPC implementation being\n used by this Service. For stubs, the \"correct type\" depends on the\n RpcChannel which the stub is using.\n\n Postconditions:\n\n * \"done\" will be called when the method is complete. This may be\n before CallMethod() returns or it may be at some point in the future.\n * If the RPC failed, the response value passed to \"done\" will be None.\n Further details about the failure can be found by querying the\n RpcController.\n \"\"\"\n raise NotImplementedError\n\n def GetRequestClass(self, method_descriptor):\n \"\"\"Returns the class of the request message for the specified method.\n\n CallMethod() requires that the request is of a particular subclass of\n Message. GetRequestClass() gets the default instance of this required\n type.\n\n Example:\n method = service.GetDescriptor().FindMethodByName(\"Foo\")\n request = stub.GetRequestClass(method)()\n request.ParseFromString(input)\n service.CallMethod(method, request, callback)\n \"\"\"\n raise NotImplementedError\n\n def GetResponseClass(self, method_descriptor):\n \"\"\"Returns the class of the response message for the specified method.\n\n This method isn't really needed, as the RpcChannel's CallMethod constructs\n the response protocol message. It's provided anyway in case it is useful\n for the caller to know the response type in advance.\n \"\"\"\n raise NotImplementedError\n\n\nclass RpcController(object):\n\n \"\"\"An RpcController mediates a single method call.\n\n The primary purpose of the controller is to provide a way to manipulate\n settings specific to the RPC implementation and to find out about RPC-level\n errors. The methods provided by the RpcController interface are intended\n to be a \"least common denominator\" set of features which we expect all\n implementations to support. Specific implementations may provide more\n advanced features (e.g. deadline propagation).\n \"\"\"\n\n # Client-side methods below\n\n def Reset(self):\n \"\"\"Resets the RpcController to its initial state.\n\n After the RpcController has been reset, it may be reused in\n a new call. Must not be called while an RPC is in progress.\n \"\"\"\n raise NotImplementedError\n\n def Failed(self):\n \"\"\"Returns true if the call failed.\n\n After a call has finished, returns true if the call failed. The possible\n reasons for failure depend on the RPC implementation. Failed() must not\n be called before a call has finished. If Failed() returns true, the\n contents of the response message are undefined.\n \"\"\"\n raise NotImplementedError\n\n def ErrorText(self):\n \"\"\"If Failed is true, returns a human-readable description of the error.\"\"\"\n raise NotImplementedError\n\n def StartCancel(self):\n \"\"\"Initiate cancellation.\n\n Advises the RPC system that the caller desires that the RPC call be\n canceled. The RPC system may cancel it immediately, may wait awhile and\n then cancel it, or may not even cancel the call at all. If the call is\n canceled, the \"done\" callback will still be called and the RpcController\n will indicate that the call failed at that time.\n \"\"\"\n raise NotImplementedError\n\n # Server-side methods below\n\n def SetFailed(self, reason):\n \"\"\"Sets a failure reason.\n\n Causes Failed() to return true on the client side. \"reason\" will be\n incorporated into the message returned by ErrorText(). If you find\n you need to return machine-readable information about failures, you\n should incorporate it into your response protocol buffer and should\n NOT call SetFailed().\n \"\"\"\n raise NotImplementedError\n\n def IsCanceled(self):\n \"\"\"Checks if the client cancelled the RPC.\n\n If true, indicates that the client canceled the RPC, so the server may\n as well give up on replying to it. The server should still call the\n final \"done\" callback.\n \"\"\"\n raise NotImplementedError\n\n def NotifyOnCancel(self, callback):\n \"\"\"Sets a callback to invoke on cancel.\n\n Asks that the given callback be called when the RPC is canceled. The\n callback will always be called exactly once. If the RPC completes without\n being canceled, the callback will be called after completion. If the RPC\n has already been canceled when NotifyOnCancel() is called, the callback\n will be called immediately.\n\n NotifyOnCancel() must be called no more than once per request.\n \"\"\"\n raise NotImplementedError\n\n\nclass RpcChannel(object):\n\n \"\"\"Abstract interface for an RPC channel.\n\n An RpcChannel represents a communication line to a service which can be used\n to call that service's methods. The service may be running on another\n machine. Normally, you should not use an RpcChannel directly, but instead\n construct a stub {@link Service} wrapping it. Example:\n\n Example:\n RpcChannel channel = rpcImpl.Channel(\"remotehost.example.com:1234\")\n RpcController controller = rpcImpl.Controller()\n MyService service = MyService_Stub(channel)\n service.MyMethod(controller, request, callback)\n \"\"\"\n\n def CallMethod(self, method_descriptor, rpc_controller,\n request, response_class, done):\n \"\"\"Calls the method identified by the descriptor.\n\n Call the given method of the remote service. The signature of this\n procedure looks the same as Service.CallMethod(), but the requirements\n are less strict in one important way: the request object doesn't have to\n be of any specific class as long as its descriptor is method.input_type.\n \"\"\"\n raise NotImplementedError\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/service_reflection.py",
"content": "# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n\"\"\"Contains metaclasses used to create protocol service and service stub\nclasses from ServiceDescriptor objects at runtime.\n\nThe GeneratedServiceType and GeneratedServiceStubType metaclasses are used to\ninject all useful functionality into the classes output by the protocol\ncompiler at compile-time.\n\"\"\"\n\n__author__ = 'petar@google.com (Petar Petrov)'\n\n\nclass GeneratedServiceType(type):\n\n \"\"\"Metaclass for service classes created at runtime from ServiceDescriptors.\n\n Implementations for all methods described in the Service class are added here\n by this class. We also create properties to allow getting/setting all fields\n in the protocol message.\n\n The protocol compiler currently uses this metaclass to create protocol service\n classes at runtime. Clients can also manually create their own classes at\n runtime, as in this example::\n\n mydescriptor = ServiceDescriptor(.....)\n class MyProtoService(service.Service):\n __metaclass__ = GeneratedServiceType\n DESCRIPTOR = mydescriptor\n myservice_instance = MyProtoService()\n # ...\n \"\"\"\n\n _DESCRIPTOR_KEY = 'DESCRIPTOR'\n\n def __init__(cls, name, bases, dictionary):\n \"\"\"Creates a message service class.\n\n Args:\n name: Name of the class (ignored, but required by the metaclass\n protocol).\n bases: Base classes of the class being constructed.\n dictionary: The class dictionary of the class being constructed.\n dictionary[_DESCRIPTOR_KEY] must contain a ServiceDescriptor object\n describing this protocol service type.\n \"\"\"\n # Don't do anything if this class doesn't have a descriptor. This happens\n # when a service class is subclassed.\n if GeneratedServiceType._DESCRIPTOR_KEY not in dictionary:\n return\n\n descriptor = dictionary[GeneratedServiceType._DESCRIPTOR_KEY]\n service_builder = _ServiceBuilder(descriptor)\n service_builder.BuildService(cls)\n cls.DESCRIPTOR = descriptor\n\n\nclass GeneratedServiceStubType(GeneratedServiceType):\n\n \"\"\"Metaclass for service stubs created at runtime from ServiceDescriptors.\n\n This class has similar responsibilities as GeneratedServiceType, except that\n it creates the service stub classes.\n \"\"\"\n\n _DESCRIPTOR_KEY = 'DESCRIPTOR'\n\n def __init__(cls, name, bases, dictionary):\n \"\"\"Creates a message service stub class.\n\n Args:\n name: Name of the class (ignored, here).\n bases: Base classes of the class being constructed.\n dictionary: The class dictionary of the class being constructed.\n dictionary[_DESCRIPTOR_KEY] must contain a ServiceDescriptor object\n describing this protocol service type.\n \"\"\"\n super(GeneratedServiceStubType, cls).__init__(name, bases, dictionary)\n # Don't do anything if this class doesn't have a descriptor. This happens\n # when a service stub is subclassed.\n if GeneratedServiceStubType._DESCRIPTOR_KEY not in dictionary:\n return\n\n descriptor = dictionary[GeneratedServiceStubType._DESCRIPTOR_KEY]\n service_stub_builder = _ServiceStubBuilder(descriptor)\n service_stub_builder.BuildServiceStub(cls)\n\n\nclass _ServiceBuilder(object):\n\n \"\"\"This class constructs a protocol service class using a service descriptor.\n\n Given a service descriptor, this class constructs a class that represents\n the specified service descriptor. One service builder instance constructs\n exactly one service class. That means all instances of that class share the\n same builder.\n \"\"\"\n\n def __init__(self, service_descriptor):\n \"\"\"Initializes an instance of the service class builder.\n\n Args:\n service_descriptor: ServiceDescriptor to use when constructing the\n service class.\n \"\"\"\n self.descriptor = service_descriptor\n\n def BuildService(builder, cls):\n \"\"\"Constructs the service class.\n\n Args:\n cls: The class that will be constructed.\n \"\"\"\n\n # CallMethod needs to operate with an instance of the Service class. This\n # internal wrapper function exists only to be able to pass the service\n # instance to the method that does the real CallMethod work.\n # Making sure to use exact argument names from the abstract interface in\n # service.py to match the type signature\n def _WrapCallMethod(self, method_descriptor, rpc_controller, request, done):\n return builder._CallMethod(self, method_descriptor, rpc_controller,\n request, done)\n\n def _WrapGetRequestClass(self, method_descriptor):\n return builder._GetRequestClass(method_descriptor)\n\n def _WrapGetResponseClass(self, method_descriptor):\n return builder._GetResponseClass(method_descriptor)\n\n builder.cls = cls\n cls.CallMethod = _WrapCallMethod\n cls.GetDescriptor = staticmethod(lambda: builder.descriptor)\n cls.GetDescriptor.__doc__ = 'Returns the service descriptor.'\n cls.GetRequestClass = _WrapGetRequestClass\n cls.GetResponseClass = _WrapGetResponseClass\n for method in builder.descriptor.methods:\n setattr(cls, method.name, builder._GenerateNonImplementedMethod(method))\n\n def _CallMethod(self, srvc, method_descriptor,\n rpc_controller, request, callback):\n \"\"\"Calls the method described by a given method descriptor.\n\n Args:\n srvc: Instance of the service for which this method is called.\n method_descriptor: Descriptor that represent the method to call.\n rpc_controller: RPC controller to use for this method's execution.\n request: Request protocol message.\n callback: A callback to invoke after the method has completed.\n \"\"\"\n if method_descriptor.containing_service != self.descriptor:\n raise RuntimeError(\n 'CallMethod() given method descriptor for wrong service type.')\n method = getattr(srvc, method_descriptor.name)\n return method(rpc_controller, request, callback)\n\n def _GetRequestClass(self, method_descriptor):\n \"\"\"Returns the class of the request protocol message.\n\n Args:\n method_descriptor: Descriptor of the method for which to return the\n request protocol message class.\n\n Returns:\n A class that represents the input protocol message of the specified\n method.\n \"\"\"\n if method_descriptor.containing_service != self.descriptor:\n raise RuntimeError(\n 'GetRequestClass() given method descriptor for wrong service type.')\n return method_descriptor.input_type._concrete_class\n\n def _GetResponseClass(self, method_descriptor):\n \"\"\"Returns the class of the response protocol message.\n\n Args:\n method_descriptor: Descriptor of the method for which to return the\n response protocol message class.\n\n Returns:\n A class that represents the output protocol message of the specified\n method.\n \"\"\"\n if method_descriptor.containing_service != self.descriptor:\n raise RuntimeError(\n 'GetResponseClass() given method descriptor for wrong service type.')\n return method_descriptor.output_type._concrete_class\n\n def _GenerateNonImplementedMethod(self, method):\n \"\"\"Generates and returns a method that can be set for a service methods.\n\n Args:\n method: Descriptor of the service method for which a method is to be\n generated.\n\n Returns:\n A method that can be added to the service class.\n \"\"\"\n return lambda inst, rpc_controller, request, callback: (\n self._NonImplementedMethod(method.name, rpc_controller, callback))\n\n def _NonImplementedMethod(self, method_name, rpc_controller, callback):\n \"\"\"The body of all methods in the generated service class.\n\n Args:\n method_name: Name of the method being executed.\n rpc_controller: RPC controller used to execute this method.\n callback: A callback which will be invoked when the method finishes.\n \"\"\"\n rpc_controller.SetFailed('Method %s not implemented.' % method_name)\n callback(None)\n\n\nclass _ServiceStubBuilder(object):\n\n \"\"\"Constructs a protocol service stub class using a service descriptor.\n\n Given a service descriptor, this class constructs a suitable stub class.\n A stub is just a type-safe wrapper around an RpcChannel which emulates a\n local implementation of the service.\n\n One service stub builder instance constructs exactly one class. It means all\n instances of that class share the same service stub builder.\n \"\"\"\n\n def __init__(self, service_descriptor):\n \"\"\"Initializes an instance of the service stub class builder.\n\n Args:\n service_descriptor: ServiceDescriptor to use when constructing the\n stub class.\n \"\"\"\n self.descriptor = service_descriptor\n\n def BuildServiceStub(self, cls):\n \"\"\"Constructs the stub class.\n\n Args:\n cls: The class that will be constructed.\n \"\"\"\n\n def _ServiceStubInit(stub, rpc_channel):\n stub.rpc_channel = rpc_channel\n self.cls = cls\n cls.__init__ = _ServiceStubInit\n for method in self.descriptor.methods:\n setattr(cls, method.name, self._GenerateStubMethod(method))\n\n def _GenerateStubMethod(self, method):\n return (lambda inst, rpc_controller, request, callback=None:\n self._StubMethod(inst, method, rpc_controller, request, callback))\n\n def _StubMethod(self, stub, method_descriptor,\n rpc_controller, request, callback):\n \"\"\"The body of all service methods in the generated stub class.\n\n Args:\n stub: Stub instance.\n method_descriptor: Descriptor of the invoked method.\n rpc_controller: Rpc controller to execute the method.\n request: Request protocol message.\n callback: A callback to execute when the method finishes.\n Returns:\n Response message (in case of blocking call).\n \"\"\"\n return stub.rpc_channel.CallMethod(\n method_descriptor, rpc_controller, request,\n method_descriptor.output_type._concrete_class, callback)\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/source_context_pb2.py",
"content": "# -*- coding: utf-8 -*-\n# Generated by the protocol buffer compiler. DO NOT EDIT!\n# source: google/protobuf/source_context.proto\n\"\"\"Generated protocol buffer code.\"\"\"\nfrom google.protobuf.internal import builder as _builder\nfrom google.protobuf import descriptor as _descriptor\nfrom google.protobuf import descriptor_pool as _descriptor_pool\nfrom google.protobuf import symbol_database as _symbol_database\n# @@protoc_insertion_point(imports)\n\n_sym_db = _symbol_database.Default()\n\n\n\n\nDESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\\n$google/protobuf/source_context.proto\\x12\\x0fgoogle.protobuf\\\"\\\"\\n\\rSourceContext\\x12\\x11\\n\\tfile_name\\x18\\x01 \\x01(\\tB\\x8a\\x01\\n\\x13\\x63om.google.protobufB\\x12SourceContextProtoP\\x01Z6google.golang.org/protobuf/types/known/sourcecontextpb\\xa2\\x02\\x03GPB\\xaa\\x02\\x1eGoogle.Protobuf.WellKnownTypesb\\x06proto3')\n\n_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())\n_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.source_context_pb2', globals())\nif _descriptor._USE_C_DESCRIPTORS == False:\n\n DESCRIPTOR._options = None\n DESCRIPTOR._serialized_options = b'\\n\\023com.google.protobufB\\022SourceContextProtoP\\001Z6google.golang.org/protobuf/types/known/sourcecontextpb\\242\\002\\003GPB\\252\\002\\036Google.Protobuf.WellKnownTypes'\n _SOURCECONTEXT._serialized_start=57\n _SOURCECONTEXT._serialized_end=91\n# @@protoc_insertion_point(module_scope)\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/struct_pb2.py",
"content": "# -*- coding: utf-8 -*-\n# Generated by the protocol buffer compiler. DO NOT EDIT!\n# source: google/protobuf/struct.proto\n\"\"\"Generated protocol buffer code.\"\"\"\nfrom google.protobuf.internal import builder as _builder\nfrom google.protobuf import descriptor as _descriptor\nfrom google.protobuf import descriptor_pool as _descriptor_pool\nfrom google.protobuf import symbol_database as _symbol_database\n# @@protoc_insertion_point(imports)\n\n_sym_db = _symbol_database.Default()\n\n\n\n\nDESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\\n\\x1cgoogle/protobuf/struct.proto\\x12\\x0fgoogle.protobuf\\\"\\x84\\x01\\n\\x06Struct\\x12\\x33\\n\\x06\\x66ields\\x18\\x01 \\x03(\\x0b\\x32#.google.protobuf.Struct.FieldsEntry\\x1a\\x45\\n\\x0b\\x46ieldsEntry\\x12\\x0b\\n\\x03key\\x18\\x01 \\x01(\\t\\x12%\\n\\x05value\\x18\\x02 \\x01(\\x0b\\x32\\x16.google.protobuf.Value:\\x02\\x38\\x01\\\"\\xea\\x01\\n\\x05Value\\x12\\x30\\n\\nnull_value\\x18\\x01 \\x01(\\x0e\\x32\\x1a.google.protobuf.NullValueH\\x00\\x12\\x16\\n\\x0cnumber_value\\x18\\x02 \\x01(\\x01H\\x00\\x12\\x16\\n\\x0cstring_value\\x18\\x03 \\x01(\\tH\\x00\\x12\\x14\\n\\nbool_value\\x18\\x04 \\x01(\\x08H\\x00\\x12/\\n\\x0cstruct_value\\x18\\x05 \\x01(\\x0b\\x32\\x17.google.protobuf.StructH\\x00\\x12\\x30\\n\\nlist_value\\x18\\x06 \\x01(\\x0b\\x32\\x1a.google.protobuf.ListValueH\\x00\\x42\\x06\\n\\x04kind\\\"3\\n\\tListValue\\x12&\\n\\x06values\\x18\\x01 \\x03(\\x0b\\x32\\x16.google.protobuf.Value*\\x1b\\n\\tNullValue\\x12\\x0e\\n\\nNULL_VALUE\\x10\\x00\\x42\\x7f\\n\\x13\\x63om.google.protobufB\\x0bStructProtoP\\x01Z/google.golang.org/protobuf/types/known/structpb\\xf8\\x01\\x01\\xa2\\x02\\x03GPB\\xaa\\x02\\x1eGoogle.Protobuf.WellKnownTypesb\\x06proto3')\n\n_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())\n_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.struct_pb2', globals())\nif _descriptor._USE_C_DESCRIPTORS == False:\n\n DESCRIPTOR._options = None\n DESCRIPTOR._serialized_options = b'\\n\\023com.google.protobufB\\013StructProtoP\\001Z/google.golang.org/protobuf/types/known/structpb\\370\\001\\001\\242\\002\\003GPB\\252\\002\\036Google.Protobuf.WellKnownTypes'\n _STRUCT_FIELDSENTRY._options = None\n _STRUCT_FIELDSENTRY._serialized_options = b'8\\001'\n _NULLVALUE._serialized_start=474\n _NULLVALUE._serialized_end=501\n _STRUCT._serialized_start=50\n _STRUCT._serialized_end=182\n _STRUCT_FIELDSENTRY._serialized_start=113\n _STRUCT_FIELDSENTRY._serialized_end=182\n _VALUE._serialized_start=185\n _VALUE._serialized_end=419\n _LISTVALUE._serialized_start=421\n _LISTVALUE._serialized_end=472\n# @@protoc_insertion_point(module_scope)\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/symbol_database.py",
"content": "# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n\"\"\"A database of Python protocol buffer generated symbols.\n\nSymbolDatabase is the MessageFactory for messages generated at compile time,\nand makes it easy to create new instances of a registered type, given only the\ntype's protocol buffer symbol name.\n\nExample usage::\n\n db = symbol_database.SymbolDatabase()\n\n # Register symbols of interest, from one or multiple files.\n db.RegisterFileDescriptor(my_proto_pb2.DESCRIPTOR)\n db.RegisterMessage(my_proto_pb2.MyMessage)\n db.RegisterEnumDescriptor(my_proto_pb2.MyEnum.DESCRIPTOR)\n\n # The database can be used as a MessageFactory, to generate types based on\n # their name:\n types = db.GetMessages(['my_proto.proto'])\n my_message_instance = types['MyMessage']()\n\n # The database's underlying descriptor pool can be queried, so it's not\n # necessary to know a type's filename to be able to generate it:\n filename = db.pool.FindFileContainingSymbol('MyMessage')\n my_message_instance = db.GetMessages([filename])['MyMessage']()\n\n # This functionality is also provided directly via a convenience method:\n my_message_instance = db.GetSymbol('MyMessage')()\n\"\"\"\n\n\nfrom google.protobuf.internal import api_implementation\nfrom google.protobuf import descriptor_pool\nfrom google.protobuf import message_factory\n\n\nclass SymbolDatabase(message_factory.MessageFactory):\n \"\"\"A database of Python generated symbols.\"\"\"\n\n def RegisterMessage(self, message):\n \"\"\"Registers the given message type in the local database.\n\n Calls to GetSymbol() and GetMessages() will return messages registered here.\n\n Args:\n message: A :class:`google.protobuf.message.Message` subclass (or\n instance); its descriptor will be registered.\n\n Returns:\n The provided message.\n \"\"\"\n\n desc = message.DESCRIPTOR\n self._classes[desc] = message\n self.RegisterMessageDescriptor(desc)\n return message\n\n def RegisterMessageDescriptor(self, message_descriptor):\n \"\"\"Registers the given message descriptor in the local database.\n\n Args:\n message_descriptor (Descriptor): the message descriptor to add.\n \"\"\"\n if api_implementation.Type() == 'python':\n # pylint: disable=protected-access\n self.pool._AddDescriptor(message_descriptor)\n\n def RegisterEnumDescriptor(self, enum_descriptor):\n \"\"\"Registers the given enum descriptor in the local database.\n\n Args:\n enum_descriptor (EnumDescriptor): The enum descriptor to register.\n\n Returns:\n EnumDescriptor: The provided descriptor.\n \"\"\"\n if api_implementation.Type() == 'python':\n # pylint: disable=protected-access\n self.pool._AddEnumDescriptor(enum_descriptor)\n return enum_descriptor\n\n def RegisterServiceDescriptor(self, service_descriptor):\n \"\"\"Registers the given service descriptor in the local database.\n\n Args:\n service_descriptor (ServiceDescriptor): the service descriptor to\n register.\n \"\"\"\n if api_implementation.Type() == 'python':\n # pylint: disable=protected-access\n self.pool._AddServiceDescriptor(service_descriptor)\n\n def RegisterFileDescriptor(self, file_descriptor):\n \"\"\"Registers the given file descriptor in the local database.\n\n Args:\n file_descriptor (FileDescriptor): The file descriptor to register.\n \"\"\"\n if api_implementation.Type() == 'python':\n # pylint: disable=protected-access\n self.pool._InternalAddFileDescriptor(file_descriptor)\n\n def GetSymbol(self, symbol):\n \"\"\"Tries to find a symbol in the local database.\n\n Currently, this method only returns message.Message instances, however, if\n may be extended in future to support other symbol types.\n\n Args:\n symbol (str): a protocol buffer symbol.\n\n Returns:\n A Python class corresponding to the symbol.\n\n Raises:\n KeyError: if the symbol could not be found.\n \"\"\"\n\n return self._classes[self.pool.FindMessageTypeByName(symbol)]\n\n def GetMessages(self, files):\n # TODO(amauryfa): Fix the differences with MessageFactory.\n \"\"\"Gets all registered messages from a specified file.\n\n Only messages already created and registered will be returned; (this is the\n case for imported _pb2 modules)\n But unlike MessageFactory, this version also returns already defined nested\n messages, but does not register any message extensions.\n\n Args:\n files (list[str]): The file names to extract messages from.\n\n Returns:\n A dictionary mapping proto names to the message classes.\n\n Raises:\n KeyError: if a file could not be found.\n \"\"\"\n\n def _GetAllMessages(desc):\n \"\"\"Walk a message Descriptor and recursively yields all message names.\"\"\"\n yield desc\n for msg_desc in desc.nested_types:\n for nested_desc in _GetAllMessages(msg_desc):\n yield nested_desc\n\n result = {}\n for file_name in files:\n file_desc = self.pool.FindFileByName(file_name)\n for msg_desc in file_desc.message_types_by_name.values():\n for desc in _GetAllMessages(msg_desc):\n try:\n result[desc.full_name] = self._classes[desc]\n except KeyError:\n # This descriptor has no registered class, skip it.\n pass\n return result\n\n\n_DEFAULT = SymbolDatabase(pool=descriptor_pool.Default())\n\n\ndef Default():\n \"\"\"Returns the default SymbolDatabase.\"\"\"\n return _DEFAULT\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/text_encoding.py",
"content": "# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n\"\"\"Encoding related utilities.\"\"\"\nimport re\n\n_cescape_chr_to_symbol_map = {}\n_cescape_chr_to_symbol_map[9] = r'\\t' # optional escape\n_cescape_chr_to_symbol_map[10] = r'\\n' # optional escape\n_cescape_chr_to_symbol_map[13] = r'\\r' # optional escape\n_cescape_chr_to_symbol_map[34] = r'\\\"' # necessary escape\n_cescape_chr_to_symbol_map[39] = r\"\\'\" # optional escape\n_cescape_chr_to_symbol_map[92] = r'\\\\' # necessary escape\n\n# Lookup table for unicode\n_cescape_unicode_to_str = [chr(i) for i in range(0, 256)]\nfor byte, string in _cescape_chr_to_symbol_map.items():\n _cescape_unicode_to_str[byte] = string\n\n# Lookup table for non-utf8, with necessary escapes at (o >= 127 or o < 32)\n_cescape_byte_to_str = ([r'\\%03o' % i for i in range(0, 32)] +\n [chr(i) for i in range(32, 127)] +\n [r'\\%03o' % i for i in range(127, 256)])\nfor byte, string in _cescape_chr_to_symbol_map.items():\n _cescape_byte_to_str[byte] = string\ndel byte, string\n\n\ndef CEscape(text, as_utf8):\n # type: (...) -> str\n \"\"\"Escape a bytes string for use in an text protocol buffer.\n\n Args:\n text: A byte string to be escaped.\n as_utf8: Specifies if result may contain non-ASCII characters.\n In Python 3 this allows unescaped non-ASCII Unicode characters.\n In Python 2 the return value will be valid UTF-8 rather than only ASCII.\n Returns:\n Escaped string (str).\n \"\"\"\n # Python's text.encode() 'string_escape' or 'unicode_escape' codecs do not\n # satisfy our needs; they encodes unprintable characters using two-digit hex\n # escapes whereas our C++ unescaping function allows hex escapes to be any\n # length. So, \"\\0011\".encode('string_escape') ends up being \"\\\\x011\", which\n # will be decoded in C++ as a single-character string with char code 0x11.\n text_is_unicode = isinstance(text, str)\n if as_utf8 and text_is_unicode:\n # We're already unicode, no processing beyond control char escapes.\n return text.translate(_cescape_chr_to_symbol_map)\n ord_ = ord if text_is_unicode else lambda x: x # bytes iterate as ints.\n if as_utf8:\n return ''.join(_cescape_unicode_to_str[ord_(c)] for c in text)\n return ''.join(_cescape_byte_to_str[ord_(c)] for c in text)\n\n\n_CUNESCAPE_HEX = re.compile(r'(\\\\+)x([0-9a-fA-F])(?![0-9a-fA-F])')\n\n\ndef CUnescape(text):\n # type: (str) -> bytes\n \"\"\"Unescape a text string with C-style escape sequences to UTF-8 bytes.\n\n Args:\n text: The data to parse in a str.\n Returns:\n A byte string.\n \"\"\"\n\n def ReplaceHex(m):\n # Only replace the match if the number of leading back slashes is odd. i.e.\n # the slash itself is not escaped.\n if len(m.group(1)) & 1:\n return m.group(1) + 'x0' + m.group(2)\n return m.group(0)\n\n # This is required because the 'string_escape' encoding doesn't\n # allow single-digit hex escapes (like '\\xf').\n result = _CUNESCAPE_HEX.sub(ReplaceHex, text)\n\n return (result.encode('utf-8') # Make it bytes to allow decode.\n .decode('unicode_escape')\n # Make it bytes again to return the proper type.\n .encode('raw_unicode_escape'))\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/text_format.py",
"content": "# Protocol Buffers - Google's data interchange format\n# Copyright 2008 Google Inc. All rights reserved.\n# https://developers.google.com/protocol-buffers/\n#\n# Redistribution and use in source and binary forms, with or without\n# modification, are permitted provided that the following conditions are\n# met:\n#\n# * Redistributions of source code must retain the above copyright\n# notice, this list of conditions and the following disclaimer.\n# * Redistributions in binary form must reproduce the above\n# copyright notice, this list of conditions and the following disclaimer\n# in the documentation and/or other materials provided with the\n# distribution.\n# * Neither the name of Google Inc. nor the names of its\n# contributors may be used to endorse or promote products derived from\n# this software without specific prior written permission.\n#\n# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS\n# \"AS IS\" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT\n# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR\n# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT\n# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,\n# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT\n# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,\n# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY\n# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT\n# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE\n# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.\n\n\"\"\"Contains routines for printing protocol messages in text format.\n\nSimple usage example::\n\n # Create a proto object and serialize it to a text proto string.\n message = my_proto_pb2.MyMessage(foo='bar')\n text_proto = text_format.MessageToString(message)\n\n # Parse a text proto string.\n message = text_format.Parse(text_proto, my_proto_pb2.MyMessage())\n\"\"\"\n\n__author__ = 'kenton@google.com (Kenton Varda)'\n\n# TODO(b/129989314) Import thread contention leads to test failures.\nimport encodings.raw_unicode_escape # pylint: disable=unused-import\nimport encodings.unicode_escape # pylint: disable=unused-import\nimport io\nimport math\nimport re\n\nfrom google.protobuf.internal import decoder\nfrom google.protobuf.internal import type_checkers\nfrom google.protobuf import descriptor\nfrom google.protobuf import text_encoding\n\n# pylint: disable=g-import-not-at-top\n__all__ = ['MessageToString', 'Parse', 'PrintMessage', 'PrintField',\n 'PrintFieldValue', 'Merge', 'MessageToBytes']\n\n_INTEGER_CHECKERS = (type_checkers.Uint32ValueChecker(),\n type_checkers.Int32ValueChecker(),\n type_checkers.Uint64ValueChecker(),\n type_checkers.Int64ValueChecker())\n_FLOAT_INFINITY = re.compile('-?inf(?:inity)?f?$', re.IGNORECASE)\n_FLOAT_NAN = re.compile('nanf?$', re.IGNORECASE)\n_QUOTES = frozenset((\"'\", '\"'))\n_ANY_FULL_TYPE_NAME = 'google.protobuf.Any'\n\n\nclass Error(Exception):\n \"\"\"Top-level module error for text_format.\"\"\"\n\n\nclass ParseError(Error):\n \"\"\"Thrown in case of text parsing or tokenizing error.\"\"\"\n\n def __init__(self, message=None, line=None, column=None):\n if message is not None and line is not None:\n loc = str(line)\n if column is not None:\n loc += ':{0}'.format(column)\n message = '{0} : {1}'.format(loc, message)\n if message is not None:\n super(ParseError, self).__init__(message)\n else:\n super(ParseError, self).__init__()\n self._line = line\n self._column = column\n\n def GetLine(self):\n return self._line\n\n def GetColumn(self):\n return self._column\n\n\nclass TextWriter(object):\n\n def __init__(self, as_utf8):\n self._writer = io.StringIO()\n\n def write(self, val):\n return self._writer.write(val)\n\n def close(self):\n return self._writer.close()\n\n def getvalue(self):\n return self._writer.getvalue()\n\n\ndef MessageToString(\n message,\n as_utf8=False,\n as_one_line=False,\n use_short_repeated_primitives=False,\n pointy_brackets=False,\n use_index_order=False,\n float_format=None,\n double_format=None,\n use_field_number=False,\n descriptor_pool=None,\n indent=0,\n message_formatter=None,\n print_unknown_fields=False,\n force_colon=False):\n # type: (...) -> str\n \"\"\"Convert protobuf message to text format.\n\n Double values can be formatted compactly with 15 digits of\n precision (which is the most that IEEE 754 \"double\" can guarantee)\n using double_format='.15g'. To ensure that converting to text and back to a\n proto will result in an identical value, double_format='.17g' should be used.\n\n Args:\n message: The protocol buffers message.\n as_utf8: Return unescaped Unicode for non-ASCII characters.\n In Python 3 actual Unicode characters may appear as is in strings.\n In Python 2 the return value will be valid UTF-8 rather than only ASCII.\n as_one_line: Don't introduce newlines between fields.\n use_short_repeated_primitives: Use short repeated format for primitives.\n pointy_brackets: If True, use angle brackets instead of curly braces for\n nesting.\n use_index_order: If True, fields of a proto message will be printed using\n the order defined in source code instead of the field number, extensions\n will be printed at the end of the message and their relative order is\n determined by the extension number. By default, use the field number\n order.\n float_format (str): If set, use this to specify float field formatting\n (per the \"Format Specification Mini-Language\"); otherwise, shortest float\n that has same value in wire will be printed. Also affect double field\n if double_format is not set but float_format is set.\n double_format (str): If set, use this to specify double field formatting\n (per the \"Format Specification Mini-Language\"); if it is not set but\n float_format is set, use float_format. Otherwise, use ``str()``\n use_field_number: If True, print field numbers instead of names.\n descriptor_pool (DescriptorPool): Descriptor pool used to resolve Any types.\n indent (int): The initial indent level, in terms of spaces, for pretty\n print.\n message_formatter (function(message, indent, as_one_line) -> unicode|None):\n Custom formatter for selected sub-messages (usually based on message\n type). Use to pretty print parts of the protobuf for easier diffing.\n print_unknown_fields: If True, unknown fields will be printed.\n force_colon: If set, a colon will be added after the field name even if the\n field is a proto message.\n\n Returns:\n str: A string of the text formatted protocol buffer message.\n \"\"\"\n out = TextWriter(as_utf8)\n printer = _Printer(\n out,\n indent,\n as_utf8,\n as_one_line,\n use_short_repeated_primitives,\n pointy_brackets,\n use_index_order,\n float_format,\n double_format,\n use_field_number,\n descriptor_pool,\n message_formatter,\n print_unknown_fields=print_unknown_fields,\n force_colon=force_colon)\n printer.PrintMessage(message)\n result = out.getvalue()\n out.close()\n if as_one_line:\n return result.rstrip()\n return result\n\n\ndef MessageToBytes(message, **kwargs):\n # type: (...) -> bytes\n \"\"\"Convert protobuf message to encoded text format. See MessageToString.\"\"\"\n text = MessageToString(message, **kwargs)\n if isinstance(text, bytes):\n return text\n codec = 'utf-8' if kwargs.get('as_utf8') else 'ascii'\n return text.encode(codec)\n\n\ndef _IsMapEntry(field):\n return (field.type == descriptor.FieldDescriptor.TYPE_MESSAGE and\n field.message_type.has_options and\n field.message_type.GetOptions().map_entry)\n\n\ndef PrintMessage(message,\n out,\n indent=0,\n as_utf8=False,\n as_one_line=False,\n use_short_repeated_primitives=False,\n pointy_brackets=False,\n use_index_order=False,\n float_format=None,\n double_format=None,\n use_field_number=False,\n descriptor_pool=None,\n message_formatter=None,\n print_unknown_fields=False,\n force_colon=False):\n printer = _Printer(\n out=out, indent=indent, as_utf8=as_utf8,\n as_one_line=as_one_line,\n use_short_repeated_primitives=use_short_repeated_primitives,\n pointy_brackets=pointy_brackets,\n use_index_order=use_index_order,\n float_format=float_format,\n double_format=double_format,\n use_field_number=use_field_number,\n descriptor_pool=descriptor_pool,\n message_formatter=message_formatter,\n print_unknown_fields=print_unknown_fields,\n force_colon=force_colon)\n printer.PrintMessage(message)\n\n\ndef PrintField(field,\n value,\n out,\n indent=0,\n as_utf8=False,\n as_one_line=False,\n use_short_repeated_primitives=False,\n pointy_brackets=False,\n use_index_order=False,\n float_format=None,\n double_format=None,\n message_formatter=None,\n print_unknown_fields=False,\n force_colon=False):\n \"\"\"Print a single field name/value pair.\"\"\"\n printer = _Printer(out, indent, as_utf8, as_one_line,\n use_short_repeated_primitives, pointy_brackets,\n use_index_order, float_format, double_format,\n message_formatter=message_formatter,\n print_unknown_fields=print_unknown_fields,\n force_colon=force_colon)\n printer.PrintField(field, value)\n\n\ndef PrintFieldValue(field,\n value,\n out,\n indent=0,\n as_utf8=False,\n as_one_line=False,\n use_short_repeated_primitives=False,\n pointy_brackets=False,\n use_index_order=False,\n float_format=None,\n double_format=None,\n message_formatter=None,\n print_unknown_fields=False,\n force_colon=False):\n \"\"\"Print a single field value (not including name).\"\"\"\n printer = _Printer(out, indent, as_utf8, as_one_line,\n use_short_repeated_primitives, pointy_brackets,\n use_index_order, float_format, double_format,\n message_formatter=message_formatter,\n print_unknown_fields=print_unknown_fields,\n force_colon=force_colon)\n printer.PrintFieldValue(field, value)\n\n\ndef _BuildMessageFromTypeName(type_name, descriptor_pool):\n \"\"\"Returns a protobuf message instance.\n\n Args:\n type_name: Fully-qualified protobuf message type name string.\n descriptor_pool: DescriptorPool instance.\n\n Returns:\n A Message instance of type matching type_name, or None if the a Descriptor\n wasn't found matching type_name.\n \"\"\"\n # pylint: disable=g-import-not-at-top\n if descriptor_pool is None:\n from google.protobuf import descriptor_pool as pool_mod\n descriptor_pool = pool_mod.Default()\n from google.protobuf import symbol_database\n database = symbol_database.Default()\n try:\n message_descriptor = descriptor_pool.FindMessageTypeByName(type_name)\n except KeyError:\n return None\n message_type = database.GetPrototype(message_descriptor)\n return message_type()\n\n\n# These values must match WireType enum in google/protobuf/wire_format.h.\nWIRETYPE_LENGTH_DELIMITED = 2\nWIRETYPE_START_GROUP = 3\n\n\nclass _Printer(object):\n \"\"\"Text format printer for protocol message.\"\"\"\n\n def __init__(\n self,\n out,\n indent=0,\n as_utf8=False,\n as_one_line=False,\n use_short_repeated_primitives=False,\n pointy_brackets=False,\n use_index_order=False,\n float_format=None,\n double_format=None,\n use_field_number=False,\n descriptor_pool=None,\n message_formatter=None,\n print_unknown_fields=False,\n force_colon=False):\n \"\"\"Initialize the Printer.\n\n Double values can be formatted compactly with 15 digits of precision\n (which is the most that IEEE 754 \"double\" can guarantee) using\n double_format='.15g'. To ensure that converting to text and back to a proto\n will result in an identical value, double_format='.17g' should be used.\n\n Args:\n out: To record the text format result.\n indent: The initial indent level for pretty print.\n as_utf8: Return unescaped Unicode for non-ASCII characters.\n In Python 3 actual Unicode characters may appear as is in strings.\n In Python 2 the return value will be valid UTF-8 rather than ASCII.\n as_one_line: Don't introduce newlines between fields.\n use_short_repeated_primitives: Use short repeated format for primitives.\n pointy_brackets: If True, use angle brackets instead of curly braces for\n nesting.\n use_index_order: If True, print fields of a proto message using the order\n defined in source code instead of the field number. By default, use the\n field number order.\n float_format: If set, use this to specify float field formatting\n (per the \"Format Specification Mini-Language\"); otherwise, shortest\n float that has same value in wire will be printed. Also affect double\n field if double_format is not set but float_format is set.\n double_format: If set, use this to specify double field formatting\n (per the \"Format Specification Mini-Language\"); if it is not set but\n float_format is set, use float_format. Otherwise, str() is used.\n use_field_number: If True, print field numbers instead of names.\n descriptor_pool: A DescriptorPool used to resolve Any types.\n message_formatter: A function(message, indent, as_one_line): unicode|None\n to custom format selected sub-messages (usually based on message type).\n Use to pretty print parts of the protobuf for easier diffing.\n print_unknown_fields: If True, unknown fields will be printed.\n force_colon: If set, a colon will be added after the field name even if\n the field is a proto message.\n \"\"\"\n self.out = out\n self.indent = indent\n self.as_utf8 = as_utf8\n self.as_one_line = as_one_line\n self.use_short_repeated_primitives = use_short_repeated_primitives\n self.pointy_brackets = pointy_brackets\n self.use_index_order = use_index_order\n self.float_format = float_format\n if double_format is not None:\n self.double_format = double_format\n else:\n self.double_format = float_format\n self.use_field_number = use_field_number\n self.descriptor_pool = descriptor_pool\n self.message_formatter = message_formatter\n self.print_unknown_fields = print_unknown_fields\n self.force_colon = force_colon\n\n def _TryPrintAsAnyMessage(self, message):\n \"\"\"Serializes if message is a google.protobuf.Any field.\"\"\"\n if '/' not in message.type_url:\n return False\n packed_message = _BuildMessageFromTypeName(message.TypeName(),\n self.descriptor_pool)\n if packed_message:\n packed_message.MergeFromString(message.value)\n colon = ':' if self.force_colon else ''\n self.out.write('%s[%s]%s ' % (self.indent * ' ', message.type_url, colon))\n self._PrintMessageFieldValue(packed_message)\n self.out.write(' ' if self.as_one_line else '\\n')\n return True\n else:\n return False\n\n def _TryCustomFormatMessage(self, message):\n formatted = self.message_formatter(message, self.indent, self.as_one_line)\n if formatted is None:\n return False\n\n out = self.out\n out.write(' ' * self.indent)\n out.write(formatted)\n out.write(' ' if self.as_one_line else '\\n')\n return True\n\n def PrintMessage(self, message):\n \"\"\"Convert protobuf message to text format.\n\n Args:\n message: The protocol buffers message.\n \"\"\"\n if self.message_formatter and self._TryCustomFormatMessage(message):\n return\n if (message.DESCRIPTOR.full_name == _ANY_FULL_TYPE_NAME and\n self._TryPrintAsAnyMessage(message)):\n return\n fields = message.ListFields()\n if self.use_index_order:\n fields.sort(\n key=lambda x: x[0].number if x[0].is_extension else x[0].index)\n for field, value in fields:\n if _IsMapEntry(field):\n for key in sorted(value):\n # This is slow for maps with submessage entries because it copies the\n # entire tree. Unfortunately this would take significant refactoring\n # of this file to work around.\n #\n # TODO(haberman): refactor and optimize if this becomes an issue.\n entry_submsg = value.GetEntryClass()(key=key, value=value[key])\n self.PrintField(field, entry_submsg)\n elif field.label == descriptor.FieldDescriptor.LABEL_REPEATED:\n if (self.use_short_repeated_primitives\n and field.cpp_type != descriptor.FieldDescriptor.CPPTYPE_MESSAGE\n and field.cpp_type != descriptor.FieldDescriptor.CPPTYPE_STRING):\n self._PrintShortRepeatedPrimitivesValue(field, value)\n else:\n for element in value:\n self.PrintField(field, element)\n else:\n self.PrintField(field, value)\n\n if self.print_unknown_fields:\n self._PrintUnknownFields(message.UnknownFields())\n\n def _PrintUnknownFields(self, unknown_fields):\n \"\"\"Print unknown fields.\"\"\"\n out = self.out\n for field in unknown_fields:\n out.write(' ' * self.indent)\n out.write(str(field.field_number))\n if field.wire_type == WIRETYPE_START_GROUP:\n if self.as_one_line:\n out.write(' { ')\n else:\n out.write(' {\\n')\n self.indent += 2\n\n self._PrintUnknownFields(field.data)\n\n if self.as_one_line:\n out.write('} ')\n else:\n self.indent -= 2\n out.write(' ' * self.indent + '}\\n')\n elif field.wire_type == WIRETYPE_LENGTH_DELIMITED:\n try:\n # If this field is parseable as a Message, it is probably\n # an embedded message.\n # pylint: disable=protected-access\n (embedded_unknown_message, pos) = decoder._DecodeUnknownFieldSet(\n memoryview(field.data), 0, len(field.data))\n except Exception: # pylint: disable=broad-except\n pos = 0\n\n if pos == len(field.data):\n if self.as_one_line:\n out.write(' { ')\n else:\n out.write(' {\\n')\n self.indent += 2\n\n self._PrintUnknownFields(embedded_unknown_message)\n\n if self.as_one_line:\n out.write('} ')\n else:\n self.indent -= 2\n out.write(' ' * self.indent + '}\\n')\n else:\n # A string or bytes field. self.as_utf8 may not work.\n out.write(': \\\"')\n out.write(text_encoding.CEscape(field.data, False))\n out.write('\\\" ' if self.as_one_line else '\\\"\\n')\n else:\n # varint, fixed32, fixed64\n out.write(': ')\n out.write(str(field.data))\n out.write(' ' if self.as_one_line else '\\n')\n\n def _PrintFieldName(self, field):\n \"\"\"Print field name.\"\"\"\n out = self.out\n out.write(' ' * self.indent)\n if self.use_field_number:\n out.write(str(field.number))\n else:\n if field.is_extension:\n out.write('[')\n if (field.containing_type.GetOptions().message_set_wire_format and\n field.type == descriptor.FieldDescriptor.TYPE_MESSAGE and\n field.label == descriptor.FieldDescriptor.LABEL_OPTIONAL):\n out.write(field.message_type.full_name)\n else:\n out.write(field.full_name)\n out.write(']')\n elif field.type == descriptor.FieldDescriptor.TYPE_GROUP:\n # For groups, use the capitalized name.\n out.write(field.message_type.name)\n else:\n out.write(field.name)\n\n if (self.force_colon or\n field.cpp_type != descriptor.FieldDescriptor.CPPTYPE_MESSAGE):\n # The colon is optional in this case, but our cross-language golden files\n # don't include it. Here, the colon is only included if force_colon is\n # set to True\n out.write(':')\n\n def PrintField(self, field, value):\n \"\"\"Print a single field name/value pair.\"\"\"\n self._PrintFieldName(field)\n self.out.write(' ')\n self.PrintFieldValue(field, value)\n self.out.write(' ' if self.as_one_line else '\\n')\n\n def _PrintShortRepeatedPrimitivesValue(self, field, value):\n \"\"\"\"Prints short repeated primitives value.\"\"\"\n # Note: this is called only when value has at least one element.\n self._PrintFieldName(field)\n self.out.write(' [')\n for i in range(len(value) - 1):\n self.PrintFieldValue(field, value[i])\n self.out.write(', ')\n self.PrintFieldValue(field, value[-1])\n self.out.write(']')\n self.out.write(' ' if self.as_one_line else '\\n')\n\n def _PrintMessageFieldValue(self, value):\n if self.pointy_brackets:\n openb = '<'\n closeb = '>'\n else:\n openb = '{'\n closeb = '}'\n\n if self.as_one_line:\n self.out.write('%s ' % openb)\n self.PrintMessage(value)\n self.out.write(closeb)\n else:\n self.out.write('%s\\n' % openb)\n self.indent += 2\n self.PrintMessage(value)\n self.indent -= 2\n self.out.write(' ' * self.indent + closeb)\n\n def PrintFieldValue(self, field, value):\n \"\"\"Print a single field value (not including name).\n\n For repeated fields, the value should be a single element.\n\n Args:\n field: The descriptor of the field to be printed.\n value: The value of the field.\n \"\"\"\n out = self.out\n if field.cpp_type == descriptor.FieldDescriptor.CPPTYPE_MESSAGE:\n self._PrintMessageFieldValue(value)\n elif field.cpp_type == descriptor.FieldDescriptor.CPPTYPE_ENUM:\n enum_value = field.enum_type.values_by_number.get(value, None)\n if enum_value is not None:\n out.write(enum_value.name)\n else:\n out.write(str(value))\n elif field.cpp_type == descriptor.FieldDescriptor.CPPTYPE_STRING:\n out.write('\\\"')\n if isinstance(value, str) and not self.as_utf8:\n out_value = value.encode('utf-8')\n else:\n out_value = value\n if field.type == descriptor.FieldDescriptor.TYPE_BYTES:\n # We always need to escape all binary data in TYPE_BYTES fields.\n out_as_utf8 = False\n else:\n out_as_utf8 = self.as_utf8\n out.write(text_encoding.CEscape(out_value, out_as_utf8))\n out.write('\\\"')\n elif field.cpp_type == descriptor.FieldDescriptor.CPPTYPE_BOOL:\n if value:\n out.write('true')\n else:\n out.write('false')\n elif field.cpp_type == descriptor.FieldDescriptor.CPPTYPE_FLOAT:\n if self.float_format is not None:\n out.write('{1:{0}}'.format(self.float_format, value))\n else:\n if math.isnan(value):\n out.write(str(value))\n else:\n out.write(str(type_checkers.ToShortestFloat(value)))\n elif (field.cpp_type == descriptor.FieldDescriptor.CPPTYPE_DOUBLE and\n self.double_format is not None):\n out.write('{1:{0}}'.format(self.double_format, value))\n else:\n out.write(str(value))\n\n\ndef Parse(text,\n message,\n allow_unknown_extension=False,\n allow_field_number=False,\n descriptor_pool=None,\n allow_unknown_field=False):\n \"\"\"Parses a text representation of a protocol message into a message.\n\n NOTE: for historical reasons this function does not clear the input\n message. This is different from what the binary msg.ParseFrom(...) does.\n If text contains a field already set in message, the value is appended if the\n field is repeated. Otherwise, an error is raised.\n\n Example::\n\n a = MyProto()\n a.repeated_field.append('test')\n b = MyProto()\n\n # Repeated fields are combined\n text_format.Parse(repr(a), b)\n text_format.Parse(repr(a), b) # repeated_field contains [\"test\", \"test\"]\n\n # Non-repeated fields cannot be overwritten\n a.singular_field = 1\n b.singular_field = 2\n text_format.Parse(repr(a), b) # ParseError\n\n # Binary version:\n b.ParseFromString(a.SerializeToString()) # repeated_field is now \"test\"\n\n Caller is responsible for clearing the message as needed.\n\n Args:\n text (str): Message text representation.\n message (Message): A protocol buffer message to merge into.\n allow_unknown_extension: if True, skip over missing extensions and keep\n parsing\n allow_field_number: if True, both field number and field name are allowed.\n descriptor_pool (DescriptorPool): Descriptor pool used to resolve Any types.\n allow_unknown_field: if True, skip over unknown field and keep\n parsing. Avoid to use this option if possible. It may hide some\n errors (e.g. spelling error on field name)\n\n Returns:\n Message: The same message passed as argument.\n\n Raises:\n ParseError: On text parsing problems.\n \"\"\"\n return ParseLines(text.split(b'\\n' if isinstance(text, bytes) else u'\\n'),\n message,\n allow_unknown_extension,\n allow_field_number,\n descriptor_pool=descriptor_pool,\n allow_unknown_field=allow_unknown_field)\n\n\ndef Merge(text,\n message,\n allow_unknown_extension=False,\n allow_field_number=False,\n descriptor_pool=None,\n allow_unknown_field=False):\n \"\"\"Parses a text representation of a protocol message into a message.\n\n Like Parse(), but allows repeated values for a non-repeated field, and uses\n the last one. This means any non-repeated, top-level fields specified in text\n replace those in the message.\n\n Args:\n text (str): Message text representation.\n message (Message): A protocol buffer message to merge into.\n allow_unknown_extension: if True, skip over missing extensions and keep\n parsing\n allow_field_number: if True, both field number and field name are allowed.\n descriptor_pool (DescriptorPool): Descriptor pool used to resolve Any types.\n allow_unknown_field: if True, skip over unknown field and keep\n parsing. Avoid to use this option if possible. It may hide some\n errors (e.g. spelling error on field name)\n\n Returns:\n Message: The same message passed as argument.\n\n Raises:\n ParseError: On text parsing problems.\n \"\"\"\n return MergeLines(\n text.split(b'\\n' if isinstance(text, bytes) else u'\\n'),\n message,\n allow_unknown_extension,\n allow_field_number,\n descriptor_pool=descriptor_pool,\n allow_unknown_field=allow_unknown_field)\n\n\ndef ParseLines(lines,\n message,\n allow_unknown_extension=False,\n allow_field_number=False,\n descriptor_pool=None,\n allow_unknown_field=False):\n \"\"\"Parses a text representation of a protocol message into a message.\n\n See Parse() for caveats.\n\n Args:\n lines: An iterable of lines of a message's text representation.\n message: A protocol buffer message to merge into.\n allow_unknown_extension: if True, skip over missing extensions and keep\n parsing\n allow_field_number: if True, both field number and field name are allowed.\n descriptor_pool: A DescriptorPool used to resolve Any types.\n allow_unknown_field: if True, skip over unknown field and keep\n parsing. Avoid to use this option if possible. It may hide some\n errors (e.g. spelling error on field name)\n\n Returns:\n The same message passed as argument.\n\n Raises:\n ParseError: On text parsing problems.\n \"\"\"\n parser = _Parser(allow_unknown_extension,\n allow_field_number,\n descriptor_pool=descriptor_pool,\n allow_unknown_field=allow_unknown_field)\n return parser.ParseLines(lines, message)\n\n\ndef MergeLines(lines,\n message,\n allow_unknown_extension=False,\n allow_field_number=False,\n descriptor_pool=None,\n allow_unknown_field=False):\n \"\"\"Parses a text representation of a protocol message into a message.\n\n See Merge() for more details.\n\n Args:\n lines: An iterable of lines of a message's text representation.\n message: A protocol buffer message to merge into.\n allow_unknown_extension: if True, skip over missing extensions and keep\n parsing\n allow_field_number: if True, both field number and field name are allowed.\n descriptor_pool: A DescriptorPool used to resolve Any types.\n allow_unknown_field: if True, skip over unknown field and keep\n parsing. Avoid to use this option if possible. It may hide some\n errors (e.g. spelling error on field name)\n\n Returns:\n The same message passed as argument.\n\n Raises:\n ParseError: On text parsing problems.\n \"\"\"\n parser = _Parser(allow_unknown_extension,\n allow_field_number,\n descriptor_pool=descriptor_pool,\n allow_unknown_field=allow_unknown_field)\n return parser.MergeLines(lines, message)\n\n\nclass _Parser(object):\n \"\"\"Text format parser for protocol message.\"\"\"\n\n def __init__(self,\n allow_unknown_extension=False,\n allow_field_number=False,\n descriptor_pool=None,\n allow_unknown_field=False):\n self.allow_unknown_extension = allow_unknown_extension\n self.allow_field_number = allow_field_number\n self.descriptor_pool = descriptor_pool\n self.allow_unknown_field = allow_unknown_field\n\n def ParseLines(self, lines, message):\n \"\"\"Parses a text representation of a protocol message into a message.\"\"\"\n self._allow_multiple_scalars = False\n self._ParseOrMerge(lines, message)\n return message\n\n def MergeLines(self, lines, message):\n \"\"\"Merges a text representation of a protocol message into a message.\"\"\"\n self._allow_multiple_scalars = True\n self._ParseOrMerge(lines, message)\n return message\n\n def _ParseOrMerge(self, lines, message):\n \"\"\"Converts a text representation of a protocol message into a message.\n\n Args:\n lines: Lines of a message's text representation.\n message: A protocol buffer message to merge into.\n\n Raises:\n ParseError: On text parsing problems.\n \"\"\"\n # Tokenize expects native str lines.\n str_lines = (\n line if isinstance(line, str) else line.decode('utf-8')\n for line in lines)\n tokenizer = Tokenizer(str_lines)\n while not tokenizer.AtEnd():\n self._MergeField(tokenizer, message)\n\n def _MergeField(self, tokenizer, message):\n \"\"\"Merges a single protocol message field into a message.\n\n Args:\n tokenizer: A tokenizer to parse the field name and values.\n message: A protocol message to record the data.\n\n Raises:\n ParseError: In case of text parsing problems.\n \"\"\"\n message_descriptor = message.DESCRIPTOR\n if (message_descriptor.full_name == _ANY_FULL_TYPE_NAME and\n tokenizer.TryConsume('[')):\n type_url_prefix, packed_type_name = self._ConsumeAnyTypeUrl(tokenizer)\n tokenizer.Consume(']')\n tokenizer.TryConsume(':')\n if tokenizer.TryConsume('<'):\n expanded_any_end_token = '>'\n else:\n tokenizer.Consume('{')\n expanded_any_end_token = '}'\n expanded_any_sub_message = _BuildMessageFromTypeName(packed_type_name,\n self.descriptor_pool)\n if not expanded_any_sub_message:\n raise ParseError('Type %s not found in descriptor pool' %\n packed_type_name)\n while not tokenizer.TryConsume(expanded_any_end_token):\n if tokenizer.AtEnd():\n raise tokenizer.ParseErrorPreviousToken('Expected \"%s\".' %\n (expanded_any_end_token,))\n self._MergeField(tokenizer, expanded_any_sub_message)\n deterministic = False\n\n message.Pack(expanded_any_sub_message,\n type_url_prefix=type_url_prefix,\n deterministic=deterministic)\n return\n\n if tokenizer.TryConsume('['):\n name = [tokenizer.ConsumeIdentifier()]\n while tokenizer.TryConsume('.'):\n name.append(tokenizer.ConsumeIdentifier())\n name = '.'.join(name)\n\n if not message_descriptor.is_extendable:\n raise tokenizer.ParseErrorPreviousToken(\n 'Message type \"%s\" does not have extensions.' %\n message_descriptor.full_name)\n # pylint: disable=protected-access\n field = message.Extensions._FindExtensionByName(name)\n # pylint: enable=protected-access\n\n\n if not field:\n if self.allow_unknown_extension:\n field = None\n else:\n raise tokenizer.ParseErrorPreviousToken(\n 'Extension \"%s\" not registered. '\n 'Did you import the _pb2 module which defines it? '\n 'If you are trying to place the extension in the MessageSet '\n 'field of another message that is in an Any or MessageSet field, '\n 'that message\\'s _pb2 module must be imported as well' % name)\n elif message_descriptor != field.containing_type:\n raise tokenizer.ParseErrorPreviousToken(\n 'Extension \"%s\" does not extend message type \"%s\".' %\n (name, message_descriptor.full_name))\n\n tokenizer.Consume(']')\n\n else:\n name = tokenizer.ConsumeIdentifierOrNumber()\n if self.allow_field_number and name.isdigit():\n number = ParseInteger(name, True, True)\n field = message_descriptor.fields_by_number.get(number, None)\n if not field and message_descriptor.is_extendable:\n field = message.Extensions._FindExtensionByNumber(number)\n else:\n field = message_descriptor.fields_by_name.get(name, None)\n\n # Group names are expected to be capitalized as they appear in the\n # .proto file, which actually matches their type names, not their field\n # names.\n if not field:\n field = message_descriptor.fields_by_name.get(name.lower(), None)\n if field and field.type != descriptor.FieldDescriptor.TYPE_GROUP:\n field = None\n\n if (field and field.type == descriptor.FieldDescriptor.TYPE_GROUP and\n field.message_type.name != name):\n field = None\n\n if not field and not self.allow_unknown_field:\n raise tokenizer.ParseErrorPreviousToken(\n 'Message type \"%s\" has no field named \"%s\".' %\n (message_descriptor.full_name, name))\n\n if field:\n if not self._allow_multiple_scalars and field.containing_oneof:\n # Check if there's a different field set in this oneof.\n # Note that we ignore the case if the same field was set before, and we\n # apply _allow_multiple_scalars to non-scalar fields as well.\n which_oneof = message.WhichOneof(field.containing_oneof.name)\n if which_oneof is not None and which_oneof != field.name:\n raise tokenizer.ParseErrorPreviousToken(\n 'Field \"%s\" is specified along with field \"%s\", another member '\n 'of oneof \"%s\" for message type \"%s\".' %\n (field.name, which_oneof, field.containing_oneof.name,\n message_descriptor.full_name))\n\n if field.cpp_type == descriptor.FieldDescriptor.CPPTYPE_MESSAGE:\n tokenizer.TryConsume(':')\n merger = self._MergeMessageField\n else:\n tokenizer.Consume(':')\n merger = self._MergeScalarField\n\n if (field.label == descriptor.FieldDescriptor.LABEL_REPEATED and\n tokenizer.TryConsume('[')):\n # Short repeated format, e.g. \"foo: [1, 2, 3]\"\n if not tokenizer.TryConsume(']'):\n while True:\n merger(tokenizer, message, field)\n if tokenizer.TryConsume(']'):\n break\n tokenizer.Consume(',')\n\n else:\n merger(tokenizer, message, field)\n\n else: # Proto field is unknown.\n assert (self.allow_unknown_extension or self.allow_unknown_field)\n _SkipFieldContents(tokenizer)\n\n # For historical reasons, fields may optionally be separated by commas or\n # semicolons.\n if not tokenizer.TryConsume(','):\n tokenizer.TryConsume(';')\n\n\n def _ConsumeAnyTypeUrl(self, tokenizer):\n \"\"\"Consumes a google.protobuf.Any type URL and returns the type name.\"\"\"\n # Consume \"type.googleapis.com/\".\n prefix = [tokenizer.ConsumeIdentifier()]\n tokenizer.Consume('.')\n prefix.append(tokenizer.ConsumeIdentifier())\n tokenizer.Consume('.')\n prefix.append(tokenizer.ConsumeIdentifier())\n tokenizer.Consume('/')\n # Consume the fully-qualified type name.\n name = [tokenizer.ConsumeIdentifier()]\n while tokenizer.TryConsume('.'):\n name.append(tokenizer.ConsumeIdentifier())\n return '.'.join(prefix), '.'.join(name)\n\n def _MergeMessageField(self, tokenizer, message, field):\n \"\"\"Merges a single scalar field into a message.\n\n Args:\n tokenizer: A tokenizer to parse the field value.\n message: The message of which field is a member.\n field: The descriptor of the field to be merged.\n\n Raises:\n ParseError: In case of text parsing problems.\n \"\"\"\n is_map_entry = _IsMapEntry(field)\n\n if tokenizer.TryConsume('<'):\n end_token = '>'\n else:\n tokenizer.Consume('{')\n end_token = '}'\n\n if field.label == descriptor.FieldDescriptor.LABEL_REPEATED:\n if field.is_extension:\n sub_message = message.Extensions[field].add()\n elif is_map_entry:\n sub_message = getattr(message, field.name).GetEntryClass()()\n else:\n sub_message = getattr(message, field.name).add()\n else:\n if field.is_extension:\n if (not self._allow_multiple_scalars and\n message.HasExtension(field)):\n raise tokenizer.ParseErrorPreviousToken(\n 'Message type \"%s\" should not have multiple \"%s\" extensions.' %\n (message.DESCRIPTOR.full_name, field.full_name))\n sub_message = message.Extensions[field]\n else:\n # Also apply _allow_multiple_scalars to message field.\n # TODO(jieluo): Change to _allow_singular_overwrites.\n if (not self._allow_multiple_scalars and\n message.HasField(field.name)):\n raise tokenizer.ParseErrorPreviousToken(\n 'Message type \"%s\" should not have multiple \"%s\" fields.' %\n (message.DESCRIPTOR.full_name, field.name))\n sub_message = getattr(message, field.name)\n sub_message.SetInParent()\n\n while not tokenizer.TryConsume(end_token):\n if tokenizer.AtEnd():\n raise tokenizer.ParseErrorPreviousToken('Expected \"%s\".' % (end_token,))\n self._MergeField(tokenizer, sub_message)\n\n if is_map_entry:\n value_cpptype = field.message_type.fields_by_name['value'].cpp_type\n if value_cpptype == descriptor.FieldDescriptor.CPPTYPE_MESSAGE:\n value = getattr(message, field.name)[sub_message.key]\n value.CopyFrom(sub_message.value)\n else:\n getattr(message, field.name)[sub_message.key] = sub_message.value\n\n @staticmethod\n def _IsProto3Syntax(message):\n message_descriptor = message.DESCRIPTOR\n return (hasattr(message_descriptor, 'syntax') and\n message_descriptor.syntax == 'proto3')\n\n def _MergeScalarField(self, tokenizer, message, field):\n \"\"\"Merges a single scalar field into a message.\n\n Args:\n tokenizer: A tokenizer to parse the field value.\n message: A protocol message to record the data.\n field: The descriptor of the field to be merged.\n\n Raises:\n ParseError: In case of text parsing problems.\n RuntimeError: On runtime errors.\n \"\"\"\n _ = self.allow_unknown_extension\n value = None\n\n if field.type in (descriptor.FieldDescriptor.TYPE_INT32,\n descriptor.FieldDescriptor.TYPE_SINT32,\n descriptor.FieldDescriptor.TYPE_SFIXED32):\n value = _ConsumeInt32(tokenizer)\n elif field.type in (descriptor.FieldDescriptor.TYPE_INT64,\n descriptor.FieldDescriptor.TYPE_SINT64,\n descriptor.FieldDescriptor.TYPE_SFIXED64):\n value = _ConsumeInt64(tokenizer)\n elif field.type in (descriptor.FieldDescriptor.TYPE_UINT32,\n descriptor.FieldDescriptor.TYPE_FIXED32):\n value = _ConsumeUint32(tokenizer)\n elif field.type in (descriptor.FieldDescriptor.TYPE_UINT64,\n descriptor.FieldDescriptor.TYPE_FIXED64):\n value = _ConsumeUint64(tokenizer)\n elif field.type in (descriptor.FieldDescriptor.TYPE_FLOAT,\n descriptor.FieldDescriptor.TYPE_DOUBLE):\n value = tokenizer.ConsumeFloat()\n elif field.type == descriptor.FieldDescriptor.TYPE_BOOL:\n value = tokenizer.ConsumeBool()\n elif field.type == descriptor.FieldDescriptor.TYPE_STRING:\n value = tokenizer.ConsumeString()\n elif field.type == descriptor.FieldDescriptor.TYPE_BYTES:\n value = tokenizer.ConsumeByteString()\n elif field.type == descriptor.FieldDescriptor.TYPE_ENUM:\n value = tokenizer.ConsumeEnum(field)\n else:\n raise RuntimeError('Unknown field type %d' % field.type)\n\n if field.label == descriptor.FieldDescriptor.LABEL_REPEATED:\n if field.is_extension:\n message.Extensions[field].append(value)\n else:\n getattr(message, field.name).append(value)\n else:\n if field.is_extension:\n if (not self._allow_multiple_scalars and\n not self._IsProto3Syntax(message) and\n message.HasExtension(field)):\n raise tokenizer.ParseErrorPreviousToken(\n 'Message type \"%s\" should not have multiple \"%s\" extensions.' %\n (message.DESCRIPTOR.full_name, field.full_name))\n else:\n message.Extensions[field] = value\n else:\n duplicate_error = False\n if not self._allow_multiple_scalars:\n if self._IsProto3Syntax(message):\n # Proto3 doesn't represent presence so we try best effort to check\n # multiple scalars by compare to default values.\n duplicate_error = bool(getattr(message, field.name))\n else:\n duplicate_error = message.HasField(field.name)\n\n if duplicate_error:\n raise tokenizer.ParseErrorPreviousToken(\n 'Message type \"%s\" should not have multiple \"%s\" fields.' %\n (message.DESCRIPTOR.full_name, field.name))\n else:\n setattr(message, field.name, value)\n\n\ndef _SkipFieldContents(tokenizer):\n \"\"\"Skips over contents (value or message) of a field.\n\n Args:\n tokenizer: A tokenizer to parse the field name and values.\n \"\"\"\n # Try to guess the type of this field.\n # If this field is not a message, there should be a \":\" between the\n # field name and the field value and also the field value should not\n # start with \"{\" or \"<\" which indicates the beginning of a message body.\n # If there is no \":\" or there is a \"{\" or \"<\" after \":\", this field has\n # to be a message or the input is ill-formed.\n if tokenizer.TryConsume(':') and not tokenizer.LookingAt(\n '{') and not tokenizer.LookingAt('<'):\n _SkipFieldValue(tokenizer)\n else:\n _SkipFieldMessage(tokenizer)\n\n\ndef _SkipField(tokenizer):\n \"\"\"Skips over a complete field (name and value/message).\n\n Args:\n tokenizer: A tokenizer to parse the field name and values.\n \"\"\"\n if tokenizer.TryConsume('['):\n # Consume extension name.\n tokenizer.ConsumeIdentifier()\n while tokenizer.TryConsume('.'):\n tokenizer.ConsumeIdentifier()\n tokenizer.Consume(']')\n else:\n tokenizer.ConsumeIdentifierOrNumber()\n\n _SkipFieldContents(tokenizer)\n\n # For historical reasons, fields may optionally be separated by commas or\n # semicolons.\n if not tokenizer.TryConsume(','):\n tokenizer.TryConsume(';')\n\n\ndef _SkipFieldMessage(tokenizer):\n \"\"\"Skips over a field message.\n\n Args:\n tokenizer: A tokenizer to parse the field name and values.\n \"\"\"\n\n if tokenizer.TryConsume('<'):\n delimiter = '>'\n else:\n tokenizer.Consume('{')\n delimiter = '}'\n\n while not tokenizer.LookingAt('>') and not tokenizer.LookingAt('}'):\n _SkipField(tokenizer)\n\n tokenizer.Consume(delimiter)\n\n\ndef _SkipFieldValue(tokenizer):\n \"\"\"Skips over a field value.\n\n Args:\n tokenizer: A tokenizer to parse the field name and values.\n\n Raises:\n ParseError: In case an invalid field value is found.\n \"\"\"\n # String/bytes tokens can come in multiple adjacent string literals.\n # If we can consume one, consume as many as we can.\n if tokenizer.TryConsumeByteString():\n while tokenizer.TryConsumeByteString():\n pass\n return\n\n if (not tokenizer.TryConsumeIdentifier() and\n not _TryConsumeInt64(tokenizer) and not _TryConsumeUint64(tokenizer) and\n not tokenizer.TryConsumeFloat()):\n raise ParseError('Invalid field value: ' + tokenizer.token)\n\n\nclass Tokenizer(object):\n \"\"\"Protocol buffer text representation tokenizer.\n\n This class handles the lower level string parsing by splitting it into\n meaningful tokens.\n\n It was directly ported from the Java protocol buffer API.\n \"\"\"\n\n _WHITESPACE = re.compile(r'\\s+')\n _COMMENT = re.compile(r'(\\s*#.*$)', re.MULTILINE)\n _WHITESPACE_OR_COMMENT = re.compile(r'(\\s|(#.*$))+', re.MULTILINE)\n _TOKEN = re.compile('|'.join([\n r'[a-zA-Z_][0-9a-zA-Z_+-]*', # an identifier\n r'([0-9+-]|(\\.[0-9]))[0-9a-zA-Z_.+-]*', # a number\n ] + [ # quoted str for each quote mark\n # Avoid backtracking! https://stackoverflow.com/a/844267\n r'{qt}[^{qt}\\n\\\\]*((\\\\.)+[^{qt}\\n\\\\]*)*({qt}|\\\\?$)'.format(qt=mark)\n for mark in _QUOTES\n ]))\n\n _IDENTIFIER = re.compile(r'[^\\d\\W]\\w*')\n _IDENTIFIER_OR_NUMBER = re.compile(r'\\w+')\n\n def __init__(self, lines, skip_comments=True):\n self._position = 0\n self._line = -1\n self._column = 0\n self._token_start = None\n self.token = ''\n self._lines = iter(lines)\n self._current_line = ''\n self._previous_line = 0\n self._previous_column = 0\n self._more_lines = True\n self._skip_comments = skip_comments\n self._whitespace_pattern = (skip_comments and self._WHITESPACE_OR_COMMENT\n or self._WHITESPACE)\n self._SkipWhitespace()\n self.NextToken()\n\n def LookingAt(self, token):\n return self.token == token\n\n def AtEnd(self):\n \"\"\"Checks the end of the text was reached.\n\n Returns:\n True iff the end was reached.\n \"\"\"\n return not self.token\n\n def _PopLine(self):\n while len(self._current_line) <= self._column:\n try:\n self._current_line = next(self._lines)\n except StopIteration:\n self._current_line = ''\n self._more_lines = False\n return\n else:\n self._line += 1\n self._column = 0\n\n def _SkipWhitespace(self):\n while True:\n self._PopLine()\n match = self._whitespace_pattern.match(self._current_line, self._column)\n if not match:\n break\n length = len(match.group(0))\n self._column += length\n\n def TryConsume(self, token):\n \"\"\"Tries to consume a given piece of text.\n\n Args:\n token: Text to consume.\n\n Returns:\n True iff the text was consumed.\n \"\"\"\n if self.token == token:\n self.NextToken()\n return True\n return False\n\n def Consume(self, token):\n \"\"\"Consumes a piece of text.\n\n Args:\n token: Text to consume.\n\n Raises:\n ParseError: If the text couldn't be consumed.\n \"\"\"\n if not self.TryConsume(token):\n raise self.ParseError('Expected \"%s\".' % token)\n\n def ConsumeComment(self):\n result = self.token\n if not self._COMMENT.match(result):\n raise self.ParseError('Expected comment.')\n self.NextToken()\n return result\n\n def ConsumeCommentOrTrailingComment(self):\n \"\"\"Consumes a comment, returns a 2-tuple (trailing bool, comment str).\"\"\"\n\n # Tokenizer initializes _previous_line and _previous_column to 0. As the\n # tokenizer starts, it looks like there is a previous token on the line.\n just_started = self._line == 0 and self._column == 0\n\n before_parsing = self._previous_line\n comment = self.ConsumeComment()\n\n # A trailing comment is a comment on the same line than the previous token.\n trailing = (self._previous_line == before_parsing\n and not just_started)\n\n return trailing, comment\n\n def TryConsumeIdentifier(self):\n try:\n self.ConsumeIdentifier()\n return True\n except ParseError:\n return False\n\n def ConsumeIdentifier(self):\n \"\"\"Consumes protocol message field identifier.\n\n Returns:\n Identifier string.\n\n Raises:\n ParseError: If an identifier couldn't be consumed.\n \"\"\"\n result = self.token\n if not self._IDENTIFIER.match(result):\n raise self.ParseError('Expected identifier.')\n self.NextToken()\n return result\n\n def TryConsumeIdentifierOrNumber(self):\n try:\n self.ConsumeIdentifierOrNumber()\n return True\n except ParseError:\n return False\n\n def ConsumeIdentifierOrNumber(self):\n \"\"\"Consumes protocol message field identifier.\n\n Returns:\n Identifier string.\n\n Raises:\n ParseError: If an identifier couldn't be consumed.\n \"\"\"\n result = self.token\n if not self._IDENTIFIER_OR_NUMBER.match(result):\n raise self.ParseError('Expected identifier or number, got %s.' % result)\n self.NextToken()\n return result\n\n def TryConsumeInteger(self):\n try:\n self.ConsumeInteger()\n return True\n except ParseError:\n return False\n\n def ConsumeInteger(self):\n \"\"\"Consumes an integer number.\n\n Returns:\n The integer parsed.\n\n Raises:\n ParseError: If an integer couldn't be consumed.\n \"\"\"\n try:\n result = _ParseAbstractInteger(self.token)\n except ValueError as e:\n raise self.ParseError(str(e))\n self.NextToken()\n return result\n\n def TryConsumeFloat(self):\n try:\n self.ConsumeFloat()\n return True\n except ParseError:\n return False\n\n def ConsumeFloat(self):\n \"\"\"Consumes an floating point number.\n\n Returns:\n The number parsed.\n\n Raises:\n ParseError: If a floating point number couldn't be consumed.\n \"\"\"\n try:\n result = ParseFloat(self.token)\n except ValueError as e:\n raise self.ParseError(str(e))\n self.NextToken()\n return result\n\n def ConsumeBool(self):\n \"\"\"Consumes a boolean value.\n\n Returns:\n The bool parsed.\n\n Raises:\n ParseError: If a boolean value couldn't be consumed.\n \"\"\"\n try:\n result = ParseBool(self.token)\n except ValueError as e:\n raise self.ParseError(str(e))\n self.NextToken()\n return result\n\n def TryConsumeByteString(self):\n try:\n self.ConsumeByteString()\n return True\n except ParseError:\n return False\n\n def ConsumeString(self):\n \"\"\"Consumes a string value.\n\n Returns:\n The string parsed.\n\n Raises:\n ParseError: If a string value couldn't be consumed.\n \"\"\"\n the_bytes = self.ConsumeByteString()\n try:\n return str(the_bytes, 'utf-8')\n except UnicodeDecodeError as e:\n raise self._StringParseError(e)\n\n def ConsumeByteString(self):\n \"\"\"Consumes a byte array value.\n\n Returns:\n The array parsed (as a string).\n\n Raises:\n ParseError: If a byte array value couldn't be consumed.\n \"\"\"\n the_list = [self._ConsumeSingleByteString()]\n while self.token and self.token[0] in _QUOTES:\n the_list.append(self._ConsumeSingleByteString())\n return b''.join(the_list)\n\n def _ConsumeSingleByteString(self):\n \"\"\"Consume one token of a string literal.\n\n String literals (whether bytes or text) can come in multiple adjacent\n tokens which are automatically concatenated, like in C or Python. This\n method only consumes one token.\n\n Returns:\n The token parsed.\n Raises:\n ParseError: When the wrong format data is found.\n \"\"\"\n text = self.token\n if len(text) < 1 or text[0] not in _QUOTES:\n raise self.ParseError('Expected string but found: %r' % (text,))\n\n if len(text) < 2 or text[-1] != text[0]:\n raise self.ParseError('String missing ending quote: %r' % (text,))\n\n try:\n result = text_encoding.CUnescape(text[1:-1])\n except ValueError as e:\n raise self.ParseError(str(e))\n self.NextToken()\n return result\n\n def ConsumeEnum(self, field):\n try:\n result = ParseEnum(field, self.token)\n except ValueError as e:\n raise self.ParseError(str(e))\n self.NextToken()\n return result\n\n def ParseErrorPreviousToken(self, message):\n \"\"\"Creates and *returns* a ParseError for the previously read token.\n\n Args:\n message: A message to set for the exception.\n\n Returns:\n A ParseError instance.\n \"\"\"\n return ParseError(message, self._previous_line + 1,\n self._previous_column + 1)\n\n def ParseError(self, message):\n \"\"\"Creates and *returns* a ParseError for the current token.\"\"\"\n return ParseError('\\'' + self._current_line + '\\': ' + message,\n self._line + 1, self._column + 1)\n\n def _StringParseError(self, e):\n return self.ParseError('Couldn\\'t parse string: ' + str(e))\n\n def NextToken(self):\n \"\"\"Reads the next meaningful token.\"\"\"\n self._previous_line = self._line\n self._previous_column = self._column\n\n self._column += len(self.token)\n self._SkipWhitespace()\n\n if not self._more_lines:\n self.token = ''\n return\n\n match = self._TOKEN.match(self._current_line, self._column)\n if not match and not self._skip_comments:\n match = self._COMMENT.match(self._current_line, self._column)\n if match:\n token = match.group(0)\n self.token = token\n else:\n self.token = self._current_line[self._column]\n\n# Aliased so it can still be accessed by current visibility violators.\n# TODO(dbarnett): Migrate violators to textformat_tokenizer.\n_Tokenizer = Tokenizer # pylint: disable=invalid-name\n\n\ndef _ConsumeInt32(tokenizer):\n \"\"\"Consumes a signed 32bit integer number from tokenizer.\n\n Args:\n tokenizer: A tokenizer used to parse the number.\n\n Returns:\n The integer parsed.\n\n Raises:\n ParseError: If a signed 32bit integer couldn't be consumed.\n \"\"\"\n return _ConsumeInteger(tokenizer, is_signed=True, is_long=False)\n\n\ndef _ConsumeUint32(tokenizer):\n \"\"\"Consumes an unsigned 32bit integer number from tokenizer.\n\n Args:\n tokenizer: A tokenizer used to parse the number.\n\n Returns:\n The integer parsed.\n\n Raises:\n ParseError: If an unsigned 32bit integer couldn't be consumed.\n \"\"\"\n return _ConsumeInteger(tokenizer, is_signed=False, is_long=False)\n\n\ndef _TryConsumeInt64(tokenizer):\n try:\n _ConsumeInt64(tokenizer)\n return True\n except ParseError:\n return False\n\n\ndef _ConsumeInt64(tokenizer):\n \"\"\"Consumes a signed 32bit integer number from tokenizer.\n\n Args:\n tokenizer: A tokenizer used to parse the number.\n\n Returns:\n The integer parsed.\n\n Raises:\n ParseError: If a signed 32bit integer couldn't be consumed.\n \"\"\"\n return _ConsumeInteger(tokenizer, is_signed=True, is_long=True)\n\n\ndef _TryConsumeUint64(tokenizer):\n try:\n _ConsumeUint64(tokenizer)\n return True\n except ParseError:\n return False\n\n\ndef _ConsumeUint64(tokenizer):\n \"\"\"Consumes an unsigned 64bit integer number from tokenizer.\n\n Args:\n tokenizer: A tokenizer used to parse the number.\n\n Returns:\n The integer parsed.\n\n Raises:\n ParseError: If an unsigned 64bit integer couldn't be consumed.\n \"\"\"\n return _ConsumeInteger(tokenizer, is_signed=False, is_long=True)\n\n\ndef _ConsumeInteger(tokenizer, is_signed=False, is_long=False):\n \"\"\"Consumes an integer number from tokenizer.\n\n Args:\n tokenizer: A tokenizer used to parse the number.\n is_signed: True if a signed integer must be parsed.\n is_long: True if a long integer must be parsed.\n\n Returns:\n The integer parsed.\n\n Raises:\n ParseError: If an integer with given characteristics couldn't be consumed.\n \"\"\"\n try:\n result = ParseInteger(tokenizer.token, is_signed=is_signed, is_long=is_long)\n except ValueError as e:\n raise tokenizer.ParseError(str(e))\n tokenizer.NextToken()\n return result\n\n\ndef ParseInteger(text, is_signed=False, is_long=False):\n \"\"\"Parses an integer.\n\n Args:\n text: The text to parse.\n is_signed: True if a signed integer must be parsed.\n is_long: True if a long integer must be parsed.\n\n Returns:\n The integer value.\n\n Raises:\n ValueError: Thrown Iff the text is not a valid integer.\n \"\"\"\n # Do the actual parsing. Exception handling is propagated to caller.\n result = _ParseAbstractInteger(text)\n\n # Check if the integer is sane. Exceptions handled by callers.\n checker = _INTEGER_CHECKERS[2 * int(is_long) + int(is_signed)]\n checker.CheckValue(result)\n return result\n\n\ndef _ParseAbstractInteger(text):\n \"\"\"Parses an integer without checking size/signedness.\n\n Args:\n text: The text to parse.\n\n Returns:\n The integer value.\n\n Raises:\n ValueError: Thrown Iff the text is not a valid integer.\n \"\"\"\n # Do the actual parsing. Exception handling is propagated to caller.\n orig_text = text\n c_octal_match = re.match(r'(-?)0(\\d+)$', text)\n if c_octal_match:\n # Python 3 no longer supports 0755 octal syntax without the 'o', so\n # we always use the '0o' prefix for multi-digit numbers starting with 0.\n text = c_octal_match.group(1) + '0o' + c_octal_match.group(2)\n try:\n return int(text, 0)\n except ValueError:\n raise ValueError('Couldn\\'t parse integer: %s' % orig_text)\n\n\ndef ParseFloat(text):\n \"\"\"Parse a floating point number.\n\n Args:\n text: Text to parse.\n\n Returns:\n The number parsed.\n\n Raises:\n ValueError: If a floating point number couldn't be parsed.\n \"\"\"\n try:\n # Assume Python compatible syntax.\n return float(text)\n except ValueError:\n # Check alternative spellings.\n if _FLOAT_INFINITY.match(text):\n if text[0] == '-':\n return float('-inf')\n else:\n return float('inf')\n elif _FLOAT_NAN.match(text):\n return float('nan')\n else:\n # assume '1.0f' format\n try:\n return float(text.rstrip('f'))\n except ValueError:\n raise ValueError('Couldn\\'t parse float: %s' % text)\n\n\ndef ParseBool(text):\n \"\"\"Parse a boolean value.\n\n Args:\n text: Text to parse.\n\n Returns:\n Boolean values parsed\n\n Raises:\n ValueError: If text is not a valid boolean.\n \"\"\"\n if text in ('true', 't', '1', 'True'):\n return True\n elif text in ('false', 'f', '0', 'False'):\n return False\n else:\n raise ValueError('Expected \"true\" or \"false\".')\n\n\ndef ParseEnum(field, value):\n \"\"\"Parse an enum value.\n\n The value can be specified by a number (the enum value), or by\n a string literal (the enum name).\n\n Args:\n field: Enum field descriptor.\n value: String value.\n\n Returns:\n Enum value number.\n\n Raises:\n ValueError: If the enum value could not be parsed.\n \"\"\"\n enum_descriptor = field.enum_type\n try:\n number = int(value, 0)\n except ValueError:\n # Identifier.\n enum_value = enum_descriptor.values_by_name.get(value, None)\n if enum_value is None:\n raise ValueError('Enum type \"%s\" has no value named %s.' %\n (enum_descriptor.full_name, value))\n else:\n # Numeric value.\n if hasattr(field.file, 'syntax'):\n # Attribute is checked for compatibility.\n if field.file.syntax == 'proto3':\n # Proto3 accept numeric unknown enums.\n return number\n enum_value = enum_descriptor.values_by_number.get(number, None)\n if enum_value is None:\n raise ValueError('Enum type \"%s\" has no value with number %d.' %\n (enum_descriptor.full_name, number))\n return enum_value.number\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/timestamp_pb2.py",
"content": "# -*- coding: utf-8 -*-\n# Generated by the protocol buffer compiler. DO NOT EDIT!\n# source: google/protobuf/timestamp.proto\n\"\"\"Generated protocol buffer code.\"\"\"\nfrom google.protobuf.internal import builder as _builder\nfrom google.protobuf import descriptor as _descriptor\nfrom google.protobuf import descriptor_pool as _descriptor_pool\nfrom google.protobuf import symbol_database as _symbol_database\n# @@protoc_insertion_point(imports)\n\n_sym_db = _symbol_database.Default()\n\n\n\n\nDESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\\n\\x1fgoogle/protobuf/timestamp.proto\\x12\\x0fgoogle.protobuf\\\"+\\n\\tTimestamp\\x12\\x0f\\n\\x07seconds\\x18\\x01 \\x01(\\x03\\x12\\r\\n\\x05nanos\\x18\\x02 \\x01(\\x05\\x42\\x85\\x01\\n\\x13\\x63om.google.protobufB\\x0eTimestampProtoP\\x01Z2google.golang.org/protobuf/types/known/timestamppb\\xf8\\x01\\x01\\xa2\\x02\\x03GPB\\xaa\\x02\\x1eGoogle.Protobuf.WellKnownTypesb\\x06proto3')\n\n_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())\n_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.timestamp_pb2', globals())\nif _descriptor._USE_C_DESCRIPTORS == False:\n\n DESCRIPTOR._options = None\n DESCRIPTOR._serialized_options = b'\\n\\023com.google.protobufB\\016TimestampProtoP\\001Z2google.golang.org/protobuf/types/known/timestamppb\\370\\001\\001\\242\\002\\003GPB\\252\\002\\036Google.Protobuf.WellKnownTypes'\n _TIMESTAMP._serialized_start=52\n _TIMESTAMP._serialized_end=95\n# @@protoc_insertion_point(module_scope)\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/type_pb2.py",
"content": "# -*- coding: utf-8 -*-\n# Generated by the protocol buffer compiler. DO NOT EDIT!\n# source: google/protobuf/type.proto\n\"\"\"Generated protocol buffer code.\"\"\"\nfrom google.protobuf.internal import builder as _builder\nfrom google.protobuf import descriptor as _descriptor\nfrom google.protobuf import descriptor_pool as _descriptor_pool\nfrom google.protobuf import symbol_database as _symbol_database\n# @@protoc_insertion_point(imports)\n\n_sym_db = _symbol_database.Default()\n\n\nfrom google.protobuf import any_pb2 as google_dot_protobuf_dot_any__pb2\nfrom google.protobuf import source_context_pb2 as google_dot_protobuf_dot_source__context__pb2\n\n\nDESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\\n\\x1agoogle/protobuf/type.proto\\x12\\x0fgoogle.protobuf\\x1a\\x19google/protobuf/any.proto\\x1a$google/protobuf/source_context.proto\\\"\\xd7\\x01\\n\\x04Type\\x12\\x0c\\n\\x04name\\x18\\x01 \\x01(\\t\\x12&\\n\\x06\\x66ields\\x18\\x02 \\x03(\\x0b\\x32\\x16.google.protobuf.Field\\x12\\x0e\\n\\x06oneofs\\x18\\x03 \\x03(\\t\\x12(\\n\\x07options\\x18\\x04 \\x03(\\x0b\\x32\\x17.google.protobuf.Option\\x12\\x36\\n\\x0esource_context\\x18\\x05 \\x01(\\x0b\\x32\\x1e.google.protobuf.SourceContext\\x12\\'\\n\\x06syntax\\x18\\x06 \\x01(\\x0e\\x32\\x17.google.protobuf.Syntax\\\"\\xd5\\x05\\n\\x05\\x46ield\\x12)\\n\\x04kind\\x18\\x01 \\x01(\\x0e\\x32\\x1b.google.protobuf.Field.Kind\\x12\\x37\\n\\x0b\\x63\\x61rdinality\\x18\\x02 \\x01(\\x0e\\x32\\\".google.protobuf.Field.Cardinality\\x12\\x0e\\n\\x06number\\x18\\x03 \\x01(\\x05\\x12\\x0c\\n\\x04name\\x18\\x04 \\x01(\\t\\x12\\x10\\n\\x08type_url\\x18\\x06 \\x01(\\t\\x12\\x13\\n\\x0boneof_index\\x18\\x07 \\x01(\\x05\\x12\\x0e\\n\\x06packed\\x18\\x08 \\x01(\\x08\\x12(\\n\\x07options\\x18\\t \\x03(\\x0b\\x32\\x17.google.protobuf.Option\\x12\\x11\\n\\tjson_name\\x18\\n \\x01(\\t\\x12\\x15\\n\\rdefault_value\\x18\\x0b \\x01(\\t\\\"\\xc8\\x02\\n\\x04Kind\\x12\\x10\\n\\x0cTYPE_UNKNOWN\\x10\\x00\\x12\\x0f\\n\\x0bTYPE_DOUBLE\\x10\\x01\\x12\\x0e\\n\\nTYPE_FLOAT\\x10\\x02\\x12\\x0e\\n\\nTYPE_INT64\\x10\\x03\\x12\\x0f\\n\\x0bTYPE_UINT64\\x10\\x04\\x12\\x0e\\n\\nTYPE_INT32\\x10\\x05\\x12\\x10\\n\\x0cTYPE_FIXED64\\x10\\x06\\x12\\x10\\n\\x0cTYPE_FIXED32\\x10\\x07\\x12\\r\\n\\tTYPE_BOOL\\x10\\x08\\x12\\x0f\\n\\x0bTYPE_STRING\\x10\\t\\x12\\x0e\\n\\nTYPE_GROUP\\x10\\n\\x12\\x10\\n\\x0cTYPE_MESSAGE\\x10\\x0b\\x12\\x0e\\n\\nTYPE_BYTES\\x10\\x0c\\x12\\x0f\\n\\x0bTYPE_UINT32\\x10\\r\\x12\\r\\n\\tTYPE_ENUM\\x10\\x0e\\x12\\x11\\n\\rTYPE_SFIXED32\\x10\\x0f\\x12\\x11\\n\\rTYPE_SFIXED64\\x10\\x10\\x12\\x0f\\n\\x0bTYPE_SINT32\\x10\\x11\\x12\\x0f\\n\\x0bTYPE_SINT64\\x10\\x12\\\"t\\n\\x0b\\x43\\x61rdinality\\x12\\x17\\n\\x13\\x43\\x41RDINALITY_UNKNOWN\\x10\\x00\\x12\\x18\\n\\x14\\x43\\x41RDINALITY_OPTIONAL\\x10\\x01\\x12\\x18\\n\\x14\\x43\\x41RDINALITY_REQUIRED\\x10\\x02\\x12\\x18\\n\\x14\\x43\\x41RDINALITY_REPEATED\\x10\\x03\\\"\\xce\\x01\\n\\x04\\x45num\\x12\\x0c\\n\\x04name\\x18\\x01 \\x01(\\t\\x12-\\n\\tenumvalue\\x18\\x02 \\x03(\\x0b\\x32\\x1a.google.protobuf.EnumValue\\x12(\\n\\x07options\\x18\\x03 \\x03(\\x0b\\x32\\x17.google.protobuf.Option\\x12\\x36\\n\\x0esource_context\\x18\\x04 \\x01(\\x0b\\x32\\x1e.google.protobuf.SourceContext\\x12\\'\\n\\x06syntax\\x18\\x05 \\x01(\\x0e\\x32\\x17.google.protobuf.Syntax\\\"S\\n\\tEnumValue\\x12\\x0c\\n\\x04name\\x18\\x01 \\x01(\\t\\x12\\x0e\\n\\x06number\\x18\\x02 \\x01(\\x05\\x12(\\n\\x07options\\x18\\x03 \\x03(\\x0b\\x32\\x17.google.protobuf.Option\\\";\\n\\x06Option\\x12\\x0c\\n\\x04name\\x18\\x01 \\x01(\\t\\x12#\\n\\x05value\\x18\\x02 \\x01(\\x0b\\x32\\x14.google.protobuf.Any*.\\n\\x06Syntax\\x12\\x11\\n\\rSYNTAX_PROTO2\\x10\\x00\\x12\\x11\\n\\rSYNTAX_PROTO3\\x10\\x01\\x42{\\n\\x13\\x63om.google.protobufB\\tTypeProtoP\\x01Z-google.golang.org/protobuf/types/known/typepb\\xf8\\x01\\x01\\xa2\\x02\\x03GPB\\xaa\\x02\\x1eGoogle.Protobuf.WellKnownTypesb\\x06proto3')\n\n_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())\n_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.type_pb2', globals())\nif _descriptor._USE_C_DESCRIPTORS == False:\n\n DESCRIPTOR._options = None\n DESCRIPTOR._serialized_options = b'\\n\\023com.google.protobufB\\tTypeProtoP\\001Z-google.golang.org/protobuf/types/known/typepb\\370\\001\\001\\242\\002\\003GPB\\252\\002\\036Google.Protobuf.WellKnownTypes'\n _SYNTAX._serialized_start=1413\n _SYNTAX._serialized_end=1459\n _TYPE._serialized_start=113\n _TYPE._serialized_end=328\n _FIELD._serialized_start=331\n _FIELD._serialized_end=1056\n _FIELD_KIND._serialized_start=610\n _FIELD_KIND._serialized_end=938\n _FIELD_CARDINALITY._serialized_start=940\n _FIELD_CARDINALITY._serialized_end=1056\n _ENUM._serialized_start=1059\n _ENUM._serialized_end=1265\n _ENUMVALUE._serialized_start=1267\n _ENUMVALUE._serialized_end=1350\n _OPTION._serialized_start=1352\n _OPTION._serialized_end=1411\n# @@protoc_insertion_point(module_scope)\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/util/__init__.py",
"content": ""
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/util/json_format_pb2.py",
"content": "# -*- coding: utf-8 -*-\n# Generated by the protocol buffer compiler. DO NOT EDIT!\n# source: google/protobuf/util/json_format.proto\n\"\"\"Generated protocol buffer code.\"\"\"\nfrom google.protobuf.internal import builder as _builder\nfrom google.protobuf import descriptor as _descriptor\nfrom google.protobuf import descriptor_pool as _descriptor_pool\nfrom google.protobuf import symbol_database as _symbol_database\n# @@protoc_insertion_point(imports)\n\n_sym_db = _symbol_database.Default()\n\n\n\n\nDESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\\n&google/protobuf/util/json_format.proto\\x12\\x11protobuf_unittest\\\"\\x89\\x01\\n\\x13TestFlagsAndStrings\\x12\\t\\n\\x01\\x41\\x18\\x01 \\x02(\\x05\\x12K\\n\\rrepeatedgroup\\x18\\x02 \\x03(\\n24.protobuf_unittest.TestFlagsAndStrings.RepeatedGroup\\x1a\\x1a\\n\\rRepeatedGroup\\x12\\t\\n\\x01\\x66\\x18\\x03 \\x02(\\t\\\"!\\n\\x14TestBase64ByteArrays\\x12\\t\\n\\x01\\x61\\x18\\x01 \\x02(\\x0c\\\"G\\n\\x12TestJavaScriptJSON\\x12\\t\\n\\x01\\x61\\x18\\x01 \\x01(\\x05\\x12\\r\\n\\x05\\x66inal\\x18\\x02 \\x01(\\x02\\x12\\n\\n\\x02in\\x18\\x03 \\x01(\\t\\x12\\x0b\\n\\x03Var\\x18\\x04 \\x01(\\t\\\"Q\\n\\x18TestJavaScriptOrderJSON1\\x12\\t\\n\\x01\\x64\\x18\\x01 \\x01(\\x05\\x12\\t\\n\\x01\\x63\\x18\\x02 \\x01(\\x05\\x12\\t\\n\\x01x\\x18\\x03 \\x01(\\x08\\x12\\t\\n\\x01\\x62\\x18\\x04 \\x01(\\x05\\x12\\t\\n\\x01\\x61\\x18\\x05 \\x01(\\x05\\\"\\x89\\x01\\n\\x18TestJavaScriptOrderJSON2\\x12\\t\\n\\x01\\x64\\x18\\x01 \\x01(\\x05\\x12\\t\\n\\x01\\x63\\x18\\x02 \\x01(\\x05\\x12\\t\\n\\x01x\\x18\\x03 \\x01(\\x08\\x12\\t\\n\\x01\\x62\\x18\\x04 \\x01(\\x05\\x12\\t\\n\\x01\\x61\\x18\\x05 \\x01(\\x05\\x12\\x36\\n\\x01z\\x18\\x06 \\x03(\\x0b\\x32+.protobuf_unittest.TestJavaScriptOrderJSON1\\\"$\\n\\x0cTestLargeInt\\x12\\t\\n\\x01\\x61\\x18\\x01 \\x02(\\x03\\x12\\t\\n\\x01\\x62\\x18\\x02 \\x02(\\x04\\\"\\xa0\\x01\\n\\x0bTestNumbers\\x12\\x30\\n\\x01\\x61\\x18\\x01 \\x01(\\x0e\\x32%.protobuf_unittest.TestNumbers.MyType\\x12\\t\\n\\x01\\x62\\x18\\x02 \\x01(\\x05\\x12\\t\\n\\x01\\x63\\x18\\x03 \\x01(\\x02\\x12\\t\\n\\x01\\x64\\x18\\x04 \\x01(\\x08\\x12\\t\\n\\x01\\x65\\x18\\x05 \\x01(\\x01\\x12\\t\\n\\x01\\x66\\x18\\x06 \\x01(\\r\\\"(\\n\\x06MyType\\x12\\x06\\n\\x02OK\\x10\\x00\\x12\\x0b\\n\\x07WARNING\\x10\\x01\\x12\\t\\n\\x05\\x45RROR\\x10\\x02\\\"T\\n\\rTestCamelCase\\x12\\x14\\n\\x0cnormal_field\\x18\\x01 \\x01(\\t\\x12\\x15\\n\\rCAPITAL_FIELD\\x18\\x02 \\x01(\\x05\\x12\\x16\\n\\x0e\\x43\\x61melCaseField\\x18\\x03 \\x01(\\x05\\\"|\\n\\x0bTestBoolMap\\x12=\\n\\x08\\x62ool_map\\x18\\x01 \\x03(\\x0b\\x32+.protobuf_unittest.TestBoolMap.BoolMapEntry\\x1a.\\n\\x0c\\x42oolMapEntry\\x12\\x0b\\n\\x03key\\x18\\x01 \\x01(\\x08\\x12\\r\\n\\x05value\\x18\\x02 \\x01(\\x05:\\x02\\x38\\x01\\\"O\\n\\rTestRecursion\\x12\\r\\n\\x05value\\x18\\x01 \\x01(\\x05\\x12/\\n\\x05\\x63hild\\x18\\x02 \\x01(\\x0b\\x32 .protobuf_unittest.TestRecursion\\\"\\x86\\x01\\n\\rTestStringMap\\x12\\x43\\n\\nstring_map\\x18\\x01 \\x03(\\x0b\\x32/.protobuf_unittest.TestStringMap.StringMapEntry\\x1a\\x30\\n\\x0eStringMapEntry\\x12\\x0b\\n\\x03key\\x18\\x01 \\x01(\\t\\x12\\r\\n\\x05value\\x18\\x02 \\x01(\\t:\\x02\\x38\\x01\\\"\\xc4\\x01\\n\\x14TestStringSerializer\\x12\\x15\\n\\rscalar_string\\x18\\x01 \\x01(\\t\\x12\\x17\\n\\x0frepeated_string\\x18\\x02 \\x03(\\t\\x12J\\n\\nstring_map\\x18\\x03 \\x03(\\x0b\\x32\\x36.protobuf_unittest.TestStringSerializer.StringMapEntry\\x1a\\x30\\n\\x0eStringMapEntry\\x12\\x0b\\n\\x03key\\x18\\x01 \\x01(\\t\\x12\\r\\n\\x05value\\x18\\x02 \\x01(\\t:\\x02\\x38\\x01\\\"$\\n\\x18TestMessageWithExtension*\\x08\\x08\\x64\\x10\\x80\\x80\\x80\\x80\\x02\\\"z\\n\\rTestExtension\\x12\\r\\n\\x05value\\x18\\x01 \\x01(\\t2Z\\n\\x03\\x65xt\\x12+.protobuf_unittest.TestMessageWithExtension\\x18\\x64 \\x01(\\x0b\\x32 .protobuf_unittest.TestExtension\\\"Q\\n\\x14TestDefaultEnumValue\\x12\\x39\\n\\nenum_value\\x18\\x01 \\x01(\\x0e\\x32\\x1c.protobuf_unittest.EnumValue:\\x07\\x44\\x45\\x46\\x41ULT*2\\n\\tEnumValue\\x12\\x0c\\n\\x08PROTOCOL\\x10\\x00\\x12\\n\\n\\x06\\x42UFFER\\x10\\x01\\x12\\x0b\\n\\x07\\x44\\x45\\x46\\x41ULT\\x10\\x02')\n\n_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())\n_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.util.json_format_pb2', globals())\nif _descriptor._USE_C_DESCRIPTORS == False:\n TestMessageWithExtension.RegisterExtension(_TESTEXTENSION.extensions_by_name['ext'])\n\n DESCRIPTOR._options = None\n _TESTBOOLMAP_BOOLMAPENTRY._options = None\n _TESTBOOLMAP_BOOLMAPENTRY._serialized_options = b'8\\001'\n _TESTSTRINGMAP_STRINGMAPENTRY._options = None\n _TESTSTRINGMAP_STRINGMAPENTRY._serialized_options = b'8\\001'\n _TESTSTRINGSERIALIZER_STRINGMAPENTRY._options = None\n _TESTSTRINGSERIALIZER_STRINGMAPENTRY._serialized_options = b'8\\001'\n _ENUMVALUE._serialized_start=1607\n _ENUMVALUE._serialized_end=1657\n _TESTFLAGSANDSTRINGS._serialized_start=62\n _TESTFLAGSANDSTRINGS._serialized_end=199\n _TESTFLAGSANDSTRINGS_REPEATEDGROUP._serialized_start=173\n _TESTFLAGSANDSTRINGS_REPEATEDGROUP._serialized_end=199\n _TESTBASE64BYTEARRAYS._serialized_start=201\n _TESTBASE64BYTEARRAYS._serialized_end=234\n _TESTJAVASCRIPTJSON._serialized_start=236\n _TESTJAVASCRIPTJSON._serialized_end=307\n _TESTJAVASCRIPTORDERJSON1._serialized_start=309\n _TESTJAVASCRIPTORDERJSON1._serialized_end=390\n _TESTJAVASCRIPTORDERJSON2._serialized_start=393\n _TESTJAVASCRIPTORDERJSON2._serialized_end=530\n _TESTLARGEINT._serialized_start=532\n _TESTLARGEINT._serialized_end=568\n _TESTNUMBERS._serialized_start=571\n _TESTNUMBERS._serialized_end=731\n _TESTNUMBERS_MYTYPE._serialized_start=691\n _TESTNUMBERS_MYTYPE._serialized_end=731\n _TESTCAMELCASE._serialized_start=733\n _TESTCAMELCASE._serialized_end=817\n _TESTBOOLMAP._serialized_start=819\n _TESTBOOLMAP._serialized_end=943\n _TESTBOOLMAP_BOOLMAPENTRY._serialized_start=897\n _TESTBOOLMAP_BOOLMAPENTRY._serialized_end=943\n _TESTRECURSION._serialized_start=945\n _TESTRECURSION._serialized_end=1024\n _TESTSTRINGMAP._serialized_start=1027\n _TESTSTRINGMAP._serialized_end=1161\n _TESTSTRINGMAP_STRINGMAPENTRY._serialized_start=1113\n _TESTSTRINGMAP_STRINGMAPENTRY._serialized_end=1161\n _TESTSTRINGSERIALIZER._serialized_start=1164\n _TESTSTRINGSERIALIZER._serialized_end=1360\n _TESTSTRINGSERIALIZER_STRINGMAPENTRY._serialized_start=1113\n _TESTSTRINGSERIALIZER_STRINGMAPENTRY._serialized_end=1161\n _TESTMESSAGEWITHEXTENSION._serialized_start=1362\n _TESTMESSAGEWITHEXTENSION._serialized_end=1398\n _TESTEXTENSION._serialized_start=1400\n _TESTEXTENSION._serialized_end=1522\n _TESTDEFAULTENUMVALUE._serialized_start=1524\n _TESTDEFAULTENUMVALUE._serialized_end=1605\n# @@protoc_insertion_point(module_scope)\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/util/json_format_proto3_pb2.py",
"content": "# -*- coding: utf-8 -*-\n# Generated by the protocol buffer compiler. DO NOT EDIT!\n# source: google/protobuf/util/json_format_proto3.proto\n\"\"\"Generated protocol buffer code.\"\"\"\nfrom google.protobuf.internal import builder as _builder\nfrom google.protobuf import descriptor as _descriptor\nfrom google.protobuf import descriptor_pool as _descriptor_pool\nfrom google.protobuf import symbol_database as _symbol_database\n# @@protoc_insertion_point(imports)\n\n_sym_db = _symbol_database.Default()\n\n\nfrom google.protobuf import any_pb2 as google_dot_protobuf_dot_any__pb2\nfrom google.protobuf import duration_pb2 as google_dot_protobuf_dot_duration__pb2\nfrom google.protobuf import field_mask_pb2 as google_dot_protobuf_dot_field__mask__pb2\nfrom google.protobuf import struct_pb2 as google_dot_protobuf_dot_struct__pb2\nfrom google.protobuf import timestamp_pb2 as google_dot_protobuf_dot_timestamp__pb2\nfrom google.protobuf import wrappers_pb2 as google_dot_protobuf_dot_wrappers__pb2\nfrom google.protobuf import unittest_pb2 as google_dot_protobuf_dot_unittest__pb2\n\n\nDESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\\n-google/protobuf/util/json_format_proto3.proto\\x12\\x06proto3\\x1a\\x19google/protobuf/any.proto\\x1a\\x1egoogle/protobuf/duration.proto\\x1a google/protobuf/field_mask.proto\\x1a\\x1cgoogle/protobuf/struct.proto\\x1a\\x1fgoogle/protobuf/timestamp.proto\\x1a\\x1egoogle/protobuf/wrappers.proto\\x1a\\x1egoogle/protobuf/unittest.proto\\\"\\x1c\\n\\x0bMessageType\\x12\\r\\n\\x05value\\x18\\x01 \\x01(\\x05\\\"\\x94\\x05\\n\\x0bTestMessage\\x12\\x12\\n\\nbool_value\\x18\\x01 \\x01(\\x08\\x12\\x13\\n\\x0bint32_value\\x18\\x02 \\x01(\\x05\\x12\\x13\\n\\x0bint64_value\\x18\\x03 \\x01(\\x03\\x12\\x14\\n\\x0cuint32_value\\x18\\x04 \\x01(\\r\\x12\\x14\\n\\x0cuint64_value\\x18\\x05 \\x01(\\x04\\x12\\x13\\n\\x0b\\x66loat_value\\x18\\x06 \\x01(\\x02\\x12\\x14\\n\\x0c\\x64ouble_value\\x18\\x07 \\x01(\\x01\\x12\\x14\\n\\x0cstring_value\\x18\\x08 \\x01(\\t\\x12\\x13\\n\\x0b\\x62ytes_value\\x18\\t \\x01(\\x0c\\x12$\\n\\nenum_value\\x18\\n \\x01(\\x0e\\x32\\x10.proto3.EnumType\\x12*\\n\\rmessage_value\\x18\\x0b \\x01(\\x0b\\x32\\x13.proto3.MessageType\\x12\\x1b\\n\\x13repeated_bool_value\\x18\\x15 \\x03(\\x08\\x12\\x1c\\n\\x14repeated_int32_value\\x18\\x16 \\x03(\\x05\\x12\\x1c\\n\\x14repeated_int64_value\\x18\\x17 \\x03(\\x03\\x12\\x1d\\n\\x15repeated_uint32_value\\x18\\x18 \\x03(\\r\\x12\\x1d\\n\\x15repeated_uint64_value\\x18\\x19 \\x03(\\x04\\x12\\x1c\\n\\x14repeated_float_value\\x18\\x1a \\x03(\\x02\\x12\\x1d\\n\\x15repeated_double_value\\x18\\x1b \\x03(\\x01\\x12\\x1d\\n\\x15repeated_string_value\\x18\\x1c \\x03(\\t\\x12\\x1c\\n\\x14repeated_bytes_value\\x18\\x1d \\x03(\\x0c\\x12-\\n\\x13repeated_enum_value\\x18\\x1e \\x03(\\x0e\\x32\\x10.proto3.EnumType\\x12\\x33\\n\\x16repeated_message_value\\x18\\x1f \\x03(\\x0b\\x32\\x13.proto3.MessageType\\\"\\x8c\\x02\\n\\tTestOneof\\x12\\x1b\\n\\x11oneof_int32_value\\x18\\x01 \\x01(\\x05H\\x00\\x12\\x1c\\n\\x12oneof_string_value\\x18\\x02 \\x01(\\tH\\x00\\x12\\x1b\\n\\x11oneof_bytes_value\\x18\\x03 \\x01(\\x0cH\\x00\\x12,\\n\\x10oneof_enum_value\\x18\\x04 \\x01(\\x0e\\x32\\x10.proto3.EnumTypeH\\x00\\x12\\x32\\n\\x13oneof_message_value\\x18\\x05 \\x01(\\x0b\\x32\\x13.proto3.MessageTypeH\\x00\\x12\\x36\\n\\x10oneof_null_value\\x18\\x06 \\x01(\\x0e\\x32\\x1a.google.protobuf.NullValueH\\x00\\x42\\r\\n\\x0boneof_value\\\"\\xe1\\x04\\n\\x07TestMap\\x12.\\n\\x08\\x62ool_map\\x18\\x01 \\x03(\\x0b\\x32\\x1c.proto3.TestMap.BoolMapEntry\\x12\\x30\\n\\tint32_map\\x18\\x02 \\x03(\\x0b\\x32\\x1d.proto3.TestMap.Int32MapEntry\\x12\\x30\\n\\tint64_map\\x18\\x03 \\x03(\\x0b\\x32\\x1d.proto3.TestMap.Int64MapEntry\\x12\\x32\\n\\nuint32_map\\x18\\x04 \\x03(\\x0b\\x32\\x1e.proto3.TestMap.Uint32MapEntry\\x12\\x32\\n\\nuint64_map\\x18\\x05 \\x03(\\x0b\\x32\\x1e.proto3.TestMap.Uint64MapEntry\\x12\\x32\\n\\nstring_map\\x18\\x06 \\x03(\\x0b\\x32\\x1e.proto3.TestMap.StringMapEntry\\x1a.\\n\\x0c\\x42oolMapEntry\\x12\\x0b\\n\\x03key\\x18\\x01 \\x01(\\x08\\x12\\r\\n\\x05value\\x18\\x02 \\x01(\\x05:\\x02\\x38\\x01\\x1a/\\n\\rInt32MapEntry\\x12\\x0b\\n\\x03key\\x18\\x01 \\x01(\\x05\\x12\\r\\n\\x05value\\x18\\x02 \\x01(\\x05:\\x02\\x38\\x01\\x1a/\\n\\rInt64MapEntry\\x12\\x0b\\n\\x03key\\x18\\x01 \\x01(\\x03\\x12\\r\\n\\x05value\\x18\\x02 \\x01(\\x05:\\x02\\x38\\x01\\x1a\\x30\\n\\x0eUint32MapEntry\\x12\\x0b\\n\\x03key\\x18\\x01 \\x01(\\r\\x12\\r\\n\\x05value\\x18\\x02 \\x01(\\x05:\\x02\\x38\\x01\\x1a\\x30\\n\\x0eUint64MapEntry\\x12\\x0b\\n\\x03key\\x18\\x01 \\x01(\\x04\\x12\\r\\n\\x05value\\x18\\x02 \\x01(\\x05:\\x02\\x38\\x01\\x1a\\x30\\n\\x0eStringMapEntry\\x12\\x0b\\n\\x03key\\x18\\x01 \\x01(\\t\\x12\\r\\n\\x05value\\x18\\x02 \\x01(\\x05:\\x02\\x38\\x01\\\"\\x85\\x06\\n\\rTestNestedMap\\x12\\x34\\n\\x08\\x62ool_map\\x18\\x01 \\x03(\\x0b\\x32\\\".proto3.TestNestedMap.BoolMapEntry\\x12\\x36\\n\\tint32_map\\x18\\x02 \\x03(\\x0b\\x32#.proto3.TestNestedMap.Int32MapEntry\\x12\\x36\\n\\tint64_map\\x18\\x03 \\x03(\\x0b\\x32#.proto3.TestNestedMap.Int64MapEntry\\x12\\x38\\n\\nuint32_map\\x18\\x04 \\x03(\\x0b\\x32$.proto3.TestNestedMap.Uint32MapEntry\\x12\\x38\\n\\nuint64_map\\x18\\x05 \\x03(\\x0b\\x32$.proto3.TestNestedMap.Uint64MapEntry\\x12\\x38\\n\\nstring_map\\x18\\x06 \\x03(\\x0b\\x32$.proto3.TestNestedMap.StringMapEntry\\x12\\x32\\n\\x07map_map\\x18\\x07 \\x03(\\x0b\\x32!.proto3.TestNestedMap.MapMapEntry\\x1a.\\n\\x0c\\x42oolMapEntry\\x12\\x0b\\n\\x03key\\x18\\x01 \\x01(\\x08\\x12\\r\\n\\x05value\\x18\\x02 \\x01(\\x05:\\x02\\x38\\x01\\x1a/\\n\\rInt32MapEntry\\x12\\x0b\\n\\x03key\\x18\\x01 \\x01(\\x05\\x12\\r\\n\\x05value\\x18\\x02 \\x01(\\x05:\\x02\\x38\\x01\\x1a/\\n\\rInt64MapEntry\\x12\\x0b\\n\\x03key\\x18\\x01 \\x01(\\x03\\x12\\r\\n\\x05value\\x18\\x02 \\x01(\\x05:\\x02\\x38\\x01\\x1a\\x30\\n\\x0eUint32MapEntry\\x12\\x0b\\n\\x03key\\x18\\x01 \\x01(\\r\\x12\\r\\n\\x05value\\x18\\x02 \\x01(\\x05:\\x02\\x38\\x01\\x1a\\x30\\n\\x0eUint64MapEntry\\x12\\x0b\\n\\x03key\\x18\\x01 \\x01(\\x04\\x12\\r\\n\\x05value\\x18\\x02 \\x01(\\x05:\\x02\\x38\\x01\\x1a\\x30\\n\\x0eStringMapEntry\\x12\\x0b\\n\\x03key\\x18\\x01 \\x01(\\t\\x12\\r\\n\\x05value\\x18\\x02 \\x01(\\x05:\\x02\\x38\\x01\\x1a\\x44\\n\\x0bMapMapEntry\\x12\\x0b\\n\\x03key\\x18\\x01 \\x01(\\t\\x12$\\n\\x05value\\x18\\x02 \\x01(\\x0b\\x32\\x15.proto3.TestNestedMap:\\x02\\x38\\x01\\\"{\\n\\rTestStringMap\\x12\\x38\\n\\nstring_map\\x18\\x01 \\x03(\\x0b\\x32$.proto3.TestStringMap.StringMapEntry\\x1a\\x30\\n\\x0eStringMapEntry\\x12\\x0b\\n\\x03key\\x18\\x01 \\x01(\\t\\x12\\r\\n\\x05value\\x18\\x02 \\x01(\\t:\\x02\\x38\\x01\\\"\\xee\\x07\\n\\x0bTestWrapper\\x12.\\n\\nbool_value\\x18\\x01 \\x01(\\x0b\\x32\\x1a.google.protobuf.BoolValue\\x12\\x30\\n\\x0bint32_value\\x18\\x02 \\x01(\\x0b\\x32\\x1b.google.protobuf.Int32Value\\x12\\x30\\n\\x0bint64_value\\x18\\x03 \\x01(\\x0b\\x32\\x1b.google.protobuf.Int64Value\\x12\\x32\\n\\x0cuint32_value\\x18\\x04 \\x01(\\x0b\\x32\\x1c.google.protobuf.UInt32Value\\x12\\x32\\n\\x0cuint64_value\\x18\\x05 \\x01(\\x0b\\x32\\x1c.google.protobuf.UInt64Value\\x12\\x30\\n\\x0b\\x66loat_value\\x18\\x06 \\x01(\\x0b\\x32\\x1b.google.protobuf.FloatValue\\x12\\x32\\n\\x0c\\x64ouble_value\\x18\\x07 \\x01(\\x0b\\x32\\x1c.google.protobuf.DoubleValue\\x12\\x32\\n\\x0cstring_value\\x18\\x08 \\x01(\\x0b\\x32\\x1c.google.protobuf.StringValue\\x12\\x30\\n\\x0b\\x62ytes_value\\x18\\t \\x01(\\x0b\\x32\\x1b.google.protobuf.BytesValue\\x12\\x37\\n\\x13repeated_bool_value\\x18\\x0b \\x03(\\x0b\\x32\\x1a.google.protobuf.BoolValue\\x12\\x39\\n\\x14repeated_int32_value\\x18\\x0c \\x03(\\x0b\\x32\\x1b.google.protobuf.Int32Value\\x12\\x39\\n\\x14repeated_int64_value\\x18\\r \\x03(\\x0b\\x32\\x1b.google.protobuf.Int64Value\\x12;\\n\\x15repeated_uint32_value\\x18\\x0e \\x03(\\x0b\\x32\\x1c.google.protobuf.UInt32Value\\x12;\\n\\x15repeated_uint64_value\\x18\\x0f \\x03(\\x0b\\x32\\x1c.google.protobuf.UInt64Value\\x12\\x39\\n\\x14repeated_float_value\\x18\\x10 \\x03(\\x0b\\x32\\x1b.google.protobuf.FloatValue\\x12;\\n\\x15repeated_double_value\\x18\\x11 \\x03(\\x0b\\x32\\x1c.google.protobuf.DoubleValue\\x12;\\n\\x15repeated_string_value\\x18\\x12 \\x03(\\x0b\\x32\\x1c.google.protobuf.StringValue\\x12\\x39\\n\\x14repeated_bytes_value\\x18\\x13 \\x03(\\x0b\\x32\\x1b.google.protobuf.BytesValue\\\"n\\n\\rTestTimestamp\\x12)\\n\\x05value\\x18\\x01 \\x01(\\x0b\\x32\\x1a.google.protobuf.Timestamp\\x12\\x32\\n\\x0erepeated_value\\x18\\x02 \\x03(\\x0b\\x32\\x1a.google.protobuf.Timestamp\\\"k\\n\\x0cTestDuration\\x12(\\n\\x05value\\x18\\x01 \\x01(\\x0b\\x32\\x19.google.protobuf.Duration\\x12\\x31\\n\\x0erepeated_value\\x18\\x02 \\x03(\\x0b\\x32\\x19.google.protobuf.Duration\\\":\\n\\rTestFieldMask\\x12)\\n\\x05value\\x18\\x01 \\x01(\\x0b\\x32\\x1a.google.protobuf.FieldMask\\\"e\\n\\nTestStruct\\x12&\\n\\x05value\\x18\\x01 \\x01(\\x0b\\x32\\x17.google.protobuf.Struct\\x12/\\n\\x0erepeated_value\\x18\\x02 \\x03(\\x0b\\x32\\x17.google.protobuf.Struct\\\"\\\\\\n\\x07TestAny\\x12#\\n\\x05value\\x18\\x01 \\x01(\\x0b\\x32\\x14.google.protobuf.Any\\x12,\\n\\x0erepeated_value\\x18\\x02 \\x03(\\x0b\\x32\\x14.google.protobuf.Any\\\"b\\n\\tTestValue\\x12%\\n\\x05value\\x18\\x01 \\x01(\\x0b\\x32\\x16.google.protobuf.Value\\x12.\\n\\x0erepeated_value\\x18\\x02 \\x03(\\x0b\\x32\\x16.google.protobuf.Value\\\"n\\n\\rTestListValue\\x12)\\n\\x05value\\x18\\x01 \\x01(\\x0b\\x32\\x1a.google.protobuf.ListValue\\x12\\x32\\n\\x0erepeated_value\\x18\\x02 \\x03(\\x0b\\x32\\x1a.google.protobuf.ListValue\\\"\\x89\\x01\\n\\rTestBoolValue\\x12\\x12\\n\\nbool_value\\x18\\x01 \\x01(\\x08\\x12\\x34\\n\\x08\\x62ool_map\\x18\\x02 \\x03(\\x0b\\x32\\\".proto3.TestBoolValue.BoolMapEntry\\x1a.\\n\\x0c\\x42oolMapEntry\\x12\\x0b\\n\\x03key\\x18\\x01 \\x01(\\x08\\x12\\r\\n\\x05value\\x18\\x02 \\x01(\\x05:\\x02\\x38\\x01\\\"+\\n\\x12TestCustomJsonName\\x12\\x15\\n\\x05value\\x18\\x01 \\x01(\\x05R\\x06@value\\\"J\\n\\x0eTestExtensions\\x12\\x38\\n\\nextensions\\x18\\x01 \\x01(\\x0b\\x32$.protobuf_unittest.TestAllExtensions\\\"\\x84\\x01\\n\\rTestEnumValue\\x12%\\n\\x0b\\x65num_value1\\x18\\x01 \\x01(\\x0e\\x32\\x10.proto3.EnumType\\x12%\\n\\x0b\\x65num_value2\\x18\\x02 \\x01(\\x0e\\x32\\x10.proto3.EnumType\\x12%\\n\\x0b\\x65num_value3\\x18\\x03 \\x01(\\x0e\\x32\\x10.proto3.EnumType*\\x1c\\n\\x08\\x45numType\\x12\\x07\\n\\x03\\x46OO\\x10\\x00\\x12\\x07\\n\\x03\\x42\\x41R\\x10\\x01\\x42,\\n\\x18\\x63om.google.protobuf.utilB\\x10JsonFormatProto3b\\x06proto3')\n\n_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())\n_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.util.json_format_proto3_pb2', globals())\nif _descriptor._USE_C_DESCRIPTORS == False:\n\n DESCRIPTOR._options = None\n DESCRIPTOR._serialized_options = b'\\n\\030com.google.protobuf.utilB\\020JsonFormatProto3'\n _TESTMAP_BOOLMAPENTRY._options = None\n _TESTMAP_BOOLMAPENTRY._serialized_options = b'8\\001'\n _TESTMAP_INT32MAPENTRY._options = None\n _TESTMAP_INT32MAPENTRY._serialized_options = b'8\\001'\n _TESTMAP_INT64MAPENTRY._options = None\n _TESTMAP_INT64MAPENTRY._serialized_options = b'8\\001'\n _TESTMAP_UINT32MAPENTRY._options = None\n _TESTMAP_UINT32MAPENTRY._serialized_options = b'8\\001'\n _TESTMAP_UINT64MAPENTRY._options = None\n _TESTMAP_UINT64MAPENTRY._serialized_options = b'8\\001'\n _TESTMAP_STRINGMAPENTRY._options = None\n _TESTMAP_STRINGMAPENTRY._serialized_options = b'8\\001'\n _TESTNESTEDMAP_BOOLMAPENTRY._options = None\n _TESTNESTEDMAP_BOOLMAPENTRY._serialized_options = b'8\\001'\n _TESTNESTEDMAP_INT32MAPENTRY._options = None\n _TESTNESTEDMAP_INT32MAPENTRY._serialized_options = b'8\\001'\n _TESTNESTEDMAP_INT64MAPENTRY._options = None\n _TESTNESTEDMAP_INT64MAPENTRY._serialized_options = b'8\\001'\n _TESTNESTEDMAP_UINT32MAPENTRY._options = None\n _TESTNESTEDMAP_UINT32MAPENTRY._serialized_options = b'8\\001'\n _TESTNESTEDMAP_UINT64MAPENTRY._options = None\n _TESTNESTEDMAP_UINT64MAPENTRY._serialized_options = b'8\\001'\n _TESTNESTEDMAP_STRINGMAPENTRY._options = None\n _TESTNESTEDMAP_STRINGMAPENTRY._serialized_options = b'8\\001'\n _TESTNESTEDMAP_MAPMAPENTRY._options = None\n _TESTNESTEDMAP_MAPMAPENTRY._serialized_options = b'8\\001'\n _TESTSTRINGMAP_STRINGMAPENTRY._options = None\n _TESTSTRINGMAP_STRINGMAPENTRY._serialized_options = b'8\\001'\n _TESTBOOLVALUE_BOOLMAPENTRY._options = None\n _TESTBOOLVALUE_BOOLMAPENTRY._serialized_options = b'8\\001'\n _ENUMTYPE._serialized_start=4849\n _ENUMTYPE._serialized_end=4877\n _MESSAGETYPE._serialized_start=277\n _MESSAGETYPE._serialized_end=305\n _TESTMESSAGE._serialized_start=308\n _TESTMESSAGE._serialized_end=968\n _TESTONEOF._serialized_start=971\n _TESTONEOF._serialized_end=1239\n _TESTMAP._serialized_start=1242\n _TESTMAP._serialized_end=1851\n _TESTMAP_BOOLMAPENTRY._serialized_start=1557\n _TESTMAP_BOOLMAPENTRY._serialized_end=1603\n _TESTMAP_INT32MAPENTRY._serialized_start=1605\n _TESTMAP_INT32MAPENTRY._serialized_end=1652\n _TESTMAP_INT64MAPENTRY._serialized_start=1654\n _TESTMAP_INT64MAPENTRY._serialized_end=1701\n _TESTMAP_UINT32MAPENTRY._serialized_start=1703\n _TESTMAP_UINT32MAPENTRY._serialized_end=1751\n _TESTMAP_UINT64MAPENTRY._serialized_start=1753\n _TESTMAP_UINT64MAPENTRY._serialized_end=1801\n _TESTMAP_STRINGMAPENTRY._serialized_start=1803\n _TESTMAP_STRINGMAPENTRY._serialized_end=1851\n _TESTNESTEDMAP._serialized_start=1854\n _TESTNESTEDMAP._serialized_end=2627\n _TESTNESTEDMAP_BOOLMAPENTRY._serialized_start=1557\n _TESTNESTEDMAP_BOOLMAPENTRY._serialized_end=1603\n _TESTNESTEDMAP_INT32MAPENTRY._serialized_start=1605\n _TESTNESTEDMAP_INT32MAPENTRY._serialized_end=1652\n _TESTNESTEDMAP_INT64MAPENTRY._serialized_start=1654\n _TESTNESTEDMAP_INT64MAPENTRY._serialized_end=1701\n _TESTNESTEDMAP_UINT32MAPENTRY._serialized_start=1703\n _TESTNESTEDMAP_UINT32MAPENTRY._serialized_end=1751\n _TESTNESTEDMAP_UINT64MAPENTRY._serialized_start=1753\n _TESTNESTEDMAP_UINT64MAPENTRY._serialized_end=1801\n _TESTNESTEDMAP_STRINGMAPENTRY._serialized_start=1803\n _TESTNESTEDMAP_STRINGMAPENTRY._serialized_end=1851\n _TESTNESTEDMAP_MAPMAPENTRY._serialized_start=2559\n _TESTNESTEDMAP_MAPMAPENTRY._serialized_end=2627\n _TESTSTRINGMAP._serialized_start=2629\n _TESTSTRINGMAP._serialized_end=2752\n _TESTSTRINGMAP_STRINGMAPENTRY._serialized_start=2704\n _TESTSTRINGMAP_STRINGMAPENTRY._serialized_end=2752\n _TESTWRAPPER._serialized_start=2755\n _TESTWRAPPER._serialized_end=3761\n _TESTTIMESTAMP._serialized_start=3763\n _TESTTIMESTAMP._serialized_end=3873\n _TESTDURATION._serialized_start=3875\n _TESTDURATION._serialized_end=3982\n _TESTFIELDMASK._serialized_start=3984\n _TESTFIELDMASK._serialized_end=4042\n _TESTSTRUCT._serialized_start=4044\n _TESTSTRUCT._serialized_end=4145\n _TESTANY._serialized_start=4147\n _TESTANY._serialized_end=4239\n _TESTVALUE._serialized_start=4241\n _TESTVALUE._serialized_end=4339\n _TESTLISTVALUE._serialized_start=4341\n _TESTLISTVALUE._serialized_end=4451\n _TESTBOOLVALUE._serialized_start=4454\n _TESTBOOLVALUE._serialized_end=4591\n _TESTBOOLVALUE_BOOLMAPENTRY._serialized_start=1557\n _TESTBOOLVALUE_BOOLMAPENTRY._serialized_end=1603\n _TESTCUSTOMJSONNAME._serialized_start=4593\n _TESTCUSTOMJSONNAME._serialized_end=4636\n _TESTEXTENSIONS._serialized_start=4638\n _TESTEXTENSIONS._serialized_end=4712\n _TESTENUMVALUE._serialized_start=4715\n _TESTENUMVALUE._serialized_end=4847\n# @@protoc_insertion_point(module_scope)\n"
},
{
"path": "openpype/hosts/hiero/vendor/google/protobuf/wrappers_pb2.py",
"content": "# -*- coding: utf-8 -*-\n# Generated by the protocol buffer compiler. DO NOT EDIT!\n# source: google/protobuf/wrappers.proto\n\"\"\"Generated protocol buffer code.\"\"\"\nfrom google.protobuf.internal import builder as _builder\nfrom google.protobuf import descriptor as _descriptor\nfrom google.protobuf import descriptor_pool as _descriptor_pool\nfrom google.protobuf import symbol_database as _symbol_database\n# @@protoc_insertion_point(imports)\n\n_sym_db = _symbol_database.Default()\n\n\n\n\nDESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\\n\\x1egoogle/protobuf/wrappers.proto\\x12\\x0fgoogle.protobuf\\\"\\x1c\\n\\x0b\\x44oubleValue\\x12\\r\\n\\x05value\\x18\\x01 \\x01(\\x01\\\"\\x1b\\n\\nFloatValue\\x12\\r\\n\\x05value\\x18\\x01 \\x01(\\x02\\\"\\x1b\\n\\nInt64Value\\x12\\r\\n\\x05value\\x18\\x01 \\x01(\\x03\\\"\\x1c\\n\\x0bUInt64Value\\x12\\r\\n\\x05value\\x18\\x01 \\x01(\\x04\\\"\\x1b\\n\\nInt32Value\\x12\\r\\n\\x05value\\x18\\x01 \\x01(\\x05\\\"\\x1c\\n\\x0bUInt32Value\\x12\\r\\n\\x05value\\x18\\x01 \\x01(\\r\\\"\\x1a\\n\\tBoolValue\\x12\\r\\n\\x05value\\x18\\x01 \\x01(\\x08\\\"\\x1c\\n\\x0bStringValue\\x12\\r\\n\\x05value\\x18\\x01 \\x01(\\t\\\"\\x1b\\n\\nBytesValue\\x12\\r\\n\\x05value\\x18\\x01 \\x01(\\x0c\\x42\\x83\\x01\\n\\x13\\x63om.google.protobufB\\rWrappersProtoP\\x01Z1google.golang.org/protobuf/types/known/wrapperspb\\xf8\\x01\\x01\\xa2\\x02\\x03GPB\\xaa\\x02\\x1eGoogle.Protobuf.WellKnownTypesb\\x06proto3')\n\n_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, globals())\n_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'google.protobuf.wrappers_pb2', globals())\nif _descriptor._USE_C_DESCRIPTORS == False:\n\n DESCRIPTOR._options = None\n DESCRIPTOR._serialized_options = b'\\n\\023com.google.protobufB\\rWrappersProtoP\\001Z1google.golang.org/protobuf/types/known/wrapperspb\\370\\001\\001\\242\\002\\003GPB\\252\\002\\036Google.Protobuf.WellKnownTypes'\n _DOUBLEVALUE._serialized_start=51\n _DOUBLEVALUE._serialized_end=79\n _FLOATVALUE._serialized_start=81\n _FLOATVALUE._serialized_end=108\n _INT64VALUE._serialized_start=110\n _INT64VALUE._serialized_end=137\n _UINT64VALUE._serialized_start=139\n _UINT64VALUE._serialized_end=167\n _INT32VALUE._serialized_start=169\n _INT32VALUE._serialized_end=196\n _UINT32VALUE._serialized_start=198\n _UINT32VALUE._serialized_end=226\n _BOOLVALUE._serialized_start=228\n _BOOLVALUE._serialized_end=254\n _STRINGVALUE._serialized_start=256\n _STRINGVALUE._serialized_end=284\n _BYTESVALUE._serialized_start=286\n _BYTESVALUE._serialized_end=313\n# @@protoc_insertion_point(module_scope)\n"
},
{
"path": "openpype/hosts/houdini/__init__.py",
"content": "from .addon import (\n HoudiniAddon,\n HOUDINI_HOST_DIR,\n)\n\n\n__all__ = (\n \"HoudiniAddon\",\n \"HOUDINI_HOST_DIR\",\n)\n"
},
{
"path": "openpype/hosts/houdini/addon.py",
"content": "import os\nfrom openpype.modules import OpenPypeModule, IHostAddon\n\nHOUDINI_HOST_DIR = os.path.dirname(os.path.abspath(__file__))\n\n\nclass HoudiniAddon(OpenPypeModule, IHostAddon):\n name = \"houdini\"\n host_name = \"houdini\"\n\n def initialize(self, module_settings):\n self.enabled = True\n\n def add_implementation_envs(self, env, _app):\n # Add requirements to HOUDINI_PATH and HOUDINI_MENU_PATH\n startup_path = os.path.join(HOUDINI_HOST_DIR, \"startup\")\n new_houdini_path = [startup_path]\n new_houdini_menu_path = [startup_path]\n\n old_houdini_path = env.get(\"HOUDINI_PATH\") or \"\"\n old_houdini_menu_path = env.get(\"HOUDINI_MENU_PATH\") or \"\"\n\n for path in old_houdini_path.split(os.pathsep):\n if not path:\n continue\n\n norm_path = os.path.normpath(path)\n if norm_path not in new_houdini_path:\n new_houdini_path.append(norm_path)\n\n for path in old_houdini_menu_path.split(os.pathsep):\n if not path:\n continue\n\n norm_path = os.path.normpath(path)\n if norm_path not in new_houdini_menu_path:\n new_houdini_menu_path.append(norm_path)\n\n # Add ampersand for unknown reason (Maybe is needed in Houdini?)\n new_houdini_path.append(\"&\")\n new_houdini_menu_path.append(\"&\")\n\n env[\"HOUDINI_PATH\"] = os.pathsep.join(new_houdini_path)\n env[\"HOUDINI_MENU_PATH\"] = os.pathsep.join(new_houdini_menu_path)\n\n def get_launch_hook_paths(self, app):\n if app.host_name != self.host_name:\n return []\n return [\n os.path.join(HOUDINI_HOST_DIR, \"hooks\")\n ]\n\n def get_workfile_extensions(self):\n return [\".hip\", \".hiplc\", \".hipnc\"]\n"
},
{
"path": "openpype/hosts/houdini/api/__init__.py",
"content": "from .pipeline import (\n HoudiniHost,\n ls,\n containerise\n)\n\nfrom .plugin import (\n Creator,\n)\n\nfrom .lib import (\n lsattr,\n lsattrs,\n read,\n\n maintained_selection\n)\n\n\n__all__ = [\n \"HoudiniHost\",\n\n \"ls\",\n \"containerise\",\n\n \"Creator\",\n\n # Utility functions\n \"lsattr\",\n \"lsattrs\",\n \"read\",\n\n \"maintained_selection\"\n]\n"
},
{
"path": "openpype/hosts/houdini/api/action.py",
"content": "import pyblish.api\nimport hou\n\nfrom openpype.pipeline.publish import get_errored_instances_from_context\n\n\nclass SelectInvalidAction(pyblish.api.Action):\n \"\"\"Select invalid nodes in Maya when plug-in failed.\n\n To retrieve the invalid nodes this assumes a static `get_invalid()`\n method is available on the plugin.\n\n \"\"\"\n label = \"Select invalid\"\n on = \"failed\" # This action is only available on a failed plug-in\n icon = \"search\" # Icon from Awesome Icon\n\n def process(self, context, plugin):\n\n errored_instances = get_errored_instances_from_context(context,\n plugin=plugin)\n\n # Get the invalid nodes for the plug-ins\n self.log.info(\"Finding invalid nodes..\")\n invalid = list()\n for instance in errored_instances:\n invalid_nodes = plugin.get_invalid(instance)\n if invalid_nodes:\n if isinstance(invalid_nodes, (list, tuple)):\n invalid.extend(invalid_nodes)\n else:\n self.log.warning(\"Plug-in returned to be invalid, \"\n \"but has no selectable nodes.\")\n\n hou.clearAllSelected()\n if invalid:\n self.log.info(\"Selecting invalid nodes: {}\".format(\n \", \".join(node.path() for node in invalid)\n ))\n for node in invalid:\n node.setSelected(True)\n node.setCurrent(True)\n else:\n self.log.info(\"No invalid nodes found.\")\n\n\nclass SelectROPAction(pyblish.api.Action):\n \"\"\"Select ROP.\n\n It's used to select the associated ROPs with the errored instances.\n \"\"\"\n\n label = \"Select ROP\"\n on = \"failed\" # This action is only available on a failed plug-in\n icon = \"mdi.cursor-default-click\"\n\n def process(self, context, plugin):\n errored_instances = get_errored_instances_from_context(context, plugin)\n\n # Get the invalid nodes for the plug-ins\n self.log.info(\"Finding ROP nodes..\")\n rop_nodes = list()\n for instance in errored_instances:\n node_path = instance.data.get(\"instance_node\")\n if not node_path:\n continue\n\n node = hou.node(node_path)\n if not node:\n continue\n\n rop_nodes.append(node)\n\n hou.clearAllSelected()\n if rop_nodes:\n self.log.info(\"Selecting ROP nodes: {}\".format(\n \", \".join(node.path() for node in rop_nodes)\n ))\n for node in rop_nodes:\n node.setSelected(True)\n node.setCurrent(True)\n else:\n self.log.info(\"No ROP nodes found.\")\n"
},
{
"path": "openpype/hosts/houdini/api/colorspace.py",
"content": "import attr\nimport hou\nfrom openpype.hosts.houdini.api.lib import get_color_management_preferences\nfrom openpype.pipeline.colorspace import get_display_view_colorspace_name\n\n@attr.s\nclass LayerMetadata(object):\n \"\"\"Data class for Render Layer metadata.\"\"\"\n frameStart = attr.ib()\n frameEnd = attr.ib()\n\n\n@attr.s\nclass RenderProduct(object):\n \"\"\"Getting Colorspace as\n Specific Render Product Parameter for submitting\n publish job.\n\n \"\"\"\n colorspace = attr.ib() # colorspace\n view = attr.ib()\n productName = attr.ib(default=None)\n\n\nclass ARenderProduct(object):\n\n def __init__(self):\n \"\"\"Constructor.\"\"\"\n # Initialize\n self.layer_data = self._get_layer_data()\n self.layer_data.products = self.get_colorspace_data()\n\n def _get_layer_data(self):\n return LayerMetadata(\n frameStart=int(hou.playbar.frameRange()[0]),\n frameEnd=int(hou.playbar.frameRange()[1]),\n )\n\n def get_colorspace_data(self):\n \"\"\"To be implemented by renderer class.\n\n This should return a list of RenderProducts.\n\n Returns:\n list: List of RenderProduct\n\n \"\"\"\n data = get_color_management_preferences()\n colorspace_data = [\n RenderProduct(\n colorspace=data[\"display\"],\n view=data[\"view\"],\n productName=\"\"\n )\n ]\n return colorspace_data\n\n\ndef get_default_display_view_colorspace():\n \"\"\"Returns the colorspace attribute of the default (display, view) pair.\n\n It's used for 'ociocolorspace' parm in OpenGL Node.\"\"\"\n\n prefs = get_color_management_preferences()\n return get_display_view_colorspace_name(\n config_path=prefs[\"config\"],\n display=prefs[\"display\"],\n view=prefs[\"view\"]\n )\n"
},
{
"path": "openpype/hosts/houdini/api/creator_node_shelves.py",
"content": "\"\"\"Library to register OpenPype Creators for Houdini TAB node search menu.\n\nThis can be used to install custom houdini tools for the TAB search\nmenu which will trigger a publish instance to be created interactively.\n\nThe Creators are automatically registered on launch of Houdini through the\nHoudini integration's `host.install()` method.\n\n\"\"\"\nimport contextlib\nimport tempfile\nimport logging\nimport os\n\nfrom openpype.client import get_asset_by_name\nfrom openpype.pipeline import registered_host\nfrom openpype.pipeline.create import CreateContext\nfrom openpype.resources import get_openpype_icon_filepath\n\nimport hou\nimport stateutils\nimport soptoolutils\nimport loptoolutils\nimport cop2toolutils\n\n\nlog = logging.getLogger(__name__)\n\nCATEGORY_GENERIC_TOOL = {\n hou.sopNodeTypeCategory(): soptoolutils.genericTool,\n hou.cop2NodeTypeCategory(): cop2toolutils.genericTool,\n hou.lopNodeTypeCategory(): loptoolutils.genericTool\n}\n\n\nCREATE_SCRIPT = \"\"\"\nfrom openpype.hosts.houdini.api.creator_node_shelves import create_interactive\ncreate_interactive(\"{identifier}\", **kwargs)\n\"\"\"\n\n\ndef create_interactive(creator_identifier, **kwargs):\n \"\"\"Create a Creator using its identifier interactively.\n\n This is used by the generated shelf tools as callback when a user selects\n the creator from the node tab search menu.\n\n The `kwargs` should be what Houdini passes to the tool create scripts\n context. For more information see:\n https://www.sidefx.com/docs/houdini/hom/tool_script.html#arguments\n\n Args:\n creator_identifier (str): The creator identifier of the Creator plugin\n to create.\n\n Return:\n list: The created instances.\n\n \"\"\"\n host = registered_host()\n context = CreateContext(host)\n creator = context.manual_creators.get(creator_identifier)\n if not creator:\n raise RuntimeError(\"Invalid creator identifier: {}\".format(\n creator_identifier)\n )\n\n # TODO Use Qt instead\n result, variant = hou.ui.readInput(\n \"Define variant name\",\n buttons=(\"Ok\", \"Cancel\"),\n initial_contents=creator.get_default_variant(),\n title=\"Define variant\",\n help=\"Set the variant for the publish instance\",\n close_choice=1\n )\n\n if result == 1:\n # User interrupted\n return\n\n variant = variant.strip()\n if not variant:\n raise RuntimeError(\"Empty variant value entered.\")\n\n # TODO: Once more elaborate unique create behavior should exist per Creator\n # instead of per network editor area then we should move this from here\n # to a method on the Creators for which this could be the default\n # implementation.\n pane = stateutils.activePane(kwargs)\n if isinstance(pane, hou.NetworkEditor):\n pwd = pane.pwd()\n subset_name = creator.get_subset_name(\n variant=variant,\n task_name=context.get_current_task_name(),\n asset_doc=get_asset_by_name(\n project_name=context.get_current_project_name(),\n asset_name=context.get_current_asset_name()\n ),\n project_name=context.get_current_project_name(),\n host_name=context.host_name\n )\n\n tool_fn = CATEGORY_GENERIC_TOOL.get(pwd.childTypeCategory())\n if tool_fn is not None:\n out_null = tool_fn(kwargs, \"null\")\n out_null.setName(\"OUT_{}\".format(subset_name), unique_name=True)\n\n before = context.instances_by_id.copy()\n\n # Create the instance\n context.create(\n creator_identifier=creator_identifier,\n variant=variant,\n pre_create_data={\"use_selection\": True}\n )\n\n # For convenience we set the new node as current since that's much more\n # familiar to the artist when creating a node interactively\n # TODO Allow to disable auto-select in studio settings or user preferences\n after = context.instances_by_id\n new = set(after) - set(before)\n if new:\n # Select the new instance\n for instance_id in new:\n instance = after[instance_id]\n node = hou.node(instance.get(\"instance_node\"))\n node.setCurrent(True)\n\n return list(new)\n\n\n@contextlib.contextmanager\ndef shelves_change_block():\n \"\"\"Write shelf changes at the end of the context.\"\"\"\n hou.shelves.beginChangeBlock()\n try:\n yield\n finally:\n hou.shelves.endChangeBlock()\n\n\ndef install():\n \"\"\"Install the Creator plug-ins to show in Houdini's TAB node search menu.\n\n This function is re-entrant and can be called again to reinstall and\n update the node definitions. For example during development it can be\n useful to call it manually:\n >>> from openpype.hosts.houdini.api.creator_node_shelves import install\n >>> install()\n\n Returns:\n list: List of `hou.Tool` instances\n\n \"\"\"\n\n host = registered_host()\n\n # Store the filepath on the host\n # TODO: Define a less hacky static shelf path for current houdini session\n filepath_attr = \"_creator_node_shelf_filepath\"\n filepath = getattr(host, filepath_attr, None)\n if filepath is None:\n f = tempfile.NamedTemporaryFile(prefix=\"houdini_creator_nodes_\",\n suffix=\".shelf\",\n delete=False)\n f.close()\n filepath = f.name\n setattr(host, filepath_attr, filepath)\n elif os.path.exists(filepath):\n # Remove any existing shelf file so that we can completey regenerate\n # and update the tools file if creator identifiers change\n os.remove(filepath)\n\n icon = get_openpype_icon_filepath()\n tab_menu_label = os.environ.get(\"AVALON_LABEL\") or \"AYON\"\n\n # Create context only to get creator plugins, so we don't reset and only\n # populate what we need to retrieve the list of creator plugins\n create_context = CreateContext(host, reset=False)\n create_context.reset_current_context()\n create_context._reset_creator_plugins()\n\n log.debug(\"Writing OpenPype Creator nodes to shelf: {}\".format(filepath))\n tools = []\n\n with shelves_change_block():\n for identifier, creator in create_context.manual_creators.items():\n\n # Allow the creator plug-in itself to override the categories\n # for where they are shown with `Creator.get_network_categories()`\n if not hasattr(creator, \"get_network_categories\"):\n log.debug(\"Creator {} has no `get_network_categories` method \"\n \"and will not be added to TAB search.\")\n continue\n\n network_categories = creator.get_network_categories()\n if not network_categories:\n continue\n\n key = \"ayon_create.{}\".format(identifier)\n log.debug(f\"Registering {key}\")\n script = CREATE_SCRIPT.format(identifier=identifier)\n data = {\n \"script\": script,\n \"language\": hou.scriptLanguage.Python,\n \"icon\": icon,\n \"help\": \"Create Ayon publish instance for {}\".format(\n creator.label\n ),\n \"help_url\": None,\n \"network_categories\": network_categories,\n \"viewer_categories\": [],\n \"cop_viewer_categories\": [],\n \"network_op_type\": None,\n \"viewer_op_type\": None,\n \"locations\": [tab_menu_label]\n }\n label = \"Create {}\".format(creator.label)\n tool = hou.shelves.tool(key)\n if tool:\n tool.setData(**data)\n tool.setLabel(label)\n else:\n tool = hou.shelves.newTool(\n file_path=filepath,\n name=key,\n label=label,\n **data\n )\n\n tools.append(tool)\n\n # Ensure the shelf is reloaded\n hou.shelves.loadFile(filepath)\n\n return tools\n"
},
{
"path": "openpype/hosts/houdini/api/lib.py",
"content": "# -*- coding: utf-8 -*-\nimport sys\nimport os\nimport errno\nimport re\nimport uuid\nimport logging\nfrom contextlib import contextmanager\nimport json\n\nimport six\n\nfrom openpype.lib import StringTemplate\nfrom openpype.client import get_project, get_asset_by_name\nfrom openpype.settings import get_current_project_settings\nfrom openpype.pipeline import (\n Anatomy,\n get_current_project_name,\n get_current_asset_name,\n registered_host,\n get_current_context,\n get_current_host_name,\n)\nfrom openpype.pipeline.create import CreateContext\nfrom openpype.pipeline.template_data import get_template_data\nfrom openpype.pipeline.context_tools import get_current_project_asset\nfrom openpype.widgets import popup\nfrom openpype.tools.utils.host_tools import get_tool_by_name\n\nimport hou\n\n\nself = sys.modules[__name__]\nself._parent = None\nlog = logging.getLogger(__name__)\nJSON_PREFIX = \"JSON:::\"\n\n\ndef get_asset_fps(asset_doc=None):\n \"\"\"Return current asset fps.\"\"\"\n\n if asset_doc is None:\n asset_doc = get_current_project_asset(fields=[\"data.fps\"])\n return asset_doc[\"data\"][\"fps\"]\n\n\ndef set_id(node, unique_id, overwrite=False):\n exists = node.parm(\"id\")\n if not exists:\n imprint(node, {\"id\": unique_id})\n\n if not exists and overwrite:\n node.setParm(\"id\", unique_id)\n\n\ndef get_id(node):\n \"\"\"Get the `cbId` attribute of the given node.\n\n Args:\n node (hou.Node): the name of the node to retrieve the attribute from\n\n Returns:\n str: cbId attribute of the node.\n\n \"\"\"\n\n if node is not None:\n return node.parm(\"id\")\n\n\ndef generate_ids(nodes, asset_id=None):\n \"\"\"Returns new unique ids for the given nodes.\n\n Note: This does not assign the new ids, it only generates the values.\n\n To assign new ids using this method:\n >>> nodes = [\"a\", \"b\", \"c\"]\n >>> for node, id in generate_ids(nodes):\n >>> set_id(node, id)\n\n To also override any existing values (and assign regenerated ids):\n >>> nodes = [\"a\", \"b\", \"c\"]\n >>> for node, id in generate_ids(nodes):\n >>> set_id(node, id, overwrite=True)\n\n Args:\n nodes (list): List of nodes.\n asset_id (str or bson.ObjectId): The database id for the *asset* to\n generate for. When None provided the current asset in the\n active session is used.\n\n Returns:\n list: A list of (node, id) tuples.\n\n \"\"\"\n\n if asset_id is None:\n project_name = get_current_project_name()\n asset_name = get_current_asset_name()\n # Get the asset ID from the database for the asset of current context\n asset_doc = get_asset_by_name(project_name, asset_name, fields=[\"_id\"])\n\n assert asset_doc, \"No current asset found in Session\"\n asset_id = asset_doc['_id']\n\n node_ids = []\n for node in nodes:\n _, uid = str(uuid.uuid4()).rsplit(\"-\", 1)\n unique_id = \"{}:{}\".format(asset_id, uid)\n node_ids.append((node, unique_id))\n\n return node_ids\n\n\ndef get_id_required_nodes():\n\n valid_types = [\"geometry\"]\n nodes = {n for n in hou.node(\"/out\").children() if\n n.type().name() in valid_types}\n\n return list(nodes)\n\n\ndef get_output_parameter(node):\n \"\"\"Return the render output parameter of the given node\n\n Example:\n root = hou.node(\"/obj\")\n my_alembic_node = root.createNode(\"alembic\")\n get_output_parameter(my_alembic_node)\n >>> \"filename\"\n\n Notes:\n I'm using node.type().name() to get on par with the creators,\n Because the return value of `node.type().name()` is the\n same string value used in creators\n e.g. instance_data.update({\"node_type\": \"alembic\"})\n\n Rop nodes in different network categories have\n the same output parameter.\n So, I took that into consideration as a hint for\n future development.\n\n Args:\n node(hou.Node): node instance\n\n Returns:\n hou.Parm\n \"\"\"\n\n node_type = node.type().name()\n\n # Figure out which type of node is being rendered\n if node_type in {\"alembic\", \"rop_alembic\"}:\n return node.parm(\"filename\")\n elif node_type == \"arnold\":\n if node_type.evalParm(\"ar_ass_export_enable\"):\n return node.parm(\"ar_ass_file\")\n return node.parm(\"ar_picture\")\n elif node_type in {\n \"geometry\",\n \"rop_geometry\",\n \"filmboxfbx\",\n \"rop_fbx\"\n }:\n return node.parm(\"sopoutput\")\n elif node_type == \"comp\":\n return node.parm(\"copoutput\")\n elif node_type in {\"karma\", \"opengl\"}:\n return node.parm(\"picture\")\n elif node_type == \"ifd\": # Mantra\n if node.evalParm(\"soho_outputmode\"):\n return node.parm(\"soho_diskfile\")\n return node.parm(\"vm_picture\")\n elif node_type == \"Redshift_Proxy_Output\":\n return node.parm(\"RS_archive_file\")\n elif node_type == \"Redshift_ROP\":\n return node.parm(\"RS_outputFileNamePrefix\")\n elif node_type in {\"usd\", \"usd_rop\", \"usdexport\"}:\n return node.parm(\"lopoutput\")\n elif node_type in {\"usdrender\", \"usdrender_rop\"}:\n return node.parm(\"outputimage\")\n elif node_type == \"vray_renderer\":\n return node.parm(\"SettingsOutput_img_file_path\")\n\n raise TypeError(\"Node type '%s' not supported\" % node_type)\n\n\ndef set_scene_fps(fps):\n hou.setFps(fps)\n\n\n# Valid FPS\ndef validate_fps():\n \"\"\"Validate current scene FPS and show pop-up when it is incorrect\n\n Returns:\n bool\n\n \"\"\"\n\n fps = get_asset_fps()\n current_fps = hou.fps() # returns float\n\n if current_fps != fps:\n\n # Find main window\n parent = hou.ui.mainQtWindow()\n if parent is None:\n pass\n else:\n dialog = popup.PopupUpdateKeys(parent=parent)\n dialog.setModal(True)\n dialog.setWindowTitle(\"Houdini scene does not match project FPS\")\n dialog.setMessage(\"Scene %i FPS does not match project %i FPS\" %\n (current_fps, fps))\n dialog.setButtonText(\"Fix\")\n\n # on_show is the Fix button clicked callback\n dialog.on_clicked_state.connect(lambda: set_scene_fps(fps))\n\n dialog.show()\n\n return False\n\n return True\n\n\ndef create_remote_publish_node(force=True):\n \"\"\"Function to create a remote publish node in /out\n\n This is a hacked \"Shell\" node that does *nothing* except for triggering\n `colorbleed.lib.publish_remote()` as pre-render script.\n\n All default attributes of the Shell node are hidden to the Artist to\n avoid confusion.\n\n Additionally some custom attributes are added that can be collected\n by a Collector to set specific settings for the publish, e.g. whether\n to separate the jobs per instance or process in one single job.\n\n \"\"\"\n\n cmd = \"import colorbleed.lib; colorbleed.lib.publish_remote()\"\n\n existing = hou.node(\"/out/REMOTE_PUBLISH\")\n if existing:\n if force:\n log.warning(\"Removing existing '/out/REMOTE_PUBLISH' node..\")\n existing.destroy()\n else:\n raise RuntimeError(\"Node already exists /out/REMOTE_PUBLISH. \"\n \"Please remove manually or set `force` to \"\n \"True.\")\n\n # Create the shell node\n out = hou.node(\"/out\")\n node = out.createNode(\"shell\", node_name=\"REMOTE_PUBLISH\")\n node.moveToGoodPosition()\n\n # Set color make it stand out (avalon/pyblish color)\n node.setColor(hou.Color(0.439, 0.709, 0.933))\n\n # Set the pre-render script\n node.setParms({\n \"prerender\": cmd,\n \"lprerender\": \"python\" # command language\n })\n\n # Lock the attributes to ensure artists won't easily mess things up.\n node.parm(\"prerender\").lock(True)\n node.parm(\"lprerender\").lock(True)\n\n # Lock up the actual shell command\n command_parm = node.parm(\"command\")\n command_parm.set(\"\")\n command_parm.lock(True)\n shellexec_parm = node.parm(\"shellexec\")\n shellexec_parm.set(False)\n shellexec_parm.lock(True)\n\n # Get the node's parm template group so we can customize it\n template = node.parmTemplateGroup()\n\n # Hide default tabs\n template.hideFolder(\"Shell\", True)\n template.hideFolder(\"Scripts\", True)\n\n # Hide default settings\n template.hide(\"execute\", True)\n template.hide(\"renderdialog\", True)\n template.hide(\"trange\", True)\n template.hide(\"f\", True)\n template.hide(\"take\", True)\n\n # Add custom settings to this node.\n parm_folder = hou.FolderParmTemplate(\"folder\", \"Submission Settings\")\n\n # Separate Jobs per Instance\n parm = hou.ToggleParmTemplate(name=\"separateJobPerInstance\",\n label=\"Separate Job per Instance\",\n default_value=False)\n parm_folder.addParmTemplate(parm)\n\n # Add our custom Submission Settings folder\n template.append(parm_folder)\n\n # Apply template back to the node\n node.setParmTemplateGroup(template)\n\n\ndef render_rop(ropnode):\n \"\"\"Render ROP node utility for Publishing.\n\n This renders a ROP node with the settings we want during Publishing.\n \"\"\"\n # Print verbose when in batch mode without UI\n verbose = not hou.isUIAvailable()\n\n # Render\n try:\n ropnode.render(verbose=verbose,\n # Allow Deadline to capture completion percentage\n output_progress=verbose)\n except hou.Error as exc:\n # The hou.Error is not inherited from a Python Exception class,\n # so we explicitly capture the houdini error, otherwise pyblish\n # will remain hanging.\n import traceback\n traceback.print_exc()\n raise RuntimeError(\"Render failed: {0}\".format(exc))\n\n\ndef imprint(node, data, update=False):\n \"\"\"Store attributes with value on a node\n\n Depending on the type of attribute it creates the correct parameter\n template. Houdini uses a template per type, see the docs for more\n information.\n\n http://www.sidefx.com/docs/houdini/hom/hou/ParmTemplate.html\n\n Because of some update glitch where you cannot overwrite existing\n ParmTemplates on node using:\n `setParmTemplates()` and `parmTuplesInFolder()`\n update is done in another pass.\n\n Args:\n node(hou.Node): node object from Houdini\n data(dict): collection of attributes and their value\n update (bool, optional): flag if imprint should update\n already existing data or leave them untouched and only\n add new.\n\n Returns:\n None\n\n \"\"\"\n if not data:\n return\n if not node:\n self.log.error(\"Node is not set, calling imprint on invalid data.\")\n return\n\n current_parms = {p.name(): p for p in node.spareParms()}\n update_parm_templates = []\n new_parm_templates = []\n\n for key, value in data.items():\n if value is None:\n continue\n\n parm_template = get_template_from_value(key, value)\n\n if key in current_parms:\n if node.evalParm(key) == value:\n continue\n if not update:\n log.debug(f\"{key} already exists on {node}\")\n else:\n log.debug(f\"replacing {key}\")\n update_parm_templates.append(parm_template)\n continue\n\n new_parm_templates.append(parm_template)\n\n if not new_parm_templates and not update_parm_templates:\n return\n\n parm_group = node.parmTemplateGroup()\n\n # Add new parm templates\n if new_parm_templates:\n parm_folder = parm_group.findFolder(\"Extra\")\n\n # if folder doesn't exist yet, create one and append to it,\n # else append to existing one\n if not parm_folder:\n parm_folder = hou.FolderParmTemplate(\"folder\", \"Extra\")\n parm_folder.setParmTemplates(new_parm_templates)\n parm_group.append(parm_folder)\n else:\n # Add to parm template folder instance then replace with updated\n # one in parm template group\n for template in new_parm_templates:\n parm_folder.addParmTemplate(template)\n parm_group.replace(parm_folder.name(), parm_folder)\n\n # Update existing parm templates\n for parm_template in update_parm_templates:\n parm_group.replace(parm_template.name(), parm_template)\n\n # When replacing a parm with a parm of the same name it preserves its\n # value if before the replacement the parm was not at the default,\n # because it has a value override set. Since we're trying to update the\n # parm by using the new value as `default` we enforce the parm is at\n # default state\n node.parm(parm_template.name()).revertToDefaults()\n\n node.setParmTemplateGroup(parm_group)\n\n\ndef lsattr(attr, value=None, root=\"/\"):\n \"\"\"Return nodes that have `attr`\n When `value` is not None it will only return nodes matching that value\n for the given attribute.\n Args:\n attr (str): Name of the attribute (hou.Parm)\n value (object, Optional): The value to compare the attribute too.\n When the default None is provided the value check is skipped.\n root (str): The root path in Houdini to search in.\n Returns:\n list: Matching nodes that have attribute with value.\n \"\"\"\n if value is None:\n # Use allSubChildren() as allNodes() errors on nodes without\n # permission to enter without a means to continue of querying\n # the rest\n nodes = hou.node(root).allSubChildren()\n return [n for n in nodes if n.parm(attr)]\n return lsattrs({attr: value})\n\n\ndef lsattrs(attrs, root=\"/\"):\n \"\"\"Return nodes matching `key` and `value`\n Arguments:\n attrs (dict): collection of attribute: value\n root (str): The root path in Houdini to search in.\n Example:\n >> lsattrs({\"id\": \"myId\"})\n [\"myNode\"]\n >> lsattr(\"id\")\n [\"myNode\", \"myOtherNode\"]\n Returns:\n list: Matching nodes that have attribute with value.\n \"\"\"\n\n matches = set()\n # Use allSubChildren() as allNodes() errors on nodes without\n # permission to enter without a means to continue of querying\n # the rest\n nodes = hou.node(root).allSubChildren()\n for node in nodes:\n for attr in attrs:\n if not node.parm(attr):\n continue\n elif node.evalParm(attr) != attrs[attr]:\n continue\n else:\n matches.add(node)\n\n return list(matches)\n\n\ndef read(node):\n \"\"\"Read the container data in to a dict\n\n Args:\n node(hou.Node): Houdini node\n\n Returns:\n dict\n\n \"\"\"\n # `spareParms` returns a tuple of hou.Parm objects\n data = {}\n if not node:\n return data\n for parameter in node.spareParms():\n value = parameter.eval()\n # test if value is json encoded dict\n if isinstance(value, six.string_types) and \\\n value.startswith(JSON_PREFIX):\n try:\n value = json.loads(value[len(JSON_PREFIX):])\n except json.JSONDecodeError:\n # not a json\n pass\n data[parameter.name()] = value\n\n return data\n\n\n@contextmanager\ndef maintained_selection():\n \"\"\"Maintain selection during context\n Example:\n >>> with maintained_selection():\n ... # Modify selection\n ... node.setSelected(on=False, clear_all_selected=True)\n >>> # Selection restored\n \"\"\"\n\n previous_selection = hou.selectedNodes()\n try:\n yield\n finally:\n # Clear the selection\n # todo: does hou.clearAllSelected() do the same?\n for node in hou.selectedNodes():\n node.setSelected(on=False)\n\n if previous_selection:\n for node in previous_selection:\n node.setSelected(on=True)\n\n\ndef reset_framerange():\n \"\"\"Set frame range and FPS to current asset\"\"\"\n\n # Get asset data\n project_name = get_current_project_name()\n asset_name = get_current_asset_name()\n # Get the asset ID from the database for the asset of current context\n asset_doc = get_asset_by_name(project_name, asset_name)\n asset_data = asset_doc[\"data\"]\n\n # Get FPS\n fps = get_asset_fps(asset_doc)\n\n # Get Start and End Frames\n frame_start = asset_data.get(\"frameStart\")\n frame_end = asset_data.get(\"frameEnd\")\n\n if frame_start is None or frame_end is None:\n log.warning(\"No edit information found for %s\" % asset_name)\n return\n\n handle_start = asset_data.get(\"handleStart\", 0)\n handle_end = asset_data.get(\"handleEnd\", 0)\n\n frame_start -= int(handle_start)\n frame_end += int(handle_end)\n\n # Set frame range and FPS\n print(\"Setting scene FPS to {}\".format(int(fps)))\n set_scene_fps(fps)\n hou.playbar.setFrameRange(frame_start, frame_end)\n hou.playbar.setPlaybackRange(frame_start, frame_end)\n hou.setFrame(frame_start)\n\n\ndef get_main_window():\n \"\"\"Acquire Houdini's main window\"\"\"\n if self._parent is None:\n self._parent = hou.ui.mainQtWindow()\n return self._parent\n\n\ndef get_template_from_value(key, value):\n if isinstance(value, float):\n parm = hou.FloatParmTemplate(name=key,\n label=key,\n num_components=1,\n default_value=(value,))\n elif isinstance(value, bool):\n parm = hou.ToggleParmTemplate(name=key,\n label=key,\n default_value=value)\n elif isinstance(value, int):\n parm = hou.IntParmTemplate(name=key,\n label=key,\n num_components=1,\n default_value=(value,))\n elif isinstance(value, six.string_types):\n parm = hou.StringParmTemplate(name=key,\n label=key,\n num_components=1,\n default_value=(value,))\n elif isinstance(value, (dict, list, tuple)):\n parm = hou.StringParmTemplate(name=key,\n label=key,\n num_components=1,\n default_value=(\n JSON_PREFIX + json.dumps(value),))\n else:\n raise TypeError(\"Unsupported type: %r\" % type(value))\n\n return parm\n\n\ndef get_frame_data(node, log=None):\n \"\"\"Get the frame data: `frameStartHandle`, `frameEndHandle`\n and `byFrameStep`.\n\n This function uses Houdini node's `trange`, `t1, `t2` and `t3`\n parameters as the source of truth for the full inclusive frame\n range to render, as such these are considered as the frame\n range including the handles.\n\n The non-inclusive frame start and frame end without handles\n can be computed by subtracting the handles from the inclusive\n frame range.\n\n Args:\n node (hou.Node): ROP node to retrieve frame range from,\n the frame range is assumed to be the frame range\n *including* the start and end handles.\n\n Returns:\n dict: frame data for `frameStartHandle`, `frameEndHandle`\n and `byFrameStep`.\n\n \"\"\"\n\n if log is None:\n log = self.log\n\n data = {}\n\n if node.parm(\"trange\") is None:\n log.debug(\n \"Node has no 'trange' parameter: {}\".format(node.path())\n )\n return data\n\n if node.evalParm(\"trange\") == 0:\n data[\"frameStartHandle\"] = hou.intFrame()\n data[\"frameEndHandle\"] = hou.intFrame()\n data[\"byFrameStep\"] = 1.0\n\n log.info(\n \"Node '{}' has 'Render current frame' set.\\n\"\n \"Asset Handles are ignored.\\n\"\n \"frameStart and frameEnd are set to the \"\n \"current frame.\".format(node.path())\n )\n else:\n data[\"frameStartHandle\"] = int(node.evalParm(\"f1\"))\n data[\"frameEndHandle\"] = int(node.evalParm(\"f2\"))\n data[\"byFrameStep\"] = node.evalParm(\"f3\")\n\n return data\n\n\ndef splitext(name, allowed_multidot_extensions):\n # type: (str, list) -> tuple\n \"\"\"Split file name to name and extension.\n\n Args:\n name (str): File name to split.\n allowed_multidot_extensions (list of str): List of allowed multidot\n extensions.\n\n Returns:\n tuple: Name and extension.\n \"\"\"\n\n for ext in allowed_multidot_extensions:\n if name.endswith(ext):\n return name[:-len(ext)], ext\n\n return os.path.splitext(name)\n\n\ndef get_top_referenced_parm(parm):\n\n processed = set() # disallow infinite loop\n while True:\n if parm.path() in processed:\n raise RuntimeError(\"Parameter references result in cycle.\")\n\n processed.add(parm.path())\n\n ref = parm.getReferencedParm()\n if ref.path() == parm.path():\n # It returns itself when it doesn't reference\n # another parameter\n return ref\n else:\n parm = ref\n\n\ndef evalParmNoFrame(node, parm, pad_character=\"#\"):\n\n parameter = node.parm(parm)\n assert parameter, \"Parameter does not exist: %s.%s\" % (node, parm)\n\n # If the parameter has a parameter reference, then get that\n # parameter instead as otherwise `unexpandedString()` fails.\n parameter = get_top_referenced_parm(parameter)\n\n # Substitute out the frame numbering with padded characters\n try:\n raw = parameter.unexpandedString()\n except hou.Error as exc:\n print(\"Failed: %s\" % parameter)\n raise RuntimeError(exc)\n\n def replace(match):\n padding = 1\n n = match.group(2)\n if n and int(n):\n padding = int(n)\n return pad_character * padding\n\n expression = re.sub(r\"(\\$F([0-9]*))\", replace, raw)\n\n with hou.ScriptEvalContext(parameter):\n return hou.expandStringAtFrame(expression, 0)\n\n\ndef get_color_management_preferences():\n \"\"\"Get default OCIO preferences\"\"\"\n return {\n \"config\": hou.Color.ocio_configPath(),\n \"display\": hou.Color.ocio_defaultDisplay(),\n \"view\": hou.Color.ocio_defaultView()\n }\n\n\ndef get_obj_node_output(obj_node):\n \"\"\"Find output node.\n\n If the node has any output node return the\n output node with the minimum `outputidx`.\n When no output is present return the node\n with the display flag set. If no output node is\n detected then None is returned.\n\n Arguments:\n node (hou.Node): The node to retrieve a single\n the output node for.\n\n Returns:\n Optional[hou.Node]: The child output node.\n\n \"\"\"\n\n outputs = obj_node.subnetOutputs()\n if not outputs:\n return\n\n elif len(outputs) == 1:\n return outputs[0]\n\n else:\n return min(outputs,\n key=lambda node: node.evalParm('outputidx'))\n\n\ndef get_output_children(output_node, include_sops=True):\n \"\"\"Recursively return a list of all output nodes\n contained in this node including this node.\n\n It works in a similar manner to output_node.allNodes().\n \"\"\"\n out_list = [output_node]\n\n if output_node.childTypeCategory() == hou.objNodeTypeCategory():\n for child in output_node.children():\n out_list += get_output_children(child, include_sops=include_sops)\n\n elif include_sops and \\\n output_node.childTypeCategory() == hou.sopNodeTypeCategory():\n out = get_obj_node_output(output_node)\n if out:\n out_list += [out]\n\n return out_list\n\n\ndef get_resolution_from_doc(doc):\n \"\"\"Get resolution from the given asset document. \"\"\"\n\n if not doc or \"data\" not in doc:\n print(\"Entered document is not valid. \\\"{}\\\"\".format(str(doc)))\n return None\n\n resolution_width = doc[\"data\"].get(\"resolutionWidth\")\n resolution_height = doc[\"data\"].get(\"resolutionHeight\")\n\n # Make sure both width and height are set\n if resolution_width is None or resolution_height is None:\n print(\"No resolution information found for \\\"{}\\\"\".format(doc[\"name\"]))\n return None\n\n return int(resolution_width), int(resolution_height)\n\n\ndef set_camera_resolution(camera, asset_doc=None):\n \"\"\"Apply resolution to camera from asset document of the publish\"\"\"\n\n if not asset_doc:\n asset_doc = get_current_project_asset()\n\n resolution = get_resolution_from_doc(asset_doc)\n\n if resolution:\n print(\"Setting camera resolution: {} -> {}x{}\".format(\n camera.name(), resolution[0], resolution[1]\n ))\n camera.parm(\"resx\").set(resolution[0])\n camera.parm(\"resy\").set(resolution[1])\n\n\ndef get_camera_from_container(container):\n \"\"\"Get camera from container node. \"\"\"\n\n cameras = container.recursiveGlob(\n \"*\",\n filter=hou.nodeTypeFilter.ObjCamera,\n include_subnets=False\n )\n\n assert len(cameras) == 1, \"Camera instance must have only one camera\"\n return cameras[0]\n\n\ndef get_current_context_template_data_with_asset_data():\n \"\"\"\n TODOs:\n Support both 'assetData' and 'folderData' in future.\n \"\"\"\n\n context = get_current_context()\n project_name = context[\"project_name\"]\n asset_name = context[\"asset_name\"]\n task_name = context[\"task_name\"]\n host_name = get_current_host_name()\n\n anatomy = Anatomy(project_name)\n project_doc = get_project(project_name)\n asset_doc = get_asset_by_name(project_name, asset_name)\n\n # get context specific vars\n asset_data = asset_doc[\"data\"]\n\n # compute `frameStartHandle` and `frameEndHandle`\n frame_start = asset_data.get(\"frameStart\")\n frame_end = asset_data.get(\"frameEnd\")\n handle_start = asset_data.get(\"handleStart\")\n handle_end = asset_data.get(\"handleEnd\")\n if frame_start is not None and handle_start is not None:\n asset_data[\"frameStartHandle\"] = frame_start - handle_start\n\n if frame_end is not None and handle_end is not None:\n asset_data[\"frameEndHandle\"] = frame_end + handle_end\n\n template_data = get_template_data(\n project_doc, asset_doc, task_name, host_name\n )\n template_data[\"root\"] = anatomy.roots\n template_data[\"assetData\"] = asset_data\n\n return template_data\n\n\ndef get_context_var_changes():\n \"\"\"get context var changes.\"\"\"\n\n houdini_vars_to_update = {}\n\n project_settings = get_current_project_settings()\n houdini_vars_settings = \\\n project_settings[\"houdini\"][\"general\"][\"update_houdini_var_context\"]\n\n if not houdini_vars_settings[\"enabled\"]:\n return houdini_vars_to_update\n\n houdini_vars = houdini_vars_settings[\"houdini_vars\"]\n\n # No vars specified - nothing to do\n if not houdini_vars:\n return houdini_vars_to_update\n\n # Get Template data\n template_data = get_current_context_template_data_with_asset_data()\n\n # Set Houdini Vars\n for item in houdini_vars:\n # For consistency reasons we always force all vars to be uppercase\n # Also remove any leading, and trailing whitespaces.\n var = item[\"var\"].strip().upper()\n\n # get and resolve template in value\n item_value = StringTemplate.format_template(\n item[\"value\"],\n template_data\n )\n\n if var == \"JOB\" and item_value == \"\":\n # sync $JOB to $HIP if $JOB is empty\n item_value = os.environ[\"HIP\"]\n\n if item[\"is_directory\"]:\n item_value = item_value.replace(\"\\\\\", \"/\")\n\n current_value = hou.hscript(\"echo -n `${}`\".format(var))[0]\n\n if current_value != item_value:\n houdini_vars_to_update[var] = (\n current_value, item_value, item[\"is_directory\"]\n )\n\n return houdini_vars_to_update\n\n\ndef update_houdini_vars_context():\n \"\"\"Update asset context variables\"\"\"\n\n for var, (_old, new, is_directory) in get_context_var_changes().items():\n if is_directory:\n try:\n os.makedirs(new)\n except OSError as e:\n if e.errno != errno.EEXIST:\n print(\n \"Failed to create ${} dir. Maybe due to \"\n \"insufficient permissions.\".format(var)\n )\n\n hou.hscript(\"set {}={}\".format(var, new))\n os.environ[var] = new\n print(\"Updated ${} to {}\".format(var, new))\n\n\ndef update_houdini_vars_context_dialog():\n \"\"\"Show pop-up to update asset context variables\"\"\"\n update_vars = get_context_var_changes()\n if not update_vars:\n # Nothing to change\n print(\"Nothing to change, Houdini vars are already up to date.\")\n return\n\n message = \"\\n\".join(\n \"${}: {} -> {}\".format(var, old or \"None\", new or \"None\")\n for var, (old, new, _is_directory) in update_vars.items()\n )\n\n # TODO: Use better UI!\n parent = hou.ui.mainQtWindow()\n dialog = popup.Popup(parent=parent)\n dialog.setModal(True)\n dialog.setWindowTitle(\"Houdini scene has outdated asset variables\")\n dialog.setMessage(message)\n dialog.setButtonText(\"Fix\")\n\n # on_show is the Fix button clicked callback\n dialog.on_clicked.connect(update_houdini_vars_context)\n\n dialog.show()\n\n\ndef publisher_show_and_publish(comment=None):\n \"\"\"Open publisher window and trigger publishing action.\n\n Args:\n comment (Optional[str]): Comment to set in publisher window.\n \"\"\"\n\n main_window = get_main_window()\n publisher_window = get_tool_by_name(\n tool_name=\"publisher\",\n parent=main_window,\n )\n publisher_window.show_and_publish(comment)\n\n\ndef find_rop_input_dependencies(input_tuple):\n \"\"\"Self publish from ROP nodes.\n\n Arguments:\n tuple (hou.RopNode.inputDependencies) which can be a nested tuples\n represents the input dependencies of the ROP node, consisting of ROPs,\n and the frames that need to be be rendered prior to rendering the ROP.\n\n Returns:\n list of the RopNode.path() that can be found inside\n the input tuple.\n \"\"\"\n\n out_list = []\n if isinstance(input_tuple[0], hou.RopNode):\n return input_tuple[0].path()\n\n if isinstance(input_tuple[0], tuple):\n for item in input_tuple:\n out_list.append(find_rop_input_dependencies(item))\n\n return out_list\n\n\ndef self_publish():\n \"\"\"Self publish from ROP nodes.\n\n Firstly, it gets the node and its dependencies.\n Then, it deactivates all other ROPs\n And finaly, it triggers the publishing action.\n \"\"\"\n\n result, comment = hou.ui.readInput(\n \"Add Publish Comment\",\n buttons=(\"Publish\", \"Cancel\"),\n title=\"Publish comment\",\n close_choice=1\n )\n\n if result:\n return\n\n current_node = hou.node(\".\")\n inputs_paths = find_rop_input_dependencies(\n current_node.inputDependencies()\n )\n inputs_paths.append(current_node.path())\n\n host = registered_host()\n context = CreateContext(host, reset=True)\n\n for instance in context.instances:\n node_path = instance.data.get(\"instance_node\")\n instance[\"active\"] = node_path and node_path in inputs_paths\n\n context.save_changes()\n\n publisher_show_and_publish(comment)\n\n\ndef add_self_publish_button(node):\n \"\"\"Adds a self publish button to the rop node.\"\"\"\n\n label = os.environ.get(\"AVALON_LABEL\") or \"AYON\"\n\n button_parm = hou.ButtonParmTemplate(\n \"ayon_self_publish\",\n \"{} Publish\".format(label),\n script_callback=\"from openpype.hosts.houdini.api.lib import \"\n \"self_publish; self_publish()\",\n script_callback_language=hou.scriptLanguage.Python,\n join_with_next=True\n )\n\n template = node.parmTemplateGroup()\n template.insertBefore((0,), button_parm)\n node.setParmTemplateGroup(template)\n"
},
{
"path": "openpype/hosts/houdini/api/pipeline.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Pipeline tools for OpenPype Houdini integration.\"\"\"\nimport os\nimport sys\nimport logging\n\nimport hou # noqa\n\nfrom openpype.host import HostBase, IWorkfileHost, ILoadHost, IPublishHost\n\nimport pyblish.api\n\nfrom openpype.pipeline import (\n register_creator_plugin_path,\n register_loader_plugin_path,\n register_inventory_action_path,\n AVALON_CONTAINER_ID,\n)\nfrom openpype.pipeline.load import any_outdated_containers\nfrom openpype.hosts.houdini import HOUDINI_HOST_DIR\nfrom openpype.hosts.houdini.api import lib, shelves, creator_node_shelves\n\nfrom openpype.lib import (\n register_event_callback,\n emit_event,\n)\n\n\nlog = logging.getLogger(\"openpype.hosts.houdini\")\n\nAVALON_CONTAINERS = \"/obj/AVALON_CONTAINERS\"\nCONTEXT_CONTAINER = \"/obj/OpenPypeContext\"\nIS_HEADLESS = not hasattr(hou, \"ui\")\n\nPLUGINS_DIR = os.path.join(HOUDINI_HOST_DIR, \"plugins\")\nPUBLISH_PATH = os.path.join(PLUGINS_DIR, \"publish\")\nLOAD_PATH = os.path.join(PLUGINS_DIR, \"load\")\nCREATE_PATH = os.path.join(PLUGINS_DIR, \"create\")\nINVENTORY_PATH = os.path.join(PLUGINS_DIR, \"inventory\")\n\n\nclass HoudiniHost(HostBase, IWorkfileHost, ILoadHost, IPublishHost):\n name = \"houdini\"\n\n def __init__(self):\n super(HoudiniHost, self).__init__()\n self._op_events = {}\n self._has_been_setup = False\n\n def install(self):\n pyblish.api.register_host(\"houdini\")\n pyblish.api.register_host(\"hython\")\n pyblish.api.register_host(\"hpython\")\n\n pyblish.api.register_plugin_path(PUBLISH_PATH)\n register_loader_plugin_path(LOAD_PATH)\n register_creator_plugin_path(CREATE_PATH)\n register_inventory_action_path(INVENTORY_PATH)\n\n log.info(\"Installing callbacks ... \")\n # register_event_callback(\"init\", on_init)\n self._register_callbacks()\n register_event_callback(\"before.save\", before_save)\n register_event_callback(\"save\", on_save)\n register_event_callback(\"open\", on_open)\n register_event_callback(\"new\", on_new)\n\n self._has_been_setup = True\n\n # Set asset settings for the empty scene directly after launch of\n # Houdini so it initializes into the correct scene FPS,\n # Frame Range, etc.\n # TODO: make sure this doesn't trigger when\n # opening with last workfile.\n _set_context_settings()\n\n if not IS_HEADLESS:\n import hdefereval # noqa, hdefereval is only available in ui mode\n # Defer generation of shelves due to issue on Windows where shelf\n # initialization during start up delays Houdini UI by minutes\n # making it extremely slow to launch.\n hdefereval.executeDeferred(shelves.generate_shelves)\n\n if not IS_HEADLESS:\n import hdefereval # noqa, hdefereval is only available in ui mode\n hdefereval.executeDeferred(creator_node_shelves.install)\n\n def workfile_has_unsaved_changes(self):\n return hou.hipFile.hasUnsavedChanges()\n\n def get_workfile_extensions(self):\n return [\".hip\", \".hiplc\", \".hipnc\"]\n\n def save_workfile(self, dst_path=None):\n # Force forwards slashes to avoid segfault\n if dst_path:\n dst_path = dst_path.replace(\"\\\\\", \"/\")\n hou.hipFile.save(file_name=dst_path,\n save_to_recent_files=True)\n return dst_path\n\n def open_workfile(self, filepath):\n # Force forwards slashes to avoid segfault\n filepath = filepath.replace(\"\\\\\", \"/\")\n\n hou.hipFile.load(filepath,\n suppress_save_prompt=True,\n ignore_load_warnings=False)\n\n return filepath\n\n def get_current_workfile(self):\n current_filepath = hou.hipFile.path()\n if (os.path.basename(current_filepath) == \"untitled.hip\" and\n not os.path.exists(current_filepath)):\n # By default a new scene in houdini is saved in the current\n # working directory as \"untitled.hip\" so we need to capture\n # that and consider it 'not saved' when it's in that state.\n return None\n\n return current_filepath\n\n def get_containers(self):\n return ls()\n\n def _register_callbacks(self):\n for event in self._op_events.copy().values():\n if event is None:\n continue\n\n try:\n hou.hipFile.removeEventCallback(event)\n except RuntimeError as e:\n log.info(e)\n\n self._op_events[on_file_event_callback] = hou.hipFile.addEventCallback(\n on_file_event_callback\n )\n\n @staticmethod\n def create_context_node():\n \"\"\"Helper for creating context holding node.\n\n Returns:\n hou.Node: context node\n\n \"\"\"\n obj_network = hou.node(\"/obj\")\n op_ctx = obj_network.createNode(\"subnet\",\n node_name=\"OpenPypeContext\",\n run_init_scripts=False,\n load_contents=False)\n\n op_ctx.moveToGoodPosition()\n op_ctx.setBuiltExplicitly(False)\n op_ctx.setCreatorState(\"OpenPype\")\n op_ctx.setComment(\"OpenPype node to hold context metadata\")\n op_ctx.setColor(hou.Color((0.081, 0.798, 0.810)))\n op_ctx.setDisplayFlag(False)\n op_ctx.hide(True)\n return op_ctx\n\n def update_context_data(self, data, changes):\n op_ctx = hou.node(CONTEXT_CONTAINER)\n if not op_ctx:\n op_ctx = self.create_context_node()\n\n lib.imprint(op_ctx, data)\n\n def get_context_data(self):\n op_ctx = hou.node(CONTEXT_CONTAINER)\n if not op_ctx:\n op_ctx = self.create_context_node()\n return lib.read(op_ctx)\n\n def save_file(self, dst_path=None):\n # Force forwards slashes to avoid segfault\n dst_path = dst_path.replace(\"\\\\\", \"/\")\n\n hou.hipFile.save(file_name=dst_path,\n save_to_recent_files=True)\n\n\ndef on_file_event_callback(event):\n if event == hou.hipFileEventType.AfterLoad:\n emit_event(\"open\")\n elif event == hou.hipFileEventType.AfterSave:\n emit_event(\"save\")\n elif event == hou.hipFileEventType.BeforeSave:\n emit_event(\"before.save\")\n elif event == hou.hipFileEventType.AfterClear:\n emit_event(\"new\")\n\n\ndef containerise(name,\n namespace,\n nodes,\n context,\n loader=None,\n suffix=\"\"):\n \"\"\"Bundle `nodes` into a subnet and imprint it with metadata\n\n Containerisation enables a tracking of version, author and origin\n for loaded assets.\n\n Arguments:\n name (str): Name of resulting assembly\n namespace (str): Namespace under which to host container\n nodes (list): Long names of nodes to containerise\n context (dict): Asset information\n loader (str, optional): Name of loader used to produce this container.\n suffix (str, optional): Suffix of container, defaults to `_CON`.\n\n Returns:\n container (str): Name of container assembly\n\n \"\"\"\n\n # Ensure AVALON_CONTAINERS subnet exists\n subnet = hou.node(AVALON_CONTAINERS)\n if subnet is None:\n obj_network = hou.node(\"/obj\")\n subnet = obj_network.createNode(\"subnet\",\n node_name=\"AVALON_CONTAINERS\")\n\n # Create proper container name\n container_name = \"{}_{}\".format(name, suffix or \"CON\")\n container = hou.node(\"/obj/{}\".format(name))\n container.setName(container_name, unique_name=True)\n\n data = {\n \"schema\": \"openpype:container-2.0\",\n \"id\": AVALON_CONTAINER_ID,\n \"name\": name,\n \"namespace\": namespace,\n \"loader\": str(loader),\n \"representation\": str(context[\"representation\"][\"_id\"]),\n }\n\n lib.imprint(container, data)\n\n # \"Parent\" the container under the container network\n hou.moveNodesTo([container], subnet)\n\n subnet.node(container_name).moveToGoodPosition()\n\n return container\n\n\ndef parse_container(container):\n \"\"\"Return the container node's full container data.\n\n Args:\n container (hou.Node): A container node name.\n\n Returns:\n dict: The container schema data for this container node.\n\n \"\"\"\n data = lib.read(container)\n\n # Backwards compatibility pre-schemas for containers\n data[\"schema\"] = data.get(\"schema\", \"openpype:container-1.0\")\n\n # Append transient data\n data[\"objectName\"] = container.path()\n data[\"node\"] = container\n\n return data\n\n\ndef ls():\n containers = []\n for identifier in (AVALON_CONTAINER_ID,\n \"pyblish.mindbender.container\"):\n containers += lib.lsattr(\"id\", identifier)\n\n for container in sorted(containers,\n # Hou 19+ Python 3 hou.ObjNode are not\n # sortable due to not supporting greater\n # than comparisons\n key=lambda node: node.path()):\n yield parse_container(container)\n\n\ndef before_save():\n return lib.validate_fps()\n\n\ndef on_save():\n\n log.info(\"Running callback on save..\")\n\n # update houdini vars\n lib.update_houdini_vars_context_dialog()\n\n nodes = lib.get_id_required_nodes()\n for node, new_id in lib.generate_ids(nodes):\n lib.set_id(node, new_id, overwrite=False)\n\n\ndef _show_outdated_content_popup():\n # Get main window\n parent = lib.get_main_window()\n if parent is None:\n log.info(\"Skipping outdated content pop-up \"\n \"because Houdini window can't be found.\")\n else:\n from openpype.widgets import popup\n\n # Show outdated pop-up\n def _on_show_inventory():\n from openpype.tools.utils import host_tools\n host_tools.show_scene_inventory(parent=parent)\n\n dialog = popup.Popup(parent=parent)\n dialog.setWindowTitle(\"Houdini scene has outdated content\")\n dialog.setMessage(\"There are outdated containers in \"\n \"your Houdini scene.\")\n dialog.on_clicked.connect(_on_show_inventory)\n dialog.show()\n\n\ndef on_open():\n\n if not hou.isUIAvailable():\n log.debug(\"Batch mode detected, ignoring `on_open` callbacks..\")\n return\n\n log.info(\"Running callback on open..\")\n\n # update houdini vars\n lib.update_houdini_vars_context_dialog()\n\n # Validate FPS after update_task_from_path to\n # ensure it is using correct FPS for the asset\n lib.validate_fps()\n\n if any_outdated_containers():\n parent = lib.get_main_window()\n if parent is None:\n # When opening Houdini with last workfile on launch the UI hasn't\n # initialized yet completely when the `on_open` callback triggers.\n # We defer the dialog popup to wait for the UI to become available.\n # We assume it will open because `hou.isUIAvailable()` returns True\n import hdefereval\n hdefereval.executeDeferred(_show_outdated_content_popup)\n else:\n _show_outdated_content_popup()\n\n log.warning(\"Scene has outdated content.\")\n\n\ndef on_new():\n \"\"\"Set project resolution and fps when create a new file\"\"\"\n\n if hou.hipFile.isLoadingHipFile():\n # This event also triggers when Houdini opens a file due to the\n # new event being registered to 'afterClear'. As such we can skip\n # 'new' logic if the user is opening a file anyway\n log.debug(\"Skipping on new callback due to scene being opened.\")\n return\n\n log.info(\"Running callback on new..\")\n _set_context_settings()\n\n # It seems that the current frame always gets reset to frame 1 on\n # new scene. So we enforce current frame to be at the start of the playbar\n # with execute deferred\n def _enforce_start_frame():\n start = hou.playbar.playbackRange()[0]\n hou.setFrame(start)\n\n if hou.isUIAvailable():\n import hdefereval\n hdefereval.executeDeferred(_enforce_start_frame)\n else:\n # Run without execute deferred when no UI is available because\n # without UI `hdefereval` is not available to import\n _enforce_start_frame()\n\n\ndef _set_context_settings():\n \"\"\"Apply the project settings from the project definition\n\n Settings can be overwritten by an asset if the asset.data contains\n any information regarding those settings.\n\n Examples of settings:\n fps\n resolution\n renderer\n\n Returns:\n None\n \"\"\"\n\n lib.reset_framerange()\n lib.update_houdini_vars_context()\n"
},
{
"path": "openpype/hosts/houdini/api/plugin.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Houdini specific Avalon/Pyblish plugin definitions.\"\"\"\nimport sys\nfrom abc import (\n ABCMeta\n)\nimport six\nimport hou\n\nfrom openpype import AYON_SERVER_ENABLED\nfrom openpype.pipeline import (\n CreatorError,\n LegacyCreator,\n Creator as NewCreator,\n CreatedInstance\n)\nfrom openpype.lib import BoolDef\nfrom .lib import imprint, read, lsattr, add_self_publish_button\n\n\nclass OpenPypeCreatorError(CreatorError):\n pass\n\n\nclass Creator(LegacyCreator):\n \"\"\"Creator plugin to create instances in Houdini\n\n To support the wide range of node types for render output (Alembic, VDB,\n Mantra) the Creator needs a node type to create the correct instance\n\n By default, if none is given, is `geometry`. An example of accepted node\n types: geometry, alembic, ifd (mantra)\n\n Please check the Houdini documentation for more node types.\n\n Tip: to find the exact node type to create press the `i` left of the node\n when hovering over a node. The information is visible under the name of\n the node.\n\n Deprecated:\n This creator is deprecated and will be removed in future version.\n\n \"\"\"\n defaults = ['Main']\n\n def __init__(self, *args, **kwargs):\n super(Creator, self).__init__(*args, **kwargs)\n self.nodes = []\n\n def process(self):\n \"\"\"This is the base functionality to create instances in Houdini\n\n The selected nodes are stored in self to be used in an override method.\n This is currently necessary in order to support the multiple output\n types in Houdini which can only be rendered through their own node.\n\n Default node type if none is given is `geometry`\n\n It also makes it easier to apply custom settings per instance type\n\n Example of override method for Alembic:\n\n def process(self):\n instance = super(CreateEpicNode, self, process()\n # Set parameters for Alembic node\n instance.setParms(\n {\"sop_path\": \"$HIP/%s.abc\" % self.nodes[0]}\n )\n\n Returns:\n hou.Node\n\n \"\"\"\n try:\n if (self.options or {}).get(\"useSelection\"):\n self.nodes = hou.selectedNodes()\n\n # Get the node type and remove it from the data, not needed\n node_type = self.data.pop(\"node_type\", None)\n if node_type is None:\n node_type = \"geometry\"\n\n # Get out node\n out = hou.node(\"/out\")\n instance = out.createNode(node_type, node_name=self.name)\n instance.moveToGoodPosition()\n\n imprint(instance, self.data)\n\n self._process(instance)\n\n except hou.Error as er:\n six.reraise(\n OpenPypeCreatorError,\n OpenPypeCreatorError(\"Creator error: {}\".format(er)),\n sys.exc_info()[2])\n\n\nclass HoudiniCreatorBase(object):\n @staticmethod\n def cache_subsets(shared_data):\n \"\"\"Cache instances for Creators to shared data.\n\n Create `houdini_cached_subsets` key when needed in shared data and\n fill it with all collected instances from the scene under its\n respective creator identifiers.\n\n Create `houdini_cached_legacy_subsets` key for any legacy instances\n detected in the scene as instances per family.\n\n Args:\n Dict[str, Any]: Shared data.\n\n Return:\n Dict[str, Any]: Shared data dictionary.\n\n \"\"\"\n if shared_data.get(\"houdini_cached_subsets\") is None:\n cache = dict()\n cache_legacy = dict()\n\n for node in lsattr(\"id\", \"pyblish.avalon.instance\"):\n\n creator_identifier_parm = node.parm(\"creator_identifier\")\n if creator_identifier_parm:\n # creator instance\n creator_id = creator_identifier_parm.eval()\n cache.setdefault(creator_id, []).append(node)\n\n else:\n # legacy instance\n family_parm = node.parm(\"family\")\n if not family_parm:\n # must be a broken instance\n continue\n\n family = family_parm.eval()\n cache_legacy.setdefault(family, []).append(node)\n\n shared_data[\"houdini_cached_subsets\"] = cache\n shared_data[\"houdini_cached_legacy_subsets\"] = cache_legacy\n\n return shared_data\n\n @staticmethod\n def create_instance_node(\n asset_name, node_name, parent, node_type=\"geometry\"\n ):\n # type: (str, str, str) -> hou.Node\n \"\"\"Create node representing instance.\n\n Arguments:\n asset_name (str): Asset name.\n node_name (str): Name of the new node.\n parent (str): Name of the parent node.\n node_type (str, optional): Type of the node.\n\n Returns:\n hou.Node: Newly created instance node.\n\n \"\"\"\n parent_node = hou.node(parent)\n instance_node = parent_node.createNode(\n node_type, node_name=node_name)\n instance_node.moveToGoodPosition()\n return instance_node\n\n\n@six.add_metaclass(ABCMeta)\nclass HoudiniCreator(NewCreator, HoudiniCreatorBase):\n \"\"\"Base class for most of the Houdini creator plugins.\"\"\"\n selected_nodes = []\n settings_name = None\n add_publish_button = False\n\n def create(self, subset_name, instance_data, pre_create_data):\n try:\n self.selected_nodes = []\n\n if pre_create_data.get(\"use_selection\"):\n self.selected_nodes = hou.selectedNodes()\n\n # Get the node type and remove it from the data, not needed\n node_type = instance_data.pop(\"node_type\", None)\n if node_type is None:\n node_type = \"geometry\"\n\n if AYON_SERVER_ENABLED:\n asset_name = instance_data[\"folderPath\"]\n else:\n asset_name = instance_data[\"asset\"]\n\n instance_node = self.create_instance_node(\n asset_name, subset_name, \"/out\", node_type)\n\n self.customize_node_look(instance_node)\n\n instance_data[\"instance_node\"] = instance_node.path()\n instance_data[\"instance_id\"] = instance_node.path()\n instance = CreatedInstance(\n self.family,\n subset_name,\n instance_data,\n self)\n self._add_instance_to_context(instance)\n self.imprint(instance_node, instance.data_to_store())\n\n if self.add_publish_button:\n add_self_publish_button(instance_node)\n\n return instance\n\n except hou.Error as er:\n six.reraise(\n OpenPypeCreatorError,\n OpenPypeCreatorError(\"Creator error: {}\".format(er)),\n sys.exc_info()[2])\n\n def lock_parameters(self, node, parameters):\n \"\"\"Lock list of specified parameters on the node.\n\n Args:\n node (hou.Node): Houdini node to lock parameters on.\n parameters (list of str): List of parameter names.\n\n \"\"\"\n for name in parameters:\n try:\n parm = node.parm(name)\n parm.lock(True)\n except AttributeError:\n self.log.debug(\"missing lock pattern {}\".format(name))\n\n def collect_instances(self):\n # cache instances if missing\n self.cache_subsets(self.collection_shared_data)\n for instance in self.collection_shared_data[\n \"houdini_cached_subsets\"].get(self.identifier, []):\n\n node_data = read(instance)\n\n # Node paths are always the full node path since that is unique\n # Because it's the node's path it's not written into attributes\n # but explicitly collected\n node_path = instance.path()\n node_data[\"instance_id\"] = node_path\n node_data[\"instance_node\"] = node_path\n\n created_instance = CreatedInstance.from_existing(\n node_data, self\n )\n self._add_instance_to_context(created_instance)\n\n def update_instances(self, update_list):\n for created_inst, changes in update_list:\n instance_node = hou.node(created_inst.get(\"instance_node\"))\n new_values = {\n key: changes[key].new_value\n for key in changes.changed_keys\n }\n # Update parm templates and values\n self.imprint(\n instance_node,\n new_values,\n update=True\n )\n\n def imprint(self, node, values, update=False):\n # Never store instance node and instance id since that data comes\n # from the node's path\n values.pop(\"instance_node\", None)\n values.pop(\"instance_id\", None)\n imprint(node, values, update=update)\n\n def remove_instances(self, instances):\n \"\"\"Remove specified instance from the scene.\n\n This is only removing `id` parameter so instance is no longer\n instance, because it might contain valuable data for artist.\n\n \"\"\"\n for instance in instances:\n instance_node = hou.node(instance.data.get(\"instance_node\"))\n if instance_node:\n instance_node.destroy()\n\n self._remove_instance_from_context(instance)\n\n def get_pre_create_attr_defs(self):\n return [\n BoolDef(\"use_selection\", label=\"Use selection\")\n ]\n\n @staticmethod\n def customize_node_look(\n node, color=None,\n shape=\"chevron_down\"):\n \"\"\"Set custom look for instance nodes.\n\n Args:\n node (hou.Node): Node to set look.\n color (hou.Color, Optional): Color of the node.\n shape (str, Optional): Shape name of the node.\n\n Returns:\n None\n\n \"\"\"\n if not color:\n color = hou.Color((0.616, 0.871, 0.769))\n node.setUserData('nodeshape', shape)\n node.setColor(color)\n\n def get_network_categories(self):\n \"\"\"Return in which network view type this creator should show.\n\n The node type categories returned here will be used to define where\n the creator will show up in the TAB search for nodes in Houdini's\n Network View.\n\n This can be overridden in inherited classes to define where that\n particular Creator should be visible in the TAB search.\n\n Returns:\n list: List of houdini node type categories\n\n \"\"\"\n return [hou.ropNodeTypeCategory()]\n\n def apply_settings(self, project_settings):\n \"\"\"Method called on initialization of plugin to apply settings.\"\"\"\n\n # Apply General Settings\n houdini_general_settings = project_settings[\"houdini\"][\"general\"]\n self.add_publish_button = houdini_general_settings.get(\n \"add_self_publish_button\", False)\n\n # Apply Creator Settings\n settings_name = self.settings_name\n if settings_name is None:\n settings_name = self.__class__.__name__\n\n settings = project_settings[\"houdini\"][\"create\"]\n settings = settings.get(settings_name)\n if settings is None:\n self.log.debug(\n \"No settings found for {}\".format(self.__class__.__name__)\n )\n return\n\n for key, value in settings.items():\n setattr(self, key, value)\n"
},
{
"path": "openpype/hosts/houdini/api/shelves.py",
"content": "import os\nimport re\nimport logging\nimport platform\n\nfrom openpype.settings import get_project_settings\nfrom openpype.pipeline import get_current_project_name\n\nfrom openpype.lib import StringTemplate\n\nimport hou\n\nfrom .lib import get_current_context_template_data_with_asset_data\n\nlog = logging.getLogger(\"openpype.hosts.houdini.shelves\")\n\n\ndef generate_shelves():\n \"\"\"This function generates complete shelves from shelf set to tools\n in Houdini from openpype project settings houdini shelf definition.\n \"\"\"\n current_os = platform.system().lower()\n\n # load configuration of houdini shelves\n project_name = get_current_project_name()\n project_settings = get_project_settings(project_name)\n shelves_configs = project_settings[\"houdini\"][\"shelves\"]\n\n if not shelves_configs:\n log.debug(\"No custom shelves found in project settings.\")\n return\n\n # Get Template data\n template_data = get_current_context_template_data_with_asset_data()\n\n for config in shelves_configs:\n selected_option = config[\"options\"]\n shelf_set_config = config[selected_option]\n\n shelf_set_filepath = shelf_set_config.get('shelf_set_source_path')\n if shelf_set_filepath:\n shelf_set_os_filepath = shelf_set_filepath[current_os]\n if shelf_set_os_filepath:\n shelf_set_os_filepath = get_path_using_template_data(\n shelf_set_os_filepath, template_data\n )\n if not os.path.isfile(shelf_set_os_filepath):\n log.error(\"Shelf path doesn't exist - \"\n \"{}\".format(shelf_set_os_filepath))\n continue\n\n hou.shelves.loadFile(shelf_set_os_filepath)\n continue\n\n shelf_set_name = shelf_set_config.get('shelf_set_name')\n if not shelf_set_name:\n log.warning(\"No name found in shelf set definition.\")\n continue\n\n shelves_definition = shelf_set_config.get('shelf_definition')\n if not shelves_definition:\n log.debug(\n \"No shelf definition found for shelf set named '{}'\".format(\n shelf_set_name\n )\n )\n continue\n\n shelf_set = get_or_create_shelf_set(shelf_set_name)\n for shelf_definition in shelves_definition:\n shelf_name = shelf_definition.get('shelf_name')\n if not shelf_name:\n log.warning(\"No name found in shelf definition.\")\n continue\n\n shelf = get_or_create_shelf(shelf_name)\n\n if not shelf_definition.get('tools_list'):\n log.debug(\n \"No tool definition found for shelf named {}\".format(\n shelf_name\n )\n )\n continue\n\n mandatory_attributes = {'label', 'script'}\n for tool_definition in shelf_definition.get('tools_list'):\n # We verify that the name and script attributes of the tool\n # are set\n if not all(\n tool_definition[key] for key in mandatory_attributes\n ):\n log.warning(\n \"You need to specify at least the name and the \"\n \"script path of the tool.\")\n continue\n\n tool = get_or_create_tool(\n tool_definition, shelf, template_data\n )\n\n if not tool:\n continue\n\n # Add the tool to the shelf if not already in it\n if tool not in shelf.tools():\n shelf.setTools(list(shelf.tools()) + [tool])\n\n # Add the shelf in the shelf set if not already in it\n if shelf not in shelf_set.shelves():\n shelf_set.setShelves(shelf_set.shelves() + (shelf,))\n\n\ndef get_or_create_shelf_set(shelf_set_label):\n \"\"\"This function verifies if the shelf set label exists. If not,\n creates a new shelf set.\n\n Arguments:\n shelf_set_label (str): The label of the shelf set\n\n Returns:\n hou.ShelfSet: The shelf set existing or the new one\n \"\"\"\n all_shelves_sets = hou.shelves.shelfSets().values()\n\n shelf_set = next((shelf for shelf in all_shelves_sets if\n shelf.label() == shelf_set_label), None)\n if shelf_set:\n return shelf_set\n\n shelf_set_name = shelf_set_label.replace(' ', '_').lower()\n new_shelf_set = hou.shelves.newShelfSet(\n name=shelf_set_name,\n label=shelf_set_label\n )\n return new_shelf_set\n\n\ndef get_or_create_shelf(shelf_label):\n \"\"\"This function verifies if the shelf label exists. If not, creates\n a new shelf.\n\n Arguments:\n shelf_label (str): The label of the shelf\n\n Returns:\n hou.Shelf: The shelf existing or the new one\n \"\"\"\n all_shelves = hou.shelves.shelves().values()\n\n shelf = next((s for s in all_shelves if s.label() == shelf_label), None)\n if shelf:\n return shelf\n\n shelf_name = shelf_label.replace(' ', '_').lower()\n new_shelf = hou.shelves.newShelf(\n name=shelf_name,\n label=shelf_label\n )\n return new_shelf\n\n\ndef get_or_create_tool(tool_definition, shelf, template_data):\n \"\"\"This function verifies if the tool exists and updates it. If not, creates\n a new one.\n\n Arguments:\n tool_definition (dict): Dict with label, script, icon and help\n shelf (hou.Shelf): The parent shelf of the tool\n\n Returns:\n hou.Tool: The tool updated or the new one\n \"\"\"\n\n tool_label = tool_definition.get(\"label\")\n if not tool_label:\n log.warning(\"Skipped shelf without label\")\n return\n\n script_path = tool_definition[\"script\"]\n script_path = get_path_using_template_data(script_path, template_data)\n if not script_path or not os.path.exists(script_path):\n log.warning(\"This path doesn't exist - {}\".format(script_path))\n return\n\n icon_path = tool_definition[\"icon\"]\n if icon_path:\n icon_path = get_path_using_template_data(icon_path, template_data)\n tool_definition[\"icon\"] = icon_path\n\n existing_tools = shelf.tools()\n existing_tool = next(\n (tool for tool in existing_tools if tool.label() == tool_label),\n None\n )\n\n with open(script_path) as stream:\n script = stream.read()\n\n tool_definition[\"script\"] = script\n\n if existing_tool:\n tool_definition.pop(\"label\", None)\n existing_tool.setData(**tool_definition)\n return existing_tool\n\n tool_name = re.sub(r\"[^\\w\\d]+\", \"_\", tool_label).lower()\n return hou.shelves.newTool(name=tool_name, **tool_definition)\n\n\ndef get_path_using_template_data(path, template_data):\n path = StringTemplate.format_template(path, template_data)\n path = path.replace(\"\\\\\", \"/\")\n\n return path\n"
},
{
"path": "openpype/hosts/houdini/api/usd.py",
"content": "\"\"\"Houdini-specific USD Library functions.\"\"\"\n\nimport contextlib\nimport logging\n\nfrom qtpy import QtWidgets, QtCore, QtGui\n\nfrom openpype import style\nfrom openpype.client import get_asset_by_name\nfrom openpype.pipeline import legacy_io\nfrom openpype.tools.utils.assets_widget import SingleSelectAssetsWidget\n\nfrom pxr import Sdf\n\n\nlog = logging.getLogger(__name__)\n\n\nclass SelectAssetDialog(QtWidgets.QWidget):\n \"\"\"Frameless assets dialog to select asset with double click.\n\n Args:\n parm: Parameter where selected asset name is set.\n \"\"\"\n\n def __init__(self, parm):\n self.setWindowTitle(\"Pick Asset\")\n self.setWindowFlags(QtCore.Qt.FramelessWindowHint | QtCore.Qt.Popup)\n\n assets_widget = SingleSelectAssetsWidget(legacy_io, parent=self)\n\n layout = QtWidgets.QHBoxLayout(self)\n layout.addWidget(assets_widget)\n\n assets_widget.double_clicked.connect(self._set_parameter)\n self._assets_widget = assets_widget\n self._parm = parm\n\n def _set_parameter(self):\n name = self._assets_widget.get_selected_asset_name()\n self._parm.set(name)\n self.close()\n\n def _on_show(self):\n pos = QtGui.QCursor.pos()\n # Select the current asset if there is any\n select_id = None\n name = self._parm.eval()\n if name:\n project_name = legacy_io.active_project()\n db_asset = get_asset_by_name(project_name, name, fields=[\"_id\"])\n if db_asset:\n select_id = db_asset[\"_id\"]\n\n # Set stylesheet\n self.setStyleSheet(style.load_stylesheet())\n # Refresh assets (is threaded)\n self._assets_widget.refresh()\n # Select asset - must be done after refresh\n if select_id is not None:\n self._assets_widget.select_asset(select_id)\n\n # Show cursor (top right of window) near cursor\n self.resize(250, 400)\n self.move(self.mapFromGlobal(pos) - QtCore.QPoint(self.width(), 0))\n\n def showEvent(self, event):\n super(SelectAssetDialog, self).showEvent(event)\n self._on_show()\n\n\ndef pick_asset(node):\n \"\"\"Show a user interface to select an Asset in the project\n\n When double clicking an asset it will set the Asset value in the\n 'asset' parameter.\n\n \"\"\"\n\n parm = node.parm(\"asset_name\")\n if not parm:\n log.error(\"Node has no 'asset' parameter: %s\", node)\n return\n\n # Construct a frameless popup so it automatically\n # closes when clicked outside of it.\n global tool\n tool = SelectAssetDialog(parm)\n tool.show()\n\n\ndef add_usd_output_processor(ropnode, processor):\n \"\"\"Add USD Output Processor to USD Rop node.\n\n Args:\n ropnode (hou.RopNode): The USD Rop node.\n processor (str): The output processor name. This is the basename of\n the python file that contains the Houdini USD Output Processor.\n\n \"\"\"\n\n import loputils\n\n loputils.handleOutputProcessorAdd(\n {\n \"node\": ropnode,\n \"parm\": ropnode.parm(\"outputprocessors\"),\n \"script_value\": processor,\n }\n )\n\n\ndef remove_usd_output_processor(ropnode, processor):\n \"\"\"Removes USD Output Processor from USD Rop node.\n\n Args:\n ropnode (hou.RopNode): The USD Rop node.\n processor (str): The output processor name. This is the basename of\n the python file that contains the Houdini USD Output Processor.\n\n \"\"\"\n import loputils\n\n parm = ropnode.parm(processor + \"_remove\")\n if not parm:\n raise RuntimeError(\n \"Output Processor %s does not \"\n \"exist on %s\" % (processor, ropnode.name())\n )\n\n loputils.handleOutputProcessorRemove({\"node\": ropnode, \"parm\": parm})\n\n\n@contextlib.contextmanager\ndef outputprocessors(ropnode, processors=tuple(), disable_all_others=True):\n \"\"\"Context manager to temporarily add Output Processors to USD ROP node.\n\n Args:\n ropnode (hou.RopNode): The USD Rop node.\n processors (tuple or list): The processors to add.\n disable_all_others (bool, Optional): Whether to disable all\n output processors currently on the ROP node that are not in the\n `processors` list passed to this function.\n\n \"\"\"\n # TODO: Add support for forcing the correct Order of the processors\n\n original = []\n prefix = \"enableoutputprocessor_\"\n processor_parms = ropnode.globParms(prefix + \"*\")\n for parm in processor_parms:\n original.append((parm, parm.eval()))\n\n if disable_all_others:\n for parm in processor_parms:\n parm.set(False)\n\n added = []\n for processor in processors:\n\n parm = ropnode.parm(prefix + processor)\n if parm:\n # If processor already exists, just enable it\n parm.set(True)\n\n else:\n # Else add the new processor\n add_usd_output_processor(ropnode, processor)\n added.append(processor)\n\n try:\n yield\n finally:\n\n # Remove newly added processors\n for processor in added:\n remove_usd_output_processor(ropnode, processor)\n\n # Revert to original values\n for parm, value in original:\n if parm:\n parm.set(value)\n\n\ndef get_usd_rop_loppath(node):\n\n # Get sop path\n node_type = node.type().name()\n if node_type == \"usd\":\n return node.parm(\"loppath\").evalAsNode()\n\n elif node_type in {\"usd_rop\", \"usdrender_rop\"}:\n # Inside Solaris e.g. /stage (not in ROP context)\n # When incoming connection is present it takes it directly\n inputs = node.inputs()\n if inputs:\n return inputs[0]\n else:\n return node.parm(\"loppath\").evalAsNode()\n\n\ndef get_layer_save_path(layer):\n \"\"\"Get custom HoudiniLayerInfo->HoudiniSavePath from SdfLayer.\n\n Args:\n layer (pxr.Sdf.Layer): The Layer to retrieve the save pah data from.\n\n Returns:\n str or None: Path to save to when data exists.\n\n \"\"\"\n hou_layer_info = layer.rootPrims.get(\"HoudiniLayerInfo\")\n if not hou_layer_info:\n return\n\n save_path = hou_layer_info.customData.get(\"HoudiniSavePath\", None)\n if save_path:\n # Unfortunately this doesn't actually resolve the full absolute path\n return layer.ComputeAbsolutePath(save_path)\n\n\ndef get_referenced_layers(layer):\n \"\"\"Return SdfLayers for all external references of the current layer\n\n Args:\n layer (pxr.Sdf.Layer): The Layer to retrieve the save pah data from.\n\n Returns:\n list: List of pxr.Sdf.Layer that are external references to this layer\n\n \"\"\"\n\n layers = []\n for layer_id in layer.GetExternalReferences():\n layer = Sdf.Layer.Find(layer_id)\n if not layer:\n # A file may not be in memory and is\n # referenced from disk. As such it cannot\n # be found. We will ignore those layers.\n continue\n\n layers.append(layer)\n\n return layers\n\n\ndef iter_layer_recursive(layer):\n \"\"\"Recursively iterate all 'external' referenced layers\"\"\"\n\n layers = get_referenced_layers(layer)\n traversed = set(layers) # Avoid recursion to itself (if even possible)\n traverse = list(layers)\n for layer in traverse:\n\n # Include children layers (recursion)\n children_layers = get_referenced_layers(layer)\n children_layers = [x for x in children_layers if x not in traversed]\n traverse.extend(children_layers)\n traversed.update(children_layers)\n\n yield layer\n\n\ndef get_configured_save_layers(usd_rop):\n\n lop_node = get_usd_rop_loppath(usd_rop)\n stage = lop_node.stage(apply_viewport_overrides=False)\n if not stage:\n raise RuntimeError(\n \"No valid USD stage for ROP node: \" \"%s\" % usd_rop.path()\n )\n\n root_layer = stage.GetRootLayer()\n\n save_layers = []\n for layer in iter_layer_recursive(root_layer):\n save_path = get_layer_save_path(layer)\n if save_path is not None:\n save_layers.append(layer)\n\n return save_layers\n"
},
{
"path": "openpype/hosts/houdini/hooks/set_paths.py",
"content": "from openpype.lib.applications import PreLaunchHook, LaunchTypes\n\n\nclass SetPath(PreLaunchHook):\n \"\"\"Set current dir to workdir.\n\n Hook `GlobalHostDataHook` must be executed before this hook.\n \"\"\"\n app_groups = {\"houdini\"}\n launch_types = {LaunchTypes.local}\n\n def execute(self):\n workdir = self.launch_context.env.get(\"AVALON_WORKDIR\", \"\")\n if not workdir:\n self.log.warning(\"BUG: Workdir is not filled.\")\n return\n\n self.launch_context.kwargs[\"cwd\"] = workdir\n"
},
{
"path": "openpype/hosts/houdini/plugins/create/convert_legacy.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Converter for legacy Houdini subsets.\"\"\"\nfrom openpype.pipeline.create.creator_plugins import SubsetConvertorPlugin\nfrom openpype.hosts.houdini.api.lib import imprint\n\n\nclass HoudiniLegacyConvertor(SubsetConvertorPlugin):\n \"\"\"Find and convert any legacy subsets in the scene.\n\n This Converter will find all legacy subsets in the scene and will\n transform them to the current system. Since the old subsets doesn't\n retain any information about their original creators, the only mapping\n we can do is based on their families.\n\n Its limitation is that you can have multiple creators creating subset\n of the same family and there is no way to handle it. This code should\n nevertheless cover all creators that came with OpenPype.\n\n \"\"\"\n identifier = \"io.openpype.creators.houdini.legacy\"\n family_to_id = {\n \"camera\": \"io.openpype.creators.houdini.camera\",\n \"ass\": \"io.openpype.creators.houdini.ass\",\n \"imagesequence\": \"io.openpype.creators.houdini.imagesequence\",\n \"hda\": \"io.openpype.creators.houdini.hda\",\n \"pointcache\": \"io.openpype.creators.houdini.pointcache\",\n \"redshiftproxy\": \"io.openpype.creators.houdini.redshiftproxy\",\n \"redshift_rop\": \"io.openpype.creators.houdini.redshift_rop\",\n \"usd\": \"io.openpype.creators.houdini.usd\",\n \"usdrender\": \"io.openpype.creators.houdini.usdrender\",\n \"vdbcache\": \"io.openpype.creators.houdini.vdbcache\"\n }\n\n def __init__(self, *args, **kwargs):\n super(HoudiniLegacyConvertor, self).__init__(*args, **kwargs)\n self.legacy_subsets = {}\n\n def find_instances(self):\n \"\"\"Find legacy subsets in the scene.\n\n Legacy subsets are the ones that doesn't have `creator_identifier`\n parameter on them.\n\n This is using cached entries done in\n :py:meth:`~HoudiniCreatorBase.cache_subsets()`\n\n \"\"\"\n self.legacy_subsets = self.collection_shared_data.get(\n \"houdini_cached_legacy_subsets\")\n if not self.legacy_subsets:\n return\n self.add_convertor_item(\"Found {} incompatible subset{}.\".format(\n len(self.legacy_subsets), \"s\" if len(self.legacy_subsets) > 1 else \"\")\n )\n\n def convert(self):\n \"\"\"Convert all legacy subsets to current.\n\n It is enough to add `creator_identifier` and `instance_node`.\n\n \"\"\"\n if not self.legacy_subsets:\n return\n\n for family, subsets in self.legacy_subsets.items():\n if family in self.family_to_id:\n for subset in subsets:\n data = {\n \"creator_identifier\": self.family_to_id[family],\n \"instance_node\": subset.path()\n }\n if family == \"pointcache\":\n data[\"families\"] = [\"abc\"]\n self.log.info(\"Converting {} to {}\".format(\n subset.path(), self.family_to_id[family]))\n imprint(subset, data)\n"
},
{
"path": "openpype/hosts/houdini/plugins/create/create_alembic_camera.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin for creating alembic camera subsets.\"\"\"\nfrom openpype.hosts.houdini.api import plugin\nfrom openpype.pipeline import CreatedInstance, CreatorError\n\nimport hou\n\n\nclass CreateAlembicCamera(plugin.HoudiniCreator):\n \"\"\"Single baked camera from Alembic ROP.\"\"\"\n\n identifier = \"io.openpype.creators.houdini.camera\"\n label = \"Camera (Abc)\"\n family = \"camera\"\n icon = \"camera\"\n\n def create(self, subset_name, instance_data, pre_create_data):\n import hou\n\n instance_data.pop(\"active\", None)\n instance_data.update({\"node_type\": \"alembic\"})\n\n instance = super(CreateAlembicCamera, self).create(\n subset_name,\n instance_data,\n pre_create_data) # type: CreatedInstance\n\n instance_node = hou.node(instance.get(\"instance_node\"))\n parms = {\n \"filename\": hou.text.expandString(\n \"$HIP/pyblish/{}.abc\".format(subset_name)),\n \"use_sop_path\": False,\n }\n\n if self.selected_nodes:\n if len(self.selected_nodes) > 1:\n raise CreatorError(\"More than one item selected.\")\n path = self.selected_nodes[0].path()\n # Split the node path into the first root and the remainder\n # So we can set the root and objects parameters correctly\n _, root, remainder = path.split(\"/\", 2)\n parms.update({\"root\": \"/\" + root, \"objects\": remainder})\n\n instance_node.setParms(parms)\n\n # Lock the Use Sop Path setting so the\n # user doesn't accidentally enable it.\n to_lock = [\"use_sop_path\"]\n self.lock_parameters(instance_node, to_lock)\n\n instance_node.parm(\"trange\").set(1)\n\n def get_network_categories(self):\n return [\n hou.ropNodeTypeCategory(),\n hou.objNodeTypeCategory()\n ]\n"
},
{
"path": "openpype/hosts/houdini/plugins/create/create_arnold_ass.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin for creating Arnold ASS files.\"\"\"\nfrom openpype.hosts.houdini.api import plugin\nfrom openpype.lib import BoolDef\n\n\nclass CreateArnoldAss(plugin.HoudiniCreator):\n \"\"\"Arnold .ass Archive\"\"\"\n\n identifier = \"io.openpype.creators.houdini.ass\"\n label = \"Arnold ASS\"\n family = \"ass\"\n icon = \"magic\"\n\n # Default extension: `.ass` or `.ass.gz`\n # however calling HoudiniCreator.create()\n # will override it by the value in the project settings\n ext = \".ass\"\n\n def create(self, subset_name, instance_data, pre_create_data):\n import hou\n\n instance_data.pop(\"active\", None)\n instance_data.update({\"node_type\": \"arnold\"})\n creator_attributes = instance_data.setdefault(\n \"creator_attributes\", dict())\n creator_attributes[\"farm\"] = pre_create_data[\"farm\"]\n\n instance = super(CreateArnoldAss, self).create(\n subset_name,\n instance_data,\n pre_create_data) # type: plugin.CreatedInstance\n\n instance_node = hou.node(instance.get(\"instance_node\"))\n\n # Hide Properties Tab on Arnold ROP since that's used\n # for rendering instead of .ass Archive Export\n parm_template_group = instance_node.parmTemplateGroup()\n parm_template_group.hideFolder(\"Properties\", True)\n instance_node.setParmTemplateGroup(parm_template_group)\n\n filepath = \"{}{}\".format(\n hou.text.expandString(\"$HIP/pyblish/\"),\n \"{}.$F4{}\".format(subset_name, self.ext)\n )\n parms = {\n # Render frame range\n \"trange\": 1,\n # Arnold ROP settings\n \"ar_ass_file\": filepath,\n \"ar_ass_export_enable\": 1\n }\n\n instance_node.setParms(parms)\n\n # Lock any parameters in this list\n to_lock = [\"ar_ass_export_enable\", \"family\", \"id\"]\n self.lock_parameters(instance_node, to_lock)\n\n def get_instance_attr_defs(self):\n return [\n BoolDef(\"farm\",\n label=\"Submitting to Farm\",\n default=False)\n ]\n\n def get_pre_create_attr_defs(self):\n attrs = super().get_pre_create_attr_defs()\n # Use same attributes as for instance attributes\n return attrs + self.get_instance_attr_defs()\n"
},
{
"path": "openpype/hosts/houdini/plugins/create/create_arnold_rop.py",
"content": "from openpype.hosts.houdini.api import plugin\nfrom openpype.lib import EnumDef, BoolDef\n\n\nclass CreateArnoldRop(plugin.HoudiniCreator):\n \"\"\"Arnold ROP\"\"\"\n\n identifier = \"io.openpype.creators.houdini.arnold_rop\"\n label = \"Arnold ROP\"\n family = \"arnold_rop\"\n icon = \"magic\"\n\n # Default extension\n ext = \"exr\"\n\n # Default to split export and render jobs\n export_job = True\n\n def create(self, subset_name, instance_data, pre_create_data):\n import hou\n\n # Remove the active, we are checking the bypass flag of the nodes\n instance_data.pop(\"active\", None)\n instance_data.update({\"node_type\": \"arnold\"})\n\n # Add chunk size attribute\n instance_data[\"chunkSize\"] = 1\n # Submit for job publishing\n instance_data[\"farm\"] = pre_create_data.get(\"farm\")\n\n instance = super(CreateArnoldRop, self).create(\n subset_name,\n instance_data,\n pre_create_data) # type: plugin.CreatedInstance\n\n instance_node = hou.node(instance.get(\"instance_node\"))\n\n ext = pre_create_data.get(\"image_format\")\n\n filepath = \"{renders_dir}{subset_name}/{subset_name}.$F4.{ext}\".format(\n renders_dir=hou.text.expandString(\"$HIP/pyblish/renders/\"),\n subset_name=subset_name,\n ext=ext,\n )\n parms = {\n # Render frame range\n \"trange\": 1,\n\n # Arnold ROP settings\n \"ar_picture\": filepath,\n \"ar_exr_half_precision\": 1 # half precision\n }\n\n if pre_create_data.get(\"export_job\"):\n ass_filepath = \\\n \"{export_dir}{subset_name}/{subset_name}.$F4.ass\".format(\n export_dir=hou.text.expandString(\"$HIP/pyblish/ass/\"),\n subset_name=subset_name,\n )\n parms[\"ar_ass_export_enable\"] = 1\n parms[\"ar_ass_file\"] = ass_filepath\n\n instance_node.setParms(parms)\n\n # Lock any parameters in this list\n to_lock = [\"family\", \"id\"]\n self.lock_parameters(instance_node, to_lock)\n\n def get_pre_create_attr_defs(self):\n attrs = super(CreateArnoldRop, self).get_pre_create_attr_defs()\n\n image_format_enum = [\n \"bmp\", \"cin\", \"exr\", \"jpg\", \"pic\", \"pic.gz\", \"png\",\n \"rad\", \"rat\", \"rta\", \"sgi\", \"tga\", \"tif\",\n ]\n\n return attrs + [\n BoolDef(\"farm\",\n label=\"Submitting to Farm\",\n default=True),\n BoolDef(\"export_job\",\n label=\"Split export and render jobs\",\n default=self.export_job),\n EnumDef(\"image_format\",\n image_format_enum,\n default=self.ext,\n label=\"Image Format Options\")\n ]\n"
},
{
"path": "openpype/hosts/houdini/plugins/create/create_bgeo.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin for creating pointcache bgeo files.\"\"\"\nfrom openpype.hosts.houdini.api import plugin\nfrom openpype.pipeline import CreatedInstance, CreatorError\nimport hou\nfrom openpype.lib import EnumDef, BoolDef\n\n\nclass CreateBGEO(plugin.HoudiniCreator):\n \"\"\"BGEO pointcache creator.\"\"\"\n identifier = \"io.openpype.creators.houdini.bgeo\"\n label = \"PointCache (Bgeo)\"\n family = \"pointcache\"\n icon = \"gears\"\n\n def create(self, subset_name, instance_data, pre_create_data):\n\n instance_data.pop(\"active\", None)\n\n instance_data.update({\"node_type\": \"geometry\"})\n creator_attributes = instance_data.setdefault(\n \"creator_attributes\", dict())\n creator_attributes[\"farm\"] = pre_create_data[\"farm\"]\n\n instance = super(CreateBGEO, self).create(\n subset_name,\n instance_data,\n pre_create_data) # type: CreatedInstance\n\n instance_node = hou.node(instance.get(\"instance_node\"))\n\n file_path = \"{}{}\".format(\n hou.text.expandString(\"$HIP/pyblish/\"),\n \"{}.$F4.{}\".format(\n subset_name,\n pre_create_data.get(\"bgeo_type\") or \"bgeo.sc\")\n )\n parms = {\n \"sopoutput\": file_path\n }\n\n instance_node.parm(\"trange\").set(1)\n if self.selected_nodes:\n # if selection is on SOP level, use it\n if isinstance(self.selected_nodes[0], hou.SopNode):\n parms[\"soppath\"] = self.selected_nodes[0].path()\n else:\n # try to find output node with the lowest index\n outputs = [\n child for child in self.selected_nodes[0].children()\n if child.type().name() == \"output\"\n ]\n if not outputs:\n instance_node.setParms(parms)\n raise CreatorError((\n \"Missing output node in SOP level for the selection. \"\n \"Please select correct SOP path in created instance.\"\n ))\n outputs.sort(key=lambda output: output.evalParm(\"outputidx\"))\n parms[\"soppath\"] = outputs[0].path()\n\n instance_node.setParms(parms)\n\n def get_instance_attr_defs(self):\n return [\n BoolDef(\"farm\",\n label=\"Submitting to Farm\",\n default=False)\n ]\n\n def get_pre_create_attr_defs(self):\n attrs = super().get_pre_create_attr_defs()\n bgeo_enum = [\n {\n \"value\": \"bgeo\",\n \"label\": \"uncompressed bgeo (.bgeo)\"\n },\n {\n \"value\": \"bgeosc\",\n \"label\": \"BLOSC compressed bgeo (.bgeosc)\"\n },\n {\n \"value\": \"bgeo.sc\",\n \"label\": \"BLOSC compressed bgeo (.bgeo.sc)\"\n },\n {\n \"value\": \"bgeo.gz\",\n \"label\": \"GZ compressed bgeo (.bgeo.gz)\"\n },\n {\n \"value\": \"bgeo.lzma\",\n \"label\": \"LZMA compressed bgeo (.bgeo.lzma)\"\n },\n {\n \"value\": \"bgeo.bz2\",\n \"label\": \"BZip2 compressed bgeo (.bgeo.bz2)\"\n }\n ]\n\n return attrs + [\n EnumDef(\"bgeo_type\", bgeo_enum, label=\"BGEO Options\"),\n ] + self.get_instance_attr_defs()\n\n def get_network_categories(self):\n return [\n hou.ropNodeTypeCategory(),\n hou.sopNodeTypeCategory()\n ]\n"
},
{
"path": "openpype/hosts/houdini/plugins/create/create_composite.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin for creating composite sequences.\"\"\"\nfrom openpype.hosts.houdini.api import plugin\nfrom openpype.pipeline import CreatedInstance, CreatorError\n\nimport hou\n\n\nclass CreateCompositeSequence(plugin.HoudiniCreator):\n \"\"\"Composite ROP to Image Sequence\"\"\"\n\n identifier = \"io.openpype.creators.houdini.imagesequence\"\n label = \"Composite (Image Sequence)\"\n family = \"imagesequence\"\n icon = \"gears\"\n\n ext = \".exr\"\n\n def create(self, subset_name, instance_data, pre_create_data):\n import hou # noqa\n\n instance_data.pop(\"active\", None)\n instance_data.update({\"node_type\": \"comp\"})\n\n instance = super(CreateCompositeSequence, self).create(\n subset_name,\n instance_data,\n pre_create_data) # type: CreatedInstance\n\n instance_node = hou.node(instance.get(\"instance_node\"))\n filepath = \"{}{}\".format(\n hou.text.expandString(\"$HIP/pyblish/\"),\n \"{}.$F4{}\".format(subset_name, self.ext)\n )\n parms = {\n \"trange\": 1,\n \"copoutput\": filepath\n }\n\n if self.selected_nodes:\n if len(self.selected_nodes) > 1:\n raise CreatorError(\"More than one item selected.\")\n path = self.selected_nodes[0].path()\n parms[\"coppath\"] = path\n\n instance_node.setParms(parms)\n\n # Manually set f1 & f2 to $FSTART and $FEND respectively\n # to match other Houdini nodes default.\n instance_node.parm(\"f1\").setExpression(\"$FSTART\")\n instance_node.parm(\"f2\").setExpression(\"$FEND\")\n\n # Lock any parameters in this list\n to_lock = [\"prim_to_detail_pattern\"]\n self.lock_parameters(instance_node, to_lock)\n\n def get_network_categories(self):\n return [\n hou.ropNodeTypeCategory(),\n hou.cop2NodeTypeCategory()\n ]\n"
},
{
"path": "openpype/hosts/houdini/plugins/create/create_hda.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin for creating publishable Houdini Digital Assets.\"\"\"\nfrom openpype.client import (\n get_asset_by_name,\n get_subsets,\n)\nfrom openpype.hosts.houdini.api import plugin\nimport hou\n\n\nclass CreateHDA(plugin.HoudiniCreator):\n \"\"\"Publish Houdini Digital Asset file.\"\"\"\n\n identifier = \"io.openpype.creators.houdini.hda\"\n label = \"Houdini Digital Asset (Hda)\"\n family = \"hda\"\n icon = \"gears\"\n maintain_selection = False\n\n def _check_existing(self, asset_name, subset_name):\n # type: (str) -> bool\n \"\"\"Check if existing subset name versions already exists.\"\"\"\n # Get all subsets of the current asset\n project_name = self.project_name\n asset_doc = get_asset_by_name(\n project_name, asset_name, fields=[\"_id\"]\n )\n subset_docs = get_subsets(\n project_name, asset_ids=[asset_doc[\"_id\"]], fields=[\"name\"]\n )\n existing_subset_names_low = {\n subset_doc[\"name\"].lower()\n for subset_doc in subset_docs\n }\n return subset_name.lower() in existing_subset_names_low\n\n def create_instance_node(\n self, asset_name, node_name, parent, node_type=\"geometry\"\n ):\n\n parent_node = hou.node(\"/obj\")\n if self.selected_nodes:\n # if we have `use selection` enabled, and we have some\n # selected nodes ...\n subnet = parent_node.collapseIntoSubnet(\n self.selected_nodes,\n subnet_name=\"{}_subnet\".format(node_name))\n subnet.moveToGoodPosition()\n to_hda = subnet\n else:\n to_hda = parent_node.createNode(\n \"subnet\", node_name=\"{}_subnet\".format(node_name))\n if not to_hda.type().definition():\n # if node type has not its definition, it is not user\n # created hda. We test if hda can be created from the node.\n if not to_hda.canCreateDigitalAsset():\n raise plugin.OpenPypeCreatorError(\n \"cannot create hda from node {}\".format(to_hda))\n\n hda_node = to_hda.createDigitalAsset(\n name=node_name,\n hda_file_name=\"$HIP/{}.hda\".format(node_name)\n )\n hda_node.layoutChildren()\n elif self._check_existing(asset_name, node_name):\n raise plugin.OpenPypeCreatorError(\n (\"subset {} is already published with different HDA\"\n \"definition.\").format(node_name))\n else:\n hda_node = to_hda\n\n hda_node.setName(node_name)\n self.customize_node_look(hda_node)\n return hda_node\n\n def create(self, subset_name, instance_data, pre_create_data):\n instance_data.pop(\"active\", None)\n\n instance = super(CreateHDA, self).create(\n subset_name,\n instance_data,\n pre_create_data) # type: plugin.CreatedInstance\n\n return instance\n\n def get_network_categories(self):\n return [\n hou.objNodeTypeCategory()\n ]\n"
},
{
"path": "openpype/hosts/houdini/plugins/create/create_karma_rop.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin to create Karma ROP.\"\"\"\nfrom openpype.hosts.houdini.api import plugin\nfrom openpype.pipeline import CreatedInstance\nfrom openpype.lib import BoolDef, EnumDef, NumberDef\n\n\nclass CreateKarmaROP(plugin.HoudiniCreator):\n \"\"\"Karma ROP\"\"\"\n identifier = \"io.openpype.creators.houdini.karma_rop\"\n label = \"Karma ROP\"\n family = \"karma_rop\"\n icon = \"magic\"\n\n def create(self, subset_name, instance_data, pre_create_data):\n import hou # noqa\n\n instance_data.pop(\"active\", None)\n instance_data.update({\"node_type\": \"karma\"})\n # Add chunk size attribute\n instance_data[\"chunkSize\"] = 10\n # Submit for job publishing\n instance_data[\"farm\"] = pre_create_data.get(\"farm\")\n\n instance = super(CreateKarmaROP, self).create(\n subset_name,\n instance_data,\n pre_create_data) # type: CreatedInstance\n\n instance_node = hou.node(instance.get(\"instance_node\"))\n\n ext = pre_create_data.get(\"image_format\")\n\n filepath = \"{renders_dir}{subset_name}/{subset_name}.$F4.{ext}\".format(\n renders_dir=hou.text.expandString(\"$HIP/pyblish/renders/\"),\n subset_name=subset_name,\n ext=ext,\n )\n checkpoint = \"{cp_dir}{subset_name}.$F4.checkpoint\".format(\n cp_dir=hou.text.expandString(\"$HIP/pyblish/\"),\n subset_name=subset_name\n )\n\n usd_directory = \"{usd_dir}{subset_name}_$RENDERID\".format(\n usd_dir=hou.text.expandString(\"$HIP/pyblish/renders/usd_renders/\"), # noqa\n subset_name=subset_name\n )\n\n parms = {\n # Render Frame Range\n \"trange\": 1,\n # Karma ROP Setting\n \"picture\": filepath,\n # Karma Checkpoint Setting\n \"productName\": checkpoint,\n # USD Output Directory\n \"savetodirectory\": usd_directory,\n }\n\n res_x = pre_create_data.get(\"res_x\")\n res_y = pre_create_data.get(\"res_y\")\n\n if self.selected_nodes:\n # If camera found in selection\n # we will use as render camera\n camera = None\n for node in self.selected_nodes:\n if node.type().name() == \"cam\":\n camera = node.path()\n has_camera = pre_create_data.get(\"cam_res\")\n if has_camera:\n res_x = node.evalParm(\"resx\")\n res_y = node.evalParm(\"resy\")\n\n if not camera:\n self.log.warning(\"No render camera found in selection\")\n\n parms.update({\n \"camera\": camera or \"\",\n \"resolutionx\": res_x,\n \"resolutiony\": res_y,\n })\n\n instance_node.setParms(parms)\n\n # Lock some Avalon attributes\n to_lock = [\"family\", \"id\"]\n self.lock_parameters(instance_node, to_lock)\n\n def get_pre_create_attr_defs(self):\n attrs = super(CreateKarmaROP, self).get_pre_create_attr_defs()\n\n image_format_enum = [\n \"bmp\", \"cin\", \"exr\", \"jpg\", \"pic\", \"pic.gz\", \"png\",\n \"rad\", \"rat\", \"rta\", \"sgi\", \"tga\", \"tif\",\n ]\n\n return attrs + [\n BoolDef(\"farm\",\n label=\"Submitting to Farm\",\n default=True),\n EnumDef(\"image_format\",\n image_format_enum,\n default=\"exr\",\n label=\"Image Format Options\"),\n NumberDef(\"res_x\",\n label=\"width\",\n default=1920,\n decimals=0),\n NumberDef(\"res_y\",\n label=\"height\",\n default=720,\n decimals=0),\n BoolDef(\"cam_res\",\n label=\"Camera Resolution\",\n default=False)\n ]\n"
},
{
"path": "openpype/hosts/houdini/plugins/create/create_mantra_ifd.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin for creating pointcache alembics.\"\"\"\nfrom openpype.hosts.houdini.api import plugin\nfrom openpype.pipeline import CreatedInstance\nfrom openpype.lib import BoolDef\n\n\nclass CreateMantraIFD(plugin.HoudiniCreator):\n \"\"\"Mantra .ifd Archive\"\"\"\n identifier = \"io.openpype.creators.houdini.mantraifd\"\n label = \"Mantra IFD\"\n family = \"mantraifd\"\n icon = \"gears\"\n\n def create(self, subset_name, instance_data, pre_create_data):\n import hou\n instance_data.pop(\"active\", None)\n instance_data.update({\"node_type\": \"ifd\"})\n creator_attributes = instance_data.setdefault(\n \"creator_attributes\", dict())\n creator_attributes[\"farm\"] = pre_create_data[\"farm\"]\n instance = super(CreateMantraIFD, self).create(\n subset_name,\n instance_data,\n pre_create_data) # type: CreatedInstance\n\n instance_node = hou.node(instance.get(\"instance_node\"))\n\n filepath = \"{}{}\".format(\n hou.text.expandString(\"$HIP/pyblish/\"),\n \"{}.$F4.ifd\".format(subset_name))\n parms = {\n # Render frame range\n \"trange\": 1,\n # Arnold ROP settings\n \"soho_diskfile\": filepath,\n \"soho_outputmode\": 1\n }\n\n instance_node.setParms(parms)\n\n # Lock any parameters in this list\n to_lock = [\"soho_outputmode\", \"family\", \"id\"]\n self.lock_parameters(instance_node, to_lock)\n\n def get_instance_attr_defs(self):\n return [\n BoolDef(\"farm\",\n label=\"Submitting to Farm\",\n default=False)\n ]\n\n def get_pre_create_attr_defs(self):\n attrs = super().get_pre_create_attr_defs()\n # Use same attributes as for instance attributes\n return attrs + self.get_instance_attr_defs()\n"
},
{
"path": "openpype/hosts/houdini/plugins/create/create_mantra_rop.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin to create Mantra ROP.\"\"\"\nfrom openpype.hosts.houdini.api import plugin\nfrom openpype.pipeline import CreatedInstance\nfrom openpype.lib import EnumDef, BoolDef\n\n\nclass CreateMantraROP(plugin.HoudiniCreator):\n \"\"\"Mantra ROP\"\"\"\n identifier = \"io.openpype.creators.houdini.mantra_rop\"\n label = \"Mantra ROP\"\n family = \"mantra_rop\"\n icon = \"magic\"\n\n # Default to split export and render jobs\n export_job = True\n\n def create(self, subset_name, instance_data, pre_create_data):\n import hou # noqa\n\n instance_data.pop(\"active\", None)\n instance_data.update({\"node_type\": \"ifd\"})\n # Add chunk size attribute\n instance_data[\"chunkSize\"] = 10\n # Submit for job publishing\n instance_data[\"farm\"] = pre_create_data.get(\"farm\")\n\n instance = super(CreateMantraROP, self).create(\n subset_name,\n instance_data,\n pre_create_data) # type: CreatedInstance\n\n instance_node = hou.node(instance.get(\"instance_node\"))\n\n ext = pre_create_data.get(\"image_format\")\n\n filepath = \"{renders_dir}{subset_name}/{subset_name}.$F4.{ext}\".format(\n renders_dir=hou.text.expandString(\"$HIP/pyblish/renders/\"),\n subset_name=subset_name,\n ext=ext,\n )\n\n parms = {\n # Render Frame Range\n \"trange\": 1,\n # Mantra ROP Setting\n \"vm_picture\": filepath,\n }\n\n if pre_create_data.get(\"export_job\"):\n ifd_filepath = \\\n \"{export_dir}{subset_name}/{subset_name}.$F4.ifd\".format(\n export_dir=hou.text.expandString(\"$HIP/pyblish/ifd/\"),\n subset_name=subset_name,\n )\n parms[\"soho_outputmode\"] = 1\n parms[\"soho_diskfile\"] = ifd_filepath\n\n if self.selected_nodes:\n # If camera found in selection\n # we will use as render camera\n camera = None\n for node in self.selected_nodes:\n if node.type().name() == \"cam\":\n camera = node.path()\n\n if not camera:\n self.log.warning(\"No render camera found in selection\")\n\n parms.update({\"camera\": camera or \"\"})\n\n custom_res = pre_create_data.get(\"override_resolution\")\n if custom_res:\n parms.update({\"override_camerares\": 1})\n instance_node.setParms(parms)\n\n # Lock some Avalon attributes\n to_lock = [\"family\", \"id\"]\n self.lock_parameters(instance_node, to_lock)\n\n def get_pre_create_attr_defs(self):\n attrs = super(CreateMantraROP, self).get_pre_create_attr_defs()\n\n image_format_enum = [\n \"bmp\", \"cin\", \"exr\", \"jpg\", \"pic\", \"pic.gz\", \"png\",\n \"rad\", \"rat\", \"rta\", \"sgi\", \"tga\", \"tif\",\n ]\n\n return attrs + [\n BoolDef(\"farm\",\n label=\"Submitting to Farm\",\n default=True),\n BoolDef(\"export_job\",\n label=\"Split export and render jobs\",\n default=self.export_job),\n EnumDef(\"image_format\",\n image_format_enum,\n default=\"exr\",\n label=\"Image Format Options\"),\n BoolDef(\"override_resolution\",\n label=\"Override Camera Resolution\",\n tooltip=\"Override the current camera \"\n \"resolution, recommended for IPR.\",\n default=False)\n ]\n"
},
{
"path": "openpype/hosts/houdini/plugins/create/create_pointcache.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin for creating pointcache alembics.\"\"\"\nfrom openpype.hosts.houdini.api import plugin\nfrom openpype.lib import BoolDef\n\nimport hou\n\n\n\nclass CreatePointCache(plugin.HoudiniCreator):\n \"\"\"Alembic ROP to pointcache\"\"\"\n identifier = \"io.openpype.creators.houdini.pointcache\"\n label = \"PointCache (Abc)\"\n family = \"pointcache\"\n icon = \"gears\"\n\n def create(self, subset_name, instance_data, pre_create_data):\n instance_data.pop(\"active\", None)\n instance_data.update({\"node_type\": \"alembic\"})\n creator_attributes = instance_data.setdefault(\n \"creator_attributes\", dict())\n creator_attributes[\"farm\"] = pre_create_data[\"farm\"]\n\n instance = super(CreatePointCache, self).create(\n subset_name,\n instance_data,\n pre_create_data)\n\n instance_node = hou.node(instance.get(\"instance_node\"))\n parms = {\n \"use_sop_path\": True,\n \"build_from_path\": True,\n \"path_attrib\": \"path\",\n \"prim_to_detail_pattern\": \"cbId\",\n \"format\": 2,\n \"facesets\": 0,\n \"filename\": hou.text.expandString(\n \"$HIP/pyblish/{}.abc\".format(subset_name))\n }\n\n if self.selected_nodes:\n selected_node = self.selected_nodes[0]\n\n # Although Houdini allows ObjNode path on `sop_path` for the\n # the ROP node we prefer it set to the SopNode path explicitly\n\n # Allow sop level paths (e.g. /obj/geo1/box1)\n if isinstance(selected_node, hou.SopNode):\n parms[\"sop_path\"] = selected_node.path()\n self.log.debug(\n \"Valid SopNode selection, 'SOP Path' in ROP will be set to '%s'.\"\n % selected_node.path()\n )\n\n # Allow object level paths to Geometry nodes (e.g. /obj/geo1)\n # but do not allow other object level nodes types like cameras, etc.\n elif isinstance(selected_node, hou.ObjNode) and \\\n selected_node.type().name() in [\"geo\"]:\n\n # get the output node with the minimum\n # 'outputidx' or the node with display flag\n sop_path = self.get_obj_output(selected_node)\n\n if sop_path:\n parms[\"sop_path\"] = sop_path.path()\n self.log.debug(\n \"Valid ObjNode selection, 'SOP Path' in ROP will be set to \"\n \"the child path '%s'.\"\n % sop_path.path()\n )\n\n if not parms.get(\"sop_path\", None):\n self.log.debug(\n \"Selection isn't valid. 'SOP Path' in ROP will be empty.\"\n )\n else:\n self.log.debug(\n \"No Selection. 'SOP Path' in ROP will be empty.\"\n )\n\n instance_node.setParms(parms)\n instance_node.parm(\"trange\").set(1)\n\n # Lock any parameters in this list\n to_lock = [\"prim_to_detail_pattern\"]\n self.lock_parameters(instance_node, to_lock)\n\n def get_network_categories(self):\n return [\n hou.ropNodeTypeCategory(),\n hou.sopNodeTypeCategory()\n ]\n\n def get_obj_output(self, obj_node):\n \"\"\"Find output node with the smallest 'outputidx'.\"\"\"\n\n outputs = obj_node.subnetOutputs()\n\n # if obj_node is empty\n if not outputs:\n return\n\n # if obj_node has one output child whether its\n # sop output node or a node with the render flag\n elif len(outputs) == 1:\n return outputs[0]\n\n # if there are more than one, then it have multiple ouput nodes\n # return the one with the minimum 'outputidx'\n else:\n return min(outputs,\n key=lambda node: node.evalParm('outputidx'))\n\n def get_instance_attr_defs(self):\n return [\n BoolDef(\"farm\",\n label=\"Submitting to Farm\",\n default=False)\n ]\n\n def get_pre_create_attr_defs(self):\n attrs = super().get_pre_create_attr_defs()\n # Use same attributes as for instance attributes\n return attrs + self.get_instance_attr_defs()\n"
},
{
"path": "openpype/hosts/houdini/plugins/create/create_redshift_proxy.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin for creating Redshift proxies.\"\"\"\nfrom openpype.hosts.houdini.api import plugin\nimport hou\nfrom openpype.lib import BoolDef\n\n\nclass CreateRedshiftProxy(plugin.HoudiniCreator):\n \"\"\"Redshift Proxy\"\"\"\n identifier = \"io.openpype.creators.houdini.redshiftproxy\"\n label = \"Redshift Proxy\"\n family = \"redshiftproxy\"\n icon = \"magic\"\n\n def create(self, subset_name, instance_data, pre_create_data):\n\n # Remove the active, we are checking the bypass flag of the nodes\n instance_data.pop(\"active\", None)\n\n # Redshift provides a `Redshift_Proxy_Output` node type which shows\n # a limited set of parameters by default and is set to extract a\n # Redshift Proxy. However when \"imprinting\" extra parameters needed\n # for OpenPype it starts showing all its parameters again. It's unclear\n # why this happens.\n # TODO: Somehow enforce so that it only shows the original limited\n # attributes of the Redshift_Proxy_Output node type\n instance_data.update({\"node_type\": \"Redshift_Proxy_Output\"})\n creator_attributes = instance_data.setdefault(\n \"creator_attributes\", dict())\n creator_attributes[\"farm\"] = pre_create_data[\"farm\"]\n\n instance = super(CreateRedshiftProxy, self).create(\n subset_name,\n instance_data,\n pre_create_data)\n\n instance_node = hou.node(instance.get(\"instance_node\"))\n\n parms = {\n \"RS_archive_file\": '$HIP/pyblish/{}.$F4.rs'.format(subset_name),\n }\n\n if self.selected_nodes:\n parms[\"RS_archive_sopPath\"] = self.selected_nodes[0].path()\n\n instance_node.setParms(parms)\n\n # Lock some Avalon attributes\n to_lock = [\"family\", \"id\", \"prim_to_detail_pattern\"]\n self.lock_parameters(instance_node, to_lock)\n\n def get_network_categories(self):\n return [\n hou.ropNodeTypeCategory(),\n hou.sopNodeTypeCategory()\n ]\n\n def get_instance_attr_defs(self):\n return [\n BoolDef(\"farm\",\n label=\"Submitting to Farm\",\n default=False)\n ]\n\n def get_pre_create_attr_defs(self):\n attrs = super().get_pre_create_attr_defs()\n # Use same attributes as for instance attributes\n return attrs + self.get_instance_attr_defs()\n"
},
{
"path": "openpype/hosts/houdini/plugins/create/create_redshift_rop.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin to create Redshift ROP.\"\"\"\nimport hou # noqa\n\nfrom openpype.hosts.houdini.api import plugin\nfrom openpype.lib import EnumDef, BoolDef\n\n\nclass CreateRedshiftROP(plugin.HoudiniCreator):\n \"\"\"Redshift ROP\"\"\"\n\n identifier = \"io.openpype.creators.houdini.redshift_rop\"\n label = \"Redshift ROP\"\n family = \"redshift_rop\"\n icon = \"magic\"\n ext = \"exr\"\n\n # Default to split export and render jobs\n split_render = True\n\n def create(self, subset_name, instance_data, pre_create_data):\n\n instance_data.pop(\"active\", None)\n instance_data.update({\"node_type\": \"Redshift_ROP\"})\n # Add chunk size attribute\n instance_data[\"chunkSize\"] = 10\n # Submit for job publishing\n instance_data[\"farm\"] = pre_create_data.get(\"farm\")\n\n instance = super(CreateRedshiftROP, self).create(\n subset_name,\n instance_data,\n pre_create_data)\n\n instance_node = hou.node(instance.get(\"instance_node\"))\n\n basename = instance_node.name()\n\n # Also create the linked Redshift IPR Rop\n try:\n ipr_rop = instance_node.parent().createNode(\n \"Redshift_IPR\", node_name=f\"{basename}_IPR\"\n )\n except hou.OperationFailed as e:\n raise plugin.OpenPypeCreatorError(\n (\n \"Cannot create Redshift node. Is Redshift \"\n \"installed and enabled?\"\n )\n ) from e\n\n # Move it to directly under the Redshift ROP\n ipr_rop.setPosition(instance_node.position() + hou.Vector2(0, -1))\n\n # Set the linked rop to the Redshift ROP\n ipr_rop.parm(\"linked_rop\").set(instance_node.path())\n\n ext = pre_create_data.get(\"image_format\")\n filepath = \"{renders_dir}{subset_name}/{subset_name}.{fmt}\".format(\n renders_dir=hou.text.expandString(\"$HIP/pyblish/renders/\"),\n subset_name=subset_name,\n fmt=\"${aov}.$F4.{ext}\".format(aov=\"AOV\", ext=ext)\n )\n\n ext_format_index = {\"exr\": 0, \"tif\": 1, \"jpg\": 2, \"png\": 3}\n\n parms = {\n # Render frame range\n \"trange\": 1,\n # Redshift ROP settings\n \"RS_outputFileNamePrefix\": filepath,\n \"RS_outputMultilayerMode\": \"1\", # no multi-layered exr\n \"RS_outputBeautyAOVSuffix\": \"beauty\",\n \"RS_outputFileFormat\": ext_format_index[ext],\n }\n\n if self.selected_nodes:\n # set up the render camera from the selected node\n camera = None\n for node in self.selected_nodes:\n if node.type().name() == \"cam\":\n camera = node.path()\n parms[\"RS_renderCamera\"] = camera or \"\"\n\n export_dir = hou.text.expandString(\"$HIP/pyblish/rs/\")\n rs_filepath = f\"{export_dir}{subset_name}/{subset_name}.$F4.rs\"\n parms[\"RS_archive_file\"] = rs_filepath\n\n if pre_create_data.get(\"split_render\", self.split_render):\n parms[\"RS_archive_enable\"] = 1\n\n instance_node.setParms(parms)\n\n # Lock some Avalon attributes\n to_lock = [\"family\", \"id\"]\n self.lock_parameters(instance_node, to_lock)\n\n def remove_instances(self, instances):\n for instance in instances:\n node = instance.data.get(\"instance_node\")\n\n ipr_node = hou.node(f\"{node}_IPR\")\n if ipr_node:\n ipr_node.destroy()\n\n return super(CreateRedshiftROP, self).remove_instances(instances)\n\n def get_pre_create_attr_defs(self):\n attrs = super(CreateRedshiftROP, self).get_pre_create_attr_defs()\n image_format_enum = [\n \"exr\", \"tif\", \"jpg\", \"png\",\n ]\n\n return attrs + [\n BoolDef(\"farm\",\n label=\"Submitting to Farm\",\n default=True),\n BoolDef(\"split_render\",\n label=\"Split export and render jobs\",\n default=self.split_render),\n EnumDef(\"image_format\",\n image_format_enum,\n default=self.ext,\n label=\"Image Format Options\")\n ]\n"
},
{
"path": "openpype/hosts/houdini/plugins/create/create_review.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin for creating openGL reviews.\"\"\"\nfrom openpype.hosts.houdini.api import plugin\nfrom openpype.lib import EnumDef, BoolDef, NumberDef\n\nimport os\nimport hou\n\n\nclass CreateReview(plugin.HoudiniCreator):\n \"\"\"Review with OpenGL ROP\"\"\"\n\n identifier = \"io.openpype.creators.houdini.review\"\n label = \"Review\"\n family = \"review\"\n icon = \"video-camera\"\n\n def create(self, subset_name, instance_data, pre_create_data):\n\n instance_data.pop(\"active\", None)\n instance_data.update({\"node_type\": \"opengl\"})\n instance_data[\"imageFormat\"] = pre_create_data.get(\"imageFormat\")\n instance_data[\"keepImages\"] = pre_create_data.get(\"keepImages\")\n\n instance = super(CreateReview, self).create(\n subset_name,\n instance_data,\n pre_create_data)\n\n instance_node = hou.node(instance.get(\"instance_node\"))\n\n frame_range = hou.playbar.frameRange()\n\n filepath = \"{root}/{subset}/{subset}.$F4.{ext}\".format(\n root=hou.text.expandString(\"$HIP/pyblish\"),\n subset=\"`chs(\\\"subset\\\")`\", # keep dynamic link to subset\n ext=pre_create_data.get(\"image_format\") or \"png\"\n )\n\n parms = {\n \"picture\": filepath,\n\n \"trange\": 1,\n\n # Unlike many other ROP nodes the opengl node does not default\n # to expression of $FSTART and $FEND so we preserve that behavior\n # but do set the range to the frame range of the playbar\n \"f1\": frame_range[0],\n \"f2\": frame_range[1],\n }\n\n override_resolution = pre_create_data.get(\"override_resolution\")\n if override_resolution:\n parms.update({\n \"tres\": override_resolution,\n \"res1\": pre_create_data.get(\"resx\"),\n \"res2\": pre_create_data.get(\"resy\"),\n \"aspect\": pre_create_data.get(\"aspect\"),\n })\n\n if self.selected_nodes:\n # The first camera found in selection we will use as camera\n # Other node types we set in force objects\n camera = None\n force_objects = []\n for node in self.selected_nodes:\n path = node.path()\n if node.type().name() == \"cam\":\n if camera:\n continue\n camera = path\n else:\n force_objects.append(path)\n\n if not camera:\n self.log.warning(\"No camera found in selection.\")\n\n parms.update({\n \"camera\": camera or \"\",\n \"scenepath\": \"/obj\",\n \"forceobjects\": \" \".join(force_objects),\n \"vobjects\": \"\" # clear candidate objects from '*' value\n })\n\n instance_node.setParms(parms)\n\n # Set OCIO Colorspace to the default output colorspace\n # if there's OCIO\n if os.getenv(\"OCIO\"):\n self.set_colorcorrect_to_default_view_space(instance_node)\n\n to_lock = [\"id\", \"family\"]\n\n self.lock_parameters(instance_node, to_lock)\n\n def get_pre_create_attr_defs(self):\n attrs = super(CreateReview, self).get_pre_create_attr_defs()\n\n image_format_enum = [\n \"bmp\", \"cin\", \"exr\", \"jpg\", \"pic\", \"pic.gz\", \"png\",\n \"rad\", \"rat\", \"rta\", \"sgi\", \"tga\", \"tif\",\n ]\n\n return attrs + [\n BoolDef(\"keepImages\",\n label=\"Keep Image Sequences\",\n default=False),\n EnumDef(\"imageFormat\",\n image_format_enum,\n default=\"png\",\n label=\"Image Format Options\"),\n BoolDef(\"override_resolution\",\n label=\"Override resolution\",\n tooltip=\"When disabled the resolution set on the camera \"\n \"is used instead.\",\n default=True),\n NumberDef(\"resx\",\n label=\"Resolution Width\",\n default=1280,\n minimum=2,\n decimals=0),\n NumberDef(\"resy\",\n label=\"Resolution Height\",\n default=720,\n minimum=2,\n decimals=0),\n NumberDef(\"aspect\",\n label=\"Aspect Ratio\",\n default=1.0,\n minimum=0.0001,\n decimals=3)\n ]\n\n def set_colorcorrect_to_default_view_space(self,\n instance_node):\n \"\"\"Set ociocolorspace to the default output space.\"\"\"\n from openpype.hosts.houdini.api.colorspace import get_default_display_view_colorspace # noqa\n\n # set Color Correction parameter to OpenColorIO\n instance_node.setParms({\"colorcorrect\": 2})\n\n # Get default view space for ociocolorspace parm.\n default_view_space = get_default_display_view_colorspace()\n instance_node.setParms(\n {\"ociocolorspace\": default_view_space}\n )\n\n self.log.debug(\n \"'OCIO Colorspace' parm on '{}' has been set to \"\n \"the default view color space '{}'\"\n .format(instance_node, default_view_space)\n )\n"
},
{
"path": "openpype/hosts/houdini/plugins/create/create_staticmesh.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator for Unreal Static Meshes.\"\"\"\nfrom openpype.hosts.houdini.api import plugin\nfrom openpype.lib import BoolDef, EnumDef\n\nimport hou\n\n\nclass CreateStaticMesh(plugin.HoudiniCreator):\n \"\"\"Static Meshes as FBX. \"\"\"\n\n identifier = \"io.openpype.creators.houdini.staticmesh.fbx\"\n label = \"Static Mesh (FBX)\"\n family = \"staticMesh\"\n icon = \"fa5s.cubes\"\n\n default_variants = [\"Main\"]\n\n def create(self, subset_name, instance_data, pre_create_data):\n\n instance_data.update({\"node_type\": \"filmboxfbx\"})\n\n instance = super(CreateStaticMesh, self).create(\n subset_name,\n instance_data,\n pre_create_data)\n\n # get the created rop node\n instance_node = hou.node(instance.get(\"instance_node\"))\n\n # prepare parms\n output_path = hou.text.expandString(\n \"$HIP/pyblish/{}.fbx\".format(subset_name)\n )\n\n parms = {\n \"startnode\": self.get_selection(),\n \"sopoutput\": output_path,\n # vertex cache format\n \"vcformat\": pre_create_data.get(\"vcformat\"),\n \"convertunits\": pre_create_data.get(\"convertunits\"),\n # set render range to use frame range start-end frame\n \"trange\": 1,\n \"createsubnetroot\": pre_create_data.get(\"createsubnetroot\")\n }\n\n # set parms\n instance_node.setParms(parms)\n\n # Lock any parameters in this list\n to_lock = [\"family\", \"id\"]\n self.lock_parameters(instance_node, to_lock)\n\n def get_network_categories(self):\n return [\n hou.ropNodeTypeCategory(),\n hou.objNodeTypeCategory(),\n hou.sopNodeTypeCategory()\n ]\n\n def get_pre_create_attr_defs(self):\n \"\"\"Add settings for users. \"\"\"\n\n attrs = super(CreateStaticMesh, self).get_pre_create_attr_defs()\n createsubnetroot = BoolDef(\"createsubnetroot\",\n tooltip=\"Create an extra root for the \"\n \"Export node when it's a \"\n \"subnetwork. This causes the \"\n \"exporting subnetwork node to be \"\n \"represented in the FBX file.\",\n default=False,\n label=\"Create Root for Subnet\")\n vcformat = EnumDef(\"vcformat\",\n items={\n 0: \"Maya Compatible (MC)\",\n 1: \"3DS MAX Compatible (PC2)\"\n },\n default=0,\n label=\"Vertex Cache Format\")\n convert_units = BoolDef(\"convertunits\",\n tooltip=\"When on, the FBX is converted\"\n \"from the current Houdini \"\n \"system units to the native \"\n \"FBX unit of centimeters.\",\n default=False,\n label=\"Convert Units\")\n\n return attrs + [createsubnetroot, vcformat, convert_units]\n\n def get_dynamic_data(\n self, variant, task_name, asset_doc, project_name, host_name, instance\n ):\n \"\"\"\n The default subset name templates for Unreal include {asset} and thus\n we should pass that along as dynamic data.\n \"\"\"\n dynamic_data = super(CreateStaticMesh, self).get_dynamic_data(\n variant, task_name, asset_doc, project_name, host_name, instance\n )\n dynamic_data[\"asset\"] = asset_doc[\"name\"]\n return dynamic_data\n\n def get_selection(self):\n \"\"\"Selection Logic.\n\n how self.selected_nodes should be processed to get\n the desirable node from selection.\n\n Returns:\n str : node path\n \"\"\"\n\n selection = \"\"\n\n if self.selected_nodes:\n selected_node = self.selected_nodes[0]\n\n # Accept sop level nodes (e.g. /obj/geo1/box1)\n if isinstance(selected_node, hou.SopNode):\n selection = selected_node.path()\n self.log.debug(\n \"Valid SopNode selection, 'Export' in filmboxfbx\"\n \" will be set to '%s'.\", selected_node\n )\n\n # Accept object level nodes (e.g. /obj/geo1)\n elif isinstance(selected_node, hou.ObjNode):\n selection = selected_node.path()\n self.log.debug(\n \"Valid ObjNode selection, 'Export' in filmboxfbx \"\n \"will be set to the child path '%s'.\", selection\n )\n\n else:\n self.log.debug(\n \"Selection isn't valid. 'Export' in \"\n \"filmboxfbx will be empty.\"\n )\n else:\n self.log.debug(\n \"No Selection. 'Export' in filmboxfbx will be empty.\"\n )\n\n return selection\n"
},
{
"path": "openpype/hosts/houdini/plugins/create/create_usd.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin for creating USDs.\"\"\"\nfrom openpype.hosts.houdini.api import plugin\nfrom openpype.pipeline import CreatedInstance\n\nimport hou\n\n\nclass CreateUSD(plugin.HoudiniCreator):\n \"\"\"Universal Scene Description\"\"\"\n identifier = \"io.openpype.creators.houdini.usd\"\n label = \"USD (experimental)\"\n family = \"usd\"\n icon = \"gears\"\n enabled = False\n\n def create(self, subset_name, instance_data, pre_create_data):\n\n instance_data.pop(\"active\", None)\n instance_data.update({\"node_type\": \"usd\"})\n\n instance = super(CreateUSD, self).create(\n subset_name,\n instance_data,\n pre_create_data) # type: CreatedInstance\n\n instance_node = hou.node(instance.get(\"instance_node\"))\n\n parms = {\n \"lopoutput\": \"$HIP/pyblish/{}.usd\".format(subset_name),\n \"enableoutputprocessor_simplerelativepaths\": False,\n }\n\n if self.selected_nodes:\n parms[\"loppath\"] = self.selected_nodes[0].path()\n\n instance_node.setParms(parms)\n\n # Lock any parameters in this list\n to_lock = [\n \"fileperframe\",\n # Lock some Avalon attributes\n \"family\",\n \"id\",\n ]\n self.lock_parameters(instance_node, to_lock)\n\n def get_network_categories(self):\n return [\n hou.ropNodeTypeCategory(),\n hou.lopNodeTypeCategory()\n ]\n"
},
{
"path": "openpype/hosts/houdini/plugins/create/create_usdrender.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin for creating USD renders.\"\"\"\nfrom openpype.hosts.houdini.api import plugin\nfrom openpype.pipeline import CreatedInstance\n\n\nclass CreateUSDRender(plugin.HoudiniCreator):\n \"\"\"USD Render ROP in /stage\"\"\"\n identifier = \"io.openpype.creators.houdini.usdrender\"\n label = \"USD Render (experimental)\"\n family = \"usdrender\"\n icon = \"magic\"\n\n def create(self, subset_name, instance_data, pre_create_data):\n import hou # noqa\n\n instance_data[\"parent\"] = hou.node(\"/stage\")\n\n # Remove the active, we are checking the bypass flag of the nodes\n instance_data.pop(\"active\", None)\n instance_data.update({\"node_type\": \"usdrender\"})\n\n instance = super(CreateUSDRender, self).create(\n subset_name,\n instance_data,\n pre_create_data) # type: CreatedInstance\n\n instance_node = hou.node(instance.get(\"instance_node\"))\n\n\n parms = {\n # Render frame range\n \"trange\": 1\n }\n if self.selected_nodes:\n parms[\"loppath\"] = self.selected_nodes[0].path()\n instance_node.setParms(parms)\n\n # Lock some Avalon attributes\n to_lock = [\"family\", \"id\"]\n self.lock_parameters(instance_node, to_lock)\n"
},
{
"path": "openpype/hosts/houdini/plugins/create/create_vbd_cache.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin for creating VDB Caches.\"\"\"\nfrom openpype.hosts.houdini.api import plugin\nfrom openpype.pipeline import CreatedInstance\nfrom openpype.lib import BoolDef\n\nimport hou\n\n\nclass CreateVDBCache(plugin.HoudiniCreator):\n \"\"\"OpenVDB from Geometry ROP\"\"\"\n identifier = \"io.openpype.creators.houdini.vdbcache\"\n name = \"vbdcache\"\n label = \"VDB Cache\"\n family = \"vdbcache\"\n icon = \"cloud\"\n\n def create(self, subset_name, instance_data, pre_create_data):\n import hou\n\n instance_data.pop(\"active\", None)\n instance_data.update({\"node_type\": \"geometry\"})\n creator_attributes = instance_data.setdefault(\n \"creator_attributes\", dict())\n creator_attributes[\"farm\"] = pre_create_data[\"farm\"]\n instance = super(CreateVDBCache, self).create(\n subset_name,\n instance_data,\n pre_create_data) # type: CreatedInstance\n\n instance_node = hou.node(instance.get(\"instance_node\"))\n file_path = \"{}{}\".format(\n hou.text.expandString(\"$HIP/pyblish/\"),\n \"{}.$F4.vdb\".format(subset_name))\n parms = {\n \"sopoutput\": file_path,\n \"initsim\": True,\n \"trange\": 1\n }\n\n if self.selected_nodes:\n parms[\"soppath\"] = self.get_sop_node_path(self.selected_nodes[0])\n\n instance_node.setParms(parms)\n\n def get_network_categories(self):\n return [\n hou.ropNodeTypeCategory(),\n hou.objNodeTypeCategory(),\n hou.sopNodeTypeCategory()\n ]\n\n def get_sop_node_path(self, selected_node):\n \"\"\"Get Sop Path of the selected node.\n\n Although Houdini allows ObjNode path on `sop_path` for the\n the ROP node, we prefer it set to the SopNode path explicitly.\n \"\"\"\n\n # Allow sop level paths (e.g. /obj/geo1/box1)\n if isinstance(selected_node, hou.SopNode):\n self.log.debug(\n \"Valid SopNode selection, 'SOP Path' in ROP will\"\n \" be set to '%s'.\", selected_node.path()\n )\n return selected_node.path()\n\n # Allow object level paths to Geometry nodes (e.g. /obj/geo1)\n # but do not allow other object level nodes types like cameras, etc.\n elif isinstance(selected_node, hou.ObjNode) and \\\n selected_node.type().name() == \"geo\":\n\n # Try to find output node.\n sop_node = self.get_obj_output(selected_node)\n if sop_node:\n self.log.debug(\n \"Valid ObjNode selection, 'SOP Path' in ROP will \"\n \"be set to the child path '%s'.\", sop_node.path()\n )\n return sop_node.path()\n\n self.log.debug(\n \"Selection isn't valid. 'SOP Path' in ROP will be empty.\"\n )\n return \"\"\n\n def get_obj_output(self, obj_node):\n \"\"\"Try to find output node.\n\n If any output nodes are present, return the output node with\n the minimum 'outputidx'\n If no output nodes are present, return the node with display flag\n If no nodes are present at all, return None\n \"\"\"\n\n outputs = obj_node.subnetOutputs()\n\n # if obj_node is empty\n if not outputs:\n return\n\n # if obj_node has one output child whether its\n # sop output node or a node with the render flag\n elif len(outputs) == 1:\n return outputs[0]\n\n # if there are more than one, then it has multiple output nodes\n # return the one with the minimum 'outputidx'\n else:\n return min(outputs,\n key=lambda node: node.evalParm('outputidx'))\n\n def get_instance_attr_defs(self):\n return [\n BoolDef(\"farm\",\n label=\"Submitting to Farm\",\n default=False)\n ]\n\n def get_pre_create_attr_defs(self):\n attrs = super().get_pre_create_attr_defs()\n # Use same attributes as for instance attributes\n return attrs + self.get_instance_attr_defs()\n"
},
{
"path": "openpype/hosts/houdini/plugins/create/create_vray_rop.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin to create VRay ROP.\"\"\"\nimport hou\n\nfrom openpype.hosts.houdini.api import plugin\nfrom openpype.pipeline import CreatedInstance\nfrom openpype.lib import EnumDef, BoolDef\n\n\nclass CreateVrayROP(plugin.HoudiniCreator):\n \"\"\"VRay ROP\"\"\"\n\n identifier = \"io.openpype.creators.houdini.vray_rop\"\n label = \"VRay ROP\"\n family = \"vray_rop\"\n icon = \"magic\"\n ext = \"exr\"\n\n # Default to split export and render jobs\n export_job = True\n\n def create(self, subset_name, instance_data, pre_create_data):\n\n instance_data.pop(\"active\", None)\n instance_data.update({\"node_type\": \"vray_renderer\"})\n # Add chunk size attribute\n instance_data[\"chunkSize\"] = 10\n # Submit for job publishing\n instance_data[\"farm\"] = pre_create_data.get(\"farm\")\n\n instance = super(CreateVrayROP, self).create(\n subset_name,\n instance_data,\n pre_create_data) # type: CreatedInstance\n\n instance_node = hou.node(instance.get(\"instance_node\"))\n\n # Add IPR for Vray\n basename = instance_node.name()\n try:\n ipr_rop = instance_node.parent().createNode(\n \"vray\", node_name=basename + \"_IPR\"\n )\n except hou.OperationFailed:\n raise plugin.OpenPypeCreatorError(\n \"Cannot create Vray render node. \"\n \"Make sure Vray installed and enabled!\"\n )\n\n ipr_rop.setPosition(instance_node.position() + hou.Vector2(0, -1))\n ipr_rop.parm(\"rop\").set(instance_node.path())\n\n parms = {\n \"trange\": 1,\n \"SettingsEXR_bits_per_channel\": \"16\" # half precision\n }\n\n if pre_create_data.get(\"export_job\"):\n scene_filepath = \\\n \"{export_dir}{subset_name}/{subset_name}.$F4.vrscene\".format(\n export_dir=hou.text.expandString(\"$HIP/pyblish/vrscene/\"),\n subset_name=subset_name,\n )\n # Setting render_export_mode to \"2\" because that's for\n # \"Export only\" (\"1\" is for \"Export & Render\")\n parms[\"render_export_mode\"] = \"2\"\n parms[\"render_export_filepath\"] = scene_filepath\n\n if self.selected_nodes:\n # set up the render camera from the selected node\n camera = None\n for node in self.selected_nodes:\n if node.type().name() == \"cam\":\n camera = node.path()\n parms.update({\n \"render_camera\": camera or \"\"\n })\n\n # Enable render element\n ext = pre_create_data.get(\"image_format\")\n instance_data[\"RenderElement\"] = pre_create_data.get(\"render_element_enabled\") # noqa\n if pre_create_data.get(\"render_element_enabled\", True):\n # Vray has its own tag for AOV file output\n filepath = \"{renders_dir}{subset_name}/{subset_name}.{fmt}\".format(\n renders_dir=hou.text.expandString(\"$HIP/pyblish/renders/\"),\n subset_name=subset_name,\n fmt=\"${aov}.$F4.{ext}\".format(aov=\"AOV\",\n ext=ext)\n )\n filepath = \"{}{}\".format(\n hou.text.expandString(\"$HIP/pyblish/renders/\"),\n \"{}/{}.${}.$F4.{}\".format(subset_name,\n subset_name,\n \"AOV\",\n ext)\n )\n re_rop = instance_node.parent().createNode(\n \"vray_render_channels\",\n node_name=basename + \"_render_element\"\n )\n # move the render element node next to the vray renderer node\n re_rop.setPosition(instance_node.position() + hou.Vector2(0, 1))\n re_path = re_rop.path()\n parms.update({\n \"use_render_channels\": 1,\n \"SettingsOutput_img_file_path\": filepath,\n \"render_network_render_channels\": re_path\n })\n\n else:\n filepath = \"{renders_dir}{subset_name}/{subset_name}.{fmt}\".format(\n renders_dir=hou.text.expandString(\"$HIP/pyblish/renders/\"),\n subset_name=subset_name,\n fmt=\"$F4.{ext}\".format(ext=ext)\n )\n parms.update({\n \"use_render_channels\": 0,\n \"SettingsOutput_img_file_path\": filepath\n })\n\n custom_res = pre_create_data.get(\"override_resolution\")\n if custom_res:\n parms.update({\"override_camerares\": 1})\n\n instance_node.setParms(parms)\n\n # lock parameters from AVALON\n to_lock = [\"family\", \"id\"]\n self.lock_parameters(instance_node, to_lock)\n\n def remove_instances(self, instances):\n for instance in instances:\n node = instance.data.get(\"instance_node\")\n # for the extra render node from the plugins\n # such as vray and redshift\n ipr_node = hou.node(\"{}{}\".format(node, \"_IPR\"))\n if ipr_node:\n ipr_node.destroy()\n re_node = hou.node(\"{}{}\".format(node,\n \"_render_element\"))\n if re_node:\n re_node.destroy()\n\n return super(CreateVrayROP, self).remove_instances(instances)\n\n def get_pre_create_attr_defs(self):\n attrs = super(CreateVrayROP, self).get_pre_create_attr_defs()\n image_format_enum = [\n \"bmp\", \"cin\", \"exr\", \"jpg\", \"pic\", \"pic.gz\", \"png\",\n \"rad\", \"rat\", \"rta\", \"sgi\", \"tga\", \"tif\",\n ]\n\n return attrs + [\n BoolDef(\"farm\",\n label=\"Submitting to Farm\",\n default=True),\n BoolDef(\"export_job\",\n label=\"Split export and render jobs\",\n default=self.export_job),\n EnumDef(\"image_format\",\n image_format_enum,\n default=self.ext,\n label=\"Image Format Options\"),\n BoolDef(\"override_resolution\",\n label=\"Override Camera Resolution\",\n tooltip=\"Override the current camera \"\n \"resolution, recommended for IPR.\",\n default=False),\n BoolDef(\"render_element_enabled\",\n label=\"Render Element\",\n tooltip=\"Create Render Element Node \"\n \"if enabled\",\n default=False)\n ]\n"
},
{
"path": "openpype/hosts/houdini/plugins/create/create_workfile.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin for creating workfiles.\"\"\"\nfrom openpype import AYON_SERVER_ENABLED\nfrom openpype.hosts.houdini.api import plugin\nfrom openpype.hosts.houdini.api.lib import read, imprint\nfrom openpype.hosts.houdini.api.pipeline import CONTEXT_CONTAINER\nfrom openpype.pipeline import CreatedInstance, AutoCreator\nfrom openpype.client import get_asset_by_name\nimport hou\n\n\nclass CreateWorkfile(plugin.HoudiniCreatorBase, AutoCreator):\n \"\"\"Workfile auto-creator.\"\"\"\n identifier = \"io.openpype.creators.houdini.workfile\"\n label = \"Workfile\"\n family = \"workfile\"\n icon = \"fa5.file\"\n\n default_variant = \"Main\"\n\n def create(self):\n variant = self.default_variant\n current_instance = next(\n (\n instance for instance in self.create_context.instances\n if instance.creator_identifier == self.identifier\n ), None)\n\n project_name = self.project_name\n asset_name = self.create_context.get_current_asset_name()\n task_name = self.create_context.get_current_task_name()\n host_name = self.host_name\n\n if current_instance is None:\n current_instance_asset = None\n elif AYON_SERVER_ENABLED:\n current_instance_asset = current_instance[\"folderPath\"]\n else:\n current_instance_asset = current_instance[\"asset\"]\n\n if current_instance is None:\n asset_doc = get_asset_by_name(project_name, asset_name)\n subset_name = self.get_subset_name(\n variant, task_name, asset_doc, project_name, host_name\n )\n data = {\n \"task\": task_name,\n \"variant\": variant\n }\n if AYON_SERVER_ENABLED:\n data[\"folderPath\"] = asset_name\n else:\n data[\"asset\"] = asset_name\n\n data.update(\n self.get_dynamic_data(\n variant, task_name, asset_doc,\n project_name, host_name, current_instance)\n )\n self.log.info(\"Auto-creating workfile instance...\")\n current_instance = CreatedInstance(\n self.family, subset_name, data, self\n )\n self._add_instance_to_context(current_instance)\n elif (\n current_instance_asset != asset_name\n or current_instance[\"task\"] != task_name\n ):\n # Update instance context if is not the same\n asset_doc = get_asset_by_name(project_name, asset_name)\n subset_name = self.get_subset_name(\n variant, task_name, asset_doc, project_name, host_name\n )\n if AYON_SERVER_ENABLED:\n current_instance[\"folderPath\"] = asset_name\n else:\n current_instance[\"asset\"] = asset_name\n current_instance[\"task\"] = task_name\n current_instance[\"subset\"] = subset_name\n\n # write workfile information to context container.\n op_ctx = hou.node(CONTEXT_CONTAINER)\n if not op_ctx:\n op_ctx = self.create_context_node()\n\n workfile_data = {\"workfile\": current_instance.data_to_store()}\n imprint(op_ctx, workfile_data)\n\n def collect_instances(self):\n op_ctx = hou.node(CONTEXT_CONTAINER)\n instance = read(op_ctx)\n if not instance:\n return\n workfile = instance.get(\"workfile\")\n if not workfile:\n return\n created_instance = CreatedInstance.from_existing(\n workfile, self\n )\n self._add_instance_to_context(created_instance)\n\n def update_instances(self, update_list):\n op_ctx = hou.node(CONTEXT_CONTAINER)\n for created_inst, _changes in update_list:\n if created_inst[\"creator_identifier\"] == self.identifier:\n workfile_data = {\"workfile\": created_inst.data_to_store()}\n imprint(op_ctx, workfile_data, update=True)\n"
},
{
"path": "openpype/hosts/houdini/plugins/inventory/set_camera_resolution.py",
"content": "from openpype.pipeline import InventoryAction\nfrom openpype.hosts.houdini.api.lib import (\n get_camera_from_container,\n set_camera_resolution\n)\nfrom openpype.pipeline.context_tools import get_current_project_asset\n\n\nclass SetCameraResolution(InventoryAction):\n\n label = \"Set Camera Resolution\"\n icon = \"desktop\"\n color = \"orange\"\n\n @staticmethod\n def is_compatible(container):\n return (\n container.get(\"loader\") == \"CameraLoader\"\n )\n\n def process(self, containers):\n asset_doc = get_current_project_asset()\n for container in containers:\n node = container[\"node\"]\n camera = get_camera_from_container(node)\n set_camera_resolution(camera, asset_doc)\n"
},
{
"path": "openpype/hosts/houdini/plugins/load/actions.py",
"content": "\"\"\"A module containing generic loader actions that will display in the Loader.\n\n\"\"\"\n\nfrom openpype.pipeline import load\n\n\nclass SetFrameRangeLoader(load.LoaderPlugin):\n \"\"\"Set frame range excluding pre- and post-handles\"\"\"\n\n families = [\n \"animation\",\n \"camera\",\n \"pointcache\",\n \"vdbcache\",\n \"usd\",\n ]\n representations = [\"abc\", \"vdb\", \"usd\"]\n\n label = \"Set frame range\"\n order = 11\n icon = \"clock-o\"\n color = \"white\"\n\n def load(self, context, name, namespace, data):\n\n import hou\n\n version = context[\"version\"]\n version_data = version.get(\"data\", {})\n\n start = version_data.get(\"frameStart\", None)\n end = version_data.get(\"frameEnd\", None)\n\n if start is None or end is None:\n print(\n \"Skipping setting frame range because start or \"\n \"end frame data is missing..\"\n )\n return\n\n hou.playbar.setFrameRange(start, end)\n hou.playbar.setPlaybackRange(start, end)\n\n\nclass SetFrameRangeWithHandlesLoader(load.LoaderPlugin):\n \"\"\"Set frame range including pre- and post-handles\"\"\"\n\n families = [\n \"animation\",\n \"camera\",\n \"pointcache\",\n \"vdbcache\",\n \"usd\",\n ]\n representations = [\"abc\", \"vdb\", \"usd\"]\n\n label = \"Set frame range (with handles)\"\n order = 12\n icon = \"clock-o\"\n color = \"white\"\n\n def load(self, context, name, namespace, data):\n\n import hou\n\n version = context[\"version\"]\n version_data = version.get(\"data\", {})\n\n start = version_data.get(\"frameStart\", None)\n end = version_data.get(\"frameEnd\", None)\n\n if start is None or end is None:\n print(\n \"Skipping setting frame range because start or \"\n \"end frame data is missing..\"\n )\n return\n\n # Include handles\n start -= version_data.get(\"handleStart\", 0)\n end += version_data.get(\"handleEnd\", 0)\n\n hou.playbar.setFrameRange(start, end)\n hou.playbar.setPlaybackRange(start, end)\n"
},
{
"path": "openpype/hosts/houdini/plugins/load/load_alembic.py",
"content": "import os\nfrom openpype.pipeline import (\n load,\n get_representation_path,\n)\nfrom openpype.hosts.houdini.api import pipeline\n\n\nclass AbcLoader(load.LoaderPlugin):\n \"\"\"Load Alembic\"\"\"\n\n families = [\"model\", \"animation\", \"pointcache\", \"gpuCache\"]\n label = \"Load Alembic\"\n representations = [\"abc\"]\n order = -10\n icon = \"code-fork\"\n color = \"orange\"\n\n def load(self, context, name=None, namespace=None, data=None):\n import hou\n\n # Format file name, Houdini only wants forward slashes\n file_path = self.filepath_from_context(context)\n file_path = os.path.normpath(file_path)\n file_path = file_path.replace(\"\\\\\", \"/\")\n\n # Get the root node\n obj = hou.node(\"/obj\")\n\n # Define node name\n namespace = namespace if namespace else context[\"asset\"][\"name\"]\n node_name = \"{}_{}\".format(namespace, name) if namespace else name\n\n # Create a new geo node\n container = obj.createNode(\"geo\", node_name=node_name)\n\n # Remove the file node, it only loads static meshes\n # Houdini 17 has removed the file node from the geo node\n file_node = container.node(\"file1\")\n if file_node:\n file_node.destroy()\n\n # Create an alembic node (supports animation)\n alembic = container.createNode(\"alembic\", node_name=node_name)\n alembic.setParms({\"fileName\": file_path})\n\n # Add unpack node\n unpack_name = \"unpack_{}\".format(name)\n unpack = container.createNode(\"unpack\", node_name=unpack_name)\n unpack.setInput(0, alembic)\n unpack.setParms({\"transfer_attributes\": \"path\"})\n\n # Add normal to points\n # Order of menu ['point', 'vertex', 'prim', 'detail']\n normal_name = \"normal_{}\".format(name)\n normal_node = container.createNode(\"normal\", node_name=normal_name)\n normal_node.setParms({\"type\": 0})\n\n normal_node.setInput(0, unpack)\n\n null = container.createNode(\"null\", node_name=\"OUT\".format(name))\n null.setInput(0, normal_node)\n\n # Ensure display flag is on the Alembic input node and not on the OUT\n # node to optimize \"debug\" displaying in the viewport.\n alembic.setDisplayFlag(True)\n\n # Set new position for unpack node else it gets cluttered\n nodes = [container, alembic, unpack, normal_node, null]\n for nr, node in enumerate(nodes):\n node.setPosition([0, (0 - nr)])\n\n self[:] = nodes\n\n return pipeline.containerise(\n node_name,\n namespace,\n nodes,\n context,\n self.__class__.__name__,\n suffix=\"\",\n )\n\n def update(self, container, representation):\n\n node = container[\"node\"]\n try:\n alembic_node = next(\n n for n in node.children() if n.type().name() == \"alembic\"\n )\n except StopIteration:\n self.log.error(\"Could not find node of type `alembic`\")\n return\n\n # Update the file path\n file_path = get_representation_path(representation)\n file_path = file_path.replace(\"\\\\\", \"/\")\n\n alembic_node.setParms({\"fileName\": file_path})\n\n # Update attribute\n node.setParms({\"representation\": str(representation[\"_id\"])})\n\n def remove(self, container):\n\n node = container[\"node\"]\n node.destroy()\n\n def switch(self, container, representation):\n self.update(container, representation)\n"
},
{
"path": "openpype/hosts/houdini/plugins/load/load_alembic_archive.py",
"content": "import os\nfrom openpype.pipeline import (\n load,\n get_representation_path,\n)\nfrom openpype.hosts.houdini.api import pipeline\n\n\nclass AbcArchiveLoader(load.LoaderPlugin):\n \"\"\"Load Alembic as full geometry network hierarchy \"\"\"\n\n families = [\"model\", \"animation\", \"pointcache\", \"gpuCache\"]\n label = \"Load Alembic as Archive\"\n representations = [\"abc\"]\n order = -5\n icon = \"code-fork\"\n color = \"orange\"\n\n def load(self, context, name=None, namespace=None, data=None):\n\n import hou\n\n # Format file name, Houdini only wants forward slashes\n file_path = self.filepath_from_context(context)\n file_path = os.path.normpath(file_path)\n file_path = file_path.replace(\"\\\\\", \"/\")\n\n # Get the root node\n obj = hou.node(\"/obj\")\n\n # Define node name\n namespace = namespace if namespace else context[\"asset\"][\"name\"]\n node_name = \"{}_{}\".format(namespace, name) if namespace else name\n\n # Create an Alembic archive node\n node = obj.createNode(\"alembicarchive\", node_name=node_name)\n node.moveToGoodPosition()\n\n # TODO: add FPS of project / asset\n node.setParms({\"fileName\": file_path,\n \"channelRef\": True})\n\n # Apply some magic\n node.parm(\"buildHierarchy\").pressButton()\n node.moveToGoodPosition()\n\n nodes = [node]\n\n self[:] = nodes\n\n return pipeline.containerise(node_name,\n namespace,\n nodes,\n context,\n self.__class__.__name__,\n suffix=\"\")\n\n def update(self, container, representation):\n\n node = container[\"node\"]\n\n # Update the file path\n file_path = get_representation_path(representation)\n file_path = file_path.replace(\"\\\\\", \"/\")\n\n # Update attributes\n node.setParms({\"fileName\": file_path,\n \"representation\": str(representation[\"_id\"])})\n\n # Rebuild\n node.parm(\"buildHierarchy\").pressButton()\n\n def remove(self, container):\n\n node = container[\"node\"]\n node.destroy()\n\n def switch(self, container, representation):\n self.update(container, representation)\n"
},
{
"path": "openpype/hosts/houdini/plugins/load/load_ass.py",
"content": "import os\nimport re\n\nfrom openpype.pipeline import (\n load,\n get_representation_path,\n)\nfrom openpype.hosts.houdini.api import pipeline\n\n\nclass AssLoader(load.LoaderPlugin):\n \"\"\"Load .ass with Arnold Procedural\"\"\"\n\n families = [\"ass\"]\n label = \"Load Arnold Procedural\"\n representations = [\"ass\"]\n order = -10\n icon = \"code-fork\"\n color = \"orange\"\n\n def load(self, context, name=None, namespace=None, data=None):\n import hou\n\n # Get the root node\n obj = hou.node(\"/obj\")\n\n # Define node name\n namespace = namespace if namespace else context[\"asset\"][\"name\"]\n node_name = \"{}_{}\".format(namespace, name) if namespace else name\n\n # Create a new geo node\n procedural = obj.createNode(\"arnold::procedural\", node_name=node_name)\n\n procedural.setParms(\n {\n \"ar_filename\": self.format_path(context[\"representation\"])\n })\n\n nodes = [procedural]\n self[:] = nodes\n\n return pipeline.containerise(\n node_name,\n namespace,\n nodes,\n context,\n self.__class__.__name__,\n suffix=\"\",\n )\n\n def update(self, container, representation):\n # Update the file path\n procedural = container[\"node\"]\n procedural.setParms({\"ar_filename\": self.format_path(representation)})\n\n # Update attribute\n procedural.setParms({\"representation\": str(representation[\"_id\"])})\n\n def remove(self, container):\n node = container[\"node\"]\n node.destroy()\n\n @staticmethod\n def format_path(representation):\n \"\"\"Format file path correctly for single ass.* or ass.* sequence.\n\n Args:\n representation (dict): representation to be loaded.\n\n Returns:\n str: Formatted path to be used by the input node.\n\n \"\"\"\n path = get_representation_path(representation)\n if not os.path.exists(path):\n raise RuntimeError(\"Path does not exist: {}\".format(path))\n\n is_sequence = bool(representation[\"context\"].get(\"frame\"))\n # The path is either a single file or sequence in a folder.\n if is_sequence:\n dir_path, file_name = os.path.split(path)\n path = os.path.join(\n dir_path,\n re.sub(r\"(.*)\\.(\\d+)\\.(ass.*)\", \"\\\\1.$F4.\\\\3\", file_name)\n )\n\n return os.path.normpath(path).replace(\"\\\\\", \"/\")\n\n def switch(self, container, representation):\n self.update(container, representation)\n"
},
{
"path": "openpype/hosts/houdini/plugins/load/load_bgeo.py",
"content": "# -*- coding: utf-8 -*-\nimport os\nimport re\n\nfrom openpype.pipeline import (\n load,\n get_representation_path,\n)\nfrom openpype.hosts.houdini.api import pipeline\n\n\nclass BgeoLoader(load.LoaderPlugin):\n \"\"\"Load bgeo files to Houdini.\"\"\"\n\n label = \"Load bgeo\"\n families = [\"model\", \"pointcache\", \"bgeo\"]\n representations = [\n \"bgeo\", \"bgeosc\", \"bgeogz\",\n \"bgeo.sc\", \"bgeo.gz\", \"bgeo.lzma\", \"bgeo.bz2\"]\n order = -10\n icon = \"code-fork\"\n color = \"orange\"\n\n def load(self, context, name=None, namespace=None, data=None):\n\n import hou\n\n # Get the root node\n obj = hou.node(\"/obj\")\n\n # Define node name\n namespace = namespace if namespace else context[\"asset\"][\"name\"]\n node_name = \"{}_{}\".format(namespace, name) if namespace else name\n\n # Create a new geo node\n container = obj.createNode(\"geo\", node_name=node_name)\n\n # Remove the file node, it only loads static meshes\n # Houdini 17 has removed the file node from the geo node\n file_node = container.node(\"file1\")\n if file_node:\n file_node.destroy()\n\n # Explicitly create a file node\n path = self.filepath_from_context(context)\n file_node = container.createNode(\"file\", node_name=node_name)\n file_node.setParms(\n {\"file\": self.format_path(path, context[\"representation\"])})\n\n # Set display on last node\n file_node.setDisplayFlag(True)\n\n nodes = [container, file_node]\n self[:] = nodes\n\n return pipeline.containerise(\n node_name,\n namespace,\n nodes,\n context,\n self.__class__.__name__,\n suffix=\"\",\n )\n\n @staticmethod\n def format_path(path, representation):\n \"\"\"Format file path correctly for single bgeo or bgeo sequence.\"\"\"\n if not os.path.exists(path):\n raise RuntimeError(\"Path does not exist: %s\" % path)\n\n is_sequence = bool(representation[\"context\"].get(\"frame\"))\n # The path is either a single file or sequence in a folder.\n if not is_sequence:\n filename = path\n else:\n filename = re.sub(r\"(.*)\\.(\\d+)\\.(bgeo.*)\", \"\\\\1.$F4.\\\\3\", path)\n\n filename = os.path.join(path, filename)\n\n filename = os.path.normpath(filename)\n filename = filename.replace(\"\\\\\", \"/\")\n\n return filename\n\n def update(self, container, representation):\n\n node = container[\"node\"]\n try:\n file_node = next(\n n for n in node.children() if n.type().name() == \"file\"\n )\n except StopIteration:\n self.log.error(\"Could not find node of type `alembic`\")\n return\n\n # Update the file path\n file_path = get_representation_path(representation)\n file_path = self.format_path(file_path, representation)\n\n file_node.setParms({\"file\": file_path})\n\n # Update attribute\n node.setParms({\"representation\": str(representation[\"_id\"])})\n\n def remove(self, container):\n\n node = container[\"node\"]\n node.destroy()\n\n def switch(self, container, representation):\n self.update(container, representation)\n"
},
{
"path": "openpype/hosts/houdini/plugins/load/load_camera.py",
"content": "from openpype.pipeline import (\n load,\n get_representation_path,\n)\nfrom openpype.hosts.houdini.api import pipeline\n\nfrom openpype.hosts.houdini.api.lib import (\n set_camera_resolution,\n get_camera_from_container\n)\n\nimport hou\n\n\nARCHIVE_EXPRESSION = ('__import__(\"_alembic_hom_extensions\")'\n '.alembicGetCameraDict')\n\n\ndef transfer_non_default_values(src, dest, ignore=None):\n \"\"\"Copy parm from src to dest.\n\n Because the Alembic Archive rebuilds the entire node\n hierarchy on triggering \"Build Hierarchy\" we want to\n preserve any local tweaks made by the user on the camera\n for ease of use. That could be a background image, a\n resolution change or even Redshift camera parameters.\n\n We try to do so by finding all Parms that exist on both\n source and destination node, include only those that both\n are not at their default value, they must be visible,\n we exclude those that have the special \"alembic archive\"\n channel expression and ignore certain Parm types.\n\n \"\"\"\n\n ignore_types = {\n hou.parmTemplateType.Toggle,\n hou.parmTemplateType.Menu,\n hou.parmTemplateType.Button,\n hou.parmTemplateType.FolderSet,\n hou.parmTemplateType.Separator,\n hou.parmTemplateType.Label,\n }\n\n src.updateParmStates()\n\n for parm in src.allParms():\n\n if ignore and parm.name() in ignore:\n continue\n\n # If destination parm does not exist, ignore..\n dest_parm = dest.parm(parm.name())\n if not dest_parm:\n continue\n\n # Ignore values that are currently at default\n if parm.isAtDefault() and dest_parm.isAtDefault():\n continue\n\n if not parm.isVisible():\n # Ignore hidden parameters, assume they\n # are implementation details\n continue\n\n expression = None\n try:\n expression = parm.expression()\n except hou.OperationFailed:\n # No expression present\n pass\n\n if expression is not None and ARCHIVE_EXPRESSION in expression:\n # Assume it's part of the automated connections that the\n # Alembic Archive makes on loading of the camera and thus we do\n # not want to transfer the expression\n continue\n\n # Ignore folders, separators, etc.\n if parm.parmTemplate().type() in ignore_types:\n continue\n\n print(\"Preserving attribute: %s\" % parm.name())\n dest_parm.setFromParm(parm)\n\n\nclass CameraLoader(load.LoaderPlugin):\n \"\"\"Load camera from an Alembic file\"\"\"\n\n families = [\"camera\"]\n label = \"Load Camera (abc)\"\n representations = [\"abc\"]\n order = -10\n\n icon = \"code-fork\"\n color = \"orange\"\n\n def load(self, context, name=None, namespace=None, data=None):\n\n # Format file name, Houdini only wants forward slashes\n file_path = self.filepath_from_context(context).replace(\"\\\\\", \"/\")\n\n # Get the root node\n obj = hou.node(\"/obj\")\n\n # Define node name\n namespace = namespace if namespace else context[\"asset\"][\"name\"]\n node_name = \"{}_{}\".format(namespace, name) if namespace else name\n\n # Create a archive node\n node = self.create_and_connect(obj, \"alembicarchive\", node_name)\n\n # TODO: add FPS of project / asset\n node.setParms({\"fileName\": file_path, \"channelRef\": True})\n\n # Apply some magic\n node.parm(\"buildHierarchy\").pressButton()\n node.moveToGoodPosition()\n\n # Create an alembic xform node\n nodes = [node]\n\n camera = get_camera_from_container(node)\n self._match_maya_render_mask(camera)\n set_camera_resolution(camera, asset_doc=context[\"asset\"])\n self[:] = nodes\n\n return pipeline.containerise(node_name,\n namespace,\n nodes,\n context,\n self.__class__.__name__,\n suffix=\"\")\n\n def update(self, container, representation):\n\n node = container[\"node\"]\n\n # Update the file path\n file_path = get_representation_path(representation)\n file_path = file_path.replace(\"\\\\\", \"/\")\n\n # Update attributes\n node.setParms({\"fileName\": file_path,\n \"representation\": str(representation[\"_id\"])})\n\n # Store the cam temporarily next to the Alembic Archive\n # so that we can preserve parm values the user set on it\n # after build hierarchy was triggered.\n old_camera = get_camera_from_container(node)\n temp_camera = old_camera.copyTo(node.parent())\n\n # Rebuild\n node.parm(\"buildHierarchy\").pressButton()\n\n # Apply values to the new camera\n new_camera = get_camera_from_container(node)\n transfer_non_default_values(temp_camera,\n new_camera,\n # The hidden uniform scale attribute\n # gets a default connection to\n # \"icon_scale\" just skip that completely\n ignore={\"scale\"})\n\n self._match_maya_render_mask(new_camera)\n set_camera_resolution(new_camera)\n\n temp_camera.destroy()\n\n def remove(self, container):\n\n node = container[\"node\"]\n node.destroy()\n\n def create_and_connect(self, node, node_type, name=None):\n \"\"\"Create a node within a node which and connect it to the input\n\n Args:\n node(hou.Node): parent of the new node\n node_type(str) name of the type of node, eg: 'alembic'\n name(str, Optional): name of the node\n\n Returns:\n hou.Node\n\n \"\"\"\n if name:\n new_node = node.createNode(node_type, node_name=name)\n else:\n new_node = node.createNode(node_type)\n\n new_node.moveToGoodPosition()\n return new_node\n\n def _match_maya_render_mask(self, camera):\n \"\"\"Workaround to match Maya render mask in Houdini\"\"\"\n\n # print(\"Setting match maya render mask \")\n parm = camera.parm(\"aperture\")\n expression = parm.expression()\n expression = expression.replace(\"return \", \"aperture = \")\n expression += \"\"\"\n# Match maya render mask (logic from Houdini's own FBX importer)\nnode = hou.pwd()\nresx = node.evalParm('resx')\nresy = node.evalParm('resy')\naspect = node.evalParm('aspect')\naperture *= min(1, (resx / resy * aspect) / 1.5)\nreturn aperture\n\"\"\"\n parm.setExpression(expression, language=hou.exprLanguage.Python)\n"
},
{
"path": "openpype/hosts/houdini/plugins/load/load_fbx.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Fbx Loader for houdini. \"\"\"\nfrom openpype.pipeline import (\n load,\n get_representation_path,\n)\nfrom openpype.hosts.houdini.api import pipeline\n\n\nclass FbxLoader(load.LoaderPlugin):\n \"\"\"Load fbx files. \"\"\"\n\n label = \"Load FBX\"\n icon = \"code-fork\"\n color = \"orange\"\n\n order = -10\n\n families = [\"*\"]\n representations = [\"*\"]\n extensions = {\"fbx\"}\n\n def load(self, context, name=None, namespace=None, data=None):\n\n # get file path from context\n file_path = self.filepath_from_context(context)\n file_path = file_path.replace(\"\\\\\", \"/\")\n\n # get necessary data\n namespace, node_name = self.get_node_name(context, name, namespace)\n\n # create load tree\n nodes = self.create_load_node_tree(file_path, node_name, name)\n\n self[:] = nodes\n\n # Call containerise function which does some automations for you\n # like moving created nodes to the AVALON_CONTAINERS subnetwork\n containerised_nodes = pipeline.containerise(\n node_name,\n namespace,\n nodes,\n context,\n self.__class__.__name__,\n suffix=\"\",\n )\n\n return containerised_nodes\n\n def update(self, container, representation):\n\n node = container[\"node\"]\n try:\n file_node = next(\n n for n in node.children() if n.type().name() == \"file\"\n )\n except StopIteration:\n self.log.error(\"Could not find node of type `file`\")\n return\n\n # Update the file path from representation\n file_path = get_representation_path(representation)\n file_path = file_path.replace(\"\\\\\", \"/\")\n\n file_node.setParms({\"file\": file_path})\n\n # Update attribute\n node.setParms({\"representation\": str(representation[\"_id\"])})\n\n def remove(self, container):\n\n node = container[\"node\"]\n node.destroy()\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def get_node_name(self, context, name=None, namespace=None):\n \"\"\"Define node name.\"\"\"\n\n if not namespace:\n namespace = context[\"asset\"][\"name\"]\n\n if namespace:\n node_name = \"{}_{}\".format(namespace, name)\n else:\n node_name = name\n\n return namespace, node_name\n\n def create_load_node_tree(self, file_path, node_name, subset_name):\n \"\"\"Create Load network.\n\n you can start building your tree at any obj level.\n it'll be much easier to build it in the root obj level.\n\n Afterwards, your tree will be automatically moved to\n '/obj/AVALON_CONTAINERS' subnetwork.\n \"\"\"\n import hou\n\n # Get the root obj level\n obj = hou.node(\"/obj\")\n\n # Create a new obj geo node\n parent_node = obj.createNode(\"geo\", node_name=node_name)\n\n # In older houdini,\n # when reating a new obj geo node, a default file node will be\n # automatically created.\n # so, we will delete it if exists.\n file_node = parent_node.node(\"file1\")\n if file_node:\n file_node.destroy()\n\n # Create a new file node\n file_node = parent_node.createNode(\"file\", node_name=node_name)\n file_node.setParms({\"file\": file_path})\n\n # Create attribute delete\n attribdelete_name = \"attribdelete_{}\".format(subset_name)\n attribdelete = parent_node.createNode(\"attribdelete\",\n node_name=attribdelete_name)\n attribdelete.setParms({\"ptdel\": \"fbx_*\"})\n attribdelete.setInput(0, file_node)\n\n # Create a Null node\n null_name = \"OUT_{}\".format(subset_name)\n null = parent_node.createNode(\"null\", node_name=null_name)\n null.setInput(0, attribdelete)\n\n # Ensure display flag is on the file_node input node and not on the OUT\n # node to optimize \"debug\" displaying in the viewport.\n file_node.setDisplayFlag(True)\n\n # Set new position for children nodes\n parent_node.layoutChildren()\n\n # Return all the nodes\n return [parent_node, file_node, attribdelete, null]\n"
},
{
"path": "openpype/hosts/houdini/plugins/load/load_hda.py",
"content": "# -*- coding: utf-8 -*-\nimport os\nfrom openpype.pipeline import (\n load,\n get_representation_path,\n)\nfrom openpype.hosts.houdini.api import pipeline\n\n\nclass HdaLoader(load.LoaderPlugin):\n \"\"\"Load Houdini Digital Asset file.\"\"\"\n\n families = [\"hda\"]\n label = \"Load Hda\"\n representations = [\"hda\"]\n order = -10\n icon = \"code-fork\"\n color = \"orange\"\n\n def load(self, context, name=None, namespace=None, data=None):\n import hou\n\n # Format file name, Houdini only wants forward slashes\n file_path = self.filepath_from_context(context)\n file_path = os.path.normpath(file_path)\n file_path = file_path.replace(\"\\\\\", \"/\")\n\n # Get the root node\n obj = hou.node(\"/obj\")\n\n # Create a unique name\n counter = 1\n namespace = namespace or context[\"asset\"][\"name\"]\n formatted = \"{}_{}\".format(namespace, name) if namespace else name\n node_name = \"{0}_{1:03d}\".format(formatted, counter)\n\n hou.hda.installFile(file_path)\n hda_node = obj.createNode(name, node_name)\n\n self[:] = [hda_node]\n\n return pipeline.containerise(\n node_name,\n namespace,\n [hda_node],\n context,\n self.__class__.__name__,\n suffix=\"\",\n )\n\n def update(self, container, representation):\n import hou\n\n hda_node = container[\"node\"]\n file_path = get_representation_path(representation)\n file_path = file_path.replace(\"\\\\\", \"/\")\n hou.hda.installFile(file_path)\n defs = hda_node.type().allInstalledDefinitions()\n def_paths = [d.libraryFilePath() for d in defs]\n new = def_paths.index(file_path)\n defs[new].setIsPreferred(True)\n hda_node.setParms({\n \"representation\": str(representation[\"_id\"])\n })\n\n def remove(self, container):\n node = container[\"node\"]\n node.destroy()\n"
},
{
"path": "openpype/hosts/houdini/plugins/load/load_image.py",
"content": "import os\n\nfrom openpype.pipeline import (\n load,\n get_representation_path,\n AVALON_CONTAINER_ID,\n)\nfrom openpype.hosts.houdini.api import lib, pipeline\n\nimport hou\n\n\ndef get_image_avalon_container():\n \"\"\"The COP2 files must be in a COP2 network.\n\n So we maintain a single entry point within AVALON_CONTAINERS,\n just for ease of use.\n\n \"\"\"\n\n path = pipeline.AVALON_CONTAINERS\n avalon_container = hou.node(path)\n if not avalon_container:\n # Let's create avalon container secretly\n # but make sure the pipeline still is built the\n # way we anticipate it was built, asserting it.\n assert path == \"/obj/AVALON_CONTAINERS\"\n\n parent = hou.node(\"/obj\")\n avalon_container = parent.createNode(\n \"subnet\", node_name=\"AVALON_CONTAINERS\"\n )\n\n image_container = hou.node(path + \"/IMAGES\")\n if not image_container:\n image_container = avalon_container.createNode(\n \"cop2net\", node_name=\"IMAGES\"\n )\n image_container.moveToGoodPosition()\n\n return image_container\n\n\nclass ImageLoader(load.LoaderPlugin):\n \"\"\"Load images into COP2\"\"\"\n\n families = [\"imagesequence\"]\n label = \"Load Image (COP2)\"\n representations = [\"*\"]\n order = -10\n\n icon = \"code-fork\"\n color = \"orange\"\n\n def load(self, context, name=None, namespace=None, data=None):\n\n # Format file name, Houdini only wants forward slashes\n file_path = self.filepath_from_context(context)\n file_path = os.path.normpath(file_path)\n file_path = file_path.replace(\"\\\\\", \"/\")\n file_path = self._get_file_sequence(file_path)\n\n # Get the root node\n parent = get_image_avalon_container()\n\n # Define node name\n namespace = namespace if namespace else context[\"asset\"][\"name\"]\n node_name = \"{}_{}\".format(namespace, name) if namespace else name\n\n node = parent.createNode(\"file\", node_name=node_name)\n node.moveToGoodPosition()\n\n node.setParms({\"filename1\": file_path})\n\n # Imprint it manually\n data = {\n \"schema\": \"openpype:container-2.0\",\n \"id\": AVALON_CONTAINER_ID,\n \"name\": node_name,\n \"namespace\": namespace,\n \"loader\": str(self.__class__.__name__),\n \"representation\": str(context[\"representation\"][\"_id\"]),\n }\n\n # todo: add folder=\"Avalon\"\n lib.imprint(node, data)\n\n return node\n\n def update(self, container, representation):\n\n node = container[\"node\"]\n\n # Update the file path\n file_path = get_representation_path(representation)\n file_path = file_path.replace(\"\\\\\", \"/\")\n file_path = self._get_file_sequence(file_path)\n\n # Update attributes\n node.setParms(\n {\n \"filename1\": file_path,\n \"representation\": str(representation[\"_id\"]),\n }\n )\n\n def remove(self, container):\n\n node = container[\"node\"]\n\n # Let's clean up the IMAGES COP2 network\n # if it ends up being empty and we deleted\n # the last file node. Store the parent\n # before we delete the node.\n parent = node.parent()\n\n node.destroy()\n\n if not parent.children():\n parent.destroy()\n\n def _get_file_sequence(self, file_path):\n root = os.path.dirname(file_path)\n files = sorted(os.listdir(root))\n\n first_fname = files[0]\n prefix, padding, suffix = first_fname.rsplit(\".\", 2)\n fname = \".\".join([prefix, \"$F{}\".format(len(padding)), suffix])\n return os.path.join(root, fname).replace(\"\\\\\", \"/\")\n\n def switch(self, container, representation):\n self.update(container, representation)\n"
},
{
"path": "openpype/hosts/houdini/plugins/load/load_redshift_proxy.py",
"content": "import os\nimport re\nfrom openpype.pipeline import (\n load,\n get_representation_path,\n)\nfrom openpype.hosts.houdini.api import pipeline\nfrom openpype.pipeline.load import LoadError\n\nimport hou\n\n\nclass RedshiftProxyLoader(load.LoaderPlugin):\n \"\"\"Load Redshift Proxy\"\"\"\n\n families = [\"redshiftproxy\"]\n label = \"Load Redshift Proxy\"\n representations = [\"rs\"]\n order = -10\n icon = \"code-fork\"\n color = \"orange\"\n\n def load(self, context, name=None, namespace=None, data=None):\n\n # Get the root node\n obj = hou.node(\"/obj\")\n\n # Define node name\n namespace = namespace if namespace else context[\"asset\"][\"name\"]\n node_name = \"{}_{}\".format(namespace, name) if namespace else name\n\n # Create a new geo node\n container = obj.createNode(\"geo\", node_name=node_name)\n\n # Check whether the Redshift parameters exist - if not, then likely\n # redshift is not set up or initialized correctly\n if not container.parm(\"RS_objprop_proxy_enable\"):\n container.destroy()\n raise LoadError(\"Unable to initialize geo node with Redshift \"\n \"attributes. Make sure you have the Redshift \"\n \"plug-in set up correctly for Houdini.\")\n\n # Enable by default\n container.setParms({\n \"RS_objprop_proxy_enable\": True,\n \"RS_objprop_proxy_file\": self.format_path(\n self.filepath_from_context(context),\n context[\"representation\"])\n })\n\n # Remove the file node, it only loads static meshes\n # Houdini 17 has removed the file node from the geo node\n file_node = container.node(\"file1\")\n if file_node:\n file_node.destroy()\n\n # Add this stub node inside so it previews ok\n proxy_sop = container.createNode(\"redshift_proxySOP\",\n node_name=node_name)\n proxy_sop.setDisplayFlag(True)\n\n nodes = [container, proxy_sop]\n\n self[:] = nodes\n\n return pipeline.containerise(\n node_name,\n namespace,\n nodes,\n context,\n self.__class__.__name__,\n suffix=\"\",\n )\n\n def update(self, container, representation):\n\n # Update the file path\n file_path = get_representation_path(representation)\n\n node = container[\"node\"]\n node.setParms({\n \"RS_objprop_proxy_file\": self.format_path(\n file_path, representation)\n })\n\n # Update attribute\n node.setParms({\"representation\": str(representation[\"_id\"])})\n\n def remove(self, container):\n\n node = container[\"node\"]\n node.destroy()\n\n @staticmethod\n def format_path(path, representation):\n \"\"\"Format file path correctly for single redshift proxy\n or redshift proxy sequence.\"\"\"\n if not os.path.exists(path):\n raise RuntimeError(\"Path does not exist: %s\" % path)\n\n is_sequence = bool(representation[\"context\"].get(\"frame\"))\n # The path is either a single file or sequence in a folder.\n if is_sequence:\n filename = re.sub(r\"(.*)\\.(\\d+)\\.(rs.*)\", \"\\\\1.$F4.\\\\3\", path)\n filename = os.path.join(path, filename)\n else:\n filename = path\n\n filename = os.path.normpath(filename)\n filename = filename.replace(\"\\\\\", \"/\")\n\n return filename\n"
},
{
"path": "openpype/hosts/houdini/plugins/load/load_usd_layer.py",
"content": "from openpype.pipeline import (\n load,\n get_representation_path,\n AVALON_CONTAINER_ID,\n)\nfrom openpype.hosts.houdini.api import lib\n\n\nclass USDSublayerLoader(load.LoaderPlugin):\n \"\"\"Sublayer USD file in Solaris\"\"\"\n\n families = [\n \"usd\",\n \"usdCamera\",\n ]\n label = \"Sublayer USD\"\n representations = [\"usd\", \"usda\", \"usdlc\", \"usdnc\", \"abc\"]\n order = 1\n\n icon = \"code-fork\"\n color = \"orange\"\n\n def load(self, context, name=None, namespace=None, data=None):\n\n import os\n import hou\n\n # Format file name, Houdini only wants forward slashes\n file_path = self.filepath_from_context(context)\n file_path = os.path.normpath(file_path)\n file_path = file_path.replace(\"\\\\\", \"/\")\n\n # Get the root node\n stage = hou.node(\"/stage\")\n\n # Define node name\n namespace = namespace if namespace else context[\"asset\"][\"name\"]\n node_name = \"{}_{}\".format(namespace, name) if namespace else name\n\n # Create USD reference\n container = stage.createNode(\"sublayer\", node_name=node_name)\n container.setParms({\"filepath1\": file_path})\n container.moveToGoodPosition()\n\n # Imprint it manually\n data = {\n \"schema\": \"openpype:container-2.0\",\n \"id\": AVALON_CONTAINER_ID,\n \"name\": node_name,\n \"namespace\": namespace,\n \"loader\": str(self.__class__.__name__),\n \"representation\": str(context[\"representation\"][\"_id\"]),\n }\n\n # todo: add folder=\"Avalon\"\n lib.imprint(container, data)\n\n return container\n\n def update(self, container, representation):\n\n node = container[\"node\"]\n\n # Update the file path\n file_path = get_representation_path(representation)\n file_path = file_path.replace(\"\\\\\", \"/\")\n\n # Update attributes\n node.setParms(\n {\n \"filepath1\": file_path,\n \"representation\": str(representation[\"_id\"]),\n }\n )\n\n # Reload files\n node.parm(\"reload\").pressButton()\n\n def remove(self, container):\n\n node = container[\"node\"]\n node.destroy()\n\n def switch(self, container, representation):\n self.update(container, representation)\n"
},
{
"path": "openpype/hosts/houdini/plugins/load/load_usd_reference.py",
"content": "from openpype.pipeline import (\n load,\n get_representation_path,\n AVALON_CONTAINER_ID,\n)\nfrom openpype.hosts.houdini.api import lib\n\n\nclass USDReferenceLoader(load.LoaderPlugin):\n \"\"\"Reference USD file in Solaris\"\"\"\n\n families = [\n \"usd\",\n \"usdCamera\",\n ]\n label = \"Reference USD\"\n representations = [\"usd\", \"usda\", \"usdlc\", \"usdnc\", \"abc\"]\n order = -8\n\n icon = \"code-fork\"\n color = \"orange\"\n\n def load(self, context, name=None, namespace=None, data=None):\n\n import os\n import hou\n\n # Format file name, Houdini only wants forward slashes\n file_path = self.filepath_from_context(context)\n file_path = os.path.normpath(file_path)\n file_path = file_path.replace(\"\\\\\", \"/\")\n\n # Get the root node\n stage = hou.node(\"/stage\")\n\n # Define node name\n namespace = namespace if namespace else context[\"asset\"][\"name\"]\n node_name = \"{}_{}\".format(namespace, name) if namespace else name\n\n # Create USD reference\n container = stage.createNode(\"reference\", node_name=node_name)\n container.setParms({\"filepath1\": file_path})\n container.moveToGoodPosition()\n\n # Imprint it manually\n data = {\n \"schema\": \"openpype:container-2.0\",\n \"id\": AVALON_CONTAINER_ID,\n \"name\": node_name,\n \"namespace\": namespace,\n \"loader\": str(self.__class__.__name__),\n \"representation\": str(context[\"representation\"][\"_id\"]),\n }\n\n # todo: add folder=\"Avalon\"\n lib.imprint(container, data)\n\n return container\n\n def update(self, container, representation):\n\n node = container[\"node\"]\n\n # Update the file path\n file_path = get_representation_path(representation)\n file_path = file_path.replace(\"\\\\\", \"/\")\n\n # Update attributes\n node.setParms(\n {\n \"filepath1\": file_path,\n \"representation\": str(representation[\"_id\"]),\n }\n )\n\n # Reload files\n node.parm(\"reload\").pressButton()\n\n def remove(self, container):\n\n node = container[\"node\"]\n node.destroy()\n\n def switch(self, container, representation):\n self.update(container, representation)\n"
},
{
"path": "openpype/hosts/houdini/plugins/load/load_vdb.py",
"content": "import os\nimport re\n\nfrom openpype.pipeline import (\n load,\n get_representation_path,\n)\nfrom openpype.hosts.houdini.api import pipeline\n\n\nclass VdbLoader(load.LoaderPlugin):\n \"\"\"Load VDB\"\"\"\n\n families = [\"vdbcache\"]\n label = \"Load VDB\"\n representations = [\"vdb\"]\n order = -10\n icon = \"code-fork\"\n color = \"orange\"\n\n def load(self, context, name=None, namespace=None, data=None):\n\n import hou\n\n # Get the root node\n obj = hou.node(\"/obj\")\n\n # Define node name\n namespace = namespace if namespace else context[\"asset\"][\"name\"]\n node_name = \"{}_{}\".format(namespace, name) if namespace else name\n\n # Create a new geo node\n container = obj.createNode(\"geo\", node_name=node_name)\n\n # Remove the file node, it only loads static meshes\n # Houdini 17 has removed the file node from the geo node\n file_node = container.node(\"file1\")\n if file_node:\n file_node.destroy()\n\n # Explicitly create a file node\n file_node = container.createNode(\"file\", node_name=node_name)\n path = self.filepath_from_context(context)\n file_node.setParms(\n {\"file\": self.format_path(path, context[\"representation\"])})\n\n # Set display on last node\n file_node.setDisplayFlag(True)\n\n nodes = [container, file_node]\n self[:] = nodes\n\n return pipeline.containerise(\n node_name,\n namespace,\n nodes,\n context,\n self.__class__.__name__,\n suffix=\"\",\n )\n\n @staticmethod\n def format_path(path, representation):\n \"\"\"Format file path correctly for single vdb or vdb sequence.\"\"\"\n if not os.path.exists(path):\n raise RuntimeError(\"Path does not exist: %s\" % path)\n\n is_sequence = bool(representation[\"context\"].get(\"frame\"))\n # The path is either a single file or sequence in a folder.\n if not is_sequence:\n filename = path\n else:\n filename = re.sub(r\"(.*)\\.(\\d+)\\.vdb$\", \"\\\\1.$F4.vdb\", path)\n\n filename = os.path.join(path, filename)\n\n filename = os.path.normpath(filename)\n filename = filename.replace(\"\\\\\", \"/\")\n\n return filename\n\n def update(self, container, representation):\n\n node = container[\"node\"]\n try:\n file_node = next(\n n for n in node.children() if n.type().name() == \"file\"\n )\n except StopIteration:\n self.log.error(\"Could not find node of type `alembic`\")\n return\n\n # Update the file path\n file_path = get_representation_path(representation)\n file_path = self.format_path(file_path, representation)\n\n file_node.setParms({\"file\": file_path})\n\n # Update attribute\n node.setParms({\"representation\": str(representation[\"_id\"])})\n\n def remove(self, container):\n\n node = container[\"node\"]\n node.destroy()\n\n def switch(self, container, representation):\n self.update(container, representation)\n"
},
{
"path": "openpype/hosts/houdini/plugins/load/show_usdview.py",
"content": "import os\nimport platform\nimport subprocess\n\nfrom openpype.lib.vendor_bin_utils import find_executable\nfrom openpype.pipeline import load\n\n\nclass ShowInUsdview(load.LoaderPlugin):\n \"\"\"Open USD file in usdview\"\"\"\n\n label = \"Show in usdview\"\n representations = [\"*\"]\n families = [\"*\"]\n extensions = {\"usd\", \"usda\", \"usdlc\", \"usdnc\", \"abc\"}\n order = 15\n\n icon = \"code-fork\"\n color = \"white\"\n\n def load(self, context, name=None, namespace=None, data=None):\n from pathlib import Path\n\n if platform.system() == \"Windows\":\n executable = \"usdview.bat\"\n else:\n executable = \"usdview\"\n\n usdview = find_executable(executable)\n if not usdview:\n raise RuntimeError(\"Unable to find usdview\")\n\n # For some reason Windows can return the path like:\n # C:/PROGRA~1/SIDEEF~1/HOUDIN~1.435/bin/usdview\n # convert to resolved path so `subprocess` can take it\n usdview = str(Path(usdview).resolve().as_posix())\n\n filepath = self.filepath_from_context(context)\n filepath = os.path.normpath(filepath)\n filepath = filepath.replace(\"\\\\\", \"/\")\n\n if not os.path.exists(filepath):\n self.log.error(\"File does not exist: %s\" % filepath)\n return\n\n self.log.info(\"Start houdini variant of usdview...\")\n\n subprocess.Popen([usdview, filepath, \"--renderer\", \"GL\"])\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/collect_active_state.py",
"content": "import pyblish.api\nimport hou\n\n\nclass CollectInstanceActiveState(pyblish.api.InstancePlugin):\n \"\"\"Collect default active state for instance from its node bypass state.\n\n This is done at the very end of the CollectorOrder so that any required\n collecting of data iterating over instances (with InstancePlugin) will\n actually collect the data for when the user enables the state in the UI.\n Otherwise potentially required data might have skipped collecting.\n\n \"\"\"\n\n order = pyblish.api.CollectorOrder + 0.299\n families = [\"*\"]\n hosts = [\"houdini\"]\n label = \"Instance Active State\"\n\n def process(self, instance):\n\n # Must have node to check for bypass state\n if len(instance) == 0:\n return\n\n # Check bypass state and reverse\n active = True\n node = hou.node(instance.data.get(\"instance_node\"))\n if hasattr(node, \"isBypassed\"):\n active = not node.isBypassed()\n\n # Set instance active state\n instance.data.update(\n {\n \"active\": active,\n # temporarily translation of `active` to `publish` till\n # issue has been resolved:\n # https://github.com/pyblish/pyblish-base/issues/307\n \"publish\": active,\n }\n )\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/collect_arnold_rop.py",
"content": "import os\nimport re\n\nimport hou\nimport pyblish.api\n\nfrom openpype.hosts.houdini.api import colorspace\nfrom openpype.hosts.houdini.api.lib import (\n evalParmNoFrame, get_color_management_preferences)\n\n\nclass CollectArnoldROPRenderProducts(pyblish.api.InstancePlugin):\n \"\"\"Collect Arnold ROP Render Products\n\n Collects the instance.data[\"files\"] for the render products.\n\n Provides:\n instance -> files\n\n \"\"\"\n\n label = \"Arnold ROP Render Products\"\n # This specific order value is used so that\n # this plugin runs after CollectFrames\n order = pyblish.api.CollectorOrder + 0.11\n hosts = [\"houdini\"]\n families = [\"arnold_rop\"]\n\n def process(self, instance):\n\n rop = hou.node(instance.data.get(\"instance_node\"))\n\n # Collect chunkSize\n chunk_size_parm = rop.parm(\"chunkSize\")\n if chunk_size_parm:\n chunk_size = int(chunk_size_parm.eval())\n instance.data[\"chunkSize\"] = chunk_size\n self.log.debug(\"Chunk Size: %s\" % chunk_size)\n\n default_prefix = evalParmNoFrame(rop, \"ar_picture\")\n render_products = []\n\n # Store whether we are splitting the render job (export + render)\n split_render = bool(rop.parm(\"ar_ass_export_enable\").eval())\n instance.data[\"splitRender\"] = split_render\n export_prefix = None\n export_products = []\n if split_render:\n export_prefix = evalParmNoFrame(\n rop, \"ar_ass_file\", pad_character=\"0\"\n )\n beauty_export_product = self.get_render_product_name(\n prefix=export_prefix,\n suffix=None)\n export_products.append(beauty_export_product)\n self.log.debug(\n \"Found export product: {}\".format(beauty_export_product)\n )\n instance.data[\"ifdFile\"] = beauty_export_product\n instance.data[\"exportFiles\"] = list(export_products)\n\n # Default beauty AOV\n beauty_product = self.get_render_product_name(prefix=default_prefix,\n suffix=None)\n render_products.append(beauty_product)\n\n files_by_aov = {\n \"\": self.generate_expected_files(instance, beauty_product)\n }\n\n num_aovs = rop.evalParm(\"ar_aovs\")\n for index in range(1, num_aovs + 1):\n # Skip disabled AOVs\n if not rop.evalParm(\"ar_enable_aov{}\".format(index)):\n continue\n\n if rop.evalParm(\"ar_aov_exr_enable_layer_name{}\".format(index)):\n label = rop.evalParm(\"ar_aov_exr_layer_name{}\".format(index))\n else:\n label = evalParmNoFrame(rop, \"ar_aov_label{}\".format(index))\n\n aov_product = self.get_render_product_name(default_prefix,\n suffix=label)\n render_products.append(aov_product)\n files_by_aov[label] = self.generate_expected_files(instance,\n aov_product)\n\n for product in render_products:\n self.log.debug(\"Found render product: {}\".format(product))\n\n instance.data[\"files\"] = list(render_products)\n instance.data[\"renderProducts\"] = colorspace.ARenderProduct()\n\n # For now by default do NOT try to publish the rendered output\n instance.data[\"publishJobState\"] = \"Suspended\"\n instance.data[\"attachTo\"] = [] # stub required data\n\n if \"expectedFiles\" not in instance.data:\n instance.data[\"expectedFiles\"] = list()\n instance.data[\"expectedFiles\"].append(files_by_aov)\n\n # update the colorspace data\n colorspace_data = get_color_management_preferences()\n instance.data[\"colorspaceConfig\"] = colorspace_data[\"config\"]\n instance.data[\"colorspaceDisplay\"] = colorspace_data[\"display\"]\n instance.data[\"colorspaceView\"] = colorspace_data[\"view\"]\n\n def get_render_product_name(self, prefix, suffix):\n \"\"\"Return the output filename using the AOV prefix and suffix\"\"\"\n\n # When AOV is explicitly defined in prefix we just swap it out\n # directly with the AOV suffix to embed it.\n # Note: ${AOV} seems to be evaluated in the parameter as %AOV%\n if \"%AOV%\" in prefix:\n # It seems that when some special separator characters are present\n # before the %AOV% token that Redshift will secretly remove it if\n # there is no suffix for the current product, for example:\n # foo_%AOV% -> foo.exr\n pattern = \"%AOV%\" if suffix else \"[._-]?%AOV%\"\n product_name = re.sub(pattern,\n suffix,\n prefix,\n flags=re.IGNORECASE)\n else:\n if suffix:\n # Add \".{suffix}\" before the extension\n prefix_base, ext = os.path.splitext(prefix)\n product_name = prefix_base + \".\" + suffix + ext\n else:\n product_name = prefix\n\n return product_name\n\n def generate_expected_files(self, instance, path):\n \"\"\"Create expected files in instance data\"\"\"\n\n dir = os.path.dirname(path)\n file = os.path.basename(path)\n\n if \"#\" in file:\n def replace(match):\n return \"%0{}d\".format(len(match.group()))\n\n file = re.sub(\"#+\", replace, file)\n\n if \"%\" not in file:\n return path\n\n expected_files = []\n start = instance.data[\"frameStartHandle\"]\n end = instance.data[\"frameEndHandle\"]\n\n for i in range(int(start), (int(end) + 1)):\n expected_files.append(\n os.path.join(dir, (file % i)).replace(\"\\\\\", \"/\"))\n\n return expected_files\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/collect_asset_handles.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Collector plugin for frames data on ROP instances.\"\"\"\nimport hou # noqa\nimport pyblish.api\nfrom openpype.lib import BoolDef\nfrom openpype.pipeline import OpenPypePyblishPluginMixin\n\n\nclass CollectAssetHandles(pyblish.api.InstancePlugin,\n OpenPypePyblishPluginMixin):\n \"\"\"Apply asset handles.\n\n If instance does not have:\n - frameStart\n - frameEnd\n - handleStart\n - handleEnd\n But it does have:\n - frameStartHandle\n - frameEndHandle\n\n Then we will retrieve the asset's handles to compute\n the exclusive frame range and actual handle ranges.\n \"\"\"\n\n hosts = [\"houdini\"]\n\n # This specific order value is used so that\n # this plugin runs after CollectAnatomyInstanceData\n order = pyblish.api.CollectorOrder + 0.499\n\n label = \"Collect Asset Handles\"\n use_asset_handles = True\n\n def process(self, instance):\n # Only process instances without already existing handles data\n # but that do have frameStartHandle and frameEndHandle defined\n # like the data collected from CollectRopFrameRange\n if \"frameStartHandle\" not in instance.data:\n return\n if \"frameEndHandle\" not in instance.data:\n return\n\n has_existing_data = {\n \"handleStart\",\n \"handleEnd\",\n \"frameStart\",\n \"frameEnd\"\n }.issubset(instance.data)\n if has_existing_data:\n return\n\n attr_values = self.get_attr_values_from_data(instance.data)\n if attr_values.get(\"use_handles\", self.use_asset_handles):\n asset_data = instance.data[\"assetEntity\"][\"data\"]\n handle_start = asset_data.get(\"handleStart\", 0)\n handle_end = asset_data.get(\"handleEnd\", 0)\n else:\n handle_start = 0\n handle_end = 0\n\n frame_start = instance.data[\"frameStartHandle\"] + handle_start\n frame_end = instance.data[\"frameEndHandle\"] - handle_end\n\n instance.data.update({\n \"handleStart\": handle_start,\n \"handleEnd\": handle_end,\n \"frameStart\": frame_start,\n \"frameEnd\": frame_end\n })\n\n # Log debug message about the collected frame range\n if attr_values.get(\"use_handles\", self.use_asset_handles):\n self.log.debug(\n \"Full Frame range with Handles \"\n \"[{frame_start_handle} - {frame_end_handle}]\"\n .format(\n frame_start_handle=instance.data[\"frameStartHandle\"],\n frame_end_handle=instance.data[\"frameEndHandle\"]\n )\n )\n else:\n self.log.debug(\n \"Use handles is deactivated for this instance, \"\n \"start and end handles are set to 0.\"\n )\n\n # Log collected frame range to the user\n message = \"Frame range [{frame_start} - {frame_end}]\".format(\n frame_start=frame_start,\n frame_end=frame_end\n )\n if handle_start or handle_end:\n message += \" with handles [{handle_start}]-[{handle_end}]\".format(\n handle_start=handle_start,\n handle_end=handle_end\n )\n self.log.info(message)\n\n if instance.data.get(\"byFrameStep\", 1.0) != 1.0:\n self.log.info(\n \"Frame steps {}\".format(instance.data[\"byFrameStep\"]))\n\n # Add frame range to label if the instance has a frame range.\n label = instance.data.get(\"label\", instance.data[\"name\"])\n instance.data[\"label\"] = (\n \"{label} [{frame_start_handle} - {frame_end_handle}]\"\n .format(\n label=label,\n frame_start_handle=instance.data[\"frameStartHandle\"],\n frame_end_handle=instance.data[\"frameEndHandle\"]\n )\n )\n\n @classmethod\n def get_attribute_defs(cls):\n return [\n BoolDef(\"use_handles\",\n tooltip=\"Disable this if you want the publisher to\"\n \" ignore start and end handles specified in the\"\n \" asset data for this publish instance\",\n default=cls.use_asset_handles,\n label=\"Use asset handles\")\n ]\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/collect_cache_farm.py",
"content": "import os\nimport pyblish.api\nimport hou\nfrom openpype.hosts.houdini.api import lib\n\n\nclass CollectDataforCache(pyblish.api.InstancePlugin):\n \"\"\"Collect data for caching to Deadline.\"\"\"\n\n order = pyblish.api.CollectorOrder + 0.04\n families = [\"ass\", \"pointcache\",\n \"mantraifd\", \"redshiftproxy\",\n \"vdbcache\"]\n hosts = [\"houdini\"]\n targets = [\"local\", \"remote\"]\n label = \"Collect Data for Cache\"\n\n def process(self, instance):\n creator_attribute = instance.data[\"creator_attributes\"]\n farm_enabled = creator_attribute[\"farm\"]\n instance.data[\"farm\"] = farm_enabled\n if not farm_enabled:\n self.log.debug(\"Caching on farm is disabled. \"\n \"Skipping farm collecting.\")\n return\n # Why do we need this particular collector to collect the expected\n # output files from a ROP node. Don't we have a dedicated collector\n # for that yet?\n # Collect expected files\n ropnode = hou.node(instance.data[\"instance_node\"])\n output_parm = lib.get_output_parameter(ropnode)\n expected_filepath = output_parm.eval()\n instance.data.setdefault(\"files\", list())\n instance.data.setdefault(\"expectedFiles\", list())\n if instance.data.get(\"frames\"):\n files = self.get_files(instance, expected_filepath)\n # list of files\n instance.data[\"files\"].extend(files)\n else:\n # single file\n instance.data[\"files\"].append(output_parm.eval())\n cache_files = {\"_\": instance.data[\"files\"]}\n # Convert instance family to pointcache if it is bgeo or abc\n # because ???\n for family in instance.data[\"families\"]:\n if family == \"bgeo\" or \"abc\":\n instance.data[\"family\"] = \"pointcache\"\n break\n instance.data.update({\n \"plugin\": \"Houdini\",\n \"publish\": True\n })\n instance.data[\"families\"].append(\"publish.hou\")\n instance.data[\"expectedFiles\"].append(cache_files)\n\n self.log.debug(\"{}\".format(instance.data))\n\n def get_files(self, instance, output_parm):\n \"\"\"Get the files with the frame range data\n\n Args:\n instance (_type_): instance\n output_parm (_type_): path of output parameter\n\n Returns:\n files: a list of files\n \"\"\"\n directory = os.path.dirname(output_parm)\n\n files = [\n os.path.join(directory, frame).replace(\"\\\\\", \"/\")\n for frame in instance.data[\"frames\"]\n ]\n\n return files\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/collect_chunk_size.py",
"content": "import pyblish.api\nfrom openpype.lib import NumberDef\nfrom openpype.pipeline import OpenPypePyblishPluginMixin\n\n\nclass CollectChunkSize(pyblish.api.InstancePlugin,\n OpenPypePyblishPluginMixin):\n \"\"\"Collect chunk size for cache submission to Deadline.\"\"\"\n\n order = pyblish.api.CollectorOrder + 0.05\n families = [\"ass\", \"pointcache\",\n \"vdbcache\", \"mantraifd\",\n \"redshiftproxy\"]\n hosts = [\"houdini\"]\n targets = [\"local\", \"remote\"]\n label = \"Collect Chunk Size\"\n chunk_size = 999999\n\n def process(self, instance):\n # need to get the chunk size info from the setting\n attr_values = self.get_attr_values_from_data(instance.data)\n instance.data[\"chunkSize\"] = attr_values.get(\"chunkSize\")\n\n @classmethod\n def get_attribute_defs(cls):\n return [\n NumberDef(\"chunkSize\",\n minimum=1,\n maximum=999999,\n decimals=0,\n default=cls.chunk_size,\n label=\"Frame Per Task\")\n ]\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/collect_current_file.py",
"content": "import os\nimport hou\n\nimport pyblish.api\n\n\nclass CollectHoudiniCurrentFile(pyblish.api.ContextPlugin):\n \"\"\"Inject the current working file into context\"\"\"\n\n order = pyblish.api.CollectorOrder - 0.1\n label = \"Houdini Current File\"\n hosts = [\"houdini\"]\n\n def process(self, context):\n \"\"\"Inject the current working file\"\"\"\n\n current_file = hou.hipFile.path()\n if not os.path.exists(current_file):\n # By default, Houdini will even point a new scene to a path.\n # However if the file is not saved at all and does not exist,\n # we assume the user never set it.\n current_file = \"\"\n\n elif os.path.basename(current_file) == \"untitled.hip\":\n # Due to even a new file being called 'untitled.hip' we are unable\n # to confirm the current scene was ever saved because the file\n # could have existed already. We will allow it if the file exists,\n # but show a warning for this edge case to clarify the potential\n # false positive.\n self.log.warning(\n \"Current file is 'untitled.hip' and we are \"\n \"unable to detect whether the current scene is \"\n \"saved correctly.\"\n )\n\n context.data[\"currentFile\"] = current_file\n self.log.info('Current workfile path: {}'.format(current_file))\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/collect_frames.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Collector plugin for frames data on ROP instances.\"\"\"\nimport os\nimport re\n\nimport hou # noqa\nimport pyblish.api\nfrom openpype.hosts.houdini.api import lib\n\n\nclass CollectFrames(pyblish.api.InstancePlugin):\n \"\"\"Collect all frames which would be saved from the ROP nodes\"\"\"\n\n # This specific order value is used so that\n # this plugin runs after CollectRopFrameRange\n order = pyblish.api.CollectorOrder + 0.1\n label = \"Collect Frames\"\n families = [\"vdbcache\", \"imagesequence\", \"ass\",\n \"mantraifd\", \"redshiftproxy\", \"review\",\n \"bgeo\"]\n\n def process(self, instance):\n\n ropnode = hou.node(instance.data[\"instance_node\"])\n\n start_frame = instance.data.get(\"frameStartHandle\", None)\n end_frame = instance.data.get(\"frameEndHandle\", None)\n\n output_parm = lib.get_output_parameter(ropnode)\n if start_frame is not None:\n # When rendering only a single frame still explicitly\n # get the name for that particular frame instead of current frame\n output = output_parm.evalAtFrame(start_frame)\n else:\n self.log.warning(\"Using current frame: {}\".format(hou.frame()))\n output = output_parm.eval()\n\n _, ext = lib.splitext(\n output, allowed_multidot_extensions=[\n \".ass.gz\", \".bgeo.sc\", \".bgeo.gz\",\n \".bgeo.lzma\", \".bgeo.bz2\"])\n file_name = os.path.basename(output)\n result = file_name\n\n # Get the filename pattern match from the output\n # path, so we can compute all frames that would\n # come out from rendering the ROP node if there\n # is a frame pattern in the name\n pattern = r\"\\w+\\.(\\d+)\" + re.escape(ext)\n match = re.match(pattern, file_name)\n\n if match and start_frame is not None:\n\n # Check if frames are bigger than 1 (file collection)\n # override the result\n if end_frame - start_frame > 0:\n result = self.create_file_list(\n match, int(start_frame), int(end_frame)\n )\n\n # todo: `frames` currently conflicts with \"explicit frames\" for a\n # for a custom frame list. So this should be refactored.\n instance.data.update({\"frames\": result})\n\n @staticmethod\n def create_file_list(match, start_frame, end_frame):\n \"\"\"Collect files based on frame range and `regex.match`\n\n Args:\n match(re.match): match object\n start_frame(int): start of the animation\n end_frame(int): end of the animation\n\n Returns:\n list\n\n \"\"\"\n\n # Get the padding length\n frame = match.group(1)\n padding = len(frame)\n\n # Get the parts of the filename surrounding the frame number,\n # so we can put our own frame numbers in.\n span = match.span(1)\n prefix = match.string[: span[0]]\n suffix = match.string[span[1]:]\n\n # Generate filenames for all frames\n result = []\n for i in range(start_frame, end_frame + 1):\n\n # Format frame number by the padding amount\n str_frame = \"{number:0{width}d}\".format(number=i, width=padding)\n\n file_name = prefix + str_frame + suffix\n result.append(file_name)\n\n return result\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/collect_inputs.py",
"content": "import pyblish.api\n\nfrom openpype.pipeline import registered_host\n\n\ndef collect_input_containers(nodes):\n \"\"\"Collect containers that contain any of the node in `nodes`.\n\n This will return any loaded Avalon container that contains at least one of\n the nodes. As such, the Avalon container is an input for it. Or in short,\n there are member nodes of that container.\n\n Returns:\n list: Input avalon containers\n\n \"\"\"\n\n # Lookup by node ids\n lookup = frozenset(nodes)\n\n containers = []\n host = registered_host()\n for container in host.ls():\n\n node = container[\"node\"]\n\n # Usually the loaded containers don't have any complex references\n # and the contained children should be all we need. So we disregard\n # checking for .references() on the nodes.\n members = set(node.allSubChildren())\n members.add(node) # include the node itself\n\n # If there's an intersection\n if not lookup.isdisjoint(members):\n containers.append(container)\n\n return containers\n\n\ndef iter_upstream(node):\n \"\"\"Yields all upstream inputs for the current node.\n\n This includes all `node.inputAncestors()` but also traverses through all\n `node.references()` for the node itself and for any of the upstream nodes.\n This method has no max-depth and will collect all upstream inputs.\n\n Yields:\n hou.Node: The upstream nodes, including references.\n\n \"\"\"\n\n upstream = node.inputAncestors(\n include_ref_inputs=True, follow_subnets=True\n )\n\n # Initialize process queue with the node's ancestors itself\n queue = list(upstream)\n collected = set(upstream)\n\n # Traverse upstream references for all nodes and yield them as we\n # process the queue.\n while queue:\n upstream_node = queue.pop()\n yield upstream_node\n\n # Find its references that are not collected yet.\n references = upstream_node.references()\n references = [n for n in references if n not in collected]\n\n queue.extend(references)\n collected.update(references)\n\n # Include the references' ancestors that have not been collected yet.\n for reference in references:\n ancestors = reference.inputAncestors(\n include_ref_inputs=True, follow_subnets=True\n )\n ancestors = [n for n in ancestors if n not in collected]\n\n queue.extend(ancestors)\n collected.update(ancestors)\n\n\nclass CollectUpstreamInputs(pyblish.api.InstancePlugin):\n \"\"\"Collect source input containers used for this publish.\n\n This will include `inputs` data of which loaded publishes were used in the\n generation of this publish. This leaves an upstream trace to what was used\n as input.\n\n \"\"\"\n\n label = \"Collect Inputs\"\n order = pyblish.api.CollectorOrder + 0.4\n hosts = [\"houdini\"]\n\n def process(self, instance):\n # We can't get the \"inputAncestors\" directly from the ROP\n # node, so we find the related output node (set in SOP/COP path)\n # and include that together with its ancestors\n output = instance.data.get(\"output_node\")\n\n if output is None:\n # If no valid output node is set then ignore it as validation\n # will be checking those cases.\n self.log.debug(\n \"No output node found, skipping collecting of inputs..\"\n )\n return\n\n # Collect all upstream parents\n nodes = list(iter_upstream(output))\n nodes.append(output)\n\n # Collect containers for the given set of nodes\n containers = collect_input_containers(nodes)\n\n inputs = [c[\"representation\"] for c in containers]\n instance.data[\"inputRepresentations\"] = inputs\n self.log.debug(\"Collected inputs: %s\" % inputs)\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/collect_instances.py",
"content": "import hou\n\nimport pyblish.api\n\nfrom openpype.hosts.houdini.api import lib\n\n\nclass CollectInstances(pyblish.api.ContextPlugin):\n \"\"\"Gather instances by all node in out graph and pre-defined attributes\n\n This collector takes into account assets that are associated with\n an specific node and marked with a unique identifier;\n\n Identifier:\n id (str): \"pyblish.avalon.instance\n\n Specific node:\n The specific node is important because it dictates in which way the\n subset is being exported.\n\n alembic: will export Alembic file which supports cascading attributes\n like 'cbId' and 'path'\n geometry: Can export a wide range of file types, default out\n\n \"\"\"\n\n order = pyblish.api.CollectorOrder - 0.01\n label = \"Collect Instances\"\n hosts = [\"houdini\"]\n\n def process(self, context):\n\n nodes = hou.node(\"/out\").children()\n nodes += hou.node(\"/obj\").children()\n\n # Include instances in USD stage only when it exists so it\n # remains backwards compatible with version before houdini 18\n stage = hou.node(\"/stage\")\n if stage:\n nodes += stage.recursiveGlob(\"*\", filter=hou.nodeTypeFilter.Rop)\n\n for node in nodes:\n\n if not node.parm(\"id\"):\n continue\n\n if node.evalParm(\"id\") != \"pyblish.avalon.instance\":\n continue\n\n # instance was created by new creator code, skip it as\n # it is already collected.\n if node.parm(\"creator_identifier\"):\n continue\n\n has_family = node.evalParm(\"family\")\n assert has_family, \"'%s' is missing 'family'\" % node.name()\n\n self.log.info(\n \"Processing legacy instance node {}\".format(node.path())\n )\n\n data = lib.read(node)\n # Check bypass state and reverse\n if hasattr(node, \"isBypassed\"):\n data.update({\"active\": not node.isBypassed()})\n\n # temporarily translation of `active` to `publish` till issue has\n # been resolved.\n # https://github.com/pyblish/pyblish-base/issues/307\n if \"active\" in data:\n data[\"publish\"] = data[\"active\"]\n\n # Create nice name if the instance has a frame range.\n label = data.get(\"name\", node.name())\n label += \" (%s)\" % data[\"asset\"] # include asset in name\n\n instance = context.create_instance(label)\n\n # Include `families` using `family` data\n instance.data[\"families\"] = [instance.data[\"family\"]]\n\n instance[:] = [node]\n instance.data[\"instance_node\"] = node.path()\n instance.data.update(data)\n\n def sort_by_family(instance):\n \"\"\"Sort by family\"\"\"\n return instance.data.get(\"families\", instance.data.get(\"family\"))\n\n # Sort/grouped by family (preserving local index)\n context[:] = sorted(context, key=sort_by_family)\n\n return context\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/collect_instances_usd_layered.py",
"content": "import hou\nimport pyblish.api\nfrom openpype.hosts.houdini.api import lib\nimport openpype.hosts.houdini.api.usd as hou_usdlib\nimport openpype.lib.usdlib as usdlib\n\n\nclass CollectInstancesUsdLayered(pyblish.api.ContextPlugin):\n \"\"\"Collect Instances from a ROP Network and its configured layer paths.\n\n The output nodes of the ROP node will only be published when *any* of the\n layers remain set to 'publish' by the user.\n\n This works differently from most of our Avalon instances in the pipeline.\n As opposed to storing `pyblish.avalon.instance` as id on the node we store\n `pyblish.avalon.usdlayered`.\n\n Additionally this instance has no need for storing family, asset, subset\n or name on the nodes. Instead all information is retrieved solely from\n the output filepath, which is an Avalon URI:\n avalon://{asset}/{subset}.{representation}\n\n Each final ROP node is considered a dependency for any of the Configured\n Save Path layers it sets along the way. As such, the instances shown in\n the Pyblish UI are solely the configured layers. The encapsulating usd\n files are generated whenever *any* of the dependencies is published.\n\n These dependency instances are stored in:\n instance.data[\"publishDependencies\"]\n\n \"\"\"\n\n order = pyblish.api.CollectorOrder - 0.01\n label = \"Collect Instances (USD Configured Layers)\"\n hosts = [\"houdini\"]\n\n def process(self, context):\n\n stage = hou.node(\"/stage\")\n if not stage:\n # Likely Houdini version <18\n return\n\n nodes = stage.recursiveGlob(\"*\", filter=hou.nodeTypeFilter.Rop)\n for node in nodes:\n\n if not node.parm(\"id\"):\n continue\n\n if node.evalParm(\"id\") != \"pyblish.avalon.usdlayered\":\n continue\n\n has_family = node.evalParm(\"family\")\n assert has_family, \"'%s' is missing 'family'\" % node.name()\n\n self.process_node(node, context)\n\n def sort_by_family(instance):\n \"\"\"Sort by family\"\"\"\n return instance.data.get(\"families\", instance.data.get(\"family\"))\n\n # Sort/grouped by family (preserving local index)\n context[:] = sorted(context, key=sort_by_family)\n\n return context\n\n def process_node(self, node, context):\n\n # Allow a single ROP node or a full ROP network of USD ROP nodes\n # to be processed as a single entry that should \"live together\" on\n # a publish.\n if node.type().name() == \"ropnet\":\n # All rop nodes inside ROP Network\n ropnodes = node.recursiveGlob(\"*\", filter=hou.nodeTypeFilter.Rop)\n else:\n # A single node\n ropnodes = [node]\n\n data = lib.read(node)\n\n # Don't use the explicit \"colorbleed.usd.layered\" family for publishing\n # instead use the \"colorbleed.usd\" family to integrate.\n data[\"publishFamilies\"] = [\"colorbleed.usd\"]\n\n # For now group ALL of them into USD Layer subset group\n # Allow this subset to be grouped into a USD Layer on creation\n data[\"subsetGroup\"] = \"USD Layer\"\n\n instances = list()\n dependencies = []\n for ropnode in ropnodes:\n\n # Create a dependency instance per ROP Node.\n lopoutput = ropnode.evalParm(\"lopoutput\")\n dependency_save_data = self.get_save_data(lopoutput)\n dependency = context.create_instance(dependency_save_data[\"name\"])\n dependency.append(ropnode)\n dependency.data.update(data)\n dependency.data.update(dependency_save_data)\n dependency.data[\"family\"] = \"colorbleed.usd.dependency\"\n dependency.data[\"optional\"] = False\n dependencies.append(dependency)\n\n # Hide the dependency instance from the context\n context.pop()\n\n # Get all configured layers for this USD ROP node\n # and create a Pyblish instance for each one\n layers = hou_usdlib.get_configured_save_layers(ropnode)\n for layer in layers:\n save_path = hou_usdlib.get_layer_save_path(layer)\n save_data = self.get_save_data(save_path)\n if not save_data:\n continue\n self.log.info(save_path)\n\n instance = context.create_instance(save_data[\"name\"])\n instance[:] = [node]\n\n # Set the instance data\n instance.data.update(data)\n instance.data.update(save_data)\n instance.data[\"usdLayer\"] = layer\n\n instances.append(instance)\n\n # Store the collected ROP node dependencies\n self.log.debug(\"Collected dependencies: %s\" % (dependencies,))\n for instance in instances:\n instance.data[\"publishDependencies\"] = dependencies\n\n def get_save_data(self, save_path):\n\n # Resolve Avalon URI\n uri_data = usdlib.parse_avalon_uri(save_path)\n if not uri_data:\n self.log.warning(\"Non Avalon URI Layer Path: %s\" % save_path)\n return {}\n\n # Collect asset + subset from URI\n name = \"{subset} ({asset})\".format(**uri_data)\n fname = \"{asset}_{subset}.{ext}\".format(**uri_data)\n\n data = dict(uri_data)\n data[\"usdSavePath\"] = save_path\n data[\"usdFilename\"] = fname\n data[\"name\"] = name\n return data\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/collect_karma_rop.py",
"content": "import re\nimport os\n\nimport hou\nimport pyblish.api\n\nfrom openpype.hosts.houdini.api.lib import (\n evalParmNoFrame,\n get_color_management_preferences\n)\nfrom openpype.hosts.houdini.api import (\n colorspace\n)\n\n\nclass CollectKarmaROPRenderProducts(pyblish.api.InstancePlugin):\n \"\"\"Collect Karma Render Products\n\n Collects the instance.data[\"files\"] for the multipart render product.\n\n Provides:\n instance -> files\n\n \"\"\"\n\n label = \"Karma ROP Render Products\"\n # This specific order value is used so that\n # this plugin runs after CollectFrames\n order = pyblish.api.CollectorOrder + 0.11\n hosts = [\"houdini\"]\n families = [\"karma_rop\"]\n\n def process(self, instance):\n\n rop = hou.node(instance.data.get(\"instance_node\"))\n\n # Collect chunkSize\n chunk_size_parm = rop.parm(\"chunkSize\")\n if chunk_size_parm:\n chunk_size = int(chunk_size_parm.eval())\n instance.data[\"chunkSize\"] = chunk_size\n self.log.debug(\"Chunk Size: %s\" % chunk_size)\n\n default_prefix = evalParmNoFrame(rop, \"picture\")\n render_products = []\n\n # Default beauty AOV\n beauty_product = self.get_render_product_name(\n prefix=default_prefix, suffix=None\n )\n render_products.append(beauty_product)\n\n files_by_aov = {\n \"beauty\": self.generate_expected_files(instance,\n beauty_product)\n }\n\n filenames = list(render_products)\n instance.data[\"files\"] = filenames\n instance.data[\"renderProducts\"] = colorspace.ARenderProduct()\n\n for product in render_products:\n self.log.debug(\"Found render product: %s\" % product)\n\n if \"expectedFiles\" not in instance.data:\n instance.data[\"expectedFiles\"] = list()\n instance.data[\"expectedFiles\"].append(files_by_aov)\n\n # update the colorspace data\n colorspace_data = get_color_management_preferences()\n instance.data[\"colorspaceConfig\"] = colorspace_data[\"config\"]\n instance.data[\"colorspaceDisplay\"] = colorspace_data[\"display\"]\n instance.data[\"colorspaceView\"] = colorspace_data[\"view\"]\n\n def get_render_product_name(self, prefix, suffix):\n product_name = prefix\n if suffix:\n # Add \".{suffix}\" before the extension\n prefix_base, ext = os.path.splitext(prefix)\n product_name = \"{}.{}{}\".format(prefix_base, suffix, ext)\n\n return product_name\n\n def generate_expected_files(self, instance, path):\n \"\"\"Create expected files in instance data\"\"\"\n\n dir = os.path.dirname(path)\n file = os.path.basename(path)\n\n if \"#\" in file:\n def replace(match):\n return \"%0{}d\".format(len(match.group()))\n\n file = re.sub(\"#+\", replace, file)\n\n if \"%\" not in file:\n return path\n\n expected_files = []\n start = instance.data[\"frameStartHandle\"]\n end = instance.data[\"frameEndHandle\"]\n\n for i in range(int(start), (int(end) + 1)):\n expected_files.append(\n os.path.join(dir, (file % i)).replace(\"\\\\\", \"/\"))\n\n return expected_files\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/collect_mantra_rop.py",
"content": "import re\nimport os\n\nimport hou\nimport pyblish.api\n\nfrom openpype.hosts.houdini.api.lib import (\n evalParmNoFrame,\n get_color_management_preferences\n)\nfrom openpype.hosts.houdini.api import (\n colorspace\n)\n\n\nclass CollectMantraROPRenderProducts(pyblish.api.InstancePlugin):\n \"\"\"Collect Mantra Render Products\n\n Collects the instance.data[\"files\"] for the render products.\n\n Provides:\n instance -> files\n\n \"\"\"\n\n label = \"Mantra ROP Render Products\"\n # This specific order value is used so that\n # this plugin runs after CollectFrames\n order = pyblish.api.CollectorOrder + 0.11\n hosts = [\"houdini\"]\n families = [\"mantra_rop\"]\n\n def process(self, instance):\n\n rop = hou.node(instance.data.get(\"instance_node\"))\n\n # Collect chunkSize\n chunk_size_parm = rop.parm(\"chunkSize\")\n if chunk_size_parm:\n chunk_size = int(chunk_size_parm.eval())\n instance.data[\"chunkSize\"] = chunk_size\n self.log.debug(\"Chunk Size: %s\" % chunk_size)\n\n default_prefix = evalParmNoFrame(rop, \"vm_picture\")\n render_products = []\n\n # Store whether we are splitting the render job (export + render)\n split_render = bool(rop.parm(\"soho_outputmode\").eval())\n instance.data[\"splitRender\"] = split_render\n export_prefix = None\n export_products = []\n if split_render:\n export_prefix = evalParmNoFrame(\n rop, \"soho_diskfile\", pad_character=\"0\"\n )\n beauty_export_product = self.get_render_product_name(\n prefix=export_prefix,\n suffix=None)\n export_products.append(beauty_export_product)\n self.log.debug(\n \"Found export product: {}\".format(beauty_export_product)\n )\n instance.data[\"ifdFile\"] = beauty_export_product\n instance.data[\"exportFiles\"] = list(export_products)\n\n # Default beauty AOV\n beauty_product = self.get_render_product_name(\n prefix=default_prefix, suffix=None\n )\n render_products.append(beauty_product)\n\n files_by_aov = {\n \"beauty\": self.generate_expected_files(instance,\n beauty_product)\n }\n\n aov_numbers = rop.evalParm(\"vm_numaux\")\n if aov_numbers > 0:\n # get the filenames of the AOVs\n for i in range(1, aov_numbers + 1):\n var = rop.evalParm(\"vm_variable_plane%d\" % i)\n if var:\n aov_name = \"vm_filename_plane%d\" % i\n aov_boolean = \"vm_usefile_plane%d\" % i\n aov_enabled = rop.evalParm(aov_boolean)\n has_aov_path = rop.evalParm(aov_name)\n if has_aov_path and aov_enabled == 1:\n aov_prefix = evalParmNoFrame(rop, aov_name)\n aov_product = self.get_render_product_name(\n prefix=aov_prefix, suffix=None\n )\n render_products.append(aov_product)\n\n files_by_aov[var] = self.generate_expected_files(instance, aov_product) # noqa\n\n for product in render_products:\n self.log.debug(\"Found render product: %s\" % product)\n\n filenames = list(render_products)\n instance.data[\"files\"] = filenames\n instance.data[\"renderProducts\"] = colorspace.ARenderProduct()\n\n # For now by default do NOT try to publish the rendered output\n instance.data[\"publishJobState\"] = \"Suspended\"\n instance.data[\"attachTo\"] = [] # stub required data\n\n if \"expectedFiles\" not in instance.data:\n instance.data[\"expectedFiles\"] = list()\n instance.data[\"expectedFiles\"].append(files_by_aov)\n\n # update the colorspace data\n colorspace_data = get_color_management_preferences()\n instance.data[\"colorspaceConfig\"] = colorspace_data[\"config\"]\n instance.data[\"colorspaceDisplay\"] = colorspace_data[\"display\"]\n instance.data[\"colorspaceView\"] = colorspace_data[\"view\"]\n\n def get_render_product_name(self, prefix, suffix):\n product_name = prefix\n if suffix:\n # Add \".{suffix}\" before the extension\n prefix_base, ext = os.path.splitext(prefix)\n product_name = prefix_base + \".\" + suffix + ext\n\n return product_name\n\n def generate_expected_files(self, instance, path):\n \"\"\"Create expected files in instance data\"\"\"\n\n dir = os.path.dirname(path)\n file = os.path.basename(path)\n\n if \"#\" in file:\n def replace(match):\n return \"%0{}d\".format(len(match.group()))\n\n file = re.sub(\"#+\", replace, file)\n\n if \"%\" not in file:\n return path\n\n expected_files = []\n start = instance.data[\"frameStartHandle\"]\n end = instance.data[\"frameEndHandle\"]\n\n for i in range(int(start), (int(end) + 1)):\n expected_files.append(\n os.path.join(dir, (file % i)).replace(\"\\\\\", \"/\"))\n\n return expected_files\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/collect_output_node.py",
"content": "import pyblish.api\n\nfrom openpype.pipeline.publish import KnownPublishError\n\n\nclass CollectOutputSOPPath(pyblish.api.InstancePlugin):\n \"\"\"Collect the out node's SOP/COP Path value.\"\"\"\n\n order = pyblish.api.CollectorOrder\n families = [\n \"pointcache\",\n \"camera\",\n \"vdbcache\",\n \"imagesequence\",\n \"usd\",\n \"usdrender\",\n \"redshiftproxy\",\n \"staticMesh\"\n ]\n\n hosts = [\"houdini\"]\n label = \"Collect Output Node Path\"\n\n def process(self, instance):\n\n import hou\n\n node = hou.node(instance.data[\"instance_node\"])\n\n # Get sop path\n node_type = node.type().name()\n if node_type == \"geometry\":\n out_node = node.parm(\"soppath\").evalAsNode()\n\n elif node_type == \"alembic\":\n\n # Alembic can switch between using SOP Path or object\n if node.parm(\"use_sop_path\").eval():\n out_node = node.parm(\"sop_path\").evalAsNode()\n else:\n root = node.parm(\"root\").eval()\n objects = node.parm(\"objects\").eval()\n path = root + \"/\" + objects\n out_node = hou.node(path)\n\n elif node_type == \"comp\":\n out_node = node.parm(\"coppath\").evalAsNode()\n\n elif node_type == \"usd\" or node_type == \"usdrender\":\n out_node = node.parm(\"loppath\").evalAsNode()\n\n elif node_type == \"usd_rop\" or node_type == \"usdrender_rop\":\n # Inside Solaris e.g. /stage (not in ROP context)\n # When incoming connection is present it takes it directly\n inputs = node.inputs()\n if inputs:\n out_node = inputs[0]\n else:\n out_node = node.parm(\"loppath\").evalAsNode()\n\n elif node_type == \"Redshift_Proxy_Output\":\n out_node = node.parm(\"RS_archive_sopPath\").evalAsNode()\n\n elif node_type == \"filmboxfbx\":\n out_node = node.parm(\"startnode\").evalAsNode()\n\n else:\n raise KnownPublishError(\n \"ROP node type '{}' is not supported.\".format(node_type)\n )\n\n if not out_node:\n self.log.warning(\"No output node collected.\")\n return\n\n self.log.debug(\"Output node: %s\" % out_node.path())\n instance.data[\"output_node\"] = out_node\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/collect_pointcache_type.py",
"content": "\"\"\"Collector for pointcache types.\n\nThis will add additional family to pointcache instance based on\nthe creator_identifier parameter.\n\"\"\"\nimport pyblish.api\n\n\nclass CollectPointcacheType(pyblish.api.InstancePlugin):\n \"\"\"Collect data type for pointcache instance.\"\"\"\n\n order = pyblish.api.CollectorOrder\n hosts = [\"houdini\"]\n families = [\"pointcache\"]\n label = \"Collect type of pointcache\"\n\n def process(self, instance):\n if instance.data[\"creator_identifier\"] == \"io.openpype.creators.houdini.bgeo\": # noqa: E501\n instance.data[\"families\"] += [\"bgeo\"]\n elif instance.data[\"creator_identifier\"] == \"io.openpype.creators.houdini.pointcache\": # noqa: E501\n instance.data[\"families\"] += [\"abc\"]\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/collect_redshift_rop.py",
"content": "import re\nimport os\n\nimport hou\nimport pyblish.api\n\nfrom openpype.hosts.houdini.api.lib import (\n evalParmNoFrame,\n get_color_management_preferences\n)\nfrom openpype.hosts.houdini.api import (\n colorspace\n)\n\n\nclass CollectRedshiftROPRenderProducts(pyblish.api.InstancePlugin):\n \"\"\"Collect USD Render Products\n\n Collects the instance.data[\"files\"] for the render products.\n\n Provides:\n instance -> files\n\n \"\"\"\n\n label = \"Redshift ROP Render Products\"\n # This specific order value is used so that\n # this plugin runs after CollectFrames\n order = pyblish.api.CollectorOrder + 0.11\n hosts = [\"houdini\"]\n families = [\"redshift_rop\"]\n\n def process(self, instance):\n rop = hou.node(instance.data.get(\"instance_node\"))\n\n # Collect chunkSize\n chunk_size_parm = rop.parm(\"chunkSize\")\n if chunk_size_parm:\n chunk_size = int(chunk_size_parm.eval())\n instance.data[\"chunkSize\"] = chunk_size\n self.log.debug(\"Chunk Size: %s\" % chunk_size)\n\n default_prefix = evalParmNoFrame(rop, \"RS_outputFileNamePrefix\")\n beauty_suffix = rop.evalParm(\"RS_outputBeautyAOVSuffix\")\n # Store whether we are splitting the render job (export + render)\n split_render = bool(rop.parm(\"RS_archive_enable\").eval())\n instance.data[\"splitRender\"] = split_render\n export_products = []\n if split_render:\n export_prefix = evalParmNoFrame(\n rop, \"RS_archive_file\", pad_character=\"0\"\n )\n beauty_export_product = self.get_render_product_name(\n prefix=export_prefix,\n suffix=None)\n export_products.append(beauty_export_product)\n self.log.debug(\n \"Found export product: {}\".format(beauty_export_product)\n )\n instance.data[\"ifdFile\"] = beauty_export_product\n instance.data[\"exportFiles\"] = list(export_products)\n\n # Default beauty AOV\n beauty_product = self.get_render_product_name(\n prefix=default_prefix, suffix=beauty_suffix\n )\n render_products = [beauty_product]\n files_by_aov = {\n \"_\": self.generate_expected_files(instance,\n beauty_product)}\n\n num_aovs = rop.evalParm(\"RS_aov\")\n for index in range(num_aovs):\n i = index + 1\n\n # Skip disabled AOVs\n if not rop.evalParm(f\"RS_aovEnable_{i}\"):\n continue\n\n aov_suffix = rop.evalParm(f\"RS_aovSuffix_{i}\")\n aov_prefix = evalParmNoFrame(rop, f\"RS_aovCustomPrefix_{i}\")\n if not aov_prefix:\n aov_prefix = default_prefix\n\n aov_product = self.get_render_product_name(aov_prefix, aov_suffix)\n render_products.append(aov_product)\n\n files_by_aov[aov_suffix] = self.generate_expected_files(instance,\n aov_product) # noqa\n\n for product in render_products:\n self.log.debug(\"Found render product: %s\" % product)\n\n filenames = list(render_products)\n instance.data[\"files\"] = filenames\n instance.data[\"renderProducts\"] = colorspace.ARenderProduct()\n\n # For now by default do NOT try to publish the rendered output\n instance.data[\"publishJobState\"] = \"Suspended\"\n instance.data[\"attachTo\"] = [] # stub required data\n\n if \"expectedFiles\" not in instance.data:\n instance.data[\"expectedFiles\"] = []\n instance.data[\"expectedFiles\"].append(files_by_aov)\n\n # update the colorspace data\n colorspace_data = get_color_management_preferences()\n instance.data[\"colorspaceConfig\"] = colorspace_data[\"config\"]\n instance.data[\"colorspaceDisplay\"] = colorspace_data[\"display\"]\n instance.data[\"colorspaceView\"] = colorspace_data[\"view\"]\n\n def get_render_product_name(self, prefix, suffix):\n \"\"\"Return the output filename using the AOV prefix and suffix\"\"\"\n\n # When AOV is explicitly defined in prefix we just swap it out\n # directly with the AOV suffix to embed it.\n # Note: ${AOV} seems to be evaluated in the parameter as %AOV%\n has_aov_in_prefix = \"%AOV%\" in prefix\n if has_aov_in_prefix:\n # It seems that when some special separator characters are present\n # before the %AOV% token that Redshift will secretly remove it if\n # there is no suffix for the current product, for example:\n # foo_%AOV% -> foo.exr\n pattern = \"%AOV%\" if suffix else \"[._-]?%AOV%\"\n product_name = re.sub(pattern, suffix, prefix, flags=re.IGNORECASE)\n else:\n if suffix:\n # Add \".{suffix}\" before the extension\n prefix_base, ext = os.path.splitext(prefix)\n product_name = prefix_base + \".\" + suffix + ext\n else:\n product_name = prefix\n\n return product_name\n\n def generate_expected_files(self, instance, path):\n \"\"\"Create expected files in instance data\"\"\"\n\n dir = os.path.dirname(path)\n file = os.path.basename(path)\n\n if \"#\" in file:\n def replace(match):\n return \"%0{}d\".format(len(match.group()))\n\n file = re.sub(\"#+\", replace, file)\n\n if \"%\" not in file:\n return path\n\n expected_files = []\n start = instance.data[\"frameStartHandle\"]\n end = instance.data[\"frameEndHandle\"]\n\n for i in range(int(start), (int(end) + 1)):\n expected_files.append(\n os.path.join(dir, (file % i)).replace(\"\\\\\", \"/\"))\n\n return expected_files\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/collect_remote_publish.py",
"content": "import pyblish.api\n\nimport hou\nfrom openpype.pipeline.publish import RepairAction\nfrom openpype.hosts.houdini.api import lib\n\n\nclass CollectRemotePublishSettings(pyblish.api.ContextPlugin):\n \"\"\"Collect custom settings of the Remote Publish node.\"\"\"\n\n order = pyblish.api.CollectorOrder\n families = [\"*\"]\n hosts = [\"houdini\"]\n targets = [\"deadline\"]\n label = \"Remote Publish Submission Settings\"\n actions = [RepairAction]\n\n def process(self, context):\n\n node = hou.node(\"/out/REMOTE_PUBLISH\")\n if not node:\n return\n\n attributes = lib.read(node)\n\n # Debug the settings we have collected\n for key, value in sorted(attributes.items()):\n self.log.debug(\"Collected %s: %s\" % (key, value))\n\n context.data.update(attributes)\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/collect_render_products.py",
"content": "import re\nimport os\n\nimport hou\nimport pxr.UsdRender\n\nimport pyblish.api\n\n\ndef get_var_changed(variable=None):\n \"\"\"Return changed variables and operators that use it.\n\n Note: `varchange` hscript states that it forces a recook of the nodes\n that use Variables. That was tested in Houdini\n 18.0.391.\n\n Args:\n variable (str, Optional): A specific variable to query the operators\n for. When None is provided it will return all variables that have\n had recent changes and require a recook. Defaults to None.\n\n Returns:\n dict: Variable that changed with the operators that use it.\n\n \"\"\"\n cmd = \"varchange -V\"\n if variable:\n cmd += \" {0}\".format(variable)\n output, _ = hou.hscript(cmd)\n\n changed = {}\n for line in output.split(\"Variable: \"):\n if not line.strip():\n continue\n\n split = line.split()\n var = split[0]\n operators = split[1:]\n changed[var] = operators\n\n return changed\n\n\nclass CollectRenderProducts(pyblish.api.InstancePlugin):\n \"\"\"Collect USD Render Products.\"\"\"\n\n label = \"Collect Render Products\"\n order = pyblish.api.CollectorOrder + 0.4\n hosts = [\"houdini\"]\n families = [\"usdrender\"]\n\n def process(self, instance):\n\n node = instance.data.get(\"output_node\")\n if not node:\n rop_path = instance.data[\"instance_node\"].path()\n raise RuntimeError(\n \"No output node found. Make sure to connect an \"\n \"input to the USD ROP: %s\" % rop_path\n )\n\n # Workaround Houdini 18.0.391 bug where $HIPNAME doesn't automatically\n # update after scene save.\n if hou.applicationVersion() == (18, 0, 391):\n self.log.debug(\n \"Checking for recook to workaround \" \"$HIPNAME refresh bug...\"\n )\n changed = get_var_changed(\"HIPNAME\").get(\"HIPNAME\")\n if changed:\n self.log.debug(\"Recooking for $HIPNAME refresh bug...\")\n for operator in changed:\n hou.node(operator).cook(force=True)\n\n # Make sure to recook any 'cache' nodes in the history chain\n chain = [node]\n chain.extend(node.inputAncestors())\n for input_node in chain:\n if input_node.type().name() == \"cache\":\n input_node.cook(force=True)\n\n stage = node.stage()\n\n filenames = []\n for prim in stage.Traverse():\n\n if not prim.IsA(pxr.UsdRender.Product):\n continue\n\n # Get Render Product Name\n product = pxr.UsdRender.Product(prim)\n\n # We force taking it from any random time sample as opposed to\n # \"default\" that the USD Api falls back to since that won't return\n # time sampled values if they were set per time sample.\n name = product.GetProductNameAttr().Get(time=0)\n dirname = os.path.dirname(name)\n basename = os.path.basename(name)\n\n dollarf_regex = r\"(\\$F([0-9]?))\"\n frame_regex = r\"^(.+\\.)([0-9]+)(\\.[a-zA-Z]+)$\"\n if re.match(dollarf_regex, basename):\n # TODO: Confirm this actually is allowed USD stages and HUSK\n # Substitute $F\n def replace(match):\n \"\"\"Replace $F4 with padded #.\"\"\"\n padding = int(match.group(2)) if match.group(2) else 1\n return \"#\" * padding\n\n filename_base = re.sub(dollarf_regex, replace, basename)\n filename = os.path.join(dirname, filename_base)\n else:\n # Substitute basename.0001.ext\n def replace(match):\n prefix, frame, ext = match.groups()\n padding = \"#\" * len(frame)\n return prefix + padding + ext\n\n filename_base = re.sub(frame_regex, replace, basename)\n filename = os.path.join(dirname, filename_base)\n filename = filename.replace(\"\\\\\", \"/\")\n\n assert \"#\" in filename, (\n \"Couldn't resolve render product name \"\n \"with frame number: %s\" % name\n )\n\n filenames.append(filename)\n\n prim_path = str(prim.GetPath())\n self.log.info(\"Collected %s name: %s\" % (prim_path, filename))\n\n # Filenames for Deadline\n instance.data[\"files\"] = filenames\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/collect_review_data.py",
"content": "import hou\nimport pyblish.api\n\n\nclass CollectHoudiniReviewData(pyblish.api.InstancePlugin):\n \"\"\"Collect Review Data.\"\"\"\n\n label = \"Collect Review Data\"\n # This specific order value is used so that\n # this plugin runs after CollectRopFrameRange\n order = pyblish.api.CollectorOrder + 0.1\n hosts = [\"houdini\"]\n families = [\"review\"]\n\n def process(self, instance):\n\n # This fixes the burnin having the incorrect start/end timestamps\n # because without this it would take it from the context instead\n # which isn't the actual frame range that this instance renders.\n instance.data[\"handleStart\"] = 0\n instance.data[\"handleEnd\"] = 0\n instance.data[\"fps\"] = instance.context.data[\"fps\"]\n\n # Enable ftrack functionality\n instance.data.setdefault(\"families\", []).append('ftrack')\n\n # Get the camera from the rop node to collect the focal length\n ropnode_path = instance.data[\"instance_node\"]\n ropnode = hou.node(ropnode_path)\n\n camera_path = ropnode.parm(\"camera\").eval()\n camera_node = hou.node(camera_path)\n if not camera_node:\n self.log.warning(\"No valid camera node found on review node: \"\n \"{}\".format(camera_path))\n return\n\n # Collect focal length.\n focal_length_parm = camera_node.parm(\"focal\")\n if not focal_length_parm:\n self.log.warning(\"No 'focal' (focal length) parameter found on \"\n \"camera: {}\".format(camera_path))\n return\n\n if focal_length_parm.isTimeDependent():\n start = instance.data[\"frameStartHandle\"]\n end = instance.data[\"frameEndHandle\"] + 1\n focal_length = [\n focal_length_parm.evalAsFloatAtFrame(t)\n for t in range(int(start), int(end))\n ]\n else:\n focal_length = focal_length_parm.evalAsFloat()\n\n # Store focal length in `burninDataMembers`\n burnin_members = instance.data.setdefault(\"burninDataMembers\", {})\n burnin_members[\"focalLength\"] = focal_length\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/collect_rop_frame_range.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Collector plugin for frames data on ROP instances.\"\"\"\nimport hou # noqa\nimport pyblish.api\nfrom openpype.hosts.houdini.api import lib\n\n\nclass CollectRopFrameRange(pyblish.api.InstancePlugin):\n \"\"\"Collect all frames which would be saved from the ROP nodes\"\"\"\n\n hosts = [\"houdini\"]\n order = pyblish.api.CollectorOrder\n label = \"Collect RopNode Frame Range\"\n\n def process(self, instance):\n\n node_path = instance.data.get(\"instance_node\")\n if node_path is None:\n # Instance without instance node like a workfile instance\n self.log.debug(\n \"No instance node found for instance: {}\".format(instance)\n )\n return\n\n ropnode = hou.node(node_path)\n frame_data = lib.get_frame_data(\n ropnode, self.log\n )\n\n if not frame_data:\n return\n\n # Log debug message about the collected frame range\n self.log.debug(\n \"Collected frame_data: {}\".format(frame_data)\n )\n\n instance.data.update(frame_data)\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/collect_staticmesh_type.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Collector for staticMesh types. \"\"\"\n\nimport pyblish.api\n\n\nclass CollectStaticMeshType(pyblish.api.InstancePlugin):\n \"\"\"Collect data type for fbx instance.\"\"\"\n\n hosts = [\"houdini\"]\n families = [\"staticMesh\"]\n label = \"Collect type of staticMesh\"\n\n order = pyblish.api.CollectorOrder\n\n def process(self, instance):\n\n if instance.data[\"creator_identifier\"] == \"io.openpype.creators.houdini.staticmesh.fbx\": # noqa: E501\n # Marking this instance as FBX triggers the FBX extractor.\n instance.data[\"families\"] += [\"fbx\"]\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/collect_usd_bootstrap.py",
"content": "import pyblish.api\n\nfrom openpype.client import (\n get_subset_by_name,\n get_asset_by_name,\n get_asset_name_identifier,\n)\nimport openpype.lib.usdlib as usdlib\n\n\nclass CollectUsdBootstrap(pyblish.api.InstancePlugin):\n \"\"\"Collect special Asset/Shot bootstrap instances if those are needed.\n\n Some specific subsets are intended to be part of the default structure\n of an \"Asset\" or \"Shot\" in our USD pipeline. For example, for an Asset\n we layer a Model and Shade USD file over each other and expose that in\n a Asset USD file, ready to use.\n\n On the first publish of any of the components of a Asset or Shot the\n missing pieces are bootstrapped and generated in the pipeline too. This\n means that on the very first publish of your model the Asset USD file\n will exist too.\n\n \"\"\"\n\n order = pyblish.api.CollectorOrder + 0.35\n label = \"Collect USD Bootstrap\"\n hosts = [\"houdini\"]\n families = [\"usd\", \"usd.layered\"]\n\n def process(self, instance):\n\n # Detect whether the current subset is a subset in a pipeline\n def get_bootstrap(instance):\n instance_subset = instance.data[\"subset\"]\n for name, layers in usdlib.PIPELINE.items():\n if instance_subset in set(layers):\n return name # e.g. \"asset\"\n break\n else:\n return\n\n bootstrap = get_bootstrap(instance)\n if bootstrap:\n self.add_bootstrap(instance, bootstrap)\n\n # Check if any of the dependencies requires a bootstrap\n for dependency in instance.data.get(\"publishDependencies\", list()):\n bootstrap = get_bootstrap(dependency)\n if bootstrap:\n self.add_bootstrap(dependency, bootstrap)\n\n def add_bootstrap(self, instance, bootstrap):\n\n self.log.debug(\"Add bootstrap for: %s\" % bootstrap)\n\n project_name = instance.context.data[\"projectName\"]\n asset_name = instance.data[\"asset\"]\n asset_doc = get_asset_by_name(project_name, asset_name)\n assert asset_doc, \"Asset must exist: %s\" % asset_name\n\n # Check which are not about to be created and don't exist yet\n required = {\"shot\": [\"usdShot\"], \"asset\": [\"usdAsset\"]}.get(bootstrap)\n\n require_all_layers = instance.data.get(\"requireAllLayers\", False)\n if require_all_layers:\n # USD files load fine in usdview and Houdini even when layered or\n # referenced files do not exist. So by default we don't require\n # the layers to exist.\n layers = usdlib.PIPELINE.get(bootstrap)\n if layers:\n required += list(layers)\n\n self.log.debug(\"Checking required bootstrap: %s\" % required)\n for subset_name in required:\n if self._subset_exists(\n project_name, instance, subset_name, asset_doc\n ):\n continue\n\n self.log.debug(\n \"Creating {0} USD bootstrap: {1} {2}\".format(\n bootstrap, asset_name, subset_name\n )\n )\n\n new = instance.context.create_instance(subset_name)\n new.data[\"subset\"] = subset_name\n new.data[\"label\"] = \"{0} ({1})\".format(subset_name, asset_name)\n new.data[\"family\"] = \"usd.bootstrap\"\n new.data[\"comment\"] = \"Automated bootstrap USD file.\"\n new.data[\"publishFamilies\"] = [\"usd\"]\n\n # Do not allow the user to toggle this instance\n new.data[\"optional\"] = False\n\n # Copy some data from the instance for which we bootstrap\n for key in [\"asset\"]:\n new.data[key] = instance.data[key]\n\n def _subset_exists(self, project_name, instance, subset_name, asset_doc):\n \"\"\"Return whether subset exists in current context or in database.\"\"\"\n # Allow it to be created during this publish session\n context = instance.context\n\n asset_doc_name = get_asset_name_identifier(asset_doc)\n for inst in context:\n if (\n inst.data[\"subset\"] == subset_name\n and inst.data[\"asset\"] == asset_doc_name\n ):\n return True\n\n # Or, if they already exist in the database we can\n # skip them too.\n if get_subset_by_name(\n project_name, subset_name, asset_doc[\"_id\"], fields=[\"_id\"]\n ):\n return True\n return False\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/collect_usd_layers.py",
"content": "import os\n\nimport pyblish.api\nimport openpype.hosts.houdini.api.usd as usdlib\n\nimport hou\n\n\nclass CollectUsdLayers(pyblish.api.InstancePlugin):\n \"\"\"Collect the USD Layers that have configured save paths.\"\"\"\n\n order = pyblish.api.CollectorOrder + 0.35\n label = \"Collect USD Layers\"\n hosts = [\"houdini\"]\n families = [\"usd\"]\n\n def process(self, instance):\n\n output = instance.data.get(\"output_node\")\n if not output:\n self.log.debug(\"No output node found..\")\n return\n\n rop_node = hou.node(instance.data[\"instance_node\"])\n\n save_layers = []\n for layer in usdlib.get_configured_save_layers(rop_node):\n\n info = layer.rootPrims.get(\"HoudiniLayerInfo\")\n save_path = info.customData.get(\"HoudiniSavePath\")\n creator = info.customData.get(\"HoudiniCreatorNode\")\n\n self.log.debug(\"Found configured save path: \"\n \"%s -> %s\" % (layer, save_path))\n\n # Log node that configured this save path\n if creator:\n self.log.debug(\"Created by: %s\" % creator)\n\n save_layers.append((layer, save_path))\n\n # Store on the instance\n instance.data[\"usdConfiguredSavePaths\"] = save_layers\n\n # Create configured layer instances so User can disable updating\n # specific configured layers for publishing.\n context = instance.context\n for layer, save_path in save_layers:\n name = os.path.basename(save_path)\n label = \"{0} -> {1}\".format(instance.data[\"name\"], name)\n layer_inst = context.create_instance(name)\n\n family = \"usdlayer\"\n layer_inst.data[\"family\"] = family\n layer_inst.data[\"families\"] = [family]\n layer_inst.data[\"subset\"] = \"__stub__\"\n layer_inst.data[\"label\"] = label\n layer_inst.data[\"asset\"] = instance.data[\"asset\"]\n layer_inst.data[\"instance_node\"] = instance.data[\"instance_node\"]\n # include same USD ROP\n layer_inst.append(rop_node)\n # include layer data\n layer_inst.append((layer, save_path))\n\n # Allow this subset to be grouped into a USD Layer on creation\n layer_inst.data[\"subsetGroup\"] = \"USD Layer\"\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/collect_vray_rop.py",
"content": "import re\nimport os\n\nimport hou\nimport pyblish.api\n\nfrom openpype.hosts.houdini.api.lib import (\n evalParmNoFrame,\n get_color_management_preferences\n)\nfrom openpype.hosts.houdini.api import (\n colorspace\n)\n\n\nclass CollectVrayROPRenderProducts(pyblish.api.InstancePlugin):\n \"\"\"Collect Vray Render Products\n\n Collects the instance.data[\"files\"] for the render products.\n\n Provides:\n instance -> files\n\n \"\"\"\n\n label = \"VRay ROP Render Products\"\n # This specific order value is used so that\n # this plugin runs after CollectFrames\n order = pyblish.api.CollectorOrder + 0.11\n hosts = [\"houdini\"]\n families = [\"vray_rop\"]\n\n def process(self, instance):\n\n rop = hou.node(instance.data.get(\"instance_node\"))\n\n # Collect chunkSize\n chunk_size_parm = rop.parm(\"chunkSize\")\n if chunk_size_parm:\n chunk_size = int(chunk_size_parm.eval())\n instance.data[\"chunkSize\"] = chunk_size\n self.log.debug(\"Chunk Size: %s\" % chunk_size)\n\n default_prefix = evalParmNoFrame(rop, \"SettingsOutput_img_file_path\")\n render_products = []\n # TODO: add render elements if render element\n\n # Store whether we are splitting the render job in an export + render\n split_render = rop.parm(\"render_export_mode\").eval() == \"2\"\n instance.data[\"splitRender\"] = split_render\n export_prefix = None\n export_products = []\n if split_render:\n export_prefix = evalParmNoFrame(\n rop, \"render_export_filepath\", pad_character=\"0\"\n )\n beauty_export_product = self.get_render_product_name(\n prefix=export_prefix,\n suffix=None)\n export_products.append(beauty_export_product)\n self.log.debug(\n \"Found export product: {}\".format(beauty_export_product)\n )\n instance.data[\"ifdFile\"] = beauty_export_product\n instance.data[\"exportFiles\"] = list(export_products)\n\n beauty_product = self.get_render_product_name(default_prefix)\n render_products.append(beauty_product)\n files_by_aov = {\n \"\": self.generate_expected_files(instance,\n beauty_product)}\n\n if instance.data.get(\"RenderElement\", True):\n render_element = self.get_render_element_name(rop, default_prefix)\n if render_element:\n for aov, renderpass in render_element.items():\n render_products.append(renderpass)\n files_by_aov[aov] = self.generate_expected_files(\n instance, renderpass)\n\n\n for product in render_products:\n self.log.debug(\"Found render product: %s\" % product)\n filenames = list(render_products)\n instance.data[\"files\"] = filenames\n instance.data[\"renderProducts\"] = colorspace.ARenderProduct()\n\n # For now by default do NOT try to publish the rendered output\n instance.data[\"publishJobState\"] = \"Suspended\"\n instance.data[\"attachTo\"] = [] # stub required data\n\n if \"expectedFiles\" not in instance.data:\n instance.data[\"expectedFiles\"] = list()\n instance.data[\"expectedFiles\"].append(files_by_aov)\n self.log.debug(\"expectedFiles:{}\".format(files_by_aov))\n\n # update the colorspace data\n colorspace_data = get_color_management_preferences()\n instance.data[\"colorspaceConfig\"] = colorspace_data[\"config\"]\n instance.data[\"colorspaceDisplay\"] = colorspace_data[\"display\"]\n instance.data[\"colorspaceView\"] = colorspace_data[\"view\"]\n\n def get_render_product_name(self, prefix, suffix=\"\"):\n \"\"\"Return the beauty output filename if render element enabled\n \"\"\"\n # Remove aov suffix from the product: `prefix.aov_suffix` -> `prefix`\n aov_parm = \".{}\".format(suffix)\n return prefix.replace(aov_parm, \"\")\n\n def get_render_element_name(self, node, prefix, suffix=\"\"):\n \"\"\"Return the output filename using the AOV prefix and suffix\n \"\"\"\n render_element_dict = {}\n # need a rewrite\n re_path = node.evalParm(\"render_network_render_channels\")\n if re_path:\n node_children = hou.node(re_path).children()\n for element in node_children:\n if element.shaderName() != \"vray:SettingsRenderChannels\":\n aov = str(element)\n render_product = prefix.replace(suffix, aov)\n render_element_dict[aov] = render_product\n return render_element_dict\n\n def generate_expected_files(self, instance, path):\n \"\"\"Create expected files in instance data\"\"\"\n\n dir = os.path.dirname(path)\n file = os.path.basename(path)\n\n if \"#\" in file:\n def replace(match):\n return \"%0{}d\".format(len(match.group()))\n\n file = re.sub(\"#+\", replace, file)\n\n if \"%\" not in file:\n return path\n\n expected_files = []\n start = instance.data[\"frameStartHandle\"]\n end = instance.data[\"frameEndHandle\"]\n\n for i in range(int(start), (int(end) + 1)):\n expected_files.append(\n os.path.join(dir, (file % i)).replace(\"\\\\\", \"/\"))\n\n return expected_files\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/collect_workfile.py",
"content": "import os\n\nimport pyblish.api\n\n\nclass CollectWorkfile(pyblish.api.InstancePlugin):\n \"\"\"Inject workfile representation into instance\"\"\"\n\n order = pyblish.api.CollectorOrder - 0.01\n label = \"Houdini Workfile Data\"\n hosts = [\"houdini\"]\n families = [\"workfile\"]\n\n def process(self, instance):\n\n current_file = instance.context.data[\"currentFile\"]\n folder, file = os.path.split(current_file)\n filename, ext = os.path.splitext(file)\n\n instance.data.update({\n \"setMembers\": [current_file],\n \"frameStart\": instance.context.data['frameStart'],\n \"frameEnd\": instance.context.data['frameEnd'],\n \"handleStart\": instance.context.data['handleStart'],\n \"handleEnd\": instance.context.data['handleEnd']\n })\n\n instance.data['representations'] = [{\n 'name': ext.lstrip(\".\"),\n 'ext': ext.lstrip(\".\"),\n 'files': file,\n \"stagingDir\": folder,\n }]\n\n self.log.debug('Collected workfile instance: {}'.format(file))\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/collect_workscene_fps.py",
"content": "import pyblish.api\nimport hou\n\n\nclass CollectWorksceneFPS(pyblish.api.ContextPlugin):\n \"\"\"Get the FPS of the work scene.\"\"\"\n\n label = \"Workscene FPS\"\n order = pyblish.api.CollectorOrder\n hosts = [\"houdini\"]\n\n def process(self, context):\n fps = hou.fps()\n self.log.info(\"Workscene FPS: %s\" % fps)\n context.data.update({\"fps\": fps})\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/extract_alembic.py",
"content": "import os\n\nimport pyblish.api\n\nfrom openpype.pipeline import publish\nfrom openpype.hosts.houdini.api.lib import render_rop\n\nimport hou\n\n\nclass ExtractAlembic(publish.Extractor):\n\n order = pyblish.api.ExtractorOrder\n label = \"Extract Alembic\"\n hosts = [\"houdini\"]\n families = [\"abc\", \"camera\"]\n targets = [\"local\", \"remote\"]\n\n def process(self, instance):\n if instance.data.get(\"farm\"):\n self.log.debug(\"Should be processed on farm, skipping.\")\n return\n\n ropnode = hou.node(instance.data[\"instance_node\"])\n\n # Get the filename from the filename parameter\n output = ropnode.evalParm(\"filename\")\n staging_dir = os.path.dirname(output)\n instance.data[\"stagingDir\"] = staging_dir\n\n file_name = os.path.basename(output)\n\n # We run the render\n self.log.info(\"Writing alembic '%s' to '%s'\" % (file_name,\n staging_dir))\n\n render_rop(ropnode)\n\n if \"representations\" not in instance.data:\n instance.data[\"representations\"] = []\n\n representation = {\n 'name': 'abc',\n 'ext': 'abc',\n 'files': file_name,\n \"stagingDir\": staging_dir,\n }\n instance.data[\"representations\"].append(representation)\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/extract_ass.py",
"content": "import os\n\nimport pyblish.api\n\nfrom openpype.pipeline import publish\nfrom openpype.hosts.houdini.api.lib import render_rop\n\nimport hou\n\n\nclass ExtractAss(publish.Extractor):\n\n order = pyblish.api.ExtractorOrder + 0.1\n label = \"Extract Ass\"\n families = [\"ass\"]\n hosts = [\"houdini\"]\n targets = [\"local\", \"remote\"]\n\n def process(self, instance):\n if instance.data.get(\"farm\"):\n self.log.debug(\"Should be processed on farm, skipping.\")\n return\n ropnode = hou.node(instance.data[\"instance_node\"])\n\n # Get the filename from the filename parameter\n # `.evalParm(parameter)` will make sure all tokens are resolved\n output = ropnode.evalParm(\"ar_ass_file\")\n staging_dir = os.path.dirname(output)\n instance.data[\"stagingDir\"] = staging_dir\n file_name = os.path.basename(output)\n\n # We run the render\n self.log.info(\"Writing ASS '%s' to '%s'\" % (file_name, staging_dir))\n\n render_rop(ropnode)\n\n # Unfortunately user interrupting the extraction does not raise an\n # error and thus still continues to the integrator. To capture that\n # we make sure all files exist\n files = instance.data[\"frames\"]\n missing = []\n for file_name in files:\n full_path = os.path.normpath(os.path.join(staging_dir, file_name))\n if not os.path.exists(full_path):\n missing.append(full_path)\n\n if missing:\n raise RuntimeError(\"Failed to complete Arnold ass extraction. \"\n \"Missing output files: {}\".format(missing))\n\n if \"representations\" not in instance.data:\n instance.data[\"representations\"] = []\n\n # Allow ass.gz extension as well\n ext = \"ass.gz\" if file_name.endswith(\".ass.gz\") else \"ass\"\n\n representation = {\n 'name': 'ass',\n 'ext': ext,\n \"files\": files,\n \"stagingDir\": staging_dir,\n \"frameStart\": instance.data[\"frameStartHandle\"],\n \"frameEnd\": instance.data[\"frameEndHandle\"],\n }\n instance.data[\"representations\"].append(representation)\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/extract_bgeo.py",
"content": "import os\n\nimport pyblish.api\n\nfrom openpype.pipeline import publish\nfrom openpype.hosts.houdini.api.lib import render_rop\nfrom openpype.hosts.houdini.api import lib\n\nimport hou\n\n\nclass ExtractBGEO(publish.Extractor):\n\n order = pyblish.api.ExtractorOrder\n label = \"Extract BGEO\"\n hosts = [\"houdini\"]\n families = [\"bgeo\"]\n\n def process(self, instance):\n if instance.data.get(\"farm\"):\n self.log.debug(\"Should be processed on farm, skipping.\")\n return\n ropnode = hou.node(instance.data[\"instance_node\"])\n\n # Get the filename from the filename parameter\n output = ropnode.evalParm(\"sopoutput\")\n staging_dir, file_name = os.path.split(output)\n instance.data[\"stagingDir\"] = staging_dir\n\n # We run the render\n self.log.info(\"Writing bgeo files '{}' to '{}'.\".format(\n file_name, staging_dir))\n\n # write files\n render_rop(ropnode)\n\n output = instance.data[\"frames\"]\n\n _, ext = lib.splitext(\n output[0], allowed_multidot_extensions=[\n \".ass.gz\", \".bgeo.sc\", \".bgeo.gz\",\n \".bgeo.lzma\", \".bgeo.bz2\"])\n\n if \"representations\" not in instance.data:\n instance.data[\"representations\"] = []\n\n representation = {\n \"name\": \"bgeo\",\n \"ext\": ext.lstrip(\".\"),\n \"files\": output,\n \"stagingDir\": staging_dir,\n \"frameStart\": instance.data[\"frameStartHandle\"],\n \"frameEnd\": instance.data[\"frameEndHandle\"]\n }\n instance.data[\"representations\"].append(representation)\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/extract_composite.py",
"content": "import os\nimport pyblish.api\n\nfrom openpype.pipeline import publish\nfrom openpype.hosts.houdini.api.lib import render_rop, splitext\n\nimport hou\n\n\nclass ExtractComposite(publish.Extractor):\n\n order = pyblish.api.ExtractorOrder\n label = \"Extract Composite (Image Sequence)\"\n hosts = [\"houdini\"]\n families = [\"imagesequence\"]\n\n def process(self, instance):\n\n ropnode = hou.node(instance.data[\"instance_node\"])\n\n # Get the filename from the copoutput parameter\n # `.evalParm(parameter)` will make sure all tokens are resolved\n output = ropnode.evalParm(\"copoutput\")\n staging_dir = os.path.dirname(output)\n instance.data[\"stagingDir\"] = staging_dir\n file_name = os.path.basename(output)\n\n self.log.info(\"Writing comp '%s' to '%s'\" % (file_name, staging_dir))\n\n render_rop(ropnode)\n\n output = instance.data[\"frames\"]\n _, ext = splitext(output[0], [])\n ext = ext.lstrip(\".\")\n\n if \"representations\" not in instance.data:\n instance.data[\"representations\"] = []\n\n representation = {\n \"name\": ext,\n \"ext\": ext,\n \"files\": output,\n \"stagingDir\": staging_dir,\n \"frameStart\": instance.data[\"frameStartHandle\"],\n \"frameEnd\": instance.data[\"frameEndHandle\"],\n }\n\n from pprint import pformat\n\n self.log.info(pformat(representation))\n\n instance.data[\"representations\"].append(representation)\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/extract_fbx.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Fbx Extractor for houdini. \"\"\"\n\nimport os\nimport pyblish.api\nfrom openpype.pipeline import publish\nfrom openpype.hosts.houdini.api.lib import render_rop\n\nimport hou\n\n\nclass ExtractFBX(publish.Extractor):\n\n label = \"Extract FBX\"\n families = [\"fbx\"]\n hosts = [\"houdini\"]\n\n order = pyblish.api.ExtractorOrder + 0.1\n\n def process(self, instance):\n\n # get rop node\n ropnode = hou.node(instance.data.get(\"instance_node\"))\n output_file = ropnode.evalParm(\"sopoutput\")\n\n # get staging_dir and file_name\n staging_dir = os.path.normpath(os.path.dirname(output_file))\n file_name = os.path.basename(output_file)\n\n # render rop\n self.log.debug(\"Writing FBX '%s' to '%s'\", file_name, staging_dir)\n render_rop(ropnode)\n\n # prepare representation\n representation = {\n \"name\": \"fbx\",\n \"ext\": \"fbx\",\n \"files\": file_name,\n \"stagingDir\": staging_dir\n }\n\n # A single frame may also be rendered without start/end frame.\n if \"frameStartHandle\" in instance.data and \"frameEndHandle\" in instance.data: # noqa\n representation[\"frameStart\"] = instance.data[\"frameStartHandle\"]\n representation[\"frameEnd\"] = instance.data[\"frameEndHandle\"]\n\n # set value type for 'representations' key to list\n if \"representations\" not in instance.data:\n instance.data[\"representations\"] = []\n\n # update instance data\n instance.data[\"stagingDir\"] = staging_dir\n instance.data[\"representations\"].append(representation)\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/extract_hda.py",
"content": "# -*- coding: utf-8 -*-\nimport os\nfrom pprint import pformat\nimport pyblish.api\nfrom openpype.pipeline import publish\nimport hou\n\n\nclass ExtractHDA(publish.Extractor):\n\n order = pyblish.api.ExtractorOrder\n label = \"Extract HDA\"\n hosts = [\"houdini\"]\n families = [\"hda\"]\n\n def process(self, instance):\n self.log.info(pformat(instance.data))\n hda_node = hou.node(instance.data.get(\"instance_node\"))\n hda_def = hda_node.type().definition()\n hda_options = hda_def.options()\n hda_options.setSaveInitialParmsAndContents(True)\n\n next_version = instance.data[\"anatomyData\"][\"version\"]\n self.log.info(\"setting version: {}\".format(next_version))\n hda_def.setVersion(str(next_version))\n hda_def.setOptions(hda_options)\n hda_def.save(hda_def.libraryFilePath(), hda_node, hda_options)\n\n if \"representations\" not in instance.data:\n instance.data[\"representations\"] = []\n\n file = os.path.basename(hda_def.libraryFilePath())\n staging_dir = os.path.dirname(hda_def.libraryFilePath())\n self.log.info(\"Using HDA from {}\".format(hda_def.libraryFilePath()))\n\n representation = {\n 'name': 'hda',\n 'ext': 'hda',\n 'files': file,\n \"stagingDir\": staging_dir,\n }\n instance.data[\"representations\"].append(representation)\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/extract_mantra_ifd.py",
"content": "import os\n\nimport pyblish.api\n\nfrom openpype.pipeline import publish\n\nimport hou\n\n\nclass ExtractMantraIFD(publish.Extractor):\n\n order = pyblish.api.ExtractorOrder\n label = \"Extract Mantra ifd\"\n hosts = [\"houdini\"]\n families = [\"mantraifd\"]\n targets = [\"local\", \"remote\"]\n\n def process(self, instance):\n if instance.data.get(\"farm\"):\n self.log.debug(\"Should be processed on farm, skipping.\")\n return\n\n ropnode = hou.node(instance.data.get(\"instance_node\"))\n output = ropnode.evalParm(\"soho_diskfile\")\n staging_dir = os.path.dirname(output)\n instance.data[\"stagingDir\"] = staging_dir\n\n files = instance.data[\"frames\"]\n missing_frames = [\n frame\n for frame in instance.data[\"frames\"]\n if not os.path.exists(\n os.path.normpath(os.path.join(staging_dir, frame)))\n ]\n if missing_frames:\n raise RuntimeError(\"Failed to complete Mantra ifd extraction. \"\n \"Missing output files: {}\".format(\n missing_frames))\n\n if \"representations\" not in instance.data:\n instance.data[\"representations\"] = []\n\n representation = {\n 'name': 'ifd',\n 'ext': 'ifd',\n 'files': files,\n \"stagingDir\": staging_dir,\n \"frameStart\": instance.data[\"frameStart\"],\n \"frameEnd\": instance.data[\"frameEnd\"],\n }\n instance.data[\"representations\"].append(representation)\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/extract_opengl.py",
"content": "import os\n\nimport pyblish.api\n\nfrom openpype.pipeline import publish\nfrom openpype.hosts.houdini.api.lib import render_rop\n\nimport hou\n\n\nclass ExtractOpenGL(publish.Extractor):\n\n order = pyblish.api.ExtractorOrder - 0.01\n label = \"Extract OpenGL\"\n families = [\"review\"]\n hosts = [\"houdini\"]\n\n def process(self, instance):\n ropnode = hou.node(instance.data.get(\"instance_node\"))\n\n output = ropnode.evalParm(\"picture\")\n staging_dir = os.path.normpath(os.path.dirname(output))\n instance.data[\"stagingDir\"] = staging_dir\n file_name = os.path.basename(output)\n\n self.log.info(\"Extracting '%s' to '%s'\" % (file_name,\n staging_dir))\n\n render_rop(ropnode)\n\n output = instance.data[\"frames\"]\n\n tags = [\"review\"]\n if not instance.data.get(\"keepImages\"):\n tags.append(\"delete\")\n\n representation = {\n \"name\": instance.data[\"imageFormat\"],\n \"ext\": instance.data[\"imageFormat\"],\n \"files\": output,\n \"stagingDir\": staging_dir,\n \"frameStart\": instance.data[\"frameStartHandle\"],\n \"frameEnd\": instance.data[\"frameEndHandle\"],\n \"tags\": tags,\n \"preview\": True,\n \"camera_name\": instance.data.get(\"review_camera\")\n }\n\n if \"representations\" not in instance.data:\n instance.data[\"representations\"] = []\n instance.data[\"representations\"].append(representation)\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/extract_redshift_proxy.py",
"content": "import os\n\nimport pyblish.api\n\nfrom openpype.pipeline import publish\nfrom openpype.hosts.houdini.api.lib import render_rop\n\nimport hou\n\n\nclass ExtractRedshiftProxy(publish.Extractor):\n\n order = pyblish.api.ExtractorOrder + 0.1\n label = \"Extract Redshift Proxy\"\n families = [\"redshiftproxy\"]\n hosts = [\"houdini\"]\n targets = [\"local\", \"remote\"]\n\n def process(self, instance):\n if instance.data.get(\"farm\"):\n self.log.debug(\"Should be processed on farm, skipping.\")\n return\n ropnode = hou.node(instance.data.get(\"instance_node\"))\n\n # Get the filename from the filename parameter\n # `.evalParm(parameter)` will make sure all tokens are resolved\n output = ropnode.evalParm(\"RS_archive_file\")\n staging_dir = os.path.normpath(os.path.dirname(output))\n instance.data[\"stagingDir\"] = staging_dir\n file_name = os.path.basename(output)\n\n self.log.info(\"Writing Redshift Proxy '%s' to '%s'\" % (file_name,\n staging_dir))\n\n render_rop(ropnode)\n\n output = instance.data[\"frames\"]\n\n if \"representations\" not in instance.data:\n instance.data[\"representations\"] = []\n\n representation = {\n \"name\": \"rs\",\n \"ext\": \"rs\",\n \"files\": output,\n \"stagingDir\": staging_dir,\n }\n\n # A single frame may also be rendered without start/end frame.\n if \"frameStartHandle\" in instance.data and \"frameEndHandle\" in instance.data: # noqa\n representation[\"frameStart\"] = instance.data[\"frameStartHandle\"]\n representation[\"frameEnd\"] = instance.data[\"frameEndHandle\"]\n\n instance.data[\"representations\"].append(representation)\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/extract_usd.py",
"content": "import os\n\nimport pyblish.api\n\nfrom openpype.pipeline import publish\nfrom openpype.hosts.houdini.api.lib import render_rop\n\nimport hou\n\nclass ExtractUSD(publish.Extractor):\n\n order = pyblish.api.ExtractorOrder\n label = \"Extract USD\"\n hosts = [\"houdini\"]\n families = [\"usd\",\n \"usdModel\",\n \"usdSetDress\"]\n\n def process(self, instance):\n\n ropnode = hou.node(instance.data.get(\"instance_node\"))\n\n # Get the filename from the filename parameter\n output = ropnode.evalParm(\"lopoutput\")\n staging_dir = os.path.dirname(output)\n instance.data[\"stagingDir\"] = staging_dir\n file_name = os.path.basename(output)\n\n self.log.info(\"Writing USD '%s' to '%s'\" % (file_name, staging_dir))\n\n render_rop(ropnode)\n\n assert os.path.exists(output), \"Output does not exist: %s\" % output\n\n if \"representations\" not in instance.data:\n instance.data[\"representations\"] = []\n\n representation = {\n 'name': 'usd',\n 'ext': 'usd',\n 'files': file_name,\n \"stagingDir\": staging_dir,\n }\n instance.data[\"representations\"].append(representation)\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/extract_usd_layered.py",
"content": "import os\nimport contextlib\nimport hou\nimport sys\nfrom collections import deque\n\nimport pyblish.api\n\nfrom openpype.client import (\n get_asset_by_name,\n get_subset_by_name,\n get_last_version_by_subset_id,\n get_representation_by_name,\n)\nfrom openpype.pipeline import (\n get_representation_path,\n publish,\n)\nimport openpype.hosts.houdini.api.usd as hou_usdlib\nfrom openpype.hosts.houdini.api.lib import render_rop\n\n\nclass ExitStack(object):\n \"\"\"Context manager for dynamic management of a stack of exit callbacks.\n\n For example:\n\n with ExitStack() as stack:\n files = [stack.enter_context(open(fname)) for fname in filenames]\n # All opened files will automatically be closed at the end of\n # the with statement, even if attempts to open files later\n # in the list raise an exception\n\n \"\"\"\n\n def __init__(self):\n self._exit_callbacks = deque()\n\n def pop_all(self):\n \"\"\"Preserve the context stack by transferring it to a new instance\"\"\"\n new_stack = type(self)()\n new_stack._exit_callbacks = self._exit_callbacks\n self._exit_callbacks = deque()\n return new_stack\n\n def _push_cm_exit(self, cm, cm_exit):\n \"\"\"Helper to correctly register callbacks to __exit__ methods\"\"\"\n\n def _exit_wrapper(*exc_details):\n return cm_exit(cm, *exc_details)\n\n _exit_wrapper.__self__ = cm\n self.push(_exit_wrapper)\n\n def push(self, exit):\n \"\"\"Registers a callback with the standard __exit__ method signature.\n\n Can suppress exceptions the same way __exit__ methods can.\n\n Also accepts any object with an __exit__ method (registering a call\n to the method instead of the object itself)\n\n \"\"\"\n # We use an unbound method rather than a bound method to follow\n # the standard lookup behaviour for special methods\n _cb_type = type(exit)\n try:\n exit_method = _cb_type.__exit__\n except AttributeError:\n # Not a context manager, so assume its a callable\n self._exit_callbacks.append(exit)\n else:\n self._push_cm_exit(exit, exit_method)\n return exit # Allow use as a decorator\n\n def callback(self, callback, *args, **kwds):\n \"\"\"Registers an arbitrary callback and arguments.\n\n Cannot suppress exceptions.\n \"\"\"\n\n def _exit_wrapper(exc_type, exc, tb):\n callback(*args, **kwds)\n\n # We changed the signature, so using @wraps is not appropriate, but\n # setting __wrapped__ may still help with introspection\n _exit_wrapper.__wrapped__ = callback\n self.push(_exit_wrapper)\n return callback # Allow use as a decorator\n\n def enter_context(self, cm):\n \"\"\"Enters the supplied context manager\n\n If successful, also pushes its __exit__ method as a callback and\n returns the result of the __enter__ method.\n \"\"\"\n # We look up the special methods on the type to match the with\n # statement\n _cm_type = type(cm)\n _exit = _cm_type.__exit__\n result = _cm_type.__enter__(cm)\n self._push_cm_exit(cm, _exit)\n return result\n\n def close(self):\n \"\"\"Immediately unwind the context stack\"\"\"\n self.__exit__(None, None, None)\n\n def __enter__(self):\n return self\n\n def __exit__(self, *exc_details):\n # We manipulate the exception state so it behaves as though\n # we were actually nesting multiple with statements\n frame_exc = sys.exc_info()[1]\n\n def _fix_exception_context(new_exc, old_exc):\n while 1:\n exc_context = new_exc.__context__\n if exc_context in (None, frame_exc):\n break\n new_exc = exc_context\n new_exc.__context__ = old_exc\n\n # Callbacks are invoked in LIFO order to match the behaviour of\n # nested context managers\n suppressed_exc = False\n while self._exit_callbacks:\n cb = self._exit_callbacks.pop()\n try:\n if cb(*exc_details):\n suppressed_exc = True\n exc_details = (None, None, None)\n except Exception:\n new_exc_details = sys.exc_info()\n # simulate the stack of exceptions by setting the context\n _fix_exception_context(new_exc_details[1], exc_details[1])\n if not self._exit_callbacks:\n raise\n exc_details = new_exc_details\n return suppressed_exc\n\n\n@contextlib.contextmanager\ndef parm_values(overrides):\n \"\"\"Override Parameter values during the context.\"\"\"\n\n originals = []\n try:\n for parm, value in overrides:\n originals.append((parm, parm.eval()))\n parm.set(value)\n yield\n finally:\n for parm, value in originals:\n # Parameter might not exist anymore so first\n # check whether it's still valid\n if hou.parm(parm.path()):\n parm.set(value)\n\n\nclass ExtractUSDLayered(publish.Extractor):\n\n order = pyblish.api.ExtractorOrder\n label = \"Extract Layered USD\"\n hosts = [\"houdini\"]\n families = [\"usdLayered\", \"usdShade\"]\n\n # Force Output Processors so it will always save any file\n # into our unique staging directory with processed Avalon paths\n output_processors = [\"avalon_uri_processor\", \"stagingdir_processor\"]\n\n def process(self, instance):\n\n self.log.info(\"Extracting: %s\" % instance)\n\n staging_dir = self.staging_dir(instance)\n fname = instance.data.get(\"usdFilename\")\n\n # The individual rop nodes are collected as \"publishDependencies\"\n dependencies = instance.data[\"publishDependencies\"]\n ropnodes = [dependency[0] for dependency in dependencies]\n assert all(\n node.type().name() in {\"usd\", \"usd_rop\"} for node in ropnodes\n )\n\n # Main ROP node, either a USD Rop or ROP network with\n # multiple USD ROPs\n node = hou.node(instance.data[\"instance_node\"])\n\n # Collect any output dependencies that have not been processed yet\n # during extraction of other instances\n outputs = [fname]\n active_dependencies = [\n dep\n for dep in dependencies\n if dep.data.get(\"publish\", True)\n and not dep.data.get(\"_isExtracted\", False)\n ]\n for dependency in active_dependencies:\n outputs.append(dependency.data[\"usdFilename\"])\n\n pattern = r\"*[/\\]{0} {0}\"\n save_pattern = \" \".join(pattern.format(fname) for fname in outputs)\n\n # Run a stack of context managers before we start the render to\n # temporarily adjust USD ROP settings for our publish output.\n rop_overrides = {\n # This sets staging directory on the processor to force our\n # output files to end up in the Staging Directory.\n \"stagingdiroutputprocessor_stagingDir\": staging_dir,\n # Force the Avalon URI Output Processor to refactor paths for\n # references, payloads and layers to published paths.\n \"avalonurioutputprocessor_use_publish_paths\": True,\n # Only write out specific USD files based on our outputs\n \"savepattern\": save_pattern,\n }\n overrides = list()\n with ExitStack() as stack:\n\n for ropnode in ropnodes:\n manager = hou_usdlib.outputprocessors(\n ropnode,\n processors=self.output_processors,\n disable_all_others=True,\n )\n stack.enter_context(manager)\n\n # Some of these must be added after we enter the output\n # processor context manager because those parameters only\n # exist when the Output Processor is added to the ROP node.\n for name, value in rop_overrides.items():\n parm = ropnode.parm(name)\n assert parm, \"Parm not found: %s.%s\" % (\n ropnode.path(),\n name,\n )\n overrides.append((parm, value))\n\n stack.enter_context(parm_values(overrides))\n\n # Render the single ROP node or the full ROP network\n render_rop(node)\n\n # Assert all output files in the Staging Directory\n for output_fname in outputs:\n path = os.path.join(staging_dir, output_fname)\n assert os.path.exists(path), \"Output file must exist: %s\" % path\n\n # Set up the dependency for publish if they have new content\n # compared to previous publishes\n project_name = instance.context.data[\"projectName\"]\n for dependency in active_dependencies:\n dependency_fname = dependency.data[\"usdFilename\"]\n\n filepath = os.path.join(staging_dir, dependency_fname)\n similar = self._compare_with_latest_publish(\n project_name, dependency, filepath\n )\n if similar:\n # Deactivate this dependency\n self.log.debug(\n \"Dependency matches previous publish version,\"\n \" deactivating %s for publish\" % dependency\n )\n dependency.data[\"publish\"] = False\n else:\n self.log.debug(\"Extracted dependency: %s\" % dependency)\n # This dependency should be published\n dependency.data[\"files\"] = [dependency_fname]\n dependency.data[\"stagingDir\"] = staging_dir\n dependency.data[\"_isExtracted\"] = True\n\n # Store the created files on the instance\n if \"files\" not in instance.data:\n instance.data[\"files\"] = []\n instance.data[\"files\"].append(fname)\n\n def _compare_with_latest_publish(self, project_name, dependency, new_file):\n import filecmp\n\n _, ext = os.path.splitext(new_file)\n\n # Compare this dependency with the latest published version\n # to detect whether we should make this into a new publish\n # version. If not, skip it.\n asset = get_asset_by_name(\n project_name, dependency.data[\"asset\"], fields=[\"_id\"]\n )\n subset = get_subset_by_name(\n project_name,\n dependency.data[\"subset\"],\n asset[\"_id\"],\n fields=[\"_id\"]\n )\n if not subset:\n # Subset doesn't exist yet. Definitely new file\n self.log.debug(\"No existing subset..\")\n return False\n\n version = get_last_version_by_subset_id(\n project_name, subset[\"_id\"], fields=[\"_id\"]\n )\n if not version:\n self.log.debug(\"No existing version..\")\n return False\n\n representation = get_representation_by_name(\n project_name, ext.lstrip(\".\"), version[\"_id\"]\n )\n if not representation:\n self.log.debug(\"No existing representation..\")\n return False\n\n old_file = get_representation_path(representation)\n if not os.path.exists(old_file):\n return False\n\n return filecmp.cmp(old_file, new_file)\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/extract_vdb_cache.py",
"content": "import os\n\nimport pyblish.api\n\nfrom openpype.pipeline import publish\nfrom openpype.hosts.houdini.api.lib import render_rop\n\nimport hou\n\n\nclass ExtractVDBCache(publish.Extractor):\n\n order = pyblish.api.ExtractorOrder + 0.1\n label = \"Extract VDB Cache\"\n families = [\"vdbcache\"]\n hosts = [\"houdini\"]\n\n def process(self, instance):\n if instance.data.get(\"farm\"):\n self.log.debug(\"Should be processed on farm, skipping.\")\n return\n ropnode = hou.node(instance.data[\"instance_node\"])\n\n # Get the filename from the filename parameter\n # `.evalParm(parameter)` will make sure all tokens are resolved\n sop_output = ropnode.evalParm(\"sopoutput\")\n staging_dir = os.path.normpath(os.path.dirname(sop_output))\n instance.data[\"stagingDir\"] = staging_dir\n file_name = os.path.basename(sop_output)\n\n self.log.info(\"Writing VDB '%s' to '%s'\" % (file_name, staging_dir))\n\n render_rop(ropnode)\n\n output = instance.data[\"frames\"]\n\n if \"representations\" not in instance.data:\n instance.data[\"representations\"] = []\n\n representation = {\n \"name\": \"vdb\",\n \"ext\": \"vdb\",\n \"files\": output,\n \"stagingDir\": staging_dir,\n \"frameStart\": instance.data[\"frameStartHandle\"],\n \"frameEnd\": instance.data[\"frameEndHandle\"],\n }\n instance.data[\"representations\"].append(representation)\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/help/validate_vdb_output_node.xml",
"content": "\n\n\nInvalid VDB\n\n## Invalid VDB output\n\nAll primitives of the output geometry must be VDBs, no other primitive\ntypes are allowed. That means that regardless of the amount of VDBs in the\ngeometry it will have an equal amount of VDBs, points, primitives and\nvertices since each VDB primitive is one point, one vertex and one VDB.\n\nThis validation only checks the geometry on the first frame of the export\nframe range.\n\n\n\n\n\n### Detailed Info\n\nROP node `{rop_path}` is set to export SOP path `{sop_path}`.\n\n{message}\n\n\n\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/increment_current_file.py",
"content": "import pyblish.api\n\nfrom openpype.lib import version_up\nfrom openpype.pipeline import registered_host\nfrom openpype.pipeline.publish import get_errored_plugins_from_context\nfrom openpype.hosts.houdini.api import HoudiniHost\nfrom openpype.pipeline.publish import KnownPublishError\n\n\nclass IncrementCurrentFile(pyblish.api.ContextPlugin):\n \"\"\"Increment the current file.\n\n Saves the current scene with an increased version number.\n\n \"\"\"\n\n label = \"Increment current file\"\n order = pyblish.api.IntegratorOrder + 9.0\n hosts = [\"houdini\"]\n families = [\"workfile\",\n \"redshift_rop\",\n \"arnold_rop\",\n \"mantra_rop\",\n \"karma_rop\",\n \"usdrender\",\n \"publish.hou\"]\n optional = True\n\n def process(self, context):\n\n errored_plugins = get_errored_plugins_from_context(context)\n if any(\n plugin.__name__ == \"HoudiniSubmitPublishDeadline\"\n for plugin in errored_plugins\n ):\n raise KnownPublishError(\n \"Skipping incrementing current file because \"\n \"submission to deadline failed.\"\n )\n\n # Filename must not have changed since collecting\n host = registered_host() # type: HoudiniHost\n current_file = host.current_file()\n if context.data[\"currentFile\"] != current_file:\n raise KnownPublishError(\n \"Collected filename mismatches from current scene name.\"\n )\n\n new_filepath = version_up(current_file)\n host.save_workfile(new_filepath)\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/save_scene.py",
"content": "import pyblish.api\n\nfrom openpype.pipeline import registered_host\n\n\nclass SaveCurrentScene(pyblish.api.ContextPlugin):\n \"\"\"Save current scene\"\"\"\n\n label = \"Save current file\"\n order = pyblish.api.ExtractorOrder - 0.49\n hosts = [\"houdini\"]\n\n def process(self, context):\n\n # Filename must not have changed since collecting\n host = registered_host()\n current_file = host.get_current_workfile()\n assert context.data['currentFile'] == current_file, (\n \"Collected filename from current scene name.\"\n )\n\n if host.workfile_has_unsaved_changes():\n self.log.info(\"Saving current file: {}\".format(current_file))\n host.save_workfile(current_file)\n else:\n self.log.debug(\"No unsaved changes, skipping file save..\")\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_abc_primitive_to_detail.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\n\nfrom collections import defaultdict\nfrom openpype.pipeline import PublishValidationError\n\n\nclass ValidateAbcPrimitiveToDetail(pyblish.api.InstancePlugin):\n \"\"\"Validate Alembic ROP Primitive to Detail attribute is consistent.\n\n The Alembic ROP crashes Houdini whenever an attribute in the \"Primitive to\n Detail\" parameter exists on only a part of the primitives that belong to\n the same hierarchy path. Whenever it encounters inconsistent values,\n specifically where some are empty as opposed to others then Houdini\n crashes. (Tested in Houdini 17.5.229)\n\n \"\"\"\n\n order = pyblish.api.ValidatorOrder + 0.1\n families = [\"abc\"]\n hosts = [\"houdini\"]\n label = \"Validate Primitive to Detail (Abc)\"\n\n def process(self, instance):\n invalid = self.get_invalid(instance)\n if invalid:\n raise PublishValidationError(\n (\"Primitives found with inconsistent primitive \"\n \"to detail attributes. See log.\"),\n title=self.label\n )\n\n @classmethod\n def get_invalid(cls, instance):\n import hou # noqa\n output_node = instance.data.get(\"output_node\")\n rop_node = hou.node(instance.data[\"instance_node\"])\n if output_node is None:\n cls.log.error(\n \"SOP Output node in '%s' does not exist. \"\n \"Ensure a valid SOP output path is set.\" % rop_node.path()\n )\n\n return [rop_node.path()]\n\n pattern = rop_node.parm(\"prim_to_detail_pattern\").eval().strip()\n if not pattern:\n cls.log.debug(\n \"Alembic ROP has no 'Primitive to Detail' pattern. \"\n \"Validation is ignored..\"\n )\n return\n\n build_from_path = rop_node.parm(\"build_from_path\").eval()\n if not build_from_path:\n cls.log.debug(\n \"Alembic ROP has 'Build from Path' disabled. \"\n \"Validation is ignored..\"\n )\n return\n\n path_attr = rop_node.parm(\"path_attrib\").eval()\n if not path_attr:\n cls.log.error(\n \"The Alembic ROP node has no Path Attribute\"\n \"value set, but 'Build Hierarchy from Attribute'\"\n \"is enabled.\"\n )\n return [rop_node.path()]\n\n # Let's assume each attribute is explicitly named for now and has no\n # wildcards for Primitive to Detail. This simplifies the check.\n cls.log.debug(\"Checking Primitive to Detail pattern: %s\" % pattern)\n cls.log.debug(\"Checking with path attribute: %s\" % path_attr)\n\n if not hasattr(output_node, \"geometry\"):\n # In the case someone has explicitly set an Object\n # node instead of a SOP node in Geometry context\n # then for now we ignore - this allows us to also\n # export object transforms.\n cls.log.warning(\"No geometry output node found, skipping check..\")\n return\n\n # Check if the primitive attribute exists\n frame = instance.data.get(\"frameStart\", 0)\n geo = output_node.geometryAtFrame(frame)\n\n # If there are no primitives on the start frame then it might be\n # something that is emitted over time. As such we can't actually\n # validate whether the attributes exist, because they won't exist\n # yet. In that case, just warn the user and allow it.\n if len(geo.iterPrims()) == 0:\n cls.log.warning(\n \"No primitives found on current frame. Validation\"\n \" for Primitive to Detail will be skipped.\"\n )\n return\n\n attrib = geo.findPrimAttrib(path_attr)\n if not attrib:\n cls.log.info(\n \"Geometry Primitives are missing \"\n \"path attribute: `%s`\" % path_attr\n )\n return [output_node.path()]\n\n # Ensure at least a single string value is present\n if not attrib.strings():\n cls.log.info(\n \"Primitive path attribute has no \"\n \"string values: %s\" % path_attr\n )\n return [output_node.path()]\n\n paths = None\n for attr in pattern.split(\" \"):\n if not attr.strip():\n # Ignore empty values\n continue\n\n # Check if the primitive attribute exists\n attrib = geo.findPrimAttrib(attr)\n if not attrib:\n # It is allowed to not have the attribute at all\n continue\n\n # The issue can only happen if at least one string attribute is\n # present. So we ignore cases with no values whatsoever.\n if not attrib.strings():\n continue\n\n check = defaultdict(set)\n values = geo.primStringAttribValues(attr)\n if paths is None:\n paths = geo.primStringAttribValues(path_attr)\n\n for path, value in zip(paths, values):\n check[path].add(value)\n\n for path, values in check.items():\n # Whenever a single path has multiple values for the\n # Primitive to Detail attribute then we consider it\n # inconsistent and invalidate the ROP node's content.\n if len(values) > 1:\n cls.log.warning(\n \"Path has multiple values: %s (path: %s)\"\n % (list(values), path)\n )\n return [output_node.path()]\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_alembic_face_sets.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\nimport hou\n\nclass ValidateAlembicROPFaceSets(pyblish.api.InstancePlugin):\n \"\"\"Validate Face Sets are disabled for extraction to pointcache.\n\n When groups are saved as Face Sets with the Alembic these show up\n as shadingEngine connections in Maya - however, with animated groups\n these connections in Maya won't work as expected, it won't update per\n frame. Additionally, it can break shader assignments in some cases\n where it requires to first break this connection to allow a shader to\n be assigned.\n\n It is allowed to include Face Sets, so only an issue is logged to\n identify that it could introduce issues down the pipeline.\n\n \"\"\"\n\n order = pyblish.api.ValidatorOrder + 0.1\n families = [\"abc\"]\n hosts = [\"houdini\"]\n label = \"Validate Alembic ROP Face Sets\"\n\n def process(self, instance):\n\n rop = hou.node(instance.data[\"instance_node\"])\n facesets = rop.parm(\"facesets\").eval()\n\n # 0 = No Face Sets\n # 1 = Save Non-Empty Groups as Face Sets\n # 2 = Save All Groups As Face Sets\n if facesets != 0:\n self.log.warning(\n \"Alembic ROP saves 'Face Sets' for Geometry. \"\n \"Are you sure you want this?\"\n )\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_alembic_input_node.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\nfrom openpype.pipeline import PublishValidationError\nimport hou\n\n\nclass ValidateAlembicInputNode(pyblish.api.InstancePlugin):\n \"\"\"Validate that the node connected to the output is correct.\n\n The connected node cannot be of the following types for Alembic:\n - VDB\n - Volume\n\n \"\"\"\n\n order = pyblish.api.ValidatorOrder + 0.1\n families = [\"abc\"]\n hosts = [\"houdini\"]\n label = \"Validate Input Node (Abc)\"\n\n def process(self, instance):\n invalid = self.get_invalid(instance)\n if invalid:\n raise PublishValidationError(\n (\"Primitive types found that are not supported \"\n \"for Alembic output.\"),\n title=self.label\n )\n\n @classmethod\n def get_invalid(cls, instance):\n\n invalid_prim_types = [\"VDB\", \"Volume\"]\n output_node = instance.data.get(\"output_node\")\n\n if output_node is None:\n node = hou.node(instance.data[\"instance_node\"])\n cls.log.error(\n \"SOP Output node in '%s' does not exist. \"\n \"Ensure a valid SOP output path is set.\" % node.path()\n )\n\n return [node.path()]\n\n if not hasattr(output_node, \"geometry\"):\n # In the case someone has explicitly set an Object\n # node instead of a SOP node in Geometry context\n # then for now we ignore - this allows us to also\n # export object transforms.\n cls.log.warning(\"No geometry output node found, skipping check..\")\n return\n\n frame = instance.data.get(\"frameStart\", 0)\n geo = output_node.geometryAtFrame(frame)\n\n invalid = False\n for prim_type in invalid_prim_types:\n if geo.countPrimType(prim_type) > 0:\n cls.log.error(\n \"Found a primitive which is of type '%s' !\" % prim_type\n )\n invalid = True\n\n if invalid:\n return [instance]\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_animation_settings.py",
"content": "import pyblish.api\n\nfrom openpype.pipeline.publish import PublishValidationError\nfrom openpype.hosts.houdini.api import lib\nimport hou\n\n\nclass ValidateAnimationSettings(pyblish.api.InstancePlugin):\n \"\"\"Validate if the unexpanded string contains the frame ('$F') token\n\n This validator will only check the output parameter of the node if\n the Valid Frame Range is not set to 'Render Current Frame'\n\n Rules:\n If you render out a frame range it is mandatory to have the\n frame token - '$F4' or similar - to ensure that each frame gets\n written. If this is not the case you will override the same file\n every time a frame is written out.\n\n Examples:\n Good: 'my_vbd_cache.$F4.vdb'\n Bad: 'my_vbd_cache.vdb'\n\n \"\"\"\n\n order = pyblish.api.ValidatorOrder\n label = \"Validate Frame Settings\"\n families = [\"vdbcache\"]\n\n def process(self, instance):\n\n invalid = self.get_invalid(instance)\n if invalid:\n raise PublishValidationError(\n \"Output settings do no match for '%s'\" % instance\n )\n\n @classmethod\n def get_invalid(cls, instance):\n\n node = hou.node(instance.data[\"instance_node\"])\n # Check trange parm, 0 means Render Current Frame\n frame_range = node.evalParm(\"trange\")\n if frame_range == 0:\n return []\n\n output_parm = lib.get_output_parameter(node)\n unexpanded_str = output_parm.unexpandedString()\n\n if \"$F\" not in unexpanded_str:\n cls.log.error(\"No frame token found in '%s'\" % node.path())\n return [instance]\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_bypass.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\nfrom openpype.pipeline import PublishValidationError\n\nimport hou\n\nclass ValidateBypassed(pyblish.api.InstancePlugin):\n \"\"\"Validate all primitives build hierarchy from attribute when enabled.\n\n The name of the attribute must exist on the prims and have the same name\n as Build Hierarchy from Attribute's `Path Attribute` value on the Alembic\n ROP node whenever Build Hierarchy from Attribute is enabled.\n\n \"\"\"\n\n order = pyblish.api.ValidatorOrder - 0.1\n families = [\"*\"]\n hosts = [\"houdini\"]\n label = \"Validate ROP Bypass\"\n\n def process(self, instance):\n\n if len(instance) == 0:\n # Ignore instances without any nodes\n # e.g. in memory bootstrap instances\n return\n\n invalid = self.get_invalid(instance)\n if invalid:\n rop = invalid[0]\n raise PublishValidationError(\n (\"ROP node {} is set to bypass, publishing cannot \"\n \"continue.\".format(rop.path())),\n title=self.label\n )\n\n @classmethod\n def get_invalid(cls, instance):\n\n rop = hou.node(instance.data[\"instance_node\"])\n if hasattr(rop, \"isBypassed\") and rop.isBypassed():\n return [rop]\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_camera_rop.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Validator plugin for Houdini Camera ROP settings.\"\"\"\nimport pyblish.api\nfrom openpype.pipeline import PublishValidationError\n\n\nclass ValidateCameraROP(pyblish.api.InstancePlugin):\n \"\"\"Validate Camera ROP settings.\"\"\"\n\n order = pyblish.api.ValidatorOrder\n families = [\"camera\"]\n hosts = [\"houdini\"]\n label = \"Camera ROP\"\n\n def process(self, instance):\n\n import hou\n\n node = hou.node(instance.data.get(\"instance_node\"))\n if node.parm(\"use_sop_path\").eval():\n raise PublishValidationError(\n (\"Alembic ROP for Camera export should not be \"\n \"set to 'Use Sop Path'. Please disable.\"),\n title=self.label\n )\n\n # Get the root and objects parameter of the Alembic ROP node\n root = node.parm(\"root\").eval()\n objects = node.parm(\"objects\").eval()\n errors = []\n if not root:\n errors.append(\"Root parameter must be set on Alembic ROP\")\n if not root.startswith(\"/\"):\n errors.append(\"Root parameter must start with slash /\")\n if not objects:\n errors.append(\"Objects parameter must be set on Alembic ROP\")\n if len(objects.split(\" \")) != 1:\n errors.append(\"Must have only a single object.\")\n\n if errors:\n for error in errors:\n self.log.error(error)\n raise PublishValidationError(\n \"Some checks failed, see validator log.\",\n title=self.label)\n\n # Check if the object exists and is a camera\n path = root + \"/\" + objects\n camera = hou.node(path)\n\n if not camera:\n raise PublishValidationError(\n \"Camera path does not exist: %s\" % path,\n title=self.label)\n\n if camera.type().name() != \"cam\":\n raise PublishValidationError(\n (\"Object set in Alembic ROP is not a camera: \"\n \"{} (type: {})\").format(camera, camera.type().name()),\n title=self.label)\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_cop_output_node.py",
"content": "# -*- coding: utf-8 -*-\nimport sys\nimport pyblish.api\nimport six\n\nfrom openpype.pipeline import PublishValidationError\n\n\nclass ValidateCopOutputNode(pyblish.api.InstancePlugin):\n \"\"\"Validate the instance COP Output Node.\n\n This will ensure:\n - The COP Path is set.\n - The COP Path refers to an existing object.\n - The COP Path node is a COP node.\n\n \"\"\"\n\n order = pyblish.api.ValidatorOrder\n families = [\"imagesequence\"]\n hosts = [\"houdini\"]\n label = \"Validate COP Output Node\"\n\n def process(self, instance):\n\n invalid = self.get_invalid(instance)\n if invalid:\n raise PublishValidationError(\n (\"Output node(s) `{}` are incorrect. \"\n \"See plug-in log for details.\").format(invalid),\n title=self.label\n )\n\n @classmethod\n def get_invalid(cls, instance):\n\n import hou\n\n try:\n output_node = instance.data[\"output_node\"]\n except KeyError:\n six.reraise(\n PublishValidationError,\n PublishValidationError(\n \"Can't determine COP output node.\",\n title=cls.__name__),\n sys.exc_info()[2]\n )\n\n if output_node is None:\n node = hou.node(instance.data.get(\"instance_node\"))\n cls.log.error(\n \"COP Output node in '%s' does not exist. \"\n \"Ensure a valid COP output path is set.\" % node.path()\n )\n\n return [node.path()]\n\n # Output node must be a Sop node.\n if not isinstance(output_node, hou.CopNode):\n cls.log.error(\n \"Output node %s is not a COP node. \"\n \"COP Path must point to a COP node, \"\n \"instead found category type: %s\"\n % (output_node.path(), output_node.type().category().name())\n )\n return [output_node.path()]\n\n # For the sake of completeness also assert the category type\n # is Cop2 to avoid potential edge case scenarios even though\n # the isinstance check above should be stricter than this category\n if output_node.type().category().name() != \"Cop2\":\n raise PublishValidationError(\n (\"Output node %s is not of category Cop2. \"\n \"This is a bug...\").format(output_node.path()),\n title=cls.label)\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_fbx_output_node.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\nfrom openpype.pipeline import PublishValidationError\nfrom openpype.hosts.houdini.api.action import (\n SelectInvalidAction,\n SelectROPAction,\n)\nfrom openpype.hosts.houdini.api.lib import get_obj_node_output\nimport hou\n\n\nclass ValidateFBXOutputNode(pyblish.api.InstancePlugin):\n \"\"\"Validate the instance Output Node.\n\n This will ensure:\n - The Output Node Path is set.\n - The Output Node Path refers to an existing object.\n - The Output Node is a Sop or Obj node.\n - The Output Node has geometry data.\n - The Output Node doesn't include invalid primitive types.\n \"\"\"\n\n order = pyblish.api.ValidatorOrder\n families = [\"fbx\"]\n hosts = [\"houdini\"]\n label = \"Validate FBX Output Node\"\n actions = [SelectROPAction, SelectInvalidAction]\n\n def process(self, instance):\n\n invalid = self.get_invalid(instance)\n if invalid:\n nodes = [n.path() for n in invalid]\n raise PublishValidationError(\n \"See log for details. \"\n \"Invalid nodes: {0}\".format(nodes),\n title=\"Invalid output node(s)\"\n )\n\n @classmethod\n def get_invalid(cls, instance):\n output_node = instance.data.get(\"output_node\")\n\n # Check if The Output Node Path is set and\n # refers to an existing object.\n if output_node is None:\n rop_node = hou.node(instance.data[\"instance_node\"])\n cls.log.error(\n \"Output node in '%s' does not exist. \"\n \"Ensure a valid output path is set.\", rop_node.path()\n )\n\n return [rop_node]\n\n # Check if the Output Node is a Sop or an Obj node\n # also, list all sop output nodes inside as well as\n # invalid empty nodes.\n all_out_sops = []\n invalid = []\n\n # if output_node is an ObjSubnet or an ObjNetwork\n if output_node.childTypeCategory() == hou.objNodeTypeCategory():\n for node in output_node.allSubChildren():\n if node.type().name() == \"geo\":\n out = get_obj_node_output(node)\n if out:\n all_out_sops.append(out)\n else:\n invalid.append(node) # empty_objs\n cls.log.error(\n \"Geo Obj Node '%s' is empty!\",\n node.path()\n )\n if not all_out_sops:\n invalid.append(output_node) # empty_objs\n cls.log.error(\n \"Output Node '%s' is empty!\",\n node.path()\n )\n\n # elif output_node is an ObjNode\n elif output_node.type().name() == \"geo\":\n out = get_obj_node_output(output_node)\n if out:\n all_out_sops.append(out)\n else:\n invalid.append(node) # empty_objs\n cls.log.error(\n \"Output Node '%s' is empty!\",\n node.path()\n )\n\n # elif output_node is a SopNode\n elif output_node.type().category().name() == \"Sop\":\n all_out_sops.append(output_node)\n\n # Then it's a wrong node type\n else:\n cls.log.error(\n \"Output node %s is not a SOP or OBJ Geo or OBJ SubNet node. \"\n \"Instead found category type: %s %s\",\n output_node.path(), output_node.type().category().name(),\n output_node.type().name()\n )\n return [output_node]\n\n # Check if all output sop nodes have geometry\n # and don't contain invalid prims\n invalid_prim_types = [\"VDB\", \"Volume\"]\n for sop_node in all_out_sops:\n # Empty Geometry test\n if not hasattr(sop_node, \"geometry\"):\n invalid.append(sop_node) # empty_geometry\n cls.log.error(\n \"Sop node '%s' doesn't include any prims.\",\n sop_node.path()\n )\n continue\n\n frame = instance.data.get(\"frameStart\", 0)\n geo = sop_node.geometryAtFrame(frame)\n if len(geo.iterPrims()) == 0:\n invalid.append(sop_node) # empty_geometry\n cls.log.error(\n \"Sop node '%s' doesn't include any prims.\",\n sop_node.path()\n )\n continue\n\n # Invalid Prims test\n for prim_type in invalid_prim_types:\n if geo.countPrimType(prim_type) > 0:\n invalid.append(sop_node) # invalid_prims\n cls.log.error(\n \"Sop node '%s' includes invalid prims of type '%s'.\",\n sop_node.path(), prim_type\n )\n\n if invalid:\n return invalid\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_file_extension.py",
"content": "# -*- coding: utf-8 -*-\nimport os\nimport pyblish.api\n\nfrom openpype.hosts.houdini.api import lib\nfrom openpype.pipeline import PublishValidationError\n\nimport hou\n\n\nclass ValidateFileExtension(pyblish.api.InstancePlugin):\n \"\"\"Validate the output file extension fits the output family.\n\n File extensions:\n - Pointcache must be .abc\n - Camera must be .abc\n - VDB must be .vdb\n\n \"\"\"\n\n order = pyblish.api.ValidatorOrder\n families = [\"camera\", \"vdbcache\"]\n hosts = [\"houdini\"]\n label = \"Output File Extension\"\n\n family_extensions = {\n \"camera\": \".abc\",\n \"vdbcache\": \".vdb\",\n }\n\n def process(self, instance):\n\n invalid = self.get_invalid(instance)\n if invalid:\n raise PublishValidationError(\n \"ROP node has incorrect file extension: {}\".format(invalid),\n title=self.label\n )\n\n @classmethod\n def get_invalid(cls, instance):\n\n # Get ROP node from instance\n node = hou.node(instance.data[\"instance_node\"])\n\n # Create lookup for current family in instance\n families = []\n family = instance.data.get(\"family\", None)\n if family:\n families.append(family)\n families = set(families)\n\n # Perform extension check\n output = lib.get_output_parameter(node).eval()\n _, output_extension = os.path.splitext(output)\n\n for family in families:\n extension = cls.family_extensions.get(family, None)\n if extension is None:\n raise PublishValidationError(\n \"Unsupported family: {}\".format(family),\n title=cls.label)\n\n if output_extension != extension:\n return [node.path()]\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_frame_range.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\nfrom openpype.pipeline import PublishValidationError\nfrom openpype.pipeline.publish import RepairAction\nfrom openpype.hosts.houdini.api.action import SelectInvalidAction\n\nimport hou\n\n\nclass DisableUseAssetHandlesAction(RepairAction):\n label = \"Disable use asset handles\"\n icon = \"mdi.toggle-switch-off\"\n\n\nclass ValidateFrameRange(pyblish.api.InstancePlugin):\n \"\"\"Validate Frame Range.\n\n Due to the usage of start and end handles,\n then Frame Range must be >= (start handle + end handle)\n which results that frameEnd be smaller than frameStart\n \"\"\"\n\n order = pyblish.api.ValidatorOrder - 0.1\n hosts = [\"houdini\"]\n label = \"Validate Frame Range\"\n actions = [DisableUseAssetHandlesAction, SelectInvalidAction]\n\n def process(self, instance):\n\n invalid = self.get_invalid(instance)\n if invalid:\n raise PublishValidationError(\n title=\"Invalid Frame Range\",\n message=(\n \"Invalid frame range because the instance \"\n \"start frame ({0[frameStart]}) is higher than \"\n \"the end frame ({0[frameEnd]})\"\n .format(instance.data)\n ),\n description=(\n \"## Invalid Frame Range\\n\"\n \"The frame range for the instance is invalid because \"\n \"the start frame is higher than the end frame.\\n\\nThis \"\n \"is likely due to asset handles being applied to your \"\n \"instance or the ROP node's start frame \"\n \"is set higher than the end frame.\\n\\nIf your ROP frame \"\n \"range is correct and you do not want to apply asset \"\n \"handles make sure to disable Use asset handles on the \"\n \"publish instance.\"\n )\n )\n\n @classmethod\n def get_invalid(cls, instance):\n\n if not instance.data.get(\"instance_node\"):\n return\n\n rop_node = hou.node(instance.data[\"instance_node\"])\n frame_start = instance.data.get(\"frameStart\")\n frame_end = instance.data.get(\"frameEnd\")\n\n if frame_start is None or frame_end is None:\n cls.log.debug(\n \"Skipping frame range validation for \"\n \"instance without frame data: {}\".format(rop_node.path())\n )\n return\n\n if frame_start > frame_end:\n cls.log.info(\n \"The ROP node render range is set to \"\n \"{0[frameStartHandle]} - {0[frameEndHandle]} \"\n \"The asset handles applied to the instance are start handle \"\n \"{0[handleStart]} and end handle {0[handleEnd]}\"\n .format(instance.data)\n )\n return [rop_node]\n\n @classmethod\n def repair(cls, instance):\n\n if not cls.get_invalid(instance):\n # Already fixed\n return\n\n # Disable use asset handles\n context = instance.context\n create_context = context.data[\"create_context\"]\n instance_id = instance.data.get(\"instance_id\")\n if not instance_id:\n cls.log.debug(\"'{}' must have instance id\"\n .format(instance))\n return\n\n created_instance = create_context.get_instance_by_id(instance_id)\n if not instance_id:\n cls.log.debug(\"Unable to find instance '{}' by id\"\n .format(instance))\n return\n\n created_instance.publish_attributes[\"CollectAssetHandles\"][\"use_handles\"] = False # noqa\n\n create_context.save_changes()\n cls.log.debug(\"use asset handles is turned off for '{}'\"\n .format(instance))\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_frame_token.py",
"content": "import pyblish.api\n\nfrom openpype.hosts.houdini.api import lib\nimport hou\n\n\nclass ValidateFrameToken(pyblish.api.InstancePlugin):\n \"\"\"Validate if the unexpanded string contains the frame ('$F') token.\n\n This validator will *only* check the output parameter of the node if\n the Valid Frame Range is not set to 'Render Current Frame'\n\n Rules:\n If you render out a frame range it is mandatory to have the\n frame token - '$F4' or similar - to ensure that each frame gets\n written. If this is not the case you will override the same file\n every time a frame is written out.\n\n Examples:\n Good: 'my_vbd_cache.$F4.vdb'\n Bad: 'my_vbd_cache.vdb'\n\n \"\"\"\n\n order = pyblish.api.ValidatorOrder\n label = \"Validate Frame Token\"\n families = [\"vdbcache\"]\n\n def process(self, instance):\n\n invalid = self.get_invalid(instance)\n if invalid:\n raise RuntimeError(\n \"Output settings do no match for '%s'\" % instance\n )\n\n @classmethod\n def get_invalid(cls, instance):\n\n node = hou.node(instance.data[\"instance_node\"])\n # Check trange parm, 0 means Render Current Frame\n frame_range = node.evalParm(\"trange\")\n if frame_range == 0:\n return []\n\n output_parm = lib.get_output_parameter(node)\n unexpanded_str = output_parm.unexpandedString()\n\n if \"$F\" not in unexpanded_str:\n cls.log.error(\"No frame token found in '%s'\" % node.path())\n return [instance]\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_houdini_license_category.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\nfrom openpype.pipeline import PublishValidationError\nimport hou\n\n\nclass ValidateHoudiniNotApprenticeLicense(pyblish.api.InstancePlugin):\n \"\"\"Validate the Houdini instance runs a non Apprentice license.\n\n USD ROPs:\n When extracting USD files from an apprentice Houdini license,\n the resulting files will get \"scrambled\" with a license protection\n and get a special .usdnc suffix.\n\n This currently breaks the Subset/representation pipeline so we disallow\n any publish with apprentice license.\n\n Alembic ROPs:\n Houdini Apprentice does not export Alembic.\n \"\"\"\n\n order = pyblish.api.ValidatorOrder\n families = [\"usd\", \"abc\", \"fbx\", \"camera\"]\n hosts = [\"houdini\"]\n label = \"Houdini Apprentice License\"\n\n def process(self, instance):\n\n if hou.isApprentice():\n # Find which family was matched with the plug-in\n families = {instance.data[\"family\"]}\n families.update(instance.data.get(\"families\", []))\n disallowed_families = families.intersection(self.families)\n families = \" \".join(sorted(disallowed_families)).title()\n\n raise PublishValidationError(\n \"{} publishing requires a non apprentice license.\"\n .format(families),\n title=self.label)\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_mesh_is_static.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Validator for correct naming of Static Meshes.\"\"\"\nimport pyblish.api\nfrom openpype.pipeline import (\n PublishValidationError,\n OptionalPyblishPluginMixin\n)\nfrom openpype.pipeline.publish import ValidateContentsOrder\n\nfrom openpype.hosts.houdini.api.action import SelectInvalidAction\nfrom openpype.hosts.houdini.api.lib import get_output_children\n\n\nclass ValidateMeshIsStatic(pyblish.api.InstancePlugin,\n OptionalPyblishPluginMixin):\n \"\"\"Validate mesh is static.\n\n It checks if output node is time dependent.\n \"\"\"\n\n families = [\"staticMesh\"]\n hosts = [\"houdini\"]\n label = \"Validate Mesh is Static\"\n order = ValidateContentsOrder + 0.1\n actions = [SelectInvalidAction]\n\n def process(self, instance):\n\n invalid = self.get_invalid(instance)\n if invalid:\n nodes = [n.path() for n in invalid]\n raise PublishValidationError(\n \"See log for details. \"\n \"Invalid nodes: {0}\".format(nodes)\n )\n\n @classmethod\n def get_invalid(cls, instance):\n\n invalid = []\n\n output_node = instance.data.get(\"output_node\")\n if output_node is None:\n cls.log.debug(\n \"No Output Node, skipping check..\"\n )\n return\n\n all_outputs = get_output_children(output_node)\n\n for output in all_outputs:\n if output.isTimeDependent():\n invalid.append(output)\n cls.log.error(\n \"Output node '%s' is time dependent.\",\n output.path()\n )\n\n return invalid\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_mkpaths_toggled.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\nfrom openpype.pipeline import PublishValidationError\n\n\nclass ValidateIntermediateDirectoriesChecked(pyblish.api.InstancePlugin):\n \"\"\"Validate Create Intermediate Directories is enabled on ROP node.\"\"\"\n\n order = pyblish.api.ValidatorOrder\n families = [\"pointcache\", \"camera\", \"vdbcache\"]\n hosts = [\"houdini\"]\n label = \"Create Intermediate Directories Checked\"\n\n def process(self, instance):\n\n invalid = self.get_invalid(instance)\n if invalid:\n raise PublishValidationError(\n (\"Found ROP node with Create Intermediate \"\n \"Directories turned off: {}\".format(invalid)),\n title=self.label)\n\n @classmethod\n def get_invalid(cls, instance):\n\n result = []\n\n for node in instance[:]:\n if node.parm(\"mkpath\").eval() != 1:\n cls.log.error(\"Invalid settings found on `%s`\" % node.path())\n result.append(node.path())\n\n return result\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_no_errors.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\nimport hou\nfrom openpype.pipeline import PublishValidationError\n\n\ndef cook_in_range(node, start, end):\n current = hou.intFrame()\n if start >= current >= end:\n # Allow cooking current frame since we're in frame range\n node.cook(force=False)\n else:\n node.cook(force=False, frame_range=(start, start))\n\n\ndef get_errors(node):\n \"\"\"Get cooking errors.\n\n If node already has errors check whether it needs to recook\n If so, then recook first to see if that solves it.\n\n \"\"\"\n if node.errors() and node.needsToCook():\n node.cook()\n\n return node.errors()\n\n\nclass ValidateNoErrors(pyblish.api.InstancePlugin):\n \"\"\"Validate the Instance has no current cooking errors.\"\"\"\n\n order = pyblish.api.ValidatorOrder\n hosts = [\"houdini\"]\n label = \"Validate no errors\"\n\n def process(self, instance):\n\n validate_nodes = []\n\n if len(instance) > 0:\n validate_nodes.append(hou.node(instance.data.get(\"instance_node\")))\n output_node = instance.data.get(\"output_node\")\n if output_node:\n validate_nodes.append(output_node)\n\n for node in validate_nodes:\n self.log.debug(\"Validating for errors: %s\" % node.path())\n errors = get_errors(node)\n\n if errors:\n # If there are current errors, then try an unforced cook\n # to see whether the error will disappear.\n self.log.debug(\n \"Recooking to revalidate error \"\n \"is up to date for: %s\" % node.path()\n )\n current_frame = hou.intFrame()\n start = instance.data.get(\"frameStart\", current_frame)\n end = instance.data.get(\"frameEnd\", current_frame)\n cook_in_range(node, start=start, end=end)\n\n # Check for errors again after the forced recook\n errors = get_errors(node)\n if errors:\n self.log.error(errors)\n raise PublishValidationError(\n \"Node has errors: {}\".format(node.path()),\n title=self.label)\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_primitive_hierarchy_paths.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\nfrom openpype.pipeline import PublishValidationError\nfrom openpype.pipeline.publish import (\n ValidateContentsOrder,\n RepairAction,\n)\n\nimport hou\n\n\nclass AddDefaultPathAction(RepairAction):\n label = \"Add a default path attribute\"\n icon = \"mdi.pencil-plus-outline\"\n\n\nclass ValidatePrimitiveHierarchyPaths(pyblish.api.InstancePlugin):\n \"\"\"Validate all primitives build hierarchy from attribute when enabled.\n\n The name of the attribute must exist on the prims and have the same name\n as Build Hierarchy from Attribute's `Path Attribute` value on the Alembic\n ROP node whenever Build Hierarchy from Attribute is enabled.\n\n \"\"\"\n\n order = ValidateContentsOrder + 0.1\n families = [\"abc\"]\n hosts = [\"houdini\"]\n label = \"Validate Prims Hierarchy Path\"\n actions = [AddDefaultPathAction]\n\n def process(self, instance):\n invalid = self.get_invalid(instance)\n if invalid:\n nodes = [n.path() for n in invalid]\n raise PublishValidationError(\n \"See log for details. \" \"Invalid nodes: {0}\".format(nodes),\n title=self.label\n )\n\n @classmethod\n def get_invalid(cls, instance):\n\n output_node = instance.data.get(\"output_node\")\n rop_node = hou.node(instance.data[\"instance_node\"])\n\n if output_node is None:\n cls.log.error(\n \"SOP Output node in '%s' does not exist. \"\n \"Ensure a valid SOP output path is set.\", rop_node.path()\n )\n\n return [rop_node]\n\n build_from_path = rop_node.parm(\"build_from_path\").eval()\n if not build_from_path:\n cls.log.debug(\n \"Alembic ROP has 'Build from Path' disabled. \"\n \"Validation is ignored..\"\n )\n return\n\n path_attr = rop_node.parm(\"path_attrib\").eval()\n if not path_attr:\n cls.log.error(\n \"The Alembic ROP node has no Path Attribute\"\n \"value set, but 'Build Hierarchy from Attribute'\"\n \"is enabled.\"\n )\n return [rop_node]\n\n cls.log.debug(\"Checking for attribute: %s\", path_attr)\n\n if not hasattr(output_node, \"geometry\"):\n # In the case someone has explicitly set an Object\n # node instead of a SOP node in Geometry context\n # then for now we ignore - this allows us to also\n # export object transforms.\n cls.log.warning(\"No geometry output node found, skipping check..\")\n return\n\n # Check if the primitive attribute exists\n frame = instance.data.get(\"frameStart\", 0)\n geo = output_node.geometryAtFrame(frame)\n\n # If there are no primitives on the current frame then we can't\n # check whether the path names are correct. So we'll just issue a\n # warning that the check can't be done consistently and skip\n # validation.\n if len(geo.iterPrims()) == 0:\n cls.log.warning(\n \"No primitives found on current frame. Validation\"\n \" for primitive hierarchy paths will be skipped,\"\n \" thus can't be validated.\"\n )\n return\n\n # Check if there are any values for the primitives\n attrib = geo.findPrimAttrib(path_attr)\n if not attrib:\n cls.log.info(\n \"Geometry Primitives are missing \"\n \"path attribute: `%s`\", path_attr\n )\n return [output_node]\n\n # Ensure at least a single string value is present\n if not attrib.strings():\n cls.log.info(\n \"Primitive path attribute has no \"\n \"string values: %s\", path_attr\n )\n return [output_node]\n\n paths = geo.primStringAttribValues(path_attr)\n # Ensure all primitives are set to a valid path\n # Collect all invalid primitive numbers\n invalid_prims = [i for i, path in enumerate(paths) if not path]\n if invalid_prims:\n num_prims = len(geo.iterPrims()) # faster than len(geo.prims())\n cls.log.info(\n \"Prims have no value for attribute `%s` \"\n \"(%s of %s prims)\", path_attr, len(invalid_prims), num_prims\n )\n return [output_node]\n\n @classmethod\n def repair(cls, instance):\n \"\"\"Add a default path attribute Action.\n\n It is a helper action more than a repair action,\n used to add a default single value for the path.\n \"\"\"\n\n rop_node = hou.node(instance.data[\"instance_node\"])\n output_node = rop_node.parm(\"sop_path\").evalAsNode()\n\n if not output_node:\n cls.log.debug(\n \"Action isn't performed, invalid SOP Path on %s\",\n rop_node\n )\n return\n\n # This check to prevent the action from running multiple times.\n # git_invalid only returns [output_node] when\n # path attribute is the problem\n if cls.get_invalid(instance) != [output_node]:\n return\n\n path_attr = rop_node.parm(\"path_attrib\").eval()\n\n path_node = output_node.parent().createNode(\"name\", \"AUTO_PATH\")\n path_node.parm(\"attribname\").set(path_attr)\n path_node.parm(\"name1\").set('`opname(\"..\")`/`opname(\"..\")`Shape')\n\n cls.log.debug(\n \"'%s' was created. It adds '%s' with a default single value\",\n path_node, path_attr\n )\n\n path_node.setGenericFlag(hou.nodeFlag.DisplayComment, True)\n path_node.setComment(\n 'Auto path node was created automatically by '\n '\"Add a default path attribute\"'\n '\\nFeel free to modify or replace it.'\n )\n\n if output_node.type().name() in [\"null\", \"output\"]:\n # Connect before\n path_node.setFirstInput(output_node.input(0))\n path_node.moveToGoodPosition()\n output_node.setFirstInput(path_node)\n output_node.moveToGoodPosition()\n else:\n # Connect after\n path_node.setFirstInput(output_node)\n rop_node.parm(\"sop_path\").set(path_node.path())\n path_node.moveToGoodPosition()\n\n cls.log.debug(\n \"SOP path on '%s' updated to new output node '%s'\",\n rop_node, path_node\n )\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_remote_publish.py",
"content": "# -*-coding: utf-8 -*-\nimport pyblish.api\n\nfrom openpype.hosts.houdini.api import lib\nfrom openpype.pipeline.publish import RepairContextAction\nfrom openpype.pipeline import PublishValidationError\n\nimport hou\n\n\nclass ValidateRemotePublishOutNode(pyblish.api.ContextPlugin):\n \"\"\"Validate the remote publish out node exists for Deadline to trigger.\"\"\"\n\n order = pyblish.api.ValidatorOrder - 0.4\n families = [\"*\"]\n hosts = [\"houdini\"]\n targets = [\"deadline\"]\n label = \"Remote Publish ROP node\"\n actions = [RepairContextAction]\n\n def process(self, context):\n\n cmd = \"import colorbleed.lib; colorbleed.lib.publish_remote()\"\n\n node = hou.node(\"/out/REMOTE_PUBLISH\")\n if not node:\n raise RuntimeError(\"Missing REMOTE_PUBLISH node.\")\n\n # We ensure it's a shell node and that it has the pre-render script\n # set correctly. Plus the shell script it will trigger should be\n # completely empty (doing nothing)\n if node.type().name() != \"shell\":\n self.raise_error(\"Must be shell ROP node\")\n if node.parm(\"command\").eval() != \"\":\n self.raise_error(\"Must have no command\")\n if node.parm(\"shellexec\").eval():\n self.raise_error(\"Must not execute in shell\")\n if node.parm(\"prerender\").eval() != cmd:\n self.raise_error(\"REMOTE_PUBLISH node does not have \"\n \"correct prerender script.\")\n if node.parm(\"lprerender\").eval() != \"python\":\n self.raise_error(\"REMOTE_PUBLISH node prerender script \"\n \"type not set to 'python'\")\n\n @classmethod\n def repair(cls, context):\n \"\"\"(Re)create the node if it fails to pass validation.\"\"\"\n lib.create_remote_publish_node(force=True)\n\n def raise_error(self, message):\n raise PublishValidationError(message)\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_remote_publish_enabled.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\n\nimport hou\nfrom openpype.pipeline.publish import RepairContextAction\nfrom openpype.pipeline import PublishValidationError\n\n\nclass ValidateRemotePublishEnabled(pyblish.api.ContextPlugin):\n \"\"\"Validate the remote publish node is *not* bypassed.\"\"\"\n\n order = pyblish.api.ValidatorOrder - 0.39\n families = [\"*\"]\n hosts = [\"houdini\"]\n targets = [\"deadline\"]\n label = \"Remote Publish ROP enabled\"\n actions = [RepairContextAction]\n\n def process(self, context):\n\n node = hou.node(\"/out/REMOTE_PUBLISH\")\n if not node:\n raise PublishValidationError(\n \"Missing REMOTE_PUBLISH node.\", title=self.label)\n\n if node.isBypassed():\n raise PublishValidationError(\n \"REMOTE_PUBLISH must not be bypassed.\", title=self.label)\n\n @classmethod\n def repair(cls, context):\n \"\"\"(Re)create the node if it fails to pass validation.\"\"\"\n\n node = hou.node(\"/out/REMOTE_PUBLISH\")\n if not node:\n raise PublishValidationError(\n \"Missing REMOTE_PUBLISH node.\", title=cls.label)\n\n cls.log.info(\"Disabling bypass on /out/REMOTE_PUBLISH\")\n node.bypass(False)\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_review_colorspace.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\nfrom openpype.pipeline import (\n PublishValidationError,\n OptionalPyblishPluginMixin\n)\nfrom openpype.pipeline.publish import RepairAction\nfrom openpype.hosts.houdini.api.action import SelectROPAction\n\nimport os\nimport hou\n\n\nclass SetDefaultViewSpaceAction(RepairAction):\n label = \"Set default view colorspace\"\n icon = \"mdi.monitor\"\n\n\nclass ValidateReviewColorspace(pyblish.api.InstancePlugin,\n OptionalPyblishPluginMixin):\n \"\"\"Validate Review Colorspace parameters.\n\n It checks if 'OCIO Colorspace' parameter was set to valid value.\n \"\"\"\n\n order = pyblish.api.ValidatorOrder + 0.1\n families = [\"review\"]\n hosts = [\"houdini\"]\n label = \"Validate Review Colorspace\"\n actions = [SetDefaultViewSpaceAction, SelectROPAction]\n\n optional = True\n\n def process(self, instance):\n\n if not self.is_active(instance.data):\n return\n\n if os.getenv(\"OCIO\") is None:\n self.log.debug(\n \"Using Houdini's Default Color Management, \"\n \" skipping check..\"\n )\n return\n\n rop_node = hou.node(instance.data[\"instance_node\"])\n if rop_node.evalParm(\"colorcorrect\") != 2:\n # any colorspace settings other than default requires\n # 'Color Correct' parm to be set to 'OpenColorIO'\n raise PublishValidationError(\n \"'Color Correction' parm on '{}' ROP must be set to\"\n \" 'OpenColorIO'\".format(rop_node.path())\n )\n\n if rop_node.evalParm(\"ociocolorspace\") not in \\\n hou.Color.ocio_spaces():\n\n raise PublishValidationError(\n \"Invalid value: Colorspace name doesn't exist.\\n\"\n \"Check 'OCIO Colorspace' parameter on '{}' ROP\"\n .format(rop_node.path())\n )\n\n @classmethod\n def repair(cls, instance):\n \"\"\"Set Default View Space Action.\n\n It is a helper action more than a repair action,\n used to set colorspace on opengl node to the default view.\n \"\"\"\n from openpype.hosts.houdini.api.colorspace import get_default_display_view_colorspace # noqa\n\n rop_node = hou.node(instance.data[\"instance_node\"])\n\n if rop_node.evalParm(\"colorcorrect\") != 2:\n rop_node.setParms({\"colorcorrect\": 2})\n cls.log.debug(\n \"'Color Correction' parm on '{}' has been set to\"\n \" 'OpenColorIO'\".format(rop_node.path())\n )\n\n # Get default view colorspace name\n default_view_space = get_default_display_view_colorspace()\n\n rop_node.setParms({\"ociocolorspace\": default_view_space})\n cls.log.info(\n \"'OCIO Colorspace' parm on '{}' has been set to \"\n \"the default view color space '{}'\"\n .format(rop_node, default_view_space)\n )\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_scene_review.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\nfrom openpype.pipeline import PublishValidationError\nimport hou\n\n\nclass ValidateSceneReview(pyblish.api.InstancePlugin):\n \"\"\"Validator Some Scene Settings before publishing the review\n 1. Scene Path\n 2. Resolution\n \"\"\"\n\n order = pyblish.api.ValidatorOrder\n families = [\"review\"]\n hosts = [\"houdini\"]\n label = \"Scene Setting for review\"\n\n def process(self, instance):\n\n report = []\n instance_node = hou.node(instance.data.get(\"instance_node\"))\n\n invalid = self.get_invalid_scene_path(instance_node)\n if invalid:\n report.append(invalid)\n\n invalid = self.get_invalid_camera_path(instance_node)\n if invalid:\n report.append(invalid)\n\n invalid = self.get_invalid_resolution(instance_node)\n if invalid:\n report.extend(invalid)\n\n if report:\n raise PublishValidationError(\n \"\\n\\n\".join(report),\n title=self.label)\n\n def get_invalid_scene_path(self, rop_node):\n scene_path_parm = rop_node.parm(\"scenepath\")\n scene_path_node = scene_path_parm.evalAsNode()\n if not scene_path_node:\n path = scene_path_parm.evalAsString()\n return \"Scene path does not exist: '{}'\".format(path)\n\n def get_invalid_camera_path(self, rop_node):\n camera_path_parm = rop_node.parm(\"camera\")\n camera_node = camera_path_parm.evalAsNode()\n path = camera_path_parm.evalAsString()\n if not camera_node:\n return \"Camera path does not exist: '{}'\".format(path)\n type_name = camera_node.type().name()\n if type_name != \"cam\":\n return \"Camera path is not a camera: '{}' (type: {})\".format(\n path, type_name\n )\n\n def get_invalid_resolution(self, rop_node):\n\n # The resolution setting is only used when Override Camera Resolution\n # is enabled. So we skip validation if it is disabled.\n override = rop_node.parm(\"tres\").eval()\n if not override:\n return\n\n invalid = []\n res_width = rop_node.parm(\"res1\").eval()\n res_height = rop_node.parm(\"res2\").eval()\n if res_width == 0:\n invalid.append(\"Override Resolution width is set to zero.\")\n if res_height == 0:\n invalid.append(\"Override Resolution height is set to zero\")\n\n return invalid\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_sop_output_node.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\nfrom openpype.pipeline import PublishValidationError\nfrom openpype.hosts.houdini.api.action import (\n SelectInvalidAction,\n SelectROPAction,\n)\n\nimport hou\n\n\nclass ValidateSopOutputNode(pyblish.api.InstancePlugin):\n \"\"\"Validate the instance SOP Output Node.\n\n This will ensure:\n - The SOP Path is set.\n - The SOP Path refers to an existing object.\n - The SOP Path node is a SOP node.\n - The SOP Path node has at least one input connection (has an input)\n - The SOP Path has geometry data.\n\n \"\"\"\n\n order = pyblish.api.ValidatorOrder\n families = [\"pointcache\", \"vdbcache\"]\n hosts = [\"houdini\"]\n label = \"Validate Output Node (SOP)\"\n actions = [SelectROPAction, SelectInvalidAction]\n\n def process(self, instance):\n\n invalid = self.get_invalid(instance)\n if invalid:\n raise PublishValidationError(\n \"Output node(s) are incorrect\",\n title=\"Invalid output node(s)\"\n )\n\n @classmethod\n def get_invalid(cls, instance):\n output_node = instance.data.get(\"output_node\")\n\n if output_node is None:\n node = hou.node(instance.data[\"instance_node\"])\n cls.log.error(\n \"SOP Output node in '%s' does not exist. \"\n \"Ensure a valid SOP output path is set.\" % node.path()\n )\n\n return [node]\n\n # Output node must be a Sop node.\n if not isinstance(output_node, hou.SopNode):\n cls.log.error(\n \"Output node %s is not a SOP node. \"\n \"SOP Path must point to a SOP node, \"\n \"instead found category type: %s\"\n % (output_node.path(), output_node.type().category().name())\n )\n return [output_node]\n\n # For the sake of completeness also assert the category type\n # is Sop to avoid potential edge case scenarios even though\n # the isinstance check above should be stricter than this category\n if output_node.type().category().name() != \"Sop\":\n raise PublishValidationError(\n (\"Output node {} is not of category Sop. \"\n \"This is a bug.\").format(output_node.path()),\n title=cls.label)\n\n # Ensure the node is cooked and succeeds to cook so we can correctly\n # check for its geometry data.\n if output_node.needsToCook():\n cls.log.debug(\"Cooking node: %s\" % output_node.path())\n try:\n output_node.cook()\n except hou.Error as exc:\n cls.log.error(\"Cook failed: %s\" % exc)\n cls.log.error(output_node.errors()[0])\n return [output_node]\n\n # Ensure the output node has at least Geometry data\n if not output_node.geometry():\n cls.log.error(\n \"Output node `%s` has no geometry data.\" % output_node.path()\n )\n return [output_node]\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_subset_name.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Validator for correct naming of Static Meshes.\"\"\"\nimport pyblish.api\nfrom openpype.pipeline import (\n PublishValidationError,\n OptionalPyblishPluginMixin\n)\nfrom openpype.pipeline.publish import (\n ValidateContentsOrder,\n RepairAction,\n)\nfrom openpype.hosts.houdini.api.action import SelectInvalidAction\nfrom openpype.pipeline.create import get_subset_name\n\nimport hou\n\n\nclass FixSubsetNameAction(RepairAction):\n label = \"Fix Subset Name\"\n\n\nclass ValidateSubsetName(pyblish.api.InstancePlugin,\n OptionalPyblishPluginMixin):\n \"\"\"Validate Subset name.\n\n \"\"\"\n\n families = [\"staticMesh\"]\n hosts = [\"houdini\"]\n label = \"Validate Subset Name\"\n order = ValidateContentsOrder + 0.1\n actions = [FixSubsetNameAction, SelectInvalidAction]\n\n optional = True\n\n def process(self, instance):\n\n if not self.is_active(instance.data):\n return\n\n invalid = self.get_invalid(instance)\n if invalid:\n nodes = [n.path() for n in invalid]\n raise PublishValidationError(\n \"See log for details. \"\n \"Invalid nodes: {0}\".format(nodes)\n )\n\n @classmethod\n def get_invalid(cls, instance):\n\n invalid = []\n\n rop_node = hou.node(instance.data[\"instance_node\"])\n\n # Check subset name\n asset_doc = instance.data[\"assetEntity\"]\n subset_name = get_subset_name(\n family=instance.data[\"family\"],\n variant=instance.data[\"variant\"],\n task_name=instance.data[\"task\"],\n asset_doc=asset_doc,\n dynamic_data={\"asset\": asset_doc[\"name\"]}\n )\n\n if instance.data.get(\"subset\") != subset_name:\n invalid.append(rop_node)\n cls.log.error(\n \"Invalid subset name on rop node '%s' should be '%s'.\",\n rop_node.path(), subset_name\n )\n\n return invalid\n\n @classmethod\n def repair(cls, instance):\n rop_node = hou.node(instance.data[\"instance_node\"])\n\n # Check subset name\n asset_doc = instance.data[\"assetEntity\"]\n subset_name = get_subset_name(\n family=instance.data[\"family\"],\n variant=instance.data[\"variant\"],\n task_name=instance.data[\"task\"],\n asset_doc=asset_doc,\n dynamic_data={\"asset\": asset_doc[\"name\"]}\n )\n\n instance.data[\"subset\"] = subset_name\n rop_node.parm(\"subset\").set(subset_name)\n\n cls.log.debug(\n \"Subset name on rop node '%s' has been set to '%s'.\",\n rop_node.path(), subset_name\n )\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_unreal_staticmesh_naming.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Validator for correct naming of Static Meshes.\"\"\"\nimport pyblish.api\nfrom openpype.pipeline import (\n PublishValidationError,\n OptionalPyblishPluginMixin\n)\nfrom openpype.pipeline.publish import ValidateContentsOrder\n\nfrom openpype.hosts.houdini.api.action import SelectInvalidAction\nfrom openpype.hosts.houdini.api.lib import get_output_children\n\nimport hou\n\n\nclass ValidateUnrealStaticMeshName(pyblish.api.InstancePlugin,\n OptionalPyblishPluginMixin):\n \"\"\"Validate name of Unreal Static Mesh.\n\n This validator checks if output node name has a collision prefix:\n - UBX\n - UCP\n - USP\n - UCX\n\n This validator also checks if subset name is correct\n - {static mesh prefix}_{Asset-Name}{Variant}.\n\n \"\"\"\n\n families = [\"staticMesh\"]\n hosts = [\"houdini\"]\n label = \"Unreal Static Mesh Name (FBX)\"\n order = ValidateContentsOrder + 0.1\n actions = [SelectInvalidAction]\n\n optional = True\n collision_prefixes = []\n static_mesh_prefix = \"\"\n\n @classmethod\n def apply_settings(cls, project_settings, system_settings):\n\n settings = (\n project_settings[\"houdini\"][\"create\"][\"CreateStaticMesh\"]\n )\n cls.collision_prefixes = settings[\"collision_prefixes\"]\n cls.static_mesh_prefix = settings[\"static_mesh_prefix\"]\n\n def process(self, instance):\n\n if not self.is_active(instance.data):\n return\n\n invalid = self.get_invalid(instance)\n if invalid:\n nodes = [n.path() for n in invalid]\n raise PublishValidationError(\n \"See log for details. \"\n \"Invalid nodes: {0}\".format(nodes)\n )\n\n @classmethod\n def get_invalid(cls, instance):\n\n invalid = []\n\n rop_node = hou.node(instance.data[\"instance_node\"])\n output_node = instance.data.get(\"output_node\")\n if output_node is None:\n cls.log.debug(\n \"No Output Node, skipping check..\"\n )\n return\n\n if rop_node.evalParm(\"buildfrompath\"):\n # This validator doesn't support naming check if\n # building hierarchy from path' is used\n cls.log.info(\n \"Using 'Build Hierarchy from Path Attribute', skipping check..\"\n )\n return\n\n # Check nodes names\n all_outputs = get_output_children(output_node, include_sops=False)\n for output in all_outputs:\n for prefix in cls.collision_prefixes:\n if output.name().startswith(prefix):\n invalid.append(output)\n cls.log.error(\n \"Invalid node name: Node '%s' \"\n \"includes a collision prefix '%s'\",\n output.path(), prefix\n )\n break\n\n return invalid\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_usd_layer_path_backslashes.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\n\nimport openpype.hosts.houdini.api.usd as hou_usdlib\nfrom openpype.pipeline import PublishValidationError\n\nimport hou\n\n\nclass ValidateUSDLayerPathBackslashes(pyblish.api.InstancePlugin):\n \"\"\"Validate USD loaded paths have no backslashes.\n\n This is a crucial validation for HUSK USD rendering as Houdini's\n USD Render ROP will fail to write out a .usd file for rendering that\n correctly preserves the backslashes, e.g. it will incorrectly convert a\n '\\t' to a TAB character disallowing HUSK to find those specific files.\n\n This validation is redundant for usdModel since that flattens the model\n before write. As such it will never have any used layers with a path.\n\n \"\"\"\n\n order = pyblish.api.ValidatorOrder\n families = [\"usdSetDress\", \"usdShade\", \"usd\", \"usdrender\"]\n hosts = [\"houdini\"]\n label = \"USD Layer path backslashes\"\n optional = True\n\n def process(self, instance):\n\n rop = hou.node(instance.data.get(\"instance_node\"))\n lop_path = hou_usdlib.get_usd_rop_loppath(rop)\n stage = lop_path.stage(apply_viewport_overrides=False)\n\n invalid = []\n for layer in stage.GetUsedLayers():\n references = layer.externalReferences\n\n for ref in references:\n\n # Ignore anonymous layers\n if ref.startswith(\"anon:\"):\n continue\n\n # If any backslashes in the path consider it invalid\n if \"\\\\\" in ref:\n self.log.error(\"Found invalid path: %s\" % ref)\n invalid.append(layer)\n\n if invalid:\n raise PublishValidationError((\n \"Loaded layers have backslashes. \"\n \"This is invalid for HUSK USD rendering.\"),\n title=self.label)\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_usd_model_and_shade.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\n\nimport openpype.hosts.houdini.api.usd as hou_usdlib\nfrom openpype.pipeline import PublishValidationError\n\nfrom pxr import UsdShade, UsdRender, UsdLux\n\nimport hou\n\n\ndef fullname(o):\n \"\"\"Get fully qualified class name\"\"\"\n module = o.__module__\n if module is None or module == str.__module__:\n return o.__name__\n return module + \".\" + o.__name__\n\n\nclass ValidateUsdModel(pyblish.api.InstancePlugin):\n \"\"\"Validate USD Model.\n\n Disallow Shaders, Render settings, products and vars and Lux lights.\n\n \"\"\"\n\n order = pyblish.api.ValidatorOrder\n families = [\"usdModel\"]\n hosts = [\"houdini\"]\n label = \"Validate USD Model\"\n optional = True\n\n disallowed = [\n UsdShade.Shader,\n UsdRender.Settings,\n UsdRender.Product,\n UsdRender.Var,\n UsdLux.Light,\n ]\n\n def process(self, instance):\n\n rop = hou.node(instance.data.get(\"instance_node\"))\n lop_path = hou_usdlib.get_usd_rop_loppath(rop)\n stage = lop_path.stage(apply_viewport_overrides=False)\n\n invalid = []\n for prim in stage.Traverse():\n\n for klass in self.disallowed:\n if klass(prim):\n # Get full class name without pxr. prefix\n name = fullname(klass).split(\"pxr.\", 1)[-1]\n path = str(prim.GetPath())\n self.log.warning(\"Disallowed %s: %s\" % (name, path))\n\n invalid.append(prim)\n\n if invalid:\n prim_paths = sorted([str(prim.GetPath()) for prim in invalid])\n raise PublishValidationError(\n \"Found invalid primitives: {}\".format(prim_paths))\n\n\nclass ValidateUsdShade(ValidateUsdModel):\n \"\"\"Validate usdShade.\n\n Disallow Render settings, products, vars and Lux lights.\n\n \"\"\"\n\n families = [\"usdShade\"]\n label = \"Validate USD Shade\"\n\n disallowed = [\n UsdRender.Settings,\n UsdRender.Product,\n UsdRender.Var,\n UsdLux.Light,\n ]\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_usd_output_node.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\nfrom openpype.pipeline import PublishValidationError\n\n\nclass ValidateUSDOutputNode(pyblish.api.InstancePlugin):\n \"\"\"Validate the instance USD LOPs Output Node.\n\n This will ensure:\n - The LOP Path is set.\n - The LOP Path refers to an existing object.\n - The LOP Path node is a LOP node.\n\n \"\"\"\n\n order = pyblish.api.ValidatorOrder\n families = [\"usd\"]\n hosts = [\"houdini\"]\n label = \"Validate Output Node (USD)\"\n\n def process(self, instance):\n\n invalid = self.get_invalid(instance)\n if invalid:\n raise PublishValidationError(\n (\"Output node(s) `{}` are incorrect. \"\n \"See plug-in log for details.\").format(invalid),\n title=self.label\n )\n\n @classmethod\n def get_invalid(cls, instance):\n\n import hou\n\n output_node = instance.data[\"output_node\"]\n\n if output_node is None:\n node = hou.node(instance.data.get(\"instance_node\"))\n cls.log.error(\n \"USD node '%s' LOP path does not exist. \"\n \"Ensure a valid LOP path is set.\" % node.path()\n )\n\n return [node.path()]\n\n # Output node must be a Sop node.\n if not isinstance(output_node, hou.LopNode):\n cls.log.error(\n \"Output node %s is not a LOP node. \"\n \"LOP Path must point to a LOP node, \"\n \"instead found category type: %s\"\n % (output_node.path(), output_node.type().category().name())\n )\n return [output_node.path()]\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_usd_render_product_names.py",
"content": "# -*- coding: utf-8 -*-\nimport os\nimport pyblish.api\n\nfrom openpype.pipeline import PublishValidationError\n\n\nclass ValidateUSDRenderProductNames(pyblish.api.InstancePlugin):\n \"\"\"Validate USD Render Product names are correctly set absolute paths.\"\"\"\n\n order = pyblish.api.ValidatorOrder\n families = [\"usdrender\"]\n hosts = [\"houdini\"]\n label = \"Validate USD Render Product Names\"\n optional = True\n\n def process(self, instance):\n\n invalid = []\n for filepath in instance.data[\"files\"]:\n\n if not filepath:\n invalid.append(\"Detected empty output filepath.\")\n\n if not os.path.isabs(filepath):\n invalid.append(\n \"Output file path is not absolute path: %s\" % filepath\n )\n\n if invalid:\n for message in invalid:\n self.log.error(message)\n raise PublishValidationError(\n \"USD Render Paths are invalid.\", title=self.label)\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_usd_setdress.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\n\nimport openpype.hosts.houdini.api.usd as hou_usdlib\nfrom openpype.pipeline import PublishValidationError\n\n\nclass ValidateUsdSetDress(pyblish.api.InstancePlugin):\n \"\"\"Validate USD Set Dress.\n\n Must only have references or payloads. May not generate new mesh or\n flattened meshes.\n\n \"\"\"\n\n order = pyblish.api.ValidatorOrder\n families = [\"usdSetDress\"]\n hosts = [\"houdini\"]\n label = \"Validate USD Set Dress\"\n optional = True\n\n def process(self, instance):\n\n from pxr import UsdGeom\n import hou\n\n rop = hou.node(instance.data.get(\"instance_node\"))\n lop_path = hou_usdlib.get_usd_rop_loppath(rop)\n stage = lop_path.stage(apply_viewport_overrides=False)\n\n invalid = []\n for node in stage.Traverse():\n\n if UsdGeom.Mesh(node):\n # This solely checks whether there is any USD involved\n # in this Prim's Stack and doesn't accurately tell us\n # whether it was generated locally or not.\n # TODO: More accurately track whether the Prim was created\n # in the local scene\n stack = node.GetPrimStack()\n for sdf in stack:\n path = sdf.layer.realPath\n if path:\n break\n else:\n prim_path = node.GetPath()\n self.log.error(\n \"%s is not referenced geometry.\" % prim_path\n )\n invalid.append(node)\n\n if invalid:\n raise PublishValidationError((\n \"SetDress contains local geometry. \"\n \"This is not allowed, it must be an assembly \"\n \"of referenced assets.\"),\n title=self.label\n )\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_usd_shade_model_exists.py",
"content": "# -*- coding: utf-8 -*-\nimport re\n\nimport pyblish.api\n\nfrom openpype.client import get_subset_by_name\nfrom openpype.pipeline.publish import ValidateContentsOrder\nfrom openpype.pipeline import PublishValidationError\n\n\nclass ValidateUSDShadeModelExists(pyblish.api.InstancePlugin):\n \"\"\"Validate the Instance has no current cooking errors.\"\"\"\n\n order = ValidateContentsOrder\n hosts = [\"houdini\"]\n families = [\"usdShade\"]\n label = \"USD Shade model exists\"\n\n def process(self, instance):\n project_name = instance.context.data[\"projectName\"]\n asset_name = instance.data[\"asset\"]\n subset = instance.data[\"subset\"]\n\n # Assume shading variation starts after a dot separator\n shade_subset = subset.split(\".\", 1)[0]\n model_subset = re.sub(\"^usdShade\", \"usdModel\", shade_subset)\n\n asset_doc = instance.data.get(\"assetEntity\")\n if not asset_doc:\n raise RuntimeError(\"Asset document is not filled on instance.\")\n\n subset_doc = get_subset_by_name(\n project_name, model_subset, asset_doc[\"_id\"], fields=[\"_id\"]\n )\n if not subset_doc:\n raise PublishValidationError(\n (\"USD Model subset not found: \"\n \"{} ({})\").format(model_subset, asset_name),\n title=self.label\n )\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_usd_shade_workspace.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\nfrom openpype.pipeline import PublishValidationError\n\nimport hou\n\n\nclass ValidateUsdShadeWorkspace(pyblish.api.InstancePlugin):\n \"\"\"Validate USD Shading Workspace is correct version.\n\n There have been some issues with outdated/erroneous Shading Workspaces\n so this is to confirm everything is set as it should.\n\n \"\"\"\n\n order = pyblish.api.ValidatorOrder\n hosts = [\"houdini\"]\n families = [\"usdShade\"]\n label = \"USD Shade Workspace\"\n\n def process(self, instance):\n\n rop = hou.node(instance.data.get(\"instance_node\"))\n workspace = rop.parent()\n\n definition = workspace.type().definition()\n name = definition.nodeType().name()\n library = definition.libraryFilePath()\n\n all_definitions = hou.hda.definitionsInFile(library)\n node_type, version = name.rsplit(\":\", 1)\n version = float(version)\n\n highest = version\n for other_definition in all_definitions:\n other_name = other_definition.nodeType().name()\n other_node_type, other_version = other_name.rsplit(\":\", 1)\n other_version = float(other_version)\n\n if node_type != other_node_type:\n continue\n\n # Get the highest version\n highest = max(highest, other_version)\n\n if version != highest:\n raise PublishValidationError(\n (\"Shading Workspace is not the latest version.\"\n \" Found {}. Latest is {}.\").format(version, highest),\n title=self.label\n )\n\n # There were some issues with the editable node not having the right\n # configured path. So for now let's assure that is correct to.from\n value = (\n 'avalon://`chs(\"../asset_name\")`/'\n 'usdShade`chs(\"../model_variantname1\")`.usd'\n )\n rop_value = rop.parm(\"lopoutput\").rawValue()\n if rop_value != value:\n raise PublishValidationError(\n (\"Shading Workspace has invalid 'lopoutput'\"\n \" parameter value. The Shading Workspace\"\n \" needs to be reset to its default values.\"),\n title=self.label\n )\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_vdb_output_node.py",
"content": "# -*- coding: utf-8 -*-\nimport contextlib\n\nimport pyblish.api\nimport hou\n\nfrom openpype.pipeline import PublishXmlValidationError\nfrom openpype.hosts.houdini.api.action import SelectInvalidAction\n\n\ndef group_consecutive_numbers(nums):\n \"\"\"\n Args:\n nums (list): List of sorted integer numbers.\n\n Yields:\n str: Group ranges as {start}-{end} if more than one number in the range\n else it yields {end}\n\n \"\"\"\n start = None\n end = None\n\n def _result(a, b):\n if a == b:\n return \"{}\".format(a)\n else:\n return \"{}-{}\".format(a, b)\n\n for num in nums:\n if start is None:\n start = num\n end = num\n elif num == end + 1:\n end = num\n else:\n yield _result(start, end)\n start = num\n end = num\n if start is not None:\n yield _result(start, end)\n\n\n@contextlib.contextmanager\ndef update_mode_context(mode):\n original = hou.updateModeSetting()\n try:\n hou.setUpdateMode(mode)\n yield\n finally:\n hou.setUpdateMode(original)\n\n\ndef get_geometry_at_frame(sop_node, frame, force=True):\n \"\"\"Return geometry at frame but force a cooked value.\"\"\"\n if not hasattr(sop_node, \"geometry\"):\n return\n with update_mode_context(hou.updateMode.AutoUpdate):\n sop_node.cook(force=force, frame_range=(frame, frame))\n return sop_node.geometryAtFrame(frame)\n\n\nclass ValidateVDBOutputNode(pyblish.api.InstancePlugin):\n \"\"\"Validate that the node connected to the output node is of type VDB.\n\n All primitives of the output geometry must be VDBs, no other primitive\n types are allowed. That means that regardless of the amount of VDBs in the\n geometry it will have an equal amount of VDBs, points, primitives and\n vertices since each VDB primitive is one point, one vertex and one VDB.\n\n This validation only checks the geometry on the first frame of the export\n frame range for optimization purposes.\n\n A VDB is an inherited type of Prim, holds the following data:\n - Primitives: 1\n - Points: 1\n - Vertices: 1\n - VDBs: 1\n\n \"\"\"\n\n order = pyblish.api.ValidatorOrder + 0.1\n families = [\"vdbcache\"]\n hosts = [\"houdini\"]\n label = \"Validate Output Node (VDB)\"\n actions = [SelectInvalidAction]\n\n def process(self, instance):\n invalid_nodes, message = self.get_invalid_with_message(instance)\n if invalid_nodes:\n\n # instance_node is str, but output_node is hou.Node so we convert\n output = instance.data.get(\"output_node\")\n output_path = output.path() if output else None\n\n raise PublishXmlValidationError(\n self,\n \"Invalid VDB content: {}\".format(message),\n formatting_data={\n \"message\": message,\n \"rop_path\": instance.data.get(\"instance_node\"),\n \"sop_path\": output_path\n }\n )\n\n @classmethod\n def get_invalid_with_message(cls, instance):\n\n node = instance.data.get(\"output_node\")\n if node is None:\n instance_node = instance.data.get(\"instance_node\")\n error = (\n \"SOP path is not correctly set on \"\n \"ROP node `{}`.\".format(instance_node)\n )\n return [hou.node(instance_node), error]\n\n frame = instance.data.get(\"frameStart\", 0)\n geometry = get_geometry_at_frame(node, frame)\n if geometry is None:\n # No geometry data on this node, maybe the node hasn't cooked?\n error = (\n \"SOP node `{}` has no geometry data. \"\n \"Was it unable to cook?\".format(node.path())\n )\n return [node, error]\n\n num_prims = geometry.intrinsicValue(\"primitivecount\")\n num_points = geometry.intrinsicValue(\"pointcount\")\n if num_prims == 0 and num_points == 0:\n # Since we are only checking the first frame it doesn't mean there\n # won't be VDB prims in a few frames. As such we'll assume for now\n # the user knows what he or she is doing\n cls.log.warning(\n \"SOP node `{}` has no primitives on start frame {}. \"\n \"Validation is skipped and it is assumed elsewhere in the \"\n \"frame range VDB prims and only VDB prims will exist.\"\n \"\".format(node.path(), int(frame))\n )\n return [None, None]\n\n num_vdb_prims = geometry.countPrimType(hou.primType.VDB)\n cls.log.debug(\"Detected {} VDB primitives\".format(num_vdb_prims))\n if num_prims != num_vdb_prims:\n # There's at least one primitive that is not a VDB.\n # Search them and report them to the artist.\n prims = geometry.prims()\n invalid_prims = [prim for prim in prims\n if not isinstance(prim, hou.VDB)]\n if invalid_prims:\n # Log prim numbers as consecutive ranges so logging isn't very\n # slow for large number of primitives\n error = (\n \"Found non-VDB primitives for `{}`. \"\n \"Primitive indices {} are not VDB primitives.\".format(\n node.path(),\n \", \".join(group_consecutive_numbers(\n prim.number() for prim in invalid_prims\n ))\n )\n )\n return [node, error]\n\n if num_points != num_vdb_prims:\n # We have points unrelated to the VDB primitives.\n error = (\n \"The number of primitives and points do not match in '{}'. \"\n \"This likely means you have unconnected points, which we do \"\n \"not allow in the VDB output.\".format(node.path()))\n return [node, error]\n\n return [None, None]\n\n @classmethod\n def get_invalid(cls, instance):\n nodes, _ = cls.get_invalid_with_message(instance)\n return nodes\n"
},
{
"path": "openpype/hosts/houdini/plugins/publish/validate_workfile_paths.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\nimport hou\nfrom openpype.pipeline import (\n PublishValidationError,\n OptionalPyblishPluginMixin\n)\nfrom openpype.pipeline.publish import RepairAction\n\n\nclass ValidateWorkfilePaths(\n pyblish.api.InstancePlugin, OptionalPyblishPluginMixin):\n \"\"\"Validate workfile paths so they are absolute.\"\"\"\n\n order = pyblish.api.ValidatorOrder\n families = [\"workfile\"]\n hosts = [\"houdini\"]\n label = \"Validate Workfile Paths\"\n actions = [RepairAction]\n optional = True\n\n node_types = [\"file\", \"alembic\"]\n prohibited_vars = [\"$HIP\", \"$JOB\"]\n\n def process(self, instance):\n if not self.is_active(instance.data):\n return\n invalid = self.get_invalid()\n self.log.debug(\n \"Checking node types: {}\".format(\", \".join(self.node_types)))\n self.log.debug(\n \"Searching prohibited vars: {}\".format(\n \", \".join(self.prohibited_vars)\n )\n )\n\n if invalid:\n all_container_vars = set()\n for param in invalid:\n value = param.unexpandedString()\n contained_vars = [\n var for var in self.prohibited_vars\n if var in value\n ]\n all_container_vars.update(contained_vars)\n\n self.log.error(\n \"Parm {} contains prohibited vars {}: {}\".format(\n param.path(),\n \", \".join(contained_vars),\n value)\n )\n\n message = (\n \"Prohibited vars {} found in parameter values\".format(\n \", \".join(all_container_vars)\n )\n )\n raise PublishValidationError(message, title=self.label)\n\n @classmethod\n def get_invalid(cls):\n invalid = []\n for param, _ in hou.fileReferences():\n # it might return None for some reason\n if not param:\n continue\n # skip nodes we are not interested in\n if param.node().type().name() not in cls.node_types:\n continue\n\n if any(\n v for v in cls.prohibited_vars\n if v in param.unexpandedString()):\n invalid.append(param)\n\n return invalid\n\n @classmethod\n def repair(cls, instance):\n invalid = cls.get_invalid()\n for param in invalid:\n cls.log.info(\"Processing: {}\".format(param.path()))\n cls.log.info(\"Replacing {} for {}\".format(\n param.unexpandedString(),\n hou.text.expandString(param.unexpandedString())))\n param.set(hou.text.expandString(param.unexpandedString()))\n"
},
{
"path": "openpype/hosts/houdini/startup/MainMenuCommon.xml",
"content": "\n\n \n \n \n\n"
},
{
"path": "openpype/hosts/houdini/startup/python2.7libs/pythonrc.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"OpenPype startup script.\"\"\"\nfrom openpype.pipeline import install_host\nfrom openpype.hosts.houdini.api import HoudiniHost\nfrom openpype import AYON_SERVER_ENABLED\n\n\ndef main():\n print(\"Installing {} ...\".format(\n \"AYON\" if AYON_SERVER_ENABLED else \"OpenPype\"))\n install_host(HoudiniHost())\n\n\nmain()\n"
},
{
"path": "openpype/hosts/houdini/startup/python3.10libs/pythonrc.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"OpenPype startup script.\"\"\"\nfrom openpype.pipeline import install_host\nfrom openpype.hosts.houdini.api import HoudiniHost\nfrom openpype import AYON_SERVER_ENABLED\n\n\ndef main():\n print(\"Installing {} ...\".format(\n \"AYON\" if AYON_SERVER_ENABLED else \"OpenPype\"))\n install_host(HoudiniHost())\n\n\nmain()\n"
},
{
"path": "openpype/hosts/houdini/startup/python3.7libs/pythonrc.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"OpenPype startup script.\"\"\"\nfrom openpype.pipeline import install_host\nfrom openpype.hosts.houdini.api import HoudiniHost\nfrom openpype import AYON_SERVER_ENABLED\n\n\ndef main():\n print(\"Installing {} ...\".format(\n \"AYON\" if AYON_SERVER_ENABLED else \"OpenPype\"))\n install_host(HoudiniHost())\n\n\nmain()\n"
},
{
"path": "openpype/hosts/houdini/startup/python3.9libs/pythonrc.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"OpenPype startup script.\"\"\"\nfrom openpype.pipeline import install_host\nfrom openpype.hosts.houdini.api import HoudiniHost\nfrom openpype import AYON_SERVER_ENABLED\n\n\ndef main():\n print(\"Installing {} ...\".format(\n \"AYON\" if AYON_SERVER_ENABLED else \"OpenPype\"))\n install_host(HoudiniHost())\n\n\nmain()\n"
},
{
"path": "openpype/hosts/max/__init__.py",
"content": "from .addon import (\n MaxAddon,\n MAX_HOST_DIR,\n)\n\n\n__all__ = (\n \"MaxAddon\",\n \"MAX_HOST_DIR\",\n)\n"
},
{
"path": "openpype/hosts/max/addon.py",
"content": "# -*- coding: utf-8 -*-\nimport os\nfrom openpype.modules import OpenPypeModule, IHostAddon\n\nMAX_HOST_DIR = os.path.dirname(os.path.abspath(__file__))\n\n\nclass MaxAddon(OpenPypeModule, IHostAddon):\n name = \"max\"\n host_name = \"max\"\n\n def initialize(self, module_settings):\n self.enabled = True\n\n def add_implementation_envs(self, env, _app):\n # Remove auto screen scale factor for Qt\n # - let 3dsmax decide it's value\n env.pop(\"QT_AUTO_SCREEN_SCALE_FACTOR\", None)\n\n def get_workfile_extensions(self):\n return [\".max\"]\n\n def get_launch_hook_paths(self, app):\n if app.host_name != self.host_name:\n return []\n return [\n os.path.join(MAX_HOST_DIR, \"hooks\")\n ]\n"
},
{
"path": "openpype/hosts/max/api/__init__.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Public API for 3dsmax\"\"\"\n\nfrom .pipeline import (\n MaxHost,\n)\n\n\nfrom .lib import (\n maintained_selection,\n lsattr,\n get_all_children\n)\n\n__all__ = [\n \"MaxHost\",\n \"maintained_selection\",\n \"lsattr\",\n \"get_all_children\"\n]\n"
},
{
"path": "openpype/hosts/max/api/action.py",
"content": "from pymxs import runtime as rt\n\nimport pyblish.api\n\nfrom openpype.pipeline.publish import get_errored_instances_from_context\n\n\nclass SelectInvalidAction(pyblish.api.Action):\n \"\"\"Select invalid objects in Blender when a publish plug-in failed.\"\"\"\n label = \"Select Invalid\"\n on = \"failed\"\n icon = \"search\"\n\n def process(self, context, plugin):\n errored_instances = get_errored_instances_from_context(context,\n plugin=plugin)\n\n # Get the invalid nodes for the plug-ins\n self.log.info(\"Finding invalid nodes...\")\n invalid = list()\n for instance in errored_instances:\n invalid_nodes = plugin.get_invalid(instance)\n if invalid_nodes:\n if isinstance(invalid_nodes, (list, tuple)):\n invalid.extend(invalid_nodes)\n else:\n self.log.warning(\n \"Failed plug-in doesn't have any selectable objects.\"\n )\n\n if not invalid:\n self.log.info(\"No invalid nodes found.\")\n return\n invalid_names = [obj.name for obj in invalid if isinstance(obj, str)]\n if not invalid_names:\n invalid_names = [obj.name for obj, _ in invalid]\n invalid = [obj for obj, _ in invalid]\n self.log.info(\n \"Selecting invalid objects: %s\", \", \".join(invalid_names)\n )\n\n rt.Select(invalid)\n"
},
{
"path": "openpype/hosts/max/api/colorspace.py",
"content": "import attr\nfrom pymxs import runtime as rt\n\n\n@attr.s\nclass LayerMetadata(object):\n \"\"\"Data class for Render Layer metadata.\"\"\"\n frameStart = attr.ib()\n frameEnd = attr.ib()\n\n\n@attr.s\nclass RenderProduct(object):\n \"\"\"Getting Colorspace as\n Specific Render Product Parameter for submitting\n publish job.\n \"\"\"\n colorspace = attr.ib() # colorspace\n view = attr.ib()\n productName = attr.ib(default=None)\n\n\nclass ARenderProduct(object):\n\n def __init__(self):\n \"\"\"Constructor.\"\"\"\n # Initialize\n self.layer_data = self._get_layer_data()\n self.layer_data.products = self.get_colorspace_data()\n\n def _get_layer_data(self):\n return LayerMetadata(\n frameStart=int(rt.rendStart),\n frameEnd=int(rt.rendEnd),\n )\n\n def get_colorspace_data(self):\n \"\"\"To be implemented by renderer class.\n This should return a list of RenderProducts.\n Returns:\n list: List of RenderProduct\n \"\"\"\n colorspace_data = [\n RenderProduct(\n colorspace=\"sRGB\",\n view=\"ACES 1.0\",\n productName=\"\"\n )\n ]\n return colorspace_data\n"
},
{
"path": "openpype/hosts/max/api/lib.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Library of functions useful for 3dsmax pipeline.\"\"\"\nimport contextlib\nimport logging\nimport json\nfrom typing import Any, Dict, Union\n\nimport six\nfrom openpype.pipeline import get_current_project_name, colorspace\nfrom openpype.settings import get_project_settings\nfrom openpype.pipeline.context_tools import (\n get_current_project, get_current_project_asset)\nfrom openpype.style import load_stylesheet\nfrom pymxs import runtime as rt\n\n\nJSON_PREFIX = \"JSON::\"\nlog = logging.getLogger(\"openpype.hosts.max\")\n\n\ndef get_main_window():\n \"\"\"Acquire Max's main window\"\"\"\n from qtpy import QtWidgets\n top_widgets = QtWidgets.QApplication.topLevelWidgets()\n name = \"QmaxApplicationWindow\"\n for widget in top_widgets:\n if (\n widget.inherits(\"QMainWindow\")\n and widget.metaObject().className() == name\n ):\n return widget\n raise RuntimeError('Count not find 3dsMax main window.')\n\n\ndef imprint(node_name: str, data: dict) -> bool:\n node = rt.GetNodeByName(node_name)\n if not node:\n return False\n\n for k, v in data.items():\n if isinstance(v, (dict, list)):\n rt.SetUserProp(node, k, f\"{JSON_PREFIX}{json.dumps(v)}\")\n else:\n rt.SetUserProp(node, k, v)\n\n return True\n\n\ndef lsattr(\n attr: str,\n value: Union[str, None] = None,\n root: Union[str, None] = None) -> list:\n \"\"\"List nodes having attribute with specified value.\n\n Args:\n attr (str): Attribute name to match.\n value (str, Optional): Value to match, of omitted, all nodes\n with specified attribute are returned no matter of value.\n root (str, Optional): Root node name. If omitted, scene root is used.\n\n Returns:\n list of nodes.\n \"\"\"\n root = rt.RootNode if root is None else rt.GetNodeByName(root)\n\n def output_node(node, nodes):\n nodes.append(node)\n for child in node.Children:\n output_node(child, nodes)\n\n nodes = []\n output_node(root, nodes)\n return [\n n for n in nodes\n if rt.GetUserProp(n, attr) == value\n ] if value else [\n n for n in nodes\n if rt.GetUserProp(n, attr)\n ]\n\n\ndef read(container) -> dict:\n data = {}\n props = rt.GetUserPropBuffer(container)\n # this shouldn't happen but let's guard against it anyway\n if not props:\n return data\n\n for line in props.split(\"\\r\\n\"):\n try:\n key, value = line.split(\"=\")\n except ValueError:\n # if the line cannot be split we can't really parse it\n continue\n\n value = value.strip()\n if isinstance(value.strip(), six.string_types) and \\\n value.startswith(JSON_PREFIX):\n with contextlib.suppress(json.JSONDecodeError):\n value = json.loads(value[len(JSON_PREFIX):])\n\n # default value behavior\n # convert maxscript boolean values\n if value == \"true\":\n value = True\n elif value == \"false\":\n value = False\n\n data[key.strip()] = value\n\n data[\"instance_node\"] = container.Name\n\n return data\n\n\n@contextlib.contextmanager\ndef maintained_selection():\n previous_selection = rt.GetCurrentSelection()\n try:\n yield\n finally:\n if previous_selection:\n rt.Select(previous_selection)\n else:\n rt.Select()\n\n\ndef get_all_children(parent, node_type=None):\n \"\"\"Handy function to get all the children of a given node\n\n Args:\n parent (3dsmax Node1): Node to get all children of.\n node_type (None, runtime.class): give class to check for\n e.g. rt.FFDBox/rt.GeometryClass etc.\n\n Returns:\n list: list of all children of the parent node\n \"\"\"\n def list_children(node):\n children = []\n for c in node.Children:\n children.append(c)\n children = children + list_children(c)\n return children\n child_list = list_children(parent)\n\n return ([x for x in child_list if rt.SuperClassOf(x) == node_type]\n if node_type else child_list)\n\n\ndef get_current_renderer():\n \"\"\"\n Notes:\n Get current renderer for Max\n\n Returns:\n \"{Current Renderer}:{Current Renderer}\"\n e.g. \"Redshift_Renderer:Redshift_Renderer\"\n \"\"\"\n return rt.renderers.production\n\n\ndef get_default_render_folder(project_setting=None):\n return (project_setting[\"max\"]\n [\"RenderSettings\"]\n [\"default_render_image_folder\"])\n\n\ndef set_render_frame_range(start_frame, end_frame):\n \"\"\"\n Note:\n Frame range can be specified in different types. Possible values are:\n * `1` - Single frame.\n * `2` - Active time segment ( animationRange ).\n * `3` - User specified Range.\n * `4` - User specified Frame pickup string (for example `1,3,5-12`).\n\n Todo:\n Current type is hard-coded, there should be a custom setting for this.\n \"\"\"\n rt.rendTimeType = 3\n if start_frame is not None and end_frame is not None:\n rt.rendStart = int(start_frame)\n rt.rendEnd = int(end_frame)\n\n\ndef get_multipass_setting(project_setting=None):\n return (project_setting[\"max\"]\n [\"RenderSettings\"]\n [\"multipass\"])\n\n\ndef set_scene_resolution(width: int, height: int):\n \"\"\"Set the render resolution\n\n Args:\n width(int): value of the width\n height(int): value of the height\n\n Returns:\n None\n\n \"\"\"\n # make sure the render dialog is closed\n # for the update of resolution\n # Changing the Render Setup dialog settings should be done\n # with the actual Render Setup dialog in a closed state.\n if rt.renderSceneDialog.isOpen():\n rt.renderSceneDialog.close()\n\n rt.renderWidth = width\n rt.renderHeight = height\n\n\ndef reset_scene_resolution():\n \"\"\"Apply the scene resolution from the project definition\n\n scene resolution can be overwritten by an asset if the asset.data contains\n any information regarding scene resolution .\n Returns:\n None\n \"\"\"\n data = [\"data.resolutionWidth\", \"data.resolutionHeight\"]\n project_resolution = get_current_project(fields=data)\n project_resolution_data = project_resolution[\"data\"]\n asset_resolution = get_current_project_asset(fields=data)\n asset_resolution_data = asset_resolution[\"data\"]\n # Set project resolution\n project_width = int(project_resolution_data.get(\"resolutionWidth\", 1920))\n project_height = int(project_resolution_data.get(\"resolutionHeight\", 1080))\n width = int(asset_resolution_data.get(\"resolutionWidth\", project_width))\n height = int(asset_resolution_data.get(\"resolutionHeight\", project_height))\n\n set_scene_resolution(width, height)\n\n\ndef get_frame_range(asset_doc=None) -> Union[Dict[str, Any], None]:\n \"\"\"Get the current assets frame range and handles.\n\n Args:\n asset_doc (dict): Asset Entity Data\n\n Returns:\n dict: with frame start, frame end, handle start, handle end.\n \"\"\"\n # Set frame start/end\n if asset_doc is None:\n asset_doc = get_current_project_asset()\n\n data = asset_doc[\"data\"]\n frame_start = data.get(\"frameStart\")\n frame_end = data.get(\"frameEnd\")\n\n if frame_start is None or frame_end is None:\n return {}\n\n frame_start = int(frame_start)\n frame_end = int(frame_end)\n handle_start = int(data.get(\"handleStart\", 0))\n handle_end = int(data.get(\"handleEnd\", 0))\n frame_start_handle = frame_start - handle_start\n frame_end_handle = frame_end + handle_end\n\n return {\n \"frameStart\": frame_start,\n \"frameEnd\": frame_end,\n \"handleStart\": handle_start,\n \"handleEnd\": handle_end,\n \"frameStartHandle\": frame_start_handle,\n \"frameEndHandle\": frame_end_handle,\n }\n\n\ndef reset_frame_range(fps: bool = True):\n \"\"\"Set frame range to current asset.\n This is part of 3dsmax documentation:\n\n animationRange: A System Global variable which lets you get and\n set an Interval value that defines the start and end frames\n of the Active Time Segment.\n frameRate: A System Global variable which lets you get\n and set an Integer value that defines the current\n scene frame rate in frames-per-second.\n \"\"\"\n if fps:\n data_fps = get_current_project(fields=[\"data.fps\"])\n fps_number = float(data_fps[\"data\"][\"fps\"])\n rt.frameRate = fps_number\n frame_range = get_frame_range()\n\n set_timeline(\n frame_range[\"frameStartHandle\"], frame_range[\"frameEndHandle\"])\n set_render_frame_range(\n frame_range[\"frameStartHandle\"], frame_range[\"frameEndHandle\"])\n\n\ndef reset_unit_scale():\n \"\"\"Apply the unit scale setting to 3dsMax\n \"\"\"\n project_name = get_current_project_name()\n settings = get_project_settings(project_name).get(\"max\")\n scene_scale = settings.get(\"unit_scale_settings\",\n {}).get(\"scene_unit_scale\")\n if scene_scale:\n rt.units.DisplayType = rt.Name(\"Metric\")\n rt.units.MetricType = rt.Name(scene_scale)\n else:\n rt.units.DisplayType = rt.Name(\"Generic\")\n\n\ndef convert_unit_scale():\n \"\"\"Convert system unit scale in 3dsMax\n for fbx export\n\n Returns:\n str: unit scale\n \"\"\"\n unit_scale_dict = {\n \"millimeters\": \"mm\",\n \"centimeters\": \"cm\",\n \"meters\": \"m\",\n \"kilometers\": \"km\"\n }\n current_unit_scale = rt.Execute(\"units.MetricType as string\")\n return unit_scale_dict[current_unit_scale]\n\n\ndef set_context_setting():\n \"\"\"Apply the project settings from the project definition\n\n Settings can be overwritten by an asset if the asset.data contains\n any information regarding those settings.\n\n Examples of settings:\n frame range\n resolution\n\n Returns:\n None\n \"\"\"\n reset_scene_resolution()\n reset_frame_range()\n reset_colorspace()\n reset_unit_scale()\n\n\ndef get_max_version():\n \"\"\"\n Args:\n get max version date for deadline\n\n Returns:\n #(25000, 62, 0, 25, 0, 0, 997, 2023, \"\")\n max_info[7] = max version date\n \"\"\"\n max_info = rt.MaxVersion()\n return max_info[7]\n\n\ndef is_headless():\n \"\"\"Check if 3dsMax runs in batch mode.\n If it returns True, it runs in 3dsbatch.exe\n If it returns False, it runs in 3dsmax.exe\n \"\"\"\n return rt.maxops.isInNonInteractiveMode()\n\n\ndef set_timeline(frameStart, frameEnd):\n \"\"\"Set frame range for timeline editor in Max\n \"\"\"\n rt.animationRange = rt.interval(frameStart, frameEnd)\n return rt.animationRange\n\n\ndef reset_colorspace():\n \"\"\"OCIO Configuration\n Supports in 3dsMax 2024+\n\n \"\"\"\n if int(get_max_version()) < 2024:\n return\n project_name = get_current_project_name()\n colorspace_mgr = rt.ColorPipelineMgr\n project_settings = get_project_settings(project_name)\n\n max_config_data = colorspace.get_imageio_config(\n project_name, \"max\", project_settings)\n if max_config_data:\n ocio_config_path = max_config_data[\"path\"]\n colorspace_mgr = rt.ColorPipelineMgr\n colorspace_mgr.Mode = rt.Name(\"OCIO_Custom\")\n colorspace_mgr.OCIOConfigPath = ocio_config_path\n\n\ndef check_colorspace():\n parent = get_main_window()\n if parent is None:\n log.info(\"Skipping outdated pop-up \"\n \"because Max main window can't be found.\")\n if int(get_max_version()) >= 2024:\n color_mgr = rt.ColorPipelineMgr\n project_name = get_current_project_name()\n project_settings = get_project_settings(project_name)\n max_config_data = colorspace.get_imageio_config(\n project_name, \"max\", project_settings)\n if max_config_data and color_mgr.Mode != rt.Name(\"OCIO_Custom\"):\n if not is_headless():\n from openpype.widgets import popup\n dialog = popup.Popup(parent=parent)\n dialog.setWindowTitle(\"Warning: Wrong OCIO Mode\")\n dialog.setMessage(\"This scene has wrong OCIO \"\n \"Mode setting.\")\n dialog.setButtonText(\"Fix\")\n dialog.setStyleSheet(load_stylesheet())\n dialog.on_clicked.connect(reset_colorspace)\n dialog.show()\n\ndef unique_namespace(namespace, format=\"%02d\",\n prefix=\"\", suffix=\"\", con_suffix=\"CON\"):\n \"\"\"Return unique namespace\n\n Arguments:\n namespace (str): Name of namespace to consider\n format (str, optional): Formatting of the given iteration number\n suffix (str, optional): Only consider namespaces with this suffix.\n con_suffix: max only, for finding the name of the master container\n\n >>> unique_namespace(\"bar\")\n # bar01\n >>> unique_namespace(\":hello\")\n # :hello01\n >>> unique_namespace(\"bar:\", suffix=\"_NS\")\n # bar01_NS:\n\n \"\"\"\n\n def current_namespace():\n current = namespace\n # When inside a namespace Max adds no trailing :\n if not current.endswith(\":\"):\n current += \":\"\n return current\n\n # Always check against the absolute namespace root\n # There's no clash with :x if we're defining namespace :a:x\n ROOT = \":\" if namespace.startswith(\":\") else current_namespace()\n\n # Strip trailing `:` tokens since we might want to add a suffix\n start = \":\" if namespace.startswith(\":\") else \"\"\n end = \":\" if namespace.endswith(\":\") else \"\"\n namespace = namespace.strip(\":\")\n if \":\" in namespace:\n # Split off any nesting that we don't uniqify anyway.\n parents, namespace = namespace.rsplit(\":\", 1)\n start += parents + \":\"\n ROOT += start\n\n iteration = 1\n increment_version = True\n while increment_version:\n nr_namespace = namespace + format % iteration\n unique = prefix + nr_namespace + suffix\n container_name = f\"{unique}:{namespace}{con_suffix}\"\n if not rt.getNodeByName(container_name):\n name_space = start + unique + end\n increment_version = False\n return name_space\n else:\n increment_version = True\n iteration += 1\n\n\ndef get_namespace(container_name):\n \"\"\"Get the namespace and name of the sub-container\n\n Args:\n container_name (str): the name of master container\n\n Raises:\n RuntimeError: when there is no master container found\n\n Returns:\n namespace (str): namespace of the sub-container\n name (str): name of the sub-container\n \"\"\"\n node = rt.getNodeByName(container_name)\n if not node:\n raise RuntimeError(\"Master Container Not Found..\")\n name = rt.getUserProp(node, \"name\")\n namespace = rt.getUserProp(node, \"namespace\")\n return namespace, name\n\n\ndef object_transform_set(container_children):\n \"\"\"A function which allows to store the transform of\n previous loaded object(s)\n Args:\n container_children(list): A list of nodes\n\n Returns:\n transform_set (dict): A dict with all transform data of\n the previous loaded object(s)\n \"\"\"\n transform_set = {}\n for node in container_children:\n name = f\"{node.name}.transform\"\n transform_set[name] = node.pos\n name = f\"{node.name}.scale\"\n transform_set[name] = node.scale\n return transform_set\n\n\ndef get_plugins() -> list:\n \"\"\"Get all loaded plugins in 3dsMax\n\n Returns:\n plugin_info_list: a list of loaded plugins\n \"\"\"\n manager = rt.PluginManager\n count = manager.pluginDllCount\n plugin_info_list = []\n for p in range(1, count + 1):\n plugin_info = manager.pluginDllName(p)\n plugin_info_list.append(plugin_info)\n\n return plugin_info_list\n\n\n@contextlib.contextmanager\ndef render_resolution(width, height):\n \"\"\"Set render resolution option during context\n\n Args:\n width (int): render width\n height (int): render height\n \"\"\"\n current_renderWidth = rt.renderWidth\n current_renderHeight = rt.renderHeight\n try:\n rt.renderWidth = width\n rt.renderHeight = height\n yield\n finally:\n rt.renderWidth = current_renderWidth\n rt.renderHeight = current_renderHeight\n\n\n@contextlib.contextmanager\ndef suspended_refresh():\n \"\"\"Suspended refresh for scene and modify panel redraw.\n \"\"\"\n if is_headless():\n yield\n return\n rt.disableSceneRedraw()\n rt.suspendEditing()\n try:\n yield\n\n finally:\n rt.enableSceneRedraw()\n rt.resumeEditing()\n"
},
{
"path": "openpype/hosts/max/api/lib_renderproducts.py",
"content": "# Render Element Example : For scanline render, VRay\n# https://help.autodesk.com/view/MAXDEV/2022/ENU/?guid=GUID-E8F75D47-B998-4800-A3A5-610E22913CFC\n# arnold\n# https://help.autodesk.com/view/ARNOL/ENU/?guid=arnold_for_3ds_max_ax_maxscript_commands_ax_renderview_commands_html\nimport os\n\nfrom pymxs import runtime as rt\n\nfrom openpype.hosts.max.api.lib import get_current_renderer\nfrom openpype.pipeline import get_current_project_name\nfrom openpype.settings import get_project_settings\n\n\nclass RenderProducts(object):\n\n def __init__(self, project_settings=None):\n self._project_settings = project_settings\n if not self._project_settings:\n self._project_settings = get_project_settings(\n get_current_project_name()\n )\n\n def get_beauty(self, container):\n render_dir = os.path.dirname(rt.rendOutputFilename)\n\n output_file = os.path.join(render_dir, container)\n\n setting = self._project_settings\n img_fmt = setting[\"max\"][\"RenderSettings\"][\"image_format\"] # noqa\n\n start_frame = int(rt.rendStart)\n end_frame = int(rt.rendEnd) + 1\n\n return {\n \"beauty\": self.get_expected_beauty(\n output_file, start_frame, end_frame, img_fmt\n )\n }\n\n def get_multiple_beauty(self, outputs, cameras):\n beauty_output_frames = dict()\n for output, camera in zip(outputs, cameras):\n filename, ext = os.path.splitext(output)\n filename = filename.replace(\".\", \"\")\n ext = ext.replace(\".\", \"\")\n start_frame = int(rt.rendStart)\n end_frame = int(rt.rendEnd) + 1\n new_beauty = self.get_expected_beauty(\n filename, start_frame, end_frame, ext\n )\n beauty_output = ({\n f\"{camera}_beauty\": new_beauty\n })\n beauty_output_frames.update(beauty_output)\n return beauty_output_frames\n\n def get_multiple_aovs(self, outputs, cameras):\n renderer_class = get_current_renderer()\n renderer = str(renderer_class).split(\":\")[0]\n aovs_frames = {}\n for output, camera in zip(outputs, cameras):\n filename, ext = os.path.splitext(output)\n filename = filename.replace(\".\", \"\")\n ext = ext.replace(\".\", \"\")\n start_frame = int(rt.rendStart)\n end_frame = int(rt.rendEnd) + 1\n\n if renderer in [\n \"ART_Renderer\",\n \"V_Ray_6_Hotfix_3\",\n \"V_Ray_GPU_6_Hotfix_3\",\n \"Default_Scanline_Renderer\",\n \"Quicksilver_Hardware_Renderer\",\n ]:\n render_name = self.get_render_elements_name()\n if render_name:\n for name in render_name:\n aovs_frames.update({\n f\"{camera}_{name}\": self.get_expected_aovs(\n filename, name, start_frame,\n end_frame, ext)\n })\n elif renderer == \"Redshift_Renderer\":\n render_name = self.get_render_elements_name()\n if render_name:\n rs_aov_files = rt.Execute(\"renderers.current.separateAovFiles\") # noqa\n # this doesn't work, always returns False\n # rs_AovFiles = rt.RedShift_Renderer().separateAovFiles\n if ext == \"exr\" and not rs_aov_files:\n for name in render_name:\n if name == \"RsCryptomatte\":\n aovs_frames.update({\n f\"{camera}_{name}\": self.get_expected_aovs(\n filename, name, start_frame,\n end_frame, ext)\n })\n else:\n for name in render_name:\n aovs_frames.update({\n f\"{camera}_{name}\": self.get_expected_aovs(\n filename, name, start_frame,\n end_frame, ext)\n })\n elif renderer == \"Arnold\":\n render_name = self.get_arnold_product_name()\n if render_name:\n for name in render_name:\n aovs_frames.update({\n f\"{camera}_{name}\": self.get_expected_arnold_product( # noqa\n filename, name, start_frame,\n end_frame, ext)\n })\n elif renderer in [\n \"V_Ray_6_Hotfix_3\",\n \"V_Ray_GPU_6_Hotfix_3\"\n ]:\n if ext != \"exr\":\n render_name = self.get_render_elements_name()\n if render_name:\n for name in render_name:\n aovs_frames.update({\n f\"{camera}_{name}\": self.get_expected_aovs(\n filename, name, start_frame,\n end_frame, ext)\n })\n\n return aovs_frames\n\n def get_aovs(self, container):\n render_dir = os.path.dirname(rt.rendOutputFilename)\n\n output_file = os.path.join(render_dir,\n container)\n\n setting = self._project_settings\n img_fmt = setting[\"max\"][\"RenderSettings\"][\"image_format\"] # noqa\n\n start_frame = int(rt.rendStart)\n end_frame = int(rt.rendEnd) + 1\n renderer_class = get_current_renderer()\n renderer = str(renderer_class).split(\":\")[0]\n render_dict = {}\n\n if renderer in [\n \"ART_Renderer\",\n \"V_Ray_6_Hotfix_3\",\n \"V_Ray_GPU_6_Hotfix_3\",\n \"Default_Scanline_Renderer\",\n \"Quicksilver_Hardware_Renderer\",\n ]:\n render_name = self.get_render_elements_name()\n if render_name:\n for name in render_name:\n render_dict.update({\n name: self.get_expected_aovs(\n output_file, name, start_frame,\n end_frame, img_fmt)\n })\n elif renderer == \"Redshift_Renderer\":\n render_name = self.get_render_elements_name()\n if render_name:\n rs_aov_files = rt.Execute(\"renderers.current.separateAovFiles\")\n # this doesn't work, always returns False\n # rs_AovFiles = rt.RedShift_Renderer().separateAovFiles\n if img_fmt == \"exr\" and not rs_aov_files:\n for name in render_name:\n if name == \"RsCryptomatte\":\n render_dict.update({\n name: self.get_expected_aovs(\n output_file, name, start_frame,\n end_frame, img_fmt)\n })\n else:\n for name in render_name:\n render_dict.update({\n name: self.get_expected_aovs(\n output_file, name, start_frame,\n end_frame, img_fmt)\n })\n\n elif renderer == \"Arnold\":\n render_name = self.get_arnold_product_name()\n if render_name:\n for name in render_name:\n render_dict.update({\n name: self.get_expected_arnold_product(\n output_file, name, start_frame,\n end_frame, img_fmt)\n })\n elif renderer in [\n \"V_Ray_6_Hotfix_3\",\n \"V_Ray_GPU_6_Hotfix_3\"\n ]:\n if img_fmt != \"exr\":\n render_name = self.get_render_elements_name()\n if render_name:\n for name in render_name:\n render_dict.update({\n name: self.get_expected_aovs(\n output_file, name, start_frame,\n end_frame, img_fmt) # noqa\n })\n\n return render_dict\n\n def get_expected_beauty(self, folder, start_frame, end_frame, fmt):\n beauty_frame_range = []\n for f in range(start_frame, end_frame):\n frame = \"%04d\" % f\n beauty_output = f\"{folder}.{frame}.{fmt}\"\n beauty_output = beauty_output.replace(\"\\\\\", \"/\")\n beauty_frame_range.append(beauty_output)\n\n return beauty_frame_range\n\n def get_arnold_product_name(self):\n \"\"\"Get all the Arnold AOVs name\"\"\"\n aov_name = []\n\n amw = rt.MaxToAOps.AOVsManagerWindow()\n aov_mgr = rt.renderers.current.AOVManager\n # Check if there is any aov group set in AOV manager\n aov_group_num = len(aov_mgr.drivers)\n if aov_group_num < 1:\n return\n for i in range(aov_group_num):\n # get the specific AOV group\n aov_name.extend(aov.name for aov in aov_mgr.drivers[i].aov_list)\n # close the AOVs manager window\n amw.close()\n\n return aov_name\n\n def get_expected_arnold_product(self, folder, name,\n start_frame, end_frame, fmt):\n \"\"\"Get all the expected Arnold AOVs\"\"\"\n aov_list = []\n for f in range(start_frame, end_frame):\n frame = \"%04d\" % f\n render_element = f\"{folder}_{name}.{frame}.{fmt}\"\n render_element = render_element.replace(\"\\\\\", \"/\")\n aov_list.append(render_element)\n\n return aov_list\n\n def get_render_elements_name(self):\n \"\"\"Get all the render element names for general \"\"\"\n render_name = []\n render_elem = rt.maxOps.GetCurRenderElementMgr()\n render_elem_num = render_elem.NumRenderElements()\n if render_elem_num < 1:\n return\n # get render elements from the renders\n for i in range(render_elem_num):\n renderlayer_name = render_elem.GetRenderElement(i)\n if renderlayer_name.enabled:\n target, renderpass = str(renderlayer_name).split(\":\")\n render_name.append(renderpass)\n\n return render_name\n\n def get_expected_aovs(self, folder, name,\n start_frame, end_frame, fmt):\n \"\"\"Get all the expected render element output files. \"\"\"\n render_elements = []\n for f in range(start_frame, end_frame):\n frame = \"%04d\" % f\n render_element = f\"{folder}_{name}.{frame}.{fmt}\"\n render_element = render_element.replace(\"\\\\\", \"/\")\n render_elements.append(render_element)\n\n return render_elements\n\n def image_format(self):\n return self._project_settings[\"max\"][\"RenderSettings\"][\"image_format\"] # noqa\n"
},
{
"path": "openpype/hosts/max/api/lib_rendersettings.py",
"content": "import os\nfrom pymxs import runtime as rt\nfrom openpype.lib import Logger\nfrom openpype.settings import get_project_settings\nfrom openpype.pipeline import get_current_project_name\nfrom openpype.pipeline.context_tools import get_current_project_asset\n\nfrom openpype.hosts.max.api.lib import (\n set_render_frame_range,\n get_current_renderer,\n get_default_render_folder\n)\n\n\nclass RenderSettings(object):\n\n log = Logger.get_logger(\"RenderSettings\")\n\n _aov_chars = {\n \"dot\": \".\",\n \"dash\": \"-\",\n \"underscore\": \"_\"\n }\n\n def __init__(self, project_settings=None):\n \"\"\"\n Set up the naming convention for the render\n elements for the deadline submission\n \"\"\"\n\n self._project_settings = project_settings\n if not self._project_settings:\n self._project_settings = get_project_settings(\n get_current_project_name()\n )\n\n def set_render_camera(self, selection):\n for sel in selection:\n # to avoid Attribute Error from pymxs wrapper\n if rt.classOf(sel) in rt.Camera.classes:\n rt.viewport.setCamera(sel)\n return\n raise RuntimeError(\"Active Camera not found\")\n\n def render_output(self, container):\n folder = rt.maxFilePath\n # hard-coded, should be customized in the setting\n file = rt.maxFileName\n folder = folder.replace(\"\\\\\", \"/\")\n # hard-coded, set the renderoutput path\n setting = self._project_settings\n render_folder = get_default_render_folder(setting)\n filename, ext = os.path.splitext(file)\n output_dir = os.path.join(folder,\n render_folder,\n filename)\n if not os.path.exists(output_dir):\n os.makedirs(output_dir)\n # hard-coded, should be customized in the setting\n context = get_current_project_asset()\n\n # get project resolution\n width = context[\"data\"].get(\"resolutionWidth\")\n height = context[\"data\"].get(\"resolutionHeight\")\n # Set Frame Range\n frame_start = context[\"data\"].get(\"frame_start\")\n frame_end = context[\"data\"].get(\"frame_end\")\n set_render_frame_range(frame_start, frame_end)\n # get the production render\n renderer_class = get_current_renderer()\n renderer = str(renderer_class).split(\":\")[0]\n\n img_fmt = self._project_settings[\"max\"][\"RenderSettings\"][\"image_format\"] # noqa\n output = os.path.join(output_dir, container)\n try:\n aov_separator = self._aov_chars[(\n self._project_settings[\"max\"]\n [\"RenderSettings\"]\n [\"aov_separator\"]\n )]\n except KeyError:\n aov_separator = \".\"\n output_filename = f\"{output}..{img_fmt}\"\n output_filename = output_filename.replace(\"{aov_separator}\",\n aov_separator)\n rt.rendOutputFilename = output_filename\n if renderer == \"VUE_File_Renderer\":\n return\n # TODO: Finish the arnold render setup\n if renderer == \"Arnold\":\n self.arnold_setup()\n\n if renderer in [\n \"ART_Renderer\",\n \"Redshift_Renderer\",\n \"V_Ray_6_Hotfix_3\",\n \"V_Ray_GPU_6_Hotfix_3\",\n \"Default_Scanline_Renderer\",\n \"Quicksilver_Hardware_Renderer\",\n ]:\n self.render_element_layer(output, width, height, img_fmt)\n\n rt.rendSaveFile = True\n\n if rt.renderSceneDialog.isOpen():\n rt.renderSceneDialog.close()\n\n def arnold_setup(self):\n # get Arnold RenderView run in the background\n # for setting up renderable camera\n arv = rt.MAXToAOps.ArnoldRenderView()\n render_camera = rt.viewport.GetCamera()\n if render_camera:\n arv.setOption(\"Camera\", str(render_camera))\n\n # TODO: add AOVs and extension\n img_fmt = self._project_settings[\"max\"][\"RenderSettings\"][\"image_format\"] # noqa\n setup_cmd = (\n f\"\"\"\n amw = MaxtoAOps.AOVsManagerWindow()\n amw.close()\n aovmgr = renderers.current.AOVManager\n aovmgr.drivers = #()\n img_fmt = \"{img_fmt}\"\n if img_fmt == \"png\" then driver = ArnoldPNGDriver()\n if img_fmt == \"jpg\" then driver = ArnoldJPEGDriver()\n if img_fmt == \"exr\" then driver = ArnoldEXRDriver()\n if img_fmt == \"tif\" then driver = ArnoldTIFFDriver()\n if img_fmt == \"tiff\" then driver = ArnoldTIFFDriver()\n append aovmgr.drivers driver\n aovmgr.drivers[1].aov_list = #()\n \"\"\")\n\n rt.execute(setup_cmd)\n arv.close()\n\n def render_element_layer(self, dir, width, height, ext):\n \"\"\"For Renderers with render elements\"\"\"\n rt.renderWidth = width\n rt.renderHeight = height\n render_elem = rt.maxOps.GetCurRenderElementMgr()\n render_elem_num = render_elem.NumRenderElements()\n if render_elem_num < 0:\n return\n\n for i in range(render_elem_num):\n renderlayer_name = render_elem.GetRenderElement(i)\n target, renderpass = str(renderlayer_name).split(\":\")\n aov_name = f\"{dir}_{renderpass}..{ext}\"\n render_elem.SetRenderElementFileName(i, aov_name)\n\n def get_render_output(self, container, output_dir):\n output = os.path.join(output_dir, container)\n img_fmt = self._project_settings[\"max\"][\"RenderSettings\"][\"image_format\"] # noqa\n output_filename = f\"{output}..{img_fmt}\"\n return output_filename\n\n def get_render_element(self):\n orig_render_elem = []\n render_elem = rt.maxOps.GetCurRenderElementMgr()\n render_elem_num = render_elem.NumRenderElements()\n if render_elem_num < 0:\n return\n\n for i in range(render_elem_num):\n render_element = render_elem.GetRenderElementFilename(i)\n orig_render_elem.append(render_element)\n\n return orig_render_elem\n\n def get_batch_render_elements(self, container,\n output_dir, camera):\n render_element_list = list()\n output = os.path.join(output_dir, container)\n render_elem = rt.maxOps.GetCurRenderElementMgr()\n render_elem_num = render_elem.NumRenderElements()\n if render_elem_num < 0:\n return\n img_fmt = self._project_settings[\"max\"][\"RenderSettings\"][\"image_format\"] # noqa\n\n for i in range(render_elem_num):\n renderlayer_name = render_elem.GetRenderElement(i)\n target, renderpass = str(renderlayer_name).split(\":\")\n aov_name = f\"{output}_{camera}_{renderpass}..{img_fmt}\"\n render_element_list.append(aov_name)\n return render_element_list\n\n def get_batch_render_output(self, camera):\n target_layer_no = rt.batchRenderMgr.FindView(camera)\n target_layer = rt.batchRenderMgr.GetView(target_layer_no)\n return target_layer.outputFilename\n\n def batch_render_elements(self, camera):\n target_layer_no = rt.batchRenderMgr.FindView(camera)\n target_layer = rt.batchRenderMgr.GetView(target_layer_no)\n outputfilename = target_layer.outputFilename\n directory = os.path.dirname(outputfilename)\n render_elem = rt.maxOps.GetCurRenderElementMgr()\n render_elem_num = render_elem.NumRenderElements()\n if render_elem_num < 0:\n return\n ext = self._project_settings[\"max\"][\"RenderSettings\"][\"image_format\"] # noqa\n\n for i in range(render_elem_num):\n renderlayer_name = render_elem.GetRenderElement(i)\n target, renderpass = str(renderlayer_name).split(\":\")\n aov_name = f\"{directory}_{camera}_{renderpass}..{ext}\"\n render_elem.SetRenderElementFileName(i, aov_name)\n\n def batch_render_layer(self, container,\n output_dir, cameras):\n outputs = list()\n output = os.path.join(output_dir, container)\n img_fmt = self._project_settings[\"max\"][\"RenderSettings\"][\"image_format\"] # noqa\n for cam in cameras:\n camera = rt.getNodeByName(cam)\n layer_no = rt.batchRenderMgr.FindView(cam)\n renderlayer = None\n if layer_no == 0:\n renderlayer = rt.batchRenderMgr.CreateView(camera)\n else:\n renderlayer = rt.batchRenderMgr.GetView(layer_no)\n # use camera name as renderlayer name\n renderlayer.name = cam\n renderlayer.outputFilename = f\"{output}_{cam}..{img_fmt}\"\n outputs.append(renderlayer.outputFilename)\n return outputs\n"
},
{
"path": "openpype/hosts/max/api/menu.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"3dsmax menu definition of AYON.\"\"\"\nimport os\nfrom qtpy import QtWidgets, QtCore\nfrom pymxs import runtime as rt\n\nfrom openpype.tools.utils import host_tools\nfrom openpype.hosts.max.api import lib\n\n\nclass OpenPypeMenu(object):\n \"\"\"Object representing OpenPype/AYON menu.\n\n This is using \"hack\" to inject itself before \"Help\" menu of 3dsmax.\n For some reason `postLoadingMenus` event doesn't fire, and main menu\n if probably re-initialized by menu templates, se we wait for at least\n 1 event Qt event loop before trying to insert.\n\n \"\"\"\n\n def __init__(self):\n super().__init__()\n self.main_widget = self.get_main_widget()\n self.menu = None\n\n timer = QtCore.QTimer()\n # set number of event loops to wait.\n timer.setInterval(1)\n timer.timeout.connect(self._on_timer)\n timer.start()\n\n self._timer = timer\n self._counter = 0\n\n def _on_timer(self):\n if self._counter < 1:\n self._counter += 1\n return\n\n self._counter = 0\n self._timer.stop()\n self.build_openpype_menu()\n\n @staticmethod\n def get_main_widget():\n \"\"\"Get 3dsmax main window.\"\"\"\n return QtWidgets.QWidget.find(rt.windows.getMAXHWND())\n\n def get_main_menubar(self) -> QtWidgets.QMenuBar:\n \"\"\"Get main Menubar by 3dsmax main window.\"\"\"\n return list(self.main_widget.findChildren(QtWidgets.QMenuBar))[0]\n\n def get_or_create_openpype_menu(\n self, name: str = \"&Openpype\",\n before: str = \"&Help\") -> QtWidgets.QAction:\n \"\"\"Create AYON menu.\n\n Args:\n name (str, Optional): AYON menu name.\n before (str, Optional): Name of the 3dsmax main menu item to\n add AYON menu before.\n\n Returns:\n QtWidgets.QAction: AYON menu action.\n\n \"\"\"\n if self.menu is not None:\n return self.menu\n\n menu_bar = self.get_main_menubar()\n menu_items = menu_bar.findChildren(\n QtWidgets.QMenu, options=QtCore.Qt.FindDirectChildrenOnly)\n help_action = None\n for item in menu_items:\n if name in item.title():\n # we already have OpenPype menu\n return item\n\n if before in item.title():\n help_action = item.menuAction()\n tab_menu_label = os.environ.get(\"AVALON_LABEL\") or \"AYON\"\n op_menu = QtWidgets.QMenu(\"&{}\".format(tab_menu_label))\n menu_bar.insertMenu(help_action, op_menu)\n\n self.menu = op_menu\n return op_menu\n\n def build_openpype_menu(self) -> QtWidgets.QAction:\n \"\"\"Build items in AYON menu.\"\"\"\n openpype_menu = self.get_or_create_openpype_menu()\n load_action = QtWidgets.QAction(\"Load...\", openpype_menu)\n load_action.triggered.connect(self.load_callback)\n openpype_menu.addAction(load_action)\n\n publish_action = QtWidgets.QAction(\"Publish...\", openpype_menu)\n publish_action.triggered.connect(self.publish_callback)\n openpype_menu.addAction(publish_action)\n\n manage_action = QtWidgets.QAction(\"Manage...\", openpype_menu)\n manage_action.triggered.connect(self.manage_callback)\n openpype_menu.addAction(manage_action)\n\n library_action = QtWidgets.QAction(\"Library...\", openpype_menu)\n library_action.triggered.connect(self.library_callback)\n openpype_menu.addAction(library_action)\n\n openpype_menu.addSeparator()\n\n workfiles_action = QtWidgets.QAction(\"Work Files...\", openpype_menu)\n workfiles_action.triggered.connect(self.workfiles_callback)\n openpype_menu.addAction(workfiles_action)\n\n openpype_menu.addSeparator()\n\n res_action = QtWidgets.QAction(\"Set Resolution\", openpype_menu)\n res_action.triggered.connect(self.resolution_callback)\n openpype_menu.addAction(res_action)\n\n frame_action = QtWidgets.QAction(\"Set Frame Range\", openpype_menu)\n frame_action.triggered.connect(self.frame_range_callback)\n openpype_menu.addAction(frame_action)\n\n colorspace_action = QtWidgets.QAction(\"Set Colorspace\", openpype_menu)\n colorspace_action.triggered.connect(self.colorspace_callback)\n openpype_menu.addAction(colorspace_action)\n\n unit_scale_action = QtWidgets.QAction(\"Set Unit Scale\", openpype_menu)\n unit_scale_action.triggered.connect(self.unit_scale_callback)\n openpype_menu.addAction(unit_scale_action)\n\n return openpype_menu\n\n def load_callback(self):\n \"\"\"Callback to show Loader tool.\"\"\"\n host_tools.show_loader(parent=self.main_widget)\n\n def publish_callback(self):\n \"\"\"Callback to show Publisher tool.\"\"\"\n host_tools.show_publisher(parent=self.main_widget)\n\n def manage_callback(self):\n \"\"\"Callback to show Scene Manager/Inventory tool.\"\"\"\n host_tools.show_scene_inventory(parent=self.main_widget)\n\n def library_callback(self):\n \"\"\"Callback to show Library Loader tool.\"\"\"\n host_tools.show_library_loader(parent=self.main_widget)\n\n def workfiles_callback(self):\n \"\"\"Callback to show Workfiles tool.\"\"\"\n host_tools.show_workfiles(parent=self.main_widget)\n\n def resolution_callback(self):\n \"\"\"Callback to reset scene resolution\"\"\"\n return lib.reset_scene_resolution()\n\n def frame_range_callback(self):\n \"\"\"Callback to reset frame range\"\"\"\n return lib.reset_frame_range()\n\n def colorspace_callback(self):\n \"\"\"Callback to reset colorspace\"\"\"\n return lib.reset_colorspace()\n\n def unit_scale_callback(self):\n \"\"\"Callback to reset unit scale\"\"\"\n return lib.reset_unit_scale()\n"
},
{
"path": "openpype/hosts/max/api/pipeline.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Pipeline tools for OpenPype Houdini integration.\"\"\"\nimport os\nimport logging\nfrom operator import attrgetter\n\nimport json\n\nfrom openpype.host import HostBase, IWorkfileHost, ILoadHost, IPublishHost\nimport pyblish.api\nfrom openpype.pipeline import (\n register_creator_plugin_path,\n register_loader_plugin_path,\n AVALON_CONTAINER_ID,\n)\nfrom openpype.hosts.max.api.menu import OpenPypeMenu\nfrom openpype.hosts.max.api import lib\nfrom openpype.hosts.max.api.plugin import MS_CUSTOM_ATTRIB\nfrom openpype.hosts.max import MAX_HOST_DIR\n\nfrom pymxs import runtime as rt # noqa\n\nlog = logging.getLogger(\"openpype.hosts.max\")\n\nPLUGINS_DIR = os.path.join(MAX_HOST_DIR, \"plugins\")\nPUBLISH_PATH = os.path.join(PLUGINS_DIR, \"publish\")\nLOAD_PATH = os.path.join(PLUGINS_DIR, \"load\")\nCREATE_PATH = os.path.join(PLUGINS_DIR, \"create\")\nINVENTORY_PATH = os.path.join(PLUGINS_DIR, \"inventory\")\n\n\nclass MaxHost(HostBase, IWorkfileHost, ILoadHost, IPublishHost):\n\n name = \"max\"\n menu = None\n\n def __init__(self):\n super(MaxHost, self).__init__()\n self._op_events = {}\n self._has_been_setup = False\n\n def install(self):\n pyblish.api.register_host(\"max\")\n\n pyblish.api.register_plugin_path(PUBLISH_PATH)\n register_loader_plugin_path(LOAD_PATH)\n register_creator_plugin_path(CREATE_PATH)\n\n # self._register_callbacks()\n self.menu = OpenPypeMenu()\n\n self._has_been_setup = True\n\n def context_setting():\n return lib.set_context_setting()\n\n rt.callbacks.addScript(rt.Name('systemPostNew'),\n context_setting)\n\n rt.callbacks.addScript(rt.Name('filePostOpen'),\n lib.check_colorspace)\n rt.callbacks.addScript(rt.Name('postWorkspaceChange'),\n self._deferred_menu_creation)\n\n def workfile_has_unsaved_changes(self):\n return rt.getSaveRequired()\n\n def get_workfile_extensions(self):\n return [\".max\"]\n\n def save_workfile(self, dst_path=None):\n rt.saveMaxFile(dst_path)\n return dst_path\n\n def open_workfile(self, filepath):\n rt.checkForSave()\n rt.loadMaxFile(filepath)\n return filepath\n\n def get_current_workfile(self):\n return os.path.join(rt.maxFilePath, rt.maxFileName)\n\n def get_containers(self):\n return ls()\n\n def _register_callbacks(self):\n rt.callbacks.removeScripts(id=rt.name(\"OpenPypeCallbacks\"))\n\n rt.callbacks.addScript(\n rt.Name(\"postLoadingMenus\"),\n self._deferred_menu_creation, id=rt.Name('OpenPypeCallbacks'))\n\n def _deferred_menu_creation(self):\n self.log.info(\"Building menu ...\")\n self.menu = OpenPypeMenu()\n\n @staticmethod\n def create_context_node():\n \"\"\"Helper for creating context holding node.\"\"\"\n\n root_scene = rt.rootScene\n\n create_attr_script = (\"\"\"\nattributes \"OpenPypeContext\"\n(\n parameters main rollout:params\n (\n context type: #string\n )\n\n rollout params \"OpenPype Parameters\"\n (\n editText editTextContext \"Context\" type: #string\n )\n)\n \"\"\")\n\n attr = rt.execute(create_attr_script)\n rt.custAttributes.add(root_scene, attr)\n\n return root_scene.OpenPypeContext.context\n\n def update_context_data(self, data, changes):\n try:\n _ = rt.rootScene.OpenPypeContext.context\n except AttributeError:\n # context node doesn't exists\n self.create_context_node()\n\n rt.rootScene.OpenPypeContext.context = json.dumps(data)\n\n def get_context_data(self):\n try:\n context = rt.rootScene.OpenPypeContext.context\n except AttributeError:\n # context node doesn't exists\n context = self.create_context_node()\n if not context:\n context = \"{}\"\n return json.loads(context)\n\n def save_file(self, dst_path=None):\n # Force forwards slashes to avoid segfault\n dst_path = dst_path.replace(\"\\\\\", \"/\")\n rt.saveMaxFile(dst_path)\n\n\ndef ls() -> list:\n \"\"\"Get all OpenPype instances.\"\"\"\n objs = rt.objects\n containers = [\n obj for obj in objs\n if rt.getUserProp(obj, \"id\") == AVALON_CONTAINER_ID\n ]\n\n for container in sorted(containers, key=attrgetter(\"name\")):\n yield lib.read(container)\n\n\ndef containerise(name: str, nodes: list, context,\n namespace=None, loader=None, suffix=\"_CON\"):\n data = {\n \"schema\": \"openpype:container-2.0\",\n \"id\": AVALON_CONTAINER_ID,\n \"name\": name,\n \"namespace\": namespace or \"\",\n \"loader\": loader,\n \"representation\": context[\"representation\"][\"_id\"],\n }\n container_name = f\"{namespace}:{name}{suffix}\"\n container = rt.container(name=container_name)\n import_custom_attribute_data(container, nodes)\n if not lib.imprint(container_name, data):\n print(f\"imprinting of {container_name} failed.\")\n return container\n\n\ndef load_custom_attribute_data():\n \"\"\"Re-loading the AYON custom parameter built by the creator\n\n Returns:\n attribute: re-loading the custom OP attributes set in Maxscript\n \"\"\"\n return rt.Execute(MS_CUSTOM_ATTRIB)\n\n\ndef import_custom_attribute_data(container: str, selections: list):\n \"\"\"Importing the Openpype/AYON custom parameter built by the creator\n\n Args:\n container (str): target container which adds custom attributes\n selections (list): nodes to be added into\n group in custom attributes\n \"\"\"\n attrs = load_custom_attribute_data()\n modifier = rt.EmptyModifier()\n rt.addModifier(container, modifier)\n container.modifiers[0].name = \"OP Data\"\n rt.custAttributes.add(container.modifiers[0], attrs)\n node_list = []\n sel_list = []\n for i in selections:\n node_ref = rt.NodeTransformMonitor(node=i)\n node_list.append(node_ref)\n sel_list.append(str(i))\n\n # Setting the property\n rt.setProperty(\n container.modifiers[0].openPypeData,\n \"all_handles\", node_list)\n rt.setProperty(\n container.modifiers[0].openPypeData,\n \"sel_list\", sel_list)\n\n\ndef update_custom_attribute_data(container: str, selections: list):\n \"\"\"Updating the AYON custom parameter built by the creator\n\n Args:\n container (str): target container which adds custom attributes\n selections (list): nodes to be added into\n group in custom attributes\n \"\"\"\n if container.modifiers[0].name == \"OP Data\":\n rt.deleteModifier(container, container.modifiers[0])\n import_custom_attribute_data(container, selections)\n\n\ndef get_previous_loaded_object(container: str):\n \"\"\"Get previous loaded_object through the OP data\n\n Args:\n container (str): the container which stores the OP data\n\n Returns:\n node_list(list): list of nodes which are previously loaded\n \"\"\"\n node_list = []\n sel_list = rt.getProperty(container.modifiers[0].openPypeData, \"sel_list\")\n for obj in rt.Objects:\n if str(obj) in sel_list:\n node_list.append(obj)\n return node_list\n"
},
{
"path": "openpype/hosts/max/api/plugin.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"3dsmax specific Avalon/Pyblish plugin definitions.\"\"\"\nfrom abc import ABCMeta\n\nimport six\nfrom pymxs import runtime as rt\n\nfrom openpype.lib import BoolDef\nfrom openpype.pipeline import CreatedInstance, Creator, CreatorError\n\nfrom .lib import imprint, lsattr, read\n\nMS_CUSTOM_ATTRIB = \"\"\"attributes \"openPypeData\"\n(\n parameters main rollout:OPparams\n (\n all_handles type:#maxObjectTab tabSize:0 tabSizeVariable:on\n sel_list type:#stringTab tabSize:0 tabSizeVariable:on\n )\n\n rollout OPparams \"OP Parameters\"\n (\n listbox list_node \"Node References\" items:#()\n button button_add \"Add to Container\"\n button button_del \"Delete from Container\"\n\n fn node_to_name the_node =\n (\n handle = the_node.handle\n obj_name = the_node.name\n handle_name = obj_name + \"<\" + handle as string + \">\"\n return handle_name\n )\n fn nodes_to_add node =\n (\n sceneObjs = #()\n if classOf node == Container do return false\n n = node as string\n for obj in Objects do\n (\n tmp_obj = obj as string\n append sceneObjs tmp_obj\n )\n if sel_list != undefined do\n (\n for obj in sel_list do\n (\n idx = findItem sceneObjs obj\n if idx do\n (\n deleteItem sceneObjs idx\n )\n )\n )\n idx = findItem sceneObjs n\n if idx then return true else false\n )\n\n fn nodes_to_rmv node =\n (\n n = node as string\n idx = findItem sel_list n\n if idx then return true else false\n )\n\n on button_add pressed do\n (\n current_sel = selectByName title:\"Select Objects to add to\n the Container\" buttontext:\"Add\" filter:nodes_to_add\n if current_sel == undefined then return False\n temp_arr = #()\n i_node_arr = #()\n for c in current_sel do\n (\n handle_name = node_to_name c\n node_ref = NodeTransformMonitor node:c\n idx = finditem list_node.items handle_name\n if idx do (\n continue\n )\n name = c as string\n append temp_arr handle_name\n append i_node_arr node_ref\n append sel_list name\n )\n all_handles = join i_node_arr all_handles\n list_node.items = join temp_arr list_node.items\n )\n\n on button_del pressed do\n (\n current_sel = selectByName title:\"Select Objects to remove\n from the Container\" buttontext:\"Remove\" filter: nodes_to_rmv\n if current_sel == undefined or current_sel.count == 0 then\n (\n return False\n )\n temp_arr = #()\n i_node_arr = #()\n new_i_node_arr = #()\n new_temp_arr = #()\n\n for c in current_sel do\n (\n node_ref = NodeTransformMonitor node:c as string\n handle_name = node_to_name c\n n = c as string\n tmp_all_handles = #()\n for i in all_handles do\n (\n tmp = i as string\n append tmp_all_handles tmp\n )\n idx = finditem tmp_all_handles node_ref\n if idx do\n (\n new_i_node_arr = DeleteItem all_handles idx\n\n )\n idx = finditem list_node.items handle_name\n if idx do\n (\n new_temp_arr = DeleteItem list_node.items idx\n )\n idx = finditem sel_list n\n if idx do\n (\n sel_list = DeleteItem sel_list idx\n )\n )\n all_handles = join i_node_arr new_i_node_arr\n list_node.items = join temp_arr new_temp_arr\n )\n\n on OPparams open do\n (\n if all_handles.count != 0 then\n (\n temp_arr = #()\n for x in all_handles do\n (\n if x.node == undefined do continue\n handle_name = node_to_name x.node\n append temp_arr handle_name\n )\n list_node.items = temp_arr\n )\n )\n )\n)\"\"\"\n\n\nclass OpenPypeCreatorError(CreatorError):\n pass\n\n\nclass MaxCreatorBase(object):\n\n @staticmethod\n def cache_subsets(shared_data):\n if shared_data.get(\"max_cached_subsets\") is not None:\n return shared_data\n\n shared_data[\"max_cached_subsets\"] = {}\n cached_instances = lsattr(\"id\", \"pyblish.avalon.instance\")\n for i in cached_instances:\n creator_id = rt.GetUserProp(i, \"creator_identifier\")\n if creator_id not in shared_data[\"max_cached_subsets\"]:\n shared_data[\"max_cached_subsets\"][creator_id] = [i.name]\n else:\n shared_data[\n \"max_cached_subsets\"][creator_id].append(i.name)\n return shared_data\n\n @staticmethod\n def create_instance_node(node):\n \"\"\"Create instance node.\n\n If the supplied node is existing node, it will be used to hold the\n instance, otherwise new node of type Dummy will be created.\n\n Args:\n node (rt.MXSWrapperBase, str): Node or node name to use.\n\n Returns:\n instance\n \"\"\"\n if isinstance(node, str):\n node = rt.Container(name=node)\n\n attrs = rt.Execute(MS_CUSTOM_ATTRIB)\n modifier = rt.EmptyModifier()\n rt.addModifier(node, modifier)\n node.modifiers[0].name = \"OP Data\"\n rt.custAttributes.add(node.modifiers[0], attrs)\n\n return node\n\n\n@six.add_metaclass(ABCMeta)\nclass MaxCreator(Creator, MaxCreatorBase):\n selected_nodes = []\n\n def create(self, subset_name, instance_data, pre_create_data):\n if pre_create_data.get(\"use_selection\"):\n self.selected_nodes = rt.GetCurrentSelection()\n if rt.getNodeByName(subset_name):\n raise CreatorError(f\"'{subset_name}' is already created..\")\n\n instance_node = self.create_instance_node(subset_name)\n instance_data[\"instance_node\"] = instance_node.name\n instance = CreatedInstance(\n self.family,\n subset_name,\n instance_data,\n self\n )\n if pre_create_data.get(\"use_selection\"):\n\n node_list = []\n sel_list = []\n for i in self.selected_nodes:\n node_ref = rt.NodeTransformMonitor(node=i)\n node_list.append(node_ref)\n sel_list.append(str(i))\n\n # Setting the property\n rt.setProperty(\n instance_node.modifiers[0].openPypeData,\n \"all_handles\", node_list)\n rt.setProperty(\n instance_node.modifiers[0].openPypeData,\n \"sel_list\", sel_list)\n\n self._add_instance_to_context(instance)\n imprint(instance_node.name, instance.data_to_store())\n\n return instance\n\n def collect_instances(self):\n self.cache_subsets(self.collection_shared_data)\n for instance in self.collection_shared_data[\"max_cached_subsets\"].get(self.identifier, []): # noqa\n created_instance = CreatedInstance.from_existing(\n read(rt.GetNodeByName(instance)), self\n )\n self._add_instance_to_context(created_instance)\n\n def update_instances(self, update_list):\n for created_inst, changes in update_list:\n instance_node = created_inst.get(\"instance_node\")\n new_values = {\n key: changes[key].new_value\n for key in changes.changed_keys\n }\n subset = new_values.get(\"subset\", \"\")\n if subset and instance_node != subset:\n node = rt.getNodeByName(instance_node)\n new_subset_name = new_values[\"subset\"]\n if rt.getNodeByName(new_subset_name):\n raise CreatorError(\n \"The subset '{}' already exists.\".format(\n new_subset_name))\n instance_node = new_subset_name\n created_inst[\"instance_node\"] = instance_node\n node.name = instance_node\n\n imprint(\n instance_node,\n created_inst.data_to_store(),\n )\n\n def remove_instances(self, instances):\n \"\"\"Remove specified instance from the scene.\n\n This is only removing `id` parameter so instance is no longer\n instance, because it might contain valuable data for artist.\n\n \"\"\"\n for instance in instances:\n instance_node = rt.GetNodeByName(\n instance.data.get(\"instance_node\"))\n if instance_node:\n count = rt.custAttributes.count(instance_node.modifiers[0])\n rt.custAttributes.delete(instance_node.modifiers[0], count)\n rt.Delete(instance_node)\n\n self._remove_instance_from_context(instance)\n\n def get_pre_create_attr_defs(self):\n return [\n BoolDef(\"use_selection\", label=\"Use selection\")\n ]\n"
},
{
"path": "openpype/hosts/max/api/preview_animation.py",
"content": "import logging\nimport contextlib\nfrom pymxs import runtime as rt\nfrom .lib import get_max_version, render_resolution\n\nlog = logging.getLogger(\"openpype.hosts.max\")\n\n\n@contextlib.contextmanager\ndef play_preview_when_done(has_autoplay):\n \"\"\"Set preview playback option during context\n\n Args:\n has_autoplay (bool): autoplay during creating\n preview animation\n \"\"\"\n current_playback = rt.preferences.playPreviewWhenDone\n try:\n rt.preferences.playPreviewWhenDone = has_autoplay\n yield\n finally:\n rt.preferences.playPreviewWhenDone = current_playback\n\n\n@contextlib.contextmanager\ndef viewport_layout_and_camera(camera, layout=\"layout_1\"):\n \"\"\"Set viewport layout and camera during context\n ***For 3dsMax 2024+\n Args:\n camera (str): viewport camera\n layout (str): layout to use in viewport, defaults to `layout_1`\n Use None to not change viewport layout during context.\n \"\"\"\n original_camera = rt.viewport.getCamera()\n original_layout = rt.viewport.getLayout()\n if not original_camera:\n # if there is no original camera\n # use the current camera as original\n original_camera = rt.getNodeByName(camera)\n review_camera = rt.getNodeByName(camera)\n try:\n if layout is not None:\n layout = rt.Name(layout)\n if rt.viewport.getLayout() != layout:\n rt.viewport.setLayout(layout)\n rt.viewport.setCamera(review_camera)\n yield\n finally:\n rt.viewport.setLayout(original_layout)\n rt.viewport.setCamera(original_camera)\n\n\n@contextlib.contextmanager\ndef viewport_preference_setting(general_viewport,\n nitrous_manager,\n nitrous_viewport,\n vp_button_mgr):\n \"\"\"Function to set viewport setting during context\n ***For Max Version < 2024\n Args:\n camera (str): Viewport camera for review render\n general_viewport (dict): General viewport setting\n nitrous_manager (dict): Nitrous graphic manager\n nitrous_viewport (dict): Nitrous setting for\n preview animation\n vp_button_mgr (dict): Viewport button manager Setting\n preview_preferences (dict): Preview Preferences Setting\n \"\"\"\n orig_vp_grid = rt.viewport.getGridVisibility(1)\n orig_vp_bkg = rt.viewport.IsSolidBackgroundColorMode()\n\n nitrousGraphicMgr = rt.NitrousGraphicsManager\n viewport_setting = nitrousGraphicMgr.GetActiveViewportSetting()\n vp_button_mgr_original = {\n key: getattr(rt.ViewportButtonMgr, key) for key in vp_button_mgr\n }\n nitrous_manager_original = {\n key: getattr(nitrousGraphicMgr, key) for key in nitrous_manager\n }\n nitrous_viewport_original = {\n key: getattr(viewport_setting, key) for key in nitrous_viewport\n }\n\n try:\n rt.viewport.setGridVisibility(1, general_viewport[\"dspGrid\"])\n rt.viewport.EnableSolidBackgroundColorMode(general_viewport[\"dspBkg\"])\n for key, value in vp_button_mgr.items():\n setattr(rt.ViewportButtonMgr, key, value)\n for key, value in nitrous_manager.items():\n setattr(nitrousGraphicMgr, key, value)\n for key, value in nitrous_viewport.items():\n if nitrous_viewport[key] != nitrous_viewport_original[key]:\n setattr(viewport_setting, key, value)\n yield\n\n finally:\n rt.viewport.setGridVisibility(1, orig_vp_grid)\n rt.viewport.EnableSolidBackgroundColorMode(orig_vp_bkg)\n for key, value in vp_button_mgr_original.items():\n setattr(rt.ViewportButtonMgr, key, value)\n for key, value in nitrous_manager_original.items():\n setattr(nitrousGraphicMgr, key, value)\n for key, value in nitrous_viewport_original.items():\n setattr(viewport_setting, key, value)\n\n\ndef _render_preview_animation_max_2024(\n filepath, start, end, percentSize, ext, viewport_options):\n \"\"\"Render viewport preview with MaxScript using `CreateAnimation`.\n ****For 3dsMax 2024+\n Args:\n filepath (str): filepath for render output without frame number and\n extension, for example: /path/to/file\n start (int): startFrame\n end (int): endFrame\n percentSize (float): render resolution multiplier by 100\n e.g. 100.0 is 1x, 50.0 is 0.5x, 150.0 is 1.5x\n viewport_options (dict): viewport setting options, e.g.\n {\"vpStyle\": \"defaultshading\", \"vpPreset\": \"highquality\"}\n Returns:\n list: Created files\n \"\"\"\n # the percentSize argument must be integer\n percent = int(percentSize)\n filepath = filepath.replace(\"\\\\\", \"/\")\n preview_output = f\"{filepath}..{ext}\"\n frame_template = f\"{filepath}.{{:04d}}.{ext}\"\n job_args = []\n for key, value in viewport_options.items():\n if isinstance(value, bool):\n if value:\n job_args.append(f\"{key}:{value}\")\n elif isinstance(value, str):\n if key == \"vpStyle\":\n if value == \"Realistic\":\n value = \"defaultshading\"\n elif value == \"Shaded\":\n log.warning(\n \"'Shaded' Mode not supported in \"\n \"preview animation in Max 2024.\\n\"\n \"Using 'defaultshading' instead.\")\n value = \"defaultshading\"\n elif value == \"ConsistentColors\":\n value = \"flatcolor\"\n else:\n value = value.lower()\n elif key == \"vpPreset\":\n if value == \"Quality\":\n value = \"highquality\"\n elif value == \"Customize\":\n value = \"userdefined\"\n else:\n value = value.lower()\n job_args.append(f\"{key}: #{value}\")\n\n job_str = (\n f'CreatePreview filename:\"{preview_output}\" outputAVI:false '\n f\"percentSize:{percent} start:{start} end:{end} \"\n f\"{' '.join(job_args)} \"\n \"autoPlay:false\"\n )\n rt.completeRedraw()\n rt.execute(job_str)\n # Return the created files\n return [frame_template.format(frame) for frame in range(start, end + 1)]\n\n\ndef _render_preview_animation_max_pre_2024(\n filepath, startFrame, endFrame,\n width, height, percentSize, ext):\n \"\"\"Render viewport animation by creating bitmaps\n ***For 3dsMax Version <2024\n Args:\n filepath (str): filepath without frame numbers and extension\n startFrame (int): start frame\n endFrame (int): end frame\n width (int): render resolution width\n height (int): render resolution height\n percentSize (float): render resolution multiplier by 100\n e.g. 100.0 is 1x, 50.0 is 0.5x, 150.0 is 1.5x\n ext (str): image extension\n Returns:\n list: Created filepaths\n \"\"\"\n\n # get the screenshot\n percent = percentSize / 100.0\n res_width = width * percent\n res_height = height * percent\n frame_template = \"{}.{{:04}}.{}\".format(filepath, ext)\n frame_template.replace(\"\\\\\", \"/\")\n files = []\n user_cancelled = False\n for frame in range(startFrame, endFrame + 1):\n rt.sliderTime = frame\n filepath = frame_template.format(frame)\n preview_res = rt.bitmap(\n res_width, res_height, filename=filepath\n )\n dib = rt.gw.getViewportDib()\n dib_width = float(dib.width)\n dib_height = float(dib.height)\n # aspect ratio\n viewportRatio = dib_width / dib_height\n renderRatio = float(res_width / res_height)\n if viewportRatio < renderRatio:\n heightCrop = (dib_width / renderRatio)\n topEdge = int((dib_height - heightCrop) / 2.0)\n tempImage_bmp = rt.bitmap(dib_width, heightCrop)\n src_box_value = rt.Box2(0, topEdge, dib_width, heightCrop)\n rt.pasteBitmap(dib, tempImage_bmp, src_box_value, rt.Point2(0, 0))\n rt.copy(tempImage_bmp, preview_res)\n rt.close(tempImage_bmp)\n elif viewportRatio > renderRatio:\n widthCrop = dib_height * renderRatio\n leftEdge = int((dib_width - widthCrop) / 2.0)\n tempImage_bmp = rt.bitmap(widthCrop, dib_height)\n src_box_value = rt.Box2(leftEdge, 0, widthCrop, dib_height)\n rt.pasteBitmap(dib, tempImage_bmp, src_box_value, rt.Point2(0, 0))\n rt.copy(tempImage_bmp, preview_res)\n rt.close(tempImage_bmp)\n else:\n rt.copy(dib, preview_res)\n rt.save(preview_res)\n rt.close(preview_res)\n rt.close(dib)\n files.append(filepath)\n if rt.keyboard.escPressed:\n user_cancelled = True\n break\n # clean up the cache\n rt.gc(delayed=True)\n if user_cancelled:\n raise RuntimeError(\"User cancelled rendering of viewport animation.\")\n return files\n\n\ndef render_preview_animation(\n filepath,\n ext,\n camera,\n start_frame=None,\n end_frame=None,\n percentSize=100.0,\n width=1920,\n height=1080,\n viewport_options=None):\n \"\"\"Render camera review animation\n Args:\n filepath (str): filepath to render to, without frame number and\n extension\n ext (str): output file extension\n camera (str): viewport camera for preview render\n start_frame (int): start frame\n end_frame (int): end frame\n percentSize (float): render resolution multiplier by 100\n e.g. 100.0 is 1x, 50.0 is 0.5x, 150.0 is 1.5x\n width (int): render resolution width\n height (int): render resolution height\n viewport_options (dict): viewport setting options\n Returns:\n list: Rendered output files\n \"\"\"\n if start_frame is None:\n start_frame = int(rt.animationRange.start)\n if end_frame is None:\n end_frame = int(rt.animationRange.end)\n\n if viewport_options is None:\n viewport_options = viewport_options_for_preview_animation()\n with play_preview_when_done(False):\n with viewport_layout_and_camera(camera):\n if int(get_max_version()) < 2024:\n with viewport_preference_setting(\n viewport_options[\"general_viewport\"],\n viewport_options[\"nitrous_manager\"],\n viewport_options[\"nitrous_viewport\"],\n viewport_options[\"vp_btn_mgr\"]\n ):\n return _render_preview_animation_max_pre_2024(\n filepath,\n start_frame,\n end_frame,\n width,\n height,\n percentSize,\n ext\n )\n else:\n with render_resolution(width, height):\n return _render_preview_animation_max_2024(\n filepath,\n start_frame,\n end_frame,\n percentSize,\n ext,\n viewport_options\n )\n\n\ndef viewport_options_for_preview_animation():\n \"\"\"Get default viewport options for `render_preview_animation`.\n\n Returns:\n dict: viewport setting options\n \"\"\"\n # viewport_options should be the dictionary\n if int(get_max_version()) < 2024:\n return {\n \"visualStyleMode\": \"defaultshading\",\n \"viewportPreset\": \"highquality\",\n \"vpTexture\": False,\n \"dspGeometry\": True,\n \"dspShapes\": False,\n \"dspLights\": False,\n \"dspCameras\": False,\n \"dspHelpers\": False,\n \"dspParticles\": True,\n \"dspBones\": False,\n \"dspBkg\": True,\n \"dspGrid\": False,\n \"dspSafeFrame\": False,\n \"dspFrameNums\": False\n }\n else:\n viewport_options = {}\n viewport_options[\"general_viewport\"] = {\n \"dspBkg\": True,\n \"dspGrid\": False\n }\n viewport_options[\"nitrous_manager\"] = {\n \"AntialiasingQuality\": \"None\"\n }\n viewport_options[\"nitrous_viewport\"] = {\n \"VisualStyleMode\": \"defaultshading\",\n \"ViewportPreset\": \"highquality\",\n \"UseTextureEnabled\": False\n }\n viewport_options[\"vp_btn_mgr\"] = {\n \"EnableButtons\": False}\n return viewport_options\n"
},
{
"path": "openpype/hosts/max/hooks/force_startup_script.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Pre-launch to force 3ds max startup script.\"\"\"\nimport os\nfrom openpype.hosts.max import MAX_HOST_DIR\nfrom openpype.lib.applications import PreLaunchHook, LaunchTypes\n\n\nclass ForceStartupScript(PreLaunchHook):\n \"\"\"Inject OpenPype environment to 3ds max.\n\n Note that this works in combination whit 3dsmax startup script that\n is translating it back to PYTHONPATH for cases when 3dsmax drops PYTHONPATH\n environment.\n\n Hook `GlobalHostDataHook` must be executed before this hook.\n \"\"\"\n app_groups = {\"3dsmax\", \"adsk_3dsmax\"}\n order = 11\n launch_types = {LaunchTypes.local}\n\n def execute(self):\n startup_args = [\n \"-U\",\n \"MAXScript\",\n os.path.join(MAX_HOST_DIR, \"startup\", \"startup.ms\"),\n ]\n self.launch_context.launch_args.append(startup_args)\n"
},
{
"path": "openpype/hosts/max/hooks/inject_python.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Pre-launch hook to inject python environment.\"\"\"\nimport os\nfrom openpype.lib.applications import PreLaunchHook, LaunchTypes\n\n\nclass InjectPythonPath(PreLaunchHook):\n \"\"\"Inject OpenPype environment to 3dsmax.\n\n Note that this works in combination whit 3dsmax startup script that\n is translating it back to PYTHONPATH for cases when 3dsmax drops PYTHONPATH\n environment.\n\n Hook `GlobalHostDataHook` must be executed before this hook.\n \"\"\"\n app_groups = {\"3dsmax\", \"adsk_3dsmax\"}\n launch_types = {LaunchTypes.local}\n\n def execute(self):\n self.launch_context.env[\"MAX_PYTHONPATH\"] = os.environ[\"PYTHONPATH\"]\n"
},
{
"path": "openpype/hosts/max/hooks/set_paths.py",
"content": "from openpype.lib.applications import PreLaunchHook, LaunchTypes\n\n\nclass SetPath(PreLaunchHook):\n \"\"\"Set current dir to workdir.\n\n Hook `GlobalHostDataHook` must be executed before this hook.\n \"\"\"\n app_groups = {\"max\"}\n launch_types = {LaunchTypes.local}\n\n def execute(self):\n workdir = self.launch_context.env.get(\"AVALON_WORKDIR\", \"\")\n if not workdir:\n self.log.warning(\"BUG: Workdir is not filled.\")\n return\n\n self.launch_context.kwargs[\"cwd\"] = workdir\n"
},
{
"path": "openpype/hosts/max/plugins/__init__.py",
"content": ""
},
{
"path": "openpype/hosts/max/plugins/create/create_camera.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin for creating camera.\"\"\"\nfrom openpype.hosts.max.api import plugin\n\n\nclass CreateCamera(plugin.MaxCreator):\n \"\"\"Creator plugin for Camera.\"\"\"\n identifier = \"io.openpype.creators.max.camera\"\n label = \"Camera\"\n family = \"camera\"\n icon = \"gear\"\n"
},
{
"path": "openpype/hosts/max/plugins/create/create_maxScene.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin for creating raw max scene.\"\"\"\nfrom openpype.hosts.max.api import plugin\n\n\nclass CreateMaxScene(plugin.MaxCreator):\n \"\"\"Creator plugin for 3ds max scenes.\"\"\"\n identifier = \"io.openpype.creators.max.maxScene\"\n label = \"Max Scene\"\n family = \"maxScene\"\n icon = \"gear\"\n"
},
{
"path": "openpype/hosts/max/plugins/create/create_model.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin for model.\"\"\"\nfrom openpype.hosts.max.api import plugin\n\n\nclass CreateModel(plugin.MaxCreator):\n \"\"\"Creator plugin for Model.\"\"\"\n identifier = \"io.openpype.creators.max.model\"\n label = \"Model\"\n family = \"model\"\n icon = \"gear\"\n"
},
{
"path": "openpype/hosts/max/plugins/create/create_pointcache.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin for creating pointcache alembics.\"\"\"\nfrom openpype.hosts.max.api import plugin\n\n\nclass CreatePointCache(plugin.MaxCreator):\n \"\"\"Creator plugin for Point caches.\"\"\"\n identifier = \"io.openpype.creators.max.pointcache\"\n label = \"Point Cache\"\n family = \"pointcache\"\n icon = \"gear\"\n"
},
{
"path": "openpype/hosts/max/plugins/create/create_pointcloud.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin for creating point cloud.\"\"\"\nfrom openpype.hosts.max.api import plugin\n\n\nclass CreatePointCloud(plugin.MaxCreator):\n \"\"\"Creator plugin for Point Clouds.\"\"\"\n identifier = \"io.openpype.creators.max.pointcloud\"\n label = \"Point Cloud\"\n family = \"pointcloud\"\n icon = \"gear\"\n"
},
{
"path": "openpype/hosts/max/plugins/create/create_redshift_proxy.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin for creating camera.\"\"\"\nfrom openpype.hosts.max.api import plugin\nfrom openpype.pipeline import CreatedInstance\n\n\nclass CreateRedshiftProxy(plugin.MaxCreator):\n identifier = \"io.openpype.creators.max.redshiftproxy\"\n label = \"Redshift Proxy\"\n family = \"redshiftproxy\"\n icon = \"gear\"\n"
},
{
"path": "openpype/hosts/max/plugins/create/create_render.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin for creating camera.\"\"\"\nimport os\nfrom openpype.hosts.max.api import plugin\nfrom openpype.lib import BoolDef\nfrom openpype.hosts.max.api.lib_rendersettings import RenderSettings\n\n\nclass CreateRender(plugin.MaxCreator):\n \"\"\"Creator plugin for Renders.\"\"\"\n identifier = \"io.openpype.creators.max.render\"\n label = \"Render\"\n family = \"maxrender\"\n icon = \"gear\"\n\n def create(self, subset_name, instance_data, pre_create_data):\n from pymxs import runtime as rt\n file = rt.maxFileName\n filename, _ = os.path.splitext(file)\n instance_data[\"AssetName\"] = filename\n instance_data[\"multiCamera\"] = pre_create_data.get(\"multi_cam\")\n num_of_renderlayer = rt.batchRenderMgr.numViews\n if num_of_renderlayer > 0:\n rt.batchRenderMgr.DeleteView(num_of_renderlayer)\n\n instance = super(CreateRender, self).create(\n subset_name,\n instance_data,\n pre_create_data)\n\n container_name = instance.data.get(\"instance_node\")\n # set output paths for rendering(mandatory for deadline)\n RenderSettings().render_output(container_name)\n # TODO: create multiple camera options\n if self.selected_nodes:\n selected_nodes_name = []\n for sel in self.selected_nodes:\n name = sel.name\n selected_nodes_name.append(name)\n RenderSettings().batch_render_layer(\n container_name, filename,\n selected_nodes_name)\n\n def get_pre_create_attr_defs(self):\n attrs = super(CreateRender, self).get_pre_create_attr_defs()\n return attrs + [\n BoolDef(\"multi_cam\",\n label=\"Multiple Cameras Submission\",\n default=False),\n ]\n"
},
{
"path": "openpype/hosts/max/plugins/create/create_review.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin for creating review in Max.\"\"\"\nfrom openpype.hosts.max.api import plugin\nfrom openpype.lib import BoolDef, EnumDef, NumberDef\n\n\nclass CreateReview(plugin.MaxCreator):\n \"\"\"Review in 3dsMax\"\"\"\n\n identifier = \"io.openpype.creators.max.review\"\n label = \"Review\"\n family = \"review\"\n icon = \"video-camera\"\n\n review_width = 1920\n review_height = 1080\n percentSize = 100\n keep_images = False\n image_format = \"png\"\n visual_style = \"Realistic\"\n viewport_preset = \"Quality\"\n vp_texture = True\n anti_aliasing = \"None\"\n\n def apply_settings(self, project_settings):\n settings = project_settings[\"max\"][\"CreateReview\"] # noqa\n\n # Take some defaults from settings\n self.review_width = settings.get(\"review_width\", self.review_width)\n self.review_height = settings.get(\"review_height\", self.review_height)\n self.percentSize = settings.get(\"percentSize\", self.percentSize)\n self.keep_images = settings.get(\"keep_images\", self.keep_images)\n self.image_format = settings.get(\"image_format\", self.image_format)\n self.visual_style = settings.get(\"visual_style\", self.visual_style)\n self.viewport_preset = settings.get(\n \"viewport_preset\", self.viewport_preset)\n self.anti_aliasing = settings.get(\n \"anti_aliasing\", self.anti_aliasing)\n self.vp_texture = settings.get(\"vp_texture\", self.vp_texture)\n\n def create(self, subset_name, instance_data, pre_create_data):\n # Transfer settings from pre create to instance\n creator_attributes = instance_data.setdefault(\n \"creator_attributes\", dict())\n for key in [\"imageFormat\",\n \"keepImages\",\n \"review_width\",\n \"review_height\",\n \"percentSize\",\n \"visualStyleMode\",\n \"viewportPreset\",\n \"antialiasingQuality\",\n \"vpTexture\"]:\n if key in pre_create_data:\n creator_attributes[key] = pre_create_data[key]\n\n super(CreateReview, self).create(\n subset_name,\n instance_data,\n pre_create_data)\n\n def get_instance_attr_defs(self):\n image_format_enum = [\"exr\", \"jpg\", \"png\", \"tga\"]\n\n visual_style_preset_enum = [\n \"Realistic\", \"Shaded\", \"Facets\",\n \"ConsistentColors\", \"HiddenLine\",\n \"Wireframe\", \"BoundingBox\", \"Ink\",\n \"ColorInk\", \"Acrylic\", \"Tech\", \"Graphite\",\n \"ColorPencil\", \"Pastel\", \"Clay\", \"ModelAssist\"\n ]\n preview_preset_enum = [\n \"Quality\", \"Standard\", \"Performance\",\n \"DXMode\", \"Customize\"]\n anti_aliasing_enum = [\"None\", \"2X\", \"4X\", \"8X\"]\n\n return [\n NumberDef(\"review_width\",\n label=\"Review width\",\n decimals=0,\n minimum=0,\n default=self.review_width),\n NumberDef(\"review_height\",\n label=\"Review height\",\n decimals=0,\n minimum=0,\n default=self.review_height),\n NumberDef(\"percentSize\",\n label=\"Percent of Output\",\n default=self.percentSize,\n minimum=1,\n decimals=0),\n BoolDef(\"keepImages\",\n label=\"Keep Image Sequences\",\n default=self.keep_images),\n EnumDef(\"imageFormat\",\n image_format_enum,\n default=self.image_format,\n label=\"Image Format Options\"),\n EnumDef(\"visualStyleMode\",\n visual_style_preset_enum,\n default=self.visual_style,\n label=\"Preference\"),\n EnumDef(\"viewportPreset\",\n preview_preset_enum,\n default=self.viewport_preset,\n label=\"Preview Preset\"),\n EnumDef(\"antialiasingQuality\",\n anti_aliasing_enum,\n default=self.anti_aliasing,\n label=\"Anti-aliasing Quality\"),\n BoolDef(\"vpTexture\",\n label=\"Viewport Texture\",\n default=self.vp_texture)\n ]\n\n def get_pre_create_attr_defs(self):\n # Use same attributes as for instance attributes\n attrs = super().get_pre_create_attr_defs()\n return attrs + self.get_instance_attr_defs()\n"
},
{
"path": "openpype/hosts/max/plugins/create/create_tycache.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin for creating TyCache.\"\"\"\nfrom openpype.hosts.max.api import plugin\n\n\nclass CreateTyCache(plugin.MaxCreator):\n \"\"\"Creator plugin for TyCache.\"\"\"\n identifier = \"io.openpype.creators.max.tycache\"\n label = \"TyCache\"\n family = \"tycache\"\n icon = \"gear\"\n"
},
{
"path": "openpype/hosts/max/plugins/create/create_workfile.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin for creating workfiles.\"\"\"\nfrom openpype import AYON_SERVER_ENABLED\nfrom openpype.pipeline import CreatedInstance, AutoCreator\nfrom openpype.client import get_asset_by_name, get_asset_name_identifier\nfrom openpype.hosts.max.api import plugin\nfrom openpype.hosts.max.api.lib import read, imprint\nfrom pymxs import runtime as rt\n\n\nclass CreateWorkfile(plugin.MaxCreatorBase, AutoCreator):\n \"\"\"Workfile auto-creator.\"\"\"\n identifier = \"io.openpype.creators.max.workfile\"\n label = \"Workfile\"\n family = \"workfile\"\n icon = \"fa5.file\"\n\n default_variant = \"Main\"\n\n def create(self):\n variant = self.default_variant\n current_instance = next(\n (\n instance for instance in self.create_context.instances\n if instance.creator_identifier == self.identifier\n ), None)\n project_name = self.project_name\n asset_name = self.create_context.get_current_asset_name()\n task_name = self.create_context.get_current_task_name()\n host_name = self.create_context.host_name\n\n if current_instance is None:\n current_instance_asset = None\n elif AYON_SERVER_ENABLED:\n current_instance_asset = current_instance[\"folderPath\"]\n else:\n current_instance_asset = current_instance[\"asset\"]\n\n if current_instance is None:\n asset_doc = get_asset_by_name(project_name, asset_name)\n subset_name = self.get_subset_name(\n variant, task_name, asset_doc, project_name, host_name\n )\n data = {\n \"task\": task_name,\n \"variant\": variant\n }\n if AYON_SERVER_ENABLED:\n data[\"folderPath\"] = asset_name\n else:\n data[\"asset\"] = asset_name\n\n data.update(\n self.get_dynamic_data(\n variant, task_name, asset_doc,\n project_name, host_name, current_instance)\n )\n self.log.info(\"Auto-creating workfile instance...\")\n instance_node = self.create_node(subset_name)\n data[\"instance_node\"] = instance_node.name\n current_instance = CreatedInstance(\n self.family, subset_name, data, self\n )\n self._add_instance_to_context(current_instance)\n imprint(instance_node.name, current_instance.data)\n elif (\n current_instance_asset != asset_name\n or current_instance[\"task\"] != task_name\n ):\n # Update instance context if is not the same\n asset_doc = get_asset_by_name(project_name, asset_name)\n subset_name = self.get_subset_name(\n variant, task_name, asset_doc, project_name, host_name\n )\n asset_name = get_asset_name_identifier(asset_doc)\n\n if AYON_SERVER_ENABLED:\n current_instance[\"folderPath\"] = asset_name\n else:\n current_instance[\"asset\"] = asset_name\n current_instance[\"task\"] = task_name\n current_instance[\"subset\"] = subset_name\n\n def collect_instances(self):\n self.cache_subsets(self.collection_shared_data)\n for instance in self.collection_shared_data[\"max_cached_subsets\"].get(self.identifier, []): # noqa\n if not rt.getNodeByName(instance):\n continue\n created_instance = CreatedInstance.from_existing(\n read(rt.GetNodeByName(instance)), self\n )\n self._add_instance_to_context(created_instance)\n\n def update_instances(self, update_list):\n for created_inst, _ in update_list:\n instance_node = created_inst.get(\"instance_node\")\n imprint(\n instance_node,\n created_inst.data_to_store()\n )\n\n def remove_instances(self, instances):\n \"\"\"Remove specified instance from the scene.\n\n This is only removing `id` parameter so instance is no longer\n instance, because it might contain valuable data for artist.\n\n \"\"\"\n for instance in instances:\n instance_node = rt.GetNodeByName(\n instance.data.get(\"instance_node\"))\n if instance_node:\n rt.Delete(instance_node)\n\n self._remove_instance_from_context(instance)\n\n def create_node(self, subset_name):\n if rt.getNodeByName(subset_name):\n node = rt.getNodeByName(subset_name)\n return node\n node = rt.Container(name=subset_name)\n node.isHidden = True\n return node\n"
},
{
"path": "openpype/hosts/max/plugins/load/load_camera_fbx.py",
"content": "import os\n\nfrom openpype.hosts.max.api import lib, maintained_selection\nfrom openpype.hosts.max.api.lib import (\n unique_namespace,\n get_namespace,\n object_transform_set\n)\nfrom openpype.hosts.max.api.pipeline import (\n containerise,\n get_previous_loaded_object,\n update_custom_attribute_data\n)\nfrom openpype.pipeline import get_representation_path, load\n\n\nclass FbxLoader(load.LoaderPlugin):\n \"\"\"Fbx Loader.\"\"\"\n\n families = [\"camera\"]\n representations = [\"fbx\"]\n order = -9\n icon = \"code-fork\"\n color = \"white\"\n\n def load(self, context, name=None, namespace=None, data=None):\n from pymxs import runtime as rt\n filepath = self.filepath_from_context(context)\n filepath = os.path.normpath(filepath)\n rt.FBXImporterSetParam(\"Animation\", True)\n rt.FBXImporterSetParam(\"Camera\", True)\n rt.FBXImporterSetParam(\"AxisConversionMethod\", True)\n rt.FBXImporterSetParam(\"Mode\", rt.Name(\"create\"))\n rt.FBXImporterSetParam(\"Preserveinstances\", True)\n rt.ImportFile(\n filepath,\n rt.name(\"noPrompt\"),\n using=rt.FBXIMP)\n\n namespace = unique_namespace(\n name + \"_\",\n suffix=\"_\",\n )\n selections = rt.GetCurrentSelection()\n\n for selection in selections:\n selection.name = f\"{namespace}:{selection.name}\"\n\n return containerise(\n name, selections, context,\n namespace, loader=self.__class__.__name__)\n\n def update(self, container, representation):\n from pymxs import runtime as rt\n\n path = get_representation_path(representation)\n node_name = container[\"instance_node\"]\n node = rt.getNodeByName(node_name)\n namespace, _ = get_namespace(node_name)\n\n node_list = get_previous_loaded_object(node)\n rt.Select(node_list)\n prev_fbx_objects = rt.GetCurrentSelection()\n transform_data = object_transform_set(prev_fbx_objects)\n for prev_fbx_obj in prev_fbx_objects:\n if rt.isValidNode(prev_fbx_obj):\n rt.Delete(prev_fbx_obj)\n\n rt.FBXImporterSetParam(\"Animation\", True)\n rt.FBXImporterSetParam(\"Camera\", True)\n rt.FBXImporterSetParam(\"Mode\", rt.Name(\"merge\"))\n rt.FBXImporterSetParam(\"AxisConversionMethod\", True)\n rt.FBXImporterSetParam(\"Preserveinstances\", True)\n rt.ImportFile(\n path, rt.name(\"noPrompt\"), using=rt.FBXIMP)\n current_fbx_objects = rt.GetCurrentSelection()\n fbx_objects = []\n for fbx_object in current_fbx_objects:\n fbx_object.name = f\"{namespace}:{fbx_object.name}\"\n fbx_objects.append(fbx_object)\n fbx_transform = f\"{fbx_object.name}.transform\"\n if fbx_transform in transform_data.keys():\n fbx_object.pos = transform_data[fbx_transform] or 0\n fbx_object.scale = transform_data[\n f\"{fbx_object.name}.scale\"] or 0\n\n update_custom_attribute_data(node, fbx_objects)\n lib.imprint(container[\"instance_node\"], {\n \"representation\": str(representation[\"_id\"])\n })\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def remove(self, container):\n from pymxs import runtime as rt\n\n node = rt.GetNodeByName(container[\"instance_node\"])\n rt.Delete(node)\n"
},
{
"path": "openpype/hosts/max/plugins/load/load_max_scene.py",
"content": "import os\n\nfrom openpype.hosts.max.api import lib\nfrom openpype.hosts.max.api.lib import (\n unique_namespace,\n get_namespace,\n object_transform_set\n)\nfrom openpype.hosts.max.api.pipeline import (\n containerise, get_previous_loaded_object,\n update_custom_attribute_data\n)\nfrom openpype.pipeline import get_representation_path, load\n\n\nclass MaxSceneLoader(load.LoaderPlugin):\n \"\"\"Max Scene Loader.\"\"\"\n\n families = [\"camera\",\n \"maxScene\",\n \"model\"]\n\n representations = [\"max\"]\n order = -8\n icon = \"code-fork\"\n color = \"green\"\n\n def load(self, context, name=None, namespace=None, data=None):\n from pymxs import runtime as rt\n path = self.filepath_from_context(context)\n path = os.path.normpath(path)\n # import the max scene by using \"merge file\"\n path = path.replace('\\\\', '/')\n rt.MergeMaxFile(path, quiet=True, includeFullGroup=True)\n max_objects = rt.getLastMergedNodes()\n max_object_names = [obj.name for obj in max_objects]\n # implement the OP/AYON custom attributes before load\n max_container = []\n namespace = unique_namespace(\n name + \"_\",\n suffix=\"_\",\n )\n for max_obj, obj_name in zip(max_objects, max_object_names):\n max_obj.name = f\"{namespace}:{obj_name}\"\n max_container.append(rt.getNodeByName(max_obj.name))\n return containerise(\n name, max_container, context,\n namespace, loader=self.__class__.__name__)\n\n def update(self, container, representation):\n from pymxs import runtime as rt\n\n path = get_representation_path(representation)\n node_name = container[\"instance_node\"]\n node = rt.getNodeByName(node_name)\n namespace, _ = get_namespace(node_name)\n # delete the old container with attribute\n # delete old duplicate\n # use the modifier OP data to delete the data\n node_list = get_previous_loaded_object(node)\n rt.select(node_list)\n prev_max_objects = rt.GetCurrentSelection()\n transform_data = object_transform_set(prev_max_objects)\n\n for prev_max_obj in prev_max_objects:\n if rt.isValidNode(prev_max_obj): # noqa\n rt.Delete(prev_max_obj)\n rt.MergeMaxFile(path, quiet=True)\n\n current_max_objects = rt.getLastMergedNodes()\n\n current_max_object_names = [obj.name for obj\n in current_max_objects]\n\n max_objects = []\n for max_obj, obj_name in zip(current_max_objects,\n current_max_object_names):\n max_obj.name = f\"{namespace}:{obj_name}\"\n max_objects.append(max_obj)\n max_transform = f\"{max_obj.name}.transform\"\n if max_transform in transform_data.keys():\n max_obj.pos = transform_data[max_transform] or 0\n max_obj.scale = transform_data[\n f\"{max_obj.name}.scale\"] or 0\n\n update_custom_attribute_data(node, max_objects)\n lib.imprint(container[\"instance_node\"], {\n \"representation\": str(representation[\"_id\"])\n })\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def remove(self, container):\n from pymxs import runtime as rt\n\n node = rt.GetNodeByName(container[\"instance_node\"])\n rt.Delete(node)\n"
},
{
"path": "openpype/hosts/max/plugins/load/load_model.py",
"content": "import os\nfrom openpype.pipeline import load, get_representation_path\nfrom openpype.hosts.max.api.pipeline import (\n containerise,\n get_previous_loaded_object\n)\nfrom openpype.hosts.max.api import lib\nfrom openpype.hosts.max.api.lib import (\n maintained_selection, unique_namespace\n)\n\n\nclass ModelAbcLoader(load.LoaderPlugin):\n \"\"\"Loading model with the Alembic loader.\"\"\"\n\n families = [\"model\"]\n label = \"Load Model with Alembic\"\n representations = [\"abc\"]\n order = -10\n icon = \"code-fork\"\n color = \"orange\"\n\n def load(self, context, name=None, namespace=None, data=None):\n from pymxs import runtime as rt\n\n file_path = os.path.normpath(self.filepath_from_context(context))\n\n abc_before = {\n c\n for c in rt.rootNode.Children\n if rt.classOf(c) == rt.AlembicContainer\n }\n\n rt.AlembicImport.ImportToRoot = False\n rt.AlembicImport.CustomAttributes = True\n rt.AlembicImport.UVs = True\n rt.AlembicImport.VertexColors = True\n rt.importFile(file_path, rt.name(\"noPrompt\"), using=rt.AlembicImport)\n\n abc_after = {\n c\n for c in rt.rootNode.Children\n if rt.classOf(c) == rt.AlembicContainer\n }\n\n # This should yield new AlembicContainer node\n abc_containers = abc_after.difference(abc_before)\n\n if len(abc_containers) != 1:\n self.log.error(\"Something failed when loading.\")\n\n abc_container = abc_containers.pop()\n\n namespace = unique_namespace(\n name + \"_\",\n suffix=\"_\",\n )\n abc_objects = []\n for abc_object in abc_container.Children:\n abc_object.name = f\"{namespace}:{abc_object.name}\"\n abc_objects.append(abc_object)\n # rename the abc container with namespace\n abc_container_name = f\"{namespace}:{name}\"\n abc_container.name = abc_container_name\n abc_objects.append(abc_container)\n\n return containerise(\n name, abc_objects, context,\n namespace, loader=self.__class__.__name__\n )\n\n def update(self, container, representation):\n from pymxs import runtime as rt\n\n path = get_representation_path(representation)\n node = rt.GetNodeByName(container[\"instance_node\"])\n node_list = [n for n in get_previous_loaded_object(node)\n if rt.ClassOf(n) == rt.AlembicContainer]\n with maintained_selection():\n rt.Select(node_list)\n\n for alembic in rt.Selection:\n abc = rt.GetNodeByName(alembic.name)\n rt.Select(abc.Children)\n for abc_con in abc.Children:\n abc_con.source = path\n rt.Select(abc_con.Children)\n for abc_obj in abc_con.Children:\n abc_obj.source = path\n lib.imprint(\n container[\"instance_node\"],\n {\"representation\": str(representation[\"_id\"])},\n )\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def remove(self, container):\n from pymxs import runtime as rt\n\n node = rt.GetNodeByName(container[\"instance_node\"])\n rt.Delete(node)\n\n @staticmethod\n def get_container_children(parent, type_name):\n from pymxs import runtime as rt\n\n def list_children(node):\n children = []\n for c in node.Children:\n children.append(c)\n children += list_children(c)\n return children\n\n filtered = []\n for child in list_children(parent):\n class_type = str(rt.ClassOf(child.baseObject))\n if class_type == type_name:\n filtered.append(child)\n\n return filtered\n"
},
{
"path": "openpype/hosts/max/plugins/load/load_model_fbx.py",
"content": "import os\nfrom openpype.pipeline import load, get_representation_path\nfrom openpype.hosts.max.api.pipeline import (\n containerise, get_previous_loaded_object,\n update_custom_attribute_data\n)\nfrom openpype.hosts.max.api import lib\nfrom openpype.hosts.max.api.lib import (\n unique_namespace,\n get_namespace,\n object_transform_set\n)\nfrom openpype.hosts.max.api.lib import maintained_selection\n\n\nclass FbxModelLoader(load.LoaderPlugin):\n \"\"\"Fbx Model Loader.\"\"\"\n\n families = [\"model\"]\n representations = [\"fbx\"]\n order = -9\n icon = \"code-fork\"\n color = \"white\"\n\n def load(self, context, name=None, namespace=None, data=None):\n from pymxs import runtime as rt\n filepath = self.filepath_from_context(context)\n filepath = os.path.normpath(filepath)\n rt.FBXImporterSetParam(\"Animation\", False)\n rt.FBXImporterSetParam(\"Cameras\", False)\n rt.FBXImporterSetParam(\"Mode\", rt.Name(\"create\"))\n rt.FBXImporterSetParam(\"Preserveinstances\", True)\n rt.importFile(\n filepath, rt.name(\"noPrompt\"), using=rt.FBXIMP)\n\n namespace = unique_namespace(\n name + \"_\",\n suffix=\"_\",\n )\n selections = rt.GetCurrentSelection()\n\n for selection in selections:\n selection.name = f\"{namespace}:{selection.name}\"\n\n return containerise(\n name, selections, context,\n namespace, loader=self.__class__.__name__)\n\n def update(self, container, representation):\n from pymxs import runtime as rt\n\n path = get_representation_path(representation)\n node_name = container[\"instance_node\"]\n node = rt.getNodeByName(node_name)\n if not node:\n rt.Container(name=node_name)\n namespace, _ = get_namespace(node_name)\n\n node_list = get_previous_loaded_object(node)\n rt.Select(node_list)\n prev_fbx_objects = rt.GetCurrentSelection()\n transform_data = object_transform_set(prev_fbx_objects)\n for prev_fbx_obj in prev_fbx_objects:\n if rt.isValidNode(prev_fbx_obj):\n rt.Delete(prev_fbx_obj)\n\n rt.FBXImporterSetParam(\"Animation\", False)\n rt.FBXImporterSetParam(\"Cameras\", False)\n rt.FBXImporterSetParam(\"Mode\", rt.Name(\"create\"))\n rt.FBXImporterSetParam(\"Preserveinstances\", True)\n rt.importFile(path, rt.name(\"noPrompt\"), using=rt.FBXIMP)\n current_fbx_objects = rt.GetCurrentSelection()\n fbx_objects = []\n for fbx_object in current_fbx_objects:\n fbx_object.name = f\"{namespace}:{fbx_object.name}\"\n fbx_objects.append(fbx_object)\n fbx_transform = f\"{fbx_object.name}.transform\"\n if fbx_transform in transform_data.keys():\n fbx_object.pos = transform_data[fbx_transform] or 0\n fbx_object.scale = transform_data[\n f\"{fbx_object.name}.scale\"] or 0\n\n with maintained_selection():\n rt.Select(node)\n update_custom_attribute_data(node, fbx_objects)\n lib.imprint(container[\"instance_node\"], {\n \"representation\": str(representation[\"_id\"])\n })\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def remove(self, container):\n from pymxs import runtime as rt\n\n node = rt.GetNodeByName(container[\"instance_node\"])\n rt.Delete(node)\n"
},
{
"path": "openpype/hosts/max/plugins/load/load_model_obj.py",
"content": "import os\n\nfrom openpype.hosts.max.api import lib\nfrom openpype.hosts.max.api.lib import (\n unique_namespace,\n get_namespace,\n maintained_selection,\n object_transform_set\n)\nfrom openpype.hosts.max.api.lib import maintained_selection\nfrom openpype.hosts.max.api.pipeline import (\n containerise,\n get_previous_loaded_object,\n update_custom_attribute_data\n)\nfrom openpype.pipeline import get_representation_path, load\n\n\nclass ObjLoader(load.LoaderPlugin):\n \"\"\"Obj Loader.\"\"\"\n\n families = [\"model\"]\n representations = [\"obj\"]\n order = -9\n icon = \"code-fork\"\n color = \"white\"\n\n def load(self, context, name=None, namespace=None, data=None):\n from pymxs import runtime as rt\n\n filepath = os.path.normpath(self.filepath_from_context(context))\n self.log.debug(\"Executing command to import..\")\n\n rt.Execute(f'importFile @\"{filepath}\" #noPrompt using:ObjImp')\n\n namespace = unique_namespace(\n name + \"_\",\n suffix=\"_\",\n )\n # create \"missing\" container for obj import\n selections = rt.GetCurrentSelection()\n # get current selection\n for selection in selections:\n selection.name = f\"{namespace}:{selection.name}\"\n return containerise(\n name, selections, context,\n namespace, loader=self.__class__.__name__)\n\n def update(self, container, representation):\n from pymxs import runtime as rt\n\n path = get_representation_path(representation)\n node_name = container[\"instance_node\"]\n node = rt.getNodeByName(node_name)\n namespace, _ = get_namespace(node_name)\n node_list = get_previous_loaded_object(node)\n rt.Select(node_list)\n previous_objects = rt.GetCurrentSelection()\n transform_data = object_transform_set(previous_objects)\n for prev_obj in previous_objects:\n if rt.isValidNode(prev_obj):\n rt.Delete(prev_obj)\n\n rt.Execute(f'importFile @\"{path}\" #noPrompt using:ObjImp')\n # get current selection\n selections = rt.GetCurrentSelection()\n for selection in selections:\n selection.name = f\"{namespace}:{selection.name}\"\n selection_transform = f\"{selection.name}.transform\"\n if selection_transform in transform_data.keys():\n selection.pos = transform_data[selection_transform] or 0\n selection.scale = transform_data[\n f\"{selection.name}.scale\"] or 0\n update_custom_attribute_data(node, selections)\n with maintained_selection():\n rt.Select(node)\n\n lib.imprint(node_name, {\n \"representation\": str(representation[\"_id\"])\n })\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def remove(self, container):\n from pymxs import runtime as rt\n\n node = rt.GetNodeByName(container[\"instance_node\"])\n rt.Delete(node)\n"
},
{
"path": "openpype/hosts/max/plugins/load/load_model_usd.py",
"content": "import os\n\nfrom pymxs import runtime as rt\nfrom openpype.pipeline.load import LoadError\nfrom openpype.hosts.max.api import lib\nfrom openpype.hosts.max.api.lib import (\n unique_namespace,\n get_namespace,\n object_transform_set,\n get_plugins\n)\nfrom openpype.hosts.max.api.lib import maintained_selection\nfrom openpype.hosts.max.api.pipeline import (\n containerise,\n get_previous_loaded_object,\n update_custom_attribute_data\n)\nfrom openpype.pipeline import get_representation_path, load\n\n\nclass ModelUSDLoader(load.LoaderPlugin):\n \"\"\"Loading model with the USD loader.\"\"\"\n\n families = [\"model\"]\n label = \"Load Model(USD)\"\n representations = [\"usda\"]\n order = -10\n icon = \"code-fork\"\n color = \"orange\"\n\n def load(self, context, name=None, namespace=None, data=None):\n # asset_filepath\n plugin_info = get_plugins()\n if \"usdimport.dli\" not in plugin_info:\n raise LoadError(\"No USDImporter loaded/installed in Max..\")\n filepath = os.path.normpath(self.filepath_from_context(context))\n import_options = rt.USDImporter.CreateOptions()\n base_filename = os.path.basename(filepath)\n _, ext = os.path.splitext(base_filename)\n log_filepath = filepath.replace(ext, \"txt\")\n\n rt.LogPath = log_filepath\n rt.LogLevel = rt.Name(\"info\")\n rt.USDImporter.importFile(filepath,\n importOptions=import_options)\n namespace = unique_namespace(\n name + \"_\",\n suffix=\"_\",\n )\n asset = rt.GetNodeByName(name)\n usd_objects = []\n\n for usd_asset in asset.Children:\n usd_asset.name = f\"{namespace}:{usd_asset.name}\"\n usd_objects.append(usd_asset)\n\n asset_name = f\"{namespace}:{name}\"\n asset.name = asset_name\n # need to get the correct container after renamed\n asset = rt.GetNodeByName(asset_name)\n usd_objects.append(asset)\n\n return containerise(\n name, usd_objects, context,\n namespace, loader=self.__class__.__name__)\n\n def update(self, container, representation):\n path = get_representation_path(representation)\n node_name = container[\"instance_node\"]\n node = rt.GetNodeByName(node_name)\n namespace, name = get_namespace(node_name)\n node_list = get_previous_loaded_object(node)\n rt.Select(node_list)\n prev_objects = [sel for sel in rt.GetCurrentSelection()\n if sel != rt.Container\n and sel.name != node_name]\n transform_data = object_transform_set(prev_objects)\n for n in prev_objects:\n rt.Delete(n)\n\n import_options = rt.USDImporter.CreateOptions()\n base_filename = os.path.basename(path)\n _, ext = os.path.splitext(base_filename)\n log_filepath = path.replace(ext, \"txt\")\n\n rt.LogPath = log_filepath\n rt.LogLevel = rt.Name(\"info\")\n rt.USDImporter.importFile(\n path, importOptions=import_options)\n\n asset = rt.GetNodeByName(name)\n usd_objects = []\n for children in asset.Children:\n children.name = f\"{namespace}:{children.name}\"\n usd_objects.append(children)\n children_transform = f\"{children.name}.transform\"\n if children_transform in transform_data.keys():\n children.pos = transform_data[children_transform] or 0\n children.scale = transform_data[\n f\"{children.name}.scale\"] or 0\n\n asset.name = f\"{namespace}:{asset.name}\"\n usd_objects.append(asset)\n update_custom_attribute_data(node, usd_objects)\n with maintained_selection():\n rt.Select(node)\n\n lib.imprint(node_name, {\n \"representation\": str(representation[\"_id\"])\n })\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def remove(self, container):\n node = rt.GetNodeByName(container[\"instance_node\"])\n rt.Delete(node)\n"
},
{
"path": "openpype/hosts/max/plugins/load/load_pointcache.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Simple alembic loader for 3dsmax.\n\nBecause of limited api, alembics can be only loaded, but not easily updated.\n\n\"\"\"\nimport os\nfrom openpype.pipeline import load, get_representation_path\nfrom openpype.hosts.max.api import lib, maintained_selection\nfrom openpype.hosts.max.api.lib import unique_namespace\nfrom openpype.hosts.max.api.pipeline import (\n containerise,\n get_previous_loaded_object\n)\n\n\nclass AbcLoader(load.LoaderPlugin):\n \"\"\"Alembic loader.\"\"\"\n\n families = [\"camera\", \"animation\", \"pointcache\"]\n label = \"Load Alembic\"\n representations = [\"abc\"]\n order = -10\n icon = \"code-fork\"\n color = \"orange\"\n\n def load(self, context, name=None, namespace=None, data=None):\n from pymxs import runtime as rt\n\n file_path = self.filepath_from_context(context)\n file_path = os.path.normpath(file_path)\n\n abc_before = {\n c\n for c in rt.rootNode.Children\n if rt.classOf(c) == rt.AlembicContainer\n }\n\n rt.AlembicImport.ImportToRoot = False\n rt.importFile(file_path, rt.name(\"noPrompt\"), using=rt.AlembicImport)\n\n abc_after = {\n c\n for c in rt.rootNode.Children\n if rt.classOf(c) == rt.AlembicContainer\n }\n\n # This should yield new AlembicContainer node\n abc_containers = abc_after.difference(abc_before)\n\n if len(abc_containers) != 1:\n self.log.error(\"Something failed when loading.\")\n\n abc_container = abc_containers.pop()\n selections = rt.GetCurrentSelection()\n for abc in selections:\n for cam_shape in abc.Children:\n cam_shape.playbackType = 0\n\n namespace = unique_namespace(\n name + \"_\",\n suffix=\"_\",\n )\n abc_objects = []\n for abc_object in abc_container.Children:\n abc_object.name = f\"{namespace}:{abc_object.name}\"\n abc_objects.append(abc_object)\n # rename the abc container with namespace\n abc_container_name = f\"{namespace}:{name}\"\n abc_container.name = abc_container_name\n abc_objects.append(abc_container)\n\n return containerise(\n name, abc_objects, context,\n namespace, loader=self.__class__.__name__\n )\n\n def update(self, container, representation):\n from pymxs import runtime as rt\n\n path = get_representation_path(representation)\n node = rt.GetNodeByName(container[\"instance_node\"])\n abc_container = [n for n in get_previous_loaded_object(node)\n if rt.ClassOf(n) == rt.AlembicContainer]\n with maintained_selection():\n rt.Select(abc_container)\n\n for alembic in rt.Selection:\n abc = rt.GetNodeByName(alembic.name)\n rt.Select(abc.Children)\n for abc_con in abc.Children:\n abc_con.source = path\n rt.Select(abc_con.Children)\n for abc_obj in abc_con.Children:\n abc_obj.source = path\n lib.imprint(\n container[\"instance_node\"],\n {\"representation\": str(representation[\"_id\"])},\n )\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def remove(self, container):\n from pymxs import runtime as rt\n\n node = rt.GetNodeByName(container[\"instance_node\"])\n rt.Delete(node)\n\n @staticmethod\n def get_container_children(parent, type_name):\n from pymxs import runtime as rt\n\n def list_children(node):\n children = []\n for c in node.Children:\n children.append(c)\n children += list_children(c)\n return children\n\n filtered = []\n for child in list_children(parent):\n class_type = str(rt.classOf(child.baseObject))\n if class_type == type_name:\n filtered.append(child)\n\n return filtered\n"
},
{
"path": "openpype/hosts/max/plugins/load/load_pointcache_ornatrix.py",
"content": "import os\nfrom openpype.pipeline import load, get_representation_path\nfrom openpype.pipeline.load import LoadError\nfrom openpype.hosts.max.api.pipeline import (\n containerise,\n get_previous_loaded_object,\n update_custom_attribute_data\n)\n\nfrom openpype.hosts.max.api.lib import (\n unique_namespace,\n get_namespace,\n object_transform_set,\n get_plugins\n)\nfrom openpype.hosts.max.api import lib\nfrom pymxs import runtime as rt\n\n\nclass OxAbcLoader(load.LoaderPlugin):\n \"\"\"Ornatrix Alembic loader.\"\"\"\n\n families = [\"camera\", \"animation\", \"pointcache\"]\n label = \"Load Alembic with Ornatrix\"\n representations = [\"abc\"]\n order = -10\n icon = \"code-fork\"\n color = \"orange\"\n postfix = \"param\"\n\n def load(self, context, name=None, namespace=None, data=None):\n plugin_list = get_plugins()\n if \"ephere.plugins.autodesk.max.ornatrix.dlo\" not in plugin_list:\n raise LoadError(\"Ornatrix plugin not \"\n \"found/installed in Max yet..\")\n\n file_path = os.path.normpath(self.filepath_from_context(context))\n rt.AlembicImport.ImportToRoot = True\n rt.AlembicImport.CustomAttributes = True\n rt.importFile(\n file_path, rt.name(\"noPrompt\"),\n using=rt.Ornatrix_Alembic_Importer)\n\n scene_object = []\n for obj in rt.rootNode.Children:\n obj_type = rt.ClassOf(obj)\n if str(obj_type).startswith(\"Ox_\"):\n scene_object.append(obj)\n\n namespace = unique_namespace(\n name + \"_\",\n suffix=\"_\",\n )\n abc_container = []\n for abc in scene_object:\n abc.name = f\"{namespace}:{abc.name}\"\n abc_container.append(abc)\n\n return containerise(\n name, abc_container, context,\n namespace, loader=self.__class__.__name__\n )\n\n def update(self, container, representation):\n path = get_representation_path(representation)\n node_name = container[\"instance_node\"]\n namespace, name = get_namespace(node_name)\n node = rt.getNodeByName(node_name)\n node_list = get_previous_loaded_object(node)\n rt.Select(node_list)\n selections = rt.getCurrentSelection()\n transform_data = object_transform_set(selections)\n for prev_obj in selections:\n if rt.isValidNode(prev_obj):\n rt.Delete(prev_obj)\n\n rt.AlembicImport.ImportToRoot = False\n rt.AlembicImport.CustomAttributes = True\n rt.importFile(\n path, rt.name(\"noPrompt\"),\n using=rt.Ornatrix_Alembic_Importer)\n\n scene_object = []\n for obj in rt.rootNode.Children:\n obj_type = rt.ClassOf(obj)\n if str(obj_type).startswith(\"Ox_\"):\n scene_object.append(obj)\n ox_abc_objects = []\n for abc in scene_object:\n abc.Parent = container\n abc.name = f\"{namespace}:{abc.name}\"\n ox_abc_objects.append(abc)\n ox_transform = f\"{abc.name}.transform\"\n if ox_transform in transform_data.keys():\n abc.pos = transform_data[ox_transform] or 0\n abc.scale = transform_data[f\"{abc.name}.scale\"] or 0\n update_custom_attribute_data(node, ox_abc_objects)\n lib.imprint(\n container[\"instance_node\"],\n {\"representation\": str(representation[\"_id\"])},\n )\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def remove(self, container):\n node = rt.GetNodeByName(container[\"instance_node\"])\n rt.Delete(node)\n"
},
{
"path": "openpype/hosts/max/plugins/load/load_pointcloud.py",
"content": "import os\n\nfrom openpype.hosts.max.api import lib, maintained_selection\nfrom openpype.hosts.max.api.lib import (\n unique_namespace,\n\n)\nfrom openpype.hosts.max.api.pipeline import (\n containerise,\n get_previous_loaded_object,\n update_custom_attribute_data\n)\nfrom openpype.pipeline import get_representation_path, load\n\n\nclass PointCloudLoader(load.LoaderPlugin):\n \"\"\"Point Cloud Loader.\"\"\"\n\n families = [\"pointcloud\"]\n representations = [\"prt\"]\n order = -8\n icon = \"code-fork\"\n color = \"green\"\n postfix = \"param\"\n\n def load(self, context, name=None, namespace=None, data=None):\n \"\"\"load point cloud by tyCache\"\"\"\n from pymxs import runtime as rt\n filepath = os.path.normpath(self.filepath_from_context(context))\n obj = rt.tyCache()\n obj.filename = filepath\n\n namespace = unique_namespace(\n name + \"_\",\n suffix=\"_\",\n )\n obj.name = f\"{namespace}:{obj.name}\"\n\n return containerise(\n name, [obj], context,\n namespace, loader=self.__class__.__name__)\n\n def update(self, container, representation):\n \"\"\"update the container\"\"\"\n from pymxs import runtime as rt\n\n path = get_representation_path(representation)\n node = rt.GetNodeByName(container[\"instance_node\"])\n node_list = get_previous_loaded_object(node)\n update_custom_attribute_data(\n node, node_list)\n with maintained_selection():\n rt.Select(node_list)\n for prt in rt.Selection:\n prt.filename = path\n lib.imprint(container[\"instance_node\"], {\n \"representation\": str(representation[\"_id\"])\n })\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def remove(self, container):\n \"\"\"remove the container\"\"\"\n from pymxs import runtime as rt\n\n node = rt.GetNodeByName(container[\"instance_node\"])\n rt.Delete(node)\n"
},
{
"path": "openpype/hosts/max/plugins/load/load_redshift_proxy.py",
"content": "import os\nimport clique\n\nfrom openpype.pipeline import (\n load,\n get_representation_path\n)\nfrom openpype.pipeline.load import LoadError\nfrom openpype.hosts.max.api.pipeline import (\n containerise,\n update_custom_attribute_data,\n get_previous_loaded_object\n)\nfrom openpype.hosts.max.api import lib\nfrom openpype.hosts.max.api.lib import (\n unique_namespace,\n get_plugins\n)\n\n\nclass RedshiftProxyLoader(load.LoaderPlugin):\n \"\"\"Load rs files with Redshift Proxy\"\"\"\n\n label = \"Load Redshift Proxy\"\n families = [\"redshiftproxy\"]\n representations = [\"rs\"]\n order = -9\n icon = \"code-fork\"\n color = \"white\"\n\n def load(self, context, name=None, namespace=None, data=None):\n from pymxs import runtime as rt\n plugin_info = get_plugins()\n if \"redshift4max.dlr\" not in plugin_info:\n raise LoadError(\"Redshift not loaded/installed in Max..\")\n filepath = self.filepath_from_context(context)\n rs_proxy = rt.RedshiftProxy()\n rs_proxy.file = filepath\n files_in_folder = os.listdir(os.path.dirname(filepath))\n collections, remainder = clique.assemble(files_in_folder)\n if collections:\n rs_proxy.is_sequence = True\n\n namespace = unique_namespace(\n name + \"_\",\n suffix=\"_\",\n )\n rs_proxy.name = f\"{namespace}:{rs_proxy.name}\"\n\n return containerise(\n name, [rs_proxy], context,\n namespace, loader=self.__class__.__name__)\n\n def update(self, container, representation):\n from pymxs import runtime as rt\n\n path = get_representation_path(representation)\n node = rt.getNodeByName(container[\"instance_node\"])\n node_list = get_previous_loaded_object(node)\n rt.Select(node_list)\n update_custom_attribute_data(\n node, rt.Selection)\n for proxy in rt.Selection:\n proxy.file = path\n\n lib.imprint(container[\"instance_node\"], {\n \"representation\": str(representation[\"_id\"])\n })\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def remove(self, container):\n from pymxs import runtime as rt\n\n node = rt.getNodeByName(container[\"instance_node\"])\n rt.delete(node)\n"
},
{
"path": "openpype/hosts/max/plugins/load/load_tycache.py",
"content": "import os\nfrom openpype.hosts.max.api import lib, maintained_selection\nfrom openpype.hosts.max.api.lib import (\n unique_namespace,\n\n)\nfrom openpype.hosts.max.api.pipeline import (\n containerise,\n get_previous_loaded_object,\n update_custom_attribute_data\n)\nfrom openpype.pipeline import get_representation_path, load\n\n\nclass TyCacheLoader(load.LoaderPlugin):\n \"\"\"TyCache Loader.\"\"\"\n\n families = [\"tycache\"]\n representations = [\"tyc\"]\n order = -8\n icon = \"code-fork\"\n color = \"green\"\n\n def load(self, context, name=None, namespace=None, data=None):\n \"\"\"Load tyCache\"\"\"\n from pymxs import runtime as rt\n filepath = os.path.normpath(self.filepath_from_context(context))\n obj = rt.tyCache()\n obj.filename = filepath\n\n namespace = unique_namespace(\n name + \"_\",\n suffix=\"_\",\n )\n obj.name = f\"{namespace}:{obj.name}\"\n\n return containerise(\n name, [obj], context,\n namespace, loader=self.__class__.__name__)\n\n def update(self, container, representation):\n \"\"\"update the container\"\"\"\n from pymxs import runtime as rt\n\n path = get_representation_path(representation)\n node = rt.GetNodeByName(container[\"instance_node\"])\n node_list = get_previous_loaded_object(node)\n update_custom_attribute_data(node, node_list)\n with maintained_selection():\n for tyc in node_list:\n tyc.filename = path\n lib.imprint(container[\"instance_node\"], {\n \"representation\": str(representation[\"_id\"])\n })\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def remove(self, container):\n \"\"\"remove the container\"\"\"\n from pymxs import runtime as rt\n\n node = rt.GetNodeByName(container[\"instance_node\"])\n rt.Delete(node)\n"
},
{
"path": "openpype/hosts/max/plugins/publish/collect_frame_range.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\nfrom pymxs import runtime as rt\n\n\nclass CollectFrameRange(pyblish.api.InstancePlugin):\n \"\"\"Collect Frame Range.\"\"\"\n\n order = pyblish.api.CollectorOrder + 0.01\n label = \"Collect Frame Range\"\n hosts = ['max']\n families = [\"camera\", \"maxrender\",\n \"pointcache\", \"pointcloud\",\n \"review\", \"redshiftproxy\"]\n\n def process(self, instance):\n if instance.data[\"family\"] == \"maxrender\":\n instance.data[\"frameStartHandle\"] = int(rt.rendStart)\n instance.data[\"frameEndHandle\"] = int(rt.rendEnd)\n else:\n instance.data[\"frameStartHandle\"] = int(rt.animationRange.start)\n instance.data[\"frameEndHandle\"] = int(rt.animationRange.end)\n"
},
{
"path": "openpype/hosts/max/plugins/publish/collect_members.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Collect instance members.\"\"\"\nimport pyblish.api\nfrom pymxs import runtime as rt\n\n\nclass CollectMembers(pyblish.api.InstancePlugin):\n \"\"\"Collect Set Members.\"\"\"\n\n order = pyblish.api.CollectorOrder + 0.01\n label = \"Collect Instance Members\"\n hosts = ['max']\n\n def process(self, instance):\n if instance.data[\"family\"] == \"workfile\":\n self.log.debug(\"Skipping Actions for workfile family.\")\n self.log.debug(\"{}\".format(instance.data[\"subset\"]))\n return\n if instance.data.get(\"instance_node\"):\n container = rt.GetNodeByName(instance.data[\"instance_node\"])\n instance.data[\"members\"] = [\n member.node for member\n in container.modifiers[0].openPypeData.all_handles\n ]\n self.log.debug(\"{}\".format(instance.data[\"members\"]))\n"
},
{
"path": "openpype/hosts/max/plugins/publish/collect_render.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Collect Render\"\"\"\nimport os\nimport pyblish.api\n\nfrom pymxs import runtime as rt\nfrom openpype.pipeline.publish import KnownPublishError\nfrom openpype.hosts.max.api import colorspace\nfrom openpype.hosts.max.api.lib import get_max_version, get_current_renderer\nfrom openpype.hosts.max.api.lib_rendersettings import RenderSettings\nfrom openpype.hosts.max.api.lib_renderproducts import RenderProducts\n\n\nclass CollectRender(pyblish.api.InstancePlugin):\n \"\"\"Collect Render for Deadline\"\"\"\n\n order = pyblish.api.CollectorOrder + 0.02\n label = \"Collect 3dsmax Render Layers\"\n hosts = ['max']\n families = [\"maxrender\"]\n\n def process(self, instance):\n context = instance.context\n folder = rt.maxFilePath\n file = rt.maxFileName\n current_file = os.path.join(folder, file)\n filepath = current_file.replace(\"\\\\\", \"/\")\n context.data['currentFile'] = current_file\n\n files_by_aov = RenderProducts().get_beauty(instance.name)\n aovs = RenderProducts().get_aovs(instance.name)\n files_by_aov.update(aovs)\n\n camera = rt.viewport.GetCamera()\n if instance.data.get(\"members\"):\n camera_list = [member for member in instance.data[\"members\"]\n if rt.ClassOf(member) == rt.Camera.Classes]\n if camera_list:\n camera = camera_list[-1]\n\n instance.data[\"cameras\"] = [camera.name] if camera else None # noqa\n\n if instance.data.get(\"multiCamera\"):\n cameras = instance.data.get(\"members\")\n if not cameras:\n raise KnownPublishError(\"There should be at least\"\n \" one renderable camera in container\")\n sel_cam = [\n c.name for c in cameras\n if rt.classOf(c) in rt.Camera.classes]\n container_name = instance.data.get(\"instance_node\")\n render_dir = os.path.dirname(rt.rendOutputFilename)\n outputs = RenderSettings().batch_render_layer(\n container_name, render_dir, sel_cam\n )\n\n instance.data[\"cameras\"] = sel_cam\n\n files_by_aov = RenderProducts().get_multiple_beauty(\n outputs, sel_cam)\n aovs = RenderProducts().get_multiple_aovs(\n outputs, sel_cam)\n files_by_aov.update(aovs)\n\n if \"expectedFiles\" not in instance.data:\n instance.data[\"expectedFiles\"] = list()\n instance.data[\"files\"] = list()\n instance.data[\"expectedFiles\"].append(files_by_aov)\n instance.data[\"files\"].append(files_by_aov)\n\n img_format = RenderProducts().image_format()\n # OCIO config not support in\n # most of the 3dsmax renderers\n # so this is currently hard coded\n # TODO: add options for redshift/vray ocio config\n instance.data[\"colorspaceConfig\"] = \"\"\n instance.data[\"colorspaceDisplay\"] = \"sRGB\"\n instance.data[\"colorspaceView\"] = \"ACES 1.0 SDR-video\"\n\n if int(get_max_version()) >= 2024:\n colorspace_mgr = rt.ColorPipelineMgr # noqa\n display = next(\n (display for display in colorspace_mgr.GetDisplayList()))\n view_transform = next(\n (view for view in colorspace_mgr.GetViewList(display)))\n instance.data[\"colorspaceConfig\"] = colorspace_mgr.OCIOConfigPath\n instance.data[\"colorspaceDisplay\"] = display\n instance.data[\"colorspaceView\"] = view_transform\n\n instance.data[\"renderProducts\"] = colorspace.ARenderProduct()\n instance.data[\"publishJobState\"] = \"Suspended\"\n instance.data[\"attachTo\"] = []\n renderer_class = get_current_renderer()\n renderer = str(renderer_class).split(\":\")[0]\n # also need to get the render dir for conversion\n data = {\n \"asset\": instance.data[\"asset\"],\n \"subset\": str(instance.name),\n \"publish\": True,\n \"maxversion\": str(get_max_version()),\n \"imageFormat\": img_format,\n \"family\": 'maxrender',\n \"families\": ['maxrender'],\n \"renderer\": renderer,\n \"source\": filepath,\n \"plugin\": \"3dsmax\",\n \"frameStart\": instance.data[\"frameStartHandle\"],\n \"frameEnd\": instance.data[\"frameEndHandle\"],\n \"farm\": True\n }\n instance.data.update(data)\n\n # TODO: this should be unified with maya and its \"multipart\" flag\n # on instance.\n if renderer == \"Redshift_Renderer\":\n instance.data.update(\n {\"separateAovFiles\": rt.Execute(\n \"renderers.current.separateAovFiles\")})\n\n self.log.info(\"data: {0}\".format(data))\n"
},
{
"path": "openpype/hosts/max/plugins/publish/collect_review.py",
"content": "# dont forget getting the focal length for burnin\n\"\"\"Collect Review\"\"\"\nimport pyblish.api\n\nfrom pymxs import runtime as rt\nfrom openpype.lib import BoolDef\nfrom openpype.hosts.max.api.lib import get_max_version\nfrom openpype.pipeline.publish import (\n OpenPypePyblishPluginMixin,\n KnownPublishError\n)\n\n\nclass CollectReview(pyblish.api.InstancePlugin,\n OpenPypePyblishPluginMixin):\n \"\"\"Collect Review Data for Preview Animation\"\"\"\n\n order = pyblish.api.CollectorOrder + 0.02\n label = \"Collect Review Data\"\n hosts = ['max']\n families = [\"review\"]\n\n def process(self, instance):\n nodes = instance.data[\"members\"]\n\n def is_camera(node):\n is_camera_class = rt.classOf(node) in rt.Camera.classes\n return is_camera_class and rt.isProperty(node, \"fov\")\n\n # Use first camera in instance\n cameras = [node for node in nodes if is_camera(node)]\n if cameras:\n if len(cameras) > 1:\n self.log.warning(\n \"Found more than one camera in instance, using first \"\n f\"one found: {cameras[0]}\"\n )\n camera = cameras[0]\n camera_name = camera.name\n focal_length = camera.fov\n else:\n raise KnownPublishError(\n \"Unable to find a valid camera in 'Review' container.\"\n \" Only native max Camera supported. \"\n f\"Found objects: {nodes}\"\n )\n creator_attrs = instance.data[\"creator_attributes\"]\n attr_values = self.get_attr_values_from_data(instance.data)\n\n general_preview_data = {\n \"review_camera\": camera_name,\n \"frameStart\": instance.data[\"frameStartHandle\"],\n \"frameEnd\": instance.data[\"frameEndHandle\"],\n \"percentSize\": creator_attrs[\"percentSize\"],\n \"imageFormat\": creator_attrs[\"imageFormat\"],\n \"keepImages\": creator_attrs[\"keepImages\"],\n \"fps\": instance.context.data[\"fps\"],\n \"review_width\": creator_attrs[\"review_width\"],\n \"review_height\": creator_attrs[\"review_height\"],\n }\n\n if int(get_max_version()) >= 2024:\n colorspace_mgr = rt.ColorPipelineMgr # noqa\n display = next(\n (display for display in colorspace_mgr.GetDisplayList()))\n view_transform = next(\n (view for view in colorspace_mgr.GetViewList(display)))\n instance.data[\"colorspaceConfig\"] = colorspace_mgr.OCIOConfigPath\n instance.data[\"colorspaceDisplay\"] = display\n instance.data[\"colorspaceView\"] = view_transform\n\n preview_data = {\n \"vpStyle\": creator_attrs[\"visualStyleMode\"],\n \"vpPreset\": creator_attrs[\"viewportPreset\"],\n \"vpTextures\": creator_attrs[\"vpTexture\"],\n \"dspGeometry\": attr_values.get(\"dspGeometry\"),\n \"dspShapes\": attr_values.get(\"dspShapes\"),\n \"dspLights\": attr_values.get(\"dspLights\"),\n \"dspCameras\": attr_values.get(\"dspCameras\"),\n \"dspHelpers\": attr_values.get(\"dspHelpers\"),\n \"dspParticles\": attr_values.get(\"dspParticles\"),\n \"dspBones\": attr_values.get(\"dspBones\"),\n \"dspBkg\": attr_values.get(\"dspBkg\"),\n \"dspGrid\": attr_values.get(\"dspGrid\"),\n \"dspSafeFrame\": attr_values.get(\"dspSafeFrame\"),\n \"dspFrameNums\": attr_values.get(\"dspFrameNums\")\n }\n else:\n general_viewport = {\n \"dspBkg\": attr_values.get(\"dspBkg\"),\n \"dspGrid\": attr_values.get(\"dspGrid\")\n }\n nitrous_manager = {\n \"AntialiasingQuality\": creator_attrs[\"antialiasingQuality\"],\n }\n nitrous_viewport = {\n \"VisualStyleMode\": creator_attrs[\"visualStyleMode\"],\n \"ViewportPreset\": creator_attrs[\"viewportPreset\"],\n \"UseTextureEnabled\": creator_attrs[\"vpTexture\"]\n }\n preview_data = {\n \"general_viewport\": general_viewport,\n \"nitrous_manager\": nitrous_manager,\n \"nitrous_viewport\": nitrous_viewport,\n \"vp_btn_mgr\": {\"EnableButtons\": False}\n }\n\n # Enable ftrack functionality\n instance.data.setdefault(\"families\", []).append('ftrack')\n\n burnin_members = instance.data.setdefault(\"burninDataMembers\", {})\n burnin_members[\"focalLength\"] = focal_length\n\n instance.data.update(general_preview_data)\n instance.data[\"viewport_options\"] = preview_data\n\n @classmethod\n def get_attribute_defs(cls):\n return [\n BoolDef(\"dspGeometry\",\n label=\"Geometry\",\n default=True),\n BoolDef(\"dspShapes\",\n label=\"Shapes\",\n default=False),\n BoolDef(\"dspLights\",\n label=\"Lights\",\n default=False),\n BoolDef(\"dspCameras\",\n label=\"Cameras\",\n default=False),\n BoolDef(\"dspHelpers\",\n label=\"Helpers\",\n default=False),\n BoolDef(\"dspParticles\",\n label=\"Particle Systems\",\n default=True),\n BoolDef(\"dspBones\",\n label=\"Bone Objects\",\n default=False),\n BoolDef(\"dspBkg\",\n label=\"Background\",\n default=True),\n BoolDef(\"dspGrid\",\n label=\"Active Grid\",\n default=False),\n BoolDef(\"dspSafeFrame\",\n label=\"Safe Frames\",\n default=False),\n BoolDef(\"dspFrameNums\",\n label=\"Frame Numbers\",\n default=False)\n ]\n"
},
{
"path": "openpype/hosts/max/plugins/publish/collect_tycache_attributes.py",
"content": "import pyblish.api\n\nfrom openpype.lib import EnumDef, TextDef\nfrom openpype.pipeline.publish import OpenPypePyblishPluginMixin\n\n\nclass CollectTyCacheData(pyblish.api.InstancePlugin,\n OpenPypePyblishPluginMixin):\n \"\"\"Collect Channel Attributes for TyCache Export\"\"\"\n\n order = pyblish.api.CollectorOrder + 0.02\n label = \"Collect tyCache attribute Data\"\n hosts = ['max']\n families = [\"tycache\"]\n\n def process(self, instance):\n attr_values = self.get_attr_values_from_data(instance.data)\n attributes = {}\n for attr_key in attr_values.get(\"tycacheAttributes\", []):\n attributes[attr_key] = True\n\n for key in [\"tycacheLayer\", \"tycacheObjectName\"]:\n attributes[key] = attr_values.get(key, \"\")\n\n # Collect the selected channel data before exporting\n instance.data[\"tyc_attrs\"] = attributes\n self.log.debug(\n f\"Found tycache attributes: {attributes}\"\n )\n\n @classmethod\n def get_attribute_defs(cls):\n # TODO: Support the attributes with maxObject array\n tyc_attr_enum = [\"tycacheChanAge\", \"tycacheChanGroups\",\n \"tycacheChanPos\", \"tycacheChanRot\",\n \"tycacheChanScale\", \"tycacheChanVel\",\n \"tycacheChanSpin\", \"tycacheChanShape\",\n \"tycacheChanMatID\", \"tycacheChanMapping\",\n \"tycacheChanMaterials\", \"tycacheChanCustomFloat\"\n \"tycacheChanCustomVector\", \"tycacheChanCustomTM\",\n \"tycacheChanPhysX\", \"tycacheMeshBackup\",\n \"tycacheCreateObject\",\n \"tycacheCreateObjectIfNotCreated\",\n \"tycacheAdditionalCloth\",\n \"tycacheAdditionalSkin\",\n \"tycacheAdditionalSkinID\",\n \"tycacheAdditionalSkinIDValue\",\n \"tycacheAdditionalTerrain\",\n \"tycacheAdditionalVDB\",\n \"tycacheAdditionalSplinePaths\",\n \"tycacheAdditionalGeo\",\n \"tycacheAdditionalGeoActivateModifiers\",\n \"tycacheSplines\",\n \"tycacheSplinesAdditionalSplines\"\n ]\n tyc_default_attrs = [\"tycacheChanGroups\", \"tycacheChanPos\",\n \"tycacheChanRot\", \"tycacheChanScale\",\n \"tycacheChanVel\", \"tycacheChanShape\",\n \"tycacheChanMatID\", \"tycacheChanMapping\",\n \"tycacheChanMaterials\",\n \"tycacheCreateObjectIfNotCreated\"]\n return [\n EnumDef(\"tycacheAttributes\",\n tyc_attr_enum,\n default=tyc_default_attrs,\n multiselection=True,\n label=\"TyCache Attributes\"),\n TextDef(\"tycacheLayer\",\n label=\"TyCache Layer\",\n tooltip=\"Name of tycache layer\",\n default=\"$(tyFlowLayer)\"),\n TextDef(\"tycacheObjectName\",\n label=\"TyCache Object Name\",\n tooltip=\"TyCache Object Name\",\n default=\"$(tyFlowName)_tyCache\")\n ]\n"
},
{
"path": "openpype/hosts/max/plugins/publish/collect_workfile.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Collect current work file.\"\"\"\nimport os\nimport pyblish.api\n\nfrom pymxs import runtime as rt\n\n\nclass CollectWorkfile(pyblish.api.InstancePlugin):\n \"\"\"Inject the current working file into context\"\"\"\n\n order = pyblish.api.CollectorOrder - 0.01\n label = \"Collect 3dsmax Workfile\"\n hosts = ['max']\n\n def process(self, instance):\n \"\"\"Inject the current working file.\"\"\"\n context = instance.context\n folder = rt.maxFilePath\n file = rt.maxFileName\n if not folder or not file:\n self.log.error(\"Scene is not saved.\")\n current_file = os.path.join(folder, file)\n\n context.data['currentFile'] = current_file\n\n ext = os.path.splitext(file)[-1].lstrip(\".\")\n\n data = {}\n\n # create instance\n subset = instance.data[\"subset\"]\n\n data.update({\n \"subset\": subset,\n \"asset\": context.data[\"asset\"],\n \"label\": subset,\n \"publish\": True,\n \"family\": 'workfile',\n \"families\": ['workfile'],\n \"setMembers\": [current_file],\n \"frameStart\": context.data['frameStart'],\n \"frameEnd\": context.data['frameEnd'],\n \"handleStart\": context.data['handleStart'],\n \"handleEnd\": context.data['handleEnd']\n })\n\n data['representations'] = [{\n 'name': ext.lstrip(\".\"),\n 'ext': ext.lstrip(\".\"),\n 'files': file,\n \"stagingDir\": folder,\n }]\n\n instance.data.update(data)\n self.log.info('Collected data: {}'.format(data))\n self.log.info('Collected instance: {}'.format(file))\n self.log.info('Scene path: {}'.format(current_file))\n self.log.info('staging Dir: {}'.format(folder))\n self.log.info('subset: {}'.format(subset))\n"
},
{
"path": "openpype/hosts/max/plugins/publish/extract_alembic.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"\nExport alembic file.\n\nNote:\n Parameters on AlembicExport (AlembicExport.Parameter):\n\n ParticleAsMesh (bool): Sets whether particle shapes are exported\n as meshes.\n AnimTimeRange (enum): How animation is saved:\n #CurrentFrame: saves current frame\n #TimeSlider: saves the active time segments on time slider (default)\n #StartEnd: saves a range specified by the Step\n StartFrame (int)\n EnFrame (int)\n ShapeSuffix (bool): When set to true, appends the string \"Shape\" to the\n name of each exported mesh. This property is set to false by default.\n SamplesPerFrame (int): Sets the number of animation samples per frame.\n Hidden (bool): When true, export hidden geometry.\n UVs (bool): When true, export the mesh UV map channel.\n Normals (bool): When true, export the mesh normals.\n VertexColors (bool): When true, export the mesh vertex color map 0 and the\n current vertex color display data when it differs\n ExtraChannels (bool): When true, export the mesh extra map channels\n (map channels greater than channel 1)\n Velocity (bool): When true, export the meh vertex and particle velocity\n data.\n MaterialIDs (bool): When true, export the mesh material ID as\n Alembic face sets.\n Visibility (bool): When true, export the node visibility data.\n LayerName (bool): When true, export the node layer name as an Alembic\n object property.\n MaterialName (bool): When true, export the geometry node material name as\n an Alembic object property\n ObjectID (bool): When true, export the geometry node g-buffer object ID as\n an Alembic object property.\n CustomAttributes (bool): When true, export the node and its modifiers\n custom attributes into an Alembic object compound property.\n\"\"\"\nimport os\nimport pyblish.api\nfrom openpype.pipeline import publish, OptionalPyblishPluginMixin\nfrom pymxs import runtime as rt\nfrom openpype.hosts.max.api import maintained_selection\nfrom openpype.hosts.max.api.lib import suspended_refresh\nfrom openpype.lib import BoolDef\n\n\nclass ExtractAlembic(publish.Extractor,\n OptionalPyblishPluginMixin):\n order = pyblish.api.ExtractorOrder\n label = \"Extract Pointcache\"\n hosts = [\"max\"]\n families = [\"pointcache\"]\n optional = True\n\n def process(self, instance):\n if not self.is_active(instance.data):\n return\n\n parent_dir = self.staging_dir(instance)\n file_name = \"{name}.abc\".format(**instance.data)\n path = os.path.join(parent_dir, file_name)\n\n with suspended_refresh():\n self._set_abc_attributes(instance)\n with maintained_selection():\n # select and export\n node_list = instance.data[\"members\"]\n rt.Select(node_list)\n rt.exportFile(\n path,\n rt.name(\"noPrompt\"),\n selectedOnly=True,\n using=rt.AlembicExport,\n )\n\n if \"representations\" not in instance.data:\n instance.data[\"representations\"] = []\n\n representation = {\n \"name\": \"abc\",\n \"ext\": \"abc\",\n \"files\": file_name,\n \"stagingDir\": parent_dir,\n }\n instance.data[\"representations\"].append(representation)\n\n def _set_abc_attributes(self, instance):\n start = instance.data[\"frameStartHandle\"]\n end = instance.data[\"frameEndHandle\"]\n attr_values = self.get_attr_values_from_data(instance.data)\n custom_attrs = attr_values.get(\"custom_attrs\", False)\n if not custom_attrs:\n self.log.debug(\n \"No Custom Attributes included in this abc export...\")\n rt.AlembicExport.ArchiveType = rt.Name(\"ogawa\")\n rt.AlembicExport.CoordinateSystem = rt.Name(\"maya\")\n rt.AlembicExport.StartFrame = start\n rt.AlembicExport.EndFrame = end\n rt.AlembicExport.CustomAttributes = custom_attrs\n\n @classmethod\n def get_attribute_defs(cls):\n return [\n BoolDef(\"custom_attrs\",\n label=\"Custom Attributes\",\n default=False),\n ]\n\n\nclass ExtractCameraAlembic(ExtractAlembic):\n \"\"\"Extract Camera with AlembicExport.\"\"\"\n\n label = \"Extract Alembic Camera\"\n families = [\"camera\"]\n\n\nclass ExtractModel(ExtractAlembic):\n \"\"\"Extract Geometry in Alembic Format\"\"\"\n label = \"Extract Geometry (Alembic)\"\n families = [\"model\"]\n\n def _set_abc_attributes(self, instance):\n attr_values = self.get_attr_values_from_data(instance.data)\n custom_attrs = attr_values.get(\"custom_attrs\", False)\n if not custom_attrs:\n self.log.debug(\n \"No Custom Attributes included in this abc export...\")\n rt.AlembicExport.ArchiveType = rt.name(\"ogawa\")\n rt.AlembicExport.CoordinateSystem = rt.name(\"maya\")\n rt.AlembicExport.CustomAttributes = custom_attrs\n rt.AlembicExport.UVs = True\n rt.AlembicExport.VertexColors = True\n rt.AlembicExport.PreserveInstances = True\n"
},
{
"path": "openpype/hosts/max/plugins/publish/extract_fbx.py",
"content": "import os\nimport pyblish.api\nfrom openpype.pipeline import publish, OptionalPyblishPluginMixin\nfrom pymxs import runtime as rt\nfrom openpype.hosts.max.api import maintained_selection\nfrom openpype.hosts.max.api.lib import convert_unit_scale\n\n\nclass ExtractModelFbx(publish.Extractor, OptionalPyblishPluginMixin):\n \"\"\"\n Extract Geometry in FBX Format\n \"\"\"\n\n order = pyblish.api.ExtractorOrder - 0.05\n label = \"Extract FBX\"\n hosts = [\"max\"]\n families = [\"model\"]\n optional = True\n\n def process(self, instance):\n if not self.is_active(instance.data):\n return\n\n stagingdir = self.staging_dir(instance)\n filename = \"{name}.fbx\".format(**instance.data)\n filepath = os.path.join(stagingdir, filename)\n self._set_fbx_attributes()\n\n with maintained_selection():\n # select and export\n node_list = instance.data[\"members\"]\n rt.Select(node_list)\n rt.exportFile(\n filepath,\n rt.name(\"noPrompt\"),\n selectedOnly=True,\n using=rt.FBXEXP,\n )\n\n if \"representations\" not in instance.data:\n instance.data[\"representations\"] = []\n\n representation = {\n \"name\": \"fbx\",\n \"ext\": \"fbx\",\n \"files\": filename,\n \"stagingDir\": stagingdir,\n }\n instance.data[\"representations\"].append(representation)\n self.log.info(\n \"Extracted instance '%s' to: %s\" % (instance.name, filepath)\n )\n\n def _set_fbx_attributes(self):\n unit_scale = convert_unit_scale()\n rt.FBXExporterSetParam(\"Animation\", False)\n rt.FBXExporterSetParam(\"Cameras\", False)\n rt.FBXExporterSetParam(\"Lights\", False)\n rt.FBXExporterSetParam(\"PointCache\", False)\n rt.FBXExporterSetParam(\"AxisConversionMethod\", \"Animation\")\n rt.FBXExporterSetParam(\"UpAxis\", \"Y\")\n rt.FBXExporterSetParam(\"Preserveinstances\", True)\n if unit_scale:\n rt.FBXExporterSetParam(\"ConvertUnit\", unit_scale)\n\n\nclass ExtractCameraFbx(ExtractModelFbx):\n \"\"\"Extract Camera with FbxExporter.\"\"\"\n\n order = pyblish.api.ExtractorOrder - 0.2\n label = \"Extract Fbx Camera\"\n families = [\"camera\"]\n optional = True\n\n def _set_fbx_attributes(self):\n unit_scale = convert_unit_scale()\n rt.FBXExporterSetParam(\"Animation\", True)\n rt.FBXExporterSetParam(\"Cameras\", True)\n rt.FBXExporterSetParam(\"AxisConversionMethod\", \"Animation\")\n rt.FBXExporterSetParam(\"UpAxis\", \"Y\")\n rt.FBXExporterSetParam(\"Preserveinstances\", True)\n if unit_scale:\n rt.FBXExporterSetParam(\"ConvertUnit\", unit_scale)\n"
},
{
"path": "openpype/hosts/max/plugins/publish/extract_max_scene_raw.py",
"content": "import os\nimport pyblish.api\nfrom openpype.pipeline import publish, OptionalPyblishPluginMixin\nfrom pymxs import runtime as rt\n\n\nclass ExtractMaxSceneRaw(publish.Extractor, OptionalPyblishPluginMixin):\n \"\"\"\n Extract Raw Max Scene with SaveSelected\n \"\"\"\n\n order = pyblish.api.ExtractorOrder - 0.2\n label = \"Extract Max Scene (Raw)\"\n hosts = [\"max\"]\n families = [\"camera\", \"maxScene\", \"model\"]\n optional = True\n\n def process(self, instance):\n if not self.is_active(instance.data):\n return\n\n # publish the raw scene for camera\n self.log.debug(\"Extracting Raw Max Scene ...\")\n\n stagingdir = self.staging_dir(instance)\n filename = \"{name}.max\".format(**instance.data)\n\n max_path = os.path.join(stagingdir, filename)\n\n if \"representations\" not in instance.data:\n instance.data[\"representations\"] = []\n\n nodes = instance.data[\"members\"]\n rt.saveNodes(nodes, max_path, quiet=True)\n\n self.log.info(\"Performing Extraction ...\")\n\n representation = {\n \"name\": \"max\",\n \"ext\": \"max\",\n \"files\": filename,\n \"stagingDir\": stagingdir,\n }\n instance.data[\"representations\"].append(representation)\n self.log.info(\n \"Extracted instance '%s' to: %s\" % (instance.name, max_path)\n )\n"
},
{
"path": "openpype/hosts/max/plugins/publish/extract_model_obj.py",
"content": "import os\nimport pyblish.api\nfrom openpype.pipeline import publish, OptionalPyblishPluginMixin\nfrom pymxs import runtime as rt\nfrom openpype.hosts.max.api import maintained_selection\nfrom openpype.hosts.max.api.lib import suspended_refresh\nfrom openpype.pipeline.publish import KnownPublishError\n\n\nclass ExtractModelObj(publish.Extractor, OptionalPyblishPluginMixin):\n \"\"\"\n Extract Geometry in OBJ Format\n \"\"\"\n\n order = pyblish.api.ExtractorOrder - 0.05\n label = \"Extract OBJ\"\n hosts = [\"max\"]\n families = [\"model\"]\n optional = True\n\n def process(self, instance):\n if not self.is_active(instance.data):\n return\n\n stagingdir = self.staging_dir(instance)\n filename = \"{name}.obj\".format(**instance.data)\n filepath = os.path.join(stagingdir, filename)\n\n with suspended_refresh():\n with maintained_selection():\n # select and export\n node_list = instance.data[\"members\"]\n rt.Select(node_list)\n rt.exportFile(\n filepath,\n rt.name(\"noPrompt\"),\n selectedOnly=True,\n using=rt.ObjExp,\n )\n if not os.path.exists(filepath):\n raise KnownPublishError(\n \"File {} wasn't produced by 3ds max, please check the logs.\")\n\n if \"representations\" not in instance.data:\n instance.data[\"representations\"] = []\n\n representation = {\n \"name\": \"obj\",\n \"ext\": \"obj\",\n \"files\": filename,\n \"stagingDir\": stagingdir,\n }\n\n instance.data[\"representations\"].append(representation)\n self.log.info(\n \"Extracted instance '%s' to: %s\" % (instance.name, filepath)\n )\n"
},
{
"path": "openpype/hosts/max/plugins/publish/extract_model_usd.py",
"content": "import os\n\nimport pyblish.api\nfrom pymxs import runtime as rt\n\nfrom openpype.hosts.max.api import maintained_selection\nfrom openpype.pipeline import OptionalPyblishPluginMixin, publish\n\n\nclass ExtractModelUSD(publish.Extractor,\n OptionalPyblishPluginMixin):\n \"\"\"Extract Geometry in USDA Format.\"\"\"\n\n order = pyblish.api.ExtractorOrder - 0.05\n label = \"Extract Geometry (USD)\"\n hosts = [\"max\"]\n families = [\"model\"]\n optional = True\n\n def process(self, instance):\n if not self.is_active(instance.data):\n return\n\n self.log.info(\"Extracting Geometry ...\")\n\n stagingdir = self.staging_dir(instance)\n asset_filename = \"{name}.usda\".format(**instance.data)\n asset_filepath = os.path.join(stagingdir,\n asset_filename)\n self.log.info(f\"Writing USD '{asset_filepath}' to '{stagingdir}'\")\n\n log_filename = \"{name}.txt\".format(**instance.data)\n log_filepath = os.path.join(stagingdir,\n log_filename)\n self.log.info(f\"Writing log '{log_filepath}' to '{stagingdir}'\")\n\n # get the nodes which need to be exported\n export_options = self.get_export_options(log_filepath)\n with maintained_selection():\n # select and export\n node_list = instance.data[\"members\"]\n rt.Select(node_list)\n rt.USDExporter.ExportFile(asset_filepath,\n exportOptions=export_options,\n contentSource=rt.Name(\"selected\"),\n nodeList=node_list)\n\n self.log.info(\"Performing Extraction ...\")\n if \"representations\" not in instance.data:\n instance.data[\"representations\"] = []\n\n representation = {\n 'name': 'usda',\n 'ext': 'usda',\n 'files': asset_filename,\n \"stagingDir\": stagingdir,\n }\n instance.data[\"representations\"].append(representation)\n\n log_representation = {\n 'name': 'txt',\n 'ext': 'txt',\n 'files': log_filename,\n \"stagingDir\": stagingdir,\n }\n instance.data[\"representations\"].append(log_representation)\n\n self.log.info(\n f\"Extracted instance '{instance.name}' to: {asset_filepath}\")\n\n @staticmethod\n def get_export_options(log_path):\n \"\"\"Set Export Options for USD Exporter\"\"\"\n\n export_options = rt.USDExporter.createOptions()\n\n export_options.Meshes = True\n export_options.Shapes = False\n export_options.Lights = False\n export_options.Cameras = False\n export_options.Materials = False\n export_options.MeshFormat = rt.Name('fromScene')\n export_options.FileFormat = rt.Name('ascii')\n export_options.UpAxis = rt.Name('y')\n export_options.LogLevel = rt.Name('info')\n export_options.LogPath = log_path\n export_options.PreserveEdgeOrientation = True\n export_options.TimeMode = rt.Name('current')\n\n rt.USDexporter.UIOptions = export_options\n\n return export_options\n"
},
{
"path": "openpype/hosts/max/plugins/publish/extract_pointcloud.py",
"content": "import os\n\nimport pyblish.api\nfrom pymxs import runtime as rt\n\nfrom openpype.hosts.max.api import maintained_selection\nfrom openpype.pipeline import publish\n\n\nclass ExtractPointCloud(publish.Extractor):\n \"\"\"\n Extract PRT format with tyFlow operators.\n\n Notes:\n Currently only works for the default partition setting\n\n Args:\n self.export_particle(): sets up all job arguments for attributes\n to be exported in MAXscript\n\n self.get_operators(): get the export_particle operator\n\n self.get_custom_attr(): get all custom channel attributes from Openpype\n setting and sets it as job arguments before exporting\n\n self.get_files(): get the files with tyFlow naming convention\n before publishing\n\n self.partition_output_name(): get the naming with partition settings.\n\n self.get_partition(): get partition value\n\n \"\"\"\n\n order = pyblish.api.ExtractorOrder - 0.2\n label = \"Extract Point Cloud\"\n hosts = [\"max\"]\n families = [\"pointcloud\"]\n settings = []\n\n def process(self, instance):\n self.settings = self.get_setting(instance)\n start = instance.data[\"frameStartHandle\"]\n end = instance.data[\"frameEndHandle\"]\n self.log.info(\"Extracting PRT...\")\n\n stagingdir = self.staging_dir(instance)\n filename = \"{name}.prt\".format(**instance.data)\n path = os.path.join(stagingdir, filename)\n\n with maintained_selection():\n job_args = self.export_particle(instance.data[\"members\"],\n start,\n end,\n path)\n\n for job in job_args:\n rt.Execute(job)\n\n self.log.info(\"Performing Extraction ...\")\n if \"representations\" not in instance.data:\n instance.data[\"representations\"] = []\n\n self.log.info(\"Writing PRT with TyFlow Plugin...\")\n filenames = self.get_files(\n instance.data[\"members\"], path, start, end)\n self.log.debug(f\"filenames: {filenames}\")\n\n partition = self.partition_output_name(\n instance.data[\"members\"])\n\n representation = {\n 'name': 'prt',\n 'ext': 'prt',\n 'files': filenames if len(filenames) > 1 else filenames[0],\n \"stagingDir\": stagingdir,\n \"outputName\": partition # partition value\n }\n instance.data[\"representations\"].append(representation)\n self.log.info(f\"Extracted instance '{instance.name}' to: {path}\")\n\n def export_particle(self,\n members,\n start,\n end,\n filepath):\n \"\"\"Sets up all job arguments for attributes.\n\n Those attributes are to be exported in MAX Script.\n\n Args:\n members (list): Member nodes of the instance.\n start (int): Start frame.\n end (int): End frame.\n filepath (str): Path to PRT file.\n\n Returns:\n list of arguments for MAX Script.\n\n \"\"\"\n job_args = []\n opt_list = self.get_operators(members)\n for operator in opt_list:\n start_frame = f\"{operator}.frameStart={start}\"\n job_args.append(start_frame)\n end_frame = f\"{operator}.frameEnd={end}\"\n job_args.append(end_frame)\n filepath = filepath.replace(\"\\\\\", \"/\")\n prt_filename = f'{operator}.PRTFilename=\"{filepath}\"'\n job_args.append(prt_filename)\n # Partition\n mode = f\"{operator}.PRTPartitionsMode=2\"\n job_args.append(mode)\n\n additional_args = self.get_custom_attr(operator)\n job_args.extend(iter(additional_args))\n prt_export = f\"{operator}.exportPRT()\"\n job_args.append(prt_export)\n\n return job_args\n\n @staticmethod\n def get_operators(members):\n \"\"\"Get Export Particles Operator.\n\n Args:\n members (list): Instance members.\n\n Returns:\n list of particle operators\n\n \"\"\"\n opt_list = []\n for member in members:\n obj = member.baseobject\n # TODO: to see if it can be used maxscript instead\n anim_names = rt.GetSubAnimNames(obj)\n for anim_name in anim_names:\n sub_anim = rt.GetSubAnim(obj, anim_name)\n boolean = rt.IsProperty(sub_anim, \"Export_Particles\")\n if boolean:\n event_name = sub_anim.Name\n opt = f\"${member.Name}.{event_name}.export_particles\"\n opt_list.append(opt)\n\n return opt_list\n\n @staticmethod\n def get_setting(instance):\n project_setting = instance.context.data[\"project_settings\"]\n return project_setting[\"max\"][\"PointCloud\"]\n\n def get_custom_attr(self, operator):\n \"\"\"Get Custom Attributes\"\"\"\n\n custom_attr_list = []\n attr_settings = self.settings[\"attribute\"]\n for key, value in attr_settings.items():\n custom_attr = \"{0}.PRTChannels_{1}=True\".format(operator,\n value)\n self.log.debug(\n \"{0} will be added as custom attribute\".format(key)\n )\n custom_attr_list.append(custom_attr)\n\n return custom_attr_list\n\n def get_files(self,\n container,\n path,\n start_frame,\n end_frame):\n \"\"\"Get file names for tyFlow.\n\n Set the filenames accordingly to the tyFlow file\n naming extension for the publishing purpose\n\n Actual File Output from tyFlow::\n __partof..prt\n\n e.g. tyFlow_cloth_CCCS_blobbyFill_001__part1of1_00004.prt\n\n Args:\n container: Instance node.\n path (str): Output directory.\n start_frame (int): Start frame.\n end_frame (int): End frame.\n\n Returns:\n list of filenames\n\n \"\"\"\n filenames = []\n filename = os.path.basename(path)\n orig_name, ext = os.path.splitext(filename)\n partition_count, partition_start = self.get_partition(container)\n for frame in range(int(start_frame), int(end_frame) + 1):\n actual_name = \"{}__part{:03}of{}_{:05}\".format(orig_name,\n partition_start,\n partition_count,\n frame)\n actual_filename = path.replace(orig_name, actual_name)\n filenames.append(os.path.basename(actual_filename))\n\n return filenames\n\n def partition_output_name(self, container):\n \"\"\"Get partition output name.\n\n Partition output name set for mapping\n the published file output.\n\n Todo:\n Customizes the setting for the output.\n\n Args:\n container: Instance node.\n\n Returns:\n str: Partition name.\n\n \"\"\"\n partition_count, partition_start = self.get_partition(container)\n return f\"_part{partition_start:03}of{partition_count}\"\n\n def get_partition(self, container):\n \"\"\"Get Partition value.\n\n Args:\n container: Instance node.\n\n \"\"\"\n opt_list = self.get_operators(container)\n # TODO: This looks strange? Iterating over\n # the opt_list but returning from inside?\n for operator in opt_list:\n count = rt.Execute(f'{operator}.PRTPartitionsCount')\n start = rt.Execute(f'{operator}.PRTPartitionsFrom')\n\n return count, start\n"
},
{
"path": "openpype/hosts/max/plugins/publish/extract_redshift_proxy.py",
"content": "import os\nimport pyblish.api\nfrom openpype.pipeline import publish\nfrom pymxs import runtime as rt\nfrom openpype.hosts.max.api import maintained_selection\n\n\nclass ExtractRedshiftProxy(publish.Extractor):\n \"\"\"\n Extract Redshift Proxy with rsProxy\n \"\"\"\n\n order = pyblish.api.ExtractorOrder - 0.1\n label = \"Extract RedShift Proxy\"\n hosts = [\"max\"]\n families = [\"redshiftproxy\"]\n\n def process(self, instance):\n start = instance.data[\"frameStartHandle\"]\n end = instance.data[\"frameEndHandle\"]\n\n self.log.debug(\"Extracting Redshift Proxy...\")\n stagingdir = self.staging_dir(instance)\n rs_filename = \"{name}.rs\".format(**instance.data)\n rs_filepath = os.path.join(stagingdir, rs_filename)\n rs_filepath = rs_filepath.replace(\"\\\\\", \"/\")\n\n rs_filenames = self.get_rsfiles(instance, start, end)\n\n with maintained_selection():\n # select and export\n node_list = instance.data[\"members\"]\n rt.Select(node_list)\n # Redshift rsProxy command\n # rsProxy fp selected compress connectivity startFrame endFrame\n # camera warnExisting transformPivotToOrigin\n rt.rsProxy(rs_filepath, 1, 0, 0, start, end, 0, 1, 1)\n\n self.log.info(\"Performing Extraction ...\")\n\n if \"representations\" not in instance.data:\n instance.data[\"representations\"] = []\n\n representation = {\n 'name': 'rs',\n 'ext': 'rs',\n 'files': rs_filenames if len(rs_filenames) > 1 else rs_filenames[0], # noqa\n \"stagingDir\": stagingdir,\n }\n instance.data[\"representations\"].append(representation)\n self.log.info(\"Extracted instance '%s' to: %s\" % (instance.name,\n stagingdir))\n\n def get_rsfiles(self, instance, startFrame, endFrame):\n rs_filenames = []\n rs_name = instance.data[\"name\"]\n for frame in range(startFrame, endFrame + 1):\n rs_filename = \"%s.%04d.rs\" % (rs_name, frame)\n rs_filenames.append(rs_filename)\n\n return rs_filenames\n"
},
{
"path": "openpype/hosts/max/plugins/publish/extract_review_animation.py",
"content": "import os\nimport pyblish.api\nfrom openpype.pipeline import publish\nfrom openpype.hosts.max.api.preview_animation import (\n render_preview_animation\n)\n\n\nclass ExtractReviewAnimation(publish.Extractor):\n \"\"\"\n Extract Review by Review Animation\n \"\"\"\n\n order = pyblish.api.ExtractorOrder + 0.001\n label = \"Extract Review Animation\"\n hosts = [\"max\"]\n families = [\"review\"]\n\n def process(self, instance):\n staging_dir = self.staging_dir(instance)\n ext = instance.data.get(\"imageFormat\")\n start = int(instance.data[\"frameStart\"])\n end = int(instance.data[\"frameEnd\"])\n filepath = os.path.join(staging_dir, instance.name)\n self.log.debug(\n \"Writing Review Animation to '{}'\".format(filepath))\n\n review_camera = instance.data[\"review_camera\"]\n viewport_options = instance.data.get(\"viewport_options\", {})\n files = render_preview_animation(\n filepath,\n ext,\n review_camera,\n start,\n end,\n percentSize=instance.data[\"percentSize\"],\n width=instance.data[\"review_width\"],\n height=instance.data[\"review_height\"],\n viewport_options=viewport_options)\n\n filenames = [os.path.basename(path) for path in files]\n\n tags = [\"review\"]\n if not instance.data.get(\"keepImages\"):\n tags.append(\"delete\")\n\n self.log.debug(\"Performing Extraction ...\")\n\n representation = {\n \"name\": instance.data[\"imageFormat\"],\n \"ext\": instance.data[\"imageFormat\"],\n \"files\": filenames,\n \"stagingDir\": staging_dir,\n \"frameStart\": instance.data[\"frameStartHandle\"],\n \"frameEnd\": instance.data[\"frameEndHandle\"],\n \"tags\": tags,\n \"preview\": True,\n \"camera_name\": review_camera\n }\n self.log.debug(f\"{representation}\")\n\n if \"representations\" not in instance.data:\n instance.data[\"representations\"] = []\n instance.data[\"representations\"].append(representation)\n"
},
{
"path": "openpype/hosts/max/plugins/publish/extract_thumbnail.py",
"content": "import os\nimport pyblish.api\nfrom openpype.pipeline import publish\nfrom openpype.hosts.max.api.preview_animation import render_preview_animation\n\n\nclass ExtractThumbnail(publish.Extractor):\n \"\"\"Extract Thumbnail for Review\n \"\"\"\n\n order = pyblish.api.ExtractorOrder\n label = \"Extract Thumbnail\"\n hosts = [\"max\"]\n families = [\"review\"]\n\n def process(self, instance):\n ext = instance.data.get(\"imageFormat\")\n frame = int(instance.data[\"frameStart\"])\n staging_dir = self.staging_dir(instance)\n filepath = os.path.join(\n staging_dir, f\"{instance.name}_thumbnail\")\n self.log.debug(\"Writing Thumbnail to '{}'\".format(filepath))\n\n review_camera = instance.data[\"review_camera\"]\n viewport_options = instance.data.get(\"viewport_options\", {})\n files = render_preview_animation(\n filepath,\n ext,\n review_camera,\n start_frame=frame,\n end_frame=frame,\n percentSize=instance.data[\"percentSize\"],\n width=instance.data[\"review_width\"],\n height=instance.data[\"review_height\"],\n viewport_options=viewport_options)\n\n thumbnail = next(os.path.basename(path) for path in files)\n\n representation = {\n \"name\": \"thumbnail\",\n \"ext\": ext,\n \"files\": thumbnail,\n \"stagingDir\": staging_dir,\n \"thumbnail\": True\n }\n\n self.log.debug(f\"{representation}\")\n\n if \"representations\" not in instance.data:\n instance.data[\"representations\"] = []\n instance.data[\"representations\"].append(representation)\n"
},
{
"path": "openpype/hosts/max/plugins/publish/extract_tycache.py",
"content": "import os\n\nimport pyblish.api\nfrom pymxs import runtime as rt\n\nfrom openpype.hosts.max.api import maintained_selection\nfrom openpype.pipeline import publish\n\n\nclass ExtractTyCache(publish.Extractor):\n \"\"\"Extract tycache format with tyFlow operators.\n Notes:\n - TyCache only works for TyFlow Pro Plugin.\n\n Methods:\n self.get_export_particles_job_args(): sets up all job arguments\n for attributes to be exported in MAXscript\n\n self.get_operators(): get the export_particle operator\n\n self.get_files(): get the files with tyFlow naming convention\n before publishing\n \"\"\"\n\n order = pyblish.api.ExtractorOrder - 0.2\n label = \"Extract TyCache\"\n hosts = [\"max\"]\n families = [\"tycache\"]\n\n def process(self, instance):\n # TODO: let user decide the param\n start = int(instance.context.data[\"frameStart\"])\n end = int(instance.context.data.get(\"frameEnd\"))\n self.log.debug(\"Extracting Tycache...\")\n\n stagingdir = self.staging_dir(instance)\n filename = \"{name}.tyc\".format(**instance.data)\n path = os.path.join(stagingdir, filename)\n filenames = self.get_files(instance, start, end)\n additional_attributes = instance.data.get(\"tyc_attrs\", {})\n\n with maintained_selection():\n job_args = self.get_export_particles_job_args(\n instance.data[\"members\"],\n start, end, path,\n additional_attributes)\n for job in job_args:\n rt.Execute(job)\n representations = instance.data.setdefault(\"representations\", [])\n representation = {\n 'name': 'tyc',\n 'ext': 'tyc',\n 'files': filenames if len(filenames) > 1 else filenames[0],\n \"stagingDir\": stagingdir,\n }\n representations.append(representation)\n\n # Get the tyMesh filename for extraction\n mesh_filename = f\"{instance.name}__tyMesh.tyc\"\n mesh_repres = {\n 'name': 'tyMesh',\n 'ext': 'tyc',\n 'files': mesh_filename,\n \"stagingDir\": stagingdir,\n \"outputName\": '__tyMesh'\n }\n representations.append(mesh_repres)\n\n # Get the material filename of which assigned in\n # tyCache for extraction\n material_filename = f\"{instance.name}__tyMtl.mat\"\n full_material_name = os.path.join(stagingdir, material_filename)\n full_material_name = full_material_name.replace(\"\\\\\", \"/\")\n if os.path.exists(full_material_name):\n mateiral_repres = {\n \"name\": 'tyMtl',\n \"ext\": 'mat',\n 'files': material_filename,\n 'stagingDir': stagingdir,\n \"outputName\": '__tyMtl'\n }\n representations.append(mateiral_repres)\n\n self.log.debug(f\"Extracted instance '{instance.name}' to: {filenames}\")\n\n def get_files(self, instance, start_frame, end_frame):\n \"\"\"Get file names for tyFlow in tyCache format.\n\n Set the filenames accordingly to the tyCache file\n naming extension(.tyc) for the publishing purpose\n\n Actual File Output from tyFlow in tyCache format:\n __tyPart_.tyc\n\n e.g. tycacheMain__tyPart_00000.tyc\n\n Args:\n instance (pyblish.api.Instance): instance.\n start_frame (int): Start frame.\n end_frame (int): End frame.\n\n Returns:\n filenames(list): list of filenames\n\n \"\"\"\n filenames = []\n for frame in range(int(start_frame), int(end_frame) + 1):\n filename = f\"{instance.name}__tyPart_{frame:05}.tyc\"\n filenames.append(filename)\n return filenames\n\n def get_export_particles_job_args(self, members, start, end,\n filepath, additional_attributes):\n \"\"\"Sets up all job arguments for attributes.\n\n Those attributes are to be exported in MAX Script.\n\n Args:\n members (list): Member nodes of the instance.\n start (int): Start frame.\n end (int): End frame.\n filepath (str): Output path of the TyCache file.\n additional_attributes (dict): channel attributes data\n which needed to be exported\n\n Returns:\n list of arguments for MAX Script.\n\n \"\"\"\n settings = {\n \"exportMode\": 2,\n \"frameStart\": start,\n \"frameEnd\": end,\n \"tyCacheFilename\": filepath.replace(\"\\\\\", \"/\")\n }\n settings.update(additional_attributes)\n\n job_args = []\n for operator in self.get_operators(members):\n for key, value in settings.items():\n if isinstance(value, str):\n # embed in quotes\n value = f'\"{value}\"'\n\n job_args.append(f\"{operator}.{key}={value}\")\n job_args.append(f\"{operator}.exportTyCache()\")\n return job_args\n\n @staticmethod\n def get_operators(members):\n \"\"\"Get Export Particles Operator.\n\n Args:\n members (list): Instance members.\n\n Returns:\n list of particle operators\n\n \"\"\"\n opt_list = []\n for member in members:\n obj = member.baseobject\n # TODO: see if it can use maxscript instead\n anim_names = rt.GetSubAnimNames(obj)\n for anim_name in anim_names:\n sub_anim = rt.GetSubAnim(obj, anim_name)\n boolean = rt.IsProperty(sub_anim, \"Export_Particles\")\n if boolean:\n event_name = sub_anim.Name\n opt = f\"${member.Name}.{event_name}.export_particles\"\n opt_list.append(opt)\n\n return opt_list\n"
},
{
"path": "openpype/hosts/max/plugins/publish/increment_workfile_version.py",
"content": "import pyblish.api\nfrom openpype.lib import version_up\nfrom pymxs import runtime as rt\n\n\nclass IncrementWorkfileVersion(pyblish.api.ContextPlugin):\n \"\"\"Increment current workfile version.\"\"\"\n\n order = pyblish.api.IntegratorOrder + 0.9\n label = \"Increment Workfile Version\"\n hosts = [\"max\"]\n families = [\"workfile\"]\n\n def process(self, context):\n path = context.data[\"currentFile\"]\n filepath = version_up(path)\n\n rt.saveMaxFile(filepath)\n self.log.info(\"Incrementing file version\")\n"
},
{
"path": "openpype/hosts/max/plugins/publish/save_scene.py",
"content": "import pyblish.api\nfrom openpype.pipeline import registered_host\n\n\nclass SaveCurrentScene(pyblish.api.ContextPlugin):\n \"\"\"Save current scene\"\"\"\n\n label = \"Save current file\"\n order = pyblish.api.ExtractorOrder - 0.49\n hosts = [\"max\"]\n families = [\"maxrender\", \"workfile\"]\n\n def process(self, context):\n host = registered_host()\n current_file = host.get_current_workfile()\n\n assert context.data[\"currentFile\"] == current_file\n\n if host.workfile_has_unsaved_changes():\n self.log.info(f\"Saving current file: {current_file}\")\n host.save_workfile(current_file)\n else:\n self.log.debug(\"No unsaved changes, skipping file save..\")\n"
},
{
"path": "openpype/hosts/max/plugins/publish/save_scenes_for_cameras.py",
"content": "import pyblish.api\nimport os\nimport sys\nimport tempfile\n\nfrom pymxs import runtime as rt\nfrom openpype.lib import run_subprocess\nfrom openpype.hosts.max.api.lib_rendersettings import RenderSettings\nfrom openpype.hosts.max.api.lib_renderproducts import RenderProducts\n\n\nclass SaveScenesForCamera(pyblish.api.InstancePlugin):\n \"\"\"Save scene files for multiple cameras without\n editing the original scene before deadline submission\n\n \"\"\"\n\n label = \"Save Scene files for cameras\"\n order = pyblish.api.ExtractorOrder - 0.48\n hosts = [\"max\"]\n families = [\"maxrender\"]\n\n def process(self, instance):\n if not instance.data.get(\"multiCamera\"):\n self.log.debug(\n \"Multi Camera disabled. \"\n \"Skipping to save scene files for cameras\")\n return\n current_folder = rt.maxFilePath\n current_filename = rt.maxFileName\n current_filepath = os.path.join(current_folder, current_filename)\n camera_scene_files = []\n scripts = []\n filename, ext = os.path.splitext(current_filename)\n fmt = RenderProducts().image_format()\n cameras = instance.data.get(\"cameras\")\n if not cameras:\n return\n new_folder = f\"{current_folder}_{filename}\"\n os.makedirs(new_folder, exist_ok=True)\n for camera in cameras:\n new_output = RenderSettings().get_batch_render_output(camera) # noqa\n new_output = new_output.replace(\"\\\\\", \"/\")\n new_filename = f\"{filename}_{camera}{ext}\"\n new_filepath = os.path.join(new_folder, new_filename)\n new_filepath = new_filepath.replace(\"\\\\\", \"/\")\n camera_scene_files.append(new_filepath)\n RenderSettings().batch_render_elements(camera)\n rt.rendOutputFilename = new_output\n rt.saveMaxFile(current_filepath)\n script = (\"\"\"\nfrom pymxs import runtime as rt\nimport os\nfilename = \"{filename}\"\nnew_filepath = \"{new_filepath}\"\nnew_output = \"{new_output}\"\ncamera = \"{camera}\"\nrt.rendOutputFilename = new_output\ndirectory = os.path.dirname(rt.rendOutputFilename)\ndirectory = os.path.join(directory, filename)\nrender_elem = rt.maxOps.GetCurRenderElementMgr()\nrender_elem_num = render_elem.NumRenderElements()\nif render_elem_num > 0:\n ext = \"{ext}\"\n for i in range(render_elem_num):\n renderlayer_name = render_elem.GetRenderElement(i)\n target, renderpass = str(renderlayer_name).split(\":\")\n aov_name = f\"{{directory}}_{camera}_{{renderpass}}..{ext}\"\n render_elem.SetRenderElementFileName(i, aov_name)\nrt.saveMaxFile(new_filepath)\n \"\"\").format(filename=instance.name,\n new_filepath=new_filepath,\n new_output=new_output,\n camera=camera,\n ext=fmt)\n scripts.append(script)\n\n maxbatch_exe = os.path.join(\n os.path.dirname(sys.executable), \"3dsmaxbatch\")\n maxbatch_exe = maxbatch_exe.replace(\"\\\\\", \"/\")\n if sys.platform == \"windows\":\n maxbatch_exe += \".exe\"\n maxbatch_exe = os.path.normpath(maxbatch_exe)\n with tempfile.TemporaryDirectory() as tmp_dir_name:\n tmp_script_path = os.path.join(\n tmp_dir_name, \"extract_scene_files.py\")\n self.log.info(\"Using script file: {}\".format(tmp_script_path))\n\n with open(tmp_script_path, \"wt\") as tmp:\n for script in scripts:\n tmp.write(script + \"\\n\")\n\n try:\n current_filepath = current_filepath.replace(\"\\\\\", \"/\")\n tmp_script_path = tmp_script_path.replace(\"\\\\\", \"/\")\n run_subprocess([maxbatch_exe, tmp_script_path,\n \"-sceneFile\", current_filepath])\n except RuntimeError:\n self.log.debug(\"Checking the scene files existing\")\n\n for camera_scene in camera_scene_files:\n if not os.path.exists(camera_scene):\n self.log.error(\"Camera scene files not existed yet!\")\n raise RuntimeError(\"MaxBatch.exe doesn't run as expected\")\n self.log.debug(f\"Found Camera scene:{camera_scene}\")\n"
},
{
"path": "openpype/hosts/max/plugins/publish/validate_attributes.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Validator for Attributes.\"\"\"\nfrom pyblish.api import ContextPlugin, ValidatorOrder\nfrom pymxs import runtime as rt\n\nfrom openpype.pipeline.publish import (\n OptionalPyblishPluginMixin,\n PublishValidationError,\n RepairContextAction\n)\n\n\ndef has_property(object_name, property_name):\n \"\"\"Return whether an object has a property with given name\"\"\"\n return rt.Execute(f'isProperty {object_name} \"{property_name}\"')\n\n\ndef is_matching_value(object_name, property_name, value):\n \"\"\"Return whether an existing property matches value `value\"\"\"\n property_value = rt.Execute(f\"{object_name}.{property_name}\")\n\n # Wrap property value if value is a string valued attributes\n # starting with a `#`\n if (\n isinstance(value, str) and\n value.startswith(\"#\") and\n not value.endswith(\")\")\n ):\n # prefix value with `#`\n # not applicable for #() array value type\n # and only applicable for enum i.e. #bob, #sally\n property_value = f\"#{property_value}\"\n\n return property_value == value\n\n\nclass ValidateAttributes(OptionalPyblishPluginMixin,\n ContextPlugin):\n \"\"\"Validates attributes in the project setting are consistent\n with the nodes from MaxWrapper Class in 3ds max.\n E.g. \"renderers.current.separateAovFiles\",\n \"renderers.production.PrimaryGIEngine\"\n Admin(s) need to put the dict below and enable this validator for a check:\n {\n \"renderers.current\":{\n \"separateAovFiles\" : True\n },\n \"renderers.production\":{\n \"PrimaryGIEngine\": \"#RS_GIENGINE_BRUTE_FORCE\"\n }\n ....\n }\n\n \"\"\"\n\n order = ValidatorOrder\n hosts = [\"max\"]\n label = \"Attributes\"\n actions = [RepairContextAction]\n optional = True\n\n @classmethod\n def get_invalid(cls, context):\n attributes = (\n context.data[\"project_settings\"][\"max\"][\"publish\"]\n [\"ValidateAttributes\"][\"attributes\"]\n )\n if not attributes:\n return\n invalid = []\n for object_name, required_properties in attributes.items():\n if not rt.Execute(f\"isValidValue {object_name}\"):\n # Skip checking if the node does not\n # exist in MaxWrapper Class\n cls.log.debug(f\"Unable to find '{object_name}'.\"\n \" Skipping validation of attributes.\")\n continue\n\n for property_name, value in required_properties.items():\n if not has_property(object_name, property_name):\n cls.log.error(\n \"Non-existing property: \"\n f\"{object_name}.{property_name}\")\n invalid.append((object_name, property_name))\n\n if not is_matching_value(object_name, property_name, value):\n cls.log.error(\n f\"Invalid value for: {object_name}.{property_name}\"\n f\" should be: {value}\")\n invalid.append((object_name, property_name))\n\n return invalid\n\n def process(self, context):\n if not self.is_active(context.data):\n self.log.debug(\"Skipping Validate Attributes...\")\n return\n invalid_attributes = self.get_invalid(context)\n if invalid_attributes:\n bullet_point_invalid_statement = \"\\n\".join(\n \"- {}\".format(invalid) for invalid\n in invalid_attributes\n )\n report = (\n \"Required Attribute(s) have invalid value(s).\\n\\n\"\n f\"{bullet_point_invalid_statement}\\n\\n\"\n \"You can use repair action to fix them if they are not\\n\"\n \"unknown property value(s).\"\n )\n raise PublishValidationError(\n report, title=\"Invalid Value(s) for Required Attribute(s)\")\n\n @classmethod\n def repair(cls, context):\n attributes = (\n context.data[\"project_settings\"][\"max\"][\"publish\"]\n [\"ValidateAttributes\"][\"attributes\"]\n )\n invalid_attributes = cls.get_invalid(context)\n for attrs in invalid_attributes:\n prop, attr = attrs\n value = attributes[prop][attr]\n if isinstance(value, str) and not value.startswith(\"#\"):\n attribute_fix = '{}.{}=\"{}\"'.format(\n prop, attr, value\n )\n else:\n attribute_fix = \"{}.{}={}\".format(\n prop, attr, value\n )\n rt.Execute(attribute_fix)\n"
},
{
"path": "openpype/hosts/max/plugins/publish/validate_camera_attributes.py",
"content": "import pyblish.api\nfrom pymxs import runtime as rt\n\nfrom openpype.pipeline.publish import (\n RepairAction,\n OptionalPyblishPluginMixin,\n PublishValidationError\n)\nfrom openpype.hosts.max.api.action import SelectInvalidAction\n\n\nclass ValidateCameraAttributes(OptionalPyblishPluginMixin,\n pyblish.api.InstancePlugin):\n \"\"\"Validates Camera has no invalid attribute properties\n or values.(For 3dsMax Cameras only)\n\n \"\"\"\n\n order = pyblish.api.ValidatorOrder\n families = ['camera']\n hosts = ['max']\n label = 'Validate Camera Attributes'\n actions = [SelectInvalidAction, RepairAction]\n optional = True\n\n DEFAULTS = [\"fov\", \"nearrange\", \"farrange\",\n \"nearclip\", \"farclip\"]\n CAM_TYPE = [\"Freecamera\", \"Targetcamera\",\n \"Physical\"]\n\n @classmethod\n def get_invalid(cls, instance):\n invalid = []\n if rt.units.DisplayType != rt.Name(\"Generic\"):\n cls.log.warning(\n \"Generic Type is not used as a scene unit\\n\\n\"\n \"sure you tweak the settings with your own values\\n\\n\"\n \"before validation.\")\n cameras = instance.data[\"members\"]\n project_settings = instance.context.data[\"project_settings\"].get(\"max\")\n cam_attr_settings = (\n project_settings[\"publish\"][\"ValidateCameraAttributes\"]\n )\n for camera in cameras:\n if str(rt.ClassOf(camera)) not in cls.CAM_TYPE:\n cls.log.debug(\n \"Skipping camera created from external plugin..\")\n continue\n for attr in cls.DEFAULTS:\n default_value = cam_attr_settings.get(attr)\n if default_value == float(0):\n cls.log.debug(\n f\"the value of {attr} in setting set to\"\n \" zero. Skipping the check.\")\n continue\n if round(rt.getProperty(camera, attr), 1) != default_value:\n cls.log.error(\n f\"Invalid attribute value for {camera.name}:{attr} \"\n f\"(should be: {default_value}))\")\n invalid.append(camera)\n\n return invalid\n\n def process(self, instance):\n if not self.is_active(instance.data):\n self.log.debug(\"Skipping Validate Camera Attributes.\")\n return\n invalid = self.get_invalid(instance)\n\n if invalid:\n raise PublishValidationError(\n \"Invalid camera attributes found. See log.\")\n\n @classmethod\n def repair(cls, instance):\n invalid_cameras = cls.get_invalid(instance)\n project_settings = instance.context.data[\"project_settings\"].get(\"max\")\n cam_attr_settings = (\n project_settings[\"publish\"][\"ValidateCameraAttributes\"]\n )\n for camera in invalid_cameras:\n for attr in cls.DEFAULTS:\n expected_value = cam_attr_settings.get(attr)\n if expected_value == float(0):\n cls.log.debug(\n f\"the value of {attr} in setting set to zero.\")\n continue\n rt.setProperty(camera, attr, expected_value)\n"
},
{
"path": "openpype/hosts/max/plugins/publish/validate_camera_contents.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\nfrom openpype.pipeline import PublishValidationError\nfrom pymxs import runtime as rt\n\n\nclass ValidateCameraContent(pyblish.api.InstancePlugin):\n \"\"\"Validates Camera instance contents.\n\n A Camera instance may only hold a SINGLE camera's transform\n \"\"\"\n\n order = pyblish.api.ValidatorOrder\n families = [\"camera\", \"review\"]\n hosts = [\"max\"]\n label = \"Camera Contents\"\n camera_type = [\"$Free_Camera\", \"$Target_Camera\",\n \"$Physical_Camera\", \"$Target\"]\n\n def process(self, instance):\n invalid = self.get_invalid(instance)\n if invalid:\n raise PublishValidationError((\"Camera instance must only include\"\n \"camera (and camera target). \"\n f\"Invalid content {invalid}\"))\n\n def get_invalid(self, instance):\n \"\"\"\n Get invalid nodes if the instance is not camera\n \"\"\"\n invalid = []\n container = instance.data[\"instance_node\"]\n self.log.info(f\"Validating camera content for {container}\")\n\n selection_list = instance.data[\"members\"]\n for sel in selection_list:\n # to avoid Attribute Error from pymxs wrapper\n sel_tmp = str(sel)\n found = any(sel_tmp.startswith(cam) for cam in self.camera_type)\n if not found:\n self.log.error(\"Camera not found\")\n invalid.append(sel)\n return invalid\n"
},
{
"path": "openpype/hosts/max/plugins/publish/validate_frame_range.py",
"content": "import pyblish.api\n\nfrom pymxs import runtime as rt\nfrom openpype.pipeline import (\n OptionalPyblishPluginMixin\n)\nfrom openpype.pipeline.publish import (\n RepairAction,\n ValidateContentsOrder,\n PublishValidationError,\n KnownPublishError\n)\nfrom openpype.hosts.max.api.lib import get_frame_range, set_timeline\n\n\nclass ValidateFrameRange(pyblish.api.InstancePlugin,\n OptionalPyblishPluginMixin):\n \"\"\"Validates the frame ranges.\n\n This is an optional validator checking if the frame range on instance\n matches the frame range specified for the asset.\n\n It also validates render frame ranges of render layers.\n\n Repair action will change everything to match the asset frame range.\n\n This can be turned off by the artist to allow custom ranges.\n \"\"\"\n\n label = \"Validate Frame Range\"\n order = ValidateContentsOrder\n families = [\"camera\", \"maxrender\",\n \"pointcache\", \"pointcloud\",\n \"review\", \"redshiftproxy\"]\n hosts = [\"max\"]\n optional = True\n actions = [RepairAction]\n\n def process(self, instance):\n if not self.is_active(instance.data):\n self.log.debug(\"Skipping Validate Frame Range...\")\n return\n\n frame_range = get_frame_range(\n asset_doc=instance.data[\"assetEntity\"])\n\n inst_frame_start = instance.data.get(\"frameStartHandle\")\n inst_frame_end = instance.data.get(\"frameEndHandle\")\n if inst_frame_start is None or inst_frame_end is None:\n raise KnownPublishError(\n \"Missing frame start and frame end on \"\n \"instance to to validate.\"\n )\n frame_start_handle = frame_range[\"frameStartHandle\"]\n frame_end_handle = frame_range[\"frameEndHandle\"]\n errors = []\n if frame_start_handle != inst_frame_start:\n errors.append(\n f\"Start frame ({inst_frame_start}) on instance does not match \" # noqa\n f\"with the start frame ({frame_start_handle}) set on the asset data. \") # noqa\n if frame_end_handle != inst_frame_end:\n errors.append(\n f\"End frame ({inst_frame_end}) on instance does not match \"\n f\"with the end frame ({frame_end_handle}) \"\n \"from the asset data. \")\n\n if errors:\n bullet_point_errors = \"\\n\".join(\n \"- {}\".format(error) for error in errors\n )\n report = (\n \"Frame range settings are incorrect.\\n\\n\"\n f\"{bullet_point_errors}\\n\\n\"\n \"You can use repair action to fix it.\"\n )\n raise PublishValidationError(report, title=\"Frame Range incorrect\")\n\n @classmethod\n def repair(cls, instance):\n frame_range = get_frame_range()\n frame_start_handle = frame_range[\"frameStartHandle\"]\n frame_end_handle = frame_range[\"frameEndHandle\"]\n\n if instance.data[\"family\"] == \"maxrender\":\n rt.rendStart = frame_start_handle\n rt.rendEnd = frame_end_handle\n else:\n set_timeline(frame_start_handle, frame_end_handle)\n"
},
{
"path": "openpype/hosts/max/plugins/publish/validate_instance_has_members.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\nfrom openpype.pipeline import PublishValidationError\n\n\nclass ValidateInstanceHasMembers(pyblish.api.InstancePlugin):\n \"\"\"Validates Instance has members.\n\n Check if MaxScene containers includes any contents underneath.\n \"\"\"\n\n order = pyblish.api.ValidatorOrder\n families = [\"camera\",\n \"model\",\n \"maxScene\",\n \"review\",\n \"pointcache\",\n \"pointcloud\",\n \"redshiftproxy\"]\n hosts = [\"max\"]\n label = \"Container Contents\"\n\n def process(self, instance):\n if not instance.data[\"members\"]:\n raise PublishValidationError(\"No content found in the container\")\n"
},
{
"path": "openpype/hosts/max/plugins/publish/validate_instance_in_context.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Validate if instance asset is the same as context asset.\"\"\"\nfrom __future__ import absolute_import\n\nimport pyblish.api\nfrom openpype import AYON_SERVER_ENABLED\nfrom openpype.pipeline.publish import (\n RepairAction,\n ValidateContentsOrder,\n PublishValidationError,\n OptionalPyblishPluginMixin\n)\nfrom openpype.hosts.max.api.action import SelectInvalidAction\nfrom pymxs import runtime as rt\n\n\nclass ValidateInstanceInContext(pyblish.api.InstancePlugin,\n OptionalPyblishPluginMixin):\n \"\"\"Validator to check if instance asset match context asset.\n\n When working in per-shot style you always publish data in context of\n current asset (shot). This validator checks if this is so. It is optional\n so it can be disabled when needed.\n\n Action on this validator will select invalid instances.\n \"\"\"\n\n order = ValidateContentsOrder\n label = \"Instance in same Context\"\n optional = True\n hosts = [\"max\"]\n actions = [SelectInvalidAction, RepairAction]\n\n def process(self, instance):\n if not self.is_active(instance.data):\n return\n\n instance_node = rt.getNodeByName(instance.data.get(\n \"instance_node\", \"\"))\n if not instance_node:\n return\n asset_name_attr = \"folderPath\" if AYON_SERVER_ENABLED else \"asset\"\n asset = rt.getUserProp(instance_node, asset_name_attr)\n context_asset = self.get_context_asset(instance)\n task = rt.getUserProp(instance_node, \"task\")\n context_task = self.get_context_task(instance)\n if asset != context_asset:\n raise PublishValidationError(\n message=(\n \"Instance '{}' publishes to different asset than current \"\n \"context: {}. Current context: {}\".format(\n instance.name, asset, context_asset\n )\n ),\n description=(\n \"## Publishing to a different asset\\n\"\n \"There are publish instances present which are publishing \"\n \"into a different asset than your current context.\\n\\n\"\n \"Usually this is not what you want but there can be cases \"\n \"where you might want to publish into another asset or \"\n \"shot. If that's the case you can disable the validation \"\n \"on the instance to ignore it.\"\n )\n )\n if task != context_task:\n raise PublishValidationError(\n message=(\n \"Instance '{}' publishes to different task than current \"\n \"context: {}. Current context: {}\".format(\n instance.name, task, context_task\n )\n ),\n description=(\n \"## Publishing to a different asset\\n\"\n \"There are publish instances present which are publishing \"\n \"into a different asset than your current context.\\n\\n\"\n \"Usually this is not what you want but there can be cases \"\n \"where you might want to publish into another asset or \"\n \"shot. If that's the case you can disable the validation \"\n \"on the instance to ignore it.\"\n )\n )\n\n @classmethod\n def get_invalid(cls, instance):\n asset_name_attr = \"folderPath\" if AYON_SERVER_ENABLED else \"asset\"\n node = rt.getNodeByName(instance.data[\"instance_node\"])\n asset = rt.getUserProp(node, asset_name_attr)\n context_asset = cls.get_context_asset(instance)\n if asset != context_asset:\n return instance.data[\"instance_node\"]\n\n @classmethod\n def repair(cls, instance):\n context_asset = cls.get_context_asset(instance)\n context_task = cls.get_context_task(instance)\n instance_node = rt.getNodeByName(instance.data.get(\n \"instance_node\", \"\"))\n if not instance_node:\n return\n asset_name_attr = \"folderPath\" if AYON_SERVER_ENABLED else \"asset\"\n rt.SetUserProp(instance_node, asset_name_attr, context_asset)\n rt.SetUserProp(instance_node, \"task\", context_task)\n\n @staticmethod\n def get_context_asset(instance):\n return instance.context.data[\"asset\"]\n\n @staticmethod\n def get_context_task(instance):\n return instance.context.data[\"task\"]\n"
},
{
"path": "openpype/hosts/max/plugins/publish/validate_loaded_plugin.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Validator for Loaded Plugin.\"\"\"\nimport os\nimport pyblish.api\nfrom pymxs import runtime as rt\n\nfrom openpype.pipeline.publish import (\n RepairAction,\n OptionalPyblishPluginMixin,\n PublishValidationError\n)\nfrom openpype.hosts.max.api.lib import get_plugins\n\n\nclass ValidateLoadedPlugin(OptionalPyblishPluginMixin,\n pyblish.api.InstancePlugin):\n \"\"\"Validates if the specific plugin is loaded in 3ds max.\n Studio Admin(s) can add the plugins they want to check in validation\n via studio defined project settings\n \"\"\"\n\n order = pyblish.api.ValidatorOrder\n hosts = [\"max\"]\n label = \"Validate Loaded Plugins\"\n optional = True\n actions = [RepairAction]\n\n family_plugins_mapping = {}\n\n @classmethod\n def get_invalid(cls, instance):\n \"\"\"Plugin entry point.\"\"\"\n family_plugins_mapping = cls.family_plugins_mapping\n if not family_plugins_mapping:\n return\n\n invalid = []\n # Find all plug-in requirements for current instance\n instance_families = {instance.data[\"family\"]}\n instance_families.update(instance.data.get(\"families\", []))\n cls.log.debug(\"Checking plug-in validation \"\n f\"for instance families: {instance_families}\")\n all_required_plugins = set()\n\n for mapping in family_plugins_mapping:\n # Check for matching families\n if not mapping:\n return\n\n match_families = {fam.strip() for fam in mapping[\"families\"]}\n has_match = \"*\" in match_families or match_families.intersection(\n instance_families)\n\n if not has_match:\n continue\n\n cls.log.debug(\n f\"Found plug-in family requirements: {match_families}\")\n required_plugins = [\n # match lowercase and format with os.environ to allow\n # plugin names defined by max version, e.g. {3DSMAX_VERSION}\n plugin.format(**os.environ).lower()\n for plugin in mapping[\"plugins\"]\n # ignore empty fields in settings\n if plugin.strip()\n ]\n\n all_required_plugins.update(required_plugins)\n\n if not all_required_plugins:\n # Instance has no plug-in requirements\n return\n\n # get all DLL loaded plugins in Max and their plugin index\n available_plugins = {\n plugin_name.lower(): index for index, plugin_name in enumerate(\n get_plugins())\n }\n # validate the required plug-ins\n for plugin in sorted(all_required_plugins):\n plugin_index = available_plugins.get(plugin)\n if plugin_index is None:\n debug_msg = (\n f\"Plugin {plugin} does not exist\"\n \" in 3dsMax Plugin List.\"\n )\n invalid.append((plugin, debug_msg))\n continue\n if not rt.pluginManager.isPluginDllLoaded(plugin_index):\n debug_msg = f\"Plugin {plugin} not loaded.\"\n invalid.append((plugin, debug_msg))\n return invalid\n\n def process(self, instance):\n if not self.is_active(instance.data):\n self.log.debug(\"Skipping Validate Loaded Plugin...\")\n return\n invalid = self.get_invalid(instance)\n if invalid:\n bullet_point_invalid_statement = \"\\n\".join(\n \"- {}\".format(message) for _, message in invalid\n )\n report = (\n \"Required plugins are not loaded.\\n\\n\"\n f\"{bullet_point_invalid_statement}\\n\\n\"\n \"You can use repair action to load the plugin.\"\n )\n raise PublishValidationError(\n report, title=\"Missing Required Plugins\")\n\n @classmethod\n def repair(cls, instance):\n # get all DLL loaded plugins in Max and their plugin index\n invalid = cls.get_invalid(instance)\n if not invalid:\n return\n\n # get all DLL loaded plugins in Max and their plugin index\n available_plugins = {\n plugin_name.lower(): index for index, plugin_name in enumerate(\n get_plugins())\n }\n\n for invalid_plugin, _ in invalid:\n plugin_index = available_plugins.get(invalid_plugin)\n\n if plugin_index is None:\n cls.log.warning(\n f\"Can't enable missing plugin: {invalid_plugin}\")\n continue\n\n if not rt.pluginManager.isPluginDllLoaded(plugin_index):\n rt.pluginManager.loadPluginDll(plugin_index)\n"
},
{
"path": "openpype/hosts/max/plugins/publish/validate_model_contents.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\nfrom pymxs import runtime as rt\n\nfrom openpype.pipeline import PublishValidationError\n\n\nclass ValidateModelContent(pyblish.api.InstancePlugin):\n \"\"\"Validates Model instance contents.\n\n A model instance may only hold either geometry-related\n object(excluding Shapes) or editable meshes.\n \"\"\"\n\n order = pyblish.api.ValidatorOrder\n families = [\"model\"]\n hosts = [\"max\"]\n label = \"Model Contents\"\n\n def process(self, instance):\n invalid = self.get_invalid(instance)\n if invalid:\n raise PublishValidationError((\"Model instance must only include\"\n \"Geometry and Editable Mesh. \"\n f\"Invalid types on: {invalid}\"))\n\n def get_invalid(self, instance):\n \"\"\"\n Get invalid nodes if the instance is not camera\n \"\"\"\n invalid = []\n container = instance.data[\"instance_node\"]\n self.log.info(f\"Validating model content for {container}\")\n\n selection_list = instance.data[\"members\"]\n for sel in selection_list:\n if rt.ClassOf(sel) in rt.Camera.classes:\n invalid.append(sel)\n if rt.ClassOf(sel) in rt.Light.classes:\n invalid.append(sel)\n if rt.ClassOf(sel) in rt.Shape.classes:\n invalid.append(sel)\n\n return invalid\n"
},
{
"path": "openpype/hosts/max/plugins/publish/validate_pointcloud.py",
"content": "import pyblish.api\nfrom openpype.pipeline import PublishValidationError\nfrom pymxs import runtime as rt\n\n\nclass ValidatePointCloud(pyblish.api.InstancePlugin):\n \"\"\"Validate that work file was saved.\"\"\"\n\n order = pyblish.api.ValidatorOrder\n families = [\"pointcloud\"]\n hosts = [\"max\"]\n label = \"Validate Point Cloud\"\n\n def process(self, instance):\n \"\"\"\n Notes:\n 1. Validate if the export mode of Export Particle is at PRT format\n 2. Validate the partition count and range set as default value\n Partition Count : 100\n Partition Range : 1 to 1\n 3. Validate if the custom attribute(s) exist as parameter(s)\n of export_particle operator\n\n \"\"\"\n report = []\n\n if self.validate_export_mode(instance):\n report.append(\"The export mode is not at PRT\")\n\n if self.validate_partition_value(instance):\n report.append((\"tyFlow Partition setting is \"\n \"not at the default value\"))\n\n invalid_attribute = self.validate_custom_attribute(instance)\n if invalid_attribute:\n report.append((\"Custom Attribute not found \"\n f\":{invalid_attribute}\"))\n\n if report:\n raise PublishValidationError(f\"{report}\")\n\n def validate_custom_attribute(self, instance):\n invalid = []\n container = instance.data[\"instance_node\"]\n self.log.info(\n f\"Validating tyFlow custom attributes for {container}\")\n\n selection_list = instance.data[\"members\"]\n\n project_settings = instance.context.data[\"project_settings\"]\n attr_settings = project_settings[\"max\"][\"PointCloud\"][\"attribute\"]\n for sel in selection_list:\n obj = sel.baseobject\n anim_names = rt.GetSubAnimNames(obj)\n for anim_name in anim_names:\n # get all the names of the related tyFlow nodes\n sub_anim = rt.GetSubAnim(obj, anim_name)\n if rt.IsProperty(sub_anim, \"Export_Particles\"):\n event_name = sub_anim.name\n opt = \"${0}.{1}.export_particles\".format(sel.name,\n event_name)\n for key, value in attr_settings.items():\n custom_attr = \"{0}.PRTChannels_{1}\".format(opt,\n value)\n try:\n rt.Execute(custom_attr)\n except RuntimeError:\n invalid.append(key)\n\n return invalid\n\n def validate_partition_value(self, instance):\n invalid = []\n container = instance.data[\"instance_node\"]\n self.log.info(\n f\"Validating tyFlow partition value for {container}\")\n\n selection_list = instance.data[\"members\"]\n for sel in selection_list:\n obj = sel.baseobject\n anim_names = rt.GetSubAnimNames(obj)\n for anim_name in anim_names:\n # get all the names of the related tyFlow nodes\n sub_anim = rt.GetSubAnim(obj, anim_name)\n if rt.IsProperty(sub_anim, \"Export_Particles\"):\n event_name = sub_anim.name\n opt = \"${0}.{1}.export_particles\".format(sel.name,\n event_name)\n count = rt.Execute(f'{opt}.PRTPartitionsCount')\n if count != 100:\n invalid.append(count)\n start = rt.Execute(f'{opt}.PRTPartitionsFrom')\n if start != 1:\n invalid.append(start)\n end = rt.Execute(f'{opt}.PRTPartitionsTo')\n if end != 1:\n invalid.append(end)\n\n return invalid\n\n def validate_export_mode(self, instance):\n invalid = []\n container = instance.data[\"instance_node\"]\n self.log.info(\n f\"Validating tyFlow export mode for {container}\")\n\n con = rt.GetNodeByName(container)\n selection_list = list(con.Children)\n for sel in selection_list:\n obj = sel.baseobject\n anim_names = rt.GetSubAnimNames(obj)\n for anim_name in anim_names:\n # get all the names of the related tyFlow nodes\n sub_anim = rt.GetSubAnim(obj, anim_name)\n # check if there is export particle operator\n boolean = rt.IsProperty(sub_anim, \"Export_Particles\")\n event_name = sub_anim.name\n if boolean:\n opt = f\"${sel.name}.{event_name}.export_particles\"\n export_mode = rt.Execute(f'{opt}.exportMode')\n if export_mode != 1:\n invalid.append(export_mode)\n\n return invalid\n"
},
{
"path": "openpype/hosts/max/plugins/publish/validate_renderable_camera.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\nfrom openpype.pipeline import (\n PublishValidationError,\n OptionalPyblishPluginMixin)\nfrom openpype.pipeline.publish import RepairAction\nfrom openpype.hosts.max.api.lib import get_current_renderer\n\nfrom pymxs import runtime as rt\n\n\nclass ValidateRenderableCamera(pyblish.api.InstancePlugin,\n OptionalPyblishPluginMixin):\n \"\"\"Validates Renderable Camera\n\n Check if the renderable camera used for rendering\n \"\"\"\n\n order = pyblish.api.ValidatorOrder\n families = [\"maxrender\"]\n hosts = [\"max\"]\n label = \"Renderable Camera\"\n optional = True\n actions = [RepairAction]\n\n def process(self, instance):\n if not self.is_active(instance.data):\n return\n if not instance.data[\"cameras\"]:\n raise PublishValidationError(\n \"No renderable Camera found in scene.\"\n )\n\n @classmethod\n def repair(cls, instance):\n\n rt.viewport.setType(rt.Name(\"view_camera\"))\n camera = rt.viewport.GetCamera()\n cls.log.info(f\"Camera {camera} set as renderable camera\")\n renderer_class = get_current_renderer()\n renderer = str(renderer_class).split(\":\")[0]\n if renderer == \"Arnold\":\n arv = rt.MAXToAOps.ArnoldRenderView()\n arv.setOption(\"Camera\", str(camera))\n arv.close()\n instance.data[\"cameras\"] = [camera.name]\n"
},
{
"path": "openpype/hosts/max/plugins/publish/validate_renderer_redshift_proxy.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\nfrom openpype.pipeline import PublishValidationError\nfrom pymxs import runtime as rt\nfrom openpype.pipeline.publish import RepairAction\nfrom openpype.hosts.max.api.lib import get_current_renderer\n\n\nclass ValidateRendererRedshiftProxy(pyblish.api.InstancePlugin):\n \"\"\"\n Validates Redshift as the current renderer for creating\n Redshift Proxy\n \"\"\"\n\n order = pyblish.api.ValidatorOrder\n families = [\"redshiftproxy\"]\n hosts = [\"max\"]\n label = \"Redshift Renderer\"\n actions = [RepairAction]\n\n def process(self, instance):\n invalid = self.get_redshift_renderer(instance)\n if invalid:\n raise PublishValidationError(\"Please install Redshift for 3dsMax\"\n \" before using the Redshift proxy instance\") # noqa\n invalid = self.get_current_renderer(instance)\n if invalid:\n raise PublishValidationError(\"The Redshift proxy extraction\"\n \"discontinued since the current renderer is not Redshift\") # noqa\n\n def get_redshift_renderer(self, instance):\n invalid = list()\n max_renderers_list = str(rt.RendererClass.classes)\n if \"Redshift_Renderer\" not in max_renderers_list:\n invalid.append(max_renderers_list)\n\n return invalid\n\n def get_current_renderer(self, instance):\n invalid = list()\n renderer_class = get_current_renderer()\n current_renderer = str(renderer_class).split(\":\")[0]\n if current_renderer != \"Redshift_Renderer\":\n invalid.append(current_renderer)\n\n return invalid\n\n @classmethod\n def repair(cls, instance):\n for Renderer in rt.RendererClass.classes:\n renderer = Renderer()\n if \"Redshift_Renderer\" in str(renderer):\n rt.renderers.production = renderer\n break\n"
},
{
"path": "openpype/hosts/max/plugins/publish/validate_renderpasses.py",
"content": "import os\nimport pyblish.api\nfrom pymxs import runtime as rt\nfrom openpype.pipeline.publish import (\n RepairAction,\n ValidateContentsOrder,\n PublishValidationError,\n OptionalPyblishPluginMixin\n)\nfrom openpype.hosts.max.api.lib_rendersettings import RenderSettings\n\n\nclass ValidateRenderPasses(OptionalPyblishPluginMixin,\n pyblish.api.InstancePlugin):\n \"\"\"Validates Render Passes before Deadline Submission\n \"\"\"\n\n order = ValidateContentsOrder\n families = [\"maxrender\"]\n hosts = [\"max\"]\n label = \"Validate Render Passes\"\n optional = True\n actions = [RepairAction]\n\n def process(self, instance):\n if not self.is_active(instance.data):\n return\n invalid = self.get_invalid(instance)\n if invalid:\n bullet_point_invalid_statement = \"\\n\".join(\n f\"- {err_type}: {filepath}\" for err_type, filepath\n in invalid\n )\n report = (\n \"Invalid render passes found.\\n\\n\"\n f\"{bullet_point_invalid_statement}\\n\\n\"\n \"You can use repair action to fix the invalid filepath.\"\n )\n raise PublishValidationError(\n report, title=\"Invalid Render Passes\")\n\n @classmethod\n def get_invalid(cls, instance):\n \"\"\"Function to get invalid beauty render outputs and\n render elements.\n\n 1. Check Render Output Folder matches the name of\n the current Max Scene, e.g.\n The name of the current Max scene:\n John_Doe.max\n The expected render output directory:\n {root[work]}/{project[name]}/{hierarchy}/{asset}/\n work/{task[name]}/render/3dsmax/John_Doe/\n\n 2. Check image extension(s) of the render output(s)\n matches the image format in OP/AYON setting, e.g.\n The current image format in settings: png\n The expected render outputs: John_Doe.png\n\n 3. Check filename of render element ends with the name of\n render element from the 3dsMax Render Element Manager.\n e.g. The name of render element: RsCryptomatte\n The expected filename: {InstanceName}_RsCryptomatte.png\n\n Args:\n instance (pyblish.api.Instance): instance\n filename (str): filename of the Max scene\n\n Returns:\n list: list of invalid filename which doesn't match\n with the project name\n \"\"\"\n invalid = []\n file = rt.maxFileName\n filename, ext = os.path.splitext(file)\n if filename not in rt.rendOutputFilename:\n cls.log.error(\n \"Render output folder must include \"\n f\" the max scene name {filename} \"\n )\n invalid_folder_name = os.path.dirname(\n rt.rendOutputFilename).replace(\n \"\\\\\", \"/\").split(\"/\")[-1]\n invalid.append((\"Invalid Render Output Folder\",\n invalid_folder_name))\n beauty_fname = os.path.basename(rt.rendOutputFilename)\n beauty_name, ext = os.path.splitext(beauty_fname)\n invalid_filenames = cls.get_invalid_filenames(\n instance, beauty_name)\n invalid.extend(invalid_filenames)\n invalid_image_format = cls.get_invalid_image_format(\n instance, ext.lstrip(\".\"))\n invalid.extend(invalid_image_format)\n renderer = instance.data[\"renderer\"]\n if renderer in [\n \"ART_Renderer\",\n \"Redshift_Renderer\",\n \"V_Ray_6_Hotfix_3\",\n \"V_Ray_GPU_6_Hotfix_3\",\n \"Default_Scanline_Renderer\",\n \"Quicksilver_Hardware_Renderer\",\n ]:\n render_elem = rt.maxOps.GetCurRenderElementMgr()\n render_elem_num = render_elem.NumRenderElements()\n for i in range(render_elem_num):\n renderlayer_name = render_elem.GetRenderElement(i)\n renderpass = str(renderlayer_name).split(\":\")[-1]\n rend_file = render_elem.GetRenderElementFilename(i)\n if not rend_file:\n cls.log.error(\n f\"No filepath for render element {renderpass}\")\n invalid.append((f\"Invalid {renderpass}\",\n \"No filepath\"))\n rend_fname, ext = os.path.splitext(\n os.path.basename(rend_file))\n invalid_filenames = cls.get_invalid_filenames(\n instance, rend_fname, renderpass=renderpass)\n invalid.extend(invalid_filenames)\n invalid_image_format = cls.get_invalid_image_format(\n instance, ext)\n invalid.extend(invalid_image_format)\n elif renderer == \"Arnold\":\n cls.log.debug(\n \"Renderpass validation does not support Arnold yet,\"\n \" validation skipped...\")\n\n return invalid\n\n @classmethod\n def get_invalid_filenames(cls, instance, file_name, renderpass=None):\n \"\"\"Function to get invalid filenames from render outputs.\n\n Args:\n instance (pyblish.api.Instance): instance\n file_name (str): name of the file\n renderpass (str, optional): name of the renderpass.\n Defaults to None.\n\n Returns:\n list: invalid filenames\n \"\"\"\n invalid = []\n if instance.name not in file_name:\n cls.log.error(\"The renderpass should have instance name inside.\")\n invalid.append((f\"Invalid instance name\",\n file_name))\n if renderpass is not None:\n if not file_name.rstrip(\".\").endswith(renderpass):\n cls.log.error(\n f\"Filename for {renderpass} should \"\n f\"end with {renderpass}\"\n )\n invalid.append((f\"Invalid {renderpass}\",\n os.path.basename(file_name)))\n return invalid\n\n @classmethod\n def get_invalid_image_format(cls, instance, ext):\n \"\"\"Function to check if the image format of the render outputs\n aligns with that in the setting.\n\n Args:\n instance (pyblish.api.Instance): instance\n ext (str): image extension\n\n Returns:\n list: list of files with invalid image format\n \"\"\"\n invalid = []\n settings = instance.context.data[\"project_settings\"].get(\"max\")\n image_format = settings[\"RenderSettings\"][\"image_format\"]\n ext = ext.lstrip(\".\")\n if ext != image_format:\n msg = (\n f\"Invalid image format {ext} for render outputs.\\n\"\n f\"Should be: {image_format}\")\n cls.log.error(msg)\n invalid.append((msg, ext))\n return invalid\n\n @classmethod\n def repair(cls, instance):\n container = instance.data.get(\"instance_node\")\n # TODO: need to rename the function of render_output\n RenderSettings().render_output(container)\n cls.log.debug(\"Finished repairing the render output \"\n \"folder and filenames.\")\n"
},
{
"path": "openpype/hosts/max/plugins/publish/validate_resolution_setting.py",
"content": "import pyblish.api\nfrom pymxs import runtime as rt\nfrom openpype.pipeline import (\n OptionalPyblishPluginMixin\n)\nfrom openpype.pipeline.publish import (\n RepairAction,\n PublishValidationError\n)\nfrom openpype.hosts.max.api.lib import reset_scene_resolution\n\n\nclass ValidateResolutionSetting(pyblish.api.InstancePlugin,\n OptionalPyblishPluginMixin):\n \"\"\"Validate the resolution setting aligned with DB\"\"\"\n\n order = pyblish.api.ValidatorOrder - 0.01\n families = [\"maxrender\"]\n hosts = [\"max\"]\n label = \"Validate Resolution Setting\"\n optional = True\n actions = [RepairAction]\n\n def process(self, instance):\n if not self.is_active(instance.data):\n return\n width, height = self.get_db_resolution(instance)\n current_width = rt.renderWidth\n current_height = rt.renderHeight\n if current_width != width and current_height != height:\n raise PublishValidationError(\"Resolution Setting \"\n \"not matching resolution \"\n \"set on asset or shot.\")\n if current_width != width:\n raise PublishValidationError(\"Width in Resolution Setting \"\n \"not matching resolution set \"\n \"on asset or shot.\")\n\n if current_height != height:\n raise PublishValidationError(\"Height in Resolution Setting \"\n \"not matching resolution set \"\n \"on asset or shot.\")\n\n def get_db_resolution(self, instance):\n asset_doc = instance.data[\"assetEntity\"]\n project_doc = instance.context.data[\"projectEntity\"]\n for data in [asset_doc[\"data\"], project_doc[\"data\"]]:\n if \"resolutionWidth\" in data and \"resolutionHeight\" in data:\n width = data[\"resolutionWidth\"]\n height = data[\"resolutionHeight\"]\n return int(width), int(height)\n\n # Defaults if not found in asset document or project document\n return 1920, 1080\n\n @classmethod\n def repair(cls, instance):\n reset_scene_resolution()\n"
},
{
"path": "openpype/hosts/max/plugins/publish/validate_scene_saved.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\nfrom openpype.pipeline import PublishValidationError\nfrom pymxs import runtime as rt\n\n\nclass ValidateSceneSaved(pyblish.api.InstancePlugin):\n \"\"\"Validate that workfile was saved.\"\"\"\n\n order = pyblish.api.ValidatorOrder\n families = [\"workfile\"]\n hosts = [\"max\"]\n label = \"Validate Workfile is saved\"\n\n def process(self, instance):\n if not rt.maxFilePath or not rt.maxFileName:\n raise PublishValidationError(\n \"Workfile is not saved\", title=self.label)\n"
},
{
"path": "openpype/hosts/max/plugins/publish/validate_tyflow_data.py",
"content": "import pyblish.api\nfrom openpype.pipeline import PublishValidationError\nfrom pymxs import runtime as rt\n\n\nclass ValidateTyFlowData(pyblish.api.InstancePlugin):\n \"\"\"Validate TyFlow plugins or relevant operators are set correctly.\"\"\"\n\n order = pyblish.api.ValidatorOrder\n families = [\"pointcloud\", \"tycache\"]\n hosts = [\"max\"]\n label = \"TyFlow Data\"\n\n def process(self, instance):\n \"\"\"\n Notes:\n 1. Validate the container only include tyFlow objects\n 2. Validate if tyFlow operator Export Particle exists\n\n \"\"\"\n\n invalid_object = self.get_tyflow_object(instance)\n if invalid_object:\n self.log.error(f\"Non tyFlow object found: {invalid_object}\")\n\n invalid_operator = self.get_tyflow_operator(instance)\n if invalid_operator:\n self.log.error(\n \"Operator 'Export Particles' not found in tyFlow editor.\")\n if invalid_object or invalid_operator:\n raise PublishValidationError(\n \"issues occurred\",\n description=\"Container should only include tyFlow object \"\n \"and tyflow operator 'Export Particle' should be in \"\n \"the tyFlow editor.\")\n\n def get_tyflow_object(self, instance):\n \"\"\"Get the nodes which are not tyFlow object(s)\n and editable mesh(es)\n\n Args:\n instance (pyblish.api.Instance): instance\n\n Returns:\n list: invalid nodes which are not tyFlow\n object(s) and editable mesh(es).\n \"\"\"\n container = instance.data[\"instance_node\"]\n self.log.debug(f\"Validating tyFlow container for {container}\")\n\n allowed_classes = [rt.tyFlow, rt.Editable_Mesh]\n return [\n member for member in instance.data[\"members\"]\n if rt.ClassOf(member) not in allowed_classes\n ]\n\n def get_tyflow_operator(self, instance):\n \"\"\"Check if the Export Particle Operators in the node\n connections.\n\n Args:\n instance (str): instance node\n\n Returns:\n invalid(list): list of invalid nodes which do\n not consist of Export Particle Operators as parts\n of the node connections\n \"\"\"\n invalid = []\n members = instance.data[\"members\"]\n for member in members:\n obj = member.baseobject\n\n # There must be at least one animation with export\n # particles enabled\n has_export_particles = False\n anim_names = rt.GetSubAnimNames(obj)\n for anim_name in anim_names:\n # get name of the related tyFlow node\n sub_anim = rt.GetSubAnim(obj, anim_name)\n # check if there is export particle operator\n if rt.IsProperty(sub_anim, \"Export_Particles\"):\n has_export_particles = True\n break\n\n if not has_export_particles:\n invalid.append(member)\n return invalid\n"
},
{
"path": "openpype/hosts/max/startup/startup.ms",
"content": "-- OpenPype Init Script\n(\n local sysPath = dotNetClass \"System.IO.Path\"\n\tlocal sysDir = dotNetClass \"System.IO.Directory\"\n\tlocal localScript = getThisScriptFilename()\n local startup = sysPath.Combine (sysPath.GetDirectoryName localScript) \"startup.py\"\n\n local pythonpath = systemTools.getEnvVariable \"MAX_PYTHONPATH\"\n systemTools.setEnvVariable \"PYTHONPATH\" pythonpath\n \n /*opens the create menu on startup to ensure users are presented with a useful default view.*/\n max create mode\n\n python.ExecuteFile startup\n)\n"
},
{
"path": "openpype/hosts/max/startup/startup.py",
"content": "# -*- coding: utf-8 -*-\nimport os\nimport sys\n\n# this might happen in some 3dsmax version where PYTHONPATH isn't added\n# to sys.path automatically\nfor path in os.environ[\"PYTHONPATH\"].split(os.pathsep):\n if path and path not in sys.path:\n sys.path.append(path)\n\nfrom openpype.hosts.max.api import MaxHost\nfrom openpype.pipeline import install_host\n\nhost = MaxHost()\ninstall_host(host)\n"
},
{
"path": "openpype/hosts/maya/__init__.py",
"content": "from .addon import (\n MayaAddon,\n MAYA_ROOT_DIR,\n)\n\n\n__all__ = (\n \"MayaAddon\",\n \"MAYA_ROOT_DIR\",\n)\n"
},
{
"path": "openpype/hosts/maya/addon.py",
"content": "import os\nfrom openpype.modules import OpenPypeModule, IHostAddon\n\nMAYA_ROOT_DIR = os.path.dirname(os.path.abspath(__file__))\n\n\nclass MayaAddon(OpenPypeModule, IHostAddon):\n name = \"maya\"\n host_name = \"maya\"\n\n def initialize(self, module_settings):\n self.enabled = True\n\n def add_implementation_envs(self, env, _app):\n # Add requirements to PYTHONPATH\n new_python_paths = [\n os.path.join(MAYA_ROOT_DIR, \"startup\")\n ]\n old_python_path = env.get(\"PYTHONPATH\") or \"\"\n for path in old_python_path.split(os.pathsep):\n if not path:\n continue\n\n norm_path = os.path.normpath(path)\n if norm_path not in new_python_paths:\n new_python_paths.append(norm_path)\n\n env[\"PYTHONPATH\"] = os.pathsep.join(new_python_paths)\n\n # Set default environments\n envs = {\n \"OPENPYPE_LOG_NO_COLORS\": \"Yes\",\n # For python module 'qtpy'\n \"QT_API\": \"PySide2\",\n # For python module 'Qt'\n \"QT_PREFERRED_BINDING\": \"PySide2\"\n }\n for key, value in envs.items():\n env[key] = value\n\n def get_launch_hook_paths(self, app):\n if app.host_name != self.host_name:\n return []\n return [\n os.path.join(MAYA_ROOT_DIR, \"hooks\")\n ]\n\n def get_workfile_extensions(self):\n return [\".ma\", \".mb\"]\n"
},
{
"path": "openpype/hosts/maya/api/__init__.py",
"content": "\"\"\"Public API\n\nAnything that isn't defined here is INTERNAL and unreliable for external use.\n\n\"\"\"\n\nfrom .pipeline import (\n uninstall,\n\n ls,\n containerise,\n MayaHost,\n)\nfrom .plugin import (\n Creator,\n Loader\n)\n\nfrom .workio import (\n open_file,\n save_file,\n current_file,\n has_unsaved_changes,\n file_extensions,\n work_root\n)\n\nfrom .lib import (\n lsattr,\n lsattrs,\n read,\n\n apply_shaders,\n maintained_selection,\n suspended_refresh,\n\n unique_namespace,\n)\n\n\n__all__ = [\n \"uninstall\",\n\n \"ls\",\n \"containerise\",\n \"MayaHost\",\n\n \"Creator\",\n \"Loader\",\n\n # Workfiles API\n \"open_file\",\n \"save_file\",\n \"current_file\",\n \"has_unsaved_changes\",\n \"file_extensions\",\n \"work_root\",\n\n # Utility functions\n \"lsattr\",\n \"lsattrs\",\n \"read\",\n\n \"unique_namespace\",\n\n \"apply_shaders\",\n \"maintained_selection\",\n \"suspended_refresh\",\n\n]\n\n# Backwards API compatibility\nopen = open_file\nsave = save_file\n"
},
{
"path": "openpype/hosts/maya/api/action.py",
"content": "# absolute_import is needed to counter the `module has no cmds error` in Maya\nfrom __future__ import absolute_import\n\nimport pyblish.api\n\nfrom openpype.client import get_asset_by_name\nfrom openpype.pipeline.publish import get_errored_instances_from_context\n\n\nclass GenerateUUIDsOnInvalidAction(pyblish.api.Action):\n \"\"\"Generate UUIDs on the invalid nodes in the instance.\n\n Invalid nodes are those returned by the plugin's `get_invalid` method.\n As such it is the plug-in's responsibility to ensure the nodes that\n receive new UUIDs are actually invalid.\n\n Requires:\n - instance.data[\"asset\"]\n\n \"\"\"\n\n label = \"Regenerate UUIDs\"\n on = \"failed\" # This action is only available on a failed plug-in\n icon = \"wrench\" # Icon from Awesome Icon\n\n def process(self, context, plugin):\n\n from maya import cmds\n\n self.log.info(\"Finding bad nodes..\")\n\n errored_instances = get_errored_instances_from_context(context)\n\n # Apply pyblish logic to get the instances for the plug-in\n instances = pyblish.api.instances_by_plugin(errored_instances, plugin)\n\n # Get the nodes from the all instances that ran through this plug-in\n all_invalid = []\n for instance in instances:\n invalid = plugin.get_invalid(instance)\n\n # Don't allow referenced nodes to get their ids regenerated to\n # avoid loaded content getting messed up with reference edits\n if invalid:\n referenced = {node for node in invalid if\n cmds.referenceQuery(node, isNodeReferenced=True)}\n if referenced:\n self.log.warning(\"Skipping UUID generation on referenced \"\n \"nodes: {}\".format(list(referenced)))\n invalid = [node for node in invalid\n if node not in referenced]\n\n if invalid:\n\n self.log.info(\"Fixing instance {}\".format(instance.name))\n self._update_id_attribute(instance, invalid)\n\n all_invalid.extend(invalid)\n\n if not all_invalid:\n self.log.info(\"No invalid nodes found.\")\n return\n\n all_invalid = list(set(all_invalid))\n self.log.info(\"Generated ids on nodes: {0}\".format(all_invalid))\n\n def _update_id_attribute(self, instance, nodes):\n \"\"\"Delete the id attribute\n\n Args:\n instance: The instance we're fixing for\n nodes (list): all nodes to regenerate ids on\n \"\"\"\n\n from . import lib\n\n # Expecting this is called on validators in which case 'assetEntity'\n # should be always available, but kept a way to query it by name.\n asset_doc = instance.data.get(\"assetEntity\")\n if not asset_doc:\n asset_name = instance.data[\"asset\"]\n project_name = instance.context.data[\"projectName\"]\n self.log.info((\n \"Asset is not stored on instance.\"\n \" Querying by name \\\"{}\\\" from project \\\"{}\\\"\"\n ).format(asset_name, project_name))\n asset_doc = get_asset_by_name(\n project_name, asset_name, fields=[\"_id\"]\n )\n\n for node, _id in lib.generate_ids(nodes, asset_id=asset_doc[\"_id\"]):\n lib.set_id(node, _id, overwrite=True)\n\n\nclass SelectInvalidAction(pyblish.api.Action):\n \"\"\"Select invalid nodes in Maya when plug-in failed.\n\n To retrieve the invalid nodes this assumes a static `get_invalid()`\n method is available on the plugin.\n\n \"\"\"\n label = \"Select invalid\"\n on = \"failed\" # This action is only available on a failed plug-in\n icon = \"search\" # Icon from Awesome Icon\n\n def process(self, context, plugin):\n\n try:\n from maya import cmds\n except ImportError:\n raise ImportError(\"Current host is not Maya\")\n\n errored_instances = get_errored_instances_from_context(context,\n plugin=plugin)\n\n # Get the invalid nodes for the plug-ins\n self.log.info(\"Finding invalid nodes..\")\n invalid = list()\n for instance in errored_instances:\n invalid_nodes = plugin.get_invalid(instance)\n if invalid_nodes:\n if isinstance(invalid_nodes, (list, tuple)):\n invalid.extend(invalid_nodes)\n else:\n self.log.warning(\"Plug-in returned to be invalid, \"\n \"but has no selectable nodes.\")\n\n # Ensure unique (process each node only once)\n invalid = list(set(invalid))\n\n if invalid:\n self.log.info(\"Selecting invalid nodes: %s\" % \", \".join(invalid))\n cmds.select(invalid, replace=True, noExpand=True)\n else:\n self.log.info(\"No invalid nodes found.\")\n cmds.select(deselect=True)\n"
},
{
"path": "openpype/hosts/maya/api/alembic.py",
"content": "import json\nimport logging\nimport os\n\nfrom maya import cmds # noqa\n\nfrom openpype.hosts.maya.api.lib import evaluation\n\nlog = logging.getLogger(__name__)\n\n# The maya alembic export types\nALEMBIC_ARGS = {\n \"attr\": (list, tuple),\n \"attrPrefix\": (list, tuple),\n \"autoSubd\": bool,\n \"dataFormat\": str,\n \"dontSkipUnwrittenFrames\": bool,\n \"endFrame\": float,\n \"eulerFilter\": bool,\n \"frameRange\": str, # \"start end\"; overrides startFrame & endFrame\n \"frameRelativeSample\": float,\n \"melPerFrameCallback\": str,\n \"melPostJobCallback\": str,\n \"noNormals\": bool,\n \"preRoll\": bool,\n \"preRollStartFrame\": int,\n \"pythonPerFrameCallback\": str,\n \"pythonPostJobCallback\": str,\n \"renderableOnly\": bool,\n \"root\": (list, tuple),\n \"selection\": bool,\n \"startFrame\": float,\n \"step\": float,\n \"stripNamespaces\": bool,\n \"userAttr\": (list, tuple),\n \"userAttrPrefix\": (list, tuple),\n \"uvWrite\": bool,\n \"uvsOnly\": bool,\n \"verbose\": bool,\n \"wholeFrameGeo\": bool,\n \"worldSpace\": bool,\n \"writeColorSets\": bool,\n \"writeCreases\": bool, # Maya 2015 Ext1+\n \"writeFaceSets\": bool,\n \"writeUVSets\": bool, # Maya 2017+\n \"writeVisibility\": bool,\n}\n\n\ndef extract_alembic(\n file,\n attr=None,\n attrPrefix=None,\n dataFormat=\"ogawa\",\n endFrame=None,\n eulerFilter=True,\n frameRange=\"\",\n noNormals=False,\n preRoll=False,\n preRollStartFrame=0,\n renderableOnly=False,\n root=None,\n selection=True,\n startFrame=None,\n step=1.0,\n stripNamespaces=True,\n uvWrite=True,\n verbose=False,\n wholeFrameGeo=False,\n worldSpace=False,\n writeColorSets=False,\n writeCreases=False,\n writeNormals=False,\n writeFaceSets=False,\n writeUVSets=False,\n writeVisibility=False\n):\n \"\"\"Extract a single Alembic Cache.\n\n This extracts an Alembic cache using the `-selection` flag to minimize\n the extracted content to solely what was Collected into the instance.\n\n Arguments:\n\n startFrame (float): Start frame of output. Ignored if `frameRange`\n provided.\n\n endFrame (float): End frame of output. Ignored if `frameRange`\n provided.\n\n frameRange (tuple or str): Two-tuple with start and end frame or a\n string formatted as: \"startFrame endFrame\". This argument\n overrides `startFrame` and `endFrame` arguments.\n\n eulerFilter (bool): When on, X, Y, and Z rotation data is filtered with\n an Euler filter. Euler filtering helps resolve irregularities in\n rotations especially if X, Y, and Z rotations exceed 360 degrees.\n Defaults to True.\n\n noNormals (bool): When on, normal data from the original polygon\n objects is not included in the exported Alembic cache file.\n\n preRoll (bool): This frame range will not be sampled.\n Defaults to False.\n\n renderableOnly (bool): When on, any non-renderable nodes or hierarchy,\n such as hidden objects, are not included in the Alembic file.\n Defaults to False.\n\n selection (bool): Write out all all selected nodes from the\n active selection list that are descendents of the roots specified\n with -root. Defaults to False.\n\n uvWrite (bool): When on, UV data from polygon meshes and subdivision\n objects are written to the Alembic file. Only the current UV map is\n included.\n\n writeColorSets (bool): Write all color sets on MFnMeshes as\n color 3 or color 4 indexed geometry parameters with face varying\n scope. Defaults to False.\n\n writeFaceSets (bool): Write all Face sets on MFnMeshes.\n Defaults to False.\n\n wholeFrameGeo (bool): Data for geometry will only be written\n out on whole frames. Defaults to False.\n\n worldSpace (bool): When on, the top node in the node hierarchy is\n stored as world space. By default, these nodes are stored as local\n space. Defaults to False.\n\n writeVisibility (bool): Visibility state will be stored in\n the Alembic file. Otherwise everything written out is treated as\n visible. Defaults to False.\n\n writeUVSets (bool): Write all uv sets on MFnMeshes as vector\n 2 indexed geometry parameters with face varying scope. Defaults to\n False.\n\n writeCreases (bool): If the mesh has crease edges or crease\n vertices, the mesh (OPolyMesh) would now be written out as an OSubD\n and crease info will be stored in the Alembic file. Otherwise,\n creases info won't be preserved in Alembic file unless a custom\n Boolean attribute SubDivisionMesh has been added to mesh node and\n its value is true. Defaults to False.\n\n dataFormat (str): The data format to use for the cache,\n defaults to \"ogawa\"\n\n step (float): The time interval (expressed in frames) at\n which the frame range is sampled. Additional samples around each\n frame can be specified with -frs. Defaults to 1.0.\n\n attr (list of str, optional): A specific geometric attribute to write\n out. Defaults to [].\n\n attrPrefix (list of str, optional): Prefix filter for determining which\n geometric attributes to write out. Defaults to [\"ABC_\"].\n\n root (list of str): Maya dag path which will be parented to\n the root of the Alembic file. Defaults to [], which means the\n entire scene will be written out.\n\n stripNamespaces (bool): When on, any namespaces associated with the\n exported objects are removed from the Alembic file. For example, an\n object with the namespace taco:foo:bar appears as bar in the\n Alembic file.\n\n verbose (bool): When on, outputs frame number information to the\n Script Editor or output window during extraction.\n\n preRollStartFrame (float): The frame to start scene\n evaluation at. This is used to set the starting frame for time\n dependent translations and can be used to evaluate run-up that\n isn't actually translated. Defaults to 0.\n \"\"\"\n\n # Ensure alembic exporter is loaded\n cmds.loadPlugin('AbcExport', quiet=True)\n\n # Alembic Exporter requires forward slashes\n file = file.replace('\\\\', '/')\n\n # Ensure list arguments are valid.\n attr = attr or []\n attrPrefix = attrPrefix or []\n root = root or []\n\n # Pass the start and end frame on as `frameRange` so that it\n # never conflicts with that argument\n if not frameRange:\n # Fallback to maya timeline if no start or end frame provided.\n if startFrame is None:\n startFrame = cmds.playbackOptions(query=True,\n animationStartTime=True)\n if endFrame is None:\n endFrame = cmds.playbackOptions(query=True,\n animationEndTime=True)\n\n # Ensure valid types are converted to frame range\n assert isinstance(startFrame, ALEMBIC_ARGS[\"startFrame\"])\n assert isinstance(endFrame, ALEMBIC_ARGS[\"endFrame\"])\n frameRange = \"{0} {1}\".format(startFrame, endFrame)\n else:\n # Allow conversion from tuple for `frameRange`\n if isinstance(frameRange, (list, tuple)):\n assert len(frameRange) == 2\n frameRange = \"{0} {1}\".format(frameRange[0], frameRange[1])\n\n # Assemble options\n options = {\n \"selection\": selection,\n \"frameRange\": frameRange,\n \"eulerFilter\": eulerFilter,\n \"noNormals\": noNormals,\n \"preRoll\": preRoll,\n \"renderableOnly\": renderableOnly,\n \"uvWrite\": uvWrite,\n \"writeColorSets\": writeColorSets,\n \"writeFaceSets\": writeFaceSets,\n \"wholeFrameGeo\": wholeFrameGeo,\n \"worldSpace\": worldSpace,\n \"writeVisibility\": writeVisibility,\n \"writeUVSets\": writeUVSets,\n \"writeCreases\": writeCreases,\n \"dataFormat\": dataFormat,\n \"step\": step,\n \"attr\": attr,\n \"attrPrefix\": attrPrefix,\n \"stripNamespaces\": stripNamespaces,\n \"verbose\": verbose,\n \"preRollStartFrame\": preRollStartFrame\n }\n\n # Validate options\n for key, value in options.copy().items():\n\n # Discard unknown options\n if key not in ALEMBIC_ARGS:\n log.warning(\"extract_alembic() does not support option '%s'. \"\n \"Flag will be ignored..\", key)\n options.pop(key)\n continue\n\n # Validate value type\n valid_types = ALEMBIC_ARGS[key]\n if not isinstance(value, valid_types):\n raise TypeError(\"Alembic option unsupported type: \"\n \"{0} (expected {1})\".format(value, valid_types))\n\n # Ignore empty values, like an empty string, since they mess up how\n # job arguments are built\n if isinstance(value, (list, tuple)):\n value = [x for x in value if x.strip()]\n\n # Ignore option completely if no values remaining\n if not value:\n options.pop(key)\n continue\n\n options[key] = value\n\n # The `writeCreases` argument was changed to `autoSubd` in Maya 2018+\n maya_version = int(cmds.about(version=True))\n if maya_version >= 2018:\n options['autoSubd'] = options.pop('writeCreases', False)\n\n # Format the job string from options\n job_args = list()\n for key, value in options.items():\n if isinstance(value, (list, tuple)):\n for entry in value:\n job_args.append(\"-{} {}\".format(key, entry))\n elif isinstance(value, bool):\n # Add only when state is set to True\n if value:\n job_args.append(\"-{0}\".format(key))\n else:\n job_args.append(\"-{0} {1}\".format(key, value))\n\n job_str = \" \".join(job_args)\n job_str += ' -file \"%s\"' % file\n\n # Ensure output directory exists\n parent_dir = os.path.dirname(file)\n if not os.path.exists(parent_dir):\n os.makedirs(parent_dir)\n\n if verbose:\n log.debug(\"Preparing Alembic export with options: %s\",\n json.dumps(options, indent=4))\n log.debug(\"Extracting Alembic with job arguments: %s\", job_str)\n\n # Perform extraction\n print(\"Alembic Job Arguments : {}\".format(job_str))\n\n # Disable the parallel evaluation temporarily to ensure no buggy\n # exports are made. (PLN-31)\n # TODO: Make sure this actually fixes the issues\n with evaluation(\"off\"):\n cmds.AbcExport(j=job_str, verbose=verbose)\n\n if verbose:\n log.debug(\"Extracted Alembic to: %s\", file)\n\n return file\n"
},
{
"path": "openpype/hosts/maya/api/commands.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"OpenPype script commands to be used directly in Maya.\"\"\"\nfrom maya import cmds\n\nfrom openpype.client import get_asset_by_name, get_project\nfrom openpype.pipeline import get_current_project_name, get_current_asset_name\n\n\nclass ToolWindows:\n\n _windows = {}\n\n @classmethod\n def get_window(cls, tool):\n \"\"\"Get widget for specific tool.\n\n Args:\n tool (str): Name of the tool.\n\n Returns:\n Stored widget.\n\n \"\"\"\n try:\n return cls._windows[tool]\n except KeyError:\n return None\n\n @classmethod\n def set_window(cls, tool, window):\n \"\"\"Set widget for the tool.\n\n Args:\n tool (str): Name of the tool.\n window (QtWidgets.QWidget): Widget\n\n \"\"\"\n cls._windows[tool] = window\n\n\ndef edit_shader_definitions():\n from qtpy import QtWidgets\n from openpype.hosts.maya.api.shader_definition_editor import (\n ShaderDefinitionsEditor\n )\n from openpype.tools.utils import qt_app_context\n\n top_level_widgets = QtWidgets.QApplication.topLevelWidgets()\n main_window = next(widget for widget in top_level_widgets\n if widget.objectName() == \"MayaWindow\")\n\n with qt_app_context():\n window = ToolWindows.get_window(\"shader_definition_editor\")\n if not window:\n window = ShaderDefinitionsEditor(parent=main_window)\n ToolWindows.set_window(\"shader_definition_editor\", window)\n window.show()\n\n\ndef _resolution_from_document(doc):\n if not doc or \"data\" not in doc:\n print(\"Entered document is not valid. \\\"{}\\\"\".format(str(doc)))\n return None\n\n resolution_width = doc[\"data\"].get(\"resolutionWidth\")\n resolution_height = doc[\"data\"].get(\"resolutionHeight\")\n # Backwards compatibility\n if resolution_width is None or resolution_height is None:\n resolution_width = doc[\"data\"].get(\"resolution_width\")\n resolution_height = doc[\"data\"].get(\"resolution_height\")\n\n # Make sure both width and height are set\n if resolution_width is None or resolution_height is None:\n cmds.warning(\n \"No resolution information found for \\\"{}\\\"\".format(doc[\"name\"])\n )\n return None\n\n return int(resolution_width), int(resolution_height)\n\n\ndef reset_resolution():\n # Default values\n resolution_width = 1920\n resolution_height = 1080\n\n # Get resolution from asset\n project_name = get_current_project_name()\n asset_name = get_current_asset_name()\n asset_doc = get_asset_by_name(project_name, asset_name)\n resolution = _resolution_from_document(asset_doc)\n # Try get resolution from project\n if resolution is None:\n # TODO go through visualParents\n print((\n \"Asset \\\"{}\\\" does not have set resolution.\"\n \" Trying to get resolution from project\"\n ).format(asset_name))\n project_doc = get_project(project_name)\n resolution = _resolution_from_document(project_doc)\n\n if resolution is None:\n msg = \"Using default resolution {}x{}\"\n else:\n resolution_width, resolution_height = resolution\n msg = \"Setting resolution to {}x{}\"\n\n print(msg.format(resolution_width, resolution_height))\n\n # set for different renderers\n # arnold, vray, redshift, renderman\n\n renderer = cmds.getAttr(\"defaultRenderGlobals.currentRenderer\").lower()\n # handle various renderman names\n if renderer.startswith(\"renderman\"):\n renderer = \"renderman\"\n\n # default attributes are usable for Arnold, Renderman and Redshift\n width_attr_name = \"defaultResolution.width\"\n height_attr_name = \"defaultResolution.height\"\n\n # Vray has its own way\n if renderer == \"vray\":\n width_attr_name = \"vraySettings.width\"\n height_attr_name = \"vraySettings.height\"\n\n cmds.setAttr(width_attr_name, resolution_width)\n cmds.setAttr(height_attr_name, resolution_height)\n"
},
{
"path": "openpype/hosts/maya/api/customize.py",
"content": "\"\"\"A set of commands that install overrides to Maya's UI\"\"\"\n\nimport os\nimport logging\n\nfrom functools import partial\n\nimport maya.cmds as cmds\nimport maya.mel as mel\n\nfrom openpype import resources\nfrom openpype.tools.utils import host_tools\nfrom .lib import get_main_window\nfrom ..tools import show_look_assigner\n\nlog = logging.getLogger(__name__)\n\nCOMPONENT_MASK_ORIGINAL = {}\n\n\ndef override_component_mask_commands():\n \"\"\"Override component mask ctrl+click behavior.\n\n This implements special behavior for Maya's component\n mask menu items where a ctrl+click will instantly make\n it an isolated behavior disabling all others.\n\n Tested in Maya 2016 and 2018\n\n \"\"\"\n log.info(\"Installing override_component_mask_commands..\")\n\n # Get all object mask buttons\n buttons = cmds.formLayout(\"objectMaskIcons\",\n query=True,\n childArray=True)\n # Skip the triangle list item\n buttons = [btn for btn in buttons if btn != \"objPickMenuLayout\"]\n\n def on_changed_callback(raw_command, state):\n \"\"\"New callback\"\"\"\n\n # If \"control\" is held force the toggled one to on and\n # toggle the others based on whether any of the buttons\n # was remaining active after the toggle, if not then\n # enable all\n if cmds.getModifiers() == 4: # = CTRL\n state = True\n active = [cmds.iconTextCheckBox(btn, query=True, value=True)\n for btn in buttons]\n if any(active):\n cmds.selectType(allObjects=False)\n else:\n cmds.selectType(allObjects=True)\n\n # Replace #1 with the current button state\n cmd = raw_command.replace(\" #1\", \" {}\".format(int(state)))\n mel.eval(cmd)\n\n for btn in buttons:\n\n # Store a reference to the original command so that if\n # we rerun this override command it doesn't recursively\n # try to implement the fix. (This also allows us to\n # \"uninstall\" the behavior later)\n if btn not in COMPONENT_MASK_ORIGINAL:\n original = cmds.iconTextCheckBox(btn, query=True, cc=True)\n COMPONENT_MASK_ORIGINAL[btn] = original\n\n # Assign the special callback\n original = COMPONENT_MASK_ORIGINAL[btn]\n new_fn = partial(on_changed_callback, original)\n cmds.iconTextCheckBox(btn, edit=True, cc=new_fn)\n\n\ndef override_toolbox_ui():\n \"\"\"Add custom buttons in Toolbox as replacement for Maya web help icon.\"\"\"\n icons = resources.get_resource(\"icons\")\n parent_widget = get_main_window()\n\n # Ensure the maya web icon on toolbox exists\n button_names = [\n # Maya 2022.1+ with maya.cmds.iconTextStaticLabel\n \"ToolBox|MainToolboxLayout|mayaHomeToolboxButton\",\n # Older with maya.cmds.iconTextButton\n \"ToolBox|MainToolboxLayout|mayaWebButton\"\n ]\n for name in button_names:\n if cmds.control(name, query=True, exists=True):\n web_button = name\n break\n else:\n # Button does not exist\n log.warning(\"Can't find Maya Home/Web button to override toolbox ui..\")\n return\n\n cmds.control(web_button, edit=True, visible=False)\n\n # real = 32, but 36 with padding - according to toolbox mel script\n icon_size = 36\n parent = web_button.rsplit(\"|\", 1)[0]\n\n # Ensure the parent is a formLayout\n if not cmds.objectTypeUI(parent) == \"formLayout\":\n return\n\n # Create our controls\n controls = []\n\n controls.append(\n cmds.iconTextButton(\n \"pype_toolbox_lookmanager\",\n annotation=\"Look Manager\",\n label=\"Look Manager\",\n image=os.path.join(icons, \"lookmanager.png\"),\n command=show_look_assigner,\n width=icon_size,\n height=icon_size,\n parent=parent\n )\n )\n\n controls.append(\n cmds.iconTextButton(\n \"pype_toolbox_workfiles\",\n annotation=\"Work Files\",\n label=\"Work Files\",\n image=os.path.join(icons, \"workfiles.png\"),\n command=lambda: host_tools.show_workfiles(\n parent=parent_widget\n ),\n width=icon_size,\n height=icon_size,\n parent=parent\n )\n )\n\n controls.append(\n cmds.iconTextButton(\n \"pype_toolbox_loader\",\n annotation=\"Loader\",\n label=\"Loader\",\n image=os.path.join(icons, \"loader.png\"),\n command=lambda: host_tools.show_loader(\n parent=parent_widget, use_context=True\n ),\n width=icon_size,\n height=icon_size,\n parent=parent\n )\n )\n\n controls.append(\n cmds.iconTextButton(\n \"pype_toolbox_manager\",\n annotation=\"Inventory\",\n label=\"Inventory\",\n image=os.path.join(icons, \"inventory.png\"),\n command=lambda: host_tools.show_scene_inventory(\n parent=parent_widget\n ),\n width=icon_size,\n height=icon_size,\n parent=parent\n )\n )\n\n # Add the buttons on the bottom and stack\n # them above each other with side padding\n controls.reverse()\n for i, control in enumerate(controls):\n previous = controls[i - 1] if i > 0 else web_button\n\n cmds.formLayout(parent, edit=True,\n attachControl=[control, \"bottom\", 0, previous],\n attachForm=([control, \"left\", 1],\n [control, \"right\", 1]))\n"
},
{
"path": "openpype/hosts/maya/api/exitstack.py",
"content": "\"\"\"Backwards compatible implementation of ExitStack for Python 2.\n\nExitStack contextmanager was implemented with Python 3.3.\nAs long as we supportPython 2 hosts we can use this backwards\ncompatible implementation to support bothPython 2 and Python 3.\n\nInstead of using ExitStack from contextlib, use it from this module:\n\n>>> from openpype.hosts.maya.api.exitstack import ExitStack\n\nIt will provide the appropriate ExitStack implementation for the current\nrunning Python version.\n\n\"\"\"\n# TODO: Remove the entire script once dropping Python 2 support.\nimport contextlib\nif getattr(contextlib, \"nested\", None):\n from contextlib import ExitStack # noqa\nelse:\n import sys\n from collections import deque\n\n class ExitStack(object):\n\n \"\"\"Context manager for dynamic management of a stack of exit callbacks\n\n For example:\n\n with ExitStack() as stack:\n files = [stack.enter_context(open(fname))\n for fname in filenames]\n # All opened files will automatically be closed at the end of\n # the with statement, even if attempts to open files later\n # in the list raise an exception\n\n \"\"\"\n def __init__(self):\n self._exit_callbacks = deque()\n\n def pop_all(self):\n \"\"\"Preserve the context stack by transferring\n it to a new instance\"\"\"\n new_stack = type(self)()\n new_stack._exit_callbacks = self._exit_callbacks\n self._exit_callbacks = deque()\n return new_stack\n\n def _push_cm_exit(self, cm, cm_exit):\n \"\"\"Helper to correctly register callbacks\n to __exit__ methods\"\"\"\n def _exit_wrapper(*exc_details):\n return cm_exit(cm, *exc_details)\n _exit_wrapper.__self__ = cm\n self.push(_exit_wrapper)\n\n def push(self, exit):\n \"\"\"Registers a callback with the standard __exit__ method signature\n\n Can suppress exceptions the same way __exit__ methods can.\n\n Also accepts any object with an __exit__ method (registering a call\n to the method instead of the object itself)\n \"\"\"\n # We use an unbound method rather than a bound method to follow\n # the standard lookup behaviour for special methods\n _cb_type = type(exit)\n try:\n exit_method = _cb_type.__exit__\n except AttributeError:\n # Not a context manager, so assume its a callable\n self._exit_callbacks.append(exit)\n else:\n self._push_cm_exit(exit, exit_method)\n return exit # Allow use as a decorator\n\n def callback(self, callback, *args, **kwds):\n \"\"\"Registers an arbitrary callback and arguments.\n\n Cannot suppress exceptions.\n \"\"\"\n def _exit_wrapper(exc_type, exc, tb):\n callback(*args, **kwds)\n # We changed the signature, so using @wraps is not appropriate, but\n # setting __wrapped__ may still help with introspection\n _exit_wrapper.__wrapped__ = callback\n self.push(_exit_wrapper)\n return callback # Allow use as a decorator\n\n def enter_context(self, cm):\n \"\"\"Enters the supplied context manager\n\n If successful, also pushes its __exit__ method as a callback and\n returns the result of the __enter__ method.\n \"\"\"\n # We look up the special methods on the type to\n # match the with statement\n _cm_type = type(cm)\n _exit = _cm_type.__exit__\n result = _cm_type.__enter__(cm)\n self._push_cm_exit(cm, _exit)\n return result\n\n def close(self):\n \"\"\"Immediately unwind the context stack\"\"\"\n self.__exit__(None, None, None)\n\n def __enter__(self):\n return self\n\n def __exit__(self, *exc_details):\n # We manipulate the exception state so it behaves as though\n # we were actually nesting multiple with statements\n frame_exc = sys.exc_info()[1]\n\n def _fix_exception_context(new_exc, old_exc):\n while 1:\n exc_context = new_exc.__context__\n if exc_context in (None, frame_exc):\n break\n new_exc = exc_context\n new_exc.__context__ = old_exc\n\n # Callbacks are invoked in LIFO order to match the behaviour of\n # nested context managers\n suppressed_exc = False\n while self._exit_callbacks:\n cb = self._exit_callbacks.pop()\n try:\n if cb(*exc_details):\n suppressed_exc = True\n exc_details = (None, None, None)\n except Exception:\n new_exc_details = sys.exc_info()\n # simulate the stack of exceptions by setting the context\n _fix_exception_context(new_exc_details[1], exc_details[1])\n if not self._exit_callbacks:\n raise\n exc_details = new_exc_details\n return suppressed_exc\n"
},
{
"path": "openpype/hosts/maya/api/fbx.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Tools to work with FBX.\"\"\"\nimport logging\n\nfrom pyblish.api import Instance\n\nfrom maya import cmds # noqa\nimport maya.mel as mel # noqa\nfrom openpype.hosts.maya.api.lib import maintained_selection\n\n\nclass FBXExtractor:\n \"\"\"Extract FBX from Maya.\n\n This extracts reproducible FBX exports ignoring any of the settings set\n on the local machine in the FBX export options window.\n\n All export settings are applied with the `FBXExport*` commands prior\n to the `FBXExport` call itself. The options can be overridden with\n their\n nice names as seen in the \"options\" property on this class.\n\n For more information on FBX exports see:\n - https://knowledge.autodesk.com/support/maya/learn-explore/caas\n /CloudHelp/cloudhelp/2016/ENU/Maya/files/GUID-6CCE943A-2ED4-4CEE-96D4\n -9CB19C28F4E0-htm.html\n - http://forums.cgsociety.org/archive/index.php?t-1032853.html\n - https://groups.google.com/forum/#!msg/python_inside_maya/cLkaSo361oE\n /LKs9hakE28kJ\n\n \"\"\"\n @property\n def options(self):\n \"\"\"Overridable options for FBX Export\n\n Given in the following format\n - {NAME: EXPECTED TYPE}\n\n If the overridden option's type does not match,\n the option is not included and a warning is logged.\n\n \"\"\"\n\n return {\n \"cameras\": bool,\n \"smoothingGroups\": bool,\n \"hardEdges\": bool,\n \"tangents\": bool,\n \"smoothMesh\": bool,\n \"instances\": bool,\n # \"referencedContainersContent\": bool, # deprecated in Maya 2016+\n \"bakeComplexAnimation\": int,\n \"bakeComplexStart\": int,\n \"bakeComplexEnd\": int,\n \"bakeComplexStep\": int,\n \"bakeResampleAnimation\": bool,\n \"useSceneName\": bool,\n \"quaternion\": str, # \"euler\"\n \"shapes\": bool,\n \"skins\": bool,\n \"constraints\": bool,\n \"lights\": bool,\n \"embeddedTextures\": bool,\n \"inputConnections\": bool,\n \"upAxis\": str, # x, y or z,\n \"triangulate\": bool,\n \"fileVersion\": str,\n \"skeletonDefinitions\": bool,\n \"referencedAssetsContent\": bool\n }\n\n @property\n def default_options(self):\n \"\"\"The default options for FBX extraction.\n\n This includes shapes, skins, constraints, lights and incoming\n connections and exports with the Y-axis as up-axis.\n\n By default this uses the time sliders start and end time.\n\n \"\"\"\n\n start_frame = int(cmds.playbackOptions(query=True,\n animationStartTime=True))\n end_frame = int(cmds.playbackOptions(query=True,\n animationEndTime=True))\n\n return {\n \"cameras\": False,\n \"smoothingGroups\": True,\n \"hardEdges\": False,\n \"tangents\": False,\n \"smoothMesh\": True,\n \"instances\": False,\n \"bakeComplexAnimation\": True,\n \"bakeComplexStart\": start_frame,\n \"bakeComplexEnd\": end_frame,\n \"bakeComplexStep\": 1,\n \"bakeResampleAnimation\": True,\n \"useSceneName\": False,\n \"quaternion\": \"euler\",\n \"shapes\": True,\n \"skins\": True,\n \"constraints\": False,\n \"lights\": True,\n \"embeddedTextures\": False,\n \"inputConnections\": True,\n \"upAxis\": \"y\",\n \"triangulate\": False,\n \"fileVersion\": \"FBX202000\",\n \"skeletonDefinitions\": False,\n \"referencedAssetsContent\": False\n }\n\n def __init__(self, log=None):\n # Ensure FBX plug-in is loaded\n self.log = log or logging.getLogger(self.__class__.__name__)\n cmds.loadPlugin(\"fbxmaya\", quiet=True)\n\n def parse_overrides(self, instance, options):\n \"\"\"Inspect data of instance to determine overridden options\n\n An instance may supply any of the overridable options\n as data, the option is then added to the extraction.\n\n \"\"\"\n\n for key in instance.data:\n if key not in self.options:\n continue\n\n # Ensure the data is of correct type\n value = instance.data[key]\n if not isinstance(value, self.options[key]):\n self.log.warning(\n \"Overridden attribute {key} was of \"\n \"the wrong type: {invalid_type} \"\n \"- should have been {valid_type}\".format(\n key=key,\n invalid_type=type(value).__name__,\n valid_type=self.options[key].__name__))\n continue\n\n options[key] = value\n\n return options\n\n def set_options_from_instance(self, instance):\n # type: (Instance) -> None\n \"\"\"Sets FBX export options from data in the instance.\n\n Args:\n instance (Instance): Instance data.\n\n \"\"\"\n # Parse export options\n options = self.default_options\n options = self.parse_overrides(instance, options)\n self.log.debug(\"Export options: {0}\".format(options))\n\n # Collect the start and end including handles\n start = instance.data.get(\"frameStartHandle\") or \\\n instance.context.data.get(\"frameStartHandle\")\n end = instance.data.get(\"frameEndHandle\") or \\\n instance.context.data.get(\"frameEndHandle\")\n\n options['bakeComplexStart'] = start\n options['bakeComplexEnd'] = end\n\n # First apply the default export settings to be fully consistent\n # each time for successive publishes\n mel.eval(\"FBXResetExport\")\n\n # Apply the FBX overrides through MEL since the commands\n # only work correctly in MEL according to online\n # available discussions on the topic\n _iteritems = getattr(options, \"iteritems\", options.items)\n for option, value in _iteritems():\n key = option[0].upper() + option[1:] # uppercase first letter\n\n # Boolean must be passed as lower-case strings\n # as to MEL standards\n if isinstance(value, bool):\n value = str(value).lower()\n\n template = \"FBXExport{0} {1}\" if key == \"UpAxis\" else \\\n \"FBXExport{0} -v {1}\" # noqa\n cmd = template.format(key, value)\n self.log.debug(cmd)\n mel.eval(cmd)\n\n # Never show the UI or generate a log\n mel.eval(\"FBXExportShowUI -v false\")\n mel.eval(\"FBXExportGenerateLog -v false\")\n\n @staticmethod\n def export(members, path):\n # type: (list, str) -> None\n \"\"\"Export members as FBX with given path.\n\n Args:\n members (list): List of members to export.\n path (str): Path to use for export.\n\n \"\"\"\n # The export requires forward slashes because we need\n # to format it into a string in a mel expression\n path = path.replace(\"\\\\\", \"/\")\n with maintained_selection():\n cmds.select(members, r=True, noExpand=True)\n mel.eval('FBXExport -f \"{}\" -s'.format(path))\n"
},
{
"path": "openpype/hosts/maya/api/gltf.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Tools to work with GLTF.\"\"\"\nimport logging\n\nfrom maya import cmds, mel # noqa\n\nlog = logging.getLogger(__name__)\n\n_gltf_options = {\n \"of\": str, # outputFolder\n \"cpr\": str, # copyright\n \"sno\": bool, # selectedNodeOnly\n \"sn\": str, # sceneName\n \"glb\": bool, # binary\n \"nbu\": bool, # niceBufferURIs\n \"hbu\": bool, # hashBufferURI\n \"ext\": bool, # externalTextures\n \"ivt\": int, # initialValuesTime\n \"acn\": str, # animationClipName\n \"ast\": int, # animationClipStartTime\n \"aet\": int, # animationClipEndTime\n \"afr\": float, # animationClipFrameRate\n \"dsa\": int, # detectStepAnimations\n \"mpa\": str, # meshPrimitiveAttributes\n \"bpa\": str, # blendPrimitiveAttributes\n \"i32\": bool, # force32bitIndices\n \"ssm\": bool, # skipStandardMaterials\n \"eut\": bool, # excludeUnusedTexcoord\n \"dm\": bool, # defaultMaterial\n \"cm\": bool, # colorizeMaterials\n \"dmy\": str, # dumpMaya\n \"dgl\": str, # dumpGLTF\n \"imd\": str, # ignoreMeshDeformers\n \"ssc\": bool, # skipSkinClusters\n \"sbs\": bool, # skipBlendShapes\n \"rvp\": bool, # redrawViewport\n \"vno\": bool # visibleNodesOnly\n}\n\n\ndef extract_gltf(parent_dir,\n filename,\n **kwargs):\n\n \"\"\"Sets GLTF export options from data in the instance.\n\n \"\"\"\n\n cmds.loadPlugin('maya2glTF', quiet=True)\n # load the UI to run mel command\n mel.eval(\"maya2glTF_UI()\")\n\n parent_dir = parent_dir.replace('\\\\', '/')\n options = {\n \"dsa\": 1,\n \"glb\": True\n }\n options.update(kwargs)\n\n for key, value in options.copy().items():\n if key not in _gltf_options:\n log.warning(\"extract_gltf() does not support option '%s'. \"\n \"Flag will be ignored..\", key)\n options.pop(key)\n options.pop(value)\n continue\n\n job_args = list()\n default_opt = \"maya2glTF -of \\\"{0}\\\" -sn \\\"{1}\\\"\".format(parent_dir, filename) # noqa\n job_args.append(default_opt)\n\n for key, value in options.items():\n if isinstance(value, str):\n job_args.append(\"-{0} \\\"{1}\\\"\".format(key, value))\n elif isinstance(value, bool):\n if value:\n job_args.append(\"-{0}\".format(key))\n else:\n job_args.append(\"-{0} {1}\".format(key, value))\n\n job_str = \" \".join(job_args)\n log.info(\"{}\".format(job_str))\n mel.eval(job_str)\n\n # close the gltf export after finish the export\n gltf_UI = \"maya2glTF_exporter_window\"\n if cmds.window(gltf_UI, q=True, exists=True):\n cmds.deleteUI(gltf_UI)\n"
},
{
"path": "openpype/hosts/maya/api/lib.py",
"content": "\"\"\"Standalone helper functions\"\"\"\n\nimport os\nimport copy\nfrom pprint import pformat\nimport sys\nimport uuid\nimport re\n\nimport json\nimport logging\nimport contextlib\nimport capture\nfrom .exitstack import ExitStack\nfrom collections import OrderedDict, defaultdict\nfrom math import ceil\nfrom six import string_types\n\nfrom maya import cmds, mel\nfrom maya.api import OpenMaya\n\nfrom openpype.client import (\n get_project,\n get_asset_by_name,\n get_subsets,\n get_last_versions,\n get_representation_by_name\n)\nfrom openpype.settings import get_project_settings\nfrom openpype.pipeline import (\n get_current_project_name,\n get_current_asset_name,\n get_current_task_name,\n discover_loader_plugins,\n loaders_from_representation,\n get_representation_path,\n load_container,\n registered_host\n)\nfrom openpype.lib import NumberDef\nfrom openpype.pipeline.context_tools import get_current_project_asset\nfrom openpype.pipeline.create import CreateContext\nfrom openpype.lib.profiles_filtering import filter_profiles\n\n\nself = sys.modules[__name__]\nself._parent = None\n\nlog = logging.getLogger(__name__)\n\nIS_HEADLESS = not hasattr(cmds, \"about\") or cmds.about(batch=True)\nATTRIBUTE_DICT = {\"int\": {\"attributeType\": \"long\"},\n \"str\": {\"dataType\": \"string\"},\n \"unicode\": {\"dataType\": \"string\"},\n \"float\": {\"attributeType\": \"double\"},\n \"bool\": {\"attributeType\": \"bool\"}}\n\nSHAPE_ATTRS = {\"castsShadows\",\n \"receiveShadows\",\n \"motionBlur\",\n \"primaryVisibility\",\n \"smoothShading\",\n \"visibleInReflections\",\n \"visibleInRefractions\",\n \"doubleSided\",\n \"opposite\"}\n\n\nDEFAULT_MATRIX = [1.0, 0.0, 0.0, 0.0,\n 0.0, 1.0, 0.0, 0.0,\n 0.0, 0.0, 1.0, 0.0,\n 0.0, 0.0, 0.0, 1.0]\n\nINT_FPS = {15, 24, 25, 30, 48, 50, 60, 44100, 48000}\nFLOAT_FPS = {23.98, 23.976, 29.97, 47.952, 59.94}\n\n\nDISPLAY_LIGHTS_ENUM = [\n {\"label\": \"Use Project Settings\", \"value\": \"project_settings\"},\n {\"label\": \"Default Lighting\", \"value\": \"default\"},\n {\"label\": \"All Lights\", \"value\": \"all\"},\n {\"label\": \"Selected Lights\", \"value\": \"selected\"},\n {\"label\": \"Flat Lighting\", \"value\": \"flat\"},\n {\"label\": \"No Lights\", \"value\": \"none\"}\n]\n\n\ndef get_main_window():\n \"\"\"Acquire Maya's main window\"\"\"\n from qtpy import QtWidgets\n\n if self._parent is None:\n self._parent = {\n widget.objectName(): widget\n for widget in QtWidgets.QApplication.topLevelWidgets()\n }[\"MayaWindow\"]\n return self._parent\n\n\n@contextlib.contextmanager\ndef suspended_refresh(suspend=True):\n \"\"\"Suspend viewport refreshes\n\n cmds.ogs(pause=True) is a toggle so we cant pass False.\n \"\"\"\n if IS_HEADLESS:\n yield\n return\n\n original_state = cmds.ogs(query=True, pause=True)\n try:\n if suspend and not original_state:\n cmds.ogs(pause=True)\n yield\n finally:\n if suspend and not original_state:\n cmds.ogs(pause=True)\n\n\n@contextlib.contextmanager\ndef maintained_selection():\n \"\"\"Maintain selection during context\n\n Example:\n >>> scene = cmds.file(new=True, force=True)\n >>> node = cmds.createNode(\"transform\", name=\"Test\")\n >>> cmds.select(\"persp\")\n >>> with maintained_selection():\n ... cmds.select(\"Test\", replace=True)\n >>> \"Test\" in cmds.ls(selection=True)\n False\n\n \"\"\"\n\n previous_selection = cmds.ls(selection=True)\n try:\n yield\n finally:\n if previous_selection:\n cmds.select(previous_selection,\n replace=True,\n noExpand=True)\n else:\n cmds.select(clear=True)\n\n\ndef reload_all_udim_tile_previews():\n \"\"\"Regenerate all UDIM tile preview in texture file\"\"\"\n for texture_file in cmds.ls(type=\"file\"):\n if cmds.getAttr(\"{}.uvTilingMode\".format(texture_file)) > 0:\n cmds.ogs(regenerateUVTilePreview=texture_file)\n\n\n@contextlib.contextmanager\ndef panel_camera(panel, camera):\n \"\"\"Set modelPanel's camera during the context.\n\n Arguments:\n panel (str): modelPanel name.\n camera (str): camera name.\n\n \"\"\"\n original_camera = cmds.modelPanel(panel, query=True, camera=True)\n try:\n cmds.modelPanel(panel, edit=True, camera=camera)\n yield\n finally:\n cmds.modelPanel(panel, edit=True, camera=original_camera)\n\n\ndef render_capture_preset(preset):\n \"\"\"Capture playblast with a preset.\n\n To generate the preset use `generate_capture_preset`.\n\n Args:\n preset (dict): preset options\n\n Returns:\n str: Output path of `capture.capture`\n \"\"\"\n\n # Force a refresh at the start of the timeline\n # TODO (Question): Why do we need to do this? What bug does it solve?\n # Is this for simulations?\n cmds.refresh(force=True)\n refresh_frame_int = int(cmds.playbackOptions(query=True, minTime=True))\n cmds.currentTime(refresh_frame_int - 1, edit=True)\n cmds.currentTime(refresh_frame_int, edit=True)\n log.debug(\n \"Using preset: {}\".format(\n json.dumps(preset, indent=4, sort_keys=True)\n )\n )\n preset = copy.deepcopy(preset)\n # not supported by `capture` so we pop it off of the preset\n reload_textures = preset[\"viewport_options\"].pop(\"loadTextures\", False)\n panel = preset.pop(\"panel\")\n with ExitStack() as stack:\n stack.enter_context(maintained_time())\n stack.enter_context(panel_camera(panel, preset[\"camera\"]))\n stack.enter_context(viewport_default_options(panel, preset))\n if reload_textures:\n # Force immediate texture loading when to ensure\n # all textures have loaded before the playblast starts\n stack.enter_context(material_loading_mode(mode=\"immediate\"))\n # Regenerate all UDIM tiles previews\n reload_all_udim_tile_previews()\n path = capture.capture(log=self.log, **preset)\n\n return path\n\n\ndef generate_capture_preset(instance, camera, path,\n start=None, end=None, capture_preset=None):\n \"\"\"Function for getting all the data of preset options for\n playblast capturing\n\n Args:\n instance (pyblish.api.Instance): instance\n camera (str): review camera\n path (str): filepath\n start (int): frameStart\n end (int): frameEnd\n capture_preset (dict): capture preset\n\n Returns:\n dict: Resulting preset\n \"\"\"\n preset = load_capture_preset(data=capture_preset)\n\n preset[\"camera\"] = camera\n preset[\"start_frame\"] = start\n preset[\"end_frame\"] = end\n preset[\"filename\"] = path\n preset[\"overwrite\"] = True\n preset[\"panel\"] = instance.data[\"panel\"]\n\n # Disable viewer since we use the rendering logic for publishing\n # We don't want to open the generated playblast in a viewer directly.\n preset[\"viewer\"] = False\n\n # \"isolate_view\" will already have been applied at creation, so we'll\n # ignore it here.\n preset.pop(\"isolate_view\")\n\n # Set resolution variables from capture presets\n width_preset = capture_preset[\"Resolution\"][\"width\"]\n height_preset = capture_preset[\"Resolution\"][\"height\"]\n\n # Set resolution variables from asset values\n asset_data = instance.data[\"assetEntity\"][\"data\"]\n asset_width = asset_data.get(\"resolutionWidth\")\n asset_height = asset_data.get(\"resolutionHeight\")\n review_instance_width = instance.data.get(\"review_width\")\n review_instance_height = instance.data.get(\"review_height\")\n\n # Use resolution from instance if review width/height is set\n # Otherwise use the resolution from preset if it has non-zero values\n # Otherwise fall back to asset width x height\n # Else define no width, then `capture.capture` will use render resolution\n if review_instance_width and review_instance_height:\n preset[\"width\"] = review_instance_width\n preset[\"height\"] = review_instance_height\n elif width_preset and height_preset:\n preset[\"width\"] = width_preset\n preset[\"height\"] = height_preset\n elif asset_width and asset_height:\n preset[\"width\"] = asset_width\n preset[\"height\"] = asset_height\n\n # Isolate view is requested by having objects in the set besides a\n # camera. If there is only 1 member it'll be the camera because we\n # validate to have 1 camera only.\n if instance.data[\"isolate\"] and len(instance.data[\"setMembers\"]) > 1:\n preset[\"isolate\"] = instance.data[\"setMembers\"]\n\n # Override camera options\n # Enforce persisting camera depth of field\n camera_options = preset.setdefault(\"camera_options\", {})\n camera_options[\"depthOfField\"] = cmds.getAttr(\n \"{0}.depthOfField\".format(camera)\n )\n\n # Use Pan/Zoom from instance data instead of from preset\n preset.pop(\"pan_zoom\", None)\n camera_options[\"panZoomEnabled\"] = instance.data[\"panZoom\"]\n\n # Override viewport options by instance data\n viewport_options = preset.setdefault(\"viewport_options\", {})\n viewport_options[\"displayLights\"] = instance.data[\"displayLights\"]\n viewport_options[\"imagePlane\"] = instance.data.get(\"imagePlane\", True)\n\n # Override transparency if requested.\n transparency = instance.data.get(\"transparency\", 0)\n if transparency != 0:\n preset[\"viewport2_options\"][\"transparencyAlgorithm\"] = transparency\n\n # Update preset with current panel setting\n # if override_viewport_options is turned off\n if not capture_preset[\"Viewport Options\"][\"override_viewport_options\"]:\n panel_preset = capture.parse_view(preset[\"panel\"])\n panel_preset.pop(\"camera\")\n preset.update(panel_preset)\n\n return preset\n\n\n@contextlib.contextmanager\ndef viewport_default_options(panel, preset):\n \"\"\"Context manager used by `render_capture_preset`.\n\n We need to explicitly enable some viewport changes so the viewport is\n refreshed ahead of playblasting.\n\n \"\"\"\n # TODO: Clarify in the docstring WHY we need to set it ahead of\n # playblasting. What issues does it solve?\n viewport_defaults = {}\n try:\n keys = [\n \"useDefaultMaterial\",\n \"wireframeOnShaded\",\n \"xray\",\n \"jointXray\",\n \"backfaceCulling\",\n \"textures\"\n ]\n for key in keys:\n viewport_defaults[key] = cmds.modelEditor(\n panel, query=True, **{key: True}\n )\n if preset[\"viewport_options\"].get(key):\n cmds.modelEditor(\n panel, edit=True, **{key: True}\n )\n yield\n finally:\n # Restoring viewport options.\n if viewport_defaults:\n cmds.modelEditor(\n panel, edit=True, **viewport_defaults\n )\n\n\n@contextlib.contextmanager\ndef material_loading_mode(mode=\"immediate\"):\n \"\"\"Set material loading mode during context\"\"\"\n original = cmds.displayPref(query=True, materialLoadingMode=True)\n cmds.displayPref(materialLoadingMode=mode)\n try:\n yield\n finally:\n cmds.displayPref(materialLoadingMode=original)\n\n\ndef get_namespace(node):\n \"\"\"Return namespace of given node\"\"\"\n node_name = node.rsplit(\"|\", 1)[-1]\n if \":\" in node_name:\n return node_name.rsplit(\":\", 1)[0]\n else:\n return \"\"\n\n\ndef strip_namespace(node, namespace):\n \"\"\"Strip given namespace from node path.\n\n The namespace will only be stripped from names\n if it starts with that namespace. If the namespace\n occurs within another namespace it's not removed.\n\n Examples:\n >>> strip_namespace(\"namespace:node\", namespace=\"namespace:\")\n \"node\"\n >>> strip_namespace(\"hello:world:node\", namespace=\"hello:world\")\n \"node\"\n >>> strip_namespace(\"hello:world:node\", namespace=\"hello\")\n \"world:node\"\n >>> strip_namespace(\"hello:world:node\", namespace=\"world\")\n \"hello:world:node\"\n >>> strip_namespace(\"ns:group|ns:node\", namespace=\"ns\")\n \"group|node\"\n\n Returns:\n str: Node name without given starting namespace.\n\n \"\"\"\n\n # Ensure namespace ends with `:`\n if not namespace.endswith(\":\"):\n namespace = \"{}:\".format(namespace)\n\n # The long path for a node can also have the namespace\n # in its parents so we need to remove it from each\n return \"|\".join(\n name[len(namespace):] if name.startswith(namespace) else name\n for name in node.split(\"|\")\n )\n\n\ndef get_custom_namespace(custom_namespace):\n \"\"\"Return unique namespace.\n\n The input namespace can contain a single group\n of '#' number tokens to indicate where the namespace's\n unique index should go. The amount of tokens defines\n the zero padding of the number, e.g ### turns into 001.\n\n Warning: Note that a namespace will always be\n prefixed with a _ if it starts with a digit\n\n Example:\n >>> get_custom_namespace(\"myspace_##_\")\n # myspace_01_\n >>> get_custom_namespace(\"##_myspace\")\n # _01_myspace\n >>> get_custom_namespace(\"myspace##\")\n # myspace01\n\n \"\"\"\n split = re.split(\"([#]+)\", custom_namespace, 1)\n\n if len(split) == 3:\n base, padding, suffix = split\n padding = \"%0{}d\".format(len(padding))\n else:\n base = split[0]\n padding = \"%02d\" # default padding\n suffix = \"\"\n\n return unique_namespace(\n base,\n format=padding,\n prefix=\"_\" if not base or base[0].isdigit() else \"\",\n suffix=suffix\n )\n\n\ndef unique_namespace(namespace, format=\"%02d\", prefix=\"\", suffix=\"\"):\n \"\"\"Return unique namespace\n\n Arguments:\n namespace (str): Name of namespace to consider\n format (str, optional): Formatting of the given iteration number\n suffix (str, optional): Only consider namespaces with this suffix.\n\n >>> unique_namespace(\"bar\")\n # bar01\n >>> unique_namespace(\":hello\")\n # :hello01\n >>> unique_namespace(\"bar:\", suffix=\"_NS\")\n # bar01_NS:\n\n \"\"\"\n\n def current_namespace():\n current = cmds.namespaceInfo(currentNamespace=True,\n absoluteName=True)\n # When inside a namespace Maya adds no trailing :\n if not current.endswith(\":\"):\n current += \":\"\n return current\n\n # Always check against the absolute namespace root\n # There's no clash with :x if we're defining namespace :a:x\n ROOT = \":\" if namespace.startswith(\":\") else current_namespace()\n\n # Strip trailing `:` tokens since we might want to add a suffix\n start = \":\" if namespace.startswith(\":\") else \"\"\n end = \":\" if namespace.endswith(\":\") else \"\"\n namespace = namespace.strip(\":\")\n if \":\" in namespace:\n # Split off any nesting that we don't uniqify anyway.\n parents, namespace = namespace.rsplit(\":\", 1)\n start += parents + \":\"\n ROOT += start\n\n def exists(n):\n # Check for clash with nodes and namespaces\n fullpath = ROOT + n\n return cmds.objExists(fullpath) or cmds.namespace(exists=fullpath)\n\n iteration = 1\n while True:\n nr_namespace = namespace + format % iteration\n unique = prefix + nr_namespace + suffix\n\n if not exists(unique):\n return start + unique + end\n\n iteration += 1\n\n\ndef read(node):\n \"\"\"Return user-defined attributes from `node`\"\"\"\n\n data = dict()\n\n for attr in cmds.listAttr(node, userDefined=True) or list():\n try:\n value = cmds.getAttr(node + \".\" + attr, asString=True)\n\n except RuntimeError:\n # For Message type attribute or others that have connections,\n # take source node name as value.\n source = cmds.listConnections(node + \".\" + attr,\n source=True,\n destination=False)\n source = cmds.ls(source, long=True) or [None]\n value = source[0]\n\n except ValueError:\n # Some attributes cannot be read directly,\n # such as mesh and color attributes. These\n # are considered non-essential to this\n # particular publishing pipeline.\n value = None\n\n data[attr] = value\n\n return data\n\n\ndef matrix_equals(a, b, tolerance=1e-10):\n \"\"\"\n Compares two matrices with an imperfection tolerance\n\n Args:\n a (list, tuple): the matrix to check\n b (list, tuple): the matrix to check against\n tolerance (float): the precision of the differences\n\n Returns:\n bool : True or False\n\n \"\"\"\n if not all(abs(x - y) < tolerance for x, y in zip(a, b)):\n return False\n return True\n\n\ndef float_round(num, places=0, direction=ceil):\n return direction(num * (10**places)) / float(10**places)\n\n\ndef pairwise(iterable):\n \"\"\"s -> (s0,s1), (s2,s3), (s4, s5), ...\"\"\"\n from six.moves import zip\n\n a = iter(iterable)\n return zip(a, a)\n\n\ndef collect_animation_defs(fps=False):\n \"\"\"Get the basic animation attribute defintions for the publisher.\n\n Returns:\n OrderedDict\n\n \"\"\"\n\n # get scene values as defaults\n frame_start = cmds.playbackOptions(query=True, minTime=True)\n frame_end = cmds.playbackOptions(query=True, maxTime=True)\n frame_start_handle = cmds.playbackOptions(\n query=True, animationStartTime=True\n )\n frame_end_handle = cmds.playbackOptions(query=True, animationEndTime=True)\n\n handle_start = frame_start - frame_start_handle\n handle_end = frame_end_handle - frame_end\n\n # build attributes\n defs = [\n NumberDef(\"frameStart\",\n label=\"Frame Start\",\n default=frame_start,\n decimals=0),\n NumberDef(\"frameEnd\",\n label=\"Frame End\",\n default=frame_end,\n decimals=0),\n NumberDef(\"handleStart\",\n label=\"Handle Start\",\n default=handle_start,\n decimals=0),\n NumberDef(\"handleEnd\",\n label=\"Handle End\",\n default=handle_end,\n decimals=0),\n NumberDef(\"step\",\n label=\"Step size\",\n tooltip=\"A smaller step size means more samples and larger \"\n \"output files.\\n\"\n \"A 1.0 step size is a single sample every frame.\\n\"\n \"A 0.5 step size is two samples per frame.\\n\"\n \"A 0.2 step size is five samples per frame.\",\n default=1.0,\n decimals=3),\n ]\n\n if fps:\n current_fps = mel.eval('currentTimeUnitToFPS()')\n fps_def = NumberDef(\n \"fps\", label=\"FPS\", default=current_fps, decimals=5\n )\n defs.append(fps_def)\n\n return defs\n\n\ndef imprint(node, data):\n \"\"\"Write `data` to `node` as userDefined attributes\n\n Arguments:\n node (str): Long name of node\n data (dict): Dictionary of key/value pairs\n\n Example:\n >>> from maya import cmds\n >>> def compute():\n ... return 6\n ...\n >>> cube, generator = cmds.polyCube()\n >>> imprint(cube, {\n ... \"regularString\": \"myFamily\",\n ... \"computedValue\": lambda: compute()\n ... })\n ...\n >>> cmds.getAttr(cube + \".computedValue\")\n 6\n\n \"\"\"\n\n for key, value in data.items():\n\n if callable(value):\n # Support values evaluated at imprint\n value = value()\n\n if isinstance(value, bool):\n add_type = {\"attributeType\": \"bool\"}\n set_type = {\"keyable\": False, \"channelBox\": True}\n elif isinstance(value, string_types):\n add_type = {\"dataType\": \"string\"}\n set_type = {\"type\": \"string\"}\n elif isinstance(value, int):\n add_type = {\"attributeType\": \"long\"}\n set_type = {\"keyable\": False, \"channelBox\": True}\n elif isinstance(value, float):\n add_type = {\"attributeType\": \"double\"}\n set_type = {\"keyable\": False, \"channelBox\": True}\n elif isinstance(value, (list, tuple)):\n add_type = {\"attributeType\": \"enum\", \"enumName\": \":\".join(value)}\n set_type = {\"keyable\": False, \"channelBox\": True}\n value = 0 # enum default\n else:\n raise TypeError(\"Unsupported type: %r\" % type(value))\n\n cmds.addAttr(node, longName=key, **add_type)\n cmds.setAttr(node + \".\" + key, value, **set_type)\n\n\ndef lsattr(attr, value=None):\n \"\"\"Return nodes matching `key` and `value`\n\n Arguments:\n attr (str): Name of Maya attribute\n value (object, optional): Value of attribute. If none\n is provided, return all nodes with this attribute.\n\n Example:\n >> lsattr(\"id\", \"myId\")\n [\"myNode\"]\n >> lsattr(\"id\")\n [\"myNode\", \"myOtherNode\"]\n\n \"\"\"\n\n if value is None:\n return cmds.ls(\"*.%s\" % attr,\n recursive=True,\n objectsOnly=True,\n long=True)\n return lsattrs({attr: value})\n\n\ndef lsattrs(attrs):\n \"\"\"Return nodes with the given attribute(s).\n\n Arguments:\n attrs (dict): Name and value pairs of expected matches\n\n Example:\n >>> # Return nodes with an `age` of five.\n >>> lsattrs({\"age\": \"five\"})\n >>> # Return nodes with both `age` and `color` of five and blue.\n >>> lsattrs({\"age\": \"five\", \"color\": \"blue\"})\n\n Return:\n list: matching nodes.\n\n \"\"\"\n\n dep_fn = OpenMaya.MFnDependencyNode()\n dag_fn = OpenMaya.MFnDagNode()\n selection_list = OpenMaya.MSelectionList()\n\n first_attr = next(iter(attrs))\n\n try:\n selection_list.add(\"*.{0}\".format(first_attr),\n searchChildNamespaces=True)\n except RuntimeError as exc:\n if str(exc).endswith(\"Object does not exist\"):\n return []\n\n matches = set()\n for i in range(selection_list.length()):\n node = selection_list.getDependNode(i)\n if node.hasFn(OpenMaya.MFn.kDagNode):\n fn_node = dag_fn.setObject(node)\n full_path_names = [path.fullPathName()\n for path in fn_node.getAllPaths()]\n else:\n fn_node = dep_fn.setObject(node)\n full_path_names = [fn_node.name()]\n\n for attr in attrs:\n try:\n plug = fn_node.findPlug(attr, True)\n if plug.asString() != attrs[attr]:\n break\n except RuntimeError:\n break\n else:\n matches.update(full_path_names)\n\n return list(matches)\n\n\n@contextlib.contextmanager\ndef attribute_values(attr_values):\n \"\"\"Remaps node attributes to values during context.\n\n Arguments:\n attr_values (dict): Dictionary with (attr, value)\n\n \"\"\"\n\n original = [(attr, cmds.getAttr(attr)) for attr in attr_values]\n try:\n for attr, value in attr_values.items():\n if isinstance(value, string_types):\n cmds.setAttr(attr, value, type=\"string\")\n else:\n cmds.setAttr(attr, value)\n yield\n finally:\n for attr, value in original:\n if isinstance(value, string_types):\n cmds.setAttr(attr, value, type=\"string\")\n elif value is None and cmds.getAttr(attr, type=True) == \"string\":\n # In some cases the maya.cmds.getAttr command returns None\n # for string attributes but this value cannot assigned.\n # Note: After setting it once to \"\" it will then return \"\"\n # instead of None. So this would only happen once.\n cmds.setAttr(attr, \"\", type=\"string\")\n else:\n cmds.setAttr(attr, value)\n\n\n@contextlib.contextmanager\ndef keytangent_default(in_tangent_type='auto',\n out_tangent_type='auto'):\n \"\"\"Set the default keyTangent for new keys during this context\"\"\"\n\n original_itt = cmds.keyTangent(query=True, g=True, itt=True)[0]\n original_ott = cmds.keyTangent(query=True, g=True, ott=True)[0]\n cmds.keyTangent(g=True, itt=in_tangent_type)\n cmds.keyTangent(g=True, ott=out_tangent_type)\n try:\n yield\n finally:\n cmds.keyTangent(g=True, itt=original_itt)\n cmds.keyTangent(g=True, ott=original_ott)\n\n\n@contextlib.contextmanager\ndef undo_chunk():\n \"\"\"Open a undo chunk during context.\"\"\"\n\n try:\n cmds.undoInfo(openChunk=True)\n yield\n finally:\n cmds.undoInfo(closeChunk=True)\n\n\n@contextlib.contextmanager\ndef evaluation(mode=\"off\"):\n \"\"\"Set the evaluation manager during context.\n\n Arguments:\n mode (str): The mode to apply during context.\n \"off\": The standard DG evaluation (stable)\n \"serial\": A serial DG evaluation\n \"parallel\": The Maya 2016+ parallel evaluation\n\n \"\"\"\n\n original = cmds.evaluationManager(query=True, mode=1)[0]\n try:\n cmds.evaluationManager(mode=mode)\n yield\n finally:\n cmds.evaluationManager(mode=original)\n\n\n@contextlib.contextmanager\ndef empty_sets(sets, force=False):\n \"\"\"Remove all members of the sets during the context\"\"\"\n\n assert isinstance(sets, (list, tuple))\n\n original = dict()\n original_connections = []\n\n # Store original state\n for obj_set in sets:\n members = cmds.sets(obj_set, query=True)\n original[obj_set] = members\n\n try:\n for obj_set in sets:\n cmds.sets(clear=obj_set)\n if force:\n # Break all connections if force is enabled, this way we\n # prevent Maya from exporting any reference nodes which are\n # connected with placeHolder[x] attributes\n plug = \"%s.dagSetMembers\" % obj_set\n connections = cmds.listConnections(plug,\n source=True,\n destination=False,\n plugs=True,\n connections=True) or []\n original_connections.extend(connections)\n for dest, src in pairwise(connections):\n cmds.disconnectAttr(src, dest)\n yield\n finally:\n\n for dest, src in pairwise(original_connections):\n cmds.connectAttr(src, dest)\n\n # Restore original members\n _iteritems = getattr(original, \"iteritems\", original.items)\n for origin_set, members in _iteritems():\n cmds.sets(members, forceElement=origin_set)\n\n\n@contextlib.contextmanager\ndef renderlayer(layer):\n \"\"\"Set the renderlayer during the context\n\n Arguments:\n layer (str): Name of layer to switch to.\n\n \"\"\"\n\n original = cmds.editRenderLayerGlobals(query=True,\n currentRenderLayer=True)\n\n try:\n cmds.editRenderLayerGlobals(currentRenderLayer=layer)\n yield\n finally:\n cmds.editRenderLayerGlobals(currentRenderLayer=original)\n\n\nclass delete_after(object):\n \"\"\"Context Manager that will delete collected nodes after exit.\n\n This allows to ensure the nodes added to the context are deleted\n afterwards. This is useful if you want to ensure nodes are deleted\n even if an error is raised.\n\n Examples:\n with delete_after() as delete_bin:\n cube = maya.cmds.polyCube()\n delete_bin.extend(cube)\n # cube exists\n # cube deleted\n\n \"\"\"\n\n def __init__(self, nodes=None):\n\n self._nodes = list()\n\n if nodes:\n self.extend(nodes)\n\n def append(self, node):\n self._nodes.append(node)\n\n def extend(self, nodes):\n self._nodes.extend(nodes)\n\n def __iter__(self):\n return iter(self._nodes)\n\n def __enter__(self):\n return self\n\n def __exit__(self, type, value, traceback):\n if self._nodes:\n cmds.delete(self._nodes)\n\n\ndef get_current_renderlayer():\n return cmds.editRenderLayerGlobals(query=True, currentRenderLayer=True)\n\n\ndef get_renderer(layer):\n with renderlayer(layer):\n return cmds.getAttr(\"defaultRenderGlobals.currentRenderer\")\n\n\n@contextlib.contextmanager\ndef no_undo(flush=False):\n \"\"\"Disable the undo queue during the context\n\n Arguments:\n flush (bool): When True the undo queue will be emptied when returning\n from the context losing all undo history. Defaults to False.\n\n \"\"\"\n original = cmds.undoInfo(query=True, state=True)\n keyword = 'state' if flush else 'stateWithoutFlush'\n\n try:\n cmds.undoInfo(**{keyword: False})\n yield\n finally:\n cmds.undoInfo(**{keyword: original})\n\n\ndef get_shader_assignments_from_shapes(shapes, components=True):\n \"\"\"Return the shape assignment per related shading engines.\n\n Returns a dictionary where the keys are shadingGroups and the values are\n lists of assigned shapes or shape-components.\n\n Since `maya.cmds.sets` returns shader members on the shapes as components\n on the transform we correct that in this method too.\n\n For the 'shapes' this will return a dictionary like:\n {\n \"shadingEngineX\": [\"nodeX\", \"nodeY\"],\n \"shadingEngineY\": [\"nodeA\", \"nodeB\"]\n }\n\n Args:\n shapes (list): The shapes to collect the assignments for.\n components (bool): Whether to include the component assignments.\n\n Returns:\n dict: The {shadingEngine: shapes} relationships\n\n \"\"\"\n\n shapes = cmds.ls(shapes,\n long=True,\n shapes=True,\n objectsOnly=True)\n if not shapes:\n return {}\n\n # Collect shading engines and their shapes\n assignments = defaultdict(list)\n for shape in shapes:\n\n # Get unique shading groups for the shape\n shading_groups = cmds.listConnections(shape,\n source=False,\n destination=True,\n plugs=False,\n connections=False,\n type=\"shadingEngine\") or []\n shading_groups = list(set(shading_groups))\n for shading_group in shading_groups:\n assignments[shading_group].append(shape)\n\n if components:\n # Note: Components returned from maya.cmds.sets are \"listed\" as if\n # being assigned to the transform like: pCube1.f[0] as opposed\n # to pCubeShape1.f[0] so we correct that here too.\n\n # Build a mapping from parent to shapes to include in lookup.\n transforms = {shape.rsplit(\"|\", 1)[0]: shape for shape in shapes}\n lookup = set(shapes) | set(transforms.keys())\n\n component_assignments = defaultdict(list)\n for shading_group in assignments.keys():\n members = cmds.ls(cmds.sets(shading_group, query=True), long=True)\n for member in members:\n\n node = member.split(\".\", 1)[0]\n if node not in lookup:\n continue\n\n # Component\n if \".\" in member:\n\n # Fix transform to shape as shaders are assigned to shapes\n if node in transforms:\n shape = transforms[node]\n component = member.split(\".\", 1)[1]\n member = \"{0}.{1}\".format(shape, component)\n\n component_assignments[shading_group].append(member)\n assignments = component_assignments\n\n return dict(assignments)\n\n\n@contextlib.contextmanager\ndef shader(nodes, shadingEngine=\"initialShadingGroup\"):\n \"\"\"Assign a shader to nodes during the context\"\"\"\n\n shapes = cmds.ls(nodes, dag=1, objectsOnly=1, shapes=1, long=1)\n original = get_shader_assignments_from_shapes(shapes)\n\n try:\n # Assign override shader\n if shapes:\n cmds.sets(shapes, edit=True, forceElement=shadingEngine)\n yield\n finally:\n\n # Assign original shaders\n for sg, members in original.items():\n if members:\n cmds.sets(members, edit=True, forceElement=sg)\n\n\n@contextlib.contextmanager\ndef displaySmoothness(nodes,\n divisionsU=0,\n divisionsV=0,\n pointsWire=4,\n pointsShaded=1,\n polygonObject=1):\n \"\"\"Set the displaySmoothness during the context\"\"\"\n\n # Ensure only non-intermediate shapes\n nodes = cmds.ls(nodes,\n dag=1,\n shapes=1,\n long=1,\n noIntermediate=True)\n\n def parse(node):\n \"\"\"Parse the current state of a node\"\"\"\n state = {}\n for key in [\"divisionsU\",\n \"divisionsV\",\n \"pointsWire\",\n \"pointsShaded\",\n \"polygonObject\"]:\n value = cmds.displaySmoothness(node, query=1, **{key: True})\n if value is not None:\n state[key] = value[0]\n return state\n\n originals = dict((node, parse(node)) for node in nodes)\n\n try:\n # Apply current state\n cmds.displaySmoothness(nodes,\n divisionsU=divisionsU,\n divisionsV=divisionsV,\n pointsWire=pointsWire,\n pointsShaded=pointsShaded,\n polygonObject=polygonObject)\n yield\n finally:\n # Revert state\n _iteritems = getattr(originals, \"iteritems\", originals.items)\n for node, state in _iteritems():\n if state:\n cmds.displaySmoothness(node, **state)\n\n\n@contextlib.contextmanager\ndef no_display_layers(nodes):\n \"\"\"Ensure nodes are not in a displayLayer during context.\n\n Arguments:\n nodes (list): The nodes to remove from any display layer.\n\n \"\"\"\n\n # Ensure long names\n nodes = cmds.ls(nodes, long=True)\n\n # Get the original state\n lookup = set(nodes)\n original = {}\n for layer in cmds.ls(type='displayLayer'):\n\n # Skip default layer\n if layer == \"defaultLayer\":\n continue\n\n members = cmds.editDisplayLayerMembers(layer,\n query=True,\n fullNames=True)\n if not members:\n continue\n members = set(members)\n\n included = lookup.intersection(members)\n if included:\n original[layer] = list(included)\n\n try:\n # Add all nodes to default layer\n cmds.editDisplayLayerMembers(\"defaultLayer\", nodes, noRecurse=True)\n yield\n finally:\n # Restore original members\n _iteritems = getattr(original, \"iteritems\", original.items)\n for layer, members in _iteritems():\n cmds.editDisplayLayerMembers(layer, members, noRecurse=True)\n\n\n@contextlib.contextmanager\ndef namespaced(namespace, new=True, relative_names=None):\n \"\"\"Work inside namespace during context\n\n Args:\n new (bool): When enabled this will rename the namespace to a unique\n namespace if the input namespace already exists.\n\n Yields:\n str: The namespace that is used during the context\n\n \"\"\"\n original = cmds.namespaceInfo(cur=True, absoluteName=True)\n original_relative_names = cmds.namespace(query=True, relativeNames=True)\n if new:\n namespace = unique_namespace(namespace)\n cmds.namespace(add=namespace)\n if relative_names is not None:\n cmds.namespace(relativeNames=relative_names)\n try:\n cmds.namespace(set=namespace)\n yield namespace\n finally:\n cmds.namespace(set=original)\n if relative_names is not None:\n cmds.namespace(relativeNames=original_relative_names)\n\n\n@contextlib.contextmanager\ndef maintained_selection_api():\n \"\"\"Maintain selection using the Maya Python API.\n\n Warning: This is *not* added to the undo stack.\n\n \"\"\"\n original = OpenMaya.MGlobal.getActiveSelectionList()\n try:\n yield\n finally:\n OpenMaya.MGlobal.setActiveSelectionList(original)\n\n\n@contextlib.contextmanager\ndef tool(context):\n \"\"\"Set a tool context during the context manager.\n\n \"\"\"\n original = cmds.currentCtx()\n try:\n cmds.setToolTo(context)\n yield\n finally:\n cmds.setToolTo(original)\n\n\ndef polyConstraint(components, *args, **kwargs):\n \"\"\"Return the list of *components* with the constraints applied.\n\n A wrapper around Maya's `polySelectConstraint` to retrieve its results as\n a list without altering selections. For a list of possible constraints\n see `maya.cmds.polySelectConstraint` documentation.\n\n Arguments:\n components (list): List of components of polygon meshes\n\n Returns:\n list: The list of components filtered by the given constraints.\n\n \"\"\"\n\n kwargs.pop('mode', None)\n\n with no_undo(flush=False):\n # Reverting selection to the original selection using\n # `maya.cmds.select` can be slow in rare cases where previously\n # `maya.cmds.polySelectConstraint` had set constrain to \"All and Next\"\n # and the \"Random\" setting was activated. To work around this we\n # revert to the original selection using the Maya API. This is safe\n # since we're not generating any undo change anyway.\n with tool(\"selectSuperContext\"):\n # Selection can be very slow when in a manipulator mode.\n # So we force the selection context which is fast.\n with maintained_selection_api():\n # Apply constraint using mode=2 (current and next) so\n # it applies to the selection made before it; because just\n # a `maya.cmds.select()` call will not trigger the constraint.\n with reset_polySelectConstraint():\n cmds.select(components, r=1, noExpand=True)\n cmds.polySelectConstraint(*args, mode=2, **kwargs)\n result = cmds.ls(selection=True)\n cmds.select(clear=True)\n return result\n\n\n@contextlib.contextmanager\ndef reset_polySelectConstraint(reset=True):\n \"\"\"Context during which the given polyConstraint settings are disabled.\n\n The original settings are restored after the context.\n\n \"\"\"\n\n original = cmds.polySelectConstraint(query=True, stateString=True)\n\n try:\n if reset:\n # Ensure command is available in mel\n # This can happen when running standalone\n if not mel.eval(\"exists resetPolySelectConstraint\"):\n mel.eval(\"source polygonConstraint\")\n\n # Reset all parameters\n mel.eval(\"resetPolySelectConstraint;\")\n cmds.polySelectConstraint(disable=True)\n yield\n finally:\n mel.eval(original)\n\n\ndef is_visible(node,\n displayLayer=True,\n intermediateObject=True,\n parentHidden=True,\n visibility=True):\n \"\"\"Is `node` visible?\n\n Returns whether a node is hidden by one of the following methods:\n - The node exists (always checked)\n - The node must be a dagNode (always checked)\n - The node's visibility is off.\n - The node is set as intermediate Object.\n - The node is in a disabled displayLayer.\n - Whether any of its parent nodes is hidden.\n\n Roughly based on: http://ewertb.soundlinker.com/mel/mel.098.php\n\n Returns:\n bool: Whether the node is visible in the scene\n\n \"\"\"\n\n # Only existing objects can be visible\n if not cmds.objExists(node):\n return False\n\n # Only dagNodes can be visible\n if not cmds.objectType(node, isAType='dagNode'):\n return False\n\n if visibility:\n if not cmds.getAttr('{0}.visibility'.format(node)):\n return False\n\n if intermediateObject and cmds.objectType(node, isAType='shape'):\n if cmds.getAttr('{0}.intermediateObject'.format(node)):\n return False\n\n if displayLayer:\n # Display layers set overrideEnabled and overrideVisibility on members\n if cmds.attributeQuery('overrideEnabled', node=node, exists=True):\n override_enabled = cmds.getAttr('{}.overrideEnabled'.format(node))\n override_visibility = cmds.getAttr('{}.overrideVisibility'.format(\n node))\n if override_enabled and not override_visibility:\n return False\n\n if parentHidden:\n parents = cmds.listRelatives(node, parent=True, fullPath=True)\n if parents:\n parent = parents[0]\n if not is_visible(parent,\n displayLayer=displayLayer,\n intermediateObject=False,\n parentHidden=parentHidden,\n visibility=visibility):\n return False\n\n return True\n\n# region ID\ndef get_id_required_nodes(referenced_nodes=False, nodes=None):\n \"\"\"Filter out any node which are locked (reference) or readOnly\n\n Args:\n referenced_nodes (bool): set True to filter out reference nodes\n nodes (list, Optional): nodes to consider\n Returns:\n nodes (set): list of filtered nodes\n \"\"\"\n\n lookup = None\n if nodes is None:\n # Consider all nodes\n nodes = cmds.ls()\n else:\n # Build a lookup for the only allowed nodes in output based\n # on `nodes` input of the function (+ ensure long names)\n lookup = set(cmds.ls(nodes, long=True))\n\n def _node_type_exists(node_type):\n try:\n cmds.nodeType(node_type, isTypeName=True)\n return True\n except RuntimeError:\n return False\n\n # `readOnly` flag is obsolete as of Maya 2016 therefore we explicitly\n # remove default nodes and reference nodes\n camera_shapes = [\"frontShape\", \"sideShape\", \"topShape\", \"perspShape\"]\n\n ignore = set()\n if not referenced_nodes:\n ignore |= set(cmds.ls(long=True, referencedNodes=True))\n\n # list all defaultNodes to filter out from the rest\n ignore |= set(cmds.ls(long=True, defaultNodes=True))\n ignore |= set(cmds.ls(camera_shapes, long=True))\n\n # Remove Turtle from the result of `cmds.ls` if Turtle is loaded\n # TODO: This should be a less specific check for a single plug-in.\n if _node_type_exists(\"ilrBakeLayer\"):\n ignore |= set(cmds.ls(type=\"ilrBakeLayer\", long=True))\n\n # Establish set of nodes types to include\n types = [\"objectSet\", \"file\", \"mesh\", \"nurbsCurve\", \"nurbsSurface\"]\n\n # Check if plugin nodes are available for Maya by checking if the plugin\n # is loaded\n if cmds.pluginInfo(\"pgYetiMaya\", query=True, loaded=True):\n types.append(\"pgYetiMaya\")\n\n # We *always* ignore intermediate shapes, so we filter them out directly\n nodes = cmds.ls(nodes, type=types, long=True, noIntermediate=True)\n\n # The items which need to pass the id to their parent\n # Add the collected transform to the nodes\n dag = cmds.ls(nodes, type=\"dagNode\", long=True) # query only dag nodes\n transforms = cmds.listRelatives(dag,\n parent=True,\n fullPath=True) or []\n\n nodes = set(nodes)\n nodes |= set(transforms)\n\n nodes -= ignore # Remove the ignored nodes\n if not nodes:\n return nodes\n\n # Ensure only nodes from the input `nodes` are returned when a\n # filter was applied on function call because we also iterated\n # to parents and alike\n if lookup is not None:\n nodes &= lookup\n\n # Avoid locked nodes\n nodes_list = list(nodes)\n locked = cmds.lockNode(nodes_list, query=True, lock=True)\n for node, lock in zip(nodes_list, locked):\n if lock:\n log.warning(\"Skipping locked node: %s\" % node)\n nodes.remove(node)\n\n return nodes\n\n\ndef get_id(node):\n \"\"\"Get the `cbId` attribute of the given node.\n\n Args:\n node (str): the name of the node to retrieve the attribute from\n Returns:\n str\n\n \"\"\"\n if node is None:\n return\n\n sel = OpenMaya.MSelectionList()\n sel.add(node)\n\n api_node = sel.getDependNode(0)\n fn = OpenMaya.MFnDependencyNode(api_node)\n\n if not fn.hasAttribute(\"cbId\"):\n return\n\n try:\n return fn.findPlug(\"cbId\", False).asString()\n except RuntimeError:\n log.warning(\"Failed to retrieve cbId on %s\", node)\n return\n\n\ndef generate_ids(nodes, asset_id=None):\n \"\"\"Returns new unique ids for the given nodes.\n\n Note: This does not assign the new ids, it only generates the values.\n\n To assign new ids using this method:\n >>> nodes = [\"a\", \"b\", \"c\"]\n >>> for node, id in generate_ids(nodes):\n >>> set_id(node, id)\n\n To also override any existing values (and assign regenerated ids):\n >>> nodes = [\"a\", \"b\", \"c\"]\n >>> for node, id in generate_ids(nodes):\n >>> set_id(node, id, overwrite=True)\n\n Args:\n nodes (list): List of nodes.\n asset_id (str or bson.ObjectId): The database id for the *asset* to\n generate for. When None provided the current asset in the\n active session is used.\n\n Returns:\n list: A list of (node, id) tuples.\n\n \"\"\"\n\n if asset_id is None:\n # Get the asset ID from the database for the asset of current context\n project_name = get_current_project_name()\n asset_name = get_current_asset_name()\n asset_doc = get_asset_by_name(project_name, asset_name, fields=[\"_id\"])\n assert asset_doc, \"No current asset found in Session\"\n asset_id = asset_doc['_id']\n\n node_ids = []\n for node in nodes:\n _, uid = str(uuid.uuid4()).rsplit(\"-\", 1)\n unique_id = \"{}:{}\".format(asset_id, uid)\n node_ids.append((node, unique_id))\n\n return node_ids\n\n\ndef set_id(node, unique_id, overwrite=False):\n \"\"\"Add cbId to `node` unless one already exists.\n\n Args:\n node (str): the node to add the \"cbId\" on\n unique_id (str): The unique node id to assign.\n This should be generated by `generate_ids`.\n overwrite (bool, optional): When True overrides the current value even\n if `node` already has an id. Defaults to False.\n\n Returns:\n None\n\n \"\"\"\n\n exists = cmds.attributeQuery(\"cbId\", node=node, exists=True)\n\n # Add the attribute if it does not exist yet\n if not exists:\n cmds.addAttr(node, longName=\"cbId\", dataType=\"string\")\n\n # Set the value\n if not exists or overwrite:\n attr = \"{0}.cbId\".format(node)\n cmds.setAttr(attr, unique_id, type=\"string\")\n\n\ndef get_attribute(plug,\n asString=False,\n expandEnvironmentVariables=False,\n **kwargs):\n \"\"\"Maya getAttr with some fixes based on `pymel.core.general.getAttr()`.\n\n Like Pymel getAttr this applies some changes to `maya.cmds.getAttr`\n - maya pointlessly returned vector results as a tuple wrapped in a list\n (ex. '[(1,2,3)]'). This command unpacks the vector for you.\n - when getting a multi-attr, maya would raise an error, but this will\n return a list of values for the multi-attr\n - added support for getting message attributes by returning the\n connections instead\n\n Note that the asString + expandEnvironmentVariables argument naming\n convention matches the `maya.cmds.getAttr` arguments so that it can\n act as a direct replacement for it.\n\n Args:\n plug (str): Node's attribute plug as `node.attribute`\n asString (bool): Return string value for enum attributes instead\n of the index. Note that the return value can be dependent on the\n UI language Maya is running in.\n expandEnvironmentVariables (bool): Expand any environment variable and\n (tilde characters on UNIX) found in string attributes which are\n returned.\n\n Kwargs:\n Supports the keyword arguments of `maya.cmds.getAttr`\n\n Returns:\n object: The value of the maya attribute.\n\n \"\"\"\n attr_type = cmds.getAttr(plug, type=True)\n if asString:\n kwargs[\"asString\"] = True\n if expandEnvironmentVariables:\n kwargs[\"expandEnvironmentVariables\"] = True\n try:\n res = cmds.getAttr(plug, **kwargs)\n except RuntimeError:\n if attr_type == \"message\":\n return cmds.listConnections(plug)\n\n node, attr = plug.split(\".\", 1)\n children = cmds.attributeQuery(attr, node=node, listChildren=True)\n if children:\n return [\n get_attribute(\"{}.{}\".format(node, child))\n for child in children\n ]\n\n raise\n\n # Convert vector result wrapped in tuple\n if isinstance(res, list) and len(res):\n if isinstance(res[0], tuple) and len(res):\n if attr_type in {'pointArray', 'vectorArray'}:\n return res\n return res[0]\n\n return res\n\n\ndef set_attribute(attribute, value, node):\n \"\"\"Adjust attributes based on the value from the attribute data\n\n If an attribute does not exists on the target it will be added with\n the dataType being controlled by the value type.\n\n Args:\n attribute (str): name of the attribute to change\n value: the value to change to attribute to\n node (str): name of the node\n\n Returns:\n None\n \"\"\"\n\n value_type = type(value).__name__\n kwargs = ATTRIBUTE_DICT[value_type]\n if not cmds.attributeQuery(attribute, node=node, exists=True):\n log.debug(\"Creating attribute '{}' on \"\n \"'{}'\".format(attribute, node))\n cmds.addAttr(node, longName=attribute, **kwargs)\n\n node_attr = \"{}.{}\".format(node, attribute)\n enum_type = cmds.attributeQuery(attribute, node=node, enum=True)\n if enum_type and value_type == \"str\":\n enum_string_values = cmds.attributeQuery(\n attribute, node=node, listEnum=True\n )[0].split(\":\")\n cmds.setAttr(\n \"{}.{}\".format(node, attribute), enum_string_values.index(value)\n )\n elif \"dataType\" in kwargs:\n attr_type = kwargs[\"dataType\"]\n cmds.setAttr(node_attr, value, type=attr_type)\n else:\n cmds.setAttr(node_attr, value)\n\n\ndef apply_attributes(attributes, nodes_by_id):\n \"\"\"Alter the attributes to match the state when publishing\n\n Apply attribute settings from the publish to the node in the scene based\n on the UUID which is stored in the cbId attribute.\n\n Args:\n attributes (list): list of dictionaries\n nodes_by_id (dict): collection of nodes based on UUID\n {uuid: [node, node]}\n\n \"\"\"\n\n for attr_data in attributes:\n nodes = nodes_by_id[attr_data[\"uuid\"]]\n attr_value = attr_data[\"attributes\"]\n for node in nodes:\n for attr, value in attr_value.items():\n set_attribute(attr, value, node)\n\n\ndef get_container_members(container):\n \"\"\"Returns the members of a container.\n This includes the nodes from any loaded references in the container.\n \"\"\"\n if isinstance(container, dict):\n # Assume it's a container dictionary\n container = container[\"objectName\"]\n\n members = cmds.sets(container, query=True) or []\n members = cmds.ls(members, long=True, objectsOnly=True) or []\n all_members = set(members)\n\n # Include any referenced nodes from any reference in the container\n # This is required since we've removed adding ALL nodes of a reference\n # into the container set and only add the reference node now.\n for ref in cmds.ls(members, exactType=\"reference\", objectsOnly=True):\n\n # Ignore any `:sharedReferenceNode`\n if ref.rsplit(\":\", 1)[-1].startswith(\"sharedReferenceNode\"):\n continue\n\n # Ignore _UNKNOWN_REF_NODE_ (PLN-160)\n if ref.rsplit(\":\", 1)[-1].startswith(\"_UNKNOWN_REF_NODE_\"):\n continue\n\n reference_members = cmds.referenceQuery(ref, nodes=True, dagPath=True)\n reference_members = cmds.ls(reference_members,\n long=True,\n objectsOnly=True)\n all_members.update(reference_members)\n\n return list(all_members)\n\n\n# region LOOKDEV\ndef list_looks(project_name, asset_id):\n \"\"\"Return all look subsets for the given asset\n\n This assumes all look subsets start with \"look*\" in their names.\n \"\"\"\n # # get all subsets with look leading in\n # the name associated with the asset\n # TODO this should probably look for family 'look' instead of checking\n # subset name that can not start with family\n subset_docs = get_subsets(project_name, asset_ids=[asset_id])\n return [\n subset_doc\n for subset_doc in subset_docs\n if subset_doc[\"name\"].startswith(\"look\")\n ]\n\n\ndef assign_look_by_version(nodes, version_id):\n \"\"\"Assign nodes a specific published look version by id.\n\n This assumes the nodes correspond with the asset.\n\n Args:\n nodes(list): nodes to assign look to\n version_id (bson.ObjectId): database id of the version\n\n Returns:\n None\n \"\"\"\n\n project_name = get_current_project_name()\n\n # Get representations of shader file and relationships\n look_representation = get_representation_by_name(\n project_name, \"ma\", version_id\n )\n json_representation = get_representation_by_name(\n project_name, \"json\", version_id\n )\n\n # See if representation is already loaded, if so reuse it.\n host = registered_host()\n representation_id = str(look_representation['_id'])\n for container in host.ls():\n if (container['loader'] == \"LookLoader\" and\n container['representation'] == representation_id):\n log.info(\"Reusing loaded look ..\")\n container_node = container['objectName']\n break\n else:\n log.info(\"Using look for the first time ..\")\n\n # Load file\n _loaders = discover_loader_plugins()\n loaders = loaders_from_representation(_loaders, representation_id)\n Loader = next((i for i in loaders if i.__name__ == \"LookLoader\"), None)\n if Loader is None:\n raise RuntimeError(\"Could not find LookLoader, this is a bug\")\n\n # Reference the look file\n with maintained_selection():\n container_node = load_container(Loader, look_representation)\n\n # Get container members\n shader_nodes = get_container_members(container_node)\n\n # Load relationships\n shader_relation = get_representation_path(json_representation)\n with open(shader_relation, \"r\") as f:\n relationships = json.load(f)\n\n # Assign relationships\n apply_shaders(relationships, shader_nodes, nodes)\n\n\ndef assign_look(nodes, subset=\"lookDefault\"):\n \"\"\"Assigns a look to a node.\n\n Optimizes the nodes by grouping by asset id and finding\n related subset by name.\n\n Args:\n nodes (list): all nodes to assign the look to\n subset (str): name of the subset to find\n \"\"\"\n\n # Group all nodes per asset id\n grouped = defaultdict(list)\n for node in nodes:\n pype_id = get_id(node)\n if not pype_id:\n continue\n\n parts = pype_id.split(\":\", 1)\n grouped[parts[0]].append(node)\n\n project_name = get_current_project_name()\n subset_docs = get_subsets(\n project_name, subset_names=[subset], asset_ids=grouped.keys()\n )\n subset_docs_by_asset_id = {\n str(subset_doc[\"parent\"]): subset_doc\n for subset_doc in subset_docs\n }\n subset_ids = {\n subset_doc[\"_id\"]\n for subset_doc in subset_docs_by_asset_id.values()\n }\n last_version_docs = get_last_versions(\n project_name,\n subset_ids=subset_ids,\n fields=[\"_id\", \"name\", \"data.families\"]\n )\n last_version_docs_by_subset_id = {\n last_version_doc[\"parent\"]: last_version_doc\n for last_version_doc in last_version_docs\n }\n\n for asset_id, asset_nodes in grouped.items():\n # create objectId for database\n subset_doc = subset_docs_by_asset_id.get(asset_id)\n if not subset_doc:\n log.warning(\"No subset '{}' found for {}\".format(subset, asset_id))\n continue\n\n last_version = last_version_docs_by_subset_id.get(subset_doc[\"_id\"])\n if not last_version:\n log.warning((\n \"Not found last version for subset '{}' on asset with id {}\"\n ).format(subset, asset_id))\n continue\n\n families = last_version.get(\"data\", {}).get(\"families\") or []\n if \"look\" not in families:\n log.warning((\n \"Last version for subset '{}' on asset with id {}\"\n \" does not have look family\"\n ).format(subset, asset_id))\n continue\n\n log.debug(\"Assigning look '{}' \".format(\n subset, last_version[\"name\"]))\n\n assign_look_by_version(asset_nodes, last_version[\"_id\"])\n\n\ndef apply_shaders(relationships, shadernodes, nodes):\n \"\"\"Link shadingEngine to the right nodes based on relationship data\n\n Relationship data is constructed of a collection of `sets` and `attributes`\n `sets` corresponds with the shaderEngines found in the lookdev.\n Each set has the keys `name`, `members` and `uuid`, the `members`\n hold a collection of node information `name` and `uuid`.\n\n Args:\n relationships (dict): relationship data\n shadernodes (list): list of nodes of the shading objectSets (includes\n VRayObjectProperties and shadingEngines)\n nodes (list): list of nodes to apply shader to\n\n Returns:\n None\n \"\"\"\n\n attributes = relationships.get(\"attributes\", [])\n shader_data = relationships.get(\"relationships\", {})\n\n shading_engines = cmds.ls(shadernodes, type=\"objectSet\", long=True)\n assert shading_engines, \"Error in retrieving objectSets from reference\"\n\n # region compute lookup\n nodes_by_id = defaultdict(list)\n for node in nodes:\n nodes_by_id[get_id(node)].append(node)\n\n shading_engines_by_id = defaultdict(list)\n for shad in shading_engines:\n shading_engines_by_id[get_id(shad)].append(shad)\n # endregion\n\n # region assign shading engines and other sets\n for data in shader_data.values():\n # collect all unique IDs of the set members\n shader_uuid = data[\"uuid\"]\n member_uuids = [member[\"uuid\"] for member in data[\"members\"]]\n\n filtered_nodes = list()\n for m_uuid in member_uuids:\n filtered_nodes.extend(nodes_by_id[m_uuid])\n\n id_shading_engines = shading_engines_by_id[shader_uuid]\n if not id_shading_engines:\n log.error(\"No shader found with cbId \"\n \"'{}'\".format(shader_uuid))\n continue\n elif len(id_shading_engines) > 1:\n log.error(\"Skipping shader assignment. \"\n \"More than one shader found with cbId \"\n \"'{}'. (found: {})\".format(shader_uuid,\n id_shading_engines))\n continue\n\n if not filtered_nodes:\n log.warning(\"No nodes found for shading engine \"\n \"'{0}'\".format(id_shading_engines[0]))\n continue\n try:\n cmds.sets(filtered_nodes, forceElement=id_shading_engines[0])\n except RuntimeError as rte:\n log.error(\"Error during shader assignment: {}\".format(rte))\n\n # endregion\n\n apply_attributes(attributes, nodes_by_id)\n\n\n# endregion LOOKDEV\ndef get_isolate_view_sets():\n \"\"\"Return isolate view sets of all modelPanels.\n\n Returns:\n list: all sets related to isolate view\n\n \"\"\"\n\n view_sets = set()\n for panel in cmds.getPanel(type=\"modelPanel\") or []:\n view_set = cmds.modelEditor(panel, query=True, viewObjects=True)\n if view_set:\n view_sets.add(view_set)\n\n return view_sets\n\n\ndef get_related_sets(node):\n \"\"\"Return objectSets that are relationships for a look for `node`.\n\n Filters out based on:\n - id attribute is NOT `pyblish.avalon.container`\n - shapes and deformer shapes (alembic creates meshShapeDeformed)\n - set name ends with any from a predefined list\n - set in not in viewport set (isolate selected for example)\n\n Args:\n node (str): name of the current node to check\n\n Returns:\n list: The related sets\n\n \"\"\"\n\n # Ignore specific suffices\n ignore_suffices = [\"out_SET\", \"controls_SET\", \"_INST\", \"_CON\"]\n\n # Default nodes to ignore\n defaults = {\"defaultLightSet\", \"defaultObjectSet\"}\n\n # Ids to ignore\n ignored = {\"pyblish.avalon.instance\", \"pyblish.avalon.container\"}\n\n view_sets = get_isolate_view_sets()\n\n sets = cmds.listSets(object=node, extendToShape=False)\n if not sets:\n return []\n\n # Fix 'no object matches name' errors on nodes returned by listSets.\n # In rare cases it can happen that a node is added to an internal maya\n # set inaccessible by maya commands, for example check some nodes\n # returned by `cmds.listSets(allSets=True)`\n sets = cmds.ls(sets)\n\n # Ignore `avalon.container`\n sets = [s for s in sets if\n not cmds.attributeQuery(\"id\", node=s, exists=True) or\n not cmds.getAttr(\"%s.id\" % s) in ignored]\n\n # Exclude deformer sets (`type=2` for `maya.cmds.listSets`)\n deformer_sets = cmds.listSets(object=node,\n extendToShape=False,\n type=2) or []\n deformer_sets = set(deformer_sets) # optimize lookup\n sets = [s for s in sets if s not in deformer_sets]\n\n # Ignore when the set has a specific suffix\n sets = [s for s in sets if not any(s.endswith(x) for x in ignore_suffices)]\n\n # Ignore viewport filter view sets (from isolate select and\n # viewports)\n sets = [s for s in sets if s not in view_sets]\n sets = [s for s in sets if s not in defaults]\n\n return sets\n\n\ndef get_container_transforms(container, members=None, root=False):\n \"\"\"Retrieve the root node of the container content\n\n When a container is created through a Loader the content\n of the file will be grouped under a transform. The name of the root\n transform is stored in the container information\n\n Args:\n container (dict): the container\n members (list): optional and convenience argument\n root (bool): return highest node in hierarchy if True\n\n Returns:\n root (list / str):\n \"\"\"\n\n if not members:\n members = get_container_members(container)\n\n results = cmds.ls(members, type=\"transform\", long=True)\n if root:\n root = get_highest_in_hierarchy(results)\n if root:\n results = root[0]\n\n return results\n\n\ndef get_highest_in_hierarchy(nodes):\n \"\"\"Return highest nodes in the hierarchy that are in the `nodes` list.\n\n The \"highest in hierarchy\" are the nodes closest to world: top-most level.\n\n Args:\n nodes (list): The nodes in which find the highest in hierarchies.\n\n Returns:\n list: The highest nodes from the input nodes.\n\n \"\"\"\n\n # Ensure we use long names\n nodes = cmds.ls(nodes, long=True)\n lookup = set(nodes)\n\n highest = []\n for node in nodes:\n # If no parents are within the nodes input list\n # then this is a highest node\n if not any(n in lookup for n in iter_parents(node)):\n highest.append(node)\n\n return highest\n\n\ndef iter_parents(node):\n \"\"\"Iter parents of node from its long name.\n\n Note: The `node` *must* be the long node name.\n\n Args:\n node (str): Node long name.\n\n Yields:\n str: All parent node names (long names)\n\n \"\"\"\n while True:\n split = node.rsplit(\"|\", 1)\n if len(split) == 1 or not split[0]:\n return\n\n node = split[0]\n yield node\n\n\ndef remove_other_uv_sets(mesh):\n \"\"\"Remove all other UV sets than the current UV set.\n\n Keep only current UV set and ensure it's the renamed to default 'map1'.\n\n \"\"\"\n\n uvSets = cmds.polyUVSet(mesh, query=True, allUVSets=True)\n current = cmds.polyUVSet(mesh, query=True, currentUVSet=True)[0]\n\n # Copy over to map1\n if current != 'map1':\n cmds.polyUVSet(mesh, uvSet=current, newUVSet='map1', copy=True)\n cmds.polyUVSet(mesh, currentUVSet=True, uvSet='map1')\n current = 'map1'\n\n # Delete all non-current UV sets\n deleteUVSets = [uvSet for uvSet in uvSets if uvSet != current]\n uvSet = None\n\n # Maya Bug (tested in 2015/2016):\n # In some cases the API's MFnMesh will report less UV sets than\n # maya.cmds.polyUVSet. This seems to happen when the deletion of UV sets\n # has not triggered a cleanup of the UVSet array attribute on the mesh\n # node. It will still have extra entries in the attribute, though it will\n # not show up in API or UI. Nevertheless it does show up in\n # maya.cmds.polyUVSet. To ensure we clean up the array we'll force delete\n # the extra remaining 'indices' that we don't want.\n\n # TODO: Implement a better fix\n # The best way to fix would be to get the UVSet indices from api with\n # MFnMesh (to ensure we keep correct ones) and then only force delete the\n # other entries in the array attribute on the node. But for now we're\n # deleting all entries except first one. Note that the first entry could\n # never be removed (the default 'map1' always exists and is supposed to\n # be undeletable.)\n try:\n for uvSet in deleteUVSets:\n cmds.polyUVSet(mesh, delete=True, uvSet=uvSet)\n except RuntimeError as exc:\n log.warning('Error uvSet: %s - %s', uvSet, exc)\n indices = cmds.getAttr('{0}.uvSet'.format(mesh),\n multiIndices=True)\n if not indices:\n log.warning(\"No uv set found indices for: %s\", mesh)\n return\n\n # Delete from end to avoid shifting indices\n # and remove the indices in the attribute\n indices = reversed(indices[1:])\n for i in indices:\n attr = '{0}.uvSet[{1}]'.format(mesh, i)\n cmds.removeMultiInstance(attr, b=True)\n\n\ndef get_node_parent(node):\n \"\"\"Return full path name for parent of node\"\"\"\n parents = cmds.listRelatives(node, parent=True, fullPath=True)\n return parents[0] if parents else None\n\n\ndef get_id_from_sibling(node, history_only=True):\n \"\"\"Return first node id in the history chain that matches this node.\n\n The nodes in history must be of the exact same node type and must be\n parented under the same parent.\n\n Optionally, if no matching node is found from the history, all the\n siblings of the node that are of the same type are checked.\n Additionally to having the same parent, the sibling must be marked as\n 'intermediate object'.\n\n Args:\n node (str): node to retrieve the history from\n history_only (bool): if True and if nothing found in history,\n look for an 'intermediate object' in all the node's siblings\n of same type\n\n Returns:\n str or None: The id from the sibling node or None when no id found\n on any valid nodes in the history or siblings.\n\n \"\"\"\n\n node = cmds.ls(node, long=True)[0]\n\n # Find all similar nodes in history\n history = cmds.listHistory(node)\n node_type = cmds.nodeType(node)\n similar_nodes = cmds.ls(history, exactType=node_type, long=True)\n\n # Exclude itself\n similar_nodes = [x for x in similar_nodes if x != node]\n\n # The node *must be* under the same parent\n parent = get_node_parent(node)\n similar_nodes = [i for i in similar_nodes if get_node_parent(i) == parent]\n\n # Check all of the remaining similar nodes and take the first one\n # with an id and assume it's the original.\n for similar_node in similar_nodes:\n _id = get_id(similar_node)\n if _id:\n return _id\n\n if not history_only:\n # Get siblings of same type\n similar_nodes = cmds.listRelatives(parent,\n type=node_type,\n fullPath=True)\n similar_nodes = cmds.ls(similar_nodes, exactType=node_type, long=True)\n\n # Exclude itself\n similar_nodes = [x for x in similar_nodes if x != node]\n\n # Get all unique ids from siblings in order since\n # we consistently take the first one found\n sibling_ids = OrderedDict()\n for similar_node in similar_nodes:\n # Check if \"intermediate object\"\n if not cmds.getAttr(similar_node + \".intermediateObject\"):\n continue\n\n _id = get_id(similar_node)\n if not _id:\n continue\n\n if _id in sibling_ids:\n sibling_ids[_id].append(similar_node)\n else:\n sibling_ids[_id] = [similar_node]\n\n if sibling_ids:\n first_id, found_nodes = next(iter(sibling_ids.items()))\n\n # Log a warning if we've found multiple unique ids\n if len(sibling_ids) > 1:\n log.warning((\"Found more than 1 intermediate shape with\"\n \" unique id for '{}'. Using id of first\"\n \" found: '{}'\".format(node, found_nodes[0])))\n\n return first_id\n\n\ndef set_scene_fps(fps, update=True):\n \"\"\"Set FPS from project configuration\n\n Args:\n fps (int, float): desired FPS\n update(bool): toggle update animation, default is True\n\n Returns:\n None\n\n \"\"\"\n\n fps_mapping = {\n '15': 'game',\n '24': 'film',\n '25': 'pal',\n '30': 'ntsc',\n '48': 'show',\n '50': 'palf',\n '60': 'ntscf',\n '23.976023976023978': '23.976fps',\n '29.97002997002997': '29.97fps',\n '47.952047952047955': '47.952fps',\n '59.94005994005994': '59.94fps',\n '44100': '44100fps',\n '48000': '48000fps'\n }\n\n unit = fps_mapping.get(str(convert_to_maya_fps(fps)), None)\n if unit is None:\n raise ValueError(\"Unsupported FPS value: `%s`\" % fps)\n\n # Get time slider current state\n start_frame = cmds.playbackOptions(query=True, minTime=True)\n end_frame = cmds.playbackOptions(query=True, maxTime=True)\n\n # Get animation data\n animation_start = cmds.playbackOptions(query=True, animationStartTime=True)\n animation_end = cmds.playbackOptions(query=True, animationEndTime=True)\n\n current_frame = cmds.currentTime(query=True)\n\n log.info(\"Setting scene FPS to: '{}'\".format(unit))\n cmds.currentUnit(time=unit, updateAnimation=update)\n\n # Set time slider data back to previous state\n cmds.playbackOptions(edit=True, minTime=start_frame)\n cmds.playbackOptions(edit=True, maxTime=end_frame)\n\n # Set animation data\n cmds.playbackOptions(edit=True, animationStartTime=animation_start)\n cmds.playbackOptions(edit=True, animationEndTime=animation_end)\n\n cmds.currentTime(current_frame, edit=True, update=True)\n\n # Force file stated to 'modified'\n cmds.file(modified=True)\n\n\ndef set_scene_resolution(width, height, pixelAspect):\n \"\"\"Set the render resolution\n\n Args:\n width(int): value of the width\n height(int): value of the height\n\n Returns:\n None\n\n \"\"\"\n\n control_node = \"defaultResolution\"\n current_renderer = cmds.getAttr(\"defaultRenderGlobals.currentRenderer\")\n aspect_ratio_attr = \"deviceAspectRatio\"\n\n # Give VRay a helping hand as it is slightly different from the rest\n if current_renderer == \"vray\":\n aspect_ratio_attr = \"aspectRatio\"\n vray_node = \"vraySettings\"\n if cmds.objExists(vray_node):\n control_node = vray_node\n else:\n log.error(\"Can't set VRay resolution because there is no node \"\n \"named: `%s`\" % vray_node)\n\n log.info(\"Setting scene resolution to: %s x %s\" % (width, height))\n cmds.setAttr(\"%s.width\" % control_node, width)\n cmds.setAttr(\"%s.height\" % control_node, height)\n\n deviceAspectRatio = ((float(width) / float(height)) * float(pixelAspect))\n cmds.setAttr(\n \"{}.{}\".format(control_node, aspect_ratio_attr), deviceAspectRatio)\n cmds.setAttr(\"%s.pixelAspect\" % control_node, pixelAspect)\n\n\ndef get_fps_for_current_context():\n \"\"\"Get fps that should be set for current context.\n\n Todos:\n - Skip project value.\n - Merge logic with 'get_frame_range' and 'reset_scene_resolution' ->\n all the values in the functions can be collected at one place as\n they have same requirements.\n\n Returns:\n Union[int, float]: FPS value.\n \"\"\"\n\n project_name = get_current_project_name()\n asset_name = get_current_asset_name()\n asset_doc = get_asset_by_name(\n project_name, asset_name, fields=[\"data.fps\"]\n ) or {}\n fps = asset_doc.get(\"data\", {}).get(\"fps\")\n if not fps:\n project_doc = get_project(project_name, fields=[\"data.fps\"]) or {}\n fps = project_doc.get(\"data\", {}).get(\"fps\")\n\n if not fps:\n fps = 25\n\n return convert_to_maya_fps(fps)\n\n\ndef get_frame_range(include_animation_range=False):\n \"\"\"Get the current assets frame range and handles.\n\n Args:\n include_animation_range (bool, optional): Whether to include\n `animationStart` and `animationEnd` keys to define the outer\n range of the timeline. It is excluded by default.\n\n Returns:\n dict: Asset's expected frame range values.\n\n \"\"\"\n\n # Set frame start/end\n project_name = get_current_project_name()\n asset_name = get_current_asset_name()\n asset = get_asset_by_name(project_name, asset_name)\n\n frame_start = asset[\"data\"].get(\"frameStart\")\n frame_end = asset[\"data\"].get(\"frameEnd\")\n\n if frame_start is None or frame_end is None:\n cmds.warning(\"No edit information found for %s\" % asset_name)\n return\n\n handle_start = asset[\"data\"].get(\"handleStart\") or 0\n handle_end = asset[\"data\"].get(\"handleEnd\") or 0\n\n frame_range = {\n \"frameStart\": frame_start,\n \"frameEnd\": frame_end,\n \"handleStart\": handle_start,\n \"handleEnd\": handle_end\n }\n if include_animation_range:\n # The animation range values are only included to define whether\n # the Maya time slider should include the handles or not.\n # Some usages of this function use the full dictionary to define\n # instance attributes for which we want to exclude the animation\n # keys. That is why these are excluded by default.\n task_name = get_current_task_name()\n settings = get_project_settings(project_name)\n include_handles_settings = settings[\"maya\"][\"include_handles\"]\n current_task = asset.get(\"data\").get(\"tasks\").get(task_name)\n\n animation_start = frame_start\n animation_end = frame_end\n\n include_handles = include_handles_settings[\"include_handles_default\"]\n for item in include_handles_settings[\"per_task_type\"]:\n if current_task[\"type\"] in item[\"task_type\"]:\n include_handles = item[\"include_handles\"]\n break\n if include_handles:\n animation_start -= int(handle_start)\n animation_end += int(handle_end)\n\n frame_range[\"animationStart\"] = animation_start\n frame_range[\"animationEnd\"] = animation_end\n\n return frame_range\n\n\ndef reset_frame_range(playback=True, render=True, fps=True):\n \"\"\"Set frame range to current asset\n\n Args:\n playback (bool, Optional): Whether to set the maya timeline playback\n frame range. Defaults to True.\n render (bool, Optional): Whether to set the maya render frame range.\n Defaults to True.\n fps (bool, Optional): Whether to set scene FPS. Defaults to True.\n \"\"\"\n if fps:\n set_scene_fps(get_fps_for_current_context())\n\n frame_range = get_frame_range(include_animation_range=True)\n if not frame_range:\n # No frame range data found for asset\n return\n\n frame_start = frame_range[\"frameStart\"]\n frame_end = frame_range[\"frameEnd\"]\n animation_start = frame_range[\"animationStart\"]\n animation_end = frame_range[\"animationEnd\"]\n\n if playback:\n cmds.playbackOptions(\n minTime=frame_start,\n maxTime=frame_end,\n animationStartTime=animation_start,\n animationEndTime=animation_end\n )\n cmds.currentTime(frame_start)\n\n if render:\n cmds.setAttr(\"defaultRenderGlobals.startFrame\", animation_start)\n cmds.setAttr(\"defaultRenderGlobals.endFrame\", animation_end)\n\n\ndef reset_scene_resolution():\n \"\"\"Apply the scene resolution from the project definition\n\n scene resolution can be overwritten by an asset if the asset.data contains\n any information regarding scene resolution .\n\n Returns:\n None\n \"\"\"\n\n project_name = get_current_project_name()\n project_doc = get_project(project_name)\n project_data = project_doc[\"data\"]\n asset_data = get_current_project_asset()[\"data\"]\n\n # Set project resolution\n width_key = \"resolutionWidth\"\n height_key = \"resolutionHeight\"\n pixelAspect_key = \"pixelAspect\"\n\n width = asset_data.get(width_key, project_data.get(width_key, 1920))\n height = asset_data.get(height_key, project_data.get(height_key, 1080))\n pixelAspect = asset_data.get(pixelAspect_key,\n project_data.get(pixelAspect_key, 1))\n\n set_scene_resolution(width, height, pixelAspect)\n\n\ndef set_context_settings():\n \"\"\"Apply the project settings from the project definition\n\n Settings can be overwritten by an asset if the asset.data contains\n any information regarding those settings.\n\n Examples of settings:\n fps\n resolution\n renderer\n\n Returns:\n None\n \"\"\"\n\n\n # Set project fps\n set_scene_fps(get_fps_for_current_context())\n\n reset_scene_resolution()\n\n # Set frame range.\n reset_frame_range()\n\n # Set colorspace\n set_colorspace()\n\n\n# Valid FPS\ndef validate_fps():\n \"\"\"Validate current scene FPS and show pop-up when it is incorrect\n\n Returns:\n bool\n\n \"\"\"\n\n expected_fps = get_fps_for_current_context()\n current_fps = mel.eval('currentTimeUnitToFPS()')\n\n fps_match = current_fps == expected_fps\n if not fps_match and not IS_HEADLESS:\n from openpype.widgets import popup\n\n parent = get_main_window()\n\n dialog = popup.PopupUpdateKeys(parent=parent)\n dialog.setModal(True)\n dialog.setWindowTitle(\"Maya scene does not match project FPS\")\n dialog.setMessage(\n \"Scene {} FPS does not match project {} FPS\".format(\n current_fps, expected_fps\n )\n )\n dialog.setButtonText(\"Fix\")\n\n # Set new text for button (add optional argument for the popup?)\n toggle = dialog.widgets[\"toggle\"]\n update = toggle.isChecked()\n dialog.on_clicked_state.connect(\n lambda: set_scene_fps(expected_fps, update)\n )\n\n dialog.show()\n\n return False\n\n return fps_match\n\n\ndef bake(nodes,\n frame_range=None,\n step=1.0,\n simulation=True,\n preserve_outside_keys=False,\n disable_implicit_control=True,\n shape=True):\n \"\"\"Bake the given nodes over the time range.\n\n This will bake all attributes of the node, including custom attributes.\n\n Args:\n nodes (list): Names of transform nodes, eg. camera, light.\n frame_range (list): frame range with start and end frame.\n or if None then takes timeSliderRange\n simulation (bool): Whether to perform a full simulation of the\n attributes over time.\n preserve_outside_keys (bool): Keep keys that are outside of the baked\n range.\n disable_implicit_control (bool): When True will disable any\n constraints to the object.\n shape (bool): When True also bake attributes on the children shapes.\n step (float): The step size to sample by.\n\n Returns:\n None\n\n \"\"\"\n\n # Parse inputs\n if not nodes:\n return\n\n assert isinstance(nodes, (list, tuple)), \"Nodes must be a list or tuple\"\n\n # If frame range is None fall back to time slider playback time range\n if frame_range is None:\n frame_range = [cmds.playbackOptions(query=True, minTime=True),\n cmds.playbackOptions(query=True, maxTime=True)]\n\n # If frame range is single frame bake one frame more,\n # otherwise maya.cmds.bakeResults gets confused\n if frame_range[1] == frame_range[0]:\n frame_range[1] += 1\n\n # Bake it\n with keytangent_default(in_tangent_type='auto',\n out_tangent_type='auto'):\n cmds.bakeResults(nodes,\n simulation=simulation,\n preserveOutsideKeys=preserve_outside_keys,\n disableImplicitControl=disable_implicit_control,\n shape=shape,\n sampleBy=step,\n time=(frame_range[0], frame_range[1]))\n\n\ndef bake_to_world_space(nodes,\n frame_range=None,\n simulation=True,\n preserve_outside_keys=False,\n disable_implicit_control=True,\n shape=True,\n step=1.0):\n \"\"\"Bake the nodes to world space transformation (incl. other attributes)\n\n Bakes the transforms to world space (while maintaining all its animated\n attributes and settings) by duplicating the node. Then parents it to world\n and constrains to the original.\n\n Other attributes are also baked by connecting all attributes directly.\n Baking is then done using Maya's bakeResults command.\n\n See `bake` for the argument documentation.\n\n Returns:\n list: The newly created and baked node names.\n\n \"\"\"\n @contextlib.contextmanager\n def _unlock_attr(attr):\n \"\"\"Unlock attribute during context if it is locked\"\"\"\n if not cmds.getAttr(attr, lock=True):\n # If not locked, do nothing\n yield\n return\n try:\n cmds.setAttr(attr, lock=False)\n yield\n finally:\n cmds.setAttr(attr, lock=True)\n\n def _get_attrs(node):\n \"\"\"Workaround for buggy shape attribute listing with listAttr\n\n This will only return keyable settable attributes that have an\n incoming connections (those that have a reason to be baked).\n\n Technically this *may* fail to return attributes driven by complex\n expressions for which maya makes no connections, e.g. doing actual\n `setAttr` calls in expressions.\n\n Arguments:\n node (str): The node to list attributes for.\n\n Returns:\n list: Keyable attributes with incoming connections.\n The attribute may be locked.\n\n \"\"\"\n attrs = cmds.listAttr(node,\n write=True,\n scalar=True,\n settable=True,\n connectable=True,\n keyable=True,\n shortNames=True) or []\n valid_attrs = []\n for attr in attrs:\n node_attr = '{0}.{1}'.format(node, attr)\n\n # Sometimes Maya returns 'non-existent' attributes for shapes\n # so we filter those out\n if not cmds.attributeQuery(attr, node=node, exists=True):\n continue\n\n # We only need those that have a connection, just to be safe\n # that it's actually keyable/connectable anyway.\n if cmds.connectionInfo(node_attr,\n isDestination=True):\n valid_attrs.append(attr)\n\n return valid_attrs\n\n transform_attrs = {\"t\", \"r\", \"s\",\n \"tx\", \"ty\", \"tz\",\n \"rx\", \"ry\", \"rz\",\n \"sx\", \"sy\", \"sz\"}\n\n world_space_nodes = []\n with ExitStack() as stack:\n delete_bin = stack.enter_context(delete_after())\n # Create the duplicate nodes that are in world-space connected to\n # the originals\n for node in nodes:\n\n # Duplicate the node\n short_name = node.rsplit(\"|\", 1)[-1]\n new_name = \"{0}_baked\".format(short_name)\n new_node = cmds.duplicate(node,\n name=new_name,\n renameChildren=True)[0] # noqa\n\n # Parent new node to world\n if cmds.listRelatives(new_node, parent=True):\n new_node = cmds.parent(new_node, world=True)[0]\n\n # Temporarily unlock and passthrough connect all attributes\n # so we can bake them over time\n # Skip transform attributes because we will constrain them later\n attrs = set(_get_attrs(node)) - transform_attrs\n for attr in attrs:\n orig_node_attr = \"{}.{}\".format(node, attr)\n new_node_attr = \"{}.{}\".format(new_node, attr)\n\n # unlock during context to avoid connection errors\n stack.enter_context(_unlock_attr(new_node_attr))\n cmds.connectAttr(orig_node_attr,\n new_node_attr,\n force=True)\n\n # If shapes are also baked then also temporarily unlock and\n # passthrough connect all shape attributes for baking\n if shape:\n children_shapes = cmds.listRelatives(new_node,\n children=True,\n fullPath=True,\n shapes=True)\n if children_shapes:\n orig_children_shapes = cmds.listRelatives(node,\n children=True,\n fullPath=True,\n shapes=True)\n for orig_shape, new_shape in zip(orig_children_shapes,\n children_shapes):\n attrs = _get_attrs(orig_shape)\n for attr in attrs:\n orig_node_attr = \"{}.{}\".format(orig_shape, attr)\n new_node_attr = \"{}.{}\".format(new_shape, attr)\n\n # unlock during context to avoid connection errors\n stack.enter_context(_unlock_attr(new_node_attr))\n cmds.connectAttr(orig_node_attr,\n new_node_attr,\n force=True)\n\n # Constraint transforms\n for attr in transform_attrs:\n transform_attr = \"{}.{}\".format(new_node, attr)\n stack.enter_context(_unlock_attr(transform_attr))\n delete_bin.extend(cmds.parentConstraint(node, new_node, mo=False))\n delete_bin.extend(cmds.scaleConstraint(node, new_node, mo=False))\n\n world_space_nodes.append(new_node)\n\n bake(world_space_nodes,\n frame_range=frame_range,\n step=step,\n simulation=simulation,\n preserve_outside_keys=preserve_outside_keys,\n disable_implicit_control=disable_implicit_control,\n shape=shape)\n\n return world_space_nodes\n\n\ndef load_capture_preset(data):\n \"\"\"Convert OpenPype Extract Playblast settings to `capture` arguments\n\n Input data is the settings from:\n `project_settings/maya/publish/ExtractPlayblast/capture_preset`\n\n Args:\n data (dict): Capture preset settings from OpenPype settings\n\n Returns:\n dict: `capture.capture` compatible keyword arguments\n\n \"\"\"\n\n options = dict()\n viewport_options = dict()\n viewport2_options = dict()\n camera_options = dict()\n\n # Straight key-value match from settings to capture arguments\n options.update(data[\"Codec\"])\n options.update(data[\"Generic\"])\n options.update(data[\"Resolution\"])\n\n camera_options.update(data['Camera Options'])\n viewport_options.update(data[\"Renderer\"])\n\n # DISPLAY OPTIONS\n disp_options = {}\n for key, value in data['Display Options'].items():\n if key.startswith('background'):\n # Convert background, backgroundTop, backgroundBottom colors\n if len(value) == 4:\n # Ignore alpha + convert RGB to float\n value = [\n float(value[0]) / 255,\n float(value[1]) / 255,\n float(value[2]) / 255\n ]\n disp_options[key] = value\n elif key == \"displayGradient\":\n disp_options[key] = value\n\n options['display_options'] = disp_options\n\n # Viewport Options has a mixture of Viewport2 Options and Viewport Options\n # to pass along to capture. So we'll need to differentiate between the two\n VIEWPORT2_OPTIONS = {\n \"textureMaxResolution\",\n \"renderDepthOfField\",\n \"ssaoEnable\",\n \"ssaoSamples\",\n \"ssaoAmount\",\n \"ssaoRadius\",\n \"ssaoFilterRadius\",\n \"hwFogStart\",\n \"hwFogEnd\",\n \"hwFogAlpha\",\n \"hwFogFalloff\",\n \"hwFogColorR\",\n \"hwFogColorG\",\n \"hwFogColorB\",\n \"hwFogDensity\",\n \"motionBlurEnable\",\n \"motionBlurSampleCount\",\n \"motionBlurShutterOpenFraction\",\n \"lineAAEnable\"\n }\n for key, value in data['Viewport Options'].items():\n\n # There are some keys we want to ignore\n if key in {\"override_viewport_options\", \"high_quality\"}:\n continue\n\n # First handle special cases where we do value conversion to\n # separate option values\n if key == 'textureMaxResolution':\n viewport2_options['textureMaxResolution'] = value\n if value > 0:\n viewport2_options['enableTextureMaxRes'] = True\n viewport2_options['textureMaxResMode'] = 1\n else:\n viewport2_options['enableTextureMaxRes'] = False\n viewport2_options['textureMaxResMode'] = 0\n\n elif key == 'multiSample':\n viewport2_options['multiSampleEnable'] = value > 0\n viewport2_options['multiSampleCount'] = value\n\n elif key == 'alphaCut':\n viewport2_options['transparencyAlgorithm'] = 5\n viewport2_options['transparencyQuality'] = 1\n\n elif key == 'hwFogFalloff':\n # Settings enum value string to integer\n viewport2_options['hwFogFalloff'] = int(value)\n\n # Then handle Viewport 2.0 Options\n elif key in VIEWPORT2_OPTIONS:\n viewport2_options[key] = value\n\n # Then assume remainder is Viewport Options\n else:\n viewport_options[key] = value\n\n options['viewport_options'] = viewport_options\n options['viewport2_options'] = viewport2_options\n options['camera_options'] = camera_options\n\n # use active sound track\n scene = capture.parse_active_scene()\n options['sound'] = scene['sound']\n\n return options\n\n\ndef get_attr_in_layer(attr, layer):\n \"\"\"Return attribute value in specified renderlayer.\n\n Same as cmds.getAttr but this gets the attribute's value in a\n given render layer without having to switch to it.\n\n Warning for parent attribute overrides:\n Attributes that have render layer overrides to their parent attribute\n are not captured correctly since they do not have a direct connection.\n For example, an override to sphere.rotate when querying sphere.rotateX\n will not return correctly!\n\n Note: This is much faster for Maya's renderLayer system, yet the code\n does no optimized query for render setup.\n\n Args:\n attr (str): attribute name, ex. \"node.attribute\"\n layer (str): layer name\n\n Returns:\n The return value from `maya.cmds.getAttr`\n\n \"\"\"\n\n try:\n if cmds.mayaHasRenderSetup():\n from . import lib_rendersetup\n return lib_rendersetup.get_attr_in_layer(attr, layer)\n except AttributeError:\n pass\n\n # Ignore complex query if we're in the layer anyway\n current_layer = cmds.editRenderLayerGlobals(query=True,\n currentRenderLayer=True)\n if layer == current_layer:\n return cmds.getAttr(attr)\n\n connections = cmds.listConnections(attr,\n plugs=True,\n source=False,\n destination=True,\n type=\"renderLayer\") or []\n connections = filter(lambda x: x.endswith(\".plug\"), connections)\n if not connections:\n return cmds.getAttr(attr)\n\n # Some value types perform a conversion when assigning\n # TODO: See if there's a maya method to allow this conversion\n # instead of computing it ourselves.\n attr_type = cmds.getAttr(attr, type=True)\n conversion = None\n if attr_type == \"time\":\n conversion = mel.eval('currentTimeUnitToFPS()') # returns float\n elif attr_type == \"doubleAngle\":\n # Radians to Degrees: 180 / pi\n # TODO: This will likely only be correct when Maya units are set\n # to degrees\n conversion = 57.2957795131\n elif attr_type == \"doubleLinear\":\n raise NotImplementedError(\"doubleLinear conversion not implemented.\")\n\n for connection in connections:\n if connection.startswith(layer + \".\"):\n attr_split = connection.split(\".\")\n if attr_split[0] == layer:\n attr = \".\".join(attr_split[0:-1])\n value = cmds.getAttr(\"%s.value\" % attr)\n if conversion:\n value *= conversion\n return value\n\n else:\n # When connections are present, but none\n # to the specific renderlayer than the layer\n # should have the \"defaultRenderLayer\"'s value\n layer = \"defaultRenderLayer\"\n for connection in connections:\n if connection.startswith(layer):\n attr_split = connection.split(\".\")\n if attr_split[0] == \"defaultRenderLayer\":\n attr = \".\".join(attr_split[0:-1])\n value = cmds.getAttr(\"%s.value\" % attr)\n if conversion:\n value *= conversion\n return value\n\n return cmds.getAttr(attr)\n\n\ndef fix_incompatible_containers():\n \"\"\"Backwards compatibility: old containers to use new ReferenceLoader\"\"\"\n old_loaders = {\n \"MayaAsciiLoader\",\n \"AbcLoader\",\n \"ModelLoader\",\n \"CameraLoader\",\n \"RigLoader\",\n \"FBXLoader\"\n }\n host = registered_host()\n for container in host.ls():\n loader = container['loader']\n if loader in old_loaders:\n log.info(\n \"Converting legacy container loader {} to \"\n \"ReferenceLoader: {}\".format(loader, container[\"objectName\"])\n )\n cmds.setAttr(container[\"objectName\"] + \".loader\",\n \"ReferenceLoader\", type=\"string\")\n\n\ndef update_content_on_context_change():\n \"\"\"\n This will update scene content to match new asset on context change\n \"\"\"\n scene_sets = cmds.listSets(allSets=True)\n asset_doc = get_current_project_asset()\n new_asset = asset_doc[\"name\"]\n new_data = asset_doc[\"data\"]\n for s in scene_sets:\n try:\n if cmds.getAttr(\"{}.id\".format(s)) == \"pyblish.avalon.instance\":\n attr = cmds.listAttr(s)\n print(s)\n if \"asset\" in attr:\n print(\" - setting asset to: [ {} ]\".format(new_asset))\n cmds.setAttr(\"{}.asset\".format(s),\n new_asset, type=\"string\")\n if \"frameStart\" in attr:\n cmds.setAttr(\"{}.frameStart\".format(s),\n new_data[\"frameStart\"])\n if \"frameEnd\" in attr:\n cmds.setAttr(\"{}.frameEnd\".format(s),\n new_data[\"frameEnd\"],)\n except ValueError:\n pass\n\n\ndef show_message(title, msg):\n from qtpy import QtWidgets\n from openpype.widgets import message_window\n\n # Find maya main window\n top_level_widgets = {w.objectName(): w for w in\n QtWidgets.QApplication.topLevelWidgets()}\n\n parent = top_level_widgets.get(\"MayaWindow\", None)\n if parent is None:\n pass\n else:\n message_window.message(title=title, message=msg, parent=parent)\n\n\ndef iter_shader_edits(relationships, shader_nodes, nodes_by_id, label=None):\n \"\"\"Yield edits as a set of actions.\"\"\"\n\n attributes = relationships.get(\"attributes\", [])\n shader_data = relationships.get(\"relationships\", {})\n\n shading_engines = cmds.ls(shader_nodes, type=\"objectSet\", long=True)\n assert shading_engines, \"Error in retrieving objectSets from reference\"\n\n # region compute lookup\n shading_engines_by_id = defaultdict(list)\n for shad in shading_engines:\n shading_engines_by_id[get_id(shad)].append(shad)\n # endregion\n\n # region assign shading engines and other sets\n for data in shader_data.values():\n # collect all unique IDs of the set members\n shader_uuid = data[\"uuid\"]\n member_uuids = [\n (member[\"uuid\"], member.get(\"components\"))\n for member in data[\"members\"]]\n\n filtered_nodes = list()\n for _uuid, components in member_uuids:\n nodes = nodes_by_id.get(_uuid, None)\n if nodes is None:\n continue\n\n if components:\n # Assign to the components\n nodes = [\".\".join([node, components]) for node in nodes]\n\n filtered_nodes.extend(nodes)\n\n id_shading_engines = shading_engines_by_id[shader_uuid]\n if not id_shading_engines:\n log.error(\"{} - No shader found with cbId \"\n \"'{}'\".format(label, shader_uuid))\n continue\n elif len(id_shading_engines) > 1:\n log.error(\"{} - Skipping shader assignment. \"\n \"More than one shader found with cbId \"\n \"'{}'. (found: {})\".format(label, shader_uuid,\n id_shading_engines))\n continue\n\n if not filtered_nodes:\n log.warning(\"{} - No nodes found for shading engine \"\n \"'{}'\".format(label, id_shading_engines[0]))\n continue\n\n yield {\"action\": \"assign\",\n \"uuid\": data[\"uuid\"],\n \"nodes\": filtered_nodes,\n \"shader\": id_shading_engines[0]}\n\n for data in attributes:\n nodes = nodes_by_id.get(data[\"uuid\"], [])\n attr_value = data[\"attributes\"]\n yield {\"action\": \"setattr\",\n \"uuid\": data[\"uuid\"],\n \"nodes\": nodes,\n \"attributes\": attr_value}\n\n\ndef set_colorspace():\n \"\"\"Set Colorspace from project configuration\"\"\"\n\n project_name = get_current_project_name()\n imageio = get_project_settings(project_name)[\"maya\"][\"imageio\"]\n\n # ocio compatibility variables\n ocio_v2_maya_version = 2022\n maya_version = int(cmds.about(version=True))\n ocio_v2_support = use_ocio_v2 = maya_version >= ocio_v2_maya_version\n is_ocio_set = bool(os.environ.get(\"OCIO\"))\n\n use_workfile_settings = imageio.get(\"workfile\", {}).get(\"enabled\")\n if use_workfile_settings:\n root_dict = imageio[\"workfile\"]\n else:\n # TODO: deprecated code from 3.15.5 - remove\n # Maya 2022+ introduces new OCIO v2 color management settings that\n # can override the old color management preferences. OpenPype has\n # separate settings for both so we fall back when necessary.\n use_ocio_v2 = imageio[\"colorManagementPreference_v2\"][\"enabled\"]\n if use_ocio_v2 and not ocio_v2_support:\n # Fallback to legacy behavior with a warning\n log.warning(\n \"Color Management Preference v2 is enabled but not \"\n \"supported by current Maya version: {} (< {}). Falling \"\n \"back to legacy settings.\".format(\n maya_version, ocio_v2_maya_version)\n )\n\n if use_ocio_v2:\n root_dict = imageio[\"colorManagementPreference_v2\"]\n else:\n root_dict = imageio[\"colorManagementPreference\"]\n\n if not isinstance(root_dict, dict):\n msg = \"set_colorspace(): argument should be dictionary\"\n log.error(msg)\n return\n\n # backward compatibility\n # TODO: deprecated code from 3.15.5 - remove with deprecated code above\n view_name = root_dict.get(\"viewTransform\")\n if view_name is None:\n view_name = root_dict.get(\"viewName\")\n\n log.debug(\">> root_dict: {}\".format(pformat(root_dict)))\n if not root_dict:\n return\n\n # set color spaces for rendering space and view transforms\n def _colormanage(**kwargs):\n \"\"\"Wrapper around `cmds.colorManagementPrefs`.\n\n This logs errors instead of raising an error so color management\n settings get applied as much as possible.\n\n \"\"\"\n assert len(kwargs) == 1, \"Must receive one keyword argument\"\n try:\n cmds.colorManagementPrefs(edit=True, **kwargs)\n log.debug(\"Setting Color Management Preference: {}\".format(kwargs))\n except RuntimeError as exc:\n log.error(exc)\n\n # enable color management\n cmds.colorManagementPrefs(edit=True, cmEnabled=True)\n cmds.colorManagementPrefs(edit=True, ocioRulesEnabled=True)\n\n if use_ocio_v2:\n log.info(\"Using Maya OCIO v2\")\n if not is_ocio_set:\n # Set the Maya 2022+ default OCIO v2 config file path\n log.info(\"Setting default Maya OCIO v2 config\")\n # Note: Setting \"\" as value also sets this default however\n # introduces a bug where launching a file on startup will prompt\n # to save the empty scene before it, so we set using the path.\n # This value has been the same for 2022, 2023 and 2024\n path = \"/OCIO-configs/Maya2022-default/config.ocio\"\n cmds.colorManagementPrefs(edit=True, configFilePath=path)\n\n # set rendering space and view transform\n _colormanage(renderingSpaceName=root_dict[\"renderSpace\"])\n _colormanage(viewName=view_name)\n _colormanage(displayName=root_dict[\"displayName\"])\n else:\n log.info(\"Using Maya OCIO v1 (legacy)\")\n if not is_ocio_set:\n # Set the Maya default config file path\n log.info(\"Setting default Maya OCIO v1 legacy config\")\n cmds.colorManagementPrefs(edit=True, configFilePath=\"legacy\")\n\n # set rendering space and view transform\n _colormanage(renderingSpaceName=root_dict[\"renderSpace\"])\n _colormanage(viewTransformName=view_name)\n\n\n@contextlib.contextmanager\ndef parent_nodes(nodes, parent=None):\n # type: (list, str) -> list\n \"\"\"Context manager to un-parent provided nodes and return them back.\"\"\"\n\n def _as_mdagpath(node):\n \"\"\"Return MDagPath for node path.\"\"\"\n if not node:\n return\n sel = OpenMaya.MSelectionList()\n sel.add(node)\n return sel.getDagPath(0)\n\n # We can only parent dag nodes so we ensure input contains only dag nodes\n nodes = cmds.ls(nodes, type=\"dagNode\", long=True)\n if not nodes:\n # opt-out early\n yield\n return\n\n parent_node_path = None\n delete_parent = False\n if parent:\n if not cmds.objExists(parent):\n parent_node = cmds.createNode(\"transform\",\n name=parent,\n skipSelect=False)\n delete_parent = True\n else:\n parent_node = parent\n parent_node_path = cmds.ls(parent_node, long=True)[0]\n\n # Store original parents\n node_parents = []\n for node in nodes:\n node_parent = get_node_parent(node)\n node_parents.append((_as_mdagpath(node), _as_mdagpath(node_parent)))\n\n try:\n for node, node_parent in node_parents:\n node_parent_path = node_parent.fullPathName() if node_parent else None # noqa\n if node_parent_path == parent_node_path:\n # Already a child\n continue\n\n if parent_node_path:\n cmds.parent(node.fullPathName(), parent_node_path)\n else:\n cmds.parent(node.fullPathName(), world=True)\n\n yield\n finally:\n # Reparent to original parents\n for node, original_parent in node_parents:\n node_path = node.fullPathName()\n if not node_path:\n # Node must have been deleted\n continue\n\n node_parent_path = get_node_parent(node_path)\n\n original_parent_path = None\n if original_parent:\n original_parent_path = original_parent.fullPathName()\n if not original_parent_path:\n # Original parent node must have been deleted\n continue\n\n if node_parent_path != original_parent_path:\n if not original_parent_path:\n cmds.parent(node_path, world=True)\n else:\n cmds.parent(node_path, original_parent_path)\n\n if delete_parent:\n cmds.delete(parent_node_path)\n\n\n@contextlib.contextmanager\ndef maintained_time():\n ct = cmds.currentTime(query=True)\n try:\n yield\n finally:\n cmds.currentTime(ct, edit=True)\n\n\ndef iter_visible_nodes_in_range(nodes, start, end):\n \"\"\"Yield nodes that are visible in start-end frame range.\n\n - Ignores intermediateObjects completely.\n - Considers animated visibility attributes + upstream visibilities.\n\n This is optimized for large scenes where some nodes in the parent\n hierarchy might have some input connections to the visibilities,\n e.g. key, driven keys, connections to other attributes, etc.\n\n This only does a single time step to `start` if current frame is\n not inside frame range since the assumption is made that changing\n a frame isn't so slow that it beats querying all visibility\n plugs through MDGContext on another frame.\n\n Args:\n nodes (list): List of node names to consider.\n start (int, float): Start frame.\n end (int, float): End frame.\n\n Returns:\n list: List of node names. These will be long full path names so\n might have a longer name than the input nodes.\n\n \"\"\"\n # States we consider per node\n VISIBLE = 1 # always visible\n INVISIBLE = 0 # always invisible\n ANIMATED = -1 # animated visibility\n\n # Ensure integers\n start = int(start)\n end = int(end)\n\n # Consider only non-intermediate dag nodes and use the \"long\" names.\n nodes = cmds.ls(nodes, long=True, noIntermediate=True, type=\"dagNode\")\n if not nodes:\n return\n\n with maintained_time():\n # Go to first frame of the range if the current time is outside\n # the queried range so can directly query all visible nodes on\n # that frame.\n current_time = cmds.currentTime(query=True)\n if not (start <= current_time <= end):\n cmds.currentTime(start)\n\n visible = cmds.ls(nodes, long=True, visible=True)\n for node in visible:\n yield node\n if len(visible) == len(nodes) or start == end:\n # All are visible on frame one, so they are at least visible once\n # inside the frame range.\n return\n\n # For the invisible ones check whether its visibility and/or\n # any of its parents visibility attributes are animated. If so, it might\n # get visible on other frames in the range.\n def memodict(f):\n \"\"\"Memoization decorator for a function taking a single argument.\n\n See: http://code.activestate.com/recipes/\n 578231-probably-the-fastest-memoization-decorator-in-the-/\n \"\"\"\n\n class memodict(dict):\n def __missing__(self, key):\n ret = self[key] = f(key)\n return ret\n\n return memodict().__getitem__\n\n @memodict\n def get_state(node):\n plug = node + \".visibility\"\n connections = cmds.listConnections(plug,\n source=True,\n destination=False)\n if connections:\n return ANIMATED\n else:\n return VISIBLE if cmds.getAttr(plug) else INVISIBLE\n\n visible = set(visible)\n invisible = [node for node in nodes if node not in visible]\n always_invisible = set()\n # Iterate over the nodes by short to long names to iterate the highest\n # in hierarchy nodes first. So the collected data can be used from the\n # cache for parent queries in next iterations.\n node_dependencies = dict()\n for node in sorted(invisible, key=len):\n\n state = get_state(node)\n if state == INVISIBLE:\n always_invisible.add(node)\n continue\n\n # If not always invisible by itself we should go through and check\n # the parents to see if any of them are always invisible. For those\n # that are \"ANIMATED\" we consider that this node is dependent on\n # that attribute, we store them as dependency.\n dependencies = set()\n if state == ANIMATED:\n dependencies.add(node)\n\n traversed_parents = list()\n for parent in iter_parents(node):\n\n if parent in always_invisible or get_state(parent) == INVISIBLE:\n # When parent is always invisible then consider this parent,\n # this node we started from and any of the parents we\n # have traversed in-between to be *always invisible*\n always_invisible.add(parent)\n always_invisible.add(node)\n always_invisible.update(traversed_parents)\n break\n\n # If we have traversed the parent before and its visibility\n # was dependent on animated visibilities then we can just extend\n # its dependencies for to those for this node and break further\n # iteration upwards.\n parent_dependencies = node_dependencies.get(parent, None)\n if parent_dependencies is not None:\n dependencies.update(parent_dependencies)\n break\n\n state = get_state(parent)\n if state == ANIMATED:\n dependencies.add(parent)\n\n traversed_parents.append(parent)\n\n if node not in always_invisible and dependencies:\n node_dependencies[node] = dependencies\n\n if not node_dependencies:\n return\n\n # Now we only have to check the visibilities for nodes that have animated\n # visibility dependencies upstream. The fastest way to check these\n # visibility attributes across different frames is with Python api 2.0\n # so we do that.\n @memodict\n def get_visibility_mplug(node):\n \"\"\"Return api 2.0 MPlug with cached memoize decorator\"\"\"\n sel = OpenMaya.MSelectionList()\n sel.add(node)\n dag = sel.getDagPath(0)\n return OpenMaya.MFnDagNode(dag).findPlug(\"visibility\", True)\n\n @contextlib.contextmanager\n def dgcontext(mtime):\n \"\"\"MDGContext context manager\"\"\"\n context = OpenMaya.MDGContext(mtime)\n try:\n previous = context.makeCurrent()\n yield context\n finally:\n previous.makeCurrent()\n\n # We skip the first frame as we already used that frame to check for\n # overall visibilities. And end+1 to include the end frame.\n scene_units = OpenMaya.MTime.uiUnit()\n for frame in range(start + 1, end + 1):\n mtime = OpenMaya.MTime(frame, unit=scene_units)\n\n # Build little cache so we don't query the same MPlug's value\n # again if it was checked on this frame and also is a dependency\n # for another node\n frame_visibilities = {}\n with dgcontext(mtime) as context:\n for node, dependencies in list(node_dependencies.items()):\n for dependency in dependencies:\n dependency_visible = frame_visibilities.get(dependency,\n None)\n if dependency_visible is None:\n mplug = get_visibility_mplug(dependency)\n dependency_visible = mplug.asBool(context)\n frame_visibilities[dependency] = dependency_visible\n\n if not dependency_visible:\n # One dependency is not visible, thus the\n # node is not visible.\n break\n\n else:\n # All dependencies are visible.\n yield node\n # Remove node with dependencies for next frame iterations\n # because it was visible at least once.\n node_dependencies.pop(node)\n\n # If no more nodes to process break the frame iterations..\n if not node_dependencies:\n break\n\n\ndef get_attribute_input(attr):\n connections = cmds.listConnections(attr, plugs=True, destination=False)\n return connections[0] if connections else None\n\n\ndef convert_to_maya_fps(fps):\n \"\"\"Convert any fps to supported Maya framerates.\"\"\"\n float_framerates = [\n 23.976023976023978,\n # WTF is 29.97 df vs fps?\n 29.97002997002997,\n 47.952047952047955,\n 59.94005994005994\n ]\n # 44100 fps evaluates as 41000.0. Why? Omitting for now.\n int_framerates = [\n 2,\n 3,\n 4,\n 5,\n 6,\n 8,\n 10,\n 12,\n 15,\n 16,\n 20,\n 24,\n 25,\n 30,\n 40,\n 48,\n 50,\n 60,\n 75,\n 80,\n 90,\n 100,\n 120,\n 125,\n 150,\n 200,\n 240,\n 250,\n 300,\n 375,\n 400,\n 500,\n 600,\n 750,\n 1200,\n 1500,\n 2000,\n 3000,\n 6000,\n 48000\n ]\n\n # If input fps is a whole number we'll return.\n if float(fps).is_integer():\n # Validate fps is part of Maya's fps selection.\n if int(fps) not in int_framerates:\n raise ValueError(\n \"Framerate \\\"{}\\\" is not supported in Maya\".format(fps)\n )\n return int(fps)\n else:\n # Differences to supported float frame rates.\n differences = []\n for i in float_framerates:\n differences.append(abs(i - fps))\n\n # Validate difference does not stray too far from supported framerates.\n min_difference = min(differences)\n min_index = differences.index(min_difference)\n supported_framerate = float_framerates[min_index]\n if min_difference > 0.1:\n raise ValueError(\n \"Framerate \\\"{}\\\" strays too far from any supported framerate\"\n \" in Maya. Closest supported framerate is \\\"{}\\\"\".format(\n fps, supported_framerate\n )\n )\n\n return supported_framerate\n\n\ndef write_xgen_file(data, filepath):\n \"\"\"Overwrites data in .xgen files.\n\n Quite naive approach to mainly overwrite \"xgDataPath\" and \"xgProjectPath\".\n\n Args:\n data (dict): Dictionary of key, value. Key matches with xgen file.\n For example:\n {\"xgDataPath\": \"some/path\"}\n filepath (string): Absolute path of .xgen file.\n \"\"\"\n # Generate regex lookup for line to key basically\n # match any of the keys in `\\t{key}\\t\\t`\n keys = \"|\".join(re.escape(key) for key in data.keys())\n re_keys = re.compile(\"^\\t({})\\t\\t\".format(keys))\n\n lines = []\n with open(filepath, \"r\") as f:\n for line in f:\n match = re_keys.match(line)\n if match:\n key = match.group(1)\n value = data[key]\n line = \"\\t{}\\t\\t{}\\n\".format(key, value)\n\n lines.append(line)\n\n with open(filepath, \"w\") as f:\n f.writelines(lines)\n\n\ndef get_color_management_preferences():\n \"\"\"Get and resolve OCIO preferences.\"\"\"\n data = {\n # Is color management enabled.\n \"enabled\": cmds.colorManagementPrefs(\n query=True, cmEnabled=True\n ),\n \"rendering_space\": cmds.colorManagementPrefs(\n query=True, renderingSpaceName=True\n ),\n \"output_transform\": cmds.colorManagementPrefs(\n query=True, outputTransformName=True\n ),\n \"output_transform_enabled\": cmds.colorManagementPrefs(\n query=True, outputTransformEnabled=True\n ),\n \"view_transform\": cmds.colorManagementPrefs(\n query=True, viewTransformName=True\n )\n }\n\n # Split view and display from view_transform. view_transform comes in\n # format of \"{view} ({display})\".\n regex = re.compile(r\"^(?P.+) \\((?P.+)\\)$\")\n if int(cmds.about(version=True)) <= 2020:\n # view_transform comes in format of \"{view} {display}\" in 2020.\n regex = re.compile(r\"^(?P.+) (?P.+)$\")\n\n match = regex.match(data[\"view_transform\"])\n if not match:\n raise ValueError(\n \"Unable to parse view and display from Maya view transform: '{}' \"\n \"using regex '{}'\".format(data[\"view_transform\"], regex.pattern)\n )\n\n data.update({\n \"display\": match.group(\"display\"),\n \"view\": match.group(\"view\")\n })\n\n # Get config absolute path.\n path = cmds.colorManagementPrefs(\n query=True, configFilePath=True\n )\n\n # The OCIO config supports a custom token.\n maya_resources_token = \"\"\n maya_resources_path = OpenMaya.MGlobal.getAbsolutePathToResources()\n path = path.replace(maya_resources_token, maya_resources_path)\n\n data[\"config\"] = path\n\n return data\n\n\ndef get_color_management_output_transform():\n preferences = get_color_management_preferences()\n colorspace = preferences[\"rendering_space\"]\n if preferences[\"output_transform_enabled\"]:\n colorspace = preferences[\"output_transform\"]\n return colorspace\n\n\ndef image_info(file_path):\n # type: (str) -> dict\n \"\"\"Based on tha texture path, get its bit depth and format information.\n Take reference from makeTx.py in Arnold:\n ImageInfo(filename): Get Image Information for colorspace\n AiTextureGetFormat(filename): Get Texture Format\n AiTextureGetBitDepth(filename): Get Texture bit depth\n Args:\n file_path (str): Path to the texture file.\n Returns:\n dict: Dictionary with the information about the texture file.\n \"\"\"\n from arnold import (\n AiTextureGetBitDepth,\n AiTextureGetFormat\n )\n # Get Texture Information\n img_info = {'filename': file_path}\n if os.path.isfile(file_path):\n img_info['bit_depth'] = AiTextureGetBitDepth(file_path) # noqa\n img_info['format'] = AiTextureGetFormat(file_path) # noqa\n else:\n img_info['bit_depth'] = 8\n img_info['format'] = \"unknown\"\n return img_info\n\n\ndef guess_colorspace(img_info):\n # type: (dict) -> str\n \"\"\"Guess the colorspace of the input image filename.\n Note:\n Reference from makeTx.py\n Args:\n img_info (dict): Image info generated by :func:`image_info`\n Returns:\n str: color space name use in the `--colorconvert`\n option of maketx.\n \"\"\"\n from arnold import (\n AiTextureInvalidate,\n # types\n AI_TYPE_BYTE,\n AI_TYPE_INT,\n AI_TYPE_UINT\n )\n try:\n if img_info['bit_depth'] <= 16:\n if img_info['format'] in (AI_TYPE_BYTE, AI_TYPE_INT, AI_TYPE_UINT): # noqa\n return 'sRGB'\n else:\n return 'linear'\n # now discard the image file as AiTextureGetFormat has loaded it\n AiTextureInvalidate(img_info['filename']) # noqa\n except ValueError:\n print((\"[maketx] Error: Could not guess\"\n \"colorspace for {}\").format(img_info[\"filename\"]))\n return \"linear\"\n\n\ndef len_flattened(components):\n \"\"\"Return the length of the list as if it was flattened.\n\n Maya will return consecutive components as a single entry\n when requesting with `maya.cmds.ls` without the `flatten`\n flag. Though enabling `flatten` on a large list (e.g. millions)\n will result in a slow result. This command will return the amount\n of entries in a non-flattened list by parsing the result with\n regex.\n\n Args:\n components (list): The non-flattened components.\n\n Returns:\n int: The amount of entries.\n\n \"\"\"\n assert isinstance(components, (list, tuple))\n n = 0\n\n pattern = re.compile(r\"\\[(\\d+):(\\d+)\\]\")\n for c in components:\n match = pattern.search(c)\n if match:\n start, end = match.groups()\n n += int(end) - int(start) + 1\n else:\n n += 1\n return n\n\n\ndef get_all_children(nodes):\n \"\"\"Return all children of `nodes` including each instanced child.\n Using maya.cmds.listRelatives(allDescendents=True) includes only the first\n instance. As such, this function acts as an optimal replacement with a\n focus on a fast query.\n\n \"\"\"\n\n sel = OpenMaya.MSelectionList()\n traversed = set()\n iterator = OpenMaya.MItDag(OpenMaya.MItDag.kDepthFirst)\n for node in nodes:\n\n if node in traversed:\n # Ignore if already processed as a child\n # before\n continue\n\n sel.clear()\n sel.add(node)\n dag = sel.getDagPath(0)\n\n iterator.reset(dag)\n # ignore self\n iterator.next() # noqa: B305\n while not iterator.isDone():\n\n path = iterator.fullPathName()\n\n if path in traversed:\n iterator.prune()\n iterator.next() # noqa: B305\n continue\n\n traversed.add(path)\n iterator.next() # noqa: B305\n\n return list(traversed)\n\n\ndef get_capture_preset(task_name, task_type, subset, project_settings, log):\n \"\"\"Get capture preset for playblasting.\n\n Logic for transitioning from old style capture preset to new capture preset\n profiles.\n\n Args:\n task_name (str): Task name.\n take_type (str): Task type.\n subset (str): Subset name.\n project_settings (dict): Project settings.\n log (object): Logging object.\n \"\"\"\n capture_preset = None\n filtering_criteria = {\n \"hosts\": \"maya\",\n \"families\": \"review\",\n \"task_names\": task_name,\n \"task_types\": task_type,\n \"subset\": subset\n }\n\n plugin_settings = project_settings[\"maya\"][\"publish\"][\"ExtractPlayblast\"]\n if plugin_settings[\"profiles\"]:\n profile = filter_profiles(\n plugin_settings[\"profiles\"],\n filtering_criteria,\n logger=log\n )\n capture_preset = profile.get(\"capture_preset\")\n else:\n log.warning(\"No profiles present for Extract Playblast\")\n\n # Backward compatibility for deprecated Extract Playblast settings\n # without profiles.\n if capture_preset is None:\n log.debug(\n \"Falling back to deprecated Extract Playblast capture preset \"\n \"because no new style playblast profiles are defined.\"\n )\n capture_preset = plugin_settings[\"capture_preset\"]\n\n return capture_preset or {}\n\n\ndef get_reference_node(members, log=None):\n \"\"\"Get the reference node from the container members\n Args:\n members: list of node names\n\n Returns:\n str: Reference node name.\n\n \"\"\"\n\n # Collect the references without .placeHolderList[] attributes as\n # unique entries (objects only) and skipping the sharedReferenceNode.\n references = set()\n for ref in cmds.ls(members, exactType=\"reference\", objectsOnly=True):\n\n # Ignore any `:sharedReferenceNode`\n if ref.rsplit(\":\", 1)[-1].startswith(\"sharedReferenceNode\"):\n continue\n\n # Ignore _UNKNOWN_REF_NODE_ (PLN-160)\n if ref.rsplit(\":\", 1)[-1].startswith(\"_UNKNOWN_REF_NODE_\"):\n continue\n\n references.add(ref)\n\n assert references, \"No reference node found in container\"\n\n # Get highest reference node (least parents)\n highest = min(references,\n key=lambda x: len(get_reference_node_parents(x)))\n\n # Warn the user when we're taking the highest reference node\n if len(references) > 1:\n if not log:\n log = logging.getLogger(__name__)\n\n log.warning(\"More than one reference node found in \"\n \"container, using highest reference node: \"\n \"%s (in: %s)\", highest, list(references))\n\n return highest\n\n\ndef get_reference_node_parents(ref):\n \"\"\"Return all parent reference nodes of reference node\n\n Args:\n ref (str): reference node.\n\n Returns:\n list: The upstream parent reference nodes.\n\n \"\"\"\n parent = cmds.referenceQuery(ref,\n referenceNode=True,\n parent=True)\n parents = []\n while parent:\n parents.append(parent)\n parent = cmds.referenceQuery(parent,\n referenceNode=True,\n parent=True)\n return parents\n\n\ndef create_rig_animation_instance(\n nodes, context, namespace, options=None, log=None\n):\n \"\"\"Create an animation publish instance for loaded rigs.\n\n See the RecreateRigAnimationInstance inventory action on how to use this\n for loaded rig containers.\n\n Arguments:\n nodes (list): Member nodes of the rig instance.\n context (dict): Representation context of the rig container\n namespace (str): Namespace of the rig container\n options (dict, optional): Additional loader data\n log (logging.Logger, optional): Logger to log to if provided\n\n Returns:\n None\n\n \"\"\"\n if options is None:\n options = {}\n name = context[\"representation\"][\"name\"]\n output = next((node for node in nodes if\n node.endswith(\"out_SET\")), None)\n controls = next((node for node in nodes if\n node.endswith(\"controls_SET\")), None)\n if name != \"fbx\":\n assert output, \"No out_SET in rig, this is a bug.\"\n assert controls, \"No controls_SET in rig, this is a bug.\"\n\n anim_skeleton = next((node for node in nodes if\n node.endswith(\"skeletonAnim_SET\")), None)\n skeleton_mesh = next((node for node in nodes if\n node.endswith(\"skeletonMesh_SET\")), None)\n\n # Find the roots amongst the loaded nodes\n roots = (\n cmds.ls(nodes, assemblies=True, long=True) or\n get_highest_in_hierarchy(nodes)\n )\n assert roots, \"No root nodes in rig, this is a bug.\"\n\n custom_subset = options.get(\"animationSubsetName\")\n if custom_subset:\n formatting_data = {\n \"asset\": context[\"asset\"],\n \"subset\": context['subset']['name'],\n \"family\": (\n context['subset']['data'].get('family') or\n context['subset']['data']['families'][0]\n )\n }\n namespace = get_custom_namespace(\n custom_subset.format(\n **formatting_data\n )\n )\n\n if log:\n log.info(\"Creating subset: {}\".format(namespace))\n\n # Fill creator identifier\n creator_identifier = \"io.openpype.creators.maya.animation\"\n\n host = registered_host()\n create_context = CreateContext(host)\n # Create the animation instance\n rig_sets = [output, controls, anim_skeleton, skeleton_mesh]\n # Remove sets that this particular rig does not have\n rig_sets = [s for s in rig_sets if s is not None]\n with maintained_selection():\n cmds.select(rig_sets + roots, noExpand=True)\n create_context.create(\n creator_identifier=creator_identifier,\n variant=namespace,\n pre_create_data={\"use_selection\": True}\n )\n"
},
{
"path": "openpype/hosts/maya/api/lib_renderproducts.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Module handling expected render output from Maya.\n\nThis module is used in :mod:`collect_render` and :mod:`collect_vray_scene`.\n\nNote:\n To implement new renderer, just create new class inheriting from\n :class:`ARenderProducts` and add it to :func:`RenderProducts.get()`.\n\nAttributes:\n R_SINGLE_FRAME (:class:`re.Pattern`): Find single frame number.\n R_FRAME_RANGE (:class:`re.Pattern`): Find frame range.\n R_FRAME_NUMBER (:class:`re.Pattern`): Find frame number in string.\n R_LAYER_TOKEN (:class:`re.Pattern`): Find layer token in image prefixes.\n R_AOV_TOKEN (:class:`re.Pattern`): Find AOV token in image prefixes.\n R_SUBSTITUTE_AOV_TOKEN (:class:`re.Pattern`): Find and substitute AOV token\n in image prefixes.\n R_REMOVE_AOV_TOKEN (:class:`re.Pattern`): Find and remove AOV token in\n image prefixes.\n R_CLEAN_FRAME_TOKEN (:class:`re.Pattern`): Find and remove unfilled\n Renderman frame token in image prefix.\n R_CLEAN_EXT_TOKEN (:class:`re.Pattern`): Find and remove unfilled Renderman\n extension token in image prefix.\n R_SUBSTITUTE_LAYER_TOKEN (:class:`re.Pattern`): Find and substitute render\n layer token in image prefixes.\n R_SUBSTITUTE_SCENE_TOKEN (:class:`re.Pattern`): Find and substitute scene\n token in image prefixes.\n R_SUBSTITUTE_CAMERA_TOKEN (:class:`re.Pattern`): Find and substitute camera\n token in image prefixes.\n IMAGE_PREFIXES (dict): Mapping between renderers and their respective\n image prefix attribute names.\n\nThanks:\n Roy Nieterau (BigRoy) / Colorbleed for overhaul of original\n *expected_files*.\n\n\"\"\"\n\nimport logging\nimport re\nimport os\nfrom abc import ABCMeta, abstractmethod\n\nimport six\nimport attr\n\nfrom . import lib\nfrom . import lib_rendersetup\nfrom openpype.pipeline.colorspace import get_ocio_config_views\n\nfrom maya import cmds, mel\n\nlog = logging.getLogger(__name__)\n\nR_SINGLE_FRAME = re.compile(r\"^(-?)\\d+$\")\nR_FRAME_RANGE = re.compile(r\"^(?P(-?)\\d+)-(?P(-?)\\d+)$\")\nR_FRAME_NUMBER = re.compile(r\".+\\.(?P[0-9]+)\\..+\")\nR_LAYER_TOKEN = re.compile(\n r\".*((?:%l)|(?:)|(?:)).*\", re.IGNORECASE\n)\nR_AOV_TOKEN = re.compile(r\".*%a.*|.*.*|.*.*\", re.IGNORECASE)\nR_SUBSTITUTE_AOV_TOKEN = re.compile(r\"%a||\", re.IGNORECASE)\nR_REMOVE_AOV_TOKEN = re.compile(\n r\"_%a|\\.%a|_|\\.|_|\\.\", re.IGNORECASE)\n# to remove unused renderman tokens\nR_CLEAN_FRAME_TOKEN = re.compile(r\"\\.?\\.?\", re.IGNORECASE)\nR_CLEAN_EXT_TOKEN = re.compile(r\"\\.?\\.?\", re.IGNORECASE)\n\nR_SUBSTITUTE_LAYER_TOKEN = re.compile(\n r\"%l||\", re.IGNORECASE\n)\nR_SUBSTITUTE_CAMERA_TOKEN = re.compile(r\"%c|\", re.IGNORECASE)\nR_SUBSTITUTE_SCENE_TOKEN = re.compile(r\"%s|\", re.IGNORECASE)\n\n# not sure about the renderman image prefix\nIMAGE_PREFIXES = {\n \"vray\": \"vraySettings.fileNamePrefix\",\n \"arnold\": \"defaultRenderGlobals.imageFilePrefix\",\n \"renderman\": \"rmanGlobals.imageFileFormat\",\n \"redshift\": \"defaultRenderGlobals.imageFilePrefix\",\n \"mayahardware2\": \"defaultRenderGlobals.imageFilePrefix\"\n}\n\nRENDERMAN_IMAGE_DIR = \"/\"\n\n\ndef has_tokens(string, tokens):\n \"\"\"Return whether any of tokens is in input string (case-insensitive)\"\"\"\n pattern = \"({})\".format(\"|\".join(re.escape(token) for token in tokens))\n match = re.search(pattern, string, re.IGNORECASE)\n return bool(match)\n\n\n@attr.s\nclass LayerMetadata(object):\n \"\"\"Data class for Render Layer metadata.\"\"\"\n frameStart = attr.ib()\n frameEnd = attr.ib()\n cameras = attr.ib()\n sceneName = attr.ib()\n layerName = attr.ib()\n renderer = attr.ib()\n defaultExt = attr.ib()\n filePrefix = attr.ib()\n frameStep = attr.ib(default=1)\n padding = attr.ib(default=4)\n\n # Render Products\n products = attr.ib(init=False, default=attr.Factory(list))\n\n # The AOV separator token. Note that not all renderers define an explicit\n # render separator but allow to put the AOV/RenderPass token anywhere in\n # the file path prefix. For those renderers we'll fall back to whatever\n # is between the last occurrences of and tokens.\n aov_separator = attr.ib(default=\"_\")\n\n\n@attr.s\nclass RenderProduct(object):\n \"\"\"Describes an image or other file-like artifact produced by a render.\n\n Warning:\n This currently does NOT return as a product PER render camera.\n A single Render Product will generate files per camera. E.g. with two\n cameras each render product generates two sequences on disk assuming\n the file path prefix correctly uses the tokens.\n\n \"\"\"\n productName = attr.ib()\n ext = attr.ib() # extension\n colorspace = attr.ib() # colorspace\n aov = attr.ib(default=None) # source aov\n driver = attr.ib(default=None) # source driver\n multipart = attr.ib(default=False) # multichannel file\n camera = attr.ib(default=None) # used only when rendering\n # from multiple cameras\n\n\ndef get(layer, render_instance=None):\n # type: (str, object) -> ARenderProducts\n \"\"\"Get render details and products for given renderer and render layer.\n\n Args:\n layer (str): Name of render layer\n render_instance (pyblish.api.Instance): Publish instance.\n If not provided an empty mock instance is used.\n\n Returns:\n ARenderProducts: The correct RenderProducts instance for that\n renderlayer.\n\n Raises:\n :exc:`UnsupportedRendererException`: If requested renderer\n is not supported. It needs to be implemented by extending\n :class:`ARenderProducts` and added to this methods ``if``\n statement.\n\n \"\"\"\n\n if render_instance is None:\n # For now produce a mock instance\n class Instance(object):\n data = {}\n render_instance = Instance()\n\n renderer_name = lib.get_attr_in_layer(\n \"defaultRenderGlobals.currentRenderer\",\n layer=layer\n )\n\n renderer = {\n \"arnold\": RenderProductsArnold,\n \"vray\": RenderProductsVray,\n \"redshift\": RenderProductsRedshift,\n \"renderman\": RenderProductsRenderman,\n \"mayahardware2\": RenderProductsMayaHardware\n }.get(renderer_name.lower(), None)\n if renderer is None:\n raise UnsupportedRendererException(\n \"Unsupported renderer: {}\".format(renderer_name)\n )\n\n return renderer(layer, render_instance)\n\n\n@six.add_metaclass(ABCMeta)\nclass ARenderProducts:\n \"\"\"Abstract class with common code for all renderers.\n\n Attributes:\n renderer (str): name of renderer.\n\n \"\"\"\n\n renderer = None\n\n def __init__(self, layer, render_instance):\n \"\"\"Constructor.\"\"\"\n self.layer = layer\n self.render_instance = render_instance\n self.multipart = self.get_multipart()\n\n # Initialize\n self.layer_data = self._get_layer_data()\n self.layer_data.products = self.get_render_products()\n\n def get_multipart(self):\n raise NotImplementedError(\n \"The render product implementation does not have a \"\n \"\\\"get_multipart\\\" method.\"\n )\n\n def has_camera_token(self):\n # type: () -> bool\n \"\"\"Check if camera token is in image prefix.\n\n Returns:\n bool: True/False if camera token is present.\n\n \"\"\"\n return \"\" in self.layer_data.filePrefix.lower()\n\n @abstractmethod\n def get_render_products(self):\n \"\"\"To be implemented by renderer class.\n\n This should return a list of RenderProducts.\n\n Returns:\n list: List of RenderProduct\n\n \"\"\"\n\n @staticmethod\n def sanitize_camera_name(camera):\n # type: (str) -> str\n \"\"\"Sanitize camera name.\n\n Remove Maya illegal characters from camera name.\n\n Args:\n camera (str): Maya camera name.\n\n Returns:\n (str): sanitized camera name\n\n Example:\n >>> ARenderProducts.sanizite_camera_name('test:camera_01')\n test_camera_01\n\n \"\"\"\n return re.sub('[^0-9a-zA-Z_]+', '_', camera)\n\n def get_renderer_prefix(self):\n # type: () -> str\n \"\"\"Return prefix for specific renderer.\n\n This is for most renderers the same and can be overridden if needed.\n\n Returns:\n str: String with image prefix containing tokens\n\n Raises:\n :exc:`UnsupportedRendererException`: If we requested image\n prefix for renderer we know nothing about.\n See :data:`IMAGE_PREFIXES` for mapping of renderers and\n image prefixes.\n\n \"\"\"\n try:\n prefix_attr = IMAGE_PREFIXES[self.renderer]\n except KeyError:\n raise UnsupportedRendererException(\n \"Unsupported renderer {}\".format(self.renderer)\n )\n\n # Note: When this attribute is never set (e.g. on maya launch) then\n # this can return None even though it is a string attribute\n prefix = self._get_attr(prefix_attr)\n\n if not prefix:\n # Fall back to scene name by default\n log.warning(\"Image prefix not set, using \")\n prefix = \"\"\n\n return prefix\n\n def get_render_attribute(self, attribute):\n \"\"\"Get attribute from render options.\n\n Args:\n attribute (str): name of attribute to be looked up.\n\n Returns:\n Attribute value\n\n \"\"\"\n return self._get_attr(\"defaultRenderGlobals\", attribute)\n\n def _get_attr(self, node_attr, attribute=None):\n \"\"\"Return the value of the attribute in the renderlayer\n\n For readability this allows passing in the attribute in two ways.\n\n As a single argument:\n _get_attr(\"node.attr\")\n Or as two arguments:\n _get_attr(\"node\", \"attr\")\n\n Returns:\n Value of the attribute inside the layer this instance is set to.\n\n \"\"\"\n\n if attribute is None:\n plug = node_attr\n else:\n plug = \"{}.{}\".format(node_attr, attribute)\n\n return lib.get_attr_in_layer(plug, layer=self.layer)\n\n @staticmethod\n def extract_separator(file_prefix):\n \"\"\"Extract AOV separator character from the prefix.\n\n Default behavior extracts the part between\n last occurrences of and \n\n Todo:\n This code also triggers for V-Ray which overrides it explicitly\n so this code will invalidly debug log it couldn't extract the\n AOV separator even though it does set it in RenderProductsVray.\n\n Args:\n file_prefix (str): File prefix with tokens.\n\n Returns:\n str or None: prefix character if it can be extracted.\n \"\"\"\n layer_tokens = [\"\", \"\"]\n aov_tokens = [\"\", \"\"]\n\n def match_last(tokens, text):\n \"\"\"regex match the last occurrence from a list of tokens\"\"\"\n pattern = \"(?:.*)({})\".format(\"|\".join(tokens))\n return re.search(pattern, text, re.IGNORECASE)\n\n layer_match = match_last(layer_tokens, file_prefix)\n aov_match = match_last(aov_tokens, file_prefix)\n separator = None\n if layer_match and aov_match:\n matches = sorted((layer_match, aov_match),\n key=lambda match: match.end(1))\n separator = file_prefix[matches[0].end(1):matches[1].start(1)]\n return separator\n\n def _get_layer_data(self):\n # type: () -> LayerMetadata\n # ______________________________________________\n # ____________________/ ____________________________________________/\n # 1 - get scene name /__________________/\n # ____________________/\n _, scene_basename = os.path.split(cmds.file(q=True, loc=True))\n scene_name, _ = os.path.splitext(scene_basename)\n kwargs = {}\n file_prefix = self.get_renderer_prefix()\n\n # If the Render Layer belongs to a Render Setup layer then the\n # output name is based on the Render Setup Layer name without\n # the `rs_` prefix.\n layer_name = self.layer\n rs_layer = lib_rendersetup.get_rendersetup_layer(layer_name)\n if rs_layer:\n layer_name = rs_layer\n\n if self.layer == \"defaultRenderLayer\":\n # defaultRenderLayer renders as masterLayer\n layer_name = \"masterLayer\"\n\n separator = self.extract_separator(file_prefix)\n if separator:\n kwargs[\"aov_separator\"] = separator\n else:\n log.debug(\"Couldn't extract aov separator from \"\n \"file prefix: {}\".format(file_prefix))\n\n # todo: Support Custom Frames sequences 0,5-10,100-120\n # Deadline allows submitting renders with a custom frame list\n # to support those cases we might want to allow 'custom frames'\n # to be overridden to `ExpectFiles` class?\n return LayerMetadata(\n frameStart=int(self.get_render_attribute(\"startFrame\")),\n frameEnd=int(self.get_render_attribute(\"endFrame\")),\n frameStep=int(self.get_render_attribute(\"byFrameStep\")),\n padding=int(self.get_render_attribute(\"extensionPadding\")),\n # if we have token in prefix path we'll expect output for\n # every renderable camera in layer.\n cameras=self.get_renderable_cameras(),\n sceneName=scene_name,\n layerName=layer_name,\n renderer=self.renderer,\n defaultExt=self._get_attr(\"defaultRenderGlobals.imfPluginKey\"),\n filePrefix=file_prefix,\n **kwargs\n )\n\n def _generate_file_sequence(\n self, layer_data,\n force_aov_name=None,\n force_ext=None,\n force_cameras=None):\n # type: (LayerMetadata, str, str, list) -> list\n expected_files = []\n cameras = force_cameras or layer_data.cameras\n ext = force_ext or layer_data.defaultExt\n for cam in cameras:\n file_prefix = layer_data.filePrefix\n mappings = (\n (R_SUBSTITUTE_SCENE_TOKEN, layer_data.sceneName),\n (R_SUBSTITUTE_LAYER_TOKEN, layer_data.layerName),\n (R_SUBSTITUTE_CAMERA_TOKEN, self.sanitize_camera_name(cam)),\n # this is required to remove unfilled aov token, for example\n # in Redshift\n (R_REMOVE_AOV_TOKEN, \"\") if not force_aov_name \\\n else (R_SUBSTITUTE_AOV_TOKEN, force_aov_name),\n\n (R_CLEAN_FRAME_TOKEN, \"\"),\n (R_CLEAN_EXT_TOKEN, \"\"),\n )\n\n for regex, value in mappings:\n file_prefix = re.sub(regex, value, file_prefix)\n\n for frame in range(\n int(layer_data.frameStart),\n int(layer_data.frameEnd) + 1,\n int(layer_data.frameStep),\n ):\n frame_str = str(frame).rjust(layer_data.padding, \"0\")\n expected_files.append(\n \"{}.{}.{}\".format(file_prefix, frame_str, ext)\n )\n return expected_files\n\n def get_files(self, product):\n # type: (RenderProduct) -> list\n \"\"\"Return list of expected files.\n\n It will translate render token strings ('', etc.) to\n their values. This task is tricky as every renderer deals with this\n differently. That's why we expose `get_files` as a method on the\n Renderer class so it can be overridden for complex cases.\n\n Args:\n product (RenderProduct): Render product to be used for file\n generation.\n\n Returns:\n List of files\n\n \"\"\"\n return self._generate_file_sequence(\n self.layer_data,\n force_aov_name=product.productName,\n force_ext=product.ext,\n force_cameras=[product.camera]\n )\n\n def get_renderable_cameras(self):\n # type: () -> list\n \"\"\"Get all renderable camera transforms.\n\n Returns:\n list: list of renderable cameras.\n\n \"\"\"\n\n renderable_cameras = [\n cam for cam in cmds.ls(cameras=True)\n if self._get_attr(cam, \"renderable\")\n ]\n\n # The output produces a sanitized name for using its\n # shortest unique path of the transform so we'll return\n # at least that unique path. This could include a parent\n # name too when two cameras have the same name but are\n # in a different hierarchy, e.g. \"group1|cam\" and \"group2|cam\"\n def get_name(camera):\n return cmds.ls(cmds.listRelatives(camera,\n parent=True,\n fullPath=True))[0]\n\n return [get_name(cam) for cam in renderable_cameras]\n\n\nclass RenderProductsArnold(ARenderProducts):\n \"\"\"Render products for Arnold renderer.\n\n References:\n mtoa.utils.getFileName()\n mtoa.utils.ui.common.updateArnoldTargetFilePreview()\n\n Notes:\n - Output Denoising AOVs are not currently included.\n - Only Frame/Animation ext: name.#.ext is supported.\n - Use Custom extension is not supported.\n - and tokens not tested\n - With Merge AOVs but in File Name Prefix Arnold\n will still NOT merge the aovs. This class correctly resolves\n it - but user should be aware.\n - File Path Prefix overrides per AOV driver are not implemented\n\n Attributes:\n aiDriverExtension (dict): Arnold AOV driver extension mapping.\n Is there a better way?\n renderer (str): name of renderer.\n\n \"\"\"\n renderer = \"arnold\"\n aiDriverExtension = {\n \"jpeg\": \"jpg\",\n \"exr\": \"exr\",\n \"deepexr\": \"exr\",\n \"png\": \"png\",\n \"tiff\": \"tif\",\n \"mtoa_shaders\": \"ass\", # TODO: research what those last two should be\n \"maya\": \"\",\n }\n\n def get_renderer_prefix(self):\n\n prefix = super(RenderProductsArnold, self).get_renderer_prefix()\n merge_aovs = self._get_attr(\"defaultArnoldDriver.mergeAOVs\")\n if not merge_aovs and \"\" not in prefix.lower():\n # When Merge AOVs is disabled and token not present\n # then Arnold prepends / to the output path.\n # todo: It's untested what happens if AOV driver has an\n # an explicit override path prefix.\n prefix = \"/\" + prefix\n\n return prefix\n\n def get_multipart(self):\n multipart = False\n multilayer = bool(self._get_attr(\"defaultArnoldDriver.multipart\"))\n merge_AOVs = bool(self._get_attr(\"defaultArnoldDriver.mergeAOVs\"))\n if multilayer or merge_AOVs:\n multipart = True\n\n return multipart\n\n def _get_aov_render_products(self, aov, cameras=None):\n \"\"\"Return all render products for the AOV\"\"\"\n\n products = []\n aov_name = self._get_attr(aov, \"name\")\n ai_drivers = cmds.listConnections(\"{}.outputs\".format(aov),\n source=True,\n destination=False,\n type=\"aiAOVDriver\") or []\n if not cameras:\n cameras = [\n self.sanitize_camera_name(\n self.get_renderable_cameras()[0]\n )\n ]\n\n for ai_driver in ai_drivers:\n colorspace = self._get_colorspace(\n ai_driver + \".colorManagement\"\n )\n # todo: check aiAOVDriver.prefix as it could have\n # a custom path prefix set for this driver\n\n # Skip Drivers set only for GUI\n # 0: GUI, 1: Batch, 2: GUI and Batch\n output_mode = self._get_attr(ai_driver, \"outputMode\")\n if output_mode == 0: # GUI only\n log.warning(\"%s has Output Mode set to GUI, \"\n \"skipping...\", ai_driver)\n continue\n\n ai_translator = self._get_attr(ai_driver, \"aiTranslator\")\n try:\n ext = self.aiDriverExtension[ai_translator]\n except KeyError:\n raise AOVError(\n \"Unrecognized arnold driver format \"\n \"for AOV - {}\".format(aov_name)\n )\n\n # If aov RGBA is selected, arnold will translate it to `beauty`\n name = aov_name\n if name == \"RGBA\":\n name = \"beauty\"\n\n # Support Arnold light groups for AOVs\n # Global AOV: When disabled the main layer is\n # not written: `{pass}`\n # All Light Groups: When enabled, a `{pass}_lgroups` file is\n # written and is always merged into a\n # single file\n # Light Groups List: When set, a product per light\n # group is written\n # e.g. {pass}_front, {pass}_rim\n global_aov = self._get_attr(aov, \"globalAov\")\n if global_aov:\n for camera in cameras:\n product = RenderProduct(\n productName=name,\n ext=ext,\n aov=aov_name,\n driver=ai_driver,\n multipart=self.multipart,\n camera=camera,\n colorspace=colorspace\n )\n products.append(product)\n\n all_light_groups = self._get_attr(aov, \"lightGroups\")\n if all_light_groups:\n # All light groups is enabled. A single multipart\n # Render Product\n for camera in cameras:\n product = RenderProduct(\n productName=name + \"_lgroups\",\n ext=ext,\n aov=aov_name,\n driver=ai_driver,\n # Always multichannel output\n multipart=True,\n camera=camera,\n colorspace=colorspace\n )\n products.append(product)\n else:\n value = self._get_attr(aov, \"lightGroupsList\")\n if not value:\n continue\n selected_light_groups = value.strip().split()\n for light_group in selected_light_groups:\n # Render Product per selected light group\n aov_light_group_name = \"{}_{}\".format(name, light_group)\n for camera in cameras:\n product = RenderProduct(\n productName=aov_light_group_name,\n aov=aov_name,\n driver=ai_driver,\n ext=ext,\n camera=camera,\n colorspace=colorspace\n )\n products.append(product)\n\n return products\n\n def _get_colorspace(self, attribute):\n \"\"\"Resolve colorspace from Arnold settings.\"\"\"\n\n def _view_transform():\n preferences = lib.get_color_management_preferences()\n views_data = get_ocio_config_views(preferences[\"config\"])\n view_data = views_data[\n \"{}/{}\".format(preferences[\"display\"], preferences[\"view\"])\n ]\n return view_data[\"colorspace\"]\n\n def _raw():\n preferences = lib.get_color_management_preferences()\n return preferences[\"rendering_space\"]\n\n resolved_values = {\n \"Raw\": _raw,\n \"Use View Transform\": _view_transform,\n # Default. Same as Maya Preferences.\n \"Use Output Transform\": lib.get_color_management_output_transform\n }\n return resolved_values[self._get_attr(attribute)]()\n\n def get_render_products(self):\n \"\"\"Get all AOVs.\n\n See Also:\n :func:`ARenderProducts.get_render_products()`\n\n Raises:\n :class:`AOVError`: If AOV cannot be determined.\n\n \"\"\"\n\n if not cmds.ls(\"defaultArnoldRenderOptions\", type=\"aiOptions\"):\n # this occurs when Render Setting windows was not opened yet. In\n # such case there are no Arnold options created so query for AOVs\n # will fail. We terminate here as there are no AOVs specified then.\n # This state will most probably fail later on some Validator\n # anyway.\n return []\n\n # check if camera token is in prefix. If so, and we have list of\n # renderable cameras, generate render product for each and every\n # of them.\n cameras = [\n self.sanitize_camera_name(c)\n for c in self.get_renderable_cameras()\n ]\n\n default_ext = self._get_attr(\"defaultRenderGlobals.imfPluginKey\")\n colorspace = self._get_colorspace(\n \"defaultArnoldDriver.colorManagement\"\n )\n beauty_products = [\n RenderProduct(\n productName=\"beauty\",\n ext=default_ext,\n driver=\"defaultArnoldDriver\",\n camera=camera,\n colorspace=colorspace\n ) for camera in cameras\n ]\n\n # AOVs > Legacy > Maya Render View > Mode\n aovs_enabled = bool(\n self._get_attr(\"defaultArnoldRenderOptions.aovMode\")\n )\n if not aovs_enabled:\n return beauty_products\n\n # Common > File Output > Merge AOVs or \n # We don't need to check for Merge AOVs due to overridden\n # `get_renderer_prefix()` behavior which forces \n has_renderpass_token = (\n \"\" in self.layer_data.filePrefix.lower()\n )\n if not has_renderpass_token:\n for product in beauty_products:\n product.multipart = True\n return beauty_products\n\n # AOVs are set to be rendered separately. We should expect\n # token in path.\n # handle aovs from references\n use_ref_aovs = self.render_instance.data.get(\n \"useReferencedAovs\", False) or False\n\n aovs = cmds.ls(type=\"aiAOV\")\n if not use_ref_aovs:\n ref_aovs = cmds.ls(type=\"aiAOV\", referencedNodes=True)\n aovs = list(set(aovs) - set(ref_aovs))\n\n products = []\n\n # Append the AOV products\n for aov in aovs:\n enabled = self._get_attr(aov, \"enabled\")\n if not enabled:\n continue\n\n # For now stick to the legacy output format.\n aov_products = self._get_aov_render_products(aov, cameras)\n products.extend(aov_products)\n\n if all(product.aov != \"RGBA\" for product in products):\n # Append default 'beauty' as this is arnolds default.\n # However, it is excluded whenever a RGBA pass is enabled.\n # For legibility add the beauty layer as first entry\n products += beauty_products\n\n # TODO: Output Denoising AOVs?\n\n return products\n\n\nclass RenderProductsVray(ARenderProducts):\n \"\"\"Expected files for V-Ray renderer.\n\n Notes:\n - \"Disabled\" animation incorrectly returns frames in filename\n - \"Renumber Frames\" is not supported\n\n Reference:\n vrayAddRenderElementImpl() in vrayCreateRenderElementsTab.mel\n\n \"\"\"\n # todo: detect whether rendering with V-Ray GPU + whether AOV is supported\n\n renderer = \"vray\"\n\n def get_multipart(self):\n multipart = False\n image_format = self._get_attr(\"vraySettings.imageFormatStr\")\n if image_format == \"exr (multichannel)\":\n multipart = True\n\n return multipart\n\n def get_renderer_prefix(self):\n # type: () -> str\n \"\"\"Get image prefix for V-Ray.\n\n This overrides :func:`ARenderProducts.get_renderer_prefix()` as\n we must add `` token manually. This is done only for\n non-multipart outputs, where `` token doesn't make sense.\n\n See also:\n :func:`ARenderProducts.get_renderer_prefix()`\n\n \"\"\"\n prefix = super(RenderProductsVray, self).get_renderer_prefix()\n if self.multipart:\n return prefix\n aov_separator = self._get_aov_separator()\n prefix = \"{}{}\".format(prefix, aov_separator)\n return prefix\n\n def _get_aov_separator(self):\n # type: () -> str\n \"\"\"Return the V-Ray AOV/Render Elements separator\"\"\"\n return self._get_attr(\n \"vraySettings.fileNameRenderElementSeparator\"\n )\n\n def _get_layer_data(self):\n # type: () -> LayerMetadata\n \"\"\"Override to get vray specific extension.\"\"\"\n layer_data = super(RenderProductsVray, self)._get_layer_data()\n\n default_ext = self._get_attr(\"vraySettings.imageFormatStr\")\n if default_ext in [\"exr (multichannel)\", \"exr (deep)\"]:\n default_ext = \"exr\"\n layer_data.defaultExt = default_ext\n layer_data.padding = self._get_attr(\"vraySettings.fileNamePadding\")\n\n layer_data.aov_separator = self._get_aov_separator()\n\n return layer_data\n\n def get_render_products(self):\n \"\"\"Get all AOVs.\n\n See Also:\n :func:`ARenderProducts.get_render_products()`\n\n \"\"\"\n if not cmds.ls(\"vraySettings\", type=\"VRaySettingsNode\"):\n # this occurs when Render Setting windows was not opened yet. In\n # such case there are no VRay options created so query for AOVs\n # will fail. We terminate here as there are no AOVs specified then.\n # This state will most probably fail later on some Validator\n # anyway.\n return []\n\n cameras = [\n self.sanitize_camera_name(c)\n for c in self.get_renderable_cameras()\n ]\n\n image_format_str = self._get_attr(\"vraySettings.imageFormatStr\")\n default_ext = image_format_str\n if default_ext in {\"exr (multichannel)\", \"exr (deep)\"}:\n default_ext = \"exr\"\n\n colorspace = lib.get_color_management_output_transform()\n products = []\n\n # add beauty as default when not disabled\n dont_save_rgb = self._get_attr(\"vraySettings.dontSaveRgbChannel\")\n if not dont_save_rgb:\n for camera in cameras:\n products.append(\n RenderProduct(\n productName=\"\",\n ext=default_ext,\n camera=camera,\n colorspace=colorspace,\n multipart=self.multipart\n )\n )\n\n # separate alpha file\n separate_alpha = self._get_attr(\"vraySettings.separateAlpha\")\n if separate_alpha:\n for camera in cameras:\n products.append(\n RenderProduct(\n productName=\"Alpha\",\n ext=default_ext,\n camera=camera,\n colorspace=colorspace,\n multipart=self.multipart\n )\n )\n if self.multipart:\n # AOVs are merged in m-channel file, only main layer is rendered\n return products\n\n # handle aovs from references\n use_ref_aovs = self.render_instance.data.get(\n \"useReferencedAovs\", False) or False\n\n # this will have list of all aovs no matter if they are coming from\n # reference or not.\n aov_types = [\"VRayRenderElement\", \"VRayRenderElementSet\"]\n aovs = cmds.ls(type=aov_types)\n if not use_ref_aovs:\n ref_aovs = cmds.ls(type=aov_types, referencedNodes=True) or []\n aovs = list(set(aovs) - set(ref_aovs))\n\n for aov in aovs:\n enabled = self._get_attr(aov, \"enabled\")\n if not enabled:\n continue\n\n class_type = self._get_attr(aov + \".vrayClassType\")\n if class_type == \"LightMixElement\":\n # Special case which doesn't define a name by itself but\n # instead seems to output multiple Render Products,\n # specifically \"Self_Illumination\" and \"Environment\"\n product_names = [\"Self_Illumination\", \"Environment\"]\n for camera in cameras:\n for name in product_names:\n product = RenderProduct(productName=name,\n ext=default_ext,\n aov=aov,\n camera=camera,\n colorspace=colorspace)\n products.append(product)\n # Continue as we've processed this special case AOV\n continue\n\n aov_name = self._get_vray_aov_name(aov)\n for camera in cameras:\n product = RenderProduct(\n productName=aov_name,\n ext=default_ext,\n aov=aov,\n camera=camera,\n colorspace=colorspace\n )\n products.append(product)\n\n return products\n\n def _get_vray_aov_attr(self, node, prefix):\n \"\"\"Get value for attribute that starts with key in name\n\n V-Ray AOVs have attribute names that include the type\n of AOV in the attribute name, for example:\n - vray_filename_rawdiffuse\n - vray_filename_velocity\n - vray_name_gi\n - vray_explicit_name_extratex\n\n To simplify querying the \"vray_filename\" or \"vray_name\"\n attributes we just find the first attribute that has\n that particular \"{prefix}_\" in the attribute name.\n\n Args:\n node (str): AOV node name\n prefix (str): Prefix of the attribute name.\n\n Returns:\n Value of the attribute if it exists, else None\n\n \"\"\"\n attrs = cmds.listAttr(node, string=\"{}_*\".format(prefix))\n if not attrs:\n return None\n\n assert len(attrs) == 1, \"Found more than one attribute: %s\" % attrs\n attr = attrs[0]\n\n return self._get_attr(node, attr)\n\n def _get_vray_aov_name(self, node):\n \"\"\"Get AOVs name from Vray.\n\n Args:\n node (str): aov node name.\n\n Returns:\n str: aov name.\n\n \"\"\"\n\n vray_explicit_name = self._get_vray_aov_attr(node,\n \"vray_explicit_name\")\n vray_filename = self._get_vray_aov_attr(node, \"vray_filename\")\n vray_name = self._get_vray_aov_attr(node, \"vray_name\")\n final_name = vray_explicit_name or vray_filename or vray_name or None\n\n class_type = self._get_attr(node, \"vrayClassType\")\n if not vray_explicit_name:\n # Explicit name takes precedence and overrides completely\n # otherwise add the connected node names to the special cases\n # Any namespace colon ':' gets replaced to underscore '_'\n # so we sanitize using `sanitize_camera_name`\n def _get_source_name(node, attr):\n \"\"\"Return sanitized name of input connection to attribute\"\"\"\n plug = \"{}.{}\".format(node, attr)\n connections = cmds.listConnections(plug,\n source=True,\n destination=False)\n if connections:\n return self.sanitize_camera_name(connections[0])\n\n if class_type == \"MaterialSelectElement\":\n # Name suffix is based on the connected material or set\n attrs = [\n \"vray_mtllist_mtlselect\",\n \"vray_mtl_mtlselect\"\n ]\n for attribute in attrs:\n name = _get_source_name(node, attribute)\n if name:\n final_name += '_{}'.format(name)\n break\n else:\n log.warning(\"Material Select Element has no \"\n \"selected materials: %s\", node)\n\n elif class_type == \"ExtraTexElement\":\n # Name suffix is based on the connected textures\n extratex_type = self._get_attr(node, \"vray_type_extratex\")\n attr = {\n 0: \"vray_texture_extratex\",\n 1: \"vray_float_texture_extratex\",\n 2: \"vray_int_texture_extratex\",\n }.get(extratex_type)\n name = _get_source_name(node, attr)\n if name:\n final_name += '_{}'.format(name)\n else:\n log.warning(\"Extratex Element has no incoming texture\")\n\n assert final_name, \"Output filename not defined for AOV: %s\" % node\n\n return final_name\n\n\nclass RenderProductsRedshift(ARenderProducts):\n \"\"\"Expected files for Redshift renderer.\n\n Notes:\n - `get_files()` only supports rendering with frames, like \"animation\"\n\n Attributes:\n\n unmerged_aovs (list): Name of aovs that are not merged into resulting\n exr and we need them specified in Render Products output.\n\n \"\"\"\n\n renderer = \"redshift\"\n unmerged_aovs = {\"Cryptomatte\"}\n\n def get_files(self, product):\n # When outputting AOVs we need to replace Redshift specific AOV tokens\n # with Maya render tokens for generating file sequences. We validate to\n # a specific AOV fileprefix so we only need to account for one\n # replacement.\n if not product.multipart and product.driver:\n file_prefix = self._get_attr(product.driver + \".filePrefix\")\n self.layer_data.filePrefix = file_prefix.replace(\n \"/\",\n \"//\"\n )\n\n return super(RenderProductsRedshift, self).get_files(product)\n\n def get_multipart(self):\n # For Redshift we don't directly return upon forcing multilayer\n # due to some AOVs still being written into separate files,\n # like Cryptomatte.\n # AOVs are merged in multi-channel file\n multipart = False\n force_layer = bool(\n self._get_attr(\"redshiftOptions.exrForceMultilayer\")\n )\n if force_layer:\n multipart = True\n\n return multipart\n\n def get_renderer_prefix(self):\n \"\"\"Get image prefix for Redshift.\n\n This overrides :func:`ARenderProducts.get_renderer_prefix()` as\n we must add `` token manually. This is done only for\n non-multipart outputs, where `` token doesn't make sense.\n\n See also:\n :func:`ARenderProducts.get_renderer_prefix()`\n\n \"\"\"\n prefix = super(RenderProductsRedshift, self).get_renderer_prefix()\n if self.multipart:\n return prefix\n separator = self.extract_separator(prefix)\n prefix = \"{}{}\".format(prefix, separator or \"_\")\n return prefix\n\n def get_render_products(self):\n \"\"\"Get all AOVs.\n\n See Also:\n :func:`ARenderProducts.get_render_products()`\n\n \"\"\"\n\n if not cmds.ls(\"redshiftOptions\", type=\"RedshiftOptions\"):\n # this occurs when Render Setting windows was not opened yet. In\n # such case there are no Redshift options created so query for AOVs\n # will fail. We terminate here as there are no AOVs specified then.\n # This state will most probably fail later on some Validator\n # anyway.\n return []\n\n cameras = [\n self.sanitize_camera_name(c)\n for c in self.get_renderable_cameras()\n ]\n\n # Get Redshift Extension from image format\n image_format = self._get_attr(\"redshiftOptions.imageFormat\") # integer\n ext = mel.eval(\"redshiftGetImageExtension(%i)\" % image_format)\n\n use_ref_aovs = self.render_instance.data.get(\n \"useReferencedAovs\", False) or False\n\n aovs = cmds.ls(type=\"RedshiftAOV\")\n if not use_ref_aovs:\n ref_aovs = cmds.ls(type=\"RedshiftAOV\", referencedNodes=True)\n aovs = list(set(aovs) - set(ref_aovs))\n\n products = []\n light_groups_enabled = False\n has_beauty_aov = False\n colorspace = lib.get_color_management_output_transform()\n for aov in aovs:\n enabled = self._get_attr(aov, \"enabled\")\n if not enabled:\n continue\n\n aov_type = self._get_attr(aov, \"aovType\")\n if self.multipart and aov_type not in self.unmerged_aovs:\n continue\n\n # Any AOVs that still get processed, like Cryptomatte\n # by themselves are not multipart files.\n\n # Redshift skips rendering of masterlayer without AOV suffix\n # when a Beauty AOV is rendered. It overrides the main layer.\n if aov_type == \"Beauty\":\n has_beauty_aov = True\n\n aov_name = self._get_attr(aov, \"name\")\n\n # Support light Groups\n light_groups = []\n if self._get_attr(aov, \"supportsLightGroups\"):\n all_light_groups = self._get_attr(aov, \"allLightGroups\")\n if all_light_groups:\n # All light groups is enabled\n light_groups = self._get_redshift_light_groups()\n else:\n value = self._get_attr(aov, \"lightGroupList\")\n # note: string value can return None when never set\n if value:\n selected_light_groups = value.strip().split()\n light_groups = selected_light_groups\n\n for light_group in light_groups:\n aov_light_group_name = \"{}_{}\".format(aov_name,\n light_group)\n for camera in cameras:\n product = RenderProduct(\n productName=aov_light_group_name,\n aov=aov_name,\n ext=ext,\n multipart=False,\n camera=camera,\n driver=aov,\n colorspace=colorspace)\n products.append(product)\n\n if light_groups:\n light_groups_enabled = True\n\n # Redshift AOV Light Select always renders the global AOV\n # even when light groups are present so we don't need to\n # exclude it when light groups are active\n for camera in cameras:\n product = RenderProduct(productName=aov_name,\n aov=aov_name,\n ext=ext,\n multipart=False,\n camera=camera,\n driver=aov,\n colorspace=colorspace)\n products.append(product)\n\n # When a Beauty AOV is added manually, it will be rendered as\n # 'Beauty_other' in file name and \"standard\" beauty will have\n # 'Beauty' in its name. When disabled, standard output will be\n # without `Beauty`. Except when using light groups.\n if light_groups_enabled:\n return products\n\n beauty_name = \"BeautyAux\" if has_beauty_aov else \"\"\n for camera in cameras:\n products.insert(0,\n RenderProduct(productName=beauty_name,\n ext=ext,\n multipart=self.multipart,\n camera=camera,\n colorspace=colorspace))\n\n return products\n\n @staticmethod\n def _get_redshift_light_groups():\n return sorted(mel.eval(\"redshiftAllAovLightGroups\"))\n\n\nclass RenderProductsRenderman(ARenderProducts):\n \"\"\"Expected files for Renderman renderer.\n\n Warning:\n This is very rudimentary and needs more love and testing.\n \"\"\"\n\n renderer = \"renderman\"\n unmerged_aovs = {\"PxrCryptomatte\"}\n\n def get_multipart(self):\n # Implemented as display specific in \"get_render_products\".\n return False\n\n def get_render_products(self):\n \"\"\"Get all AOVs.\n\n See Also:\n :func:`ARenderProducts.get_render_products()`\n\n \"\"\"\n from rfm2.api.displays import get_displays # noqa\n\n colorspace = lib.get_color_management_output_transform()\n\n cameras = [\n self.sanitize_camera_name(c)\n for c in self.get_renderable_cameras()\n ]\n\n if not cameras:\n cameras = [\n self.sanitize_camera_name(\n self.get_renderable_cameras()[0])\n ]\n products = []\n\n # NOTE: This is guessing extensions from renderman display types.\n # Some of them are just framebuffers, d_texture format can be\n # set in display setting. We set those now to None, but it\n # should be handled more gracefully.\n display_types = {\n \"d_deepexr\": \"exr\",\n \"d_it\": None,\n \"d_null\": None,\n \"d_openexr\": \"exr\",\n \"d_png\": \"png\",\n \"d_pointcloud\": \"ptc\",\n \"d_targa\": \"tga\",\n \"d_texture\": None,\n \"d_tiff\": \"tif\"\n }\n\n displays = get_displays(override_dst=\"render\")[\"displays\"]\n for name, display in displays.items():\n enabled = display[\"params\"][\"enable\"][\"value\"]\n if not enabled:\n continue\n\n # Skip display types not producing any file output.\n # Is there a better way to do it?\n if not display_types.get(display[\"driverNode\"][\"type\"]):\n continue\n\n has_cryptomatte = cmds.ls(type=self.unmerged_aovs)\n matte_enabled = False\n if has_cryptomatte:\n for cryptomatte in has_cryptomatte:\n cryptomatte_aov = cryptomatte\n matte_name = \"cryptomatte\"\n rman_globals = cmds.listConnections(cryptomatte +\n \".message\")\n if rman_globals:\n matte_enabled = True\n\n aov_name = name\n if aov_name == \"rmanDefaultDisplay\":\n aov_name = \"beauty\"\n\n extensions = display_types.get(\n display[\"driverNode\"][\"type\"], \"exr\")\n\n for camera in cameras:\n # Create render product and set it as multipart only on\n # display types supporting it. In all other cases, Renderman\n # will create separate output per channel.\n if display[\"driverNode\"][\"type\"] in [\"d_openexr\", \"d_deepexr\", \"d_tiff\"]: # noqa\n product = RenderProduct(\n productName=aov_name,\n ext=extensions,\n camera=camera,\n multipart=True,\n colorspace=colorspace\n )\n\n if has_cryptomatte and matte_enabled:\n cryptomatte = RenderProduct(\n productName=matte_name,\n aov=cryptomatte_aov,\n ext=extensions,\n camera=camera,\n multipart=True,\n colorspace=colorspace\n )\n else:\n # this code should handle the case where no multipart\n # capable format is selected. But since it involves\n # shady logic to determine what channel become what\n # lets not do that as all productions will use exr anyway.\n \"\"\"\n for channel in display['params']['displayChannels']['value']: # noqa\n product = RenderProduct(\n productName=\"{}_{}\".format(aov_name, channel),\n ext=extensions,\n camera=camera,\n multipart=False\n )\n \"\"\"\n raise UnsupportedImageFormatException(\n \"Only exr, deep exr and tiff formats are supported.\")\n\n products.append(product)\n\n if has_cryptomatte and matte_enabled:\n products.append(cryptomatte)\n\n return products\n\n def get_files(self, product):\n \"\"\"Get expected files.\n\n \"\"\"\n files = super(RenderProductsRenderman, self).get_files(product)\n\n layer_data = self.layer_data\n new_files = []\n\n resolved_image_dir = re.sub(\"\", layer_data.sceneName, RENDERMAN_IMAGE_DIR, flags=re.IGNORECASE) # noqa: E501\n resolved_image_dir = re.sub(\"\", layer_data.layerName, resolved_image_dir, flags=re.IGNORECASE) # noqa: E501\n for file in files:\n new_file = \"{}/{}\".format(resolved_image_dir, file)\n new_files.append(new_file)\n\n return new_files\n\n\nclass RenderProductsMayaHardware(ARenderProducts):\n \"\"\"Expected files for MayaHardware renderer.\"\"\"\n\n renderer = \"mayahardware2\"\n\n extensions = [\n {\"label\": \"JPEG\", \"index\": 8, \"extension\": \"jpg\"},\n {\"label\": \"PNG\", \"index\": 32, \"extension\": \"png\"},\n {\"label\": \"EXR(exr)\", \"index\": 40, \"extension\": \"exr\"}\n ]\n\n def get_multipart(self):\n # MayaHardware does not support multipart EXRs.\n return False\n\n def _get_extension(self, value):\n result = None\n if isinstance(value, int):\n extensions = {\n extension[\"index\"]: extension[\"extension\"]\n for extension in self.extensions\n }\n try:\n result = extensions[value]\n except KeyError:\n raise NotImplementedError(\n \"Could not find extension for {}\".format(value)\n )\n\n if isinstance(value, six.string_types):\n extensions = {\n extension[\"label\"]: extension[\"extension\"]\n for extension in self.extensions\n }\n try:\n result = extensions[value]\n except KeyError:\n raise NotImplementedError(\n \"Could not find extension for {}\".format(value)\n )\n\n if not result:\n raise NotImplementedError(\n \"Could not find extension for {}\".format(value)\n )\n\n return result\n\n def get_render_products(self):\n \"\"\"Get all AOVs.\n See Also:\n :func:`ARenderProducts.get_render_products()`\n \"\"\"\n ext = self._get_extension(\n self._get_attr(\"defaultRenderGlobals.imageFormat\")\n )\n\n products = []\n for cam in self.get_renderable_cameras():\n product = RenderProduct(\n productName=\"beauty\",\n ext=ext,\n camera=cam,\n colorspace=lib.get_color_management_output_transform()\n )\n products.append(product)\n\n return products\n\n\nclass AOVError(Exception):\n \"\"\"Custom exception for determining AOVs.\"\"\"\n\n\nclass UnsupportedRendererException(Exception):\n \"\"\"Custom exception.\n\n Raised when requesting data from unsupported renderer.\n \"\"\"\n\n\nclass UnsupportedImageFormatException(Exception):\n \"\"\"Custom exception to report unsupported output image format.\"\"\"\n"
},
{
"path": "openpype/hosts/maya/api/lib_rendersettings.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Class for handling Render Settings.\"\"\"\nimport six\nimport sys\n\nfrom openpype.lib import Logger\nfrom openpype.settings import get_project_settings\n\nfrom openpype.pipeline import CreatorError, get_current_project_name\nfrom openpype.pipeline.context_tools import get_current_project_asset\nfrom openpype.hosts.maya.api.lib import reset_frame_range\n\n\nclass RenderSettings(object):\n\n _image_prefix_nodes = {\n 'vray': 'vraySettings.fileNamePrefix',\n 'arnold': 'defaultRenderGlobals.imageFilePrefix',\n 'renderman': 'rmanGlobals.imageFileFormat',\n 'redshift': 'defaultRenderGlobals.imageFilePrefix',\n 'mayahardware2': 'defaultRenderGlobals.imageFilePrefix'\n }\n\n _aov_chars = {\n \"dot\": \".\",\n \"dash\": \"-\",\n \"underscore\": \"_\"\n }\n\n log = Logger.get_logger(\"RenderSettings\")\n\n @classmethod\n def get_image_prefix_attr(cls, renderer):\n return cls._image_prefix_nodes[renderer]\n\n @staticmethod\n def get_padding_attr(renderer):\n \"\"\"Return attribute for renderer that defines frame padding amount\"\"\"\n if renderer == \"vray\":\n return \"vraySettings.fileNamePadding\"\n else:\n return \"defaultRenderGlobals.extensionPadding\"\n\n def __init__(self, project_settings=None):\n if not project_settings:\n project_settings = get_project_settings(\n get_current_project_name()\n )\n render_settings = project_settings[\"maya\"][\"RenderSettings\"]\n image_prefixes = {\n \"vray\": render_settings[\"vray_renderer\"][\"image_prefix\"],\n \"arnold\": render_settings[\"arnold_renderer\"][\"image_prefix\"],\n \"renderman\": render_settings[\"renderman_renderer\"][\"image_prefix\"],\n \"redshift\": render_settings[\"redshift_renderer\"][\"image_prefix\"]\n }\n\n # TODO probably should be stored to more explicit attribute\n # Renderman only\n renderman_settings = render_settings[\"renderman_renderer\"]\n _image_dir = {\n \"renderman\": renderman_settings[\"image_dir\"],\n \"cryptomatte\": renderman_settings[\"cryptomatte_dir\"],\n \"imageDisplay\": renderman_settings[\"imageDisplay_dir\"],\n \"watermark\": renderman_settings[\"watermark_dir\"]\n }\n self._image_prefixes = image_prefixes\n self._image_dir = _image_dir\n self._project_settings = project_settings\n\n def set_default_renderer_settings(self, renderer=None):\n \"\"\"Set basic settings based on renderer.\"\"\"\n # Not all hosts can import this module.\n from maya import cmds # noqa: F401\n import maya.mel as mel # noqa: F401\n\n if not renderer:\n renderer = cmds.getAttr(\n 'defaultRenderGlobals.currentRenderer').lower()\n\n asset_doc = get_current_project_asset()\n # project_settings/maya/create/CreateRender/aov_separator\n try:\n aov_separator = self._aov_chars[(\n self._project_settings[\"maya\"]\n [\"RenderSettings\"]\n [\"aov_separator\"]\n )]\n except KeyError:\n aov_separator = \"_\"\n reset_frame = self._project_settings[\"maya\"][\"RenderSettings\"][\"reset_current_frame\"] # noqa\n\n if reset_frame:\n start_frame = cmds.getAttr(\"defaultRenderGlobals.startFrame\")\n cmds.currentTime(start_frame, edit=True)\n\n if renderer in self._image_prefix_nodes:\n prefix = self._image_prefixes[renderer]\n prefix = prefix.replace(\"{aov_separator}\", aov_separator)\n cmds.setAttr(self._image_prefix_nodes[renderer],\n prefix, type=\"string\") # noqa\n else:\n print(\"{0} isn't a supported renderer to autoset settings.\".format(renderer)) # noqa\n # TODO: handle not having res values in the doc\n width = asset_doc[\"data\"].get(\"resolutionWidth\")\n height = asset_doc[\"data\"].get(\"resolutionHeight\")\n\n if renderer == \"arnold\":\n # set renderer settings for Arnold from project settings\n self._set_arnold_settings(width, height)\n\n if renderer == \"vray\":\n self._set_vray_settings(aov_separator, width, height)\n\n if renderer == \"redshift\":\n self._set_redshift_settings(width, height)\n mel.eval(\"redshiftUpdateActiveAovList\")\n\n if renderer == \"renderman\":\n image_dir = self._image_dir[\"renderman\"]\n cmds.setAttr(\"rmanGlobals.imageOutputDir\",\n image_dir, type=\"string\")\n self._set_renderman_settings(width, height,\n aov_separator)\n\n def _set_arnold_settings(self, width, height):\n \"\"\"Sets settings for Arnold.\"\"\"\n from mtoa.core import createOptions # noqa\n from mtoa.aovs import AOVInterface # noqa\n # Not all hosts can import this module.\n from maya import cmds # noqa: F401\n import maya.mel as mel # noqa: F401\n\n createOptions()\n render_settings = self._project_settings[\"maya\"][\"RenderSettings\"]\n arnold_render_presets = render_settings[\"arnold_renderer\"] # noqa\n # Force resetting settings and AOV list to avoid having to deal with\n # AOV checking logic, for now.\n # This is a work around because the standard\n # function to revert render settings does not reset AOVs list in MtoA\n # Fetch current aovs in case there's any.\n current_aovs = AOVInterface().getAOVs()\n remove_aovs = render_settings[\"remove_aovs\"]\n if remove_aovs:\n # Remove fetched AOVs\n AOVInterface().removeAOVs(current_aovs)\n mel.eval(\"unifiedRenderGlobalsRevertToDefault\")\n img_ext = arnold_render_presets[\"image_format\"]\n img_prefix = arnold_render_presets[\"image_prefix\"]\n aovs = arnold_render_presets[\"aov_list\"]\n img_tiled = arnold_render_presets[\"tiled\"]\n multi_exr = arnold_render_presets[\"multilayer_exr\"]\n additional_options = arnold_render_presets[\"additional_options\"]\n for aov in aovs:\n if aov in current_aovs and not remove_aovs:\n continue\n AOVInterface('defaultArnoldRenderOptions').addAOV(aov)\n\n cmds.setAttr(\"defaultResolution.width\", width)\n cmds.setAttr(\"defaultResolution.height\", height)\n\n self._set_global_output_settings()\n\n cmds.setAttr(\n \"defaultRenderGlobals.imageFilePrefix\", img_prefix, type=\"string\")\n\n cmds.setAttr(\n \"defaultArnoldDriver.ai_translator\", img_ext, type=\"string\")\n\n cmds.setAttr(\n \"defaultArnoldDriver.exrTiled\", img_tiled)\n\n cmds.setAttr(\n \"defaultArnoldDriver.mergeAOVs\", multi_exr)\n self._additional_attribs_setter(additional_options)\n reset_frame_range(playback=False, fps=False, render=True)\n\n def _set_redshift_settings(self, width, height):\n \"\"\"Sets settings for Redshift.\"\"\"\n # Not all hosts can import this module.\n from maya import cmds # noqa: F401\n import maya.mel as mel # noqa: F401\n\n render_settings = self._project_settings[\"maya\"][\"RenderSettings\"]\n redshift_render_presets = render_settings[\"redshift_renderer\"]\n\n remove_aovs = render_settings[\"remove_aovs\"]\n all_rs_aovs = cmds.ls(type='RedshiftAOV')\n if remove_aovs:\n for aov in all_rs_aovs:\n enabled = cmds.getAttr(\"{}.enabled\".format(aov))\n if enabled:\n cmds.delete(aov)\n\n redshift_aovs = redshift_render_presets[\"aov_list\"]\n # list all the aovs\n all_rs_aovs = cmds.ls(type='RedshiftAOV')\n for rs_aov in redshift_aovs:\n rs_layername = \"rsAov_{}\".format(rs_aov.replace(\" \", \"\"))\n if rs_layername in all_rs_aovs:\n continue\n cmds.rsCreateAov(type=rs_aov)\n # update the AOV list\n mel.eval(\"redshiftUpdateActiveAovList\")\n\n rs_p_engine = redshift_render_presets[\"primary_gi_engine\"]\n rs_s_engine = redshift_render_presets[\"secondary_gi_engine\"]\n\n if int(rs_p_engine) or int(rs_s_engine) != 0:\n cmds.setAttr(\"redshiftOptions.GIEnabled\", 1)\n if int(rs_p_engine) == 0:\n # reset the primary GI Engine as default\n cmds.setAttr(\"redshiftOptions.primaryGIEngine\", 4)\n if int(rs_s_engine) == 0:\n # reset the secondary GI Engine as default\n cmds.setAttr(\"redshiftOptions.secondaryGIEngine\", 2)\n else:\n cmds.setAttr(\"redshiftOptions.GIEnabled\", 0)\n\n cmds.setAttr(\"redshiftOptions.primaryGIEngine\", int(rs_p_engine))\n cmds.setAttr(\"redshiftOptions.secondaryGIEngine\", int(rs_s_engine))\n\n additional_options = redshift_render_presets[\"additional_options\"]\n ext = redshift_render_presets[\"image_format\"]\n img_exts = [\"iff\", \"exr\", \"tif\", \"png\", \"tga\", \"jpg\"]\n img_ext = img_exts.index(ext)\n\n self._set_global_output_settings()\n cmds.setAttr(\"redshiftOptions.imageFormat\", img_ext)\n cmds.setAttr(\"defaultResolution.width\", width)\n cmds.setAttr(\"defaultResolution.height\", height)\n self._additional_attribs_setter(additional_options)\n\n def _set_renderman_settings(self, width, height, aov_separator):\n \"\"\"Sets settings for Renderman\"\"\"\n # Not all hosts can import this module.\n from maya import cmds # noqa: F401\n import maya.mel as mel # noqa: F401\n\n rman_render_presets = (\n self._project_settings\n [\"maya\"]\n [\"RenderSettings\"]\n [\"renderman_renderer\"]\n )\n display_filters = rman_render_presets[\"display_filters\"]\n d_filters_number = len(display_filters)\n for i in range(d_filters_number):\n d_node = cmds.ls(typ=display_filters[i])\n if len(d_node) > 0:\n filter_nodes = d_node[0]\n else:\n filter_nodes = cmds.createNode(display_filters[i])\n\n cmds.connectAttr(filter_nodes + \".message\",\n \"rmanGlobals.displayFilters[%i]\" % i,\n force=True)\n if filter_nodes.startswith(\"PxrImageDisplayFilter\"):\n imageDisplay_dir = self._image_dir[\"imageDisplay\"]\n imageDisplay_dir = imageDisplay_dir.replace(\"{aov_separator}\",\n aov_separator)\n cmds.setAttr(filter_nodes + \".filename\",\n imageDisplay_dir, type=\"string\")\n\n sample_filters = rman_render_presets[\"sample_filters\"]\n s_filters_number = len(sample_filters)\n for n in range(s_filters_number):\n s_node = cmds.ls(typ=sample_filters[n])\n if len(s_node) > 0:\n filter_nodes = s_node[0]\n else:\n filter_nodes = cmds.createNode(sample_filters[n])\n\n cmds.connectAttr(filter_nodes + \".message\",\n \"rmanGlobals.sampleFilters[%i]\" % n,\n force=True)\n\n if filter_nodes.startswith(\"PxrCryptomatte\"):\n matte_dir = self._image_dir[\"cryptomatte\"]\n matte_dir = matte_dir.replace(\"{aov_separator}\",\n aov_separator)\n cmds.setAttr(filter_nodes + \".filename\",\n matte_dir, type=\"string\")\n elif filter_nodes.startswith(\"PxrWatermarkFilter\"):\n watermark_dir = self._image_dir[\"watermark\"]\n watermark_dir = watermark_dir.replace(\"{aov_separator}\",\n aov_separator)\n cmds.setAttr(filter_nodes + \".filename\",\n watermark_dir, type=\"string\")\n\n additional_options = rman_render_presets[\"additional_options\"]\n\n self._set_global_output_settings()\n cmds.setAttr(\"defaultResolution.width\", width)\n cmds.setAttr(\"defaultResolution.height\", height)\n self._additional_attribs_setter(additional_options)\n\n def _set_vray_settings(self, aov_separator, width, height):\n # type: (str, int, int) -> None\n \"\"\"Sets important settings for Vray.\"\"\"\n # Not all hosts can import this module.\n from maya import cmds # noqa: F401\n import maya.mel as mel # noqa: F401\n\n\n settings = cmds.ls(type=\"VRaySettingsNode\")\n node = settings[0] if settings else cmds.createNode(\"VRaySettingsNode\")\n render_settings = self._project_settings[\"maya\"][\"RenderSettings\"]\n vray_render_presets = render_settings[\"vray_renderer\"]\n # vrayRenderElement\n remove_aovs = render_settings[\"remove_aovs\"]\n all_vray_aovs = cmds.ls(type='VRayRenderElement')\n lightSelect_aovs = cmds.ls(type='VRayRenderElementSet')\n if remove_aovs:\n for aov in all_vray_aovs:\n # remove all aovs except LightSelect\n enabled = cmds.getAttr(\"{}.enabled\".format(aov))\n if enabled:\n cmds.delete(aov)\n # remove LightSelect\n for light_aovs in lightSelect_aovs:\n light_enabled = cmds.getAttr(\"{}.enabled\".format(light_aovs))\n if light_enabled:\n cmds.delete(lightSelect_aovs)\n\n vray_aovs = vray_render_presets[\"aov_list\"]\n for renderlayer in vray_aovs:\n renderElement = \"vrayAddRenderElement {}\".format(renderlayer)\n RE_name = mel.eval(renderElement)\n # if there is more than one same render element\n if RE_name.endswith(\"1\"):\n cmds.delete(RE_name)\n # Set aov separator\n # First we need to explicitly set the UI items in Render Settings\n # because that is also what V-Ray updates to when that Render Settings\n # UI did initialize before and refreshes again.\n MENU = \"vrayRenderElementSeparator\"\n if cmds.optionMenuGrp(MENU, query=True, exists=True):\n items = cmds.optionMenuGrp(MENU, query=True, ill=True)\n separators = [cmds.menuItem(i, query=True, label=True) for i in items] # noqa: E501\n try:\n sep_idx = separators.index(aov_separator)\n except ValueError:\n six.reraise(\n CreatorError,\n CreatorError(\n \"AOV character {} not in {}\".format(\n aov_separator, separators)),\n sys.exc_info()[2])\n\n cmds.optionMenuGrp(MENU, edit=True, select=sep_idx + 1)\n\n # Set the render element attribute as string. This is also what V-Ray\n # sets whenever the `vrayRenderElementSeparator` menu items switch\n cmds.setAttr(\n \"{}.fileNameRenderElementSeparator\".format(node),\n aov_separator,\n type=\"string\"\n )\n\n # Set render file format to exr\n ext = vray_render_presets[\"image_format\"]\n cmds.setAttr(\"{}.imageFormatStr\".format(node), ext, type=\"string\")\n\n # animType\n cmds.setAttr(\"{}.animType\".format(node), 1)\n\n # resolution\n cmds.setAttr(\"{}.width\".format(node), width)\n cmds.setAttr(\"{}.height\".format(node), height)\n\n additional_options = vray_render_presets[\"additional_options\"]\n\n self._additional_attribs_setter(additional_options)\n\n @staticmethod\n def _set_global_output_settings():\n # Not all hosts can import this module.\n from maya import cmds # noqa: F401\n import maya.mel as mel # noqa: F401\n\n # enable animation\n cmds.setAttr(\"defaultRenderGlobals.outFormatControl\", 0)\n cmds.setAttr(\"defaultRenderGlobals.animation\", 1)\n cmds.setAttr(\"defaultRenderGlobals.putFrameBeforeExt\", 1)\n cmds.setAttr(\"defaultRenderGlobals.extensionPadding\", 4)\n\n def _additional_attribs_setter(self, additional_attribs):\n # Not all hosts can import this module.\n from maya import cmds # noqa: F401\n import maya.mel as mel # noqa: F401\n\n for item in additional_attribs:\n attribute, value = item\n attribute = str(attribute) # ensure str conversion from settings\n attribute_type = cmds.getAttr(attribute, type=True)\n if attribute_type in {\"long\", \"bool\"}:\n cmds.setAttr(attribute, int(value))\n elif attribute_type == \"string\":\n cmds.setAttr(attribute, str(value), type=\"string\")\n elif attribute_type in {\"double\", \"doubleAngle\", \"doubleLinear\"}:\n cmds.setAttr(attribute, float(value))\n else:\n self.log.error(\n \"Attribute {attribute} can not be set due to unsupported \"\n \"type: {attribute_type}\".format(\n attribute=attribute,\n attribute_type=attribute_type)\n )\n"
},
{
"path": "openpype/hosts/maya/api/lib_rendersetup.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Code to get attributes from render layer without switching to it.\n\nhttps://github.com/Colorbleed/colorbleed-config/blob/acre/colorbleed/maya/lib_rendersetup.py\nCredits: Roy Nieterau (BigRoy) / Colorbleed\nModified for use in OpenPype\n\n\"\"\"\n\nfrom maya import cmds\nimport maya.api.OpenMaya as om\nimport logging\n\nimport maya.app.renderSetup.model.utils as utils\nfrom maya.app.renderSetup.model import renderSetup\nfrom maya.app.renderSetup.model.override import (\n AbsOverride,\n RelOverride,\n UniqueOverride\n)\n\nfrom openpype.hosts.maya.api.lib import get_attribute\n\nEXACT_MATCH = 0\nPARENT_MATCH = 1\nCLIENT_MATCH = 2\n\nDEFAULT_RENDER_LAYER = \"defaultRenderLayer\"\n\nlog = logging.getLogger(__name__)\n\n\ndef get_rendersetup_layer(layer):\n \"\"\"Return render setup layer name.\n\n This also converts names from legacy renderLayer node name to render setup\n name.\n\n Note: `defaultRenderLayer` is not a renderSetupLayer node but it is however\n the valid layer name for Render Setup - so we return that as is.\n\n Example:\n >>> for legacy_layer in cmds.ls(type=\"renderLayer\"):\n >>> layer = get_rendersetup_layer(legacy_layer)\n\n Returns:\n str or None: Returns renderSetupLayer node name if `layer` is a valid\n layer name in legacy renderlayers or render setup layers.\n Returns None if the layer can't be found or Render Setup is\n currently disabled.\n\n\n \"\"\"\n if layer == DEFAULT_RENDER_LAYER:\n # defaultRenderLayer doesn't have a `renderSetupLayer`\n return layer\n\n if not cmds.mayaHasRenderSetup():\n return None\n\n if not cmds.objExists(layer):\n return None\n\n if cmds.nodeType(layer) == \"renderSetupLayer\":\n return layer\n\n # By default Render Setup renames the legacy renderlayer\n # to `rs_` but lets not rely on that as the\n # layer node can be renamed manually\n connections = cmds.listConnections(layer + \".message\",\n type=\"renderSetupLayer\",\n exactType=True,\n source=False,\n destination=True,\n plugs=True) or []\n return next((conn.split(\".\", 1)[0] for conn in connections\n if conn.endswith(\".legacyRenderLayer\")), None)\n\n\ndef get_attr_in_layer(node_attr, layer):\n \"\"\"Return attribute value in Render Setup layer.\n\n This will only work for attributes which can be\n retrieved with `maya.cmds.getAttr` and for which\n Relative and Absolute overrides are applicable.\n\n Examples:\n >>> get_attr_in_layer(\"defaultResolution.width\", layer=\"layer1\")\n >>> get_attr_in_layer(\"defaultRenderGlobals.startFrame\", layer=\"layer\")\n >>> get_attr_in_layer(\"transform.translate\", layer=\"layer3\")\n\n Args:\n attr (str): attribute name as 'node.attribute'\n layer (str): layer name\n\n Returns:\n object: attribute value in layer\n\n \"\"\"\n\n def _layer_needs_update(layer):\n \"\"\"Return whether layer needs updating.\"\"\"\n # Use `getattr` as e.g. DEFAULT_RENDER_LAYER does not have\n # the attribute\n return getattr(layer, \"needsMembershipUpdate\", False) or \\\n getattr(layer, \"needsApplyUpdate\", False)\n\n def get_default_layer_value(node_attr_):\n \"\"\"Return attribute value in `DEFAULT_RENDER_LAYER`.\"\"\"\n inputs = cmds.listConnections(node_attr_,\n source=True,\n destination=False,\n # We want to skip conversion nodes since\n # an override to `endFrame` could have\n # a `unitToTimeConversion` node\n # in-between\n skipConversionNodes=True,\n type=\"applyOverride\") or []\n if inputs:\n override = inputs[0]\n history_overrides = cmds.ls(cmds.listHistory(override,\n pruneDagObjects=True),\n type=\"applyOverride\")\n node = history_overrides[-1] if history_overrides else override\n node_attr_ = node + \".original\"\n\n return get_attribute(node_attr_, asString=True)\n\n layer = get_rendersetup_layer(layer)\n rs = renderSetup.instance()\n current_layer = rs.getVisibleRenderLayer()\n if current_layer.name() == layer:\n\n # Ensure layer is up-to-date\n if _layer_needs_update(current_layer):\n try:\n rs.switchToLayer(current_layer)\n except RuntimeError:\n # Some cases can cause errors on switching\n # the first time with Render Setup layers\n # e.g. different overrides to compounds\n # and its children plugs. So we just force\n # it another time. If it then still fails\n # we will let it error out.\n rs.switchToLayer(current_layer)\n\n return get_attribute(node_attr, asString=True)\n\n overrides = get_attr_overrides(node_attr, layer)\n default_layer_value = get_default_layer_value(node_attr)\n if not overrides:\n return default_layer_value\n\n value = default_layer_value\n for match, layer_override, index in overrides:\n if isinstance(layer_override, AbsOverride):\n # Absolute override\n value = get_attribute(layer_override.name() + \".attrValue\")\n if match == EXACT_MATCH:\n # value = value\n pass\n elif match == PARENT_MATCH:\n value = value[index]\n elif match == CLIENT_MATCH:\n value[index] = value\n\n elif isinstance(layer_override, RelOverride):\n # Relative override\n # Value = Original * Multiply + Offset\n multiply = get_attribute(layer_override.name() + \".multiply\")\n offset = get_attribute(layer_override.name() + \".offset\")\n\n if match == EXACT_MATCH:\n value = value * multiply + offset\n elif match == PARENT_MATCH:\n value = value * multiply[index] + offset[index]\n elif match == CLIENT_MATCH:\n value[index] = value[index] * multiply + offset\n\n else:\n raise TypeError(\"Unsupported override: %s\" % layer_override)\n\n return value\n\n\ndef get_attr_overrides(node_attr, layer,\n skip_disabled=True,\n skip_local_render=True,\n stop_at_absolute_override=True):\n \"\"\"Return all Overrides applicable to the attribute.\n\n Overrides are returned as a 3-tuple:\n (Match, Override, Index)\n\n Match:\n This is any of EXACT_MATCH, PARENT_MATCH, CLIENT_MATCH\n and defines whether the override is exactly on the\n plug, on the parent or on a child plug.\n\n Override:\n This is the RenderSetup Override instance.\n\n Index:\n This is the Plug index under the parent or for\n the child that matches. The EXACT_MATCH index will\n always be None. For PARENT_MATCH the index is which\n index the plug is under the parent plug. For CLIENT_MATCH\n the index is which child index matches the plug.\n\n Args:\n node_attr (str): attribute name as 'node.attribute'\n layer (str): layer name\n skip_disabled (bool): exclude disabled overrides\n skip_local_render (bool): exclude overrides marked\n as local render.\n stop_at_absolute_override: exclude overrides prior\n to the last absolute override as they have\n no influence on the resulting value.\n\n Returns:\n list: Ordered Overrides in order of strength\n\n \"\"\"\n\n def get_mplug_children(plug):\n \"\"\"Return children MPlugs of compound `MPlug`.\"\"\"\n children = []\n if plug.isCompound:\n for i in range(plug.numChildren()):\n children.append(plug.child(i))\n return children\n\n def get_mplug_names(mplug):\n \"\"\"Return long and short name of `MPlug`.\"\"\"\n long_name = mplug.partialName(useLongNames=True)\n short_name = mplug.partialName(useLongNames=False)\n return {long_name, short_name}\n\n def iter_override_targets(override):\n try:\n for target in override._targets():\n yield target\n except AssertionError:\n # Workaround: There is a bug where the private `_targets()` method\n # fails on some attribute plugs. For example overrides\n # to the defaultRenderGlobals.endFrame\n # (Tested in Maya 2020.2)\n log.debug(\"Workaround for %s\" % override)\n from maya.app.renderSetup.common.utils import findPlug\n\n attr = override.attributeName()\n if isinstance(override, UniqueOverride):\n node = override.targetNodeName()\n yield findPlug(node, attr)\n else:\n nodes = override.parent().selector().nodes()\n for node in nodes:\n if cmds.attributeQuery(attr, node=node, exists=True):\n yield findPlug(node, attr)\n\n # Get the MPlug for the node.attr\n sel = om.MSelectionList()\n sel.add(node_attr)\n plug = sel.getPlug(0)\n\n layer = get_rendersetup_layer(layer)\n if layer == DEFAULT_RENDER_LAYER:\n # DEFAULT_RENDER_LAYER will never have overrides\n # since it's the default layer\n return []\n\n rs_layer = renderSetup.instance().getRenderLayer(layer)\n if rs_layer is None:\n # Renderlayer does not exist\n return\n\n # Get any parent or children plugs as we also\n # want to include them in the attribute match\n # for overrides\n parent = plug.parent() if plug.isChild else None\n parent_index = None\n if parent:\n parent_index = get_mplug_children(parent).index(plug)\n\n children = get_mplug_children(plug)\n\n # Create lookup for the attribute by both long\n # and short names\n attr_names = get_mplug_names(plug)\n for child in children:\n attr_names.update(get_mplug_names(child))\n if parent:\n attr_names.update(get_mplug_names(parent))\n\n # Get all overrides of the layer\n # And find those that are relevant to the attribute\n plug_overrides = []\n\n # Iterate over the overrides in reverse so we get the last\n # overrides first and can \"break\" whenever an absolute\n # override is reached\n layer_overrides = list(utils.getOverridesRecursive(rs_layer))\n for layer_override in reversed(layer_overrides):\n\n if skip_disabled and not layer_override.isEnabled():\n # Ignore disabled overrides\n continue\n\n if skip_local_render and layer_override.isLocalRender():\n continue\n\n # The targets list can be very large so we'll do\n # a quick filter by attribute name to detect whether\n # it matches the attribute name, or its parent or child\n if layer_override.attributeName() not in attr_names:\n continue\n\n override_match = None\n for override_plug in iter_override_targets(layer_override):\n\n override_match = None\n if plug == override_plug:\n override_match = (EXACT_MATCH, layer_override, None)\n\n elif parent and override_plug == parent:\n override_match = (PARENT_MATCH, layer_override, parent_index)\n\n elif children and override_plug in children:\n child_index = children.index(override_plug)\n override_match = (CLIENT_MATCH, layer_override, child_index)\n\n if override_match:\n plug_overrides.append(override_match)\n break\n\n if (\n override_match and\n stop_at_absolute_override and\n isinstance(layer_override, AbsOverride) and\n # When the override is only on a child plug then it doesn't\n # override the entire value so we not stop at this override\n not override_match[0] == CLIENT_MATCH\n ):\n # If override is absolute override, then BREAK out\n # of parent loop we don't need to look any further as\n # this is the absolute override\n break\n\n return reversed(plug_overrides)\n\n\ndef get_shader_in_layer(node, layer):\n \"\"\"Return the assigned shader in a renderlayer without switching layers.\n\n This has been developed and tested for Legacy Renderlayers and *not* for\n Render Setup.\n\n Note: This will also return the shader for any face assignments, however\n it will *not* return the components they are assigned to. This could\n be implemented, but since Maya's renderlayers are famous for breaking\n with face assignments there has been no need for this function to\n support that.\n\n Returns:\n list: The list of assigned shaders in the given layer.\n\n \"\"\"\n\n def _get_connected_shader(plug):\n \"\"\"Return current shader\"\"\"\n return cmds.listConnections(plug,\n source=False,\n destination=True,\n plugs=False,\n connections=False,\n type=\"shadingEngine\") or []\n\n # We check the instObjGroups (shader connection) for layer overrides.\n plug = node + \".instObjGroups\"\n\n # Ignore complex query if we're in the layer anyway (optimization)\n current_layer = cmds.editRenderLayerGlobals(query=True,\n currentRenderLayer=True)\n if layer == current_layer:\n return _get_connected_shader(plug)\n\n connections = cmds.listConnections(plug,\n plugs=True,\n source=False,\n destination=True,\n type=\"renderLayer\") or []\n connections = filter(lambda x: x.endswith(\".outPlug\"), connections)\n if not connections:\n # If no overrides anywhere on the shader, just get the current shader\n return _get_connected_shader(plug)\n\n def _get_override(connections, layer):\n \"\"\"Return the overridden connection for that layer in connections\"\"\"\n # If there's an override on that layer, return that.\n for connection in connections:\n if (connection.startswith(layer + \".outAdjustments\") and\n connection.endswith(\".outPlug\")):\n\n # This is a shader override on that layer so get the shader\n # connected to .outValue of the .outAdjustment[i]\n out_adjustment = connection.rsplit(\".\", 1)[0]\n connection_attr = out_adjustment + \".outValue\"\n override = cmds.listConnections(connection_attr) or []\n\n return override\n\n override_shader = _get_override(connections, layer)\n if override_shader is not None:\n return override_shader\n else:\n # Get the override for \"defaultRenderLayer\" (=masterLayer)\n return _get_override(connections, layer=\"defaultRenderLayer\")\n"
},
{
"path": "openpype/hosts/maya/api/menu.py",
"content": "import os\nimport logging\nfrom functools import partial\n\nfrom qtpy import QtWidgets, QtGui\n\nimport maya.utils\nimport maya.cmds as cmds\n\nfrom openpype.pipeline import (\n get_current_asset_name,\n get_current_task_name\n)\nfrom openpype.pipeline.workfile import BuildWorkfile\nfrom openpype.tools.utils import host_tools\nfrom openpype.hosts.maya.api import lib, lib_rendersettings\nfrom .lib import get_main_window, IS_HEADLESS\nfrom ..tools import show_look_assigner\n\nfrom .workfile_template_builder import (\n create_placeholder,\n update_placeholder,\n build_workfile_template,\n update_workfile_template,\n)\n\nlog = logging.getLogger(__name__)\n\nMENU_NAME = \"op_maya_menu\"\n\n\ndef _get_menu(menu_name=None):\n \"\"\"Return the menu instance if it currently exists in Maya\"\"\"\n if menu_name is None:\n menu_name = MENU_NAME\n\n widgets = {w.objectName(): w for w in QtWidgets.QApplication.allWidgets()}\n return widgets.get(menu_name)\n\n\ndef get_context_label():\n return \"{}, {}\".format(\n get_current_asset_name(),\n get_current_task_name()\n )\n\n\ndef install(project_settings):\n if cmds.about(batch=True):\n log.info(\"Skipping openpype.menu initialization in batch mode..\")\n return\n\n def add_menu():\n pyblish_icon = host_tools.get_pyblish_icon()\n parent_widget = get_main_window()\n cmds.menu(\n MENU_NAME,\n label=os.environ.get(\"AVALON_LABEL\") or \"OpenPype\",\n tearOff=True,\n parent=\"MayaWindow\"\n )\n\n # Create context menu\n cmds.menuItem(\n \"currentContext\",\n label=get_context_label(),\n parent=MENU_NAME,\n enable=False\n )\n\n cmds.setParent(\"..\", menu=True)\n\n cmds.menuItem(divider=True)\n\n cmds.menuItem(\n \"Create...\",\n command=lambda *args: host_tools.show_publisher(\n parent=parent_widget,\n tab=\"create\"\n )\n )\n\n cmds.menuItem(\n \"Load...\",\n command=lambda *args: host_tools.show_loader(\n parent=parent_widget,\n use_context=True\n )\n )\n\n cmds.menuItem(\n \"Publish...\",\n command=lambda *args: host_tools.show_publisher(\n parent=parent_widget,\n tab=\"publish\"\n ),\n image=pyblish_icon\n )\n\n cmds.menuItem(\n \"Manage...\",\n command=lambda *args: host_tools.show_scene_inventory(\n parent=parent_widget\n )\n )\n\n cmds.menuItem(\n \"Library...\",\n command=lambda *args: host_tools.show_library_loader(\n parent=parent_widget\n )\n )\n\n cmds.menuItem(divider=True)\n\n cmds.menuItem(\n \"Work Files...\",\n command=lambda *args: host_tools.show_workfiles(\n parent=parent_widget\n ),\n )\n\n cmds.menuItem(\n \"Set Frame Range\",\n command=lambda *args: lib.reset_frame_range()\n )\n\n cmds.menuItem(\n \"Set Resolution\",\n command=lambda *args: lib.reset_scene_resolution()\n )\n\n cmds.menuItem(\n \"Set Colorspace\",\n command=lambda *args: lib.set_colorspace(),\n )\n\n cmds.menuItem(\n \"Set Render Settings\",\n command=lambda *args: lib_rendersettings.RenderSettings().set_default_renderer_settings() # noqa\n )\n\n cmds.menuItem(divider=True, parent=MENU_NAME)\n cmds.menuItem(\n \"Build First Workfile\",\n parent=MENU_NAME,\n command=lambda *args: BuildWorkfile().process()\n )\n\n cmds.menuItem(\n \"Look assigner...\",\n command=lambda *args: show_look_assigner(\n parent_widget\n )\n )\n\n cmds.menuItem(\n \"Experimental tools...\",\n command=lambda *args: host_tools.show_experimental_tools_dialog(\n parent_widget\n )\n )\n\n builder_menu = cmds.menuItem(\n \"Template Builder\",\n subMenu=True,\n tearOff=True,\n parent=MENU_NAME\n )\n cmds.menuItem(\n \"Create Placeholder\",\n parent=builder_menu,\n command=create_placeholder\n )\n cmds.menuItem(\n \"Update Placeholder\",\n parent=builder_menu,\n command=update_placeholder\n )\n cmds.menuItem(\n \"Build Workfile from template\",\n parent=builder_menu,\n command=build_workfile_template\n )\n cmds.menuItem(\n \"Update Workfile from template\",\n parent=builder_menu,\n command=update_workfile_template\n )\n\n cmds.setParent(MENU_NAME, menu=True)\n\n def add_scripts_menu(project_settings):\n try:\n import scriptsmenu.launchformaya as launchformaya\n except ImportError:\n log.warning(\n \"Skipping studio.menu install, because \"\n \"'scriptsmenu' module seems unavailable.\"\n )\n return\n\n config = project_settings[\"maya\"][\"scriptsmenu\"][\"definition\"]\n _menu = project_settings[\"maya\"][\"scriptsmenu\"][\"name\"]\n\n if not config:\n log.warning(\"Skipping studio menu, no definition found.\")\n return\n\n # run the launcher for Maya menu\n studio_menu = launchformaya.main(\n title=_menu.title(),\n objectName=_menu.title().lower().replace(\" \", \"_\")\n )\n\n # apply configuration\n studio_menu.build_from_configuration(studio_menu, config)\n\n # Allow time for uninstallation to finish.\n # We use Maya's executeDeferred instead of QTimer.singleShot\n # so that it only gets called after Maya UI has initialized too.\n # This is crucial with Maya 2020+ which initializes without UI\n # first as a QCoreApplication\n maya.utils.executeDeferred(add_menu)\n cmds.evalDeferred(partial(add_scripts_menu, project_settings),\n lowestPriority=True)\n\n\ndef uninstall():\n menu = _get_menu()\n if menu:\n log.info(\"Attempting to uninstall ...\")\n\n try:\n menu.deleteLater()\n del menu\n except Exception as e:\n log.error(e)\n\n\ndef popup():\n \"\"\"Pop-up the existing menu near the mouse cursor.\"\"\"\n menu = _get_menu()\n cursor = QtGui.QCursor()\n point = cursor.pos()\n menu.exec_(point)\n\n\ndef update_menu_task_label():\n \"\"\"Update the task label in Avalon menu to current session\"\"\"\n\n if IS_HEADLESS:\n return\n\n object_name = \"{}|currentContext\".format(MENU_NAME)\n if not cmds.menuItem(object_name, query=True, exists=True):\n log.warning(\"Can't find menuItem: {}\".format(object_name))\n return\n\n label = get_context_label()\n cmds.menuItem(object_name, edit=True, label=label)\n"
},
{
"path": "openpype/hosts/maya/api/pipeline.py",
"content": "import json\nimport base64\nimport os\nimport errno\nimport logging\nimport contextlib\nimport shutil\n\nfrom maya import utils, cmds, OpenMaya\nimport maya.api.OpenMaya as om\n\nimport pyblish.api\n\nfrom openpype.settings import get_project_settings\nfrom openpype.host import (\n HostBase,\n IWorkfileHost,\n ILoadHost,\n IPublishHost,\n HostDirmap,\n)\nfrom openpype.tools.utils import host_tools\nfrom openpype.tools.workfiles.lock_dialog import WorkfileLockDialog\nfrom openpype.lib import (\n register_event_callback,\n emit_event\n)\nfrom openpype.pipeline import (\n legacy_io,\n get_current_project_name,\n register_loader_plugin_path,\n register_inventory_action_path,\n register_creator_plugin_path,\n deregister_loader_plugin_path,\n deregister_inventory_action_path,\n deregister_creator_plugin_path,\n AVALON_CONTAINER_ID,\n)\nfrom openpype.pipeline.load import any_outdated_containers\nfrom openpype.pipeline.workfile.lock_workfile import (\n create_workfile_lock,\n remove_workfile_lock,\n is_workfile_locked,\n is_workfile_lock_enabled\n)\nfrom openpype.hosts.maya import MAYA_ROOT_DIR\nfrom openpype.hosts.maya.lib import create_workspace_mel\n\nfrom . import menu, lib\nfrom .workfile_template_builder import MayaPlaceholderLoadPlugin\nfrom .workio import (\n open_file,\n save_file,\n file_extensions,\n has_unsaved_changes,\n work_root,\n current_file\n)\n\nlog = logging.getLogger(\"openpype.hosts.maya\")\n\nPLUGINS_DIR = os.path.join(MAYA_ROOT_DIR, \"plugins\")\nPUBLISH_PATH = os.path.join(PLUGINS_DIR, \"publish\")\nLOAD_PATH = os.path.join(PLUGINS_DIR, \"load\")\nCREATE_PATH = os.path.join(PLUGINS_DIR, \"create\")\nINVENTORY_PATH = os.path.join(PLUGINS_DIR, \"inventory\")\n\nAVALON_CONTAINERS = \":AVALON_CONTAINERS\"\n\n\nclass MayaHost(HostBase, IWorkfileHost, ILoadHost, IPublishHost):\n name = \"maya\"\n\n def __init__(self):\n super(MayaHost, self).__init__()\n self._op_events = {}\n\n def install(self):\n project_name = get_current_project_name()\n project_settings = get_project_settings(project_name)\n # process path mapping\n dirmap_processor = MayaDirmap(\"maya\", project_name, project_settings)\n dirmap_processor.process_dirmap()\n\n pyblish.api.register_plugin_path(PUBLISH_PATH)\n pyblish.api.register_host(\"mayabatch\")\n pyblish.api.register_host(\"mayapy\")\n pyblish.api.register_host(\"maya\")\n\n register_loader_plugin_path(LOAD_PATH)\n register_creator_plugin_path(CREATE_PATH)\n register_inventory_action_path(INVENTORY_PATH)\n self.log.info(PUBLISH_PATH)\n\n self.log.info(\"Installing callbacks ... \")\n register_event_callback(\"init\", on_init)\n\n _set_project()\n\n if lib.IS_HEADLESS:\n self.log.info((\n \"Running in headless mode, skipping Maya save/open/new\"\n \" callback installation..\"\n ))\n\n return\n\n self._register_callbacks()\n\n menu.install(project_settings)\n\n register_event_callback(\"save\", on_save)\n register_event_callback(\"open\", on_open)\n register_event_callback(\"new\", on_new)\n register_event_callback(\"before.save\", on_before_save)\n register_event_callback(\"after.save\", on_after_save)\n register_event_callback(\"before.close\", on_before_close)\n register_event_callback(\"before.file.open\", before_file_open)\n register_event_callback(\"taskChanged\", on_task_changed)\n register_event_callback(\"workfile.open.before\", before_workfile_open)\n register_event_callback(\"workfile.save.before\", before_workfile_save)\n register_event_callback(\n \"workfile.save.before\", workfile_save_before_xgen\n )\n register_event_callback(\"workfile.save.after\", after_workfile_save)\n\n def open_workfile(self, filepath):\n return open_file(filepath)\n\n def save_workfile(self, filepath=None):\n return save_file(filepath)\n\n def work_root(self, session):\n return work_root(session)\n\n def get_current_workfile(self):\n return current_file()\n\n def workfile_has_unsaved_changes(self):\n return has_unsaved_changes()\n\n def get_workfile_extensions(self):\n return file_extensions()\n\n def get_containers(self):\n return ls()\n\n def get_workfile_build_placeholder_plugins(self):\n return [\n MayaPlaceholderLoadPlugin\n ]\n\n @contextlib.contextmanager\n def maintained_selection(self):\n with lib.maintained_selection():\n yield\n\n def get_context_data(self):\n data = cmds.fileInfo(\"OpenPypeContext\", query=True)\n if not data:\n return {}\n\n data = data[0] # Maya seems to return a list\n decoded = base64.b64decode(data).decode(\"utf-8\")\n return json.loads(decoded)\n\n def update_context_data(self, data, changes):\n json_str = json.dumps(data)\n encoded = base64.b64encode(json_str.encode(\"utf-8\"))\n return cmds.fileInfo(\"OpenPypeContext\", encoded)\n\n def _register_callbacks(self):\n for handler, event in self._op_events.copy().items():\n if event is None:\n continue\n\n try:\n OpenMaya.MMessage.removeCallback(event)\n self._op_events[handler] = None\n except RuntimeError as exc:\n self.log.info(exc)\n\n self._op_events[_on_scene_save] = OpenMaya.MSceneMessage.addCallback(\n OpenMaya.MSceneMessage.kBeforeSave, _on_scene_save\n )\n\n self._op_events[_after_scene_save] = (\n OpenMaya.MSceneMessage.addCallback(\n OpenMaya.MSceneMessage.kAfterSave,\n _after_scene_save\n )\n )\n\n self._op_events[_before_scene_save] = (\n OpenMaya.MSceneMessage.addCheckCallback(\n OpenMaya.MSceneMessage.kBeforeSaveCheck,\n _before_scene_save\n )\n )\n\n self._op_events[_on_scene_new] = OpenMaya.MSceneMessage.addCallback(\n OpenMaya.MSceneMessage.kAfterNew, _on_scene_new\n )\n\n self._op_events[_on_maya_initialized] = (\n OpenMaya.MSceneMessage.addCallback(\n OpenMaya.MSceneMessage.kMayaInitialized,\n _on_maya_initialized\n )\n )\n\n self._op_events[_on_scene_open] = (\n OpenMaya.MSceneMessage.addCallback(\n OpenMaya.MSceneMessage.kAfterOpen,\n _on_scene_open\n )\n )\n\n self._op_events[_before_scene_open] = (\n OpenMaya.MSceneMessage.addCallback(\n OpenMaya.MSceneMessage.kBeforeOpen,\n _before_scene_open\n )\n )\n\n self._op_events[_before_close_maya] = (\n OpenMaya.MSceneMessage.addCallback(\n OpenMaya.MSceneMessage.kMayaExiting,\n _before_close_maya\n )\n )\n\n self.log.info(\"Installed event handler _on_scene_save..\")\n self.log.info(\"Installed event handler _before_scene_save..\")\n self.log.info(\"Installed event handler _on_after_save..\")\n self.log.info(\"Installed event handler _on_scene_new..\")\n self.log.info(\"Installed event handler _on_maya_initialized..\")\n self.log.info(\"Installed event handler _on_scene_open..\")\n self.log.info(\"Installed event handler _check_lock_file..\")\n self.log.info(\"Installed event handler _before_close_maya..\")\n\n\ndef _set_project():\n \"\"\"Sets the maya project to the current Session's work directory.\n\n Returns:\n None\n\n \"\"\"\n workdir = legacy_io.Session[\"AVALON_WORKDIR\"]\n\n try:\n os.makedirs(workdir)\n except OSError as e:\n # An already existing working directory is fine.\n if e.errno == errno.EEXIST:\n pass\n else:\n raise\n\n cmds.workspace(workdir, openWorkspace=True)\n\n\ndef _on_maya_initialized(*args):\n emit_event(\"init\")\n\n if cmds.about(batch=True):\n log.warning(\"Running batch mode ...\")\n return\n\n # Keep reference to the main Window, once a main window exists.\n lib.get_main_window()\n\n\ndef _on_scene_new(*args):\n emit_event(\"new\")\n\n\ndef _after_scene_save(*arg):\n emit_event(\"after.save\")\n\n\ndef _on_scene_save(*args):\n emit_event(\"save\")\n\n\ndef _on_scene_open(*args):\n emit_event(\"open\")\n\n\ndef _before_close_maya(*args):\n emit_event(\"before.close\")\n\n\ndef _before_scene_open(*args):\n emit_event(\"before.file.open\")\n\n\ndef _before_scene_save(return_code, client_data):\n\n # Default to allowing the action. Registered\n # callbacks can optionally set this to False\n # in order to block the operation.\n OpenMaya.MScriptUtil.setBool(return_code, True)\n\n emit_event(\n \"before.save\",\n {\"return_code\": return_code}\n )\n\n\ndef _remove_workfile_lock():\n \"\"\"Remove workfile lock on current file\"\"\"\n if not handle_workfile_locks():\n return\n filepath = current_file()\n log.info(\"Removing lock on current file {}...\".format(filepath))\n if filepath:\n remove_workfile_lock(filepath)\n\n\ndef handle_workfile_locks():\n if lib.IS_HEADLESS:\n return False\n project_name = get_current_project_name()\n return is_workfile_lock_enabled(MayaHost.name, project_name)\n\n\ndef uninstall():\n pyblish.api.deregister_plugin_path(PUBLISH_PATH)\n pyblish.api.deregister_host(\"mayabatch\")\n pyblish.api.deregister_host(\"mayapy\")\n pyblish.api.deregister_host(\"maya\")\n\n deregister_loader_plugin_path(LOAD_PATH)\n deregister_creator_plugin_path(CREATE_PATH)\n deregister_inventory_action_path(INVENTORY_PATH)\n\n menu.uninstall()\n\n\ndef parse_container(container):\n \"\"\"Return the container node's full container data.\n\n Args:\n container (str): A container node name.\n\n Returns:\n dict: The container schema data for this container node.\n\n \"\"\"\n data = lib.read(container)\n\n # Backwards compatibility pre-schemas for containers\n data[\"schema\"] = data.get(\"schema\", \"openpype:container-1.0\")\n\n # Append transient data\n data[\"objectName\"] = container\n\n return data\n\n\ndef _ls():\n \"\"\"Yields Avalon container node names.\n\n Used by `ls()` to retrieve the nodes and then query the full container's\n data.\n\n Yields:\n str: Avalon container node name (objectSet)\n\n \"\"\"\n\n def _maya_iterate(iterator):\n \"\"\"Helper to iterate a maya iterator\"\"\"\n while not iterator.isDone():\n yield iterator.thisNode()\n iterator.next()\n\n ids = {AVALON_CONTAINER_ID,\n # Backwards compatibility\n \"pyblish.mindbender.container\"}\n\n # Iterate over all 'set' nodes in the scene to detect whether\n # they have the avalon container \".id\" attribute.\n fn_dep = om.MFnDependencyNode()\n iterator = om.MItDependencyNodes(om.MFn.kSet)\n for mobject in _maya_iterate(iterator):\n if mobject.apiTypeStr != \"kSet\":\n # Only match by exact type\n continue\n\n fn_dep.setObject(mobject)\n if not fn_dep.hasAttribute(\"id\"):\n continue\n\n plug = fn_dep.findPlug(\"id\", True)\n value = plug.asString()\n if value in ids:\n yield fn_dep.name()\n\n\ndef ls():\n \"\"\"Yields containers from active Maya scene\n\n This is the host-equivalent of api.ls(), but instead of listing\n assets on disk, it lists assets already loaded in Maya; once loaded\n they are called 'containers'\n\n Yields:\n dict: container\n\n \"\"\"\n container_names = _ls()\n for container in sorted(container_names):\n yield parse_container(container)\n\n\ndef containerise(name,\n namespace,\n nodes,\n context,\n loader=None,\n suffix=\"CON\"):\n \"\"\"Bundle `nodes` into an assembly and imprint it with metadata\n\n Containerisation enables a tracking of version, author and origin\n for loaded assets.\n\n Arguments:\n name (str): Name of resulting assembly\n namespace (str): Namespace under which to host container\n nodes (list): Long names of nodes to containerise\n context (dict): Asset information\n loader (str, optional): Name of loader used to produce this container.\n suffix (str, optional): Suffix of container, defaults to `_CON`.\n\n Returns:\n container (str): Name of container assembly\n\n \"\"\"\n container = cmds.sets(nodes, name=\"%s_%s_%s\" % (namespace, name, suffix))\n\n data = [\n (\"schema\", \"openpype:container-2.0\"),\n (\"id\", AVALON_CONTAINER_ID),\n (\"name\", name),\n (\"namespace\", namespace),\n (\"loader\", loader),\n (\"representation\", context[\"representation\"][\"_id\"]),\n ]\n\n for key, value in data:\n cmds.addAttr(container, longName=key, dataType=\"string\")\n cmds.setAttr(container + \".\" + key, str(value), type=\"string\")\n\n main_container = cmds.ls(AVALON_CONTAINERS, type=\"objectSet\")\n if not main_container:\n main_container = cmds.sets(empty=True, name=AVALON_CONTAINERS)\n\n # Implement #399: Maya 2019+ hide AVALON_CONTAINERS on creation..\n if cmds.attributeQuery(\"hiddenInOutliner\",\n node=main_container,\n exists=True):\n cmds.setAttr(main_container + \".hiddenInOutliner\", True)\n else:\n main_container = main_container[0]\n\n cmds.sets(container, addElement=main_container)\n\n # Implement #399: Maya 2019+ hide containers in outliner\n if cmds.attributeQuery(\"hiddenInOutliner\",\n node=container,\n exists=True):\n cmds.setAttr(container + \".hiddenInOutliner\", True)\n\n return container\n\n\ndef on_init():\n log.info(\"Running callback on init..\")\n\n def safe_deferred(fn):\n \"\"\"Execute deferred the function in a try-except\"\"\"\n\n def _fn():\n \"\"\"safely call in deferred callback\"\"\"\n try:\n fn()\n except Exception as exc:\n print(exc)\n\n try:\n utils.executeDeferred(_fn)\n except Exception as exc:\n print(exc)\n\n # Force load Alembic so referenced alembics\n # work correctly on scene open\n cmds.loadPlugin(\"AbcImport\", quiet=True)\n cmds.loadPlugin(\"AbcExport\", quiet=True)\n\n # Force load objExport plug-in (requested by artists)\n cmds.loadPlugin(\"objExport\", quiet=True)\n\n if not lib.IS_HEADLESS:\n launch_workfiles = os.environ.get(\"WORKFILES_STARTUP\")\n if launch_workfiles:\n safe_deferred(host_tools.show_workfiles)\n\n from .customize import (\n override_component_mask_commands,\n override_toolbox_ui\n )\n safe_deferred(override_component_mask_commands)\n safe_deferred(override_toolbox_ui)\n\n\ndef on_before_save():\n \"\"\"Run validation for scene's FPS prior to saving\"\"\"\n return lib.validate_fps()\n\n\ndef on_after_save():\n \"\"\"Check if there is a lockfile after save\"\"\"\n check_lock_on_current_file()\n\n\ndef check_lock_on_current_file():\n\n \"\"\"Check if there is a user opening the file\"\"\"\n if not handle_workfile_locks():\n return\n log.info(\"Running callback on checking the lock file...\")\n\n # add the lock file when opening the file\n filepath = current_file()\n # Skip if current file is 'untitled'\n if not filepath:\n return\n\n if is_workfile_locked(filepath):\n # add lockfile dialog\n workfile_dialog = WorkfileLockDialog(filepath)\n if not workfile_dialog.exec_():\n cmds.file(new=True)\n return\n\n create_workfile_lock(filepath)\n\n\ndef on_before_close():\n \"\"\"Delete the lock file after user quitting the Maya Scene\"\"\"\n log.info(\"Closing Maya...\")\n # delete the lock file\n filepath = current_file()\n if handle_workfile_locks():\n remove_workfile_lock(filepath)\n\n\ndef before_file_open():\n \"\"\"check lock file when the file changed\"\"\"\n # delete the lock file\n _remove_workfile_lock()\n\n\ndef on_save():\n \"\"\"Automatically add IDs to new nodes\n\n Any transform of a mesh, without an existing ID, is given one\n automatically on file save.\n \"\"\"\n log.info(\"Running callback on save..\")\n # remove lockfile if users jumps over from one scene to another\n _remove_workfile_lock()\n\n # Generate ids of the current context on nodes in the scene\n nodes = lib.get_id_required_nodes(referenced_nodes=False)\n for node, new_id in lib.generate_ids(nodes):\n lib.set_id(node, new_id, overwrite=False)\n\n\ndef on_open():\n \"\"\"On scene open let's assume the containers have changed.\"\"\"\n\n from openpype.widgets import popup\n\n # Validate FPS after update_task_from_path to\n # ensure it is using correct FPS for the asset\n lib.validate_fps()\n lib.fix_incompatible_containers()\n\n if any_outdated_containers():\n log.warning(\"Scene has outdated content.\")\n\n # Find maya main window\n parent = lib.get_main_window()\n if parent is None:\n log.info(\"Skipping outdated content pop-up \"\n \"because Maya window can't be found.\")\n else:\n\n # Show outdated pop-up\n def _on_show_inventory():\n host_tools.show_scene_inventory(parent=parent)\n\n dialog = popup.Popup(parent=parent)\n dialog.setWindowTitle(\"Maya scene has outdated content\")\n dialog.setMessage(\"There are outdated containers in \"\n \"your Maya scene.\")\n dialog.on_clicked.connect(_on_show_inventory)\n dialog.show()\n\n # create lock file for the maya scene\n check_lock_on_current_file()\n\n\ndef on_new():\n \"\"\"Set project resolution and fps when create a new file\"\"\"\n log.info(\"Running callback on new..\")\n with lib.suspended_refresh():\n lib.set_context_settings()\n\n _remove_workfile_lock()\n\n\ndef on_task_changed():\n \"\"\"Wrapped function of app initialize and maya's on task changed\"\"\"\n # Run\n menu.update_menu_task_label()\n\n workdir = legacy_io.Session[\"AVALON_WORKDIR\"]\n if os.path.exists(workdir):\n log.info(\"Updating Maya workspace for task change to %s\", workdir)\n _set_project()\n\n # Set Maya fileDialog's start-dir to /scenes\n frule_scene = cmds.workspace(fileRuleEntry=\"scene\")\n cmds.optionVar(stringValue=(\"browserLocationmayaBinaryscene\",\n workdir + \"/\" + frule_scene))\n\n else:\n log.warning((\n \"Can't set project for new context because path does not exist: {}\"\n ).format(workdir))\n\n with lib.suspended_refresh():\n lib.set_context_settings()\n lib.update_content_on_context_change()\n\n\ndef before_workfile_open():\n if handle_workfile_locks():\n _remove_workfile_lock()\n\n\ndef before_workfile_save(event):\n project_name = get_current_project_name()\n if handle_workfile_locks():\n _remove_workfile_lock()\n workdir_path = event[\"workdir_path\"]\n if workdir_path:\n create_workspace_mel(workdir_path, project_name)\n\n\ndef workfile_save_before_xgen(event):\n \"\"\"Manage Xgen external files when switching context.\n\n Xgen has various external files that needs to be unique and relative to the\n workfile, so we need to copy and potentially overwrite these files when\n switching context.\n\n Args:\n event (Event) - openpype/lib/events.py\n \"\"\"\n if not cmds.pluginInfo(\"xgenToolkit\", query=True, loaded=True):\n return\n\n import xgenm\n\n current_work_dir = legacy_io.Session[\"AVALON_WORKDIR\"].replace(\"\\\\\", \"/\")\n expected_work_dir = event.data[\"workdir_path\"].replace(\"\\\\\", \"/\")\n if current_work_dir == expected_work_dir:\n return\n\n palettes = cmds.ls(type=\"xgmPalette\", long=True)\n if not palettes:\n return\n\n transfers = []\n overwrites = []\n attribute_changes = {}\n attrs = [\"xgFileName\", \"xgBaseFile\"]\n for palette in palettes:\n sanitized_palette = palette.replace(\"|\", \"\")\n project_path = xgenm.getAttr(\"xgProjectPath\", sanitized_palette)\n _, maya_extension = os.path.splitext(event.data[\"filename\"])\n\n for attr in attrs:\n node_attr = \"{}.{}\".format(palette, attr)\n attr_value = cmds.getAttr(node_attr)\n\n if not attr_value:\n continue\n\n source = os.path.join(project_path, attr_value)\n\n attr_value = event.data[\"filename\"].replace(\n maya_extension,\n \"__{}{}\".format(\n sanitized_palette.replace(\":\", \"__\"),\n os.path.splitext(attr_value)[1]\n )\n )\n target = os.path.join(expected_work_dir, attr_value)\n\n transfers.append((source, target))\n attribute_changes[node_attr] = attr_value\n\n relative_path = xgenm.getAttr(\n \"xgDataPath\", sanitized_palette\n ).split(os.pathsep)[0]\n absolute_path = relative_path.replace(\"${PROJECT}\", project_path)\n for root, _, files in os.walk(absolute_path):\n for f in files:\n source = os.path.join(root, f).replace(\"\\\\\", \"/\")\n target = source.replace(project_path, expected_work_dir + \"/\")\n transfers.append((source, target))\n if os.path.exists(target):\n overwrites.append(target)\n\n # Ask user about overwriting files.\n if overwrites:\n log.warning(\n \"WARNING! Potential loss of data.\\n\\n\"\n \"Found duplicate Xgen files in new context.\\n{}\".format(\n \"\\n\".join(overwrites)\n )\n )\n return\n\n for source, destination in transfers:\n if not os.path.exists(os.path.dirname(destination)):\n os.makedirs(os.path.dirname(destination))\n shutil.copy(source, destination)\n\n for attribute, value in attribute_changes.items():\n cmds.setAttr(attribute, value, type=\"string\")\n\n\ndef after_workfile_save(event):\n workfile_name = event[\"filename\"]\n if (\n handle_workfile_locks()\n and workfile_name\n and not is_workfile_locked(workfile_name)\n ):\n create_workfile_lock(workfile_name)\n\n\nclass MayaDirmap(HostDirmap):\n def on_enable_dirmap(self):\n cmds.dirmap(en=True)\n\n def dirmap_routine(self, source_path, destination_path):\n cmds.dirmap(m=(source_path, destination_path))\n cmds.dirmap(m=(destination_path, source_path))\n"
},
{
"path": "openpype/hosts/maya/api/plugin.py",
"content": "import json\nimport os\nfrom abc import ABCMeta\n\nimport qargparse\nimport six\nfrom maya import cmds\nfrom maya.app.renderSetup.model import renderSetup\n\nfrom openpype import AYON_SERVER_ENABLED\nfrom openpype.lib import BoolDef, Logger\nfrom openpype.settings import get_project_settings\nfrom openpype.pipeline import (\n AVALON_CONTAINER_ID,\n Anatomy,\n\n CreatedInstance,\n Creator as NewCreator,\n AutoCreator,\n HiddenCreator,\n\n CreatorError,\n LegacyCreator,\n LoaderPlugin,\n get_representation_path,\n)\nfrom openpype.pipeline.load import LoadError\nfrom openpype.client import get_asset_by_name\nfrom openpype.pipeline.create import get_subset_name\n\nfrom . import lib\nfrom .lib import imprint, read\nfrom .pipeline import containerise\n\nlog = Logger.get_logger()\n\n\ndef _get_attr(node, attr, default=None):\n \"\"\"Helper to get attribute which allows attribute to not exist.\"\"\"\n if not cmds.attributeQuery(attr, node=node, exists=True):\n return default\n return cmds.getAttr(\"{}.{}\".format(node, attr))\n\n\n# Backwards compatibility: these functions has been moved to lib.\ndef get_reference_node(*args, **kwargs):\n \"\"\"Get the reference node from the container members\n\n Deprecated:\n This function was moved and will be removed in 3.16.x.\n \"\"\"\n msg = \"Function 'get_reference_node' has been moved.\"\n log.warning(msg)\n cmds.warning(msg)\n return lib.get_reference_node(*args, **kwargs)\n\n\ndef get_reference_node_parents(*args, **kwargs):\n \"\"\"\n Deprecated:\n This function was moved and will be removed in 3.16.x.\n \"\"\"\n msg = \"Function 'get_reference_node_parents' has been moved.\"\n log.warning(msg)\n cmds.warning(msg)\n return lib.get_reference_node_parents(*args, **kwargs)\n\n\nclass Creator(LegacyCreator):\n defaults = ['Main']\n\n def process(self):\n nodes = list()\n\n with lib.undo_chunk():\n if (self.options or {}).get(\"useSelection\"):\n nodes = cmds.ls(selection=True)\n\n instance = cmds.sets(nodes, name=self.name)\n lib.imprint(instance, self.data)\n\n return instance\n\n\n@six.add_metaclass(ABCMeta)\nclass MayaCreatorBase(object):\n\n @staticmethod\n def cache_subsets(shared_data):\n \"\"\"Cache instances for Creators to shared data.\n\n Create `maya_cached_subsets` key when needed in shared data and\n fill it with all collected instances from the scene under its\n respective creator identifiers.\n\n If legacy instances are detected in the scene, create\n `maya_cached_legacy_subsets` there and fill it with\n all legacy subsets under family as a key.\n\n Args:\n Dict[str, Any]: Shared data.\n\n Return:\n Dict[str, Any]: Shared data dictionary.\n\n \"\"\"\n if shared_data.get(\"maya_cached_subsets\") is None:\n cache = dict()\n cache_legacy = dict()\n\n for node in cmds.ls(type=\"objectSet\"):\n\n if _get_attr(node, attr=\"id\") != \"pyblish.avalon.instance\":\n continue\n\n creator_id = _get_attr(node, attr=\"creator_identifier\")\n if creator_id is not None:\n # creator instance\n cache.setdefault(creator_id, []).append(node)\n else:\n # legacy instance\n family = _get_attr(node, attr=\"family\")\n if family is None:\n # must be a broken instance\n continue\n\n cache_legacy.setdefault(family, []).append(node)\n\n shared_data[\"maya_cached_subsets\"] = cache\n shared_data[\"maya_cached_legacy_subsets\"] = cache_legacy\n return shared_data\n\n def get_publish_families(self):\n \"\"\"Return families for the instances of this creator.\n\n Allow a Creator to define multiple families so that a creator can\n e.g. specify `usd` and `usdMaya` and another USD creator can also\n specify `usd` but apply different extractors like `usdMultiverse`.\n\n There is no need to override this method if you only have the\n primary family defined by the `family` property as that will always\n be set.\n\n Returns:\n list: families for instances of this creator\n\n \"\"\"\n return []\n\n def imprint_instance_node(self, node, data):\n\n # We never store the instance_node as value on the node since\n # it's the node name itself\n data.pop(\"instance_node\", None)\n data.pop(\"instance_id\", None)\n\n # Don't store `families` since it's up to the creator itself\n # to define the initial publish families - not a stored attribute of\n # `families`\n data.pop(\"families\", None)\n\n # We store creator attributes at the root level and assume they\n # will not clash in names with `subset`, `task`, etc. and other\n # default names. This is just so these attributes in many cases\n # are still editable in the maya UI by artists.\n # note: pop to move to end of dict to sort attributes last on the node\n creator_attributes = data.pop(\"creator_attributes\", {})\n\n # We only flatten value types which `imprint` function supports\n json_creator_attributes = {}\n for key, value in dict(creator_attributes).items():\n if isinstance(value, (list, tuple, dict)):\n creator_attributes.pop(key)\n json_creator_attributes[key] = value\n\n # Flatten remaining creator attributes to the node itself\n data.update(creator_attributes)\n\n # We know the \"publish_attributes\" will be complex data of\n # settings per plugins, we'll store this as a flattened json structure\n # pop to move to end of dict to sort attributes last on the node\n data[\"publish_attributes\"] = json.dumps(\n data.pop(\"publish_attributes\", {})\n )\n\n # Persist the non-flattened creator attributes (special value types,\n # like multiselection EnumDef)\n data[\"creator_attributes\"] = json.dumps(json_creator_attributes)\n\n # Since we flattened the data structure for creator attributes we want\n # to correctly detect which flattened attributes should end back in the\n # creator attributes when reading the data from the node, so we store\n # the relevant keys as a string\n data[\"__creator_attributes_keys\"] = \",\".join(creator_attributes.keys())\n\n # Kill any existing attributes just so we can imprint cleanly again\n for attr in data.keys():\n if cmds.attributeQuery(attr, node=node, exists=True):\n cmds.deleteAttr(\"{}.{}\".format(node, attr))\n\n return imprint(node, data)\n\n def read_instance_node(self, node):\n node_data = read(node)\n\n # Never care about a cbId attribute on the object set\n # being read as 'data'\n node_data.pop(\"cbId\", None)\n\n # Make sure we convert any creator attributes from the json string\n creator_attributes = node_data.get(\"creator_attributes\")\n if creator_attributes:\n node_data[\"creator_attributes\"] = json.loads(creator_attributes)\n else:\n node_data[\"creator_attributes\"] = {}\n\n # Move the relevant attributes into \"creator_attributes\" that\n # we flattened originally\n creator_attribute_keys = node_data.pop(\"__creator_attributes_keys\",\n \"\").split(\",\")\n for key in creator_attribute_keys:\n if key in node_data:\n node_data[\"creator_attributes\"][key] = node_data.pop(key)\n\n # Make sure we convert any publish attributes from the json string\n publish_attributes = node_data.get(\"publish_attributes\")\n if publish_attributes:\n node_data[\"publish_attributes\"] = json.loads(publish_attributes)\n\n # Explicitly re-parse the node name\n node_data[\"instance_node\"] = node\n node_data[\"instance_id\"] = node\n\n # If the creator plug-in specifies\n families = self.get_publish_families()\n if families:\n node_data[\"families\"] = families\n\n return node_data\n\n def _default_collect_instances(self):\n self.cache_subsets(self.collection_shared_data)\n cached_subsets = self.collection_shared_data[\"maya_cached_subsets\"]\n for node in cached_subsets.get(self.identifier, []):\n node_data = self.read_instance_node(node)\n\n created_instance = CreatedInstance.from_existing(node_data, self)\n self._add_instance_to_context(created_instance)\n\n def _default_update_instances(self, update_list):\n for created_inst, _changes in update_list:\n data = created_inst.data_to_store()\n node = data.get(\"instance_node\")\n\n self.imprint_instance_node(node, data)\n\n def _default_remove_instances(self, instances):\n \"\"\"Remove specified instance from the scene.\n\n This is only removing `id` parameter so instance is no longer\n instance, because it might contain valuable data for artist.\n\n \"\"\"\n for instance in instances:\n node = instance.data.get(\"instance_node\")\n if node:\n cmds.delete(node)\n\n self._remove_instance_from_context(instance)\n\n\n@six.add_metaclass(ABCMeta)\nclass MayaCreator(NewCreator, MayaCreatorBase):\n\n settings_category = \"maya\"\n\n def create(self, subset_name, instance_data, pre_create_data):\n\n members = list()\n if pre_create_data.get(\"use_selection\"):\n members = cmds.ls(selection=True)\n\n # Allow a Creator to define multiple families\n publish_families = self.get_publish_families()\n if publish_families:\n families = instance_data.setdefault(\"families\", [])\n for family in self.get_publish_families():\n if family not in families:\n families.append(family)\n\n with lib.undo_chunk():\n instance_node = cmds.sets(members, name=subset_name)\n instance_data[\"instance_node\"] = instance_node\n instance = CreatedInstance(\n self.family,\n subset_name,\n instance_data,\n self)\n self._add_instance_to_context(instance)\n\n self.imprint_instance_node(instance_node,\n data=instance.data_to_store())\n return instance\n\n def collect_instances(self):\n return self._default_collect_instances()\n\n def update_instances(self, update_list):\n return self._default_update_instances(update_list)\n\n def remove_instances(self, instances):\n return self._default_remove_instances(instances)\n\n def get_pre_create_attr_defs(self):\n return [\n BoolDef(\"use_selection\",\n label=\"Use selection\",\n default=True)\n ]\n\n\nclass MayaAutoCreator(AutoCreator, MayaCreatorBase):\n \"\"\"Automatically triggered creator for Maya.\n\n The plugin is not visible in UI, and 'create' method does not expect\n any arguments.\n \"\"\"\n\n settings_category = \"maya\"\n\n def collect_instances(self):\n return self._default_collect_instances()\n\n def update_instances(self, update_list):\n return self._default_update_instances(update_list)\n\n def remove_instances(self, instances):\n return self._default_remove_instances(instances)\n\n\nclass MayaHiddenCreator(HiddenCreator, MayaCreatorBase):\n \"\"\"Hidden creator for Maya.\n\n The plugin is not visible in UI, and it does not have strictly defined\n arguments for 'create' method.\n \"\"\"\n\n settings_category = \"maya\"\n\n def create(self, *args, **kwargs):\n return MayaCreator.create(self, *args, **kwargs)\n\n def collect_instances(self):\n return self._default_collect_instances()\n\n def update_instances(self, update_list):\n return self._default_update_instances(update_list)\n\n def remove_instances(self, instances):\n return self._default_remove_instances(instances)\n\n\ndef ensure_namespace(namespace):\n \"\"\"Make sure the namespace exists.\n\n Args:\n namespace (str): The preferred namespace name.\n\n Returns:\n str: The generated or existing namespace\n\n \"\"\"\n exists = cmds.namespace(exists=namespace)\n if exists:\n return namespace\n else:\n return cmds.namespace(add=namespace)\n\n\nclass RenderlayerCreator(NewCreator, MayaCreatorBase):\n \"\"\"Creator which creates an instance per renderlayer in the workfile.\n\n Create and manages renderlayer subset per renderLayer in workfile.\n This generates a singleton node in the scene which, if it exists, tells the\n Creator to collect Maya rendersetup renderlayers as individual instances.\n As such, triggering create doesn't actually create the instance node per\n layer but only the node which tells the Creator it may now collect\n an instance per renderlayer.\n\n \"\"\"\n\n # These are required to be overridden in subclass\n singleton_node_name = \"\"\n\n # These are optional to be overridden in subclass\n layer_instance_prefix = None\n\n def _get_singleton_node(self, return_all=False):\n nodes = lib.lsattr(\"pre_creator_identifier\", self.identifier)\n if nodes:\n return nodes if return_all else nodes[0]\n\n def create(self, subset_name, instance_data, pre_create_data):\n # A Renderlayer is never explicitly created using the create method.\n # Instead, renderlayers from the scene are collected. Thus \"create\"\n # would only ever be called to say, 'hey, please refresh collect'\n self.create_singleton_node()\n\n # if no render layers are present, create default one with\n # asterisk selector\n rs = renderSetup.instance()\n if not rs.getRenderLayers():\n render_layer = rs.createRenderLayer(\"Main\")\n collection = render_layer.createCollection(\"defaultCollection\")\n collection.getSelector().setPattern('*')\n\n # By RenderLayerCreator.create we make it so that the renderlayer\n # instances directly appear even though it just collects scene\n # renderlayers. This doesn't actually 'create' any scene contents.\n self.collect_instances()\n\n def create_singleton_node(self):\n if self._get_singleton_node():\n raise CreatorError(\"A Render instance already exists - only \"\n \"one can be configured.\")\n\n with lib.undo_chunk():\n node = cmds.sets(empty=True, name=self.singleton_node_name)\n lib.imprint(node, data={\n \"pre_creator_identifier\": self.identifier\n })\n\n return node\n\n def collect_instances(self):\n\n # We only collect if the global render instance exists\n if not self._get_singleton_node():\n return\n\n rs = renderSetup.instance()\n layers = rs.getRenderLayers()\n for layer in layers:\n layer_instance_node = self.find_layer_instance_node(layer)\n if layer_instance_node:\n data = self.read_instance_node(layer_instance_node)\n instance = CreatedInstance.from_existing(data, creator=self)\n else:\n # No existing scene instance node for this layer. Note that\n # this instance will not have the `instance_node` data yet\n # until it's been saved/persisted at least once.\n project_name = self.create_context.get_current_project_name()\n asset_name = self.create_context.get_current_asset_name()\n instance_data = {\n \"task\": self.create_context.get_current_task_name(),\n \"variant\": layer.name(),\n }\n if AYON_SERVER_ENABLED:\n instance_data[\"folderPath\"] = asset_name\n else:\n instance_data[\"asset\"] = asset_name\n asset_doc = get_asset_by_name(project_name, asset_name)\n subset_name = self.get_subset_name(\n layer.name(),\n instance_data[\"task\"],\n asset_doc,\n project_name)\n\n instance = CreatedInstance(\n family=self.family,\n subset_name=subset_name,\n data=instance_data,\n creator=self\n )\n\n instance.transient_data[\"layer\"] = layer\n self._add_instance_to_context(instance)\n\n def find_layer_instance_node(self, layer):\n connected_sets = cmds.listConnections(\n \"{}.message\".format(layer.name()),\n source=False,\n destination=True,\n type=\"objectSet\"\n ) or []\n\n for node in connected_sets:\n if not cmds.attributeQuery(\"creator_identifier\",\n node=node,\n exists=True):\n continue\n\n creator_identifier = cmds.getAttr(node + \".creator_identifier\")\n if creator_identifier == self.identifier:\n self.log.info(\"Found node: {}\".format(node))\n return node\n\n def _create_layer_instance_node(self, layer):\n\n # We only collect if a CreateRender instance exists\n create_render_set = self._get_singleton_node()\n if not create_render_set:\n raise CreatorError(\"Creating a renderlayer instance node is not \"\n \"allowed if no 'CreateRender' instance exists\")\n\n namespace = \"_{}\".format(self.singleton_node_name)\n namespace = ensure_namespace(namespace)\n\n name = \"{}:{}\".format(namespace, layer.name())\n render_set = cmds.sets(name=name, empty=True)\n\n # Keep an active link with the renderlayer so we can retrieve it\n # later by a physical maya connection instead of relying on the layer\n # name\n cmds.addAttr(render_set, longName=\"renderlayer\", at=\"message\")\n cmds.connectAttr(\"{}.message\".format(layer.name()),\n \"{}.renderlayer\".format(render_set), force=True)\n\n # Add the set to the 'CreateRender' set.\n cmds.sets(render_set, forceElement=create_render_set)\n\n return render_set\n\n def update_instances(self, update_list):\n # We only generate the persisting layer data into the scene once\n # we save with the UI on e.g. validate or publish\n for instance, _changes in update_list:\n instance_node = instance.data.get(\"instance_node\")\n\n # Ensure a node exists to persist the data to\n if not instance_node:\n layer = instance.transient_data[\"layer\"]\n instance_node = self._create_layer_instance_node(layer)\n instance.data[\"instance_node\"] = instance_node\n\n self.imprint_instance_node(instance_node,\n data=instance.data_to_store())\n\n def imprint_instance_node(self, node, data):\n # Do not ever try to update the `renderlayer` since it'll try\n # to remove the attribute and recreate it but fail to keep it a\n # message attribute link. We only ever imprint that on the initial\n # node creation.\n # TODO: Improve how this is handled\n data.pop(\"renderlayer\", None)\n data.get(\"creator_attributes\", {}).pop(\"renderlayer\", None)\n\n return super(RenderlayerCreator, self).imprint_instance_node(node,\n data=data)\n\n def remove_instances(self, instances):\n \"\"\"Remove specified instances from the scene.\n\n This is only removing `id` parameter so instance is no longer\n instance, because it might contain valuable data for artist.\n\n \"\"\"\n # Instead of removing the single instance or renderlayers we instead\n # remove the CreateRender node this creator relies on to decide whether\n # it should collect anything at all.\n nodes = self._get_singleton_node(return_all=True)\n if nodes:\n cmds.delete(nodes)\n\n # Remove ALL the instances even if only one gets deleted\n for instance in list(self.create_context.instances):\n if instance.get(\"creator_identifier\") == self.identifier:\n self._remove_instance_from_context(instance)\n\n # Remove the stored settings per renderlayer too\n node = instance.data.get(\"instance_node\")\n if node and cmds.objExists(node):\n cmds.delete(node)\n\n def get_subset_name(\n self,\n variant,\n task_name,\n asset_doc,\n project_name,\n host_name=None,\n instance=None\n ):\n # creator.family != 'render' as expected\n return get_subset_name(self.layer_instance_prefix,\n variant,\n task_name,\n asset_doc,\n project_name)\n\n\nclass Loader(LoaderPlugin):\n hosts = [\"maya\"]\n\n load_settings = {} # defined in settings\n\n @classmethod\n def apply_settings(cls, project_settings, system_settings):\n super(Loader, cls).apply_settings(project_settings, system_settings)\n cls.load_settings = project_settings['maya']['load']\n\n def get_custom_namespace_and_group(self, context, options, loader_key):\n \"\"\"Queries Settings to get custom template for namespace and group.\n\n Group template might be empty >> this forces to not wrap imported items\n into separate group.\n\n Args:\n context (dict)\n options (dict): artist modifiable options from dialog\n loader_key (str): key to get separate configuration from Settings\n ('reference_loader'|'import_loader')\n \"\"\"\n\n options[\"attach_to_root\"] = True\n try:\n custom_naming = self.load_settings[loader_key]\n except KeyError:\n self.log.warning(\n \"No settings found for {} in settings, falling back to \"\n \"ReferenceLoader defaults.\".format(loader_key))\n custom_naming = self.load_settings[\"reference_loader\"]\n\n if not custom_naming['namespace']:\n raise LoadError(\"No namespace specified in \"\n \"Maya ReferenceLoader settings\")\n elif not custom_naming['group_name']:\n self.log.debug(\"No custom group_name, no group will be created.\")\n options[\"attach_to_root\"] = False\n\n asset = context['asset']\n subset = context['subset']\n formatting_data = {\n \"asset_name\": asset['name'],\n \"asset_type\": asset['type'],\n \"folder\": {\n \"name\": asset[\"name\"],\n },\n \"subset\": subset['name'],\n \"family\": (\n subset['data'].get('family') or\n subset['data']['families'][0]\n )\n }\n\n custom_namespace = custom_naming['namespace'].format(\n **formatting_data\n )\n\n custom_group_name = custom_naming['group_name'].format(\n **formatting_data\n )\n\n return custom_group_name, custom_namespace, options\n\n\nclass ReferenceLoader(Loader):\n \"\"\"A basic ReferenceLoader for Maya\n\n This will implement the basic behavior for a loader to inherit from that\n will containerize the reference and will implement the `remove` and\n `update` logic.\n\n \"\"\"\n\n options = [\n qargparse.Integer(\n \"count\",\n label=\"Count\",\n default=1,\n min=1,\n help=\"How many times to load?\"\n ),\n qargparse.Double3(\n \"offset\",\n label=\"Position Offset\",\n help=\"Offset loaded models for easier selection.\"\n ),\n qargparse.Boolean(\n \"attach_to_root\",\n label=\"Group imported asset\",\n default=True,\n help=\"Should a group be created to encapsulate\"\n \" imported representation ?\"\n )\n ]\n\n def load(\n self,\n context,\n name=None,\n namespace=None,\n options=None\n ):\n path = self.filepath_from_context(context)\n assert os.path.exists(path), \"%s does not exist.\" % path\n\n custom_group_name, custom_namespace, options = \\\n self.get_custom_namespace_and_group(context, options,\n \"reference_loader\")\n\n count = options.get(\"count\") or 1\n\n loaded_containers = []\n for c in range(0, count):\n namespace = lib.get_custom_namespace(custom_namespace)\n group_name = \"{}:{}\".format(\n namespace,\n custom_group_name\n )\n\n options['group_name'] = group_name\n\n # Offset loaded subset\n if \"offset\" in options:\n offset = [i * c for i in options[\"offset\"]]\n options[\"translate\"] = offset\n\n self.log.info(options)\n\n self.process_reference(\n context=context,\n name=name,\n namespace=namespace,\n options=options\n )\n\n # Only containerize if any nodes were loaded by the Loader\n nodes = self[:]\n if not nodes:\n return\n\n ref_node = lib.get_reference_node(nodes, self.log)\n container = containerise(\n name=name,\n namespace=namespace,\n nodes=[ref_node],\n context=context,\n loader=self.__class__.__name__\n )\n loaded_containers.append(container)\n self._organize_containers(nodes, container)\n c += 1\n\n return loaded_containers\n\n def process_reference(self, context, name, namespace, options):\n \"\"\"To be implemented by subclass\"\"\"\n raise NotImplementedError(\"Must be implemented by subclass\")\n\n def update(self, container, representation):\n from maya import cmds\n\n from openpype.hosts.maya.api.lib import get_container_members\n\n node = container[\"objectName\"]\n\n path = get_representation_path(representation)\n\n # Get reference node from container members\n members = get_container_members(node)\n reference_node = lib.get_reference_node(members, self.log)\n namespace = cmds.referenceQuery(reference_node, namespace=True)\n\n file_type = {\n \"ma\": \"mayaAscii\",\n \"mb\": \"mayaBinary\",\n \"abc\": \"Alembic\",\n \"fbx\": \"FBX\",\n \"usd\": \"USD Import\"\n }.get(representation[\"name\"])\n\n assert file_type, \"Unsupported representation: %s\" % representation\n\n assert os.path.exists(path), \"%s does not exist.\" % path\n\n # Need to save alembic settings and reapply, cause referencing resets\n # them to incoming data.\n alembic_attrs = [\"speed\", \"offset\", \"cycleType\", \"time\"]\n alembic_data = {}\n if representation[\"name\"] == \"abc\":\n alembic_nodes = cmds.ls(\n \"{}:*\".format(namespace), type=\"AlembicNode\"\n )\n if alembic_nodes:\n for attr in alembic_attrs:\n node_attr = \"{}.{}\".format(alembic_nodes[0], attr)\n data = {\n \"input\": lib.get_attribute_input(node_attr),\n \"value\": cmds.getAttr(node_attr)\n }\n\n alembic_data[attr] = data\n else:\n self.log.debug(\"No alembic nodes found in {}\".format(members))\n\n try:\n path = self.prepare_root_value(path,\n representation[\"context\"]\n [\"project\"]\n [\"name\"])\n content = cmds.file(path,\n loadReference=reference_node,\n type=file_type,\n returnNewNodes=True)\n except RuntimeError as exc:\n # When changing a reference to a file that has load errors the\n # command will raise an error even if the file is still loaded\n # correctly (e.g. when raising errors on Arnold attributes)\n # When the file is loaded and has content, we consider it's fine.\n if not cmds.referenceQuery(reference_node, isLoaded=True):\n raise\n\n content = cmds.referenceQuery(reference_node,\n nodes=True,\n dagPath=True)\n if not content:\n raise\n\n self.log.warning(\"Ignoring file read error:\\n%s\", exc)\n\n self._organize_containers(content, container[\"objectName\"])\n\n # Reapply alembic settings.\n if representation[\"name\"] == \"abc\" and alembic_data:\n alembic_nodes = cmds.ls(\n \"{}:*\".format(namespace), type=\"AlembicNode\"\n )\n if alembic_nodes:\n alembic_node = alembic_nodes[0] # assume single AlembicNode\n for attr, data in alembic_data.items():\n node_attr = \"{}.{}\".format(alembic_node, attr)\n input = lib.get_attribute_input(node_attr)\n if data[\"input\"]:\n if data[\"input\"] != input:\n cmds.connectAttr(\n data[\"input\"], node_attr, force=True\n )\n else:\n if input:\n cmds.disconnectAttr(input, node_attr)\n cmds.setAttr(node_attr, data[\"value\"])\n\n # Fix PLN-40 for older containers created with Avalon that had the\n # `.verticesOnlySet` set to True.\n if cmds.getAttr(\"{}.verticesOnlySet\".format(node)):\n self.log.info(\"Setting %s.verticesOnlySet to False\", node)\n cmds.setAttr(\"{}.verticesOnlySet\".format(node), False)\n\n # Remove any placeHolderList attribute entries from the set that\n # are remaining from nodes being removed from the referenced file.\n members = cmds.sets(node, query=True)\n invalid = [x for x in members if \".placeHolderList\" in x]\n if invalid:\n cmds.sets(invalid, remove=node)\n\n # Update metadata\n cmds.setAttr(\"{}.representation\".format(node),\n str(representation[\"_id\"]),\n type=\"string\")\n\n # When an animation or pointcache gets connected to an Xgen container,\n # the compound attribute \"xgenContainers\" gets created. When animation\n # containers gets updated we also need to update the cacheFileName on\n # the Xgen collection.\n compound_name = \"xgenContainers\"\n if cmds.objExists(\"{}.{}\".format(node, compound_name)):\n import xgenm\n container_amount = cmds.getAttr(\n \"{}.{}\".format(node, compound_name), size=True\n )\n # loop through all compound children\n for i in range(container_amount):\n attr = \"{}.{}[{}].container\".format(node, compound_name, i)\n objectset = cmds.listConnections(attr)[0]\n reference_node = cmds.sets(objectset, query=True)[0]\n palettes = cmds.ls(\n cmds.referenceQuery(reference_node, nodes=True),\n type=\"xgmPalette\"\n )\n for palette in palettes:\n for description in xgenm.descriptions(palette):\n xgenm.setAttr(\n \"cacheFileName\",\n path.replace(\"\\\\\", \"/\"),\n palette,\n description,\n \"SplinePrimitive\"\n )\n\n # Refresh UI and viewport.\n de = xgenm.xgGlobal.DescriptionEditor\n de.refresh(\"Full\")\n\n def remove(self, container):\n \"\"\"Remove an existing `container` from Maya scene\n\n Deprecated; this functionality is replaced by `api.remove()`\n\n Arguments:\n container (openpype:container-1.0): Which container\n to remove from scene.\n\n \"\"\"\n from maya import cmds\n\n node = container[\"objectName\"]\n\n # Assume asset has been referenced\n members = cmds.sets(node, query=True)\n reference_node = lib.get_reference_node(members, self.log)\n\n assert reference_node, (\"Imported container not supported; \"\n \"container must be referenced.\")\n\n self.log.info(\"Removing '%s' from Maya..\" % container[\"name\"])\n\n namespace = cmds.referenceQuery(reference_node, namespace=True)\n fname = cmds.referenceQuery(reference_node, filename=True)\n cmds.file(fname, removeReference=True)\n\n try:\n cmds.delete(node)\n except ValueError:\n # Already implicitly deleted by Maya upon removing reference\n pass\n\n try:\n # If container is not automatically cleaned up by May (issue #118)\n cmds.namespace(removeNamespace=namespace,\n deleteNamespaceContent=True)\n except RuntimeError:\n pass\n\n def prepare_root_value(self, file_url, project_name):\n \"\"\"Replace root value with env var placeholder.\n\n Use ${OPENPYPE_ROOT_WORK} (or any other root) instead of proper root\n value when storing referenced url into a workfile.\n Useful for remote workflows with SiteSync.\n\n Args:\n file_url (str)\n project_name (dict)\n Returns:\n (str)\n \"\"\"\n settings = get_project_settings(project_name)\n use_env_var_as_root = (settings[\"maya\"]\n [\"maya-dirmap\"]\n [\"use_env_var_as_root\"])\n if use_env_var_as_root:\n anatomy = Anatomy(project_name)\n file_url = anatomy.replace_root_with_env_key(file_url, '${{{}}}')\n\n return file_url\n\n @staticmethod\n def _organize_containers(nodes, container):\n # type: (list, str) -> None\n \"\"\"Put containers in loaded data to correct hierarchy.\"\"\"\n for node in nodes:\n id_attr = \"{}.id\".format(node)\n if not cmds.attributeQuery(\"id\", node=node, exists=True):\n continue\n if cmds.getAttr(id_attr) == AVALON_CONTAINER_ID:\n cmds.sets(node, forceElement=container)\n"
},
{
"path": "openpype/hosts/maya/api/render_setup_tools.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Export stuff in render setup layer context.\n\nExport Maya nodes from Render Setup layer as if flattened in that layer instead\nof exporting the defaultRenderLayer as Maya forces by default\n\nCredits: Roy Nieterau (BigRoy) / Colorbleed\nModified for use in OpenPype\n\n\"\"\"\n\nimport os\nimport contextlib\n\nfrom maya import cmds\nfrom maya.app.renderSetup.model import renderSetup\n\nfrom .lib import pairwise\n\n\n@contextlib.contextmanager\ndef _allow_export_from_render_setup_layer():\n \"\"\"Context manager to override Maya settings to allow RS layer export\"\"\"\n try:\n\n rs = renderSetup.instance()\n\n # Exclude Render Setup nodes from the export\n rs._setAllRSNodesDoNotWrite(True)\n\n # Disable Render Setup forcing the switch to master layer\n os.environ[\"MAYA_BATCH_RENDER_EXPORT\"] = \"1\"\n\n yield\n\n finally:\n # Reset original state\n rs._setAllRSNodesDoNotWrite(False)\n os.environ.pop(\"MAYA_BATCH_RENDER_EXPORT\", None)\n\n\ndef export_in_rs_layer(path, nodes, export=None):\n \"\"\"Export nodes from Render Setup layer.\n\n When exporting from Render Setup layer Maya by default\n forces a switch to the defaultRenderLayer as such making\n it impossible to export the contents of a Render Setup\n layer. Maya presents this warning message:\n # Warning: Exporting Render Setup master layer content #\n\n This function however avoids the renderlayer switch and\n exports from the Render Setup layer as if the edits were\n 'flattened' in the master layer.\n\n It does so by:\n - Allowing export from Render Setup Layer\n - Enforce Render Setup nodes to NOT be written on export\n - Disconnect connections from any `applyOverride` nodes\n to flatten the values (so they are written correctly)*\n *Connection overrides like Shader Override and Material\n Overrides export correctly out of the box since they don't\n create an intermediate connection to an 'applyOverride' node.\n However, any scalar override (absolute or relative override)\n will get input connections in the layer so we'll break those\n to 'store' the values on the attribute itself and write value\n out instead.\n\n Args:\n path (str): File path to export to.\n nodes (list): Maya nodes to export.\n export (callable, optional): Callback to be used for exporting. If\n not specified, default export to `.ma` will be called.\n\n Returns:\n None\n\n Raises:\n AssertionError: When not in a Render Setup layer an\n AssertionError is raised. This command assumes\n you are currently in a Render Setup layer.\n\n \"\"\"\n rs = renderSetup.instance()\n assert rs.getVisibleRenderLayer().name() != \"defaultRenderLayer\", \\\n (\"Export in Render Setup layer is only supported when in \"\n \"Render Setup layer\")\n\n # Break connection to any value overrides\n history = cmds.listHistory(nodes) or []\n nodes_all = list(\n set(cmds.ls(nodes + history, long=True, objectsOnly=True)))\n overrides = cmds.listConnections(nodes_all,\n source=True,\n destination=False,\n type=\"applyOverride\",\n plugs=True,\n connections=True) or []\n for dest, src in pairwise(overrides):\n # Even after disconnecting the values\n # should be preserved as they were\n # Note: animated overrides would be lost for export\n cmds.disconnectAttr(src, dest)\n\n # Export Selected\n with _allow_export_from_render_setup_layer():\n cmds.select(nodes, noExpand=True)\n if export:\n export()\n else:\n cmds.file(path,\n force=True,\n typ=\"mayaAscii\",\n exportSelected=True,\n preserveReferences=False,\n channels=True,\n constraints=True,\n expressions=True,\n constructionHistory=True)\n\n if overrides:\n # If we have broken override connections then Maya\n # is unaware that the Render Setup layer is in an\n # invalid state. So let's 'hard reset' the state\n # by going to default render layer and switching back\n layer = rs.getVisibleRenderLayer()\n rs.switchToLayer(None)\n rs.switchToLayer(layer)\n"
},
{
"path": "openpype/hosts/maya/api/setdress.py",
"content": "import logging\nimport json\nimport os\n\nimport contextlib\nimport copy\n\nimport six\n\nfrom maya import cmds\n\nfrom openpype.client import (\n get_version_by_name,\n get_last_version_by_subset_id,\n get_representation_by_id,\n get_representation_by_name,\n get_representation_parents,\n)\nfrom openpype.pipeline import (\n schema,\n discover_loader_plugins,\n loaders_from_representation,\n load_container,\n update_container,\n remove_container,\n get_representation_path,\n get_current_project_name,\n)\nfrom openpype.hosts.maya.api.lib import (\n matrix_equals,\n unique_namespace,\n get_container_transforms,\n DEFAULT_MATRIX\n)\n\nlog = logging.getLogger(\"PackageLoader\")\n\n\ndef to_namespace(node, namespace):\n \"\"\"Return node name as if it's inside the namespace.\n\n Args:\n node (str): Node name\n namespace (str): Namespace\n\n Returns:\n str: The node in the namespace.\n\n \"\"\"\n namespace_prefix = \"|{}:\".format(namespace)\n node = namespace_prefix.join(node.split(\"|\"))\n return node\n\n\n@contextlib.contextmanager\ndef namespaced(namespace, new=True):\n \"\"\"Work inside namespace during context\n\n Args:\n new (bool): When enabled this will rename the namespace to a unique\n namespace if the input namespace already exists.\n\n Yields:\n str: The namespace that is used during the context\n\n \"\"\"\n original = cmds.namespaceInfo(cur=True)\n if new:\n namespace = unique_namespace(namespace)\n cmds.namespace(add=namespace)\n\n try:\n cmds.namespace(set=namespace)\n yield namespace\n finally:\n cmds.namespace(set=original)\n\n\n@contextlib.contextmanager\ndef unlocked(nodes):\n\n # Get node state by Maya's uuid\n nodes = cmds.ls(nodes, long=True)\n uuids = cmds.ls(nodes, uuid=True)\n states = cmds.lockNode(nodes, query=True, lock=True)\n states = {uuid: state for uuid, state in zip(uuids, states)}\n originals = {uuid: node for uuid, node in zip(uuids, nodes)}\n\n try:\n cmds.lockNode(nodes, lock=False)\n yield\n finally:\n # Reapply original states\n _iteritems = getattr(states, \"iteritems\", states.items)\n for uuid, state in _iteritems():\n nodes_from_id = cmds.ls(uuid, long=True)\n if nodes_from_id:\n node = nodes_from_id[0]\n else:\n log.debug(\"Falling back to node name: %s\", node)\n node = originals[uuid]\n if not cmds.objExists(node):\n log.warning(\"Unable to find: %s\", node)\n continue\n cmds.lockNode(node, lock=state)\n\n\ndef load_package(filepath, name, namespace=None):\n \"\"\"Load a package that was gathered elsewhere.\n\n A package is a group of published instances, possibly with additional data\n in a hierarchy.\n\n \"\"\"\n\n if namespace is None:\n # Define a unique namespace for the package\n namespace = os.path.basename(filepath).split(\".\")[0]\n unique_namespace(namespace)\n assert isinstance(namespace, six.string_types)\n\n # Load the setdress package data\n with open(filepath, \"r\") as fp:\n data = json.load(fp)\n\n # Load the setdress alembic hierarchy\n # We import this into the namespace in which we'll load the package's\n # instances into afterwards.\n alembic = filepath.replace(\".json\", \".abc\")\n hierarchy = cmds.file(alembic,\n reference=True,\n namespace=namespace,\n returnNewNodes=True,\n groupReference=True,\n groupName=\"{}:{}\".format(namespace, name),\n typ=\"Alembic\")\n\n # Get the top root node (the reference group)\n root = \"{}:{}\".format(namespace, name)\n\n containers = []\n all_loaders = discover_loader_plugins()\n for representation_id, instances in data.items():\n\n # Find the compatible loaders\n loaders = loaders_from_representation(\n all_loaders, representation_id\n )\n\n for instance in instances:\n container = _add(instance=instance,\n representation_id=representation_id,\n loaders=loaders,\n namespace=namespace,\n root=root)\n containers.append(container)\n\n # TODO: Do we want to cripple? Or do we want to add a 'parent' parameter?\n # Cripple the original avalon containers so they don't show up in the\n # manager\n # for container in containers:\n # cmds.setAttr(\"%s.id\" % container,\n # \"setdress.container\",\n # type=\"string\")\n\n # TODO: Lock all loaded nodes\n # This is to ensure the hierarchy remains unaltered by the artists\n # for node in nodes:\n # cmds.lockNode(node, lock=True)\n\n return containers + hierarchy\n\n\ndef _add(instance, representation_id, loaders, namespace, root=\"|\"):\n \"\"\"Add an item from the package\n\n Args:\n instance (dict):\n representation_id (str):\n loaders (list):\n namespace (str):\n\n Returns:\n str: The created Avalon container.\n\n \"\"\"\n\n # Process within the namespace\n with namespaced(namespace, new=False) as namespace:\n\n # Get the used loader\n Loader = next((x for x in loaders if\n x.__name__ == instance['loader']),\n None)\n\n if Loader is None:\n log.warning(\"Loader is missing: %s. Skipping %s\",\n instance['loader'], instance)\n raise RuntimeError(\"Loader is missing.\")\n\n container = load_container(\n Loader,\n representation_id,\n namespace=instance['namespace']\n )\n\n # Get the root from the loaded container\n loaded_root = get_container_transforms({\"objectName\": container},\n root=True)\n\n # Apply matrix to root node (if any matrix edits)\n matrix = instance.get(\"matrix\", None)\n if matrix:\n cmds.xform(loaded_root, objectSpace=True, matrix=matrix)\n\n # Parent into the setdress hierarchy\n # Namespace is missing from parent node(s), add namespace\n # manually\n parent = root + to_namespace(instance[\"parent\"], namespace)\n cmds.parent(loaded_root, parent, relative=True)\n\n return container\n\n\n# Store root nodes based on representation and namespace\ndef _instances_by_namespace(data):\n \"\"\"Rebuild instance data so we can look it up by namespace.\n\n Note that the `representation` is added into the instance's\n data with a `representation` key.\n\n Args:\n data (dict): scene build data\n\n Returns:\n dict\n\n \"\"\"\n result = {}\n # Add new assets\n for representation_id, instances in data.items():\n\n # Ensure we leave the source data unaltered\n instances = copy.deepcopy(instances)\n for instance in instances:\n instance['representation'] = representation_id\n result[instance['namespace']] = instance\n\n return result\n\n\ndef get_contained_containers(container):\n \"\"\"Get the Avalon containers in this container\n\n Args:\n container (dict): The container dict.\n\n Returns:\n list: A list of member container dictionaries.\n\n \"\"\"\n\n from .pipeline import parse_container\n\n # Get avalon containers in this package setdress container\n containers = []\n members = cmds.sets(container['objectName'], query=True)\n for node in cmds.ls(members, type=\"objectSet\"):\n try:\n member_container = parse_container(node)\n containers.append(member_container)\n except schema.ValidationError:\n pass\n\n return containers\n\n\ndef update_package_version(container, version):\n \"\"\"\n Update package by version number\n\n Args:\n container (dict): container data of the container node\n version (int): the new version number of the package\n\n Returns:\n None\n\n \"\"\"\n\n # Versioning (from `core.maya.pipeline`)\n project_name = get_current_project_name()\n current_representation = get_representation_by_id(\n project_name, container[\"representation\"]\n )\n\n assert current_representation is not None, \"This is a bug\"\n\n version_doc, subset_doc, asset_doc, project_doc = (\n get_representation_parents(project_name, current_representation)\n )\n\n if version == -1:\n new_version = get_last_version_by_subset_id(\n project_name, subset_doc[\"_id\"]\n )\n else:\n new_version = get_version_by_name(\n project_name, version, subset_doc[\"_id\"]\n )\n\n assert new_version is not None, \"This is a bug\"\n\n # Get the new representation (new file)\n new_representation = get_representation_by_name(\n project_name, current_representation[\"name\"], new_version[\"_id\"]\n )\n\n update_package(container, new_representation)\n\n\ndef update_package(set_container, representation):\n \"\"\"Update any matrix changes in the scene based on the new data\n\n Args:\n set_container (dict): container data from `ls()`\n representation (dict): the representation document from the database\n\n Returns:\n None\n\n \"\"\"\n\n # Load the original package data\n project_name = get_current_project_name()\n current_representation = get_representation_by_id(\n project_name, set_container[\"representation\"]\n )\n\n current_file = get_representation_path(current_representation)\n assert current_file.endswith(\".json\")\n with open(current_file, \"r\") as fp:\n current_data = json.load(fp)\n\n # Load the new package data\n new_file = get_representation_path(representation)\n assert new_file.endswith(\".json\")\n with open(new_file, \"r\") as fp:\n new_data = json.load(fp)\n\n # Update scene content\n containers = get_contained_containers(set_container)\n update_scene(set_container, containers, current_data, new_data, new_file)\n\n # TODO: This should be handled by the pipeline itself\n cmds.setAttr(set_container['objectName'] + \".representation\",\n str(representation['_id']), type=\"string\")\n\n\ndef update_scene(set_container, containers, current_data, new_data, new_file):\n \"\"\"Updates the hierarchy, assets and their matrix\n\n Updates the following within the scene:\n * Setdress hierarchy alembic\n * Matrix\n * Parenting\n * Representations\n\n It removes any assets which are not present in the new build data\n\n Args:\n set_container (dict): the setdress container of the scene\n containers (list): the list of containers under the setdress container\n current_data (dict): the current build data of the setdress\n new_data (dict): the new build data of the setdres\n\n Returns:\n processed_containers (list): all new and updated containers\n\n \"\"\"\n\n set_namespace = set_container['namespace']\n project_name = get_current_project_name()\n\n # Update the setdress hierarchy alembic\n set_root = get_container_transforms(set_container, root=True)\n set_hierarchy_root = cmds.listRelatives(set_root, fullPath=True)[0]\n set_hierarchy_reference = cmds.referenceQuery(set_hierarchy_root,\n referenceNode=True)\n new_alembic = new_file.replace(\".json\", \".abc\")\n assert os.path.exists(new_alembic), \"%s does not exist.\" % new_alembic\n with unlocked(cmds.listRelatives(set_root, ad=True, fullPath=True)):\n cmds.file(new_alembic,\n loadReference=set_hierarchy_reference,\n type=\"Alembic\")\n\n identity = DEFAULT_MATRIX[:]\n\n processed_namespaces = set()\n processed_containers = list()\n\n new_lookup = _instances_by_namespace(new_data)\n old_lookup = _instances_by_namespace(current_data)\n for container in containers:\n container_ns = container['namespace']\n\n # Consider it processed here, even it it fails we want to store that\n # the namespace was already available.\n processed_namespaces.add(container_ns)\n processed_containers.append(container['objectName'])\n\n if container_ns in new_lookup:\n root = get_container_transforms(container, root=True)\n if not root:\n log.error(\"Can't find root for %s\", container['objectName'])\n continue\n\n old_instance = old_lookup.get(container_ns, {})\n new_instance = new_lookup[container_ns]\n\n # Update the matrix\n # check matrix against old_data matrix to find local overrides\n current_matrix = cmds.xform(root,\n query=True,\n matrix=True,\n objectSpace=True)\n\n original_matrix = old_instance.get(\"matrix\", identity)\n has_matrix_override = not matrix_equals(current_matrix,\n original_matrix)\n\n if has_matrix_override:\n log.warning(\"Matrix override preserved on %s\", container_ns)\n else:\n new_matrix = new_instance.get(\"matrix\", identity)\n cmds.xform(root, matrix=new_matrix, objectSpace=True)\n\n # Update the parenting\n if old_instance.get(\"parent\", None) != new_instance[\"parent\"]:\n\n parent = to_namespace(new_instance['parent'], set_namespace)\n if not cmds.objExists(parent):\n log.error(\"Can't find parent %s\", parent)\n continue\n\n # Set the new parent\n cmds.lockNode(root, lock=False)\n root = cmds.parent(root, parent, relative=True)\n cmds.lockNode(root, lock=True)\n\n # Update the representation\n representation_current = container['representation']\n representation_old = old_instance['representation']\n representation_new = new_instance['representation']\n has_representation_override = (representation_current !=\n representation_old)\n\n if representation_new != representation_current:\n\n if has_representation_override:\n log.warning(\"Your scene had local representation \"\n \"overrides within the set. New \"\n \"representations not loaded for %s.\",\n container_ns)\n continue\n\n # We check it against the current 'loader' in the scene instead\n # of the original data of the package that was loaded because\n # an Artist might have made scene local overrides\n if new_instance['loader'] != container['loader']:\n log.warning(\"Loader is switched - local edits will be \"\n \"lost. Removing: %s\",\n container_ns)\n\n # Remove this from the \"has been processed\" list so it's\n # considered as new element and added afterwards.\n processed_containers.pop()\n processed_namespaces.remove(container_ns)\n remove_container(container)\n continue\n\n # Check whether the conversion can be done by the Loader.\n # They *must* use the same asset, subset and Loader for\n # `update_container` to make sense.\n old = get_representation_by_id(\n project_name, representation_current\n )\n new = get_representation_by_id(\n project_name, representation_new\n )\n is_valid = compare_representations(old=old, new=new)\n if not is_valid:\n log.error(\"Skipping: %s. See log for details.\",\n container_ns)\n continue\n\n new_version = new[\"context\"][\"version\"]\n update_container(container, version=new_version)\n\n else:\n # Remove this container because it's not in the new data\n log.warning(\"Removing content: %s\", container_ns)\n remove_container(container)\n\n # Add new assets\n all_loaders = discover_loader_plugins()\n for representation_id, instances in new_data.items():\n\n # Find the compatible loaders\n loaders = loaders_from_representation(\n all_loaders, representation_id\n )\n for instance in instances:\n\n # Already processed in update functionality\n if instance['namespace'] in processed_namespaces:\n continue\n\n container = _add(instance=instance,\n representation_id=representation_id,\n loaders=loaders,\n namespace=set_container['namespace'],\n root=set_root)\n\n # Add to the setdress container\n cmds.sets(container,\n addElement=set_container['objectName'])\n\n processed_containers.append(container)\n\n return processed_containers\n\n\ndef compare_representations(old, new):\n \"\"\"Check if the old representation given can be updated\n\n Due to limitations of the `update_container` function we cannot allow\n differences in the following data:\n\n * Representation name (extension)\n * Asset name\n * Subset name (variation)\n\n If any of those data values differs, the function will raise an\n RuntimeError\n\n Args:\n old(dict): representation data from the database\n new(dict): representation data from the database\n\n Returns:\n bool: False if the representation is not invalid else True\n \"\"\"\n\n if new[\"name\"] != old[\"name\"]:\n log.error(\"Cannot switch extensions\")\n return False\n\n new_context = new[\"context\"]\n old_context = old[\"context\"]\n\n if new_context[\"asset\"] != old_context[\"asset\"]:\n log.error(\"Changing assets between updates is \"\n \"not supported.\")\n return False\n\n if new_context[\"subset\"] != old_context[\"subset\"]:\n log.error(\"Changing subsets between updates is \"\n \"not supported.\")\n return False\n\n return True\n"
},
{
"path": "openpype/hosts/maya/api/shader_definition_editor.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Editor for shader definitions.\n\nShader names are stored as simple text file over GridFS in mongodb.\n\n\"\"\"\nimport os\nfrom qtpy import QtWidgets, QtCore, QtGui\nfrom openpype.client.mongo import OpenPypeMongoConnection\nfrom openpype import resources\nimport gridfs\n\n\nDEFINITION_FILENAME = \"{}/maya/shader_definition.txt\".format(\n os.getenv(\"AVALON_PROJECT\"))\n\n\nclass ShaderDefinitionsEditor(QtWidgets.QWidget):\n \"\"\"Widget serving as simple editor for shader name definitions.\"\"\"\n\n # name of the file used to store definitions\n\n def __init__(self, parent=None):\n super(ShaderDefinitionsEditor, self).__init__(parent)\n self._mongo = OpenPypeMongoConnection.get_mongo_client()\n self._gridfs = gridfs.GridFS(\n self._mongo[os.getenv(\"OPENPYPE_DATABASE_NAME\")])\n self._editor = None\n\n self._original_content = self._read_definition_file()\n\n self.setObjectName(\"shaderDefinitionEditor\")\n self.setWindowTitle(\"OpenPype shader name definition editor\")\n icon = QtGui.QIcon(resources.get_openpype_icon_filepath())\n self.setWindowIcon(icon)\n self.setWindowFlags(QtCore.Qt.Window)\n self.setParent(parent)\n self.setAttribute(QtCore.Qt.WA_DeleteOnClose)\n self.resize(750, 500)\n\n self._setup_ui()\n self._reload()\n\n def _setup_ui(self):\n \"\"\"Setup UI of Widget.\"\"\"\n layout = QtWidgets.QVBoxLayout(self)\n label = QtWidgets.QLabel()\n label.setText(\"Put shader names here - one name per line:\")\n layout.addWidget(label)\n self._editor = QtWidgets.QPlainTextEdit()\n self._editor.setStyleSheet(\"border: none;\")\n layout.addWidget(self._editor)\n\n btn_layout = QtWidgets.QHBoxLayout()\n save_btn = QtWidgets.QPushButton(\"Save\")\n save_btn.clicked.connect(self._save)\n\n reload_btn = QtWidgets.QPushButton(\"Reload\")\n reload_btn.clicked.connect(self._reload)\n\n exit_btn = QtWidgets.QPushButton(\"Exit\")\n exit_btn.clicked.connect(self._close)\n\n btn_layout.addWidget(reload_btn)\n btn_layout.addWidget(save_btn)\n btn_layout.addWidget(exit_btn)\n\n layout.addLayout(btn_layout)\n\n def _read_definition_file(self, file=None):\n \"\"\"Read definition file from database.\n\n Args:\n file (gridfs.grid_file.GridOut, Optional): File to read. If not\n set, new query will be issued to find it.\n\n Returns:\n str: Content of the file or empty string if file doesn't exist.\n\n \"\"\"\n content = \"\"\n if not file:\n file = self._gridfs.find_one(\n {\"filename\": DEFINITION_FILENAME})\n if not file:\n print(\">>> [SNDE]: nothing in database yet\")\n return content\n content = file.read()\n file.close()\n return content\n\n def _write_definition_file(self, content, force=False):\n \"\"\"Write content as definition to file in database.\n\n Before file is written, check is made if its content has not\n changed. If is changed, warning is issued to user if he wants\n it to overwrite. Note: GridFs doesn't allow changing file content.\n You need to delete existing file and create new one.\n\n Args:\n content (str): Content to write.\n\n Raises:\n ContentException: If file is changed in database while\n editor is running.\n \"\"\"\n file = self._gridfs.find_one(\n {\"filename\": DEFINITION_FILENAME})\n if file:\n content_check = self._read_definition_file(file)\n if content == content_check:\n print(\">>> [SNDE]: content not changed\")\n return\n if self._original_content != content_check:\n if not force:\n raise ContentException(\"Content changed\")\n print(\">>> [SNDE]: overwriting data\")\n file.close()\n self._gridfs.delete(file._id)\n\n file = self._gridfs.new_file(\n filename=DEFINITION_FILENAME,\n content_type='text/plain',\n encoding='utf-8')\n file.write(content)\n file.close()\n QtCore.QTimer.singleShot(200, self._reset_style)\n self._editor.setStyleSheet(\"border: 1px solid #33AF65;\")\n self._original_content = content\n\n def _reset_style(self):\n \"\"\"Reset editor style back.\n\n Used to visually indicate save.\n\n \"\"\"\n self._editor.setStyleSheet(\"border: none;\")\n\n def _close(self):\n self.hide()\n\n def closeEvent(self, event):\n event.ignore()\n self.hide()\n\n def _reload(self):\n print(\">>> [SNDE]: reloading\")\n self._set_content(self._read_definition_file())\n\n def _save(self):\n try:\n self._write_definition_file(content=self._editor.toPlainText())\n except ContentException:\n # content has changed meanwhile\n print(\">>> [SNDE]: content has changed\")\n self._show_overwrite_warning()\n\n def _set_content(self, content):\n self._editor.setPlainText(content)\n\n def _show_overwrite_warning(self):\n reply = QtWidgets.QMessageBox.question(\n self,\n \"Warning\",\n (\"Content you are editing was changed meanwhile in database.\\n\"\n \"Please, reload and solve the conflict.\"),\n QtWidgets.QMessageBox.OK)\n\n if reply == QtWidgets.QMessageBox.OK:\n # do nothing\n pass\n\n\nclass ContentException(Exception):\n \"\"\"This is risen during save if file is changed in database.\"\"\"\n pass\n"
},
{
"path": "openpype/hosts/maya/api/workfile_template_builder.py",
"content": "import json\n\nfrom maya import cmds\n\nfrom openpype.pipeline import registered_host, get_current_asset_name\nfrom openpype.pipeline.workfile.workfile_template_builder import (\n TemplateAlreadyImported,\n AbstractTemplateBuilder,\n PlaceholderPlugin,\n LoadPlaceholderItem,\n PlaceholderLoadMixin,\n)\nfrom openpype.tools.workfile_template_build import (\n WorkfileBuildPlaceholderDialog,\n)\n\nfrom .lib import read, imprint, get_reference_node, get_main_window\n\nPLACEHOLDER_SET = \"PLACEHOLDERS_SET\"\n\n\nclass MayaTemplateBuilder(AbstractTemplateBuilder):\n \"\"\"Concrete implementation of AbstractTemplateBuilder for maya\"\"\"\n\n use_legacy_creators = True\n\n def import_template(self, path):\n \"\"\"Import template into current scene.\n Block if a template is already loaded.\n\n Args:\n path (str): A path to current template (usually given by\n get_template_preset implementation)\n\n Returns:\n bool: Whether the template was successfully imported or not\n \"\"\"\n\n if cmds.objExists(PLACEHOLDER_SET):\n raise TemplateAlreadyImported((\n \"Build template already loaded\\n\"\n \"Clean scene if needed (File > New Scene)\"\n ))\n\n cmds.sets(name=PLACEHOLDER_SET, empty=True)\n new_nodes = cmds.file(\n path,\n i=True,\n returnNewNodes=True,\n preserveReferences=True,\n loadReferenceDepth=\"all\",\n )\n\n # make default cameras non-renderable\n default_cameras = [cam for cam in cmds.ls(cameras=True)\n if cmds.camera(cam, query=True, startupCamera=True)]\n for cam in default_cameras:\n if not cmds.attributeQuery(\"renderable\", node=cam, exists=True):\n self.log.debug(\n \"Camera {} has no attribute 'renderable'\".format(cam)\n )\n continue\n cmds.setAttr(\"{}.renderable\".format(cam), 0)\n\n cmds.setAttr(PLACEHOLDER_SET + \".hiddenInOutliner\", True)\n\n imported_sets = cmds.ls(new_nodes, set=True)\n if not imported_sets:\n return True\n\n # update imported sets information\n asset_name = get_current_asset_name()\n for node in imported_sets:\n if not cmds.attributeQuery(\"id\", node=node, exists=True):\n continue\n if cmds.getAttr(\"{}.id\".format(node)) != \"pyblish.avalon.instance\":\n continue\n if not cmds.attributeQuery(\"asset\", node=node, exists=True):\n continue\n\n cmds.setAttr(\n \"{}.asset\".format(node), asset_name, type=\"string\")\n\n return True\n\n\nclass MayaPlaceholderLoadPlugin(PlaceholderPlugin, PlaceholderLoadMixin):\n identifier = \"maya.load\"\n label = \"Maya load\"\n\n def _collect_scene_placeholders(self):\n # Cache placeholder data to shared data\n placeholder_nodes = self.builder.get_shared_populate_data(\n \"placeholder_nodes\"\n )\n if placeholder_nodes is None:\n attributes = cmds.ls(\"*.plugin_identifier\", long=True)\n placeholder_nodes = {}\n for attribute in attributes:\n node_name = attribute.rpartition(\".\")[0]\n placeholder_nodes[node_name] = (\n self._parse_placeholder_node_data(node_name)\n )\n\n self.builder.set_shared_populate_data(\n \"placeholder_nodes\", placeholder_nodes\n )\n return placeholder_nodes\n\n def _parse_placeholder_node_data(self, node_name):\n placeholder_data = read(node_name)\n parent_name = (\n cmds.getAttr(node_name + \".parent\", asString=True)\n or node_name.rpartition(\"|\")[0]\n or \"\"\n )\n if parent_name:\n siblings = cmds.listRelatives(parent_name, children=True)\n else:\n siblings = cmds.ls(assemblies=True)\n node_shortname = node_name.rpartition(\"|\")[2]\n current_index = cmds.getAttr(node_name + \".index\", asString=True)\n if current_index < 0:\n current_index = siblings.index(node_shortname)\n\n placeholder_data.update({\n \"parent\": parent_name,\n \"index\": current_index\n })\n return placeholder_data\n\n def _create_placeholder_name(self, placeholder_data):\n placeholder_name_parts = placeholder_data[\"builder_type\"].split(\"_\")\n\n pos = 1\n # add family in any\n placeholder_family = placeholder_data[\"family\"]\n if placeholder_family:\n placeholder_name_parts.insert(pos, placeholder_family)\n pos += 1\n\n # add loader arguments if any\n loader_args = placeholder_data[\"loader_args\"]\n if loader_args:\n loader_args = json.loads(loader_args.replace('\\'', '\\\"'))\n values = [v for v in loader_args.values()]\n for value in values:\n placeholder_name_parts.insert(pos, value)\n pos += 1\n\n placeholder_name = \"_\".join(placeholder_name_parts)\n\n return placeholder_name.capitalize()\n\n def _get_loaded_repre_ids(self):\n loaded_representation_ids = self.builder.get_shared_populate_data(\n \"loaded_representation_ids\"\n )\n if loaded_representation_ids is None:\n try:\n containers = cmds.sets(\"AVALON_CONTAINERS\", q=True)\n except ValueError:\n containers = []\n\n loaded_representation_ids = {\n cmds.getAttr(container + \".representation\")\n for container in containers\n }\n self.builder.set_shared_populate_data(\n \"loaded_representation_ids\", loaded_representation_ids\n )\n return loaded_representation_ids\n\n def create_placeholder(self, placeholder_data):\n selection = cmds.ls(selection=True)\n if len(selection) > 1:\n raise ValueError(\"More then one item are selected\")\n\n parent = selection[0] if selection else None\n\n placeholder_data[\"plugin_identifier\"] = self.identifier\n\n placeholder_name = self._create_placeholder_name(placeholder_data)\n\n placeholder = cmds.spaceLocator(name=placeholder_name)[0]\n if parent:\n placeholder = cmds.parent(placeholder, selection[0])[0]\n\n imprint(placeholder, placeholder_data)\n\n # Add helper attributes to keep placeholder info\n cmds.addAttr(\n placeholder,\n longName=\"parent\",\n hidden=True,\n dataType=\"string\"\n )\n cmds.addAttr(\n placeholder,\n longName=\"index\",\n hidden=True,\n attributeType=\"short\",\n defaultValue=-1\n )\n\n cmds.setAttr(placeholder + \".parent\", \"\", type=\"string\")\n\n def update_placeholder(self, placeholder_item, placeholder_data):\n node_name = placeholder_item.scene_identifier\n new_values = {}\n for key, value in placeholder_data.items():\n placeholder_value = placeholder_item.data.get(key)\n if value != placeholder_value:\n new_values[key] = value\n placeholder_item.data[key] = value\n\n for key in new_values.keys():\n cmds.deleteAttr(node_name + \".\" + key)\n\n imprint(node_name, new_values)\n\n def collect_placeholders(self):\n output = []\n scene_placeholders = self._collect_scene_placeholders()\n for node_name, placeholder_data in scene_placeholders.items():\n if placeholder_data.get(\"plugin_identifier\") != self.identifier:\n continue\n\n # TODO do data validations and maybe upgrades if they are invalid\n output.append(\n LoadPlaceholderItem(node_name, placeholder_data, self)\n )\n\n return output\n\n def populate_placeholder(self, placeholder):\n self.populate_load_placeholder(placeholder)\n\n def repopulate_placeholder(self, placeholder):\n repre_ids = self._get_loaded_repre_ids()\n self.populate_load_placeholder(placeholder, repre_ids)\n\n def get_placeholder_options(self, options=None):\n return self.get_load_plugin_options(options)\n\n def post_placeholder_process(self, placeholder, failed):\n \"\"\"Cleanup placeholder after load of its corresponding representations.\n\n Args:\n placeholder (PlaceholderItem): Item which was just used to load\n representation.\n failed (bool): Loading of representation failed.\n \"\"\"\n # Hide placeholder and add them to placeholder set\n node = placeholder.scene_identifier\n\n cmds.sets(node, addElement=PLACEHOLDER_SET)\n cmds.hide(node)\n cmds.setAttr(node + \".hiddenInOutliner\", True)\n\n def delete_placeholder(self, placeholder):\n \"\"\"Remove placeholder if building was successful\"\"\"\n cmds.delete(placeholder.scene_identifier)\n\n def load_succeed(self, placeholder, container):\n self._parent_in_hierarchy(placeholder, container)\n\n def _parent_in_hierarchy(self, placeholder, container):\n \"\"\"Parent loaded container to placeholder's parent.\n\n ie : Set loaded content as placeholder's sibling\n\n Args:\n container (str): Placeholder loaded containers\n \"\"\"\n\n if not container:\n return\n\n roots = cmds.sets(container, q=True)\n ref_node = None\n try:\n ref_node = get_reference_node(roots)\n except AssertionError as e:\n self.log.info(e.args[0])\n\n nodes_to_parent = []\n for root in roots:\n if ref_node:\n ref_root = cmds.referenceQuery(root, nodes=True)[0]\n ref_root = (\n cmds.listRelatives(ref_root, parent=True, path=True) or\n [ref_root]\n )\n nodes_to_parent.extend(ref_root)\n continue\n if root.endswith(\"_RN\"):\n # Backwards compatibility for hardcoded reference names.\n refRoot = cmds.referenceQuery(root, n=True)[0]\n refRoot = cmds.listRelatives(refRoot, parent=True) or [refRoot]\n nodes_to_parent.extend(refRoot)\n elif root not in cmds.listSets(allSets=True):\n nodes_to_parent.append(root)\n\n elif not cmds.sets(root, q=True):\n return\n\n # Move loaded nodes to correct index in outliner hierarchy\n placeholder_form = cmds.xform(\n placeholder.scene_identifier,\n q=True,\n matrix=True,\n worldSpace=True\n )\n scene_parent = cmds.listRelatives(\n placeholder.scene_identifier, parent=True, fullPath=True\n )\n for node in set(nodes_to_parent):\n cmds.reorder(node, front=True)\n cmds.reorder(node, relative=placeholder.data[\"index\"])\n cmds.xform(node, matrix=placeholder_form, ws=True)\n if scene_parent:\n cmds.parent(node, scene_parent)\n else:\n cmds.parent(node, world=True)\n\n holding_sets = cmds.listSets(object=placeholder.scene_identifier)\n if not holding_sets:\n return\n for holding_set in holding_sets:\n cmds.sets(roots, forceElement=holding_set)\n\n\ndef build_workfile_template(*args):\n builder = MayaTemplateBuilder(registered_host())\n builder.build_template()\n\n\ndef update_workfile_template(*args):\n builder = MayaTemplateBuilder(registered_host())\n builder.rebuild_template()\n\n\ndef create_placeholder(*args):\n host = registered_host()\n builder = MayaTemplateBuilder(host)\n window = WorkfileBuildPlaceholderDialog(host, builder,\n parent=get_main_window())\n window.show()\n\n\ndef update_placeholder(*args):\n host = registered_host()\n builder = MayaTemplateBuilder(host)\n placeholder_items_by_id = {\n placeholder_item.scene_identifier: placeholder_item\n for placeholder_item in builder.get_placeholders()\n }\n placeholder_items = []\n for node_name in cmds.ls(selection=True, long=True):\n if node_name in placeholder_items_by_id:\n placeholder_items.append(placeholder_items_by_id[node_name])\n\n # TODO show UI at least\n if len(placeholder_items) == 0:\n raise ValueError(\"No node selected\")\n\n if len(placeholder_items) > 1:\n raise ValueError(\"Too many selected nodes\")\n\n placeholder_item = placeholder_items[0]\n window = WorkfileBuildPlaceholderDialog(host, builder,\n parent=get_main_window())\n window.set_update_mode(placeholder_item)\n window.exec_()\n"
},
{
"path": "openpype/hosts/maya/api/workio.py",
"content": "\"\"\"Host API required Work Files tool\"\"\"\nimport os\nfrom maya import cmds\n\n\ndef file_extensions():\n return [\".ma\", \".mb\"]\n\n\ndef has_unsaved_changes():\n return cmds.file(query=True, modified=True)\n\n\ndef save_file(filepath):\n cmds.file(rename=filepath)\n ext = os.path.splitext(filepath)[1]\n if ext == \".mb\":\n file_type = \"mayaBinary\"\n else:\n file_type = \"mayaAscii\"\n cmds.file(save=True, type=file_type)\n\n\ndef open_file(filepath):\n return cmds.file(filepath, open=True, force=True)\n\n\ndef current_file():\n\n current_filepath = cmds.file(query=True, sceneName=True)\n if not current_filepath:\n return None\n\n return current_filepath\n\n\ndef work_root(session):\n work_dir = session[\"AVALON_WORKDIR\"]\n scene_dir = None\n\n # Query scene file rule from workspace.mel if it exists in WORKDIR\n # We are parsing the workspace.mel manually as opposed to temporarily\n # setting the Workspace in Maya in a context manager since Maya had a\n # tendency to crash on frequently changing the workspace when this\n # function was called many times as one scrolled through Work Files assets.\n workspace_mel = os.path.join(work_dir, \"workspace.mel\")\n if os.path.exists(workspace_mel):\n scene_rule = 'workspace -fr \"scene\" '\n # We need to use builtins as `open` is overridden by the workio API\n open_file = __builtins__[\"open\"]\n with open_file(workspace_mel, \"r\") as f:\n for line in f:\n if line.strip().startswith(scene_rule):\n # remainder == \"rule\";\n remainder = line[len(scene_rule):]\n # scene_dir == rule\n scene_dir = remainder.split('\"')[1]\n else:\n # We can't query a workspace that does not exist\n # so we return similar to what we do in other hosts.\n scene_dir = session.get(\"AVALON_SCENEDIR\")\n\n if scene_dir:\n return os.path.join(work_dir, scene_dir)\n else:\n return work_dir\n"
},
{
"path": "openpype/hosts/maya/hooks/pre_auto_load_plugins.py",
"content": "from openpype.lib.applications import PreLaunchHook, LaunchTypes\n\n\nclass MayaPreAutoLoadPlugins(PreLaunchHook):\n \"\"\"Define -noAutoloadPlugins command flag.\"\"\"\n\n # Before AddLastWorkfileToLaunchArgs\n order = 9\n app_groups = {\"maya\"}\n launch_types = {LaunchTypes.local}\n\n def execute(self):\n\n # Ignore if there's no last workfile to start.\n if not self.data.get(\"start_last_workfile\"):\n return\n\n maya_settings = self.data[\"project_settings\"][\"maya\"]\n enabled = maya_settings[\"explicit_plugins_loading\"][\"enabled\"]\n if enabled:\n # Force disable the `AddLastWorkfileToLaunchArgs`.\n self.data.pop(\"start_last_workfile\")\n\n # Force post initialization so our dedicated plug-in load can run\n # prior to Maya opening a scene file.\n key = \"OPENPYPE_OPEN_WORKFILE_POST_INITIALIZATION\"\n self.launch_context.env[key] = \"1\"\n\n self.log.debug(\"Explicit plugins loading.\")\n self.launch_context.launch_args.append(\"-noAutoloadPlugins\")\n"
},
{
"path": "openpype/hosts/maya/hooks/pre_copy_mel.py",
"content": "from openpype.lib.applications import PreLaunchHook, LaunchTypes\nfrom openpype.hosts.maya.lib import create_workspace_mel\n\n\nclass PreCopyMel(PreLaunchHook):\n \"\"\"Copy workspace.mel to workdir.\n\n Hook `GlobalHostDataHook` must be executed before this hook.\n \"\"\"\n app_groups = {\"maya\", \"mayapy\"}\n launch_types = {LaunchTypes.local}\n\n def execute(self):\n project_doc = self.data[\"project_doc\"]\n workdir = self.launch_context.env.get(\"AVALON_WORKDIR\")\n if not workdir:\n self.log.warning(\"BUG: Workdir is not filled.\")\n return\n\n project_settings = self.data[\"project_settings\"]\n create_workspace_mel(workdir, project_doc[\"name\"], project_settings)\n"
},
{
"path": "openpype/hosts/maya/hooks/pre_open_workfile_post_initialization.py",
"content": "from openpype.lib.applications import PreLaunchHook, LaunchTypes\n\n\nclass MayaPreOpenWorkfilePostInitialization(PreLaunchHook):\n \"\"\"Define whether open last workfile should run post initialize.\"\"\"\n\n # Before AddLastWorkfileToLaunchArgs.\n order = 9\n app_groups = {\"maya\"}\n launch_types = {LaunchTypes.local}\n\n def execute(self):\n\n # Ignore if there's no last workfile to start.\n if not self.data.get(\"start_last_workfile\"):\n return\n\n maya_settings = self.data[\"project_settings\"][\"maya\"]\n enabled = maya_settings[\"open_workfile_post_initialization\"]\n if enabled:\n # Force disable the `AddLastWorkfileToLaunchArgs`.\n self.data.pop(\"start_last_workfile\")\n\n self.log.debug(\"Opening workfile post initialization.\")\n key = \"OPENPYPE_OPEN_WORKFILE_POST_INITIALIZATION\"\n self.launch_context.env[key] = \"1\"\n"
},
{
"path": "openpype/hosts/maya/lib.py",
"content": "import os\nfrom openpype.settings import get_project_settings\nfrom openpype.lib import Logger\n\n\ndef create_workspace_mel(workdir, project_name, project_settings=None):\n dst_filepath = os.path.join(workdir, \"workspace.mel\")\n if os.path.exists(dst_filepath):\n return\n\n if not os.path.exists(workdir):\n os.makedirs(workdir)\n\n if not project_settings:\n project_settings = get_project_settings(project_name)\n mel_script = project_settings[\"maya\"].get(\"mel_workspace\")\n\n # Skip if mel script in settings is empty\n if not mel_script:\n log = Logger.get_logger(\"create_workspace_mel\")\n log.debug(\"File 'workspace.mel' not created. Settings value is empty.\")\n return\n\n with open(dst_filepath, \"w\") as mel_file:\n mel_file.write(mel_script)\n"
},
{
"path": "openpype/hosts/maya/plugins/__init__.py",
"content": ""
},
{
"path": "openpype/hosts/maya/plugins/create/convert_legacy.py",
"content": "from openpype.pipeline.create.creator_plugins import SubsetConvertorPlugin\nfrom openpype.hosts.maya.api import plugin\nfrom openpype.hosts.maya.api.lib import read\n\nfrom openpype.client import get_asset_by_name\n\nfrom maya import cmds\nfrom maya.app.renderSetup.model import renderSetup\n\n\nclass MayaLegacyConvertor(SubsetConvertorPlugin,\n plugin.MayaCreatorBase):\n \"\"\"Find and convert any legacy subsets in the scene.\n\n This Convertor will find all legacy subsets in the scene and will\n transform them to the current system. Since the old subsets doesn't\n retain any information about their original creators, the only mapping\n we can do is based on their families.\n\n Its limitation is that you can have multiple creators creating subset\n of the same family and there is no way to handle it. This code should\n nevertheless cover all creators that came with OpenPype.\n\n \"\"\"\n identifier = \"io.openpype.creators.maya.legacy\"\n\n # Cases where the identifier or new family doesn't correspond to the\n # original family on the legacy instances\n special_family_conversions = {\n \"rendering\": \"io.openpype.creators.maya.renderlayer\",\n }\n\n def find_instances(self):\n\n self.cache_subsets(self.collection_shared_data)\n legacy = self.collection_shared_data.get(\"maya_cached_legacy_subsets\")\n if not legacy:\n return\n\n self.add_convertor_item(\"Convert legacy instances\")\n\n def convert(self):\n self.remove_convertor_item()\n\n # We can't use the collected shared data cache here\n # we re-query it here directly to convert all found.\n cache = {}\n self.cache_subsets(cache)\n legacy = cache.get(\"maya_cached_legacy_subsets\")\n if not legacy:\n return\n\n # From all current new style manual creators find the mapping\n # from family to identifier\n family_to_id = {}\n for identifier, creator in self.create_context.creators.items():\n family = getattr(creator, \"family\", None)\n if not family:\n continue\n\n if family in family_to_id:\n # We have a clash of family -> identifier. Multiple\n # new style creators use the same family\n self.log.warning(\"Clash on family->identifier: \"\n \"{}\".format(identifier))\n family_to_id[family] = identifier\n\n family_to_id.update(self.special_family_conversions)\n\n # We also embed the current 'task' into the instance since legacy\n # instances didn't store that data on the instances. The old style\n # logic was thus to be live to the current task to begin with.\n data = dict()\n data[\"task\"] = self.create_context.get_current_task_name()\n for family, instance_nodes in legacy.items():\n if family not in family_to_id:\n self.log.warning(\n \"Unable to convert legacy instance with family '{}'\"\n \" because there is no matching new creator's family\"\n \"\".format(family)\n )\n continue\n\n creator_id = family_to_id[family]\n creator = self.create_context.creators[creator_id]\n data[\"creator_identifier\"] = creator_id\n\n if isinstance(creator, plugin.RenderlayerCreator):\n self._convert_per_renderlayer(instance_nodes, data, creator)\n else:\n self._convert_regular(instance_nodes, data)\n\n def _convert_regular(self, instance_nodes, data):\n # We only imprint the creator identifier for it to identify\n # as the new style creator\n for instance_node in instance_nodes:\n self.imprint_instance_node(instance_node,\n data=data.copy())\n\n def _convert_per_renderlayer(self, instance_nodes, data, creator):\n # Split the instance into an instance per layer\n rs = renderSetup.instance()\n layers = rs.getRenderLayers()\n if not layers:\n self.log.error(\n \"Can't convert legacy renderlayer instance because no existing\"\n \" renderSetup layers exist in the scene.\"\n )\n return\n\n creator_attribute_names = {\n attr_def.key for attr_def in creator.get_instance_attr_defs()\n }\n\n for instance_node in instance_nodes:\n\n # Ensure we have the new style singleton node generated\n # TODO: Make function public\n singleton_node = creator._get_singleton_node()\n if singleton_node:\n self.log.error(\n \"Can't convert legacy renderlayer instance '{}' because\"\n \" new style instance '{}' already exists\".format(\n instance_node,\n singleton_node\n )\n )\n continue\n\n creator.create_singleton_node()\n\n # We are creating new nodes to replace the original instance\n # Copy the attributes of the original instance to the new node\n original_data = read(instance_node)\n\n # The family gets converted to the new family (this is due to\n # \"rendering\" family being converted to \"renderlayer\" family)\n original_data[\"family\"] = creator.family\n\n # recreate subset name as without it would be\n # `renderingMain` vs correct `renderMain`\n project_name = self.create_context.get_current_project_name()\n asset_doc = get_asset_by_name(project_name,\n original_data[\"asset\"])\n subset_name = creator.get_subset_name(\n original_data[\"variant\"],\n data[\"task\"],\n asset_doc,\n project_name)\n original_data[\"subset\"] = subset_name\n\n # Convert to creator attributes when relevant\n creator_attributes = {}\n for key in list(original_data.keys()):\n # Iterate in order of the original attributes to preserve order\n # in the output creator attributes\n if key in creator_attribute_names:\n creator_attributes[key] = original_data.pop(key)\n original_data[\"creator_attributes\"] = creator_attributes\n\n # For layer in maya layers\n for layer in layers:\n layer_instance_node = creator.find_layer_instance_node(layer)\n if not layer_instance_node:\n # TODO: Make function public\n layer_instance_node = creator._create_layer_instance_node(\n layer\n )\n\n # Transfer the main attributes of the original instance\n layer_data = original_data.copy()\n layer_data.update(data)\n\n self.imprint_instance_node(layer_instance_node,\n data=layer_data)\n\n # Delete the legacy instance node\n cmds.delete(instance_node)\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_animation_pointcache.py",
"content": "from maya import cmds\n\nfrom openpype.hosts.maya.api import lib, plugin\n\nfrom openpype.lib import (\n BoolDef,\n NumberDef,\n)\nfrom openpype.pipeline import CreatedInstance\n\n\ndef _get_animation_attr_defs(cls):\n \"\"\"Get Animation generic definitions.\"\"\"\n defs = lib.collect_animation_defs()\n defs.extend(\n [\n BoolDef(\"farm\", label=\"Submit to Farm\"),\n NumberDef(\"priority\", label=\"Farm job Priority\", default=50),\n BoolDef(\"refresh\", label=\"Refresh viewport during export\"),\n BoolDef(\n \"includeParentHierarchy\", label=\"Include Parent Hierarchy\"\n ),\n BoolDef(\n \"includeUserDefinedAttributes\",\n label=\"Include User Defined Attributes\"\n ),\n ]\n )\n\n return defs\n\n\ndef extract_alembic_attributes(node_data, class_name):\n \"\"\"This is a legacy transfer of creator attributes to publish attributes\n for ExtractAlembic/ExtractAnimation plugin.\n \"\"\"\n publish_attributes = node_data[\"publish_attributes\"]\n\n if class_name in publish_attributes:\n return node_data\n\n extract_alembic_flags = [\n \"writeColorSets\",\n \"writeFaceSets\",\n \"writeNormals\",\n \"renderableOnly\",\n \"visibleOnly\",\n \"worldSpace\",\n \"renderableOnly\"\n ]\n extract_alembic_attributes = [\n \"attr\",\n \"attrPrefix\",\n \"visibleOnly\"\n ]\n attributes = extract_alembic_flags + extract_alembic_attributes\n plugin_attributes = {\"flags\": []}\n for attr in attributes:\n if attr not in node_data[\"creator_attributes\"].keys():\n continue\n value = node_data[\"creator_attributes\"].pop(attr)\n\n if value and attr in extract_alembic_flags:\n plugin_attributes[\"flags\"].append(attr)\n\n if attr in extract_alembic_attributes:\n plugin_attributes[attr] = value\n\n publish_attributes[class_name] = plugin_attributes\n\n return node_data\n\n\nclass CreateAnimation(plugin.MayaHiddenCreator):\n \"\"\"Animation output for character rigs\n\n We hide the animation creator from the UI since the creation of it is\n automated upon loading a rig. There's an inventory action to recreate it\n for loaded rigs if by chance someone deleted the animation instance.\n \"\"\"\n\n identifier = \"io.openpype.creators.maya.animation\"\n name = \"animationDefault\"\n label = \"Animation\"\n family = \"animation\"\n icon = \"male\"\n\n write_color_sets = False\n write_face_sets = False\n include_parent_hierarchy = False\n include_user_defined_attributes = False\n\n def collect_instances(self):\n try:\n cached_subsets = self.collection_shared_data[\"maya_cached_subsets\"]\n except KeyError:\n self.cache_subsets(self.collection_shared_data)\n cached_subsets = self.collection_shared_data[\"maya_cached_subsets\"]\n\n for node in cached_subsets.get(self.identifier, []):\n node_data = self.read_instance_node(node)\n\n node_data = extract_alembic_attributes(\n node_data, \"ExtractAnimation\"\n )\n\n created_instance = CreatedInstance.from_existing(node_data, self)\n self._add_instance_to_context(created_instance)\n\n def get_instance_attr_defs(self):\n super(CreateAnimation, self).get_instance_attr_defs()\n defs = _get_animation_attr_defs(self)\n return defs\n\n\nclass CreatePointCache(plugin.MayaCreator):\n \"\"\"Alembic pointcache for animated data\"\"\"\n\n identifier = \"io.openpype.creators.maya.pointcache\"\n label = \"Pointcache\"\n family = \"pointcache\"\n icon = \"gears\"\n write_color_sets = False\n write_face_sets = False\n include_user_defined_attributes = False\n\n def collect_instances(self):\n try:\n cached_subsets = self.collection_shared_data[\"maya_cached_subsets\"]\n except KeyError:\n self.cache_subsets(self.collection_shared_data)\n cached_subsets = self.collection_shared_data[\"maya_cached_subsets\"]\n\n for node in cached_subsets.get(self.identifier, []):\n node_data = self.read_instance_node(node)\n\n node_data = extract_alembic_attributes(node_data, \"ExtractAlembic\")\n\n created_instance = CreatedInstance.from_existing(node_data, self)\n self._add_instance_to_context(created_instance)\n\n def get_instance_attr_defs(self):\n super(CreatePointCache, self).get_instance_attr_defs()\n defs = _get_animation_attr_defs(self)\n return defs\n\n def create(self, subset_name, instance_data, pre_create_data):\n instance = super(CreatePointCache, self).create(\n subset_name, instance_data, pre_create_data\n )\n instance_node = instance.get(\"instance_node\")\n\n # For Arnold standin proxy\n proxy_set = cmds.sets(name=instance_node + \"_proxy_SET\", empty=True)\n cmds.sets(proxy_set, forceElement=instance_node)\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_arnold_scene_source.py",
"content": "from maya import cmds\n\nfrom openpype.hosts.maya.api import (\n lib,\n plugin\n)\nfrom openpype.lib import (\n NumberDef,\n BoolDef\n)\n\n\nclass CreateArnoldSceneSource(plugin.MayaCreator):\n \"\"\"Arnold Scene Source\"\"\"\n\n identifier = \"io.openpype.creators.maya.ass\"\n label = \"Arnold Scene Source\"\n family = \"ass\"\n icon = \"cube\"\n settings_name = \"CreateAss\"\n\n expandProcedurals = False\n motionBlur = True\n motionBlurKeys = 2\n motionBlurLength = 0.5\n maskOptions = False\n maskCamera = False\n maskLight = False\n maskShape = False\n maskShader = False\n maskOverride = False\n maskDriver = False\n maskFilter = False\n maskColor_manager = False\n maskOperator = False\n\n def get_instance_attr_defs(self):\n\n defs = lib.collect_animation_defs()\n\n defs.extend([\n BoolDef(\"expandProcedural\",\n label=\"Expand Procedural\",\n default=self.expandProcedurals),\n BoolDef(\"motionBlur\",\n label=\"Motion Blur\",\n default=self.motionBlur),\n NumberDef(\"motionBlurKeys\",\n label=\"Motion Blur Keys\",\n decimals=0,\n default=self.motionBlurKeys),\n NumberDef(\"motionBlurLength\",\n label=\"Motion Blur Length\",\n decimals=3,\n default=self.motionBlurLength),\n\n # Masks\n BoolDef(\"maskOptions\",\n label=\"Export Options\",\n default=self.maskOptions),\n BoolDef(\"maskCamera\",\n label=\"Export Cameras\",\n default=self.maskCamera),\n BoolDef(\"maskLight\",\n label=\"Export Lights\",\n default=self.maskLight),\n BoolDef(\"maskShape\",\n label=\"Export Shapes\",\n default=self.maskShape),\n BoolDef(\"maskShader\",\n label=\"Export Shaders\",\n default=self.maskShader),\n BoolDef(\"maskOverride\",\n label=\"Export Override Nodes\",\n default=self.maskOverride),\n BoolDef(\"maskDriver\",\n label=\"Export Drivers\",\n default=self.maskDriver),\n BoolDef(\"maskFilter\",\n label=\"Export Filters\",\n default=self.maskFilter),\n BoolDef(\"maskOperator\",\n label=\"Export Operators\",\n default=self.maskOperator),\n BoolDef(\"maskColor_manager\",\n label=\"Export Color Managers\",\n default=self.maskColor_manager),\n ])\n\n return defs\n\n\nclass CreateArnoldSceneSourceProxy(CreateArnoldSceneSource):\n \"\"\"Arnold Scene Source Proxy\n\n This product type facilitates working with proxy geometry in the viewport.\n \"\"\"\n\n identifier = \"io.openpype.creators.maya.assproxy\"\n label = \"Arnold Scene Source Proxy\"\n family = \"assProxy\"\n icon = \"cube\"\n\n def create(self, subset_name, instance_data, pre_create_data):\n instance = super(CreateArnoldSceneSource, self).create(\n subset_name, instance_data, pre_create_data\n )\n\n instance_node = instance.get(\"instance_node\")\n\n proxy = cmds.sets(name=instance_node + \"_proxy_SET\", empty=True)\n cmds.sets([proxy], forceElement=instance_node)\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_assembly.py",
"content": "from openpype.hosts.maya.api import plugin\n\n\nclass CreateAssembly(plugin.MayaCreator):\n \"\"\"A grouped package of loaded content\"\"\"\n\n identifier = \"io.openpype.creators.maya.assembly\"\n label = \"Assembly\"\n family = \"assembly\"\n icon = \"cubes\"\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_camera.py",
"content": "from openpype.hosts.maya.api import (\n lib,\n plugin\n)\nfrom openpype.lib import BoolDef\n\n\nclass CreateCamera(plugin.MayaCreator):\n \"\"\"Single baked camera\"\"\"\n\n identifier = \"io.openpype.creators.maya.camera\"\n label = \"Camera\"\n family = \"camera\"\n icon = \"video-camera\"\n\n def get_instance_attr_defs(self):\n\n defs = lib.collect_animation_defs()\n\n defs.extend([\n BoolDef(\"bakeToWorldSpace\",\n label=\"Bake to World-Space\",\n tooltip=\"Bake to World-Space\",\n default=True),\n ])\n\n return defs\n\n\nclass CreateCameraRig(plugin.MayaCreator):\n \"\"\"Complex hierarchy with camera.\"\"\"\n\n identifier = \"io.openpype.creators.maya.camerarig\"\n label = \"Camera Rig\"\n family = \"camerarig\"\n icon = \"video-camera\"\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_layout.py",
"content": "from openpype.hosts.maya.api import plugin\nfrom openpype.lib import BoolDef\n\n\nclass CreateLayout(plugin.MayaCreator):\n \"\"\"A grouped package of loaded content\"\"\"\n\n identifier = \"io.openpype.creators.maya.layout\"\n label = \"Layout\"\n family = \"layout\"\n icon = \"cubes\"\n\n def get_instance_attr_defs(self):\n\n return [\n BoolDef(\"groupLoadedAssets\",\n label=\"Group Loaded Assets\",\n tooltip=\"Enable this when you want to publish group of \"\n \"loaded asset\",\n default=False)\n ]\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_look.py",
"content": "from openpype.hosts.maya.api import (\n plugin,\n lib\n)\nfrom openpype.lib import (\n BoolDef,\n TextDef\n)\n\n\nclass CreateLook(plugin.MayaCreator):\n \"\"\"Shader connections defining shape look\"\"\"\n\n identifier = \"io.openpype.creators.maya.look\"\n label = \"Look\"\n family = \"look\"\n icon = \"paint-brush\"\n\n make_tx = True\n rs_tex = False\n\n def get_instance_attr_defs(self):\n\n return [\n # TODO: This value should actually get set on create!\n TextDef(\"renderLayer\",\n # TODO: Bug: Hidden attribute's label is still shown in UI?\n hidden=True,\n default=lib.get_current_renderlayer(),\n label=\"Renderlayer\",\n tooltip=\"Renderlayer to extract the look from\"),\n BoolDef(\"maketx\",\n label=\"MakeTX\",\n tooltip=\"Whether to generate .tx files for your textures\",\n default=self.make_tx),\n BoolDef(\"rstex\",\n label=\"Convert textures to .rstex\",\n tooltip=\"Whether to generate Redshift .rstex files for \"\n \"your textures\",\n default=self.rs_tex)\n ]\n\n def get_pre_create_attr_defs(self):\n # Show same attributes on create but include use selection\n defs = super(CreateLook, self).get_pre_create_attr_defs()\n defs.extend(self.get_instance_attr_defs())\n return defs\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_matchmove.py",
"content": "from openpype.hosts.maya.api import (\n lib,\n plugin\n)\nfrom openpype.lib import BoolDef\n\n\nclass CreateMatchmove(plugin.MayaCreator):\n \"\"\"Instance for more complex setup of cameras.\n\n Might contain multiple cameras, geometries etc.\n\n It is expected to be extracted into .abc or .ma\n \"\"\"\n\n identifier = \"io.openpype.creators.maya.matchmove\"\n label = \"Matchmove\"\n family = \"matchmove\"\n icon = \"video-camera\"\n\n def get_instance_attr_defs(self):\n\n defs = lib.collect_animation_defs()\n\n defs.extend([\n BoolDef(\"bakeToWorldSpace\",\n label=\"Bake Cameras to World-Space\",\n tooltip=\"Bake Cameras to World-Space\",\n default=True),\n ])\n\n return defs\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_maya_usd.py",
"content": "from openpype.hosts.maya.api import plugin, lib\nfrom openpype.lib import (\n BoolDef,\n EnumDef,\n TextDef\n)\n\nfrom maya import cmds\n\n\nclass CreateMayaUsd(plugin.MayaCreator):\n \"\"\"Create Maya USD Export\"\"\"\n\n identifier = \"io.openpype.creators.maya.mayausd\"\n label = \"Maya USD\"\n family = \"usd\"\n icon = \"cubes\"\n description = \"Create Maya USD Export\"\n\n cache = {}\n\n def get_publish_families(self):\n return [\"usd\", \"mayaUsd\"]\n\n def get_instance_attr_defs(self):\n\n if \"jobContextItems\" not in self.cache:\n # Query once instead of per instance\n job_context_items = {}\n try:\n cmds.loadPlugin(\"mayaUsdPlugin\", quiet=True)\n job_context_items = {\n cmds.mayaUSDListJobContexts(jobContext=name): name\n for name in cmds.mayaUSDListJobContexts(export=True) or []\n }\n except RuntimeError:\n # Likely `mayaUsdPlugin` plug-in not available\n self.log.warning(\"Unable to retrieve available job \"\n \"contexts for `mayaUsdPlugin` exports\")\n\n if not job_context_items:\n # enumdef multiselection may not be empty\n job_context_items = [\"\"]\n\n self.cache[\"jobContextItems\"] = job_context_items\n\n defs = lib.collect_animation_defs()\n defs.extend([\n EnumDef(\"defaultUSDFormat\",\n label=\"File format\",\n items={\n \"usdc\": \"Binary\",\n \"usda\": \"ASCII\"\n },\n default=\"usdc\"),\n BoolDef(\"stripNamespaces\",\n label=\"Strip Namespaces\",\n tooltip=(\n \"Remove namespaces during export. By default, \"\n \"namespaces are exported to the USD file in the \"\n \"following format: nameSpaceExample_pPlatonic1\"\n ),\n default=True),\n BoolDef(\"mergeTransformAndShape\",\n label=\"Merge Transform and Shape\",\n tooltip=(\n \"Combine Maya transform and shape into a single USD\"\n \"prim that has transform and geometry, for all\"\n \" \\\"geometric primitives\\\" (gprims).\\n\"\n \"This results in smaller and faster scenes. Gprims \"\n \"will be \\\"unpacked\\\" back into transform and shape \"\n \"nodes when imported into Maya from USD.\"\n ),\n default=True),\n BoolDef(\"includeUserDefinedAttributes\",\n label=\"Include User Defined Attributes\",\n tooltip=(\n \"Whether to include all custom maya attributes found \"\n \"on nodes as metadata (userProperties) in USD.\"\n ),\n default=False),\n TextDef(\"attr\",\n label=\"Custom Attributes\",\n default=\"\",\n placeholder=\"attr1, attr2\"),\n TextDef(\"attrPrefix\",\n label=\"Custom Attributes Prefix\",\n default=\"\",\n placeholder=\"prefix1, prefix2\"),\n EnumDef(\"jobContext\",\n label=\"Job Context\",\n items=self.cache[\"jobContextItems\"],\n tooltip=(\n \"Specifies an additional export context to handle.\\n\"\n \"These usually contain extra schemas, primitives,\\n\"\n \"and materials that are to be exported for a \"\n \"specific\\ntask, a target renderer for example.\"\n ),\n multiselection=True),\n ])\n\n return defs\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_mayascene.py",
"content": "from openpype.hosts.maya.api import plugin\n\n\nclass CreateMayaScene(plugin.MayaCreator):\n \"\"\"Raw Maya Scene file export\"\"\"\n\n identifier = \"io.openpype.creators.maya.mayascene\"\n name = \"mayaScene\"\n label = \"Maya Scene\"\n family = \"mayaScene\"\n icon = \"file-archive-o\"\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_model.py",
"content": "from openpype.hosts.maya.api import plugin\nfrom openpype.lib import (\n BoolDef,\n TextDef\n)\n\n\nclass CreateModel(plugin.MayaCreator):\n \"\"\"Polygonal static geometry\"\"\"\n\n identifier = \"io.openpype.creators.maya.model\"\n label = \"Model\"\n family = \"model\"\n icon = \"cube\"\n default_variants = [\"Main\", \"Proxy\", \"_MD\", \"_HD\", \"_LD\"]\n\n write_color_sets = False\n write_face_sets = False\n\n def get_instance_attr_defs(self):\n\n return [\n BoolDef(\"writeColorSets\",\n label=\"Write vertex colors\",\n tooltip=\"Write vertex colors with the geometry\",\n default=self.write_color_sets),\n BoolDef(\"writeFaceSets\",\n label=\"Write face sets\",\n tooltip=\"Write face sets with the geometry\",\n default=self.write_face_sets),\n BoolDef(\"includeParentHierarchy\",\n label=\"Include Parent Hierarchy\",\n tooltip=\"Whether to include parent hierarchy of nodes in \"\n \"the publish instance\",\n default=False),\n TextDef(\"attr\",\n label=\"Custom Attributes\",\n default=\"\",\n placeholder=\"attr1, attr2\"),\n TextDef(\"attrPrefix\",\n label=\"Custom Attributes Prefix\",\n placeholder=\"prefix1, prefix2\")\n ]\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_multishot_layout.py",
"content": "from ayon_api import (\n get_folder_by_name,\n get_folder_by_path,\n get_folders,\n)\nfrom maya import cmds # noqa: F401\n\nfrom openpype import AYON_SERVER_ENABLED\nfrom openpype.client import get_assets\nfrom openpype.hosts.maya.api import plugin\nfrom openpype.lib import BoolDef, EnumDef, TextDef\nfrom openpype.pipeline import (\n Creator,\n get_current_asset_name,\n get_current_project_name,\n)\nfrom openpype.pipeline.create import CreatorError\n\n\nclass CreateMultishotLayout(plugin.MayaCreator):\n \"\"\"Create a multi-shot layout in the Maya scene.\n\n This creator will create a Camera Sequencer in the Maya scene based on\n the shots found under the specified folder. The shots will be added to\n the sequencer in the order of their clipIn and clipOut values. For each\n shot a Layout will be created.\n\n \"\"\"\n identifier = \"io.openpype.creators.maya.multishotlayout\"\n label = \"Multi-shot Layout\"\n family = \"layout\"\n icon = \"project-diagram\"\n\n def get_pre_create_attr_defs(self):\n # Present artist with a list of parents of the current context\n # to choose from. This will be used to get the shots under the\n # selected folder to create the Camera Sequencer.\n\n \"\"\"\n Todo: `get_folder_by_name` should be switched to `get_folder_by_path`\n once the fork to pure AYON is done.\n\n Warning: this will not work for projects where the asset name\n is not unique across the project until the switch mentioned\n above is done.\n \"\"\"\n\n project_name = get_current_project_name()\n folder_path = get_current_asset_name()\n if \"/\" in folder_path:\n current_folder = get_folder_by_path(project_name, folder_path)\n else:\n current_folder = get_folder_by_name(\n project_name, folder_name=folder_path\n )\n\n current_path_parts = current_folder[\"path\"].split(\"/\")\n\n # populate the list with parents of the current folder\n # this will create menu items like:\n # [\n # {\n # \"value\": \"\",\n # \"label\": \"project (shots directly under the project)\"\n # }, {\n # \"value\": \"shots/shot_01\", \"label\": \"shot_01 (current)\"\n # }, {\n # \"value\": \"shots\", \"label\": \"shots\"\n # }\n # ]\n\n # add the project as the first item\n items_with_label = [\n {\n \"label\": f\"{self.project_name} \"\n \"(shots directly under the project)\",\n \"value\": \"\"\n }\n ]\n\n # go through the current folder path and add each part to the list,\n # but mark the current folder.\n for part_idx, part in enumerate(current_path_parts):\n label = part\n if label == current_folder[\"name\"]:\n label = f\"{label} (current)\"\n\n value = \"/\".join(current_path_parts[:part_idx + 1])\n\n items_with_label.append({\"label\": label, \"value\": value})\n\n return [\n EnumDef(\"shotParent\",\n default=current_folder[\"name\"],\n label=\"Shot Parent Folder\",\n items=items_with_label,\n ),\n BoolDef(\"groupLoadedAssets\",\n label=\"Group Loaded Assets\",\n tooltip=\"Enable this when you want to publish group of \"\n \"loaded asset\",\n default=False),\n TextDef(\"taskName\",\n label=\"Associated Task Name\",\n tooltip=(\"Task name to be associated \"\n \"with the created Layout\"),\n default=\"layout\"),\n ]\n\n def create(self, subset_name, instance_data, pre_create_data):\n shots = list(\n self.get_related_shots(folder_path=pre_create_data[\"shotParent\"])\n )\n if not shots:\n # There are no shot folders under the specified folder.\n # We are raising an error here but in the future we might\n # want to create a new shot folders by publishing the layouts\n # and shot defined in the sequencer. Sort of editorial publish\n # in side of Maya.\n raise CreatorError((\n \"No shots found under the specified \"\n f\"folder: {pre_create_data['shotParent']}.\"))\n\n # Get layout creator\n layout_creator_id = \"io.openpype.creators.maya.layout\"\n layout_creator: Creator = self.create_context.creators.get(\n layout_creator_id)\n if not layout_creator:\n raise CreatorError(\n f\"Creator {layout_creator_id} not found.\")\n\n # Get OpenPype style asset documents for the shots\n op_asset_docs = get_assets(\n self.project_name, [s[\"id\"] for s in shots])\n asset_docs_by_id = {doc[\"_id\"]: doc for doc in op_asset_docs}\n for shot in shots:\n # we are setting shot name to be displayed in the sequencer to\n # `shot name (shot label)` if the label is set, otherwise just\n # `shot name`. So far, labels are used only when the name is set\n # with characters that are not allowed in the shot name.\n if not shot[\"active\"]:\n continue\n\n # get task for shot\n asset_doc = asset_docs_by_id[shot[\"id\"]]\n\n tasks = asset_doc.get(\"data\").get(\"tasks\").keys()\n layout_task = None\n if pre_create_data[\"taskName\"] in tasks:\n layout_task = pre_create_data[\"taskName\"]\n\n shot_name = f\"{shot['name']}%s\" % (\n f\" ({shot['label']})\" if shot[\"label\"] else \"\")\n cmds.shot(sequenceStartTime=shot[\"attrib\"][\"clipIn\"],\n sequenceEndTime=shot[\"attrib\"][\"clipOut\"],\n shotName=shot_name)\n\n # Create layout instance by the layout creator\n\n instance_data = {\n \"folderPath\": shot[\"path\"],\n \"variant\": layout_creator.get_default_variant()\n }\n if layout_task:\n instance_data[\"task\"] = layout_task\n\n layout_creator.create(\n subset_name=layout_creator.get_subset_name(\n layout_creator.get_default_variant(),\n self.create_context.get_current_task_name(),\n asset_doc,\n self.project_name),\n instance_data=instance_data,\n pre_create_data={\n \"groupLoadedAssets\": pre_create_data[\"groupLoadedAssets\"]\n }\n )\n\n def get_related_shots(self, folder_path: str):\n \"\"\"Get all shots related to the current asset.\n\n Get all folders of type Shot under specified folder.\n\n Args:\n folder_path (str): Path of the folder.\n\n Returns:\n list: List of dicts with folder data.\n\n \"\"\"\n # if folder_path is None, project is selected as a root\n # and its name is used as a parent id\n parent_id = self.project_name\n if folder_path:\n current_folder = get_folder_by_path(\n project_name=self.project_name,\n folder_path=folder_path,\n )\n parent_id = current_folder[\"id\"]\n\n # get all child folders of the current one\n return get_folders(\n project_name=self.project_name,\n parent_ids=[parent_id],\n fields=[\n \"attrib.clipIn\", \"attrib.clipOut\",\n \"attrib.frameStart\", \"attrib.frameEnd\",\n \"name\", \"label\", \"path\", \"folderType\", \"id\"\n ]\n )\n\n\n# blast this creator if Ayon server is not enabled\nif not AYON_SERVER_ENABLED:\n del CreateMultishotLayout\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_multiverse_look.py",
"content": "from openpype.hosts.maya.api import plugin\nfrom openpype.lib import (\n BoolDef,\n EnumDef\n)\n\n\nclass CreateMultiverseLook(plugin.MayaCreator):\n \"\"\"Create Multiverse Look\"\"\"\n\n identifier = \"io.openpype.creators.maya.mvlook\"\n label = \"Multiverse Look\"\n family = \"mvLook\"\n icon = \"cubes\"\n\n def get_instance_attr_defs(self):\n\n return [\n EnumDef(\"fileFormat\",\n label=\"File Format\",\n tooltip=\"USD export file format\",\n items=[\"usda\", \"usd\"],\n default=\"usda\"),\n BoolDef(\"publishMipMap\",\n label=\"Publish MipMap\",\n default=True),\n ]\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_multiverse_usd.py",
"content": "from openpype.hosts.maya.api import plugin, lib\nfrom openpype.lib import (\n BoolDef,\n NumberDef,\n TextDef,\n EnumDef\n)\n\n\nclass CreateMultiverseUsd(plugin.MayaCreator):\n \"\"\"Create Multiverse USD Asset\"\"\"\n\n identifier = \"io.openpype.creators.maya.mvusdasset\"\n label = \"Multiverse USD Asset\"\n family = \"usd\"\n icon = \"cubes\"\n description = \"Create Multiverse USD Asset\"\n\n def get_publish_families(self):\n return [\"usd\", \"mvUsd\"]\n\n def get_instance_attr_defs(self):\n\n defs = lib.collect_animation_defs(fps=True)\n defs.extend([\n EnumDef(\"fileFormat\",\n label=\"File format\",\n items=[\"usd\", \"usda\", \"usdz\"],\n default=\"usd\"),\n BoolDef(\"stripNamespaces\",\n label=\"Strip Namespaces\",\n default=True),\n BoolDef(\"mergeTransformAndShape\",\n label=\"Merge Transform and Shape\",\n default=False),\n BoolDef(\"writeAncestors\",\n label=\"Write Ancestors\",\n default=True),\n BoolDef(\"flattenParentXforms\",\n label=\"Flatten Parent Xforms\",\n default=False),\n BoolDef(\"writeSparseOverrides\",\n label=\"Write Sparse Overrides\",\n default=False),\n BoolDef(\"useMetaPrimPath\",\n label=\"Use Meta Prim Path\",\n default=False),\n TextDef(\"customRootPath\",\n label=\"Custom Root Path\",\n default=''),\n TextDef(\"customAttributes\",\n label=\"Custom Attributes\",\n tooltip=\"Comma-separated list of attribute names\",\n default=''),\n TextDef(\"nodeTypesToIgnore\",\n label=\"Node Types to Ignore\",\n tooltip=\"Comma-separated list of node types to be ignored\",\n default=''),\n BoolDef(\"writeMeshes\",\n label=\"Write Meshes\",\n default=True),\n BoolDef(\"writeCurves\",\n label=\"Write Curves\",\n default=True),\n BoolDef(\"writeParticles\",\n label=\"Write Particles\",\n default=True),\n BoolDef(\"writeCameras\",\n label=\"Write Cameras\",\n default=False),\n BoolDef(\"writeLights\",\n label=\"Write Lights\",\n default=False),\n BoolDef(\"writeJoints\",\n label=\"Write Joints\",\n default=False),\n BoolDef(\"writeCollections\",\n label=\"Write Collections\",\n default=False),\n BoolDef(\"writePositions\",\n label=\"Write Positions\",\n default=True),\n BoolDef(\"writeNormals\",\n label=\"Write Normals\",\n default=True),\n BoolDef(\"writeUVs\",\n label=\"Write UVs\",\n default=True),\n BoolDef(\"writeColorSets\",\n label=\"Write Color Sets\",\n default=False),\n BoolDef(\"writeTangents\",\n label=\"Write Tangents\",\n default=False),\n BoolDef(\"writeRefPositions\",\n label=\"Write Ref Positions\",\n default=True),\n BoolDef(\"writeBlendShapes\",\n label=\"Write BlendShapes\",\n default=False),\n BoolDef(\"writeDisplayColor\",\n label=\"Write Display Color\",\n default=True),\n BoolDef(\"writeSkinWeights\",\n label=\"Write Skin Weights\",\n default=False),\n BoolDef(\"writeMaterialAssignment\",\n label=\"Write Material Assignment\",\n default=False),\n BoolDef(\"writeHardwareShader\",\n label=\"Write Hardware Shader\",\n default=False),\n BoolDef(\"writeShadingNetworks\",\n label=\"Write Shading Networks\",\n default=False),\n BoolDef(\"writeTransformMatrix\",\n label=\"Write Transform Matrix\",\n default=True),\n BoolDef(\"writeUsdAttributes\",\n label=\"Write USD Attributes\",\n default=True),\n BoolDef(\"writeInstancesAsReferences\",\n label=\"Write Instances as References\",\n default=False),\n BoolDef(\"timeVaryingTopology\",\n label=\"Time Varying Topology\",\n default=False),\n TextDef(\"customMaterialNamespace\",\n label=\"Custom Material Namespace\",\n default=''),\n NumberDef(\"numTimeSamples\",\n label=\"Num Time Samples\",\n default=1),\n NumberDef(\"timeSamplesSpan\",\n label=\"Time Samples Span\",\n default=0.0),\n ])\n\n return defs\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_multiverse_usd_comp.py",
"content": "from openpype.hosts.maya.api import plugin, lib\nfrom openpype.lib import (\n BoolDef,\n NumberDef,\n EnumDef\n)\n\n\nclass CreateMultiverseUsdComp(plugin.MayaCreator):\n \"\"\"Create Multiverse USD Composition\"\"\"\n\n identifier = \"io.openpype.creators.maya.mvusdcomposition\"\n label = \"Multiverse USD Composition\"\n family = \"mvUsdComposition\"\n icon = \"cubes\"\n\n def get_instance_attr_defs(self):\n\n defs = lib.collect_animation_defs(fps=True)\n defs.extend([\n EnumDef(\"fileFormat\",\n label=\"File format\",\n items=[\"usd\", \"usda\"],\n default=\"usd\"),\n BoolDef(\"stripNamespaces\",\n label=\"Strip Namespaces\",\n default=False),\n BoolDef(\"mergeTransformAndShape\",\n label=\"Merge Transform and Shape\",\n default=False),\n BoolDef(\"flattenContent\",\n label=\"Flatten Content\",\n default=False),\n BoolDef(\"writeAsCompoundLayers\",\n label=\"Write As Compound Layers\",\n default=False),\n BoolDef(\"writePendingOverrides\",\n label=\"Write Pending Overrides\",\n default=False),\n NumberDef(\"numTimeSamples\",\n label=\"Num Time Samples\",\n default=1),\n NumberDef(\"timeSamplesSpan\",\n label=\"Time Samples Span\",\n default=0.0),\n ])\n\n return defs\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_multiverse_usd_over.py",
"content": "from openpype.hosts.maya.api import plugin, lib\nfrom openpype.lib import (\n BoolDef,\n NumberDef,\n EnumDef\n)\n\n\nclass CreateMultiverseUsdOver(plugin.MayaCreator):\n \"\"\"Create Multiverse USD Override\"\"\"\n\n identifier = \"io.openpype.creators.maya.mvusdoverride\"\n label = \"Multiverse USD Override\"\n family = \"mvUsdOverride\"\n icon = \"cubes\"\n\n def get_instance_attr_defs(self):\n defs = lib.collect_animation_defs(fps=True)\n defs.extend([\n EnumDef(\"fileFormat\",\n label=\"File format\",\n items=[\"usd\", \"usda\"],\n default=\"usd\"),\n BoolDef(\"writeAll\",\n label=\"Write All\",\n default=False),\n BoolDef(\"writeTransforms\",\n label=\"Write Transforms\",\n default=True),\n BoolDef(\"writeVisibility\",\n label=\"Write Visibility\",\n default=True),\n BoolDef(\"writeAttributes\",\n label=\"Write Attributes\",\n default=True),\n BoolDef(\"writeMaterials\",\n label=\"Write Materials\",\n default=True),\n BoolDef(\"writeVariants\",\n label=\"Write Variants\",\n default=True),\n BoolDef(\"writeVariantsDefinition\",\n label=\"Write Variants Definition\",\n default=True),\n BoolDef(\"writeActiveState\",\n label=\"Write Active State\",\n default=True),\n BoolDef(\"writeNamespaces\",\n label=\"Write Namespaces\",\n default=False),\n NumberDef(\"numTimeSamples\",\n label=\"Num Time Samples\",\n default=1),\n NumberDef(\"timeSamplesSpan\",\n label=\"Time Samples Span\",\n default=0.0),\n ])\n\n return defs\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_proxy_abc.py",
"content": "from openpype.hosts.maya.api import (\n lib,\n plugin\n)\nfrom openpype.lib import (\n BoolDef,\n TextDef\n)\n\n\nclass CreateProxyAlembic(plugin.MayaCreator):\n \"\"\"Proxy Alembic for animated data\"\"\"\n\n identifier = \"io.openpype.creators.maya.proxyabc\"\n label = \"Proxy Alembic\"\n family = \"proxyAbc\"\n icon = \"gears\"\n write_color_sets = False\n write_face_sets = False\n\n def get_instance_attr_defs(self):\n\n defs = lib.collect_animation_defs()\n\n defs.extend([\n BoolDef(\"writeColorSets\",\n label=\"Write vertex colors\",\n tooltip=\"Write vertex colors with the geometry\",\n default=self.write_color_sets),\n BoolDef(\"writeFaceSets\",\n label=\"Write face sets\",\n tooltip=\"Write face sets with the geometry\",\n default=self.write_face_sets),\n BoolDef(\"worldSpace\",\n label=\"World-Space Export\",\n default=True),\n TextDef(\"nameSuffix\",\n label=\"Name Suffix for Bounding Box\",\n default=\"_BBox\",\n placeholder=\"_BBox\"),\n TextDef(\"attr\",\n label=\"Custom Attributes\",\n default=\"\",\n placeholder=\"attr1, attr2\"),\n TextDef(\"attrPrefix\",\n label=\"Custom Attributes Prefix\",\n placeholder=\"prefix1, prefix2\")\n ])\n\n return defs\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_redshift_proxy.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator of Redshift proxy subset types.\"\"\"\n\nfrom openpype.hosts.maya.api import plugin, lib\nfrom openpype.lib import BoolDef\n\n\nclass CreateRedshiftProxy(plugin.MayaCreator):\n \"\"\"Create instance of Redshift Proxy subset.\"\"\"\n\n identifier = \"io.openpype.creators.maya.redshiftproxy\"\n label = \"Redshift Proxy\"\n family = \"redshiftproxy\"\n icon = \"gears\"\n\n def get_instance_attr_defs(self):\n\n defs = [\n BoolDef(\"animation\",\n label=\"Export animation\",\n default=False)\n ]\n\n defs.extend(lib.collect_animation_defs())\n return defs\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_render.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Create ``Render`` instance in Maya.\"\"\"\n\nfrom openpype.hosts.maya.api import (\n lib_rendersettings,\n plugin\n)\nfrom openpype.pipeline import CreatorError\nfrom openpype.lib import (\n BoolDef,\n NumberDef,\n)\n\n\nclass CreateRenderlayer(plugin.RenderlayerCreator):\n \"\"\"Create and manages renderlayer subset per renderLayer in workfile.\n\n This generates a single node in the scene which tells the Creator to if\n it exists collect Maya rendersetup renderlayers as individual instances.\n As such, triggering create doesn't actually create the instance node per\n layer but only the node which tells the Creator it may now collect\n the renderlayers.\n\n \"\"\"\n\n identifier = \"io.openpype.creators.maya.renderlayer\"\n family = \"renderlayer\"\n label = \"Render\"\n icon = \"eye\"\n\n layer_instance_prefix = \"render\"\n singleton_node_name = \"renderingMain\"\n\n render_settings = {}\n\n @classmethod\n def apply_settings(cls, project_settings):\n cls.render_settings = project_settings[\"maya\"][\"RenderSettings\"]\n\n def create(self, subset_name, instance_data, pre_create_data):\n # Only allow a single render instance to exist\n if self._get_singleton_node():\n raise CreatorError(\"A Render instance already exists - only \"\n \"one can be configured.\")\n\n # Apply default project render settings on create\n if self.render_settings.get(\"apply_render_settings\"):\n lib_rendersettings.RenderSettings().set_default_renderer_settings()\n\n super(CreateRenderlayer, self).create(subset_name,\n instance_data,\n pre_create_data)\n\n def get_instance_attr_defs(self):\n \"\"\"Create instance settings.\"\"\"\n\n return [\n BoolDef(\"review\",\n label=\"Review\",\n tooltip=\"Mark as reviewable\",\n default=True),\n BoolDef(\"extendFrames\",\n label=\"Extend Frames\",\n tooltip=\"Extends the frames on top of the previous \"\n \"publish.\\nIf the previous was 1001-1050 and you \"\n \"would now submit 1020-1070 only the new frames \"\n \"1051-1070 would be rendered and published \"\n \"together with the previously rendered frames.\\n\"\n \"If 'overrideExistingFrame' is enabled it *will* \"\n \"render any existing frames.\",\n default=False),\n BoolDef(\"overrideExistingFrame\",\n label=\"Override Existing Frame\",\n tooltip=\"Override existing rendered frames \"\n \"(if they exist).\",\n default=True),\n\n # TODO: Should these move to submit_maya_deadline plugin?\n # Tile rendering\n BoolDef(\"tileRendering\",\n label=\"Enable tiled rendering\",\n default=False),\n NumberDef(\"tilesX\",\n label=\"Tiles X\",\n default=2,\n minimum=1,\n decimals=0),\n NumberDef(\"tilesY\",\n label=\"Tiles Y\",\n default=2,\n minimum=1,\n decimals=0),\n\n # Additional settings\n BoolDef(\"convertToScanline\",\n label=\"Convert to Scanline\",\n tooltip=\"Convert the output images to scanline images\",\n default=False),\n BoolDef(\"useReferencedAovs\",\n label=\"Use Referenced AOVs\",\n tooltip=\"Consider the AOVs from referenced scenes as well\",\n default=False),\n\n BoolDef(\"renderSetupIncludeLights\",\n label=\"Render Setup Include Lights\",\n default=self.render_settings.get(\"enable_all_lights\",\n False))\n ]\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_rendersetup.py",
"content": "from openpype.hosts.maya.api import plugin\nfrom openpype.pipeline import CreatorError\n\n\nclass CreateRenderSetup(plugin.MayaCreator):\n \"\"\"Create rendersetup template json data\"\"\"\n\n identifier = \"io.openpype.creators.maya.rendersetup\"\n label = \"Render Setup Preset\"\n family = \"rendersetup\"\n icon = \"tablet\"\n\n def get_pre_create_attr_defs(self):\n # Do not show the \"use_selection\" setting from parent class\n return []\n\n def create(self, subset_name, instance_data, pre_create_data):\n\n existing_instance = None\n for instance in self.create_context.instances:\n if instance.family == self.family:\n existing_instance = instance\n break\n\n if existing_instance:\n raise CreatorError(\"A RenderSetup instance already exists - only \"\n \"one can be configured.\")\n\n super(CreateRenderSetup, self).create(subset_name,\n instance_data,\n pre_create_data)\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_review.py",
"content": "import json\n\nfrom maya import cmds\n\nfrom openpype import AYON_SERVER_ENABLED\nfrom openpype.hosts.maya.api import (\n lib,\n plugin\n)\nfrom openpype.lib import (\n BoolDef,\n NumberDef,\n EnumDef\n)\nfrom openpype.pipeline import CreatedInstance\nfrom openpype.client import get_asset_by_name\n\nTRANSPARENCIES = [\n \"preset\",\n \"simple\",\n \"object sorting\",\n \"weighted average\",\n \"depth peeling\",\n \"alpha cut\"\n]\n\n\nclass CreateReview(plugin.MayaCreator):\n \"\"\"Playblast reviewable\"\"\"\n\n identifier = \"io.openpype.creators.maya.review\"\n label = \"Review\"\n family = \"review\"\n icon = \"video-camera\"\n\n useMayaTimeline = True\n panZoom = False\n\n # Overriding \"create\" method to prefill values from settings.\n def create(self, subset_name, instance_data, pre_create_data):\n\n members = list()\n if pre_create_data.get(\"use_selection\"):\n members = cmds.ls(selection=True)\n\n project_name = self.project_name\n if AYON_SERVER_ENABLED:\n asset_name = instance_data[\"folderPath\"]\n else:\n asset_name = instance_data[\"asset\"]\n asset_doc = get_asset_by_name(project_name, asset_name)\n task_name = instance_data[\"task\"]\n preset = lib.get_capture_preset(\n task_name,\n asset_doc[\"data\"][\"tasks\"][task_name][\"type\"],\n subset_name,\n self.project_settings,\n self.log\n )\n self.log.debug(\n \"Using preset: {}\".format(\n json.dumps(preset, indent=4, sort_keys=True)\n )\n )\n\n with lib.undo_chunk():\n instance_node = cmds.sets(members, name=subset_name)\n instance_data[\"instance_node\"] = instance_node\n instance = CreatedInstance(\n self.family,\n subset_name,\n instance_data,\n self)\n\n creator_attribute_defs_by_key = {\n x.key: x for x in instance.creator_attribute_defs\n }\n mapping = {\n \"review_width\": preset[\"Resolution\"][\"width\"],\n \"review_height\": preset[\"Resolution\"][\"height\"],\n \"isolate\": preset[\"Generic\"][\"isolate_view\"],\n \"imagePlane\": preset[\"Viewport Options\"][\"imagePlane\"],\n \"panZoom\": preset[\"Generic\"][\"pan_zoom\"]\n }\n for key, value in mapping.items():\n creator_attribute_defs_by_key[key].default = value\n\n self._add_instance_to_context(instance)\n\n self.imprint_instance_node(instance_node,\n data=instance.data_to_store())\n return instance\n\n def get_instance_attr_defs(self):\n\n defs = lib.collect_animation_defs()\n\n # Option for using Maya or asset frame range in settings.\n if not self.useMayaTimeline:\n # Update the defaults to be the asset frame range\n frame_range = lib.get_frame_range()\n defs_by_key = {attr_def.key: attr_def for attr_def in defs}\n for key, value in frame_range.items():\n if key not in defs_by_key:\n raise RuntimeError(\"Attribute definition not found to be \"\n \"updated for key: {}\".format(key))\n attr_def = defs_by_key[key]\n attr_def.default = value\n\n defs.extend([\n NumberDef(\"review_width\",\n label=\"Review width\",\n tooltip=\"A value of zero will use the asset resolution.\",\n decimals=0,\n minimum=0,\n default=0),\n NumberDef(\"review_height\",\n label=\"Review height\",\n tooltip=\"A value of zero will use the asset resolution.\",\n decimals=0,\n minimum=0,\n default=0),\n BoolDef(\"keepImages\",\n label=\"Keep Images\",\n tooltip=\"Whether to also publish along the image sequence \"\n \"next to the video reviewable.\",\n default=False),\n BoolDef(\"isolate\",\n label=\"Isolate render members of instance\",\n tooltip=\"When enabled only the members of the instance \"\n \"will be included in the playblast review.\",\n default=False),\n BoolDef(\"imagePlane\",\n label=\"Show Image Plane\",\n default=True),\n EnumDef(\"transparency\",\n label=\"Transparency\",\n items=TRANSPARENCIES),\n BoolDef(\"panZoom\",\n label=\"Enable camera pan/zoom\",\n default=True),\n EnumDef(\"displayLights\",\n label=\"Display Lights\",\n items=lib.DISPLAY_LIGHTS_ENUM),\n ])\n\n return defs\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_rig.py",
"content": "from maya import cmds\n\nfrom openpype.hosts.maya.api import plugin\n\n\nclass CreateRig(plugin.MayaCreator):\n \"\"\"Artist-friendly rig with controls to direct motion\"\"\"\n\n identifier = \"io.openpype.creators.maya.rig\"\n label = \"Rig\"\n family = \"rig\"\n icon = \"wheelchair\"\n\n def create(self, subset_name, instance_data, pre_create_data):\n\n instance = super(CreateRig, self).create(subset_name,\n instance_data,\n pre_create_data)\n\n instance_node = instance.get(\"instance_node\")\n\n self.log.info(\"Creating Rig instance set up ...\")\n # TODO：change name (_controls_SET -> _rigs_SET)\n controls = cmds.sets(name=subset_name + \"_controls_SET\", empty=True)\n # TODO：change name (_out_SET -> _geo_SET)\n pointcache = cmds.sets(name=subset_name + \"_out_SET\", empty=True)\n skeleton = cmds.sets(\n name=subset_name + \"_skeletonAnim_SET\", empty=True)\n skeleton_mesh = cmds.sets(\n name=subset_name + \"_skeletonMesh_SET\", empty=True)\n cmds.sets([controls, pointcache,\n skeleton, skeleton_mesh], forceElement=instance_node)\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_setdress.py",
"content": "from openpype.hosts.maya.api import plugin\nfrom openpype.lib import BoolDef\n\n\nclass CreateSetDress(plugin.MayaCreator):\n \"\"\"A grouped package of loaded content\"\"\"\n\n identifier = \"io.openpype.creators.maya.setdress\"\n label = \"Set Dress\"\n family = \"setdress\"\n icon = \"cubes\"\n default_variants = [\"Main\", \"Anim\"]\n\n def get_instance_attr_defs(self):\n return [\n BoolDef(\"exactSetMembersOnly\",\n label=\"Exact Set Members Only\",\n default=True)\n ]\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_unreal_skeletalmesh.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator for Unreal Skeletal Meshes.\"\"\"\nfrom openpype.hosts.maya.api import plugin, lib\nfrom openpype.lib import (\n BoolDef,\n TextDef\n)\n\nfrom maya import cmds # noqa\n\n\nclass CreateUnrealSkeletalMesh(plugin.MayaCreator):\n \"\"\"Unreal Static Meshes with collisions.\"\"\"\n\n identifier = \"io.openpype.creators.maya.unrealskeletalmesh\"\n label = \"Unreal - Skeletal Mesh\"\n family = \"skeletalMesh\"\n icon = \"thumbs-up\"\n dynamic_subset_keys = [\"asset\"]\n\n # Defined in settings\n joint_hints = set()\n\n def apply_settings(self, project_settings):\n \"\"\"Apply project settings to creator\"\"\"\n settings = (\n project_settings[\"maya\"][\"create\"][\"CreateUnrealSkeletalMesh\"]\n )\n self.joint_hints = set(settings.get(\"joint_hints\", []))\n\n def get_dynamic_data(\n self, variant, task_name, asset_doc, project_name, host_name, instance\n ):\n \"\"\"\n The default subset name templates for Unreal include {asset} and thus\n we should pass that along as dynamic data.\n \"\"\"\n dynamic_data = super(CreateUnrealSkeletalMesh, self).get_dynamic_data(\n variant, task_name, asset_doc, project_name, host_name, instance\n )\n dynamic_data[\"asset\"] = asset_doc[\"name\"]\n return dynamic_data\n\n def create(self, subset_name, instance_data, pre_create_data):\n\n with lib.undo_chunk():\n instance = super(CreateUnrealSkeletalMesh, self).create(\n subset_name, instance_data, pre_create_data)\n instance_node = instance.get(\"instance_node\")\n\n # We reorganize the geometry that was originally added into the\n # set into either 'joints_SET' or 'geometry_SET' based on the\n # joint_hints from project settings\n members = cmds.sets(instance_node, query=True) or []\n cmds.sets(clear=instance_node)\n\n geometry_set = cmds.sets(name=\"geometry_SET\", empty=True)\n joints_set = cmds.sets(name=\"joints_SET\", empty=True)\n\n cmds.sets([geometry_set, joints_set], forceElement=instance_node)\n\n for node in members:\n if node in self.joint_hints:\n cmds.sets(node, forceElement=joints_set)\n else:\n cmds.sets(node, forceElement=geometry_set)\n\n def get_instance_attr_defs(self):\n\n defs = lib.collect_animation_defs()\n\n defs.extend([\n BoolDef(\"renderableOnly\",\n label=\"Renderable Only\",\n tooltip=\"Only export renderable visible shapes\",\n default=False),\n BoolDef(\"visibleOnly\",\n label=\"Visible Only\",\n tooltip=\"Only export dag objects visible during \"\n \"frame range\",\n default=False),\n BoolDef(\"includeParentHierarchy\",\n label=\"Include Parent Hierarchy\",\n tooltip=\"Whether to include parent hierarchy of nodes in \"\n \"the publish instance\",\n default=False),\n BoolDef(\"worldSpace\",\n label=\"World-Space Export\",\n default=True),\n BoolDef(\"refresh\",\n label=\"Refresh viewport during export\",\n default=False),\n TextDef(\"attr\",\n label=\"Custom Attributes\",\n default=\"\",\n placeholder=\"attr1, attr2\"),\n TextDef(\"attrPrefix\",\n label=\"Custom Attributes Prefix\",\n placeholder=\"prefix1, prefix2\")\n ])\n\n return defs\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_unreal_staticmesh.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator for Unreal Static Meshes.\"\"\"\nfrom openpype.hosts.maya.api import plugin, lib\nfrom maya import cmds # noqa\n\n\nclass CreateUnrealStaticMesh(plugin.MayaCreator):\n \"\"\"Unreal Static Meshes with collisions.\"\"\"\n\n identifier = \"io.openpype.creators.maya.unrealstaticmesh\"\n label = \"Unreal - Static Mesh\"\n family = \"staticMesh\"\n icon = \"cube\"\n dynamic_subset_keys = [\"asset\"]\n\n # Defined in settings\n collision_prefixes = []\n\n def apply_settings(self, project_settings):\n \"\"\"Apply project settings to creator\"\"\"\n settings = project_settings[\"maya\"][\"create\"][\"CreateUnrealStaticMesh\"]\n self.collision_prefixes = settings[\"collision_prefixes\"]\n\n def get_dynamic_data(\n self, variant, task_name, asset_doc, project_name, host_name, instance\n ):\n \"\"\"\n The default subset name templates for Unreal include {asset} and thus\n we should pass that along as dynamic data.\n \"\"\"\n dynamic_data = super(CreateUnrealStaticMesh, self).get_dynamic_data(\n variant, task_name, asset_doc, project_name, host_name, instance\n )\n dynamic_data[\"asset\"] = asset_doc[\"name\"]\n return dynamic_data\n\n def create(self, subset_name, instance_data, pre_create_data):\n\n with lib.undo_chunk():\n instance = super(CreateUnrealStaticMesh, self).create(\n subset_name, instance_data, pre_create_data)\n instance_node = instance.get(\"instance_node\")\n\n # We reorganize the geometry that was originally added into the\n # set into either 'collision_SET' or 'geometry_SET' based on the\n # collision_prefixes from project settings\n members = cmds.sets(instance_node, query=True)\n cmds.sets(clear=instance_node)\n\n geometry_set = cmds.sets(name=\"geometry_SET\", empty=True)\n collisions_set = cmds.sets(name=\"collisions_SET\", empty=True)\n\n cmds.sets([geometry_set, collisions_set],\n forceElement=instance_node)\n\n members = cmds.ls(members, long=True) or []\n children = cmds.listRelatives(members, allDescendents=True,\n fullPath=True) or []\n transforms = cmds.ls(members + children, type=\"transform\")\n for transform in transforms:\n\n if not cmds.listRelatives(transform,\n type=\"shape\",\n noIntermediate=True):\n # Exclude all transforms that have no direct shapes\n continue\n\n if self.has_collision_prefix(transform):\n cmds.sets(transform, forceElement=collisions_set)\n else:\n cmds.sets(transform, forceElement=geometry_set)\n\n def has_collision_prefix(self, node_path):\n \"\"\"Return whether node name of path matches collision prefix.\n\n If the node name matches the collision prefix we add it to the\n `collisions_SET` instead of the `geometry_SET`.\n\n Args:\n node_path (str): Maya node path.\n\n Returns:\n bool: Whether the node should be considered a collision mesh.\n\n \"\"\"\n node_name = node_path.rsplit(\"|\", 1)[-1]\n for prefix in self.collision_prefixes:\n if node_name.startswith(prefix):\n return True\n return False\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_unreal_yeticache.py",
"content": "from openpype.hosts.maya.api import (\n lib,\n plugin\n)\nfrom openpype.lib import NumberDef\n\n\nclass CreateYetiCache(plugin.MayaCreator):\n \"\"\"Output for procedural plugin nodes of Yeti \"\"\"\n\n identifier = \"io.openpype.creators.maya.unrealyeticache\"\n label = \"Unreal - Yeti Cache\"\n family = \"yeticacheUE\"\n icon = \"pagelines\"\n\n def get_instance_attr_defs(self):\n\n defs = [\n NumberDef(\"preroll\",\n label=\"Preroll\",\n minimum=0,\n default=0,\n decimals=0)\n ]\n\n # Add animation data without step and handles\n defs.extend(lib.collect_animation_defs())\n remove = {\"step\", \"handleStart\", \"handleEnd\"}\n defs = [attr_def for attr_def in defs if attr_def.key not in remove]\n\n # Add samples after frame range\n defs.append(\n NumberDef(\"samples\",\n label=\"Samples\",\n default=3,\n decimals=0)\n )\n\n return defs\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_vrayproxy.py",
"content": "from openpype.hosts.maya.api import (\n plugin,\n lib\n)\nfrom openpype.lib import BoolDef\n\n\nclass CreateVrayProxy(plugin.MayaCreator):\n \"\"\"Alembic pointcache for animated data\"\"\"\n\n identifier = \"io.openpype.creators.maya.vrayproxy\"\n label = \"VRay Proxy\"\n family = \"vrayproxy\"\n icon = \"gears\"\n\n vrmesh = True\n alembic = True\n\n def get_instance_attr_defs(self):\n\n defs = [\n BoolDef(\"animation\",\n label=\"Export Animation\",\n default=False)\n ]\n\n # Add time range attributes but remove some attributes\n # which this instance actually doesn't use\n defs.extend(lib.collect_animation_defs())\n remove = {\"handleStart\", \"handleEnd\", \"step\"}\n defs = [attr_def for attr_def in defs if attr_def.key not in remove]\n\n defs.extend([\n BoolDef(\"vertexColors\",\n label=\"Write vertex colors\",\n tooltip=\"Write vertex colors with the geometry\",\n default=False),\n BoolDef(\"vrmesh\",\n label=\"Export VRayMesh\",\n tooltip=\"Publish a .vrmesh (VRayMesh) file for \"\n \"this VRayProxy\",\n default=self.vrmesh),\n BoolDef(\"alembic\",\n label=\"Export Alembic\",\n tooltip=\"Publish a .abc (Alembic) file for \"\n \"this VRayProxy\",\n default=self.alembic),\n ])\n\n return defs\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_vrayscene.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Create instance of vrayscene.\"\"\"\n\nfrom openpype.hosts.maya.api import (\n lib_rendersettings,\n plugin\n)\nfrom openpype.pipeline import CreatorError\nfrom openpype.lib import BoolDef\n\n\nclass CreateVRayScene(plugin.RenderlayerCreator):\n \"\"\"Create Vray Scene.\"\"\"\n\n identifier = \"io.openpype.creators.maya.vrayscene\"\n\n family = \"vrayscene\"\n label = \"VRay Scene\"\n icon = \"cubes\"\n\n render_settings = {}\n singleton_node_name = \"vraysceneMain\"\n\n @classmethod\n def apply_settings(cls, project_settings):\n cls.render_settings = project_settings[\"maya\"][\"RenderSettings\"]\n\n def create(self, subset_name, instance_data, pre_create_data):\n # Only allow a single render instance to exist\n if self._get_singleton_node():\n raise CreatorError(\"A Render instance already exists - only \"\n \"one can be configured.\")\n\n super(CreateVRayScene, self).create(subset_name,\n instance_data,\n pre_create_data)\n\n # Apply default project render settings on create\n if self.render_settings.get(\"apply_render_settings\"):\n lib_rendersettings.RenderSettings().set_default_renderer_settings()\n\n def get_instance_attr_defs(self):\n \"\"\"Create instance settings.\"\"\"\n\n return [\n BoolDef(\"vraySceneMultipleFiles\",\n label=\"V-Ray Scene Multiple Files\",\n default=False),\n BoolDef(\"exportOnFarm\",\n label=\"Export on farm\",\n default=False)\n ]\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_workfile.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Creator plugin for creating workfiles.\"\"\"\nfrom openpype import AYON_SERVER_ENABLED\nfrom openpype.pipeline import CreatedInstance, AutoCreator\nfrom openpype.client import get_asset_by_name, get_asset_name_identifier\nfrom openpype.hosts.maya.api import plugin\nfrom maya import cmds\n\n\nclass CreateWorkfile(plugin.MayaCreatorBase, AutoCreator):\n \"\"\"Workfile auto-creator.\"\"\"\n identifier = \"io.openpype.creators.maya.workfile\"\n label = \"Workfile\"\n family = \"workfile\"\n icon = \"fa5.file\"\n\n default_variant = \"Main\"\n\n def create(self):\n\n variant = self.default_variant\n current_instance = next(\n (\n instance for instance in self.create_context.instances\n if instance.creator_identifier == self.identifier\n ), None)\n\n project_name = self.project_name\n asset_name = self.create_context.get_current_asset_name()\n task_name = self.create_context.get_current_task_name()\n host_name = self.create_context.host_name\n\n if current_instance is None:\n current_instance_asset = None\n elif AYON_SERVER_ENABLED:\n current_instance_asset = current_instance[\"folderPath\"]\n else:\n current_instance_asset = current_instance[\"asset\"]\n\n if current_instance is None:\n asset_doc = get_asset_by_name(project_name, asset_name)\n subset_name = self.get_subset_name(\n variant, task_name, asset_doc, project_name, host_name\n )\n data = {\n \"task\": task_name,\n \"variant\": variant\n }\n if AYON_SERVER_ENABLED:\n data[\"folderPath\"] = asset_name\n else:\n data[\"asset\"] = asset_name\n\n data.update(\n self.get_dynamic_data(\n variant, task_name, asset_doc,\n project_name, host_name, current_instance)\n )\n self.log.info(\"Auto-creating workfile instance...\")\n current_instance = CreatedInstance(\n self.family, subset_name, data, self\n )\n self._add_instance_to_context(current_instance)\n elif (\n current_instance_asset != asset_name\n or current_instance[\"task\"] != task_name\n ):\n # Update instance context if is not the same\n asset_doc = get_asset_by_name(project_name, asset_name)\n subset_name = self.get_subset_name(\n variant, task_name, asset_doc, project_name, host_name\n )\n asset_name = get_asset_name_identifier(asset_doc)\n\n if AYON_SERVER_ENABLED:\n current_instance[\"folderPath\"] = asset_name\n else:\n current_instance[\"asset\"] = asset_name\n current_instance[\"task\"] = task_name\n current_instance[\"subset\"] = subset_name\n\n def collect_instances(self):\n self.cache_subsets(self.collection_shared_data)\n cached_subsets = self.collection_shared_data[\"maya_cached_subsets\"]\n for node in cached_subsets.get(self.identifier, []):\n node_data = self.read_instance_node(node)\n\n created_instance = CreatedInstance.from_existing(node_data, self)\n self._add_instance_to_context(created_instance)\n\n def update_instances(self, update_list):\n for created_inst, _changes in update_list:\n data = created_inst.data_to_store()\n node = data.get(\"instance_node\")\n if not node:\n node = self.create_node()\n created_inst[\"instance_node\"] = node\n data = created_inst.data_to_store()\n\n self.imprint_instance_node(node, data)\n\n def create_node(self):\n node = cmds.sets(empty=True, name=\"workfileMain\")\n cmds.setAttr(node + \".hiddenInOutliner\", True)\n return node\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_xgen.py",
"content": "from openpype.hosts.maya.api import plugin\n\n\nclass CreateXgen(plugin.MayaCreator):\n \"\"\"Xgen\"\"\"\n\n identifier = \"io.openpype.creators.maya.xgen\"\n label = \"Xgen\"\n family = \"xgen\"\n icon = \"pagelines\"\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_yeti_cache.py",
"content": "from openpype.hosts.maya.api import (\n lib,\n plugin\n)\nfrom openpype.lib import NumberDef\n\n\nclass CreateYetiCache(plugin.MayaCreator):\n \"\"\"Output for procedural plugin nodes of Yeti \"\"\"\n\n identifier = \"io.openpype.creators.maya.yeticache\"\n label = \"Yeti Cache\"\n family = \"yeticache\"\n icon = \"pagelines\"\n\n def get_instance_attr_defs(self):\n\n defs = [\n NumberDef(\"preroll\",\n label=\"Preroll\",\n minimum=0,\n default=0,\n decimals=0)\n ]\n\n # Add animation data without step and handles\n defs.extend(lib.collect_animation_defs())\n remove = {\"step\", \"handleStart\", \"handleEnd\"}\n defs = [attr_def for attr_def in defs if attr_def.key not in remove]\n\n # Add samples after frame range\n defs.append(\n NumberDef(\"samples\",\n label=\"Samples\",\n default=3,\n decimals=0)\n )\n\n return defs\n"
},
{
"path": "openpype/hosts/maya/plugins/create/create_yeti_rig.py",
"content": "from maya import cmds\n\nfrom openpype.hosts.maya.api import (\n lib,\n plugin\n)\n\n\nclass CreateYetiRig(plugin.MayaCreator):\n \"\"\"Output for procedural plugin nodes ( Yeti / XGen / etc)\"\"\"\n\n identifier = \"io.openpype.creators.maya.yetirig\"\n label = \"Yeti Rig\"\n family = \"yetiRig\"\n icon = \"usb\"\n\n def create(self, subset_name, instance_data, pre_create_data):\n\n with lib.undo_chunk():\n instance = super(CreateYetiRig, self).create(subset_name,\n instance_data,\n pre_create_data)\n instance_node = instance.get(\"instance_node\")\n\n self.log.info(\"Creating Rig instance set up ...\")\n input_meshes = cmds.sets(name=\"input_SET\", empty=True)\n cmds.sets(input_meshes, forceElement=instance_node)\n"
},
{
"path": "openpype/hosts/maya/plugins/inventory/connect_geometry.py",
"content": "from maya import cmds\n\nfrom openpype.pipeline import InventoryAction, get_representation_context\nfrom openpype.hosts.maya.api.lib import get_id\n\n\nclass ConnectGeometry(InventoryAction):\n \"\"\"Connect geometries within containers.\n\n Source container will connect to the target containers, by searching for\n matching geometry IDs (cbid).\n Source containers are of family; \"animation\" and \"pointcache\".\n The connection with be done with a live world space blendshape.\n \"\"\"\n\n label = \"Connect Geometry\"\n icon = \"link\"\n color = \"white\"\n\n def process(self, containers):\n # Validate selection is more than 1.\n message = (\n \"Only 1 container selected. 2+ containers needed for this action.\"\n )\n if len(containers) == 1:\n self.display_warning(message)\n return\n\n # Categorize containers by family.\n containers_by_family = {}\n for container in containers:\n family = get_representation_context(\n container[\"representation\"]\n )[\"subset\"][\"data\"][\"family\"]\n try:\n containers_by_family[family].append(container)\n except KeyError:\n containers_by_family[family] = [container]\n\n # Validate to only 1 source container.\n source_containers = containers_by_family.get(\"animation\", [])\n source_containers += containers_by_family.get(\"pointcache\", [])\n source_container_namespaces = [\n x[\"namespace\"] for x in source_containers\n ]\n message = (\n \"{} animation containers selected:\\n\\n{}\\n\\nOnly select 1 of type \"\n \"\\\"animation\\\" or \\\"pointcache\\\".\".format(\n len(source_containers), source_container_namespaces\n )\n )\n if len(source_containers) != 1:\n self.display_warning(message)\n return\n\n source_object = source_containers[0][\"objectName\"]\n\n # Collect matching geometry transforms based cbId attribute.\n target_containers = []\n for family, containers in containers_by_family.items():\n if family in [\"animation\", \"pointcache\"]:\n continue\n\n target_containers.extend(containers)\n\n source_data = self.get_container_data(source_object)\n matches = []\n node_types = set()\n for target_container in target_containers:\n target_data = self.get_container_data(\n target_container[\"objectName\"]\n )\n node_types.update(target_data[\"node_types\"])\n for id, transform in target_data[\"ids\"].items():\n source_match = source_data[\"ids\"].get(id)\n if source_match:\n matches.append([source_match, transform])\n\n # Message user about what is about to happen.\n if not matches:\n self.display_warning(\"No matching geometries found.\")\n return\n\n message = \"Connecting geometries:\\n\\n\"\n for match in matches:\n message += \"{} > {}\\n\".format(match[0], match[1])\n\n choice = self.display_warning(message, show_cancel=True)\n if choice is False:\n return\n\n # Setup live worldspace blendshape connection.\n for source, target in matches:\n blendshape = cmds.blendShape(source, target)[0]\n cmds.setAttr(blendshape + \".origin\", 0)\n cmds.setAttr(blendshape + \".\" + target.split(\":\")[-1], 1)\n\n # Update Xgen if in any of the containers.\n if \"xgmPalette\" in node_types:\n cmds.xgmPreview()\n\n def get_container_data(self, container):\n \"\"\"Collects data about the container nodes.\n\n Args:\n container (dict): Container instance.\n\n Returns:\n data (dict):\n \"node_types\": All node types in container nodes.\n \"ids\": If the node is a mesh, we collect its parent transform\n id.\n \"\"\"\n data = {\"node_types\": set(), \"ids\": {}}\n ref_node = cmds.sets(container, query=True, nodesOnly=True)[0]\n for node in cmds.referenceQuery(ref_node, nodes=True):\n node_type = cmds.nodeType(node)\n data[\"node_types\"].add(node_type)\n\n # Only interested in mesh transforms for connecting geometry with\n # blendshape.\n if node_type != \"mesh\":\n continue\n\n transform = cmds.listRelatives(node, parent=True)[0]\n data[\"ids\"][get_id(transform)] = transform\n\n return data\n\n def display_warning(self, message, show_cancel=False):\n \"\"\"Show feedback to user.\n\n Returns:\n bool\n \"\"\"\n\n from qtpy import QtWidgets\n\n accept = QtWidgets.QMessageBox.Ok\n if show_cancel:\n buttons = accept | QtWidgets.QMessageBox.Cancel\n else:\n buttons = accept\n\n state = QtWidgets.QMessageBox.warning(\n None,\n \"\",\n message,\n buttons=buttons,\n defaultButton=accept\n )\n\n return state == accept\n"
},
{
"path": "openpype/hosts/maya/plugins/inventory/connect_xgen.py",
"content": "from maya import cmds\nimport xgenm\n\nfrom openpype.pipeline import (\n InventoryAction, get_representation_context, get_representation_path\n)\n\n\nclass ConnectXgen(InventoryAction):\n \"\"\"Connect Xgen with an animation or pointcache.\n \"\"\"\n\n label = \"Connect Xgen\"\n icon = \"link\"\n color = \"white\"\n\n def process(self, containers):\n # Validate selection is more than 1.\n message = (\n \"Only 1 container selected. 2+ containers needed for this action.\"\n )\n if len(containers) == 1:\n self.display_warning(message)\n return\n\n # Categorize containers by family.\n containers_by_family = {}\n for container in containers:\n family = get_representation_context(\n container[\"representation\"]\n )[\"subset\"][\"data\"][\"family\"]\n try:\n containers_by_family[family].append(container)\n except KeyError:\n containers_by_family[family] = [container]\n\n # Validate to only 1 source container.\n source_containers = containers_by_family.get(\"animation\", [])\n source_containers += containers_by_family.get(\"pointcache\", [])\n source_container_namespaces = [\n x[\"namespace\"] for x in source_containers\n ]\n message = (\n \"{} animation containers selected:\\n\\n{}\\n\\nOnly select 1 of type \"\n \"\\\"animation\\\" or \\\"pointcache\\\".\".format(\n len(source_containers), source_container_namespaces\n )\n )\n if len(source_containers) != 1:\n self.display_warning(message)\n return\n\n source_container = source_containers[0]\n source_object = source_container[\"objectName\"]\n\n # Validate source representation is an alembic.\n source_path = get_representation_path(\n get_representation_context(\n source_container[\"representation\"]\n )[\"representation\"]\n ).replace(\"\\\\\", \"/\")\n message = \"Animation container \\\"{}\\\" is not an alembic:\\n{}\".format(\n source_container[\"namespace\"], source_path\n )\n if not source_path.endswith(\".abc\"):\n self.display_warning(message)\n return\n\n # Target containers.\n target_containers = []\n for family, containers in containers_by_family.items():\n if family in [\"animation\", \"pointcache\"]:\n continue\n\n target_containers.extend(containers)\n\n # Inform user of connections from source representation to target\n # descriptions.\n descriptions_data = []\n connections_msg = \"\"\n for target_container in target_containers:\n reference_node = cmds.sets(\n target_container[\"objectName\"], query=True\n )[0]\n palettes = cmds.ls(\n cmds.referenceQuery(reference_node, nodes=True),\n type=\"xgmPalette\"\n )\n for palette in palettes:\n for description in xgenm.descriptions(palette):\n descriptions_data.append([palette, description])\n connections_msg += \"\\n{}/{}\".format(palette, description)\n\n message = \"Connecting \\\"{}\\\" to:\\n\".format(\n source_container[\"namespace\"]\n )\n message += connections_msg\n choice = self.display_warning(message, show_cancel=True)\n if choice is False:\n return\n\n # Recreate \"xgenContainers\" attribute to reset.\n compound_name = \"xgenContainers\"\n attr = \"{}.{}\".format(source_object, compound_name)\n if cmds.objExists(attr):\n cmds.deleteAttr(attr)\n\n cmds.addAttr(\n source_object,\n longName=compound_name,\n attributeType=\"compound\",\n numberOfChildren=1,\n multi=True\n )\n\n # Connect target containers.\n for target_container in target_containers:\n cmds.addAttr(\n source_object,\n longName=\"container\",\n attributeType=\"message\",\n parent=compound_name\n )\n index = target_containers.index(target_container)\n cmds.connectAttr(\n target_container[\"objectName\"] + \".message\",\n source_object + \".{}[{}].container\".format(\n compound_name, index\n )\n )\n\n # Setup cache on Xgen\n object = \"SplinePrimitive\"\n for palette, description in descriptions_data:\n xgenm.setAttr(\"useCache\", \"true\", palette, description, object)\n xgenm.setAttr(\"liveMode\", \"false\", palette, description, object)\n xgenm.setAttr(\n \"cacheFileName\", source_path, palette, description, object\n )\n\n # Refresh UI and viewport.\n de = xgenm.xgGlobal.DescriptionEditor\n de.refresh(\"Full\")\n\n def display_warning(self, message, show_cancel=False):\n \"\"\"Show feedback to user.\n\n Returns:\n bool\n \"\"\"\n\n from qtpy import QtWidgets\n\n accept = QtWidgets.QMessageBox.Ok\n if show_cancel:\n buttons = accept | QtWidgets.QMessageBox.Cancel\n else:\n buttons = accept\n\n state = QtWidgets.QMessageBox.warning(\n None,\n \"\",\n message,\n buttons=buttons,\n defaultButton=accept\n )\n\n return state == accept\n"
},
{
"path": "openpype/hosts/maya/plugins/inventory/connect_yeti_rig.py",
"content": "import os\nimport json\nfrom collections import defaultdict\n\nfrom maya import cmds\n\nfrom openpype.pipeline import (\n InventoryAction, get_representation_context, get_representation_path\n)\nfrom openpype.hosts.maya.api.lib import get_container_members, get_id\n\n\nclass ConnectYetiRig(InventoryAction):\n \"\"\"Connect Yeti Rig with an animation or pointcache.\"\"\"\n\n label = \"Connect Yeti Rig\"\n icon = \"link\"\n color = \"white\"\n\n def process(self, containers):\n # Validate selection is more than 1.\n message = (\n \"Only 1 container selected. 2+ containers needed for this action.\"\n )\n if len(containers) == 1:\n self.display_warning(message)\n return\n\n # Categorize containers by family.\n containers_by_family = defaultdict(list)\n for container in containers:\n family = get_representation_context(\n container[\"representation\"]\n )[\"subset\"][\"data\"][\"family\"]\n containers_by_family[family].append(container)\n\n # Validate to only 1 source container.\n source_containers = containers_by_family.get(\"animation\", [])\n source_containers += containers_by_family.get(\"pointcache\", [])\n source_container_namespaces = [\n x[\"namespace\"] for x in source_containers\n ]\n message = (\n \"{} animation containers selected:\\n\\n{}\\n\\nOnly select 1 of type \"\n \"\\\"animation\\\" or \\\"pointcache\\\".\".format(\n len(source_containers), source_container_namespaces\n )\n )\n if len(source_containers) != 1:\n self.display_warning(message)\n return\n\n source_container = source_containers[0]\n source_ids = self.nodes_by_id(source_container)\n\n # Target containers.\n target_ids = {}\n inputs = []\n\n yeti_rig_containers = containers_by_family.get(\"yetiRig\")\n if not yeti_rig_containers:\n self.display_warning(\n \"Select at least one yetiRig container\"\n )\n return\n\n for container in yeti_rig_containers:\n target_ids.update(self.nodes_by_id(container))\n\n maya_file = get_representation_path(\n get_representation_context(\n container[\"representation\"]\n )[\"representation\"]\n )\n _, ext = os.path.splitext(maya_file)\n settings_file = maya_file.replace(ext, \".rigsettings\")\n if not os.path.exists(settings_file):\n continue\n\n with open(settings_file) as f:\n inputs.extend(json.load(f)[\"inputs\"])\n\n # Compare loaded connections to scene.\n for input in inputs:\n source_node = source_ids.get(input[\"sourceID\"])\n target_node = target_ids.get(input[\"destinationID\"])\n\n if not source_node or not target_node:\n self.log.debug(\n \"Could not find nodes for input:\\n\" +\n json.dumps(input, indent=4, sort_keys=True)\n )\n continue\n source_attr, target_attr = input[\"connections\"]\n\n if not cmds.attributeQuery(\n source_attr, node=source_node, exists=True\n ):\n self.log.debug(\n \"Could not find attribute {} on node {} for \"\n \"input:\\n{}\".format(\n source_attr,\n source_node,\n json.dumps(input, indent=4, sort_keys=True)\n )\n )\n continue\n\n if not cmds.attributeQuery(\n target_attr, node=target_node, exists=True\n ):\n self.log.debug(\n \"Could not find attribute {} on node {} for \"\n \"input:\\n{}\".format(\n target_attr,\n target_node,\n json.dumps(input, indent=4, sort_keys=True)\n )\n )\n continue\n\n source_plug = \"{}.{}\".format(\n source_node, source_attr\n )\n target_plug = \"{}.{}\".format(\n target_node, target_attr\n )\n if cmds.isConnected(\n source_plug, target_plug, ignoreUnitConversion=True\n ):\n self.log.debug(\n \"Connection already exists: {} -> {}\".format(\n source_plug, target_plug\n )\n )\n continue\n\n cmds.connectAttr(source_plug, target_plug, force=True)\n self.log.debug(\n \"Connected attributes: {} -> {}\".format(\n source_plug, target_plug\n )\n )\n\n def nodes_by_id(self, container):\n ids = {}\n for member in get_container_members(container):\n id = get_id(member)\n if not id:\n continue\n ids[id] = member\n\n return ids\n\n def display_warning(self, message, show_cancel=False):\n \"\"\"Show feedback to user.\n\n Returns:\n bool\n \"\"\"\n\n from qtpy import QtWidgets\n\n accept = QtWidgets.QMessageBox.Ok\n if show_cancel:\n buttons = accept | QtWidgets.QMessageBox.Cancel\n else:\n buttons = accept\n\n state = QtWidgets.QMessageBox.warning(\n None,\n \"\",\n message,\n buttons=buttons,\n defaultButton=accept\n )\n\n return state == accept\n"
},
{
"path": "openpype/hosts/maya/plugins/inventory/import_modelrender.py",
"content": "import re\nimport json\n\nfrom openpype.client import (\n get_representation_by_id,\n get_representations\n)\nfrom openpype.pipeline import (\n InventoryAction,\n get_representation_context,\n get_current_project_name,\n)\nfrom openpype.hosts.maya.api.lib import (\n maintained_selection,\n apply_shaders\n)\n\n\nclass ImportModelRender(InventoryAction):\n\n label = \"Import Model Render Sets\"\n icon = \"industry\"\n color = \"#55DDAA\"\n\n scene_type_regex = \"meta.render.m[ab]\"\n look_data_type = \"meta.render.json\"\n\n @staticmethod\n def is_compatible(container):\n return (\n container.get(\"loader\") == \"ReferenceLoader\"\n and container.get(\"name\", \"\").startswith(\"model\")\n )\n\n def process(self, containers):\n from maya import cmds # noqa: F401\n\n project_name = get_current_project_name()\n for container in containers:\n con_name = container[\"objectName\"]\n nodes = []\n for n in cmds.sets(con_name, query=True, nodesOnly=True) or []:\n if cmds.nodeType(n) == \"reference\":\n nodes += cmds.referenceQuery(n, nodes=True)\n else:\n nodes.append(n)\n\n repr_doc = get_representation_by_id(\n project_name, container[\"representation\"], fields=[\"parent\"]\n )\n version_id = repr_doc[\"parent\"]\n\n print(\"Importing render sets for model %r\" % con_name)\n self.assign_model_render_by_version(nodes, version_id)\n\n def assign_model_render_by_version(self, nodes, version_id):\n \"\"\"Assign nodes a specific published model render data version by id.\n\n This assumes the nodes correspond with the asset.\n\n Args:\n nodes(list): nodes to assign render data to\n version_id (bson.ObjectId): database id of the version of model\n\n Returns:\n None\n \"\"\"\n\n from maya import cmds # noqa: F401\n\n project_name = get_current_project_name()\n repre_docs = get_representations(\n project_name, version_ids=[version_id], fields=[\"_id\", \"name\"]\n )\n # Get representations of shader file and relationships\n json_repre = None\n look_repres = []\n scene_type_regex = re.compile(self.scene_type_regex)\n for repre_doc in repre_docs:\n repre_name = repre_doc[\"name\"]\n if repre_name == self.look_data_type:\n json_repre = repre_doc\n continue\n\n if scene_type_regex.fullmatch(repre_name):\n look_repres.append(repre_doc)\n\n look_repre = look_repres[0] if look_repres else None\n # QUESTION shouldn't be json representation validated too?\n if not look_repre:\n print(\"No model render sets for this model version..\")\n return\n\n context = get_representation_context(look_repre[\"_id\"])\n maya_file = self.filepath_from_context(context)\n\n context = get_representation_context(json_repre[\"_id\"])\n json_file = self.filepath_from_context(context)\n\n # Import the look file\n with maintained_selection():\n shader_nodes = cmds.file(maya_file,\n i=True, # import\n returnNewNodes=True)\n # imprint context data\n\n # Load relationships\n shader_relation = json_file\n with open(shader_relation, \"r\") as f:\n relationships = json.load(f)\n\n # Assign relationships\n apply_shaders(relationships, shader_nodes, nodes)\n"
},
{
"path": "openpype/hosts/maya/plugins/inventory/import_reference.py",
"content": "from maya import cmds\n\nfrom openpype.pipeline import InventoryAction\nfrom openpype.hosts.maya.api.lib import get_reference_node\n\n\nclass ImportReference(InventoryAction):\n \"\"\"Imports selected reference to inside of the file.\"\"\"\n\n label = \"Import Reference\"\n icon = \"download\"\n color = \"#d8d8d8\"\n\n def process(self, containers):\n for container in containers:\n if container[\"loader\"] != \"ReferenceLoader\":\n print(\"Not a reference, skipping\")\n continue\n\n node = container[\"objectName\"]\n members = cmds.sets(node, query=True, nodesOnly=True)\n ref_node = get_reference_node(members)\n\n ref_file = cmds.referenceQuery(ref_node, f=True)\n cmds.file(ref_file, importReference=True)\n\n return True # return anything to trigger model refresh\n"
},
{
"path": "openpype/hosts/maya/plugins/inventory/rig_recreate_animation_instance.py",
"content": "from openpype.pipeline import (\n InventoryAction,\n get_representation_context\n)\nfrom openpype.hosts.maya.api.lib import (\n create_rig_animation_instance,\n get_container_members,\n)\n\n\nclass RecreateRigAnimationInstance(InventoryAction):\n \"\"\"Recreate animation publish instance for loaded rigs\"\"\"\n\n label = \"Recreate rig animation instance\"\n icon = \"wrench\"\n color = \"#888888\"\n\n @staticmethod\n def is_compatible(container):\n return (\n container.get(\"loader\") == \"ReferenceLoader\"\n and container.get(\"name\", \"\").startswith(\"rig\")\n )\n\n def process(self, containers):\n\n for container in containers:\n # todo: delete an existing entry if it exist or skip creation\n\n namespace = container[\"namespace\"]\n representation_id = container[\"representation\"]\n context = get_representation_context(representation_id)\n nodes = get_container_members(container)\n\n create_rig_animation_instance(nodes, context, namespace)\n"
},
{
"path": "openpype/hosts/maya/plugins/inventory/select_containers.py",
"content": "from maya import cmds\n\nfrom openpype.pipeline import InventoryAction, registered_host\nfrom openpype.hosts.maya.api.lib import get_container_members\n\n\nclass SelectInScene(InventoryAction):\n \"\"\"Select nodes in the scene from selected containers in scene inventory\"\"\"\n\n label = \"Select in scene\"\n icon = \"search\"\n color = \"#888888\"\n order = 99\n\n def process(self, containers):\n\n all_members = []\n for container in containers:\n members = get_container_members(container)\n all_members.extend(members)\n cmds.select(all_members, replace=True, noExpand=True)\n\n\nclass HighlightBySceneSelection(InventoryAction):\n \"\"\"Select containers in scene inventory from the current scene selection\"\"\"\n\n label = \"Highlight by scene selection\"\n icon = \"search\"\n color = \"#888888\"\n order = 100\n\n def process(self, containers):\n\n selection = set(cmds.ls(selection=True, long=True, objectsOnly=True))\n host = registered_host()\n\n to_select = []\n for container in host.get_containers():\n members = get_container_members(container)\n if any(member in selection for member in members):\n to_select.append(container[\"objectName\"])\n\n return {\n \"objectNames\": to_select,\n \"options\": {\"clear\": True}\n }\n"
},
{
"path": "openpype/hosts/maya/plugins/load/_load_animation.py",
"content": "import openpype.hosts.maya.api.plugin\nimport maya.cmds as cmds\n\n\ndef _process_reference(file_url, name, namespace, options):\n \"\"\"Load files by referencing scene in Maya.\n\n Args:\n file_url (str): fileapth of the objects to be loaded\n name (str): subset name\n namespace (str): namespace\n options (dict): dict of storing the param\n\n Returns:\n list: list of object nodes\n \"\"\"\n from openpype.hosts.maya.api.lib import unique_namespace\n # Get name from asset being loaded\n # Assuming name is subset name from the animation, we split the number\n # suffix from the name to ensure the namespace is unique\n name = name.split(\"_\")[0]\n ext = file_url.split(\".\")[-1]\n namespace = unique_namespace(\n \"{}_\".format(name),\n format=\"%03d\",\n suffix=\"_{}\".format(ext)\n )\n\n attach_to_root = options.get(\"attach_to_root\", True)\n group_name = options[\"group_name\"]\n\n # no group shall be created\n if not attach_to_root:\n group_name = namespace\n\n nodes = cmds.file(file_url,\n namespace=namespace,\n sharedReferenceFile=False,\n groupReference=attach_to_root,\n groupName=group_name,\n reference=True,\n returnNewNodes=True)\n return nodes\n\n\nclass AbcLoader(openpype.hosts.maya.api.plugin.ReferenceLoader):\n \"\"\"Loader to reference an Alembic file\"\"\"\n\n families = [\"animation\",\n \"camera\",\n \"pointcache\"]\n representations = [\"abc\"]\n\n label = \"Reference animation\"\n order = -10\n icon = \"code-fork\"\n color = \"orange\"\n\n def process_reference(self, context, name, namespace, options):\n\n cmds.loadPlugin(\"AbcImport.mll\", quiet=True)\n # hero_001 (abc)\n # asset_counter{optional}\n path = self.filepath_from_context(context)\n file_url = self.prepare_root_value(path,\n context[\"project\"][\"name\"])\n\n nodes = _process_reference(file_url, name, namespace, options)\n # load colorbleed ID attribute\n self[:] = nodes\n\n return nodes\n\n\nclass FbxLoader(openpype.hosts.maya.api.plugin.ReferenceLoader):\n \"\"\"Loader to reference an Fbx files\"\"\"\n\n families = [\"animation\",\n \"camera\"]\n representations = [\"fbx\"]\n\n label = \"Reference animation\"\n order = -10\n icon = \"code-fork\"\n color = \"orange\"\n\n def process_reference(self, context, name, namespace, options):\n\n cmds.loadPlugin(\"fbx4maya.mll\", quiet=True)\n\n path = self.filepath_from_context(context)\n file_url = self.prepare_root_value(path,\n context[\"project\"][\"name\"])\n\n nodes = _process_reference(file_url, name, namespace, options)\n\n self[:] = nodes\n\n return nodes\n"
},
{
"path": "openpype/hosts/maya/plugins/load/actions.py",
"content": "\"\"\"A module containing generic loader actions that will display in the Loader.\n\n\"\"\"\nimport qargparse\nfrom openpype.pipeline import load\nfrom openpype.hosts.maya.api.lib import (\n maintained_selection,\n get_custom_namespace\n)\nimport openpype.hosts.maya.api.plugin\n\n\nclass SetFrameRangeLoader(load.LoaderPlugin):\n \"\"\"Set frame range excluding pre- and post-handles\"\"\"\n\n families = [\"animation\",\n \"camera\",\n \"proxyAbc\",\n \"pointcache\"]\n representations = [\"abc\"]\n\n label = \"Set frame range\"\n order = 11\n icon = \"clock-o\"\n color = \"white\"\n\n def load(self, context, name, namespace, data):\n\n import maya.cmds as cmds\n\n version = context['version']\n version_data = version.get(\"data\", {})\n\n start = version_data.get(\"frameStart\", None)\n end = version_data.get(\"frameEnd\", None)\n\n if start is None or end is None:\n print(\"Skipping setting frame range because start or \"\n \"end frame data is missing..\")\n return\n\n cmds.playbackOptions(minTime=start,\n maxTime=end,\n animationStartTime=start,\n animationEndTime=end)\n\n\nclass SetFrameRangeWithHandlesLoader(load.LoaderPlugin):\n \"\"\"Set frame range including pre- and post-handles\"\"\"\n\n families = [\"animation\",\n \"camera\",\n \"proxyAbc\",\n \"pointcache\"]\n representations = [\"abc\"]\n\n label = \"Set frame range (with handles)\"\n order = 12\n icon = \"clock-o\"\n color = \"white\"\n\n def load(self, context, name, namespace, data):\n\n import maya.cmds as cmds\n\n version = context['version']\n version_data = version.get(\"data\", {})\n\n start = version_data.get(\"frameStart\", None)\n end = version_data.get(\"frameEnd\", None)\n\n if start is None or end is None:\n print(\"Skipping setting frame range because start or \"\n \"end frame data is missing..\")\n return\n\n # Include handles\n start -= version_data.get(\"handleStart\", 0)\n end += version_data.get(\"handleEnd\", 0)\n\n cmds.playbackOptions(minTime=start,\n maxTime=end,\n animationStartTime=start,\n animationEndTime=end)\n\n\nclass ImportMayaLoader(openpype.hosts.maya.api.plugin.Loader):\n \"\"\"Import action for Maya (unmanaged)\n\n Warning:\n The loaded content will be unmanaged and is *not* visible in the\n scene inventory. It's purely intended to merge content into your scene\n so you could also use it as a new base.\n\n \"\"\"\n representations = [\"ma\", \"mb\", \"obj\"]\n families = [\n \"model\",\n \"pointcache\",\n \"proxyAbc\",\n \"animation\",\n \"mayaAscii\",\n \"mayaScene\",\n \"setdress\",\n \"layout\",\n \"camera\",\n \"rig\",\n \"camerarig\",\n \"staticMesh\",\n \"workfile\"\n ]\n\n label = \"Import\"\n order = 10\n icon = \"arrow-circle-down\"\n color = \"#775555\"\n\n options = [\n qargparse.Boolean(\n \"clean_import\",\n label=\"Clean import\",\n default=False,\n help=\"Should all occurrences of cbId be purged?\"\n )\n ]\n\n def load(self, context, name=None, namespace=None, data=None):\n import maya.cmds as cmds\n\n choice = self.display_warning()\n if choice is False:\n return\n\n custom_group_name, custom_namespace, options = \\\n self.get_custom_namespace_and_group(context, data,\n \"import_loader\")\n\n namespace = get_custom_namespace(custom_namespace)\n\n if not options.get(\"attach_to_root\", True):\n custom_group_name = namespace\n\n path = self.filepath_from_context(context)\n with maintained_selection():\n nodes = cmds.file(path,\n i=True,\n preserveReferences=True,\n namespace=namespace,\n returnNewNodes=True,\n groupReference=options.get(\"attach_to_root\",\n True),\n groupName=custom_group_name)\n\n if data.get(\"clean_import\", False):\n remove_attributes = [\"cbId\"]\n for node in nodes:\n for attr in remove_attributes:\n if cmds.attributeQuery(attr, node=node, exists=True):\n full_attr = \"{}.{}\".format(node, attr)\n print(\"Removing {}\".format(full_attr))\n cmds.deleteAttr(full_attr)\n\n # We do not containerize imported content, it remains unmanaged\n return\n\n def display_warning(self):\n \"\"\"Show warning to ensure the user can't import models by accident\n\n Returns:\n bool\n\n \"\"\"\n\n from qtpy import QtWidgets\n\n accept = QtWidgets.QMessageBox.Ok\n buttons = accept | QtWidgets.QMessageBox.Cancel\n\n message = \"Are you sure you want import this\"\n state = QtWidgets.QMessageBox.warning(None,\n \"Are you sure?\",\n message,\n buttons=buttons,\n defaultButton=accept)\n\n return state == accept\n"
},
{
"path": "openpype/hosts/maya/plugins/load/load_arnold_standin.py",
"content": "import os\nimport clique\n\nimport maya.cmds as cmds\n\nfrom openpype.settings import get_project_settings\nfrom openpype.pipeline import (\n load,\n legacy_io,\n get_representation_path\n)\nfrom openpype.hosts.maya.api.lib import (\n unique_namespace,\n get_attribute_input,\n maintained_selection,\n convert_to_maya_fps\n)\nfrom openpype.hosts.maya.api.pipeline import containerise\n\n\ndef is_sequence(files):\n sequence = False\n collections, remainder = clique.assemble(files, minimum_items=1)\n if collections:\n sequence = True\n return sequence\n\n\ndef get_current_session_fps():\n session_fps = float(legacy_io.Session.get('AVALON_FPS', 25))\n return convert_to_maya_fps(session_fps)\n\n\nclass ArnoldStandinLoader(load.LoaderPlugin):\n \"\"\"Load as Arnold standin\"\"\"\n\n families = [\n \"ass\",\n \"assProxy\",\n \"animation\",\n \"model\",\n \"proxyAbc\",\n \"pointcache\",\n \"usd\"\n ]\n representations = [\"ass\", \"abc\", \"usda\", \"usdc\", \"usd\"]\n\n label = \"Load as Arnold standin\"\n order = -5\n icon = \"code-fork\"\n color = \"orange\"\n\n def load(self, context, name, namespace, options):\n if not cmds.pluginInfo(\"mtoa\", query=True, loaded=True):\n cmds.loadPlugin(\"mtoa\")\n # Create defaultArnoldRenderOptions before creating aiStandin\n # which tries to connect it. Since we load the plugin and directly\n # create aiStandin without the defaultArnoldRenderOptions,\n # we need to create the render options for aiStandin creation.\n from mtoa.core import createOptions\n createOptions()\n\n import mtoa.ui.arnoldmenu\n\n version = context['version']\n version_data = version.get(\"data\", {})\n\n self.log.info(\"version_data: {}\\n\".format(version_data))\n\n asset = context['asset']['name']\n namespace = namespace or unique_namespace(\n asset + \"_\",\n prefix=\"_\" if asset[0].isdigit() else \"\",\n suffix=\"_\",\n )\n\n # Root group\n label = \"{}:{}\".format(namespace, name)\n root = cmds.group(name=label, empty=True)\n\n # Set color.\n settings = get_project_settings(context[\"project\"][\"name\"])\n color = settings['maya']['load']['colors'].get('ass')\n if color is not None:\n cmds.setAttr(root + \".useOutlinerColor\", True)\n cmds.setAttr(\n root + \".outlinerColor\", color[0], color[1], color[2]\n )\n\n with maintained_selection():\n # Create transform with shape\n transform_name = label + \"_standin\"\n\n standin_shape = mtoa.ui.arnoldmenu.createStandIn()\n standin = cmds.listRelatives(standin_shape, parent=True)[0]\n standin = cmds.rename(standin, transform_name)\n standin_shape = cmds.listRelatives(standin, shapes=True)[0]\n\n cmds.parent(standin, root)\n\n # Set the standin filepath\n repre_path = self.filepath_from_context(context)\n path, operator = self._setup_proxy(\n standin_shape, repre_path, namespace\n )\n cmds.setAttr(standin_shape + \".dso\", path, type=\"string\")\n sequence = is_sequence(os.listdir(os.path.dirname(repre_path)))\n cmds.setAttr(standin_shape + \".useFrameExtension\", sequence)\n\n fps = version[\"data\"].get(\"fps\") or get_current_session_fps()\n cmds.setAttr(standin_shape + \".abcFPS\", float(fps))\n\n nodes = [root, standin, standin_shape]\n if operator is not None:\n nodes.append(operator)\n self[:] = nodes\n\n return containerise(\n name=name,\n namespace=namespace,\n nodes=nodes,\n context=context,\n loader=self.__class__.__name__)\n\n def get_next_free_multi_index(self, attr_name):\n \"\"\"Find the next unconnected multi index at the input attribute.\"\"\"\n for index in range(10000000):\n connection_info = cmds.connectionInfo(\n \"{}[{}]\".format(attr_name, index),\n sourceFromDestination=True\n )\n if len(connection_info or []) == 0:\n return index\n\n def _get_proxy_path(self, path):\n basename_split = os.path.basename(path).split(\".\")\n proxy_basename = (\n basename_split[0] + \"_proxy.\" + \".\".join(basename_split[1:])\n )\n proxy_path = \"/\".join([os.path.dirname(path), proxy_basename])\n return proxy_basename, proxy_path\n\n def _update_operators(self, string_replace_operator, proxy_basename, path):\n cmds.setAttr(\n string_replace_operator + \".match\",\n proxy_basename.split(\".\")[0],\n type=\"string\"\n )\n cmds.setAttr(\n string_replace_operator + \".replace\",\n os.path.basename(path).split(\".\")[0],\n type=\"string\"\n )\n\n def _setup_proxy(self, shape, path, namespace):\n proxy_basename, proxy_path = self._get_proxy_path(path)\n\n options_node = \"defaultArnoldRenderOptions\"\n merge_operator = get_attribute_input(options_node + \".operator\")\n if merge_operator is None:\n merge_operator = cmds.createNode(\"aiMerge\")\n cmds.connectAttr(\n merge_operator + \".message\", options_node + \".operator\"\n )\n\n merge_operator = merge_operator.split(\".\")[0]\n\n string_replace_operator = cmds.createNode(\n \"aiStringReplace\", name=namespace + \":string_replace_operator\"\n )\n node_type = \"alembic\" if path.endswith(\".abc\") else \"procedural\"\n cmds.setAttr(\n string_replace_operator + \".selection\",\n \"*.(@node=='{}')\".format(node_type),\n type=\"string\"\n )\n self._update_operators(string_replace_operator, proxy_basename, path)\n\n cmds.connectAttr(\n string_replace_operator + \".out\",\n \"{}.inputs[{}]\".format(\n merge_operator,\n self.get_next_free_multi_index(merge_operator + \".inputs\")\n )\n )\n\n # We setup the string operator no matter whether there is a proxy or\n # not. This makes it easier to update since the string operator will\n # always be created. Return original path to use for standin.\n if not os.path.exists(proxy_path):\n return path, string_replace_operator\n\n return proxy_path, string_replace_operator\n\n def update(self, container, representation):\n # Update the standin\n members = cmds.sets(container['objectName'], query=True)\n for member in members:\n if cmds.nodeType(member) == \"aiStringReplace\":\n string_replace_operator = member\n\n shapes = cmds.listRelatives(member, shapes=True)\n if not shapes:\n continue\n if cmds.nodeType(shapes[0]) == \"aiStandIn\":\n standin = shapes[0]\n\n path = get_representation_path(representation)\n proxy_basename, proxy_path = self._get_proxy_path(path)\n\n # Whether there is proxy or not, we still update the string operator.\n # If no proxy exists, the string operator won't replace anything.\n self._update_operators(string_replace_operator, proxy_basename, path)\n\n dso_path = path\n if os.path.exists(proxy_path):\n dso_path = proxy_path\n cmds.setAttr(standin + \".dso\", dso_path, type=\"string\")\n\n sequence = is_sequence(os.listdir(os.path.dirname(path)))\n cmds.setAttr(standin + \".useFrameExtension\", sequence)\n\n cmds.setAttr(\n container[\"objectName\"] + \".representation\",\n str(representation[\"_id\"]),\n type=\"string\"\n )\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def remove(self, container):\n members = cmds.sets(container['objectName'], query=True)\n cmds.lockNode(members, lock=False)\n cmds.delete([container['objectName']] + members)\n\n # Clean up the namespace\n try:\n cmds.namespace(removeNamespace=container['namespace'],\n deleteNamespaceContent=True)\n except RuntimeError:\n pass\n"
},
{
"path": "openpype/hosts/maya/plugins/load/load_assembly.py",
"content": "import maya.cmds as cmds\n\nfrom openpype.pipeline import (\n load,\n remove_container\n)\n\nfrom openpype.hosts.maya.api.pipeline import containerise\nfrom openpype.hosts.maya.api.lib import unique_namespace\nfrom openpype.hosts.maya.api import setdress\n\n\nclass AssemblyLoader(load.LoaderPlugin):\n\n families = [\"assembly\"]\n representations = [\"json\"]\n\n label = \"Load Set Dress\"\n order = -9\n icon = \"code-fork\"\n color = \"orange\"\n\n def load(self, context, name, namespace, data):\n\n asset = context['asset']['name']\n namespace = namespace or unique_namespace(\n asset + \"_\",\n prefix=\"_\" if asset[0].isdigit() else \"\",\n suffix=\"_\",\n )\n\n containers = setdress.load_package(\n filepath=self.filepath_from_context(context),\n name=name,\n namespace=namespace\n )\n\n self[:] = containers\n\n # Only containerize if any nodes were loaded by the Loader\n nodes = self[:]\n if not nodes:\n return\n\n return containerise(\n name=name,\n namespace=namespace,\n nodes=nodes,\n context=context,\n loader=self.__class__.__name__)\n\n def update(self, container, representation):\n\n return setdress.update_package(container, representation)\n\n def remove(self, container):\n \"\"\"Remove all sub containers\"\"\"\n\n # Remove all members\n member_containers = setdress.get_contained_containers(container)\n for member_container in member_containers:\n self.log.info(\"Removing container %s\",\n member_container['objectName'])\n remove_container(member_container)\n\n # Remove alembic hierarchy reference\n # TODO: Check whether removing all contained references is safe enough\n members = cmds.sets(container['objectName'], query=True) or []\n references = cmds.ls(members, type=\"reference\")\n for reference in references:\n self.log.info(\"Removing %s\", reference)\n fname = cmds.referenceQuery(reference, filename=True)\n cmds.file(fname, removeReference=True)\n\n # Delete container and its contents\n if cmds.objExists(container['objectName']):\n members = cmds.sets(container['objectName'], query=True) or []\n cmds.delete([container['objectName']] + members)\n\n # TODO: Ensure namespace is gone\n"
},
{
"path": "openpype/hosts/maya/plugins/load/load_audio.py",
"content": "from maya import cmds, mel\n\nfrom openpype.pipeline import (\n load,\n get_representation_path,\n)\nfrom openpype.hosts.maya.api.pipeline import containerise\nfrom openpype.hosts.maya.api.lib import unique_namespace, get_container_members\n\n\nclass AudioLoader(load.LoaderPlugin):\n \"\"\"Specific loader of audio.\"\"\"\n\n families = [\"audio\"]\n label = \"Load audio\"\n representations = [\"wav\"]\n icon = \"volume-up\"\n color = \"orange\"\n\n def load(self, context, name, namespace, data):\n\n start_frame = cmds.playbackOptions(query=True, min=True)\n sound_node = cmds.sound(\n file=self.filepath_from_context(context), offset=start_frame\n )\n cmds.timeControl(\n mel.eval(\"$gPlayBackSlider=$gPlayBackSlider\"),\n edit=True,\n sound=sound_node,\n displaySound=True\n )\n\n asset = context[\"asset\"][\"name\"]\n namespace = namespace or unique_namespace(\n asset + \"_\",\n prefix=\"_\" if asset[0].isdigit() else \"\",\n suffix=\"_\",\n )\n\n return containerise(\n name=name,\n namespace=namespace,\n nodes=[sound_node],\n context=context,\n loader=self.__class__.__name__\n )\n\n def update(self, container, representation):\n\n members = get_container_members(container)\n audio_nodes = cmds.ls(members, type=\"audio\")\n\n assert audio_nodes is not None, \"Audio node not found.\"\n audio_node = audio_nodes[0]\n\n current_sound = cmds.timeControl(\n mel.eval(\"$gPlayBackSlider=$gPlayBackSlider\"),\n query=True,\n sound=True\n )\n activate_sound = current_sound == audio_node\n\n path = get_representation_path(representation)\n\n cmds.sound(\n audio_node,\n edit=True,\n file=path\n )\n\n # The source start + end does not automatically update itself to the\n # length of thew new audio file, even though maya does do that when\n # creating a new audio node. So to update we compute it manually.\n # This would however override any source start and source end a user\n # might have done on the original audio node after load.\n audio_frame_count = cmds.getAttr(\"{}.frameCount\".format(audio_node))\n audio_sample_rate = cmds.getAttr(\"{}.sampleRate\".format(audio_node))\n duration_in_seconds = audio_frame_count / audio_sample_rate\n fps = mel.eval('currentTimeUnitToFPS()') # workfile FPS\n source_start = 0\n source_end = (duration_in_seconds * fps)\n cmds.setAttr(\"{}.sourceStart\".format(audio_node), source_start)\n cmds.setAttr(\"{}.sourceEnd\".format(audio_node), source_end)\n\n if activate_sound:\n # maya by default deactivates it from timeline on file change\n cmds.timeControl(\n mel.eval(\"$gPlayBackSlider=$gPlayBackSlider\"),\n edit=True,\n sound=audio_node,\n displaySound=True\n )\n\n cmds.setAttr(\n container[\"objectName\"] + \".representation\",\n str(representation[\"_id\"]),\n type=\"string\"\n )\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def remove(self, container):\n members = cmds.sets(container['objectName'], query=True)\n cmds.lockNode(members, lock=False)\n cmds.delete([container['objectName']] + members)\n\n # Clean up the namespace\n try:\n cmds.namespace(removeNamespace=container['namespace'],\n deleteNamespaceContent=True)\n except RuntimeError:\n pass\n"
},
{
"path": "openpype/hosts/maya/plugins/load/load_gpucache.py",
"content": "import os\n\nimport maya.cmds as cmds\n\nfrom openpype.hosts.maya.api.pipeline import containerise\nfrom openpype.hosts.maya.api.lib import unique_namespace\nfrom openpype.pipeline import (\n load,\n get_representation_path\n)\nfrom openpype.settings import get_project_settings\n\n\nclass GpuCacheLoader(load.LoaderPlugin):\n \"\"\"Load Alembic as gpuCache\"\"\"\n\n families = [\"model\", \"animation\", \"proxyAbc\", \"pointcache\"]\n representations = [\"abc\", \"gpu_cache\"]\n\n label = \"Load Gpu Cache\"\n order = -5\n icon = \"code-fork\"\n color = \"orange\"\n\n def load(self, context, name, namespace, data):\n\n asset = context['asset']['name']\n namespace = namespace or unique_namespace(\n asset + \"_\",\n prefix=\"_\" if asset[0].isdigit() else \"\",\n suffix=\"_\",\n )\n\n cmds.loadPlugin(\"gpuCache\", quiet=True)\n\n # Root group\n label = \"{}:{}\".format(namespace, name)\n root = cmds.group(name=label, empty=True)\n\n project_name = context[\"project\"][\"name\"]\n settings = get_project_settings(project_name)\n colors = settings['maya']['load']['colors']\n c = colors.get('model')\n if c is not None:\n cmds.setAttr(root + \".useOutlinerColor\", 1)\n cmds.setAttr(\n root + \".outlinerColor\",\n (float(c[0]) / 255), (float(c[1]) / 255), (float(c[2]) / 255)\n )\n\n # Create transform with shape\n transform_name = label + \"_GPU\"\n transform = cmds.createNode(\"transform\", name=transform_name,\n parent=root)\n cache = cmds.createNode(\"gpuCache\",\n parent=transform,\n name=\"{0}Shape\".format(transform_name))\n\n # Set the cache filepath\n path = self.filepath_from_context(context)\n cmds.setAttr(cache + '.cacheFileName', path, type=\"string\")\n cmds.setAttr(cache + '.cacheGeomPath', \"|\", type=\"string\") # root\n\n # Lock parenting of the transform and cache\n cmds.lockNode([transform, cache], lock=True)\n\n nodes = [root, transform, cache]\n self[:] = nodes\n\n return containerise(\n name=name,\n namespace=namespace,\n nodes=nodes,\n context=context,\n loader=self.__class__.__name__)\n\n def update(self, container, representation):\n path = get_representation_path(representation)\n\n # Update the cache\n members = cmds.sets(container['objectName'], query=True)\n caches = cmds.ls(members, type=\"gpuCache\", long=True)\n\n assert len(caches) == 1, \"This is a bug\"\n\n for cache in caches:\n cmds.setAttr(cache + \".cacheFileName\", path, type=\"string\")\n\n cmds.setAttr(container[\"objectName\"] + \".representation\",\n str(representation[\"_id\"]),\n type=\"string\")\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def remove(self, container):\n members = cmds.sets(container['objectName'], query=True)\n cmds.lockNode(members, lock=False)\n cmds.delete([container['objectName']] + members)\n\n # Clean up the namespace\n try:\n cmds.namespace(removeNamespace=container['namespace'],\n deleteNamespaceContent=True)\n except RuntimeError:\n pass\n"
},
{
"path": "openpype/hosts/maya/plugins/load/load_image.py",
"content": "import os\nimport copy\n\nfrom openpype.lib import EnumDef\nfrom openpype.pipeline import (\n load,\n get_representation_context,\n get_current_host_name,\n)\nfrom openpype.pipeline.load.utils import get_representation_path_from_context\nfrom openpype.pipeline.colorspace import (\n get_imageio_file_rules_colorspace_from_filepath,\n get_imageio_config,\n get_imageio_file_rules\n)\nfrom openpype.settings import get_project_settings\n\nfrom openpype.hosts.maya.api.pipeline import containerise\nfrom openpype.hosts.maya.api.lib import (\n unique_namespace,\n namespaced\n)\n\nfrom maya import cmds\n\n\ndef create_texture():\n \"\"\"Create place2dTexture with file node with uv connections\n\n Mimics Maya \"file [Texture]\" creation.\n \"\"\"\n\n place = cmds.shadingNode(\"place2dTexture\", asUtility=True, name=\"place2d\")\n file = cmds.shadingNode(\"file\", asTexture=True, name=\"file\")\n\n connections = [\"coverage\", \"translateFrame\", \"rotateFrame\", \"rotateUV\",\n \"mirrorU\", \"mirrorV\", \"stagger\", \"wrapV\", \"wrapU\",\n \"repeatUV\", \"offset\", \"noiseUV\", \"vertexUvThree\",\n \"vertexUvTwo\", \"vertexUvOne\", \"vertexCameraOne\"]\n for attr in connections:\n src = \"{}.{}\".format(place, attr)\n dest = \"{}.{}\".format(file, attr)\n cmds.connectAttr(src, dest)\n\n cmds.connectAttr(place + '.outUV', file + '.uvCoord')\n cmds.connectAttr(place + '.outUvFilterSize', file + '.uvFilterSize')\n\n return file, place\n\n\ndef create_projection():\n \"\"\"Create texture with place3dTexture and projection\n\n Mimics Maya \"file [Projection]\" creation.\n \"\"\"\n\n file, place = create_texture()\n projection = cmds.shadingNode(\"projection\", asTexture=True,\n name=\"projection\")\n place3d = cmds.shadingNode(\"place3dTexture\", asUtility=True,\n name=\"place3d\")\n\n cmds.connectAttr(place3d + '.worldInverseMatrix[0]',\n projection + \".placementMatrix\")\n cmds.connectAttr(file + '.outColor', projection + \".image\")\n\n return file, place, projection, place3d\n\n\ndef create_stencil():\n \"\"\"Create texture with extra place2dTexture offset and stencil\n\n Mimics Maya \"file [Stencil]\" creation.\n \"\"\"\n\n file, place = create_texture()\n\n place_stencil = cmds.shadingNode(\"place2dTexture\", asUtility=True,\n name=\"place2d_stencil\")\n stencil = cmds.shadingNode(\"stencil\", asTexture=True, name=\"stencil\")\n\n for src_attr, dest_attr in [\n (\"outUV\", \"uvCoord\"),\n (\"outUvFilterSize\", \"uvFilterSize\")\n ]:\n src_plug = \"{}.{}\".format(place_stencil, src_attr)\n cmds.connectAttr(src_plug, \"{}.{}\".format(place, dest_attr))\n cmds.connectAttr(src_plug, \"{}.{}\".format(stencil, dest_attr))\n\n return file, place, stencil, place_stencil\n\n\nclass FileNodeLoader(load.LoaderPlugin):\n \"\"\"File node loader.\"\"\"\n\n families = [\"image\", \"plate\", \"render\"]\n label = \"Load file node\"\n representations = [\"exr\", \"tif\", \"png\", \"jpg\"]\n icon = \"image\"\n color = \"orange\"\n order = 2\n\n options = [\n EnumDef(\n \"mode\",\n items={\n \"texture\": \"Texture\",\n \"projection\": \"Projection\",\n \"stencil\": \"Stencil\"\n },\n default=\"texture\",\n label=\"Texture Mode\"\n )\n ]\n\n def load(self, context, name, namespace, data):\n\n asset = context['asset']['name']\n namespace = namespace or unique_namespace(\n asset + \"_\",\n prefix=\"_\" if asset[0].isdigit() else \"\",\n suffix=\"_\",\n )\n\n with namespaced(namespace, new=True) as namespace:\n # Create the nodes within the namespace\n nodes = {\n \"texture\": create_texture,\n \"projection\": create_projection,\n \"stencil\": create_stencil\n }[data.get(\"mode\", \"texture\")]()\n\n file_node = cmds.ls(nodes, type=\"file\")[0]\n\n self._apply_representation_context(context, file_node)\n\n # For ease of access for the user select all the nodes and select\n # the file node last so that UI shows its attributes by default\n cmds.select(list(nodes) + [file_node], replace=True)\n\n return containerise(\n name=name,\n namespace=namespace,\n nodes=nodes,\n context=context,\n loader=self.__class__.__name__\n )\n\n def update(self, container, representation):\n\n members = cmds.sets(container['objectName'], query=True)\n file_node = cmds.ls(members, type=\"file\")[0]\n\n context = get_representation_context(representation)\n self._apply_representation_context(context, file_node)\n\n # Update representation\n cmds.setAttr(\n container[\"objectName\"] + \".representation\",\n str(representation[\"_id\"]),\n type=\"string\"\n )\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def remove(self, container):\n members = cmds.sets(container['objectName'], query=True)\n cmds.lockNode(members, lock=False)\n cmds.delete([container['objectName']] + members)\n\n # Clean up the namespace\n try:\n cmds.namespace(removeNamespace=container['namespace'],\n deleteNamespaceContent=True)\n except RuntimeError:\n pass\n\n def _apply_representation_context(self, context, file_node):\n \"\"\"Update the file node to match the context.\n\n This sets the file node's attributes for:\n - file path\n - udim tiling mode (if it is an udim tile)\n - use frame extension (if it is a sequence)\n - colorspace\n\n \"\"\"\n\n repre_context = context[\"representation\"][\"context\"]\n has_frames = repre_context.get(\"frame\") is not None\n has_udim = repre_context.get(\"udim\") is not None\n\n # Set UV tiling mode if UDIM tiles\n if has_udim:\n cmds.setAttr(file_node + \".uvTilingMode\", 3) # UDIM-tiles\n else:\n cmds.setAttr(file_node + \".uvTilingMode\", 0) # off\n\n # Enable sequence if publish has `startFrame` and `endFrame` and\n # `startFrame != endFrame`\n if has_frames and self._is_sequence(context):\n # When enabling useFrameExtension maya automatically\n # connects an expression to .frameExtension to set\n # the current frame. However, this expression is generated\n # with some delay and thus it'll show a warning if frame 0\n # doesn't exist because we're explicitly setting the \n # token.\n cmds.setAttr(file_node + \".useFrameExtension\", True)\n else:\n cmds.setAttr(file_node + \".useFrameExtension\", False)\n\n # Set the file node path attribute\n path = self._format_path(context)\n cmds.setAttr(file_node + \".fileTextureName\", path, type=\"string\")\n\n # Set colorspace\n colorspace = self._get_colorspace(context)\n if colorspace:\n cmds.setAttr(file_node + \".colorSpace\", colorspace, type=\"string\")\n else:\n self.log.debug(\"Unknown colorspace - setting colorspace skipped.\")\n\n def _is_sequence(self, context):\n \"\"\"Check whether frameStart and frameEnd are not the same.\"\"\"\n version = context.get(\"version\", {})\n representation = context.get(\"representation\", {})\n\n for doc in [representation, version]:\n # Frame range can be set on version or representation.\n # When set on representation it overrides version data.\n data = doc.get(\"data\", {})\n start = data.get(\"frameStartHandle\", data.get(\"frameStart\", None))\n end = data.get(\"frameEndHandle\", data.get(\"frameEnd\", None))\n\n if start is None or end is None:\n continue\n\n if start != end:\n return True\n else:\n return False\n\n return False\n\n def _get_colorspace(self, context):\n \"\"\"Return colorspace of the file to load.\n\n Retrieves the explicit colorspace from the publish. If no colorspace\n data is stored with published content then project imageio settings\n are used to make an assumption of the colorspace based on the file\n rules. If no file rules match then None is returned.\n\n Returns:\n str or None: The colorspace of the file or None if not detected.\n\n \"\"\"\n\n # We can't apply color spaces if management is not enabled\n if not cmds.colorManagementPrefs(query=True, cmEnabled=True):\n return\n\n representation = context[\"representation\"]\n colorspace_data = representation.get(\"data\", {}).get(\"colorspaceData\")\n if colorspace_data:\n return colorspace_data[\"colorspace\"]\n\n # Assume colorspace from filepath based on project settings\n project_name = context[\"project\"][\"name\"]\n host_name = get_current_host_name()\n project_settings = get_project_settings(project_name)\n\n config_data = get_imageio_config(\n project_name, host_name,\n project_settings=project_settings\n )\n\n # ignore if host imageio is not enabled\n if not config_data:\n return\n\n file_rules = get_imageio_file_rules(\n project_name, host_name,\n project_settings=project_settings\n )\n\n path = get_representation_path_from_context(context)\n colorspace = get_imageio_file_rules_colorspace_from_filepath(\n path,\n host_name,\n project_name,\n config_data=config_data,\n file_rules=file_rules,\n project_settings=project_settings\n )\n\n return colorspace\n\n def _format_path(self, context):\n \"\"\"Format the path with correct tokens for frames and udim tiles.\"\"\"\n\n context = copy.deepcopy(context)\n representation = context[\"representation\"]\n template = representation.get(\"data\", {}).get(\"template\")\n if not template:\n # No template to find token locations for\n return get_representation_path_from_context(context)\n\n def _placeholder(key):\n # Substitute with a long placeholder value so that potential\n # custom formatting with padding doesn't find its way into\n # our formatting, so that wouldn't be padded as 0\n return \"___{}___\".format(key)\n\n # We format UDIM and Frame numbers with their specific tokens. To do so\n # we in-place change the representation context data to format the path\n # with our own data\n tokens = {\n \"frame\": \"\",\n \"udim\": \"\"\n }\n has_tokens = False\n repre_context = representation[\"context\"]\n for key, _token in tokens.items():\n if key in repre_context:\n repre_context[key] = _placeholder(key)\n has_tokens = True\n\n # Replace with our custom template that has the tokens set\n representation[\"data\"][\"template\"] = template\n path = get_representation_path_from_context(context)\n\n if has_tokens:\n for key, token in tokens.items():\n if key in repre_context:\n path = path.replace(_placeholder(key), token)\n\n return path\n"
},
{
"path": "openpype/hosts/maya/plugins/load/load_image_plane.py",
"content": "from qtpy import QtWidgets, QtCore\n\nfrom openpype.client import (\n get_asset_by_id,\n get_subset_by_id,\n get_version_by_id,\n)\nfrom openpype.pipeline import (\n load,\n get_representation_path,\n get_current_project_name,\n)\nfrom openpype.hosts.maya.api.pipeline import containerise\nfrom openpype.hosts.maya.api.lib import (\n unique_namespace,\n namespaced,\n pairwise,\n get_container_members\n)\n\nfrom maya import cmds\n\n\ndef disconnect_inputs(plug):\n overrides = cmds.listConnections(plug,\n source=True,\n destination=False,\n plugs=True,\n connections=True) or []\n for dest, src in pairwise(overrides):\n cmds.disconnectAttr(src, dest)\n\n\nclass CameraWindow(QtWidgets.QDialog):\n\n def __init__(self, cameras):\n super(CameraWindow, self).__init__()\n self.setWindowFlags(self.windowFlags() | QtCore.Qt.FramelessWindowHint)\n\n self.camera = None\n\n self.widgets = {\n \"label\": QtWidgets.QLabel(\"Select camera for image plane.\"),\n \"list\": QtWidgets.QListWidget(),\n \"staticImagePlane\": QtWidgets.QCheckBox(),\n \"showInAllViews\": QtWidgets.QCheckBox(),\n \"warning\": QtWidgets.QLabel(\"No cameras selected!\"),\n \"buttons\": QtWidgets.QWidget(),\n \"okButton\": QtWidgets.QPushButton(\"Ok\"),\n \"cancelButton\": QtWidgets.QPushButton(\"Cancel\")\n }\n\n # Build warning.\n self.widgets[\"warning\"].setVisible(False)\n self.widgets[\"warning\"].setStyleSheet(\"color: red\")\n\n # Build list.\n for camera in cameras:\n self.widgets[\"list\"].addItem(camera)\n\n\n # Build buttons.\n layout = QtWidgets.QHBoxLayout(self.widgets[\"buttons\"])\n layout.addWidget(self.widgets[\"okButton\"])\n layout.addWidget(self.widgets[\"cancelButton\"])\n\n # Build layout.\n layout = QtWidgets.QVBoxLayout(self)\n layout.addWidget(self.widgets[\"label\"])\n layout.addWidget(self.widgets[\"list\"])\n layout.addWidget(self.widgets[\"buttons\"])\n layout.addWidget(self.widgets[\"warning\"])\n\n self.widgets[\"okButton\"].pressed.connect(self.on_ok_pressed)\n self.widgets[\"cancelButton\"].pressed.connect(self.on_cancel_pressed)\n self.widgets[\"list\"].itemPressed.connect(self.on_list_itemPressed)\n\n def on_list_itemPressed(self, item):\n self.camera = item.text()\n\n def on_ok_pressed(self):\n if self.camera is None:\n self.widgets[\"warning\"].setVisible(True)\n return\n\n self.close()\n\n def on_cancel_pressed(self):\n self.camera = None\n self.close()\n\n\nclass ImagePlaneLoader(load.LoaderPlugin):\n \"\"\"Specific loader of plate for image planes on selected camera.\"\"\"\n\n families = [\"image\", \"plate\", \"render\"]\n label = \"Load imagePlane\"\n representations = [\"mov\", \"exr\", \"preview\", \"png\", \"jpg\"]\n icon = \"image\"\n color = \"orange\"\n\n def load(self, context, name, namespace, data, options=None):\n\n image_plane_depth = 1000\n asset = context['asset']['name']\n namespace = namespace or unique_namespace(\n asset + \"_\",\n prefix=\"_\" if asset[0].isdigit() else \"\",\n suffix=\"_\",\n )\n\n # Get camera from user selection.\n # is_static_image_plane = None\n # is_in_all_views = None\n camera = data.get(\"camera\") if data else None\n\n if not camera:\n cameras = cmds.ls(type=\"camera\")\n\n # Cameras by names\n camera_names = {}\n for camera in cameras:\n parent = cmds.listRelatives(camera, parent=True, path=True)[0]\n camera_names[parent] = camera\n\n camera_names[\"Create new camera.\"] = \"create-camera\"\n window = CameraWindow(camera_names.keys())\n window.exec_()\n # Skip if no camera was selected (Dialog was closed)\n if window.camera not in camera_names:\n return\n camera = camera_names[window.camera]\n\n if camera == \"create-camera\":\n camera = cmds.createNode(\"camera\")\n\n if camera is None:\n return\n\n try:\n cmds.setAttr(\"{}.displayResolution\".format(camera), True)\n cmds.setAttr(\"{}.farClipPlane\".format(camera),\n image_plane_depth * 10)\n except RuntimeError:\n pass\n\n # Create image plane\n with namespaced(namespace):\n # Create inside the namespace\n image_plane_transform, image_plane_shape = cmds.imagePlane(\n fileName=context[\"representation\"][\"data\"][\"path\"],\n camera=camera\n )\n start_frame = cmds.playbackOptions(query=True, min=True)\n end_frame = cmds.playbackOptions(query=True, max=True)\n\n for attr, value in {\n \"depth\": image_plane_depth,\n \"frameOffset\": 0,\n \"frameIn\": start_frame,\n \"frameOut\": end_frame,\n \"frameCache\": end_frame,\n \"useFrameExtension\": True\n }.items():\n plug = \"{}.{}\".format(image_plane_shape, attr)\n cmds.setAttr(plug, value)\n\n movie_representations = [\"mov\", \"preview\"]\n if context[\"representation\"][\"name\"] in movie_representations:\n cmds.setAttr(image_plane_shape + \".type\", 2)\n\n # Ask user whether to use sequence or still image.\n if context[\"representation\"][\"name\"] == \"exr\":\n # Ensure OpenEXRLoader plugin is loaded.\n cmds.loadPlugin(\"OpenEXRLoader\", quiet=True)\n\n message = (\n \"Hold image sequence on first frame?\"\n \"\\n{} files available.\".format(\n len(context[\"representation\"][\"files\"])\n )\n )\n reply = QtWidgets.QMessageBox.information(\n None,\n \"Frame Hold.\",\n message,\n QtWidgets.QMessageBox.Yes,\n QtWidgets.QMessageBox.No\n )\n if reply == QtWidgets.QMessageBox.Yes:\n frame_extension_plug = \"{}.frameExtension\".format(image_plane_shape) # noqa\n\n # Remove current frame expression\n disconnect_inputs(frame_extension_plug)\n\n cmds.setAttr(frame_extension_plug, start_frame)\n\n new_nodes = [image_plane_transform, image_plane_shape]\n\n return containerise(\n name=name,\n namespace=namespace,\n nodes=new_nodes,\n context=context,\n loader=self.__class__.__name__\n )\n\n def update(self, container, representation):\n\n members = get_container_members(container)\n image_planes = cmds.ls(members, type=\"imagePlane\")\n assert image_planes, \"Image plane not found.\"\n image_plane_shape = image_planes[0]\n\n path = get_representation_path(representation)\n cmds.setAttr(\"{}.imageName\".format(image_plane_shape),\n path,\n type=\"string\")\n cmds.setAttr(\"{}.representation\".format(container[\"objectName\"]),\n str(representation[\"_id\"]),\n type=\"string\")\n\n # Set frame range.\n project_name = get_current_project_name()\n version = get_version_by_id(\n project_name, representation[\"parent\"], fields=[\"parent\"]\n )\n subset = get_subset_by_id(\n project_name, version[\"parent\"], fields=[\"parent\"]\n )\n asset = get_asset_by_id(\n project_name, subset[\"parent\"], fields=[\"parent\"]\n )\n start_frame = asset[\"data\"][\"frameStart\"]\n end_frame = asset[\"data\"][\"frameEnd\"]\n\n for attr, value in {\n \"frameOffset\": 0,\n \"frameIn\": start_frame,\n \"frameOut\": end_frame,\n \"frameCache\": end_frame\n }:\n plug = \"{}.{}\".format(image_plane_shape, attr)\n cmds.setAttr(plug, value)\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def remove(self, container):\n members = cmds.sets(container['objectName'], query=True)\n cmds.lockNode(members, lock=False)\n cmds.delete([container['objectName']] + members)\n\n # Clean up the namespace\n try:\n cmds.namespace(removeNamespace=container['namespace'],\n deleteNamespaceContent=True)\n except RuntimeError:\n pass\n"
},
{
"path": "openpype/hosts/maya/plugins/load/load_look.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Look loader.\"\"\"\nimport json\nfrom collections import defaultdict\n\nfrom qtpy import QtWidgets\n\nfrom openpype.client import get_representation_by_name\nfrom openpype.pipeline import (\n get_current_project_name,\n get_representation_path,\n)\nimport openpype.hosts.maya.api.plugin\nfrom openpype.hosts.maya.api import lib\nfrom openpype.widgets.message_window import ScrollMessageBox\n\nfrom openpype.hosts.maya.api.lib import get_reference_node\n\n\nclass LookLoader(openpype.hosts.maya.api.plugin.ReferenceLoader):\n \"\"\"Specific loader for lookdev\"\"\"\n\n families = [\"look\"]\n representations = [\"ma\"]\n\n label = \"Reference look\"\n order = -10\n icon = \"code-fork\"\n color = \"orange\"\n\n def process_reference(self, context, name, namespace, options):\n from maya import cmds\n\n with lib.maintained_selection():\n file_url = self.prepare_root_value(\n file_url=self.filepath_from_context(context),\n project_name=context[\"project\"][\"name\"]\n )\n nodes = cmds.file(file_url,\n namespace=namespace,\n reference=True,\n returnNewNodes=True)\n\n self[:] = nodes\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def update(self, container, representation):\n \"\"\"\n Called by Scene Inventory when look should be updated to current\n version.\n If any reference edits cannot be applied, eg. shader renamed and\n material not present, reference is unloaded and cleaned.\n All failed edits are highlighted to the user via message box.\n\n Args:\n container: object that has look to be updated\n representation: (dict): relationship data to get proper\n representation from DB and persisted\n data in .json\n Returns:\n None\n \"\"\"\n from maya import cmds\n\n # Get reference node from container members\n members = lib.get_container_members(container)\n reference_node = get_reference_node(members, log=self.log)\n\n shader_nodes = cmds.ls(members, type='shadingEngine')\n orig_nodes = set(self._get_nodes_with_shader(shader_nodes))\n\n # Trigger the regular reference update on the ReferenceLoader\n super(LookLoader, self).update(container, representation)\n\n # get new applied shaders and nodes from new version\n shader_nodes = cmds.ls(members, type='shadingEngine')\n nodes = set(self._get_nodes_with_shader(shader_nodes))\n\n project_name = get_current_project_name()\n json_representation = get_representation_by_name(\n project_name, \"json\", representation[\"parent\"]\n )\n\n # Load relationships\n shader_relation = get_representation_path(json_representation)\n with open(shader_relation, \"r\") as f:\n json_data = json.load(f)\n\n # update of reference could result in failed edits - material is not\n # present because of renaming etc. If so highlight failed edits to user\n failed_edits = cmds.referenceQuery(reference_node,\n editStrings=True,\n failedEdits=True,\n successfulEdits=False)\n if failed_edits:\n # clean references - removes failed reference edits\n cmds.file(cr=reference_node) # cleanReference\n\n # reapply shading groups from json representation on orig nodes\n lib.apply_shaders(json_data, shader_nodes, orig_nodes)\n\n msg = [\"During reference update some edits failed.\",\n \"All successful edits were kept intact.\\n\",\n \"Failed and removed edits:\"]\n msg.extend(failed_edits)\n\n msg = ScrollMessageBox(QtWidgets.QMessageBox.Warning,\n \"Some reference edit failed\",\n msg)\n msg.exec_()\n\n attributes = json_data.get(\"attributes\", [])\n\n # region compute lookup\n nodes_by_id = defaultdict(list)\n for node in nodes:\n nodes_by_id[lib.get_id(node)].append(node)\n lib.apply_attributes(attributes, nodes_by_id)\n\n def _get_nodes_with_shader(self, shader_nodes):\n \"\"\"\n Returns list of nodes belonging to specific shaders\n Args:\n shader_nodes: of Shader groups\n Returns\n node names\n \"\"\"\n from maya import cmds\n\n for shader in shader_nodes:\n future = cmds.listHistory(shader, future=True)\n connections = cmds.listConnections(future,\n type='mesh')\n if connections:\n # Ensure unique entries only to optimize query and results\n connections = list(set(connections))\n return cmds.listRelatives(connections,\n shapes=True,\n fullPath=True) or []\n return []\n"
},
{
"path": "openpype/hosts/maya/plugins/load/load_matchmove.py",
"content": "# -*- coding: utf-8 -*-\nfrom maya import cmds, mel # noqa: F401\n\nfrom openpype.hosts.maya.api.pipeline import containerise\nfrom openpype.hosts.maya.api import lib, Loader\nfrom openpype.pipeline.load import get_representation_path, LoadError\n\n\nclass MatchmoveLoader(Loader):\n \"\"\"Run matchmove script to create track in scene.\n\n Supported script types are .py and .mel\n\n TODO: there might be error in the scripts exported from\n 3DEqualizer that it is trying to set frame attribute\n on camera image plane and then add expression for\n image sequence. Maya will throw RuntimeError at that\n point that will stop processing rest of the script and\n the container will not be created. We should somehow handle\n this - maybe even by patching the mel script on-the-fly.\n\n \"\"\"\n\n families = [\"matchmove\"]\n representations = [\"py\", \"mel\"]\n defaults = [\"Camera\", \"Object\", \"Mocap\"]\n\n label = \"Run matchmove script\"\n icon = \"empire\"\n color = \"orange\"\n\n def load(self, context, name, namespace, options):\n\n path = self.filepath_from_context(context)\n custom_group_name, custom_namespace, options = \\\n self.get_custom_namespace_and_group(\n context, options, \"matchmove_loader\")\n\n namespace = lib.get_custom_namespace(custom_namespace)\n\n nodes = self._load_nodes_from_script(path)\n\n return containerise(\n name=name,\n namespace=namespace,\n nodes=nodes,\n context=context,\n loader=self.__class__.__name__\n )\n\n def update(self, container, representation):\n # type: (dict, dict) -> None\n \"\"\"Update container with specified representation.\"\"\"\n self.remove(container)\n\n path = get_representation_path(representation)\n namespace = container[\"namespace\"]\n print(f\">>> loading from {path}\")\n try:\n nodes = self._load_nodes_from_script(path)\n except RuntimeError as e:\n raise LoadError(\"Failed to load matchmove script.\") from e\n\n return containerise(\n name=container[\"name\"],\n namespace=namespace,\n nodes=nodes,\n context=representation[\"context\"],\n loader=self.__class__.__name__\n )\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def remove(self, container):\n \"\"\"Delete container and its contents.\"\"\"\n\n if cmds.objExists(container['objectName']):\n members = cmds.sets(container['objectName'], query=True) or []\n cmds.delete([container['objectName']] + members)\n\n def _load_nodes_from_script(self, path):\n # type: (str) -> list\n \"\"\"Load nodes from script.\n\n This will execute py or mel script and resulting\n nodes will be returned.\n\n Args:\n path (str): path to script\n\n Returns:\n list: list of created nodes\n\n \"\"\"\n previous_nodes = set(cmds.ls(long=True))\n\n if path.lower().endswith(\".py\"):\n exec(open(path).read())\n\n elif path.lower().endswith(\".mel\"):\n mel.eval(open(path).read())\n\n else:\n self.log.error(\"Unsupported script type\")\n\n current_nodes = set(cmds.ls(long=True))\n return list(current_nodes - previous_nodes)\n"
},
{
"path": "openpype/hosts/maya/plugins/load/load_maya_usd.py",
"content": "# -*- coding: utf-8 -*-\nimport maya.cmds as cmds\n\nfrom openpype.pipeline import (\n load,\n get_representation_path,\n)\nfrom openpype.pipeline.load import get_representation_path_from_context\nfrom openpype.hosts.maya.api.lib import (\n namespaced,\n unique_namespace\n)\nfrom openpype.hosts.maya.api.pipeline import containerise\n\n\nclass MayaUsdLoader(load.LoaderPlugin):\n \"\"\"Read USD data in a Maya USD Proxy\"\"\"\n\n families = [\"model\", \"usd\", \"pointcache\", \"animation\"]\n representations = [\"usd\", \"usda\", \"usdc\", \"usdz\", \"abc\"]\n\n label = \"Load USD to Maya Proxy\"\n order = -1\n icon = \"code-fork\"\n color = \"orange\"\n\n def load(self, context, name=None, namespace=None, options=None):\n asset = context['asset']['name']\n namespace = namespace or unique_namespace(\n asset + \"_\",\n prefix=\"_\" if asset[0].isdigit() else \"\",\n suffix=\"_\",\n )\n\n # Make sure we can load the plugin\n cmds.loadPlugin(\"mayaUsdPlugin\", quiet=True)\n\n path = get_representation_path_from_context(context)\n\n # Create the shape\n cmds.namespace(addNamespace=namespace)\n with namespaced(namespace, new=False):\n transform = cmds.createNode(\"transform\",\n name=name,\n skipSelect=True)\n proxy = cmds.createNode('mayaUsdProxyShape',\n name=\"{}Shape\".format(name),\n parent=transform,\n skipSelect=True)\n\n cmds.connectAttr(\"time1.outTime\", \"{}.time\".format(proxy))\n cmds.setAttr(\"{}.filePath\".format(proxy), path, type=\"string\")\n\n # By default, we force the proxy to not use a shared stage because\n # when doing so Maya will quite easily allow to save into the\n # loaded usd file. Since we are loading published files we want to\n # avoid altering them. Unshared stages also save their edits into\n # the workfile as an artist might expect it to do.\n cmds.setAttr(\"{}.shareStage\".format(proxy), False)\n # cmds.setAttr(\"{}.shareStage\".format(proxy), lock=True)\n\n nodes = [transform, proxy]\n self[:] = nodes\n\n return containerise(\n name=name,\n namespace=namespace,\n nodes=nodes,\n context=context,\n loader=self.__class__.__name__)\n\n def update(self, container, representation):\n # type: (dict, dict) -> None\n \"\"\"Update container with specified representation.\"\"\"\n node = container['objectName']\n assert cmds.objExists(node), \"Missing container\"\n\n members = cmds.sets(node, query=True) or []\n shapes = cmds.ls(members, type=\"mayaUsdProxyShape\")\n\n path = get_representation_path(representation)\n for shape in shapes:\n cmds.setAttr(\"{}.filePath\".format(shape), path, type=\"string\")\n\n cmds.setAttr(\"{}.representation\".format(node),\n str(representation[\"_id\"]),\n type=\"string\")\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def remove(self, container):\n # type: (dict) -> None\n \"\"\"Remove loaded container.\"\"\"\n # Delete container and its contents\n if cmds.objExists(container['objectName']):\n members = cmds.sets(container['objectName'], query=True) or []\n cmds.delete([container['objectName']] + members)\n\n # Remove the namespace, if empty\n namespace = container['namespace']\n if cmds.namespace(exists=namespace):\n members = cmds.namespaceInfo(namespace, listNamespace=True)\n if not members:\n cmds.namespace(removeNamespace=namespace)\n else:\n self.log.warning(\"Namespace not deleted because it \"\n \"still has members: %s\", namespace)\n"
},
{
"path": "openpype/hosts/maya/plugins/load/load_multiverse_usd.py",
"content": "# -*- coding: utf-8 -*-\nimport maya.cmds as cmds\nfrom maya import mel\nimport os\n\nfrom openpype.pipeline import (\n load,\n get_representation_path\n)\nfrom openpype.hosts.maya.api.lib import (\n maintained_selection,\n namespaced,\n unique_namespace\n)\nfrom openpype.hosts.maya.api.pipeline import containerise\nfrom openpype.client import get_representation_by_id\n\n\nclass MultiverseUsdLoader(load.LoaderPlugin):\n \"\"\"Read USD data in a Multiverse Compound\"\"\"\n\n families = [\"model\", \"usd\", \"mvUsdComposition\", \"mvUsdOverride\",\n \"pointcache\", \"animation\"]\n representations = [\"usd\", \"usda\", \"usdc\", \"usdz\", \"abc\"]\n\n label = \"Load USD to Multiverse\"\n order = -10\n icon = \"code-fork\"\n color = \"orange\"\n\n def load(self, context, name=None, namespace=None, options=None):\n asset = context['asset']['name']\n namespace = namespace or unique_namespace(\n asset + \"_\",\n prefix=\"_\" if asset[0].isdigit() else \"\",\n suffix=\"_\",\n )\n\n path = self.filepath_from_context(context)\n\n # Make sure we can load the plugin\n cmds.loadPlugin(\"MultiverseForMaya\", quiet=True)\n import multiverse\n\n # Create the shape\n with maintained_selection():\n cmds.namespace(addNamespace=namespace)\n with namespaced(namespace, new=False):\n shape = multiverse.CreateUsdCompound(path)\n transform = cmds.listRelatives(\n shape, parent=True, fullPath=True)[0]\n\n nodes = [transform, shape]\n self[:] = nodes\n\n return containerise(\n name=name,\n namespace=namespace,\n nodes=nodes,\n context=context,\n loader=self.__class__.__name__)\n\n def update(self, container, representation):\n # type: (dict, dict) -> None\n \"\"\"Update container with specified representation.\"\"\"\n node = container['objectName']\n assert cmds.objExists(node), \"Missing container\"\n\n members = cmds.sets(node, query=True) or []\n shapes = cmds.ls(members, type=\"mvUsdCompoundShape\")\n assert shapes, \"Cannot find mvUsdCompoundShape in container\"\n\n project_name = representation[\"context\"][\"project\"][\"name\"]\n prev_representation_id = cmds.getAttr(\"{}.representation\".format(node))\n prev_representation = get_representation_by_id(project_name,\n prev_representation_id)\n prev_path = os.path.normpath(prev_representation[\"data\"][\"path\"])\n\n # Make sure we can load the plugin\n cmds.loadPlugin(\"MultiverseForMaya\", quiet=True)\n import multiverse\n\n for shape in shapes:\n\n asset_paths = multiverse.GetUsdCompoundAssetPaths(shape)\n asset_paths = [os.path.normpath(p) for p in asset_paths]\n\n assert asset_paths.count(prev_path) == 1, \\\n \"Couldn't find matching path (or too many)\"\n prev_path_idx = asset_paths.index(prev_path)\n\n path = get_representation_path(representation)\n asset_paths[prev_path_idx] = path\n\n multiverse.SetUsdCompoundAssetPaths(shape, asset_paths)\n\n cmds.setAttr(\"{}.representation\".format(node),\n str(representation[\"_id\"]),\n type=\"string\")\n mel.eval('refreshEditorTemplates;')\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def remove(self, container):\n # type: (dict) -> None\n \"\"\"Remove loaded container.\"\"\"\n # Delete container and its contents\n if cmds.objExists(container['objectName']):\n members = cmds.sets(container['objectName'], query=True) or []\n cmds.delete([container['objectName']] + members)\n\n # Remove the namespace, if empty\n namespace = container['namespace']\n if cmds.namespace(exists=namespace):\n members = cmds.namespaceInfo(namespace, listNamespace=True)\n if not members:\n cmds.namespace(removeNamespace=namespace)\n else:\n self.log.warning(\"Namespace not deleted because it \"\n \"still has members: %s\", namespace)\n"
},
{
"path": "openpype/hosts/maya/plugins/load/load_multiverse_usd_over.py",
"content": "# -*- coding: utf-8 -*-\nimport maya.cmds as cmds\nfrom maya import mel\nimport os\n\nimport qargparse\n\nfrom openpype.pipeline import (\n load,\n get_representation_path\n)\nfrom openpype.hosts.maya.api.lib import (\n maintained_selection\n)\nfrom openpype.hosts.maya.api.pipeline import containerise\nfrom openpype.client import get_representation_by_id\n\n\nclass MultiverseUsdOverLoader(load.LoaderPlugin):\n \"\"\"Reference file\"\"\"\n\n families = [\"mvUsdOverride\"]\n representations = [\"usda\", \"usd\", \"udsz\"]\n\n label = \"Load Usd Override into Compound\"\n order = -10\n icon = \"code-fork\"\n color = \"orange\"\n\n options = [\n qargparse.String(\n \"Which Compound\",\n label=\"Compound\",\n help=\"Select which compound to add this as a layer to.\"\n )\n ]\n\n def load(self, context, name=None, namespace=None, options=None):\n current_usd = cmds.ls(selection=True,\n type=\"mvUsdCompoundShape\",\n dag=True,\n long=True)\n if len(current_usd) != 1:\n self.log.error(\"Current selection invalid: '{}', \"\n \"must contain exactly 1 mvUsdCompoundShape.\"\n \"\".format(current_usd))\n return\n\n # Make sure we can load the plugin\n cmds.loadPlugin(\"MultiverseForMaya\", quiet=True)\n import multiverse\n\n path = self.filepath_from_context(context)\n nodes = current_usd\n with maintained_selection():\n multiverse.AddUsdCompoundAssetPath(current_usd[0], path)\n\n namespace = current_usd[0].split(\"|\")[1].split(\":\")[0]\n\n container = containerise(\n name=name,\n namespace=namespace,\n nodes=nodes,\n context=context,\n loader=self.__class__.__name__)\n\n cmds.addAttr(container, longName=\"mvUsdCompoundShape\",\n niceName=\"mvUsdCompoundShape\", dataType=\"string\")\n cmds.setAttr(container + \".mvUsdCompoundShape\",\n current_usd[0], type=\"string\")\n\n return container\n\n def update(self, container, representation):\n # type: (dict, dict) -> None\n \"\"\"Update container with specified representation.\"\"\"\n\n cmds.loadPlugin(\"MultiverseForMaya\", quiet=True)\n import multiverse\n\n node = container['objectName']\n assert cmds.objExists(node), \"Missing container\"\n\n members = cmds.sets(node, query=True) or []\n shapes = cmds.ls(members, type=\"mvUsdCompoundShape\")\n assert shapes, \"Cannot find mvUsdCompoundShape in container\"\n\n mvShape = container['mvUsdCompoundShape']\n assert mvShape, \"Missing mv source\"\n\n project_name = representation[\"context\"][\"project\"][\"name\"]\n prev_representation_id = cmds.getAttr(\"{}.representation\".format(node))\n prev_representation = get_representation_by_id(project_name,\n prev_representation_id)\n prev_path = os.path.normpath(prev_representation[\"data\"][\"path\"])\n\n path = get_representation_path(representation)\n\n for shape in shapes:\n asset_paths = multiverse.GetUsdCompoundAssetPaths(shape)\n asset_paths = [os.path.normpath(p) for p in asset_paths]\n\n assert asset_paths.count(prev_path) == 1, \\\n \"Couldn't find matching path (or too many)\"\n prev_path_idx = asset_paths.index(prev_path)\n asset_paths[prev_path_idx] = path\n multiverse.SetUsdCompoundAssetPaths(shape, asset_paths)\n\n cmds.setAttr(\"{}.representation\".format(node),\n str(representation[\"_id\"]),\n type=\"string\")\n mel.eval('refreshEditorTemplates;')\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def remove(self, container):\n # type: (dict) -> None\n \"\"\"Remove loaded container.\"\"\"\n # Delete container and its contents\n if cmds.objExists(container['objectName']):\n members = cmds.sets(container['objectName'], query=True) or []\n cmds.delete([container['objectName']] + members)\n\n # Remove the namespace, if empty\n namespace = container['namespace']\n if cmds.namespace(exists=namespace):\n members = cmds.namespaceInfo(namespace, listNamespace=True)\n if not members:\n cmds.namespace(removeNamespace=namespace)\n else:\n self.log.warning(\"Namespace not deleted because it \"\n \"still has members: %s\", namespace)\n"
},
{
"path": "openpype/hosts/maya/plugins/load/load_redshift_proxy.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Loader for Redshift proxy.\"\"\"\nimport os\nimport clique\n\nimport maya.cmds as cmds\n\nfrom openpype.settings import get_project_settings\nfrom openpype.pipeline import (\n load,\n get_representation_path\n)\nfrom openpype.hosts.maya.api.lib import (\n namespaced,\n maintained_selection,\n unique_namespace\n)\nfrom openpype.hosts.maya.api.pipeline import containerise\n\n\nclass RedshiftProxyLoader(load.LoaderPlugin):\n \"\"\"Load Redshift proxy\"\"\"\n\n families = [\"redshiftproxy\"]\n representations = [\"rs\"]\n\n label = \"Import Redshift Proxy\"\n order = -10\n icon = \"code-fork\"\n color = \"orange\"\n\n def load(self, context, name=None, namespace=None, options=None):\n \"\"\"Plugin entry point.\"\"\"\n try:\n family = context[\"representation\"][\"context\"][\"family\"]\n except ValueError:\n family = \"redshiftproxy\"\n\n asset_name = context['asset'][\"name\"]\n namespace = namespace or unique_namespace(\n asset_name + \"_\",\n prefix=\"_\" if asset_name[0].isdigit() else \"\",\n suffix=\"_\",\n )\n\n # Ensure Redshift for Maya is loaded.\n cmds.loadPlugin(\"redshift4maya\", quiet=True)\n\n path = self.filepath_from_context(context)\n with maintained_selection():\n cmds.namespace(addNamespace=namespace)\n with namespaced(namespace, new=False):\n nodes, group_node = self.create_rs_proxy(name, path)\n\n self[:] = nodes\n if not nodes:\n return\n\n # colour the group node\n project_name = context[\"project\"][\"name\"]\n settings = get_project_settings(project_name)\n colors = settings['maya']['load']['colors']\n c = colors.get(family)\n if c is not None:\n cmds.setAttr(\"{0}.useOutlinerColor\".format(group_node), 1)\n cmds.setAttr(\"{0}.outlinerColor\".format(group_node),\n c[0], c[1], c[2])\n\n return containerise(\n name=name,\n namespace=namespace,\n nodes=nodes,\n context=context,\n loader=self.__class__.__name__)\n\n def update(self, container, representation):\n\n node = container['objectName']\n assert cmds.objExists(node), \"Missing container\"\n\n members = cmds.sets(node, query=True) or []\n rs_meshes = cmds.ls(members, type=\"RedshiftProxyMesh\")\n assert rs_meshes, \"Cannot find RedshiftProxyMesh in container\"\n\n filename = get_representation_path(representation)\n\n for rs_mesh in rs_meshes:\n cmds.setAttr(\"{}.fileName\".format(rs_mesh),\n filename,\n type=\"string\")\n\n # Update metadata\n cmds.setAttr(\"{}.representation\".format(node),\n str(representation[\"_id\"]),\n type=\"string\")\n\n def remove(self, container):\n\n # Delete container and its contents\n if cmds.objExists(container['objectName']):\n members = cmds.sets(container['objectName'], query=True) or []\n cmds.delete([container['objectName']] + members)\n\n # Remove the namespace, if empty\n namespace = container['namespace']\n if cmds.namespace(exists=namespace):\n members = cmds.namespaceInfo(namespace, listNamespace=True)\n if not members:\n cmds.namespace(removeNamespace=namespace)\n else:\n self.log.warning(\"Namespace not deleted because it \"\n \"still has members: %s\", namespace)\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def create_rs_proxy(self, name, path):\n \"\"\"Creates Redshift Proxies showing a proxy object.\n\n Args:\n name (str): Proxy name.\n path (str): Path to proxy file.\n\n Returns:\n (str, str): Name of mesh with Redshift proxy and its parent\n transform.\n\n \"\"\"\n rs_mesh = cmds.createNode(\n 'RedshiftProxyMesh', name=\"{}_RS\".format(name))\n mesh_shape = cmds.createNode(\"mesh\", name=\"{}_GEOShape\".format(name))\n\n cmds.setAttr(\"{}.fileName\".format(rs_mesh),\n path,\n type=\"string\")\n\n cmds.connectAttr(\"{}.outMesh\".format(rs_mesh),\n \"{}.inMesh\".format(mesh_shape))\n\n # TODO: use the assigned shading group as shaders if existed\n # assign default shader to redshift proxy\n if cmds.ls(\"initialShadingGroup\", type=\"shadingEngine\"):\n cmds.sets(mesh_shape, forceElement=\"initialShadingGroup\")\n\n group_node = cmds.group(empty=True, name=\"{}_GRP\".format(name))\n mesh_transform = cmds.listRelatives(mesh_shape,\n parent=True, fullPath=True)\n cmds.parent(mesh_transform, group_node)\n nodes = [rs_mesh, mesh_shape, group_node]\n\n # determine if we need to enable animation support\n files_in_folder = os.listdir(os.path.dirname(path))\n collections, remainder = clique.assemble(files_in_folder)\n\n if collections:\n cmds.setAttr(\"{}.useFrameExtension\".format(rs_mesh), 1)\n\n return nodes, group_node\n"
},
{
"path": "openpype/hosts/maya/plugins/load/load_reference.py",
"content": "import os\nimport difflib\nimport contextlib\n\nfrom maya import cmds\nimport qargparse\n\nfrom openpype.settings import get_project_settings\nimport openpype.hosts.maya.api.plugin\nfrom openpype.hosts.maya.api.lib import (\n maintained_selection,\n get_container_members,\n parent_nodes,\n create_rig_animation_instance\n)\n\n\n@contextlib.contextmanager\ndef preserve_modelpanel_cameras(container, log=None):\n \"\"\"Preserve camera members of container in the modelPanels.\n\n This is used to ensure a camera remains in the modelPanels after updating\n to a new version.\n\n \"\"\"\n\n # Get the modelPanels that used the old camera\n members = get_container_members(container)\n old_cameras = set(cmds.ls(members, type=\"camera\", long=True))\n if not old_cameras:\n # No need to manage anything\n yield\n return\n\n panel_cameras = {}\n for panel in cmds.getPanel(type=\"modelPanel\"):\n cam = cmds.ls(cmds.modelPanel(panel, query=True, camera=True),\n long=True)[0]\n\n # Often but not always maya returns the transform from the\n # modelPanel as opposed to the camera shape, so we convert it\n # to explicitly be the camera shape\n if cmds.nodeType(cam) != \"camera\":\n cam = cmds.listRelatives(cam,\n children=True,\n fullPath=True,\n type=\"camera\")[0]\n if cam in old_cameras:\n panel_cameras[panel] = cam\n\n if not panel_cameras:\n # No need to manage anything\n yield\n return\n\n try:\n yield\n finally:\n new_members = get_container_members(container)\n new_cameras = set(cmds.ls(new_members, type=\"camera\", long=True))\n if not new_cameras:\n return\n\n for panel, cam_name in panel_cameras.items():\n new_camera = None\n if cam_name in new_cameras:\n new_camera = cam_name\n elif len(new_cameras) == 1:\n new_camera = next(iter(new_cameras))\n else:\n # Multiple cameras in the updated container but not an exact\n # match detected by name. Find the closest match\n matches = difflib.get_close_matches(word=cam_name,\n possibilities=new_cameras,\n n=1)\n if matches:\n new_camera = matches[0] # best match\n if log:\n log.info(\"Camera in '{}' restored with \"\n \"closest match camera: {} (before: {})\"\n .format(panel, new_camera, cam_name))\n\n if not new_camera:\n # Unable to find the camera to re-apply in the modelpanel\n continue\n\n cmds.modelPanel(panel, edit=True, camera=new_camera)\n\n\nclass ReferenceLoader(openpype.hosts.maya.api.plugin.ReferenceLoader):\n \"\"\"Reference file\"\"\"\n\n families = [\"model\",\n \"pointcache\",\n \"proxyAbc\",\n \"animation\",\n \"mayaAscii\",\n \"mayaScene\",\n \"setdress\",\n \"layout\",\n \"camera\",\n \"rig\",\n \"camerarig\",\n \"staticMesh\",\n \"skeletalMesh\",\n \"mvLook\",\n \"matchmove\"]\n\n representations = [\"ma\", \"abc\", \"fbx\", \"mb\"]\n\n label = \"Reference\"\n order = -10\n icon = \"code-fork\"\n color = \"orange\"\n\n def process_reference(self, context, name, namespace, options):\n import maya.cmds as cmds\n\n try:\n family = context[\"representation\"][\"context\"][\"family\"]\n except ValueError:\n family = \"model\"\n\n project_name = context[\"project\"][\"name\"]\n # True by default to keep legacy behaviours\n attach_to_root = options.get(\"attach_to_root\", True)\n group_name = options[\"group_name\"]\n\n # no group shall be created\n if not attach_to_root:\n group_name = namespace\n\n kwargs = {}\n if \"file_options\" in options:\n kwargs[\"options\"] = options[\"file_options\"]\n if \"file_type\" in options:\n kwargs[\"type\"] = options[\"file_type\"]\n\n path = self.filepath_from_context(context)\n with maintained_selection():\n cmds.loadPlugin(\"AbcImport.mll\", quiet=True)\n\n file_url = self.prepare_root_value(path, project_name)\n nodes = cmds.file(file_url,\n namespace=namespace,\n sharedReferenceFile=False,\n reference=True,\n returnNewNodes=True,\n groupReference=attach_to_root,\n groupName=group_name,\n **kwargs)\n\n shapes = cmds.ls(nodes, shapes=True, long=True)\n\n new_nodes = (list(set(nodes) - set(shapes)))\n\n # if there are cameras, try to lock their transforms\n self._lock_camera_transforms(new_nodes)\n\n current_namespace = cmds.namespaceInfo(currentNamespace=True)\n\n if current_namespace != \":\":\n group_name = current_namespace + \":\" + group_name\n\n self[:] = new_nodes\n\n if attach_to_root:\n group_name = \"|\" + group_name\n roots = cmds.listRelatives(group_name,\n children=True,\n fullPath=True) or []\n\n if family not in {\"layout\", \"setdress\",\n \"mayaAscii\", \"mayaScene\"}:\n # QUESTION Why do we need to exclude these families?\n with parent_nodes(roots, parent=None):\n cmds.xform(group_name, zeroTransformPivots=True)\n\n settings = get_project_settings(project_name)\n\n display_handle = settings['maya']['load'].get(\n 'reference_loader', {}\n ).get('display_handle', True)\n cmds.setAttr(\n \"{}.displayHandle\".format(group_name), display_handle\n )\n\n colors = settings['maya']['load']['colors']\n c = colors.get(family)\n if c is not None:\n cmds.setAttr(\"{}.useOutlinerColor\".format(group_name), 1)\n cmds.setAttr(\"{}.outlinerColor\".format(group_name),\n (float(c[0]) / 255),\n (float(c[1]) / 255),\n (float(c[2]) / 255))\n\n cmds.setAttr(\n \"{}.displayHandle\".format(group_name), display_handle\n )\n # get bounding box\n bbox = cmds.exactWorldBoundingBox(group_name)\n # get pivot position on world space\n pivot = cmds.xform(group_name, q=True, sp=True, ws=True)\n # center of bounding box\n cx = (bbox[0] + bbox[3]) / 2\n cy = (bbox[1] + bbox[4]) / 2\n cz = (bbox[2] + bbox[5]) / 2\n # add pivot position to calculate offset\n cx = cx + pivot[0]\n cy = cy + pivot[1]\n cz = cz + pivot[2]\n # set selection handle offset to center of bounding box\n cmds.setAttr(\"{}.selectHandleX\".format(group_name), cx)\n cmds.setAttr(\"{}.selectHandleY\".format(group_name), cy)\n cmds.setAttr(\"{}.selectHandleZ\".format(group_name), cz)\n\n if family == \"rig\":\n self._post_process_rig(namespace, context, options)\n else:\n if \"translate\" in options:\n if not attach_to_root and new_nodes:\n root_nodes = cmds.ls(new_nodes, assemblies=True,\n long=True)\n # we assume only a single root is ever loaded\n group_name = root_nodes[0]\n cmds.setAttr(\"{}.translate\".format(group_name),\n *options[\"translate\"])\n return new_nodes\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def update(self, container, representation):\n with preserve_modelpanel_cameras(container, log=self.log):\n super(ReferenceLoader, self).update(container, representation)\n\n # We also want to lock camera transforms on any new cameras in the\n # reference or for a camera which might have changed names.\n members = get_container_members(container)\n self._lock_camera_transforms(members)\n\n def _post_process_rig(self, namespace, context, options):\n\n nodes = self[:]\n create_rig_animation_instance(\n nodes, context, namespace, options=options, log=self.log\n )\n\n def _lock_camera_transforms(self, nodes):\n cameras = cmds.ls(nodes, type=\"camera\")\n if not cameras:\n return\n\n # Check the Maya version, lockTransform has been introduced since\n # Maya 2016.5 Ext 2\n version = int(cmds.about(version=True))\n if version >= 2016:\n for camera in cameras:\n cmds.camera(camera, edit=True, lockTransform=True)\n else:\n self.log.warning(\"This version of Maya does not support locking of\"\n \" transforms of cameras.\")\n\n\nclass MayaUSDReferenceLoader(ReferenceLoader):\n \"\"\"Reference USD file to native Maya nodes using MayaUSDImport reference\"\"\"\n\n label = \"Reference Maya USD\"\n families = [\"usd\"]\n representations = [\"usd\"]\n extensions = {\"usd\", \"usda\", \"usdc\"}\n\n options = ReferenceLoader.options + [\n qargparse.Boolean(\n \"readAnimData\",\n label=\"Load anim data\",\n default=True,\n help=\"Load animation data from USD file\"\n ),\n qargparse.Boolean(\n \"useAsAnimationCache\",\n label=\"Use as animation cache\",\n default=True,\n help=(\n \"Imports geometry prims with time-sampled point data using a \"\n \"point-based deformer that references the imported \"\n \"USD file.\\n\"\n \"This provides better import and playback performance when \"\n \"importing time-sampled geometry from USD, and should \"\n \"reduce the weight of the resulting Maya scene.\"\n )\n ),\n qargparse.Boolean(\n \"importInstances\",\n label=\"Import instances\",\n default=True,\n help=(\n \"Import USD instanced geometries as Maya instanced shapes. \"\n \"Will flatten the scene otherwise.\"\n )\n ),\n qargparse.String(\n \"primPath\",\n label=\"Prim Path\",\n default=\"/\",\n help=(\n \"Name of the USD scope where traversing will begin.\\n\"\n \"The prim at the specified primPath (including the prim) will \"\n \"be imported.\\n\"\n \"Specifying the pseudo-root (/) means you want \"\n \"to import everything in the file.\\n\"\n \"If the passed prim path is empty, it will first try to \"\n \"import the defaultPrim for the rootLayer if it exists.\\n\"\n \"Otherwise, it will behave as if the pseudo-root was passed \"\n \"in.\"\n )\n )\n ]\n\n file_type = \"USD Import\"\n\n def process_reference(self, context, name, namespace, options):\n cmds.loadPlugin(\"mayaUsdPlugin\", quiet=True)\n\n def bool_option(key, default):\n # Shorthand for getting optional boolean file option from options\n value = int(bool(options.get(key, default)))\n return \"{}={}\".format(key, value)\n\n def string_option(key, default):\n # Shorthand for getting optional string file option from options\n value = str(options.get(key, default))\n return \"{}={}\".format(key, value)\n\n options[\"file_options\"] = \";\".join([\n string_option(\"primPath\", default=\"/\"),\n bool_option(\"importInstances\", default=True),\n bool_option(\"useAsAnimationCache\", default=True),\n bool_option(\"readAnimData\", default=True),\n # TODO: Expose more parameters\n # \"preferredMaterial=none\",\n # \"importRelativeTextures=Automatic\",\n # \"useCustomFrameRange=0\",\n # \"startTime=0\",\n # \"endTime=0\",\n # \"importUSDZTextures=0\"\n ])\n options[\"file_type\"] = self.file_type\n\n return super(MayaUSDReferenceLoader, self).process_reference(\n context, name, namespace, options\n )\n"
},
{
"path": "openpype/hosts/maya/plugins/load/load_rendersetup.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Load and update RenderSetup settings.\n\nWorking with RenderSetup setting is Maya is done utilizing json files.\nWhen this json is loaded, it will overwrite all settings on RenderSetup\ninstance.\n\"\"\"\n\nimport json\nimport sys\nimport six\n\nfrom openpype.pipeline import (\n load,\n get_representation_path\n)\nfrom openpype.hosts.maya.api import lib\nfrom openpype.hosts.maya.api.pipeline import containerise\n\nfrom maya import cmds\nimport maya.app.renderSetup.model.renderSetup as renderSetup\n\n\nclass RenderSetupLoader(load.LoaderPlugin):\n \"\"\"Load json preset for RenderSetup overwriting current one.\"\"\"\n\n families = [\"rendersetup\"]\n representations = [\"json\"]\n defaults = ['Main']\n\n label = \"Load RenderSetup template\"\n icon = \"tablet\"\n color = \"orange\"\n\n def load(self, context, name, namespace, data):\n \"\"\"Load RenderSetup settings.\"\"\"\n\n # from openpype.hosts.maya.api.lib import namespaced\n\n asset = context['asset']['name']\n namespace = namespace or lib.unique_namespace(\n asset + \"_\",\n prefix=\"_\" if asset[0].isdigit() else \"\",\n suffix=\"_\",\n )\n path = self.filepath_from_context(context)\n self.log.info(\">>> loading json [ {} ]\".format(path))\n with open(path, \"r\") as file:\n renderSetup.instance().decode(\n json.load(file), renderSetup.DECODE_AND_OVERWRITE, None)\n\n nodes = []\n null = cmds.sets(name=\"null_SET\", empty=True)\n nodes.append(null)\n\n self[:] = nodes\n if not nodes:\n return\n\n self.log.info(\">>> containerising [ {} ]\".format(name))\n return containerise(\n name=name,\n namespace=namespace,\n nodes=nodes,\n context=context,\n loader=self.__class__.__name__)\n\n def remove(self, container):\n \"\"\"Remove RenderSetup settings instance.\"\"\"\n from maya import cmds\n\n container_name = container[\"objectName\"]\n\n self.log.info(\"Removing '%s' from Maya..\" % container[\"name\"])\n\n container_content = cmds.sets(container_name, query=True)\n nodes = cmds.ls(container_content, long=True)\n\n nodes.append(container_name)\n\n try:\n cmds.delete(nodes)\n except ValueError:\n # Already implicitly deleted by Maya upon removing reference\n pass\n\n def update(self, container, representation):\n \"\"\"Update RenderSetup setting by overwriting existing settings.\"\"\"\n lib.show_message(\n \"Render setup update\",\n \"Render setup setting will be overwritten by new version. All \"\n \"setting specified by user not included in loaded version \"\n \"will be lost.\")\n path = get_representation_path(representation)\n with open(path, \"r\") as file:\n try:\n renderSetup.instance().decode(\n json.load(file), renderSetup.DECODE_AND_OVERWRITE, None)\n except Exception:\n self.log.error(\"There were errors during loading\")\n six.reraise(*sys.exc_info())\n\n # Update metadata\n node = container[\"objectName\"]\n cmds.setAttr(\"{}.representation\".format(node),\n str(representation[\"_id\"]),\n type=\"string\")\n self.log.info(\"... updated\")\n\n def switch(self, container, representation):\n \"\"\"Switch representations.\"\"\"\n self.update(container, representation)\n"
},
{
"path": "openpype/hosts/maya/plugins/load/load_vdb_to_arnold.py",
"content": "import os\n\nfrom openpype.settings import get_project_settings\nfrom openpype.pipeline import (\n load,\n get_representation_path\n)\n# TODO aiVolume doesn't automatically set velocity fps correctly, set manual?\n\n\nclass LoadVDBtoArnold(load.LoaderPlugin):\n \"\"\"Load OpenVDB for Arnold in aiVolume\"\"\"\n\n families = [\"vdbcache\"]\n representations = [\"vdb\"]\n\n label = \"Load VDB to Arnold\"\n icon = \"cloud\"\n color = \"orange\"\n\n def load(self, context, name, namespace, data):\n\n from maya import cmds\n from openpype.hosts.maya.api.pipeline import containerise\n from openpype.hosts.maya.api.lib import unique_namespace\n\n try:\n family = context[\"representation\"][\"context\"][\"family\"]\n except ValueError:\n family = \"vdbcache\"\n\n # Check if the plugin for arnold is available on the pc\n try:\n cmds.loadPlugin(\"mtoa\", quiet=True)\n except Exception as exc:\n self.log.error(\"Encountered exception:\\n%s\" % exc)\n return\n\n asset = context['asset']\n asset_name = asset[\"name\"]\n namespace = namespace or unique_namespace(\n asset_name + \"_\",\n prefix=\"_\" if asset_name[0].isdigit() else \"\",\n suffix=\"_\",\n )\n\n # Root group\n label = \"{}:{}\".format(namespace, name)\n root = cmds.group(name=label, empty=True)\n\n project_name = context[\"project\"][\"name\"]\n settings = get_project_settings(project_name)\n colors = settings['maya']['load']['colors']\n\n c = colors.get(family)\n if c is not None:\n cmds.setAttr(root + \".useOutlinerColor\", 1)\n cmds.setAttr(root + \".outlinerColor\",\n (float(c[0]) / 255),\n (float(c[1]) / 255),\n (float(c[2]) / 255)\n )\n\n # Create VRayVolumeGrid\n grid_node = cmds.createNode(\"aiVolume\",\n name=\"{}Shape\".format(root),\n parent=root)\n\n path = self.filepath_from_context(context)\n self._set_path(grid_node,\n path=path,\n representation=context[\"representation\"])\n\n # Lock the shape node so the user can't delete the transform/shape\n # as if it was referenced\n cmds.lockNode(grid_node, lock=True)\n\n nodes = [root, grid_node]\n self[:] = nodes\n\n return containerise(\n name=name,\n namespace=namespace,\n nodes=nodes,\n context=context,\n loader=self.__class__.__name__)\n\n def update(self, container, representation):\n\n from maya import cmds\n\n path = get_representation_path(representation)\n\n # Find VRayVolumeGrid\n members = cmds.sets(container['objectName'], query=True)\n grid_nodes = cmds.ls(members, type=\"aiVolume\", long=True)\n assert len(grid_nodes) == 1, \"This is a bug\"\n\n # Update the VRayVolumeGrid\n self._set_path(grid_nodes[0], path=path, representation=representation)\n\n # Update container representation\n cmds.setAttr(container[\"objectName\"] + \".representation\",\n str(representation[\"_id\"]),\n type=\"string\")\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def remove(self, container):\n\n from maya import cmds\n\n # Get all members of the avalon container, ensure they are unlocked\n # and delete everything\n members = cmds.sets(container['objectName'], query=True)\n cmds.lockNode(members, lock=False)\n cmds.delete([container['objectName']] + members)\n\n # Clean up the namespace\n try:\n cmds.namespace(removeNamespace=container['namespace'],\n deleteNamespaceContent=True)\n except RuntimeError:\n pass\n\n @staticmethod\n def _set_path(grid_node,\n path,\n representation):\n \"\"\"Apply the settings for the VDB path to the aiVolume node\"\"\"\n from maya import cmds\n\n if not os.path.exists(path):\n raise RuntimeError(\"Path does not exist: %s\" % path)\n\n is_sequence = bool(representation[\"context\"].get(\"frame\"))\n cmds.setAttr(grid_node + \".useFrameExtension\", is_sequence)\n\n # Set file path\n cmds.setAttr(grid_node + \".filename\", path, type=\"string\")\n"
},
{
"path": "openpype/hosts/maya/plugins/load/load_vdb_to_redshift.py",
"content": "import os\n\nfrom openpype.settings import get_project_settings\nfrom openpype.pipeline import (\n load,\n get_representation_path\n)\n\n\nclass LoadVDBtoRedShift(load.LoaderPlugin):\n \"\"\"Load OpenVDB in a Redshift Volume Shape\n\n Note that the RedshiftVolumeShape is created without a RedshiftVolume\n shader assigned. To get the Redshift volume to render correctly assign\n a RedshiftVolume shader (in the Hypershade) and set the density, scatter\n and emission channels to the channel names of the volumes in the VDB file.\n\n \"\"\"\n\n families = [\"vdbcache\"]\n representations = [\"vdb\"]\n\n label = \"Load VDB to RedShift\"\n icon = \"cloud\"\n color = \"orange\"\n\n def load(self, context, name=None, namespace=None, data=None):\n\n from maya import cmds\n from openpype.hosts.maya.api.pipeline import containerise\n from openpype.hosts.maya.api.lib import unique_namespace\n\n try:\n family = context[\"representation\"][\"context\"][\"family\"]\n except ValueError:\n family = \"vdbcache\"\n\n # Check if the plugin for redshift is available on the pc\n try:\n cmds.loadPlugin(\"redshift4maya\", quiet=True)\n except Exception as exc:\n self.log.error(\"Encountered exception:\\n%s\" % exc)\n return\n\n # Check if viewport drawing engine is Open GL Core (compat)\n render_engine = None\n compatible = \"OpenGL\"\n if cmds.optionVar(exists=\"vp2RenderingEngine\"):\n render_engine = cmds.optionVar(query=\"vp2RenderingEngine\")\n\n if not render_engine or not render_engine.startswith(compatible):\n raise RuntimeError(\"Current scene's settings are incompatible.\"\n \"See Preferences > Display > Viewport 2.0 to \"\n \"set the render engine to '%s'\"\n % compatible)\n\n asset = context['asset']\n\n asset_name = asset[\"name\"]\n namespace = namespace or unique_namespace(\n asset_name + \"_\",\n prefix=\"_\" if asset_name[0].isdigit() else \"\",\n suffix=\"_\",\n )\n\n # Root group\n label = \"{}:{}\".format(namespace, name)\n root = cmds.createNode(\"transform\", name=label)\n\n project_name = context[\"project\"][\"name\"]\n settings = get_project_settings(project_name)\n colors = settings['maya']['load']['colors']\n\n c = colors.get(family)\n if c is not None:\n cmds.setAttr(root + \".useOutlinerColor\", 1)\n cmds.setAttr(root + \".outlinerColor\",\n (float(c[0])/255),\n (float(c[1])/255),\n (float(c[2])/255)\n )\n\n # Create VR\n volume_node = cmds.createNode(\"RedshiftVolumeShape\",\n name=\"{}RVSShape\".format(label),\n parent=root)\n\n self._set_path(volume_node,\n path=self.filepath_from_context(context),\n representation=context[\"representation\"])\n\n nodes = [root, volume_node]\n self[:] = nodes\n\n return containerise(\n name=name,\n namespace=namespace,\n nodes=nodes,\n context=context,\n loader=self.__class__.__name__)\n\n def update(self, container, representation):\n from maya import cmds\n\n path = get_representation_path(representation)\n\n # Find VRayVolumeGrid\n members = cmds.sets(container['objectName'], query=True)\n grid_nodes = cmds.ls(members, type=\"RedshiftVolumeShape\", long=True)\n assert len(grid_nodes) == 1, \"This is a bug\"\n\n # Update the VRayVolumeGrid\n self._set_path(grid_nodes[0], path=path, representation=representation)\n\n # Update container representation\n cmds.setAttr(container[\"objectName\"] + \".representation\",\n str(representation[\"_id\"]),\n type=\"string\")\n\n def remove(self, container):\n from maya import cmds\n\n # Get all members of the avalon container, ensure they are unlocked\n # and delete everything\n members = cmds.sets(container['objectName'], query=True)\n cmds.lockNode(members, lock=False)\n cmds.delete([container['objectName']] + members)\n\n # Clean up the namespace\n try:\n cmds.namespace(removeNamespace=container['namespace'],\n deleteNamespaceContent=True)\n except RuntimeError:\n pass\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n @staticmethod\n def _set_path(grid_node,\n path,\n representation):\n \"\"\"Apply the settings for the VDB path to the RedshiftVolumeShape\"\"\"\n from maya import cmds\n\n if not os.path.exists(path):\n raise RuntimeError(\"Path does not exist: %s\" % path)\n\n is_sequence = bool(representation[\"context\"].get(\"frame\"))\n cmds.setAttr(grid_node + \".useFrameExtension\", is_sequence)\n\n # Set file path\n cmds.setAttr(grid_node + \".fileName\", path, type=\"string\")\n"
},
{
"path": "openpype/hosts/maya/plugins/load/load_vdb_to_vray.py",
"content": "import os\n\nfrom openpype.settings import get_project_settings\nfrom openpype.pipeline import (\n load,\n get_representation_path\n)\n\nfrom maya import cmds\n\n# List of 3rd Party Channels Mapping names for VRayVolumeGrid\n# See: https://docs.chaosgroup.com/display/VRAY4MAYA/Input\n# #Input-3rdPartyChannelsMapping\nTHIRD_PARTY_CHANNELS = {\n 2: \"Smoke\",\n 1: \"Temperature\",\n 10: \"Fuel\",\n 4: \"Velocity.x\",\n 5: \"Velocity.y\",\n 6: \"Velocity.z\",\n 7: \"Red\",\n 8: \"Green\",\n 9: \"Blue\",\n 14: \"Wavelet Energy\",\n 19: \"Wavelet.u\",\n 20: \"Wavelet.v\",\n 21: \"Wavelet.w\",\n # These are not in UI or documentation but V-Ray does seem to set these.\n 15: \"AdvectionOrigin.x\",\n 16: \"AdvectionOrigin.y\",\n 17: \"AdvectionOrigin.z\",\n\n}\n\n\ndef _fix_duplicate_vvg_callbacks():\n \"\"\"Workaround to kill duplicate VRayVolumeGrids attribute callbacks.\n\n This fixes a huge lag in Maya on switching 3rd Party Channels Mappings\n or to different .vdb file paths because it spams an attribute changed\n callback: `vvgUserChannelMappingsUpdateUI`.\n\n ChaosGroup bug ticket: 154-008-9890\n\n Found with:\n - Maya 2019.2 on Windows 10\n - V-Ray: V-Ray Next for Maya, update 1 version 4.12.01.00001\n\n Bug still present in:\n - Maya 2022.1 on Windows 10\n - V-Ray 5 for Maya, Update 2.1 (v5.20.01 from Dec 16 2021)\n\n \"\"\"\n # todo(roy): Remove when new V-Ray release fixes duplicate calls\n\n jobs = cmds.scriptJob(listJobs=True)\n\n matched = set()\n for entry in jobs:\n # Remove the number\n index, callback = entry.split(\":\", 1)\n callback = callback.strip()\n\n # Detect whether it is a `vvgUserChannelMappingsUpdateUI`\n # attribute change callback\n if callback.startswith('\"-runOnce\" 1 \"-attributeChange\" \"'):\n if '\"vvgUserChannelMappingsUpdateUI(' in callback:\n if callback in matched:\n # If we've seen this callback before then\n # delete the duplicate callback\n cmds.scriptJob(kill=int(index))\n else:\n matched.add(callback)\n\n\nclass LoadVDBtoVRay(load.LoaderPlugin):\n \"\"\"Load OpenVDB in a V-Ray Volume Grid\"\"\"\n\n families = [\"vdbcache\"]\n representations = [\"vdb\"]\n\n label = \"Load VDB to VRay\"\n icon = \"cloud\"\n color = \"orange\"\n\n def load(self, context, name, namespace, data):\n\n from openpype.hosts.maya.api.lib import unique_namespace\n from openpype.hosts.maya.api.pipeline import containerise\n\n path = self.filepath_from_context(context)\n assert os.path.exists(path), (\n \"Path does not exist: %s\" % path\n )\n\n try:\n family = context[\"representation\"][\"context\"][\"family\"]\n except ValueError:\n family = \"vdbcache\"\n\n # Ensure V-ray is loaded with the vrayvolumegrid\n if not cmds.pluginInfo(\"vrayformaya\", query=True, loaded=True):\n cmds.loadPlugin(\"vrayformaya\")\n if not cmds.pluginInfo(\"vrayvolumegrid\", query=True, loaded=True):\n cmds.loadPlugin(\"vrayvolumegrid\")\n\n # Check if viewport drawing engine is Open GL Core (compat)\n render_engine = None\n compatible = \"OpenGLCoreProfileCompat\"\n if cmds.optionVar(exists=\"vp2RenderingEngine\"):\n render_engine = cmds.optionVar(query=\"vp2RenderingEngine\")\n\n if not render_engine or render_engine != compatible:\n self.log.warning(\"Current scene's settings are incompatible.\"\n \"See Preferences > Display > Viewport 2.0 to \"\n \"set the render engine to '%s'\" % compatible)\n\n asset = context['asset']\n asset_name = asset[\"name\"]\n namespace = namespace or unique_namespace(\n asset_name + \"_\",\n prefix=\"_\" if asset_name[0].isdigit() else \"\",\n suffix=\"_\",\n )\n\n # Root group\n label = \"{}:{}_VDB\".format(namespace, name)\n root = cmds.group(name=label, empty=True)\n\n project_name = context[\"project\"][\"name\"]\n settings = get_project_settings(project_name)\n colors = settings['maya']['load']['colors']\n\n c = colors.get(family)\n if c is not None:\n cmds.setAttr(root + \".useOutlinerColor\", 1)\n cmds.setAttr(root + \".outlinerColor\",\n float(c[0]) / 255,\n float(c[1]) / 255,\n float(c[2]) / 255)\n\n # Create VRayVolumeGrid\n grid_node = cmds.createNode(\"VRayVolumeGrid\",\n name=\"{}Shape\".format(label),\n parent=root)\n\n # Ensure .currentTime is connected to time1.outTime\n cmds.connectAttr(\"time1.outTime\", grid_node + \".currentTime\")\n\n # Set path\n self._set_path(grid_node, path, show_preset_popup=True)\n\n # Lock the shape node so the user can't delete the transform/shape\n # as if it was referenced\n cmds.lockNode(grid_node, lock=True)\n\n nodes = [root, grid_node]\n self[:] = nodes\n\n return containerise(\n name=name,\n namespace=namespace,\n nodes=nodes,\n context=context,\n loader=self.__class__.__name__)\n\n def _set_path(self, grid_node, path, show_preset_popup=True):\n\n from openpype.hosts.maya.api.lib import attribute_values\n from maya import cmds\n\n def _get_filename_from_folder(path):\n # Using the sequence of .vdb files we check the frame range, etc.\n # to set the filename with #### padding.\n files = sorted(x for x in os.listdir(path) if x.endswith(\".vdb\"))\n if not files:\n raise RuntimeError(\"Couldn't find .vdb files in: %s\" % path)\n\n if len(files) == 1:\n # Ensure check for single file is also done in folder\n fname = files[0]\n else:\n # Sequence\n import clique\n # todo: check support for negative frames as input\n collections, remainder = clique.assemble(files)\n assert len(collections) == 1, (\n \"Must find a single image sequence, \"\n \"found: %s\" % (collections,)\n )\n collection = collections[0]\n\n fname = collection.format('{head}{{padding}}{tail}')\n padding = collection.padding\n if padding == 0:\n # Clique doesn't provide padding if the frame number never\n # starts with a zero and thus has never any visual padding.\n # So we fall back to the smallest frame number as padding.\n padding = min(len(str(i)) for i in collection.indexes)\n\n # Supply frame/padding with # signs\n padding_str = \"#\" * padding\n fname = fname.format(padding=padding_str)\n\n return os.path.join(path, fname)\n\n # The path is either a single file or sequence in a folder so\n # we do a quick lookup for our files\n if os.path.isfile(path):\n path = os.path.dirname(path)\n path = _get_filename_from_folder(path)\n\n # Even when not applying a preset V-Ray will reset the 3rd Party\n # Channels Mapping of the VRayVolumeGrid when setting the .inPath\n # value. As such we try and preserve the values ourselves.\n # Reported as ChaosGroup bug ticket: 154-011-2909 \n # todo(roy): Remove when new V-Ray release preserves values\n original_user_mapping = cmds.getAttr(grid_node + \".usrchmap\") or \"\"\n\n # Workaround for V-Ray bug: fix lag on path change, see function\n _fix_duplicate_vvg_callbacks()\n\n # Suppress preset pop-up if we want.\n popup_attr = \"{0}.inDontOfferPresets\".format(grid_node)\n popup = {popup_attr: not show_preset_popup}\n with attribute_values(popup):\n cmds.setAttr(grid_node + \".inPath\", path, type=\"string\")\n\n # Reapply the 3rd Party channels user mapping when no preset popup\n # was shown to the user\n if not show_preset_popup:\n channels = cmds.getAttr(grid_node + \".usrchmapallch\").split(\";\")\n channels = set(channels) # optimize lookup\n restored_mapping = \"\"\n for entry in original_user_mapping.split(\";\"):\n if not entry:\n # Ignore empty entries\n continue\n\n # If 3rd Party Channels selection channel still exists then\n # add it again.\n index, channel = entry.split(\",\")\n attr = THIRD_PARTY_CHANNELS.get(int(index),\n # Fallback for when a mapping\n # was set that is not in the\n # documentation\n \"???\")\n if channel in channels:\n restored_mapping += entry + \";\"\n else:\n self.log.warning(\"Can't preserve '%s' mapping due to \"\n \"missing channel '%s' on node: \"\n \"%s\" % (attr, channel, grid_node))\n\n if restored_mapping:\n cmds.setAttr(grid_node + \".usrchmap\",\n restored_mapping,\n type=\"string\")\n\n def update(self, container, representation):\n\n path = get_representation_path(representation)\n\n # Find VRayVolumeGrid\n members = cmds.sets(container['objectName'], query=True)\n grid_nodes = cmds.ls(members, type=\"VRayVolumeGrid\", long=True)\n assert len(grid_nodes) > 0, \"This is a bug\"\n\n # Update the VRayVolumeGrid\n for grid_node in grid_nodes:\n self._set_path(grid_node, path=path, show_preset_popup=False)\n\n # Update container representation\n cmds.setAttr(container[\"objectName\"] + \".representation\",\n str(representation[\"_id\"]),\n type=\"string\")\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def remove(self, container):\n\n # Get all members of the avalon container, ensure they are unlocked\n # and delete everything\n members = cmds.sets(container['objectName'], query=True)\n cmds.lockNode(members, lock=False)\n cmds.delete([container['objectName']] + members)\n\n # Clean up the namespace\n try:\n cmds.namespace(removeNamespace=container['namespace'],\n deleteNamespaceContent=True)\n except RuntimeError:\n pass\n"
},
{
"path": "openpype/hosts/maya/plugins/load/load_vrayproxy.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Loader for Vray Proxy files.\n\nIf there are Alembics published along vray proxy (in the same version),\nloader will use them instead of native vray vrmesh format.\n\n\"\"\"\nimport os\n\nimport maya.cmds as cmds\n\nfrom openpype.client import get_representation_by_name\nfrom openpype.settings import get_project_settings\nfrom openpype.pipeline import (\n load,\n get_current_project_name,\n get_representation_path,\n)\nfrom openpype.hosts.maya.api.lib import (\n maintained_selection,\n namespaced,\n unique_namespace\n)\nfrom openpype.hosts.maya.api.pipeline import containerise\n\n\nclass VRayProxyLoader(load.LoaderPlugin):\n \"\"\"Load VRay Proxy with Alembic or VrayMesh.\"\"\"\n\n families = [\"vrayproxy\", \"model\", \"pointcache\", \"animation\"]\n representations = [\"vrmesh\", \"abc\"]\n\n label = \"Import VRay Proxy\"\n order = -10\n icon = \"code-fork\"\n color = \"orange\"\n\n def load(self, context, name=None, namespace=None, options=None):\n # type: (dict, str, str, dict) -> None\n \"\"\"Loader entry point.\n\n Args:\n context (dict): Loaded representation context.\n name (str): Name of container.\n namespace (str): Optional namespace name.\n options (dict): Optional loader options.\n\n \"\"\"\n\n try:\n family = context[\"representation\"][\"context\"][\"family\"]\n except ValueError:\n family = \"vrayproxy\"\n\n # get all representations for this version\n filename = self._get_abc(context[\"version\"][\"_id\"])\n if not filename:\n filename = self.filepath_from_context(context)\n\n asset_name = context['asset'][\"name\"]\n namespace = namespace or unique_namespace(\n asset_name + \"_\",\n prefix=\"_\" if asset_name[0].isdigit() else \"\",\n suffix=\"_\",\n )\n\n # Ensure V-Ray for Maya is loaded.\n cmds.loadPlugin(\"vrayformaya\", quiet=True)\n\n with maintained_selection():\n cmds.namespace(addNamespace=namespace)\n with namespaced(namespace, new=False):\n nodes, group_node = self.create_vray_proxy(\n name, filename=filename)\n\n self[:] = nodes\n if not nodes:\n return\n\n # colour the group node\n project_name = context[\"project\"][\"name\"]\n settings = get_project_settings(project_name)\n colors = settings['maya']['load']['colors']\n c = colors.get(family)\n if c is not None:\n cmds.setAttr(\"{0}.useOutlinerColor\".format(group_node), 1)\n cmds.setAttr(\n \"{0}.outlinerColor\".format(group_node),\n (float(c[0]) / 255),\n (float(c[1]) / 255),\n (float(c[2]) / 255)\n )\n\n return containerise(\n name=name,\n namespace=namespace,\n nodes=nodes,\n context=context,\n loader=self.__class__.__name__)\n\n def update(self, container, representation):\n # type: (dict, dict) -> None\n \"\"\"Update container with specified representation.\"\"\"\n node = container['objectName']\n assert cmds.objExists(node), \"Missing container\"\n\n members = cmds.sets(node, query=True) or []\n vraymeshes = cmds.ls(members, type=\"VRayProxy\")\n assert vraymeshes, \"Cannot find VRayMesh in container\"\n\n # get all representations for this version\n filename = (\n self._get_abc(representation[\"parent\"])\n or get_representation_path(representation)\n )\n\n for vray_mesh in vraymeshes:\n cmds.setAttr(\"{}.fileName\".format(vray_mesh),\n filename,\n type=\"string\")\n\n # Update metadata\n cmds.setAttr(\"{}.representation\".format(node),\n str(representation[\"_id\"]),\n type=\"string\")\n\n def remove(self, container):\n # type: (dict) -> None\n \"\"\"Remove loaded container.\"\"\"\n # Delete container and its contents\n if cmds.objExists(container['objectName']):\n members = cmds.sets(container['objectName'], query=True) or []\n cmds.delete([container['objectName']] + members)\n\n # Remove the namespace, if empty\n namespace = container['namespace']\n if cmds.namespace(exists=namespace):\n members = cmds.namespaceInfo(namespace, listNamespace=True)\n if not members:\n cmds.namespace(removeNamespace=namespace)\n else:\n self.log.warning(\"Namespace not deleted because it \"\n \"still has members: %s\", namespace)\n\n def switch(self, container, representation):\n # type: (dict, dict) -> None\n \"\"\"Switch loaded representation.\"\"\"\n self.update(container, representation)\n\n def create_vray_proxy(self, name, filename):\n # type: (str, str) -> (list, str)\n \"\"\"Re-create the structure created by VRay to support vrmeshes\n\n Args:\n name (str): Name of the asset.\n filename (str): File name of vrmesh.\n\n Returns:\n nodes(list)\n\n \"\"\"\n\n if name is None:\n name = os.path.splitext(os.path.basename(filename))[0]\n\n parent = cmds.createNode(\"transform\", name=name)\n proxy = cmds.createNode(\n \"VRayProxy\", name=\"{}Shape\".format(name), parent=parent)\n cmds.setAttr(proxy + \".fileName\", filename, type=\"string\")\n cmds.connectAttr(\"time1.outTime\", proxy + \".currentFrame\")\n\n return [parent, proxy], parent\n\n def _get_abc(self, version_id):\n # type: (str) -> str\n \"\"\"Get abc representation file path if present.\n\n If here is published Alembic (abc) representation published along\n vray proxy, get is file path.\n\n Args:\n version_id (str): Version hash id.\n\n Returns:\n str: Path to file.\n None: If abc not found.\n\n \"\"\"\n self.log.debug(\n \"Looking for abc in published representations of this version.\")\n project_name = get_current_project_name()\n abc_rep = get_representation_by_name(project_name, \"abc\", version_id)\n if abc_rep:\n self.log.debug(\"Found, we'll link alembic to vray proxy.\")\n file_name = get_representation_path(abc_rep)\n self.log.debug(\"File: {}\".format(file_name))\n return file_name\n\n return \"\"\n"
},
{
"path": "openpype/hosts/maya/plugins/load/load_vrayscene.py",
"content": "# -*- coding: utf-8 -*-\nimport os\nimport maya.cmds as cmds # noqa\nfrom openpype.settings import get_project_settings\nfrom openpype.pipeline import (\n load,\n get_representation_path\n)\nfrom openpype.hosts.maya.api.lib import (\n maintained_selection,\n namespaced,\n unique_namespace\n)\nfrom openpype.hosts.maya.api.pipeline import containerise\n\n\nclass VRaySceneLoader(load.LoaderPlugin):\n \"\"\"Load Vray scene\"\"\"\n\n families = [\"vrayscene_layer\"]\n representations = [\"vrscene\"]\n\n label = \"Import VRay Scene\"\n order = -10\n icon = \"code-fork\"\n color = \"orange\"\n\n def load(self, context, name, namespace, data):\n\n\n try:\n family = context[\"representation\"][\"context\"][\"family\"]\n except ValueError:\n family = \"vrayscene_layer\"\n\n asset_name = context['asset'][\"name\"]\n namespace = namespace or unique_namespace(\n asset_name + \"_\",\n prefix=\"_\" if asset_name[0].isdigit() else \"\",\n suffix=\"_\",\n )\n\n # Ensure V-Ray for Maya is loaded.\n cmds.loadPlugin(\"vrayformaya\", quiet=True)\n\n with maintained_selection():\n cmds.namespace(addNamespace=namespace)\n with namespaced(namespace, new=False):\n nodes, root_node = self.create_vray_scene(\n name,\n filename=self.filepath_from_context(context)\n )\n\n self[:] = nodes\n if not nodes:\n return\n\n # colour the group node\n project_name = context[\"project\"][\"name\"]\n settings = get_project_settings(project_name)\n colors = settings['maya']['load']['colors']\n c = colors.get(family)\n if c is not None:\n cmds.setAttr(\"{0}.useOutlinerColor\".format(root_node), 1)\n cmds.setAttr(\"{0}.outlinerColor\".format(root_node),\n (float(c[0])/255),\n (float(c[1])/255),\n (float(c[2])/255)\n )\n\n return containerise(\n name=name,\n namespace=namespace,\n nodes=nodes,\n context=context,\n loader=self.__class__.__name__)\n\n def update(self, container, representation):\n\n node = container['objectName']\n assert cmds.objExists(node), \"Missing container\"\n\n members = cmds.sets(node, query=True) or []\n vraymeshes = cmds.ls(members, type=\"VRayScene\")\n assert vraymeshes, \"Cannot find VRayScene in container\"\n\n filename = get_representation_path(representation)\n\n for vray_mesh in vraymeshes:\n cmds.setAttr(\"{}.FilePath\".format(vray_mesh),\n filename,\n type=\"string\")\n\n # Update metadata\n cmds.setAttr(\"{}.representation\".format(node),\n str(representation[\"_id\"]),\n type=\"string\")\n\n def remove(self, container):\n\n # Delete container and its contents\n if cmds.objExists(container['objectName']):\n members = cmds.sets(container['objectName'], query=True) or []\n cmds.delete([container['objectName']] + members)\n\n # Remove the namespace, if empty\n namespace = container['namespace']\n if cmds.namespace(exists=namespace):\n members = cmds.namespaceInfo(namespace, listNamespace=True)\n if not members:\n cmds.namespace(removeNamespace=namespace)\n else:\n self.log.warning(\"Namespace not deleted because it \"\n \"still has members: %s\", namespace)\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n def create_vray_scene(self, name, filename):\n \"\"\"Re-create the structure created by VRay to support vrscenes\n\n Args:\n name(str): name of the asset\n\n Returns:\n nodes(list)\n \"\"\"\n\n # Create nodes\n mesh_node_name = \"VRayScene_{}\".format(name)\n\n trans = cmds.createNode(\n \"transform\", name=mesh_node_name)\n vray_scene = cmds.createNode(\n \"VRayScene\", name=\"{}_VRSCN\".format(mesh_node_name), parent=trans)\n mesh = cmds.createNode(\n \"mesh\", name=\"{}_Shape\".format(mesh_node_name), parent=trans)\n\n cmds.connectAttr(\n \"{}.outMesh\".format(vray_scene), \"{}.inMesh\".format(mesh))\n\n cmds.setAttr(\"{}.FilePath\".format(vray_scene), filename, type=\"string\")\n\n # Lock the shape nodes so the user cannot delete these\n cmds.lockNode(mesh, lock=True)\n cmds.lockNode(vray_scene, lock=True)\n\n # Create important connections\n cmds.connectAttr(\"time1.outTime\",\n \"{0}.inputTime\".format(trans))\n\n # Connect mesh to initialShadingGroup\n cmds.sets([mesh], forceElement=\"initialShadingGroup\")\n\n nodes = [trans, vray_scene, mesh]\n\n # Fix: Force refresh so the mesh shows correctly after creation\n cmds.refresh()\n\n return nodes, trans\n"
},
{
"path": "openpype/hosts/maya/plugins/load/load_xgen.py",
"content": "import os\nimport shutil\n\nimport maya.cmds as cmds\nimport xgenm\n\nfrom qtpy import QtWidgets\n\nimport openpype.hosts.maya.api.plugin\nfrom openpype.hosts.maya.api.lib import (\n maintained_selection,\n get_container_members,\n attribute_values,\n write_xgen_file\n)\nfrom openpype.hosts.maya.api import current_file\nfrom openpype.pipeline import get_representation_path\n\n\nclass XgenLoader(openpype.hosts.maya.api.plugin.ReferenceLoader):\n \"\"\"Load Xgen as reference\"\"\"\n\n families = [\"xgen\"]\n representations = [\"ma\", \"mb\"]\n\n label = \"Reference Xgen\"\n icon = \"code-fork\"\n color = \"orange\"\n\n def get_xgen_xgd_paths(self, palette):\n _, maya_extension = os.path.splitext(current_file())\n xgen_file = current_file().replace(\n maya_extension,\n \"__{}.xgen\".format(palette.replace(\"|\", \"\").replace(\":\", \"__\"))\n )\n xgd_file = xgen_file.replace(\".xgen\", \".xgd\")\n return xgen_file, xgd_file\n\n def process_reference(self, context, name, namespace, options):\n # Validate workfile has a path.\n if current_file() is None:\n QtWidgets.QMessageBox.warning(\n None,\n \"\",\n \"Current workfile has not been saved. Please save the workfile\"\n \" before loading an Xgen.\"\n )\n return\n\n maya_filepath = self.prepare_root_value(\n file_url=self.filepath_from_context(context),\n project_name=context[\"project\"][\"name\"]\n )\n\n # Reference xgen. Xgen does not like being referenced in under a group.\n with maintained_selection():\n nodes = cmds.file(\n maya_filepath,\n namespace=namespace,\n sharedReferenceFile=False,\n reference=True,\n returnNewNodes=True\n )\n\n xgen_palette = cmds.ls(\n nodes, type=\"xgmPalette\", long=True\n )[0].replace(\"|\", \"\")\n\n xgen_file, xgd_file = self.get_xgen_xgd_paths(xgen_palette)\n self.set_palette_attributes(xgen_palette, xgen_file, xgd_file)\n\n # Change the cache and disk values of xgDataPath and xgProjectPath\n # to ensure paths are setup correctly.\n project_path = os.path.dirname(current_file()).replace(\"\\\\\", \"/\")\n xgenm.setAttr(\"xgProjectPath\", project_path, xgen_palette)\n data_path = \"${{PROJECT}}xgen/collections/{};{}\".format(\n xgen_palette.replace(\":\", \"__ns__\"),\n xgenm.getAttr(\"xgDataPath\", xgen_palette)\n )\n xgenm.setAttr(\"xgDataPath\", data_path, xgen_palette)\n\n data = {\"xgProjectPath\": project_path, \"xgDataPath\": data_path}\n write_xgen_file(data, xgen_file)\n\n # This create an expression attribute of float. If we did not add\n # any changes to collection, then Xgen does not create an xgd file\n # on save. This gives errors when launching the workfile again due\n # to trying to find the xgd file.\n name = \"custom_float_ignore\"\n if name not in xgenm.customAttrs(xgen_palette):\n xgenm.addCustomAttr(\n \"custom_float_ignore\", xgen_palette\n )\n\n shapes = cmds.ls(nodes, shapes=True, long=True)\n\n new_nodes = (list(set(nodes) - set(shapes)))\n\n self[:] = new_nodes\n\n return new_nodes\n\n def set_palette_attributes(self, xgen_palette, xgen_file, xgd_file):\n cmds.setAttr(\n \"{}.xgBaseFile\".format(xgen_palette),\n os.path.basename(xgen_file),\n type=\"string\"\n )\n cmds.setAttr(\n \"{}.xgFileName\".format(xgen_palette),\n os.path.basename(xgd_file),\n type=\"string\"\n )\n cmds.setAttr(\"{}.xgExportAsDelta\".format(xgen_palette), True)\n\n def update(self, container, representation):\n \"\"\"Workflow for updating Xgen.\n\n - Export changes to delta file.\n - Copy and overwrite the workspace .xgen file.\n - Set collection attributes to not include delta files.\n - Update xgen maya file reference.\n - Apply the delta file changes.\n - Reset collection attributes to include delta files.\n\n We have to do this workflow because when using referencing of the xgen\n collection, Maya implicitly imports the Xgen data from the xgen file so\n we dont have any control over when adding the delta file changes.\n\n There is an implicit increment of the xgen and delta files, due to\n using the workfile basename.\n \"\"\"\n # Storing current description to try and maintain later.\n current_description = (\n xgenm.xgGlobal.DescriptionEditor.currentDescription()\n )\n\n container_node = container[\"objectName\"]\n members = get_container_members(container_node)\n xgen_palette = cmds.ls(\n members, type=\"xgmPalette\", long=True\n )[0].replace(\"|\", \"\")\n xgen_file, xgd_file = self.get_xgen_xgd_paths(xgen_palette)\n\n # Export current changes to apply later.\n xgenm.createDelta(xgen_palette.replace(\"|\", \"\"), xgd_file)\n\n self.set_palette_attributes(xgen_palette, xgen_file, xgd_file)\n\n maya_file = get_representation_path(representation)\n _, extension = os.path.splitext(maya_file)\n new_xgen_file = maya_file.replace(extension, \".xgen\")\n data_path = \"\"\n with open(new_xgen_file, \"r\") as f:\n for line in f:\n if line.startswith(\"\\txgDataPath\"):\n line = line.rstrip()\n data_path = line.split(\"\\t\")[-1]\n break\n\n project_path = os.path.dirname(current_file()).replace(\"\\\\\", \"/\")\n data_path = \"${{PROJECT}}xgen/collections/{};{}\".format(\n xgen_palette.replace(\":\", \"__ns__\"),\n data_path\n )\n data = {\"xgProjectPath\": project_path, \"xgDataPath\": data_path}\n shutil.copy(new_xgen_file, xgen_file)\n write_xgen_file(data, xgen_file)\n\n attribute_data = {\n \"{}.xgFileName\".format(xgen_palette): os.path.basename(xgen_file),\n \"{}.xgBaseFile\".format(xgen_palette): \"\",\n \"{}.xgExportAsDelta\".format(xgen_palette): False\n }\n with attribute_values(attribute_data):\n super().update(container, representation)\n\n xgenm.applyDelta(xgen_palette.replace(\"|\", \"\"), xgd_file)\n\n # Restore current selected description if it exists.\n if cmds.objExists(current_description):\n xgenm.xgGlobal.DescriptionEditor.setCurrentDescription(\n current_description\n )\n # Full UI refresh.\n xgenm.xgGlobal.DescriptionEditor.refresh(\"Full\")\n"
},
{
"path": "openpype/hosts/maya/plugins/load/load_yeti_cache.py",
"content": "import os\nimport json\nimport re\nfrom collections import defaultdict\n\nimport clique\nfrom maya import cmds\n\nfrom openpype.settings import get_project_settings\nfrom openpype.pipeline import (\n load,\n get_representation_path\n)\nfrom openpype.hosts.maya.api import lib\nfrom openpype.hosts.maya.api.pipeline import containerise\n\n\n# Do not reset these values on update but only apply on first load\n# to preserve any potential local overrides\nSKIP_UPDATE_ATTRS = {\n \"displayOutput\",\n \"viewportDensity\",\n \"viewportWidth\",\n \"viewportLength\",\n}\n\n\ndef set_attribute(node, attr, value):\n \"\"\"Wrapper of set attribute which ignores None values\"\"\"\n if value is None:\n return\n lib.set_attribute(node, attr, value)\n\n\nclass YetiCacheLoader(load.LoaderPlugin):\n \"\"\"Load Yeti Cache with one or more Yeti nodes\"\"\"\n\n families = [\"yeticache\", \"yetiRig\"]\n representations = [\"fur\"]\n\n label = \"Load Yeti Cache\"\n order = -9\n icon = \"code-fork\"\n color = \"orange\"\n\n def load(self, context, name=None, namespace=None, data=None):\n \"\"\"Loads a .fursettings file defining how to load .fur sequences\n\n A single yeticache or yetiRig can have more than a single pgYetiMaya\n nodes and thus load more than a single yeti.fur sequence.\n\n The .fursettings file defines what the node names should be and also\n what \"cbId\" attribute they should receive to match the original source\n and allow published looks to also work for Yeti rigs and its caches.\n\n \"\"\"\n\n try:\n family = context[\"representation\"][\"context\"][\"family\"]\n except ValueError:\n family = \"yeticache\"\n\n # Build namespace\n asset = context[\"asset\"]\n if namespace is None:\n namespace = self.create_namespace(asset[\"name\"])\n\n # Ensure Yeti is loaded\n if not cmds.pluginInfo(\"pgYetiMaya\", query=True, loaded=True):\n cmds.loadPlugin(\"pgYetiMaya\", quiet=True)\n\n # Create Yeti cache nodes according to settings\n path = self.filepath_from_context(context)\n settings = self.read_settings(path)\n nodes = []\n for node in settings[\"nodes\"]:\n nodes.extend(self.create_node(namespace, node))\n\n group_name = \"{}:{}\".format(namespace, name)\n group_node = cmds.group(nodes, name=group_name)\n project_name = context[\"project\"][\"name\"]\n\n settings = get_project_settings(project_name)\n colors = settings['maya']['load']['colors']\n\n c = colors.get(family)\n if c is not None:\n cmds.setAttr(group_node + \".useOutlinerColor\", 1)\n cmds.setAttr(group_node + \".outlinerColor\",\n (float(c[0])/255),\n (float(c[1])/255),\n (float(c[2])/255)\n )\n\n nodes.append(group_node)\n\n self[:] = nodes\n\n return containerise(\n name=name,\n namespace=namespace,\n nodes=nodes,\n context=context,\n loader=self.__class__.__name__\n )\n\n def remove(self, container):\n\n from maya import cmds\n\n namespace = container[\"namespace\"]\n container_name = container[\"objectName\"]\n\n self.log.info(\"Removing '%s' from Maya..\" % container[\"name\"])\n\n container_content = cmds.sets(container_name, query=True)\n nodes = cmds.ls(container_content, long=True)\n\n nodes.append(container_name)\n\n try:\n cmds.delete(nodes)\n except ValueError:\n # Already implicitly deleted by Maya upon removing reference\n pass\n\n cmds.namespace(removeNamespace=namespace, deleteNamespaceContent=True)\n\n def update(self, container, representation):\n\n namespace = container[\"namespace\"]\n container_node = container[\"objectName\"]\n\n path = get_representation_path(representation)\n settings = self.read_settings(path)\n\n # Collect scene information of asset\n set_members = lib.get_container_members(container)\n container_root = lib.get_container_transforms(container,\n members=set_members,\n root=True)\n scene_nodes = cmds.ls(set_members, type=\"pgYetiMaya\", long=True)\n\n # Build lookup with cbId as keys\n scene_lookup = defaultdict(list)\n for node in scene_nodes:\n cb_id = lib.get_id(node)\n scene_lookup[cb_id].append(node)\n\n # Re-assemble metadata with cbId as keys\n meta_data_lookup = {n[\"cbId\"]: n for n in settings[\"nodes\"]}\n\n # Delete nodes by \"cbId\" that are not in the updated version\n to_delete_lookup = {cb_id for cb_id in scene_lookup.keys() if\n cb_id not in meta_data_lookup}\n if to_delete_lookup:\n\n # Get nodes and remove entry from lookup\n to_remove = []\n for _id in to_delete_lookup:\n # Get all related nodes\n shapes = scene_lookup[_id]\n # Get the parents of all shapes under the ID\n transforms = cmds.listRelatives(shapes,\n parent=True,\n fullPath=True) or []\n to_remove.extend(shapes + transforms)\n\n # Remove id from lookup\n scene_lookup.pop(_id, None)\n\n cmds.delete(to_remove)\n\n for cb_id, node_settings in meta_data_lookup.items():\n\n if cb_id not in scene_lookup:\n # Create new nodes\n self.log.info(\"Creating new nodes ..\")\n\n new_nodes = self.create_node(namespace, node_settings)\n cmds.sets(new_nodes, addElement=container_node)\n cmds.parent(new_nodes, container_root)\n\n else:\n # Update the matching nodes\n scene_nodes = scene_lookup[cb_id]\n lookup_result = meta_data_lookup[cb_id][\"name\"]\n\n # Remove namespace if any (e.g.: \"character_01_:head_YNShape\")\n node_name = lookup_result.rsplit(\":\", 1)[-1]\n\n for scene_node in scene_nodes:\n\n # Get transform node, this makes renaming easier\n transforms = cmds.listRelatives(scene_node,\n parent=True,\n fullPath=True) or []\n assert len(transforms) == 1, \"This is a bug!\"\n\n # Get scene node's namespace and rename the transform node\n lead = scene_node.rsplit(\":\", 1)[0]\n namespace = \":{}\".format(lead.rsplit(\"|\")[-1])\n\n new_shape_name = \"{}:{}\".format(namespace, node_name)\n new_trans_name = new_shape_name.rsplit(\"Shape\", 1)[0]\n\n transform_node = transforms[0]\n cmds.rename(transform_node,\n new_trans_name,\n ignoreShape=False)\n\n # Get the newly named shape node\n yeti_nodes = cmds.listRelatives(new_trans_name,\n children=True)\n yeti_node = yeti_nodes[0]\n\n for attr, value in node_settings[\"attrs\"].items():\n if attr in SKIP_UPDATE_ATTRS:\n continue\n set_attribute(attr, value, yeti_node)\n\n cmds.setAttr(\"{}.representation\".format(container_node),\n str(representation[\"_id\"]),\n typ=\"string\")\n\n def switch(self, container, representation):\n self.update(container, representation)\n\n # helper functions\n def create_namespace(self, asset):\n \"\"\"Create a unique namespace\n Args:\n asset (dict): asset information\n\n \"\"\"\n\n asset_name = \"{}_\".format(asset)\n prefix = \"_\" if asset_name[0].isdigit()else \"\"\n namespace = lib.unique_namespace(\n asset_name,\n prefix=prefix,\n suffix=\"_\"\n )\n\n return namespace\n\n def get_cache_node_filepath(self, root, node_name):\n \"\"\"Get the cache file path for one of the yeti nodes.\n\n All caches with more than 1 frame need cache file name set with `%04d`\n If the cache has only one frame we return the file name as we assume\n it is a snapshot.\n\n This expects the files to be named after the \"node name\" through\n exports with in Yeti.\n\n Args:\n root(str): Folder containing cache files to search in.\n node_name(str): Node name to search cache files for\n\n Returns:\n str: Cache file path value needed for cacheFileName attribute\n\n \"\"\"\n\n name = node_name.replace(\":\", \"_\")\n pattern = r\"^({name})(\\.[0-9]+)?(\\.fur)$\".format(name=re.escape(name))\n\n files = [fname for fname in os.listdir(root) if re.match(pattern,\n fname)]\n if not files:\n self.log.error(\"Could not find cache files for '{}' \"\n \"with pattern {}\".format(node_name, pattern))\n return\n\n if len(files) == 1:\n # Single file\n return os.path.join(root, files[0])\n\n # Get filename for the sequence with padding\n collections, remainder = clique.assemble(files)\n assert not remainder, \"This is a bug\"\n assert len(collections) == 1, \"This is a bug\"\n collection = collections[0]\n\n # Formats name as {head}%d{tail} like cache.%04d.fur\n fname = collection.format(\"{head}{padding}{tail}\")\n return os.path.join(root, fname)\n\n def create_node(self, namespace, node_settings):\n \"\"\"Create nodes with the correct namespace and settings\n\n Args:\n namespace(str): namespace\n node_settings(dict): Single \"nodes\" entry from .fursettings file.\n\n Returns:\n list: Created nodes\n\n \"\"\"\n nodes = []\n\n # Get original names and ids\n orig_transform_name = node_settings[\"transform\"][\"name\"]\n orig_shape_name = node_settings[\"name\"]\n\n # Add namespace\n transform_name = \"{}:{}\".format(namespace, orig_transform_name)\n shape_name = \"{}:{}\".format(namespace, orig_shape_name)\n\n # Create pgYetiMaya node\n transform_node = cmds.createNode(\"transform\",\n name=transform_name)\n yeti_node = cmds.createNode(\"pgYetiMaya\",\n name=shape_name,\n parent=transform_node)\n\n lib.set_id(transform_node, node_settings[\"transform\"][\"cbId\"])\n lib.set_id(yeti_node, node_settings[\"cbId\"])\n\n nodes.extend([transform_node, yeti_node])\n\n # Update attributes with defaults\n attributes = node_settings[\"attrs\"]\n attributes.update({\n \"verbosity\": 2,\n \"fileMode\": 1,\n\n # Fix render stats, like Yeti's own\n # ../scripts/pgYetiNode.mel script\n \"visibleInReflections\": True,\n \"visibleInRefractions\": True\n })\n\n if \"viewportDensity\" not in attributes:\n attributes[\"viewportDensity\"] = 0.1\n\n # Apply attributes to pgYetiMaya node\n for attr, value in attributes.items():\n set_attribute(attr, value, yeti_node)\n\n # Connect to the time node\n cmds.connectAttr(\"time1.outTime\", \"%s.currentTime\" % yeti_node)\n\n return nodes\n\n def read_settings(self, path):\n \"\"\"Read .fursettings file and compute some additional attributes\"\"\"\n\n with open(path, \"r\") as fp:\n fur_settings = json.load(fp)\n\n if \"nodes\" not in fur_settings:\n raise RuntimeError(\"Encountered invalid data, \"\n \"expected 'nodes' in fursettings.\")\n\n # Compute the cache file name values we want to set for the nodes\n root = os.path.dirname(path)\n for node in fur_settings[\"nodes\"]:\n cache_filename = self.get_cache_node_filepath(\n root=root, node_name=node[\"name\"])\n\n attrs = node.get(\"attrs\", {}) # allow 'attrs' to not exist\n attrs[\"cacheFileName\"] = cache_filename\n node[\"attrs\"] = attrs\n\n return fur_settings\n"
},
{
"path": "openpype/hosts/maya/plugins/load/load_yeti_rig.py",
"content": "import maya.cmds as cmds\n\nfrom openpype.settings import get_current_project_settings\nimport openpype.hosts.maya.api.plugin\nfrom openpype.hosts.maya.api import lib\n\n\nclass YetiRigLoader(openpype.hosts.maya.api.plugin.ReferenceLoader):\n \"\"\"This loader will load Yeti rig.\"\"\"\n\n families = [\"yetiRig\"]\n representations = [\"ma\"]\n\n label = \"Load Yeti Rig\"\n order = -9\n icon = \"code-fork\"\n color = \"orange\"\n\n def process_reference(\n self, context, name=None, namespace=None, options=None\n ):\n path = self.filepath_from_context(context)\n\n attach_to_root = options.get(\"attach_to_root\", True)\n group_name = options[\"group_name\"]\n\n # no group shall be created\n if not attach_to_root:\n group_name = namespace\n\n with lib.maintained_selection():\n file_url = self.prepare_root_value(\n path, context[\"project\"][\"name\"]\n )\n nodes = cmds.file(\n file_url,\n namespace=namespace,\n reference=True,\n returnNewNodes=True,\n groupReference=attach_to_root,\n groupName=group_name\n )\n\n settings = get_current_project_settings()\n colors = settings[\"maya\"][\"load\"][\"colors\"]\n c = colors.get(\"yetiRig\")\n if c is not None:\n cmds.setAttr(group_name + \".useOutlinerColor\", 1)\n cmds.setAttr(\n group_name + \".outlinerColor\",\n (float(c[0]) / 255), (float(c[1]) / 255), (float(c[2]) / 255)\n )\n self[:] = nodes\n\n return nodes\n"
},
{
"path": "openpype/hosts/maya/plugins/publish/__init__.py",
"content": ""
},
{
"path": "openpype/hosts/maya/plugins/publish/collect_animation.py",
"content": "import pyblish.api\n\nimport maya.cmds as cmds\n\n\nclass CollectAnimationOutputGeometry(pyblish.api.InstancePlugin):\n \"\"\"Collect out hierarchy data for instance.\n\n Collect all hierarchy nodes which reside in the out_SET of the animation\n instance or point cache instance. This is to unify the logic of retrieving\n that specific data. This eliminates the need to write two separate pieces\n of logic to fetch all hierarchy nodes.\n\n Results in a list of nodes from the content of the instances\n\n \"\"\"\n\n order = pyblish.api.CollectorOrder + 0.4\n families = [\"animation\"]\n label = \"Collect Animation\"\n hosts = [\"maya\"]\n\n ignore_type = [\"constraints\"]\n\n def process(self, instance):\n \"\"\"Collect the hierarchy nodes\"\"\"\n\n family = instance.data[\"family\"]\n out_set = next((i for i in instance.data[\"setMembers\"] if\n i.endswith(\"out_SET\")), None)\n\n if out_set is None:\n warning = \"Expecting out_SET for instance of family '%s'\" % family\n self.log.warning(warning)\n return\n\n members = cmds.ls(cmds.sets(out_set, query=True), long=True)\n\n # Get all the relatives of the members\n descendants = cmds.listRelatives(members,\n allDescendents=True,\n fullPath=True) or []\n descendants = cmds.ls(descendants, noIntermediate=True, long=True)\n\n # Add members and descendants together for a complete overview\n\n hierarchy = members + descendants\n\n # Ignore certain node types (e.g. constraints)\n ignore = cmds.ls(hierarchy, type=self.ignore_type, long=True)\n if ignore:\n ignore = set(ignore)\n hierarchy = [node for node in hierarchy if node not in ignore]\n\n # Store data in the instance for the validator\n instance.data[\"out_hierarchy\"] = hierarchy\n\n if instance.data.get(\"farm\"):\n instance.data[\"families\"].append(\"publish.farm\")\n\n # User defined attributes.\n instance.data[\"includeUserDefinedAttributes\"] = (\n instance.data[\"creator_attributes\"][\"includeUserDefinedAttributes\"]\n )\n"
},
{
"path": "openpype/hosts/maya/plugins/publish/collect_arnold_scene_source.py",
"content": "from maya import cmds\n\nimport pyblish.api\nfrom openpype.hosts.maya.api.lib import get_all_children\n\n\nclass CollectArnoldSceneSource(pyblish.api.InstancePlugin):\n \"\"\"Collect Arnold Scene Source data.\"\"\"\n\n # Offset to be after renderable camera collection.\n order = pyblish.api.CollectorOrder + 0.2\n label = \"Collect Arnold Scene Source\"\n families = [\"ass\", \"assProxy\"]\n\n def process(self, instance):\n instance.data[\"members\"] = []\n for set_member in instance.data[\"setMembers\"]:\n if cmds.nodeType(set_member) != \"objectSet\":\n instance.data[\"members\"].extend(self.get_hierarchy(set_member))\n continue\n\n members = cmds.sets(set_member, query=True)\n members = cmds.ls(members, long=True)\n if members is None:\n self.log.warning(\n \"Skipped empty instance: \\\"%s\\\" \" % set_member\n )\n continue\n if set_member.endswith(\"proxy_SET\"):\n instance.data[\"proxy\"] = self.get_hierarchy(members)\n\n # Use camera in object set if present else default to render globals\n # camera.\n cameras = cmds.ls(type=\"camera\", long=True)\n renderable = [c for c in cameras if cmds.getAttr(\"%s.renderable\" % c)]\n if renderable:\n camera = renderable[0]\n for node in instance.data[\"members\"]:\n camera_shapes = cmds.listRelatives(\n node, shapes=True, type=\"camera\"\n )\n if camera_shapes:\n camera = node\n instance.data[\"camera\"] = camera\n else:\n self.log.debug(\"No renderable cameras found.\")\n\n self.log.debug(\"data: {}\".format(instance.data))\n\n def get_hierarchy(self, nodes):\n \"\"\"Return nodes with all their children\"\"\"\n nodes = cmds.ls(nodes, long=True)\n if not nodes:\n return []\n children = get_all_children(nodes)\n # Make sure nodes merged with children only\n # contains unique entries\n return list(set(nodes + children))\n"
},
{
"path": "openpype/hosts/maya/plugins/publish/collect_assembly.py",
"content": "from collections import defaultdict\nimport pyblish.api\n\nfrom maya import cmds, mel\nfrom openpype.hosts.maya import api\nfrom openpype.hosts.maya.api import lib\n\n# TODO : Publish of assembly: -unique namespace for all assets, VALIDATOR!\n\n\nclass CollectAssembly(pyblish.api.InstancePlugin):\n \"\"\"Collect all relevant assembly items\n\n Collected data:\n\n * File name\n * Compatible loader\n * Matrix per instance\n * Namespace\n\n Note: GPU caches are currently not supported in the pipeline. There is no\n logic yet which supports the swapping of GPU cache to renderable objects.\n\n \"\"\"\n\n order = pyblish.api.CollectorOrder + 0.49\n label = \"Assembly\"\n families = [\"assembly\"]\n\n def process(self, instance):\n\n # Find containers\n containers = api.ls()\n\n # Get all content from the instance\n instance_lookup = set(cmds.ls(instance, type=\"transform\", long=True))\n data = defaultdict(list)\n\n hierarchy_nodes = []\n for container in containers:\n\n root = lib.get_container_transforms(container, root=True)\n if not root or root not in instance_lookup:\n continue\n\n # Retrieve the hierarchy\n parent = cmds.listRelatives(root, parent=True, fullPath=True)[0]\n hierarchy_nodes.append(parent)\n\n # Temporary warning for GPU cache which are not supported yet\n loader = container[\"loader\"]\n if loader == \"GpuCacheLoader\":\n self.log.warning(\"GPU Cache Loader is currently not supported\"\n \"in the pipeline, we will export it tho\")\n\n # Gather info for new data entry\n representation_id = container[\"representation\"]\n instance_data = {\"loader\": loader,\n \"parent\": parent,\n \"namespace\": container[\"namespace\"]}\n\n # Check if matrix differs from default and store changes\n matrix_data = self.get_matrix_data(root)\n if matrix_data:\n instance_data[\"matrix\"] = matrix_data\n\n data[representation_id].append(instance_data)\n\n instance.data[\"scenedata\"] = dict(data)\n instance.data[\"nodesHierarchy\"] = list(set(hierarchy_nodes))\n\n def get_file_rule(self, rule):\n return mel.eval('workspace -query -fileRuleEntry \"{}\"'.format(rule))\n\n def get_matrix_data(self, node):\n \"\"\"Get the matrix of all members when they are not default\n\n Each matrix which differs from the default will be stored in a\n dictionary\n\n Args:\n members (list): list of transform nmodes\n Returns:\n dict\n \"\"\"\n\n matrix = cmds.xform(node, query=True, matrix=True)\n if matrix == lib.DEFAULT_MATRIX:\n return\n\n return matrix\n"
},
{
"path": "openpype/hosts/maya/plugins/publish/collect_current_file.py",
"content": "\nimport pyblish.api\n\nfrom maya import cmds\n\n\nclass CollectCurrentFile(pyblish.api.ContextPlugin):\n \"\"\"Inject the current working file.\"\"\"\n\n order = pyblish.api.CollectorOrder - 0.4\n label = \"Maya Current File\"\n hosts = ['maya']\n\n def process(self, context):\n \"\"\"Inject the current working file\"\"\"\n context.data['currentFile'] = cmds.file(query=True, sceneName=True)\n"
},
{
"path": "openpype/hosts/maya/plugins/publish/collect_fbx_animation.py",
"content": "# -*- coding: utf-8 -*-\nfrom maya import cmds # noqa\nimport pyblish.api\nfrom openpype.pipeline import OptionalPyblishPluginMixin\n\n\nclass CollectFbxAnimation(pyblish.api.InstancePlugin,\n OptionalPyblishPluginMixin):\n \"\"\"Collect Animated Rig Data for FBX Extractor.\"\"\"\n\n order = pyblish.api.CollectorOrder + 0.2\n label = \"Collect Fbx Animation\"\n hosts = [\"maya\"]\n families = [\"animation\"]\n optional = True\n\n def process(self, instance):\n if not self.is_active(instance.data):\n return\n skeleton_sets = [\n i for i in instance\n if i.endswith(\"skeletonAnim_SET\")\n ]\n if not skeleton_sets:\n return\n\n instance.data[\"families\"].append(\"animation.fbx\")\n instance.data[\"animated_skeleton\"] = []\n for skeleton_set in skeleton_sets:\n skeleton_content = cmds.sets(skeleton_set, query=True)\n self.log.debug(\n \"Collected animated skeleton data: {}\".format(\n skeleton_content\n ))\n if skeleton_content:\n instance.data[\"animated_skeleton\"] = skeleton_content\n"
},
{
"path": "openpype/hosts/maya/plugins/publish/collect_fbx_camera.py",
"content": "# -*- coding: utf-8 -*-\nfrom maya import cmds # noqa\nimport pyblish.api\n\n\nclass CollectFbxCamera(pyblish.api.InstancePlugin):\n \"\"\"Collect Camera for FBX export.\"\"\"\n\n order = pyblish.api.CollectorOrder + 0.2\n label = \"Collect Camera for FBX export\"\n families = [\"camera\"]\n\n def process(self, instance):\n if not instance.data.get(\"families\"):\n instance.data[\"families\"] = []\n\n if \"fbx\" not in instance.data[\"families\"]:\n instance.data[\"families\"].append(\"fbx\")\n\n instance.data[\"cameras\"] = True\n"
},
{
"path": "openpype/hosts/maya/plugins/publish/collect_file_dependencies.py",
"content": "import json\n\nfrom maya import cmds\n\nimport pyblish.api\n\n\nclass CollectFileDependencies(pyblish.api.ContextPlugin):\n \"\"\"Gather all files referenced in this scene.\"\"\"\n\n label = \"Collect File Dependencies\"\n order = pyblish.api.CollectorOrder - 0.49\n hosts = [\"maya\"]\n\n def process(self, context):\n dependencies = []\n for node in cmds.ls(type=\"file\"):\n path = cmds.getAttr(\"{}.{}\".format(node, \"fileTextureName\"))\n if path not in dependencies:\n dependencies.append(path)\n\n for node in cmds.ls(type=\"AlembicNode\"):\n path = cmds.getAttr(\"{}.{}\".format(node, \"abc_File\"))\n if path not in dependencies:\n dependencies.append(path)\n\n context.data[\"fileDependencies\"] = dependencies\n self.log.debug(json.dumps(dependencies, indent=4))\n"
},
{
"path": "openpype/hosts/maya/plugins/publish/collect_gltf.py",
"content": "# -*- coding: utf-8 -*-\nimport pyblish.api\n\n\nclass CollectGLTF(pyblish.api.InstancePlugin):\n \"\"\"Collect Assets for GLTF/GLB export.\"\"\"\n\n order = pyblish.api.CollectorOrder + 0.2\n label = \"Collect Asset for GLTF/GLB export\"\n families = [\"model\", \"animation\", \"pointcache\"]\n\n def process(self, instance):\n if not instance.data.get(\"families\"):\n instance.data[\"families\"] = []\n\n if \"gltf\" not in instance.data[\"families\"]:\n instance.data[\"families\"].append(\"gltf\")\n"
},
{
"path": "openpype/hosts/maya/plugins/publish/collect_history.py",
"content": "from maya import cmds\n\nimport pyblish.api\n\n\nclass CollectMayaHistory(pyblish.api.InstancePlugin):\n \"\"\"Collect history for instances from the Maya scene\n\n Note:\n This removes render layers collected in the history\n\n This is separate from Collect Instances so we can target it towards only\n specific family types.\n\n \"\"\"\n\n order = pyblish.api.CollectorOrder + 0.1\n hosts = [\"maya\"]\n label = \"Maya History\"\n families = [\"rig\"]\n\n def process(self, instance):\n\n kwargs = {}\n if int(cmds.about(version=True)) >= 2020:\n # New flag since Maya 2020 which makes cmds.listHistory faster\n kwargs = {\"fastIteration\": True}\n else:\n self.log.debug(\"Ignoring `fastIteration` flag before Maya 2020..\")\n\n # Collect the history with long names\n history = set(cmds.listHistory(instance, leaf=False, **kwargs) or [])\n history = cmds.ls(list(history), long=True)\n\n # Exclude invalid nodes (like renderlayers)\n exclude = cmds.ls(type=\"renderLayer\", long=True)\n if exclude:\n exclude = set(exclude) # optimize lookup\n history = [x for x in history if x not in exclude]\n\n # Combine members with history\n members = instance[:] + history\n members = list(set(members)) # ensure unique\n\n # Update the instance\n instance[:] = members\n"
},
{
"path": "openpype/hosts/maya/plugins/publish/collect_inputs.py",
"content": "import copy\n\nfrom maya import cmds\nimport maya.api.OpenMaya as om\nimport pyblish.api\n\nfrom openpype.pipeline import registered_host\nfrom openpype.hosts.maya.api.lib import get_container_members\nfrom openpype.hosts.maya.api.lib_rendersetup import get_shader_in_layer\n\n\ndef iter_history(nodes,\n filter=om.MFn.kInvalid,\n direction=om.MItDependencyGraph.kUpstream):\n \"\"\"Iterate unique upstream history for list of nodes.\n\n This acts as a replacement to maya.cmds.listHistory.\n It's faster by about 2x-3x. It returns less than\n maya.cmds.listHistory as it excludes the input nodes\n from the output (unless an input node was history\n for another input node). It also excludes duplicates.\n\n Args:\n nodes (list): Maya node names to start search from.\n filter (om.MFn.Type): Filter to only specific types.\n e.g. to dag nodes using om.MFn.kDagNode\n direction (om.MItDependencyGraph.Direction): Direction to traverse in.\n Defaults to upstream.\n\n Yields:\n str: Node names in upstream history.\n\n \"\"\"\n if not nodes:\n return\n\n sel = om.MSelectionList()\n for node in nodes:\n sel.add(node)\n\n it = om.MItDependencyGraph(sel.getDependNode(0)) # init iterator\n handle = om.MObjectHandle\n\n traversed = set()\n fn_dep = om.MFnDependencyNode()\n fn_dag = om.MFnDagNode()\n for i in range(sel.length()):\n\n start_node = sel.getDependNode(i)\n start_node_hash = handle(start_node).hashCode()\n if start_node_hash in traversed:\n continue\n\n it.resetTo(start_node,\n filter=filter,\n direction=direction)\n while not it.isDone():\n\n node = it.currentNode()\n node_hash = handle(node).hashCode()\n\n if node_hash in traversed:\n it.prune()\n it.next() # noqa: B305\n continue\n\n traversed.add(node_hash)\n\n if node.hasFn(om.MFn.kDagNode):\n fn_dag.setObject(node)\n yield fn_dag.fullPathName()\n else:\n fn_dep.setObject(node)\n yield fn_dep.name()\n\n it.next() # noqa: B305\n\n\ndef collect_input_containers(containers, nodes):\n \"\"\"Collect containers that contain any of the node in `nodes`.\n\n This will return any loaded Avalon container that contains at least one of\n the nodes. As such, the Avalon container is an input for it. Or in short,\n there are member nodes of that container.\n\n Returns:\n list: Input avalon containers\n\n \"\"\"\n # Assume the containers have collected their cached '_members' data\n # in the collector.\n return [container for container in containers\n if any(node in container[\"_members\"] for node in nodes)]\n\n\nclass CollectUpstreamInputs(pyblish.api.InstancePlugin):\n \"\"\"Collect input source inputs for this publish.\n\n This will include `inputs` data of which loaded publishes were used in the\n generation of this publish. This leaves an upstream trace to what was used\n as input.\n\n \"\"\"\n\n label = \"Collect Inputs\"\n order = pyblish.api.CollectorOrder + 0.34\n hosts = [\"maya\"]\n\n def process(self, instance):\n\n # For large scenes the querying of \"host.ls()\" can be relatively slow\n # e.g. up to a second. Many instances calling it easily slows this\n # down. As such, we cache it so we trigger it only once.\n # todo: Instead of hidden cache make \"CollectContainers\" plug-in\n cache_key = \"__cache_containers\"\n scene_containers = instance.context.data.get(cache_key, None)\n if scene_containers is None:\n # Query the scenes' containers if there's no cache yet\n host = registered_host()\n scene_containers = list(host.ls())\n for container in scene_containers:\n # Embed the members into the container dictionary\n container_members = set(get_container_members(container))\n container[\"_members\"] = container_members\n instance.context.data[\"__cache_containers\"] = scene_containers\n\n # Collect the relevant input containers for this instance\n if \"renderlayer\" in set(instance.data.get(\"families\", [])):\n # Special behavior for renderlayers\n self.log.debug(\"Collecting renderlayer inputs....\")\n containers = self._collect_renderlayer_inputs(scene_containers,\n instance)\n\n else:\n # Basic behavior\n nodes = instance[:]\n\n # Include any input connections of history with long names\n # For optimization purposes only trace upstream from shape nodes\n # looking for used dag nodes. This way having just a constraint\n # on a transform is also ignored which tended to give irrelevant\n # inputs for the majority of our use cases. We tend to care more\n # about geometry inputs.\n shapes = cmds.ls(nodes,\n type=(\"mesh\", \"nurbsSurface\", \"nurbsCurve\"),\n noIntermediate=True)\n if shapes:\n history = list(iter_history(shapes, filter=om.MFn.kShape))\n history = cmds.ls(history, long=True)\n\n # Include the transforms in the collected history as shapes\n # are excluded from containers\n transforms = cmds.listRelatives(cmds.ls(history, shapes=True),\n parent=True,\n fullPath=True,\n type=\"transform\")\n if transforms:\n history.extend(transforms)\n\n if history:\n nodes = list(set(nodes + history))\n\n # Collect containers for the given set of nodes\n containers = collect_input_containers(scene_containers,\n nodes)\n\n inputs = [c[\"representation\"] for c in containers]\n instance.data[\"inputRepresentations\"] = inputs\n self.log.debug(\"Collected inputs: %s\" % inputs)\n\n def _collect_renderlayer_inputs(self, scene_containers, instance):\n \"\"\"Collects inputs from nodes in renderlayer, incl. shaders + camera\"\"\"\n\n # Get the renderlayer\n renderlayer = instance.data.get(\"renderlayer\")\n\n if renderlayer == \"defaultRenderLayer\":\n # Assume all loaded containers in the scene are inputs\n # for the masterlayer\n return copy.deepcopy(scene_containers)\n else:\n # Get the members of the layer\n members = cmds.editRenderLayerMembers(renderlayer,\n query=True,\n fullNames=True) or []\n\n # In some cases invalid objects are returned from\n # `editRenderLayerMembers` so we filter them out\n members = cmds.ls(members, long=True)\n\n # Include all children\n children = cmds.listRelatives(members,\n allDescendents=True,\n fullPath=True) or []\n members.extend(children)\n\n # Include assigned shaders in renderlayer\n shapes = cmds.ls(members, shapes=True, long=True)\n shaders = set()\n for shape in shapes:\n shape_shaders = get_shader_in_layer(shape, layer=renderlayer)\n if not shape_shaders:\n continue\n shaders.update(shape_shaders)\n members.extend(shaders)\n\n # Explicitly include the camera being rendered in renderlayer\n cameras = instance.data.get(\"cameras\")\n members.extend(cameras)\n\n containers = collect_input_containers(scene_containers, members)\n\n return containers\n"
},
{
"path": "openpype/hosts/maya/plugins/publish/collect_instances.py",
"content": "from maya import cmds\n\nimport pyblish.api\nfrom openpype.hosts.maya.api.lib import get_all_children\n\n\nclass CollectNewInstances(pyblish.api.InstancePlugin):\n \"\"\"Gather members for instances and pre-defined attribute\n\n This collector takes into account assets that are associated with\n an objectSet and marked with a unique identifier;\n\n Identifier:\n id (str): \"pyblish.avalon.instance\"\n\n Limitations:\n - Does not take into account nodes connected to those\n within an objectSet. Extractors are assumed to export\n with history preserved, but this limits what they will\n be able to achieve and the amount of data available\n to validators. An additional collector could also\n append this input data into the instance, as we do\n for `pype.rig` with collect_history.\n\n \"\"\"\n\n label = \"Collect New Instance Data\"\n order = pyblish.api.CollectorOrder\n hosts = [\"maya\"]\n\n valid_empty_families = {\"workfile\", \"renderlayer\"}\n\n def process(self, instance):\n\n objset = instance.data.get(\"instance_node\")\n if not objset:\n self.log.debug(\"Instance has no `instance_node` data\")\n\n # TODO: We might not want to do this in the future\n # Merge creator attributes into instance.data just backwards compatible\n # code still runs as expected\n creator_attributes = instance.data.get(\"creator_attributes\", {})\n if creator_attributes:\n instance.data.update(creator_attributes)\n\n members = cmds.sets(objset, query=True) or []\n if members:\n # Collect members\n members = cmds.ls(members, long=True) or []\n\n dag_members = cmds.ls(members, type=\"dagNode\", long=True)\n children = get_all_children(dag_members)\n children = cmds.ls(children, noIntermediate=True, long=True)\n parents = (\n self.get_all_parents(members)\n if creator_attributes.get(\"includeParentHierarchy\", True)\n else []\n )\n members_hierarchy = list(set(members + children + parents))\n\n instance[:] = members_hierarchy\n\n elif instance.data[\"family\"] not in self.valid_empty_families:\n self.log.warning(\"Empty instance: \\\"%s\\\" \" % objset)\n # Store the exact members of the object set\n instance.data[\"setMembers\"] = members\n\n # TODO: This might make more sense as a separate collector\n # Convert frame values to integers\n for attr_name in (\n \"handleStart\", \"handleEnd\", \"frameStart\", \"frameEnd\",\n ):\n value = instance.data.get(attr_name)\n if value is not None:\n instance.data[attr_name] = int(value)\n\n # Append start frame and end frame to label if present\n if \"frameStart\" in instance.data and \"frameEnd\" in instance.data:\n # Take handles from context if not set locally on the instance\n for key in [\"handleStart\", \"handleEnd\"]:\n if key not in instance.data:\n value = instance.context.data[key]\n if value is not None:\n value = int(value)\n instance.data[key] = value\n\n instance.data[\"frameStartHandle\"] = int(\n instance.data[\"frameStart\"] - instance.data[\"handleStart\"]\n )\n instance.data[\"frameEndHandle\"] = int(\n instance.data[\"frameEnd\"] + instance.data[\"handleEnd\"]\n )\n\n def get_all_parents(self, nodes):\n \"\"\"Get all parents by using string operations (optimization)\n\n Args:\n nodes (list): the nodes which are found in the objectSet\n\n Returns:\n list\n \"\"\"\n\n parents = []\n for node in nodes:\n splitted = node.split(\"|\")\n items = [\"|\".join(splitted[0:i]) for i in range(2, len(splitted))]\n parents.extend(items)\n\n return list(set(parents))\n"
},
{
"path": "openpype/hosts/maya/plugins/publish/collect_look.py",
"content": "# -*- coding: utf-8 -*-\n\"\"\"Maya look collector.\"\"\"\nimport re\nimport os\nimport glob\n\nfrom maya import cmds # noqa\nimport pyblish.api\nfrom openpype.hosts.maya.api import lib\n\nSHAPE_ATTRS = [\"castsShadows\",\n \"receiveShadows\",\n \"motionBlur\",\n \"primaryVisibility\",\n \"smoothShading\",\n \"visibleInReflections\",\n \"visibleInRefractions\",\n \"doubleSided\",\n \"opposite\"]\nSHAPE_ATTRS = set(SHAPE_ATTRS)\n\n\ndef get_pxr_multitexture_file_attrs(node):\n attrs = []\n for i in range(9):\n if cmds.attributeQuery(\"filename{}\".format(i), node=node, ex=True):\n file = cmds.getAttr(\"{}.filename{}\".format(node, i))\n if file:\n attrs.append(\"filename{}\".format(i))\n return attrs\n\n\nFILE_NODES = {\n # maya\n \"file\": \"fileTextureName\",\n # arnold (mtoa)\n \"aiImage\": \"filename\",\n # redshift\n \"RedshiftNormalMap\": \"tex0\",\n # renderman\n \"PxrBump\": \"filename\",\n \"PxrNormalMap\": \"filename\",\n \"PxrMultiTexture\": get_pxr_multitexture_file_attrs,\n \"PxrPtexture\": \"filename\",\n \"PxrTexture\": \"filename\"\n}\n\nRENDER_SET_TYPES = [\n \"VRayDisplacement\",\n \"VRayLightMesh\",\n \"VRayObjectProperties\",\n \"RedshiftObjectId\",\n \"RedshiftMeshParameters\",\n]\n\n# Keep only node types that actually exist\nall_node_types = set(cmds.allNodeTypes())\nfor node_type in list(FILE_NODES.keys()):\n if node_type not in all_node_types:\n FILE_NODES.pop(node_type)\n\nfor node_type in RENDER_SET_TYPES:\n if node_type not in all_node_types:\n RENDER_SET_TYPES.remove(node_type)\ndel all_node_types\n\n# Cache pixar dependency node types so we can perform a type lookup against it\nPXR_NODES = set()\nif cmds.pluginInfo(\"RenderMan_for_Maya\", query=True, loaded=True):\n PXR_NODES = set(\n cmds.pluginInfo(\"RenderMan_for_Maya\",\n query=True,\n dependNode=True)\n )\n\n\ndef get_attributes(dictionary, attr, node=None):\n # type: (dict, str, str) -> list\n if callable(dictionary[attr]):\n val = dictionary[attr](node)\n else:\n val = dictionary.get(attr, [])\n\n return val if isinstance(val, list) else [val]\n\n\ndef get_look_attrs(node):\n \"\"\"Returns attributes of a node that are important for the look.\n\n These are the \"changed\" attributes (those that have edits applied\n in the current scene).\n\n Returns:\n list: Attribute names to extract\n\n \"\"\"\n # When referenced get only attributes that are \"changed since file open\"\n # which includes any reference edits, otherwise take *all* user defined\n # attributes\n is_referenced = cmds.referenceQuery(node, isNodeReferenced=True)\n result = cmds.listAttr(node, userDefined=True,\n changedSinceFileOpen=is_referenced) or []\n\n # `cbId` is added when a scene is saved, ignore by default\n if \"cbId\" in result:\n result.remove(\"cbId\")\n\n # For shapes allow render stat changes\n if cmds.objectType(node, isAType=\"shape\"):\n attrs = cmds.listAttr(node, changedSinceFileOpen=True) or []\n for attr in attrs:\n if attr in SHAPE_ATTRS or \\\n attr not in SHAPE_ATTRS and attr.startswith('ai'):\n result.append(attr)\n return result\n\n\ndef node_uses_image_sequence(node, node_path):\n # type: (str, str) -> bool\n \"\"\"Return whether file node uses an image sequence or single image.\n\n Determine if a node uses an image sequence or just a single image,\n not always obvious from its file path alone.\n\n Args:\n node (str): Name of the Maya node\n node_path (str): The file path of the node\n\n Returns:\n bool: True if node uses an image sequence\n\n \"\"\"\n\n # useFrameExtension indicates an explicit image sequence\n try:\n use_frame_extension = cmds.getAttr('%s.useFrameExtension' % node)\n except ValueError:\n use_frame_extension = False\n if use_frame_extension:\n return True\n\n # The following tokens imply a sequence\n patterns = [\"\", \"\", \"\",\n \"u_v\", \"\"]\n node_path_lowered = node_path.lower()\n return any(pattern in node_path_lowered for pattern in patterns)\n\n\ndef seq_to_glob(path):\n \"\"\"Takes an image sequence path and returns it in glob format,\n with the frame number replaced by a '*'.\n\n Image sequences may be numerical sequences, e.g. /path/to/file.1001.exr\n will return as /path/to/file.*.exr.\n\n Image sequences may also use tokens to denote sequences, e.g.\n /path/to/texture..tif will return as /path/to/texture.*.tif.\n\n Args:\n path (str): the image sequence path\n\n Returns:\n str: Return glob string that matches the filename pattern.\n\n \"\"\"\n\n if path is None:\n return path\n\n # If any of the patterns, convert the pattern\n patterns = {\n \"\": \"\",\n \"\": \"\",\n \"\": \"\",\n \"#\": \"#\",\n \"u_v\": \"|\",\n \"\",\n \"\": \"\"\n }\n\n lower = path.lower()\n has_pattern = False\n for pattern, regex_pattern in patterns.items():\n if pattern in lower:\n path = re.sub(regex_pattern, \"*\", path, flags=re.IGNORECASE)\n has_pattern = True\n\n if has_pattern:\n return path\n\n base = os.path.basename(path)\n matches = list(re.finditer(r'\\d+', base))\n if matches:\n match = matches[-1]\n new_base = '{0}*{1}'.format(base[:match.start()],\n base[match.end():])\n head = os.path.dirname(path)\n return os.path.join(head, new_base)\n else:\n return path\n\n\ndef get_file_node_paths(node):\n # type: (str) -> list\n \"\"\"Get the file path used by a Maya file node.\n\n Args:\n node (str): Name of the Maya file node\n\n Returns:\n list: the file paths in use\n\n \"\"\"\n # if the path appears to be sequence, use computedFileTextureNamePattern,\n # this preserves the <> tag\n if cmds.attributeQuery('computedFileTextureNamePattern',\n node=node,\n exists=True):\n plug = '{0}.computedFileTextureNamePattern'.format(node)\n texture_pattern = cmds.getAttr(plug)\n\n patterns = [\"\",\n \"\",\n \"u_v\",\n \"
![\"Milan](\"https://avatars.githubusercontent.com/u/3333008?v=4?s=100\")
![\"Jakub](\"https://avatars.githubusercontent.com/u/40640033?v=4?s=100\")
![\"Ondřej](\"https://avatars.githubusercontent.com/u/33513211?v=4?s=100\")
![\"Jakub](\"https://avatars.githubusercontent.com/u/43494761?v=4?s=100\")
![\"Petr](\"https://avatars.githubusercontent.com/u/4457962?v=4?s=100\")
![\"64qam\"/](\"https://avatars.githubusercontent.com/u/26925793?v=4?s=100\")
![\"Roy](\"https://avatars.githubusercontent.com/u/2439881?v=4?s=100\")
![\"Toke](\"https://avatars.githubusercontent.com/u/1860085?v=4?s=100\")
![\"Jiri](\"https://avatars.githubusercontent.com/u/45896205?v=4?s=100\")
![\"Simone](\"https://avatars.githubusercontent.com/u/1087869?v=4?s=100\")
![\"karimmozilla\"/](\"https://avatars.githubusercontent.com/u/82811760?v=4?s=100\")
![\"Allan](\"https://avatars.githubusercontent.com/u/76656700?v=4?s=100\")
![\"murphy\"/](\"https://avatars.githubusercontent.com/u/352795?v=4?s=100\")
![\"Wijnand](\"https://avatars.githubusercontent.com/u/26920875?v=4?s=100\")
![\"Bo](\"https://avatars.githubusercontent.com/u/1798206?v=4?s=100\")
![\"Clément](\"https://avatars.githubusercontent.com/u/7068597?v=4?s=100\")
![\"David](\"https://avatars.githubusercontent.com/u/3357009?v=4?s=100\")
![\"Derek](\"https://avatars.githubusercontent.com/u/42170307?v=4?s=100\")
![\"Gábor](\"https://avatars.githubusercontent.com/u/8620515?v=4?s=100\")
![\"icyvapor\"/](\"https://avatars.githubusercontent.com/u/1195278?v=4?s=100\")
![\"Jérôme](\"https://avatars.githubusercontent.com/u/7955673?v=4?s=100\")
![\"David](\"https://avatars.githubusercontent.com/u/89823400?v=4?s=100\")
![\"BenoitConnan\"/](\"https://avatars.githubusercontent.com/u/82808268?v=4?s=100\")
![\"Malthaldar\"/](\"https://avatars.githubusercontent.com/u/33671694?v=4?s=100\")
![\"Sven](\"https://avatars.githubusercontent.com/u/2472863?v=4?s=100\")
![\"zafrs\"/](\"https://avatars.githubusercontent.com/u/26890002?v=4?s=100\")
![\"Félix](\"https://avatars.githubusercontent.com/u/22875539?v=4?s=100\")
![\"Alexey](\"https://avatars.githubusercontent.com/u/11698866?v=4?s=100\")