/data/git/CMSSW_5_3_11_patch5/src/CondCore/BeamSpotPlugins/plugins/BeamSpotObjectsPyWrapper.cc

Go to the documentation of this file.

#include "CondFormats/BeamSpotObjects/interface/BeamSpotObjects.h"

#include "CondCore/DBCommon/interface/DbConnection.h"
#include "CondCore/DBCommon/interface/DbTransaction.h"
#include "CondCore/Utilities/interface/PayLoadInspector.h"
#include "CondCore/Utilities/interface/InspectorPythonWrapper.h"

#include "TROOT.h"
#include "TCanvas.h"
#include "TStyle.h"
#include "TColor.h"
#include "TLine.h"
#include "TGraph.h"
#include "TAxis.h"
#include "TMultiGraph.h"
#include "TLegend.h"

#include <string>
#include <sstream>

namespace cond {

        template<>
        class ValueExtractor<BeamSpotObjects>: public  BaseValueExtractor<BeamSpotObjects> {
        public:
                typedef BeamSpotObjects 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<BeamSpotObjects>::summary() const {
                std::stringstream ss;
                object().print(ss);
                return ss.str();
        }

        template<>
        std::string PayLoadInspector<BeamSpotObjects>::plot(std::string const & filename,
                std::string const &,
                std::vector<int> const &,
                std::vector<float> const& ) const 
        {

                TCanvas canvas("iC","iC",800,800);    

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

        template<>
        std::string PayLoadInspector<BeamSpotObjects>::trend_plot(std::string const & filename,
                std::string const & opt_string,
                std::vector<int> const& ints,
                std::vector<float> const & floats,
                std::vector<std::string> const& strings) const 
        {
                std::stringstream ss("");

                if (strings.size() < 2)
                        return ("Error! Not enough data for initializing connection for making plots! (from \
                                        template<>\
                                        std::string PayLoadInspector<BeamSpotObjects>::trend_plot)");

                std::vector<std::string>::const_iterator iter_beg = strings.begin();

                std::string conString= (*iter_beg); 
                ++iter_beg;
                std::string authPath = (*iter_beg);
                ++iter_beg;

                //make connection object
                DbConnection dbConn;

                //set in configuration object authentication path
                dbConn.configuration().setAuthenticationPath(authPath);
                dbConn.configure();

                //create session object from connection
                DbSession dbSes=dbConn.createSession();

                //try to make connection
                dbSes.open(conString,true);

                //start a transaction (true=readOnly)
                dbSes.transaction().start(true);

                //get the actual object
                boost::shared_ptr<BeamSpotObjects> ptrBeamSpot;

                //iter_beg now stands on first token in the vector
                std::string token;

                std::vector<float> vecX;
                std::vector<float> vecY;
                std::vector<float> vecZ;
                for (std::vector<float>::const_iterator iter_float = floats.begin() ;iter_beg != strings.end();++iter_beg, ++iter_float){
                        token = (*iter_beg);
                        //std::cout << token << " ";
                        
                        ptrBeamSpot = dbSes.getTypedObject<BeamSpotObjects>(token);

                        //get data from the objects:
                        vecX.push_back(ptrBeamSpot->GetX());
                        vecY.push_back(ptrBeamSpot->GetY());
                        vecZ.push_back(ptrBeamSpot->GetZ());
                        std::cout << "since: "<<(*iter_float)<< " X: "<< ptrBeamSpot->GetX() << " Y: "<< ptrBeamSpot->GetY() << " Z: " << ptrBeamSpot->GetZ() << std::endl;


                }
                
                //close db session
                dbSes.close();

                TCanvas canvas("iC","iC",1200,1200); 
                //canvas.UseCurrentStyle();
                //gStyle->SetPalette(1);
                std::cout << *(floats.end() -1) << "   " << *(floats.begin());
                float max = *(floats.end() -1);
                float min =  *(floats.begin());

                unsigned int lineWidth = 2;
                unsigned int startColor = 2;

                float result = ((max - min) / max) ;
                if ( result >= 0.1 ){
                        canvas.SetLogx(1);
                }
                std::cout << "result: " << result << std::endl;

        
        //1)
                TGraph graphdataX(vecX.size(), static_cast<const float *>(&floats[0]), static_cast<const float *>(&vecX[0]));
                //graphdataX.GetXaxis()->SetRangeUser(*(floats.begin()),*(floats.end() -1));
                //graphdataX.GetXaxis()-> SetLabelSize(0.03);
                graphdataX.SetLineColor(startColor++);
                graphdataX.SetLineWidth(lineWidth);

        //2)
                TGraph graphdataY(vecY.size(), static_cast<const float *>(&floats[0]), static_cast<const float *>(&vecY[0]));
                //graphdataY.GetXaxis()->SetRangeUser(*(floats.begin()),*(floats.end() -1));
                //graphdataY.GetXaxis()-> SetLabelSize(0.03);
                graphdataY.SetLineColor(startColor++);
                graphdataY.SetLineWidth(lineWidth);

        //3)
                TGraph graphdataZ(vecZ.size(), static_cast<const float *>(&floats[0]), static_cast<const float *>(&vecZ[0]));
                //graphdataZ.GetXaxis()->SetRangeUser(*(floats.begin()),*(floats.end() -1));
                //graphdataZ.GetXaxis()-> SetLabelSize(0.03);
                graphdataZ.SetLineColor(startColor++);
                graphdataZ.SetLineWidth(lineWidth);


                TMultiGraph mg;
                std::stringstream plotTitle;
                plotTitle.precision(0);
                plotTitle << std::fixed;
                //plotTitle << "BeamSpot trend graph. X0 = black, Y0 = red, Z0 = green; since(first="
                plotTitle << "BeamSpot trend graph (first="
                                        <<(double)min 
                        << ", last=" 
                        << (double) max 
                        << ", total="
                        << floats.size()
                        <<");since;value";
                mg.SetTitle( plotTitle.str().c_str());

                //graphdataX.GetXaxis()->SetBinLabel(2,"mylabel");
                //graphdataY.GetXaxis()->SetBinLabel(2,"mylabel");
                //graphdataZ.GetXaxis()->SetBinLabel(2,"mylabel");

                mg.Add(&graphdataX);
                mg.Add(&graphdataY);
                mg.Add(&graphdataZ);
                

                //float size1 = mg.GetXaxis()-> GetBinWidth();
                //std::cout <<" BinWidth: " << size1 << std::endl;

                mg.Draw("LA*");

                TLegend leg(0.7,0.7,0.9,0.9,"Beam Spot Legend");
                leg.AddEntry(&graphdataX,"X_{0}","lpf");
                leg.AddEntry(&graphdataY,"Y_{0}","lpf");
                leg.AddEntry(&graphdataZ,"Z_{0}","lpf");

                leg.Draw();

                ss.str("");
                ss << filename << ".png";
                canvas.SaveAs((ss.str()).c_str());
                return ss.str();
        }
}


PYTHON_WRAPPER(BeamSpotObjects,BeamSpotObjects);