-#include "TObjArray.h"
-#include "TFile.h"
-#include "TTree.h"
-#include "TBrowser.h"
-#include "TTreeStream.h"
-
+/**************************************************************************
+ * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
+ * *
+ * Author: The ALICE Off-line Project. *
+ * Contributors are mentioned in the code where appropriate. *
+ * *
+ * Permission to use, copy, modify and distribute this software and its *
+ * documentation strictly for non-commercial purposes is hereby granted *
+ * without fee, provided that the above copyright notice appears in all *
+ * copies and that both the copyright notice and this permission notice *
+ * appear in the supporting documentation. The authors make no claims *
+ * about the suitability of this software for any purpose. It is *
+ * provided "as is" without express or implied warranty. *
+ **************************************************************************/
+
+/* $Id$ */
+
+//
+// marian.ivanov@cern.ch
+//
+// ------------------------------------------------------------------------------------------------
+// TTreeStream
+// Standard stream (cout) like input for the tree
+// Run and see TTreeStreamer::Test() - to see TTreeStreamer functionality
+// ------------------------------------------------------------------------------------------------
+//
+// -------------------------------------------------------------------------------------------------
+// TTreeSRedirector
+// Redirect file to different TTreeStreams
+// Run and see TTreeSRedirector::Test() as an example of TTreeSRedirectorer functionality
+//
+
+#include <TClass.h>
+#include <TFile.h>
+#include <TDirectory.h>
+#include <TObjArray.h>
+#include <TTree.h>
+#include <TTreeStream.h>
ClassImp(TTreeDataElement)
ClassImp(TTreeStream)
ClassImp(TTreeSRedirector)
-/*
- marian.ivanov@cern.ch
- //
- ------------------------------------------------------------------------------------------------
- TTreeStream
- Standard stream (cout) like input for the tree
- Run and see TTreeStreamer::Test() - to see TTreeStreamer functionality
- ------------------------------------------------------------------------------------------------
- //
- -------------------------------------------------------------------------------------------------
- TTreeSRedirector
- Redirect file to different TTreeStreams
- Run and see TTreeSRedirector::Test() as an example of TTreeSRedirectorer functionality
- //
-*/
void TTreeStream::Test()
fFile->Close();
delete fFile;
}
+void TTreeSRedirector::StoreObject(TObject* object){
+ //
+ //
+ //
+ TDirectory * backup = gDirectory;
+ fFile->cd();
+ object->Write();
+ if (backup) backup->cd();
+}
+
+
TTreeStream & TTreeSRedirector::operator<<(Int_t id)
{
}
}
if (!clayout){
+ TDirectory * backup = gDirectory;
fFile->cd();
char chname[100];
- sprintf(chname,"Tree%d",id);
+ snprintf(chname,100,"Tree%d",id);
clayout = new TTreeStream(chname);
clayout->fId=id;
fDataLayouts->AddAt(clayout,entries);
+ if (backup) backup->cd();
}
return *clayout;
}
Int_t entries = fDataLayouts->GetEntriesFast();
if (!clayout){
+ TDirectory * backup = gDirectory;
fFile->cd();
clayout = new TTreeStream(name);
clayout->fId=-1;
clayout->SetName(name);
fDataLayouts->AddAt(clayout,entries);
+ if (backup) backup->cd();
}
return *clayout;
}
void TTreeSRedirector::Close(){
//
//
- TFile * backup = gFile;
+ TDirectory * backup = gDirectory;
fFile->cd();
if (fDataLayouts){
Int_t entries = fDataLayouts->GetEntriesFast();
delete fDataLayouts;
fDataLayouts=0;
}
- backup->cd();
+ if (backup) backup->cd();
}
//-------------------------------------------------------------
TTreeDataElement:: TTreeDataElement(Char_t type) :
- fName(),
+ TNamed(),
fType(type),
fDType(0),
fClass(0),
}
TTreeDataElement:: TTreeDataElement(TDataType* type) :
- fName(),
- fType(' '),
+ TNamed(),
+ fType(0),
fDType(type),
fClass(0),
fPointer(0)
}
TTreeDataElement:: TTreeDataElement(TClass* cl) :
- fName(),
- fType(' '),
+ TNamed(),
+ fType(0),
fDType(0),
fClass(cl),
fPointer(0)
fBranches(0),
fTree(new TTree(treename, treename)),
fCurrentIndex(0),
+ fId(0),
fNextName(),
fNextNameCounter(),
fStatus(0)
{
+ //
+ // Standard ctor
+ //
}
TTreeStream::~TTreeStream()
{
+ //
+ // Class dtor
+ //
fElements->Delete();
fBranches->Clear();
delete fElements;
void TTreeStream::Close()
{
+ //
+ // Flush data to disk and close
//
fTree->Write();
}
Int_t TTreeStream::CheckIn(Char_t type, void *pointer)
{
- if (!fElements) fElements = new TObjArray(1000);
+ //
+ // Insert object of given type
+ //
+ if (!fElements) fElements = new TObjArray(10000);
+ if (fElements->GetSize()<=fCurrentIndex) fElements->Expand(fCurrentIndex*2);
TTreeDataElement* element = (TTreeDataElement*)fElements->At(fCurrentIndex);
if (!element) {
element = new TTreeDataElement(type);
char name[1000];
if (fNextName.Length()>0){
if (fNextNameCounter==0){
- sprintf(name,"%s",(const char*)fNextName);
+ snprintf(name,1000,"%s",(const char*)fNextName);
}
if (fNextNameCounter>0){
- sprintf(name,"%s%d",(const char*)fNextName,fNextNameCounter);
+ snprintf(name,1000,"%s%d",(const char*)fNextName,fNextNameCounter);
}
}
else{
- sprintf(name,"B%d.",fCurrentIndex);
+ snprintf(name,1000,"B%d.",fCurrentIndex);
}
element->SetName(name);
//
Int_t TTreeStream::CheckIn(TObject *o){
//
+ // Insert TObject
//
if (!o) return 0;
if (!fElements) fElements = new TObjArray(1000);
char name[1000];
if (fNextName.Length()>0){
if (fNextNameCounter==0){
- sprintf(name,"%s",(const char*)fNextName);
+ snprintf(name,1000,"%s",(const char*)fNextName);
}
if (fNextNameCounter>0){
- sprintf(name,"%s%d",(const char*)fNextName,fNextNameCounter);
+ snprintf(name,1000,"%s%d",(const char*)fNextName,fNextNameCounter);
}
}
else{
- sprintf(name,"B%d",fCurrentIndex);
+ snprintf(name,1000,"B%d",fCurrentIndex);
}
element->SetName(name);
void TTreeStream::BuildTree(){
//
+ // Build the Tree
//
- if (fTree->GetEntries()>0) return;
- fTree = new TTree(GetName(),GetName());
+ if (fTree && fTree->GetEntries()>0) return;
+ if (!fTree) fTree = new TTree(GetName(),GetName());
Int_t entries = fElements->GetEntriesFast();
fBranches = new TObjArray(entries);
TTreeDataElement* element = (TTreeDataElement*)fElements->At(i);
char bname1[1000];
if (element->GetName()[0]==0){
- sprintf(bname1,"B%d",i);
+ snprintf(bname1,1000,"B%d",i);
}
else{
- sprintf(bname1,element->GetName());
+ snprintf(bname1,1000,element->GetName());
}
if (element->fClass){
if (element->fClass->GetBaseClass("TClonesArray")){
- TBranch * br = fTree->Branch(bname1,&(element->fPointer));
+ TBranch * br = fTree->Branch(bname1,element->fClass->GetName(),&(element->fPointer));
fBranches->AddAt(br,i);
}else
{
}
if (element->GetType()>0){
char bname2[1000];
- sprintf(bname2,"B%d/%c",i,element->GetType());
+ snprintf(bname2,1000,"B%d/%c",i,element->GetType());
TBranch * br = fTree->Branch(bname1,element->fPointer,bname2);
fBranches->AddAt(br,i);
}
void TTreeStream::Fill(){
//
+ // Fill the tree
//
if (fTree) {
Int_t entries=fElements->GetEntriesFast();
}
}
-TTreeStream & TTreeStream::Endl(){
+TTreeStream & TTreeStream::Endl()
+{
+ //
+ // Perform pseudo endl operation
+ //
if (fTree->GetNbranches()==0) BuildTree();
Fill();
fStatus =0;
}
-TTreeStream &TTreeStream::operator<<(Char_t *name)
+TTreeStream &TTreeStream::operator<<(const Char_t *name)
{
//
+ // Endl
//
- //
- //Endl
if (name[0]=='\n'){
return Endl();
}