PHG4TpcDisplayAction.cc

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file PHG4TpcDisplayAction.cc

#include "PHG4TpcDisplayAction.h"

#include <g4main/PHG4ColorDefs.h>
#include "g4main/PHG4DisplayAction.h"  // for PHG4DisplayAction

#include <TSystem.h>

#include <Geant4/G4Colour.hh>
#include <Geant4/G4LogicalVolume.hh>
#include <Geant4/G4VisAttributes.hh>
#include "Geant4/G4String.hh"  // for G4String

#include <cstdlib>
#include <iostream>
#include <utility>  // for pair

PHG4TpcDisplayAction::PHG4TpcDisplayAction(const std::string &name)
  : PHG4DisplayAction(name)
{
}

PHG4TpcDisplayAction::~PHG4TpcDisplayAction()
{
  for (auto &it : m_VisAttVec)
  {
    delete it;
  }
  m_VisAttVec.clear();
}

void PHG4TpcDisplayAction::ApplyDisplayAction(G4VPhysicalVolume * /*physvol*/)
{
  static const G4Colour color[] = {PHG4TpcColorDefs::tpc_cu_color,
                                   PHG4TpcColorDefs::tpc_pcb_color,
                                   PHG4TpcColorDefs::tpc_honeycomb_color,
                                   PHG4TpcColorDefs::tpc_cu_color,
                                   PHG4TpcColorDefs::tpc_pcb_color,
                                   PHG4TpcColorDefs::tpc_kapton_color,
                                   PHG4TpcColorDefs::tpc_cu_color,
                                   PHG4TpcColorDefs::tpc_kapton_color,
                                   PHG4TpcColorDefs::tpc_cu_color};
  // check if vis attributes exist, if so someone else has set them and we do nothing
  for (auto it : m_LogicalVolumeMap)
  {
    G4LogicalVolume *logvol = it.first;
    if (logvol->GetVisAttributes())
    {
      continue;
    }
    G4VisAttributes *visatt = new G4VisAttributes();
    visatt->SetVisibility(true);
    visatt->SetForceSolid(true);
    m_VisAttVec.push_back(visatt);  // for later deletion
    if (it.second == "TpcEnvelope")
    {
      visatt->SetVisibility(false);
    }
    else if (it.second == "TpcGas")
    {
      visatt->SetColor(PHG4TpcColorDefs::tpc_gas_color);
    }
    else if (it.second == "TpcHoneyComb")
    {
      visatt->SetColor(PHG4TpcColorDefs::tpc_honeycomb_color);
    }
    else if (it.second == "TpcWindow")
    {
      visatt->SetColor(PHG4TpcColorDefs::tpc_pcb_color);
    }
    else
    {
      std::cout << "did not assign specific color to " << it.first->GetName()
                << " under " << it.second << ".  Defaulting to TpcWindow color." << std::endl;
      visatt->SetColor(PHG4TpcColorDefs::tpc_pcb_color);

      //gSystem->Exit(1);
      //exit(1);
    }
    logvol->SetVisAttributes(visatt);
  }
  for (unsigned int i = 0; i < m_TpcInnerLayersVec.size(); i++)
  {
    G4VisAttributes *visatt = new G4VisAttributes();
    visatt->SetVisibility(true);
    visatt->SetForceSolid(true);
    m_VisAttVec.push_back(visatt);  // for later deletion
    visatt->SetColor(color[i]);
    m_TpcInnerLayersVec[i]->SetVisAttributes(visatt);
  }
  for (unsigned int i = 0; i < m_TpcOuterLayersVec.size(); i++)
  {
    G4VisAttributes *visatt = new G4VisAttributes();
    visatt->SetVisibility(true);
    visatt->SetForceSolid(true);
    m_VisAttVec.push_back(visatt);  // for later deletion
    visatt->SetColor(color[i]);
    m_TpcOuterLayersVec[i]->SetVisAttributes(visatt);
  }

  return;
}