Skip to content

Commit

Permalink
fix tab/space bug in sos_filesequencer
Browse files Browse the repository at this point in the history
  • Loading branch information
@Sean-Morrison
Sean-Morrison committed Sep 7, 2022
1 parent 7bd4130 commit d116557
Showing 1 changed file with 172 additions and 172 deletions.
344 changes: 172 additions & 172 deletions bin/sos_filesequencer
View file
Original file line number Diff line number Diff line change
Expand Up @@ -30,11 +30,11 @@ Written by Gary Kushner (LBL). May 2010.

####
def usage():
"""Display usage and exit"""
print("""
usage: sos_filesequencer -f name -i path -p name -l path -s path -m 00000 -e T
"""Display usage and exit"""
print("""
usage: sos_filesequencer -f name -i path -p name -l path -s path -m 00000 -e T
-f Fits file name
-i Fits file directory path
-p Plugmap file name
Expand All @@ -43,215 +43,215 @@ def usage():
-m MJD
-e FPS mode (T for true, F for false)
-a no cal mode (T for true, F for false)
""")
sys.exit(1)
""")
sys.exit(1)


####
class Config:
"""Config Info"""
def __init__(self):
self.fitname = ""
self.fitdir = ""
self.plugname = ""
self.plugdir = ""
self.sosdir = ""
self.mjd = ""
self.fps = "T"
self.nocal = "F"
def __str__(self):
return ("fitname: " + self.fitname + "\n" +
"fitdir: " + self.fitdir + "\n" +
"plugname: " + self.plugname + "\n" +
"plugdir: " + self.plugdir + "\n" +
"sosdir: " + self.sosdir + "\n" +
"mjd: " + self.mjd + "\n" +
"fps: " + self.fps + "\n" +
"nocal: " + self.nocal);
"""Config Info"""
def __init__(self):
self.fitname = ""
self.fitdir = ""
self.plugname = ""
self.plugdir = ""
self.sosdir = ""
self.mjd = ""
self.fps = "T"
self.nocal = "F"
def __str__(self):
return ("fitname: " + self.fitname + "\n" +
"fitdir: " + self.fitdir + "\n" +
"plugname: " + self.plugname + "\n" +
"plugdir: " + self.plugdir + "\n" +
"sosdir: " + self.sosdir + "\n" +
"mjd: " + self.mjd + "\n" +
"fps: " + self.fps + "\n" +
"nocal: " + self.nocal);



####
def parseCmdLine(args):
"""Parse command line arguments"""
"""Parse command line arguments"""

cfg = Config()
# parse with options
try:
opts, pargs = getopt.gnu_getopt(args, "f:i:p:l:s:m:e:a:")
except Exception as e:
print("Illegal option specified.")
print(" ")
print(str(e))
print(" ")
usage()
if len(pargs) != 0:
print("All arguments should be parameters (start with '-')")
print("found " + str(pargs))
usage()
# Fill in the config
for (opt, value) in opts:
if opt == "-f":
cfg.fitname = value
if opt == "-i":
cfg.fitpath = value
if opt == "-p":
cfg.plugname = value
if opt == "-l":
cfg.plugpath = value
if opt == "-s":
cfg.sosdir = value
if opt == "-m":
cfg.mjd = value
if opt == "-e":
cfg.fps = value
if opt == "-a":
cfg.nocal = value
print("Config values: \n" + str(cfg))
return cfg
cfg = Config()
# parse with options
try:
opts, pargs = getopt.gnu_getopt(args, "f:i:p:l:s:m:e:a:")
except Exception as e:
print("Illegal option specified.")
print(" ")
print(str(e))
print(" ")
usage()
if len(pargs) != 0:
print("All arguments should be parameters (start with '-')")
print("found " + str(pargs))
usage()
# Fill in the config
for (opt, value) in opts:
if opt == "-f":
cfg.fitname = value
if opt == "-i":
cfg.fitpath = value
if opt == "-p":
cfg.plugname = value
if opt == "-l":
cfg.plugpath = value
if opt == "-s":
cfg.sosdir = value
if opt == "-m":
cfg.mjd = value
if opt == "-e":
cfg.fps = value
if opt == "-a":
cfg.nocal = value
print("Config values: \n" + str(cfg))
return cfg


####
def flavor(cfg):
"""return the flavor of the fits file"""
return sxpar(os.path.join(cfg.fitpath,cfg.fitname), "flavor")[0].lower()
"""return the flavor of the fits file"""
return sxpar(os.path.join(cfg.fitpath,cfg.fitname), "flavor")[0].lower()

####
def cart(cfg):
"""return the cart of the fits file"""
# return int(sxpar(os.path.join(cfg.fitpath,cfg.fitname), "cartid")[0])
#return sxpar(os.path.join(cfg.fitpath,cfg.fitname), "cartid")[0] #### SSM need to replace when updated traces for FPS-N
return '9'
"""return the cart of the fits file"""
# return int(sxpar(os.path.join(cfg.fitpath,cfg.fitname), "cartid")[0])
#return sxpar(os.path.join(cfg.fitpath,cfg.fitname), "cartid")[0] #### SSM need to replace when updated traces for FPS-N
return '9'

####
####
def plugging(cfg):
"""return the plugging of the fits file"""
return sxpar(os.path.join(cfg.fitpath,cfg.fitname), "CONFID")[0].lower()
"""return the plugging of the fits file"""
return sxpar(os.path.join(cfg.fitpath,cfg.fitname), "CONFID")[0].lower()
####
def samePlugging(cfg1, cfg2):
"""returns True if the fits files use the same plugging"""
return plugging(cfg1) == plugging(cfg2)
"""returns True if the fits files use the same plugging"""
return plugging(cfg1) == plugging(cfg2)
####
def fitsExist(cfg):
"""Return true if the fits file exists"""
return os.path.exists(os.path.join(cfg.fitpath, cfg.fitname))
"""Return true if the fits file exists"""
return os.path.exists(os.path.join(cfg.fitpath, cfg.fitname))

####
def previousExposure(cfg):
"""return a config for the previous exposure"""
# Parse: sdR-b1-00114186.fit.gz
left = cfg.fitname[:9]
right = cfg.fitname[15:]
exp = str(int(cfg.fitname[9:15]) - 1)
prevcfg = copy.copy(cfg)
prevcfg.fitname = left + exp + right
print("previous cfg:\n" + str(prevcfg))
return prevcfg
"""return a config for the previous exposure"""
# Parse: sdR-b1-00114186.fit.gz
left = cfg.fitname[:9]
right = cfg.fitname[15:]
exp = str(int(cfg.fitname[9:15]) - 1)
prevcfg = copy.copy(cfg)
prevcfg.fitname = left + exp + right
print("previous cfg:\n" + str(prevcfg))
return prevcfg

####
def processFile(cfg, flavor=""):
"""call sos_apocommand on the file. Will exit with error code if the command failts."""
cmd = "sos_apocommand"
cmd += " -f " + cfg.fitname
cmd += " -i " + cfg.fitpath
cmd += " -p " + cfg.plugname
cmd += " -l " + cfg.plugpath
cmd += " -s " + cfg.sosdir
cmd += " -m " + cfg.mjd
if cfg.fps == "T":
cmd += " -e "
if cfg.nocal == "T":
cmd += " -a "
"""call sos_apocommand on the file. Will exit with error code if the command failts."""
cmd = "sos_apocommand"
cmd += " -f " + cfg.fitname
cmd += " -i " + cfg.fitpath
cmd += " -p " + cfg.plugname
cmd += " -l " + cfg.plugpath
cmd += " -s " + cfg.sosdir
cmd += " -m " + cfg.mjd
if cfg.fps == "T":
cmd += " -e "
if cfg.nocal == "T":
cmd += " -a "

prefix = "sos_apocommand(" + flavor + "): "
print("executing: " + cmd)
rv = putils.runCommand(cmd, echo=True, prefix=prefix)
if rv[0] != 0:
print("\nCommand failed with rc = " + str(rv[0]) + "\n")
sys.exit(1)
prefix = "sos_apocommand(" + flavor + "): "
print("executing: " + cmd)
rv = putils.runCommand(cmd, echo=True, prefix=prefix)
if rv[0] != 0:
print("\nCommand failed with rc = " + str(rv[0]) + "\n")
sys.exit(1)
####
def rule1(cfg):
"""Handle arc/flat ordering"""
prevcfg = previousExposure(cfg)
print("Exposure Flavor: " + flavor(cfg))
print("Previous exposure exists: " + str(fitsExist(prevcfg)))
if fitsExist(prevcfg):
print("Previous Flavor: " + flavor(prevcfg))
print("Same Plugging: " + str(samePlugging(cfg, prevcfg)))
# Handle flats -- process, and arc if was previous
if flavor(cfg) == "flat":
print("Exposure is a flat")
processFile(cfg, "flat")
#if fitsExist(prevcfg) and samePlugging(cfg, prevcfg) and flavor(prevcfg) == "arc":
"""Handle arc/flat ordering"""
prevcfg = previousExposure(cfg)
print("Exposure Flavor: " + flavor(cfg))
print("Previous exposure exists: " + str(fitsExist(prevcfg)))
if fitsExist(prevcfg):
print("Previous Flavor: " + flavor(prevcfg))
print("Same Plugging: " + str(samePlugging(cfg, prevcfg)))
# Handle flats -- process, and arc if was previous
if flavor(cfg) == "flat":
print("Exposure is a flat")
processFile(cfg, "flat")
#if fitsExist(prevcfg) and samePlugging(cfg, prevcfg) and flavor(prevcfg) == "arc":
if fitsExist(prevcfg) and flavor(prevcfg) == "arc":
print("Processing previous arc")
processFile(prevcfg, "arc")
return True
# Not one of ours
return False
print("Processing previous arc")
processFile(prevcfg, "arc")
return True
# Not one of ours
return False
####
def rule2(cfg):
"""Don't process arc/flat/science from non-BOSS carts"""
"""Don't process arc/flat/science from non-BOSS carts"""

thisflavor = flavor(cfg)
thiscart = cart(cfg)
thisflavor = flavor(cfg)
thiscart = cart(cfg)
# if thisflavor in ('arc', 'flat', 'science') and \
# (thiscart.casefold() == ('FPS-N).casefold() OR \
# (thiscart.casefold() == ('FPS-S).casefold())
if thisflavor in ('arc', 'flat', 'science') and thiscart != '9':#10: This need change, this number is for testing only
print("Skipping %s exposure for non-BOSS cart %d file %s" % (thisflavor, thiscart, cfg.fitname))
return True
else:
return False
if thisflavor in ('arc', 'flat', 'science') and thiscart != '9':#10: This need change, this number is for testing only
print("Skipping %s exposure for non-BOSS cart %d file %s" % (thisflavor, thiscart, cfg.fitname))
return True
else:
return False
####
def main(args):
"""The program"""
# Initialize
config = parseCmdLine(args)
# Rules return true if they processed the file and processing should stop
# process rules
"""The program"""
# Initialize
config = parseCmdLine(args)
# Rules return true if they processed the file and processing should stop
# process rules
# Rule2 checks for arc/flat/science on non-BOSS cart; check this
# first so that it doesn't trigger arc/flat pre-exp logic in rule 1
# which can call processFile before returning.
print("Checking Rule 2")
if rule2(config):
return
print("Checking Rule 1")
if rule1(config):
return
print("Passed Rules; let's go!")
processFile(config, flavor(config))
# first so that it doesn't trigger arc/flat pre-exp logic in rule 1
# which can call processFile before returning.
print("Checking Rule 2")
if rule2(config):
return
print("Checking Rule 1")
if rule1(config):
return
print("Passed Rules; let's go!")
processFile(config, flavor(config))
#### Start of script

if __name__=='__main__':
main(sys.argv[1:])
main(sys.argv[1:])

0 comments on commit d116557

Please sign in to comment.