From 3ae0a4e81b7790194bcf27142a1deb29fa548b9d Mon Sep 17 00:00:00 2001
From: Grant Sanderson <grant@3blue1brown.com>
Date: Sat, 23 Apr 2022 18:51:03 -0700
Subject: [PATCH] Add equality for ShaderWrapper

---
 manimlib/shader_wrapper.py | 17 +++++++++++++++++
 1 file changed, 17 insertions(+)

diff --git a/manimlib/shader_wrapper.py b/manimlib/shader_wrapper.py
index 5ed2364f..e94d8a73 100644
--- a/manimlib/shader_wrapper.py
+++ b/manimlib/shader_wrapper.py
@@ -45,6 +45,23 @@ class ShaderWrapper(object):
         self.init_program_code()
         self.refresh_id()
 
+    def __eq__(self, shader_wrapper: ShaderWrapper):
+        return all((
+            np.all(self.vert_data == shader_wrapper.vert_data),
+            np.all(self.vert_indices == shader_wrapper.vert_indices),
+            self.shader_folder == shader_wrapper.shader_folder,
+            all(
+                np.all(self.uniforms[key] == shader_wrapper.uniforms[key])
+                for key in self.uniforms
+            ),
+            all(
+                self.texture_paths[key] == shader_wrapper.texture_paths[key]
+                for key in self.texture_paths
+            ),
+            self.depth_test == shader_wrapper.depth_test,
+            self.render_primitive == shader_wrapper.render_primitive,
+        ))
+
     def copy(self):
         result = copy.copy(self)
         result.vert_data = np.array(self.vert_data)