NodeModel.h

Go to the documentation of this file.

/*
-----------------------------------------------------------------------------
This source file is part of FRAPPER
research.animationsinstitut.de
sourceforge.net/projects/frapper

Copyright (c) 2008-2009 Filmakademie Baden-Wuerttemberg, Institute of Animation 

This program is free software; you can redistribute it and/or modify it under
the terms of the GNU Lesser General Public License as published by the Free Software
Foundation; version 2.1 of the License.

This program is distributed in the hope that it will be useful, but WITHOUT
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License for more details.

You should have received a copy of the GNU Lesser General Public License along with
this program; if not, write to the Free Software Foundation, Inc., 59 Temple
Place - Suite 330, Boston, MA 02111-1307, USA, or go to
http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html
-----------------------------------------------------------------------------
*/


#ifndef NODEMODEL_H
#define NODEMODEL_H

#include "FrapperPrerequisites.h"
#include <QtGui/QStandardItemModel>
#include <QtGui/QGraphicsItem>
#include <QtGui/QItemSelection>
#include <QtCore/QMutex>
#include "Node.h"

namespace Frapper {

    class FRAPPER_CORE_EXPORT NodeModel : public QStandardItemModel
    {

        Q_OBJECT

    public: // constructors and destructors

        NodeModel ();

        ~NodeModel ();

    public: // functions

        void beginUpdate ();

        void endUpdate ();

        void beginSelectingAll ();

        void endSelectingAll ( bool emitSignal = true);

        Node * createNode ( const QString &typeName, const QString &name = "", bool visible = true );

        Connection * createConnection ( Parameter *sourceParameter, Parameter *targetParameter );


        Node * getNode ( const QString &name ) const;

        QList<Node *> getNodes ( const QString &typeName = "" ) const;

        QList<Connection *> getConnections () const;

                Connection * getConnection (const QString &name) const;

        QStringList getNodeNames ( const QString &typeName = "" ) const;

        QStandardItem * getStandardItemNode ( const QString &name ) const;

        QStandardItem * getStandardItemConnection ( const QString &name ) const;

        void setSelected ( const QItemSelection &selection, bool selected );

        void deleteNode ( const QString &name );

        void deleteConnection ( const QString &name );

public slots:

            void setSelected ( const QString &name, bool selected );

            void renameNode ( const QString &oldName, const QString &newName );

            void changeParameter ( const QString &name, const QString &parameterName, const QVariant &value );

signals: //

            void modified ();

            void clearSelectionRequested ();

            void nodeSelected ( Node *node );

            void nodeDeleted ( const QString &name );

    private: // functions

        QStandardItem * getCategoryItem ( const QString &nodeCategory );

    private: // data

        bool m_updating;

        bool m_selectingAll;

        QMutex m_mutex;

        QHash<QString, Node *> m_nodeMap;

        QHash<QString, Connection *> m_connectionMap;

        QHash<QString, QStandardItem *> m_standardItemNodeMap;

        QHash<QString, QStandardItem *> m_standardItemConnectionMap;

        QList<QStandardItem *> m_categoryItems;

    };

} // end namespace Frapper

#endif