AgataTrackDisplay.C

Go to the documentation of this file.

/***************************************************************************
 *   Copyright (C) 2004-2006 by Olivier Stezowski & Christian Finck        *
 *   stezow(AT)ipnl.in2p3.fr, cfinck(AT)ires.in2p3.fr                      *
 *                                                                         *
 *   This program is free software; you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation; either version 2 of the License, or     *
 *   (at your option) any later version.                                   *
 *                                                                         *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *                                                                         *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program; if not, write to the                         *
 *   Free Software Foundation, Inc.,                                       *
 *   59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.             *
 ***************************************************************************/


#if !defined(__CINT__) || defined(__MAKECINT__)

#include "Riostream.h"

#include "TMath.h"
#include "TRandom.h"
#include "TEveManager.h"

#include "AgataGeometryTransformer.h"
#include "AgataEventDisplay.h"
#include "AgataEventContainer.h"

#include "GeantLMOF.h"
#include "GEM.h"
#include "GammaLink.h"

#endif

void AgataTrackDisplay(Int_t nEvents = 100)
{
  Float_t metricConv = 10.;
  
  Gw::AgataEventDisplay* dis = Gw::AgataEventDisplay::Instance();
  Gw::AgataEventContainer* agataContainer = dis->GetEventContainer(); 
  dis->SetTrackStyle("Cone");

  // GEM part
  Gw::GEM gem;
 
  Gw::BaseGEM *random = 0x0;
  random = new Gw::BaseGEM(); 
  // needs ags file
  random->Import("../../ndb/198Pb.ags"); 
  
  TList cas; Gw::GammaLink *gam;
  
  // loop over events
  for (Int_t i = 0; i < nEvents; ++i) {
    
    // to get the next cascade
    random->DoCascade(cas,""); 
    
    for (Int_t j = 0; j < cas.GetSize(); j++ ) { 
            
      gam = (Gw::GammaLink *)cas.At(j);
      Double_t e =  gRandom->Gaus(gam->GetEnergy().Get(),1);
      
      Double_t x,y,z;
      Double_t theta, phi, rho, drho;
      theta = gRandom->Uniform(TMath::Pi()/4.);
      phi   = gRandom->Uniform(2*TMath::Pi());
      rho   = gRandom->Uniform(235, 329);
      drho  = gRandom->Uniform(94);
      x = rho*TMath::Sin(theta)*TMath::Cos(phi);
      y = rho*TMath::Sin(theta)*TMath::Sin(phi);
      z = drho*TMath::Cos(theta) + 235;
      
      // new track hit belonging to list
      Gw::TrackHit* hit = agataContainer->NewTrackHit();
      hit->SetX(x/metricConv);
      hit->SetY(y/metricConv);
      hit->SetZ(z/metricConv);
      hit->SetE(e);

      // new hits inside track
      Gw::StdHit* shit = hit->NewHit();
      shit->SetX(x/metricConv);
      shit->SetY(y/metricConv);
      shit->SetZ(z/metricConv);
      shit->SetE(e*0.6);
      
      Int_t ndiff = Int_t(gRandom->Uniform(0, 1) + 0.5);
      for (Int_t k = 0; k < ndiff; ++k) {
        shit = hit->NewHit();
        theta = gRandom->Uniform(TMath::Pi()/2.);
        phi   = gRandom->Uniform(2*TMath::Pi());
        rho   = gRandom->Uniform(-30, 30);
        x += rho*TMath::Sin(theta)*TMath::Cos(phi);
        y += rho*TMath::Sin(theta)*TMath::Sin(phi);
        z += rho*TMath::Cos(theta);
        shit->SetXYZ(x/metricConv,y/metricConv,z/metricConv);
        shit->SetE(e*0.4/(ndiff+1) );
      }
    }
        // fill tracks for Agata
        agataContainer->FillTracks("Agata");
  }
  
  //--------------------- display ---------------------  
 
  // add geometry of Agata via AgataGeometryTransformer
  // need aclust, aeuler, asolid files
  dis->AddGeometry(Gw::AgataGeometryTransformer::ImportAgata("../gem/asolid", "../gem/aclust", "../gem/aeuler")); 
  dis->ShowDisplay();
}