From 228c5136d857216d5732eb2f988340bd37261ca6 Mon Sep 17 00:00:00 2001 From: Garritt McCune Date: Tue, 21 Jul 2020 14:43:27 -0500 Subject: [PATCH] Minor tweaks to some of the objects to make indexing them easier. --- engine.py | 6 ++++++ triangle.py | 3 +++ 2 files changed, 9 insertions(+) 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]