CorrelatedSpaceTree2.h

Go to the documentation of this file.

/***************************************************************************
 *   Copyright (C) 2004 by Olivier Stezowski                               *
 *   Copyright (C) 2010 by Christian Finck                                 * 
 *   stezow(AT)ipnl.in2p3.fr                                               *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/

#ifndef Gw_CorrelatedSpaceTree2
#define Gw_CorrelatedSpaceTree2

#include "GwLogMessage.h"

#ifndef Gw_SymCorrelatedSpace
#include "SymCorrelatedSpace.h"
#endif

#ifndef ROOT_TObject
#include "TObject.h"
#endif

#ifndef ROOT_TString
#include "TString.h"
#endif

class TTree;
class TFile;
class TH2;
class TH1;
class TBranch;

namespace Gw {
        
class CSFilter;

class CorrelatedSpaceTree2 : public SymCorrelatedSpace
{

protected:
  
  TTree* fTree;               
  TString fBranchName;        // name of branch where the matrix is stored  
  
  mutable LogMessage fLog;    

public:
  CorrelatedSpaceTree2(const char* name = CorrelatedSpace::GetDefaultName(), const char* title = "Tree");
  virtual ~CorrelatedSpaceTree2();
  
  virtual TBranch* SetAddress() = 0;
  
  virtual void SetBranch() = 0;
  
  virtual void FillHisto(Int_t j, TH1* h) = 0;
  
  virtual void FillCounts(Int_t i, Int_t j, TH2* h2) = 0;
  
  virtual void FillCounts(Int_t i, Int_t j, Int_t dimX, const Double_t* Xn, const Double_t weight = 1.) = 0;
  
  virtual void SetCounts(Double_t weight) = 0;
  
  virtual Double_t GetCounts() = 0;
  
  virtual void FillCounts(Int_t j) = 0;
  
  virtual void CheckH2InTree(TH2* h2) = 0;
    
  void Fill(const Double_t* Xn, const Double_t weight = 1.);
  
  void FillRandom(Int_t dim);
  
  void FillFromH2(TH2* h2);
  
  void Project(CSFilter* filter, Option_t* axis = "X");

  virtual void Project(Option_t* axis, Option_t* gateName = "Gate0");
  
   virtual void Browse(TBrowser* b);
   
  void SetBranchName(TString name = "mat1") { fBranchName = name; }
  
  const Char_t* GetBranchName() { return fBranchName.Data(); }
  
  void AssertTree();
  
  const Char_t* GetTreeName() const;
  
  const Char_t* GetTreeTitle() const;

  void DeleteTree(Option_t* opt = "");
  
  Int_t GetNbin(Int_t axis) const;
  
  void SetCellContent(Int_t iCell, Double_t weight);
  
  Double_t GetContent(Int_t i, Int_t j);
  
  Double_t GetCellContent(Int_t iCell);  
  
  ClassDef(CorrelatedSpaceTree2,1)  // Correlated Space Tree in 2 dim
};
        

  
} // end namespace

#endif