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                         cout << "Action: " << safe[0].toString() << " ";
  45                         al.push_back(safe[0]);
  46                 }else
  47                         return ActionList(); //TODO: Need to handle hurdles.
  48         }
  49         cout << "Distance: " << inversionDistance(temp) << " : ";
  50         copy(temp.begin(), temp.end(), ostream_iterator<int>(cout, " "));
  51         cout << endl;
  52         return al;
  53 }
  54
  55 struct ScoreCmp {
  56         template<typename T>
  57         bool operator()(T s1, T s2){
  58                 return s1.first < s2.first;
  59         }
  60 };
  61
  62 GeneSorter::ActionList GeneSorter::safeActions(const GeneOrder& go){
  63         if (countCycles(go) == go.size() - 1)
  64                 return ActionList();
  65         ActionList al;
  66         vector<Interval> intervals = findIntervals(go);
  67         priority_queue<pair<size_t,SortAction>,vector<pair<size_t,SortAction> >, ScoreCmp > pq;
  68         for (size_t i = 0; i < intervals.size(); ++i){
  69                 if (intervals[i].oriented && intervals[i].first != intervals[i].second){
  70                         SortAction sa(new ReverseAction(intervals[i]));
  71                         size_t score = scoreAction(go,sa);
  72                         cout << "Inversion: " << min(intervals[i].first,intervals[i].second) << ":" << max(intervals[i].first,intervals[i].second)-1 << " Score: " << score <<  endl;
  73                         pq.push(pair<size_t,SortAction>(score,sa));
  74                 }
  75         }
  76         while (pq.size() > 0){
  77                 al.push_back(pq.top().second);
  78                 pq.pop();
  79         }
  80         return al;
  81 }
  82
  83 size_t GeneSorter::scoreAction(const GeneOrder& go, SortAction& sa){
  84         GeneOrder temp(go);
  85         sa(temp);
  86         vector<Interval> intervals = findIntervals(temp);
  87         int o = 0;
  88         for (vector<Interval>::iterator in = intervals.begin(); in != intervals.end(); ++in){
  89                 if (in->oriented)
  90                         ++o;
  91         }
  92         return o;
  93 }
  94