import sqlite3
import pandas as pd
import logging
import sys
[docs]
def PPReadPointingDatabase(bsdbname, observing_filters, dbquery, surveyname):
"""
Reads in the pointing database as a Pandas dataframe.
Parameters
-----------
bsdbname : string
File location of pointing database.
observing_filters : list of strings
List of observation filters of interest.
dbquery : string
Databse query to perform on pointing database.
surveyname : string
"Name of survey being simulated"
Returns
-----------
dfo : pandas dataframe
Dataframe of pointing database.
"""
pplogger = logging.getLogger(__name__)
con = sqlite3.connect("file:" + bsdbname + "?mode=ro", uri=True)
try:
df = pd.read_sql_query(dbquery, con)
except pd.errors.DatabaseError:
pplogger.error(
"ERROR: PPReadPointingDatabase: SQL query on pointing database failed. Check that the query is correct in the config file."
)
sys.exit(
"ERROR: PPReadPointingDatabase: SQL query on pointing database failed. Check that the query is correct in the config file."
)
except Exception as e: # pragma: no cover
pplogger.error(f"ERROR: PPReadPointingDatabase: error reading from pointing database: {e}")
sys.exit(f"ERROR: PPReadPointingDatabase: error reading from pointing database: {e}")
df["observationId_"] = df["observationId"]
df = df.rename(columns={"observationId": "FieldID"})
df = df.rename(columns={"filter": "optFilter"}) # not to confuse with the pandas filter command
df["optFilter"] = df["optFilter"].astype("category") # save memory
dfo = df[df.optFilter.isin(observing_filters)].copy()
# at the moment the RubinSim pointing databases don't record the observation
# midpoint, so we calculate it. the actual pointings might.
# once we have the actual pointings this check could be changed to, eg,
# lsst_sim for the RubinSim pointings, and 'lsst' would produce different
# behaviour.
if surveyname in ["rubin_sim", "RUBIN_SIM"]:
dfo["observationMidpointMJD_TAI"] = dfo["observationStartMJD_TAI"] + (
(dfo["visitTime"] / 2.0) / 86400.0
)
else:
pplogger.error("ERROR: PPReadPointingDatabase: survey name not recognised.")
sys.exit("ERROR: PPReadPointingDatabase: survey name not recognised.")
return dfo