ObjectListItemModel.cc

/*===========================================================================*\
*                                                                            *
*                              OpenFlipper                                   *
 *           Copyright (c) 2001-2015, RWTH-Aachen University                 *
 *           Department of Computer Graphics and Multimedia                  *
 *                          All rights reserved.                             *
 *                            www.openflipper.org                            *
 *                                                                           *
 *---------------------------------------------------------------------------*
 * This file is part of OpenFlipper.                                         *
 *---------------------------------------------------------------------------*
 *                                                                           *
 * Redistribution and use in source and binary forms, with or without        *
 * modification, are permitted provided that the following conditions        *
 * are met:                                                                  *
 *                                                                           *
 * 1. Redistributions of source code must retain the above copyright notice, *
 *    this list of conditions and the following disclaimer.                  *
 *                                                                           *
 * 2. Redistributions in binary form must reproduce the above copyright      *
 *    notice, this list of conditions and the following disclaimer in the    *
 *    documentation and/or other materials provided with the distribution.   *
 *                                                                           *
 * 3. Neither the name of the copyright holder nor the names of its          *
 *    contributors may be used to endorse or promote products derived from   *
 *    this software without specific prior written permission.               *
 *                                                                           *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS       *
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED *
 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A           *
 * PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER *
 * OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,  *
 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,       *
 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR        *
 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF    *
 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING      *
 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS        *
 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.              *
*                                                                            *
\*===========================================================================*/



#include "ObjectListItemModel.hh"

#include <OpenFlipper/BasePlugin/PluginFunctions.hh>
#include <OpenFlipper/common/DataTypes.hh>

ObjectListItemModel::ObjectListItemModel() {
}

ObjectListItemModel::~ObjectListItemModel() {
}

int ObjectListItemModel::rowCount(const QModelIndex & parent) const {
    return objects_.size() + 1;
}

QVariant ObjectListItemModel::data(const QModelIndex & index, int role) const {
    switch (role) {
        case Qt::DisplayRole:
            if (index.row() <= 0) return tr("<none>");
            return objects_[index.row() - 1].getName();
        case Qt::UserRole:
            if (index.row() <= 0) return QVariant::fromValue<int>(-1);
            return QVariant::fromValue(objects_[index.row() - 1].getId());
        default:
            return QVariant::Invalid;
    }
}

void ObjectListItemModel::removeObject(int _id)
{
    beginResetModel();
    for (int i = (int)objects_.size()-1; i>=0; i--)
    {
        if (objects_[i].getId() == _id)
            objects_.erase(objects_.begin()+i);
    }
    endResetModel();
}

void ObjectListItemModel::refresh(const DataType &datatype) {
    std::vector<ObjectInfo> objects;
        objects.push_back(ObjectInfo("ALL_OBJECTS", -13));
        objects.push_back(ObjectInfo("TARGET_OBJECTS", -14));
        objects.push_back(ObjectInfo("SOURCE_OBJECTS", -15));
        
    for (PluginFunctions::ObjectIterator o_it(PluginFunctions::ALL_OBJECTS, datatype);
            o_it != PluginFunctions::objectsEnd(); ++o_it) {
        objects.push_back(ObjectInfo(o_it->name(), o_it->id()));
    }

    std::sort(objects.begin(), objects.end());

    if (objects != objects_) {
        beginResetModel();
        objects.swap(objects_);
        endResetModel();
    }
}