X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=geom.cpp;h=a23352c0f11d588c6254e3b16c4d1932a381b37e;hb=b7cce6e0a45013919e76a266533fcca4052cf157;hp=1a7c2bfd601340c85cf5b73b06b280550b8f1f60;hpb=c5c7502f435e1413c19e373dab1dfebcaa67588d;p=mothur.git

diff --git a/geom.cpp b/geom.cpp
index 1a7c2bf..a23352c 100644
--- a/geom.cpp
+++ b/geom.cpp
@@ -3,7 +3,7 @@
  *  Mothur
  *
  *  Created by Thomas Ryabin on 2/23/09.
- *  Copyright 2009 __MyCompanyName__. All rights reserved.
+ *  Copyright 2009 Schloss Lab UMASS Amherst. All rights reserved.
  *
  */
 
@@ -21,8 +21,7 @@ RAbundVector Geom::getRAbundVector(SAbundVector* rank){
 		int nb = 0;
 		int ns = 0;
 		
-		for(int i = rank->size()-1; i > 0; i--)
-		{
+		for(int i = rank->size()-1; i > 0; i--) {
 			int cur = rank->get(i);
 			if(mr == 1 && cur > 0)
 				mr = i;
@@ -42,7 +41,7 @@ RAbundVector Geom::getRAbundVector(SAbundVector* rank){
 /***********************************************************************************/
 EstOutput Geom::getValues(SAbundVector* rank){
 	try {
-		data.resize(2,0);
+		data.resize(3,0);
 		
 		rdata = getRAbundVector(rank);
 		int numInd = rdata.getNumSeqs();
@@ -51,8 +50,7 @@ EstOutput Geom::getValues(SAbundVector* rank){
 		double k = .5;
 		double step = .49999;
 		
-		while(fabs(min - numInd*kEq(k, (double)numSpec)) > .0001) //This uses a binary search to find the value of k.
-		{
+		while(fabs(min - numInd*kEq(k, (double)numSpec)) > .0001) { //This uses a binary search to find the value of k.
 			if(numInd*kEq(k, numSpec) > min)
 				k += step;
 			else
@@ -64,47 +62,41 @@ EstOutput Geom::getValues(SAbundVector* rank){
 		double sumExp = 0;
 		double sumObs = 0;
 		double maxDiff = 0;
-
-		for(int i = 0; i < rdata.size(); i++)
+		
+		for(int i = 0; i < numSpec; i++)
 		{
 			sumObs += rdata.get(i);
 			sumExp += numInd*cK*k*pow(1-k, i);
+			
 			double diff = fabs(sumObs-sumExp);
-			if(diff > maxDiff)
-				maxDiff = diff;
+			if(diff > maxDiff)	{	maxDiff = diff;		}
+			
 		}
 
-		double DStatistic = maxDiff/numInd;
-		double critVal = 0;
+
 		/*cout << "Geom:\n";
 		cout << "D-Statistic = " << DStatistic << "\n";
 		cout << "Critical value for 95% confidence interval = ";*/
-		if(rdata.size() > 20)
-		{
-			critVal = .886/sqrt(rdata.size());
-		}
-		else
-		{
-			KOSTable table;
-			critVal = table.getConfLimit(numSpec);
-		}
+
+		data[0] = maxDiff/numInd;
+		data[1] = 0.886/sqrt(numSpec);
+		data[2] = 1.031/sqrt(numSpec);
+
 		/*cout << critVal << "\n";
 		cout << "If D-Statistic is less than the critical value then the data fits the Geometric Series model w/ 95% confidence.\n\n";*/
-		
-		data[0] = DStatistic;
-		data[1] = critVal;
-		
+
 		if (isnan(data[0]) || isinf(data[0])) { data[0] = 0; }
 		if (isnan(data[1]) || isinf(data[1])) { data[1] = 0; }
+		if (isnan(data[2]) || isinf(data[2])) { data[2] = 0; }
 		
 		return data;
 	}
 	catch(exception& e) {
-		cout << "Standard Error: " << e.what() << " has occurred in the NPShannon class Function getValues. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
+		cout << "Standard Error: " << e.what() << " has occurred in the Geom class Function getValues. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
 		exit(1);
 	}
 	catch(...) {
-		cout << "An unknown error has occurred in the NPShannon class function getValues. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
+		cout << "An unknown error has occurred in the Geom class function getValues. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
 		exit(1);
 	}	
 }