findmatesalgo.cc

Go to the documentation of this file.

#include "findmatesalgo.h"
#include "algoparam.h"
#include <iostream>
#include "readdata.h"
#include "generaldata.h"
#include <cassert>
#include <qmessagebox.h>
#include <qinputdialog.h>
#include <limits>

using namespace std;

void FindMatesAlgo::start()
{
  if ( selectedReads.empty() ) {
    return;
  }

  int res = QMessageBox::question( 0, "Find mates", "All mates, or mates within an interval?", "All", "Interval", "Cancel", 0, 2 );
  
  size_t min_ = 0;
  size_t max_ = numeric_limits<size_t>::max();

  if( res == 2 ) {
    return;
  }
  else if ( res == 1 ) {
    bool ok;
    int low = QInputDialog::getInteger(
                                       "Find mates", "Lower bound", 0, 0, numeric_limits<int>::max(), 1,
                                       &ok, 0 );
    
    if ( ok ) {
      // user entered something and pressed OK
      min_ = low;
    } else {
      // user pressed Cancel
      return;
    }
    int high = QInputDialog::getInteger(
                                        "Find mates", "Upper bound", numeric_limits<int>::max(), low, numeric_limits<int>::max(), 1,
                                        &ok, 0 );
    
    if ( ok ) {
      // user entered something and pressed OK
      max_ = high;
    } else {
      // user pressed Cancel
      return;
    }
    
  }
  

  set< db_recno_t > selected_temp = selectedReads;
  
  selectedReads.clear();

  for (set< db_recno_t >::iterator it = selected_temp.begin(); it != selected_temp.end(); it++ ){

    Database::PrimaryIterator<ReadData>* read_it = new Database::PrimaryIterator<ReadData>(pDoc, "ReadData");

    int ret_read = read_it->setFromRecno(*it);
    ReadData* r_test = (ret_read != DB_NOTFOUND) ? read_it->answer() : 0;
    assert( r_test );
    

    //Find mate (if any) and get its recno
    
    Database::SecondaryIterator<ReadData>* name_it =
      new Database::SecondaryIterator<ReadData>( "name", pDoc, "ReadData" );
    name_it->key()->setName( r_test->mate() );
    if ( name_it->set() == 0 && 
         name_it->answer()->mateLength() >= min_ && 
         name_it->answer()->mateLength() <= max_ ) {
      selectedReads.insert(*it);
      selectedReads.insert(name_it->answer()->getRecno());
      
    }
    
    delete name_it;
    delete read_it;
    

  }
  
}

---