def __init__(self, varname, raw_name, target_name, option='noskim', max_entries=-1, prescale=1, prescale_phase=0, rho=0.7): #print 'DEBUG', self.__class__.__name__, '__init__' common_args = [varname, option, max_entries, prescale, prescale_phase] self.raw = DataSource(raw_name, *common_args) self.target = DataSource(target_name, *common_args) self.corrector = PhotonIdCorrector(self.raw.data, self.target.data, rho) self.postprocess_corrector()
def test(max_entries=-1): ''' Tests the PhotonIdCorrector class. ''' global raw_data, target_data, corr, vplot varname = 'setab' option = 'skim10k' raw_name = 's12-zllm50-v7n' target_name = 'r12a-pho-j22-v1' raw_data = get_dataset(raw_name, varname, max_entries, option) target_data = get_dataset(target_name, varname, max_entries, option) xvar = raw_data.get().first() raw_data.SetTitle('Raw ' + raw_name.split('-')[0].capitalize()) target_data.SetTitle('Raw ' + target_name.split('-')[0].capitalize()) corr = PhotonIdCorrector(raw_data, target_data, rho=0.7) corr.SetName('_'.join([ raw_name.split('-')[0], 'to', target_name.split('-')[0], varname, 'qqcorrector' ])) corr.SetTitle(' '.join([ raw_name.split('-')[0].capitalize(), 'to', target_name.split('-')[0].capitalize(), xvar.GetTitle(), 'Q-Q Corrector' ])) plot = xvar.frame(roo.Title(raw_data.GetTitle())) raw_data.plotOn(plot) corr.xpdf.plotOn(plot) canvases.next(varname + '_' + raw_name.split('-')[0]).SetGrid() draw_and_append(plot) plot = xvar.frame(roo.Title(target_data.GetTitle())) target_data.plotOn(plot) corr.ypdf.plotOn(plot) canvases.next(varname + '_' + target_name.split('-')[0]).SetGrid() draw_and_append(plot) canvases.next(corr.GetName()).SetGrid() draw_and_append(corr.get_correction_plot()) canvases.next(corr.GetName() + '_validation').SetGrid() draw_and_append(corr.get_validation_plot()) canvases.update()