# Path: pyDFTutils/vasp/vasp_dos.py
# def plot_all_pdos(element_types=None,filename='DOSCAR',ispin=2,ymin=-2.0,ymax=2.0,xmin=-15.0,xmax=5.0,output_dir='PDOS',orbs=['s','p','eg','t2g']):
#     for site in orbs:
#         plot_pdos(sites=site,element_types=element_types,filename=filename,ispin=ispin,ymin=ymin,ymax=ymax,xmin=xmin,xmax=xmax,output_dir=output_dir)


from pyDFTutils.vasp.vasp_dos import plot_all_pdos
import argparse
#!/usr/bin/env python

def plot_pdos():
    parser=argparse.ArgumentParser(description='Plot the partial dos')
    parser.add_argument('-f','--filename',type=str,help='DOSCAR filename',default='DOSCAR')
    parser.add_argument('-n','--ispin',type=int,help='number os spins, 1 or 2',default=2)
    parser.add_argument('-o','--output',type=str,help='output dir',default='PDOS')
    parser.add_argument('--xmin',type=float,help='xmin',default=-15.0)
    parser.add_argument('--xmax',type=float,help='xmax',default=5.0)
    parser.add_argument('--ymin',type=float,help='ymin',default=-2.0)
    parser.add_argument('--ymax',type=float,help='xmax',default=2.0)
    parser.add_argument('-s','--sites',type=str,help='sites',default=['s','p','eg','t2g'],nargs='+')
    parser.add_argument('-e','--element',type=str,help='element',default=None,nargs='+')
    args=parser.parse_args()

    plot_all_pdos(element_types=args.element,filename=args.filename,ispin=args.ispin,ymin=args.ymin,ymax=args.ymax,xmin=args.xmin,xmax=args.xmax,output_dir=args.output,orbs=args.sites)


# Path: pyDFTutils/vasp/vasp_dos.py
# def plot_all_pdos(element_types=None,filename='DOSCAR',ispin=2,ymin=-2.0,ymax=2.0,xmin=-15.0,xmax=5.0,output_dir='PDOS',orbs=['s','p','eg','t2g']):
#     for site in orbs:
#         plot_pdos(sites=site,element_types=element_types,filename=filename,ispin=ispin,ymin=ymin,ymax=ymax,xmin=xmin,xmax=xmax,output_dir=output_dir)


# Path: scripts/plotpdos.py
from pyDFTutils.vasp.vasp_dos import plot_all_pdos
import argparse

#!/usr/bin/env python

def plot_pdos():
    parser=argparse.ArgumentParser(description='Plot the partial dos')
    parser.add_argument('-f','--filename',type=str,help='DOSCAR filename',default='DOSCAR')
    parser.add_argument('-n','--ispin',type=int,help='number os spins, 1 or 2',default=2)
    parser.add_argument('-o','--output',type=str,help='output dir',default='PDOS')
    parser.add_argument('--xmin',type=float,help='xmin',default=-15.0)
    parser.add_argument('--xmax',type=float,help='xmax',default=5.0)
    parser.add_argument('--ymin',type=float,help='ymin',default=-2.0)
    parser.add_argument('--ymax',type=float,help='xmax',default=2.0)
    parser.add_argument('-s','--sites',type=str,help='sites',default=['s','p','eg','t2g'],nargs='+')
    parser.add_argument('-e','--element',type=str,help='element',default=None,nargs='+')
    args=parser.parse_args()

    plot_all_pdos(element_types=args.element,filename=args.filename,ispin=args.ispin,ymin=args.ymin,ymax=args.ymax,xmin=args.xmin,xmax=args.xmax,output_dir=args.output,orbs=args.sites)


if __name__=='__main__':