Skip to content

Commit

Permalink
Add equality for ShaderWrapper
Browse files Browse the repository at this point in the history
  • Loading branch information
3b1b committed Apr 24, 2022
1 parent c3c5717 commit 3ae0a4e
Showing 1 changed file with 17 additions and 0 deletions.
17 changes: 17 additions & 0 deletions manimlib/shader_wrapper.py
Expand Up @@ -45,6 +45,23 @@ def __init__(
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)
Expand Down

0 comments on commit 3ae0a4e

Please sign in to comment.