@@ -444,7 +444,7 @@ def parse_args():
444
444
parser .add_argument ('-rm' , \
445
445
help = 'removes processed files' , \
446
446
action = "store_true" )
447
- parser .add_argument ('-alt' , default = 2 , type = int , \
447
+ parser .add_argument ('-alt' , default = - 1 , type = int , \
448
448
help = 'altitude to plot (-1 for no plot!)' )
449
449
parser .add_argument ('-v' , \
450
450
help = 'turn on verbose mode' , \
@@ -463,11 +463,11 @@ def get_core_file(filename):
463
463
isEnsemble = False
464
464
ensembleFile = ''
465
465
ensembleNumber = - 1
466
- m = re .match ('.*([0123]D.*)(_g\d *)(\..*)' ,filename )
466
+ m = re .match (r '.*([0123]D.*)(_g[0-9] *)(\..*)' ,filename )
467
467
if m :
468
468
coreFile = m .group (1 )
469
469
# check if file is a member of an ensemble:
470
- check = re .match ('.*([0123]D.*)(_m)(\d *)' ,coreFile )
470
+ check = re .match ('.*([0123]D.*)(_m)([0-9] *)' ,coreFile )
471
471
if (check ):
472
472
ensembleFile = check .group (1 )
473
473
isEnsemble = True
@@ -934,7 +934,7 @@ def write_and_plot_data(dataToWrite,
934
934
print (' --> Outputting hdf5 file : ' , hdf5File )
935
935
write_hdf5 (dataToWrite , hdf5File , isVerbose = isVerbose )
936
936
937
- if (iAlt > - 1 ):
937
+ if (( iAlt > - 1 ) & ( len ( dataToWrite ) > 1 ) ):
938
938
plotFile = fileStart + fileAddon + '.png'
939
939
var = dataToWrite [0 ]['vars' ][iVar ]
940
940
plot_all_blocks (dataToWrite , var , iAlt , plotFile )
0 commit comments