#ifndef DM_ITEMDRAWABLETREEVIEWMODELRESEARCHERT_HPP
#define DM_ITEMDRAWABLETREEVIEWMODELRESEARCHERT_HPP

#include "tools/treeview/dm_itemdrawabletreeviewmodelresearchert.h"

template<class Item>
DM_ItemDrawableTreeViewModelResearcherT<Item>::DM_ItemDrawableTreeViewModelResearcherT() : DM_AbstractWorker()
{
    m_collection = NULL;
    m_treeViewManager = NULL;
}

template<class Item>
void DM_ItemDrawableTreeViewModelResearcherT<Item>::setCollection(const QVector<QList<Item*> > *collection)
{
    m_collection = (QVector<QList<Item*> >*)collection;
}

template<class Item>
void DM_ItemDrawableTreeViewModelResearcherT<Item>::setTreeViewManager(const DM_ITreeViewManagerT<Item> *man)
{
    m_treeViewManager = (DM_ITreeViewManagerT<Item>*)man;
}

template<class Item>
void DM_ItemDrawableTreeViewModelResearcherT<Item>::setItemDrawable(const QList<CT_AbstractItemDrawable *> &list)
{
    m_items = list;
}

template<class Item>
void DM_ItemDrawableTreeViewModelResearcherT<Item>::apply()
{
    int size = m_items.size();
    int i = 0;

    m_collection->resize(size);

    QListIterator<CT_AbstractItemDrawable *> it(m_items);

    while(it.hasNext())
    {
        CT_AbstractItemDrawable *item = it.next();

        Item* sItem = m_treeViewManager->itemFromItemDrawable(item);

        if(sItem != NULL)
            (*m_collection)[i].append(sItem);

        ++i;
        setProgress((i*100)/size);
    }

    setFinished();
}


#endif // DM_ITEMDRAWABLETREEVIEWMODELRESEARCHERT_HPP