Support UDIM images

blender_asset_tracer/cdefs.py

@@ -32,6 +32,7 @@ SEQ_TYPE_EFFECT = 8
 IMA_SRC_FILE = 1
 IMA_SRC_SEQUENCE = 2
 IMA_SRC_MOVIE = 3
+IMA_SRC_TILED = 6  # UDIM
 
 # DNA_modifier_types.h
 eModifierType_Wave = 7

blender_asset_tracer/pack/__init__.py

@@ -555,7 +555,7 @@ class Packer:
     def _copy_asset_and_deps(self, asset_path: pathlib.Path, action: AssetAction):
         # Copy the asset itself, but only if it's not a sequence (sequences are
         # handled below in the for-loop).
-        if "*" not in str(asset_path):
+        if "*" not in str(asset_path) and '<UDIM>' not in asset_path.name:
             packed_path = action.new_path
             assert packed_path is not None
             read_path = action.read_from or asset_path

blender_asset_tracer/trace/blocks2assets.py

@@ -96,11 +96,13 @@ def image(block: blendfile.BlendFileBlock) -> typing.Iterator[result.BlockUsage]
         cdefs.IMA_SRC_FILE,
         cdefs.IMA_SRC_SEQUENCE,
         cdefs.IMA_SRC_MOVIE,
+        cdefs.IMA_SRC_TILED,
     }:
+        log.debug("skiping image source type %s", image_source)
         return
 
     pathname, field = block.get(b"name", return_field=True)
-    is_sequence = image_source == cdefs.IMA_SRC_SEQUENCE
+    is_sequence = image_source in {cdefs.IMA_SRC_SEQUENCE, cdefs.IMA_SRC_TILED}
 
     yield result.BlockUsage(block, pathname, is_sequence, path_full_field=field)
 

blender_asset_tracer/trace/file_sequence.py

@@ -39,6 +39,12 @@ def expand_sequence(path: pathlib.Path) -> typing.Iterator[pathlib.Path]:
         or the path of the first file in the sequence.
     """
 
+    if "<UDIM>" in path.name: # UDIM tiles
+        # Change <UDIM> marker to a glob pattern, then let the glob case handle it.
+        # This assumes that all files that match the glob are actually UDIM
+        # tiles; this could cause some false-positives.
+        path = path.with_name(path.name.replace('<UDIM>', '*'))
+
     if "*" in str(path):  # assume it is a glob
         import glob
 
@@ -46,7 +52,6 @@ def expand_sequence(path: pathlib.Path) -> typing.Iterator[pathlib.Path]:
         for fname in sorted(glob.glob(str(path), recursive=True)):
             yield pathlib.Path(fname)
         return
-
     if not path.exists():
         raise DoesNotExist(path)