SelectorModel.hpp

/* ***** BEGIN LICENSE BLOCK *****
 * FW4SPL - Copyright (C) IRCAD, 2009-2017.
 * Distributed under the terms of the GNU Lesser General Public License (LGPL) as
 * published by the Free Software Foundation.
 * ****** END LICENSE BLOCK ****** */

#ifndef __UIMEDDATAQT_WIDGET_SELECTORMODEL_HPP__
#define __UIMEDDATAQT_WIDGET_SELECTORMODEL_HPP__

#include "uiMedDataQt/config.hpp"

#include <fwMedData/Series.hpp>
#include <fwMedData/Study.hpp>

#include <QPointer>
#include <QStandardItem>
#include <QStandardItemModel>
#include <QString>
#include <QTreeView>

#include <map>

namespace uiMedDataQt
{
namespace widget
{

class UIMEDDATAQT_CLASS_API SelectorModel : public QStandardItemModel
{
Q_OBJECT

public:

    typedef enum
    {
        ITEM_TYPE = Qt::UserRole, 
        UID                       
    }Role;

    typedef enum
    {
        STUDY = 1,  
        SERIES      
    }ItemType;

    typedef std::map<std::string, std::string> SeriesIconType;

    UIMEDDATAQT_API SelectorModel(QWidget* parent = 0);

    UIMEDDATAQT_API ~SelectorModel();

    UIMEDDATAQT_API void addSeries(::fwMedData::Series::sptr series);

    UIMEDDATAQT_API void removeSeries(::fwMedData::Series::sptr series);

    UIMEDDATAQT_API void clear();

    Qt::ItemFlags flags(const QModelIndex& index) const
    {
        return (QStandardItemModel::flags(index) & ~Qt::ItemIsEditable);
    }

    UIMEDDATAQT_API ItemType getItemType(const QModelIndex& index);

    UIMEDDATAQT_API QModelIndex getIndex(const QModelIndex& index, int column );

    UIMEDDATAQT_API void removeRows(const QModelIndexList indexes);

    UIMEDDATAQT_API QStandardItem* findSeriesItem(::fwMedData::Series::sptr series);

    UIMEDDATAQT_API QStandardItem* findStudyItem(::fwMedData::Study::sptr study);

    UIMEDDATAQT_API void setSeriesIcons(const SeriesIconType& seriesIcons);

    UIMEDDATAQT_API void setInsertMode(bool insert);

private:

    typedef std::map< ::fwMedData::DicomValueType, QStandardItem* > StudyUidItemMapType;

    template <typename T>
    QStandardItem* getInfo(T data, QString separator);

    bool removeStudyItem(QStandardItem* item);

    bool removeSeriesItem(QStandardItem* item);

    void addSeriesIcon(::fwMedData::Series::sptr series, QStandardItem* item);

    void init();

    int m_studyRowCount;

    StudyUidItemMapType m_items;

    bool m_insert;

    SeriesIconType m_seriesIcons;
};

//------------------------------------------------------------------------------

template <typename T>
QStandardItem* SelectorModel::getInfo(T data, QString separator)
{
    QString dataStr;
    if(!data.empty())
    {
        typename T::iterator itr = data.begin();
        std::ostringstream str;
        str << *itr++;
        dataStr = QString::fromStdString(str.str());

        for(; itr != data.end(); ++itr)
        {
            str.str("");
            str << *itr;
            dataStr.append(separator + QString::fromStdString(str.str()));
        }
    }
    QStandardItem* item = new QStandardItem(dataStr);
    return item;
}

} // namespace widget
} // namespace uiMedDataQt

#endif // __UIMEDDATAQT_WIDGET_SELECTORMODEL_HPP__