]> git.donarmstrong.com Git - mothur.git/blobdiff - maligner.cpp
chimera.slayer debugging
[mothur.git] / maligner.cpp
index 009f1e923c9fd7449a70409867e4cffe8bc7763b..1205cef5e36b8bb4381f5ae8bb516c30e934318d 100644 (file)
@@ -73,18 +73,47 @@ string Maligner::chimeraMaligner(int chimeraPenalty, DeCalculator* decalc) {
                if (query->getAligned() == "") { return "no"; }
 
                vector<Sequence*> temp = refSeqs;
+               
+//             for(int i=0;i<refSeqs.size();i++){
+//                     cout << refSeqs[i]->getName() << endl;
+//             }
+               
                temp.push_back(query);
                        
                verticalFilter(temp);
                
-               //for (int i = 0; i < refSeqs.size(); i++) { cout << refSeqs[i]->getName() << endl << refSeqs[i]->getAligned() << endl; }
+               //for (int i = 0; i < refSeqs.size(); i++) { cout << refSeqs[i]->getName() << endl ; }//<< refSeqs[i]->getAligned() << endl
 
                vector< vector<score_struct> > matrix = buildScoreMatrix(query->getAligned().length(), refSeqs.size()); //builds and initializes
                
                if (m->control_pressed) { return chimera;  }
                
                fillScoreMatrix(matrix, refSeqs, chimeraPenalty);
-       
+               
+               vector<score_struct> path = extractHighestPath(matrix);
+               
+               if (m->control_pressed) { return chimera;  }
+               
+               vector<trace_struct> trace = mapTraceRegionsToAlignment(path, refSeqs);
+                       
+               if (trace.size() > 1) {         chimera = "yes";        }
+               else { chimera = "no";  return chimera; }
+               
+               int traceStart = path[0].col;
+               int traceEnd = path[path.size()-1].col; 
+               string queryInRange = query->getAligned();
+               queryInRange = queryInRange.substr(traceStart, (traceEnd-traceStart+1));
+//             cout << queryInRange << endl;
+               string chimeraSeq = constructChimericSeq(trace, refSeqs);
+//             cout << chimeraSeq << endl;
+               
+//             cout << queryInRange.length() << endl;
+//             cout << chimeraSeq.length() << endl;
+               
+               percentIdenticalQueryChimera = computePercentID(queryInRange, chimeraSeq);
+               
+//             cout << percentIdenticalQueryChimera << endl;
+       /*      
                vector<trace_struct> trace = extractHighestPath(matrix);
                                
                //cout << "traces\n";
@@ -98,7 +127,7 @@ string Maligner::chimeraMaligner(int chimeraPenalty, DeCalculator* decalc) {
                int traceStart = trace[0].col;
                int traceEnd = trace[trace.size()-1].oldCol;    
                string queryInRange = query->getAligned();
-               queryInRange = queryInRange.substr(traceStart, (traceEnd-traceStart));
+               queryInRange = queryInRange.substr(traceStart, (traceEnd-traceStart));*/
                
                if (m->control_pressed) { return chimera;  }
                
@@ -108,6 +137,7 @@ string Maligner::chimeraMaligner(int chimeraPenalty, DeCalculator* decalc) {
                        int regionEnd = trace[i].oldCol;
                        int seqIndex = trace[i].row;
                        
+//                     cout << regionStart << '\t' << regionEnd << '\t' << seqIndex << endl;
                        results temp;
                        
                        temp.parent = refSeqs[seqIndex]->getName();
@@ -118,20 +148,21 @@ string Maligner::chimeraMaligner(int chimeraPenalty, DeCalculator* decalc) {
                        temp.regionEnd = regionEnd;
                        
                        string parentInRange = refSeqs[seqIndex]->getAligned();
-                       parentInRange = parentInRange.substr(traceStart, (traceEnd-traceStart));
+                       parentInRange = parentInRange.substr(traceStart, (traceEnd-traceStart+1));
                        
                        temp.queryToParent = computePercentID(queryInRange, parentInRange);
                        temp.divR = (percentIdenticalQueryChimera / temp.queryToParent);
 
                        string queryInRegion = query->getAligned();
-                       queryInRegion = queryInRegion.substr(regionStart, (regionEnd-regionStart));
+                       queryInRegion = queryInRegion.substr(regionStart, (regionEnd-regionStart+1));
                        
                        string parentInRegion = refSeqs[seqIndex]->getAligned();
-                       parentInRegion = parentInRegion.substr(regionStart, (regionEnd-regionStart));
+                       parentInRegion = parentInRegion.substr(regionStart, (regionEnd-regionStart+1));
                        
                        temp.queryToParentLocal = computePercentID(queryInRegion, parentInRegion);
                        
-                       cout << temp.parent << '\t' << "NAST:" << temp.nastRegionStart << '-' << temp.nastRegionEnd << " G:" << temp.queryToParent << " L:" << temp.queryToParentLocal << endl;
+//                     cout << query->getName() << '\t' << temp.parent << '\t' << "NAST:" << temp.nastRegionStart << '-' << temp.nastRegionEnd << " G:" << temp.queryToParent << " L:" << temp.queryToParentLocal << ", " <<  temp.divR << endl;
+
                        outputResults.push_back(temp);
                }
                
@@ -294,6 +325,8 @@ void Maligner::fillScoreMatrix(vector<vector<score_struct> >& ms, vector<Sequenc
                int numCols = query->getAligned().length();
                int numRows = seqs.size();
                
+//             cout << numRows << endl;
+               
                //initialize first col
                string queryAligned = query->getAligned();
                for (int i = 0; i < numRows; i++) {
@@ -315,6 +348,7 @@ void Maligner::fillScoreMatrix(vector<vector<score_struct> >& ms, vector<Sequenc
                //fill rest of matrix
                for (int j = 1; j < numCols; j++) {  //iterate through matrix columns
                
+//                     for (int i = 0; i < 1; i++) {  //iterate through matrix rows
                        for (int i = 0; i < numRows; i++) {  //iterate through matrix rows
                                
                                string subjectAligned = seqs[i]->getAligned();
@@ -324,16 +358,13 @@ void Maligner::fillScoreMatrix(vector<vector<score_struct> >& ms, vector<Sequenc
                                if ((!isalpha(queryAligned[j])) && (!isalpha(subjectAligned[j]))) {
                                        //leave the same
                                }else if ((toupper(queryAligned[j]) == 'N') || (toupper(subjectAligned[j]) == 'N')) {
-                                       //matchMisMatchScore = matchScore;
                                        //leave the same
                                }else if (queryAligned[j] == subjectAligned[j]) {
                                        matchMisMatchScore = matchScore;
-//                                     ms[i][j].mismatches = ms[i][j-1].mismatches;
                                }else if (queryAligned[j] != subjectAligned[j]) {
                                        matchMisMatchScore = misMatchPenalty;
-//                                     ms[i][j].mismatches = ms[i][j-1].mismatches + 1;
                                }
-                               
+
                                //compute score based on previous columns scores
                                for (int prevIndex = 0; prevIndex < numRows; prevIndex++) { //iterate through rows
                                        
@@ -348,45 +379,136 @@ void Maligner::fillScoreMatrix(vector<vector<score_struct> >& ms, vector<Sequenc
                                                ms[i][j].prev = prevIndex;
                                        }
                                }
+//                             cout << i << '\t' << j << '\t' << queryAligned[j] << '\t' << subjectAligned[j] << '\t' << matchMisMatchScore << '\t' << ms[i][j].score << endl;
+
                        }
+                       
                }
                
-       /*      for(int i=0;i<numRows;i++){
-                       cout << seqs[i]->getName();
-                       for(int j=0;j<numCols;j++){
-                               cout << '\t' << ms[i][j].mismatches;
+               
+               
+               
+               
+//             cout << numRows << '\t' << numCols << endl;
+//             for(int i=0;i<numRows;i++){
+//                     cout << seqs[i]->getName();
+//                     for(int j=0;j<numCols;j++){
+//                             cout << '\t' << ms[i][j].score;
+//                     }
+//                     cout << endl;
+//             }
+//             cout << endl;
+//             
+//             for(int i=0;i<numRows;i++){
+//                     cout << seqs[i]->getName();
+//                     for(int j=0;j<numCols;j++){
+//                             cout << '\t' << ms[i][j].prev;
+//                     }
+//                     cout << endl;
+//             }
+               
+               
+       }
+       catch(exception& e) {
+               m->errorOut(e, "Maligner", "fillScoreMatrix");
+               exit(1);
+       }
+}
+//***************************************************************************************************************
+vector<score_struct> Maligner::extractHighestPath(vector<vector<score_struct> > ms) {
+       try {
+               
+               //get matrix dimensions
+               int numCols = query->getAligned().length();
+               int numRows = ms.size();
+               
+               
+               //find highest score scoring matrix
+               score_struct highestStruct;
+               int highestScore = 0;
+               
+               for (int i = 0; i < numRows; i++) {
+                       for (int j = 0; j < numCols; j++) {
+                               if (ms[i][j].score > highestScore) {
+                                       highestScore = ms[i][j].score;
+                                       highestStruct = ms[i][j];
+                               }
                        }
-                       cout << endl;
                }
-               cout << endl;*/
-               /*cout << numRows << '\t' << numCols << endl;
-               for(int i=0;i<numRows;i++){
-                       cout << seqs[i]->getName() << endl << seqs[i]->getAligned() << endl << endl;
-                       if ((seqs[i]->getName() == "S000003470") || (seqs[i]->getName() == "S000383265") || (seqs[i]->getName() == "7000004128191054")) {
-                       for(int j=0;j<numCols;j++){
-                               cout << '\t' << ms[i][j].score;
-                       }
-                       cout << endl;
-                       }
+               
+//             cout << highestScore << endl;
+               vector<score_struct> path;
+               
+               int rowIndex = highestStruct.row;
+               int pos = highestStruct.col;
+               int score = highestStruct.score;
+               
+               while (pos >= 0 && score > 0) {
+                       score_struct temp = ms[rowIndex][pos];
+                       score = temp.score;
+                       
+                       if (score > 0) {        path.push_back(temp);   }
+                       
+                       rowIndex = temp.prev;
+                       pos--;
                }
-               cout << endl;*/
-               /*for(int i=0;i<numRows;i++){
-                       cout << seqs[i]->getName();
-                       for(int j=0;j<numCols;j++){
-                               cout << '\t' << ms[i][j].prev;
+               
+               reverse(path.begin(), path.end());
+               
+               return path;
+               
+       }
+       catch(exception& e) {
+               m->errorOut(e, "Maligner", "extractHighestPath");
+               exit(1);
+       }
+}
+//***************************************************************************************************************
+vector<trace_struct> Maligner::mapTraceRegionsToAlignment(vector<score_struct> path, vector<Sequence*> seqs) {
+       try {
+               vector<trace_struct> trace;
+               
+               int region_index = path[0].row;
+               int region_start = path[0].col;
+               
+               for (int i = 1; i < path.size(); i++) {
+                       
+                       int next_region_index = path[i].row;
+//                     cout << i << '\t' << next_region_index << endl;
+
+                       if (next_region_index != region_index) {
+                               
+                               // add trace region
+                               int col_index = path[i].col;
+                               trace_struct temp;
+                               temp.col = region_start;
+                               temp.oldCol = col_index-1;
+                               temp.row = region_index;
+                               
+                               trace.push_back(temp);
+                               
+                               region_index = path[i].row;
+                               region_start = col_index;
                        }
-                       cout << endl;
-               }*/
+               }
                
+               // get last one
+               trace_struct temp;
+               temp.col = region_start;
+               temp.oldCol = path[path.size()-1].col;
+               temp.row = region_index;
+               trace.push_back(temp);
+               
+               return trace;
                
        }
        catch(exception& e) {
-               m->errorOut(e, "Maligner", "fillScoreMatrix");
+               m->errorOut(e, "Maligner", "mapTraceRegionsToAlignment");
                exit(1);
        }
 }
 
-//***************************************************************************************************************
+/***************************************************************************************************************
 
 vector<trace_struct> Maligner::extractHighestPath(vector<vector<score_struct> > ms) {
        try {
@@ -527,7 +649,7 @@ vector<trace_struct> Maligner::mapTraceRegionsToAlignment(vector<score_struct> p
                exit(1);
        }
 }
-
+*/
 //***************************************************************************************************************
 
 string Maligner::constructChimericSeq(vector<trace_struct> trace, vector<Sequence*> seqs) {
@@ -541,7 +663,9 @@ string Maligner::constructChimericSeq(vector<trace_struct> trace, vector<Sequenc
                        seqAlign = seqAlign.substr(trace[i].col, (trace[i].oldCol-trace[i].col+1));
                        chimera += seqAlign;
                }
-                       
+//             cout << chimera << endl;
+//             if (chimera != "") { chimera = chimera.substr(0, (chimera.length()-1)); }       //this was introducing a fence post error
+//             cout << chimera << endl;
                return chimera;
        }
        catch(exception& e) {
@@ -580,11 +704,12 @@ float Maligner::computePercentID(string queryAlign, string chimera) {
        
                if (queryAlign.length() != chimera.length()) {
                        m->mothurOut("Error, alignment strings are of different lengths: "); m->mothurOutEndLine();
-                       m->mothurOut(toString(queryAlign.length())); m->mothurOutEndLine(); m->mothurOutEndLine(); m->mothurOutEndLine(); m->mothurOutEndLine();
+                       m->mothurOut(toString(queryAlign.length())); m->mothurOutEndLine(); 
                        m->mothurOut(toString(chimera.length())); m->mothurOutEndLine();
                        return -1.0;
                }
 
+//             cout << queryAlign.length() << endl;
                int numIdentical = 0;
                int countA = 0;
                int countB = 0;
@@ -597,6 +722,7 @@ float Maligner::computePercentID(string queryAlign, string chimera) {
                                if ((queryAlign[i] == 'G') || (queryAlign[i] == 'T') || (queryAlign[i] == 'A') || (queryAlign[i] == 'C')) { charA = true; }
                                if ((chimera[i] == 'G') || (chimera[i] == 'T') || (chimera[i] == 'A') || (chimera[i] == 'C')) { charB = true; }
                                        
+                               
                                if (charA || charB) {
                                                
                                        if (charA) { countA++; }
@@ -606,15 +732,24 @@ float Maligner::computePercentID(string queryAlign, string chimera) {
                                                numIdentical++;
                                        }
                                }
+//                             cout << queryAlign[i] << '\t' << chimera[i] << '\t' << countA << '\t' << countB << endl;
+
                        }
                }
                
+//             cout << "pat\t" << countA << '\t' << countB << '\t' << numIdentical << endl;
+
+               
                float numBases = (countA + countB) /(float) 2;
                
                if (numBases == 0) { return 0; }
        
+//             cout << numIdentical << '\t' << numBases << endl;
+               
                float percentIdentical = (numIdentical/(float)numBases) * 100;
-
+               
+//             cout << percentIdentical << endl;
+               
                return percentIdentical;
                
        }