/afs/cern.ch/work/a/aaltunda/public/www/CMSSW_6_2_7/src/CondCore/HcalPlugins/plugins/HcalChannelQualityPyWrapper.cc

Go to the documentation of this file.

#include "CondFormats/HcalObjects/interface/HcalChannelQuality.h"
#include "CondCore/Utilities/interface/PayLoadInspector.h"
#include "CondCore/Utilities/interface/InspectorPythonWrapper.h"

#include <string>
#include <fstream>
#include <sstream>

#include "TH1F.h"
#include "TH2F.h"
#include "DataFormats/DetId/interface/DetId.h"
#include "DataFormats/HcalDetId/interface/HcalDetId.h"

#include "TROOT.h"
#include "TCanvas.h"
#include "TStyle.h"
#include "TColor.h"
#include "TLine.h"


//functions for correct representation of data in summary and plot:
#include "CondCore/HcalPlugins/interface/HcalObjRepresent.h"
using namespace HcalObjRepresent;

namespace cond {

        template<>
        class ValueExtractor<HcalChannelQuality>: public  BaseValueExtractor<HcalChannelQuality> {
        public:
                typedef HcalChannelQuality Class;
                typedef ExtractWhat<Class> What;
                static What what() { return What();}

                ValueExtractor(){}

                ValueExtractor(What const & what)
                {
                        // here one can make stuff really complicated...
                }

                void compute(Class const & it){
                }
        private:

        };

        template<>
        std::string PayLoadInspector<HcalChannelQuality>::summary() const {
                std::stringstream ss;

                //setting map for representing errors
                std::string statusBitArray[20]; 
                short unsigned int bitMap[9];
                statusBitArray[0] = std::string("cell is off" );
                statusBitArray[1] = std::string("cell is masked/to be masked at RecHit Level" );
                statusBitArray[5] = std::string("cell is dead (from DQM algo)");
                statusBitArray[6] = std::string("cell is hot (from DQM algo)" );
                statusBitArray[7] = std::string("cell has stability error");
                statusBitArray[8] = std::string("cell has timing error" );
                statusBitArray[15] = std::string("cell is masked from the Trigger ");
                statusBitArray[18] = std::string("cell is always excluded from the CaloTower regardless of other bit settings.");
                statusBitArray[19] = std::string("cell is counted as problematic within the tower.");
                bitMap[0] = 0; //{0, 1, 5, 6, 7, 8, 15, 18, 19};
                bitMap[1] = 1;
                bitMap[2] = 5;
                bitMap[3] = 6;
                bitMap[4] = 7;
                bitMap[5] = 8;
                bitMap[6] = 15;
                bitMap[7] = 18;
                bitMap[8] = 19;

                // get all containers with names
                HcalChannelQuality::tAllContWithNames allContainers = object().getAllContainers();

                // initializing iterators
                HcalChannelQuality::tAllContWithNames::const_iterator iter;
                std::vector<HcalChannelStatus>::const_iterator contIter;
                ss << "Total HCAL containers: " << allContainers.size() << std::endl;

                //run trough all pair containers, print error values if any.
                for (iter = allContainers.begin(); iter != allContainers.end(); ++iter){
                        ss << "---------------------------------------------" << std::endl;
                        ss << "Detector: " << (*iter).first << ";    Total values: "<< (*iter).second.size() << std::endl;
                        unsigned int j = 0;
                        for (contIter = (*iter).second.begin(); contIter != (*iter).second.end(); ++contIter){

                                //if not 0, it have error, print it:
                                if ((*contIter).getValue() != 0){//HcalDetId::HcalDetId(uint32_t rawid)
                                        ss << "     Id["<< j << "]: " <<  
                                                " rawId: " << (uint32_t)(*contIter).rawId() << " "<< HcalDetId((uint32_t)(*contIter).rawId())<<"; Channel bits: " <<
                                                (uint32_t)(*contIter).getValue()<< "; Binary format: " << IntToBinary((uint32_t)(*contIter).getValue()) << "; Errors: "
                                                << getBitsSummary((uint32_t)((*contIter).getValue()), statusBitArray, bitMap);
                                }
                                ++j;
                        }
                }
                return ss.str();
        }

        template<>
        std::string PayLoadInspector<HcalChannelQuality>::plot(std::string const & filename,//
                std::string const &,
                std::vector<int> const&,
                std::vector<float> const& ) const 
        {
                std::vector<TH2F> graphData;
                setup(graphData, "ChannelStatus"); 

                std::stringstream x;
                // Change the titles of each individual histogram
                for (unsigned int d=0;d < graphData.size();++d){
                        graphData[d].Reset();
                        x << "1+log2(status) for HCAL depth " << d+1;

                        //BUG CAN BE HERE:
                        //if (ChannelStatus->depth[d]) 
                        graphData[d].SetTitle(x.str().c_str());  // replace "setTitle" with "SetTitle", since you are using TH2F objects instead of MonitorElements
                        x.str("");
                }

                HcalDetId hcal_id;
                int ieta, depth, iphi, channelBits;
                double logstatus;

                //main loop
                // get all containers with names
                HcalChannelQuality::tAllContWithNames allContainers = object().getAllContainers();

                // initializing iterators
                HcalChannelQuality::tAllContWithNames::const_iterator iter;
                std::vector<HcalChannelStatus>::const_iterator contIter;

                //run trough all pair containers
                for (iter = allContainers.begin(); iter != allContainers.end(); ++iter){
                        for (contIter = (*iter).second.begin(); contIter != (*iter).second.end(); ++contIter){
                                hcal_id = HcalDetId((uint32_t)(*contIter).rawId());

                                channelBits = (uint32_t)(*contIter).getValue();
                                if (channelBits == 0) 
                                        continue;

                                depth = hcal_id.depth();
                                if (depth<1 || depth>4) 
                                        continue;

                                ieta=hcal_id.ieta();
                                iphi=hcal_id.iphi();

                                if (hcal_id.subdet() == HcalForward)
                                        ieta>0 ? ++ieta : --ieta;

                                logstatus = log2(1.*channelBits)+1;
                                //FILLING GOES HERE:
                                graphData[depth-1].Fill(ieta,iphi,logstatus);

                                //FOR DEBUGGING:
                                //std::cout << "ieta: " << ieta << "; iphi: " << iphi << "; logstatus: " << logstatus << "; channelBits: " << channelBits<< std::endl;
                        }
                }
                FillUnphysicalHEHFBins(graphData);



                //Drawing...
                // use David's palette
                gStyle->SetPalette(1);
                const Int_t NCont = 999;
                gStyle->SetNumberContours(NCont);
                TCanvas canvas("CC map","CC map",840,369*4);

                TPad pad1("pad1","pad1", 0.0, 0.75, 1.0, 1.0);
                pad1.Draw();
                TPad pad2("pad2","pad2", 0.0, 0.5, 1.0, 0.75);
                pad2.Draw();
                TPad pad3("pad3","pad3", 0.0, 0.25, 1.0, 0.5);
                pad3.Draw();
                TPad pad4("pad4","pad4", 0.0, 0.0, 1.0, 0.25);
                pad4.Draw();


                pad1.cd();
                graphData[0].SetStats(0);
                graphData[0].Draw("colz");

                pad2.cd();
                graphData[1].SetStats(0);
                graphData[1].Draw("colz");

                pad3.cd();
                graphData[2].SetStats(0);
                graphData[2].Draw("colz");

                pad4.cd();
                graphData[3].SetStats(0);
                graphData[3].Draw("colz");


                std::stringstream ss;
                ss <<filename << ".png";

                canvas.SaveAs((ss.str()).c_str());

                return (ss.str()).c_str();
        }


}
PYTHON_WRAPPER(HcalChannelQuality,HcalChannelQuality);