GeomConverter/src/org/lcsim/geometry/compact/converter/lcdd
diff -u -r1.9 -r1.10
--- PolyhedraBarrelCalorimeter.java 12 Sep 2006 01:20:56 -0000 1.9
+++ PolyhedraBarrelCalorimeter.java 16 Oct 2006 23:36:42 -0000 1.10
@@ -1,230 +1,309 @@
package org.lcsim.geometry.compact.converter.lcdd;
+import static java.lang.Math.PI;
+import static java.lang.Math.cos;
+import static java.lang.Math.sin;
+import static java.lang.Math.tan;
+
import java.util.Iterator;
-import org.jdom.Element;
+
import org.jdom.Attribute;
+import org.jdom.Element;
import org.jdom.JDOMException;
+import org.lcsim.geometry.compact.converter.lcdd.util.Box;
import org.lcsim.geometry.compact.converter.lcdd.util.Define;
-import static java.lang.Math.PI;
-import static java.lang.Math.tan;
-import static java.lang.Math.sin;
-import static java.lang.Math.cos;
import org.lcsim.geometry.compact.converter.lcdd.util.LCDD;
import org.lcsim.geometry.compact.converter.lcdd.util.Material;
-import org.lcsim.geometry.compact.converter.lcdd.util.SensitiveDetector;
-import org.lcsim.geometry.compact.converter.lcdd.util.Solids;
-import org.lcsim.geometry.compact.converter.lcdd.util.Structure;
import org.lcsim.geometry.compact.converter.lcdd.util.PhysVol;
import org.lcsim.geometry.compact.converter.lcdd.util.PolyhedraRegular;
+import org.lcsim.geometry.compact.converter.lcdd.util.Position;
import org.lcsim.geometry.compact.converter.lcdd.util.Rotation;
+import org.lcsim.geometry.compact.converter.lcdd.util.SensitiveDetector;
+import org.lcsim.geometry.compact.converter.lcdd.util.Solids;
+import org.lcsim.geometry.compact.converter.lcdd.util.Structure;
import org.lcsim.geometry.compact.converter.lcdd.util.Trapezoid;
-import org.lcsim.geometry.compact.converter.lcdd.util.Box;
import org.lcsim.geometry.compact.converter.lcdd.util.Volume;
-import org.lcsim.geometry.compact.converter.lcdd.util.Position;
+import org.lcsim.geometry.layer.Layer;
+import org.lcsim.geometry.layer.Layering;
/**
*
+ * Convert a PolyhedraBarrelCalorimeter to LCDD format.
+ *
* @author jeremym
*/
public class PolyhedraBarrelCalorimeter extends LCDDSubdetector
{
- private Element node;
-
- public PolyhedraBarrelCalorimeter(Element node) throws JDOMException
- {
- super(node);
- this.node = node;
- }
-
- public void addToLCDD(LCDD lcdd, SensitiveDetector sens) throws JDOMException
- {
- /* local refs to LCDD objects */
- Solids solids = lcdd.getSolids();
- Structure structure = lcdd.getStructure();
- Volume motherVolume = lcdd.pickMotherVolume(this);
- Material air = lcdd.getMaterial("Air");
- Define define = lcdd.getDefine();
-
- /* name and id */
- String detName = node.getAttributeValue("name");
- int id = node.getAttribute("id").getIntValue();
-
- /* get dimensions */
- Element dimensions = node.getChild("dimensions");
- double detZ = dimensions.getAttribute("z").getDoubleValue();
- double rmin = dimensions.getAttribute("rmin").getDoubleValue();
- int numsides = dimensions.getAttribute("numsides").getIntValue();
-
- /* polyhedra rotation so it lays "flat" */
- double zrot = Math.PI / numsides;
- Rotation rot = new Rotation(detName + "_rotation");
- rot.setZ(zrot);
- define.addRotation(rot);
-
- /* total thickness of the detector */
- double totalThickness = org.lcsim.geometry.layer.LayerFromCompactCnv.computeDetectorTotalThickness(node);
-
- /* envelope volume */
- PolyhedraRegular polyhedra = new PolyhedraRegular(
- detName + "_polyhedra",
- numsides, rmin, rmin+totalThickness, detZ);
- solids.addSolid(polyhedra);
-
- Volume envelopeVolume = new Volume(detName + "_envelope");
- envelopeVolume.setSolid(polyhedra);
- envelopeVolume.setMaterial(air);
-
- PhysVol envelopePhysvol = new PhysVol(envelopeVolume);
- envelopePhysvol.setRotation(rot);
- envelopePhysvol.addPhysVolID("system",id);
- envelopePhysvol.addPhysVolID("barrel",0);
- motherVolume.addPhysVol(envelopePhysvol);
-
- /* single stave trapezoid */
- double innerAngle = Math.PI * 2 / numsides;
- double halfInnerAngle = innerAngle/2;
-
- Trapezoid sectTrd = new Trapezoid(detName + "_stave_trapezoid");
- sectTrd.setY2(detZ);
- sectTrd.setY1(detZ);
- sectTrd.setZ(totalThickness);
-
- double innerFaceLength = rmin * tan(halfInnerAngle) * 2;
- sectTrd.setX1(innerFaceLength);
- double rmax = rmin + totalThickness;
- double outerFaceLength = rmax * tan(halfInnerAngle) * 2;
- sectTrd.setX2(outerFaceLength);
-
- solids.addSolid(sectTrd);
- Volume sectVolume = new Volume(detName + "_stave");
- sectVolume.setMaterial(air);
- sectVolume.setSolid(sectTrd);
-
- /* build the box layers in the stave */
- double layerOuterAngle = (PI - innerAngle) / 2;
- double layerInnerAngle = (PI/2 - layerOuterAngle);
-
- int layerNum = 0;
- double posZ = -( totalThickness / 2 );
- double dimX = innerFaceLength;
- for (Iterator i = node.getChildren("layer").iterator(); i.hasNext();)
- {
- Element layer = (Element) i.next();
- int repeat = layer.getAttribute("repeat").getIntValue();
- for ( int j=0; j<repeat; j++)
- {
- int sliceNum = 0;
- for ( Iterator k = layer.getChildren("slice").iterator(); k.hasNext();)
- {
- String sliceName = detName + "_stave_layer" + layerNum + "_slice" + sliceNum;
-
- Element slice = (Element) k.next();
-
- Attribute s = slice.getAttribute("sensitive");
- boolean sensitive = s != null && s.getBooleanValue();
-
- double thickness = slice.getAttribute("thickness").getDoubleValue();
- posZ += thickness / 2;
-
- /* slice position */
- Position slicePosition = new Position(sliceName + "_position");
- slicePosition.setZ(posZ);
- define.addPosition(slicePosition);
-
- /* slice box */
- Box sliceBox = new Box(sliceName + "_box");
- sliceBox.setX(dimX);
- sliceBox.setY(detZ);
- sliceBox.setZ(thickness);
- solids.addSolid(sliceBox);
-
- /* slice volume */
- Volume sliceVolume = new Volume(sliceName);
- sliceVolume.setSolid(sliceBox);
- Material sliceMaterial = lcdd.getMaterial(slice.getAttributeValue("material"));
- sliceVolume.setMaterial(sliceMaterial);
- if ( sensitive ) sliceVolume.setSensitiveDetector(sens);
-
- /* FIXME: these need to be called automatically whenever a new volume is created --JM */
- setRegion(lcdd, slice, sliceVolume);
- setLimitSet(lcdd, slice, sliceVolume);
- setVisAttributes(lcdd, node, sliceVolume);
-
- structure.addVolume(sliceVolume);
-
- /* slice PhysVol */
- PhysVol slicePhysVol = new PhysVol(sliceVolume);
- slicePhysVol.setPosition(slicePosition);
- slicePhysVol.addPhysVolID("layer", layerNum);
- slicePhysVol.addPhysVolID("slice", sliceNum);
- sectVolume.addPhysVol(slicePhysVol);
-
- /* incr X dim */
- dimX += thickness * tan( layerInnerAngle ) * 2;
-
- /* incr Z position */
- posZ += thickness / 2;
-
- /* incr slice number */
- ++sliceNum;
- }
-
- /* incr layer number */
- ++layerNum;
- }
-
- setVisAttributes(lcdd,node,sectVolume);
- }
-
- /* add section volume after all slices */
- structure.addVolume(sectVolume);
-
- /* place the staves */
- double innerRotation = innerAngle;
- double offsetRotation = -innerRotation / 2;
- double placementRotation = -offsetRotation;
-
- double sectCenterRadius = rmin + totalThickness / 2;
- double rotY = -offsetRotation;
- double rotX = PI / 2;
- double posX = -sectCenterRadius * sin(rotY);
- double posY = sectCenterRadius * cos(rotY);
- for ( int i=0; i < numsides; i++)
- {
- int moduleNumber=i;
-
- Position position = new Position(detName + "_stave0_module" + moduleNumber + "_position");
- position.setX(posX);
- position.setY(posY);
-
- Rotation rotation = new Rotation(detName + "_stave0_module" + moduleNumber + "_rotation");
- rotation.setX(rotX);
- rotation.setY(rotY);
-
- define.addPosition(position);
- define.addRotation(rotation);
-
- PhysVol sectPhysVol = new PhysVol(sectVolume);
- sectPhysVol.setPosition(position);
- sectPhysVol.setRotation(rotation);
-
- envelopeVolume.addPhysVol(sectPhysVol);
- sectPhysVol.addPhysVolID("stave",0);
- sectPhysVol.addPhysVolID("module",moduleNumber);
-
- rotY -= innerRotation;
- posX = -sectCenterRadius * sin(rotY);
- posY = sectCenterRadius * cos(rotY);
- }
-
- setVisAttributes(lcdd,node,envelopeVolume);
-
- /* add envelope volume after staves */
- structure.addVolume(envelopeVolume);
- }
-
- public boolean isCalorimeter()
- {
- return true;
- }
+ private Element node;
+
+ public PolyhedraBarrelCalorimeter(Element node) throws JDOMException
+ {
+ super(node);
+ this.node = node;
+ }
+
+ public void addToLCDD(LCDD lcdd, SensitiveDetector sens) throws JDOMException
+ {
+ // Get some important LCDD references.
+ Solids solids = lcdd.getSolids();
+ Structure structure = lcdd.getStructure();
+ Volume motherVolume = lcdd.pickMotherVolume(this);
+ Material air = lcdd.getMaterial("Air");
+ Define define = lcdd.getDefine();
+
+ // Subdetector name and ID.
+ String detName = node.getAttributeValue("name");
+ int id = node.getAttribute("id").getIntValue();
+
+ // Subdetector envelope dimensions.
+ Element dimensions = node.getChild("dimensions");
+ double detZ = dimensions.getAttribute("z").getDoubleValue();
+ double rmin = dimensions.getAttribute("rmin").getDoubleValue();
+ int numsides = dimensions.getAttribute("numsides").getIntValue();
+
+ // Rotation of the envelope to make flat side down.
+ double zrot = Math.PI / numsides;
+ Rotation rot = new Rotation(detName + "_rotation");
+ rot.setZ(zrot);
+ define.addRotation(rot);
+
+ // Create layering object for this subdetector.
+ Layering layering = Layering.makeLayering(this.node);
+
+ //
+ //double totalThickness = org.lcsim.geometry.layer.LayerFromCompactCnv.computeDetectorTotalThickness(node);
+ double total_thickness = layering.getLayerStack().getTotalThickness();
+
+ // Envelope volume for subdetector.
+ PolyhedraRegular polyhedra = new PolyhedraRegular(detName + "_polyhedra", numsides, rmin, rmin + total_thickness, detZ);
+ solids.addSolid(polyhedra);
+
+ Volume envelopeVolume = new Volume(detName + "_envelope");
+ envelopeVolume.setSolid(polyhedra);
+ envelopeVolume.setMaterial(air);
+
+ PhysVol envelopePhysvol = new PhysVol(envelopeVolume);
+ envelopePhysvol.setRotation(rot);
+ envelopePhysvol.addPhysVolID("system", id);
+ envelopePhysvol.addPhysVolID("barrel", 0);
+ motherVolume.addPhysVol(envelopePhysvol);
+
+ /* single stave trapezoid */
+ double innerAngle = Math.PI * 2 / numsides;
+ double halfInnerAngle = innerAngle / 2;
+
+ Trapezoid sectTrd = new Trapezoid(detName + "_stave_trapezoid");
+ sectTrd.setY2(detZ);
+ sectTrd.setY1(detZ);
+ sectTrd.setZ(total_thickness);
+
+ double innerFaceLength = rmin * tan(halfInnerAngle) * 2;
+ sectTrd.setX1(innerFaceLength);
+ double rmax = rmin + total_thickness;
+ double outerFaceLength = rmax * tan(halfInnerAngle) * 2;
+ sectTrd.setX2(outerFaceLength);
+
+ solids.addSolid(sectTrd);
+ Volume sectVolume = new Volume(detName + "_stave");
+ sectVolume.setMaterial(air);
+ sectVolume.setSolid(sectTrd);
+
+ //
+ // Create layers.
+ //
+
+ double layerOuterAngle = (PI - innerAngle) / 2;
+ double layerInnerAngle = (PI / 2 - layerOuterAngle);
+
+ //System.out.println("total_thickness="+total_thickness);
+
+ int layer_number = 0;
+ double layer_position_z = -(total_thickness / 2);
+ double layer_dim_x = innerFaceLength;
+ for (Iterator i = node.getChildren("layer").iterator(); i.hasNext();)
+ {
+ Element layer_element = (Element) i.next();
+
+ // Get the layer from the layering engine.
+ Layer layer = layering.getLayer(layer_number);
+
+ // Get number of times to repeat this layer.
+ int repeat = layer_element.getAttribute("repeat").getIntValue();
+
+ // Loop over repeats for this layer.
+ for (int j = 0; j < repeat; j++)
+ {
+ //System.out.println("layer_number="+layer_number);
+
+ // Name of the layer.
+ String layer_name = detName + "_stave_layer" + layer_number;
+
+ // Layer thickness.
+ double layer_thickness = layer.getThickness();
+
+ //System.out.println("layer_thickness="+layer_thickness);
+
+ // Layer position in Z within the stave.
+ layer_position_z += layer_thickness / 2;
+
+ //System.out.println("layer_position_z="+layer_position_z);
+
+ // Position of layer.
+ Position layer_position = new Position(layer_name + "_position");
+ layer_position.setZ(layer_position_z);
+ define.addPosition(layer_position);
+
+ // Layer box.
+ Box layer_box = new Box(layer_name + "_box");
+ layer_box.setX(layer_dim_x);
+ layer_box.setY(detZ);
+ layer_box.setZ(layer_thickness);
+ solids.addSolid(layer_box);
+
+ // Layer volume.
+ Volume layer_volume = new Volume(layer_name);
+ layer_volume.setSolid(layer_box);
+ layer_volume.setMaterial(air);
+
+ // Set region, limitset, and vis.
+ setRegion(lcdd, layer_element, layer_volume);
+ setLimitSet(lcdd, layer_element, layer_volume);
+ setVisAttributes(lcdd, node, layer_volume);
+
+ // Create the slices (sublayers) within the layer.
+ double slice_position_z = -(layer_thickness / 2);
+
+ //System.out.println("start slice_position_z="+slice_position_z);
+
+ int slice_number = 0;
+ for (Iterator k = layer_element.getChildren("slice").iterator(); k.hasNext();)
+ {
+ Element slice_element = (Element) k.next();
+
+ String slice_name = layer_name + "_slice" + slice_number;
+
+ Attribute s = slice_element.getAttribute("sensitive");
+ boolean sensitive = s != null && s.getBooleanValue();
+
+ double slice_thickness = slice_element.getAttribute("thickness").getDoubleValue();
+
+ slice_position_z += slice_thickness / 2;
+
+ //System.out.println("slice_position_z="+slice_position_z);
+
+ Material slice_material = lcdd.getMaterial(slice_element.getAttributeValue("material"));
+
+ // Slice Position.
+ Position slice_position = new Position(slice_name + "_position");
+ slice_position.setZ(slice_position_z);
+ define.addPosition(slice_position);
+
+ // Slice box.
+ Box slice_box = new Box(slice_name + "_box");
+ slice_box.setX(layer_dim_x);
+ slice_box.setY(detZ);
+ slice_box.setZ(slice_thickness);
+ solids.addSolid(slice_box);
+
+ // Slice volume.
+ Volume slice_volume = new Volume(slice_name);
+ slice_volume.setSolid(slice_box);
+ slice_volume.setMaterial(slice_material);
+ if (sensitive)
+ slice_volume.setSensitiveDetector(sens);
+ structure.addVolume(slice_volume);
+
+ // Set region, limitset, and vis.
+ setRegion(lcdd, slice_element, slice_volume);
+ setLimitSet(lcdd, slice_element, slice_volume);
+ setVisAttributes(lcdd, node, slice_volume);
+
+ // slice PhysVol
+ PhysVol slice_physvol = new PhysVol(slice_volume);
+ slice_physvol.setPosition(slice_position);
+ slice_physvol.addPhysVolID("slice", slice_number);
+ layer_volume.addPhysVol(slice_physvol);
+
+ // Increment Z position for next slice.
+ slice_position_z += slice_thickness / 2;
+
+ // Increment slice number.
+ ++slice_number;
+ }
+
+ // Add the layer logical volume to the structure.
+ structure.addVolume(layer_volume);
+
+ // Layer physical volume.
+ PhysVol layer_physvol = new PhysVol(layer_volume);
+ layer_physvol.setPosition(layer_position);
+ layer_physvol.addPhysVolID("layer", layer_number);
+ sectVolume.addPhysVol(layer_physvol);
+
+ // Increment the layer X dimension.
+ layer_dim_x += layer_thickness * tan(layerInnerAngle) * 2;
+
+ // Increment the layer Z position.
+ layer_position_z += layer_thickness / 2;
+
+ // Increment the layer number.
+ ++layer_number;
+ }
+ }
+
+ // Set the vis of the section.
+ setVisAttributes(lcdd,node,sectVolume);
+
+ // Add the section volume to the structure.
+ structure.addVolume(sectVolume);
+
+ // Place the staves.
+ double innerRotation = innerAngle;
+ double offsetRotation = -innerRotation / 2;
+ //double placementRotation = -offsetRotation;
+
+ double sectCenterRadius = rmin + total_thickness / 2;
+ double rotY = -offsetRotation;
+ double rotX = PI / 2;
+ double posX = -sectCenterRadius * sin(rotY);
+ double posY = sectCenterRadius * cos(rotY);
+ for (int i = 0; i < numsides; i++)
+ {
+ int moduleNumber = i;
+
+ Position position = new Position(detName + "_stave0_module" + moduleNumber + "_position");
+ position.setX(posX);
+ position.setY(posY);
+
+ Rotation rotation = new Rotation(detName + "_stave0_module" + moduleNumber + "_rotation");
+ rotation.setX(rotX);
+ rotation.setY(rotY);
+
+ define.addPosition(position);
+ define.addRotation(rotation);
+
+ PhysVol sectPhysVol = new PhysVol(sectVolume);
+ sectPhysVol.setPosition(position);
+ sectPhysVol.setRotation(rotation);
+
+ envelopeVolume.addPhysVol(sectPhysVol);
+ sectPhysVol.addPhysVolID("stave", 0);
+ sectPhysVol.addPhysVolID("module", moduleNumber);
+
+ rotY -= innerRotation;
+ posX = -sectCenterRadius * sin(rotY);
+ posY = sectCenterRadius * cos(rotY);
+ }
+
+ // Set vis of envelope.
+ setVisAttributes(lcdd, node, envelopeVolume);
+
+ // Add the subdetector envelope to the structure.
+ structure.addVolume(envelopeVolume);
+ }
+
+ public boolean isCalorimeter()
+ {
+ return true;
+ }
}
\ No newline at end of file
