OP-2860 - added possibility to get number of frames from video file w…

…ith ffprobe

Previously wrong hardcoded value was used. This implementation needs to be monitored for weird format of published video files.

openpype/hosts/webpublisher/plugins/publish/collect_published_files.py

@@ -10,14 +10,18 @@
 import os
 import clique
 import tempfile
+import math
+
 from avalon import io
 import pyblish.api
-from openpype.lib import prepare_template_data
+from openpype.lib import prepare_template_data, get_asset, ffprobe_streams
+from openpype.lib.vendor_bin_utils import get_fps
 from openpype.lib.plugin_tools import (
  parse_json,
  get_subset_name_with_asset_doc
 )
 
+
 class CollectPublishedFiles(pyblish.api.ContextPlugin):
  """
  This collector will try to find json files in provided
@@ -49,10 +53,7 @@ def process(self, context):
  self.log.info("task_sub:: {}".format(task_subfolders))
 
  asset_name = context.data["asset"]
- asset_doc = io.find_one({
- "type": "asset",
- "name": asset_name
- })
+ asset_doc = get_asset()
  task_name = context.data["task"]
  task_type = context.data["taskType"]
  project_name = context.data["project_name"]
@@ -97,11 +98,26 @@ def process(self, context):
  instance.data["frameEnd"] = \
  instance.data["representations"][0]["frameEnd"]
  else:
- instance.data["frameStart"] = 0
- instance.data["frameEnd"] = 1
+ frame_start = asset_doc["data"]["frameStart"]
+ instance.data["frameStart"] = frame_start
+ instance.data["frameEnd"] = asset_doc["data"]["frameEnd"]
  instance.data["representations"] = self._get_single_repre(
  task_dir, task_data["files"], tags
  )
+ file_url = os.path.join(task_dir, task_data["files"][0])
+ duration = self._get_duration(file_url)
+ if duration:
+ try:
+ frame_end = int(frame_start) + math.ceil(duration)
+ instance.data["frameEnd"] = math.ceil(frame_end)
+ self.log.debug("frameEnd:: {}".format(
+ instance.data["frameEnd"]))
+ except ValueError:
+ self.log.warning("Unable to count frames "
+ "duration {}".format(duration))
+
+ instance.data["handleStart"] = asset_doc["data"]["handleStart"]
+ instance.data["handleEnd"] = asset_doc["data"]["handleEnd"]
 
  self.log.info("instance.data:: {}".format(instance.data))
 
@@ -127,7 +143,7 @@ def _get_single_repre(self, task_dir, files, tags):
  return [repre_data]
 
  def _process_sequence(self, files, task_dir, tags):
- """Prepare reprentations for sequence of files."""
+ """Prepare representation for sequence of files."""
  collections, remainder = clique.assemble(files)
  assert len(collections) == 1, \
  "Too many collections in {}".format(files)
@@ -188,6 +204,7 @@ def _get_family(self, settings, task_type, is_sequence, extension):
  msg = "No family found for combination of " +\
  "task_type: {}, is_sequence:{}, extension: {}".format(
  task_type, is_sequence, extension)
+ found_family = "render"
  assert found_family, msg
 
  return (found_family,
@@ -243,3 +260,41 @@ def _get_last_version(self, asset_name, subset_name):
  return version[0].get("version") or 0
  else:
  return 0
+
+ def _get_duration(self, file_url):
+ """Return duration in frames"""
+ try:
+ streams = ffprobe_streams(file_url, self.log)
+ except Exception as exc:
+ raise AssertionError((
+ "FFprobe couldn't read information about input file: \"{}\"."
+ " Error message: {}"
+ ).format(file_url, str(exc)))
+
+ first_video_stream = None
+ for stream in streams:
+ if "width" in stream and "height" in stream:
+ first_video_stream = stream
+ break
+
+ if first_video_stream:
+ nb_frames = stream.get("nb_frames")
+ if nb_frames:
+ try:
+ return int(nb_frames)
+ except ValueError:
+ self.log.warning(
+ "nb_frames {} not convertible".format(nb_frames))
+
+ duration = stream.get("duration")
+ frame_rate = get_fps(stream.get("r_frame_rate", '0/0'))
+ self.log.debu("duration:: {} frame_rate:: {}".format(
+ duration, frame_rate))
+ try:
+ return float(duration) * float(frame_rate)
+ except ValueError:
+ self.log.warning(
+ "{} or {} cannot be converted".format(duration,
+ frame_rate))
+
+ self.log.warning("Cannot get number of frames")