!
! Copyright (C) 2001-2003 PWSCF group
! 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 .
!
#include "f_defs.h"
!
!----------------------------------------------------------------------------
SUBROUTINE allocate_wfc()
  !----------------------------------------------------------------------------
  !
  ! ... dynamical allocation of arrays: wavefunctions and eigenvectors
  ! ... this is a wrapper that calls specific routines
  !
  USE io_global, ONLY : stdout
  USE wvfct,     ONLY : gamma_only
  USE wvfct,     ONLY : npwx, nbnd, nbndx
  USE klist,     ONLY : nelec
  USE basis,     ONLY : natomwfc
  USE gvect,     ONLY : ngl
  USE uspp,      ONLY : nkb
  !
  IMPLICIT NONE
  !
  !    
  IF ( gamma_only ) THEN
     !
     CALL allocate_wfc_gamma()
     !
  ELSE
     !
     CALL allocate_wfc_k()
     !
  END IF
  !
  WRITE( stdout, 100) nbndx, nbnd, natomwfc, npwx, nelec, nkb, ngl
  !
100 FORMAT(/5X,'nbndx  = ',I5,'  nbnd   = ',I5,'  natomwfc = ',I5, &
       &       '  npwx   = ',I7, &
       &   /5X,'nelec  = ',F7.2,' nkb   = ',I5,'  ngl    = ',I7)
  !       
  RETURN
  !
  CONTAINS
     !-----------------------------------------------------------------------
     SUBROUTINE allocate_wfc_gamma()
       !-----------------------------------------------------------------------
       !
       ! ... routine specific for calculations at gamma point
       !
       USE wvfct,                ONLY : et, wg
       USE klist,                ONLY : nkstot
       USE ldaU,                 ONLY : swfcatom, lda_plus_u
       USE wavefunctions_module, ONLY : evc
       !
       IMPLICIT NONE
       !
       ! ... allocate memory
       !
       ALLOCATE( et( nbnd, nkstot ) )    
       ALLOCATE( wg( nbnd, nkstot ) )    
       ALLOCATE( evc( npwx, nbnd ) )    
       !
       ! ... needed for LDA+U
       !
       IF ( lda_plus_u ) ALLOCATE( swfcatom( npwx, natomwfc) )    
       !  
       et(:,:) = 0.D0
       !
       RETURN
       !
     END SUBROUTINE allocate_wfc_gamma
     !
     !
     !-----------------------------------------------------------------------
     SUBROUTINE allocate_wfc_k()
       !-----------------------------------------------------------------------
       !
       ! ... routine for calculations with general BZ samplig
       !
       USE wvfct,                ONLY : et, wg
       USE klist,                ONLY : nkstot
       USE ldaU,                 ONLY : swfcatom, lda_plus_u
       USE noncollin_module,     ONLY : noncolin, npol
       USE wavefunctions_module, ONLY : evc, evc_nc
       !
       IMPLICIT NONE
       !
       ! ... allocate memory
       !
       ALLOCATE( et( nbnd, nkstot ) )    
       ALLOCATE( wg( nbnd, nkstot ) )    
       IF (noncolin) THEN
          ALLOCATE( evc_nc( npwx, npol, nbnd ) )    
       ELSE
          ALLOCATE( evc( npwx, nbnd ) )    
       ENDIF
       !
       ! ... needed for LDA+U
       !
       IF ( lda_plus_u ) ALLOCATE( swfcatom( npwx, natomwfc) )    
       !
       et(:,:) = 0.D0
       !  
       RETURN
       !
     END SUBROUTINE allocate_wfc_k  
     !      
END subroutine allocate_wfc