Browse Source

Class generation now also adds an equality operator

Caleb Fangmeier 6 years ago
parent
commit
de09db9ee9
1 changed files with 4 additions and 0 deletions
  1. 4 0
      tools/generate_class.py

+ 4 - 0
tools/generate_class.py

@@ -75,6 +75,10 @@ struct {obj_name} {{
     src.append('};')
 
     src.append(f'''
+bool operator==(const {obj_name}& obj1, const {obj_name}& obj2) {{
+    return obj1.idx == obj2.idx;
+}}
+
 {obj_name} {obj_name}Collection::iter::operator*() {{
     return {{collection, idx}};
 }}