]> git.donarmstrong.com Git - mothur.git/blob - errorchecking.cpp
59cfb66fcc3d0f3e7136eabfebeacb4cd419d76c
[mothur.git] / errorchecking.cpp
1 /*
2  *  errorchecking.cpp
3  *  Dotur
4  *
5  *  Created by Sarah Westcott on 1/2/09.
6  *  Copyright 2009 Schloss Lab UMASS Amherst. All rights reserved.
7  *
8  */
9
10 #include "errorchecking.h"
11
12 /*******************************************************/
13
14 /******************************************************/
15
16 ErrorCheck::ErrorCheck() {
17         globaldata = GlobalData::getInstance();
18         validCommand = new ValidCommands();
19         validParameter = new ValidParameters();
20 }
21 /*******************************************************/
22
23 /******************************************************/
24
25 void ErrorCheck::refresh() {
26         columnfile = globaldata->getColumnFile();
27         phylipfile = globaldata->getPhylipFile();
28         listfile = globaldata->getListFile();
29         rabundfile = globaldata->getRabundFile();
30         sabundfile = globaldata->getSabundFile();
31         namefile = globaldata->getNameFile();
32         groupfile = globaldata->getGroupFile();
33         orderfile = globaldata->getOrderFile();
34         fastafile = globaldata->getFastaFile();
35         treefile = globaldata->getTreeFile();
36         cutoff = globaldata->getCutOff();
37         format = globaldata->getFormat();
38         method = globaldata->getMethod();
39         randomtree = globaldata->getRandomTree();
40         sharedfile = globaldata->getSharedFile();
41 }
42
43 /*******************************************************/
44
45 /******************************************************/
46
47 ErrorCheck::~ErrorCheck() {
48         delete validCommand;
49         delete validParameter;
50 }
51
52 /*******************************************************/
53
54 /******************************************************/
55
56 bool ErrorCheck::checkInput(string input) {
57                 errorFree = true;
58                 clear();
59                 
60                 //refresh variable
61                 refresh();
62                 
63                 //get command name and parameters
64                 int openParen = input.find_first_of('(');
65                 int closeParen = input.find_last_of(')');
66
67                 if(openParen != -1 && closeParen != -1){                        
68                         commandName = input.substr(0, openParen);   //commandName contains everything before "("
69                         optionText = input.substr(openParen+1, closeParen-openParen-1); //optionString contains everything between "(" and ")".
70                 }else if (openParen == -1) { //there is no parenthesis
71                         cout << input << " is not a valid command. You are missing the ()." << endl;
72                         return false;
73                 }
74                 
75                 //is it a valid command
76                 if (validCommand->isValidCommand(commandName) != true) { return false; }
77                 
78                 string parameter, value;
79                 
80                 //reads in parameters and values
81                 if((optionText != "") && (commandName != "help")){
82                         while((optionText.find_first_of(',') != -1) && (errorFree)) {  //while there are parameters
83                                 splitAtComma(value, optionText);
84                                 splitAtEquals(parameter, value);
85                                 
86                                 //is it a valid parameter
87                                 if (validParameter->isValidParameter(parameter, commandName, value) != true) { return false; }
88
89
90                                 
91                                 if (parameter == "phylip" )             { phylipfile = value; }
92                                 if (parameter == "column" )             { columnfile = value; }
93                                 if (parameter == "list" )               { listfile = value; }
94                                 if (parameter == "rabund" )             { rabundfile = value; }
95                                 if (parameter == "sabund" )             { sabundfile = value; }
96                                 if (parameter == "name" )               { namefile = value; }
97                                 if (parameter == "order" )              { orderfile = value; }
98                                 if (parameter == "fasta" )              { fastafile = value; }
99                                 if (parameter == "tree" )               { treefile = value; }
100                                 if (parameter == "group" )              { groupfile = value; }
101                                 if (parameter == "shared" )             { sharedfile = value; }
102                                 if (parameter == "cutoff" )                     { cutoff = value; }
103                                 if (parameter == "precision" )          { precision = value; }
104                                 if (parameter == "iters" )                      { iters = value; }
105                                 if (parameter == "jumble" )                     { jumble = value; }
106                                 if (parameter == "freq" )                       { freq = value; }
107                                 if (parameter == "method" )                     { method = value; }
108                                 if (parameter == "fileroot" )           { fileroot = value; }
109                                 if (parameter == "line" )                       { line = value; }
110                                 if (parameter == "label" )                      { label = value; }
111                                 if (parameter == "abund" )          { abund = value; }
112                                 if (parameter == "random" )                     { randomtree = value;   }
113                                 if (parameter == "sorted" )                     { sorted = value;       }
114                         }
115                         
116                         //gets the last parameter and value
117                         if (errorFree)  { //gets the last parameter and value
118                                 value = optionText;
119                                 splitAtEquals(parameter, value);
120                                 //is it a valid parameter
121                                 if (validParameter->isValidParameter(parameter, commandName, value) != true) { return false; }
122         
123                                 
124                                 if (parameter == "phylip" )             { phylipfile = value; }
125                                 if (parameter == "column" )             { columnfile = value; }                         
126                                 if (parameter == "list" )               { listfile = value; }
127                                 if (parameter == "rabund" )             { rabundfile = value; }
128                                 if (parameter == "sabund" )             { sabundfile = value; }
129                                 if (parameter == "name" )               { namefile = value; }
130                                 if (parameter == "order" )              { orderfile = value; }
131                                 if (parameter == "group" )              { groupfile = value; }
132                                 if (parameter == "shared" )             { sharedfile = value; }
133                                 if (parameter == "fasta" )              { fastafile = value; }
134                                 if (parameter == "tree" )               { treefile = value; }
135                                 if (parameter == "cutoff" )                     { cutoff = value; }
136                                 if (parameter == "precision" )          { precision = value; }
137                                 if (parameter == "iters" )                      { iters = value; }
138                                 if (parameter == "jumble" )                     { jumble = value; }
139                                 if (parameter == "freq" )                       { freq = value; }
140                                 if (parameter == "method" )                     { method = value; }
141                                 if (parameter == "fileroot" )           { fileroot = value; }
142                                 if (parameter == "line" )                       { line = value; }
143                                 if (parameter == "label" )                      { label = value; }
144                                 if (parameter == "random" )                     { randomtree = value;   }
145                                 if (parameter == "abund" )          { abund = value; }
146                                 if (parameter == "sorted" )                     { sorted = value;       }
147                         }
148                 }
149                 
150                 //make sure the user does not use both the line and label parameters
151                 if ((line != "") && (label != "")) { cout << "You may use either the line or label parameters, but not both." << endl; return false; }
152                 
153                 //check for valid files 
154                 if (commandName == "read.dist") { 
155                         validateReadFiles();
156                         validateReadDist();
157                 }else if (commandName == "read.otu") { 
158                         //you want to do shared commands
159                         if ((listfile != "") && (groupfile != ""))      {
160                                 validateParseFiles(); //checks the listfile and groupfile parameters
161                         //you want to do single commands
162                         }else if ((listfile != "") || (rabundfile != "") || (sabundfile != "")){ 
163                                 validateReadFiles();
164                                 validateReadPhil();
165                         //you have not given a file
166                         }else if ((listfile == "") && (sharedfile == "") && (rabundfile == "") && (sabundfile == "")) {
167                                 cout << "You must enter either a listfile, rabundfile, sabundfile or a sharedfile with the read.otu command. " << endl; return false; 
168                         //you want to do shared commands with a shared file
169                         }else if (sharedfile != "") {//you are reading a shared file
170                                 validateReadFiles();
171                         }
172                 }else if (commandName == "read.tree") { 
173                         validateTreeFiles(); //checks the treefile and groupfile parameters
174                 }else if (commandName == "deconvolute") {
175                         if (fastafile == "") { cout << "You must enter a fastafile with the deconvolute() command." << endl; return false; }
176                         validateReadFiles();
177                 }
178                 
179                 //are you trying to cluster before you have read something                      
180                 if (((commandName == "cluster") && (globaldata->gSparseMatrix == NULL)) ||
181                         ((commandName == "cluster") && (globaldata->gListVector == NULL))) {
182                                 cout << "Before you use the cluster command, you first need to read in a distance matrix." << endl; 
183                                 errorFree = false;
184                 } 
185                 
186                 if ((commandName == "libshuff") && ((globaldata->gMatrix == NULL) || (globaldata->gGroupmap == NULL))) {
187                          cout << "You must read in a matrix and groupfile using the read.dist command, before you use the libshuff command. " << endl; return false; 
188                 }
189                 
190                 if (commandName == "parsimony") {
191                         //are you trying to use parsimony without reading a tree or saying you want random distribution
192                         if (randomtree == "")  {
193                                 if (globaldata->gTree.size() == 0) {
194                                         cout << "You must read a treefile and a groupfile or set the randomtree parameter to the output filename you wish, before you may execute the parsimony command." << endl; return false;  }
195                         }
196                 }
197                 
198                 if ((commandName == "unifrac.weighted") || (commandName == "unifrac.unweighted")) {
199                         if (globaldata->gTree.size() == 0) {//no trees were read
200                                 cout << "You must execute the read.tree command, before you may execute the unifrac.weighted or unifrac.unweighted command." << endl; return false;  }
201                 }
202                 
203                 //check for valid method
204                 if(commandName == "get.group") {
205                         if ((globaldata->getGroupFile() == "")) { cout << "You must read a group before you can use the get.group command." << endl; return false; }
206                 }
207                 if (commandName == "get.label" || commandName == "get.line") {
208                         if ((globaldata->getListFile() == "") && (globaldata->getRabundFile() == "") && (globaldata->getSabundFile() == "")) { cout << "You must read a list, sabund or rabund before you can use the get.label or get.line command." << endl; return false; }
209                 }
210                 if (commandName == "cluster") {
211                         if ((method == "furthest") || (method == "nearest") || (method == "average")) { }
212                         else {cout << "Not a valid clustering method.  Valid clustering algorithms are furthest, nearest or average." << endl; return false; }
213                 }
214                 
215                 if ((commandName == "collect.single") || (commandName == "rarefaction.single") || (commandName == "summary.single") ){ 
216                         if ((globaldata->getListFile() == "") && (globaldata->getRabundFile() == "") && (globaldata->getSabundFile() == "")) { cout << "You must read a list, sabund or rabund before you can use the collect.single, rarefaction.single or summary.single commands." << endl; return false; }
217                 }
218                 
219                 if ((commandName == "collect.shared") || (commandName == "rarefaction.shared") || (commandName == "summary.shared") ){ 
220                         if (globaldata->getSharedFile() == "") {
221                                 if (globaldata->getListFile() == "") { cout << "You must read a list and a group, or a shared before you can use the collect.shared, rarefaction.shared or summary.shared commands." << endl; return false; }
222                                 else if (globaldata->getGroupFile() == "") { cout << "You must read a list and a group, or a shared before you can use the collect.shared, rarefaction.shared or summary.shared commands." << endl; return false; }
223                         }
224                 }
225                 
226                 if ((commandName == "heatmap") || (commandName == "venn")) { 
227                         if ((globaldata->getListFile() == "") && (globaldata->getSharedFile() == "")) {
228                                  cout << "You must read a list, or a list and a group, or a shared before you can use the heatmap or venn commands." << endl; return false; 
229                         }
230                 }
231
232                 return errorFree;
233 }
234
235 /*******************************************************/
236
237 /******************************************************/
238 //This function checks to make sure the user entered a file to 
239 // read and that the file exists and can be opened.
240 void ErrorCheck::validateReadFiles() {
241         try {
242                 //Validating files for read
243                 ifstream filehandle;
244                 int ableToOpen;
245         
246                 //are we reading a phylipfile
247                 if (phylipfile != "") {
248                         ableToOpen = openInputFile(phylipfile, filehandle);
249                         filehandle.close();
250                         //unable to open
251                         if (ableToOpen == 1) { errorFree = false; }
252                         else { globaldata->inputFileName = phylipfile; }
253                 //are we reading a columnfile
254                 }else if (columnfile != "") {
255                         ableToOpen = openInputFile(columnfile, filehandle);
256                         filehandle.close();
257                         //unable to open
258                         if (ableToOpen == 1) { errorFree = false; }
259                         else { globaldata->inputFileName = columnfile; }
260                 //are we reading a listfile
261                 }else if (listfile!= "") {
262                         ableToOpen = openInputFile(listfile, filehandle);
263                         filehandle.close();
264                         //unable to open
265                         if (ableToOpen == 1) {  errorFree = false; }
266                         else { globaldata->inputFileName = listfile; }
267                 //are we reading a rabundfile
268                 }else if (rabundfile != "") {
269                         ableToOpen = openInputFile(rabundfile, filehandle);
270                         filehandle.close();
271                         //unable to open
272                         if (ableToOpen == 1) {  errorFree = false; }
273                         else { globaldata->inputFileName = rabundfile; }
274                 //are we reading a sabundfile
275                 }else if (sabundfile != "") {
276                         ableToOpen = openInputFile(sabundfile, filehandle);
277                         filehandle.close();
278                         //unable to open
279                         if (ableToOpen == 1) {  errorFree = false; }
280                         else { globaldata->inputFileName = sabundfile; }
281                 }else if (fastafile != "") {
282                         ableToOpen = openInputFile(fastafile, filehandle);
283                         filehandle.close();
284                         //unable to open
285                         if (ableToOpen == 1) {  errorFree = false; }
286                         else { globaldata->inputFileName = fastafile; }
287                 }else if (sharedfile != "") {
288                         ableToOpen = openInputFile(sharedfile, filehandle);
289                         filehandle.close();
290                         //unable to open
291                         if (ableToOpen == 1) {  errorFree = false; }
292                         else { globaldata->inputFileName = sharedfile; }
293                 }else{ //no file given
294                         errorFree = false;
295                 }
296         }
297         catch(exception& e) {
298                 cout << "Standard Error: " << e.what() << " has occurred in the ErrorCheck class Function validateReadFiles. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
299                 exit(1);
300         }
301         catch(...) {
302                 cout << "An unknown error has occurred in the ErrorCheck class function validateReadFiles. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
303                 exit(1);
304         }
305         
306 }
307 /*******************************************************/
308
309 /******************************************************/
310 //This function checks to make sure the user entered appropriate
311 // format parameters on a distfile read
312 void ErrorCheck::validateReadDist() {
313         try {
314                 ifstream filehandle;
315                 int ableToOpen;
316                 
317                 if (groupfile != "") {
318                         ableToOpen = openInputFile(groupfile, filehandle);
319                         filehandle.close();
320                         //unable to open
321                         if (ableToOpen == 1) {  errorFree = false; }
322                 }
323                 
324                 if ((phylipfile == "") && (columnfile == "")) { cout << "When executing a read.dist you must enter a phylip or a column." << endl; errorFree = false; }
325                 else if ((phylipfile != "") && (columnfile != "")) { cout << "When executing a read.dist you must enter ONLY ONE of the following: phylip or column." << endl; errorFree = false; }
326                 
327                 if (columnfile != "") {
328                         if (namefile == "") {
329                                 cout << "You need to provide a namefile if you are going to use the column format." << endl;
330                                 errorFree = false; 
331                         }else {
332                                 ableToOpen = openInputFile(namefile, filehandle);
333                                 filehandle.close();
334                                 //unable to open
335                                 if (ableToOpen == 1) { errorFree = false; }
336                         }
337                 }
338         }
339         catch(exception& e) {
340                 cout << "Standard Error: " << e.what() << " has occurred in the ErrorCheck class Function validateReadDist. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
341                 exit(1);
342         }
343         catch(...) {
344                 cout << "An unknown error has occurred in the ErrorCheck class function validateReadDist. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
345                 exit(1);
346         }
347 }
348 /*******************************************************/
349
350 /******************************************************/
351 //This function checks to make sure the user entered appropriate
352 // format parameters on a parselistcommand
353 void ErrorCheck::validateParseFiles() {
354         try {
355                 ifstream filehandle;
356                 int ableToOpen;
357                 
358                 //checks for valid files
359         
360                 if (listfile == "") { cout << "When executing a read.otu for groups you must enter a list and a group." << endl; errorFree = false; }
361                 else if (groupfile == "") { cout << "When executing a read.otu for groups you must enter a list and a group." << endl; errorFree = false; }
362         
363                 //checks parameters on the read command
364                 if (listfile != "") {
365                         ableToOpen = openInputFile(listfile, filehandle);
366                         filehandle.close();
367                         if (ableToOpen == 1) { //unable to open
368                                 errorFree = false;
369                         }
370                         if (groupfile != "") {
371                                 ableToOpen = openInputFile(groupfile, filehandle);
372                                 filehandle.close();
373                                 if (ableToOpen == 1) { //unable to open
374                                         errorFree = false;;
375                                 }
376                         }
377                 }
378         }
379         catch(exception& e) {
380                 cout << "Standard Error: " << e.what() << " has occurred in the ErrorCheck class Function validateReadPhil. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
381                 exit(1);
382         }
383         catch(...) {
384                 cout << "An unknown error has occurred in the ErrorCheck class function validateReadPhil. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
385                 exit(1);
386         }
387 }
388 /*******************************************************/
389
390 /******************************************************/
391 //This function checks to make sure the user entered appropriate
392 // format parameters on a parselistcommand
393 void ErrorCheck::validateTreeFiles() {
394         try {
395                 ifstream filehandle;
396                 int ableToOpen;
397                 
398                 //checks for valid files
399         
400                 if (treefile == "") { cout << "When executing a read.tree you must enter a treefile and a groupfile." << endl; errorFree = false; }
401                 else if (groupfile == "") { cout << "When executing a read.tree you must enter a treefile and a groupfile." << endl; errorFree = false; }
402         
403                 //checks parameters on the read command
404                 if (treefile != "") {
405                         ableToOpen = openInputFile(treefile, filehandle);
406                         filehandle.close();
407                         if (ableToOpen == 1) { //unable to open
408                                 errorFree = false;
409                         }
410                         if (groupfile != "") {
411                                 ableToOpen = openInputFile(groupfile, filehandle);
412                                 filehandle.close();
413                                 if (ableToOpen == 1) { //unable to open
414                                         errorFree = false;;
415                                 }
416                         }
417                 }
418         }
419         catch(exception& e) {
420                 cout << "Standard Error: " << e.what() << " has occurred in the ErrorCheck class Function validateTreeFiles. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
421                 exit(1);
422         }
423         catch(...) {
424                 cout << "An unknown error has occurred in the ErrorCheck class function validateTreeFiles. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
425                 exit(1);
426         }
427 }
428
429 /*******************************************************/
430
431 /******************************************************/
432 //This function checks to make sure the user entered appropriate
433 // format parameters on a distfile read
434 void ErrorCheck::validateReadPhil() {
435         try {
436                 ifstream filehandle;
437                 int ableToOpen;
438                 
439                 //checks to make sure only one file type is given
440                 if (listfile != "") { 
441                         if ((rabundfile != "") || (sabundfile != "")) { 
442                                 cout << "When executing a read.otu you must enter ONLY ONE of the following: list, rabund or sabund." << endl; errorFree = false; }
443                 }else if (rabundfile != "") { 
444                         if ((listfile != "") || (sabundfile != "")) { 
445                                 cout << "When executing a read.otu you must enter ONLY ONE of the following: list, rabund or sabund." << endl; errorFree = false; }
446                 }else if (sabundfile != "") { 
447                         if ((listfile != "") || (rabundfile != "")) { 
448                                 cout << "When executing a read.otu you must enter ONLY ONE of the following: list, rabund or sabund." << endl; errorFree = false; }
449                 }else if ((listfile == "") && (rabundfile == "") && (sabundfile == "") && (sharedfile == "")) {
450                             cout << "When executing a read.otu you must enter one of the following: list, rabund or sabund." << endl; errorFree = false; 
451                 }
452                 
453                 //checks parameters on the read command
454                 if (orderfile != "") {
455                         ableToOpen = openInputFile(orderfile, filehandle);
456                         filehandle.close();
457                         if (ableToOpen == 1) { //unable to open
458                                 errorFree = false;
459                         }
460                 }       
461         }
462         catch(exception& e) {
463                 cout << "Standard Error: " << e.what() << " has occurred in the ErrorCheck class Function validateReadPhil. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
464                 exit(1);
465         }
466         catch(...) {
467                 cout << "An unknown error has occurred in the ErrorCheck class function validateReadPhil. Please contact Pat Schloss at pschloss@microbio.umass.edu." << "\n";
468                 exit(1);
469         }
470 }
471 /*******************************************************/
472
473 /******************************************************/
474
475 void ErrorCheck::clear() {
476         //option definitions should go here...
477         phylipfile              =       "";
478         columnfile              =       "";
479         listfile                =       "";
480         rabundfile              =       "";
481         sabundfile              =       "";
482         namefile                =       "";
483         groupfile               =       ""; 
484         orderfile               =       "";
485         sharedfile              =       "";
486         line                    =       "";
487         label                   =       "";
488         method                  =   "furthest";
489 }
490 /*******************************************************/
491
492 /******************************************************/
493