From 16a092ddf1f211d6f3098cd383b1df4e2b706f14 Mon Sep 17 00:00:00 2001
From: jonasdichelle <jonasdichelle@gmail.com>
Date: Mon, 2 Sep 2024 18:19:07 +0200
Subject: [PATCH] Add Support for Dynamic Paint Modifier
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Add support for the Dynamic Paint modifier point cache.

I added a walker to iterate through all surfaces on a canvas to get each surface's point cache.

Reviewed-on: https://projects.blender.org/blender/blender-asset-tracer/pulls/92889
Reviewed-by: Sybren A. Stüvel <sybren@blender.org>
---
 blender_asset_tracer/cdefs.py                 |  1 +
 .../trace/modifier_walkers.py                 | 34 +++++++++++++++++++
 2 files changed, 35 insertions(+)

diff --git a/blender_asset_tracer/cdefs.py b/blender_asset_tracer/cdefs.py
index 099a638..c5b7fd1 100644
--- a/blender_asset_tracer/cdefs.py
+++ b/blender_asset_tracer/cdefs.py
@@ -46,6 +46,7 @@ eModifierType_WeightVGEdit = 36
 eModifierType_WeightVGMix = 37
 eModifierType_WeightVGProximity = 38
 eModifierType_Ocean = 39
+eModifierType_DynamicPaint = 40
 eModifierType_MeshCache = 46
 eModifierType_MeshSequenceCache = 52
 eModifierType_Fluid = 56
diff --git a/blender_asset_tracer/trace/modifier_walkers.py b/blender_asset_tracer/trace/modifier_walkers.py
index 12a0468..597c287 100644
--- a/blender_asset_tracer/trace/modifier_walkers.py
+++ b/blender_asset_tracer/trace/modifier_walkers.py
@@ -22,6 +22,7 @@
 The modifier_xxx() functions all yield result.BlockUsage objects for external
 files used by the modifiers.
 """
+
 import logging
 import typing
 
@@ -317,3 +318,36 @@ def modifier_cloth(
     yield from _walk_point_cache(
         ctx, block_name, modifier.bfile, pointcache, cdefs.PTCACHE_EXT
     )
+
+
+@mod_handler(cdefs.eModifierType_DynamicPaint)
+def modifier_dynamic_paint(
+    ctx: ModifierContext, modifier: blendfile.BlendFileBlock, block_name: bytes
+) -> typing.Iterator[result.BlockUsage]:
+    my_log = log.getChild("modifier_dynamic_paint")
+
+    canvas_settings = modifier.get_pointer(b"canvas")
+    if canvas_settings is None:
+        my_log.debug(
+            "Modifier %r (%r) has no canvas_settings",
+            modifier[b"modifier", b"name"],
+            block_name,
+        )
+        return
+
+    surfaces = canvas_settings.get_pointer((b"surfaces", b"first"))
+
+    for surf_idx, surface in enumerate(blendfile.iterators.listbase(surfaces)):
+        surface_block_name = block_name + b".canvas_settings.surfaces[%d]" % (surf_idx)
+        point_cache = surface.get_pointer(b"pointcache")
+        if point_cache is None:
+            my_log.debug(
+                "Surface %r (%r) has no pointcache",
+                surface[b"surface", b"name"],
+                surface_block_name,
+            )
+            continue
+
+        yield from _walk_point_cache(
+            ctx, surface_block_name, modifier.bfile, point_cache, cdefs.PTCACHE_EXT
+        )