Flag Outdated containers

pype/hosts/harmony/__init__.py

@@ -1,8 +1,9 @@
 import os
 import sys
 
-from avalon import api, harmony
+from avalon import api, io, harmony
 from avalon.vendor import Qt
+import avalon.tools.sceneinventory
 import pyblish.api
 from pype import lib
 
@@ -92,6 +93,56 @@ def ensure_scene_settings():
  set_scene_settings(valid_settings)
 
 
+def check_inventory():
+ if not lib.any_outdated():
+ return
+
+ host = avalon.api.registered_host()
+ outdated_containers = []
+ for container in host.ls():
+ representation = container['representation']
+ representation_doc = io.find_one(
+ {
+ "_id": io.ObjectId(representation),
+ "type": "representation"
+ },
+ projection={"parent": True}
+ )
+ if representation_doc and not lib.is_latest(representation_doc):
+ outdated_containers.append(container)
+
+ # Colour nodes.
+ func = """function func(args){
+ for( var i =0; i <= args[0].length - 1; ++i)
+ {
+ var red_color = new ColorRGBA(255, 0, 0, 255);
+ node.setColor(args[0][i], red_color);
+ }
+ }
+ func
+ """
+ outdated_nodes = [x["node"] for x in outdated_containers]
+ harmony.send({"function": func, "args": [outdated_nodes]})
+
+ # Warn about outdated containers.
+ print("Starting new QApplication..")
+ app = Qt.QtWidgets.QApplication(sys.argv)
+
+ message_box = Qt.QtWidgets.QMessageBox()
+ message_box.setIcon(Qt.QtWidgets.QMessageBox.Warning)
+ msg = "There are outdated containers in the scene."
+ message_box.setText(msg)
+ message_box.exec_()
+
+ # Garbage collect QApplication.
+ del app
+
+
+def application_launch():
+ ensure_scene_settings()
+ check_inventory()
+
+
 def export_template(backdrops, nodes, filepath):
  func = """function func(args)
  {
@@ -161,7 +212,7 @@ def install():
  "instanceToggled", on_pyblish_instance_toggled
  )
 
- api.on("application.launched", ensure_scene_settings)
+ api.on("application.launched", application_launch)
 
 
 def on_pyblish_instance_toggled(instance, old_value, new_value):

pype/plugins/global/publish/validate_containers.py

@@ -19,7 +19,7 @@ class ValidateContainers(pyblish.api.ContextPlugin):
 
  label = "Validate Containers"
  order = pyblish.api.ValidatorOrder
- hosts = ["maya", "houdini", "nuke"]
+ hosts = ["maya", "houdini", "nuke", "harmony"]
  optional = True
  actions = [ShowInventory]
 

pype/plugins/harmony/load/load_imagesequence.py

@@ -98,33 +98,63 @@
  transparencyModeAttr.setValue(SGITransparencyMode);
  if (extension == "psd")
  transparencyModeAttr.setValue(FlatPSDTransparencyMode);
+ if (extension == "jpg")
+ transparencyModeAttr.setValue(LayeredPSDTransparencyMode);
 
  node.linkAttr(read, "DRAWING.ELEMENT", uniqueColumnName);
 
- // Create a drawing for each file.
- for( var i =0; i <= files.length - 1; ++i)
+ if (files.length == 1)
  {
- timing = start_frame + i
  // Create a drawing drawing, 'true' indicate that the file exists.
- Drawing.create(elemId, timing, true);
+ Drawing.create(elemId, 1, true);
  // Get the actual path, in tmp folder.
- var drawingFilePath = Drawing.filename(elemId, timing.toString());
- copyFile( files[i], drawingFilePath );
-
- column.setEntry(uniqueColumnName, 1, timing, timing.toString());
+ var drawingFilePath = Drawing.filename(elemId, "1");
+ copyFile(files[0], drawingFilePath);
+ // Expose the image for the entire frame range.
+ for( var i =0; i <= frame.numberOf() - 1; ++i)
+ {
+ timing = start_frame + i
+ column.setEntry(uniqueColumnName, 1, timing, "1");
+ }
+ } else {
+ // Create a drawing for each file.
+ for( var i =0; i <= files.length - 1; ++i)
+ {
+ timing = start_frame + i
+ // Create a drawing drawing, 'true' indicate that the file exists.
+ Drawing.create(elemId, timing, true);
+ // Get the actual path, in tmp folder.
+ var drawingFilePath = Drawing.filename(elemId, timing.toString());
+ copyFile( files[i], drawingFilePath );
+
+ column.setEntry(uniqueColumnName, 1, timing, timing.toString());
+ }
  }
+
+ var green_color = new ColorRGBA(0, 255, 0, 255);
+ node.setColor(read, green_color);
+
  return read;
 }
 import_files
 """
 
-replace_files = """function replace_files(args)
+replace_files = """var PNGTransparencyMode = 0; //Premultiplied wih Black
+var TGATransparencyMode = 0; //Premultiplied wih Black
+var SGITransparencyMode = 0; //Premultiplied wih Black
+var LayeredPSDTransparencyMode = 1; //Straight
+var FlatPSDTransparencyMode = 2; //Premultiplied wih White
+
+function replace_files(args)
 {
  var files = args[0];
+ MessageLog.trace(files);
+ MessageLog.trace(files.length);
  var _node = args[1];
  var start_frame = args[2];
 
  var _column = node.linkedColumn(_node, "DRAWING.ELEMENT");
+ var elemId = column.getElementIdOfDrawing(_column);
 
  // Delete existing drawings.
  var timings = column.getDrawingTimings(_column);
@@ -133,20 +163,62 @@
  column.deleteDrawingAt(_column, parseInt(timings[i]));
  }
 
- // Create new drawings.
- for( var i =0; i <= files.length - 1; ++i)
+
+ var filename = files[0];
+ var pos = filename.lastIndexOf(".");
+ if( pos < 0 )
+ return null;
+ var extension = filename.substr(pos+1).toLowerCase();
+
+ if(extension == "jpeg")
+ extension = "jpg";
+
+ var transparencyModeAttr = node.getAttr(
+ _node, frame.current(), "applyMatteToColor"
+ );
+ if (extension == "png")
+ transparencyModeAttr.setValue(PNGTransparencyMode);
+ if (extension == "tga")
+ transparencyModeAttr.setValue(TGATransparencyMode);
+ if (extension == "sgi")
+ transparencyModeAttr.setValue(SGITransparencyMode);
+ if (extension == "psd")
+ transparencyModeAttr.setValue(FlatPSDTransparencyMode);
+ if (extension == "jpg")
+ transparencyModeAttr.setValue(LayeredPSDTransparencyMode);
+
+ if (files.length == 1)
  {
- timing = start_frame + i
  // Create a drawing drawing, 'true' indicate that the file exists.
- Drawing.create(node.getElementId(_node), timing, true);
+ Drawing.create(elemId, 1, true);
  // Get the actual path, in tmp folder.
- var drawingFilePath = Drawing.filename(
- node.getElementId(_node), timing.toString()
- );
- copyFile( files[i], drawingFilePath );
-
- column.setEntry(_column, 1, timing, timing.toString());
+ var drawingFilePath = Drawing.filename(elemId, "1");
+ copyFile(files[0], drawingFilePath);
+ MessageLog.trace(files[0]);
+ MessageLog.trace(drawingFilePath);
+ // Expose the image for the entire frame range.
+ for( var i =0; i <= frame.numberOf() - 1; ++i)
+ {
+ timing = start_frame + i
+ column.setEntry(_column, 1, timing, "1");
+ }
+ } else {
+ // Create a drawing for each file.
+ for( var i =0; i <= files.length - 1; ++i)
+ {
+ timing = start_frame + i
+ // Create a drawing drawing, 'true' indicate that the file exists.
+ Drawing.create(elemId, timing, true);
+ // Get the actual path, in tmp folder.
+ var drawingFilePath = Drawing.filename(elemId, timing.toString());
+ copyFile( files[i], drawingFilePath );
+
+ column.setEntry(_column, 1, timing, timing.toString());
+ }
  }
+
+ var green_color = new ColorRGBA(0, 255, 0, 255);
+ node.setColor(_node, green_color);
 }
 replace_files
 """
@@ -156,18 +228,27 @@ class ImageSequenceLoader(api.Loader):
  """Load images
  Stores the imported asset in a container named after the asset.
  """
- families = ["shot", "render"]
- representations = ["jpeg", "png"]
+ families = ["shot", "render", "image"]
+ representations = ["jpeg", "png", "jpg"]
 
  def load(self, context, name=None, namespace=None, data=None):
 
  collections, remainder = clique.assemble(
  os.listdir(os.path.dirname(self.fname))
  )
  files = []
- for f in list(collections[0]):
+ if collections:
+ for f in list(collections[0]):
+ files.append(
+ os.path.join(
+ os.path.dirname(self.fname), f
+ ).replace("\\", "/")
+ )
+ else:
  files.append(
- os.path.join(os.path.dirname(self.fname), f).replace("\\", "/")
+ os.path.join(
+ os.path.dirname(self.fname), remainder[0]
+ ).replace("\\", "/")
  )
 
  read_node = harmony.send(
@@ -190,15 +271,23 @@ def load(self, context, name=None, namespace=None, data=None):
  def update(self, container, representation):
  node = container.pop("node")
 
+ path = api.get_representation_path(representation)
  collections, remainder = clique.assemble(
- os.listdir(
- os.path.dirname(api.get_representation_path(representation))
- )
+ os.listdir(os.path.dirname(path))
  )
  files = []
- for f in list(collections[0]):
+ if collections:
+ for f in list(collections[0]):
+ files.append(
+ os.path.join(
+ os.path.dirname(path), f
+ ).replace("\\", "/")
+ )
+ else:
  files.append(
- os.path.join(os.path.dirname(self.fname), f).replace("\\", "/")
+ os.path.join(
+ os.path.dirname(path), remainder[0]
+ ).replace("\\", "/")
  )
 
  harmony.send(