Example #1
0
 def parser(self, ):
     """ parse poscar to system
     @todo : handle coordinates acoording to Direct tag
     """
     s = System()
     scale = float(self.scale)
     a = [scale * float(i) for i in self.a]
     b = [scale * float(i) for i in self.b]
     c = [scale * float(i) for i in self.c]
     s.name = self.name
     s.pbc = v2lattice(a, b, c)
     s.natoms = self.natoms
     s.atomtypes = self.atomtypes
     s.scaleFactor = self.scale
     s.geotag = "XTLGRF 200"
     for i in range(len(self.natoms)):
         if i == 0:
             prev = 0
             now = int(self.natoms[0])
         else:
             prev = now
             now += int(self.natoms[i])
         for j in range(prev, now):
             atom = Atom()
             #print self.atomtypes, i
             atom.name = self.atomtypes[i]
             x = float(self.coords[j][0])
             y = float(self.coords[j][1])
             z = float(self.coords[j][2])
             atom.xFrac = [x, y, z]
             atom.x[0] = a[0] * x + b[0] * y + c[0] * z
             atom.x[1] = a[1] * x + b[1] * y + c[1] * z
             atom.x[2] = a[2] * x + b[2] * y + c[2] * z
             if len(self.coords[j]) == 6 and self.sd == 1:
                 xr = self.coords[j][3]
                 yr = self.coords[j][4]
                 zr = self.coords[j][5]
                 if xr == "F":
                     atom.xr[0] = 1
                 if yr == "F":
                     atom.xr[1] = 1
                 if zr == "F":
                     atom.xr[2] = 1
             if len(self.coords[j]) == 3:
                 atom.xr = [0, 0, 0]
             s.atoms.append(atom)
     return s
Example #2
0
 def parser(self,):
     """ parse poscar to system
     @todo : handle coordinates acoording to Direct tag
     """
     s = System()
     scale = float(self.scale)
     a = [scale*float(i) for i in self.a]
     b = [scale*float(i) for i in self.b]
     c = [scale*float(i) for i in self.c]
     s.name = self.name
     s.pbc = v2lattice(a, b, c)
     s.natoms = self.natoms
     s.atomtypes = self.atomtypes
     s.scaleFactor = self.scale
     s.geotag = "XTLGRF 200"
     for i in range(len(self.natoms)):
         if i == 0:
             prev = 0
             now = int(self.natoms[0])
         else:
             prev = now
             now += int(self.natoms[i])
         for j in range(prev, now):
             atom = Atom()
             #print self.atomtypes, i
             atom.name = self.atomtypes[i]
             x = float(self.coords[j][0])
             y = float(self.coords[j][1])
             z = float(self.coords[j][2])
             atom.xFrac = [x, y, z]
             atom.x[0] = a[0]*x + b[0]*y + c[0]*z
             atom.x[1] = a[1]*x + b[1]*y + c[1]*z
             atom.x[2] = a[2]*x + b[2]*y + c[2]*z
             if len(self.coords[j]) == 6 and self.sd == 1:
                 xr = self.coords[j][3]
                 yr = self.coords[j][4]
                 zr = self.coords[j][5]
                 if xr == "F":
                     atom.xr[0] = 1
                 if yr == "F":
                     atom.xr[1] = 1
                 if zr == "F":
                     atom.xr[2] = 1
             s.atoms.append(atom)
     return s
Example #3
0
    def parser(self):
        s = System()
        s.pbc = v2lattice(self.a, self.b, self.c)
        s.atomtypes = self.atomtypes
        s.n_atoms = self.n_atoms
        s.n_bonds = self.n_bonds
        s.n_angles = self.n_angles
        s.n_dihedrals = self.n_dihedrals
        s.n_impropers = self.n_impropers
        s.n_atomtypes = self.n_atomtypes
        s.n_bondtypes = self.n_bondtypes
        s.n_angletypes = self.n_angletypes
        s.n_dihedraltypes = self.n_dihedraltypes
        s.n_impropertypes = self.n_impropertypes
        s.bonds = self.bonds
        s.angles = self.angles
        s.dihedrals = self.dihedrals
        s.ffparams = self.ffparams

        for i in self.coords:
            # n is used to distinguish the type of data file
            n = 0
            for j in i:
                # ignore the comments
                if "#" in j:
                    break
                n += 1
            atom = Atom()
            atom.an = int(i[0])
            if n == 7:
                # n = 7 is full type
                atom.resn = int(i[1])
                atom.name = self.elements[int(i[2]) - 1]
                atom.element = self.elements[int(i[2]) - 1]
                atom.type2 = int(i[2])
                atom.charge = float(i[3])
                atom.x[0] = float(i[4])
                atom.x[1] = float(i[5])
                atom.x[2] = float(i[6])
                s.atoms.append(atom)
        return s
Example #4
0
    def parser(self):
        s = System()
        s.pbc = v2lattice(self.a, self.b, self.c)
        s.atomtypes = self.atomtypes
        s.n_atoms = self.n_atoms
        s.n_bonds = self.n_bonds
        s.n_angles = self.n_angles
        s.n_dihedrals = self.n_dihedrals
        s.n_impropers = self.n_impropers
        s.n_atomtypes = self.n_atomtypes
        s.n_bondtypes = self.n_bondtypes
        s.n_angletypes = self.n_angletypes
        s.n_dihedraltypes = self.n_dihedraltypes
        s.n_impropertypes = self.n_impropertypes
        s.bonds = self.bonds
        s.angles = self.angles
        s.dihedrals = self.dihedrals
        s.ffparams = self.ffparams

        for i in self.coords:
            # n is used to distinguish the type of data file
            n = 0
            for j in i:
                # ignore the comments
                if "#" in j:
                    break
                n += 1
            atom = Atom()
            atom.an = int(i[0])
            if n == 7:
                # n = 7 is full type
                atom.resn = int(i[1])
                atom.name = self.elements[int(i[2]) - 1]
                atom.element = self.elements[int(i[2]) - 1]
                atom.type2 = int(i[2])
                atom.charge = float(i[3])
                atom.x[0] = float(i[4]) 
                atom.x[1] = float(i[5]) 
                atom.x[2] = float(i[6]) 
                s.atoms.append(atom)
        return s