LISTSERV 16.5 - LCD-CVS Archives

+01 February 2007 Fred Blanc V01-01-00 + - using the track fit parameters (d0,phi0,omega,z0,s) at all stages of the + propagation and filtering. + - added effects of: + 1) multiple scattering: add MS error matrix to track covariance matrix before filtering + 2) energy loss: subtract or add dE/dx at the end of the filtering step + - calculate the seed track using the first 3 hits on a track +

+ KFConstants.java stores a list of constants + KFFitterDriver.java main driver + KFMain.java used to run on the command line + java org.lcsim.contrib.KFFitter.KFMain <data file> <number of events to run> + KFMatrix.java unused (using Jama instead) + KFPoint.java defines a point in space and its error matrix + KFSite.java defines a Kalman site and handles the filtering of propagates track + KFSites can be a measurement and/or a material site + KFTrack.java contains a list of KFSites. + handles the swimming trough list of KFSites + KFTrackParameters.java defines the track parametrization and its error matrix + Supports three parameterizations: + 1) fit parameters: d0, phi0, omega, z0, tanl (+ path length) + 2) point parameters: x, y, z, px, py, pz + 3) helix parameters: xc, yc, rc, z0, tanl +

+ public void setLocation(double x, double y, double z) + { + // ************************ + // For testing purpose only + // ************************ + System.out.println(" ************************"); + System.out.println("KFSite: setLocation ** DEBUG MODE **"); + System.out.println(" ************************"); + + _siteLocation.makePoint(x,y,z); + + System.out.print("KFSite: Add Pt at coord = "); + System.out.print(x); System.out.print(" "); + System.out.print(y); System.out.print(" "); + System.out.println(z); + + _PhiZErrorMatrix = _siteLocation.getPhiZErrorMatrix(); + + _PhiZErrorMatrixInverse = _PhiZErrorMatrix.inverse(); + + System.out.println("KFSite: PhiZ cov matrix:"); + _PhiZErrorMatrix.print(12,10); + + //System.out.println("KFSite: PhiZ cov matrix inverted:"); + //_PhiZErrorMatrixInverse.print(12,10); + } +

public KFTrackParameters filter(KFTrackParameters trk) { double[] tpp = trk.getFitParameters(); Matrix stateVect = new Matrix(tpp,5); System.out.println("KFSite: initial stateVect:");

+ double[] ptParam = trk.getPointParameters(); + double refPhi = Math.atan2(ptParam[1],ptParam[0]); + double refZ = ptParam[2]; + double[] mp = {_siteLocation.getPhi()-refPhi,_siteLocation.getZ()-refZ}; + //double[] mp = {_siteLocation.getPhi(),_siteLocation.getZ()};

+ //Matrix tmp1 = trkCovInverse.times(stateVect); + //Matrix tmp2 = HT.times(_PhiZErrorMatrixInverse).times(measVect); + //Matrix tmp3 = tmp1.plus(tmp2); + Matrix tmp1 = filteredCovInv.times(stateVect);

Matrix tmp2 = HT.times(_PhiZErrorMatrixInverse).times(measVect); Matrix tmp3 = tmp1.plus(tmp2); Matrix filteredTrkPar = filteredCov.times(tmp3); System.out.println("KFSite: Filtered State Vector:");

- double dEdx = dEdx(tmpPtot); - System.out.print("KFSite: Ptot = "); System.out.println(tmpPtot); - System.out.print("KFSite: dEdx = "); System.out.println(dEdx); - - double tmpEtot = Math.sqrt(_cst.pionMass()*_cst.pionMass()+tmpPtot*tmpPtot); - double tmpEtotCorr = tmpEtot+0.01*dEdx; // (fix me) change sign depending on fit direction - double ratioCorr = tmpEtotCorr/tmpEtot;

+ double ratioCorr = 1.; + if(_applyDEDXCorrection) { + double dEdx = dEdx(tmpPtot); + System.out.print("KFSite: Ptot = "); System.out.println(tmpPtot); + System.out.print("KFSite: dEdx = "); System.out.println(dEdx); + + double tmpEtot = Math.sqrt(_cst.pionMass()*_cst.pionMass()+tmpPtot*tmpPtot); + double tmpEtotCorr = tmpEtot; + if(_fitOutward) { + tmpEtotCorr -= _thickness*dEdx; // apply energy _loss_ when fitting inside-out + } else { + tmpEtotCorr += _thickness*dEdx; // apply energy _gain_ when fitting outside-in + } + ratioCorr = tmpEtotCorr/tmpEtot; + }

+ public void applyMSCorrection(boolean bool) {_applyMSCorrection=bool;} + public void applyDEDXCorrection(boolean bool) { + System.out.print("KFSite: Setting dedx bool from: "); + System.out.print(_applyDEDXCorrection); + System.out.print(" to "); + _applyDEDXCorrection=bool; + System.out.println(_applyDEDXCorrection); + } +

lcsim/src/org/lcsim/contrib/KFFitter

KFTrack.java 1.2 -> 1.3

diff -u -r1.2 -r1.3
--- KFTrack.java	30 Jan 2007 17:01:58 -0000	1.2
+++ KFTrack.java	1 Feb 2007 22:29:34 -0000	1.3
@@ -27,9 +27,51 @@

 	kfhit.isMeasurementSite(true);
 	double radius = kfhit.getSiteLocation().getR();
 	double radThickness = 0.00107;

+	if (radius>10.) {radThickness = 0.00532;}
+	double thickness = 0.01;
+	if (radius>10.) {thickness = 0.03;}
+	kfhit.isMaterialSite(true,radThickness,thickness);
+
+	kfhit.applyMSCorrection(true);
+	kfhit.applyDEDXCorrection(true);
+
+	_siteList.add(kfhit);
+	//System.out.print("KFTrack: read back radius = ");
+	//System.out.println(kfhit.getSiteLocation().getR());
+
+    }
+
+    public void addHitDebug(SimTrackerHit hit)
+    {
+
+	System.out.println("                      ************************");
+	System.out.println("KFTrack: addHitDebug  **     DEBUG MODE     **");
+	System.out.println("                      ************************");
+
+
+	double[] pos = hit.getPoint();
+	// convert to cm
+	for(int i=0; i<3; i++) {pos[i]*=0.1;}
+	double radius = Math.sqrt(pos[0]*pos[0]+pos[1]*pos[1]);
+
+	KFTrackParameters seedExtrap = _seedTrack.makePredictionAtRadius(radius);
+	double[] p = seedExtrap.getPointParameters();
+
+	// define site for this hit
+	KFSite kfhit = new KFSite();
+	kfhit.setLocation(p[0],p[1],p[2]);
+
+	kfhit.isMeasurementSite(true);
+	double radThickness = 0.00107;

 	if (radius>10.) {radThickness=0.00532;}

-	kfhit.isMaterialSite(true,radThickness);

+	double thickness = 0.01;
+	if (radius>10.) {thickness = 0.03;}
+	kfhit.isMaterialSite(false,radThickness,thickness);

 	_siteList.add(kfhit);

+
+	kfhit.applyMSCorrection(false);
+	kfhit.applyDEDXCorrection(false);
+

 	//System.out.print("KFTrack: read back radius = ");
 	//System.out.println(kfhit.getSiteLocation().getR());

@@ -85,27 +127,91 @@

+    public void defineSeedTrack()
+    {
+	// ************************
+	// For testing purpose only
+	// ************************
+
+	System.out.println("                          ************************");
+	System.out.println("KFTrack: defineSeedTrack  **     DEBUG MODE     **");
+	System.out.println("                          ************************");
+
+	// load track parameters d0,omega,...
+	double d = 0.;
+	double phi = 0;
+	//double omega = 0.00300; // 5GeV
+	//double omega = 0.00075; // 20GeV
+	double omega = 0.00015; // 100GeV
+	double z = 0.;
+	double s = 0.;
+
+	double l = 0.0; // path length set to zero (=POCA)
+
+	System.out.print("KFTrack: define track par = ");
+	System.out.print(d); System.out.print(" ");
+	System.out.print(phi); System.out.print(" ");
+	System.out.print(omega); System.out.print(" ");
+	System.out.print(z); System.out.print(" ");
+	System.out.println(s);
+
+	_seedTrack.setTrackFitParameters(d,phi,omega,z,s,l);
+
+    }
+

     public void fit()
     {
 	// sort list of hits by distance from origin
 	sortSiteList();

+	int initChg = _seedTrack.getCharge();
+	// initialize track param correlation matrix
+	double[][] cor = new double[5][5];
+	cor[0][0] = 1.; // d0
+	cor[0][1] = cor[1][0] =  initChg*0.6; // d0 - phi0
+	cor[0][2] = cor[2][0] = -initChg*0.3; // d0 - omega
+	cor[1][1] = 1.; // phi0
+	cor[1][2] = cor[2][1] = -0.8; // phi0 - omega
+	cor[2][2] = 1.; // omega
+	cor[3][3] = 1.; // z0
+	cor[3][4] = cor[4][3] = -initChg*0.9; // z0 - tanl
+	cor[4][4] = 1.; // tanl
+	Matrix initCor = new Matrix(cor);
+

 	// initialize track param covariance matrix
 	double[][] diag = new double[5][5];
 	diag[0][0] =  10.; // d0
 	diag[1][1] =   1.; // phi0
 	diag[2][2] = 100.; // omega

-	diag[3][3] =  10.; // z0
-	diag[4][4] =  10.; // tanl
-	Matrix initCov = new Matrix(diag);

+	diag[3][3] = 100.; // z0
+	diag[4][4] = 100.; // tanl
+	Matrix initDiag = new Matrix(diag);
+
+	//Matrix initCov = new Matrix();
+	Matrix initCov = initDiag.times(initCor).times(initDiag);

 	System.out.println("KFTrack: init cov matrix");
 	initCov.print(6,2);
 
 	System.out.print("KFTrack: ");
 	_seedTrack.printFitParams();

+	ListIterator hits = _orderedSiteList.listIterator();
+
+	//***************
+	// Fit inside-out
+	//***************
+	System.out.println("KFTrack: Fitting Track outward");
+
+	// Determine seed track parameters from 3 innermost hits
+	KFTrackParameters inOutSeedTrk =
+	    seedTrkFromHits(_seedTrack.getCharge(),
+			    (KFSite)_orderedSiteList.get(0),
+			    (KFSite)_orderedSiteList.get(1),
+			    (KFSite)_orderedSiteList.get(2));
+

 	// initialize track parameters for outward fit

-	KFTrackParameters currentTrackOut = new KFTrackParameters(_seedTrack);

+	//KFTrackParameters currentTrackOut = new KFTrackParameters(_seedTrack);
+	KFTrackParameters currentTrackOut = new KFTrackParameters(inOutSeedTrk);

 	// determine track params at fixed small radius
 	//_seedTrack.setPointToPOCA();
 	double refRadius = 1.0;

@@ -115,24 +221,6 @@

 	// initialize error matrix
 	currentTrackOut.setErrorMatrix(initCov);

-
-	// initialize track parameters for inward fit
-	KFTrackParameters currentTrackIn = new KFTrackParameters(_seedTrack);
-	// determine track params at fixed small radius
-	refRadius = 130.0;
-	currentTrackIn.setPointToRadius(refRadius);
-	System.out.print("KFTrack: set radius to 130 : ");
-	currentTrackIn.printFitParams();
-	// initialize error matrix
-	currentTrackIn.setErrorMatrix(initCov);
-
-	ListIterator hits = _orderedSiteList.listIterator();
-
-	//***************
-	// Fit inside-out
-	//***************
-	System.out.println("KFTrack: Fitting Track outward");
-

 	for (; hits.hasNext();){
 
 	    KFSite hit = (KFSite)hits.next();

@@ -153,6 +241,7 @@

 	    predictedTrack.printPointParams();
 
 	    // filter predicted track parameters with hit at radius R

+	    hit.setFitOutward();

 	    currentTrackOut = hit.filter(predictedTrack);
 
 	    // record track parameters in KFSite

@@ -164,6 +253,25 @@

 	//***************
 	System.out.println("KFTrack: Fitting Track inward");

+	// Determine seed track parameters from 3 outtermost hits
+	//int hitListSize = _orderedSiteList.size();
+	//KFTrackParameters outInSeedTrk =
+	//    seedTrkFromHits(_seedTrack.getCharge(),
+	//		    (KFSite)_orderedSiteList.get(hitListSize-1),
+	//		    (KFSite)_orderedSiteList.get(hitListSize-2),
+	//		    (KFSite)_orderedSiteList.get(hitListSize-3));
+	//KFTrackParameters currentTrackIn = new KFTrackParameters(outInSeedTrk);
+
+	// initialize track parameters for inward fit
+	KFTrackParameters currentTrackIn = new KFTrackParameters(currentTrackOut);
+	// determine track params at fixed small radius
+	refRadius = 130.0;
+	currentTrackIn.setPointToRadius(refRadius);
+	System.out.print("KFTrack: set radius to 130 : ");
+	currentTrackIn.printFitParams();
+	// initialize error matrix
+	currentTrackIn.setErrorMatrix(initCov);
+

 	for (; hits.hasPrevious();){
 
 	    KFSite hit = (KFSite)hits.previous();

@@ -183,6 +291,7 @@

 	    predictedTrack.printFitParams();
 
 	    // filter predicted track parameters with hit at radius R

+	    hit.setFitInward();

 	    currentTrackIn = hit.filter(predictedTrack);
 
 	    // record track parameters in KFSite

@@ -191,8 +300,76 @@

 	}
 
 	_fittedPtAtPOCA = currentTrackIn.getPt();

-	System.out.print("KFTrack: Final fitted pt = ");

+	System.out.print("KFTrack: Final fitted pt  = ");

 	System.out.println(_fittedPtAtPOCA);

+	System.out.print("KFTrack: Final fitted par = ");
+	currentTrackIn.printFitParams();
+	System.out.print("KFTrack: Final cov matrix = ");
+	currentTrackIn.getErrorMatrix().print(15,12);
+	Matrix cov = currentTrackIn.getErrorMatrix();
+	double[][] covArray = cov.getArrayCopy();
+	double[] error = new double[5];
+	for(int c=0; c<5; c++) {error[c] = Math.sqrt(covArray[c][c]);}
+	for(int a=0; a<5; a++) {
+	    for(int b=0; b<5; b++) {
+		covArray[a][b] /= error[a]*error[b];
+	    }
+	}
+	Matrix corMat = new Matrix(covArray);
+	System.out.print("KFTrack: Final corrlation matrix = ");
+	corMat.print(15,12);
+    }
+
+    private KFTrackParameters seedTrkFromHits(int q, KFSite p1, KFSite p2, KFSite p3)
+    {
+	double x1 = p1.getSiteLocation().getX();
+	double y1 = p1.getSiteLocation().getY();
+	double z1 = p1.getSiteLocation().getZ();
+	double x2 = p2.getSiteLocation().getX();
+	double y2 = p2.getSiteLocation().getY();
+	double z2 = p2.getSiteLocation().getZ();
+	double x3 = p3.getSiteLocation().getX();
+	double y3 = p3.getSiteLocation().getY();
+
+	double r1 = Math.sqrt(x1*x1+y1*y1);
+	double r2 = Math.sqrt(x2*x2+y2*y2);
+
+	System.out.println("KFTrack: hit coordinates = ");
+	System.out.print(x1); System.out.print(" ");
+	System.out.print(y1); System.out.print(" ");
+	System.out.print(z1); System.out.print(" ");
+	System.out.println(r1);
+	System.out.print(x2); System.out.print(" ");
+	System.out.print(y2); System.out.print(" ");
+	System.out.print(z2); System.out.print(" ");
+	System.out.println(r2);
+	System.out.print(x3); System.out.print(" ");
+	System.out.print(y3); System.out.print(" ");
+	System.out.println(Math.sqrt(x3*x3+y3*y3));
+
+	double denom = (x1-x2)*(y3-y2)-(x2-x3)*(y2-y1);
+	double alpha = 0.5*((x1-x3)*(x2-x3)+(y1-y3)*(y2-y3))/denom;
+	double beta  = 0.5*((x1-x3)*(x1-x2)+(y1-y3)*(y1-y2))/denom;
+
+	double xc = 0.5*(x1+x2)+alpha*(y2-y1);
+	double yc = 0.5*(y1+y2)+beta*(x1-x2);
+
+	double r0 = Math.sqrt(xc*xc+yc*yc);
+	double rc = Math.sqrt((xc-x1)*(xc-x1)+(yc-y1)*(yc-y1));
+
+	double d0 = r0-rc;
+	double phi0 = Math.atan2(q*xc,-q*yc);
+	double omega = -q/rc;
+
+	double s = (z2-z1)/(r2-r1);
+	double z0 = z1+(Math.abs(d0)-r1)*s;
+
+	double l = 0.;
+
+	KFTrackParameters seedTrk = new KFTrackParameters();
+	seedTrk.setTrackFitParameters(d0,phi0,omega,z0,s,l);
+
+	return seedTrk;

     }
 
     public KFTrackParameters getTrackParameters() { return _seedTrack; }

Commit in `lcsim/src/org/lcsim/contrib/KFFitter` on MAIN
`History`	+26	-1	1.1 -> 1.2
`KFFitterDriver.java`	+2		1.2 -> 1.3
`KFPoint.java`		-2	1.2 -> 1.3
`KFSite.java`	+85	-24	1.2 -> 1.3
`KFTrack.java`	+201	-24	1.2 -> 1.3
`KFTrackParameters.java`	+1	-1	1.2 -> 1.3
	+315	-52