def setBatches(opsdb, colmap, args):
bdict = batches.scienceRadarBatch(colmap=colmap,
runName=args.runName,
extraSql=args.sqlConstraint,
extraMetadata=None,
nside=64,
benchmarkArea=18000,
benchmarkNvisits=825,
DDF=True)
return bdict
parser.add_argument("--db", type=str, default=None)
args = parser.parse_args()
if args.db is None:
if os.path.isfile('trackingDb_sqlite.db'):
os.remove('trackingDb_sqlite.db')
db_files = glob.glob('*.db')
else:
db_files = [args.db]
run_names = [
os.path.basename(name).replace('.db', '') for name in db_files
]
for filename, name in zip(db_files, run_names):
opsdb = db.OpsimDatabaseV4(filename)
colmap = batches.ColMapDict()
if os.path.isdir(name + '_sci'):
shutil.rmtree(name + '_sci')
bdict = batches.scienceRadarBatch(runName=name)
resultsDb = db.ResultsDb(outDir=name + '_sci')
group = mb.MetricBundleGroup(bdict,
opsdb,
outDir=name + '_sci',
resultsDb=resultsDb,
saveEarly=False)
group.runAll(clearMemory=True, plotNow=True)
resultsDb.close()
opsdb.close()
db.addRunToDatabase(name + '_sci', 'trackingDb_sqlite.db', None, name,
'', '', name + '.db')
import glob
import shutil
import lsst.sims.maf.batches as batches
import lsst.sims.maf.db as db
import lsst.sims.maf.metricBundles as mb
if __name__ == "__main__":
"""
Run the science batch on all .db files in a directory.
"""
db_files = glob.glob('*.db')
run_names = [name.replace('.db', '') for name in db_files]
for name in run_names:
opsdb = db.OpsimDatabaseV4(name + '.db')
colmap = batches.ColMapDict('OpsimV4')
if os.path.isdir('sci_' + name):
shutil.rmtree('sci_' + name)
bdict = batches.scienceRadarBatch()
resultsDb = db.ResultsDb(outDir='sci_' + name)
group = mb.MetricBundleGroup(bdict,
opsdb,
outDir='sci_' + name,
resultsDb=resultsDb,
saveEarly=False)
group.runAll(clearMemory=True, plotNow=True)
resultsDb.close()
opsdb.close()
db.addRunToDatabase('sci_' + name, 'trackingDb_sqlite.db', None, name,
'', '', name + '.db')
"""
parser = argparse.ArgumentParser()
parser.add_argument("filename", type=str, help="OpSim database")
args = parser.parse_args()
filename = args.filename
name = filename.replace('.db', '')
outDir = 'radar_' + name
if os.path.isdir(name):
shutil.rmtree(name)
opsdb = db.OpsimDatabaseV4(name + '.db')
colmap = batches.ColMapDict('OpsimV4')
bdict = {}
bdict.update(batches.scienceRadarBatch(colmap, name))
resultsDb = db.ResultsDb(outDir=outDir)
group = mb.MetricBundleGroup(bdict,
opsdb,
outDir=outDir,
resultsDb=resultsDb,
saveEarly=False)
group.runAll(clearMemory=True, plotNow=True)
resultsDb.close()
opsdb.close()
db.addRunToDatabase(outDir, 'trackingDb_sqlite.db', None, name, '', '',
name + '.db')