LISTSERV 16.5 - LCD-CVS Archives

GeomConverter/src/org/lcsim/geometry/segmentation

diff -N CartesianGridXY.java
--- /dev/null	1 Jan 1970 00:00:00 -0000
+++ CartesianGridXY.java	15 Sep 2009 00:57:52 -0000	1.1
@@ -0,0 +1,251 @@

+package org.lcsim.geometry.segmentation;
+
+import hep.physics.vec.BasicHep3Vector;
+import hep.physics.vec.Hep3Vector;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import org.jdom.DataConversionException;
+import org.jdom.Element;
+import org.lcsim.detector.IDetectorElement;
+import org.lcsim.detector.identifier.ExpandedIdentifier;
+import org.lcsim.detector.identifier.IExpandedIdentifier;
+import org.lcsim.detector.identifier.IIdentifier;
+import org.lcsim.detector.identifier.IIdentifierHelper;
+import org.lcsim.detector.identifier.Identifier;
+import org.lcsim.geometry.util.IDDescriptor;
+
+/**
+ * Simplified XY grid based on GridXYZ segmentation.
+ * 
+ * @author jeremym
+ */
+public class CartesianGridXY extends SegmentationBase
+{	
+	private double gridSizeX = 0;
+	private double gridSizeY = 0;
+
+	private int xIndex = -1;
+	private int yIndex = -1;
+	private int layerIndex = -1;
+
+	private double[] localPosition =
+	{ 0, 0, 0 };
+
+	private double[] globalPosition =
+	{ 0, 0, 0 };
+	
+	List<Integer> geomFields;
+
+	public CartesianGridXY(Element node) throws DataConversionException
+	{
+		super(node);
+
+		if (node.getAttribute("gridSizeX") != null)
+		{
+			gridSizeX = node.getAttribute("gridSizeX").getDoubleValue();
+		}
+		else
+		{
+			throw new RuntimeException("Missing gridSizeX parameter.");
+		}
+
+		if (node.getAttribute("gridSizeY") != null)
+		{
+			gridSizeY = node.getAttribute("gridSizeY").getDoubleValue();
+		}
+		else
+		{
+			throw new RuntimeException("Missing gridSizeY parameter.");
+		}
+	}
+
+	public boolean supportsNeighbours()
+	{
+		return true;
+	}
+	
+	// TODO: Bounds check on x and y indices.
+	public long[] getNeighbourIDs(int layerRange, int xRange, int yRange)
+	{
+		// Get the IdentifierHelper.
+		IIdentifierHelper helper = detector.getDetectorElement().getIdentifierHelper();
+		
+		// Get number of layers.
+		int nlayers = this.getNumberOfLayers();
+		
+		// Current packed id.
+		IIdentifier currId = new Identifier(this.getDecoder().getID()); 
+		
+		// Set values for current id.
+		int currLayer = helper.getValue(currId, layerIndex);
+		int currX = helper.getValue(currId, xIndex);
+		int currY = helper.getValue(currId, yIndex);
+		
+		// Create an ExpandedIdentifier for the current id.
+		IExpandedIdentifier thisId = helper.unpack(currId);
+		
+		// Compute maximum number of neighbors.
+		//int nMax = (2*layerRange + 1)*(2*xRange + 1)*(2*yRange + 1) - 1;
+		
+		// Create return array.
+		List<Long> neighbors = new ArrayList<Long>();
+		
+		// Loop over layer range.
+		//int ineighbor = 0;
+		for (int ilayer=-layerRange; ilayer<=layerRange; ilayer++)
+		{
+			// Loop over X range.
+			for (int ix=-xRange; ix<=xRange; ix++)
+			{
+				// Loop over Y range.
+				for (int iy=-yRange; iy<=yRange; iy++)
+				{								
+					// Compute layer value.
+					int neighborLayer = currLayer + ilayer;
+					
+					if (neighborLayer >= 0 && neighborLayer < nlayers)
+					{						
+						// Compute x value.
+						int neighborX = currX + ix;
+						
+						// Compute y value.
+						int neighborY = currY + iy;
+						
+						// Create a new ExpandedIdenfier from the base Id.
+						ExpandedIdentifier neighborId = new ExpandedIdentifier(thisId);
+					
+						// Set the neighbor fields.
+						neighborId.setValue(layerIndex, neighborLayer);
+						neighborId.setValue(xIndex, neighborX);
+						neighborId.setValue(yIndex, neighborY);
+					
+						// Add the neighbor id to the return array.
+						neighbors.add(helper.pack(neighborId).getValue());					
+					}
+				}
+			}		
+		}
+		
+		long result[] = new long[neighbors.size()];		
+		int i = 0;
+		for (Long id : neighbors)
+		{
+			result[i] = id;
+			i++;
+		}
+		return result;
+	}
+	
+	// TODO: Implement this method.  Need proper bounds checking first. 
+	public long findCellContainingXYZ(double x, double y, double z) 
+	{
+		if (true)
+			throw new RuntimeException("Not implemented");
+		return 0;
+	}
+	
+	public double[] getPosition()
+	{
+		return globalPosition;
+	}
+	
+	private void computeGlobalPosition()
+	{											
+		// Make an id only containing geometric fields and no segmentation fields.
+		IExpandedIdentifier geomIdExp = detector.getDetectorElement().getIdentifierHelper().unpack(new Identifier(this.getDecoder().getID()), geomFields);
+		IIdentifier geomId = detector.getDetectorElement().getIdentifierHelper().pack(geomIdExp);
+
+		// Search for the the DetectorElement associated with the geometry id.
+		List<IDetectorElement> deSearch = detector.getDetectorElement().findDetectorElement(geomId);
+
+		// Check if the lookup failed.
+		if (deSearch == null || deSearch.size() == 0)
+		{
+			throw new RuntimeException("Failed to find DetectorElement with geometry id <" + geomIdExp.toString() + "> !");
+		}
+
+		// Set the DetectorElement to use for local to global transform.
+		IDetectorElement sensor = deSearch.get(0);
+
+		// Create a vector of the local position.
+		Hep3Vector posVecLocal = new BasicHep3Vector(localPosition[0], localPosition[1], localPosition[2]);
+		
+		// Compute the global position of the hit using the DetectorElement.
+		Hep3Vector posVecGlobal = sensor.getGeometry().transformLocalToGlobal(posVecLocal);
+
+		// Set the internal global position array.
+		globalPosition[0] = posVecGlobal.x();
+		globalPosition[1] = posVecGlobal.y();
+		globalPosition[2] = posVecGlobal.z();
+	}
+					
+	public void setID(long id)
+	{
+		super.setID(id);
+		computePosition();
+	}
+
+	public void setIDDescription(IDDescriptor id)
+	{
+		super.setIDDescription(id);
+					
+		if (geomFields == null)
+		{
+			geomFields = new ArrayList<Integer>();
+			
+			xIndex = id.indexOf("x");
+			yIndex = id.indexOf("y");
+			layerIndex = id.indexOf("layer");
+
+			// Set geometry field list.
+			for (int i=0; i < id.fieldCount(); i++)
+			{
+				String fname = id.fieldName(i);
+				if (!fname.equals("x") && !fname.equals("y"))
+				{
+					geomFields.add(i);
+				}
+			}
+		}
+	}	
+	
+	public double getX()
+	{
+		return getPosition()[0];
+	}
+
+	public double getY()
+	{
+		return getPosition()[1];
+	}
+
+	public double getZ()
+	{
+		return getPosition()[2];
+	}
+	
+	private void computeLocalX()
+	{	
+		localPosition[0] = (((double) getValue(xIndex)) + 0.5) * gridSizeX;	
+	}
+
+	private void computeLocalY()
+	{
+		localPosition[1] = (((double) getValue(yIndex)) + 0.5) * gridSizeY;
+	}
+	
+	private void computePosition()
+	{
+		computeLocalPosition();
+		computeGlobalPosition();
+	}
+
+	private void computeLocalPosition()
+	{
+		localPosition[0] = localPosition[1] = localPosition[2] = 0.0;
+		computeLocalX();
+		computeLocalY();
+	}			
+}

\ No newline at end of file

Commit in `GeomConverter/src/org/lcsim/geometry` on MAIN
`compact/converter/lcdd/CartesianGridXY.java`	+45		added 1.1
`/GridXYZ.java`	+64	-62	1.2 -> 1.3
`compact/converter/lcdd/util/GridXYZ.java`	+5	-3	1.1 -> 1.2
`segmentation/CartesianGridXY.java`	+251		added 1.1
`/GridXYZ.java`	+6	-4	1.26 -> 1.27
	+371	-69