readtreecommand.cpp

   1 /*
   2  *  readtreecommand.cpp
   3  *  Mothur
   4  *
   5  *  Created by Sarah Westcott on 1/23/09.
   6  *  Copyright 2009 Schloss Lab UMASS Amherst. All rights reserved.
   7  *
   8  */
   9
  10 #include "readtreecommand.h"
  11
  12 //**********************************************************************************************************************
  13 ReadTreeCommand::ReadTreeCommand(string option){
  14         try {
  15                 globaldata = GlobalData::getInstance();
  16                 abort = false;
  17
  18                 //allow user to run help
  19                 if(option == "help") { help(); abort = true; }
  20
  21                 else {
  22                         //valid paramters for this command
  23                         string Array[] =  {"tree","group"};
  24                         vector<string> myArray (Array, Array+(sizeof(Array)/sizeof(string)));
  25
  26                         OptionParser parser(option);
  27                         map<string, string> parameters = parser.getParameters();
  28
  29                         ValidParameters validParameter;
  30
  31                         //check to make sure all parameters are valid for command
  32                         for (map<string, string>::iterator it = parameters.begin(); it != parameters.end(); it++) {
  33                                 if (validParameter.isValidParameter(it->first, myArray, it->second) != true) {  abort = true;  }
  34                         }
  35
  36                         globaldata->newRead();
  37
  38                         //check for required parameters
  39                         treefile = validParameter.validFile(parameters, "tree", true);
  40                         if (treefile == "not open") { abort = true; }
  41                         else if (treefile == "not found") { treefile = ""; cout << "tree is a required parameter for the read.tree command." << endl; abort = true;  }
  42 //                      else {  globaldata->setTreeFile(treefile);  globaldata->setFormat("tree");      }
  43
  44                         groupfile = validParameter.validFile(parameters, "group", true);
  45                         if (groupfile == "not open") { abort = true; }
  46                         else if (groupfile == "not found") { groupfile = ""; cout << "group is a required parameter for the read.tree command." << endl; abort = true;  }
  47                         else {
  48                                 globaldata->setGroupFile(groupfile);
  49                                 //read in group map info.
  50                                 treeMap = new TreeMap(groupfile);
  51                                 treeMap->readMap();
  52                                 globaldata->gTreemap = treeMap;
  53                         }
  54
  55                         if (abort == false) {
  56                                 filename = treefile;
  57                                 read = new ReadNewickTree(filename);
  58                         }
  59
  60                 }
  61         }
  62         catch(exception& e) {
  63                 cout << "Standard Error: " << e.what() << " has occurred in the ReadTreeCommand class Function ReadTreeCommand. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
  64                 exit(1);
  65         }
  66         catch(...) {
  67                 cout << "An unknown error has occurred in the ReadTreeCommand class function ReadTreeCommand. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
  68                 exit(1);
  69         }
  70 }
  71 //**********************************************************************************************************************
  72
  73 void ReadTreeCommand::help(){
  74         try {
  75                 cout << "The read.tree command must be run before you execute a unifrac.weighted, unifrac.unweighted. " << "\n";
  76                 cout << "It also must be run before using the parsimony command, unless you are using the randomtree parameter." << "\n";
  77                 cout << "The read.tree command should be in the following format: read.tree(tree=yourTreeFile, group=yourGroupFile)." << "\n";
  78                 cout << "The tree and group parameters are both required." << "\n";
  79                 cout << "Note: No spaces between parameter labels (i.e. tree), '=' and parameters (i.e.yourTreefile)." << "\n" << "\n";
  80         }
  81         catch(exception& e) {
  82                 cout << "Standard Error: " << e.what() << " has occurred in the ReadTreeCommand class Function help. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
  83                 exit(1);
  84         }
  85         catch(...) {
  86                 cout << "An unknown error has occurred in the ReadTreeCommand class function help. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
  87                 exit(1);
  88         }
  89 }
  90
  91 //**********************************************************************************************************************
  92
  93 ReadTreeCommand::~ReadTreeCommand(){
  94         delete read;
  95 }
  96
  97 //**********************************************************************************************************************
  98
  99 int ReadTreeCommand::execute(){
 100         try {
 101
 102                 if (abort == true) { return 0; }
 103
 104                 int readOk;
 105
 106                 readOk = read->read();
 107
 108                 if (readOk != 0) { cout << "Read Terminated." << endl; globaldata->gTree.clear(); delete globaldata->gTreemap; return 0; }
 109
 110                 vector<Tree*> T = globaldata->gTree;
 111
 112                 //assemble users trees
 113                 for (int i = 0; i < T.size(); i++) {
 114                         T[i]->assembleTree();
 115                 }
 116
 117                 //output any names that are in names file but not in tree
 118                 if (globaldata->Treenames.size() < treeMap->getNumSeqs()) {
 119                         for (int i = 0; i < treeMap->namesOfSeqs.size(); i++) {
 120                                 //is that name in the tree?
 121                                 int count = 0;
 122                                 for (int j = 0; j < globaldata->Treenames.size(); j++) {
 123                                         if (treeMap->namesOfSeqs[i] == globaldata->Treenames[j]) { break; } //found it
 124                                         count++;
 125                                 }
 126
 127                                 //then you did not find it so report it
 128                                 if (count == globaldata->Treenames.size()) {
 129                                         cout << treeMap->namesOfSeqs[i] << " is in your namefile and not in your tree. It will be disregarded." << endl;
 130                                 }
 131                         }
 132                 }
 133
 134                 return 0;
 135         }
 136         catch(exception& e) {
 137                 cout << "Standard Error: " << e.what() << " has occurred in the ReadTreeCommand class Function execute. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
 138                 exit(1);
 139         }
 140         catch(...) {
 141                 cout << "An unknown error has occurred in the ReadTreeCommand class function execute. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
 142                 exit(1);
 143         }
 144 }
 145
 146 //**********************************************************************************************************************