/data/refman/pasoursint/CMSSW_6_1_2_SLHC2/src/DataFormats/ParticleFlowReco/interface/CaloWindow.h

Go to the documentation of this file.

#ifndef CALOWINDOW_H_
#define CALOWINDOW_H_

//#include <boost/shared_ptr.hpp>

//#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include <vector>
#include <iostream>
#include <map>
namespace pftools {

class CaloRing {
public:

        CaloRing();

        CaloRing(unsigned nPanes);

        void reset();

        double getEnergy(unsigned pane) const;

        bool setEnergy(unsigned pane, double energy);

        bool addEnergy(unsigned pane, double energy);

        double totalE() const;

        std::vector<double> getEnergies() const {
                return myPanes_;
        }

        unsigned getNPanes() const {
                return panes_;
        }

        void printEnergies(std::ostream& s, double range = 1.0);


private:
        unsigned panes_;
        std::vector<double> myPanes_;

        friend std::ostream& operator<<(std::ostream& s, const CaloRing& caloRing);

};

class CaloWindow {
public:
        /* Default constructor - do not use (this has to be here for Reflex to work?)*/
        CaloWindow();

        CaloWindow(double eta, double phi, unsigned nRings, double deltaR, unsigned nPanes, double axis = 0.0);

        void init(double eta, double phi, unsigned nRings, double deltaR, unsigned nPanes, double axis = 0.0);

        bool addHit(double eta, double phi, double energy);

        std::map<unsigned, CaloRing> getRingDepositions() const {
                return energies_;
        }

        std::map<unsigned, double> getRingEnergySummations() const;


        virtual ~CaloWindow();

        void reset();

        void printEnergies(std::ostream& s, double range = 1.0);

        std::vector<double> stream(double normalisation = 1.0) const;

private:
        //Where is the barycentre of this window?
        double baryEta_;
        double baryPhi_;
        //How many rings is it composed of?
        unsigned nRings_;
        //What is the deltaR separation of the rings?
        double deltaR_;
        //How many panels will each calo ring be composed of? (central panel = 1)
        unsigned nPanes_;

        //Angle in radians fromm which we start counting rings (offset from zero)
        double axis_;

        /*
         * Determines which window pane to put the hit in.
         *
         */
        std::pair<unsigned, unsigned> relativePosition(double eta, double phi) const;

        //std::vector<boost::shared_ptr<std::vector<double> > > energies_;
        std::map<unsigned, CaloRing> energies_;


        friend std::ostream& operator<<(std::ostream& s,
                        const CaloWindow& caloWindow);

};

class TestCaloWindow {
public:
        TestCaloWindow() {
        }

        virtual ~TestCaloWindow() {
        }

        void doTest();
};

//typedef boost::shared_ptr<CaloWindow> CalowWindowPtr;

}
#endif /* CALOWINDOW_H_ */