From 7e45558c55e869c06e0ba4506bda276ed9b82f4f Mon Sep 17 00:00:00 2001
From: Grant Sanderson <grant@3blue1brown.com>
Date: Sun, 15 Jan 2023 17:40:05 -0800
Subject: [PATCH] Allow Mobject subclasses to specify which parts of data
 should act like points

---
 manimlib/mobject/mobject.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/manimlib/mobject/mobject.py b/manimlib/mobject/mobject.py
index 0c12d6fb..51018d6a 100644
--- a/manimlib/mobject/mobject.py
+++ b/manimlib/mobject/mobject.py
@@ -73,6 +73,7 @@ class Mobject(object):
         ('rgba', '<f4', (4,)),
     ])
     aligned_data_keys = ['points']
+    pointlike_data_keys = ['points']
 
     def __init__(
         self,
@@ -205,7 +206,7 @@ class Mobject(object):
     def apply_points_function(
         self,
         func: Callable[[np.ndarray], np.ndarray],
-        about_point: Vect3 = None,
+        about_point: Vect3 | None = None,
         about_edge: Vect3 = ORIGIN,
         works_on_bounding_box: bool = False
     ):
@@ -215,7 +216,8 @@ class Mobject(object):
         for mob in self.get_family():
             arrs = []
             if mob.has_points():
-                arrs.append(mob.get_points())
+                for key in mob.pointlike_data_keys:
+                    arrs.append(mob.data[key])
             if works_on_bounding_box:
                 arrs.append(mob.get_bounding_box())