/[PAMELA software]/PamCut/CollectionActions/Histo2DActions/Histo2DAction/Histo2DAction.h
ViewVC logotype

Diff of /PamCut/CollectionActions/Histo2DActions/Histo2DAction/Histo2DAction.h

Parent Directory Parent Directory | Revision Log Revision Log | View Patch Patch

revision 1.1 by pam-fi, Fri Sep 25 15:36:45 2009 UTC revision 1.6 by pam-fi, Thu Mar 11 19:14:03 2010 UTC
# Line 106  public: Line 106  public:
106     */     */
107    void SetYAxis(TString label, unsigned int nBins, float min, float max, bool logBinning = false);    void SetYAxis(TString label, unsigned int nBins, float min, float max, bool logBinning = false);
108    
109      /*! @brief Sets the ROOT histogram's title.
110       *
111       * @param title The histogram title as it will appear on the histogram itself.
112       */
113      void SetTitle(TString &title) {
114        _title = title;
115      }
116    /*! @brief Sets up the histogram    /*! @brief Sets up the histogram
117     *     *
118     * This routine effectively prepares the histogram, after the desired parameters has been set by #SetXAxis() and #SetYAxis().     * This routine effectively prepares the histogram, after the desired parameters has been set by #SetXAxis() and #SetYAxis().
119     *     *
120     * @param events Pointer to PamLevel2 events (unused).     * @param events Pointer to PamLevel2 events (unused).
121     */     */
   
   /*! @brief Sets the ROOT histogram's title. */  
   void SetTitle(TString &title){  
     _title = title;  
   }  
122    void Setup(PamLevel2 *events) {    void Setup(PamLevel2 *events) {
123      CollectionAction::Setup(events);      CollectionAction::Setup(events);
124      _InitHistos();      _InitHistos();
# Line 141  public: Line 143  public:
143      return _histo;      return _histo;
144    }    }
145    
146    /*! Fills the ROOT and the vector histogram. */    /*! @brief Returns a pointer to the ROOT histogram.
147       *
148       * @return A pointer to the root histogram
149       */
150      TH2 *GetRootHisto() {
151        return _rootHisto;
152      }
153    
154      /*! Fills the ROOT and the vector histogram.
155       *
156       * @param xValue The value of the X coordinate associated to the event.
157       * @param yValue The value of the Y coordinate associated to the event.
158       * @param weight The weight which will be applied to the event.
159       */
160    void Fill(double xValue, double yValue, double weight = 1.);    void Fill(double xValue, double yValue, double weight = 1.);
161    
162    /*! @brief Gets the X overflow histogram.    /*! @brief Gets the X overflow histogram.
# Line 236  public: Line 251  public:
251    
252  protected:  protected:
253    
254      /*! @brief The vector containing the limits of the X bins(from lower to higher). */
255    std::vector<float> _xBins;    std::vector<float> _xBins;
256      /*! @brief The vector containing the limits of the Y bins(from lower to higher). */
257    std::vector<float> _yBins;    std::vector<float> _yBins;
258      /*! @brief A matrix containing the value of the histogram for each X-Y bin. */
259    SimpleMatrix<HistoType> _histo;    SimpleMatrix<HistoType> _histo;
260      /*! @brief The ROOT histogram. */
261    TH2 *_rootHisto;    TH2 *_rootHisto;
262    
263      TString _outFileBase;
264      TString _mode;
265      TString _title, _xLabel, _yLabel;
266    
267  private:  private:
268    
269    vector<HistoType> _xUnderflow, _xOverflow, _yUnderflow, _yOverflow;    vector<HistoType> _xUnderflow, _xOverflow, _yUnderflow, _yOverflow;
270    HistoType _xUnderYUnderflow, _xOverYOverflow, _xUnderYOverflow, _xOverYUnderflow;    HistoType _xUnderYUnderflow, _xOverYOverflow, _xUnderYOverflow, _xOverYUnderflow;
   TString _outFileBase;  
   TString _mode;  
   TString _title, _xLabel, _yLabel;  
271    bool _outRoot;    bool _outRoot;
272    bool _outText;    bool _outText;
   
273    void _CreateHisto();    void _CreateHisto();
274    void _InitHistos();    void _InitHistos();
275  };  };
# Line 262  void Histo2DAction<HistoType>::_CreateHi Line 281  void Histo2DAction<HistoType>::_CreateHi
281    _rootHisto = NULL;    _rootHisto = NULL;
282  }  }
283    
284    // Specializations for _CreateHistos(). See Histo2DAction.cpp
285    template<>
286    void Histo2DAction<Int_t>::_CreateHisto();
287    
288    template<>
289    void Histo2DAction<Float_t>::_CreateHisto();
290    
291    template<>
292    void Histo2DAction<Double_t>::_CreateHisto();
293    
294  template<class HistoType>  template<class HistoType>
295  void Histo2DAction<HistoType>::_InitHistos() {  void Histo2DAction<HistoType>::_InitHistos() {
296    
297    _CreateHisto();    _CreateHisto();
298      if (_xBins.size() < 2) // SetXAxis not called by the main program, or wrongly filled (only 1 bin limit)
299    if (_xBins.size() < 2){      SetXAxis("Default X", 10, 0., 1.);
300      _xBins.resize(2);    if (_yBins.size() < 2) // SetYAxis not called by the main program, or wrongly filled (only 1 bin limit)
301      _xBins[0] = 0.;      SetYAxis("Default Y", 10, 0., 1.);
     _xBins[1] = 1.;  
   }  
   
   if (_yBins.size() < 2){  
     _yBins.resize(2);  
     _yBins[0] = 0.;  
     _yBins[1] = 1.;  
   }  
302    
303    if (_rootHisto) {    if (_rootHisto) {
304      Double_t *auxXArray = new Double_t[_xBins.size()];      Double_t *auxXArray = new Double_t[_xBins.size()];
# Line 305  void Histo2DAction<HistoType>::_InitHist Line 326  void Histo2DAction<HistoType>::_InitHist
326    /* The row index (first) corresponds to the position on the vertical (Y) axis. */    /* The row index (first) corresponds to the position on the vertical (Y) axis. */
327    _histo.Resize(_yBins.size() - 1, _xBins.size() - 1);    _histo.Resize(_yBins.size() - 1, _xBins.size() - 1);
328    
329    _xUnderflow.resize(_yBins.size());    _xUnderflow.resize(_yBins.size() - 1);
330    _xOverflow.resize(_yBins.size());    _xOverflow.resize(_yBins.size() - 1);
331    _yUnderflow.resize(_xBins.size());    _yUnderflow.resize(_xBins.size() - 1);
332    _yOverflow.resize(_xBins.size());    _yOverflow.resize(_xBins.size() - 1);
333    
334  }  }
335    
336  template<class HistoType>  template<class HistoType>
337  Histo2DAction<HistoType>::~Histo2DAction() {  Histo2DAction<HistoType>::~Histo2DAction() {
338    
339      delete _rootHisto;
340      _rootHisto = NULL;
341  }  }
342    
343  template<class HistoType>  template<class HistoType>
344  Histo2DAction<HistoType>::Histo2DAction(const char *actionName, TString title, TString outFileBase, TString mode,  Histo2DAction<HistoType>::Histo2DAction(const char *actionName, TString title, TString outFileBase, TString mode,
345      bool outRoot, bool outText) :      bool outRoot, bool outText) :
346    CollectionAction(actionName), _xBins(0), _yBins(0), _histo(0, 0), _rootHisto(NULL), _xUnderflow(0), _xOverflow(0),    CollectionAction(actionName), _xBins(0), _yBins(0), _histo(0, 0), _rootHisto(NULL), _outFileBase(outFileBase), _mode(
347        _yUnderflow(0), _yOverflow(0), _outFileBase(outFileBase), _mode(mode), _title(title), _xLabel(""), _yLabel(""),        mode), _title(title), _xLabel(""), _yLabel(""), _xUnderflow(0), _xOverflow(0), _yUnderflow(0), _yOverflow(0),
348        _outRoot(outRoot), _outText(outText) {        _xUnderYUnderflow((HistoType) 0), _xOverYOverflow((HistoType) 0), _xUnderYOverflow((HistoType) 0),
349          _xOverYUnderflow((HistoType) 0), _outRoot(outRoot), _outText(outText) {
350    
351  }  }
352    
# Line 444  void Histo2DAction<HistoType>::SetYAxis( Line 469  void Histo2DAction<HistoType>::SetYAxis(
469  }  }
470    
471  template<class HistoType>  template<class HistoType>
472  inline void Histo2DAction<HistoType>::Fill(double xValue, double yValue, double weight) {  void Histo2DAction<HistoType>::Fill(double xValue, double yValue, double weight) {
473    
474    _rootHisto->Fill(xValue, yValue, weight);    _rootHisto->Fill(xValue, yValue, weight);
475    
# Line 457  inline void Histo2DAction<HistoType>::Fi Line 482  inline void Histo2DAction<HistoType>::Fi
482      UOflow = true;      UOflow = true;
483    }    }
484    
485    if (xValue > _xBins.back()) {    if (xValue >= _xBins.back()) {
486      xBin = _xBins.size();      xBin = _xBins.size();
487      UOflow = true;      UOflow = true;
488    }    }
# Line 467  inline void Histo2DAction<HistoType>::Fi Line 492  inline void Histo2DAction<HistoType>::Fi
492      UOflow = true;      UOflow = true;
493    }    }
494    
495    if (yValue > _yBins.back()) {    if (yValue >= _yBins.back()) {
496      yBin = _yBins.size();      yBin = _yBins.size();
497      UOflow = true;      UOflow = true;
498    }    }
# Line 501  inline void Histo2DAction<HistoType>::Fi Line 526  inline void Histo2DAction<HistoType>::Fi
526          return;          return;
527        }        }
528        else {        else {
529          if (yBin == (int)_yBins.size()) {          if (yBin == (int) _yBins.size()) {
530            _xUnderYOverflow += (HistoType) weight;            _xUnderYOverflow += (HistoType) weight;
531              cout << "_xUnderYOverflow: " << GetXUnderYOverflow() << ", weight: " << weight << ", (HistoType) weight: "
532                  << (HistoType) weight << endl;
533            return;            return;
534          }          }
535          else {          else {
# Line 512  inline void Histo2DAction<HistoType>::Fi Line 539  inline void Histo2DAction<HistoType>::Fi
539        }        }
540      }      }
541    
542      if (xBin == (int)_xBins.size()) {      if (xBin == (int) _xBins.size()) {
543        if (yBin == -1) {        if (yBin == -1) {
544          _xOverYUnderflow += (HistoType) weight;          _xOverYUnderflow += (HistoType) weight;
545          return;          return;
546        }        }
547        else {        else {
548          if (yBin ==(int) _yBins.size()) {          if (yBin == (int) _yBins.size()) {
549            _xOverYOverflow += (HistoType) weight;            _xOverYOverflow += (HistoType) weight;
550            return;            return;
551          }          }
# Line 534  inline void Histo2DAction<HistoType>::Fi Line 561  inline void Histo2DAction<HistoType>::Fi
561        return;        return;
562      }      }
563    
564      if (yBin == (int)_yBins.size()) {      if (yBin == (int) _yBins.size()) {
565        _yOverflow[xBin] += (HistoType) weight;        _yOverflow[xBin] += (HistoType) weight;
566        return;        return;
567      }      }
# Line 559  void Histo2DAction<HistoType>::Finalize( Line 586  void Histo2DAction<HistoType>::Finalize(
586        ofstream outTextFile((_outFileBase + ".txt").Data(), ios_base::out);        ofstream outTextFile((_outFileBase + ".txt").Data(), ios_base::out);
587        for (unsigned int i = 0; i < _histo.GetNRows(); i++) {        for (unsigned int i = 0; i < _histo.GetNRows(); i++) {
588          for (unsigned int j = 0; j < _histo.GetNCols(); j++) {          for (unsigned int j = 0; j < _histo.GetNCols(); j++) {
589            outTextFile << _histo[i][j] << "  ";            outTextFile << setw(7) << _histo[i][j] << "  ";
590          }          }
591          outTextFile << "\n";          outTextFile << "\n";
592        }        }
       outTextFile << endl;  
593        outTextFile.close();        outTextFile.close();
594      }      }
595    }    }
Legend:
Removed from v.1.1  
changed lines
  Added in v.1.6
  ViewVC Help
Powered by ViewVC 1.1.23