GeoShield.cxx

Go to the documentation of this file.

//
// GeoShield
//
// GeoShield is a helper class for associating shield planes into groups.
//
// Author:  S. Kasahara 10/05

#include <iostream>
using namespace std;

#include "TGeoVolume.h"

#include "GeoGeometry/GeoShield.h"
#include "GeoGeometry/GeoShieldGroup.h"
#include "Conventions/PlaneView.h"
#include "UgliGeometry/UgliDbiSteelPln.h"
#include "MessageService/MsgService.h"

ClassImp(GeoShield)

CVSID("$Id: GeoShield.cxx,v 1.2 2007/03/01 17:13:57 rhatcher Exp $");

//_____________________________________________________________________________
GeoShield::GeoShield(GeoGeometry* geo) : fGeoGeometry(geo) {

  // Normal constructor
  MSG("Geo",Msg::kDebug) << "GeoShield normal ctor @ " << this << endl;

}

//_____________________________________________________________________________
GeoShield::~GeoShield() {
  // Destructor, delete all owned objects
  MSG("Geo",Msg::kDebug) << "GeoShield dtor @ " << this << endl;

  // reference geometry is not owned

  // Shield groups stored in map *are* owned
  GroupMapItr mapitr;
  for ( mapitr = fGroupMap.begin(); mapitr != fGroupMap.end(); mapitr++ ) {
    GeoShieldGroup* shieldgrp = mapitr -> second;
    // reclaim group memory before map is destructed
    if ( shieldgrp ) delete shieldgrp; shieldgrp = 0;
  }

}

//_____________________________________________________________________________
void GeoShield::Print(Option_t* /* option */) const {
   // print to cout

  GroupMapConstItr mapitr;
  for ( mapitr=fGroupMap.begin(); mapitr != fGroupMap.end(); mapitr++ ) {
    GeoShieldGroup* shieldgrp = mapitr -> second;
    shieldgrp -> Print();
  }
  
}

//_____________________________________________________________________________
std::list<const TGeoVolume*> 
                GeoShield::GetListOfVolumes(EGroupType shieldgrp) const {
   // retrieve list of volumes in group from map

  GroupMapConstItr mapitr;
  std::list<const TGeoVolume*> vollist; // empty list
  
  mapitr = fGroupMap.find(shieldgrp);
  if ( mapitr != fGroupMap.end() ) {
    GeoShieldGroup* shieldgrp = mapitr->second;
    return (shieldgrp -> GetListOfVolumes()); // return a copy to the caller
  }

  return vollist; // retun empty list if group not found in map
  
}

//_____________________________________________________________________________
const char* GeoShield::AsString(EGroupType shieldgrp) {
  // Convert enumerated shield group type to a string

  switch (shieldgrp ) {
  case kFarInnerE:
    return "FarInnerE";
    
  case kFarInnerW:
    return "FarInnerW";

  case kFarOuterE:
    return "FarOuterE";

  case kFarOuterW:
    return "FarOuterW";

  case kFarTop:
    return "FarTop";

  default:
    MSG("GeoShield",Msg::kWarning) 
    << "GeoShield::AsString called with unknown shield group type "
    << (int)shieldgrp << "." << endl;
    return "UNKNOWN";

  } // end of switch
  
}

//_____________________________________________________________________________
GeoShield::EGroupType GeoShield::AddVolume(const TGeoVolume* vol,
                                           PlexPlaneId steelId, 
                                           const UgliDbiSteelPln* stRow) {
  // The stRow is used to extract positional information in order to
  // determine size of bounding box.  Return the shield group to which
  // this volume belongs.

  if ( !steelId.IsVetoShield() ) {
    MSG("Geo",Msg::kFatal) 
      << "GeoShield::AddVolume called with non-vetoshield volume!" << endl;
    abort();
  }
  // Determine the type of veto-shield plane from the placement and
  // db data
  GeoShield::EGroupType shieldgrp = kUnknown;
  switch ( steelId.GetPlaneView() ) {

  case PlaneView::kVSTopFlat:
  case PlaneView::kVSTopEastSlant:
  case PlaneView::kVSTopWestSlant:
    shieldgrp = kFarTop;
    break;

  case PlaneView::kVSWallEastSlant:
    shieldgrp = kFarOuterE;
    break;

  case PlaneView::kVSWallWestSlant:
    shieldgrp = kFarOuterW;
    break;
    
  case PlaneView::kVSWallOnEdge: 
    {
      Float_t x0 = stRow->GetX0(); // meters
      if ( TMath::Abs(x0) < 5. ) {
        // inner
        if ( x0 < 0. ) shieldgrp = kFarInnerE;
        else shieldgrp = kFarInnerW;
      }
      else {
        // outer
        if ( x0 < 0. ) shieldgrp = kFarOuterE;
        else shieldgrp = kFarOuterW;
      }
    }
    break;
    
  default:
    MSG("Geo",Msg::kFatal) << "PlaneView " 
                           << PlaneView::AsString(steelId.GetPlaneView())
                           << " unknown!. Abort." << endl;
    abort();

  } // end of plane view switch

  GeoShieldGroup* grp = fGroupMap[shieldgrp];
  if ( grp == 0 ) { 
    grp = new GeoShieldGroup(fGeoGeometry,shieldgrp);
    fGroupMap[shieldgrp] = grp;
  }
  grp -> AddVolume(vol,stRow);
  
  return shieldgrp;
  
}

//_____________________________________________________________________________
void GeoShield::BuildGroupNodes(TGeoVolume* hallVol) {
  // Build shield group nodes.  Protected method called by GeoGeometry
  // when all shield volumes have been added with GeoShield::AddVolume()
  // Shield group nodes will be added to hall volume

  
  GroupMapConstItr mapitr;
  for ( mapitr=fGroupMap.begin(); mapitr!=fGroupMap.end(); mapitr++ ) {
    GeoShieldGroup* grp = mapitr -> second;
    grp -> BuildNode(hallVol);
  }
  
  return;
  
}

---