mirror of
https://github.com/triqs/dft_tools
synced 2024-11-18 12:03:50 +01:00
126 lines
3.8 KiB
Python
126 lines
3.8 KiB
Python
|
|
################################################################################
|
|
#
|
|
# TRIQS: a Toolbox for Research in Interacting Quantum Systems
|
|
#
|
|
# Copyright (C) 2011 by M. Ferrero, O. Parcollet
|
|
#
|
|
# DFT tools: Copyright (C) 2011 by M. Aichhorn, L. Pourovskii, V. Vildosola
|
|
#
|
|
# PLOVasp: Copyright (C) 2015 by O. E. Peil
|
|
#
|
|
# TRIQS is free software: you can redistribute it and/or modify it under the
|
|
# terms of the GNU General Public License as published by the Free Software
|
|
# Foundation, either version 3 of the License, or (at your option) any later
|
|
# version.
|
|
#
|
|
# TRIQS is distributed in the hope that it will be useful, but WITHOUT ANY
|
|
# WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
|
|
# FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
|
|
# details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License along with
|
|
# TRIQS. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
################################################################################
|
|
r"""
|
|
plovasp.converter
|
|
=================
|
|
|
|
PLOVASP is a tool to transform raw, non-normalized
|
|
projectors generated by VASP into normalized projectors
|
|
corresponding to user-defined projected localized orbitals (PLO).
|
|
|
|
Runs routines in proper order to generate and store PLOs.
|
|
|
|
Usage: python converter.py <conf-file> [<path-to-vasp-calculation>]
|
|
"""
|
|
import sys
|
|
from . import vaspio
|
|
from .inpconf import ConfigParameters
|
|
from .elstruct import ElectronicStructure
|
|
from .plotools import generate_plo, output_as_text
|
|
import logging
|
|
|
|
class PloFormatter(logging.Formatter):
|
|
"""
|
|
custom event logger for all output, warnings and debug info
|
|
"""
|
|
def __init__(self, default):
|
|
self._default_formatter = default
|
|
|
|
def format(self, record):
|
|
# Save the original format
|
|
_style = self._style
|
|
|
|
# Customized WARNING format
|
|
if record.levelno == logging.WARNING:
|
|
self._style = logging.PercentStyle("\n!!! WARNING !!!: %(msg)s\n")
|
|
|
|
result = super().format(record)
|
|
|
|
# Restore the original format
|
|
self._style = _style
|
|
|
|
return result
|
|
|
|
# Uncomment this to get extra output
|
|
#logging.basicConfig(level=logging.DEBUG)
|
|
|
|
# Main logger from which all other loggers should be inherited
|
|
main_log = logging.getLogger('plovasp')
|
|
main_log.propagate = False
|
|
|
|
handler = logging.StreamHandler(sys.stdout)
|
|
formatter = PloFormatter("[%(levelname)s]:[%(name)s]: %(message)s")
|
|
handler.setFormatter(formatter)
|
|
main_log.addHandler(handler)
|
|
|
|
|
|
def generate_and_output_as_text(conf_filename, vasp_dir):
|
|
"""
|
|
Parse config file, process VASP data, and store as text.
|
|
"""
|
|
# Prepare input-file parameters
|
|
pars = ConfigParameters(conf_filename, verbosity=0)
|
|
pars.parse_input()
|
|
|
|
# Read VASP data
|
|
if 'efermi' in pars.general:
|
|
efermi_required = False
|
|
else:
|
|
efermi_required = True
|
|
vasp_data = vaspio.VaspData(vasp_dir, efermi_required=efermi_required)
|
|
el_struct = ElectronicStructure(vasp_data)
|
|
el_struct.debug_density_matrix()
|
|
if 'efermi' in pars.general:
|
|
el_struct.efermi = pars.general['efermi']
|
|
|
|
# Generate and store PLOs
|
|
pshells, pgroups = generate_plo(pars, el_struct)
|
|
output_as_text(pars, el_struct, pshells, pgroups)
|
|
|
|
def main():
|
|
"""
|
|
This function should not be called directly but via a bash script
|
|
'plovasp' invoking the main function as follows:
|
|
|
|
python -m applications.dft.converters.plovasp.converter $@
|
|
"""
|
|
narg = len(sys.argv)
|
|
if narg < 2:
|
|
raise SystemExit(" Usage: plovasp <conf-file> [<path-to-vasp-calcultaion>]")
|
|
else:
|
|
filename = sys.argv[1]
|
|
if narg > 2:
|
|
vasp_dir = sys.argv[2]
|
|
if vasp_dir[-1] != '/':
|
|
vasp_dir += '/'
|
|
else:
|
|
vasp_dir = './'
|
|
|
|
generate_and_output_as_text(filename, vasp_dir)
|
|
|
|
if __name__ == '__main__':
|
|
main()
|