Added xspec_timeout argument to inv_abel_fitted_model and inv_abel_data.

Author: HallJoseph

xga/sourcetools/density.py

@@ -392,7 +392,8 @@ def inv_abel_fitted_model(sources: Union[GalaxyCluster, ClusterSample],
                           conv_temp: Union[Quantity, Dict[str, Quantity]] = None, conv_outer_radius: Quantity = "r500",
                           conv_inner_radius: Quantity = Quantity(0, 'arcsec'), inv_abel_method: str = None,
                           num_cores: int = NUM_CORES, show_warn: bool = True, stacked_spectra: bool = False,
-                          telescope: Union[str, List[str]] = None) -> Dict[str, List[Union[GasDensity3D, None]]]:
+                          telescope: Union[str, List[str]] = None,
+                          xspec_timeout: Quantity = Quantity(300, "s")) -> Dict[str, List[Union[GasDensity3D, None]]]:
     """
     A count-rate-map-based galaxy cluster gas density calculation method where a surface brightness profile
     is fit with a model and an inverse abel transform is used to infer the 3D count-rate/volume
@@ -482,6 +483,7 @@ def inv_abel_fitted_model(sources: Union[GalaxyCluster, ClusterSample],
         instead use individual spectra for an ObsID. The default is False.
     :param str/List[str] telescope: Telescope(s) to produce density profiles from. Default is None, in which
         case density profiles will be produced from all telescopes associated with a source.
+    :parma Quantity xspec_timeout: Timeout argument for xspec functions.
     :return: A dictionary of 3D gas density profile lists measured by this function - the keys are telescope
         names. The values are lists with one entry per source, even if the source in question doesn't have
         that telescope associated or the profile construction process failed.
@@ -492,7 +494,7 @@ def inv_abel_fitted_model(sources: Union[GalaxyCluster, ClusterSample],
     sources, conv_factors, obs_id, inst, telescope = _dens_setup(sources, abund_table, lo_en, hi_en, group_spec,
                                                                  min_counts, min_sn, over_sample, obs_id, inst,
                                                                  conv_temp, conv_outer_radius, conv_inner_radius,
-                                                                 num_cores, stacked_spectra, telescope)
+                                                                 num_cores, stacked_spectra, telescope, xspec_timeout)
 
     # Calls the handy spectrum region setup function to make a predictable set of outer radius
     # values
@@ -644,7 +646,8 @@ def inv_abel_data(sources: Union[GalaxyCluster, ClusterSample], outer_radius: Un
                   obs_id: Union[str, list] = None, inst: Union[str, list] = None, conv_temp: Quantity = None,
                   conv_outer_radius: Quantity = "r500", conv_inner_radius: Quantity = Quantity(0, 'arcsec'),
                   num_cores: int = NUM_CORES, stacked_spectra: bool = False,
-                  telescope: Union[str, List[str]] = None) -> Dict[str, List[Union[GasDensity3D, None]]]:
+                  telescope: Union[str, List[str]] = None,
+                  xspec_timeout: Quantity = Quantity(300, "s") ) -> Dict[str, List[Union[GasDensity3D, None]]]:
     """
     A count-rate-map-based galaxy cluster gas density calculation method where a surface brightness profile inverse
     abel transformed, thus inferring the 3D count-rate/volume profile. Then a conversion factor calculated from
@@ -750,6 +753,7 @@ def inv_abel_data(sources: Union[GalaxyCluster, ClusterSample], outer_radius: Un
         instead use individual spectra for an ObsID. The default is False.
     :param str/List[str] telescope: Telescope(s) to produce density profiles from. Default is None, in which
         case density profiles will be produced from all telescopes associated with a source.
+    :parma Quantity xspec_timeout: Timeout argument for xspec functions.
     :return: A dictionary of 3D gas density profile lists measured by this function - the keys are telescope
         names. The values are lists with one entry per source, even if the source in question doesn't have
         that telescope associated or the profile construction process failed.
@@ -760,7 +764,7 @@ def inv_abel_data(sources: Union[GalaxyCluster, ClusterSample], outer_radius: Un
     sources, conv_factors, obs_id, inst, telescope = _dens_setup(sources, abund_table, lo_en, hi_en, group_spec,
                                                                  min_counts, min_sn, over_sample, obs_id, inst,
                                                                  conv_temp, conv_outer_radius, conv_inner_radius,
-                                                                 num_cores, stacked_spectra, telescope)
+                                                                 num_cores, stacked_spectra, telescope, xspec_timeout)
 
     # Calls the handy spectrum region setup function to make a predictable set of outer radius values
     out_rads = region_setup(sources, outer_radius, Quantity(0, 'arcsec'), False, '')[-1]