AtHeartEngineer/kaipy
1
0
Fork 0
mirror of https://github.com/JHUAPL/kaipy.git synced 2026-01-09 14:28:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

Merged in dev_312 (pull request #7)

Browse Source 
Dev 312
This commit is contained in:
Nikhil Rao
2025-06-19 16:39:58 +00:00
parent 67d48fba39 c9dd4218c0
commit 3fff43d379
1 changed files with 26 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
kaipy/scripts/quicklook/msphpic.py
View File

@@ -38,7 +38,7 @@ import kaipy.kaiViz as kv
import kaipy.kaiTools as ktools
import kaipy.kdefs as kdefs
import kaipy.remix.remix as remix
import kaipy.raiju.raijuViz as rv
# Program constants and defaults
@@ -158,7 +158,7 @@ def create_command_line_parser():
)
parser.add_argument(
"-bigIM", action="store_true", default=False,
help="Show entire RCM domain (default: %(default)s)."
help="Show entire Inner Mag domain (default: %(default)s)."
)
parser.add_argument(
"-src", action="store_true", default=False,
@@ -205,8 +205,8 @@ def makePlot(i,spacecraft,nStp, args, varDict):
doEphi = args.ephi
doSrc = args.src
doBz = args.bz
noRCM = args.norcm
doBigRCM = args.bigrcm
noIM = args.noIM
bigIM = args.bigIM
do_vid = args.vid
do_overwrite = args.overwrite
do_hash = not args.nohash
@@ -217,14 +217,17 @@ def makePlot(i,spacecraft,nStp, args, varDict):
outDir = varDict.get('outDir', 'msphVid')
doIon = varDict.get('doIon', not noIon)
doMIX = varDict.get('doMIX', False)
doRCM = varDict.get('doRCM', not noRCM)
doIM = varDict.get('doIM', not noIM)
doBigIM = varDict.get('doBigIM', bigIM)
rcmdata = varDict.get('rcmdata', None)
raiI = varDict.get('raiI', None)
rmxChk = varDict.get('rmxChk', None)
gsph = varDict.get('gsph', None)
xyBds = varDict.get('xyBds', None)
branch = varDict.get('branch', 'unknown')
githash = varDict.get('githash', 'unknown')
figSz = varDict.get('figSz', (12, 7.5))
imName = varDict.get('imName', None)
# Init figure
@@ -299,6 +302,7 @@ def makePlot(i,spacecraft,nStp, args, varDict):
# If available, add the inset Inner Mag plot.
if imName is not None:
mviz.vP = kv.genNorm(1.0e-2, 100.0, doLog=True)
if imName=="RAIJU":
stepStr = "Step#"+str(nStp)
AxIM = inset_axes(AxL, width="30%", height="30%", loc=3)
@@ -473,7 +477,7 @@ def main():
import kaipy.raiju.raijuUtils as ru
import kaipy.raiju.raijuViz as rv
raiI = ru.RAIJUInfo(os.path.join(fdir, ftag + ".raiju.h5"))
mviz.vP = kv.genNorm(1.0e-2, 100.0, doLog=True)
else:
raiI = None
if imName=="RCM":
@@ -496,21 +500,24 @@ def main():
if not do_vid: # If we are making a single image, keep original functionality
# If needed, fetch the number of the last step.
if nStp < 0:
nStp = gsph.sFin
nStp = gsph.sFin - 1
print("Using Step %d" % nStp)
varDict = {
'n_pad': 0,
'outDir': 'msphVid',
'doIon': doIon,
'doMIX': doMIX,
'doRCM': doRCM,
'doIM': doIM,
'doBigIM': doBigIM,
'rcmdata': rcmdata,
'raiI': raiI,
'rmxChk': rmxChk,
'gsph': gsph,
'xyBds': xyBds,
'branch': branch,
'githash': githash,
'figSz': figSz
'figSz': figSz,
'imName': imName
}
makePlot(nStp,spacecraft,nStp, args, varDict)
@@ -534,14 +541,17 @@ def main():
'outDir': outDir,
'doIon': doIon,
'doMIX': doMIX,
'doRCM': doRCM,
'doIM': doIM,
'doBigIM': doBigIM,
'rcmdata': rcmdata,
'raiI': raiI,
'rmxChk': rmxChk,
'gsph': gsph,
'xyBds': xyBds,
'branch': branch,
'githash': githash,
'figSz': figSz
'figSz': figSz,
'imName': imName
}
makePlot(i, spacecraft, nStp, args, varDict)
else:
@@ -550,14 +560,17 @@ def main():
'outDir': outDir,
'doIon': doIon,
'doMIX': doMIX,
'doRCM': doRCM,
'doIM': doIM,
'doBigIM': doBigIM,
'rcmdata': rcmdata,
'raiI': raiI,
'rmxChk': rmxChk,
'gsph': gsph,
'xyBds': xyBds,
'branch': branch,
'githash': githash,
'figSz': figSz
'figSz': figSz,
'imName': imName
}
# Make list of parallel arguments
ag = ((i,spacecraft,nStp, args, varDict) for i, nStp in enumerate(sIds) )