crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-27 21:55:41 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

Adding getitem method to _Position.

Browse source
This commit is contained in:
Patrick Shriwise 2019-03-14 12:13:38 -05:00
parent 14e4665301
commit af54f3512e
1 changed files with 10 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
openmc/capi/plot.py
View file

@ -23,6 +23,16 @@ class _Position(Structure):
('y', c_double),
('z', c_double)]
def __getitem__(self, key):
if key == 0:
return self.x
elif key == 1:
return self.y
elif key == 2:
return self.z
raise ValueError("{} index is invalid for _Position".format(key))
def __repr__(self):
return "({}, {}, {})".format(self.x, self.y, self.z)