From 9287d6f15f28234c5d2375ce25f8996eef389d31 Mon Sep 17 00:00:00 2001 From: Chen Zhang Date: Wed, 29 Jul 2015 19:39:30 +0000 Subject: [PATCH] make the interface for Symmetry class compatible with pure python version. --- lib/damask/corientation.pyx | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/lib/damask/corientation.pyx b/lib/damask/corientation.pyx index ed56be0a9..002313604 100644 --- a/lib/damask/corientation.pyx +++ b/lib/damask/corientation.pyx @@ -604,20 +604,19 @@ cdef class Symmetry: # HEXAGONAL = 3 # CUBIC = 4 - def __init__(self, int symmetry): - if symmetry == 0: + def __init__(self, symmetry): + if symmetry == 0 or symmetry == None: self.lattice = NONE - elif symmetry == 1: + elif symmetry == 1 or symmetry == 'orthorhombic': self.lattice = ORTHORHOMBIC - elif symmetry == 2: + elif symmetry == 2 or symmetry == 'tetragonal': self.lattice = TETRAGONAL - elif symmetry == 3: + elif symmetry == 3 or symmetry == 'hexagonal': self.lattice = HEXAGONAL - elif symmetry == 4: + elif symmetry == 4 or symmetry == 'cubic': self.lattice = CUBIC else: self.lattice = NONE - # self.lattice = symmetry # use code here def __copy__(self): return self.__class__(self.lattice)