PMLComponent.h

Go to the documentation of this file.

/*****************************************************************************
 * $CAMITK_LICENCE_BEGIN$
 *
 * CamiTK - Computer Assisted Medical Intervention ToolKit
 * (c) 2001-2025 Univ. Grenoble Alpes, CNRS, Grenoble INP - UGA, 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 propertyValueChanged(QString name);

    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