quantum-espresso/PP/xsf.f90

207 lines
6.1 KiB
Fortran

!
! This file holds XSF (=Xcrysden Structure File) utilities.
! Routines written by Tone Kokalj on Mon Jan 27 18:51:17 CET 2003
!
! This file is distributed under the terms of the
! GNU General Public License. See the file `License'
! in the root directory of the present distribution,
! or http://www.gnu.org/copyleft/gpl.txt .
! -------------------------------------------------------------------
! this routine writes the crystal structure in XSF format
! -------------------------------------------------------------------
subroutine xsf_struct (alat, at, nat, tau, atm, ityp, ounit)
use parameters, only : DP
implicit none
integer :: nat, ityp (nat), ounit
character(len=3) :: atm(*)
real(kind=DP) :: alat, tau (3, nat), at (3, 3)
! --
integer :: i, j, n
real(kind=DP) :: at1 (3, 3)
! convert lattice vectors to ANGSTROM units ...
do i=1,3
do j=1,3
at1(j,i) = at(j,i)*alat*0.529177d0
enddo
enddo
write(ounit,*) 'CRYSTAL'
write(ounit,*) 'PRIMVEC'
write(ounit,'(2(3F15.9/),3f15.9)') at1
write(ounit,*) 'PRIMCOORD'
write(ounit,*) nat, 1
do n=1,nat
! positions are in Angstroms
write(ounit,'(a3,3x,3f15.9)') atm(ityp(n)), &
tau(1,n)*alat*0.529177d0, &
tau(2,n)*alat*0.529177d0, &
tau(3,n)*alat*0.529177d0
enddo
return
end subroutine xsf_struct
! -------------------------------------------------------------------
! this routine writes the 3D scalar field (i.e. uniform mesh of points)
! in XSF format using the FFT mesh (i.e. fast write)
! -------------------------------------------------------------------
subroutine xsf_fast_datagrid_3d &
(rho, nr1, nr2, nr3, nrx1, nrx2, nrx3, at, alat, ounit)
use parameters, only : DP
implicit none
integer :: nrx1, nrx2, nrx3, nr1, nr2, nr3, ounit
real(kind=DP) :: alat, at (3, 3), rho(nrx1,nrx2,nrx3)
! --
integer :: i1, i2, i3, ix, iy, iz, count, i, ii, &
ind_x(10), ind_y(10),ind_z(10)
! XSF scalar-field header
write(ounit,'(a)') 'BEGIN_BLOCK_DATAGRID_3D'
write(ounit,'(a)') '3D_PWSCF'
write(ounit,'(a)') 'DATAGRID_3D_UNKNOWN'
! number of points in each direction
write(ounit,*) nr1+1, nr2+1, nr3+1
! origin
write(ounit,'(3f10.6)') 0.0, 0.0, 0.0
! 1st spanning (=lattice) vector
write(ounit,'(3f10.6)') (0.529177d0*alat*at(i,1),i=1,3) ! in ANSTROMS
! 2nd spanning (=lattice) vector
write(ounit,'(3f10.6)') (0.529177d0*alat*at(i,2),i=1,3)
! 3rd spanning (=lattice) vector
write(ounit,'(3f10.6)') (0.529177d0*alat*at(i,3),i=1,3)
count=0
do i3=0,nr3
!iz = mod(i3,nr3)
iz = mod(i3,nr3) + 1
do i2=0,nr2
!iy = mod(i2,nr2)
iy = mod(i2,nr2) + 1
do i1=0,nr1
!ix = mod(i1,nr1)
ix = mod(i1,nr1) + 1
!ii = (1+ix) + iy*nrx1 + iz*nrx1*nrx2
if (count.lt.6) then
count = count + 1
!ind(count) = ii
else
write(ounit,'(6e13.5)') &
(rho(ind_x(i),ind_y(i),ind_z(i)),i=1,6)
count=1
!ind(count) = ii
endif
ind_x(count) = ix
ind_y(count) = iy
ind_z(count) = iz
enddo
enddo
enddo
write(ounit,'(6e13.5:)') (rho(ind_x(i),ind_y(i),ind_z(i)),i=1,count)
write(ounit,'(a)') 'END_DATAGRID_3D'
write(ounit,'(a)') 'END_BLOCK_DATAGRID_3D'
return
end subroutine xsf_fast_datagrid_3d
subroutine xsf_datagrid_2d (rho, nx, ny, m1, m2, x0, e1, e2, alat, ounit)
use parameters, only : DP
implicit none
integer :: nx, ny, ounit
real(kind=DP) :: m1, m2, alat, x0(3), e1(3), e2(3), rho(2, nx, ny)
! --
integer :: ix, iy, count, i, ind_x(10), ind_y(10)
! XSF scalar-field header
write(ounit,'(a)') 'BEGIN_BLOCK_DATAGRID_2D'
write(ounit,'(a)') '2D_PWSCF'
write(ounit,'(a)') 'DATAGRID_2D_UNKNOWN'
! number of points in each direction
write(ounit,*) nx, ny
! origin
write(ounit,'(3f10.6)') (0.529177d0*alat*x0(i),i=1,3) ! in ANSTROMS
! 1st spanning (=lattice) vector
write(ounit,'(3f10.6)') (0.529177d0*alat*e1(i)*m1,i=1,3) ! in ANSTROMS
! 2nd spanning (=lattice) vector
write(ounit,'(3f10.6)') (0.529177d0*alat*e2(i)*m2,i=1,3) ! in ANSTROMS
count=0
do iy=1,ny
do ix=1,nx
if (count < 6) then
count = count + 1
else
write(ounit,'(6e13.5)') (rho(1,ind_x(i),ind_y(i)),i=1,6)
count=1
endif
ind_x(count) = ix
ind_y(count) = iy
enddo
enddo
write(ounit,'(6e13.5:)') (rho(1,ind_x(i),ind_y(i)),i=1,count)
write(ounit,'(a)') 'END_DATAGRID_2D'
write(ounit,'(a)') 'END_BLOCK_DATAGRID_2D'
return
end subroutine xsf_datagrid_2d
subroutine xsf_datagrid_3d &
(rho, nx, ny, nz, m1, m2, m3, x0, e1, e2, e3, alat, ounit)
use parameters, only : DP
implicit none
integer :: nx, ny, nz, ounit
real(kind=DP) :: m1, m2, m3, alat, x0(3), e1(3),e2(3),e3(3), rho(nx, ny, nz)
! --
integer :: ix, iy, iz, count, i, ind_x(10), ind_y(10), ind_z(10)
! XSF scalar-field header
write(ounit,'(a)') 'BEGIN_BLOCK_DATAGRID_3D'
write(ounit,'(a)') '3D_PWSCF'
write(ounit,'(a)') 'DATAGRID_3D_UNKNOWN'
! number of points in each direction
write(ounit,*) nx, ny, nz
! origin
write(ounit,'(3f10.6)') (0.529177d0*alat*x0(i),i=1,3) ! in ANSTROMS
! 1st spanning (=lattice) vector
write(ounit,'(3f10.6)') (0.529177d0*alat*e1(i)*m1,i=1,3) ! in ANSTROMS
! 2nd spanning (=lattice) vector
write(ounit,'(3f10.6)') (0.529177d0*alat*e2(i)*m2,i=1,3) ! in ANSTROMS
! 3rd spanning (=lattice) vector
write(ounit,'(3f10.6)') (0.529177d0*alat*e3(i)*m3,i=1,3)
count=0
do iz=1,nz
do iy=1,ny
do ix=1,nx
if (count.lt.6) then
count = count + 1
else
write(ounit,'(6e13.5)') (rho(ind_x(i),ind_y(i),ind_z(i)),i=1,6)
count=1
endif
ind_x(count) = ix
ind_y(count) = iy
ind_z(count) = iz
enddo
enddo
enddo
write(ounit,'(6e13.5:)') (rho(ind_x(i),ind_y(i),ind_z(i)),i=1,count)
write(ounit,'(a)') 'END_DATAGRID_3D'
write(ounit,'(a)') 'END_BLOCK_DATAGRID_3D'
return
end subroutine xsf_datagrid_3d