]> git.donarmstrong.com Git - mothur.git/blob - mothurout.h
changed reading of name file to use buffered reads. note the splitAtWhiteSpace functi...
[mothur.git] / mothurout.h
1 #ifndef MOTHUROUT_H
2 #define MOTHUROUT_H
3
4 /*
5  *  mothurOut.h
6  *  Mothur
7  *
8  *  Created by westcott on 2/25/10.
9  *  Copyright 2010 Schloss Lab. All rights reserved.
10  *
11  */
12
13 #include "mothur.h"
14
15 /***********************************************/
16 struct logger {
17     
18     logger() {}
19     ~logger() {}
20     
21     template< class T >
22     logger& operator <<( const T& o ) {
23         cout << o; return *this;
24     }
25     
26     logger& operator<<(ostream& (*m)(ostream&) ) {
27         cout << m; return *this;
28     }
29     
30 }; 
31 /***********************************************/
32
33 class MothurOut {
34         
35         public:
36                 static MothurOut* getInstance();
37                 void setFileName(string);
38                 
39                 void mothurOut(string); //writes to cout and the logfile
40                 void mothurOutEndLine(); //writes to cout and the logfile
41                 void mothurOut(string, ofstream&); //writes to the ofstream, cout and the logfile
42                 void mothurOutEndLine(ofstream&); //writes to the ofstream, cout and the logfile
43                 void mothurOutJustToLog(string);
44                 void errorOut(exception&, string, string);
45                 void closeLog();
46                 string getDefaultPath() { return defaultPath; }
47                 void setDefaultPath(string);
48                 string getOutputDir() { return outputDir; }
49                 void setOutputDir(string);
50                 
51                 string getReleaseDate() { return releaseDate; }
52                 void setReleaseDate(string r) { releaseDate = r; }
53                 string getVersion() { return version; }
54                 void setVersion(string r) { version = r; }
55         
56                 void addGroup(string g) { Groups.push_back(g); }
57                 void setGroups(vector<string>& g) { sort(g.begin(), g.end()); Groups = g; }
58                 void clearGroups() { Groups.clear(); }
59             int getNumGroups() { return Groups.size(); }
60                 vector<string> getGroups() { sort(Groups.begin(), Groups.end()); return Groups; }
61                 void addAllGroup(string g) { namesOfGroups.push_back(g); }
62                 void setAllGroups(vector<string>& g) { sort(g.begin(), g.end()); namesOfGroups = g; }
63                 void clearAllGroups() { namesOfGroups.clear(); }
64                 int getNumAllGroups() { return namesOfGroups.size(); }
65         
66                 vector<string> getAllGroups() { sort(namesOfGroups.begin(), namesOfGroups.end()); return namesOfGroups; }
67                 vector<string> Treenames;
68                 //map<string, string> names;
69                 vector<string> binLabelsInFile;
70                 vector<string> currentBinLabels;
71                 string saveNextLabel, argv, sharedHeaderMode;
72                 bool printedHeaders, commandInputsConvertError;
73                 
74                 //functions from mothur.h
75                 //file operations
76         bool dirCheck(string&); //completes path, appends appropriate / or \, makes sure dir is writable.
77                 vector<unsigned long long> divideFile(string, int&);
78                 int divideFile(string, int&, vector<string>&);
79                 vector<unsigned long long> setFilePosEachLine(string, int&);
80                 vector<unsigned long long> setFilePosFasta(string, int&);
81                 string sortFile(string, string);
82                 int appendFiles(string, string);
83                 int renameFile(string, string); //oldname, newname
84                 string getFullPathName(string);
85         string findProgramPath(string programName);
86                 string hasPath(string);
87                 string getExtension(string);
88                 string getPathName(string);
89                 string getSimpleName(string);
90                 string getRootName(string);
91                 bool isBlank(string);
92                 int openOutputFile(string, ofstream&);
93                 int openOutputFileAppend(string, ofstream&);
94                 int openInputFile(string, ifstream&);
95                 int openInputFile(string, ifstream&, string); //no error given 
96                 string getline(ifstream&);
97                 string getline(istringstream&);
98                 void gobble(istream&);
99                 void gobble(istringstream&);
100         vector<string> splitWhiteSpace(string& rest, char[], int);
101                 map<string, int> readNames(string);
102                 int readNames(string, map<string, string>&);
103                 int readNames(string, map<string, vector<string> >&);
104                 int readNames(string, vector<seqPriorityNode>&, map<string, string>&);
105                 int mothurRemove(string);
106                 bool mothurConvert(string, int&); //use for converting user inputs. Sets commandInputsConvertError to true if error occurs. Engines check this.
107                 bool mothurConvert(string, float&); //use for converting user inputs. Sets commandInputsConvertError to true if error occurs. Engines check this.
108                 bool mothurConvert(string, double&); //use for converting user inputs. Sets commandInputsConvertError to true if error occurs. Engines check this.
109         
110                 
111                 //searchs and checks
112                 bool checkReleaseVersion(ifstream&, string);
113                 bool anyLabelsToProcess(string, set<string>&, string);
114                 bool inUsersGroups(vector<string>, vector<string>);
115                 bool inUsersGroups(string, vector<string>);
116                 void getNumSeqs(ifstream&, int&);
117                 int getNumSeqs(ifstream&);
118                 int getNumNames(string);
119                 int getNumChar(string, char);
120                 bool isTrue(string);
121                 bool isContainingOnlyDigits(string);
122                 bool isNumeric1(string);
123         
124                 
125                 //string manipulation
126                 void splitAtEquals(string&, string&);
127                 void splitAtComma(string&, string&);    
128                 void splitAtComma(string&, vector<string>&);
129                 void splitAtDash(string&, set<int>&);
130                 void splitAtDash(string&, set<string>&);
131                 void splitAtDash(string&, vector<string>&);
132                 void splitAtChar(string&, vector<string>&, char);
133         void splitAtChar(string&, string&, char);
134                 int removeConfidences(string&);
135         string makeList(vector<string>&);
136                 
137                 //math operation
138                 int factorial(int num);
139                 vector<vector<double> > binomial(int);
140                 float ceilDist(float, int);
141                 float roundDist(float, int);
142                 unsigned int fromBase36(string);
143                 int getRandomIndex(int); //highest
144
145                 int control_pressed;
146                 bool executing, runParse, jumble, gui, mothurCalling, debug;
147                 
148                 //current files - if you add a new type you must edit optionParser->getParameters, get.current command and mothurOut->printCurrentFiles/clearCurrentFiles.
149                 string getPhylipFile()          { return phylipfile;            }
150                 string getColumnFile()          { return columnfile;            }
151                 string getListFile()            { return listfile;                      }
152                 string getRabundFile()          { return rabundfile;            }
153                 string getSabundFile()          { return sabundfile;            }
154                 string getNameFile()            { return namefile;                      }       
155                 string getGroupFile()           { return groupfile;                     }       
156                 string getOrderFile()           { return orderfile;                     }
157                 string getOrderGroupFile()      { return ordergroupfile;        }
158                 string getTreeFile()            { return treefile;                      }
159                 string getSharedFile()          { return sharedfile;            }
160                 string getRelAbundFile()        { return relabundfile;          }
161                 string getDesignFile()          { return designfile;            }
162                 string getFastaFile()           { return fastafile;                     }
163                 string getSFFFile()                     { return sfffile;                       }
164                 string getQualFile()            { return qualfile;                      }
165                 string getOligosFile()          { return oligosfile;            }
166                 string getAccnosFile()          { return accnosfile;            }
167                 string getTaxonomyFile()        { return taxonomyfile;          }
168                 string getFlowFile()            { return flowfile;                      }
169         string getBiomFile()            { return biomfile;                      }
170                 string getProcessors()          { return processors;            }
171                 
172                 void setListFile(string f)                      { listfile = getFullPathName(f);                        }
173                 void setTreeFile(string f)                      { treefile = getFullPathName(f);                        }
174                 void setGroupFile(string f)                     { groupfile = getFullPathName(f);                       }               
175                 void setPhylipFile(string f)            { phylipfile = getFullPathName(f);                      }
176                 void setColumnFile(string f)            { columnfile = getFullPathName(f);                      }
177                 void setNameFile(string f)                      { namefile = getFullPathName(f);                        }       
178                 void setRabundFile(string f)            { rabundfile = getFullPathName(f);                      }
179                 void setSabundFile(string f)            { sabundfile = getFullPathName(f);                      }
180                 void setSharedFile(string f)            { sharedfile = getFullPathName(f);                      }
181                 void setRelAbundFile(string f)          { relabundfile = getFullPathName(f);            }
182                 void setOrderFile(string f)                     { orderfile = getFullPathName(f);                       }
183                 void setOrderGroupFile(string f)        { ordergroupfile = getFullPathName(f);          }
184                 void setDesignFile(string f)            { designfile = getFullPathName(f);                      }
185                 void setFastaFile(string f)                     { fastafile = getFullPathName(f);                       }
186                 void setSFFFile(string f)                       { sfffile = getFullPathName(f);                         }
187                 void setQualFile(string f)                      { qualfile = getFullPathName(f);                        }
188                 void setOligosFile(string f)            { oligosfile = getFullPathName(f);                      }
189                 void setAccnosFile(string f)            { accnosfile = getFullPathName(f);                      }
190                 void setTaxonomyFile(string f)          { taxonomyfile = getFullPathName(f);            }
191                 void setFlowFile(string f)                      { flowfile = getFullPathName(f);                        }
192         void setBiomFile(string f)                      { biomfile = getFullPathName(f);                        }
193                 void setProcessors(string p)            { processors = p;                                                       }
194                 
195                 void printCurrentFiles();
196                 bool hasCurrentFiles();
197                 void clearCurrentFiles();
198                 
199         private:
200                 static MothurOut* _uniqueInstance;
201                 MothurOut( const MothurOut& ); // Disable copy constructor
202                 void operator=( const MothurOut& ); // Disable assignment operator
203                 MothurOut() { 
204                         control_pressed = false; defaultPath=""; 
205                         phylipfile = "";
206                         columnfile = "";
207                         listfile = "";
208                         rabundfile = "";
209                         sabundfile = "";
210                         namefile = "";
211                         groupfile = "";
212                         designfile = "";
213                         orderfile = "";
214                         treefile = "";
215                         sharedfile = "";
216                         ordergroupfile = "";
217                         relabundfile = "";
218                         fastafile = "";
219                         qualfile = "";
220                         sfffile = "";
221                         oligosfile = "";
222                         accnosfile = "";
223                         taxonomyfile = "";
224                         processors = "1";
225                         flowfile = "";
226             biomfile = "";
227                         gui = false;
228                         printedHeaders = false;
229                         commandInputsConvertError = false;
230             mothurCalling = false;
231             debug = false;
232                         sharedHeaderMode = "";
233                 }
234                 ~MothurOut();
235
236                 string logFileName;
237                 string defaultPath, outputDir;
238                 string releaseDate, version;
239         
240                 string accnosfile, phylipfile, columnfile, listfile, rabundfile, sabundfile, namefile, groupfile, designfile, taxonomyfile, biomfile;
241                 string orderfile, treefile, sharedfile, ordergroupfile, relabundfile, fastafile, qualfile, sfffile, oligosfile, processors, flowfile;
242
243                 vector<string> Groups;
244                 vector<string> namesOfGroups;
245                 ofstream out;
246                 
247                 int mem_usage(double&, double&);
248
249 };
250 /***********************************************/
251
252 #endif
253