PTGuiRollGMinos.cxx

Go to the documentation of this file.

//
// PTGuiRollGMinos
//
// A conversion class to roll parameters (media, materials, cuts, etc.)
// from a TGeant3 simulation to a format used by the PTGui
//
// This code is based on that of geant3_vmc/TGeant3/G3toRoot class:
//   * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
//   *                                                                        *
//   * Author: The ALICE Off-line Project.                                    *
//   * Contributors are mentioned in the code where appropriate.              *
//   Primary Author: Andreas Morsch
//
// but has been adapted for the MINOS PTSim simulation.
//
// Adapted for MINOS: S. Kasahara 8/2006
//

#include <iostream>
using namespace std;

#include <TClonesArray.h>
#include <TFile.h>
#include <TGraph.h>
#include <TH1F.h>

#include "MessageService/MsgService.h"
#include "ParticleTransportSim/Gui/PTGuiRollGMinos.h"
#include "ParticleTransportSim/Gui/PTGuiMedium.h"
#include "ParticleTransportSim/Gui/PTGuiMaterial.h"

CVSID("$Id: PTGuiRollGMinos.cxx,v 1.3 2007/08/28 23:37:37 schubert Exp $");

ClassImp(PTGuiRollGMinos)

// Definition of methods
// *********************

//_____________________________________________________________________________
PTGuiRollGMinos::PTGuiRollGMinos(TFile* file) : fFile(file),
                                   fMaterials(0), fMedia(0) {
  // Normal Constructor
  
  if ( !fFile ) {
    MSG("PTGui",Msg::kError) << "PTGuiRollGMinos ctor received null ptr." 
                             << endl;
    abort();
  }
  
  // Convert GMinos objects to PTGui objects
  RollMaterials();
  RollMedia();
    
}

//_____________________________________________________________________________
PTGuiRollGMinos::~PTGuiRollGMinos() {
  // Destructor

  // Delete TFile and contained objects
  if ( fFile ) { 
    fFile -> Close();
    delete fFile; fFile = 0;
  }
  
  // Delete TClonesArrays and contained objects
  //if ( fMaterials ) { fMaterials->Delete(); delete fMaterials; }
  //if ( fMedia ) { fMedia -> Delete(); delete fMedia; }
  
}

//_____________________________________________________________________________
PTGuiMedium* PTGuiRollGMinos::GetMedium(std::string ptname) const {
  // Method to find corresponding gminos medium
  // during construction of medium list.

  PTGuiMedium* medium = 0;
  std::map<std::string,std::string>::const_iterator itr 
                                           = fPTtoGMinosMap.find(ptname);
  if ( itr != fPTtoGMinosMap.end() ) {
    std::string gminosname = itr->second;
    medium 
        = dynamic_cast<PTGuiMedium*>(fMedia->FindObject(gminosname.c_str()));
  }

  return medium;
  
}

//_____________________________________________________________________________
TGraph* PTGuiRollGMinos::GetGraph(Int_t imat, Int_t ipart, Int_t imec) const {
  // Method to pull corresponding gminos histogram from file

  TGraph* graph = 0;
  Int_t id = 10000*imat + 100*ipart + imec;

  std::string histname;
  
  if ( imat < 10 ) {
    // This is kind of stupid but I'm tired
    char buf[6];
    sprintf(buf,"%5i",id);
    buf[5] = '\0';
    histname = "h"+std::string(buf);
  }
  else {
    char buf[7];
    sprintf(buf,"%6i",id);
    buf[6] = '\0';
    histname = "h"+std::string(buf);
  }
  
  TH1* hist = dynamic_cast<TH1*>(fFile->Get(histname.c_str()));
  if ( hist ) {
    graph = new TGraph(hist);
  }
  else {
    cout << "Unable to find histogram." << endl;
  }
   
  return graph; // graph is owned by caller
  
}

  
//_____________________________________________________________________________
void PTGuiRollGMinos::RollMaterials() {
  // Read materials from TGeoManager arrays and puts in a TClonesArray
  // of PTGuiMaterial objects

  Int_t nentries = 0;
  
  fMaterials = dynamic_cast<TClonesArray*>(fFile->Get("GMinosMaterial"));
  TIter iter(fMaterials);
  while ( TObject* obj = iter.Next() ) {
    PTGuiMaterial* mat = dynamic_cast<PTGuiMaterial*>(obj);
    std::string matName = mat->GetName();
    // Fuss with material name to remove trailing spaces
    
    int pos(matName.size());
    for ( ; pos && matName[pos-1]==' '; --pos );
    matName.erase(pos,matName.size()-pos);
    mat->SetName(matName.c_str());
      
    mat -> Print();
    nentries++;
    
  }

}
  
//_____________________________________________________________________________
void PTGuiRollGMinos::RollMedia() {
  // Read materials from TGeoManager arrays and puts in a TClonesArray
  // of PTGuiMedia objects

  // Set up media map names to navigate from PT to GMINOS names
  fPTtoGMinosMap.insert(make_pair("ALUM","ALUMINIUM"));
  fPTtoGMinosMap.insert(make_pair("ALUM_B","ALUB MAGNETIC AL"));
  fPTtoGMinosMap.insert(make_pair("IRON","STEEL"));
  fPTtoGMinosMap.insert(make_pair("IRON_B","IRON"));
  fPTtoGMinosMap.insert(make_pair("AIR","AIR"));
  fPTtoGMinosMap.insert(make_pair("AIR_B","AIRB MAGNETIC AIR"));
  fPTtoGMinosMap.insert(make_pair("CONCRETE","CONCRETE"));
  fPTtoGMinosMap.insert(make_pair("ROCK","ROCK"));
  fPTtoGMinosMap.insert(make_pair("PSTYRENE SCINT","PSTYRENE SCINT"));
  fPTtoGMinosMap.insert(make_pair("COEX TIO2 PSTYRENE",
                                  "COEX TIO2 PSTYRENE"));
  fPTtoGMinosMap.insert(make_pair("FARCOIL","<no match>"));
  fPTtoGMinosMap.insert(make_pair("FARCOIL_B","FARCOIL"));
  fPTtoGMinosMap.insert(make_pair("DOLOMITE","DOLOMITE"));
  fPTtoGMinosMap.insert(make_pair("FEPL","STPL"));
  fPTtoGMinosMap.insert(make_pair("FEPL_B","FEPL"));
  fPTtoGMinosMap.insert(make_pair("WATER","WATER"));
  fPTtoGMinosMap.insert(make_pair("WATER_B","WATB MATGNETIC WATER"));

  Int_t nentries = 0;
  
  fMedia = dynamic_cast<TClonesArray*>(fFile->Get("GMinosMedium"));
  TIter iter(fMedia);
  while ( TObject* obj = iter.Next() ) {

    PTGuiMedium* med = dynamic_cast<PTGuiMedium*>(obj);
    std::string medName = med->GetName();

    // Fuss with the medium name to remove trailing spaces
    int pos(medName.size());
    for ( ; pos && medName[pos-1]==' '; --pos );
    medName.erase(pos,medName.size()-pos);
    med -> SetName(medName.c_str());
    
    med -> Print();
    nentries++;
  }

  PTGuiMedium* medium = new PTGuiMedium("<no match>",-1,-1,-1,-1,
                                        -1,-1,-1,-1,-1,-1);
  (*fMedia)[nentries] = medium;
  
}

  

---