milxQtRegistrationPlugin.h

#ifndef MILXQTRegistrationPLUGIN_H
#define MILXQTRegistrationPLUGIN_H

#include <QThread>
#include <QMenu>
#include <QDockWidget>
//VTK
#include <vtkPolyDataCollection.h>

#include "milxQtAliases.h"
#include "milxQtPluginInterface.h"
#include "milxQtRenderWindow.h"
#include "milxQtImage.h"
#include "milxQtModel.h"
#include "milxQtMain.h"
#include "milxQtRegistrationWindow.h"


class MILXQT_PLUGIN_EXPORT milxQtRegistrationPlugin : public milxQtPluginInterface
{
    Q_OBJECT

public:
    milxQtRegistrationPlugin(QObject *theParent = 0);
    virtual ~milxQtRegistrationPlugin();

    virtual QString name();

    inline virtual bool hasOpenSupport()
    {
        return false;
    }
    virtual QString openFileSupport();
    virtual QStringList openExtensions();
    inline virtual bool hasSaveSupport()
    {
        return false;
    }
    virtual QString saveFileSupport();
    virtual QStringList saveExtensions();

    inline virtual bool hasCollectionSupport()
    {
        return false;
    }
    virtual void SetInputCollection(vtkPolyDataCollection* collection, QStringList &filenames);

    virtual void open(QString filename);
    virtual void save(QString filename);

    virtual milxQtRenderWindow* genericResult();
    virtual milxQtModel* modelResult();
    virtual milxQtImage* imageResult();
    virtual QDockWidget* dockWidget();

    inline virtual Qt::DockWidgetArea dockDefaultArea()
    {
        return Qt::LeftDockWidgetArea;
    }

    virtual bool isPluginWindow(QWidget *window);

    void registration(RegType type);

public slots:
    virtual void loadExtension();
    virtual void update() {}
    virtual void preStartTasks() {}
    virtual void postStartTasks() {}

    void ItkAffineRegistrationSlot();

    void ItkDemonRegistrationSlot();

#ifdef USE_NIFTI_REG

    void F3DNiftiRegistrationSlot();

    void AladinNiftiRegistrationSlot();
#endif

#ifdef USE_ELASTIX

    void ElastixAffineRegistrationSlot();

    void ElastixBSplineRegistrationSlot();
#endif

protected:

    QPointer<milxQtMain> MainWindow;

    QMenu* menu; 
    QAction* actionItkAffine; 
    QAction* actionItkDemon; 

#ifdef USE_NIFTI_REG
    QAction* actionF3DNifti; 
    QAction* actionAladinNifti; 
#endif

#ifdef USE_ELASTIX
    QAction* actionElastixAffine; 
    QAction* actionElastixBSpline; 
#endif

    QPointer<milxQtRegistrationWindow> regWindow; 

    void createActions();
    void createMenu();
    void createConnections();

};

class MILXQT_PLUGIN_EXPORT milxQtRegistrationPluginFactory : public QObject, public milxQtPluginFactory
{
    Q_OBJECT
    Q_INTERFACES(milxQtPluginFactory)

public:
    milxQtPluginInterface* newPlugin(QObject *theParent = 0)
    {
        return new milxQtRegistrationPlugin(theParent);
    }
};

#endif // MILXQTRegistrationPLUGIN_H