/[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.2 by pam-fi, Tue Oct 27 10:17:48 2009 UTC revision 1.8 by pam-fi, Thu Aug 12 14:37:03 2010 UTC
# Line 143  public: Line 143  public:
143      return _histo;      return _histo;
144    }    }
145    
146      /*! @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.    /*! Fills the ROOT and the vector histogram.
155     *     *
156     * @param value The value of the X coordinate associated to the event.     * @param xValue The value of the X coordinate associated to the event.
157     * @param value The value of the Y coordinate associated to the event.     * @param yValue The value of the Y coordinate associated to the event.
158     * @param weight The weight which will be applied to the event.     * @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.);
# Line 252  protected: Line 260  protected:
260    /*! @brief The ROOT histogram. */    /*! @brief The ROOT histogram. */
261    TH2 *_rootHisto;    TH2 *_rootHisto;
262    
263      /*! @brief Base name of the output file. */
264      TString _outFileBase;
265      /*! @brief Output file open mode (UPDATE or RECREATE, see documentation of TFile). */
266      TString _mode;
267      /*! @brief Title for the ROOT histogram. */
268      TString _title;
269      /*! @brief X axis label for the ROOT histogram. */
270      TString _xLabel;
271      /*! @brief Y axis label for the ROOT histogram. */
272      TString _yLabel;
273    
274  private:  private:
275    
276    vector<HistoType> _xUnderflow, _xOverflow, _yUnderflow, _yOverflow;    vector<HistoType> _xUnderflow, _xOverflow, _yUnderflow, _yOverflow;
277    HistoType _xUnderYUnderflow, _xOverYOverflow, _xUnderYOverflow, _xOverYUnderflow;    HistoType _xUnderYUnderflow, _xOverYOverflow, _xUnderYOverflow, _xOverYUnderflow;
   TString _outFileBase;  
   TString _mode;  
   TString _title, _xLabel, _yLabel;  
278    bool _outRoot;    bool _outRoot;
279    bool _outText;    bool _outText;
   
280    void _CreateHisto();    void _CreateHisto();
281    void _InitHistos();    void _InitHistos();
282  };  };
# Line 273  void Histo2DAction<HistoType>::_CreateHi Line 288  void Histo2DAction<HistoType>::_CreateHi
288    _rootHisto = NULL;    _rootHisto = NULL;
289  }  }
290    
291    // Specializations for _CreateHistos(). See Histo2DAction.cpp
292    template<>
293    void Histo2DAction<Int_t>::_CreateHisto();
294    
295    template<>
296    void Histo2DAction<Float_t>::_CreateHisto();
297    
298    template<>
299    void Histo2DAction<Double_t>::_CreateHisto();
300    
301  template<class HistoType>  template<class HistoType>
302  void Histo2DAction<HistoType>::_InitHistos() {  void Histo2DAction<HistoType>::_InitHistos() {
303    
304    _CreateHisto();    _CreateHisto();
305      if (_xBins.size() < 2) // SetXAxis not called by the main program, or wrongly filled (only 1 bin limit)
306    if (_xBins.size() < 2) {      SetXAxis("Default X", 10, 0., 1.);
307      _xBins.resize(2);    if (_yBins.size() < 2) // SetYAxis not called by the main program, or wrongly filled (only 1 bin limit)
308      _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.;  
   }  
309    
310    if (_rootHisto) {    if (_rootHisto) {
311      Double_t *auxXArray = new Double_t[_xBins.size()];      Double_t *auxXArray = new Double_t[_xBins.size()];
# Line 316  void Histo2DAction<HistoType>::_InitHist Line 333  void Histo2DAction<HistoType>::_InitHist
333    /* 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. */
334    _histo.Resize(_yBins.size() - 1, _xBins.size() - 1);    _histo.Resize(_yBins.size() - 1, _xBins.size() - 1);
335    
336    _xUnderflow.resize(_yBins.size());    _xUnderflow.resize(_yBins.size() - 1);
337    _xOverflow.resize(_yBins.size());    _xOverflow.resize(_yBins.size() - 1);
338    _yUnderflow.resize(_xBins.size());    _yUnderflow.resize(_xBins.size() - 1);
339    _yOverflow.resize(_xBins.size());    _yOverflow.resize(_xBins.size() - 1);
340    
341  }  }
342    
343  template<class HistoType>  template<class HistoType>
344  Histo2DAction<HistoType>::~Histo2DAction() {  Histo2DAction<HistoType>::~Histo2DAction() {
345    
346      delete _rootHisto;
347      _rootHisto = NULL;
348  }  }
349    
350  template<class HistoType>  template<class HistoType>
351  Histo2DAction<HistoType>::Histo2DAction(const char *actionName, TString title, TString outFileBase, TString mode,  Histo2DAction<HistoType>::Histo2DAction(const char *actionName, TString title, TString outFileBase, TString mode,
352      bool outRoot, bool outText) :      bool outRoot, bool outText) :
353    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(
354        _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),
355        _outRoot(outRoot), _outText(outText) {        _xUnderYUnderflow((HistoType) 0), _xOverYOverflow((HistoType) 0), _xUnderYOverflow((HistoType) 0),
356          _xOverYUnderflow((HistoType) 0), _outRoot(outRoot), _outText(outText) {
357    
358  }  }
359    
# Line 455  void Histo2DAction<HistoType>::SetYAxis( Line 476  void Histo2DAction<HistoType>::SetYAxis(
476  }  }
477    
478  template<class HistoType>  template<class HistoType>
479  inline void Histo2DAction<HistoType>::Fill(double xValue, double yValue, double weight) {  void Histo2DAction<HistoType>::Fill(double xValue, double yValue, double weight) {
480    
481    _rootHisto->Fill(xValue, yValue, weight);    _rootHisto->Fill(xValue, yValue, weight);
482    
# Line 468  inline void Histo2DAction<HistoType>::Fi Line 489  inline void Histo2DAction<HistoType>::Fi
489      UOflow = true;      UOflow = true;
490    }    }
491    
492    if (xValue > _xBins.back()) {    if (xValue >= _xBins.back()) {
493      xBin = _xBins.size();      xBin = _xBins.size();
494      UOflow = true;      UOflow = true;
495    }    }
# Line 478  inline void Histo2DAction<HistoType>::Fi Line 499  inline void Histo2DAction<HistoType>::Fi
499      UOflow = true;      UOflow = true;
500    }    }
501    
502    if (yValue > _yBins.back()) {    if (yValue >= _yBins.back()) {
503      yBin = _yBins.size();      yBin = _yBins.size();
504      UOflow = true;      UOflow = true;
505    }    }
# Line 570  void Histo2DAction<HistoType>::Finalize( Line 591  void Histo2DAction<HistoType>::Finalize(
591        ofstream outTextFile((_outFileBase + ".txt").Data(), ios_base::out);        ofstream outTextFile((_outFileBase + ".txt").Data(), ios_base::out);
592        for (unsigned int i = 0; i < _histo.GetNRows(); i++) {        for (unsigned int i = 0; i < _histo.GetNRows(); i++) {
593          for (unsigned int j = 0; j < _histo.GetNCols(); j++) {          for (unsigned int j = 0; j < _histo.GetNCols(); j++) {
594            outTextFile << _histo[i][j] << "  ";            outTextFile << setw(7) << _histo[i][j] << "  ";
595          }          }
596          outTextFile << "\n";          outTextFile << "\n";
597        }        }
       outTextFile << endl;  
598        outTextFile.close();        outTextFile.close();
599      }      }
600    }    }
Legend:
Removed from v.1.2  
changed lines
  Added in v.1.8
  ViewVC Help
Powered by ViewVC 1.1.23