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_ */