milxQtRobustShapeModel.h

/*=========================================================================
  The Software is copyright (c) Commonwealth Scientific and Industrial Research Organisation (CSIRO)
  ABN 41 687 119 230.
  All rights reserved.

  Licensed under the CSIRO BSD 3-Clause License
  You may not use this file except in compliance with the License.
  You may obtain a copy of the License in the file LICENSE.md or at

  https://stash.csiro.au/projects/SMILI/repos/smili/browse/license.txt

  Unless required by applicable law or agreed to in writing, software
  distributed under the License is distributed on an "AS IS" BASIS,
  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  See the License for the specific language governing permissions and
  limitations under the License.
=========================================================================*/
#ifndef milxQtRobustShapeModel_H
#define milxQtRobustShapeModel_H

//milxSSM3D
#include <itkRobustStatisticalShapeModel.h>

#include <milxQtAliases.h>
#include <milxQtShapeModel.h>

typedef itk::RobustStatisticalShapeModel<double> RobustShapeModelType;

class MILXQT_PLUGIN_EXPORT milxQtRobustShapeModel : public milxQtShapeModel
{
    Q_OBJECT

public:
    milxQtRobustShapeModel(QWidget *theParent = 0, bool contextSystem = true);
    virtual ~milxQtRobustShapeModel();

public slots:
    void LoadModel(const QString filename);
    inline virtual void openModel(const QString filename)
    {
        LoadModel(filename);
    }
    inline virtual void SaveModel(const QString filename)
    {
        m_RobustSSM->Update();
        m_RobustSSM->SaveCompactModel(filename.toStdString().c_str());
    }
    inline virtual void saveModel(const QString filename)
    {
        SaveModel(filename);
    }

    virtual void SetInputCollection(vtkPolyDataCollection* meshes);
    virtual void SetInputCollection(vtkPolyDataCollection* meshes, QStringList &filenames);
    virtual void SetInputCollection(vtkPolyDataCollection* meshes, vtkPolyData *atlasSurface, QStringList &filenames);

    virtual void createMenu(QMenu *menu);

    virtual void generateSSM();
    virtual void generateMeanModel(vtkSmartPointer<vtkPolyData> shape = NULL);
    virtual void generateModes();
    virtual void generateCollectionBasedOnMode();
    virtual void generateCorrespondences();

    void compactness();
    void specificity();
    void generalisability();
    void eigenvalues();
    void eigenmodes();
    void parameters();

signals:
    void collectionAvailable(vtkPolyDataCollection*, QStringList&);
    void resultAvailable(milxQtRenderWindow*);
    void resultAvailable(milxQtModel*);

protected:
    RobustShapeModelType::Pointer m_RobustSSM; 
    QList< int > m_caseIDs; 

    virtual void reset();
    void createActions();
    void contextMenuEvent(QContextMenuEvent *event);

private:

};

#endif // milxQtRobustShapeModel_H