sim/testsim.cxx

Go to the documentation of this file.

#include "TFile.h"
//#include "TTree.h"
#include "TClonesArray.h"
#include "TCanvas.h"
//#include "T.h"

#include "TpcSimGeometry.h"
#include "HarpoSimChamber.h"
#include "HarpoSimStrips.h"
#include "HarpoSimEvent.h"
#include "TpcSimIonisationChamber.h"
#include "TpcSimDriftChamber.h"
#include "TpcSimIonisationTrack.h"
#include "TpcSimMCEvent.h"
#include "TpcSimEventGenerator.h"
#include "HarpoSimElectronics.h"
#include "HarpoDccMap.h"

#include <iostream>

using namespace std;

int main()
{

  //  TFile* rootfile = new TFile("test.root","recreate");

  TpcSimIonisationChamber* fIonisationChamber = new TpcSimIonisationChamber();
  // // TCanvas* c = chamber->GetGeom()->DrawDetector();
  // // c->SaveAs("geom.png");
  TpcSimEventGenerator* fEventGenerator = new TpcSimEventGenerator();

  TpcSimDriftChamber* fDriftChamber = new TpcSimDriftChamber();

  HarpoSimChamber* fHarpoChamber = new HarpoSimChamber();
  HarpoSimReadout* fHarpoReadout = new HarpoSimReadout();
  HarpoSimElectronics* fHarpoElectronics = new HarpoSimElectronics();
  fHarpoChamber->SetDriftChamber(fDriftChamber);
  fHarpoChamber->SetReadout(fHarpoReadout);
  fHarpoChamber->SetElectronics(fHarpoElectronics);


  //  TTree*  tree = new TTree("tree","tree");
  
  //  TpcSimMCEvent* fMCEvent = 0;
  HarpoSimEvent* fEvent = 0;
  //tree->Bronch("SimEvents", "HarpoSimEvent", &fEvent);



  // rootfile->Close();
  // return 0;

  // fDigi->Reset();
  // fDriftChamber->Reset();

  for(Int_t iEv = 0; iEv<2; iEv++){
    //fIonisationChamber->Reset();
    cout << "Event " << iEv << endl;
    //    TpcSimMCEvent* ev 
    cout << "Generate MCEvent" << endl;
    TpcSimMCEvent* fMCEvent = fEventGenerator->GenerateEvent();
    cout << "Create Event" << endl;
    //HarpoSimEvent* 
      fEvent = new HarpoSimEvent();
    fEvent->SetMCEvent(fMCEvent);
    //fNevts++;
    cout << "GetNtracks" << endl;
    Int_t nTr = fMCEvent->GetNtracks();
    cout << nTr << endl;
    // if(nTr<=0){
    //   if(gHarpoDebug>0) Info("NextEvent","Empty event %d",iEv);
    //   return false;
    // }
    fHarpoChamber->StartEvent();
    
    //fEvent->SetMCEvent(ev);
    for(Int_t iTr = 0; iTr<nTr; iTr++){
      
      cout << "  Generate Track " << iTr << endl;
      TpcSimIonisationTrack* tr = fIonisationChamber->GenerateTrack(fMCEvent->GetTrack(iTr));
      cout << "  Add Track in Event" << endl;
      fEvent->AddIonisationTrack(iTr,tr);
      //new((*fIonisation)[iTr]) TpcSimIonisationTrack(*tr);

      //HarpoSimStrips* strips = new   HarpoSimStrips();
      fHarpoReadout->Reset();
      //fHarpoReadout->SetStrips(strips);
      fHarpoChamber->ProcessTrack(tr);
      
      HarpoSimStrips* strips = fHarpoReadout->GetStrips();
      strips->SetTrackId(iTr);
      //      new((*fStrips)[iTr]) HarpoSimStrips(*strips);
      //fEvent->AddStrips(iTr,strips);

      fHarpoElectronics->ProcessStrips(strips);

    }
    cout << "Finish Event" << endl;
    fHarpoChamber->FinishEvent();
    cout << "GetDccMapX" << endl;
    HarpoDccMap* mapX = fHarpoElectronics->GetDccMapX();
    fEvent->SetDccMapX(mapX);

    // TCanvas* c = new TCanvas();
    // mapX->map.Draw("colz");
    // c->SaveAs(Form("test%d.png",iEv));

    cout << "GetDccMapY" << endl;
    HarpoDccMap* mapY = new HarpoDccMap();//fHarpoElectronics->GetDccMapX();
    fEvent->SetDccMapY(mapY);

    fEvent->SetHeader(0,iEv);
    //    cout << "Fill Tree" << endl;
    //tree->Fill();
    //    array->Delete();
    cout << "Delete MCEvent" << endl;
    //fMCEvent->Delete();
    cout << "Delete Event" << endl;
    fEvent->Delete();
    cout << "Delete Ionisation tracks" << endl;
    //fIonisation->Delete();
    //fStrips->Delete();
  }
  // for(Int_t ev = 0; ev<10; ev++){
  //   cout << "EVENT " << ev << endl;
  //   event = evgen->GenerateEvent();
  //   event->GenerateEvent();
  //   Int_t ntracks = event->GetNtracks();
  //   for(Int_t tr = 0; tr<ntracks; tr++){
  //     cout << "  " << tr << endl;
      
  //     TpcSimTrack* trackA = new((*array)[tr]) TpcSimTrack(0,0,10,1,1,1,0,1,ev,tr);
  //     TpcSimTrack* trackB = event->GetSimTrack(tr);

  //     trackA->SetChamber(chamber);
  //     trackB->SetChamber(chamber);

  //     trackA->GenerateTrack();
  //     trackB->GenerateTrack();
  //   }
  //   tree->Fill();
  //   array->Delete();
  //   event->Delete();

  // }

  // tree->Write();
  //  rootfile->Close();

  return 0;
}