#! /usr/bin/env python

import sys
import string
import math
from random import gauss
from numpy import linalg, dot
import numpy as np

POSCAR = open('POSCAR', 'r').readlines()

N=38
g=open('cfg_lammps.RuN.config','w')
g.write('Written by Genlammps_AIMD.py, original comment line\n\n')
g.write('   {0:.0f}  atoms\n'.format(N))
g.write('   3  atom types\n\n')
g.write( "  0.0000000000000000 8.6930937287162351 xlo xhi\n" )
g.write( "  0.0000000000000000 7.5284400065474486 ylo yhi\n" )
g.write( "  0.0000000000000000 22.2281052000000017 zlo zhi\n" )
g.write( "  4.3465468643581175 0.0000000000000000 0.0000000000000000 xy xz yz\n\n" )
g.write('Atoms\n\n')
for k in range(0,N):
	if k < 36:
		g.write( " {0:4d}  3  {1:16.12f}  {2:16.12f}  {3:16.12f}\n".format( k+1, float(POSCAR[k+9].split()[0]), float(POSCAR[k+9].split()[1]), float(POSCAR[k+9].split()[2]) ) )
	elif k == 36:
		g.write( " {0:4d}  1  {1:16.12f}  {2:16.12f}  {3:16.12f}\n".format( k+1, float(POSCAR[k+9].split()[0]), float(POSCAR[k+9].split()[1]), float(POSCAR[k+9].split()[2]) ) )
	else:
		g.write( " {0:4d}  2  {1:16.12f}  {2:16.12f}  {3:16.12f}\n".format( k+1, float(POSCAR[k+9].split()[0]), float(POSCAR[k+9].split()[1]), float(POSCAR[k+9].split()[2]) ) )

g.write('\n')
g.write('Velocities\n\n')
for k in range(0,N):
	g.write( " {0:4d}  {1:16.12f}  {2:16.12f}  {3:16.12f}\n".format( k+1, float(POSCAR[k+48].split()[0])*1000, float(POSCAR[k+48].split()[1])*1000, float(POSCAR[k+48].split()[2])*1000 ) )
g.close()