Plotter Class Reference

Public Member Functions
	__init__ (self, location)
	plotEcmsComparison (self, limits=None, tag='')
	plotSpreadComparison (self, limits=None)
	plotShift (self, limits=None)
	plotEcms (self, limits=None, tag='')
	plotPulls (self, limits=None)

Static Public Member Functions
	plotLine (df, var, color, label)
	plotCurve (df, var, label, withBand=True, withCurve=True)

Public Attributes
	dfB = pd.read_csv(f'{location}/BonlyEcmsCalib.txt', delim_whitespace=True)
	read B-only calibration
	dfC = pd.read_csv(f'{location}/finalEcmsCalib.txt', delim_whitespace=True)
	read combined calibration
	dfM = pd.read_csv(f'{location}/mumuEcalib.txt', delim_whitespace=True)
	read mumu calibration

Detailed Description

Plotting class

Definition at line 126 of file ecms.py.

Constructor & Destructor Documentation

__init__()

__init__	(		self,
			location )

Data are loaded from text files to pandas

Definition at line 129 of file ecms.py.

    def __init__(self, location):
        """
        Data are loaded from text files to pandas
        """
 
        
        self.dfB = pd.read_csv(f'{location}/BonlyEcmsCalib.txt', delim_whitespace=True)
 
        
        self.dfC = pd.read_csv(f'{location}/finalEcmsCalib.txt', delim_whitespace=True)
 
        
        self.dfM = pd.read_csv(f'{location}/mumuEcalib.txt', delim_whitespace=True)
 
        # add the state
        dfM = self.dfM
        dfC = self.dfC
        state = -1
        dfM['type'] = 1
        dfC['type'] = 1
        for i in range(len(dfM)):
            if (dfM['id'][i]) == 0:
                state *= -1
 
            dfM.at[i, 'type'] = state
            dfC.at[i, 'type'] = state
 

Member Function Documentation

plotCurve()

plotCurve ( df, var, label, withBand = True, withCurve = True )

static

Plot curve with possible error band where large intervals are distinguished by color

Definition at line 235 of file ecms.py.

    def plotCurve(df, var, label, withBand=True, withCurve=True):
        """
        Plot curve with possible error band where large intervals are distinguished by color
        """
 
        def plotBands(df, var, color, withBand=True, withCurve=True):
            varUnc = var + 'Unc'
            nan = np.full(len(df), np.nan)
            avg = (df['t1']+df['t2']).to_numpy()/2
            times = np.c_[df[['t1',    't2']].to_numpy(), avg].ravel()
            eCMS = np.c_[df[[var, var]].to_numpy(), nan].ravel()
            times = toJST(times)
 
            if withBand:
                eCMSu = np.c_[df[[varUnc, varUnc]].to_numpy(), nan].ravel()
                plt.fill_between(times, eCMS-eCMSu, eCMS+eCMSu, alpha=0.15, color=color)
            if withCurve:
                plt.plot(times, eCMS, linewidth=2, color=color)
 
        nan = np.full(len(df), np.nan)
        avg = (df['t1']+df['t2']).to_numpy()/2
        timesg = np.c_[df['t1'].to_numpy(),   avg,  df['t2'].to_numpy()].ravel()
        eCMSg = np.c_[df[var].to_numpy(), nan,  df[var].to_numpy()].ravel()
        timesg = toJST(timesg)
 
        if withCurve:
            plt.plot(timesg, eCMSg, linewidth=2, color='gray', alpha=0.35)
 
        plotBands(df[df['type'] == 1], var, 'red', withBand, withCurve)
        plotBands(df[df['type'] == -1], var, 'blue', withBand, withCurve)
 

plotEcms()

plotEcms	(		self,
			limits = None,
			tag = '' )

Plot Ecms of the combined fit, 'tag' can be '4S' means that y-axis is zoomed to 4S mass region, 'Off' to off-resonance

Definition at line 288 of file ecms.py.

    def plotEcms(self, limits=None,  tag=''):
        """
        Plot Ecms of the combined fit, 'tag' can be '4S' means that y-axis is zoomed to 4S mass region, 'Off' to off-resonance
        """
 
        Plotter.plotCurve(self.dfC, 'Ecms',  label='Combined method')
 
        plotSplitLines(self.dfC)
 
        plt.xlabel('time')
        plt.ylabel('Ecms [MeV]')
 
        setPlotRange(self.dfC, tag)
 
        loc = 'plots/allData'
        if limits is not None:
            plt.xlim(datetime.strptime(limits[0], '%Y-%m-%d'), datetime.strptime(limits[1], '%Y-%m-%d'))
            loc = 'plots/' + limits[0] + 'to' + limits[1]
 
        os.makedirs(loc, exist_ok=True)
        plt.savefig(f'{loc}/Ecms{tag}.png')
        plt.clf()
 

plotEcmsComparison()

plotEcmsComparison	(		self,
			limits = None,
			tag = '' )

Plot Ecms obtained from combined, hadB and mumu method

Definition at line 179 of file ecms.py.

    def plotEcmsComparison(self, limits=None, tag=''):
        """
        Plot Ecms obtained from combined, hadB and mumu method
        """
 
        Plotter.plotLine(self.dfB, 'Ecms', 'green', label='B decay method')
        Plotter.plotLine(self.dfC, 'Ecms', 'red', label='Combined method')
 
        d = np.nanmedian(self.dfC['Ecms']-self.dfM['Ecms'])
 
        dfMc = self.dfM.copy()
        dfMc['Ecms'] += d
        Plotter.plotLine(dfMc, 'Ecms', 'blue', label=f'mumu method (+{round(d,1)} MeV)')
 
        plotSplitLines(self.dfC)
        setPlotRange(self.dfC, tag)
 
        plt.xlabel('time')
        plt.ylabel('Ecms [MeV]')
 
        plt.legend()
 
        loc = 'plots/allData'
        if limits is not None:
            plt.xlim(datetime.strptime(limits[0], '%Y-%m-%d'), datetime.strptime(limits[1], '%Y-%m-%d'))
            loc = 'plots/' + limits[0] + 'to' + limits[1]
 
        os.makedirs(loc, exist_ok=True)
        plt.savefig(f'{loc}/EcmsComparison{tag}.png')
        plt.clf()
 

plotLine()

plotLine ( df, var, color, label )

static

Plot single line with error band

Definition at line 157 of file ecms.py.

    def plotLine(df, var, color, label):
        """
        Plot single line with error band
        """
 
        varUnc = var + 'Unc'
        nan = np.full(len(df), np.nan)
        avg = (df['t1']+df['t2']).to_numpy()/2
        times = np.c_[df[['t1',    't2']].to_numpy(), avg].ravel()
        eCMS = np.c_[df[[var, var]].to_numpy(), nan].ravel()
        eCMSu = np.c_[df[[varUnc, varUnc]].to_numpy(), nan].ravel()
 
        timesg = np.c_[df['t1'].to_numpy(),   avg,  df['t2'].to_numpy()].ravel()
        eCMSg = np.c_[df[var].to_numpy(), nan,  df[var].to_numpy()].ravel()
 
        times = toJST(times)
        timesg = toJST(timesg)
 
        plt.fill_between(times, eCMS-eCMSu, eCMS+eCMSu, alpha=0.2, color=color)
        plt.plot(times, eCMS, linewidth=2, color=color, label=label)
        plt.plot(timesg, eCMSg, linewidth=2, color=color, alpha=0.35)
 

plotPulls()

plotPulls	(		self,
			limits = None )

Plot pulls of the combined fit.

Definition at line 311 of file ecms.py.

    def plotPulls(self, limits=None):
        """
        Plot pulls of the combined fit.
        """
 
        dfC = self.dfC.copy()
        dfC['ref'] = 0
        dfC['refUnc'] = 1
        Plotter.plotCurve(dfC, 'pull', label='Combined method', withBand=False)
        Plotter.plotCurve(dfC, 'ref',  label='Combined method', withCurve=False)
 
        plotSplitLines(self.dfC)
 
        plt.ylim(-1.5, 1.5)
        plt.xlabel('time')
        plt.ylabel('pull')
 
        loc = 'plots/allData'
        if limits is not None:
            plt.xlim(datetime.strptime(limits[0], '%Y-%m-%d'), datetime.strptime(limits[1], '%Y-%m-%d'))
            loc = 'plots/' + limits[0] + 'to' + limits[1]
 
        os.makedirs(loc, exist_ok=True)
        plt.savefig(f'{loc}/Pull.png')
        plt.clf()
 
 
# Get the results from the combined calibration

plotShift()

plotShift	(		self,
			limits = None )

Plot shift between Bhad and mumu method

Definition at line 266 of file ecms.py.

    def plotShift(self, limits=None):
        """
        Plot shift between Bhad and mumu method
        """
 
        Plotter.plotCurve(self.dfC, 'shift',  label='Combined method')
 
        plotSplitLines(self.dfC)
 
        plt.xlabel('time')
        plt.ylabel('shift [MeV]')
 
        loc = 'plots/allData'
        if limits is not None:
            plt.xlim(datetime.strptime(limits[0], '%Y-%m-%d'), datetime.strptime(limits[1], '%Y-%m-%d'))
            loc = 'plots/' + limits[0] + 'to' + limits[1]
 
        os.makedirs(loc, exist_ok=True)
        plt.savefig(f'{loc}/Shift.png')
 
        plt.clf()
 

plotSpreadComparison()

plotSpreadComparison	(		self,
			limits = None )

Plot Ecms spread estimated from the combined method and B-decay method

Definition at line 210 of file ecms.py.

    def plotSpreadComparison(self, limits=None):
        """
        Plot Ecms spread estimated from the combined method and B-decay method
        """
 
        Plotter.plotLine(self.dfB, 'spread', 'green', label='B decay method')
        Plotter.plotLine(self.dfC, 'spread', 'red', label='Combined method')
 
        plotSplitLines(self.dfC)
 
        plt.legend()
 
        plt.xlabel('time')
        plt.ylabel('spread [MeV]')
 
        loc = 'plots/allData'
        if limits is not None:
            plt.xlim(datetime.strptime(limits[0], '%Y-%m-%d'), datetime.strptime(limits[1], '%Y-%m-%d'))
            loc = 'plots/' + limits[0] + 'to' + limits[1]
 
        os.makedirs(loc, exist_ok=True)
        plt.savefig(f'{loc}/SpreadComparison.png')
        plt.clf()
 

Member Data Documentation

dfB

dfB = pd.read_csv(f'{location}/BonlyEcmsCalib.txt', delim_whitespace=True)

read B-only calibration

Definition at line 135 of file ecms.py.

dfC

dfC = pd.read_csv(f'{location}/finalEcmsCalib.txt', delim_whitespace=True)

read combined calibration

Definition at line 138 of file ecms.py.

dfM

dfM = pd.read_csv(f'{location}/mumuEcalib.txt', delim_whitespace=True)

read mumu calibration

Definition at line 141 of file ecms.py.

---

The documentation for this class was generated from the following file:

• calibration/scripts/prompt/validations/ecms.py