EcalDQMChannelStatusPyWrapper.cc

Go to the documentation of this file.

#include "CondFormats/EcalObjects/interface/EcalDQMChannelStatus.h"
//#include "CondTools/Ecal/interface/EcalPedestalsXMLTranslator.h"
#include "CondTools/Ecal/interface/EcalCondHeader.h"
#include "TH2F.h"
#include "TCanvas.h"
#include "TStyle.h"
#include "TLine.h"
#include "DataFormats/EcalDetId/interface/EBDetId.h"
#include "DataFormats/EcalDetId/interface/EEDetId.h"

#include "CondCore/Utilities/interface/PayLoadInspector.h"
#include "CondCore/Utilities/interface/InspectorPythonWrapper.h"

#include <string>
#include <sstream>
#include <algorithm>
#include <numeric>
#include <iterator>
#include <boost/ref.hpp>
#include <boost/bind.hpp>
#include <boost/function.hpp>
#include <boost/iterator/transform_iterator.hpp>
#include <fstream>

#include "CondCore/EcalPlugins/plugins/EcalPyWrapperFunctions.h"
namespace cond {

    namespace ecalcond {

        typedef EcalDQMChannelStatus::Items  Items;
        typedef EcalDQMChannelStatus::value_type  value_type;

        enum How { singleChannel, bySuperModule, all};

        int bad(Items const & cont) {
            return  std::count_if(cont.begin(),cont.end(),
                boost::bind(std::greater<int>(),
                boost::bind(&value_type::getStatusCode,_1),0)
                );
        }

        void extractBarrel(EcalDQMChannelStatus const & cont, std::vector<int> const &,  std::vector<float> & result) {
            result.resize(1);
            result[0] =  bad(cont.barrelItems());
        }

        void extractEndcap(EcalDQMChannelStatus const & cont, std::vector<int> const &,  std::vector<float> & result) {
            result.resize(1);
            result[0] = bad(cont.endcapItems());
        }
        void extractAll(EcalDQMChannelStatus const & cont, std::vector<int> const &,  std::vector<float> & result) {
            result.resize(1);
            result[0] = bad(cont.barrelItems())+bad(cont.endcapItems());
        }

        void extractSuperModules(EcalDQMChannelStatus const & cont, std::vector<int> const & which,  std::vector<float> & result) {
            // bho...
        }

        void extractSingleChannel(EcalDQMChannelStatus const & cont, std::vector<int> const & which,  std::vector<float> & result) {
            result.reserve(which.size());
            for (unsigned int i=0; i<which.size();i++) {
                result.push_back(cont[which[i]].getStatusCode());
            }
        }

        typedef boost::function<void(EcalDQMChannelStatus const & cont, std::vector<int> const & which,  std::vector<float> & result)> CondExtractor;
    }  // namespace ecalcond

    template<>
    struct ExtractWhat<EcalDQMChannelStatus> {

        ecalcond::How m_how;
        std::vector<int> m_which;

        ecalcond::How const & how() const { return m_how;}
        std::vector<int> const & which() const { return m_which;}

        void set_how(ecalcond::How i) {m_how=i;}
        void set_which(std::vector<int> & i) { m_which.swap(i);}
    };


    template<>
    class ValueExtractor<EcalDQMChannelStatus>: public  BaseValueExtractor<EcalDQMChannelStatus> {
    public:

        static ecalcond::CondExtractor & extractor(ecalcond::How how) {
            static  ecalcond::CondExtractor fun[3] = { 
                ecalcond::CondExtractor(ecalcond::extractSingleChannel),
                ecalcond::CondExtractor(ecalcond::extractSuperModules),
                ecalcond::CondExtractor(ecalcond::extractAll)
            };
            return fun[how];
        }

        typedef EcalDQMChannelStatus Class;
        typedef ExtractWhat<Class> What;
        static What what() { return What();}

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

        void compute(Class const & it){
            std::vector<float> res;
            extractor(m_what.how())(it,m_what.which(),res);
            swap(res);
        }

    private:
        What  m_what;  
    };


    //template<>
    //std::string PayLoadInspector<EcalDQMChannelStatus>::dump() const {
    //  std::stringstream ss;
    //  EcalCondHeader h;
    //  ss << EcalDQMChannelStatusXMLTranslator::dumpXML(h,object());
    //  return ss.str();
    //}

    class EcalDQMChannelStatusHelper: public EcalPyWrapperHelper<EcalDQMStatusCode>{
    public:
        //change me
        EcalDQMChannelStatusHelper():EcalPyWrapperHelper<EcalObject>(30, STATUS, "-Errors total: "){}
    protected:

        //change me
        typedef EcalDQMStatusCode EcalObject;

        type_vValues getValues( const std::vector<EcalObject> & vItems)
        {
            //change me
            //unsigned int totalValues = 2; 

            type_vValues vValues(total_values);

            std::stringstream ss;

            std::string valueNames[] = {
                "CH_ID_ERROR                          ",
                "CH_GAIN_ZERO_ERROR                   ",
                "CH_GAIN_SWITCH_ERROR                 ",
                "TT_ID_ERROR                          ",
                "TT_SIZE_ERROR                        ",
                "PEDESTAL_LOW_GAIN_MEAN_ERROR         ",
                "PEDESTAL_MIDDLE_GAIN_MEAN_ERROR      ",
                "PEDESTAL_HIGH_GAIN_MEAN_ERROR        ",
                "PEDESTAL_LOW_GAIN_RMS_ERROR          ",
                "PEDESTAL_MIDDLE_GAIN_RMS_ERROR       ",
                "PEDESTAL_HIGH_GAIN_RMS_ERROR         ",
                "PEDESTAL_ONLINE_HIGH_GAIN_MEAN_ERROR ",
                "PEDESTAL_ONLINE_HIGH_GAIN_RMS_ERROR  ",
                "TESTPULSE_LOW_GAIN_MEAN_ERROR        ",
                "TESTPULSE_MIDDLE_GAIN_MEAN_ERROR     ",
                "TESTPULSE_HIGH_GAIN_MEAN_ERROR       ",
                "TESTPULSE_LOW_GAIN_RMS_ERROR         ",
                "TESTPULSE_MIDDLE_GAIN_RMS_ERROR      ",
                "TESTPULSE_HIGH_GAIN_RMS_ERROR        ",
                "LASER_MEAN_ERROR                     ",
                "LASER_RMS_ERROR                      ",
                "LASER_TIMING_MEAN_ERROR              ",
                "LASER_TIMING_RMS_ERROR               ",
                "LED_MEAN_ERROR                       ",
                "LED_RMS_ERROR                        ",
                "LED_TIMING_MEAN_ERROR                ",
                "LED_TIMING_RMS_ERROR                 ",
                "STATUS_FLAG_ERROR                    ",
                "PHYSICS_BAD_CHANNEL_WARNING          ",
                "PHYSICS_BAD_CHANNEL_ERROR            "
            };
            for (unsigned int i = 0; i < total_values; ++i){
                ss.str(""); ss << "[" << i << "]" << valueNames[i];
                vValues[i].first = ss.str();
                vValues[i].second = .0;
            }

            //get info:
            unsigned int shift = 0, mask = 1;
            unsigned int statusCode;
            for(std::vector<EcalObject>::const_iterator iItems = vItems.begin(); iItems != vItems.end(); ++iItems){
                //change us
                statusCode = iItems->getStatusCode();
                for (shift = 0; shift < total_values; ++shift){
                    mask = 1 << (shift);
                    //std::cout << "; statuscode: " << statusCode;
                    if (statusCode & mask){
                        vValues[shift].second += 1;
                    }
                }
            }
            return vValues;
        }
    };

    template<>
    std::string PayLoadInspector<EcalDQMChannelStatus>::summary() const {
        std::stringstream ss;
        EcalDQMChannelStatusHelper helper;
        ss << helper.printBarrelsEndcaps(object().barrelItems(), object().endcapItems());
        return ss.str();
    }


    template<>
    std::string PayLoadInspector<EcalDQMChannelStatus>::plot(std::string const & filename,
        std::string const &, 
        std::vector<int> const&, 
        std::vector<float> const& ) const {
            gStyle->SetPalette(1);
            const int TOTAL_IMAGES = 1;
            const int TOTAL_PADS = 3;

            const float IMG_SIZE = 1.5;
            TCanvas canvas("CC map","CC map",800*IMG_SIZE, 200 * TOTAL_IMAGES*IMG_SIZE);//800, 1200

            float xmi[3] = {0.0 , 0.22, 0.78};
            float xma[3] = {0.22, 0.78, 1.00};


            TPad*** pad = new TPad**[TOTAL_IMAGES];
            for (int gId = 0; gId < TOTAL_IMAGES; gId++) {
                pad[gId] = new TPad*[TOTAL_PADS];
                for (int obj = 0; obj < TOTAL_PADS; obj++) {
                    float yma = 1. - 0;//1.- (0.17 * gId);
                    float ymi = yma - 1;//0.15;
                    pad[gId][obj] = new TPad(Form("p_%i_%i", obj, gId),Form("p_%i_%i", obj, gId),
                        xmi[obj], ymi, xma[obj], yma);
                    pad[gId][obj]->Draw();
                }
            }

            const int kGains       = 3;
            const int gainValues[3] = {12, 6, 1};
            const int kSides       = 2;
            const int kBarlRings   = EBDetId::MAX_IETA;
            const int kBarlWedges  = EBDetId::MAX_IPHI;
            const int kEndcWedgesX = EEDetId::IX_MAX;
            const int kEndcWedgesY = EEDetId::IY_MAX;

            TH2F** barrel_m = new TH2F*[3];
            TH2F** endc_p_m = new TH2F*[3];
            TH2F** endc_m_m = new TH2F*[3];
            std::string variableName = "DQMChannelStatus";
            for (int gainId = 0; gainId < kGains; gainId++) {
                barrel_m[gainId] = new TH2F(Form((variableName + " EBm%i").c_str(),gainId), Form((variableName + " EB").c_str(),gainValues[gainId]),360,0,360, 170, -85,85);
                endc_p_m[gainId] = new TH2F(Form((variableName + " EE+m%i").c_str(),gainId), Form((variableName + " EE+").c_str(),gainValues[gainId]),100,1,101,100,1,101);
                endc_m_m[gainId] = new TH2F(Form((variableName + " EE-m%i").c_str(),gainId), Form((variableName + " EE-").c_str(),gainValues[gainId]),100,1,101,100,1,101);
            }

            for (int sign=0; sign < kSides; sign++) {
                int thesign = sign==1 ? 1:-1;

                for (int ieta=0; ieta<kBarlRings; ieta++) {
                    for (int iphi=0; iphi<kBarlWedges; iphi++) {
                        EBDetId id((ieta+1)*thesign, iphi+1);
                        float y = -1 - ieta;
                        if(sign == 1) y = ieta;
                        barrel_m[0]->Fill(iphi, y, object()[id.rawId()].getStatusCode());
                    }  // iphi
                }   // ieta

                for (int ix=0; ix<kEndcWedgesX; ix++) {
                    for (int iy=0; iy<kEndcWedgesY; iy++) {
                        if (! EEDetId::validDetId(ix+1,iy+1,thesign)) continue;
                        EEDetId id(ix+1,iy+1,thesign);
                        if (thesign==1) {
                            endc_p_m[0]->Fill(ix+1,iy+1,object()[id.rawId()].getStatusCode());
                        }
                        else{ 
                            endc_m_m[0]->Fill(ix+1,iy+1,object()[id.rawId()].getStatusCode());
                        }
                    }  // iy
                }   // ix
            }    // side

            //canvas.cd(1);
            TLine* l = new TLine(0., 0., 0., 0.);
            l->SetLineWidth(1);
            int ixSectorsEE[202] = {
                62, 62, 61, 61, 60, 60, 59, 59, 58, 58, 56, 56, 46, 46, 44, 44, 43, 43, 42, 42, 
                41, 41, 40, 40, 41, 41, 42, 42, 43, 43, 44, 44, 46, 46, 56, 56, 58, 58, 59, 59, 
                60, 60, 61, 61, 62, 62,  0,101,101, 98, 98, 96, 96, 93, 93, 88, 88, 86, 86, 81, 
                81, 76, 76, 66, 66, 61, 61, 41, 41, 36, 36, 26, 26, 21, 21, 16, 16, 14, 14,  9,
                9,  6,  6,  4,  4,  1,  1,  4,  4,  6,  6,  9,  9, 14, 14, 16, 16, 21, 21, 26, 
                26, 36, 36, 41, 41, 61, 61, 66, 66, 76, 76, 81, 81, 86, 86, 88, 88, 93, 93, 96, 
                96, 98, 98,101,101,  0, 62, 66, 66, 71, 71, 81, 81, 91, 91, 93,  0, 62, 66, 66, 
                91, 91, 98,  0, 58, 61, 61, 66, 66, 71, 71, 76, 76, 81, 81,  0, 51, 51,  0, 44, 
                41, 41, 36, 36, 31, 31, 26, 26, 21, 21,  0, 40, 36, 36, 11, 11,  4,  0, 40, 36, 
                36, 31, 31, 21, 21, 11, 11,  9,  0, 46, 46, 41, 41, 36, 36,  0, 56, 56, 61, 61, 66, 66};

                int iySectorsEE[202] = {
                    51, 56, 56, 58, 58, 59, 59, 60, 60, 61, 61, 62, 62, 61, 61, 60, 60, 59, 59, 58, 
                    58, 56, 56, 46, 46, 44, 44, 43, 43, 42, 42, 41, 41, 40, 40, 41, 41, 42, 42, 43, 
                    43, 44, 44, 46, 46, 51,  0, 51, 61, 61, 66, 66, 76, 76, 81, 81, 86, 86, 88, 88, 
                    93, 93, 96, 96, 98, 98,101,101, 98, 98, 96, 96, 93, 93, 88, 88, 86, 86, 81, 81, 
                    76, 76, 66, 66, 61, 61, 41, 41, 36, 36, 26, 26, 21, 21, 16, 16, 14, 14,  9,  9, 
                    6,  6,  4,  4,  1,  1,  4,  4,  6,  6,  9,  9, 14, 14, 16, 16, 21, 21, 26, 26, 
                    36, 36, 41, 41, 51,  0, 46, 46, 41, 41, 36, 36, 31, 31, 26, 26,  0, 51, 51, 56, 
                    56, 61, 61,  0, 61, 61, 66, 66, 71, 71, 76, 76, 86, 86, 88,  0, 62,101,  0, 61, 
                    61, 66, 66, 71, 71, 76, 76, 86, 86, 88,  0, 51, 51, 56, 56, 61, 61,  0, 46, 46, 
                    41, 41, 36, 36, 31, 31, 26, 26,  0, 40, 31, 31, 16, 16,  6,  0, 40, 31, 31, 16, 16,  6};

                    for (int gId = 0; gId < TOTAL_IMAGES; gId++) {//was 3
                        pad[gId][0]->cd();
                        endc_m_m[gId]->SetStats(0);
                        //endc_m_m[gId]->SetMaximum(225);
                        //endc_m_m[gId]->SetMinimum(175);
                        endc_m_m[gId]->Draw("colz");
                        for ( int i=0; i<201; i=i+1) {
                            if ( (ixSectorsEE[i]!=0 || iySectorsEE[i]!=0) && 
                                (ixSectorsEE[i+1]!=0 || iySectorsEE[i+1]!=0) ) {
                                    l->DrawLine(ixSectorsEE[i], iySectorsEE[i], 
                                        ixSectorsEE[i+1], iySectorsEE[i+1]);
                                    l->SetLineWidth(0.2);
                            }
                        }

                        //canvas.cd(2);
                        pad[gId][1]->cd();
                        barrel_m[gId]->SetStats(0);
                        //barrel_m[gId]->SetMaximum(225);
                        //barrel_m[gId]->SetMinimum(175);
                        barrel_m[gId]->Draw("colz");
                        for(int i = 0; i <17; i++) {
                            Double_t x = 20.+ (i *20);
                            l = new TLine(x,-85.,x,86.);
                            l->Draw();
                        }
                        l = new TLine(0.,0.,360.,0.);
                        l->Draw();

                        //canvas.cd(3);
                        pad[gId][2]->cd();
                        endc_p_m[gId]->SetStats(0);
                        //endc_p_m[gId]->SetMaximum(225);
                        //endc_p_m[gId]->SetMinimum(175);
                        endc_p_m[gId]->Draw("colz");
                        for ( int i=0; i<201; i=i+1) {
                            if ( (ixSectorsEE[i]!=0 || iySectorsEE[i]!=0) && 
                                (ixSectorsEE[i+1]!=0 || iySectorsEE[i+1]!=0) ) {
                                    l->DrawLine(ixSectorsEE[i], iySectorsEE[i], 
                                        ixSectorsEE[i+1], iySectorsEE[i+1]);
                            }
                        }
                    }

                    canvas.SaveAs(filename.c_str());
                    return filename;
    }  // plot


}


namespace condPython {
    template<>
    void defineWhat<EcalDQMChannelStatus>() {
        enum_<cond::ecalcond::How>("How")
            .value("singleChannel",cond::ecalcond::singleChannel)
            .value("bySuperModule",cond::ecalcond::bySuperModule) 
            .value("all",cond::ecalcond::all)
            ;

        typedef cond::ExtractWhat<EcalDQMChannelStatus> What;
        class_<What>("What",init<>())
            .def("set_how",&What::set_how)
            .def("set_which",&What::set_which)
            .def("how",&What::how, return_value_policy<copy_const_reference>())
            .def("which",&What::which, return_value_policy<copy_const_reference>())
            ;
    }
}

PYTHON_WRAPPER(EcalDQMChannelStatus,EcalDQMChannelStatus);