]> git.donarmstrong.com Git - mothur.git/blob - chimeraslayercommand.cpp
8647e7510088014e48153f33a81fec8fd62ee64e
[mothur.git] / chimeraslayercommand.cpp
1 /*
2  *  chimeraslayercommand.cpp
3  *  Mothur
4  *
5  *  Created by westcott on 3/31/10.
6  *  Copyright 2010 Schloss Lab. All rights reserved.
7  *
8  */
9
10 #include "chimeraslayercommand.h"
11 #include "deconvolutecommand.h"
12 #include "referencedb.h"
13 #include "sequenceparser.h"
14
15 //**********************************************************************************************************************
16 vector<string> ChimeraSlayerCommand::setParameters(){   
17         try {
18                 CommandParameter ptemplate("reference", "InputTypes", "", "", "none", "none", "none",false,true); parameters.push_back(ptemplate);
19                 CommandParameter pfasta("fasta", "InputTypes", "", "", "none", "none", "none",false,true); parameters.push_back(pfasta);
20                 CommandParameter pname("name", "InputTypes", "", "", "none", "none", "none",false,false); parameters.push_back(pname);
21                 CommandParameter pgroup("group", "InputTypes", "", "", "none", "none", "none",false,false); parameters.push_back(pgroup);
22                 CommandParameter pwindow("window", "Number", "", "50", "", "", "",false,false); parameters.push_back(pwindow);
23                 CommandParameter pksize("ksize", "Number", "", "7", "", "", "",false,false); parameters.push_back(pksize);
24                 CommandParameter pmatch("match", "Number", "", "5.0", "", "", "",false,false); parameters.push_back(pmatch);
25                 CommandParameter pmismatch("mismatch", "Number", "", "-4.0", "", "", "",false,false); parameters.push_back(pmismatch);
26                 CommandParameter pminsim("minsim", "Number", "", "90", "", "", "",false,false); parameters.push_back(pminsim);
27                 CommandParameter pmincov("mincov", "Number", "", "70", "", "", "",false,false); parameters.push_back(pmincov);
28                 CommandParameter pminsnp("minsnp", "Number", "", "10", "", "", "",false,false); parameters.push_back(pminsnp);
29                 CommandParameter pminbs("minbs", "Number", "", "90", "", "", "",false,false); parameters.push_back(pminbs);
30                 CommandParameter psearch("search", "Multiple", "kmer-blast", "blast", "", "", "",false,false); parameters.push_back(psearch);
31                 CommandParameter pprocessors("processors", "Number", "", "1", "", "", "",false,false); parameters.push_back(pprocessors);
32                 CommandParameter prealign("realign", "Boolean", "", "T", "", "", "",false,false); parameters.push_back(prealign);
33                 CommandParameter ptrim("trim", "Boolean", "", "F", "", "", "",false,false); parameters.push_back(ptrim);
34                 CommandParameter psplit("split", "Boolean", "", "F", "", "", "",false,false); parameters.push_back(psplit);
35                 CommandParameter pnumwanted("numwanted", "Number", "", "15", "", "", "",false,false); parameters.push_back(pnumwanted);
36                 CommandParameter piters("iters", "Number", "", "1000", "", "", "",false,false); parameters.push_back(piters);
37                 CommandParameter pdivergence("divergence", "Number", "", "1.007", "", "", "",false,false); parameters.push_back(pdivergence);
38                 CommandParameter pparents("parents", "Number", "", "3", "", "", "",false,false); parameters.push_back(pparents);
39                 CommandParameter pincrement("increment", "Number", "", "5", "", "", "",false,false); parameters.push_back(pincrement);
40                 CommandParameter pblastlocation("blastlocation", "String", "", "", "", "", "",false,false); parameters.push_back(pblastlocation);
41                 CommandParameter pinputdir("inputdir", "String", "", "", "", "", "",false,false); parameters.push_back(pinputdir);
42                 CommandParameter poutputdir("outputdir", "String", "", "", "", "", "",false,false); parameters.push_back(poutputdir);
43                 CommandParameter psave("save", "Boolean", "", "F", "", "", "",false,false); parameters.push_back(psave);
44
45                 vector<string> myArray;
46                 for (int i = 0; i < parameters.size(); i++) {   myArray.push_back(parameters[i].name);          }
47                 return myArray;
48         }
49         catch(exception& e) {
50                 m->errorOut(e, "ChimeraSlayerCommand", "setParameters");
51                 exit(1);
52         }
53 }
54 //**********************************************************************************************************************
55 string ChimeraSlayerCommand::getHelpString(){   
56         try {
57                 string helpString = "";
58                 helpString += "The chimera.slayer command reads a fastafile and referencefile and outputs potentially chimeric sequences.\n";
59                 helpString += "This command was modeled after the chimeraSlayer written by the Broad Institute.\n";
60                 helpString += "The chimera.slayer command parameters are fasta, name, template, processors, trim, ksize, window, match, mismatch, divergence. minsim, mincov, minbs, minsnp, parents, search, iters, increment, numwanted, blastlocation and realign.\n";
61                 helpString += "The fasta parameter allows you to enter the fasta file containing your potentially chimeric sequences, and is required, unless you have a valid current fasta file. \n";
62                 helpString += "The name parameter allows you to provide a name file, if you are using reference=self. \n";
63                 helpString += "The group parameter allows you to provide a group file. The group file can be used with a namesfile and reference=self. When checking sequences, only sequences from the same group as the query sequence will be used as the reference. \n";
64                 helpString += "You may enter multiple fasta files by separating their names with dashes. ie. fasta=abrecovery.fasta-amazon.fasta \n";
65                 helpString += "The reference parameter allows you to enter a reference file containing known non-chimeric sequences, and is required. You may also set template=self, in this case the abundant sequences will be used as potential parents. \n";
66                 helpString += "The processors parameter allows you to specify how many processors you would like to use.  The default is 1. \n";
67 #ifdef USE_MPI
68                 helpString += "When using MPI, the processors parameter is set to the number of MPI processes running. \n";
69 #endif
70                 helpString += "The trim parameter allows you to output a new fasta file containing your sequences with the chimeric ones trimmed to include only their longest piece, default=F. \n";
71                 helpString += "The split parameter allows you to check both pieces of non-chimeric sequence for chimeras, thus looking for trimeras and quadmeras. default=F. \n";
72                 helpString += "The window parameter allows you to specify the window size for searching for chimeras, default=50. \n";
73                 helpString += "The increment parameter allows you to specify how far you move each window while finding chimeric sequences, default=5.\n";
74                 helpString += "The numwanted parameter allows you to specify how many sequences you would each query sequence compared with, default=15.\n";
75                 helpString += "The ksize parameter allows you to input kmersize, default is 7, used if search is kmer. \n";
76                 helpString += "The match parameter allows you to reward matched bases in blast search, default is 5. \n";
77                 helpString += "The parents parameter allows you to select the number of potential parents to investigate from the numwanted best matches after rating them, default is 3. \n";
78                 helpString += "The mismatch parameter allows you to penalize mismatched bases in blast search, default is -4. \n";
79                 helpString += "The divergence parameter allows you to set a cutoff for chimera determination, default is 1.007. \n";
80                 helpString += "The iters parameter allows you to specify the number of bootstrap iters to do with the chimeraslayer method, default=1000.\n";
81                 helpString += "The minsim parameter allows you to specify a minimum similarity with the parent fragments, default=90. \n";
82                 helpString += "The mincov parameter allows you to specify minimum coverage by closest matches found in template. Default is 70, meaning 70%. \n";
83                 helpString += "The minbs parameter allows you to specify minimum bootstrap support for calling a sequence chimeric. Default is 90, meaning 90%. \n";
84                 helpString += "The minsnp parameter allows you to specify percent of SNPs to sample on each side of breakpoint for computing bootstrap support (default: 10) \n";
85                 helpString += "The search parameter allows you to specify search method for finding the closest parent. Choices are blast, and kmer, default blast. \n";
86                 helpString += "The realign parameter allows you to realign the query to the potential parents. Choices are true or false, default true.  \n";
87                 helpString += "The blastlocation parameter allows you to specify the location of your blast executable. By default mothur will look in ./blast/bin relative to mothur's executable.  \n";
88                 helpString += "If the save parameter is set to true the reference sequences will be saved in memory, to clear them later you can use the clear.memory command. Default=f.";
89                 helpString += "The chimera.slayer command should be in the following format: \n";
90                 helpString += "chimera.slayer(fasta=yourFastaFile, reference=yourTemplate, search=yourSearch) \n";
91                 helpString += "Example: chimera.slayer(fasta=AD.align, reference=core_set_aligned.imputed.fasta, search=kmer) \n";
92                 helpString += "Note: No spaces between parameter labels (i.e. fasta), '=' and parameters (i.e.yourFastaFile).\n";       
93                 return helpString;
94         }
95         catch(exception& e) {
96                 m->errorOut(e, "ChimeraSlayerCommand", "getHelpString");
97                 exit(1);
98         }
99 }
100 //**********************************************************************************************************************
101 ChimeraSlayerCommand::ChimeraSlayerCommand(){   
102         try {
103                 abort = true; calledHelp = true;
104                 setParameters();
105                 vector<string> tempOutNames;
106                 outputTypes["chimera"] = tempOutNames;
107                 outputTypes["accnos"] = tempOutNames;
108                 outputTypes["fasta"] = tempOutNames;
109         }
110         catch(exception& e) {
111                 m->errorOut(e, "ChimeraSlayerCommand", "ChimeraSlayerCommand");
112                 exit(1);
113         }
114 }
115 //***************************************************************************************************************
116 ChimeraSlayerCommand::ChimeraSlayerCommand(string option)  {
117         try {
118                 abort = false; calledHelp = false;   
119                 ReferenceDB* rdb = ReferenceDB::getInstance();
120                 
121                 //allow user to run help
122                 if(option == "help") { help(); abort = true; calledHelp = true; }
123                 else if(option == "citation") { citation(); abort = true; calledHelp = true;}
124                 
125                 else {
126                         vector<string> myArray = setParameters();
127                         
128                         OptionParser parser(option);
129                         map<string,string> parameters = parser.getParameters();
130                         
131                         ValidParameters validParameter("chimera.slayer");
132                         map<string,string>::iterator it;
133                         
134                         //check to make sure all parameters are valid for command
135                         for (it = parameters.begin(); it != parameters.end(); it++) { 
136                                 if (validParameter.isValidParameter(it->first, myArray, it->second) != true) {  abort = true;  }
137                         }
138                         
139                         vector<string> tempOutNames;
140                         outputTypes["chimera"] = tempOutNames;
141                         outputTypes["accnos"] = tempOutNames;
142                         outputTypes["fasta"] = tempOutNames;
143                 
144                         //if the user changes the input directory command factory will send this info to us in the output parameter 
145                         string inputDir = validParameter.validFile(parameters, "inputdir", false);              
146                         if (inputDir == "not found"){   inputDir = "";          }
147                                                 
148                         //check for required parameters
149                         fastafile = validParameter.validFile(parameters, "fasta", false);
150                         if (fastafile == "not found") {                                 
151                                 //if there is a current fasta file, use it
152                                 string filename = m->getFastaFile(); 
153                                 if (filename != "") { fastaFileNames.push_back(filename); m->mothurOut("Using " + filename + " as input file for the fasta parameter."); m->mothurOutEndLine(); }
154                                 else {  m->mothurOut("You have no current fastafile and the fasta parameter is required."); m->mothurOutEndLine(); abort = true; }
155                         }else { 
156                                 m->splitAtDash(fastafile, fastaFileNames);
157                                 
158                                 //go through files and make sure they are good, if not, then disregard them
159                                 for (int i = 0; i < fastaFileNames.size(); i++) {
160                                         
161                                         bool ignore = false;
162                                         if (fastaFileNames[i] == "current") { 
163                                                 fastaFileNames[i] = m->getFastaFile(); 
164                                                 if (fastaFileNames[i] != "") {  m->mothurOut("Using " + fastaFileNames[i] + " as input file for the fasta parameter where you had given current."); m->mothurOutEndLine(); }
165                                                 else {  
166                                                         m->mothurOut("You have no current fastafile, ignoring current."); m->mothurOutEndLine(); ignore=true; 
167                                                         //erase from file list
168                                                         fastaFileNames.erase(fastaFileNames.begin()+i);
169                                                         i--;
170                                                 }
171                                         }
172                                         
173                                         if (!ignore) {
174                                                 
175                                                 if (inputDir != "") {
176                                                         string path = m->hasPath(fastaFileNames[i]);
177                                                         //if the user has not given a path then, add inputdir. else leave path alone.
178                                                         if (path == "") {       fastaFileNames[i] = inputDir + fastaFileNames[i];               }
179                                                 }
180                 
181                                                 int ableToOpen;
182                                                 ifstream in;
183                                                 
184                                                 ableToOpen = m->openInputFile(fastaFileNames[i], in, "noerror");
185                                         
186                                                 //if you can't open it, try default location
187                                                 if (ableToOpen == 1) {
188                                                         if (m->getDefaultPath() != "") { //default path is set
189                                                                 string tryPath = m->getDefaultPath() + m->getSimpleName(fastaFileNames[i]);
190                                                                 m->mothurOut("Unable to open " + fastaFileNames[i] + ". Trying default " + tryPath); m->mothurOutEndLine();
191                                                                 ifstream in2;
192                                                                 ableToOpen = m->openInputFile(tryPath, in2, "noerror");
193                                                                 in2.close();
194                                                                 fastaFileNames[i] = tryPath;
195                                                         }
196                                                 }
197                                                 
198                                                 if (ableToOpen == 1) {
199                                                         if (m->getOutputDir() != "") { //default path is set
200                                                                 string tryPath = m->getOutputDir() + m->getSimpleName(fastaFileNames[i]);
201                                                                 m->mothurOut("Unable to open " + fastaFileNames[i] + ". Trying output directory " + tryPath); m->mothurOutEndLine();
202                                                                 ifstream in2;
203                                                                 ableToOpen = m->openInputFile(tryPath, in2, "noerror");
204                                                                 in2.close();
205                                                                 fastaFileNames[i] = tryPath;
206                                                         }
207                                                 }
208                                                 
209                                                 in.close();
210                                                 
211                                                 if (ableToOpen == 1) { 
212                                                         m->mothurOut("Unable to open " + fastaFileNames[i] + ". It will be disregarded."); m->mothurOutEndLine(); 
213                                                         //erase from file list
214                                                         fastaFileNames.erase(fastaFileNames.begin()+i);
215                                                         i--;
216                                                 }else {
217                                                         m->setFastaFile(fastaFileNames[i]);
218                                                 }
219                                         }
220                                 }
221                                 
222                                 //make sure there is at least one valid file left
223                                 if (fastaFileNames.size() == 0) { m->mothurOut("[ERROR]: no valid files."); m->mothurOutEndLine(); abort = true; }
224                         }
225                         
226                         
227                         //check for required parameters
228                         bool hasName = true;
229                         namefile = validParameter.validFile(parameters, "name", false);
230                         if (namefile == "not found") { namefile = "";  hasName = false; }
231                         else { 
232                                 m->splitAtDash(namefile, nameFileNames);
233                                 
234                                 //go through files and make sure they are good, if not, then disregard them
235                                 for (int i = 0; i < nameFileNames.size(); i++) {
236                                         
237                                         bool ignore = false;
238                                         if (nameFileNames[i] == "current") { 
239                                                 nameFileNames[i] = m->getNameFile(); 
240                                                 if (nameFileNames[i] != "") {  m->mothurOut("Using " + nameFileNames[i] + " as input file for the name parameter where you had given current."); m->mothurOutEndLine(); }
241                                                 else {  
242                                                         m->mothurOut("You have no current namefile, ignoring current."); m->mothurOutEndLine(); ignore=true; 
243                                                         //erase from file list
244                                                         nameFileNames.erase(nameFileNames.begin()+i);
245                                                         i--;
246                                                 }
247                                         }
248                                         
249                                         if (!ignore) {
250                                                 
251                                                 if (inputDir != "") {
252                                                         string path = m->hasPath(nameFileNames[i]);
253                                                         //if the user has not given a path then, add inputdir. else leave path alone.
254                                                         if (path == "") {       nameFileNames[i] = inputDir + nameFileNames[i];         }
255                                                 }
256                                                 
257                                                 int ableToOpen;
258                                                 ifstream in;
259                                                 
260                                                 ableToOpen = m->openInputFile(nameFileNames[i], in, "noerror");
261                                                 
262                                                 //if you can't open it, try default location
263                                                 if (ableToOpen == 1) {
264                                                         if (m->getDefaultPath() != "") { //default path is set
265                                                                 string tryPath = m->getDefaultPath() + m->getSimpleName(nameFileNames[i]);
266                                                                 m->mothurOut("Unable to open " + nameFileNames[i] + ". Trying default " + tryPath); m->mothurOutEndLine();
267                                                                 ifstream in2;
268                                                                 ableToOpen = m->openInputFile(tryPath, in2, "noerror");
269                                                                 in2.close();
270                                                                 nameFileNames[i] = tryPath;
271                                                         }
272                                                 }
273                                                 
274                                                 if (ableToOpen == 1) {
275                                                         if (m->getOutputDir() != "") { //default path is set
276                                                                 string tryPath = m->getOutputDir() + m->getSimpleName(nameFileNames[i]);
277                                                                 m->mothurOut("Unable to open " + nameFileNames[i] + ". Trying output directory " + tryPath); m->mothurOutEndLine();
278                                                                 ifstream in2;
279                                                                 ableToOpen = m->openInputFile(tryPath, in2, "noerror");
280                                                                 in2.close();
281                                                                 nameFileNames[i] = tryPath;
282                                                         }
283                                                 }
284                                                 
285                                                 in.close();
286                                                 
287                                                 if (ableToOpen == 1) { 
288                                                         m->mothurOut("Unable to open " + nameFileNames[i] + ". It will be disregarded."); m->mothurOutEndLine(); 
289                                                         //erase from file list
290                                                         nameFileNames.erase(nameFileNames.begin()+i);
291                                                         i--;
292                                                 }else {
293                                                         m->setNameFile(nameFileNames[i]);
294                                                 }
295                                         }
296                                 }
297                                 
298                                 //make sure there is at least one valid file left
299                                 if (nameFileNames.size() == 0) { m->mothurOut("[ERROR]: no valid name files."); m->mothurOutEndLine(); abort = true; }
300                         }
301                         
302                         if (hasName && (nameFileNames.size() != fastaFileNames.size())) { m->mothurOut("[ERROR]: The number of namefiles does not match the number of fastafiles, please correct."); m->mothurOutEndLine(); abort=true; }
303                         
304                         bool hasGroup = true;
305                         groupfile = validParameter.validFile(parameters, "group", false);
306                         if (groupfile == "not found") { groupfile = "";  hasGroup = false; }
307                         else { 
308                                 m->splitAtDash(groupfile, groupFileNames);
309                                 
310                                 //go through files and make sure they are good, if not, then disregard them
311                                 for (int i = 0; i < groupFileNames.size(); i++) {
312                                         
313                                         bool ignore = false;
314                                         if (groupFileNames[i] == "current") { 
315                                                 groupFileNames[i] = m->getGroupFile(); 
316                                                 if (groupFileNames[i] != "") {  m->mothurOut("Using " + groupFileNames[i] + " as input file for the group parameter where you had given current."); m->mothurOutEndLine(); }
317                                                 else {  
318                                                         m->mothurOut("You have no current namefile, ignoring current."); m->mothurOutEndLine(); ignore=true; 
319                                                         //erase from file list
320                                                         groupFileNames.erase(groupFileNames.begin()+i);
321                                                         i--;
322                                                 }
323                                         }
324                                         
325                                         if (!ignore) {
326                                                 
327                                                 if (inputDir != "") {
328                                                         string path = m->hasPath(groupFileNames[i]);
329                                                         //if the user has not given a path then, add inputdir. else leave path alone.
330                                                         if (path == "") {       groupFileNames[i] = inputDir + groupFileNames[i];               }
331                                                 }
332                                                 
333                                                 int ableToOpen;
334                                                 ifstream in;
335                                                 
336                                                 ableToOpen = m->openInputFile(groupFileNames[i], in, "noerror");
337                                                 
338                                                 //if you can't open it, try default location
339                                                 if (ableToOpen == 1) {
340                                                         if (m->getDefaultPath() != "") { //default path is set
341                                                                 string tryPath = m->getDefaultPath() + m->getSimpleName(groupFileNames[i]);
342                                                                 m->mothurOut("Unable to open " + groupFileNames[i] + ". Trying default " + tryPath); m->mothurOutEndLine();
343                                                                 ifstream in2;
344                                                                 ableToOpen = m->openInputFile(tryPath, in2, "noerror");
345                                                                 in2.close();
346                                                                 groupFileNames[i] = tryPath;
347                                                         }
348                                                 }
349                                                 
350                                                 if (ableToOpen == 1) {
351                                                         if (m->getOutputDir() != "") { //default path is set
352                                                                 string tryPath = m->getOutputDir() + m->getSimpleName(groupFileNames[i]);
353                                                                 m->mothurOut("Unable to open " + groupFileNames[i] + ". Trying output directory " + tryPath); m->mothurOutEndLine();
354                                                                 ifstream in2;
355                                                                 ableToOpen = m->openInputFile(tryPath, in2, "noerror");
356                                                                 in2.close();
357                                                                 groupFileNames[i] = tryPath;
358                                                         }
359                                                 }
360                                                 
361                                                 in.close();
362                                                 
363                                                 if (ableToOpen == 1) { 
364                                                         m->mothurOut("Unable to open " + groupFileNames[i] + ". It will be disregarded."); m->mothurOutEndLine(); 
365                                                         //erase from file list
366                                                         groupFileNames.erase(groupFileNames.begin()+i);
367                                                         i--;
368                                                 }else {
369                                                         m->setGroupFile(groupFileNames[i]);
370                                                 }
371                                         }
372                                 }
373                                 
374                                 //make sure there is at least one valid file left
375                                 if (groupFileNames.size() == 0) { m->mothurOut("[ERROR]: no valid group files."); m->mothurOutEndLine(); abort = true; }
376                         }
377                         
378                         if (hasGroup && (groupFileNames.size() != fastaFileNames.size())) { m->mothurOut("[ERROR]: The number of groupfiles does not match the number of fastafiles, please correct."); m->mothurOutEndLine(); abort=true; }
379                         
380                         
381                         //if the user changes the output directory command factory will send this info to us in the output parameter 
382                         outputDir = validParameter.validFile(parameters, "outputdir", false);           if (outputDir == "not found"){  outputDir = ""; }
383                         
384                         string temp = validParameter.validFile(parameters, "processors", false);        if (temp == "not found"){       temp = m->getProcessors();      }
385                         m->setProcessors(temp);
386                         m->mothurConvert(temp, processors);
387                         
388                         temp = validParameter.validFile(parameters, "save", false);                     if (temp == "not found"){       temp = "f";                             }
389                         save = m->isTrue(temp); 
390                         rdb->save = save; 
391                         if (save) { //clear out old references
392                                 rdb->clearMemory();     
393                         }
394                         
395                         string path;
396                         it = parameters.find("reference");
397                         //user has given a template file
398                         if(it != parameters.end()){ 
399                                 if (it->second == "self") { 
400                                         templatefile = "self"; 
401                                         if (save) {
402                                                 m->mothurOut("[WARNING]: You can't save reference=self, ignoring save."); 
403                                                 m->mothurOutEndLine();
404                                                 save = false;
405                                         }
406                                 }
407                                 else {
408                                         path = m->hasPath(it->second);
409                                         //if the user has not given a path then, add inputdir. else leave path alone.
410                                         if (path == "") {       parameters["reference"] = inputDir + it->second;                }
411                                         
412                                         templatefile = validParameter.validFile(parameters, "reference", true);
413                                         if (templatefile == "not open") { abort = true; }
414                                         else if (templatefile == "not found") { //check for saved reference sequences
415                                                 if (rdb->referenceSeqs.size() != 0) {
416                                                         templatefile = "saved";
417                                                 }else {
418                                                         m->mothurOut("[ERROR]: You don't have any saved reference sequences and the reference parameter is a required."); 
419                                                         m->mothurOutEndLine();
420                                                         abort = true; 
421                                                 }
422                                         }else { if (save) {     rdb->setSavedReference(templatefile);   }       }       
423                                 }
424                         }else if (hasName) {  templatefile = "self"; 
425                                 if (save) {
426                                         m->mothurOut("[WARNING]: You can't save reference=self, ignoring save."); 
427                                         m->mothurOutEndLine();
428                                         save = false;
429                                 }
430                         }
431                         else { 
432                                 if (rdb->referenceSeqs.size() != 0) {
433                                         templatefile = "saved";
434                                 }else {
435                                         m->mothurOut("[ERROR]: You don't have any saved reference sequences and the reference parameter is a required."); 
436                                         m->mothurOutEndLine();
437                                         templatefile = ""; abort = true; 
438                                 } 
439                         }
440                         
441                         
442                         
443                         temp = validParameter.validFile(parameters, "ksize", false);                    if (temp == "not found") { temp = "7"; }
444                         m->mothurConvert(temp, ksize);
445                                                 
446                         temp = validParameter.validFile(parameters, "window", false);                   if (temp == "not found") { temp = "50"; }                       
447                         m->mothurConvert(temp, window);
448                         
449                         temp = validParameter.validFile(parameters, "match", false);                    if (temp == "not found") { temp = "5"; }
450                         m->mothurConvert(temp, match);
451                         
452                         temp = validParameter.validFile(parameters, "mismatch", false);                 if (temp == "not found") { temp = "-4"; }
453                         m->mothurConvert(temp, mismatch);
454                         
455                         temp = validParameter.validFile(parameters, "divergence", false);               if (temp == "not found") { temp = "1.007"; }
456                         m->mothurConvert(temp, divR);
457                         
458                         temp = validParameter.validFile(parameters, "minsim", false);                   if (temp == "not found") { temp = "90"; }
459                         m->mothurConvert(temp, minSimilarity);
460                         
461                         temp = validParameter.validFile(parameters, "mincov", false);                   if (temp == "not found") { temp = "70"; }
462                         m->mothurConvert(temp, minCoverage);
463                         
464                         temp = validParameter.validFile(parameters, "minbs", false);                    if (temp == "not found") { temp = "90"; }
465                         m->mothurConvert(temp, minBS);
466                         
467                         temp = validParameter.validFile(parameters, "minsnp", false);                   if (temp == "not found") { temp = "10"; }
468                         m->mothurConvert(temp, minSNP);
469
470                         temp = validParameter.validFile(parameters, "parents", false);                  if (temp == "not found") { temp = "3"; }
471                         m->mothurConvert(temp, parents); 
472                         
473                         temp = validParameter.validFile(parameters, "realign", false);                  if (temp == "not found") { temp = "t"; }
474                         realign = m->isTrue(temp); 
475                         
476                         temp = validParameter.validFile(parameters, "trim", false);                             if (temp == "not found") { temp = "f"; }
477                         trim = m->isTrue(temp); 
478                         
479                         temp = validParameter.validFile(parameters, "split", false);                    if (temp == "not found") { temp = "f"; }
480                         trimera = m->isTrue(temp); 
481                         
482                         search = validParameter.validFile(parameters, "search", false);                 if (search == "not found") { search = "blast"; }
483                         
484                         temp = validParameter.validFile(parameters, "iters", false);                    if (temp == "not found") { temp = "1000"; }             
485                         m->mothurConvert(temp, iters); 
486                          
487                         temp = validParameter.validFile(parameters, "increment", false);                if (temp == "not found") { temp = "5"; }
488                         m->mothurConvert(temp, increment);
489                         
490                         temp = validParameter.validFile(parameters, "numwanted", false);                if (temp == "not found") { temp = "15"; }               
491                         m->mothurConvert(temp, numwanted);
492                         
493                         blastlocation = validParameter.validFile(parameters, "blastlocation", false);   
494                         if (blastlocation == "not found") { blastlocation = ""; }
495                         else {
496                                 //add / to name if needed
497                                 string lastChar = blastlocation.substr(blastlocation.length()-1);
498 #if defined (__APPLE__) || (__MACH__) || (linux) || (__linux) || (__linux__) || (__unix__) || (__unix)
499                                 if (lastChar != "/") { blastlocation += "/"; }
500 #else
501                                 if (lastChar != "\\") { blastlocation += "\\"; }        
502 #endif
503                                 blastlocation = m->getFullPathName(blastlocation);
504                                 string formatdbCommand = "";
505 #if defined (__APPLE__) || (__MACH__) || (linux) || (__linux) || (__linux__) || (__unix__) || (__unix)
506                                 formatdbCommand = blastlocation + "formatdb";   
507 #else
508                                 formatdbCommand = blastlocation + "formatdb.exe";
509 #endif
510                                 
511                                 //test to make sure formatdb exists
512                                 ifstream in;
513                                 formatdbCommand = m->getFullPathName(formatdbCommand);
514                                 int ableToOpen = m->openInputFile(formatdbCommand, in, "no error"); in.close();
515                                 if(ableToOpen == 1) {   m->mothurOut("[ERROR]: " + formatdbCommand + " file does not exist. mothur requires formatdb.exe to run chimera.slayer."); m->mothurOutEndLine(); abort = true; }
516                                 
517                                 string blastCommand = "";
518 #if defined (__APPLE__) || (__MACH__) || (linux) || (__linux) || (__linux__) || (__unix__) || (__unix)
519                                 blastCommand = blastlocation + "megablast";     
520 #else
521                                 blastCommand = blastlocation + "megablast.exe";
522 #endif
523                                 //test to make sure formatdb exists
524                                 ifstream in2;
525                                 blastCommand = m->getFullPathName(blastCommand);
526                                 ableToOpen = m->openInputFile(blastCommand, in2, "no error"); in2.close();
527                                 if(ableToOpen == 1) {   m->mothurOut("[ERROR]: " + blastCommand + " file does not exist. mothur requires blastall.exe to run chimera.slayer."); m->mothurOutEndLine(); abort = true; }
528                         }
529
530                         if ((search != "blast") && (search != "kmer")) { m->mothurOut(search + " is not a valid search."); m->mothurOutEndLine(); abort = true;  }
531                         
532                         if (hasName && (templatefile != "self")) { m->mothurOut("You have provided a namefile and the reference parameter is not set to self. I am not sure what reference you are trying to use, aborting."); m->mothurOutEndLine(); abort=true; }
533                         if (hasGroup && (templatefile != "self")) { m->mothurOut("You have provided a group file and the reference parameter is not set to self. I am not sure what reference you are trying to use, aborting."); m->mothurOutEndLine(); abort=true; }
534
535                         //until we resolve the issue 10-18-11
536 #if defined (__APPLE__) || (__MACH__) || (linux) || (__linux) || (__linux__) || (__unix__) || (__unix)
537 #else
538                         //processors=1;
539 #endif
540                 }
541         }
542         catch(exception& e) {
543                 m->errorOut(e, "ChimeraSlayerCommand", "ChimeraSlayerCommand");
544                 exit(1);
545         }
546 }
547 //***************************************************************************************************************
548
549 int ChimeraSlayerCommand::execute(){
550         try{
551                 if (abort == true) { if (calledHelp) { return 0; }  return 2;   }
552
553                 for (int s = 0; s < fastaFileNames.size(); s++) {
554                                 
555                         m->mothurOut("Checking sequences from " + fastaFileNames[s] + " ..." ); m->mothurOutEndLine();
556                 
557                         int start = time(NULL); 
558                         if (outputDir == "") { outputDir = m->hasPath(fastaFileNames[s]);  }//if user entered a file with a path then preserve it                               
559                         string outputFileName = outputDir + m->getRootName(m->getSimpleName(fastaFileNames[s])) + "slayer.chimera";
560                         string accnosFileName = outputDir + m->getRootName(m->getSimpleName(fastaFileNames[s]))  + "slayer.accnos";
561                         string trimFastaFileName = outputDir + m->getRootName(m->getSimpleName(fastaFileNames[s]))  + "slayer.fasta";           
562                         
563                         //clears files
564                         ofstream out, out1, out2;
565                         m->openOutputFile(outputFileName, out); out.close(); 
566                         m->openOutputFile(accnosFileName, out1); out1.close();
567                         if (trim) { m->openOutputFile(trimFastaFileName, out2); out2.close(); }
568                         outputNames.push_back(outputFileName); outputTypes["chimera"].push_back(outputFileName);
569                         outputNames.push_back(accnosFileName); outputTypes["accnos"].push_back(accnosFileName);
570                         if (trim) {  outputNames.push_back(trimFastaFileName); outputTypes["fasta"].push_back(trimFastaFileName); }                     
571                         
572                         //maps a filename to priority map. 
573                         //if no groupfile this is fastafileNames[s] -> prioirity
574                         //if groupfile then this is each groups seqs -> priority
575                         map<string, map<string, int> > fileToPriority; 
576                         map<string, map<string, int> >::iterator itFile;
577                         map<string, string> fileGroup;
578                         fileToPriority[fastaFileNames[s]] = priority; //default
579                         fileGroup[fastaFileNames[s]] = "noGroup";
580                         SequenceParser* parser = NULL;
581                         int totalChimeras = 0;
582                         lines.clear();
583                         
584                         if (templatefile == "self") { setUpForSelfReference(parser, fileGroup, fileToPriority, s); }
585                         
586                         if (m->control_pressed) {  if (parser != NULL) { delete parser; } for (int j = 0; j < outputNames.size(); j++) {        m->mothurRemove(outputNames[j]);        }  return 0;    }
587
588                         if (fileToPriority.size() == 1) { //you running without a groupfile
589                                 itFile = fileToPriority.begin();
590                                 string thisFastaName = itFile->first;
591                                 map<string, int> thisPriority = itFile->second;
592 #ifdef USE_MPI  
593                                 MPIExecute(thisFastaName, outputFileName, accnosFileName, trimFastaFileName, thisPriority);
594 #else
595                                 //break up file
596                                 vector<unsigned long long> positions; 
597 #if defined (__APPLE__) || (__MACH__) || (linux) || (__linux) || (__linux__) || (__unix__) || (__unix)
598                                 positions = m->divideFile(thisFastaName, processors);
599                                 for (int i = 0; i < (positions.size()-1); i++) {        lines.push_back(linePair(positions[i], positions[(i+1)]));      }
600 #else
601                                 if (processors == 1) {  lines.push_back(linePair(0, 1000)); }
602                                 else {
603                                         positions = m->setFilePosFasta(thisFastaName, numSeqs); 
604                     if (positions.size() < processors) { processors = positions.size(); }
605                                         
606                                         //figure out how many sequences you have to process
607                                         int numSeqsPerProcessor = numSeqs / processors;
608                                         for (int i = 0; i < processors; i++) {
609                                                 int startIndex =  i * numSeqsPerProcessor;
610                                                 if(i == (processors - 1)){      numSeqsPerProcessor = numSeqs - i * numSeqsPerProcessor;        }
611                                                 lines.push_back(linePair(positions[startIndex], numSeqsPerProcessor));
612                                         }
613                                 }
614 #endif
615                                 if(processors == 1){ numSeqs = driver(lines[0], outputFileName, thisFastaName, accnosFileName, trimFastaFileName, thisPriority);  }
616                                 else{ numSeqs = createProcesses(outputFileName, thisFastaName, accnosFileName, trimFastaFileName, thisPriority); }
617                                 
618                                 if (m->control_pressed) { if (parser != NULL) { delete parser; }  outputTypes.clear(); if (trim) { m->mothurRemove(trimFastaFileName); } m->mothurRemove(outputFileName); m->mothurRemove(accnosFileName); for (int j = 0; j < outputNames.size(); j++) {       m->mothurRemove(outputNames[j]);        }  return 0; }                          
619 #endif
620                         }else { //you have provided a groupfile
621 #ifdef USE_MPI  
622                                 MPIExecuteGroups(outputFileName, accnosFileName, trimFastaFileName, fileToPriority, fileGroup);
623 #else
624                                 if (processors == 1) { numSeqs = driverGroups(outputFileName, accnosFileName, trimFastaFileName, fileToPriority, fileGroup);    }
625                                 else {  numSeqs = createProcessesGroups(outputFileName, accnosFileName, trimFastaFileName, fileToPriority, fileGroup);          } //destroys fileToPriority
626 #endif
627
628 #ifdef USE_MPI  
629                                 int pid; 
630                                 MPI_Comm_rank(MPI_COMM_WORLD, &pid); //find out who we are
631                                 
632                                 if (pid == 0) {
633 #endif
634                                 
635                                 totalChimeras = deconvoluteResults(parser, outputFileName, accnosFileName, trimFastaFileName);
636 #ifdef USE_MPI  
637                                 }
638                                 MPI_Barrier(MPI_COMM_WORLD); //make everyone wait
639 #endif
640                         }
641         
642                         if (parser != NULL) { delete parser; } 
643                         
644                         m->mothurOutEndLine(); m->mothurOut(toString(totalChimeras) + " chimera found."); m->mothurOutEndLine(); m->mothurOut("It took " + toString(time(NULL) - start) + " secs to check " + toString(numSeqs) + " sequences.");       m->mothurOutEndLine();
645                 }
646                 
647                 //set accnos file as new current accnosfile
648                 string current = "";
649                 itTypes = outputTypes.find("accnos");
650                 if (itTypes != outputTypes.end()) {
651                         if ((itTypes->second).size() != 0) { current = (itTypes->second)[0]; m->setAccnosFile(current); }
652                 }
653                 
654                 if (trim) {
655                         itTypes = outputTypes.find("fasta");
656                         if (itTypes != outputTypes.end()) {
657                                 if ((itTypes->second).size() != 0) { current = (itTypes->second)[0]; m->setFastaFile(current); }
658                         }
659                 }
660                 
661                 m->mothurOutEndLine();
662                 m->mothurOut("Output File Names: "); m->mothurOutEndLine();
663                 for (int i = 0; i < outputNames.size(); i++) {  m->mothurOut(outputNames[i]); m->mothurOutEndLine();    }       
664                 m->mothurOutEndLine();
665
666                 return 0;
667                 
668         }
669         catch(exception& e) {
670                 m->errorOut(e, "ChimeraSlayerCommand", "execute");
671                 exit(1);
672         }
673 }
674 //**********************************************************************************************************************
675 int ChimeraSlayerCommand::MPIExecuteGroups(string outputFileName, string accnosFileName, string trimFastaFileName, map<string, map<string, int> >& fileToPriority, map<string, string>& fileGroup){
676         try {
677 #ifdef USE_MPI  
678                 int pid; 
679                 int tag = 2001;
680                 
681                 MPI_Status status; 
682                 MPI_Comm_rank(MPI_COMM_WORLD, &pid); //find out who we are
683                 MPI_Comm_size(MPI_COMM_WORLD, &processors); 
684         
685                 //put filenames in a vector, then pass each process a starting and ending point in the vector
686                 //all processes already have the fileToPriority and fileGroup, they just need to know which files to process
687                 map<string, map<string, int> >::iterator itFile;
688                 vector<string> filenames;
689                 for(itFile = fileToPriority.begin(); itFile != fileToPriority.end(); itFile++) { filenames.push_back(itFile->first); }
690                 
691                 int numGroupsPerProcessor = filenames.size() / processors;
692                 int startIndex =  pid * numGroupsPerProcessor;
693                 int endIndex = (pid+1) * numGroupsPerProcessor;
694                 if(pid == (processors - 1)){    endIndex = filenames.size();    }
695                 
696                 vector<unsigned long long> MPIPos;
697                 
698                 MPI_File outMPI;
699                 MPI_File outMPIAccnos;
700                 MPI_File outMPIFasta;
701                 
702                 int outMode=MPI_MODE_CREATE|MPI_MODE_WRONLY; 
703                 int inMode=MPI_MODE_RDONLY; 
704                 
705                 char outFilename[1024];
706                 strcpy(outFilename, outputFileName.c_str());
707                 
708                 char outAccnosFilename[1024];
709                 strcpy(outAccnosFilename, accnosFileName.c_str());
710                 
711                 char outFastaFilename[1024];
712                 strcpy(outFastaFilename, trimFastaFileName.c_str());
713                 
714                 MPI_File_open(MPI_COMM_WORLD, outFilename, outMode, MPI_INFO_NULL, &outMPI);
715                 MPI_File_open(MPI_COMM_WORLD, outAccnosFilename, outMode, MPI_INFO_NULL, &outMPIAccnos);
716                 if (trim) { MPI_File_open(MPI_COMM_WORLD, outFastaFilename, outMode, MPI_INFO_NULL, &outMPIFasta); }
717                 
718                 if (m->control_pressed) {   MPI_File_close(&outMPI); if (trim) {  MPI_File_close(&outMPIFasta);  } MPI_File_close(&outMPIAccnos);  return 0;  }
719                 
720                 //print headers
721                 if (pid == 0) { //you are the root process 
722                         m->mothurOutEndLine();
723                         m->mothurOut("Only reporting sequence supported by " + toString(minBS) + "% of bootstrapped results.");
724                         m->mothurOutEndLine();
725                         
726                         string outTemp = "Name\tLeftParent\tRightParent\tDivQLAQRB\tPerIDQLAQRB\tBootStrapA\tDivQLBQRA\tPerIDQLBQRA\tBootStrapB\tFlag\tLeftWindow\tRightWindow\n";
727                         
728                         //print header
729                         int length = outTemp.length();
730                         char* buf2 = new char[length];
731                         memcpy(buf2, outTemp.c_str(), length);
732                         
733                         MPI_File_write_shared(outMPI, buf2, length, MPI_CHAR, &status);
734                         delete buf2;
735                 }
736                 MPI_Barrier(MPI_COMM_WORLD); //make everyone wait
737                 
738                 for (int i = startIndex; i < endIndex; i++) {
739                         
740                         int start = time(NULL);
741                         int num = 0;
742                         string thisFastaName = filenames[i];
743                         map<string, int> thisPriority = fileToPriority[thisFastaName];
744                         
745                         char inFileName[1024];
746                         strcpy(inFileName, thisFastaName.c_str());
747                         MPI_File inMPI;
748                         MPI_File_open(MPI_COMM_SELF, inFileName, inMode, MPI_INFO_NULL, &inMPI);  //comm, filename, mode, info, filepointer
749                         
750                         MPIPos = m->setFilePosFasta(thisFastaName, num); //fills MPIPos, returns numSeqs
751                         
752                         cout << endl << "Checking sequences from group: " << fileGroup[thisFastaName] << "." << endl; 
753                         
754                         driverMPI(0, num, inMPI, outMPI, outMPIAccnos, outMPIFasta, MPIPos, thisFastaName, thisPriority, true);
755                         numSeqs += num;
756                         
757                         MPI_File_close(&inMPI);
758                         m->mothurRemove(thisFastaName);
759                                                 
760                         cout << endl << "It took " << toString(time(NULL) - start) << " secs to check " + toString(num) + " sequences from group " << fileGroup[thisFastaName] << "." << endl;
761                 }
762                 
763                 if (pid == 0) {
764                         for(int i = 1; i < processors; i++) { 
765                                 int temp = 0;
766                                 MPI_Recv(&temp, 1, MPI_INT, i, 2001, MPI_COMM_WORLD, &status);
767                                 numSeqs += temp;
768                         }
769                 }else{ MPI_Send(&numSeqs, 1, MPI_INT, 0, 2001, MPI_COMM_WORLD); }
770                 
771                 MPI_File_close(&outMPI);
772                 MPI_File_close(&outMPIAccnos); 
773                 if (trim) { MPI_File_close(&outMPIFasta); }
774                 
775                 MPI_Barrier(MPI_COMM_WORLD); //make everyone wait
776 #endif
777                 return 0;
778                 
779         }catch(exception& e) {
780                 m->errorOut(e, "ChimeraSlayerCommand", "MPIExecuteGroups");
781                 exit(1);
782         }
783 }               
784 //**********************************************************************************************************************
785 int ChimeraSlayerCommand::MPIExecute(string inputFile, string outputFileName, string accnosFileName, string trimFastaFileName, map<string, int>& priority){
786         try {
787                 
788 #ifdef USE_MPI  
789                 int pid, numSeqsPerProcessor; 
790                 int tag = 2001;
791                 vector<unsigned long long> MPIPos;
792                 
793                 MPI_Status status; 
794                 MPI_Comm_rank(MPI_COMM_WORLD, &pid); //find out who we are
795                 MPI_Comm_size(MPI_COMM_WORLD, &processors); 
796                 
797                 MPI_File inMPI;
798                 MPI_File outMPI;
799                 MPI_File outMPIAccnos;
800                 MPI_File outMPIFasta;
801                 
802                 int outMode=MPI_MODE_CREATE|MPI_MODE_WRONLY; 
803                 int inMode=MPI_MODE_RDONLY; 
804                 
805                 char outFilename[1024];
806                 strcpy(outFilename, outputFileName.c_str());
807                 
808                 char outAccnosFilename[1024];
809                 strcpy(outAccnosFilename, accnosFileName.c_str());
810                 
811                 char outFastaFilename[1024];
812                 strcpy(outFastaFilename, trimFastaFileName.c_str());
813                 
814                 char inFileName[1024];
815                 strcpy(inFileName, inputFile.c_str());
816                 
817                 MPI_File_open(MPI_COMM_WORLD, inFileName, inMode, MPI_INFO_NULL, &inMPI);  //comm, filename, mode, info, filepointer
818                 MPI_File_open(MPI_COMM_WORLD, outFilename, outMode, MPI_INFO_NULL, &outMPI);
819                 MPI_File_open(MPI_COMM_WORLD, outAccnosFilename, outMode, MPI_INFO_NULL, &outMPIAccnos);
820                 if (trim) { MPI_File_open(MPI_COMM_WORLD, outFastaFilename, outMode, MPI_INFO_NULL, &outMPIFasta); }
821                 
822                 if (m->control_pressed) {  MPI_File_close(&inMPI);  MPI_File_close(&outMPI); if (trim) {  MPI_File_close(&outMPIFasta);  } MPI_File_close(&outMPIAccnos);  return 0;  }
823                 
824                 if (pid == 0) { //you are the root process 
825                         m->mothurOutEndLine();
826                         m->mothurOut("Only reporting sequence supported by " + toString(minBS) + "% of bootstrapped results.");
827                         m->mothurOutEndLine();
828                         
829                         string outTemp = "Name\tLeftParent\tRightParent\tDivQLAQRB\tPerIDQLAQRB\tBootStrapA\tDivQLBQRA\tPerIDQLBQRA\tBootStrapB\tFlag\tLeftWindow\tRightWindow\n";
830                         
831                         //print header
832                         int length = outTemp.length();
833                         char* buf2 = new char[length];
834                         memcpy(buf2, outTemp.c_str(), length);
835                         
836                         MPI_File_write_shared(outMPI, buf2, length, MPI_CHAR, &status);
837                         delete buf2;
838                         
839                         MPIPos = m->setFilePosFasta(inputFile, numSeqs); //fills MPIPos, returns numSeqs
840                         
841                         if (templatefile != "self") { //if template=self we can only use 1 processor
842                                 //send file positions to all processes
843                                 for(int i = 1; i < processors; i++) { 
844                                         MPI_Send(&numSeqs, 1, MPI_INT, i, tag, MPI_COMM_WORLD);
845                                         MPI_Send(&MPIPos[0], (numSeqs+1), MPI_LONG, i, tag, MPI_COMM_WORLD);
846                                 }
847                         }
848                         //figure out how many sequences you have to align
849                         numSeqsPerProcessor = numSeqs / processors;
850                         int startIndex =  pid * numSeqsPerProcessor;
851                         if(pid == (processors - 1)){    numSeqsPerProcessor = numSeqs - pid * numSeqsPerProcessor;      }
852                         
853                         if (templatefile == "self") { //if template=self we can only use 1 processor
854                                 startIndex = 0;
855                                 numSeqsPerProcessor = numSeqs;
856                         }
857                         
858                         //do your part
859                         driverMPI(startIndex, numSeqsPerProcessor, inMPI, outMPI, outMPIAccnos, outMPIFasta, MPIPos, inputFile, priority, false);
860                                                 
861                         if (m->control_pressed) {  MPI_File_close(&inMPI);  MPI_File_close(&outMPI); if (trim) { MPI_File_close(&outMPIFasta); }  MPI_File_close(&outMPIAccnos);   return 0;  }
862                         
863                 }else{ //you are a child process
864                         if (templatefile != "self") { //if template=self we can only use 1 processor
865                                 MPI_Recv(&numSeqs, 1, MPI_INT, 0, tag, MPI_COMM_WORLD, &status);
866                                 MPIPos.resize(numSeqs+1);
867                                 MPI_Recv(&MPIPos[0], (numSeqs+1), MPI_LONG, 0, tag, MPI_COMM_WORLD, &status);
868                                 
869                                 //figure out how many sequences you have to align
870                                 numSeqsPerProcessor = numSeqs / processors;
871                                 int startIndex =  pid * numSeqsPerProcessor;
872                                 if(pid == (processors - 1)){    numSeqsPerProcessor = numSeqs - pid * numSeqsPerProcessor;      }
873                                 
874                                 //do your part
875                                 driverMPI(startIndex, numSeqsPerProcessor, inMPI, outMPI, outMPIAccnos, outMPIFasta, MPIPos, inputFile, priority, false);
876                                 
877                                 if (m->control_pressed) {  MPI_File_close(&inMPI);  MPI_File_close(&outMPI); if (trim) { MPI_File_close(&outMPIFasta); }  MPI_File_close(&outMPIAccnos);  return 0;  }
878                                 
879                         }
880                 }
881                 
882                 //close files 
883                 MPI_File_close(&inMPI);
884                 MPI_File_close(&outMPI);
885                 MPI_File_close(&outMPIAccnos); 
886                 if (trim) { MPI_File_close(&outMPIFasta); }
887                 MPI_Barrier(MPI_COMM_WORLD); //make everyone wait - just in case
888                 
889                 
890 #endif          
891                 return numSeqs;
892         }
893         catch(exception& e) {
894                 m->errorOut(e, "ChimeraSlayerCommand", "MPIExecute");
895                 exit(1);
896         }
897 }
898 //**********************************************************************************************************************
899 int ChimeraSlayerCommand::deconvoluteResults(SequenceParser* parser, string outputFileName, string accnosFileName, string trimFileName){
900         try {
901                 map<string, string> uniqueNames = parser->getAllSeqsMap();
902                 map<string, string>::iterator itUnique;
903                 int total = 0;
904         
905         if (trimera) { //add in more potential uniqueNames
906             map<string, string> newUniqueNames = uniqueNames;
907             for (map<string, string>::iterator it = uniqueNames.begin(); it != uniqueNames.end(); it++) {
908                 newUniqueNames[(it->first)+"_LEFT"] = (it->first)+"_LEFT";
909                 newUniqueNames[(it->first)+"_RIGHT"] = (it->first)+"_RIGHT";
910             }
911             uniqueNames = newUniqueNames;
912             newUniqueNames.clear();
913         }
914                 
915                 //edit accnos file
916                 ifstream in2; 
917                 m->openInputFile(accnosFileName, in2, "no error");
918                 
919                 ofstream out2;
920                 m->openOutputFile(accnosFileName+".temp", out2);
921                 
922                 string name; name = "";
923                 set<string> chimerasInFile;
924                 set<string>::iterator itChimeras;
925                 
926                 while (!in2.eof()) {
927                         if (m->control_pressed) { in2.close(); out2.close(); m->mothurRemove(outputFileName); m->mothurRemove((accnosFileName+".temp")); return 0; }
928                         
929                         in2 >> name; m->gobble(in2);
930                         
931                         //find unique name
932                         itUnique = uniqueNames.find(name);
933                         
934                         if (itUnique == uniqueNames.end()) { m->mothurOut("[ERROR]: trouble parsing accnos results. Cannot find "+ name + "."); m->mothurOutEndLine(); m->control_pressed = true; }
935                         else {
936                                 itChimeras = chimerasInFile.find((itUnique->second));
937                                 
938                                 if (itChimeras == chimerasInFile.end()) {
939                                         out2 << itUnique->second << endl;
940                                         chimerasInFile.insert((itUnique->second));
941                                         total++;
942                                 }
943                         }
944                 }
945                 in2.close();
946                 out2.close();
947                 
948                 m->mothurRemove(accnosFileName);
949                 rename((accnosFileName+".temp").c_str(), accnosFileName.c_str());
950                 
951                 
952                 //edit chimera file
953                 ifstream in; 
954                 m->openInputFile(outputFileName, in);
955                 
956                 ofstream out;
957                 m->openOutputFile(outputFileName+".temp", out); out.setf(ios::fixed, ios::floatfield); out.setf(ios::showpoint);
958
959                 string rest, parent1, parent2, line;
960                 set<string> namesInFile; //this is so if a sequence is found to be chimera in several samples we dont write it to the results file more than once
961                 set<string>::iterator itNames;
962                 
963                 //assumptions - in file each read will always look like...
964                 /*
965                  F11Fcsw_92754  no
966                  F11Fcsw_63104  F11Fcsw_33372   F11Fcsw_37007   0.89441 80.4469 0.2     1.03727 93.2961 52.2    no      0-241   243-369 
967                  */
968                 
969                 //get header line
970                 if (!in.eof()) {
971                         line = m->getline(in); m->gobble(in);
972                         out << line << endl;
973                 }
974                 
975                 //for the chimera file, we want to make sure if any group finds a sequence to be chimeric then all groups do, 
976                 //so if this is a report that did not find it to be chimeric, but it appears in the accnos file, 
977                 //then ignore this report and continue until we find the report that found it to be chimeric
978                 
979                 while (!in.eof()) {
980                         
981                         if (m->control_pressed) { in.close(); out.close(); m->mothurRemove((outputFileName+".temp")); return 0; }
982                         
983                         in >> name;             m->gobble(in);
984                         in >> parent1;  m->gobble(in);
985                         
986                         if (name == "Name") { //name = "Name" because we append the header line each time we add results from the groups
987                                 line = m->getline(in); m->gobble(in);
988                         }else {
989                                 if (parent1 == "no") {
990                                         //find unique name
991                                         itUnique = uniqueNames.find(name);
992                                         
993                                         if (itUnique == uniqueNames.end()) { m->mothurOut("[ERROR]: trouble parsing chimera results. Cannot find "+ name + "."); m->mothurOutEndLine(); m->control_pressed = true; }
994                                         else {
995                                                 //is this sequence really not chimeric??
996                                                 itChimeras = chimerasInFile.find(itUnique->second);
997                                                 
998                                                 if (itChimeras == chimerasInFile.end()) {
999                                                         //is this sequence not already in the file
1000                                                         itNames = namesInFile.find((itUnique->second));
1001                                                         
1002                                                         if (itNames == namesInFile.end()) { out << itUnique->second << '\t' << "no" << endl; namesInFile.insert(itUnique->second); }
1003                                                 }
1004                                         }
1005                                 }else { //read the rest of the line
1006                                         double DivQLAQRB,PerIDQLAQRB,BootStrapA,DivQLBQRA,PerIDQLBQRA,BootStrapB;
1007                                         string flag, range1, range2;
1008                                         bool print = false;
1009                                         in >> parent2 >> DivQLAQRB >> PerIDQLAQRB >> BootStrapA >> DivQLBQRA >> PerIDQLBQRA >> BootStrapB >> flag >> range1 >> range2;  m->gobble(in);
1010                                         
1011                                         //find unique name
1012                                         itUnique = uniqueNames.find(name);
1013                                         
1014                                         if (itUnique == uniqueNames.end()) { m->mothurOut("[ERROR]: trouble parsing chimera results. Cannot find "+ name + "."); m->mothurOutEndLine(); m->control_pressed = true; }
1015                                         else {
1016                                                 name = itUnique->second;
1017                                                 //is this name already in the file
1018                                                 itNames = namesInFile.find((name));
1019                                                 
1020                                                 if (itNames == namesInFile.end()) { //no not in file
1021                                                         if (flag == "no") { //are you really a no??
1022                                                                 //is this sequence really not chimeric??
1023                                                                 itChimeras = chimerasInFile.find(name);
1024                                                                 
1025                                                                 //then you really are a no so print, otherwise skip
1026                                                                 if (itChimeras == chimerasInFile.end()) { print = true; }
1027                                                                 
1028                                                         }else{ print = true; }
1029                                                 }
1030                                         }
1031                                         
1032                                         if (print) {
1033                                                 out << name << '\t';
1034                                                 
1035                                                 namesInFile.insert(name);
1036
1037                                                 //output parent1's name
1038                                                 itUnique = uniqueNames.find(parent1);
1039                                                 if (itUnique == uniqueNames.end()) { m->mothurOut("[ERROR]: trouble parsing chimera results. Cannot find parentA "+ parent1 + "."); m->mothurOutEndLine(); m->control_pressed = true; }
1040                                                 else { out << itUnique->second << '\t'; }
1041                                                 
1042                                                 //output parent2's name
1043                                                 itUnique = uniqueNames.find(parent2);
1044                                                 if (itUnique == uniqueNames.end()) { m->mothurOut("[ERROR]: trouble parsing chimera results. Cannot find parentA "+ parent2 + "."); m->mothurOutEndLine(); m->control_pressed = true; }
1045                                                 else { out << itUnique->second << '\t'; }
1046                                                 
1047                                                 out << DivQLAQRB << '\t' << PerIDQLAQRB << '\t' << BootStrapA << '\t' << DivQLBQRA << '\t' << PerIDQLBQRA << '\t' << BootStrapB << '\t' << flag << '\t' << range1 << '\t' << range2 << endl;
1048                                         }
1049                                 }                               
1050                         }
1051                 }
1052                 in.close();
1053                 out.close();
1054                 
1055                 m->mothurRemove(outputFileName);
1056                 rename((outputFileName+".temp").c_str(), outputFileName.c_str());
1057                 
1058                 //edit fasta file
1059                 if (trim) {
1060                         ifstream in3; 
1061                         m->openInputFile(trimFileName, in3);
1062                         
1063                         ofstream out3;
1064                         m->openOutputFile(trimFileName+".temp", out3);
1065                         
1066                         namesInFile.clear();
1067                         
1068                         while (!in3.eof()) {
1069                                 if (m->control_pressed) { in3.close(); out3.close(); m->mothurRemove(outputFileName); m->mothurRemove(accnosFileName); m->mothurRemove((trimFileName+".temp")); return 0; }
1070                                 
1071                                 Sequence seq(in3); m->gobble(in3);
1072                                 
1073                                 if (seq.getName() != "") {
1074                                         //find unique name
1075                                         itUnique = uniqueNames.find(seq.getName());
1076                                         
1077                                         if (itUnique == uniqueNames.end()) { m->mothurOut("[ERROR]: trouble parsing accnos results. Cannot find "+ seq.getName() + "."); m->mothurOutEndLine(); m->control_pressed = true; }
1078                                         else {
1079                                                 itNames = namesInFile.find((itUnique->second));
1080                                                 
1081                                                 if (itNames == namesInFile.end()) {
1082                                                         seq.printSequence(out3);
1083                                                 }
1084                                         }
1085                                 }
1086                         }
1087                         in3.close();
1088                         out3.close();
1089                         
1090                         m->mothurRemove(trimFileName);
1091                         rename((trimFileName+".temp").c_str(), trimFileName.c_str());
1092                 }
1093                 
1094                 return total;
1095         }
1096         catch(exception& e) {
1097                 m->errorOut(e, "ChimeraSlayerCommand", "deconvoluteResults");
1098                 exit(1);
1099         }
1100 }
1101 //**********************************************************************************************************************
1102 int ChimeraSlayerCommand::setUpForSelfReference(SequenceParser*& parser, map<string, string>& fileGroup, map<string, map<string, int> >& fileToPriority, int s){
1103         try {
1104                 fileGroup.clear();
1105                 fileToPriority.clear();
1106                 
1107                 string nameFile = "";
1108                 if (nameFileNames.size() != 0) { //you provided a namefile and we don't need to create one
1109                         nameFile = nameFileNames[s];
1110                 }else {  nameFile = getNamesFile(fastaFileNames[s]); }
1111                 
1112                 //you provided a groupfile
1113                 string groupFile = "";
1114                 if (groupFileNames.size() != 0) { groupFile = groupFileNames[s]; }
1115                 
1116                 if (groupFile == "") { 
1117                         if (processors != 1) { m->mothurOut("When using template=self, mothur can only use 1 processor, continuing."); m->mothurOutEndLine(); processors = 1; }
1118                                                 
1119                         //sort fastafile by abundance, returns new sorted fastafile name
1120                         m->mothurOut("Sorting fastafile according to abundance..."); cout.flush(); 
1121                         priority = sortFastaFile(fastaFileNames[s], nameFile);
1122                         m->mothurOut("Done."); m->mothurOutEndLine();
1123                         
1124                         fileToPriority[fastaFileNames[s]] = priority;
1125                         fileGroup[fastaFileNames[s]] = "noGroup";
1126                 }else {
1127                         //Parse sequences by group
1128                         parser = new SequenceParser(groupFile, fastaFileNames[s], nameFile);
1129                         vector<string> groups = parser->getNamesOfGroups();
1130                         
1131                         for (int i = 0; i < groups.size(); i++) {
1132                                 vector<Sequence> thisGroupsSeqs = parser->getSeqs(groups[i]);
1133                                 map<string, string> thisGroupsMap = parser->getNameMap(groups[i]);
1134                                 string newFastaFile = outputDir + m->getRootName(m->getSimpleName(fastaFileNames[s])) + groups[i] + "-sortedTemp.fasta";
1135                                 priority = sortFastaFile(thisGroupsSeqs, thisGroupsMap, newFastaFile); 
1136                                 fileToPriority[newFastaFile] = priority;
1137                                 fileGroup[newFastaFile] = groups[i];
1138                         }
1139                 }
1140                 
1141                 
1142                 return 0;
1143         }
1144         catch(exception& e) {
1145                 m->errorOut(e, "ChimeraSlayerCommand", "setUpForSelfReference");
1146                 exit(1);
1147         }
1148 }
1149
1150 //**********************************************************************************************************************
1151 string ChimeraSlayerCommand::getNamesFile(string& inputFile){
1152         try {
1153                 string nameFile = "";
1154                 
1155                 m->mothurOutEndLine(); m->mothurOut("No namesfile given, running unique.seqs command to generate one."); m->mothurOutEndLine(); m->mothurOutEndLine();
1156                 
1157                 //use unique.seqs to create new name and fastafile
1158                 string inputString = "fasta=" + inputFile;
1159                 m->mothurOut("/******************************************/"); m->mothurOutEndLine(); 
1160                 m->mothurOut("Running command: unique.seqs(" + inputString + ")"); m->mothurOutEndLine(); 
1161                 
1162                 Command* uniqueCommand = new DeconvoluteCommand(inputString);
1163                 uniqueCommand->execute();
1164                 
1165                 map<string, vector<string> > filenames = uniqueCommand->getOutputFiles();
1166                 
1167                 delete uniqueCommand;
1168                 
1169                 m->mothurOut("/******************************************/"); m->mothurOutEndLine(); 
1170                 
1171                 nameFile = filenames["name"][0];
1172                 inputFile = filenames["fasta"][0];
1173                 
1174                 return nameFile;
1175         }
1176         catch(exception& e) {
1177                 m->errorOut(e, "ChimeraSlayerCommand", "getNamesFile");
1178                 exit(1);
1179         }
1180 }
1181 //**********************************************************************************************************************
1182
1183 int ChimeraSlayerCommand::driverGroups(string outputFName, string accnos, string fasta, map<string, map<string, int> >& fileToPriority, map<string, string>& fileGroup){
1184         try {
1185                 int totalSeqs = 0;
1186                 
1187                 for (map<string, map<string, int> >::iterator itFile = fileToPriority.begin(); itFile != fileToPriority.end(); itFile++) {
1188                         
1189                         if (m->control_pressed) {  return 0;  }
1190                         
1191                         int start = time(NULL);
1192                         string thisFastaName = itFile->first;
1193                         map<string, int> thisPriority = itFile->second;
1194                         string thisoutputFileName = outputDir + m->getRootName(m->getSimpleName(thisFastaName)) + fileGroup[thisFastaName] + "slayer.chimera";
1195                         string thisaccnosFileName = outputDir + m->getRootName(m->getSimpleName(thisFastaName)) + fileGroup[thisFastaName] + "slayer.accnos";
1196                         string thistrimFastaFileName = outputDir + m->getRootName(m->getSimpleName(thisFastaName)) + fileGroup[thisFastaName] + "slayer.fasta";
1197                         
1198                         m->mothurOutEndLine(); m->mothurOut("Checking sequences from group: " + fileGroup[thisFastaName] + "."); m->mothurOutEndLine(); 
1199                         
1200                         lines.clear();
1201 #if defined (__APPLE__) || (__MACH__) || (linux) || (__linux) || (__linux__) || (__unix__) || (__unix)
1202                         int proc = 1;
1203                         vector<unsigned long long> positions = m->divideFile(thisFastaName, proc);
1204                         lines.push_back(linePair(positions[0], positions[1]));  
1205 #else
1206                         lines.push_back(linePair(0, 1000)); 
1207 #endif                  
1208                         int numSeqs = driver(lines[0], thisoutputFileName, thisFastaName, thisaccnosFileName, thistrimFastaFileName, thisPriority);
1209                         
1210                         //append files
1211                         m->appendFiles(thisoutputFileName, outputFName); m->mothurRemove(thisoutputFileName); 
1212                         m->appendFiles(thisaccnosFileName, accnos); m->mothurRemove(thisaccnosFileName);
1213                         if (trim) { m->appendFiles(thistrimFastaFileName, fasta); m->mothurRemove(thistrimFastaFileName); }
1214                         m->mothurRemove(thisFastaName);
1215                         
1216                         totalSeqs += numSeqs;
1217                         
1218                         m->mothurOutEndLine(); m->mothurOut("It took " + toString(time(NULL) - start) + " secs to check " + toString(numSeqs) + " sequences from group " + fileGroup[thisFastaName] + ".");     m->mothurOutEndLine();
1219                 }
1220                 
1221                 return totalSeqs;
1222         }
1223         catch(exception& e) {
1224                 m->errorOut(e, "ChimeraSlayerCommand", "driverGroups");
1225                 exit(1);
1226         }
1227 }
1228 /**************************************************************************************************/
1229 int ChimeraSlayerCommand::createProcessesGroups(string outputFName, string accnos, string fasta, map<string, map<string, int> >& fileToPriority, map<string, string>& fileGroup) {
1230         try {
1231                 int process = 1;
1232                 int num = 0;
1233                 processIDS.clear();
1234                 
1235                 if (fileToPriority.size() < processors) { processors = fileToPriority.size(); }
1236                 
1237                 int groupsPerProcessor = fileToPriority.size() / processors;
1238                 int remainder = fileToPriority.size() % processors;
1239                 
1240                 vector< map<string, map<string, int> > > breakUp;
1241                 
1242                 for (int i = 0; i < processors; i++) {
1243                         map<string, map<string, int> > thisFileToPriority;
1244                         map<string, map<string, int> >::iterator itFile;
1245                         int count = 0;
1246                         int enough = groupsPerProcessor;
1247                         if (i == 0) { enough = groupsPerProcessor + remainder; }
1248                         
1249                         for (itFile = fileToPriority.begin(); itFile != fileToPriority.end();) {
1250                                 thisFileToPriority[itFile->first] = itFile->second;
1251                                 fileToPriority.erase(itFile++);
1252                                 count++;
1253                                 if (count == enough) { break; }
1254                         }       
1255                         breakUp.push_back(thisFileToPriority);
1256                 }
1257                                 
1258 #if defined (__APPLE__) || (__MACH__) || (linux) || (__linux) || (__linux__) || (__unix__) || (__unix)
1259                 //loop through and create all the processes you want
1260                 while (process != processors) {
1261                         int pid = fork();
1262                         
1263                         if (pid > 0) {
1264                                 processIDS.push_back(pid);  //create map from line number to pid so you can append files in correct order later
1265                                 process++;
1266                         }else if (pid == 0){
1267                                 num = driverGroups(outputFName + toString(getpid()) + ".temp", accnos + toString(getpid()) + ".temp", fasta + toString(getpid()) + ".temp", breakUp[process], fileGroup);
1268                                 
1269                                 //pass numSeqs to parent
1270                                 ofstream out;
1271                                 string tempFile = outputFName + toString(getpid()) + ".num.temp";
1272                                 m->openOutputFile(tempFile, out);
1273                                 out << num << endl;
1274                                 out.close();
1275                                 exit(0);
1276                         }else { 
1277                                 m->mothurOut("[ERROR]: unable to spawn the necessary processes."); m->mothurOutEndLine(); 
1278                                 for (int i = 0; i < processIDS.size(); i++) { kill (processIDS[i], SIGINT); }
1279                                 exit(0);
1280                         }
1281                 }
1282                 
1283                 num = driverGroups(outputFName, accnos, fasta, breakUp[0], fileGroup);
1284
1285                 //force parent to wait until all the processes are done
1286                 for (int i=0;i<processors;i++) { 
1287                         int temp = processIDS[i];
1288                         wait(&temp);
1289                 }
1290                 
1291                 for (int i = 0; i < processIDS.size(); i++) {
1292                         ifstream in;
1293                         string tempFile =  outputFName + toString(processIDS[i]) + ".num.temp";
1294                         m->openInputFile(tempFile, in);
1295                         if (!in.eof()) { int tempNum = 0;  in >> tempNum; num += tempNum; }
1296                         in.close(); m->mothurRemove(tempFile);
1297                 }
1298 #else
1299                 
1300                 //////////////////////////////////////////////////////////////////////////////////////////////////////
1301                 //Windows version shared memory, so be careful when passing variables through the slayerData struct. 
1302                 //Above fork() will clone, so memory is separate, but that's not the case with windows, 
1303                 //////////////////////////////////////////////////////////////////////////////////////////////////////
1304                 
1305                 vector<slayerData*> pDataArray; 
1306                 DWORD   dwThreadIdArray[processors-1];
1307                 HANDLE  hThreadArray[processors-1]; 
1308                 
1309                 //Create processor worker threads.
1310                 for(int i=1; i<processors; i++ ){
1311                         string extension = toString(i) + ".temp";
1312                         slayerData* tempslayer = new slayerData((outputFName + extension), (fasta + extension), (accnos + extension), templatefile, search, blastlocation, trimera, trim, realign, m, breakUp[i], fileGroup, ksize, match, mismatch, window, minSimilarity, minCoverage, minBS, minSNP, parents, iters, increment, numwanted, divR, priority, i);
1313                         pDataArray.push_back(tempslayer);
1314                         processIDS.push_back(i);
1315                         
1316                         //MySlayerThreadFunction is in header. It must be global or static to work with the threads.
1317                         //default security attributes, thread function name, argument to thread function, use default creation flags, returns the thread identifier
1318                         hThreadArray[i-1] = CreateThread(NULL, 0, MySlayerGroupThreadFunction, pDataArray[i-1], 0, &dwThreadIdArray[i-1]);   
1319                 }
1320                 
1321                 num = driverGroups(outputFName, accnos, fasta, breakUp[0], fileGroup);
1322                 
1323                 //Wait until all threads have terminated.
1324                 WaitForMultipleObjects(processors-1, hThreadArray, TRUE, INFINITE);
1325                 
1326                 //Close all thread handles and free memory allocations.
1327                 for(int i=0; i < pDataArray.size(); i++){
1328                         num += pDataArray[i]->count;
1329                         CloseHandle(hThreadArray[i]);
1330                         delete pDataArray[i];
1331                 }
1332 #endif  
1333                 
1334                 //append output files
1335                 for(int i=0;i<processIDS.size();i++){
1336                         m->appendFiles((outputFName + toString(processIDS[i]) + ".temp"), outputFName);
1337                         m->mothurRemove((outputFName + toString(processIDS[i]) + ".temp"));
1338                         
1339                         m->appendFiles((accnos + toString(processIDS[i]) + ".temp"), accnos);
1340                         m->mothurRemove((accnos + toString(processIDS[i]) + ".temp"));
1341                         
1342                         if (trim) {
1343                                 m->appendFiles((fasta + toString(processIDS[i]) + ".temp"), fasta);
1344                                 m->mothurRemove((fasta + toString(processIDS[i]) + ".temp"));
1345                         }
1346                 }
1347                 
1348                 
1349                 return num;
1350         }
1351         catch(exception& e) {
1352                 m->errorOut(e, "ChimeraSlayerCommand", "createProcessesGroups");
1353                 exit(1);
1354         }
1355 }
1356 //**********************************************************************************************************************
1357
1358 int ChimeraSlayerCommand::driver(linePair filePos, string outputFName, string filename, string accnos, string fasta, map<string, int>& priority){
1359         try {
1360                 
1361                 Chimera* chimera;
1362                 if (templatefile != "self") { //you want to run slayer with a reference template
1363                         chimera = new ChimeraSlayer(filename, templatefile, trim, search, ksize, match, mismatch, window, divR, minSimilarity, minCoverage, minBS, minSNP, parents, iters, increment, numwanted, realign, blastlocation, rand());       
1364                 }else {
1365                         chimera = new ChimeraSlayer(filename, templatefile, trim, priority, search, ksize, match, mismatch, window, divR, minSimilarity, minCoverage, minBS, minSNP, parents, iters, increment, numwanted, realign, blastlocation, rand());     
1366                 }
1367                 
1368                 if (m->control_pressed) { delete chimera; return 0; }
1369                 
1370                 if (chimera->getUnaligned()) { delete chimera; m->mothurOut("Your template sequences are different lengths, please correct."); m->mothurOutEndLine(); m->control_pressed = true; return 0; }
1371                 templateSeqsLength = chimera->getLength();
1372                 
1373                 ofstream out;
1374                 m->openOutputFile(outputFName, out);
1375                 
1376                 ofstream out2;
1377                 m->openOutputFile(accnos, out2);
1378                 
1379                 ofstream out3;
1380                 if (trim) {  m->openOutputFile(fasta, out3); }
1381                 
1382                 ifstream inFASTA;
1383                 m->openInputFile(filename, inFASTA);
1384
1385                 inFASTA.seekg(filePos.start);
1386                 
1387                 if (filePos.start == 0) { chimera->printHeader(out); }
1388
1389                 bool done = false;
1390                 int count = 0;
1391         
1392                 while (!done) {
1393                 
1394                         if (m->control_pressed) {       delete chimera; out.close(); out2.close(); if (trim) { out3.close(); } inFASTA.close(); return 1;       }
1395                 
1396                         Sequence* candidateSeq = new Sequence(inFASTA);  m->gobble(inFASTA);
1397                         string candidateAligned = candidateSeq->getAligned();
1398                         
1399                         if (candidateSeq->getName() != "") { //incase there is a commented sequence at the end of a file
1400                                 if (candidateSeq->getAligned().length() != templateSeqsLength) {  
1401                                         m->mothurOut(candidateSeq->getName() + " is not the same length as the template sequences. Skipping."); m->mothurOutEndLine();
1402                                 }else{
1403                                         //find chimeras
1404                                         chimera->getChimeras(candidateSeq);
1405                                         
1406                                         if (m->control_pressed) {       delete chimera; delete candidateSeq; return 1;  }
1407                                                 
1408                                         //if you are not chimeric, then check each half
1409                                         data_results wholeResults = chimera->getResults();
1410                                         
1411                                         //determine if we need to split
1412                                         bool isChimeric = false;
1413                                         
1414                                         if (wholeResults.flag == "yes") {
1415                                                 string chimeraFlag = "no";
1416                                                 if(  (wholeResults.results[0].bsa >= minBS && wholeResults.results[0].divr_qla_qrb >= divR)
1417                                                    ||
1418                                                    (wholeResults.results[0].bsb >= minBS && wholeResults.results[0].divr_qlb_qra >= divR) ) { chimeraFlag = "yes"; }
1419                                                 
1420                                                 
1421                                                 if (chimeraFlag == "yes") {     
1422                                                         if ((wholeResults.results[0].bsa >= minBS) || (wholeResults.results[0].bsb >= minBS)) { isChimeric = true; }
1423                                                 }
1424                                         }
1425                                         
1426                                         if ((!isChimeric) && trimera) {
1427                                                 
1428                                                 //split sequence in half by bases
1429                                                 string leftQuery, rightQuery;
1430                                                 Sequence tempSeq(candidateSeq->getName(), candidateAligned);
1431                                                 divideInHalf(tempSeq, leftQuery, rightQuery);
1432                                                 
1433                                                 //run chimeraSlayer on each piece
1434                                                 Sequence* left = new Sequence(candidateSeq->getName(), leftQuery);
1435                                                 Sequence* right = new Sequence(candidateSeq->getName(), rightQuery);
1436                                                 
1437                                                 //find chimeras
1438                                                 chimera->getChimeras(left);
1439                                                 data_results leftResults = chimera->getResults();
1440                                                 
1441                                                 chimera->getChimeras(right);
1442                                                 data_results rightResults = chimera->getResults();
1443                                                 
1444                                                 //if either piece is chimeric then report
1445                                                 Sequence trimmed = chimera->print(out, out2, leftResults, rightResults);
1446                                                 if (trim) { trimmed.printSequence(out3);  }
1447                                                 
1448                                                 delete left; delete right;
1449                                                 
1450                                         }else { //already chimeric
1451                                                 //print results
1452                                                 Sequence trimmed = chimera->print(out, out2);
1453                                                 if (trim) { trimmed.printSequence(out3);  }
1454                                         }
1455                                         
1456                                         
1457                                 }
1458                                 count++;
1459                         }
1460                         
1461                         #if defined (__APPLE__) || (__MACH__) || (linux) || (__linux) || (__linux__) || (__unix__) || (__unix)
1462                                 unsigned long long pos = inFASTA.tellg();
1463                                 if ((pos == -1) || (pos >= filePos.end)) { break; }
1464                         #else
1465                                 if (inFASTA.eof()) { break; }
1466                         #endif
1467                         
1468                         delete candidateSeq;
1469                         //report progress
1470                         if((count) % 100 == 0){ m->mothurOut("Processing sequence: " + toString(count)); m->mothurOutEndLine();         }
1471                 }
1472                 //report progress
1473                 if((count) % 100 != 0){ m->mothurOut("Processing sequence: " + toString(count)); m->mothurOutEndLine();         }
1474                 
1475                 int numNoParents = chimera->getNumNoParents();
1476                 if (numNoParents == count) { m->mothurOut("[WARNING]: megablast returned 0 potential parents for all your sequences. This could be due to formatdb.exe not being setup properly, please check formatdb.log for errors."); m->mothurOutEndLine(); } 
1477                 
1478                 out.close();
1479                 out2.close();
1480                 if (trim) { out3.close(); }
1481                 inFASTA.close();
1482                 delete chimera;
1483                                 
1484                 return count;
1485                 
1486                 
1487         }
1488         catch(exception& e) {
1489                 m->errorOut(e, "ChimeraSlayerCommand", "driver");
1490                 exit(1);
1491         }
1492 }
1493 //**********************************************************************************************************************
1494 #ifdef USE_MPI
1495 int ChimeraSlayerCommand::driverMPI(int start, int num, MPI_File& inMPI, MPI_File& outMPI, MPI_File& outAccMPI, MPI_File& outFastaMPI, vector<unsigned long long>& MPIPos, string filename, map<string, int>& priority, bool byGroup){
1496         try {
1497                 MPI_Status status; 
1498                 int pid;
1499                 MPI_Comm_rank(MPI_COMM_WORLD, &pid); //find out who we are
1500                 
1501                 Chimera* chimera;
1502                 if (templatefile != "self") { //you want to run slayer with a reference template
1503                         chimera = new ChimeraSlayer(filename, templatefile, trim, search, ksize, match, mismatch, window, divR, minSimilarity, minCoverage, minBS, minSNP, parents, iters, increment, numwanted, realign, blastlocation, rand());       
1504                 }else {
1505                         chimera = new ChimeraSlayer(filename, templatefile, trim, priority, search, ksize, match, mismatch, window, divR, minSimilarity, minCoverage, minBS, minSNP, parents, iters, increment, numwanted, realign, blastlocation, rand(), byGroup);    
1506                 }
1507                 
1508                 if (m->control_pressed) { delete chimera; return 0; }
1509                 
1510                 if (chimera->getUnaligned()) { delete chimera; m->mothurOut("Your template sequences are different lengths, please correct."); m->mothurOutEndLine(); m->control_pressed = true; return 0; }
1511                 templateSeqsLength = chimera->getLength();
1512                 
1513                 for(int i=0;i<num;i++){
1514                         
1515                         if (m->control_pressed) {       delete chimera; return 1;       }
1516                         
1517                         //read next sequence
1518                         int length = MPIPos[start+i+1] - MPIPos[start+i];
1519
1520                         char* buf4 = new char[length];
1521                         MPI_File_read_at(inMPI, MPIPos[start+i], buf4, length, MPI_CHAR, &status);
1522         
1523                         string tempBuf = buf4;
1524                         if (tempBuf.length() > length) { tempBuf = tempBuf.substr(0, length);  }
1525                         istringstream iss (tempBuf,istringstream::in);
1526
1527                         delete buf4;
1528
1529                         Sequence* candidateSeq = new Sequence(iss);  m->gobble(iss);
1530                         string candidateAligned = candidateSeq->getAligned();
1531                 
1532                         if (candidateSeq->getName() != "") { //incase there is a commented sequence at the end of a file
1533                                 
1534                                 if (candidateSeq->getAligned().length() != templateSeqsLength) {  
1535                                         m->mothurOut(candidateSeq->getName() + " is not the same length as the template sequences. Skipping."); m->mothurOutEndLine();
1536                                 }else{
1537                 
1538                                         //find chimeras
1539                                         chimera->getChimeras(candidateSeq);
1540                         
1541                                         if (m->control_pressed) {       delete chimera; delete candidateSeq; return 1;  }
1542                                         
1543                                         //if you are not chimeric, then check each half
1544                                         data_results wholeResults = chimera->getResults();
1545                                         
1546                                         //determine if we need to split
1547                                         bool isChimeric = false;
1548                                         
1549                                         if (wholeResults.flag == "yes") {
1550                                                 string chimeraFlag = "no";
1551                                                 if(  (wholeResults.results[0].bsa >= minBS && wholeResults.results[0].divr_qla_qrb >= divR)
1552                                                    ||
1553                                                    (wholeResults.results[0].bsb >= minBS && wholeResults.results[0].divr_qlb_qra >= divR) ) { chimeraFlag = "yes"; }
1554                                                 
1555                                                 
1556                                                 if (chimeraFlag == "yes") {     
1557                                                         if ((wholeResults.results[0].bsa >= minBS) || (wholeResults.results[0].bsb >= minBS)) { isChimeric = true; }
1558                                                 }
1559                                         }
1560                                         
1561                                         if ((!isChimeric) && trimera) {                                                 
1562                                                 //split sequence in half by bases
1563                                                 string leftQuery, rightQuery;
1564                                                 Sequence tempSeq(candidateSeq->getName(), candidateAligned);
1565                                                 divideInHalf(tempSeq, leftQuery, rightQuery);
1566                                                 
1567                                                 //run chimeraSlayer on each piece
1568                                                 Sequence* left = new Sequence(candidateSeq->getName(), leftQuery);
1569                                                 Sequence* right = new Sequence(candidateSeq->getName(), rightQuery);
1570                                                 
1571                                                 //find chimeras
1572                                                 chimera->getChimeras(left);
1573                                                 data_results leftResults = chimera->getResults();
1574                                                 
1575                                                 chimera->getChimeras(right);
1576                                                 data_results rightResults = chimera->getResults();
1577                                                 
1578                                                 //if either piece is chimeric then report
1579                                                 Sequence trimmed = chimera->print(outMPI, outAccMPI, leftResults, rightResults);
1580                                                 if (trim) {  
1581                                                         string outputString = ">" + trimmed.getName() + "\n" + trimmed.getAligned() + "\n";
1582                                                         
1583                                                         //write to accnos file
1584                                                         int length = outputString.length();
1585                                                         char* buf2 = new char[length];
1586                                                         memcpy(buf2, outputString.c_str(), length);
1587                                                         
1588                                                         MPI_File_write_shared(outFastaMPI, buf2, length, MPI_CHAR, &status);
1589                                                         delete buf2;
1590                                                 }
1591                                                 
1592                                                 delete left; delete right;
1593                                                 
1594                                         }else { 
1595                                                 //print results
1596                                                 Sequence trimmed = chimera->print(outMPI, outAccMPI);
1597                                                 
1598                                                 if (trim) {  
1599                                                         string outputString = ">" + trimmed.getName() + "\n" + trimmed.getAligned() + "\n";
1600                                                         
1601                                                         //write to accnos file
1602                                                         int length = outputString.length();
1603                                                         char* buf2 = new char[length];
1604                                                         memcpy(buf2, outputString.c_str(), length);
1605                                                         
1606                                                         MPI_File_write_shared(outFastaMPI, buf2, length, MPI_CHAR, &status);
1607                                                         delete buf2;
1608                                                 }
1609                                         }
1610                                         
1611                                 }
1612                         }
1613                         delete candidateSeq;
1614                         
1615                         //report progress
1616                         if((i+1) % 100 == 0){  cout << "Processing sequence: " << (i+1) << endl;        m->mothurOutJustToLog("Processing sequence: " + toString(i+1) + "\n");          }
1617                 }
1618                 //report progress
1619                 if(num % 100 != 0){             cout << "Processing sequence: " << num << endl; m->mothurOutJustToLog("Processing sequence: " + toString(num) + "\n");  }
1620                 
1621                 int numNoParents = chimera->getNumNoParents();
1622                 if (numNoParents == num) { cout << "[WARNING]: megablast returned 0 potential parents for all your sequences. This could be due to formatdb.exe not being setup properly, please check formatdb.log for errors." << endl; }
1623                 
1624                 delete chimera;         
1625                 return 0;
1626         }
1627         catch(exception& e) {
1628                 m->errorOut(e, "ChimeraSlayerCommand", "driverMPI");
1629                 exit(1);
1630         }
1631 }
1632 #endif
1633
1634 /**************************************************************************************************/
1635
1636 int ChimeraSlayerCommand::createProcesses(string outputFileName, string filename, string accnos, string fasta, map<string, int>& thisPriority) {
1637         try {
1638                 int process = 0;
1639                 int num = 0;
1640                 processIDS.clear();
1641                 
1642 #if defined (__APPLE__) || (__MACH__) || (linux) || (__linux) || (__linux__) || (__unix__) || (__unix)
1643                 //loop through and create all the processes you want
1644                 while (process != processors) {
1645                         int pid = fork();
1646                         
1647                         if (pid > 0) {
1648                                 processIDS.push_back(pid);  //create map from line number to pid so you can append files in correct order later
1649                                 process++;
1650                         }else if (pid == 0){
1651                                 num = driver(lines[process], outputFileName + toString(getpid()) + ".temp", filename, accnos + toString(getpid()) + ".temp", fasta + toString(getpid()) + ".temp", thisPriority);
1652                                 
1653                                 //pass numSeqs to parent
1654                                 ofstream out;
1655                                 string tempFile = outputFileName + toString(getpid()) + ".num.temp";
1656                                 m->openOutputFile(tempFile, out);
1657                                 out << num << endl;
1658                                 out.close();
1659                                 exit(0);
1660                         }else { 
1661                                 m->mothurOut("[ERROR]: unable to spawn the necessary processes."); m->mothurOutEndLine(); 
1662                                 for (int i = 0; i < processIDS.size(); i++) { kill (processIDS[i], SIGINT); }
1663                                 exit(0);
1664                         }
1665                 }
1666                 
1667                 //force parent to wait until all the processes are done
1668                 for (int i=0;i<processors;i++) { 
1669                         int temp = processIDS[i];
1670                         wait(&temp);
1671                 }
1672                 
1673                 for (int i = 0; i < processIDS.size(); i++) {
1674                         ifstream in;
1675                         string tempFile =  outputFileName + toString(processIDS[i]) + ".num.temp";
1676                         m->openInputFile(tempFile, in);
1677                         if (!in.eof()) { int tempNum = 0;  in >> tempNum; num += tempNum; }
1678                         in.close(); m->mothurRemove(tempFile);
1679                 }
1680 #else
1681                 
1682                 //////////////////////////////////////////////////////////////////////////////////////////////////////
1683                 //Windows version shared memory, so be careful when passing variables through the slayerData struct. 
1684                 //Above fork() will clone, so memory is separate, but that's not the case with windows, 
1685                 //////////////////////////////////////////////////////////////////////////////////////////////////////
1686                 
1687                 vector<slayerData*> pDataArray; 
1688                 DWORD   dwThreadIdArray[processors];
1689                 HANDLE  hThreadArray[processors]; 
1690                 
1691                 //Create processor worker threads.
1692                 for( int i=0; i<processors; i++ ){
1693                         string extension = toString(i) + ".temp";
1694                         slayerData* tempslayer = new slayerData((outputFileName + extension), (fasta + extension), (accnos + extension), filename, templatefile, search, blastlocation, trimera, trim, realign, m, lines[i].start, lines[i].end, ksize, match, mismatch, window, minSimilarity, minCoverage, minBS, minSNP, parents, iters, increment, numwanted, divR, priority, i);
1695                         pDataArray.push_back(tempslayer);
1696                         processIDS.push_back(i);
1697                         
1698                         //MySlayerThreadFunction is in header. It must be global or static to work with the threads.
1699                         //default security attributes, thread function name, argument to thread function, use default creation flags, returns the thread identifier
1700                         hThreadArray[i] = CreateThread(NULL, 0, MySlayerThreadFunction, pDataArray[i], 0, &dwThreadIdArray[i]);   
1701                 }
1702                                 
1703                 //Wait until all threads have terminated.
1704                 WaitForMultipleObjects(processors, hThreadArray, TRUE, INFINITE);
1705                 
1706                 //Close all thread handles and free memory allocations.
1707                 for(int i=0; i < pDataArray.size(); i++){
1708                         num += pDataArray[i]->count;
1709                         CloseHandle(hThreadArray[i]);
1710                         delete pDataArray[i];
1711                 }
1712 #endif  
1713                 
1714                 rename((outputFileName + toString(processIDS[0]) + ".temp").c_str(), outputFileName.c_str());
1715                 rename((accnos + toString(processIDS[0]) + ".temp").c_str(), accnos.c_str());
1716                 if (trim) {  rename((fasta + toString(processIDS[0]) + ".temp").c_str(), fasta.c_str()); }
1717                 
1718                 //append output files
1719                 for(int i=1;i<processIDS.size();i++){
1720                         m->appendFiles((outputFileName + toString(processIDS[i]) + ".temp"), outputFileName);
1721                         m->mothurRemove((outputFileName + toString(processIDS[i]) + ".temp"));
1722                         
1723                         m->appendFiles((accnos + toString(processIDS[i]) + ".temp"), accnos);
1724                         m->mothurRemove((accnos + toString(processIDS[i]) + ".temp"));
1725                         
1726                         if (trim) {
1727                                 m->appendFiles((fasta + toString(processIDS[i]) + ".temp"), fasta);
1728                                 m->mothurRemove((fasta + toString(processIDS[i]) + ".temp"));
1729                         }
1730                 }
1731                 
1732                 
1733                 return num;
1734         }
1735         catch(exception& e) {
1736                 m->errorOut(e, "ChimeraSlayerCommand", "createProcesses");
1737                 exit(1);
1738         }
1739 }
1740
1741 /**************************************************************************************************/
1742
1743 int ChimeraSlayerCommand::divideInHalf(Sequence querySeq, string& leftQuery, string& rightQuery) {
1744         try {
1745                 
1746                 string queryUnAligned = querySeq.getUnaligned();
1747                 int numBases = int(queryUnAligned.length() * 0.5);
1748                 
1749                 string queryAligned = querySeq.getAligned();
1750                 leftQuery = querySeq.getAligned();
1751                 rightQuery = querySeq.getAligned();
1752                 
1753                 int baseCount = 0;
1754                 int leftSpot = 0;
1755                 for (int i = 0; i < queryAligned.length(); i++) {
1756                         //if you are a base
1757                         if (isalpha(queryAligned[i])) {         
1758                                 baseCount++; 
1759                         }
1760                         
1761                         //if you have half
1762                         if (baseCount >= numBases) {  leftSpot = i; break; } //first half
1763                 }
1764                 
1765                 //blank out right side
1766                 for (int i = leftSpot; i < leftQuery.length(); i++) { leftQuery[i] = '.'; }
1767                 
1768                 //blank out left side
1769                 for (int i = 0; i < leftSpot; i++) { rightQuery[i] = '.'; }
1770                 
1771                 return 0;
1772                 
1773         }
1774         catch(exception& e) {
1775                 m->errorOut(e, "ChimeraSlayerCommand", "divideInHalf");
1776                 exit(1);
1777         }
1778 }
1779 /**************************************************************************************************/
1780 map<string, int> ChimeraSlayerCommand::sortFastaFile(string fastaFile, string nameFile) {
1781         try {
1782                 map<string, int> nameAbund;
1783                 
1784                 //read through fastafile and store info
1785                 map<string, string> seqs;
1786                 ifstream in;
1787                 m->openInputFile(fastaFile, in);
1788                 
1789                 while (!in.eof()) {
1790                         
1791                         if (m->control_pressed) { in.close(); return nameAbund; }
1792                         
1793                         Sequence seq(in); m->gobble(in);
1794                         seqs[seq.getName()] = seq.getAligned();
1795                 }
1796                 
1797                 in.close();
1798                 
1799                 //read namefile
1800                 vector<seqPriorityNode> nameMapCount;
1801                 int error = m->readNames(nameFile, nameMapCount, seqs);
1802                 
1803                 if (m->control_pressed) { return nameAbund; }
1804                 
1805                 if (error == 1) { m->control_pressed = true; return nameAbund; }
1806                 if (seqs.size() != nameMapCount.size()) { m->mothurOut( "The number of sequences in your fastafile does not match the number of sequences in your namefile, aborting."); m->mothurOutEndLine(); m->control_pressed = true; return nameAbund; }
1807
1808                 sort(nameMapCount.begin(), nameMapCount.end(), compareSeqPriorityNodes);
1809                 
1810                 string newFasta = fastaFile + ".temp";
1811                 ofstream out;
1812                 m->openOutputFile(newFasta, out);
1813                 
1814                 //print new file in order of
1815                 for (int i = 0; i < nameMapCount.size(); i++) {
1816                         out << ">" << nameMapCount[i].name << endl << nameMapCount[i].seq << endl;
1817                         nameAbund[nameMapCount[i].name] = nameMapCount[i].numIdentical;
1818                 }
1819                 out.close();
1820                 
1821                 rename(newFasta.c_str(), fastaFile.c_str());
1822                                 
1823                 return nameAbund;
1824                 
1825         }
1826         catch(exception& e) {
1827                 m->errorOut(e, "ChimeraSlayerCommand", "sortFastaFile");
1828                 exit(1);
1829         }
1830 }
1831 /**************************************************************************************************/
1832 map<string, int> ChimeraSlayerCommand::sortFastaFile(vector<Sequence>& thisseqs, map<string, string>& nameMap, string newFile) {
1833         try {
1834                 map<string, int> nameAbund;
1835                 vector<seqPriorityNode> nameVector;
1836                 
1837                 //read through fastafile and store info
1838                 map<string, string> seqs;
1839                                 
1840                 for (int i = 0; i < thisseqs.size(); i++) {
1841                         
1842                         if (m->control_pressed) { return nameAbund; }
1843                         
1844                         map<string, string>::iterator itNameMap = nameMap.find(thisseqs[i].getName());
1845                         
1846                         if (itNameMap == nameMap.end()){
1847                                 m->control_pressed = true;
1848                                 m->mothurOut("[ERROR]: " + thisseqs[i].getName() + " is in your fastafile, but is not in your namesfile, please correct."); m->mothurOutEndLine();
1849                         }else {
1850                                 int num = m->getNumNames(itNameMap->second);
1851                                 
1852                                 seqPriorityNode temp(num, thisseqs[i].getAligned(), thisseqs[i].getName());
1853                                 nameVector.push_back(temp);
1854                         }
1855                 }
1856         
1857                 //sort by num represented
1858                 sort(nameVector.begin(), nameVector.end(), compareSeqPriorityNodes);
1859         
1860                 if (m->control_pressed) { return nameAbund; }
1861                 
1862                 if (thisseqs.size() != nameVector.size()) { m->mothurOut( "The number of sequences in your fastafile does not match the number of sequences in your namefile, aborting."); m->mothurOutEndLine(); m->control_pressed = true; return nameAbund; }
1863                                 
1864                 ofstream out;
1865                 m->openOutputFile(newFile, out);
1866                 
1867                 //print new file in order of
1868                 for (int i = 0; i < nameVector.size(); i++) {
1869                         out << ">" << nameVector[i].name << endl << nameVector[i].seq << endl;
1870                         nameAbund[nameVector[i].name] = nameVector[i].numIdentical;
1871                 }
1872                 out.close();
1873                 
1874                 return nameAbund;
1875                 
1876         }
1877         catch(exception& e) {
1878                 m->errorOut(e, "ChimeraSlayerCommand", "sortFastaFile");
1879                 exit(1);
1880         }
1881 }
1882 /**************************************************************************************************/
1883