2025-12-01

scripts/addons/cache_manager/propsdata.py

@@ -0,0 +1,133 @@
+# SPDX-FileCopyrightText: 2021 Blender Studio Tools Authors
+#
+# SPDX-License-Identifier: GPL-3.0-or-later
+
+import os
+from pathlib import Path
+from typing import Any
+
+import bpy
+
+from .prefs import addon_prefs_get
+from .logger import LoggerFactory
+
+logger = LoggerFactory.getLogger(__name__)
+
+
+def ui_redraw() -> None:
+    """
+    Forces blender to redraw the UI.
+    """
+    for screen in bpy.data.screens:
+        for area in screen.areas:
+            area.tag_redraw()
+
+def _get_scene_name() -> str:
+    if not bpy.data.filepath:
+        return ""
+
+    filepath = Path(os.path.abspath(bpy.path.abspath(bpy.data.filepath)))
+    return filepath.parents[1].name
+
+
+def _get_shot_name() -> str:
+    if not bpy.data.filepath:
+        return ""
+
+    filepath = Path(os.path.abspath(bpy.path.abspath(bpy.data.filepath)))
+    return filepath.parents[0].name
+
+
+def _gen_cacheconfig_filename() -> str:
+    return f"{_get_shot_name()}.cacheconfig.{bpy.context.scene.cm.cache_version}.json"
+
+
+def gen_cachedir_path_str(self: Any) -> str:
+    addon_prefs = addon_prefs_get()
+
+    if not addon_prefs.is_cachedir_root_valid:
+        return ""
+
+    p = (
+        Path(addon_prefs.cachedir_root_path)
+        / _get_scene_name()
+        / _get_shot_name()
+        / bpy.context.scene.cm.cache_version
+    )
+
+    return p.absolute().as_posix()
+
+
+def gen_cacheconfig_path_str(self: Any) -> str:
+
+    cachedir_str = gen_cachedir_path_str(None)
+
+    if not cachedir_str:
+        return ""
+
+    p = Path(cachedir_str) / _gen_cacheconfig_filename()
+
+    return p.absolute().as_posix()
+
+
+def gen_cache_coll_filename(collection: bpy.types.Collection) -> str:
+    return (
+        f"{_get_shot_name()}.{collection.name}.{bpy.context.scene.cm.cache_version}.abc"
+    )
+
+
+def gen_cachepath_collection(
+    collection: bpy.types.Collection, context: bpy.types.Context
+) -> Path:
+    cachedir_path = Path(gen_cachedir_path_str(None))
+
+    if not cachedir_path:
+        raise ValueError(
+            f"Failed to generate cachepath for collection: {collection.name}. Invalid cachepath: {str(cachedir_path)}"
+        )
+    return cachedir_path.joinpath(gen_cache_coll_filename(collection)).absolute()
+
+
+def get_cache_version_dir_path_str(self: Any) -> str:
+    addon_prefs = addon_prefs_get()
+
+    if not addon_prefs.is_cachedir_root_valid:
+        return ""
+
+    p = Path(addon_prefs.cachedir_root_path) / _get_scene_name() / _get_shot_name()
+
+    return p.absolute().as_posix()
+
+
+def rm_deleted_colls_from_list(context: bpy.types.Context) -> None:
+
+    for category in [context.scene.cm.colls_export, context.scene.cm.colls_import]:
+
+        category_name = context.scene.cm.category
+        colls = [item.coll_ptr for item in category]
+        colls.reverse()
+
+        # Make sure to remove list from the back to not throw off the subsequent indexes.
+        for idx, coll in enumerate(colls):
+            org_idx = len(colls) - 1 - idx
+            if not coll:
+                # Remove item for that category at that index.
+                category.remove(org_idx)
+                logger.info(
+                    "Removed index %i from %s list. Does not exists anymore.",
+                    org_idx,
+                    category_name.lower(),
+                )
+                # Update selection index.
+                curr_index = context.scene.cm.colls_export_index
+                if category_name == "IMPORT":
+                    curr_index = context.scene.cm.colls_import_index
+
+                if curr_index == org_idx and curr_index > 0:
+
+                    if category_name == "IMPORT":
+                        context.scene.cm.colls_import_index = curr_index - 1
+                    else:
+                        context.scene.cm.colls_export_index = curr_index - 1
+    ui_redraw()
+