Skip to content

GitLab

Commit d1334405 authored by Robert McGibbon's avatar Robert McGibbon
Browse files

Add support for pickle protocol 2 for Vec3

parent 49cb91c5
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
wrappers/python/simtk/openmm/vec3.py
View file @ d1334405
...@@ -40,6 +40,10 @@ class Vec3(tuple): ...@@ -40,6 +40,10 @@ class Vec3(tuple):
"""Create a new Vec3.""" """Create a new Vec3."""
return tuple.__new__(cls, (x, y, z)) return tuple.__new__(cls, (x, y, z))
def __getnewargs__(self):
"Support for pickle protocol 2: http://docs.python.org/2/library/pickle.html#pickling-and-unpickling-normal-class-instances"
return self[0], self[1], self[2]
def __add__(self, other): def __add__(self, other):
"""Add two Vec3s.""" """Add two Vec3s."""
return Vec3(self[0]+other[0], self[1]+other[1], self[2]+other[2]) return Vec3(self[0]+other[0], self[1]+other[1], self[2]+other[2])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment