X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=collectdisplay.h;h=15cc5a75a1f32edac6d234ea934ceffdad608301;hb=8ef6687c1f586285d01c000cc5e359bf9c07c717;hp=4fe358800dba8dd40a1132f8330ca7ebe5671956;hpb=0470f6d037aacb3563c3f7010708120a4a67d4e6;p=mothur.git

diff --git a/collectdisplay.h b/collectdisplay.h
index 4fe3588..15cc5a7 100644
--- a/collectdisplay.h
+++ b/collectdisplay.h
@@ -15,6 +15,8 @@ class CollectDisplay : public Display {
 public:
 	CollectDisplay(Calculator* calc, FileOutput* file) : estimate(calc), output(file) {timesCalled = 0;};
 	~CollectDisplay()	{	delete estimate; delete output;		}
+	
+	
 	void update(SAbundVector* rank){
 		nSeqs=rank->getNumSeqs();
 		data = estimate->getValues(rank);
@@ -45,7 +47,7 @@ public:
 			n++;
 		}
 		
-		if (estimate->getMultiple() == true) { 
+		if ((estimate->getMultiple() == true) && all) { 
 			numGroupComb++; 
 			groupData.resize((numGroupComb*data.size()), 0);
 			//is this the time its called with all values
@@ -74,14 +76,23 @@ public:
 	void init(string s)		{	output->initFile(s);	};
 	void reset()			{	output->resetFile();	};
 	void close()			{	output->resetFile();	};
+	void setAll(bool a)		{	all = a;				}
+	bool getAll()			{	return all;				}
+	
 	bool isCalcMultiple() { return estimate->getMultiple(); }
 	
+	string getName()	{  return estimate->getName();  }
+	
+	
 private:
+	
 	Calculator* estimate;
 	FileOutput* output;
 	int nSeqs, timesCalled, numGroupComb;
 	vector<double> data;
 	vector<double> groupData;
+	bool all;
+	
 };
 
 /***********************************************************************/