src/genesorter.cpp

   1 /***************************************************************************
   2  *   Copyright (C) 2006 by Michael Andreen                                 *
   3  *   andreen@student.chalmers.se                                           *
   4  *                                                                         *
   5  *   This program is free software; you can redistribute it and/or modify  *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation; either version 2 of the License, or     *
   8  *   (at your option) any later version.                                   *
   9  *                                                                         *
  10  *   This program is distributed in the hope that it will be useful,       *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with this program; if not, write to the                         *
  17  *   Free Software Foundation, Inc.,                                       *
  18  *   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA          *
  19  ***************************************************************************/
  20
  21 #include "genesorter.h"
  22
  23 #include "geneorder.h"
  24 #include "sortaction.h"
  25 #include "reverseaction.h"
  26
  27 #include "genealgorithms.h"
  28
  29 #include <queue>
  30 #include <iostream>
  31 #include <iterator>
  32 using namespace std;
  33
  34 GeneSorter::ActionList GeneSorter::sort(const GeneOrder& go){
  35         ActionList al;
  36         GeneOrder temp(go);
  37         while(inversionDistance(temp) > 0){
  38                 cout << "Distance: " << inversionDistance(temp) << " : ";
  39                 copy(temp.begin(), temp.end(), ostream_iterator<int>(cout, " "));
  40                 cout << endl;
  41                 ActionList safe = safeActions(temp);
  42                 if (safe.size() > 0){
  43                         safe[0](temp);
  44                         al.push_back(safe[0]);
  45                 }else
  46                         return ActionList(); //TODO: Need to handle hurdles.
  47         }
  48         cout << "Distance: " << inversionDistance(temp) << " : ";
  49         copy(temp.begin(), temp.end(), ostream_iterator<int>(cout, " "));
  50         cout << endl;
  51         return al;
  52 }
  53
  54 struct ScoreCmp {
  55         template<typename T>
  56         bool operator()(T s1, T s2){
  57                 return s1.first < s2.first;
  58         }
  59 };
  60
  61 GeneSorter::ActionList GeneSorter::safeActions(const GeneOrder& go){
  62         if (countCycles(go) == go.size() - 1)
  63                 return ActionList();
  64         ActionList al;
  65         vector<Interval> intervals = findIntervals(go);
  66         priority_queue<pair<size_t,SortAction>,vector<pair<size_t,SortAction> >, ScoreCmp > pq;
  67         for (size_t i = 0; i < intervals.size(); ++i){
  68                 if (intervals[i].oriented && intervals[i].first != intervals[i].second){
  69                         SortAction sa(new ReverseAction(intervals[i]));
  70                         size_t score = scoreAction(go,sa);
  71                         cout << "Inversion: " << min(intervals[i].first,intervals[i].second) << ":" << max(intervals[i].first,intervals[i].second)-1 << " Score: " << score <<  endl;
  72                         pq.push(pair<size_t,SortAction>(score,sa));
  73                 }
  74         }
  75         while (pq.size() > 0){
  76                 al.push_back(pq.top().second);
  77                 pq.pop();
  78         }
  79         return al;
  80 }
  81
  82 size_t GeneSorter::scoreAction(const GeneOrder& go, SortAction& sa){
  83         GeneOrder temp(go);
  84         sa(temp);
  85         vector<Interval> intervals = findIntervals(temp);
  86         int o = 0;
  87         for (vector<Interval>::iterator in = intervals.begin(); in != intervals.end(); ++in){
  88                 if (in->oriented)
  89                         ++o;
  90         }
  91         return o;
  92 }
  93