---

AlephNtupleWriter.cpp

Go to the documentation of this file.

#include "AlephNtupleWriter.h"
#include <stdlib.h>

//constructor
AlephNtupleWriter::AlephNtupleWriter()
{
cards = NULL;
size = 0;
booked = false;
type = 0;
}

// destructor
AlephNtupleWriter::~AlephNtupleWriter() {}

// initialize
void AlephNtupleWriter::Initialize(AlephCardsReader* thecards, int mode)
{
        cards = thecards;
        // mode: 1 = ROOT
        //       2 = HBOOK RW Ntuple 
        type = mode;
        switch (type)
        {
                case 1 :
                        ROOTfile = new TFile(cards->getHist().c_str(),"RECREATE");
                        ROOTtree = new TTree("analysis","alpha++ analysis");
                        break;
                case 2 :
                        // done by ALPHA
                        break;
                default :
                        cerr << "ERROR: Bad Ntuple Type." << endl;
                        exit(0);
        }
}

// fill
void AlephNtupleWriter::Fill()
{
        switch (type)
        {
                case 1 :
                        ROOTfile->cd();
                        ROOTtree->Fill();
                        break;
                case 2 :
                        {
                                if(!booked)
                                {
                                        char d[NTMAXIN][8];
                                        for (int i =0;i<NTMAXIN;i++) for(int j=0;j<8;j++) d[i][j]=0;
                                                int cnt = 0;
                                        for(map<string,int>::iterator mapel = NamesIndexes.begin(); mapel != NamesIndexes.end(); mapel++)
                                        {
                                                for(int i=0;i<8;i++)
                                                {
                                                        d[cnt][i] = mapel->first.c_str()[i];
                                                }
                                                cnt++;
                                        }
                                        HBOOKN(1000,"analysis",size," ",5000,d);
                                        booked = true;
                                }
                                float* myvars = new float[size];
                                int cnt = 0;
                                for(map<string,int>::iterator mapel=NamesIndexes.begin();mapel != NamesIndexes.end();mapel++)
                                        myvars[cnt++] = data[mapel->second];
                                HFN(1000,myvars);
                                delete [] myvars;
                        }
                        break;
                default :
                        cerr << "ERROR: Bad Ntuple Type." << endl;
                        exit(0);                
        }
}

// terminate
void AlephNtupleWriter::Terminate()
{
        switch (type)
        {
                case 1 :
                        ROOTfile->cd();
                        ROOTtree->Write();
                        ROOTfile->Close();
//   The program seems to hang at the end is the tree is deleted.
//                      delete ROOTtree;
                        delete ROOTfile;
                        for(map<string,int*>::iterator mapel=SizeAddresses.begin();mapel != SizeAddresses.end();mapel++)
                                delete mapel->second;
                        for(vector<smart_void*>::iterator vecel=theObjects.begin();vecel < theObjects.end();vecel++)
                                delete *vecel;
                        break;
                case 2 :
                        // done by ALPHA
                        break;
                default :
                        cerr << "ERROR: Bad Ntuple Type." << endl;
                        exit(0);
        }       
}
/*
// user methods to keep the variable
template <typename entry>
void AlephNtupleWriter::Keep(char* name, entry* value)
{
        switch(type)
        {
                case 1 :
                        *(entry*)NamesAddresses[name] = *value;
                        break;
                case 2 :
                        data[ NamesIndexes[name] ] = (float)*value;
                        break;
                default:
                        cerr << "ERROR: Bad Ntuple Type." << endl;
                        exit(0);
        }
}

template <typename entry>
void AlephNtupleWriter::Keep(char* name, entry* valuearray, int size)
{
        switch(type)
        {
                case 1 :
                        *SizeAddresses[name] = size;
                        entry* array = (entry*)NamesAddresses[name];
                        for(int i=0;i<size;i++) array[i] = valuearray[i];
                        break;
                case 2 :
                        cerr << "ERROR: column-wise ntuple still not handled... Keep(char*, entry*, int) unavailable." << endl;
                        break;
                default:
                        cerr << "ERROR: Bad Ntuple Type." << endl;
                        exit(0);
        }               
}
*/

---