X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=heatmapsimcommand.cpp;h=28b0221ba8458cb1a760fade14426ea6a7123568;hb=4169642e8a8d45f71a4a7241ee02f1b1aae29520;hp=1b4b9d79010b22391ed29e148c8b3f9137ac1b3f;hpb=74844a60d80c6dd06e3fb02ee9b928424f9019b0;p=mothur.git diff --git a/heatmapsimcommand.cpp b/heatmapsimcommand.cpp index 1b4b9d7..28b0221 100644 --- a/heatmapsimcommand.cpp +++ b/heatmapsimcommand.cpp @@ -19,20 +19,65 @@ #include "sharedmorisitahorn.h" #include "sharedbraycurtis.h" - +//********************************************************************************************************************** +vector HeatMapSimCommand::getValidParameters(){ + try { + string Array[] = {"groups","label", "calc","phylip","column","name","outputdir","inputdir"}; + vector myArray (Array, Array+(sizeof(Array)/sizeof(string))); + return myArray; + } + catch(exception& e) { + m->errorOut(e, "HeatMapSimCommand", "getValidParameters"); + exit(1); + } +} +//********************************************************************************************************************** +HeatMapSimCommand::HeatMapSimCommand(){ + try { + abort = true; calledHelp = true; + vector tempOutNames; + outputTypes["svg"] = tempOutNames; + } + catch(exception& e) { + m->errorOut(e, "HeatMapSimCommand", "HeatMapSimCommand"); + exit(1); + } +} +//********************************************************************************************************************** +vector HeatMapSimCommand::getRequiredParameters(){ + try { + vector myArray; + return myArray; + } + catch(exception& e) { + m->errorOut(e, "HeatMapSimCommand", "getRequiredParameters"); + exit(1); + } +} +//********************************************************************************************************************** +vector HeatMapSimCommand::getRequiredFiles(){ + try { + vector myArray; + return myArray; + } + catch(exception& e) { + m->errorOut(e, "HeatMapSimCommand", "getRequiredFiles"); + exit(1); + } +} //********************************************************************************************************************** HeatMapSimCommand::HeatMapSimCommand(string option) { try { globaldata = GlobalData::getInstance(); - abort = false; + abort = false; calledHelp = false; allLines = 1; labels.clear(); Groups.clear(); Estimators.clear(); //allow user to run help - if(option == "help") { validCalculator = new ValidCalculators(); help(); abort = true; } + if(option == "help") { validCalculator = new ValidCalculators(); help(); abort = true; calledHelp = true; } else { //valid paramters for this command @@ -50,6 +95,10 @@ HeatMapSimCommand::HeatMapSimCommand(string option) { if (validParameter.isValidParameter(it->first, myArray, it->second) != true) { abort = true; } } + //initialize outputTypes + vector tempOutNames; + outputTypes["svg"] = tempOutNames; + format = ""; //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 = ""; } @@ -62,7 +111,7 @@ HeatMapSimCommand::HeatMapSimCommand(string option) { it = parameters.find("phylip"); //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["phylip"] = inputDir + it->second; } } @@ -70,7 +119,7 @@ HeatMapSimCommand::HeatMapSimCommand(string option) { it = parameters.find("column"); //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["column"] = inputDir + it->second; } } @@ -78,7 +127,7 @@ HeatMapSimCommand::HeatMapSimCommand(string option) { it = parameters.find("name"); //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["name"] = inputDir + it->second; } } @@ -88,12 +137,12 @@ HeatMapSimCommand::HeatMapSimCommand(string option) { phylipfile = validParameter.validFile(parameters, "phylip", true); if (phylipfile == "not open") { abort = true; } else if (phylipfile == "not found") { phylipfile = ""; } - else { format = "phylip"; if (outputDir == "") { outputDir += hasPath(phylipfile); } } + else { format = "phylip"; if (outputDir == "") { outputDir += m->hasPath(phylipfile); } } columnfile = validParameter.validFile(parameters, "column", true); if (columnfile == "not open") { abort = true; } else if (columnfile == "not found") { columnfile = ""; } - else { format = "column"; if (outputDir == "") { outputDir += hasPath(columnfile); } } + else { format = "column"; if (outputDir == "") { outputDir += m->hasPath(columnfile); } } namefile = validParameter.validFile(parameters, "name", true); if (namefile == "not open") { abort = true; } @@ -113,12 +162,12 @@ HeatMapSimCommand::HeatMapSimCommand(string option) { //check for optional parameter and set defaults // ...at some point should added some additional type checking... if (format == "shared") { - if (outputDir == "") { outputDir += hasPath(globaldata->getSharedFile()); } + if (outputDir == "") { outputDir += m->hasPath(globaldata->getSharedFile()); } label = validParameter.validFile(parameters, "label", false); if (label == "not found") { label = ""; } else { - if(label != "all") { splitAtDash(label, labels); allLines = 0; } + if(label != "all") { m->splitAtDash(label, labels); allLines = 0; } else { allLines = 1; } } @@ -133,12 +182,12 @@ HeatMapSimCommand::HeatMapSimCommand(string option) { else { if (calc == "default") { calc = "jest-thetayc"; } } - splitAtDash(calc, Estimators); + m->splitAtDash(calc, Estimators); groups = validParameter.validFile(parameters, "groups", false); if (groups == "not found") { groups = ""; } else { - splitAtDash(groups, Groups); + m->splitAtDash(groups, Groups); globaldata->Groups = Groups; } } @@ -223,7 +272,7 @@ HeatMapSimCommand::~HeatMapSimCommand(){} int HeatMapSimCommand::execute(){ try { - if (abort == true) { return 0; } + if (abort == true) { if (calledHelp) { return 0; } return 2; } heatmap = new HeatMapSim(outputDir); @@ -240,6 +289,8 @@ int HeatMapSimCommand::execute(){ delete heatmap; delete validCalculator; + if (m->control_pressed) { for (int i = 0; i < outputNames.size(); i++) { remove(outputNames[i].c_str()); } outputTypes.clear(); 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(); } @@ -262,31 +313,35 @@ int HeatMapSimCommand::runCommandShared() { //you have groups read = new ReadOTUFile(globaldata->inputFileName); read->read(&*globaldata); - + input = globaldata->ginput; lookup = input->getSharedRAbundVectors(); string lastLabel = lookup[0]->getLabel(); - + if (lookup.size() < 2) { m->mothurOut("You have not provided enough valid groups. I cannot run the command."); m->mothurOutEndLine(); return 0;} //if the users enters label "0.06" and there is no "0.06" in their file use the next lowest label. set processedLabels; set userLabels = labels; + if (m->control_pressed) { delete read; delete input; globaldata->ginput = NULL; for (int i = 0; i < lookup.size(); i++) { delete lookup[i]; } globaldata->Groups.clear(); return 0; } + //as long as you are not at the end of the file or done wih the lines you want while((lookup[0] != NULL) && ((allLines == 1) || (userLabels.size() != 0))) { - + + if (m->control_pressed) { delete read; delete input; globaldata->ginput = NULL; for (int i = 0; i < lookup.size(); i++) { delete lookup[i]; } globaldata->Groups.clear(); return 0; } + if(allLines == 1 || labels.count(lookup[0]->getLabel()) == 1){ m->mothurOut(lookup[0]->getLabel()); m->mothurOutEndLine(); vector outfilenames = heatmap->getPic(lookup, heatCalculators); - for(int i = 0; i < outfilenames.size(); i++) { outputNames.push_back(outfilenames[i]); } + for(int i = 0; i < outfilenames.size(); i++) { outputNames.push_back(outfilenames[i]); outputTypes["svg"].push_back(outfilenames[i]); } processedLabels.insert(lookup[0]->getLabel()); userLabels.erase(lookup[0]->getLabel()); } - if ((anyLabelsToProcess(lookup[0]->getLabel(), userLabels, "") == true) && (processedLabels.count(lastLabel) != 1)) { + if ((m->anyLabelsToProcess(lookup[0]->getLabel(), userLabels, "") == true) && (processedLabels.count(lastLabel) != 1)) { string saveLabel = lookup[0]->getLabel(); for (int i = 0; i < lookup.size(); i++) { delete lookup[i]; } @@ -294,7 +349,7 @@ int HeatMapSimCommand::runCommandShared() { m->mothurOut(lookup[0]->getLabel()); m->mothurOutEndLine(); vector outfilenames = heatmap->getPic(lookup, heatCalculators); - for(int i = 0; i < outfilenames.size(); i++) { outputNames.push_back(outfilenames[i]); } + for(int i = 0; i < outfilenames.size(); i++) { outputNames.push_back(outfilenames[i]); outputTypes["svg"].push_back(outfilenames[i]); } processedLabels.insert(lookup[0]->getLabel()); userLabels.erase(lookup[0]->getLabel()); @@ -309,9 +364,13 @@ int HeatMapSimCommand::runCommandShared() { //get next line to process for (int i = 0; i < lookup.size(); i++) { delete lookup[i]; } - lookup = input->getSharedRAbundVectors(); + lookup = input->getSharedRAbundVectors(); + } + + if (m->control_pressed) { delete read; delete input; globaldata->ginput = NULL; globaldata->Groups.clear(); return 0; } + //output error messages about any remaining user labels set::iterator it; bool needToRun = false; @@ -325,6 +384,8 @@ int HeatMapSimCommand::runCommandShared() { } } + if (m->control_pressed) { delete read; delete input; globaldata->ginput = NULL; globaldata->Groups.clear(); return 0; } + //run last label if you need to if (needToRun == true) { for (int i = 0; i < lookup.size(); i++) { if (lookup[i] != NULL) { delete lookup[i]; } } @@ -332,18 +393,19 @@ int HeatMapSimCommand::runCommandShared() { m->mothurOut(lookup[0]->getLabel()); m->mothurOutEndLine(); vector outfilenames = heatmap->getPic(lookup, heatCalculators); - for(int i = 0; i < outfilenames.size(); i++) { outputNames.push_back(outfilenames[i]); } + for(int i = 0; i < outfilenames.size(); i++) { outputNames.push_back(outfilenames[i]); outputTypes["svg"].push_back(outfilenames[i]); } for (int i = 0; i < lookup.size(); i++) { delete lookup[i]; } } + if (m->control_pressed) { delete read; delete input; globaldata->ginput = NULL; globaldata->Groups.clear(); return 0; } //reset groups parameter globaldata->Groups.clear(); - + delete input; globaldata->ginput = NULL; delete read; - + return 0; } catch(exception& e) { @@ -362,7 +424,7 @@ int HeatMapSimCommand::runCommandDist() { //read distance file and create distance vector and names vector if (format == "phylip") { //read phylip file - openInputFile(phylipfile, in); + m->openInputFile(phylipfile, in); string name; int numSeqs; @@ -407,8 +469,10 @@ int HeatMapSimCommand::runCommandDist() { in >> name; names.push_back(name); - for(int j=0;j> matrix[i][j]; } - gobble(in); + if (m->control_pressed) { return 0; } + + for(int j=0;j> matrix[i][j]; } + m->gobble(in); } }else { double dist; @@ -416,11 +480,13 @@ int HeatMapSimCommand::runCommandDist() { in >> name; names.push_back(name); + if (m->control_pressed) { return 0; } + for(int j=0;j> dist; matrix[i][j] = dist; matrix[j][i] = dist; } - gobble(in); + m->gobble(in); } } in.close(); @@ -443,10 +509,12 @@ int HeatMapSimCommand::runCommandDist() { //read column file string first, second; double dist; - openInputFile(columnfile, in); + m->openInputFile(columnfile, in); while (!in.eof()) { - in >> first >> second >> dist; gobble(in); + in >> first >> second >> dist; m->gobble(in); + + if (m->control_pressed) { return 0; } map::iterator itA = nameMap->find(first); map::iterator itB = nameMap->find(second); @@ -463,8 +531,10 @@ int HeatMapSimCommand::runCommandDist() { delete nameMap; } - - outputNames.push_back(heatmap->getPic(matrix, names)); //vector>, vector + + string outputFileName = heatmap->getPic(matrix, names); + outputNames.push_back(outputFileName); //vector>, vector + outputTypes["svg"].push_back(outputFileName); return 0; }