From c315293dc3f7b9db33c915e8ee8a884377bebaf2 Mon Sep 17 00:00:00 2001
From: Jihee Kim <jihee.kim@anl.gov>
Date: Thu, 15 Apr 2021 13:45:01 +0000
Subject: [PATCH] Resolve "Implement cb_VTX"

---
 compact/cb_VTX_Barrel.xml |  39 +++++++++++++
 compact/definitions.xml   |  13 ++++-
 compact/display.xml       |   2 +
 reference_detector.xml    |   3 +-
 src/cb_VTX_Barrel.cpp     | 117 ++++++++++++++++++++++++++++++++++++++
 5 files changed, 171 insertions(+), 3 deletions(-)
 create mode 100644 compact/cb_VTX_Barrel.xml
 create mode 100644 src/cb_VTX_Barrel.cpp

diff --git a/compact/cb_VTX_Barrel.xml b/compact/cb_VTX_Barrel.xml
new file mode 100644
index 00000000..4481871a
--- /dev/null
+++ b/compact/cb_VTX_Barrel.xml
@@ -0,0 +1,39 @@
+<lccdd>
+
+  <comment> Central Barrel Vertex Detector </comment>        
+
+  <define>
+  </define>
+
+  <limits>
+  </limits>
+
+  <regions>
+  </regions>
+
+  <display>
+  </display>
+
+  <detectors>
+    <detector id="cb_VTX_Barrel_ID" name="cb_VTX_Barrel" type="cb_VTX_Barrel" readout="cb_VTX_Barrel_Hits" vis="cb_VTX_BarrelVis" insideTrackingVloume="true">
+      <dimensions rmin="cb_VTX_Barrel_rmin" rmax="cb_VTX_Barrel_rmax" length="cb_VTX_Barrel_length"/>
+      <position x="cb_VTX_Barrel_x_pos" y="cb_VTX_Barrel_y_pos" z="cb_VTX_Barrel_z_pos"/>
+      <layer repeat="cb_VTX_Barrel_layer" vis="cb_VTX_Barrel_layerVis">
+	<slice name="Silicon_slice" material="Si" sensitive="true"/>
+      </layer>
+    </detector>
+  </detectors>
+
+  <readouts>
+    <readout name="cb_VTX_Barrel_Hits">
+      <segmentation type="CartesianGridXY" grid_size_x="1.0*mm" grid_size_y="1.0*mm" />
+      <id>system:8,barrel:3,layer:4,module:8,x:32:-16,y:-16</id>  
+    </readout>
+  </readouts>
+
+  <plugins>
+  </plugins>
+
+  <fields>
+  </fields>
+</lccdd>
diff --git a/compact/definitions.xml b/compact/definitions.xml
index 01a5eeba..5b7f7253 100644
--- a/compact/definitions.xml
+++ b/compact/definitions.xml
@@ -68,7 +68,7 @@
       Layer 4 subassembly ID: 29
       Layer 5 subassembly ID: 30
     </comment>
-    <constant name="SiVertexSubAssembly_ID" value="25"/>
+    <constant name="cb_VTX_Barrel_ID" value="25"/>
     <comment>
       Unused values reserved for vertex:
       constant name="SiVertexSubAssemblyLayer1_ID" value="121"
@@ -337,7 +337,16 @@
     <constant name="VertexTrackerOuterRadius"       value="90.0*mm"/>
     <constant name="VertexTrackerOuterBarrelLength" value="400.0*mm"/>
 
+    
+    <constant name="cb_VTX_Barrel_rmin"   value="33.0*mm"/>
+    <constant name="cb_VTX_Barrel_rmax"   value="165.0*mm"/>
+    <constant name="cb_VTX_Barrel_length" value="740.0*mm"/>
+
+    <constant name="cb_VTX_Barrel_x_pos"   value="0.0*mm"/>
+    <constant name="cb_VTX_Barrel_y_pos"   value="0.0*mm"/>
+    <constant name="cb_VTX_Barrel_z_pos"   value="0.0*mm"/>
 
+    <constant name="cb_VTX_Barrel_layer"   value="6"/>
     <comment>
       --------------------------
       Silicon Tracker Parameters
@@ -370,7 +379,7 @@
 
       The cb_CTD_* parameters can probably be moved to the detector specific xml file.
     </comment>
-    <constant name="cb_CTD_rmin"   value="SiliconTrackerInnerRadius"/>
+    <constant name="cb_CTD_rmin"   value="SiliconTrackerInnerRadius + 70.0 * mm"/>
     <constant name="cb_CTD_rmax"   value="SiliconTrackerOuterRadius"/>
     <constant name="cb_CTD_zmax"   value="SiliconTrackerOuterBarrel_zmax"/>
     <constant name="cb_CTD_length" value="SiliconTrackerOuterBarrelLength"/>
diff --git a/compact/display.xml b/compact/display.xml
index d0733f6d..a51b7690 100644
--- a/compact/display.xml
+++ b/compact/display.xml
@@ -55,6 +55,8 @@
     <vis name="cb_CTD_Si_layerVis" alpha="1.0"  r= "0.9"  g="1.0"  b="0.1"  showDaughters="true" visible="true"/>
     <vis name="ce_GEMVis"          alpha="0.1"  r= "0.1"  g="0.0"  b="1.0"  showDaughters="true" visible="true"/>
     <vis name="cb_GEM_layerVis"    alpha="0.8"  r= "0.8"  g="0.4"  b="0.3"  showDaughters="true" visible="true"/>
+    <vis name="cb_VTX_BarrelVis"          alpha="0.1"  r= "0.1"  g="0.0"  b="1.0"  showDaughters="true" visible="true"/>
+    <vis name="cb_VTX_Barrel_layerVis"    alpha="2.0"  r= "0.0"  g="0.2"  b="0.8"  showDaughters="true" visible="true"/>
 
     <vis name="ci_GEMVis"  r= "0.8"  g="0.4"  b="0.3" alpha="0.8" showDaughters="true" visible="true"/>
     <vis name="ci_HCALVis"  r= "0.6"  g="0"  b="0.6" alpha="1.0" showDaughters="true" visible="true"/>
diff --git a/reference_detector.xml b/reference_detector.xml
index 1ff50a25..518c5ff1 100644
--- a/reference_detector.xml
+++ b/reference_detector.xml
@@ -8,7 +8,7 @@
 	url="https://eicweb.phy.anl.gov/EIC/detectors/reference_detector.git"
 	status="development"
 	version="v1 2021-03-16">
-    <comment>Test Detector</comment>        
+    <comment>Reference Detector</comment>        
   </info>
 
   <define>
@@ -125,6 +125,7 @@
   <include ref="compact/hcal.xml"/>
   <include ref="compact/forward_trd.xml"/>
   <include ref="compact/B0_tracker.xml"/>
+  <include ref="compact/cb_VTX_Barrel.xml"/>
   <!--
   <include ref="compact/ci_HCAL.xml"/>
   <include ref="compact/forward_rich.xml"/>
diff --git a/src/cb_VTX_Barrel.cpp b/src/cb_VTX_Barrel.cpp
new file mode 100644
index 00000000..511f4a26
--- /dev/null
+++ b/src/cb_VTX_Barrel.cpp
@@ -0,0 +1,117 @@
+#include <XML/Helper.h>
+#include "DDRec/Surface.h"
+#include "DDRec/DetectorData.h"
+#include "DD4hep/OpticalSurfaces.h"
+#include "DD4hep/DetFactoryHelper.h"
+#include "DD4hep/Printout.h"
+//////////////////////////////////
+// Central Barrel Vertex Detector
+//////////////////////////////////
+
+using namespace std;
+using namespace dd4hep;
+
+static Ref_t createDetector(Detector& desc, xml_h e, SensitiveDetector sens)
+{
+  xml_det_t  x_det      = e;
+  string     detName    = x_det.nameStr();
+  int        detID      = x_det.id();
+
+  xml_dim_t  dim        = x_det.dimensions();
+  double     RIn        = dim.rmin();
+  double     ROut       = dim.rmax();
+  double     SizeZ      = dim.length();
+
+  xml_dim_t  pos        = x_det.position();
+
+  Material   Vacuum     = desc.material("Vacuum");
+
+  // Create Global Volume 
+  Tube cb_VTX_Barrel_GVol_Solid(RIn, ROut, SizeZ / 2.0, 0., 360.0 * deg);
+  Volume detVol("cb_VTX_Barrel_GVol_Logic", cb_VTX_Barrel_GVol_Solid, Vacuum);
+  detVol.setVisAttributes(desc.visAttributes(x_det.visStr()));
+
+  //////////////////
+  // Barrel Ladder
+  //////////////////
+  xml_comp_t x_layer = x_det.child(_U(layer));
+  const int repeat   = x_layer.repeat();
+  
+  xml_comp_t x_slice = x_layer.child(_U(slice));
+  Material slice_mat = desc.material(x_slice.materialStr());
+
+  double x = 0.0 * cm;
+  double y = 0.0 * cm;
+  double z = 0.0 * cm;
+  int FDIV = 0;
+  double dR = 0.0;
+  double length = 0.0;
+  double phi = 0.0;
+  // Ladder Layer Parameters
+  double lay_Dx[6];
+  double lay_Dy[6];
+  double lay_Dz[6];
+  double lay_Rin[6];
+  lay_Dx[0] = 0.050 * mm; lay_Dy[0] = 2.0 * cm; lay_Dz[0] = 10.0 * cm; lay_Rin[0] = 3.5 * cm;
+  lay_Dx[1] = 0.050 * mm; lay_Dy[1] = 2.0 * cm; lay_Dz[1] = 11.0 * cm; lay_Rin[1] = 4.5 * cm;
+  lay_Dx[2] = 0.150 * mm; lay_Dy[2] = 4.0 * cm; lay_Dz[2] = 18.0 * cm; lay_Rin[2] = 6.5 * cm;
+  lay_Dx[3] = 0.150 * mm; lay_Dy[3] = 4.0 * cm; lay_Dz[3] = 24.0 * cm; lay_Rin[3] = 10.5 * cm;
+  lay_Dx[4] = 0.150 * mm; lay_Dy[4] = 4.0 * cm; lay_Dz[4] = 36.0 * cm; lay_Rin[4] = 13.5 * cm;
+  lay_Dx[5] = 0.150 * mm; lay_Dy[5] = 4.0 * cm; lay_Dz[5] = 48.0 * cm; lay_Rin[5] = 15.5 * cm;
+
+  int i_layer = 0;
+  int i_module = 0;
+  // Loop over layers
+  for(int i = 0; i < repeat; i++) {
+    double cb_VTX_Barrel_ladder_DZ = lay_Dz[i];
+    double cb_VTX_Barrel_ladder_DY = lay_Dy[i];
+    double cb_VTX_Barrel_ladder_Thickness = lay_Dx[i];
+    dR = lay_Rin[i]; 
+    length = 2.0 * 3.1415 * dR;
+    int laddersCount = length / cb_VTX_Barrel_ladder_DY;
+
+    for (int i = 0; i < 2; i++) {
+      double LN = cb_VTX_Barrel_ladder_DY * laddersCount;
+      double LN1 = cb_VTX_Barrel_ladder_DY * (laddersCount + 1.0 + i);
+
+      if (LN/LN1 > 0.8)
+        laddersCount = laddersCount + 1;
+    }
+
+    double cb_VTX_Barrel_ladder_deltaphi = 2.0 * 3.1415926 / laddersCount;
+
+    string ladderBoxName = detName + _toString(i, "_ladder_Solid_%d");
+    string ladderName = detName + _toString(i, "_ladder_Logic_%d");
+    Volume ladderVol(ladderName, Box(cb_VTX_Barrel_ladder_Thickness * 0.5, cb_VTX_Barrel_ladder_DY * 0.5, cb_VTX_Barrel_ladder_DZ * 0.5), slice_mat);
+    ladderVol.setVisAttributes(desc,x_layer.visStr());
+    sens.setType("tracker");
+    ladderVol.setSensitiveDetector(sens);
+    i_layer++;
+
+    for (int ia = 0; ia < laddersCount; ia++) {
+      phi = (ia * (cb_VTX_Barrel_ladder_deltaphi));
+      x = - dR * cos(phi);
+      y = - dR * sin(phi);
+
+      RotationZYX ladder_rot = RotationZYX(cb_VTX_Barrel_ladder_deltaphi * ia, 0.0, 0.0);
+      Position ladder_pos = Position(x, y, z);
+      string ladderName = detName + _toString(i, "_ladder_Phys_%d") + _toString(ia, "_%d"); 
+      PlacedVolume ladderPV = detVol.placeVolume(ladderVol, Transform3D(ladder_rot, ladder_pos));
+      i_module++;
+      ladderPV.addPhysVolID("layer", i_layer).addPhysVolID("module", i_module);
+    }
+  }
+
+  // TODO: Pixels
+
+  DetElement   det(detName, detID);
+  Volume       motherVol = desc.pickMotherVolume(det);
+  Transform3D  tr(RotationZYX(0.0, 0.0, 0.0), Position(0.0, 0.0, pos.z()));
+  PlacedVolume detPV = motherVol.placeVolume(detVol, tr);
+  detPV.addPhysVolID("system", detID);
+  detPV.addPhysVolID("barrel", 1);
+  det.setPlacement(detPV);
+  return det;
+}
+
+DECLARE_DETELEMENT(cb_VTX_Barrel, createDetector)
-- 
GitLab