SimpleBeamlineMagnet_geo.cpp

Go to the documentation of this file.

#include "DD4hep/DetFactoryHelper.h"
#include "DD4hep/Printout.h"
#include "DD4hep/Shapes.h"
#include "TMath.h"
#include "DDRec/Surface.h"
#include "DDRec/DetectorData.h"
#include "XML/Layering.h"
 
using namespace std;
using namespace dd4hep;
using namespace dd4hep::rec;
using namespace ROOT::Math;
 
static Ref_t build_magnet(Detector& dtor, xml_h e, SensitiveDetector sens) {
    xml_det_t x_det     = e;
    int       det_id    = x_det.id();
    string    det_name  = x_det.nameStr();
    xml_dim_t pos       = x_det.child(_U(placement));
    double    pos_x     = pos.x();
    double    pos_y     = pos.y();
    double    pos_z     = pos.z();
    double    pos_theta = pos.attr<double>(_U(theta));
    xml_dim_t dims      = x_det.dimensions();
    double    dim_x     = dims.x();
    double    dim_y     = dims.y();
    double    dim_z     = dims.z();
    xml_dim_t apperture = x_det.child(_Unicode(apperture));
    double    app_x     = apperture.x();
    double    app_y     = apperture.y();
 
    //xml_dim_t coil      = x_det.child(_Unicode(coil));
    //double    coil_x    = coil.dx();
    //double    coil_y    = coil.dy();
    
    Material  iron      = dtor.material("Air");
    Material  niobium   = dtor.material("Air");
 
    DetElement sdet(det_name, det_id);
    Assembly   assembly(det_name + "_assembly");
 
    string module_name = "SimpleBeamlineMagnet";
    string vis0 = dd4hep::getAttrOrDefault(x_det, _Unicode(vis), "BlueVis");
 
    sdet.setAttributes(dtor, assembly, x_det.regionStr(), x_det.limitsStr(), x_det.visStr());
 
    // -- yoke --
    Box              yoke_outer(0.5 * dim_x, 0.5 * dim_y, 0.5 * dim_z);
    Box              app_hole(0.5 * app_x, 0.5 * app_y, 0.5 * dim_z + 1.0 * m);
    SubtractionSolid yoke(yoke_outer, app_hole);
    Volume           yoke_vol("yoke_vol", yoke, iron);
    DetElement       yoke_DE(sdet, "yoke1", 1);
 
    auto yoke_pv = assembly.placeVolume(yoke_vol, Transform3D(RotationY(-pos_theta)));
    yoke_pv.addPhysVolID("element", 1);
    yoke_vol.setVisAttributes("RedVis");
    yoke_DE.setPlacement(yoke_pv);
    yoke_DE.setAttributes(dtor, yoke_vol, x_det.regionStr(), x_det.limitsStr(), x_det.visStr());
 
    //auto rod_pos_up = Transform3D(Translation3D(0.5 * (app_x - coil_y), 0.5 * (app_y - coil_x), 0.0));
    //auto rod_pos_down = Transform3D(RotationZ(M_PI) * Translation3D(0.5 * (app_x - coil_y), 0.5 * (app_y - coil_x), 0.0));
 
    //Box long_rod(0.5 * coil_y, 0.5 * coil_x, 0.5 * dim_z);
    //Box short_rod(0.5 * coil_y, coil_x, 0.5 * coil_x);
    //Box top_rod(0.5 * app_x, 0.5 * coil_y, 0.5 * coil_x);
    //UnionSolid half_rod1(long_rod, short_rod, Position(0.0, 0.5 * coil_x, -0.5 * (dim_z + coil_x)));
    //UnionSolid half_rod2(half_rod1, short_rod, Position(0.0, 0.5 * coil_x, 0.5 * (dim_z + coil_x)));
    //UnionSolid half_rod3(half_rod2, top_rod, Position(-0.5 * (app_x - coil_y), coil_x + 1.5*coil_y, -0.5 * (dim_z + coil_x)));
    //UnionSolid half_rod4(half_rod3, top_rod, Position(-0.5 * (app_x - coil_y), coil_x + 1.5*coil_y, 0.5 * (dim_z + coil_x)));
    //UnionSolid half_rod5(half_rod4, long_rod, Position(-app_x + coil_y, 0.0, 0.0));
    //UnionSolid half_rod6(half_rod5, short_rod, Position(-app_x + coil_y, 0.5 * coil_x, -0.5 * (dim_z + coil_x)));
    //UnionSolid half_rod7(half_rod6, short_rod, Position(-app_x + coil_y, 0.5 * coil_x, 0.5 * (dim_z + coil_x)));
    //Volume coil_up_vol("coil_up_vol", half_rod7, niobium);
    //Volume coil_down_vol("coil_down_vol", half_rod7, niobium);
    //coil_up_vol.setVisAttributes("BlueVis");
    //coil_down_vol.setVisAttributes("BlueVis");
    //auto rod_pv = yoke_vol.placeVolume(coil_up_vol, rod_pos_up);
    //rod_pv = yoke_vol.placeVolume(coil_down_vol, rod_pos_down);
    
 
    // -- finishing steps
    auto pv = dtor.pickMotherVolume(sdet).placeVolume(assembly, Position(pos_x, pos_y, pos_z));
    pv.addPhysVolID("system", det_id);
    sdet.setPlacement(pv);
 
    assembly->GetShape()->ComputeBBox();
    return sdet;
}
 
DECLARE_DETELEMENT(SimpleBeamlineMagnet, build_magnet)