#include "AliHLTExternalTrackParam.h"
#include "AliHLTTrackMCLabel.h"
#include "AliHLTCTPData.h"
+#include "AliHLTTPCDefinitions.h"
+#include "AliHLTTPCSpacePointData.h"
+#include "AliHLTTPCClusterDataFormat.h"
+#include "AliTPCclusterMI.h"
+#include "AliTPCseed.h"
+#include "AliITStrackV2.h"
+#include "AliESDfriend.h"
+#include "AliESDfriendTrack.h"
+#include "AliHLTTPCTransform.h"
#include "AliHLTErrorGuard.h"
#include "AliESDEvent.h"
#include "AliESDtrack.h"
#include "TTree.h"
#include "TList.h"
#include "TClonesArray.h"
+#include "TTimeStamp.h"
+#include "THnSparse.h"
#include "AliHLTESDCaloClusterMaker.h"
#include "AliHLTCaloClusterDataStruct.h"
#include "AliHLTCaloClusterReader.h"
#include "AliESDVZERO.h"
#include "AliHLTGlobalVertexerComponent.h"
#include "AliHLTVertexFinderBase.h"
+#include "AliSysInfo.h"
+#include "AliHLTSAPTrackerData.h"
+#include "AliFlatESDVertex.h"
/** ROOT macro for the implementation of ROOT specific class methods */
ClassImp(AliHLTGlobalEsdConverterComponent)
, fWriteTree(0)
, fVerbosity(0)
, fESD(NULL)
+ , fESDfriend(NULL)
, fSolenoidBz(-5.00668)
+ , fMakeFriends(0)
, fBenchmark("EsdConverter")
{
// see header file for class documentation
// refer to README to build package
// or
// visit http://web.ift.uib.no/~kjeks/doc/alice-hlt
+ for( int i=0; i<fkNPartition; i++ ){
+ fPartitionClusters[i] = 0;
+ fNPartitionClusters[i] = 0;
+ }
}
AliHLTGlobalEsdConverterComponent::~AliHLTGlobalEsdConverterComponent()
{
// see header file for class documentation
if (fESD) delete fESD;
- fESD=NULL;
+ if (fESDfriend) delete fESDfriend;
+ for( int i=0; i<fkNPartition; i++ ){
+ delete[] fPartitionClusters[i];
+ }
}
int AliHLTGlobalEsdConverterComponent::Configure(const char* arguments)
list.push_back(kAliHLTDataTypeKFVertex); // KFVertex object from vertexer
list.push_back(kAliHLTDataTypePrimaryFinder); // array of track ids for prim vertex
list.push_back(kAliHLTDataTypeESDContent);
+ list.push_back( AliHLTTPCDefinitions::fgkClustersDataType );
+ list.push_back(kAliHLTDataTypeFlatESDVertex); // VertexTracks resonctructed using SAP ITS tracks
+ list.push_back(kAliHLTDataTypeITSSAPData); // SAP ITS tracks
}
AliHLTComponentDataType AliHLTGlobalEsdConverterComponent::GetOutputDataType()
{
// see header file for class documentation
- return kAliHLTDataTypeESDObject|kAliHLTDataOriginOut;
+ return kAliHLTMultipleDataType;
+}
+
+int AliHLTGlobalEsdConverterComponent::GetOutputDataTypes(AliHLTComponentDataTypeList& tgtList){
+// see header file for class documentation
+
+ tgtList.clear();
+ tgtList.push_back( kAliHLTDataTypeESDObject|kAliHLTDataOriginOut );
+ tgtList.push_back( kAliHLTDataTypeESDfriendObject|kAliHLTDataOriginOut );
+ return tgtList.size();
}
void AliHLTGlobalEsdConverterComponent::GetOutputDataSize(unsigned long& constBase, double& inputMultiplier)
fSolenoidBz=GetBz();
+ if(fESD) delete fESD;
+ fESD = NULL;
+ if(fESDfriend) delete fESDfriend;
+ fESDfriend=NULL;
+
if (iResult>=0) {
fESD = new AliESDEvent;
if (fESD) {
SetupCTPData();
}
+
+ if( iResult>=0 && fMakeFriends ){
+ fESDfriend = new AliESDfriend();
+ }
fBenchmark.SetTimer(0,"total");
int AliHLTGlobalEsdConverterComponent::DoDeinit()
{
// see header file for class documentation
- if (fESD) delete fESD;
+ if(fESD) delete fESD;
fESD=NULL;
-
+ if(fESDfriend) delete fESDfriend;
+ fESDfriend = NULL;
return 0;
}
-int AliHLTGlobalEsdConverterComponent::DoEvent(const AliHLTComponentEventData& /*evtData*/,
+int AliHLTGlobalEsdConverterComponent::DoEvent(const AliHLTComponentEventData& evtData,
AliHLTComponentTriggerData& trigData)
{
// see header file for class documentation
int iResult=0;
+
if (!fESD) return -ENODEV;
- if (IsDataEvent()) fBenchmark.StartNewEvent();
+ if (!IsDataEvent()) return iResult;
+ fBenchmark.StartNewEvent();
fBenchmark.Start(0);
+ for(Int_t i=0; i<fkNPartition; i++){
+ delete[] fPartitionClusters[i];
+ fPartitionClusters[i] = 0;
+ fNPartitionClusters[i] = 0;
+ }
+
AliESDEvent* pESD = fESD;
-
+
pESD->Reset();
pESD->SetMagneticField(fSolenoidBz);
pESD->SetRunNumber(GetRunNo());
mask50.set(); // set all bits
mask50 >>= 50; // shift 50 right
pESD->SetTriggerMask((mask&mask50).to_ulong());
- /* next 50, to be implemented
+ //next 50
pESD->SetTriggerMaskNext50((mask>>50).to_ulong());
- */
}
TTree* pTree = NULL;
pTree->SetDirectory(0);
}
- if ((iResult=ProcessBlocks(pTree, pESD))>=0) {
+ if( fESDfriend ) fESDfriend->Reset();
+
+ if ((iResult=ProcessBlocks(pTree, pESD, fESDfriend))>=0) {
// TODO: set the specification correctly
if (pTree) {
// the esd structure is written to the user info and is
iResult=PushBack(pESD, kAliHLTDataTypeESDObject|kAliHLTDataOriginOut, 0);
}
fBenchmark.AddOutput(GetLastObjectSize());
+ if( iResult>=0 && fMakeFriends ){
+ iResult=PushBack(fESDfriend, kAliHLTDataTypeESDfriendObject|kAliHLTDataOriginOut, 0);
+ fBenchmark.AddOutput(GetLastObjectSize());
+ }
}
if (pTree) {
// clear user info list to prevent objects from being deleted
}
fBenchmark.Stop(0);
- HLTInfo( fBenchmark.GetStatistics() );
+ HLTBenchmark( fBenchmark.GetStatistics() );
+
+ for(Int_t i=0; i<fkNPartition; i++){
+ delete[] fPartitionClusters[i];
+ fPartitionClusters[i] = 0;
+ fNPartitionClusters[i] = 0;
+ }
+
+ if( fESDfriend ) fESDfriend->Reset();
+ if( fESD ) fESD->Reset();
return iResult;
}
-int AliHLTGlobalEsdConverterComponent::ProcessBlocks(TTree* pTree, AliESDEvent* pESD)
+int AliHLTGlobalEsdConverterComponent::ProcessBlocks(TTree* pTree, AliESDEvent* pESD, AliESDfriend *pESDfriend )
{
// see header file for class documentation
// 5) Add Trigger Detectors
// VZERO, ZDC
+ // read the clusters
+ // ---------- Access to clusters --------------------//
+ /*
+ const AliHLTComponentBlockData* pBl0=GetFirstInputBlock();
+ int cnt = 0;
+ while (pBl0) {
+ char tp[9],org[5];
+ strncpy(tp,pBl0->fDataType.fID,8);
+ tp[8] = '0';
+ strncpy(org,pBl0->fDataType.fOrigin,4);
+ org[4] = '0';
+ //
+ printf(">>> Bl%3d %8s|%4s of size %d\n",cnt++,tp,org,pBl0->fSize);
+ pBl0 = GetNextInputBlock();
+ };
+ */
+ for(Int_t i=0; i<fkNPartition; i++){
+ delete[] fPartitionClusters[i];
+ fPartitionClusters[i] = 0;
+ fNPartitionClusters[i] = 0;
+ }
+
+ if( pESDfriend ){
+
+ int nInputClusters = 0;
+
+ for(const AliHLTComponentBlockData *iter = GetFirstInputBlock(AliHLTTPCDefinitions::fgkClustersDataType); iter != NULL; iter = GetNextInputBlock()){
+
+ if(iter->fDataType != AliHLTTPCDefinitions::fgkClustersDataType) continue;
+ Int_t slice = AliHLTTPCDefinitions::GetMinSliceNr(iter->fSpecification);
+ Int_t partition = AliHLTTPCDefinitions::GetMinPatchNr(iter->fSpecification);
+ Int_t slicepartition = slice*6+partition;
+ if(slicepartition<0 || slicepartition > fkNPartition){
+ HLTWarning("Wrong header of TPC cluster data, slice %d, partition %d", slice, partition );
+ continue;
+ }
+
+ AliHLTTPCClusterData *inPtrSP = ( AliHLTTPCClusterData* )( iter->fPtr );
+ nInputClusters += inPtrSP->fSpacePointCnt;
+
+ delete[] fPartitionClusters[slicepartition];
+ fPartitionClusters[slicepartition] = new AliTPCclusterMI[inPtrSP->fSpacePointCnt];
+ fNPartitionClusters[slicepartition] = inPtrSP->fSpacePointCnt;
+
+ // create offline clusters out of the HLT clusters
+
+ for ( unsigned int i = 0; i < inPtrSP->fSpacePointCnt; i++ ) {
+ AliHLTTPCSpacePointData *chlt = &( inPtrSP->fSpacePoints[i] );
+ AliTPCclusterMI *c = fPartitionClusters[slicepartition]+i;
+ c->SetX(chlt->fX);
+ c->SetY(chlt->fY);
+ c->SetZ(chlt->fZ);
+ c->SetSigmaY2(chlt->fSigmaY2);
+ c->SetSigmaYZ( 0 );
+ c->SetSigmaZ2(chlt->fSigmaZ2);
+ c->SetQ( chlt->fCharge );
+ c->SetMax( chlt->fQMax );
+ Int_t sector, row;
+ Float_t padtime[3] = {0,chlt->fY,chlt->fZ};
+ AliHLTTPCTransform::Slice2Sector(slice,chlt->fPadRow, sector, row);
+ AliHLTTPCTransform::Local2Raw( padtime, sector, row);
+ c->SetDetector( sector );
+ c->SetRow( row );
+ c->SetPad( (Int_t) padtime[1] );
+ c->SetTimeBin( (Int_t) padtime[2] );
+ }
+ } // end of loop over blocks of clusters
+
+
+ // fill event info
+ {
+ for( Int_t iSlice=0; iSlice<36; iSlice++ ){
+ int iSector = iSlice;
+ int nclu = 0;
+ for( Int_t iPartition=0; iPartition<3; iPartition++){
+ int slicepartition = iSlice*6+iPartition;
+ nclu+= fNPartitionClusters[slicepartition];
+ }
+ pESDfriend->SetNclustersTPC( iSector, nclu );
+ iSector = 36+iSlice;
+ nclu = 0;
+ for( Int_t iPartition=3; iPartition<6; iPartition++){
+ int slicepartition = iSlice*6+iPartition;
+ nclu+= fNPartitionClusters[slicepartition];
+ }
+ pESDfriend->SetNclustersTPC( iSector, nclu );
+ }
+ }
+
+
+
+ }
+
// 1) first read MC information (if present)
std::map<int,int> mcLabelsTPC;
std::map<int,int> mcLabelsITS;
iotrack.SetLabel(mcLabel);
pESD->AddTrack(&iotrack);
if (fVerbosity>0) element->Print();
+
+ if( pESDfriend ){ // create friend track
+
+ AliHLTGlobalBarrelTrack gb(*element);
+ AliTPCseed tTPC;
+ tTPC.Set( gb.GetX(), gb.GetAlpha(), gb.GetParameter(), gb.GetCovariance() );
+ tTPC.SetLabel(mcLabel);
+
+ // set the clusters
+ UInt_t nClusters = element->GetNumberOfPoints();
+ const UInt_t*clusterIDs = element->GetPoints();
+
+ tTPC.SetNumberOfClusters(nClusters);
+
+ for(UInt_t ic=0; ic<nClusters; ic++){
+
+ UInt_t id = clusterIDs[ic];
+ int iSlice = AliHLTTPCSpacePointData::GetSlice(id);
+ int iPartition = AliHLTTPCSpacePointData::GetPatch(id);
+ int iCluster = AliHLTTPCSpacePointData::GetNumber(id);
+
+ if(iSlice<0 || iSlice>36 || iPartition<0 || iPartition>5){
+ HLTError("Corrupted TPC cluster Id: slice %d, partition %d, cluster %d", iSlice, iPartition, iCluster);
+ continue;
+ }
+
+ AliTPCclusterMI *patchClusters = fPartitionClusters[iSlice*6 + iPartition];
+ if(!patchClusters){
+ HLTError("Clusters are missed for slice %d, partition %d", iSlice, iPartition );
+ continue;
+ }
+
+ if(iCluster >= fNPartitionClusters[iSlice*6 + iPartition]){
+ HLTError("TPC slice %d, partition %d: ClusterID==%d >= N Cluaters==%d ", iSlice, iPartition,iCluster, fNPartitionClusters[iSlice*6 + iPartition] );
+ continue;
+ }
+
+ AliTPCclusterMI *c = &(patchClusters[iCluster]);
+ int sec = c->GetDetector();
+ int row = c->GetRow();
+ if(sec >= 36) row = row + AliHLTTPCTransform::GetNRowLow();
+
+ tTPC.SetClusterPointer(row, c);
+
+ AliTPCTrackerPoint &point = *( tTPC.GetTrackPoint( row ) );
+ //tTPC.Propagate( TMath::DegToRad()*(sec%18*20.+10.), c->GetX(), fSolenoidBz );
+ Double_t angle2 = tTPC.GetSnp()*tTPC.GetSnp();
+ angle2 = (angle2<1) ?TMath::Sqrt(angle2/(1-angle2)) :10.;
+ point.SetAngleY( angle2 );
+ point.SetAngleZ( tTPC.GetTgl() );
+ } // end of associated cluster loop
+
+ // Cook dEdx
+
+ //AliTPCseed *seed = &(tTPC);
+ //fSeedArray->AddAt( seed, TMath::Abs(seed->GetLabel()) );
+ //fdEdx->Fill( seed->P()*seed->Charge(), seed->CookdEdx(0.02, 0.6) );
+
+ AliESDfriendTrack friendTrack;
+ friendTrack.AddCalibObject(&tTPC);
+ pESDfriend->AddTrack(&friendTrack);
+ }
}
+
HLTInfo("converted %d track(s) to AliESDtrack and added to ESD", tracks.size());
iAddedDataBlocks++;
} else if (iResult<0) {
pESD->SetPrimaryVertexSPD( vtx );
}
+ // Get ITS Standalone primaries (SAP) vertexTracks
+ {
+ const AliHLTComponentBlockData* pBlock=GetFirstInputBlock(kAliHLTDataTypeFlatESDVertex|kAliHLTDataOriginITS);
+ if (pBlock) {
+ fBenchmark.AddInput(pBlock->fSize);
+ AliFlatESDVertex *vtxFlat = reinterpret_cast<AliFlatESDVertex*>( pBlock->fPtr );
+ if (vtxFlat->GetNContributors()>0) {
+ AliESDVertex vtx;
+ vtxFlat->GetESDVertex(vtx);
+ vtx.SetTitle("vertexITSSAP");
+ pESD->SetPrimaryVertexSPD( &vtx );
+ }
+ }
+ }
+
+ // Get ITS Standalone primary (SAP) Tracks
+ {
+ const AliHLTComponentBlockData* pBlock=GetFirstInputBlock(kAliHLTDataTypeITSSAPData|kAliHLTDataOriginITS);
+ if (pBlock) {
+ fBenchmark.AddInput(pBlock->fSize);
+ const AliHLTITSSAPTrackerDataContainer *dataSAP = reinterpret_cast<const AliHLTITSSAPTrackerDataContainer*>(pBlock->fPtr);
+ AliITStrackV2 trcV2;
+ int ntrITSSAP = dataSAP->fCount;
+ for (int itr=0;itr<ntrITSSAP;itr++) {
+ const AliHLTITSSAPTrackerData& trcFlatSAP = dataSAP->fTracks[itr];
+ AliESDtrack inpESDtrc;
+ inpESDtrc.SetID(pESD->GetNumberOfTracks());
+ trcFlatSAP.paramInw.GetExternalTrackParam(trcV2); // track at the vertex
+ trcV2.SetLabel(trcFlatSAP.label);
+ trcV2.SetNumberOfClusters(trcFlatSAP.ncl);
+ trcV2.SetChi2(trcFlatSAP.chi2);
+ inpESDtrc.UpdateTrackParams(&trcV2,AliESDtrack::kITSrefit);
+ inpESDtrc.SetStatus( (AliESDtrack::kITSin|AliESDtrack::kITSout|AliESDtrack::kITSpureSA) );
+ pESD->AddTrack(&inpESDtrc);
+ //
+ if( pESDfriend ) {
+ AliESDfriendTrack friendTrack;
+ trcFlatSAP.paramOut.GetExternalTrackParam(trcV2); // track at the vertex
+ friendTrack.SetITSOut(trcV2);
+ pESDfriend->AddTrack(&friendTrack);
+ }
+ }
+ }
+ }
+
+
+
// 3.1. now update ESD tracks with the ITSOut info
// updating track parameters with flag kITSout will overwrite parameter set above with flag kTPCout
// TODO 2010-07-12 there are some issues with this updating sequence, for the moment update with
// update with vertices and vertex-fitted tracks
// output of PrimaryVertexer and V0Finder components
+
TObject* pBase = (TObject*)GetFirstInputObject(kAliHLTDataTypeKFVertex | kAliHLTDataOriginOut);
if (pBase) {
AliKFVertex* kfVertex = dynamic_cast<AliKFVertex *>(pBase);
}
}
}
+
+ if( fMakeFriends && pESDfriend ){ // create friend track
+ pESD->SetESDfriend( pESDfriend );
+ }
+
if (iAddedDataBlocks>0 && pTree) {
pTree->Fill();
- }
+ }
if (iResult>=0) iResult=iAddedDataBlocks;
return iResult;
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff