/usr/include/root/TMVA/DataInputHandler.h is in libroot-tmva-dev 5.34.19+dfsg-1.2.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 | // @(#)root/tmva $Id$
// Author: Andreas Hoecker, Peter Speckmayer, Joerg Stelzer, Helge Voss
/**********************************************************************************
* Project: TMVA - a Root-integrated toolkit for multivariate data analysis *
* Package: TMVA *
* Class : DataInputHandler *
* Web : http://tmva.sourceforge.net *
* *
* Description: *
* Contains all the data information *
* *
* Authors (alphabetical): *
* Andreas Hoecker <Andreas.Hocker@cern.ch> - CERN, Switzerland *
* Joerg Stelzer <Joerg.Stelzer@cern.ch> - CERN, Switzerland *
* Peter Speckmayer <Peter.Speckmayer@cern.ch> - CERN, Switzerland *
* Helge Voss <Helge.Voss@cern.ch> - MPI-K Heidelberg, Germany *
* *
* Copyright (c) 2006: *
* CERN, Switzerland *
* MPI-K Heidelberg, Germany *
* *
* Redistribution and use in source and binary forms, with or without *
* modification, are permitted according to the terms listed in LICENSE *
* (http://tmva.sourceforge.net/LICENSE) *
**********************************************************************************/
#ifndef ROOT_TMVA_DataInputHandler
#define ROOT_TMVA_DataInputHandler
//////////////////////////////////////////////////////////////////////////
// //
// DataInputHandler //
// //
// Class that contains all the data information //
// //
//////////////////////////////////////////////////////////////////////////
#include <vector>
#include <map>
#include <string>
#include <fstream>
#ifndef ROOT_TTree
#include "TTree.h"
#endif
#ifndef ROOT_TCut
#include "TCut.h"
#endif
#ifndef ROOT_TMVA_Types
#include "TMVA/Types.h"
#endif
namespace TMVA {
class MsgLogger;
class TreeInfo {
public:
TreeInfo( TTree* tr, const TString& className, Double_t weight=1.0, Types::ETreeType tt = Types::kMaxTreeType, Bool_t own=kFALSE )
: fTree(tr), fClassName(className), fWeight(weight), fTreeType(tt), fOwner(own) {}
~TreeInfo() { if (fOwner) delete fTree; }
TTree* GetTree() const { return fTree; }
Double_t GetWeight() const { return fWeight; }
UInt_t GetEntries() const { if( !fTree ) return 0; else return fTree->GetEntries(); }
Types::ETreeType GetTreeType() const { return fTreeType; }
const TString& GetClassName() const { return fClassName; }
private:
TTree* fTree; //! pointer to the tree
TString fClassName;//! name of the class the tree belongs to
Double_t fWeight; //! weight for the tree
Types::ETreeType fTreeType; //! tree is for training/testing/both
Bool_t fOwner; //! true if created from file
};
class DataInputHandler {
public:
DataInputHandler();
~DataInputHandler();
// setters
void AddSignalTree ( TTree* tr, Double_t weight=1.0, Types::ETreeType tt = Types::kMaxTreeType );
void AddBackgroundTree( TTree* tr, Double_t weight=1.0, Types::ETreeType tt = Types::kMaxTreeType );
void AddSignalTree ( const TString& tr, Double_t weight=1.0, Types::ETreeType tt = Types::kMaxTreeType );
void AddBackgroundTree( const TString& tr, Double_t weight=1.0, Types::ETreeType tt = Types::kMaxTreeType );
void AddInputTrees ( TTree* inputTree, const TCut& SigCut, const TCut& BgCut);
void AddTree ( TTree* tree, const TString& className, Double_t weight=1.0,
const TCut& cut = "", Types::ETreeType tt = Types::kMaxTreeType );
void AddTree ( const TString& tr, const TString& className, Double_t weight=1.0,
const TCut& cut = "", Types::ETreeType tt = Types::kMaxTreeType );
// accessors
std::vector< TString >* GetClassList() const;
UInt_t GetEntries( const TString& name ) const { return GetEntries( fInputTrees[name] ); }
UInt_t GetNTrees ( const TString& name ) const { return fInputTrees[name].size(); }
UInt_t GetNSignalTrees() const { return fInputTrees["Signal"].size(); }
UInt_t GetNBackgroundTrees() const { return fInputTrees["Background"].size(); }
UInt_t GetSignalEntries() const { return GetEntries(fInputTrees["Signal"]); }
UInt_t GetBackgroundEntries() const { return GetEntries(fInputTrees["Background"]); }
UInt_t GetEntries() const;
const TreeInfo& SignalTreeInfo(Int_t i) const { return fInputTrees["Signal"][i]; }
const TreeInfo& BackgroundTreeInfo(Int_t i) const { return fInputTrees["Background"][i]; }
std::vector<TreeInfo>::const_iterator begin( const TString& className ) const { return fInputTrees[className].begin(); }
std::vector<TreeInfo>::const_iterator end( const TString& className ) const { return fInputTrees[className].end(); }
std::vector<TreeInfo>::const_iterator Sbegin() const { return begin("Signal"); }
std::vector<TreeInfo>::const_iterator Send() const { return end ("Signal"); }
std::vector<TreeInfo>::const_iterator Bbegin() const { return begin("Background"); }
std::vector<TreeInfo>::const_iterator Bend() const { return end ("Background"); }
// reset the list of trees
void ClearSignalTreeList() { ClearTreeList("Signal"); }
void ClearBackgroundTreeList() { ClearTreeList("Background"); }
void ClearTreeList( const TString& className );
private:
UInt_t GetEntries(const std::vector<TreeInfo>& tiV) const;
TTree * ReadInputTree( const TString& dataFile );
mutable std::map< TString, std::vector<TreeInfo> > fInputTrees; //! list of input trees per class (classname is given as first parameter in the map)
std::map< std::string, Bool_t > fExplicitTrainTest; //! if set to true the user has specified training and testing data explicitly
mutable MsgLogger* fLogger; // message logger
MsgLogger& Log() const { return *fLogger; }
};
}
#endif
|