[u/mrichter/AliRoot.git] / HLT / TPCLib / AliHLTTPCEsdWriterComponent.cxx

// @(#) $Id$

/**************************************************************************
 * This file is property of and copyright by the ALICE HLT Project        * 
 * ALICE Experiment at CERN, All rights reserved.                         *
 *                                                                        *
 * Primary Authors: Matthias Richter <Matthias.Richter@ift.uib.no>        *
 *                  for The ALICE HLT Project.                            *
 *                                                                        *
 * 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.                  *
 **************************************************************************/

/** @file   AliHLTTPCEsdWriterComponent.cxx
    @author Matthias Richter
    @date   
    @brief  Writer component to store tracks of the HLT TPC conformal
            mapping tracker in the AliESD format

                                                                          */
// see header file for class documentation
// or
// refer to README to build package
// or
// visit http://web.ift.uib.no/~kjeks/doc/alice-hlt

#include <cassert>
#include "AliHLTTPCEsdWriterComponent.h"
#include "AliESDEvent.h"
#include "AliESDtrack.h"
#include "TTree.h"
#include "AliHLTTPCTrack.h"
#include "AliHLTTPCTrackArray.h"
#include "AliHLTTPCTrackletDataFormat.h"
#include "AliHLTTPCDefinitions.h"
#include "AliHLTTPCTransform.h"

/** ROOT macro for the implementation of ROOT specific class methods */
ClassImp(AliHLTTPCEsdWriterComponent)

AliHLTTPCEsdWriterComponent::AliHLTTPCEsdWriterComponent()
{
  // see header file for class documentation
  // or
  // refer to README to build package
  // or
  // visit http://web.ift.uib.no/~kjeks/doc/alice-hlt
}

AliHLTTPCEsdWriterComponent::~AliHLTTPCEsdWriterComponent()
{
  // see header file for class documentation
}

AliHLTTPCEsdWriterComponent::AliWriter::AliWriter()
  :
  fTree(NULL),
  fESD(NULL),
  fBase(new AliHLTTPCEsdWriterComponent)
{
  // see header file for class documentation
  // or
  // refer to README to build package
  // or
  // visit http://web.ift.uib.no/~kjeks/doc/alice-hlt
}

AliHLTTPCEsdWriterComponent::AliWriter::~AliWriter()
{
  // see header file for class documentation
  if (fBase) delete fBase;
  fBase=NULL;
}

void AliHLTTPCEsdWriterComponent::AliWriter::GetInputDataTypes(AliHLTComponentDataTypeList& list)
{
  // see header file for class documentation
  list.push_back(AliHLTTPCDefinitions::fgkTrackSegmentsDataType);
}

int AliHLTTPCEsdWriterComponent::AliWriter::InitWriter()
{
  // see header file for class documentation
  int iResult=0;
  fESD = new AliESDEvent;
  if (fESD) {
    fESD->CreateStdContent();
    fTree = new TTree("esdTree", "Tree with HLT ESD objects");
    if (fTree) {
      fTree->SetDirectory(0);
      fESD->WriteToTree(fTree);
    }
  }
  if (fTree==NULL) {
    iResult=-ENOMEM;
  }
  return iResult;
}

int AliHLTTPCEsdWriterComponent::AliWriter::CloseWriter()
{
  // see header file for class documentation
  int iResult=0;
  if (fTree) {
    WriteObject(kAliHLTVoidEventID, fTree);
    TTree* pTree=fTree;
    fTree=NULL;
    delete pTree;
  } else {
    HLTWarning("not initialized");
  }
  iResult=AliHLTRootFileWriterComponent::CloseWriter();
  return iResult;
}

int AliHLTTPCEsdWriterComponent::AliWriter::DumpEvent( const AliHLTComponentEventData& evtData,
					    const AliHLTComponentBlockData* blocks, 
					    AliHLTComponentTriggerData& /*trigData*/ )
{
  // see header file for class documentation
  int iResult=0;
  TTree* pTree=fTree;
  assert(fBase);
  if (pTree && fBase) {
    if (fESD) {
      AliESDEvent* pESD=fESD;

      iResult=fBase->ProcessBlocks(pTree, pESD, blocks, (int)evtData.fBlockCnt);

    } else {
      iResult=-ENOMEM;
    }
  }
  return iResult;
}

int AliHLTTPCEsdWriterComponent::AliWriter::ScanArgument(int argc, const char** argv)
{
  // see header file for class documentation
  int iResult=AliHLTRootFileWriterComponent::ScanArgument(argc, argv);
  return iResult;
}

int AliHLTTPCEsdWriterComponent::ProcessBlocks(TTree* pTree, AliESDEvent* pESD,
					       const AliHLTComponentBlockData* blocks,
					       int nBlocks, int* pMinSlice,
					       int* pMaxSlice)
{
  // see header file for class documentation
  int iResult=0;
  if (pESD && blocks) {
      const AliHLTComponentBlockData* iter = NULL;
      AliHLTTPCTrackletData* inPtr=NULL;
      int bIsTrackSegs=0;
 
      for (int ndx=0; ndx<nBlocks && iResult>=0; ndx++) {
	iter = blocks+ndx;
	if ( (bIsTrackSegs=(iter->fDataType == AliHLTTPCDefinitions::fgkTrackSegmentsDataType))==1 ||
	     iter->fDataType == AliHLTTPCDefinitions::fgkTracksDataType ) {
	  Int_t minslice=AliHLTTPCDefinitions::GetMinSliceNr(iter->fSpecification);
	  Int_t maxslice=AliHLTTPCDefinitions::GetMaxSliceNr(iter->fSpecification);
	  if (bIsTrackSegs==0) {
	    // slice parameter and data specification ignored, tracks already in global coordinates
	    minslice=-1;
	    maxslice=-1;
	    if (pMinSlice) *pMinSlice=0;
	    if (pMaxSlice) *pMaxSlice=AliHLTTPCTransform::GetNSlice()-1;
	  } else {
	    if (pMinSlice && (*pMinSlice==-1 || *pMinSlice>minslice)) *pMinSlice=minslice;
	    if (pMaxSlice && (*pMaxSlice==-1 || *pMaxSlice<maxslice)) *pMaxSlice=maxslice;
	  }
	  //HLTDebug("dataspec %#x minslice %d", iter->fSpecification, minslice);
	  if (minslice >=-1 && minslice<AliHLTTPCTransform::GetNSlice()) {
	    if (minslice!=maxslice) {
	      HLTWarning("data from multiple sectors in one block: "
			 "possible missmatch in treatment of local coordinate system");
	    }
	    AliHLTTPCTrackArray tracks;
	    inPtr=(AliHLTTPCTrackletData*)iter->fPtr;
	    HLTDebug("reading block %d (slice %d): %d tracklets", ndx, minslice, inPtr->fTrackletCnt);
	    tracks.FillTracks(inPtr->fTrackletCnt, inPtr->fTracklets, minslice, 0/*don't rotate*/);
	    if ((iResult=Tracks2ESD(&tracks, pESD))>=0) {
	    }
	  } else {
	    HLTError("invalid sector number");
	    iResult=-EBADF;
	  }
	}
      }
      if (iResult>=0 && pTree) {
	pTree->Fill();
      }

      pESD->Reset();
    
  } else {
    iResult=-EINVAL;
  }
  return iResult;
}

int AliHLTTPCEsdWriterComponent::Tracks2ESD(AliHLTTPCTrackArray* pTracks, AliESDEvent* pESD)
{
  // see header file for class documentation
  int iResult=0;
  if (pTracks && pESD) {
    HLTDebug("converting %d tracks from track array", pTracks->GetNTracks());
    for (int i=0; i<pTracks->GetNTracks() && iResult>=0; i++) {
      AliHLTTPCTrack* pTrack=(*pTracks)[i];
      if (pTrack) {
	//HLTDebug("convert track %d", i);
	//pTrack->Print();
	int iLocal=pTrack->Convert2AliKalmanTrack();
	if (iLocal>=0) {
	AliESDtrack iotrack;
	iotrack.UpdateTrackParams(pTrack,AliESDtrack::kTPCin);
	iotrack.SetTPCPoints(pTrack->GetPoints());
	pESD->AddTrack(&iotrack);
	} else {
	  HLTError("conversion to AliKalmanTrack failed for track %d of %d", i, pTracks->GetNTracks());
	}
      } else {
	HLTError("internal missmatch in array");
	iResult=-EFAULT;
      }
    }
    
  } else {
    iResult=-EINVAL;
  }
  return iResult;
}

AliHLTTPCEsdWriterComponent::AliConverter::AliConverter()
  :
  fBase(new AliHLTTPCEsdWriterComponent),
  fWriteTree(1)
{
  // see header file for class documentation
  // or
  // refer to README to build package
  // or
  // visit http://web.ift.uib.no/~kjeks/doc/alice-hlt
}

AliHLTTPCEsdWriterComponent::AliConverter::~AliConverter()
{
  // see header file for class documentation
  if (fBase) delete fBase;
  fBase=NULL;
}

void AliHLTTPCEsdWriterComponent::AliConverter::GetInputDataTypes(AliHLTComponentDataTypeList& list)
{
  // see header file for class documentation
  list.push_back(AliHLTTPCDefinitions::fgkTrackSegmentsDataType);
}

AliHLTComponentDataType AliHLTTPCEsdWriterComponent::AliConverter::GetOutputDataType()
{
  // see header file for class documentation
  return kAliHLTDataTypeESDTree;
}

void AliHLTTPCEsdWriterComponent::AliConverter::GetOutputDataSize(unsigned long& constBase, double& inputMultiplier)
{
  // see header file for class documentation
  constBase=1000000;
  inputMultiplier=1.0;
}

int AliHLTTPCEsdWriterComponent::AliConverter::DoInit(int argc, const char** argv)
{
  // see header file for class documentation
  int iResult=0;
  TString argument="";
  int bMissingParam=0;
  for (int i=0; i<argc && iResult>=0; i++) {
    argument=argv[i];
    if (argument.IsNull()) continue;

    // -notree
    if (argument.CompareTo("-notree")==0) {
      fWriteTree=0;

      // -tree
    } else if (argument.CompareTo("-tree")==0) {
      fWriteTree=1;

    } else {
      HLTError("unknown argument %s", argument.Data());
      break;
    }
  }
  if (bMissingParam) {
    HLTError("missing parameter for argument %s", argument.Data());
    iResult=-EINVAL;
  }

  return iResult;
}

int AliHLTTPCEsdWriterComponent::AliConverter::DoDeinit()
{
  // see header file for class documentation
  return 0;
}

int AliHLTTPCEsdWriterComponent::AliConverter::DoEvent(const AliHLTComponentEventData& evtData, 
						       const AliHLTComponentBlockData* blocks, 
						       AliHLTComponentTriggerData& /*trigData*/,
						       AliHLTUInt8_t* /*outputPtr*/, 
						       AliHLTUInt32_t& /*size*/,
						       AliHLTComponentBlockDataList& /*outputBlocks*/ )
{
  // see header file for class documentation
  int iResult=0;
  assert(fBase);
  AliESDEvent* pESD = new AliESDEvent;
  if (pESD && fBase) {
    pESD->CreateStdContent();
    TTree* pTree = NULL;
    // TODO: Matthias 06.12.2007
    // Tried to write the ESD directly instead to a tree, but this did not work
    // out. Information in the ESD is different, needs investigation.
    if (fWriteTree)
      pTree = new TTree("esdTree", "Tree with HLT ESD objects");
    if (pTree) {
      pTree->SetDirectory(0);
      pESD->WriteToTree(pTree);
    }

    if ((iResult=fBase->ProcessBlocks(pTree, pESD, blocks, (int)evtData.fBlockCnt))>=0) {
	// TODO: set the specification correctly
      if (pTree)
	iResult=PushBack(pTree, kAliHLTDataTypeESDTree|kAliHLTDataOriginTPC, 0);
      else
	iResult=PushBack(pESD, kAliHLTDataTypeESDObject|kAliHLTDataOriginTPC, 0);
    }
    if (pTree)
      delete pTree;

    delete pESD;
  }
  return iResult;
}
Commit	Line	Data
3cde846d	1	// @(#) $Id$
3cde846d	2
2a083ac4	3	/**************************************************************************
9be2600f	4	* This file is property of and copyright by the ALICE HLT Project *
9be2600f	5	* ALICE Experiment at CERN, All rights reserved. *
2a083ac4	6	* *
9be2600f	7	* Primary Authors: Matthias Richter <Matthias.Richter@ift.uib.no> *
9be2600f	8	* for The ALICE HLT Project. *
2a083ac4	9	* *
	10	* Permission to use, copy, modify and distribute this software and its *
	11	* documentation strictly for non-commercial purposes is hereby granted *
	12	* without fee, provided that the above copyright notice appears in all *
	13	* copies and that both the copyright notice and this permission notice *
	14	* appear in the supporting documentation. The authors make no claims *
	15	* about the suitability of this software for any purpose. It is *
	16	* provided "as is" without express or implied warranty. *
	17	**************************************************************************/
	18
3cde846d	19	/** @file AliHLTTPCEsdWriterComponent.cxx
	20	@author Matthias Richter
	21	@date
	22	@brief Writer component to store tracks of the HLT TPC conformal
	23	mapping tracker in the AliESD format
	24
	25	*/
f32b83e1	26	// see header file for class documentation
	27	// or
	28	// refer to README to build package
	29	// or
	30	// visit http://web.ift.uib.no/~kjeks/doc/alice-hlt
	31
37513ae4	32	#include <cassert>
3cde846d	33	#include "AliHLTTPCEsdWriterComponent.h"
af885e0f	34	#include "AliESDEvent.h"
d523f52c	35	#include "AliESDtrack.h"
3cde846d	36	#include "TTree.h"
	37	#include "AliHLTTPCTrack.h"
	38	#include "AliHLTTPCTrackArray.h"
	39	#include "AliHLTTPCTrackletDataFormat.h"
	40	#include "AliHLTTPCDefinitions.h"
37513ae4	41	#include "AliHLTTPCTransform.h"
3cde846d	42
3cde846d	43	/** ROOT macro for the implementation of ROOT specific class methods */
	44	ClassImp(AliHLTTPCEsdWriterComponent)
	45
	46	AliHLTTPCEsdWriterComponent::AliHLTTPCEsdWriterComponent()
37513ae4	47	{
	48	// see header file for class documentation
	49	// or
	50	// refer to README to build package
	51	// or
	52	// visit http://web.ift.uib.no/~kjeks/doc/alice-hlt
	53	}
	54
	55	AliHLTTPCEsdWriterComponent::~AliHLTTPCEsdWriterComponent()
	56	{
	57	// see header file for class documentation
	58	}
	59
	60	AliHLTTPCEsdWriterComponent::AliWriter::AliWriter()
3cde846d	61	:
3cde846d	62	fTree(NULL),
37513ae4	63	fESD(NULL),
37513ae4	64	fBase(new AliHLTTPCEsdWriterComponent)
3cde846d	65	{
2a083ac4	66	// see header file for class documentation
	67	// or
	68	// refer to README to build package
	69	// or
	70	// visit http://web.ift.uib.no/~kjeks/doc/alice-hlt
3cde846d	71	}
3cde846d	72
37513ae4	73	AliHLTTPCEsdWriterComponent::AliWriter::~AliWriter()
	74	{
	75	// see header file for class documentation
	76	if (fBase) delete fBase;
	77	fBase=NULL;
	78	}
	79
	80	void AliHLTTPCEsdWriterComponent::AliWriter::GetInputDataTypes(AliHLTComponentDataTypeList& list)
3cde846d	81	{
2a083ac4	82	// see header file for class documentation
37513ae4	83	list.push_back(AliHLTTPCDefinitions::fgkTrackSegmentsDataType);
3cde846d	84	}
3cde846d	85
37513ae4	86	int AliHLTTPCEsdWriterComponent::AliWriter::InitWriter()
3cde846d	87	{
2a083ac4	88	// see header file for class documentation
3cde846d	89	int iResult=0;
af885e0f	90	fESD = new AliESDEvent;
3cde846d	91	if (fESD) {
76e61c6e	92	fESD->CreateStdContent();
3cde846d	93	fTree = new TTree("esdTree", "Tree with HLT ESD objects");
3cde846d	94	if (fTree) {
68f62aaa	95	fTree->SetDirectory(0);
af885e0f	96	fESD->WriteToTree(fTree);
3cde846d	97	}
3cde846d	98	}
	99	if (fTree==NULL) {
	100	iResult=-ENOMEM;
	101	}
	102	return iResult;
	103	}
	104
37513ae4	105	int AliHLTTPCEsdWriterComponent::AliWriter::CloseWriter()
3cde846d	106	{
2a083ac4	107	// see header file for class documentation
3cde846d	108	int iResult=0;
	109	if (fTree) {
	110	WriteObject(kAliHLTVoidEventID, fTree);
	111	TTree* pTree=fTree;
	112	fTree=NULL;
	113	delete pTree;
	114	} else {
	115	HLTWarning("not initialized");
	116	}
2a083ac4	117	iResult=AliHLTRootFileWriterComponent::CloseWriter();
2a083ac4	118	return iResult;
3cde846d	119	}
3cde846d	120
37513ae4	121	int AliHLTTPCEsdWriterComponent::AliWriter::DumpEvent( const AliHLTComponentEventData& evtData,
3cde846d	122	const AliHLTComponentBlockData* blocks,
5d2abf3b	123	AliHLTComponentTriggerData& /trigData/ )
3cde846d	124	{
2a083ac4	125	// see header file for class documentation
3cde846d	126	int iResult=0;
3cde846d	127	TTree* pTree=fTree;
37513ae4	128	assert(fBase);
37513ae4	129	if (pTree && fBase) {
3cde846d	130	if (fESD) {
af885e0f	131	AliESDEvent* pESD=fESD;
3cde846d	132
37513ae4	133	iResult=fBase->ProcessBlocks(pTree, pESD, blocks, (int)evtData.fBlockCnt);
	134
	135	} else {
	136	iResult=-ENOMEM;
	137	}
	138	}
	139	return iResult;
	140	}
	141
	142	int AliHLTTPCEsdWriterComponent::AliWriter::ScanArgument(int argc, const char** argv)
	143	{
	144	// see header file for class documentation
	145	int iResult=AliHLTRootFileWriterComponent::ScanArgument(argc, argv);
	146	return iResult;
	147	}
	148
	149	int AliHLTTPCEsdWriterComponent::ProcessBlocks(TTree* pTree, AliESDEvent* pESD,
	150	const AliHLTComponentBlockData* blocks,
	151	int nBlocks, int* pMinSlice,
	152	int* pMaxSlice)
	153	{
	154	// see header file for class documentation
	155	int iResult=0;
a978c0d5	156	if (pESD && blocks) {
3cde846d	157	const AliHLTComponentBlockData* iter = NULL;
3cde846d	158	AliHLTTPCTrackletData* inPtr=NULL;
de554e07	159	int bIsTrackSegs=0;
3cde846d	160
37513ae4	161	for (int ndx=0; ndx<nBlocks && iResult>=0; ndx++) {
3cde846d	162	iter = blocks+ndx;
96bda103	163	if ( (bIsTrackSegs=(iter->fDataType == AliHLTTPCDefinitions::fgkTrackSegmentsDataType))==1 \|\|
96bda103	164	iter->fDataType == AliHLTTPCDefinitions::fgkTracksDataType ) {
4fdaad1e	165	Int_t minslice=AliHLTTPCDefinitions::GetMinSliceNr(iter->fSpecification);
4fdaad1e	166	Int_t maxslice=AliHLTTPCDefinitions::GetMaxSliceNr(iter->fSpecification);
de554e07	167	if (bIsTrackSegs==0) {
de554e07	168	// slice parameter and data specification ignored, tracks already in global coordinates
b09663cd	169	minslice=-1;
b09663cd	170	maxslice=-1;
37513ae4	171	if (pMinSlice) *pMinSlice=0;
	172	if (pMaxSlice) *pMaxSlice=AliHLTTPCTransform::GetNSlice()-1;
	173	} else {
	174	if (pMinSlice && (pMinSlice==-1 \|\| pMinSlice>minslice)) *pMinSlice=minslice;
	175	if (pMaxSlice && (pMaxSlice==-1 \|\| pMaxSlice<maxslice)) *pMaxSlice=maxslice;
de554e07	176	}
4fdaad1e	177	//HLTDebug("dataspec %#x minslice %d", iter->fSpecification, minslice);
37513ae4	178	if (minslice >=-1 && minslice<AliHLTTPCTransform::GetNSlice()) {
4fdaad1e	179	if (minslice!=maxslice) {
	180	HLTWarning("data from multiple sectors in one block: "
	181	"possible missmatch in treatment of local coordinate system");
	182	}
	183	AliHLTTPCTrackArray tracks;
	184	inPtr=(AliHLTTPCTrackletData*)iter->fPtr;
	185	HLTDebug("reading block %d (slice %d): %d tracklets", ndx, minslice, inPtr->fTrackletCnt);
	186	tracks.FillTracks(inPtr->fTrackletCnt, inPtr->fTracklets, minslice, 0/don't rotate/);
	187	if ((iResult=Tracks2ESD(&tracks, pESD))>=0) {
3cde846d	188	}
3cde846d	189	} else {
4fdaad1e	190	HLTError("invalid sector number");
4fdaad1e	191	iResult=-EBADF;
3cde846d	192	}
	193	}
	194	}
a978c0d5	195	if (iResult>=0 && pTree) {
4fdaad1e	196	pTree->Fill();
4fdaad1e	197	}
3cde846d	198
37513ae4	199	pESD->Reset();
	200
	201	} else {
	202	iResult=-EINVAL;
3cde846d	203	}
	204	return iResult;
	205	}
	206
af885e0f	207	int AliHLTTPCEsdWriterComponent::Tracks2ESD(AliHLTTPCTrackArray* pTracks, AliESDEvent* pESD)
3cde846d	208	{
2a083ac4	209	// see header file for class documentation
3cde846d	210	int iResult=0;
	211	if (pTracks && pESD) {
	212	HLTDebug("converting %d tracks from track array", pTracks->GetNTracks());
	213	for (int i=0; i<pTracks->GetNTracks() && iResult>=0; i++) {
	214	AliHLTTPCTrack* pTrack=(*pTracks)[i];
	215	if (pTrack) {
04dbc9e4	216	//HLTDebug("convert track %d", i);
04dbc9e4	217	//pTrack->Print();
3cde846d	218	int iLocal=pTrack->Convert2AliKalmanTrack();
	219	if (iLocal>=0) {
	220	AliESDtrack iotrack;
	221	iotrack.UpdateTrackParams(pTrack,AliESDtrack::kTPCin);
	222	iotrack.SetTPCPoints(pTrack->GetPoints());
	223	pESD->AddTrack(&iotrack);
	224	} else {
	225	HLTError("conversion to AliKalmanTrack failed for track %d of %d", i, pTracks->GetNTracks());
	226	}
	227	} else {
	228	HLTError("internal missmatch in array");
	229	iResult=-EFAULT;
	230	}
	231	}
	232
	233	} else {
	234	iResult=-EINVAL;
	235	}
	236	return iResult;
	237	}
37513ae4	238
	239	AliHLTTPCEsdWriterComponent::AliConverter::AliConverter()
	240	:
a978c0d5	241	fBase(new AliHLTTPCEsdWriterComponent),
a978c0d5	242	fWriteTree(1)
37513ae4	243	{
	244	// see header file for class documentation
	245	// or
	246	// refer to README to build package
	247	// or
	248	// visit http://web.ift.uib.no/~kjeks/doc/alice-hlt
	249	}
	250
	251	AliHLTTPCEsdWriterComponent::AliConverter::~AliConverter()
	252	{
	253	// see header file for class documentation
	254	if (fBase) delete fBase;
	255	fBase=NULL;
	256	}
	257
	258	void AliHLTTPCEsdWriterComponent::AliConverter::GetInputDataTypes(AliHLTComponentDataTypeList& list)
	259	{
	260	// see header file for class documentation
	261	list.push_back(AliHLTTPCDefinitions::fgkTrackSegmentsDataType);
	262	}
	263
	264	AliHLTComponentDataType AliHLTTPCEsdWriterComponent::AliConverter::GetOutputDataType()
	265	{
	266	// see header file for class documentation
	267	return kAliHLTDataTypeESDTree;
	268	}
	269
	270	void AliHLTTPCEsdWriterComponent::AliConverter::GetOutputDataSize(unsigned long& constBase, double& inputMultiplier)
	271	{
	272	// see header file for class documentation
	273	constBase=1000000;
	274	inputMultiplier=1.0;
	275	}
	276
	277	int AliHLTTPCEsdWriterComponent::AliConverter::DoInit(int argc, const char** argv)
	278	{
	279	// see header file for class documentation
a978c0d5	280	int iResult=0;
	281	TString argument="";
	282	int bMissingParam=0;
	283	for (int i=0; i<argc && iResult>=0; i++) {
	284	argument=argv[i];
	285	if (argument.IsNull()) continue;
	286
	287	// -notree
	288	if (argument.CompareTo("-notree")==0) {
	289	fWriteTree=0;
	290
	291	// -tree
	292	} else if (argument.CompareTo("-tree")==0) {
	293	fWriteTree=1;
	294
	295	} else {
	296	HLTError("unknown argument %s", argument.Data());
	297	break;
	298	}
	299	}
	300	if (bMissingParam) {
	301	HLTError("missing parameter for argument %s", argument.Data());
	302	iResult=-EINVAL;
	303	}
	304
	305	return iResult;
37513ae4	306	}
	307
	308	int AliHLTTPCEsdWriterComponent::AliConverter::DoDeinit()
	309	{
	310	// see header file for class documentation
	311	return 0;
	312	}
	313
	314	int AliHLTTPCEsdWriterComponent::AliConverter::DoEvent(const AliHLTComponentEventData& evtData,
	315	const AliHLTComponentBlockData* blocks,
	316	AliHLTComponentTriggerData& /trigData/,
	317	AliHLTUInt8_t* /outputPtr/,
	318	AliHLTUInt32_t& /size/,
	319	AliHLTComponentBlockDataList& /outputBlocks/ )
	320	{
	321	// see header file for class documentation
	322	int iResult=0;
	323	assert(fBase);
	324	AliESDEvent* pESD = new AliESDEvent;
	325	if (pESD && fBase) {
	326	pESD->CreateStdContent();
a978c0d5	327	TTree* pTree = NULL;
	328	// TODO: Matthias 06.12.2007
	329	// Tried to write the ESD directly instead to a tree, but this did not work
	330	// out. Information in the ESD is different, needs investigation.
	331	if (fWriteTree)
	332	pTree = new TTree("esdTree", "Tree with HLT ESD objects");
37513ae4	333	if (pTree) {
68f62aaa	334	pTree->SetDirectory(0);
37513ae4	335	pESD->WriteToTree(pTree);
a978c0d5	336	}
37513ae4	337
a978c0d5	338	if ((iResult=fBase->ProcessBlocks(pTree, pESD, blocks, (int)evtData.fBlockCnt))>=0) {
37513ae4	339	// TODO: set the specification correctly
a978c0d5	340	if (pTree)
37513ae4	341	iResult=PushBack(pTree, kAliHLTDataTypeESDTree\|kAliHLTDataOriginTPC, 0);
a978c0d5	342	else
a978c0d5	343	iResult=PushBack(pESD, kAliHLTDataTypeESDObject\|kAliHLTDataOriginTPC, 0);
37513ae4	344	}
a978c0d5	345	if (pTree)
	346	delete pTree;
	347
37513ae4	348	delete pESD;
	349	}
	350	return iResult;
	351	}
	352