Merge of EVE-dev branch.

[u/mrichter/AliRoot.git] / EVE / alice-macros / trd_friend_tracks.C

TEveElementList* trd_friend_tracks(TEveElement *cont = 0)
{

  // Link data containers
  AliESDfriend *eventESDfriend = 0x0;
  if(!(eventESDfriend = AliEveEventManager::AssertESDfriend())){
    Warning("trd_friend_tracks", "AliESDfriend not found");
    return 0x0;
  }

  AliESDEvent* esd = AliEveEventManager::AssertESD();

  AliEveEventManager::AssertGeometry();

  AliTRDReconstructor::SetRecoParam(AliTRDrecoParam::GetLowFluxParam());

  TEveElementList *tracks = new TEveElementList("TRD Tracks");
  for (Int_t n=0; n<esd->GetNumberOfTracks(); n++){
    AliESDtrack* esdTrack = esd->GetTrack(n);
    AliESDfriendTrack *friendTrack = eventESDfriend->GetTrack(n);
  
    TObject *cal = 0x0;
    Int_t ical = 0;
    while(cal = friendTrack->GetCalibObject(ical++)){
      if(strcmp(cal->IsA()->GetName(), "AliTRDtrackV1") != 0) continue;
      AliTRDtrackV1 *trackObj = dynamic_cast<AliTRDtrackV1 *>(cal);
      
      tracks->AddElement(new AliEveTRDTrack(trackObj));
    }
  }
	
  tracks->SetTitle(Form("Tracks %d", tracks->NumChildren()));
  tracks->SetStamp(TEveElement::kCBObjProps);

  gEve->AddElement(tracks, cont);

  gEve->Redraw3D();
  
  return tracks;
}