diff --git a/engine.py b/engine.py index 5a22538..a4d5bc5 100644 --- a/engine.py +++ b/engine.py @@ -43,3 +43,9 @@ class Mesh: row3 = self.vertices[int(tmp[3]) - 1] self.triangles.append(Triangle(Matrix3x3(row1, row2, row3))) + + def __getitem__(self, key): + return self.triangles[key] + + def __len__(self): + return len(self.triangles) diff --git a/triangle.py b/triangle.py index cdb920c..e10b227 100644 --- a/triangle.py +++ b/triangle.py @@ -5,3 +5,6 @@ class Triangle: self.coords = coords # Matrix3x3 self.object_id = None self.color = None + + def __getitem__(self, key): + return self.coords[key]