test MotifModel and MotifElement

[mussa.git] / qui / motif_editor / MotifModel.cpp

diff --git a/qui/motif_editor/MotifModel.cpp b/qui/motif_editor/MotifModel.cpp
index 5bf0ad1db4b43445f714642c54baf8f28f325efc..d07798abd3f1241591b5892aa60ecf265f6cab89 100644 (file)
--- a/qui/motif_editor/MotifModel.cpp
+++ b/qui/motif_editor/MotifModel.cpp
@@ -2,8 +2,11 @@
 
 #include <QColor>
 
+const size_t MotifModel::model_column_size = 4;
+
 MotifModel::MotifModel(MussaRef m, QObject *parent) 
-  : QAbstractTableModel(parent)
+  : analysis(m),
+    QAbstractTableModel(parent)
 {
   const std::set<Sequence>& motif_set = m->motifs();
   boost::shared_ptr<AnnotationColors> cm = m->colorMapper();
@@ -20,6 +23,10 @@ MotifModel::MotifModel(MussaRef m, QObject *parent)
   push_empty();
 }
 
+const MussaRef MotifModel::getAnalysis() const {
+  return analysis;
+}
+
 void MotifModel::assign(
     MotifModel::size_type num, 
     const MotifElement& val
@@ -119,7 +126,7 @@ MotifModel::rowCount( const QModelIndex& parent) const
 int 
 MotifModel::columnCount(const QModelIndex& parent) const
 {
-  return 4;
+  return model_column_size;
 }
 
 QVariant 
@@ -131,7 +138,7 @@ MotifModel::data(const QModelIndex &index, int role) const
   if (index.row() >= motifs.size())
     return QVariant();
 
-  if (index.column() >= 4)
+  if (index.column() >= model_column_size)
     return QVariant();
 
   const MotifElement& motif = motifs[index.row()];