X-Git-Url: https://git.donarmstrong.com/?a=blobdiff_plain;f=matrixoutputcommand.cpp;h=05cd18a720e567e3e7c1852b436b68b1a38814f0;hb=050e1297eaf24fabbbe0e9b76c3a5acfb90eed7d;hp=c879d1b621b871a5f7a9e3dbf48df70a7d928320;hpb=2b11c674bb4d57c4542ecb89989e1590dd9f07a7;p=mothur.git diff --git a/matrixoutputcommand.cpp b/matrixoutputcommand.cpp index c879d1b..05cd18a 100644 --- a/matrixoutputcommand.cpp +++ b/matrixoutputcommand.cpp @@ -434,18 +434,18 @@ void MatrixOutputCommand::printSims(ostream& out, vector< vector >& simM out << simMatrix.size() << endl; if (output == "lt") { - for (int m = 0; m < simMatrix.size(); m++) { - out << lookup[m]->getGroup() << '\t'; - for (int n = 0; n < m; n++) { - out << simMatrix[m][n] << '\t'; + for (int b = 0; b < simMatrix.size(); b++) { + out << lookup[b]->getGroup() << '\t'; + for (int n = 0; n < b; n++) { + out << simMatrix[b][n] << '\t'; } out << endl; } }else{ - for (int m = 0; m < simMatrix.size(); m++) { - out << lookup[m]->getGroup() << '\t'; - for (int n = 0; n < simMatrix[m].size(); n++) { - out << simMatrix[m][n] << '\t'; + for (int b = 0; b < simMatrix.size(); m++) { + out << lookup[b]->getGroup() << '\t'; + for (int n = 0; n < simMatrix[b].size(); n++) { + out << simMatrix[b][n] << '\t'; } out << endl; } @@ -462,11 +462,11 @@ int MatrixOutputCommand::process(vector thisLookup){ vector< vector< vector > > calcDistsTotals; //each iter, one for each calc, then each groupCombos dists. this will be used to make .dist files vector< vector > calcDists; calcDists.resize(matrixCalculators.size()); - for (int thisIter = 0; thisIter < iters; thisIter++) { + for (int thisIter = 0; thisIter < iters+1; thisIter++) { vector thisItersLookup = thisLookup; - if (subsample) { + if (subsample && (thisIter != 0)) { SubSample sample; vector tempLabels; //dont need since we arent printing the sampled sharedRabunds @@ -619,14 +619,40 @@ int MatrixOutputCommand::process(vector thisLookup){ #endif } - calcDistsTotals.push_back(calcDists); - - if (subsample) { - + if (subsample && (thisIter != 0)) { + calcDistsTotals.push_back(calcDists); //clean up memory for (int i = 0; i < thisItersLookup.size(); i++) { delete thisItersLookup[i]; } thisItersLookup.clear(); for (int i = 0; i < calcDists.size(); i++) { calcDists[i].clear(); } + }else { //print results for whole dataset + for (int i = 0; i < calcDists.size(); i++) { + if (m->control_pressed) { break; } + + //initialize matrix + vector< vector > matrix; //square matrix to represent the distance + matrix.resize(thisLookup.size()); + for (int k = 0; k < thisLookup.size(); k++) { matrix[k].resize(thisLookup.size(), 0.0); } + + for (int j = 0; j < calcDists[i].size(); j++) { + int row = calcDists[i][j].seq1; + int column = calcDists[i][j].seq2; + double dist = calcDists[i][j].dist; + + matrix[row][column] = dist; + matrix[column][row] = dist; + } + + string distFileName = outputDir + m->getRootName(m->getSimpleName(sharedfile)) + matrixCalculators[i]->getName() + "." + thisLookup[0]->getLabel() + "." + output + ".dist"; + outputNames.push_back(distFileName); outputTypes["phylip"].push_back(distFileName); + ofstream outDist; + m->openOutputFile(distFileName, outDist); + outDist.setf(ios::fixed, ios::floatfield); outDist.setf(ios::showpoint); + + printSims(outDist, matrix); + + outDist.close(); + } } } @@ -729,35 +755,6 @@ int MatrixOutputCommand::process(vector thisLookup){ outSTD.close(); } - }else { - - for (int i = 0; i < calcDists.size(); i++) { - if (m->control_pressed) { break; } - - //initialize matrix - vector< vector > matrix; //square matrix to represent the distance - matrix.resize(thisLookup.size()); - for (int k = 0; k < thisLookup.size(); k++) { matrix[k].resize(thisLookup.size(), 0.0); } - - for (int j = 0; j < calcDists[i].size(); j++) { - int row = calcDists[i][j].seq1; - int column = calcDists[i][j].seq2; - double dist = calcDists[i][j].dist; - - matrix[row][column] = dist; - matrix[column][row] = dist; - } - - string distFileName = outputDir + m->getRootName(m->getSimpleName(sharedfile)) + matrixCalculators[i]->getName() + "." + thisLookup[0]->getLabel() + "." + output + ".dist"; - outputNames.push_back(distFileName); outputTypes["phylip"].push_back(distFileName); - ofstream outDist; - m->openOutputFile(distFileName, outDist); - outDist.setf(ios::fixed, ios::floatfield); outDist.setf(ios::showpoint); - - printSims(outDist, matrix); - - outDist.close(); - } } return 0;