Commit fe70e9d8 by Marcus Wirtz

### [simulations/healpytools] Fix pylint issues

parent 72da1380
 ... @@ -77,7 +77,7 @@ def rand_exposure_vec_in_pix(nside, ipix, a0=-35.25, zmax=60, coord_system='gal' ... @@ -77,7 +77,7 @@ def rand_exposure_vec_in_pix(nside, ipix, a0=-35.25, zmax=60, coord_system='gal' """ """ ipix = np.atleast_1d(ipix) ipix = np.atleast_1d(ipix) vecs = np.zeros((3, ipix.size)) vecs = np.zeros((3, ipix.size)) mask = check_problematic_exposure_pixel(nside, ipix, a0, zmax, deviation) mask = check_problematic_pixel(nside, ipix, a0, zmax, deviation) vecs[:, ~mask] = rand_vec_in_pix(nside, ipix[~mask], nest) vecs[:, ~mask] = rand_vec_in_pix(nside, ipix[~mask], nest) if not nest: if not nest: ipix = hp.ring2nest(nside, ipix=ipix) ipix = hp.ring2nest(nside, ipix=ipix) ... @@ -97,7 +97,7 @@ def rand_exposure_vec_in_pix(nside, ipix, a0=-35.25, zmax=60, coord_system='gal' ... @@ -97,7 +97,7 @@ def rand_exposure_vec_in_pix(nside, ipix, a0=-35.25, zmax=60, coord_system='gal' return np.array(vecs) return np.array(vecs) def check_problematic_exposure_pixel(nside, ipix, a0, zmax, deviation=0.5, coord_system='gal'): def check_problematic_pixel(nside, ipix, a0, zmax, deviation=0.5, coord_system='gal'): """ """ Checks input pixel for exposure deviation within the corner points from more than certain Checks input pixel for exposure deviation within the corner points from more than certain threshold (default: 0.5). threshold (default: 0.5). ... ...
 ... @@ -368,7 +368,8 @@ class ObservedBound(BaseSimulation): ... @@ -368,7 +368,8 @@ class ObservedBound(BaseSimulation): elif (self.cr_map is None): elif (self.cr_map is None): self._fix_point_source() self._fix_point_source() if self.cr_map.size == self.npix: if self.cr_map.size == self.npix: pixel[:, signal_label] = np.random.choice(self.npix, (self.nsets, n_sig), p=np.hstack(self.cr_map)/np.sum(np.hstack(self.cr_map))) pixel[:, signal_label] = np.random.choice(self.npix, (self.nsets, n_sig), p=np.hstack(self.cr_map)/np.sum(np.hstack(self.cr_map))) else: else: for i, rig in enumerate(self.rig_bins): for i, rig in enumerate(self.rig_bins): mask_rig = (rig == self.rigidities) * signal_label # type: np.ndarray mask_rig = (rig == self.rigidities) * signal_label # type: np.ndarray ... @@ -674,9 +675,9 @@ class SourceBound(BaseSimulation): ... @@ -674,9 +675,9 @@ class SourceBound(BaseSimulation): # Checks if for all sets there is at least one explicitly simulated source WITHOUT cosmic-ray contribution # Checks if for all sets there is at least one explicitly simulated source WITHOUT cosmic-ray contribution range_src = np.arange(self.universe.n_src) range_src = np.arange(self.universe.n_src) enough_sources = np.apply_along_axis(lambda x: np.in1d(range_src, x, invert=True).any(), axis=1, arr=source_labels).all() check = np.apply_along_axis(lambda x: np.in1d(range_src, x, invert=True).any(), axis=1, arr=source_labels).all() if (not enough_sources): if (not check): print("Warning: you might have not enough sources simulated. Set keyword 'n_src' (currently %s) higher?" % self.universe.n_src) print("Warning: not enough sources. Set keyword 'n_src' (currently %s) higher?" % self.universe.n_src) self.crs['source_labels'] = source_labels self.crs['source_labels'] = source_labels occ = np.apply_along_axis(lambda x: np.bincount(x+1)[x+1], axis=1, arr=source_labels) occ = np.apply_along_axis(lambda x: np.bincount(x+1)[x+1], axis=1, arr=source_labels) ... @@ -829,22 +830,23 @@ class SourceBound(BaseSimulation): ... @@ -829,22 +830,23 @@ class SourceBound(BaseSimulation): from scipy.stats import binned_statistic from scipy.stats import binned_statistic if opath is None: if opath is None: opath = '/tmp/composition%s__emin_%s__ecut_%s.pdf' % (self._get_charge_id(), self.energy_setting['log10e_min'], e_id = 'emin_%s__ecut_%s' % (self.energy_setting['log10e_min'], self.energy_setting['log10_cut']) self.energy_setting['log10_cut']) opath = '/tmp/composition%s__%s.pdf' % (self._get_charge_id(), e_id) crs = self.crs crs = self.crs idx = self._select_representative_set() if idx is None else idx idx = self._select_representative_set() if idx is None else idx bins = np.array(np.concatenate((np.arange(18.5, 20.1, 0.1), np.array([20.5])))) bins = np.array(np.concatenate((np.arange(18.5, 20.1, 0.1), np.array([20.5])))) mean_xmax, _, _ = binned_statistic(crs['log10e'][idx, :], values=crs['xmax'][idx, :], statistic='mean', bins=bins) mean_xmax = binned_statistic(crs['log10e'][idx, :], values=crs['xmax'][idx, :], statistic='mean', bins=bins)[0] std_xmax, _, _ = binned_statistic(crs['log10e'][idx, :], values=crs['xmax'][idx, :], statistic='std', bins=bins) std_xmax = binned_statistic(crs['log10e'][idx, :], values=crs['xmax'][idx, :], statistic='std', bins=bins)[0] n_xmax, _, _ = binned_statistic(crs['log10e'][idx, :], values=crs['xmax'][idx, :], statistic='count', bins=bins) n_xmax = binned_statistic(crs['log10e'][idx, :], values=crs['xmax'][idx, :], statistic='count', bins=bins)[0] mids, _, _ = binned_statistic(crs['log10e'][idx, :], values=crs['log10e'][idx, :], statistic='mean', bins=bins) mids = binned_statistic(crs['log10e'][idx, :], values=crs['log10e'][idx, :], statistic='mean', bins=bins)[0] fig = plt.figure(figsize=(6, 4)) fig = plt.figure(figsize=(6, 4)) ax = fig.add_subplot(111) ax = fig.add_subplot(111) auger.plot_mean_xmax(ax=ax, models=['EPOS-LHC', 'QGSJetII-04']) auger.plot_mean_xmax(ax=ax, models=['EPOS-LHC', 'QGSJetII-04']) ax.errorbar(mids, mean_xmax, yerr=std_xmax/np.sqrt(n_xmax), marker='o', color='darkgray', label='simulation', fmt='') xmax_err = std_xmax/np.sqrt(n_xmax) ax.errorbar(mids, mean_xmax, yerr=xmax_err, marker='o', color='darkgray', label='simulation', fmt='') ax.set_xlim(right=20.6) ax.set_xlim(right=20.6) fig.savefig(opath, bbox_inches='tight') fig.savefig(opath, bbox_inches='tight') ... @@ -907,7 +909,8 @@ class SourceBound(BaseSimulation): ... @@ -907,7 +909,8 @@ class SourceBound(BaseSimulation): if emin is not None: if emin is not None: mask = self.crs['log10e'][idx, :] > np.log10(emin)+18. mask = self.crs['log10e'][idx, :] > np.log10(emin)+18. skymap.eventmap(self.crs['vecs'][:, idx, labels_p >= 0][:, mask[labels_p >= 0]], c=charges[labels_p >= 0][mask[labels_p >= 0]], skymap.eventmap(self.crs['vecs'][:, idx, labels_p >= 0][:, mask[labels_p >= 0]], c=charges[labels_p >= 0][mask[labels_p >= 0]], cmap=cmap, cblabel='\$Z\$', cmap=cmap, cblabel='\$Z\$', cticks=np.array([1, 2, 6, 12, 20, 26]), vmin=0.5, vmax=26.5, cticks=np.array([1, 2, 6, 12, 20, 26]), vmin=0.5, vmax=26.5, s=25, alpha=0.6, marker='v') s=25, alpha=0.6, marker='v') ... @@ -928,7 +931,7 @@ class SourceBound(BaseSimulation): ... @@ -928,7 +931,7 @@ class SourceBound(BaseSimulation): plt.close() plt.close() def plot_distance(self, sets='all', opath=None, emin=None, sig='both'): # pragma: no cover def plot_distance(self, sets='all', opath=None, emin=None, sig='both'): # pragma: no cover # pylint: disable=too-many-statements # pylint: disable=too-many-statements,invalid-unary-operand-type """ Plot histogram of travel distances (either mean of all sets or one specific) """ """ Plot histogram of travel distances (either mean of all sets or one specific) """ import matplotlib.pyplot as plt import matplotlib.pyplot as plt ... @@ -1046,6 +1049,7 @@ class SourceGeometry: ... @@ -1046,6 +1049,7 @@ class SourceGeometry: self.sources = None self.sources = None self.source_fluxes = None self.source_fluxes = None self.distances = None self.distances = None self.background_horizon = None def set_sources(self, source_density, fluxes, n_src, background_horizon=1.): def set_sources(self, source_density, fluxes, n_src, background_horizon=1.): """ """ ... ...
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!