GeoVXDAssembly.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2010 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Martin Ritter                                            *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#ifndef GEOVXDASSEMBLY_H
#define GEOVXDASSEMBLY_H
 
#include <vector>
#include <G4Transform3D.hh>
 
class G4LogicalVolume;
 
namespace Belle2 {
  namespace VXD {
 
    class GeoVXDAssembly {
    public:
      typedef std::pair<G4LogicalVolume*, G4Transform3D> Placement;
 
      void place(G4LogicalVolume* mother, const G4Transform3D& transform);
 
      void add(G4LogicalVolume* volume, const G4Transform3D& transform = G4Transform3D())
      {
        m_volumes.push_back(std::make_pair(volume, transform));
      }
 
      void add(const GeoVXDAssembly& assembly, const G4Transform3D& transform = G4Transform3D())
      {
        for (const Placement& p : assembly.m_volumes) {
          m_volumes.push_back(std::make_pair(p.first, transform * p.second));
        }
      }
 
    private:
      std::vector<Placement> m_volumes;
    };
  } //VXD namespace
} //Belle2 namespace
#endif