PMLComponent.h

Go to the documentation of this file.

/*****************************************************************************
 * $CAMITK_LICENCE_BEGIN$
 *
 * CamiTK - Computer Assisted Medical Intervention ToolKit
 * (c) 2001-2021 Univ. Grenoble Alpes, CNRS, Grenoble INP, TIMC, 38000 Grenoble, France
 *
 * 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
// disable warning generated by clang about the surrounded headers
#include <CamiTKDisableWarnings>
#include <vtkExtractSelection.h>
#include <CamiTKReEnableWarnings>
 
#include <vtkSelectionNode.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();
 
    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