This Notebook describes how the logic in mtl.py
(the Merged Target List) uses priorities and numbers of observations set by the targeting bitmasks to determine the observational "state" of a target.
If you identify any errors or have requests for additional functionality please create a new issue at https://github.com/desihub/desitarget/issues or send a note to desi-data@desi.lbl.gov.
Last updated May 2020 using DESI software release:
In [ ]:
release="19.12"
The easiest way to get started is to use the jupyter server at NERSC so that you don't need to install any code or download any data locally.
If you need a NERSC account, see https://desi.lbl.gov/trac/wiki/Computing/AccessNersc
Then do the one-time jupyter configuration described at https://desi.lbl.gov/trac/wiki/Computing/JupyterAtNERSC
From a NERSC command line, checkout a copy of the tutorial code, e.g. from cori.nersc.gov
mkdir -p $HOME/desi/
cd $HOME/desi/
git clone https://github.com/desihub/tutorials
And then go to https://jupyter.nersc.gov, login, navigate to where you checked out this package (e.g. $HOME/desi/tutorials
), and double-click on Intro_to_mtl.ipynb
.
This tutorial has been tested using the:
In [ ]:
"DESI {}".format(release)
kernel installed at NERSC. To get an equivalent environment from a cori command line use the command printed out below:
In [ ]:
print('source /global/common/software/desi/desi_environment.sh {}'.format(release))
In [ ]:
import os
import numpy as np
from astropy.table import Table
import matplotlib.pyplot as plt
%pylab inline
# ADM import the mtl code and a function used to initialize target states.
from desitarget import mtl
from desitarget.targets import initial_priority_numobs
# ADM import the masks that define observing layers and observing states.
from desitarget.targetmask import obsconditions, obsmask
# ADM import the Main Survey targeting bit mask.
from desitarget.targetmask import desi_mask
Note that we will focus on the Main Survey targeting bit mask for this tutorial. If you instead wanted to study the behavior of MTL for the SV bit-mask, you would use:
from desitarget.sv1.sv1_targetmask import desi_mask
If you are running locally and any of these fail,
you should go back through the installation instructions and/or email desi-data@desi.lbl.gov
if you get stuck.
If you are running from jupyter.nersc.gov and have problems, double check that your kernel is as printed below:
In [ ]:
print("DESI {}".format(release))
This tutorial will focus on aspects of the targeting bitmasks and coding logic that control the observational state of a target. To understand the more basic aspects of the targeting bitmasks (how we know what is a QSO, or ELG, target, etc.) you might want to try the bits and bitmasks tutorial.
How a target is observed (how it is handled by the desitarget fiberassign
code) depends on three main targeting concepts:
Initially, all of these quantities are set by static information in the desitarget bitmask yaml file but they can subsequently be altered by the desitarget mtl code depending on each target's observational state.
When a target is initally flagged for DESI observations, the information in the desitarget bitmask yaml file is used to set an initial observational layer, priority and number of observations. Consider the following example for a quasar target:
In [ ]:
bitname="QSO"
print('Bit value associated with a DESI target with bit name "{}": {}'.format(bitname, desi_mask.QSO))
print('Conditions (layers) in which a {} is allowed to be observed: {}'.format(bitname, desi_mask.QSO.obsconditions))
print('Initial priorities set for a {}: {}'.format(bitname, desi_mask.QSO.priorities))
print('Initial number of observations set for a {}: {}'.format(bitname, desi_mask.QSO.numobs))
Or for an ELG or LRG target:
In [ ]:
for bitname in ["ELG", "LRG"]:
print('Bit value associated with a DESI target with bit name "{}": {}'.format(bitname, desi_mask[bitname]))
print('Conditions (layers) in which a {} is allowed to be observed: {}'.format(bitname, desi_mask[bitname].obsconditions))
print('Initial priorities set for an {}: {}'.format(bitname, desi_mask[bitname].priorities))
print('Initial number of observations set for an {}: {}'.format(bitname, desi_mask[bitname].numobs))
These quantities define the "state-of-play" for a given target before any DESI observations have occurred. So, for example, if an ELG is being observed for the first time, we will request 1 observation of that target at a priority of 3000, if a QSO is being observed for the first time, we will request 4 observations of such a target at a priority of 3400. The actual relative values of the priorites are unimportant, but the target with the highest priority will always be assigned a fiber first.
The values in the priorities
dictionary merit some further explanation. As we will see later in this notebook, the priority of a target can change depending on its observational state. The keys in the priorities
dictionary define what priority to set for a target that has transitioned to that observational state.
So, for example, an LRG will start with a priority of 3200
(UNOBS
) but will transition to a priority of 2
(DONE
) as soon as we have one observation of it. For target classes that request more than one observation, such as Lyman-alpha QSOs, the behavior can be more complex. If we have observed the QSO and request more observations, and the redshift is flagged as problematic, the QSO target will be assigned a priority of 3400 (MORE_ZWARN
). If we have observed the QSO and request more observations, and the redshift is flagged as good, the QSO target will be assigned a priority of 3500 (MORE_ZGOOD
). The full suite of allowed observational states for a DESI target can be retrieved using the obsmask
bitmask (which we imported earlier):
In [ ]:
print(obsmask)
Similarly, the full suite of allowed observational layers (dark-time, bright-time, etc.) can be retrieved using the obsconditions
bitmask:
In [ ]:
print(obsconditions)
No target is an island, and it is certainly possible for a DESI target to satisfy two sets of target selection criteria and be assigned a bit value consistent with two targets. Again, you might want to try the bits and bitmasks tutorial. for more insight into bit values. For example, though, consider the first 8 possible bit values and what they signify for the various DESI target classes:
In [ ]:
for i in range(8):
print('{}: {}'.format(i, desi_mask.names(i)))
Bit value (2 + 4 =) 6, for instance, which corresponds to 21 ("ELG") + 22 ("QSO"), denotes a target that satisfies the selection criteria for two different target classes. How do we set priorities and numbers of observations in such a case?
Clearly, these targets need to have their priorities "merged" in some sense. If you've been paying attention, you've probably worked out that this is part of what the desitarget mtl (Merged Target List; henceforth MTL) code achieves.
Let's set up a specific example for a target that is both an ELG and QSO, and see how MTL processes such a target.
In [ ]:
targets = Table()
ELGbit, QSObit = desi_mask["ELG"], desi_mask["QSO"]
targets["DESI_TARGET"] = np.array([ELGbit, QSObit, QSObit | ELGbit])
print(targets)
bitnames = []
for dt in targets["DESI_TARGET"]:
# ADM we'll store these bit names for later use, too!
bitnames.append(desi_mask.names(dt))
print(dt, desi_mask.names(dt))
So, now we have a simple set of targets defined. We'll also need to add some standard columns, as these are expected by the MTL code:
In [ ]:
n = len(targets)
targets['BITNAMES'] = bitnames
targets['BGS_TARGET'] = np.zeros(n, dtype=np.int64)
targets['MWS_TARGET'] = np.zeros(n, dtype=np.int64)
targets['TARGETID'] = np.arange(n)
targets["PRIORITY_INIT"] = 0
targets["NUMOBS_INIT"] = 0
Let's see what happened to the priorities and observing conditions (layers) for these targets after they were merged by calling the MTL code:
In [ ]:
mtltargets = mtl.make_mtl(targets, obscon="DARK|GRAY")
# ADM make the observing conditions more human-readable:
obscon = []
for oc in mtltargets["OBSCONDITIONS"]:
obscon.append(obsconditions.names(oc))
mtltargets["LAYERS"] = np.array(obscon)
print(mtltargets["BITNAMES", "PRIORITY", "OBSCONDITIONS", "LAYERS"])
Some important things to note:
obscon="DARK|GRAY"
(for the dark-time survey) or obscon="BRIGHT"
(for the bright-time survey).obscon="SOME_SPECIAL_LAYER"
).Note that this example was purely to show you why a mechanism for merging targets is critical. In reality, desitarget
sets initial priorities, observing conditions, and numbers of observations in advance of running MTL. This makes the initial values of these parameters more traceable (as they then appear as column names in the desitarget
initial targeting files). Critically, this step has to be done to correctly initialize the numbers of observations (NUMOBS_INIT
) for DESI targets. So a more complete example is:
In [ ]:
targets = Table()
targets["DESI_TARGET"] = np.array([ELGbit, QSObit, QSObit | ELGbit])
targets['BITNAMES'] = bitnames
n = len(targets)
targets['BGS_TARGET'] = np.zeros(n, dtype=np.int64)
targets['MWS_TARGET'] = np.zeros(n, dtype=np.int64)
targets['TARGETID'] = np.arange(n)
# ADM use function outside of MTL to more transparental initialize priorities and numobs.
pinit, ninit = initial_priority_numobs(targets)
targets["PRIORITY_INIT"] = pinit
targets["NUMOBS_INIT"] = ninit
print(targets["BITNAMES", "DESI_TARGET", "PRIORITY_INIT", "NUMOBS_INIT"])
Note, for instance, that the merged "ELG/QSO" target requires 4 observations (as it is potentially a Lyman-alpha QSO target).
In "official" DESI targeting files, e.g. as stored in the following NERSC directories:
/global/cfs/projectdirs/desi/target/catalogs/dr8
/global/cfs/projectdirs/desi/target/catalogs/dr9
PRIORITY_INIT
and NUMOBS_INIT
have already been set in this manner.
With reasonable initial values of PRIORITY_INIT
and NUMOBS_INIT
, MTL will pass through the number of additional observations required for each target (NUMOBS_MORE
). For example:
In [ ]:
mtltargets = mtl.make_mtl(targets, obscon="DARK|GRAY")
print(mtltargets["DESI_TARGET", "PRIORITY_INIT", "NUMOBS_INIT", "NUMOBS_MORE"])
The MTL code also contains logic to update the priorities of, and number of observations for, a target based on each target's current observational state. Most of this logic is contained in the desitarget.targets module.
As the DESI survey progresses, classifications and redshifts of each target will be included in a redshift catalog (henceforth a zcat) passed back to MTL from the DESI spectroscopic pipeline. Passing this zcat as an input to MTL, changes the observational state of each target, updating the number of required additional observations and transitioning between the priorites described earlier in this tutorial ("UNOBS"
, "DONE"
, etc.). Let's look at an example. First, let's construct a set of targets in a manner similar to what we did in the previous section of this tutorial:
In [ ]:
targets = Table()
# ADM we have 7 targets, two ELGs, an LRG, and four quasars.
classes = np.array(['ELG', 'ELG', 'LRG', 'QSO', 'QSO', 'QSO', 'QSO'])
n = len(classes)
# ADM pull the appropriate bit value for each target type from the desi_mask.
targets['DESI_TARGET'] = [desi_mask[c].mask for c in classes]
# ADM the BGS and MWS target bits need to be set, but we'll ignore them (set them to zero) for this tutorial.
targets['BGS_TARGET'] = np.zeros(n, dtype=np.int64)
targets['MWS_TARGET'] = np.zeros(n, dtype=np.int64)
# ADM this needs to be a unique TARGETID. For this tutorial, we'll just use the integers 0-6.
targets['TARGETID'] = np.arange(n)
# ADM determine the initial PRIORITY and NUMOBS for the input target classes.
pinit, ninit = initial_priority_numobs(targets)
targets["PRIORITY_INIT"] = pinit
targets["NUMOBS_INIT"] = ninit
Now, let's also construct a zcat:
In [ ]:
zcat = Table()
# ADM MTL matches the targets and the zcat on TARGETID.
# ADM but let's just assume everything matches row-by-row.
zcat['TARGETID'] = targets['TARGETID']
# ADM the spectroscopic pipeline assigned the following redshifts...
zcat['Z'] = [0.0, 1.2, 0.9, 2.16, 2.7, 2.14, 1.4]
# ADM ...and the following classifications.
zcat['SPECTYPE'] = ['STAR', 'GALAXY', 'GALAXY', 'QSO', 'QSO', 'QSO', 'QSO']
# ADM three of the classifications/redshifts were dubious (ZWARN=4).
zcat['ZWARN'] = [4, 0, 0, 0, 4, 0, 4]
# ADM each of our targets has one spectroscopic observation.
zcat['NUMOBS'] = [1, 1, 1, 1, 1, 1, 1]
So, here's the initial list of target properties (a static set of assignations):
In [ ]:
print(targets)
and here's the spectroscopic information we gleaned from observing these targets once in DESI:
In [ ]:
print(zcat)
Let's see what MTL makes of all of this:
In [ ]:
mtltargets = mtl.make_mtl(targets, obscon="DARK|GRAY", zcat=zcat)
print(mtltargets['DESI_TARGET', 'TARGETID', 'PRIORITY_INIT', 'NUMOBS_INIT', 'PRIORITY', 'NUMOBS_MORE'])
To summarize the output:
"DONE"
.NUMOBS_MORE
set to 0
and its priority set to the equivalent of "DONE"
."MORE_ZGOOD"
and NUMOBS_MORE
is set to 3
.NUMOBS_MORE
set to 3
but has retained its initial priority of 3400
. By extension of the last example, we can test how MTL affects every target class individually, remembering the general precepts that for merged targets (e.g. a target that is both an ELG and a QSO) the highest PRIORITY
and NUMOBS
for the individual classes will characterize the behavior, and all observing conditions (layers) will be merged across classes. Here goes:
In [ ]:
for kind in ["QSO", "LRG", "ELG"]:
targets = Table()
# ADM 4 targets of this kind
classes = np.array([kind, kind, kind, kind])
n = len(classes)
# ADM pull the appropriate bit value for each target type from the desi_mask.
targets['DESI_TARGET'] = [desi_mask[c].mask for c in classes]
# ADM the BGS and MWS target bits need to be set, but we'll ignore them (set them to zero) for this tutorial.
targets['BGS_TARGET'] = np.zeros(n, dtype=np.int64)
targets['MWS_TARGET'] = np.zeros(n, dtype=np.int64)
# ADM this needs to be a unique TARGETID. For this tutorial, we'll just use the integers 0-4.
targets['TARGETID'] = np.arange(n)
# ADM determine the initial PRIORITY and NUMOBS for the input target classes.
pinit, ninit = initial_priority_numobs(targets)
targets["PRIORITY_INIT"] = pinit
targets["NUMOBS_INIT"] = ninit
zcat = Table()
# ADM MTL matches the targets and the zcat on TARGETID.
# ADM but let's just assume everything matches row-by-row.
zcat['TARGETID'] = targets['TARGETID']
# ADM pick two redshifts above the Lyman-Alpha cutoff and two below.
zcat['Z'] = [2.5, 2.7, 1.5, 1.2]
# ADM MTL doesn't care about classifications, so everything can be a GALAXY.
zcat['SPECTYPE'] = ['GALAXY', 'GALAXY', 'GALAXY', 'GALAXY']
# ADM flag warnings in one Lyman-alpha QSO and one tracer.
zcat['ZWARN'] = [4, 0, 4, 0]
# ADM each of our targets has one spectroscopic observation.
zcat['NUMOBS'] = [1, 1, 1, 1]
print("...{}...".format(kind))
print(zcat)
mtltargets = mtl.make_mtl(targets, obscon="DARK|GRAY", zcat=zcat)
print(mtltargets['DESI_TARGET', 'TARGETID', 'PRIORITY_INIT', 'NUMOBS_INIT', 'PRIORITY', 'NUMOBS_MORE'])