5 * Created by Sarah Westcott on 1/26/09.
6 * Copyright 2009 Schloss Lab UMASS Amherst. All rights reserved.
10 #include "parsimonycommand.h"
12 /***********************************************************/
13 ParsimonyCommand::ParsimonyCommand() {
15 globaldata = GlobalData::getInstance();
17 //randomtree will tell us if user had their own treefile or if they just want the random distribution
18 randomtree = globaldata->getRandomTree();
20 //user has entered their own tree
21 if (randomtree == "") {
22 T = globaldata->gTree;
23 tmap = globaldata->gTreemap;
24 parsFile = globaldata->getTreeFile() + ".parsimony";
25 parsFileout = globaldata->getTreeFile() + "temp" + ".parsimony";
26 sumFile = globaldata->getTreeFile() + ".psummary";
27 openOutputFile(sumFile, outSum);
28 }else { //user wants random distribution
29 savetmap = globaldata->gTreemap;
31 parsFile = randomtree + ".rd_parsimony";
32 openOutputFile(parsFile, out);
35 //set users groups to analyze
37 convert(globaldata->getIters(), iters); //how many random trees to generate
38 pars = new Parsimony(tmap);
43 cout << "Standard Error: " << e.what() << " has occurred in the ParsimonyCommand class Function ParsimonyCommand. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
47 cout << "An unknown error has occurred in the ParsimonyCommand class function ParsimonyCommand. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
51 /***********************************************************/
52 int ParsimonyCommand::execute() {
55 //get pscore for users tree
56 userData.resize(numComp,0); //data = AB, AC, BC, ABC.
57 randomData.resize(numComp,0); //data = AB, AC, BC, ABC.
58 rscoreFreq.resize(numComp);
59 uscoreFreq.resize(numComp);
60 rCumul.resize(numComp);
61 uCumul.resize(numComp);
62 userTreeScores.resize(numComp);
63 UScoreSig.resize(numComp);
65 if (randomtree == "") {
66 //get pscores for users trees
67 for (int i = 0; i < T.size(); i++) {
68 cout << "Processing tree " << i+1 << endl;
69 userData = pars->getValues(T[i]); //data = AB, AC, BC, ABC.
71 //output scores for each combination
72 for(int k = 0; k < numComp; k++) {
74 it = uscoreFreq[k].find(userData[k]);
75 if (it == uscoreFreq[k].end()) {//new score
76 uscoreFreq[k][userData[k]] = 1;
77 }else{ uscoreFreq[k][userData[k]]++; }
79 //add users score to valid scores
80 validScores[userData[k]] = userData[k];
82 //save score for summary file
83 userTreeScores[k].push_back(userData[k]);
87 //get pscores for random trees
88 for (int j = 0; j < iters; j++) {
89 //create new tree with same num nodes and leaves as users
91 //create random relationships between nodes
92 randT->assembleRandomTree();
93 //get pscore of random tree
94 randomData = pars->getValues(randT);
96 for(int r = 0; r < numComp; r++) {
97 //add trees pscore to map of scores
98 it2 = rscoreFreq[r].find(randomData[r]);
99 if (it2 != rscoreFreq[r].end()) {//already have that score
100 rscoreFreq[r][randomData[r]]++;
101 }else{//first time we have seen this score
102 rscoreFreq[r][randomData[r]] = 1;
105 //add randoms score to validscores
106 validScores[randomData[r]] = randomData[r];
112 //get pscores for random trees
113 for (int j = 0; j < iters; j++) {
114 //create new tree with same num nodes and leaves as users
116 //create random relationships between nodes
117 randT->assembleRandomTree();
118 //get pscore of random tree
119 randomData = pars->getValues(randT);
121 for(int r = 0; r < numComp; r++) {
122 //add trees pscore to map of scores
123 it2 = rscoreFreq[r].find(randomData[r]);
124 if (it2 != rscoreFreq[r].end()) {//already have that score
125 rscoreFreq[r][randomData[r]]++;
126 }else{//first time we have seen this score
127 rscoreFreq[r][randomData[r]] = 1;
130 //add randoms score to validscores
131 validScores[randomData[r]] = randomData[r];
138 for(int a = 0; a < numComp; a++) {
139 float rcumul = 0.0000;
140 float ucumul = 0.0000;
141 //this loop fills the cumulative maps and put 0.0000 in the score freq map to make it easier to print.
142 for (it = validScores.begin(); it != validScores.end(); it++) {
143 if (randomtree == "") {
144 it2 = uscoreFreq[a].find(it->first);
145 //user data has that score
146 if (it2 != uscoreFreq[a].end()) { uscoreFreq[a][it->first] /= T.size(); ucumul+= it2->second; }
147 else { uscoreFreq[a][it->first] = 0.0000; } //no user trees with that score
149 uCumul[a][it->first] = ucumul;
152 //make rscoreFreq map and rCumul
153 it2 = rscoreFreq[a].find(it->first);
154 //get percentage of random trees with that info
155 if (it2 != rscoreFreq[a].end()) { rscoreFreq[a][it->first] /= iters; rcumul+= it2->second; }
156 else { rscoreFreq[a][it->first] = 0.0000; } //no random trees with that score
157 rCumul[a][it->first] = rcumul;
160 //find the signifigance of each user trees score when compared to the random trees and save for printing the summary file
161 for (int h = 0; h < userTreeScores[a].size(); h++) {
162 UScoreSig[a].push_back(rCumul[a][userTreeScores[a][h]]);
166 printParsimonyFile();
167 if (randomtree == "") { printUSummaryFile(); }
169 //reset globaldata's treemap if you just did random distrib
170 if (randomtree != "") { globaldata->gTreemap = savetmap; }
172 //reset randomTree parameter to ""
173 globaldata->setRandomTree("");
174 //reset groups parameter
175 globaldata->Groups.clear(); globaldata->setGroups("");
180 catch(exception& e) {
181 cout << "Standard Error: " << e.what() << " has occurred in the ParsimonyCommand class Function execute. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
185 cout << "An unknown error has occurred in the ParsimonyCommand class function execute. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
190 /***********************************************************/
191 void ParsimonyCommand::printParsimonyFile() {
196 out.setf(ios::fixed, ios::floatfield); out.setf(ios::showpoint);
198 for(int a = 0; a < numComp; a++) {
199 initFile(groupComb[a]);
201 for (it = validScores.begin(); it != validScores.end(); it++) {
202 if (randomtree == "") {
203 data.push_back(it->first); data.push_back(uscoreFreq[a][it->first]); data.push_back(uCumul[a][it->first]); data.push_back(rscoreFreq[a][it->first]); data.push_back(rCumul[a][it->first]);
205 data.push_back(it->first); data.push_back(rscoreFreq[a][it->first]); data.push_back(rCumul[a][it->first]);
215 remove(parsFileout.c_str());
217 catch(exception& e) {
218 cout << "Standard Error: " << e.what() << " has occurred in the ParsimonyCommand class Function printParsimonyFile. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
222 cout << "An unknown error has occurred in the ParsimonyCommand class function printParsimonyFile. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
226 /***********************************************************/
227 void ParsimonyCommand::printUSummaryFile() {
230 outSum << "Tree#" << '\t' << "Comb" << '\t' << "ParsScore" << '\t' << '\t' << "ParsSig" << endl;
233 outSum.setf(ios::fixed, ios::floatfield); outSum.setf(ios::showpoint);
237 for (int i = 0; i< T.size(); i++) {
238 for(int a = 0; a < numComp; a++) {
239 outSum << setprecision(6) << i+1 << '\t' << groupComb[a] << '\t' << '\t' << userTreeScores[a][i] << '\t' << UScoreSig[a][i] << endl;
240 cout << setprecision(6) << i+1 << '\t' << groupComb[a] << '\t' << '\t' << userTreeScores[a][i] << '\t' << UScoreSig[a][i] << endl;
246 catch(exception& e) {
247 cout << "Standard Error: " << e.what() << " has occurred in the ParsimonyCommand class Function printUSummaryFile. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
251 cout << "An unknown error has occurred in the ParsimonyCommand class function printUSummaryFile. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
256 /***********************************************************/
257 void ParsimonyCommand::getUserInput() {
261 tmap = new TreeMap();
263 cout << "Please enter the number of groups you would like to analyze: ";
268 numEachGroup.resize(numGroups, 0);
270 for (int i = 1; i <= numGroups; i++) {
271 cout << "Please enter the number of sequences in group " << i << ": ";
274 //set tmaps seqsPerGroup
275 tmap->seqsPerGroup[toString(i)] = num;
276 tmap->namesOfGroups.push_back(toString(i));
278 //set tmaps namesOfSeqs
279 for (int j = 0; j < num; j++) {
280 tmap->namesOfSeqs.push_back(toString(count));
281 tmap->treemap[toString(count)].groupname = toString(i);
286 //clears buffer so next command doesn't have error
290 //save tmap for later
291 globaldata->gTreemap = tmap;
294 catch(exception& e) {
295 cout << "Standard Error: " << e.what() << " has occurred in the ParsimonyCommand class Function getUserInput. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
299 cout << "An unknown error has occurred in the ParsimonyCommand class function getUserInput. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
303 /***********************************************************/
305 void ParsimonyCommand::setGroups() {
307 string allGroups = "";
309 //if the user has not entered specific groups to analyze then do them all
310 if (globaldata->Groups.size() != 0) {
311 if (globaldata->Groups[0] != "all") {
312 //check that groups are valid
313 for (int i = 0; i < globaldata->Groups.size(); i++) {
314 if (tmap->isValidGroup(globaldata->Groups[i]) != true) {
315 cout << globaldata->Groups[i] << " is not a valid group, and will be disregarded." << endl;
316 // erase the invalid group from globaldata->Groups
317 globaldata->Groups.erase(globaldata->Groups.begin()+i);
321 //if the user only entered invalid groups
322 if (globaldata->Groups.size() == 0) {
323 cout << "When using the groups parameter you must have at least 1 valid group. I will run the command using all the groups in your groupfile." << endl;
324 for (int i = 0; i < tmap->namesOfGroups.size(); i++) {
325 globaldata->Groups.push_back(tmap->namesOfGroups[i]);
327 allGroups += tmap->namesOfGroups[i];
330 for (int i = 0; i < globaldata->Groups.size(); i++) {
331 allGroups += tmap->namesOfGroups[i];
335 }else{//user has enter "all" and wants the default groups
336 for (int i = 0; i < tmap->namesOfGroups.size(); i++) {
337 globaldata->Groups.push_back(tmap->namesOfGroups[i]);
339 allGroups += tmap->namesOfGroups[i];
341 globaldata->setGroups("");
344 for (int i = 0; i < tmap->namesOfGroups.size(); i++) {
345 allGroups += tmap->namesOfGroups[i];
350 //calculate number of comparsions
352 for (int r=0; r<numGroups; r++) {
353 for (int l = r+1; l < numGroups; l++) {
354 groupComb.push_back(globaldata->Groups[r]+globaldata->Groups[l]);
361 groupComb.push_back(allGroups);
366 catch(exception& e) {
367 cout << "Standard Error: " << e.what() << " has occurred in the ParsimonyCommand class Function setGroups. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
371 cout << "An unknown error has occurred in the ParsimonyCommand class function setGroups. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
376 /*****************************************************************/
378 void ParsimonyCommand::initFile(string label){
381 openOutputFile(parsFileout, out);
382 openInputFile(parsFile, inFile);
385 getline(inFile, inputBuffer);
387 if (randomtree == "") {
388 out << inputBuffer << '\t' << label + "Score" << '\t' << label + "UserFreq" << '\t' << label + "UserCumul" << '\t' << label + "RandFreq" << '\t' << label + "RandCumul" << endl;
390 out << inputBuffer << '\t' << label + "Score" << '\t' << label + "RandFreq" << '\t' << label + "RandCumul" << endl;
393 openOutputFile(parsFileout, out);
395 if (randomtree == "") {
396 out << label + "Score" << '\t' << label + "UserFreq" << '\t' << label + "UserCumul" << '\t' << label + "RandFreq" << '\t' << label + "RandCumul" << endl;
398 out << label + "Score" << '\t' << label + "RandFreq" << '\t' << label + "RandCumul" << endl;
402 out.setf(ios::fixed, ios::floatfield);
403 out.setf(ios::showpoint);
405 catch(exception& e) {
406 cout << "Standard Error: " << e.what() << " has occurred in the ParsimonyCommand class Function initFile. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
410 cout << "An unknown error has occurred in the ParsimonyCommand class function initFile. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
415 /***********************************************************************/
417 void ParsimonyCommand::output(vector<double> data){
421 getline(inFile, inputBuffer);
423 if (randomtree == "") {
424 out << inputBuffer << '\t' << setprecision(6) << data[0] << '\t' << data[1] << '\t' << data[2] << '\t' << data[3] << '\t' << data[4] << endl;
426 out << inputBuffer << '\t' << setprecision(6) << data[0] << '\t' << data[1] << '\t' << data[2] << endl;
430 if (randomtree == "") {
431 out << setprecision(6) << data[0] << '\t' << data[1] << '\t' << data[2] << '\t' << data[3] << '\t' << data[4] << endl;
433 out << setprecision(6) << data[0] << '\t' << data[1] << '\t' << data[2] << endl;
438 catch(exception& e) {
439 cout << "Standard Error: " << e.what() << " has occurred in the ParsimonyCommand class Function output. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
443 cout << "An unknown error has occurred in the ParsimonyCommand class function output. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
448 /***********************************************************************/
450 void ParsimonyCommand::resetFile(){
461 remove(parsFile.c_str());
462 rename(parsFileout.c_str(), parsFile.c_str());
464 catch(exception& e) {
465 cout << "Standard Error: " << e.what() << " has occurred in the ParsimonyCommand class Function resetFile. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
469 cout << "An unknown error has occurred in the ParsimonyCommand class function resetFile. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";