#include "rabundvector.hpp"
#include "sabundvector.hpp"
+//**********************************************************************************************************************
+vector<string> PhylotypeCommand::getValidParameters(){
+ try {
+ string Array[] = {"taxonomy","cutoff","label","name","outputdir","inputdir"};
+ vector<string> myArray (Array, Array+(sizeof(Array)/sizeof(string)));
+ return myArray;
+ }
+ catch(exception& e) {
+ m->errorOut(e, "PhylotypeCommand", "getValidParameters");
+ exit(1);
+ }
+}
+//**********************************************************************************************************************
+PhylotypeCommand::PhylotypeCommand(){
+ try {
+ abort = true;
+ //initialize outputTypes
+ vector<string> tempOutNames;
+ outputTypes["list"] = tempOutNames;
+ outputTypes["sabund"] = tempOutNames;
+ outputTypes["rabund"] = tempOutNames;
+ }
+ catch(exception& e) {
+ m->errorOut(e, "PhylotypeCommand", "PhylotypeCommand");
+ exit(1);
+ }
+}
+//**********************************************************************************************************************
+vector<string> PhylotypeCommand::getRequiredParameters(){
+ try {
+ string Array[] = {"taxonomy"};
+ vector<string> myArray (Array, Array+(sizeof(Array)/sizeof(string)));
+ return myArray;
+ }
+ catch(exception& e) {
+ m->errorOut(e, "PhylotypeCommand", "getRequiredParameters");
+ exit(1);
+ }
+}
+//**********************************************************************************************************************
+vector<string> PhylotypeCommand::getRequiredFiles(){
+ try {
+ vector<string> myArray;
+ return myArray;
+ }
+ catch(exception& e) {
+ m->errorOut(e, "PhylotypeCommand", "getRequiredFiles");
+ exit(1);
+ }
+}
/**********************************************************************************************************************/
PhylotypeCommand::PhylotypeCommand(string option) {
try {
else {
//valid paramters for this command
- string AlignArray[] = {"taxonomy","cutoff","label","outputdir","inputdir"};
+ string AlignArray[] = {"taxonomy","cutoff","label","name","outputdir","inputdir"};
vector<string> myArray (AlignArray, AlignArray+(sizeof(AlignArray)/sizeof(string)));
OptionParser parser(option);
if (validParameter.isValidParameter(it->first, myArray, it->second) != true) { abort = true; }
}
+ //initialize outputTypes
+ vector<string> tempOutNames;
+ outputTypes["list"] = tempOutNames;
+ outputTypes["sabund"] = tempOutNames;
+ outputTypes["rabund"] = tempOutNames;
+
//if the user changes the input directory command factory will send this info to us in the output parameter
string inputDir = validParameter.validFile(parameters, "inputdir", false);
if (inputDir == "not found"){ inputDir = ""; }
it = parameters.find("taxonomy");
//user has given a template file
if(it != parameters.end()){
- path = hasPath(it->second);
+ path = m->hasPath(it->second);
//if the user has not given a path then, add inputdir. else leave path alone.
if (path == "") { parameters["taxonomy"] = inputDir + it->second; }
}
+
+ it = parameters.find("name");
+ //user has given a template file
+ if(it != parameters.end()){
+ path = m->hasPath(it->second);
+ //if the user has not given a path then, add inputdir. else leave path alone.
+ if (path == "") { parameters["name"] = inputDir + it->second; }
+ }
}
taxonomyFileName = validParameter.validFile(parameters, "taxonomy", true);
abort = true;
}else if (taxonomyFileName == "not open") { abort = true; }
+ namefile = validParameter.validFile(parameters, "name", true);
+ if (namefile == "not open") { abort = true; }
+ else if (namefile == "not found") { namefile = ""; }
+ else { readNamesFile(); }
+
//if the user changes the output directory command factory will send this info to us in the output parameter
outputDir = validParameter.validFile(parameters, "outputdir", false); if (outputDir == "not found"){
outputDir = "";
- outputDir += hasPath(taxonomyFileName); //if user entered a file with a path then preserve it
+ outputDir += m->hasPath(taxonomyFileName); //if user entered a file with a path then preserve it
}
string temp = validParameter.validFile(parameters, "cutoff", false);
label = validParameter.validFile(parameters, "label", false);
if (label == "not found") { label = ""; allLines = 1; }
else {
- if(label != "all") { splitAtDash(label, labels); allLines = 0; }
+ if(label != "all") { m->splitAtDash(label, labels); allLines = 0; }
else { allLines = 1; }
}
if (abort == true) { return 0; }
- vector<string> outputNames;
-
//reads in taxonomy file and makes all the taxonomies the same length
//by appending the last taxon to a given taxonomy as many times as needed to
//make it as long as the longest taxonomy in the file
- TaxEqualizer* taxEqual = new TaxEqualizer(taxonomyFileName, cutoff);
+ TaxEqualizer* taxEqual = new TaxEqualizer(taxonomyFileName, cutoff, outputDir);
+
+ if (m->control_pressed) { delete taxEqual; return 0; }
string equalizedTaxFile = taxEqual->getEqualizedTaxFile();
bool done = false;
if (tree->get(leaves[0]).parent == -1) { m->mothurOut("Empty Tree"); m->mothurOutEndLine(); done = true; }
- string fileroot = outputDir + getRootName(getSimpleName(taxonomyFileName));
+ if (m->control_pressed) { delete tree; return 0; }
+
+ string fileroot = outputDir + m->getRootName(m->getSimpleName(taxonomyFileName));
ofstream outList;
string outputListFile = fileroot + "tx.list";
- openOutputFile(outputListFile, outList);
+ m->openOutputFile(outputListFile, outList);
ofstream outSabund;
string outputSabundFile = fileroot + "tx.sabund";
- openOutputFile(outputSabundFile, outSabund);
+ m->openOutputFile(outputSabundFile, outSabund);
ofstream outRabund;
string outputRabundFile = fileroot + "tx.rabund";
- openOutputFile(outputRabundFile, outRabund);
+ m->openOutputFile(outputRabundFile, outRabund);
- outputNames.push_back(outputListFile);
- outputNames.push_back(outputSabundFile);
- outputNames.push_back(outputRabundFile);
+ outputNames.push_back(outputListFile); outputTypes["list"].push_back(outputListFile);
+ outputNames.push_back(outputSabundFile); outputTypes["sabund"].push_back(outputSabundFile);
+ outputNames.push_back(outputRabundFile); outputTypes["rabund"].push_back(outputRabundFile);
int count = 1;
//start at leaves of tree and work towards root, processing the labels the user wants
string level = toString(count);
count++;
+ if (m->control_pressed) {
+ outRabund.close(); outSabund.close(); outList.close();
+ for (int i = 0; i < outputNames.size(); i++) { remove(outputNames[i].c_str()); }
+ delete tree; return 0;
+ }
+
//is this a level the user want output for
if(allLines == 1 || labels.count(level) == 1){
//make the names compatable with listvector
string name = "";
- for (int i = 0; i < names.size(); i++) { name += names[i] + ","; }
+ for (int i = 0; i < names.size(); i++) {
+ if (namefile != "") {
+ map<string, string>::iterator itNames = namemap.find(names[i]); //make sure this name is in namefile
+
+ if (itNames != namemap.end()) { name += namemap[names[i]] + ","; } //you found it in namefile
+ else { m->mothurOut(names[i] + " is not in your namefile, please correct."); m->mothurOutEndLine(); exit(1); }
+
+ }else{ name += names[i] + ","; }
+ }
name = name.substr(0, name.length()-1); //rip off extra ','
//add bin to list vector
delete tree;
+ if (m->control_pressed) {
+ for (int i = 0; i < outputNames.size(); i++) { remove(outputNames[i].c_str()); }
+ return 0;
+ }
+
m->mothurOutEndLine();
m->mothurOut("Output File Names: "); m->mothurOutEndLine();
for (int i = 0; i < outputNames.size(); i++) { m->mothurOut(outputNames[i]); m->mothurOutEndLine(); }
exit(1);
}
}
+/*****************************************************************/
+int PhylotypeCommand::readNamesFile() {
+ try {
+
+ ifstream in;
+ m->openInputFile(namefile, in);
+
+ string first, second;
+ map<string, string>::iterator itNames;
+
+ while(!in.eof()) {
+ in >> first >> second; m->gobble(in);
+
+ itNames = namemap.find(first);
+ if (itNames == namemap.end()) {
+ namemap[first] = second;
+ }else { m->mothurOut(first + " has already been seen in namefile, disregarding names file."); m->mothurOutEndLine(); in.close(); namemap.clear(); namefile = ""; return 1; }
+ }
+ in.close();
+
+ return 0;
+ }
+ catch(exception& e) {
+ m->errorOut(e, "PhylotypeCommand", "readNamesFile");
+ exit(1);
+ }
+}
+
/**********************************************************************************************************************/
projects / mothur.git / blobdiff