hawc_hal.response.response_bin.ResponseBin._get_en_th1d() - Code Metrics - Inspection of "Fixing energy fitting" - giacomov/hawc_hal - Measure and Improve Code Quality continuously with Scrutinizer

Passed

Push — master ( 573345...9597ff )

by Giacomo

created 2018-07-06 21:54 UTC

ResponseBin._get_en_th1d() A

↳ Parent: hawc_hal.response.response_bin

Complexity

Conditions

Size

Total Lines	23
Code Lines	12

Duplication

Lines	0
Ratio	0 %

Importance

Changes

Metric	Value
eloc	12
dl	0
loc	23
rs	9.8
c	0
b	0
f	0
cc	3
nop	6

from ..psf_fast import PSFWrapper, InvalidPSF
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
import numpy as np

import pandas as pd


from threeML.exceptions.custom_exceptions import custom_warnings



class ResponseBin(object):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""

    def __init__(self, name, min_dec, max_dec, dec_center,

                 sim_n_sig_events, sim_n_bg_events,
                 sim_energy_bin_low, sim_energy_bin_centers, sim_energy_bin_hi,
                 sim_differential_photon_fluxes, sim_signal_events_per_bin,
                 psf):

        self._name = name
        self._min_dec = min_dec
        self._max_dec = max_dec
        self._dec_center = dec_center
        self._sim_n_sig_events = sim_n_sig_events
        self._sim_n_bg_events = sim_n_bg_events
        self._sim_energy_bin_low = sim_energy_bin_low
        self._sim_energy_bin_centers = sim_energy_bin_centers
        self._sim_energy_bin_hi = sim_energy_bin_hi
        self._sim_differential_photon_fluxes = sim_differential_photon_fluxes
        self._sim_signal_events_per_bin = sim_signal_events_per_bin
        self._psf = psf  # type: PSFWrapper

    @staticmethod
    def _get_en_th1d(open_ttree, dec_id, dec_id_label, analysis_bin_id, analysis_bin_id_label, prefix):


        nh_name = "nh%i" % analysis_bin_id

        en_sig_label = "En%s_dec%i_%s" % (prefix, dec_id, nh_name)

        this_en_th1d = open_ttree.Get("%s/%s/%s" % (dec_id_label, analysis_bin_id_label, en_sig_label))


        if not this_en_th1d:

            # Some old responses have a slightly different label. Try with that
            nh_name = "nh%02i" % analysis_bin_id

            en_sig_label = "En%s_dec%i_%s" % (prefix, dec_id, nh_name)

            this_en_th1d = open_ttree.Get("%s/%s/%s" % (dec_id_label, analysis_bin_id_label, en_sig_label))


            if not this_en_th1d:

                raise IOError("Could not find TH1D named %s" % this_en_th1d)

        return this_en_th1d, nh_name

    @classmethod
    def from_ttree(cls, open_ttree, dec_id, analysis_bin_id, log_log_spectrum, min_dec, dec_center, max_dec):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""

        from ..root_handler import ROOT

        # Compute the labels as used in the response file
        dec_id_label = "dec_%02i" % dec_id
        analysis_bin_id_label = "nh_%02i" % analysis_bin_id

        # Read the histogram of the simulated events detected in this bin_name
        # NOTE: we do not copy this TH1D instance because we won't use it after the
        # file is closed

        this_en_sig_th1d, nh_name = cls._get_en_th1d(open_ttree, dec_id, dec_id_label, analysis_bin_id, analysis_bin_id_label,

                                                     'Sig')

        # The sum of the histogram is the total number of simulated events detected
        # in this analysis bin_name
        sim_n_sig_events = this_en_sig_th1d.Integral()

        # Now let's see what has been simulated, i.e., the differential flux
        # at the center of each bin_name of the en_sig histogram
        sim_energy_bin_low = np.zeros(this_en_sig_th1d.GetNbinsX())
        sim_energy_bin_centers = np.zeros(this_en_sig_th1d.GetNbinsX())
        sim_energy_bin_hi = np.zeros(this_en_sig_th1d.GetNbinsX())
        sim_signal_events_per_bin = np.zeros_like(sim_energy_bin_centers)
        sim_differential_photon_fluxes = np.zeros_like(sim_energy_bin_centers)

        for i in range(sim_energy_bin_centers.shape[0]):
            # Remember: bin_name 0 is the underflow bin_name, that is why there
            # is a "i+1" and not just "i"
            bin_lo = this_en_sig_th1d.GetBinLowEdge(i+1)
            bin_center = this_en_sig_th1d.GetBinCenter(i + 1)
            bin_hi = this_en_sig_th1d.GetBinWidth(i+1) + bin_lo

            # Store the center of the logarithmic bin_name
            sim_energy_bin_low[i] = 10 ** bin_lo  # TeV
            sim_energy_bin_centers[i] = 10 ** bin_center  # TeV
            sim_energy_bin_hi[i] = 10 ** bin_hi  # TeV

            # Get from the simulated spectrum the value of the differential flux
            # at the center energy
            sim_differential_photon_fluxes[i] = 10 ** log_log_spectrum.Eval(bin_center)  # TeV^-1 cm^-1 s^-1


            # Get from the histogram the detected events in each log-energy bin_name
            sim_signal_events_per_bin[i] = this_en_sig_th1d.GetBinContent(i + 1)

        # Read the histogram of the bkg events detected in this bin_name
        # NOTE: we do not copy this TH1D instance because we won't use it after the
        # file is closed

        this_en_bg_th1d, _ = cls._get_en_th1d(open_ttree, dec_id, dec_id_label, analysis_bin_id, analysis_bin_id_label,

                                              'Bg')

        # The sum of the histogram is the total number of simulated events detected
        # in this analysis bin_name
        sim_n_bg_events = this_en_bg_th1d.Integral()

        # Now read the various TF1(s) for PSF, signal and background

        # Read the PSF and make a copy (so it will stay when we close the file)

        psf_label_tf1 = "PSF_dec%i_%s_fit" % (dec_id, nh_name)

        psf_path = "%s/%s/%s" % (dec_id_label, analysis_bin_id_label, psf_label_tf1)

        tf1 = ROOT.TF1()

        open_ttree.GetObject(psf_path, tf1)

        psf_fun = PSFWrapper.from_TF1(tf1)

        return cls(analysis_bin_id, min_dec, max_dec, dec_center, sim_n_sig_events, sim_n_bg_events,
                   sim_energy_bin_low,
                   sim_energy_bin_centers,
                   sim_energy_bin_hi,
                   sim_differential_photon_fluxes, sim_signal_events_per_bin,
                   psf_fun)

    def to_pandas(self):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""

        # In the metadata let's save all single values (floats)
        meta = {'min_dec': self._min_dec,
                'max_dec': self._max_dec,
                'declination_center': self._dec_center,
                'n_sim_signal_events': self._sim_n_sig_events,
                'n_sim_bkg_events': self._sim_n_bg_events
                }


        # Now make a dataframe containing the elements of the simulation
        items = (
            ('sim_energy_bin_low', pd.Series(self.sim_energy_bin_low)),
            ('sim_energy_bin_centers', pd.Series(self.sim_energy_bin_centers)),
            ('sim_energy_bin_hi', pd.Series(self.sim_energy_bin_hi)),
            ('sim_differential_photon_fluxes', pd.Series(self.sim_differential_photon_fluxes)),
            ('sim_signal_events_per_bin', pd.Series(self.sim_signal_events_per_bin))
        )

        df = pd.DataFrame.from_dict(dict(items))


        return df, meta, self.psf.to_pandas()

    def combine_with_weights(self, other_response_bin, dec_center, w1, w2):

        """
        Produce another response bin which is the weighted sum of this one and the other one passed.

        :param other_response_bin:
        :param w1:
        :param w2:
        :return:
        """

        assert np.isclose(w1+w2, 1.0), "Weights are not properly normalized"

        new_name = "interpolated_%s" % self._name

        # Use np.nan as declination boundaries to indicate that this is actually interpolated
        min_dec, max_dec = np.nan, np.nan

        n_sim_signal_events = w1 * self._sim_n_sig_events + w2 * other_response_bin._sim_n_sig_events

        n_sim_bkg_events = w1 * self._sim_n_bg_events + w2 * other_response_bin._sim_n_bg_events

        # We assume that the bin centers are the same
        assert np.allclose(self._sim_energy_bin_centers, other_response_bin._sim_energy_bin_centers)
class MyParent:
    def __init__(self):
        self._x = 1;
        self.y = 2;

class MyChild(MyParent):
    def some_method(self):
        return self._x    # Ok, since accessed from a child class

class AnotherClass:
    def some_method(self, instance_of_my_child):
        return instance_of_my_child._x   # Would be flagged as AnotherClass is not
                                         # a child class of MyParent

        sim_differential_photon_fluxes = w1 * self._sim_differential_photon_fluxes + \
                                         w2 * other_response_bin._sim_differential_photon_fluxes

        sim_signal_events_per_bin = w1 * self._sim_signal_events_per_bin + \
                                    w2 * other_response_bin._sim_signal_events_per_bin

        # Now interpolate the psf
        new_psf = self._psf.combine_with_other_psf(other_response_bin._psf, w1, w2)
class MyParent:
    def __init__(self):
        self._x = 1;
        self.y = 2;

class MyChild(MyParent):
    def some_method(self):
        return self._x    # Ok, since accessed from a child class

class AnotherClass:
    def some_method(self, instance_of_my_child):
        return instance_of_my_child._x   # Would be flagged as AnotherClass is not
                                         # a child class of MyParent

        new_response_bin = ResponseBin(new_name, min_dec, max_dec, dec_center,
                                       n_sim_signal_events, n_sim_bkg_events,
                                       self._sim_energy_bin_low,
                                       self._sim_energy_bin_centers,
                                       self._sim_energy_bin_hi,
                                       sim_differential_photon_fluxes,
                                       sim_signal_events_per_bin,
                                       new_psf)

        return new_response_bin

    @property
    def name(self):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        return self._name

    @property
    def declination_boundaries(self):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        return (self._min_dec, self._max_dec)

    @property
    def declination_center(self):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        return self._dec_center

    @property
    def psf(self):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        return self._psf

    @property
    def n_sim_signal_events(self):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        return self._sim_n_sig_events

    @property
    def n_sim_bkg_events(self):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        return self._sim_n_bg_events

    @property
    def sim_energy_bin_low(self):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        return self._sim_energy_bin_low

    @property
    def sim_energy_bin_centers(self):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        return self._sim_energy_bin_centers

    @property
    def sim_energy_bin_hi(self):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        return self._sim_energy_bin_hi

    @property
    def sim_differential_photon_fluxes(self):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        return self._sim_differential_photon_fluxes

    @property
    def sim_signal_events_per_bin(self):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        return self._sim_signal_events_per_bin


1			from ..psf_fast import PSFWrapper, InvalidPSF
			0 ignored issues – show Coding Style introduced 2018-07-06 22:00 UTC by Report Bug Copy Issue Report This module should have a docstring. The coding style of this project requires that you add a docstring to this code element. Below, you find an example for methods: class SomeClass: def some_method(self): """Do x and return foo.""" If you would like to know more about docstrings, we recommend to read PEP-257: Docstring Conventions. Loading history... Unused Code introduced 2018-07-06 22:00 UTC by Report Bug Copy Issue Report Unused InvalidPSF imported from psf_fast Loading history...
2			import numpy as np
			0 ignored issues – show introduced 2018-06-02 00:48 UTC by Report Bug Copy Issue Report Unable to import 'numpy' Loading history... introduced 2018-07-06 22:00 UTC by Report Bug Copy Issue Report external import "import numpy as np" should be placed before "from ..psf_fast import PSFWrapper, InvalidPSF" Loading history...
3			import pandas as pd
			0 ignored issues – show introduced 2018-07-06 22:00 UTC by Report Bug Copy Issue Report Unable to import 'pandas' Loading history... introduced 2018-07-06 22:00 UTC by Report Bug Copy Issue Report external import "import pandas as pd" should be placed before "from ..psf_fast import PSFWrapper, InvalidPSF" Loading history...
4
5			from threeML.exceptions.custom_exceptions import custom_warnings
			0 ignored issues – show introduced 2018-07-06 22:00 UTC by Report Bug Copy Issue Report Unable to import 'threeML.exceptions.custom_exceptions' Loading history... Unused Code introduced 2018-07-06 22:00 UTC by Report Bug Copy Issue Report Unused custom_warnings imported from threeML.exceptions.custom_exceptions Loading history... introduced 2018-07-06 22:00 UTC by Report Bug Copy Issue Report external import "from threeML.exceptions.custom_exceptions import custom_warnings" should be placed before "from ..psf_fast import PSFWrapper, InvalidPSF" Loading history...
6
7
8			class ResponseBin(object):
			0 ignored issues – show Coding Style introduced 2018-07-06 22:00 UTC by Report Bug Copy Issue Report This class should have a docstring. The coding style of this project requires that you add a docstring to this code element. Below, you find an example for methods: class SomeClass: def some_method(self): """Do x and return foo.""" If you would like to know more about docstrings, we recommend to read PEP-257: Docstring Conventions. Loading history... best-practice introduced 2018-07-06 22:00 UTC by Report Bug Copy Issue Report Too many instance attributes (12/7) Loading history...
9
10			def __init__(self, name, min_dec, max_dec, dec_center,
			0 ignored issues – show best-practice introduced 2018-07-06 22:00 UTC by Report Bug Copy Issue Report Too many arguments (13/5) Loading history...
11			sim_n_sig_events, sim_n_bg_events,
12			sim_energy_bin_low, sim_energy_bin_centers, sim_energy_bin_hi,
13			sim_differential_photon_fluxes, sim_signal_events_per_bin,
14			psf):
15
16			self._name = name
17			self._min_dec = min_dec
18			self._max_dec = max_dec
19			self._dec_center = dec_center
20			self._sim_n_sig_events = sim_n_sig_events
21			self._sim_n_bg_events = sim_n_bg_events
22			self._sim_energy_bin_low = sim_energy_bin_low
23			self._sim_energy_bin_centers = sim_energy_bin_centers
24			self._sim_energy_bin_hi = sim_energy_bin_hi
25			self._sim_differential_photon_fluxes = sim_differential_photon_fluxes
26			self._sim_signal_events_per_bin = sim_signal_events_per_bin
27			self._psf = psf # type: PSFWrapper
28
29			@staticmethod
30			def _get_en_th1d(open_ttree, dec_id, dec_id_label, analysis_bin_id, analysis_bin_id_label, prefix):
			0 ignored issues – show Coding Style introduced 2018-07-06 22:00 UTC by Report Bug Copy Issue Report This line is too long as per the coding-style (103/100). This check looks for lines that are too long. You can specify the maximum line length. Loading history... best-practice introduced 2018-07-06 22:00 UTC by Report Bug Copy Issue Report Too many arguments (6/5) Loading history...
31
32			nh_name = "nh%i" % analysis_bin_id
33

giacomov / hawc_hal

Push — master ( 573345...9597ff )

ResponseBin._get_en_th1d() A

Complexity

Size

Duplication

Importance

Duplication Side-by-Side

Filter issues like