Exemplo n.º 1
0
Arquivo: example.cpp Projeto: alecw/TS
int main(int argc, char* argv[])
{
	// Open a BAM file:
	char* bamFile  = argv[1];
	//	char* bamIndex = argv[2];

	BAMReader reader(bamFile); // , bamIndex);
	reader.open();
	assert(reader);

	// Print out list of reference sequences, and their lengths:
	cout << "Found " << reader.num_refs() << " reference sequences:" << endl;
	for(int i=0; i<reader.num_refs(); ++i)
		cout << setw(9) << reader.refs()[i] << "    " << reader.lens()[i] << endl;

	// Print out list of reads:
	for (BAMReader::iterator i = reader.get_iterator(); i.good(); i.next()) {		
	  BAMRead read = i.get();
	  cout << read.to_string();
	  for (Sequence::iterator s_iter = read.get_seq().get_iterator(); s_iter.good(); s_iter.next())
	    cout << s_iter.get(); // nuc from SEQ
	  cout << endl;

	  for (Cigar::iterator c_iter = read.get_cigar().get_iterator(); c_iter.good(); c_iter.next())
	    cout << c_iter.len() << ":" << c_iter.op() << "; ";
	  cout << endl;

	  // Don't use MD directly.  Use BAMUtils.
	  BAMUtils utils(read);
	  cout << utils.get_qdna() << endl << utils.get_matcha() << endl << utils.get_tdna() << endl << endl;

	}
}
Exemplo n.º 2
0
int main(int argc, const char *argv[])
{
#ifdef _DEBUG
  atexit(memstatus);
  dbgmemInit();
#endif /* _DEBUG */

  printf ("%s - %s-%s (%s)\n", argv[0], IonVersion::GetVersion().c_str(), IonVersion::GetRelease().c_str(), IonVersion::GetSvnRev().c_str());

  string bamInputFilename;
  string fastaInputFilename;
  string jsonOutputFilename;
  bool help;

  OptArgs opts;
  opts.ParseCmdLine(argc, argv);
  opts.GetOption(bamInputFilename,    "",             '-',  "bam");
  opts.GetOption(fastaInputFilename,  "",             '-',  "ref");
  opts.GetOption(jsonOutputFilename,  "TFStats.json", '-',  "output-json");
  opts.GetOption(help,                "false",        'h',  "help");
  opts.CheckNoLeftovers();

  if (help || bamInputFilename.empty() || fastaInputFilename.empty())
    return showHelp();


  // Parse BAM header

  BAMReader bamReader(bamInputFilename);
  bamReader.open();
  bam_header_t *header = (bam_header_t *)bamReader.get_header_ptr();

  int numFlows = 0;
  string flowOrder;
  string key;

  if (header->l_text >= 3) {
    if (header->dict == 0)
      header->dict = sam_header_parse2(header->text);
    int nEntries = 0;
    char **tmp = sam_header2list(header->dict, "RG", "FO", &nEntries);
    if (nEntries) {
      flowOrder = tmp[0];
      numFlows = flowOrder.length();
    }
    if (tmp)
      free(tmp);
    nEntries = 0;
    tmp = sam_header2list(header->dict, "RG", "KS", &nEntries);
    if (nEntries) {
      key = tmp[0];
    }
    if (tmp)
      free(tmp);
  }

  if (numFlows <= 0) {
    fprintf(stderr, "[TFMapper] Could not retrieve flow order from FO BAM tag. SFF-specific tags absent?\n");
    exit(1);
  }
  if (key.empty()) {
    fprintf(stderr, "[TFMapper] Could not retrieve key sequence from KS BAM tag. SFF-specific tags absent?\n");
    exit(1);
  }
  //printf("Retrieved flow order from bam: %s (%d)\n", flowOrder.c_str(), numFlows);
  //printf("Retrieved key from bam: %s\n", key.c_str());


  // Retrieve test fragment sequences

  vector<string>  referenceSequences;
  PopulateReferenceSequences(referenceSequences, fastaInputFilename, header->n_targets, header->target_name, string(""));


  //  Process the BAM reads and generate metrics

  int numTFs = header->n_targets;
  vector<int>     TFCount(numTFs,0);
  MetricGeneratorQualityHistograms  metricGeneratorQualityHistograms[numTFs];
  MetricGeneratorHPAccuracy         metricGeneratorHPAccuracy[numTFs];
  MetricGeneratorSNR                metricGeneratorSNR[numTFs];
  MetricGeneratorAvgIonogram        metricGeneratorAvgIonogram[numTFs];

  for (BAMReader::iterator i = bamReader.get_iterator(); i.good(); i.next()) {

    BAMRead bamRead = i.get();
    int bestTF = bamRead.get_tid();
    if (bestTF < 0)
      continue;
    BAMUtils bamUtil(bamRead);
    TFCount[bestTF]++;

    // Extract flowspace signal from FZ BAM tag

    uint16_t *bam_flowgram = NULL;
    uint8_t *fz = bam_aux_get(bamRead.get_bam_ptr(), "FZ");
    if (fz != NULL) {
      if (fz[0] == (uint8_t)'B' && fz[1] == (uint8_t)'S' && *((uint32_t *)(fz+2)) == (uint32_t)numFlows)
        bam_flowgram = (uint16_t *)(fz+6);
    }
    if (bam_flowgram == NULL) {
      fprintf(stderr, "[TFMapper] Could not retrieve flow signal from FZ BAM tag. SFF-specific tags absent?\n");
      exit(1);
    }


    // Use alignments to generate "synchronized" flowspace reference and read ionograms
    // TODO: Do proper flowspace alignment

    string genome = key + bamUtil.get_tdna();
    string calls = key + bamUtil.get_qdna();

    int numBases = min(genome.length(),calls.length());
    vector<int> refIonogram(numFlows, 0);
    vector<int> readIonogram(numFlows, 0);

    int numFlowsRead = 0;
    int numFlowsRef = 0;
    char gC = flowOrder[0];
    int gBC = 0;

    for (int iBase = 0; (iBase < numBases) && (numFlowsRead < numFlows) && (numFlowsRef < numFlows); iBase++) {

      // Conversion for reads (independent of reference)
      if (calls[iBase] != '-') {
        while ((calls[iBase] != flowOrder[numFlowsRead]) && (numFlowsRead < numFlows))
          numFlowsRead++;
        if (numFlowsRead < numFlows)
          readIonogram[numFlowsRead]++;
      }

      if (genome[iBase] != '-') {

        if (genome[iBase] != gC) {
          // Since a new homopolymer begins, need to drop off the old one
          while ((gC != flowOrder[numFlowsRef]) && (numFlowsRef < numFlows)) {
            numFlowsRef++;
            if (numFlowsRef < numFlows)
              refIonogram[numFlowsRef] = 0;
          }
          if (numFlowsRef < numFlows)
            refIonogram[numFlowsRef] = gBC;

          gC = genome[iBase];
          gBC = 0;
        }
        gBC++;

        if (genome[iBase] == calls[iBase])
          numFlowsRef = numFlowsRead;
      }
    }

    int validFlows = min(numFlowsRef, numFlowsRead);


    metricGeneratorSNR[bestTF].AddElement(bam_flowgram ,key.c_str(), flowOrder);
    metricGeneratorAvgIonogram[bestTF].AddElement(bam_flowgram, numFlows);
    metricGeneratorQualityHistograms[bestTF].AddElement(bamUtil.get_phred_len(10),bamUtil.get_phred_len(17));
    for (int iFlow = 0; iFlow < validFlows-20; iFlow++)
      metricGeneratorHPAccuracy[bestTF].AddElement(refIonogram[iFlow],readIonogram[iFlow]);
  }


  // Save stats to a json file

  Json::Value outputJson(Json::objectValue);

  for(int i = 0; i < numTFs; i++) {
    if (TFCount[i] < minTFCount)
      continue;

    Json::Value currentTFJson(Json::objectValue);
    currentTFJson["TF Name"] = header->target_name[i];
    currentTFJson["TF Seq"] = referenceSequences[i];
    currentTFJson["Num"] = TFCount[i];
    currentTFJson["Top Reads"] = Json::Value(Json::arrayValue); // Obsolete

    metricGeneratorSNR[i].PrintSNR(currentTFJson);
    metricGeneratorHPAccuracy[i].PrintHPAccuracy(currentTFJson);
    metricGeneratorQualityHistograms[i].PrintMetrics(currentTFJson);
    metricGeneratorAvgIonogram[i].PrintIonograms(currentTFJson);

    outputJson[header->target_name[i]] = currentTFJson;
  }

  bamReader.close();  // Closing invalidates the header pointers

  if (!jsonOutputFilename.empty()) {
    ofstream out(jsonOutputFilename.c_str(), ios::out);
    if (out.good())
      out << outputJson.toStyledString();
  }

  return 0;
}