#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 "AliESDfriend.h"
+#include "AliESDfriendTrack.h"
+#include "AliHLTTPCTransform.h"
#include "AliHLTErrorGuard.h"
#include "AliESDEvent.h"
#include "AliESDtrack.h"
#include "AliESDMuonTrack.h"
+#include "AliESDMuonCluster.h"
#include "AliCDBEntry.h"
#include "AliCDBManager.h"
#include "AliPID.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"
/** ROOT macro for the implementation of ROOT specific class methods */
ClassImp(AliHLTGlobalEsdConverterComponent)
, fWriteTree(0)
, fVerbosity(0)
, fESD(NULL)
+ , fESDfriend(NULL)
, fSolenoidBz(-5.00668)
+ , fMakeFriends(1)
, 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;
+ delete fESD;
+ 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 );
}
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();
+ delete fESD;
+ fESD = NULL;
+ delete fESDfriend;
+ fESDfriend=0;
+
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;
+ delete fESD;
fESD=NULL;
-
+ 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;
+
+
+ AliSysInfo::AddStamp("DoEvent.Start", evtData.fStructSize);
+
+ bool benchmark = true;
+
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());
const AliHLTCTPData* pCTPData=CTPData();
if (pCTPData) {
- AliHLTUInt64_t mask=pCTPData->ActiveTriggers(trigData);
+ AliHLTTriggerMask_t mask=pCTPData->ActiveTriggers(trigData);
for (int index=0; index<gkNCTPTriggerClasses; index++) {
- if ((mask&((AliHLTUInt64_t)0x1<<index)) == 0) continue;
+ if ((mask&(AliHLTTriggerMask_t(0x1)<<index)) == 0) continue;
pESD->SetTriggerClass(pCTPData->Name(index), index);
}
- pESD->SetTriggerMask(mask);
+ //first 50 triggers
+ AliHLTTriggerMask_t mask50;
+ mask50.set(); // set all bits
+ mask50 >>= 50; // shift 50 right
+ pESD->SetTriggerMask((mask&mask50).to_ulong());
+ //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() );
+ HLTWarning( fBenchmark.GetStatistics() );
+
+
+
+
+
+
+ if(benchmark){
+
+ Int_t nV0s = pESD->GetNumberOfV0s();
+ Int_t nTracks = pESD->GetNumberOfTracks();
+
+
+ Double_t statistics[10];
+ TString names[10];
+ fBenchmark.GetStatisticsData(statistics, names);
+ // statistics[5] = nTracks;
+ // statistics[6] = time;
+ // statistics[7] = nV0s;
+
+ // FillBenchmarkHistos( statistics, names);
+ fBenchmark.Reset();
+
+ AliSysInfo::AddStamp("DoEvent.Stop", (int)(statistics[1]), (int)(statistics[2]),pESD->GetNumberOfV0s(),pESD->GetNumberOfTracks() );
+ }
+
+ 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 --------------------//
+
+ 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
+ }
+
// 1) first read MC information (if present)
std::map<int,int> mcLabelsTPC;
std::map<int,int> mcLabelsITS;
for (vector<AliHLTGlobalBarrelTrack>::iterator element=tracks.begin();
element!=tracks.end(); element++) {
Float_t points[4] = {
- element->GetX(),
- element->GetY(),
- element->GetLastPointX(),
- element->GetLastPointY()
+ static_cast<Float_t>(element->GetX()),
+ static_cast<Float_t>(element->GetY()),
+ static_cast<Float_t>(element->GetLastPointX()),
+ static_cast<Float_t>(element->GetLastPointY())
};
Int_t mcLabel = -1;
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) {
// 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);
}
}
- // Add tracks from MUON.
+ // Add tracks and clusters from MUON.
for( const AliHLTComponentBlockData *i= GetFirstInputBlock(kAliHLTAnyDataType | kAliHLTDataOriginMUON); i!=NULL; i=GetNextInputBlock() ){
fBenchmark.AddInput(i->fSize);
}
)
{
const TClonesArray* tracklist = NULL;
+ const TClonesArray* clusterlist = NULL;
if (obj->IsA() == AliESDEvent::Class())
{
const AliESDEvent* event = static_cast<const AliESDEvent*>(obj);
if (event->GetList() == NULL) continue;
tracklist = dynamic_cast<const TClonesArray*>(event->GetList()->FindObject("MuonTracks"));
if (tracklist == NULL) continue;
+ clusterlist = dynamic_cast<const TClonesArray*>(event->GetList()->FindObject("MuonClusters"));
+ if (clusterlist == NULL) continue;
}
else if (obj->IsA() == TClonesArray::Class())
{
- tracklist = static_cast<const TClonesArray*>(obj);
- HLTDebug("Received a MUON TClonesArray of tracks with specification: 0x%X", GetSpecification(obj));
+ if (!strcmp(obj->GetName(), "MuonTracks")) {
+ tracklist = static_cast<const TClonesArray*>(obj);
+ HLTDebug("Received a MUON TClonesArray of tracks with specification: 0x%X", GetSpecification(obj));
+ } else {
+ clusterlist = static_cast<const TClonesArray*>(obj);
+ HLTDebug("Received a MUON TClonesArray of clusters with specification: 0x%X", GetSpecification(obj));
+ }
}
else
{
// Cannot handle this object type.
continue;
}
- HLTDebug("Received %d MUON tracks.", tracklist->GetEntriesFast());
- if (tracklist->GetEntriesFast() > 0)
- {
- const AliESDMuonTrack* track = dynamic_cast<const AliESDMuonTrack*>(tracklist->UncheckedAt(0));
- if (track == NULL)
+ // copy tracks
+ if (tracklist) {
+ HLTDebug("Received %d MUON tracks.", tracklist->GetEntriesFast());
+ if (tracklist->GetEntriesFast() > 0)
+ {
+ const AliESDMuonTrack* track = dynamic_cast<const AliESDMuonTrack*>(tracklist->UncheckedAt(0));
+ if (track == NULL)
+ {
+ HLTError(Form("%s from MUON does not contain AliESDMuonTrack objects.", obj->ClassName()));
+ continue;
+ }
+ }
+ for (Int_t i = 0; i < tracklist->GetEntriesFast(); ++i)
{
- HLTError(Form("%s from MUON does not contain AliESDMuonTrack objects.", obj->ClassName()));
- continue;
+ AliESDMuonTrack* track = pESD->NewMuonTrack();
+ *track = *(static_cast<const AliESDMuonTrack*>(tracklist->UncheckedAt(i)));
}
}
- for (Int_t i = 0; i < tracklist->GetEntriesFast(); ++i)
- {
- const AliESDMuonTrack* track = static_cast<const AliESDMuonTrack*>(tracklist->UncheckedAt(i));
- pESD->AddMuonTrack(track);
+ // copy clusters
+ if (clusterlist) {
+ HLTDebug("Received %d MUON clusters.", clusterlist->GetEntriesFast());
+ if (clusterlist->GetEntriesFast() > 0)
+ {
+ const AliESDMuonCluster* cluster = dynamic_cast<const AliESDMuonCluster*>(clusterlist->UncheckedAt(0));
+ if (cluster == NULL)
+ {
+ HLTError(Form("%s from MUON does not contain AliESDMuonCluster objects.", obj->ClassName()));
+ continue;
+ }
+ }
+ for (Int_t i = 0; i < clusterlist->GetEntriesFast(); ++i)
+ {
+ AliESDMuonCluster* cluster = pESD->NewMuonCluster();
+ *cluster = *(static_cast<const AliESDMuonCluster*>(clusterlist->UncheckedAt(i)));
+ }
}
}
+
+ if( fMakeFriends && pESDfriend ){ // create friend track
+ pESD->SetESDfriend( pESDfriend );
+ }
+ cout<<"\n\n ESD Friend: "<<pESDfriend->GetNumberOfTracks()<<endl;
+
if (iAddedDataBlocks>0 && pTree) {
pTree->Fill();
- }
+ }
if (iResult>=0) iResult=iAddedDataBlocks;
return iResult;
}
+
+
+
+/*
+void AliHLTGlobalEsdConverterComponent::FillBenchmarkHistos(Double_t *statistics, TString *names){
+return;
+
+// cout<<"Now writing benchmarks to " << fBenchmarkHistosFilename <<endl<<endl;
+
+ TFile *f = TFile::Open(fBenchmarkHistosFilename,"UPDATE");
+ THnSparseD *s = (THnSparseD*)f->Get("benchmarkInformation");
+ TNamed *t = (TNamed*)f->Get("time");
+
+ if(!s){
+ HLTWarning( "Benchmark Histograms not available!" );
+ return;
+ }
+ s->Fill(statistics);
+
+ TList histosList;
+ histosList.Add(s);
+ histosList.Add(t);
+ histosList.SaveAs(fBenchmarkHistosFilename);
+}
+*/
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff