#ifndef ROOTNTUPLEPDFTESTER_H
#define ROOTNTUPLEPDFTESTER_H
#include "TTree.h"
#include "TFile.h"
#include <string>
#include "StatTools/Pdf/RandomGenerator.h"
#include "StatTools/Pdf/Tuples.h"

namespace Pdf {

template<typename T>
void addBranchSuffix( std::string& s );

template<>
inline void addBranchSuffix<int>( std::string& s )
{ s+="/I"; }

template<>
inline void addBranchSuffix<unsigned int>( std::string& s )
{ s+="/I"; }

template<>
inline void addBranchSuffix<float>( std::string& s )
{ s+="/F"; }

template<>
inline void addBranchSuffix<double>( std::string& s )
{ s+="/D"; }

template<typename T>
inline void addBranchN( TTree& tree, T& x, int n, const std::string& name ) {
  std::string s( name ); s +=( '0' + n );
  std::string sX( s );
  addBranchSuffix<T>( sX );
  tree.Branch( s.c_str(), & x, sX.c_str() );
}

template<int n>
struct iterate {
  template<typename ntuple>
  static void addBranch( TTree& tree, ntuple& x, const std::string& name ) {
    iterate< n - 1 >::addBranch( tree, x, name );
    addBranchN( tree, boost::get< n - 1 >( x ), n - 1, name );
  }
};

template<>
struct iterate< 1 > {
  template<typename ntuple>
  static void addBranch( TTree& tree, ntuple& x, const std::string& name ) {
    addBranchN( tree, boost::get< 0 >( x ), 0, name );
  }
};

class RootNtuplePdfTester {
 public:
  RootNtuplePdfTester( const char * fileName ) : _file( fileName, "RECREATE" ) {}
  ~RootNtuplePdfTester() { _file.Close(); }
  template<typename Pdf>
  void test( const Pdf& pdf, 
	     const char * treeName, const char * treeDescr,
	     int n = 10000 ) {
    RandomGenerator<Pdf> generator( pdf );
    TTree tree( treeName, treeDescr );
    
    typedef typename tuple_from_typelist< typename Types<Pdf>::type >::type ntuple;
    ntuple x;

    iterate<Pdf::vars>::addBranch( tree, x, "x" );
    
    for( int i = 0; i < n; i ++ ) {
      tuple_adapter<ntuple>::generate( generator, x );
      tree.Fill();
    }
    tree.Write();
  }
private:
  TFile _file;
};

}

#endif