/data/doxygen/doxygen-1.7.3/gen/CMSSW_4_2_8/src/Validation/RecoB/src/BTagValidator.cc

Go to the documentation of this file.

#include "Validation/RecoB/interface/BTagValidator.h"

// root include files
#include "TClass.h"
#include "TDirectory.h"
#include "TH1.h"
#include "TH2.h"
#include "TH3.h"
#include "TString.h"

#include "FWCore/Utilities/interface/Exception.h"
#include "FWCore/Version/interface/GetReleaseVersion.h"

#include "Validation/RecoB/interface/HistoCompare.h"
#include "DQMServices/Core/interface/DQMStore.h"
#include "DQMServices/Core/interface/MonitorElement.h"

//
// constructors and destructor
//
BTagValidator::BTagValidator(const edm::ParameterSet& iConfig) {


        algorithm_ = iConfig.getParameter<std::string>( "algorithm" );
        rootFile_ = iConfig.getParameter<std::string>( "rootfile" );
        DQMFile_ = iConfig.getParameter<std::string>( "DQMFile" );
        TString tversion(edm::getReleaseVersion());
        tversion = tversion.Remove(0,1);
        tversion = tversion.Remove(tversion.Length()-1,tversion.Length());
        DQMFile_  = std::string(tversion)+"_"+DQMFile_;
        histogramList_ = iConfig.getParameter<vstring>( "histogramList" );
        referenceFilename_ = iConfig.getParameter<std::string>( "referenceFilename" );
        doCompare_ = iConfig.getParameter<bool>( "compareHistograms");
}


BTagValidator::~BTagValidator() {

}


//
// member functions
//

// ------------ method called to for each event  ------------
void
BTagValidator::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup)
{ }


void 
BTagValidator::endJob() 
{
        // Validation section
        TObject * tObject ;

        // DQM element
        MonitorElement* monElement ;
        MonitorElement* monElementRes;

        TFile * file = new TFile ( TString ( rootFile_ ) ) ;

        // comparison
        HistoCompare hcompare;

        if (doCompare_) hcompare.SetReferenceFilename(TString(referenceFilename_) );
        
        file->cd();

        // get hold of back-end interface
        DQMStore * dbe = edm::Service<DQMStore>().operator->();
        
        dbe->setCurrentFolder( "RecoBV/"+algorithm_ );
   
        for (std::size_t i=0; i<histogramList_.size(); ++i) {
                
                tObject = gDirectory->Get( TString( histogramList_[i] ) ) ;

                if ( tObject == 0 ) 
                        throw cms::Exception("BTagValidator") << "Histogram " << histogramList_[i] << " was not produced in the analysis. ";    
                                                                
                if ( tObject->IsA()->InheritsFrom( "TH1" ) ) {
                        
                        TH1 * histogram = (TH1*) tObject ;  
                        
                        TH1* hresiduals = 0;
                        
                        if (doCompare_) 
                                hresiduals = hcompare.Compare(histogram, "/DQMData/"+TString(algorithm_)+"/"+TString(histogram->GetName()) );
                        
                        file->cd();
                        
                        monElement = dbe->book1D (
                                std::string( histogram->GetName() ),
                                std::string( histogram->GetTitle() ),
                                histogram->GetXaxis()->GetNbins(),
                                histogram->GetXaxis()->GetXmin(),
                                histogram->GetXaxis()->GetXmax()
                                );

                        monElement->setAxisTitle( std::string ( histogram->GetXaxis()->GetTitle()) , 1);
                        monElement->setAxisTitle( std::string ( histogram->GetYaxis()->GetTitle()) , 2);


                        for(Int_t x=0; x<histogram->GetXaxis()->GetNbins(); x++) {
                          monElement->setBinContent ( x, histogram->GetBinContent( x ) ) ;
                          monElement->setBinError ( x, histogram->GetBinError( x ) ) ;
                        }

                        if (doCompare_ && hresiduals!= 0 ) {
                          monElementRes = dbe->book1D (
                                                       std::string( hresiduals->GetName() ),
                                                       std::string( hresiduals->GetTitle() ),
                                                       hresiduals->GetXaxis()->GetNbins(),
                                                       hresiduals->GetXaxis()->GetXmin(),
                                                       hresiduals->GetXaxis()->GetXmax()
                                                       );

                        
                          for(Int_t x=0; x<hresiduals->GetXaxis()->GetNbins(); x++) {
                            monElementRes->setBinContent ( x, hresiduals->GetBinContent( x ) ) ; 
                            monElementRes->setBinError ( x, hresiduals->GetBinError( x ) ) ;
                          }  
                        }
                }
                else if ( tObject->IsA()->InheritsFrom( "TH2" ) ) {
                        
                        TH2 * histogram = (TH2*) tObject ;  
               
                        monElement = dbe->book2D (
                                std::string( histogram->GetName() ),
                                std::string( histogram->GetTitle() ),
                                histogram->GetXaxis()->GetNbins(),
                                histogram->GetXaxis()->GetXmin(),
                                histogram->GetXaxis()->GetXmax(),
                                histogram->GetYaxis()->GetNbins(),
                                histogram->GetYaxis()->GetXmin(),
                                histogram->GetYaxis()->GetXmax()          
                                );

                        monElement->setAxisTitle( std::string ( histogram->GetXaxis()->GetTitle()) , 1);
                        monElement->setAxisTitle( std::string ( histogram->GetYaxis()->GetTitle()) , 2);
                        
                        for(Int_t x=0; x<histogram->GetXaxis()->GetNbins(); x++)
                                for(Int_t y=0; y<histogram->GetYaxis()->GetNbins(); y++) {
                                        monElement->setBinContent ( x, y, histogram->GetBinContent( x, y ) ) ;                 
                                        monElement->setBinError ( x, y, histogram->GetBinError( x, y ) ) ;                 
                                }
                }
                else if ( tObject->IsA()->InheritsFrom( "TH3" ) ) {
                        
                        TH3 * histogram = (TH3*) tObject ;  
                        
                        monElement = dbe->book3D (
                                std::string( histogram->GetName() ),
                                std::string( histogram->GetTitle() ),
                                histogram->GetXaxis()->GetNbins(),
                                histogram->GetXaxis()->GetXmin(),
                                histogram->GetXaxis()->GetXmax(),
                                histogram->GetYaxis()->GetNbins(),
                                histogram->GetYaxis()->GetXmin(),
                                histogram->GetYaxis()->GetXmax(),         
                                histogram->GetZaxis()->GetNbins(),
                                histogram->GetZaxis()->GetXmin(),
                                histogram->GetZaxis()->GetXmax()          
                                );

                        monElement->setAxisTitle( std::string ( histogram->GetXaxis()->GetTitle()) , 1);
                        monElement->setAxisTitle( std::string ( histogram->GetYaxis()->GetTitle()) , 2);
                        
                        for(Int_t x=0; x<histogram->GetXaxis()->GetNbins(); x++)
                                for(Int_t y=0; y<histogram->GetYaxis()->GetNbins(); y++)
                                        for(Int_t z=0; z<histogram->GetZaxis()->GetNbins(); z++) {
                                                monElement->setBinContent ( x, y, z, histogram->GetBinContent( x, y, z ) ) ;                 
                                                monElement->setBinError ( x, y, z, histogram->GetBinError( x, y, z ) ) ;                 
                                        }
                }
        }

        dbe->showDirStructure() ;
        dbe->save(DQMFile_) ;  
        file->Close() ;
        
}

//define this as a plug-in
DEFINE_FWK_MODULE(BTagValidator);