@@ -354,7 +354,6 @@ def write_slater(basis,eigsys,kpt,outfn,orbfn,basisfn,maxmo_spin=-1):
354
354
outlines = [
355
355
"slater" ,
356
356
"{0} {{" .format (orbstr ),
357
- "cutoff_mo" ,
358
357
" magnify 1" ,
359
358
" nmo {0}" .format (dnorbs [- 1 ]),
360
359
" orbfile {0}" .format (orbfn ),
@@ -373,16 +372,17 @@ def write_slater(basis,eigsys,kpt,outfn,orbfn,basisfn,maxmo_spin=-1):
373
372
outf .write ("\n " .join (outlines ))
374
373
375
374
###############################################################################
376
- def write_orbplot (basis ,eigsys ,kpt ,outfn ,orbfn ,basisfn ,sysfn ):
377
- # Warning: the orbfn naming convention is different because there are two files that get produced.
375
+ def write_orbplot (basis ,eigsys ,kpt ,outfn ,orbfn ,basisfn ,sysfn ,maxmo_spin = - 1 ):
378
376
ntot = basis ['ntot' ]
379
377
nmo = basis ['nmo' ]
380
378
nup = eigsys ['nup' ]
381
379
ndn = eigsys ['ndn' ]
382
380
uporbs = np .arange (nup )+ 1
383
381
dnorbs = np .arange (ndn )+ 1
384
- if eigsys ['nspin' ] > 1 :
382
+ if maxmo_spin < 0 and eigsys ['nspin' ] > 1 :
385
383
dnorbs += nmo
384
+ else :
385
+ dnorbs += maxmo_spin
386
386
if eigsys ['ikpt_iscmpx' ][kpt ]: orbstr = "corbitals"
387
387
else : orbstr = "orbitals"
388
388
uporblines = ["{:5d}" .format (orb ) for orb in uporbs ]
@@ -396,7 +396,6 @@ def write_orbplot(basis,eigsys,kpt,outfn,orbfn,basisfn,sysfn):
396
396
"method { " ,
397
397
"plot" ,
398
398
"{0} {{" .format (orbstr ),
399
- "cutoff_mo" ,
400
399
" magnify 1" ,
401
400
" nmo {0}" .format (dnorbs [- 1 ]),
402
401
" orbfile {0}" .format (orbfn ),
@@ -786,7 +785,8 @@ def convert_crystal(
786
785
outfn = files ['orbplot' ][kidx ],
787
786
orbfn = files ['orb' ][kidx ],
788
787
basisfn = files ['basis' ],
789
- sysfn = files ['sys' ][kidx ])
788
+ sysfn = files ['sys' ][kidx ],
789
+ maxmo_spin = maxmo_spin )
790
790
normalize_eigvec (eigsys ,basis ,kpt )
791
791
write_orb (eigsys ,basis ,ions ,kpt ,files ['orb' ][kidx ],maxmo_spin )
792
792
write_sys (lat_parm ,basis ,eigsys ,pseudo ,ions ,kpt ,files ['sys' ][kidx ])
0 commit comments