PMLComponent.h

Go to the documentation of this file.

/*****************************************************************************
 * $CAMITK_LICENCE_BEGIN$
 *
 * CamiTK - Computer Assisted Medical Intervention ToolKit
 * (c) 2001-2018 Univ. Grenoble Alpes, CNRS, TIMC-IMAG UMR 5525 (GMCAO)
 *
 * Visit http://camitk.imag.fr for more information
 *
 * This file is part of CamiTK.
 *
 * CamiTK is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License version 3
 * only, as published by the Free Software Foundation.
 *
 * CamiTK 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 version 3 for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * version 3 along with CamiTK.  If not, see <http://www.gnu.org/licenses/>.
 *
 * $CAMITK_LICENCE_END$
 ****************************************************************************/
#ifndef PML_COMPONENT_H
#define PML_COMPONENT_H

// stl includes
#include <map>

// QT forward declaration
class QString;

// Qt includes
#include <QObject>

// lib pml forward declaration
class PhysicalModel;
class StructuralComponent;
class MultiComponent;
class Atom;
class Cell;

// vtk forward declarations
class vtkUnstructuredGrid;
class vtkPolyVertex;
class vtkSelection;
class vtkIdTypeArray;

// CamiTK forward declaration
namespace camitk {
class AbortException;
}

// CamiTK include
#include <MeshComponent.h>
#include "PMLComponentAPI.h"

// VTK includes
#include <vtkSelectionNode.h>
#include <vtkExtractSelection.h>

namespace std {
using AtomPointIdPair = std::pair<const Atom*, const vtkIdType>;
using AtomPointIdMap = std::map <const Atom*, const vtkIdType>;
using AtomPointIdMapIterator = std::map <const Atom*, const vtkIdType>::iterator;
}

class PML_COMPONENT_API PMLComponent : public camitk::MeshComponent {
    Q_OBJECT

public:
    PMLComponent(const QString& file);

    PMLComponent(PhysicalModel* p, const QString& originalFile);

    virtual ~PMLComponent();

    virtual QPixmap getIcon();

    virtual void updateProperty(QString name, QVariant value);

    virtual void setSelected(const bool b, const bool recursive = true);

    PhysicalModel* getPhysicalModel() {
        return this->physicalModel;
    }

    vtkIdType getPointId(const Atom* a);

    void init();

    void selectCell(Cell* cell, bool showAtomGlyph);

    void selectAtom(const Atom* atom);

    void selectSC(StructuralComponent* sc, bool showAtomGlyph);

    void selectMC(MultiComponent* mc, bool showAtomGlyph);

    void unselectItems();

    void updateSelection();

    void refreshDisplay();

    virtual int addSelection(const QString& name, int fieldType, int contentType,  vtkSmartPointer< vtkAbstractArray > array, camitk::MeshSelectionModel::InsertionPolicy policy = camitk::MeshSelectionModel::REPLACE);



protected:
    virtual void initDynamicProperties();

private:

    PhysicalModel* physicalModel;

    static QPixmap* myPixmap;

    bool neverSelected;

    vtkSmartPointer<vtkPoints> thePoints;

    vtkSmartPointer<vtkUnstructuredGrid> mainGrid;

    std::AtomPointIdMap atomPointIdMap;

    void create3DStructure();

    void parseMultiComponent(MultiComponent* mc);

    vtkSmartPointer<vtkCell> cellToVTK(Cell* cell);

    vtkSmartPointer<vtkPolyVertex> atomSCToVTK(StructuralComponent* sc);



    vtkSmartPointer<vtkIdTypeArray> selectedAtomIdArray;

    vtkSmartPointer<vtkSelection> selectedAtomSelection;

    QMap<QString, vtkSmartPointer<vtkIdTypeArray> > scCellIdArrayMap;

    QMap<QString, vtkSmartPointer<vtkSelection> > scCellSelectionMap;

    void createCellSelectionVTKPipeline(const StructuralComponent* sc);

    void createAtomSelectionVTKPipeline();


};

// -------------------- Atom / VtkPoint Id Map --------------------
inline vtkIdType PMLComponent::getPointId(const Atom* a) {
    if (!a) {
        return 0;
    }
    std::AtomPointIdMapIterator result = atomPointIdMap.find(a);
    return (result == atomPointIdMap.end()) ? 0 : (result->second);
}


#endif