added 'beast_idx' to obs files for simulations being split

beast/fitting/fit.py

@@ -902,6 +902,10 @@ def IAU_names_and_extra_info(obsdata, surveyname="PHAT", extraInfo=False):
  obsfiltname = obsdata.filter_aliases[filtername]
  r[filtername] = (obsdata.data[obsfiltname] * obsdata.vega_flux[k]).astype(float)
 
+ # if running a simulation, propagate beast_idx numbers to resort to input obs file
+ if "beast_idx" in list(obsdata.data.keys()):
+ r["beast_idx"] = obsdata.data["beast_idx"]
+
  return r
 
 

beast/tools/split_simulated_catalog.py

@@ -92,6 +92,9 @@ def split_simulated_catalog(
  """
  cat = Table.read(catfile)
 
+ # add index column to help sort merged files later
+ cat["beast_idx"] = np.arange(len(cat))
+
  # write out sub-files, if chosen
  if (n_per_file is not None) or (min_n_subfile is not None):