static Ref_t create_detector(Detector& theDetector,xml_h e,SensitiveDetector sens){
typedef vector<PlacedVolume>Placements;
xml_det_t x_det=e;
Material vacuum=theDetector.vacuum();
int det_id= x_det.id();
string det_name=x_det.nameStr();
bool reflect = x_det.reflect(false);
DetElement sdet(det_name,det_id);
int m_id=0,c_id=0,n_sensor=0;
map<string,Volume>modules;
map<string,Placements>sensitives;
PlacedVolume pv;
//encoding that was missing
std::string cellIDEncoding=sens.readout().idSpec().fieldDescription();
UTIL::BitField64 encoder(cellIDEncoding);
encoder.reset();
encoder[lcio::LCTrackerCellID::subdet()]=det_id;
// --- create an envelope volume and position it into the world ---------------------
Volume envelope=dd4hep::xml::createPlacedEnvelope(theDetector,e,sdet);
dd4hep::xml::setDetectorTypeFlag(e,sdet);
if(theDetector.buildType() == BUILD_ENVELOPE)return sdet;
//-----------------------------------------------------------------------------------
dd4hep::rec::ZDiskPetalsData* zDiskPetalsData=new dd4hep::rec::ZDiskPetalsData;
//neighbour surfaces added
dd4hep::rec::NeighbourSurfacesData* neighbourSurfacesData=new dd4hep::rec::NeighbourSurfacesData();
//
std::map< std::string, double > moduleSensThickness;
envelope.setVisAttributes(theDetector.invisible());
sens.setType("tracker");
for(xml_coll_t mi(x_det,_U(module));mi;++mi,++m_id){
xml_comp_t x_mod=mi;
string m_nam=x_mod.nameStr();
xml_comp_t trd=x_mod.trd();
double posY;
double x1=trd.x1();
double x2=trd.x2();
double z=trd.z();
double y1,y2,total_thickness=0.;
xml_coll_t ci(x_mod,_U(module_component));
for(ci.reset(),total_thickness=0.0;ci;++ci)
total_thickness += xml_comp_t(ci).thickness();
y1 = y2 = total_thickness / 2;
Volume m_volume(m_nam, Trapezoid(x1, x2, y1, y2, z), vacuum);
m_volume.setVisAttributes(theDetector.visAttributes(x_mod.visStr()));
// Loop over slices
// The first slice (top in the xml) is placed at the "bottom" of the module
for(ci.reset(), n_sensor=1, c_id=0, posY=-y1; ci; ++ci, ++c_id){
xml_comp_t c=ci;
double c_thick=c.thickness();
Material c_mat=theDetector.material(c.materialStr());
string c_name=_toString(c_id,"component%d");
Volume c_vol(c_name, Trapezoid(x1,x2,c_thick/2e0,c_thick/2e0,z), c_mat);
c_vol.setVisAttributes(theDetector.visAttributes(c.visStr()));
pv = m_volume.placeVolume(c_vol,Position(0,posY+c_thick/2,0));
if (c.isSensitive()){
c_vol.setSensitiveDetector(sens);
sensitives[m_nam].push_back(pv);
++n_sensor;
}
posY += c_thick;
}
modules[m_nam] = m_volume;
}
for(xml_coll_t li(x_det,_U(layer));li;++li){
xml_comp_t x_layer(li);
int l_id=x_layer.id();
int mod_num=0;
int ring_num=0;
double sumZ(0.),innerR(1e100),outerR(0.);
//loop only to count the number of rings in a disk - it is then needed for looking for neighborous when you are in a "border" cell
int nrings = 0;
for(xml_coll_t ri(x_layer,_U(ring)); ri; ++ri) {
nrings++;
}
dd4hep::rec::ZDiskPetalsData::LayerLayout thisLayer;
for(xml_coll_t ri(x_layer,_U(ring)); ri; ++ri) {
xml_comp_t x_ring = ri;
double r=x_ring.r();
double phi0=x_ring.phi0(0);
double zstart=x_ring.zstart();
double dz=x_ring.dz(0);
int nmodules=x_ring.nmodules();
string m_nam=x_ring.moduleStr();
Volume m_vol=modules[m_nam];
double iphi=2*M_PI/nmodules;
double phi=phi0;
Placements& sensVols=sensitives[m_nam];
Box mod_shape(m_vol.solid());
if(r-mod_shape->GetDZ()<innerR)
innerR=r-mod_shape->GetDZ();
if(r+mod_shape->GetDZ()>outerR)
outerR=r+mod_shape->GetDZ();
sumZ+=zstart;
r=r+mod_shape->GetDY();
for(int k=0;k<nmodules;++k){
string m_base=_toString(l_id,"layer%d")+_toString(mod_num,"_module%d")+_toString(k,"_sensor%d");
double x=-r*std::cos(phi);
double y=-r*std::sin(phi);
DetElement module(sdet,m_base+"_pos",det_id);
pv=envelope.placeVolume(m_vol,Transform3D(RotationZYX(0,-M_PI/2-phi,-M_PI/2),Position(x,y,zstart+dz)));
pv.addPhysVolID("side",1).addPhysVolID("layer", l_id).addPhysVolID("module",mod_num).addPhysVolID("sensor",k);
module.setPlacement(pv);
for(size_t ic=0;ic<sensVols.size();++ic){
PlacedVolume sens_pv=sensVols[ic];
DetElement comp_elt(module,sens_pv.volume().name(),mod_num);
comp_elt.setPlacement(sens_pv);
}
if(reflect){
pv = envelope.placeVolume(m_vol,Transform3D(RotationZYX(M_PI,-M_PI/2-phi,-M_PI/2),Position(x,y,-zstart-dz)));
pv.addPhysVolID("side",-1).addPhysVolID("layer",l_id).addPhysVolID("module",mod_num).addPhysVolID("sensor",k);
DetElement r_module(sdet,m_base+"_neg",det_id);
r_module.setPlacement(pv);
for(size_t ic=0;ic<sensVols.size();++ic){
PlacedVolume sens_pv=sensVols[ic];
DetElement comp_elt(r_module,sens_pv.volume().name(),mod_num);
comp_elt.setPlacement(sens_pv);
}
}
//modified on comparison with TrackerEndcap_o2_v06_geo.cpp
//get cellID and fill map< cellID of surface, vector of cellID of neighbouring surfaces >
dd4hep::long64 cellID_reflect;
if(reflect){
encoder[lcio::LCTrackerCellID::side()]=lcio::ILDDetID::bwd;
encoder[lcio::LCTrackerCellID::layer()]=l_id;
encoder[lcio::LCTrackerCellID::module()]=mod_num;
encoder[lcio::LCTrackerCellID::sensor()]=k;
cellID_reflect=encoder.lowWord(); // 32 bits
}
encoder[lcio::LCTrackerCellID::side()]=lcio::ILDDetID::fwd;
encoder[lcio::LCTrackerCellID::layer()]=l_id;
encoder[lcio::LCTrackerCellID::module()]=mod_num;
encoder[lcio::LCTrackerCellID::sensor()]=k;
dd4hep::long64 cellID = encoder.lowWord(); // 32 bits
//compute neighbours
int n_neighbours_module = 1; // 1 gives the adjacent modules (i do not think we would like to change this)
int n_neighbours_sensor = 1;
int newmodule=0,newsensor=0;
for(int imodule=-n_neighbours_module; imodule<=n_neighbours_module; imodule++){ // neighbouring modules
for(int isensor=-n_neighbours_sensor; isensor<=n_neighbours_sensor; isensor++){ // neighbouring sensors
if (imodule==0 && isensor==0) continue; // cellID we started with
newmodule = mod_num + imodule;
newsensor = k + isensor;
//compute special case at the boundary
//general computation to allow (if necessary) more then adiacent neighbours (ie: +-2)
if (newsensor < 0) newsensor = nmodules + newsensor;
if (newsensor >= nmodules) newsensor = newsensor - nmodules;
if (newmodule < 0 || newmodule >= nrings)continue; //out of disk
//encoding
encoder[lcio::LCTrackerCellID::module()] = newmodule;
encoder[lcio::LCTrackerCellID::sensor()] = newsensor;
neighbourSurfacesData->sameLayer[cellID].push_back(encoder.lowWord());
if (reflect){
encoder[lcio::LCTrackerCellID::side()]=lcio::ILDDetID::bwd;
encoder[lcio::LCTrackerCellID::layer()]=l_id;
encoder[lcio::LCTrackerCellID::module()]=newmodule;
encoder[lcio::LCTrackerCellID::sensor()]=newsensor;
neighbourSurfacesData->sameLayer[cellID_reflect].push_back(encoder.lowWord());
}
}
}
dz = -dz;
phi += iphi;
}
++mod_num;
++ring_num;
}
// Only filling what is needed for CED/DDMarlinPandora
thisLayer.zPosition=sumZ/ring_num; // average z
thisLayer.distanceSensitive=innerR;
thisLayer.lengthSensitive=outerR - innerR;
thisLayer.petalNumber=ring_num; // number of rings in petalNumber, needed for tracking
zDiskPetalsData->layers.push_back(thisLayer);
}
sdet.setAttributes(theDetector,envelope,x_det.regionStr(),x_det.limitsStr(),x_det.visStr());
sdet.addExtension<dd4hep::rec::ZDiskPetalsData>(zDiskPetalsData);
//added extension
sdet.addExtension<dd4hep::rec::NeighbourSurfacesData>(neighbourSurfacesData);
std::cout<<"XXX Tracker endcap layers:"<<zDiskPetalsData->layers.size()<<std::endl;
return sdet;
}
static Ref_t create_detector(Detector& theDetector, xml_h e, SensitiveDetector sens) {
typedef vector<PlacedVolume> Placements;
xml_det_t x_det = e;
Material vacuum = theDetector.vacuum();
int det_id = x_det.id();
string det_name = x_det.nameStr();
bool reflect = x_det.reflect(false);
DetElement sdet (det_name,det_id);
int m_id=0, c_id=0, n_sensor=0;
map<string, Volume> modules;
map<string, Placements> sensitives;
PlacedVolume pv;
// --- create an envelope volume and position it into the world ---------------------
Volume envelope = dd4hep::xml::createPlacedEnvelope(theDetector, e, sdet);
dd4hep::xml::setDetectorTypeFlag(e, sdet);
if( theDetector.buildType() == BUILD_ENVELOPE ) return sdet;
//-----------------------------------------------------------------------------------
envelope.setVisAttributes(theDetector.invisible());
sens.setType("tracker");
// Build the sensor units
// Loop over 'modules' as defined in the XML
for(xml_coll_t mi(x_det,_U(module)); mi; ++mi, ++m_id) {
xml_comp_t x_mod = mi;
string m_nam = x_mod.nameStr();
xml_comp_t trd = x_mod.trd();
double posY;
double x1 = trd.x1();
double x2 = trd.x2();
double z = trd.z();
double y1, y2, total_thickness=0.;
xml_coll_t ci(x_mod, _U(module_component));
for(ci.reset(), total_thickness=0.0; ci; ++ci)
total_thickness += xml_comp_t(ci).thickness();
y1 = y2 = total_thickness / 2;
Volume m_volume(m_nam, Trapezoid(x1, x2, y1, y2, z), vacuum);
m_volume.setVisAttributes(theDetector.visAttributes(x_mod.visStr()));
std::cout << m_nam << ", thickness=" << total_thickness << std::endl;
// Loop over the module_components ('slices') in the 'module'
// The first component (top in the XML) is placed at the 'bottom'
for(ci.reset(), n_sensor=1, c_id=0, posY=-y1; ci; ++ci, ++c_id) {
xml_comp_t c = ci;
double c_thick = c.thickness();
Material c_mat = theDetector.material(c.materialStr());
string c_name = _toString(c_id, "component%d");
Volume c_vol(c_name, Trapezoid(x1,x2,c_thick/2e0,c_thick/2e0,z), c_mat);
std::cout << " + sensor " << n_sensor << " " << c_name;
c_vol.setVisAttributes(theDetector.visAttributes(c.visStr()));
pv = m_volume.placeVolume(c_vol, Position(0, posY + c_thick/2, 0));
if ( c.isSensitive() ) {
sdet.check(n_sensor > 2, "SiTrackerEndcap::fromCompact: " + c_name + " Max of 2 modules allowed!");
pv.addPhysVolID("sensor", n_sensor);
c_vol.setSensitiveDetector(sens);
sensitives[m_nam].push_back(pv);
std::cout << " (" << n_sensor << " is sensitive) ";
++n_sensor;
}
std::cout << std::endl;
posY += c_thick;
}
modules[m_nam] = m_volume;
}
// done building the 2 modules, of 12 layers each
int mod_count[12] = {0};
// Build now the detector itself
// Loop over layers as defined in the XML
for(xml_coll_t li(x_det, _U(layer)); li; ++li) {
xml_comp_t x_layer(li);
int l_id = x_layer.id();
int ring_num = 0;
std::cout << "Layer " << l_id << ":" << std::endl;
// Loop over rings, as defined in the XML
for(xml_coll_t ri(x_layer, _U(ring)); ri; ++ri) {
xml_comp_t x_ring = ri;
double r = x_ring.r();
double phi0 = x_ring.phi0(0);
double zstart = x_ring.zstart();
double dz = x_ring.dz(0);
int nmodules = x_ring.nmodules();
string m_nam = x_ring.moduleStr();
Volume m_vol = modules[m_nam];
double iphi = 2*M_PI/nmodules;
double phi = phi0;
Placements& sensVols = sensitives[m_nam];
// This driver version encodes the rings as layers and the
// petals as modules, such that 'layer' 1 contains all innermost rings
// and last 'layer' contains the outermost rings in the tracker
// farthest away on z from the IP (unintuititive, but works)
std::cout << " Ring " << ring_num << ":" << std::endl;
// Loop over modules in each ring, modules are either type 1 or 2
for(int k=0; k < nmodules; ++k) {
double x = -r*std::cos(phi);
double y = -r*std::sin(phi);
for(int s=1-2*int(reflect); s<2; s+=1+int(reflect)){
string e_name = _toString(s, "side%d") + _toString(l_id, "_layer%d") + _toString(ring_num, "_ring%d") + _toString(k, "_sensor%d");
DetElement module(sdet, e_name, det_id);
pv = envelope.placeVolume(m_vol, Transform3D(RotationZYX(0,-M_PI/2-phi,-M_PI/2), Position(x, y, s*(zstart+dz) )));
pv.addPhysVolID("side", s).addPhysVolID("layer", ring_num).addPhysVolID("module", mod_count[ring_num] + k);
module.setPlacement(pv);
for(size_t ic=0; ic<sensVols.size(); ++ic) {
PlacedVolume sens_pv = sensVols[ic];
DetElement comp_elt(module, sens_pv.volume().name(), det_id);
comp_elt.setPlacement(sens_pv);
std::cout << "Name: " << e_name << "_" << sens_pv.volume().name() << std::endl;
std::cout << " ID: side " << s << ", layer " << ring_num << ", module " << mod_count[ring_num] + k << ", sensor" << ic+1 << std::endl;
}
}
dz = -dz;
phi += iphi;
}
mod_count[ring_num] += nmodules;
++ring_num;
}
}
std::cout << "Number of modules per 'layer':" << std::endl;
for(int ii=0; ii<12; ii++){
std::cout << " mod_count[" << ii << "] = " << mod_count[ii] << std::endl;
}
return sdet;
}