mirror of
https://github.com/ynput/ayon-core.git
synced 2026-01-01 08:24:53 +01:00
OP-4643 - updated docstring, standardized arguments
This commit is contained in:
parent
3dba4f3eb1
commit
7e9d707226
2 changed files with 17 additions and 23 deletions
|
|
@ -1049,7 +1049,7 @@ def convert_ffprobe_fps_to_float(value):
|
||||||
|
|
||||||
def convert_colorspace(
|
def convert_colorspace(
|
||||||
input_path,
|
input_path,
|
||||||
out_filepath,
|
output_path,
|
||||||
config_path,
|
config_path,
|
||||||
source_colorspace,
|
source_colorspace,
|
||||||
target_colorspace,
|
target_colorspace,
|
||||||
|
|
@ -1057,18 +1057,13 @@ def convert_colorspace(
|
||||||
display,
|
display,
|
||||||
logger=None
|
logger=None
|
||||||
):
|
):
|
||||||
"""Convert source files from one color space to another.
|
"""Convert source file from one color space to another.
|
||||||
|
|
||||||
Filenames of input files are kept so make sure that output directory
|
|
||||||
is not the same directory as input files have.
|
|
||||||
- This way it can handle gaps and can keep input filenames without handling
|
|
||||||
frame template
|
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
input_path (str): Paths that should be converted. It is expected that
|
input_path (str): Path that should be converted. It is expected that
|
||||||
contains single file or image sequence of samy type.
|
contains single file or image sequence of same type
|
||||||
out_filepath (str): Path to directory where output will be rendered.
|
(sequence in format 'file.FRAMESTART-FRAMEEND#.exr', see oiio docs)
|
||||||
Must not be same as input's directory.
|
output_path (str): Path to output filename.
|
||||||
config_path (str): path to OCIO config file
|
config_path (str): path to OCIO config file
|
||||||
source_colorspace (str): ocio valid color space of source files
|
source_colorspace (str): ocio valid color space of source files
|
||||||
target_colorspace (str): ocio valid target color space
|
target_colorspace (str): ocio valid target color space
|
||||||
|
|
@ -1087,7 +1082,7 @@ def convert_colorspace(
|
||||||
# Don't add any additional attributes
|
# Don't add any additional attributes
|
||||||
"--nosoftwareattrib",
|
"--nosoftwareattrib",
|
||||||
"--colorconfig", config_path,
|
"--colorconfig", config_path,
|
||||||
"-o", out_filepath
|
"-o", output_path
|
||||||
]
|
]
|
||||||
|
|
||||||
if all([target_colorspace, view, display]):
|
if all([target_colorspace, view, display]):
|
||||||
|
|
|
||||||
|
|
@ -119,13 +119,13 @@ class ExtractOIIOTranscode(publish.Extractor):
|
||||||
files_to_convert = self._translate_to_sequence(
|
files_to_convert = self._translate_to_sequence(
|
||||||
files_to_convert)
|
files_to_convert)
|
||||||
for file_name in files_to_convert:
|
for file_name in files_to_convert:
|
||||||
input_filepath = os.path.join(original_staging_dir,
|
input_path = os.path.join(original_staging_dir,
|
||||||
file_name)
|
file_name)
|
||||||
output_path = self._get_output_file_path(input_filepath,
|
output_path = self._get_output_file_path(input_path,
|
||||||
new_staging_dir,
|
new_staging_dir,
|
||||||
output_extension)
|
output_extension)
|
||||||
convert_colorspace(
|
convert_colorspace(
|
||||||
input_filepath,
|
input_path,
|
||||||
output_path,
|
output_path,
|
||||||
config_path,
|
config_path,
|
||||||
source_colorspace,
|
source_colorspace,
|
||||||
|
|
@ -156,16 +156,17 @@ class ExtractOIIOTranscode(publish.Extractor):
|
||||||
self._mark_original_repre_for_deletion(repre, profile)
|
self._mark_original_repre_for_deletion(repre, profile)
|
||||||
|
|
||||||
def _translate_to_sequence(self, files_to_convert):
|
def _translate_to_sequence(self, files_to_convert):
|
||||||
"""Returns original list of files or single sequence format filename.
|
"""Returns original list or list with filename formatted in single
|
||||||
|
sequence format.
|
||||||
|
|
||||||
Uses clique to find frame sequence, in this case it merges all frames
|
Uses clique to find frame sequence, in this case it merges all frames
|
||||||
into sequence format (%0X) and returns it.
|
into sequence format (FRAMESTART-FRAMEEND#) and returns it.
|
||||||
If sequence not found, it returns original list
|
If sequence not found, it returns original list
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
files_to_convert (list): list of file names
|
files_to_convert (list): list of file names
|
||||||
Returns:
|
Returns:
|
||||||
(list) of [file.%04.exr] or [fileA.exr, fileB.exr]
|
(list) of [file.1001-1010#.exr] or [fileA.exr, fileB.exr]
|
||||||
"""
|
"""
|
||||||
pattern = [clique.PATTERNS["frames"]]
|
pattern = [clique.PATTERNS["frames"]]
|
||||||
collections, remainder = clique.assemble(
|
collections, remainder = clique.assemble(
|
||||||
|
|
@ -178,8 +179,6 @@ class ExtractOIIOTranscode(publish.Extractor):
|
||||||
"Too many collections {}".format(collections))
|
"Too many collections {}".format(collections))
|
||||||
|
|
||||||
collection = collections[0]
|
collection = collections[0]
|
||||||
padding = collection.padding
|
|
||||||
padding_str = "%0{}".format(padding)
|
|
||||||
frames = list(collection.indexes)
|
frames = list(collection.indexes)
|
||||||
frame_str = "{}-{}#".format(frames[0], frames[-1])
|
frame_str = "{}-{}#".format(frames[0], frames[-1])
|
||||||
file_name = "{}{}{}".format(collection.head, frame_str,
|
file_name = "{}{}{}".format(collection.head, frame_str,
|
||||||
|
|
@ -189,10 +188,10 @@ class ExtractOIIOTranscode(publish.Extractor):
|
||||||
|
|
||||||
return files_to_convert
|
return files_to_convert
|
||||||
|
|
||||||
def _get_output_file_path(self, input_filepath, output_dir,
|
def _get_output_file_path(self, input_path, output_dir,
|
||||||
output_extension):
|
output_extension):
|
||||||
"""Create output file name path."""
|
"""Create output file name path."""
|
||||||
file_name = os.path.basename(input_filepath)
|
file_name = os.path.basename(input_path)
|
||||||
file_name, input_extension = os.path.splitext(file_name)
|
file_name, input_extension = os.path.splitext(file_name)
|
||||||
if not output_extension:
|
if not output_extension:
|
||||||
output_extension = input_extension
|
output_extension = input_extension
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue