Exemplo n.º 1
0
  /**
   * This need to be thread safe
   *
   * @param seq
   * @param outStream
   * @throws IOException
   */
  private void processReads(Sequence seq, PrintStream outStream) throws IOException {
    NuclKmerGenerator kmerGenerator;
    Kmer kmer;

    boolean found = false;
    kmerGenerator = new NuclKmerGenerator(seq.getSeqString(), kmerSize);
    while (kmerGenerator.hasNext()) {

      kmer = kmerGenerator.next();
      for (int i = 0; i < kmerMaps.length; i++) { // for forward and reverse direction
        KmerAbund kmerAbund = kmerMaps[i].get(kmer);
        if (kmerAbund != null) {
          // increment the count
          kmerAbund.count.addAndGet(1);
          found = true;
        }
      }
    }
    if (found) {
      totalReads.incrementAndGet();
    }
    if (outStream != null && found) {
      writeSeq(seq, outStream);
    }
  }
Exemplo n.º 2
0
  /**
   * find the kmers in the contigs
   *
   * @param reader
   * @throws IOException
   */
  private void processContigFile(SequenceReader reader) throws IOException {
    Sequence seq;
    NuclKmerGenerator kmerGenerator;
    Kmer kmer;
    int contigIdx = 0;
    while ((seq = reader.readNextSequence()) != null) {
      if (seq.getSeqString().length() < kmerSize) {
        continue;
      }
      // use int to represent seqname in case contig names are too long
      contigMap.put(
          contigIdx, new Contig(seq.getSeqName(), seq.getSeqString().length() - kmerSize + 1));
      // forward direction
      kmerGenerator = new NuclKmerGenerator(seq.getSeqString(), kmerSize);
      while (kmerGenerator.hasNext()) {
        kmer = kmerGenerator.next();
        KmerAbund kmerAbund = kmerMaps[0].get(kmer);

        if (kmerAbund == null) {
          kmerAbund = new KmerAbund();
          kmerMaps[0].put(kmer, kmerAbund);
        }
        kmerAbund.contigList.add(new ContigCoverage(contigIdx, kmerGenerator.getPosition() - 1));
      }

      // reverse direction
      kmerGenerator =
          new NuclKmerGenerator(IUBUtilities.reverseComplement(seq.getSeqString()), kmerSize);
      while (kmerGenerator.hasNext()) {
        kmer = kmerGenerator.next();
        KmerAbund kmerAbund = kmerMaps[1].get(kmer);

        if (kmerAbund == null) {
          kmerAbund = new KmerAbund();
          kmerMaps[1].put(kmer, kmerAbund);
        }
        kmerAbund.contigList.add(
            new ContigCoverage(
                contigIdx,
                seq.getSeqString().length() - kmerGenerator.getPosition() - kmerSize + 1));
      }
      contigIdx++;
    }
    reader.close();
  }