make the interface for Symmetry class compatible with pure python version.

This commit is contained in:
Chen Zhang 2015-07-29 19:39:30 +00:00
parent 5f74f3e95e
commit 9287d6f15f
1 changed files with 6 additions and 7 deletions

View File

@ -604,20 +604,19 @@ cdef class Symmetry:
# HEXAGONAL = 3 # HEXAGONAL = 3
# CUBIC = 4 # CUBIC = 4
def __init__(self, int symmetry): def __init__(self, symmetry):
if symmetry == 0: if symmetry == 0 or symmetry == None:
self.lattice = NONE self.lattice = NONE
elif symmetry == 1: elif symmetry == 1 or symmetry == 'orthorhombic':
self.lattice = ORTHORHOMBIC self.lattice = ORTHORHOMBIC
elif symmetry == 2: elif symmetry == 2 or symmetry == 'tetragonal':
self.lattice = TETRAGONAL self.lattice = TETRAGONAL
elif symmetry == 3: elif symmetry == 3 or symmetry == 'hexagonal':
self.lattice = HEXAGONAL self.lattice = HEXAGONAL
elif symmetry == 4: elif symmetry == 4 or symmetry == 'cubic':
self.lattice = CUBIC self.lattice = CUBIC
else: else:
self.lattice = NONE self.lattice = NONE
# self.lattice = symmetry # use code here
def __copy__(self): def __copy__(self):
return self.__class__(self.lattice) return self.__class__(self.lattice)