ImageComponent.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 IMAGE_COMPONENT_H
#define IMAGE_COMPONENT_H
 
// -- Core image component stuff
#include "CamiTKAPI.h"
#include "Component.h"
#include "ImageOrientationHelper.h"
 
// -- vtk stuff
#include <vtkPolyData.h>
#include <vtkSmartPointer.h>
#include <vtkImageData.h>
#include <vtkTransform.h>
#include <vtkImageFlip.h>
#include <vtkWindowLevelLookupTable.h>
#include <vtkMatrix4x4.h>
#include <vtkUnstructuredGrid.h>
#include <vtkActor.h>
 
// -- QT stuff classes
class QMenu;
class QTableView;
class QVector3D;
class QVariant;
class QTableView;
class QStandardItemModel;
 
namespace camitk {
 
class MeshComponent;
class SingleImageComponent;
class ArbitrarySingleImageComponent;
 
class CAMITK_API ImageComponent : public camitk::Component {
    Q_OBJECT
 
public:
 
    ImageComponent(const QString& file, const QString& name);
 
    ImageComponent(vtkSmartPointer<vtkImageData> anImageData, const QString& name, bool copy = false, ImageOrientationHelper::PossibleImageOrientations initialOrientation = ImageOrientationHelper::RAI);
 
    ~ImageComponent() override;
 
    void setSelected(const bool b, const bool recursive = false) override;
 
    virtual void setVisibility(QString, bool) override;
 
    vtkSmartPointer<vtkImageData> getImageData() const override {
        return originalImageData;
    }
 
    vtkSmartPointer<vtkImageData> getImageDataWithFrameTransform();
 
    ImageOrientationHelper::PossibleImageOrientations getInitialOrientation() const {
        return initialOrientation;
    }
 
    void pixelPicked(double x, double y, double z) override;
 
    virtual vtkSmartPointer<vtkActor> get3DCursor() override;
 
    void getLastPixelPicked(int* x, int* y, int* z);
 
    void getLastPointPickedDataFrame(double* x, double* y, double* z);
 
    void getLastPointPickedWorldFrame(double* x, double* y, double* z);
 
    void setLastPointPickedFromPixel(int x, int y, int z);
 
    int getNumberOfColors() const override;
 
    double getMinColor() const;
 
    double getMaxColor() const;
 
    int getActualNumberOfColors() const;
 
    double getActualMinColor() const;
 
    double getActualMaxColor() const;
 
    int getNumberOfSlices() const override;
 
    virtual void setLut(vtkSmartPointer<vtkWindowLevelLookupTable> lookupTable);
 
    virtual vtkSmartPointer<vtkWindowLevelLookupTable>  getLut();
 
    virtual const vtkSmartPointer<vtkWindowLevelLookupTable>  getLut() const;
 
    void refresh() override;
 
    SingleImageComponent* getAxialSlices();
    SingleImageComponent* getCoronalSlices();
    SingleImageComponent* getSagittalSlices();
    ArbitrarySingleImageComponent* getArbitrarySlices();
    MeshComponent* getVolumeRenderingChild();
 
    virtual void replaceImageData(vtkSmartPointer<vtkImageData> anImageData, bool copy = false, ImageOrientationHelper::PossibleImageOrientations initialOrientation = ImageOrientationHelper::RAI);
 
    void propertyValueChanged(QString) override;
 
    unsigned int getNumberOfPropertyWidget() override;
 
    QWidget* getPropertyWidgetAt(unsigned int i) override;
 
    const vtkSmartPointer<vtkMatrix4x4> getRotationMatrix() {
        return rotationMatrix;
    }
 
    virtual void setFramesAndTransformation(const std::shared_ptr<FrameOfReference>& mainFrame, const std::shared_ptr<FrameOfReference>& dataFrame,  const std::shared_ptr<Transformation>& mainTransformation);
 
    const FrameOfReference* getDataFrame() const {
        return dataFrame.get();
    };
 
    virtual void setFrame(const std::shared_ptr<FrameOfReference>& frame) override;
 
    // Modify this object's frame using the given object's frame.
    virtual void setFrameFrom(const InterfaceFrame*) override;
 
    virtual void resetFrame() override;
 
    virtual const std::vector<std::shared_ptr<FrameOfReference>>& getAdditionalFrames() const {
        return additionalFrames;
    };
 
    virtual void addAdditionalFrame(const std::shared_ptr<FrameOfReference>& fr) {
        additionalFrames.push_back(fr);
    }
 
    virtual void removeAdditionalFrame(const FrameOfReference* fr) {
        std::erase_if(additionalFrames, [fr](auto & f) {
            return f.get() == fr;
        });
    }
 
    virtual const std::vector<std::shared_ptr<Transformation>>& getAdditionalTransformations() const {
        return additionalTransformations;
    }
 
    virtual void addAdditionalTransformation(const std::shared_ptr<Transformation> tr) {
        additionalTransformations.push_back(tr);
    }
 
    virtual void removeAdditionalTransformation(const Transformation* tr) {
        std::erase_if(additionalTransformations, [tr](auto t) {
            return t.get() == tr;
        });
    }
 
    virtual QMultiMap<const FrameOfReference*, Component*> getAllFrames(bool includeChildrenFrames = true) override;
 
    virtual QMultiMap<const Transformation*, Component*> getAllTransformations(bool includeChildrenTransformations = true) override;
 
    virtual Transformation* getMainTransformation() const {
        return mainTransformation.get();
    }
 
    virtual void updateMainTransformation(vtkSmartPointer<vtkMatrix4x4>);
 
    virtual QVariant toVariant() const override;
 
    virtual void fromVariant(const QVariant&) override;
 
protected:
 
    void setImageData(vtkSmartPointer<vtkImageData> anImageData,
                      bool copy,
                      ImageOrientationHelper::PossibleImageOrientations initialOrientation = ImageOrientationHelper::RAI,
                      vtkSmartPointer<vtkMatrix4x4> initialTransformMatrix = nullptr);
 
    void setSingleImageComponents(SingleImageComponent* axialSlices, SingleImageComponent* sagittalSlices, SingleImageComponent* coronalSlices, ArbitrarySingleImageComponent* arbitrarySlices);
 
    bool setMainTransformation(const std::shared_ptr<Transformation>& tr);
 
private:
    void initImageProperties();
 
    void initRepresentation() override {}
 
    // builds default lookup table
    void initLookupTable();
 
    void initCursor();
 
    void updateCursor();
 
    void buildImageComponents();
 
    void updateImageComponents();
 
    void update3DViewer();
 
    vtkSmartPointer<vtkPolyData> getBoundingBox();
 
    vtkSmartPointer<vtkImageData> originalImageData;
 
    SingleImageComponent* axialSlices;
 
    SingleImageComponent* sagittalSlices;
 
    SingleImageComponent* coronalSlices;
 
    ArbitrarySingleImageComponent* arbitrarySlices;
 
    MeshComponent* volumeRenderingChild;
 
    vtkSmartPointer<vtkWindowLevelLookupTable> lut;
 
    double currentPixelPicked[3];
 
    vtkSmartPointer<vtkActor> cursorActor;
 
    vtkSmartPointer<vtkUnstructuredGrid> cursorActorPointSet;
 
    void init();
 
    QTableView* selectionView;
 
    QStandardItemModel* model;
 
    ImageOrientationHelper::PossibleImageOrientations initialOrientation;
 
    vtkSmartPointer<vtkTransform> initialImageDataTransform;
 
    vtkSmartPointer<vtkTransform> initialFrameTransform;
 
    vtkSmartPointer<vtkMatrix4x4> rotationMatrix;
 
    // InterfaceFrame extension for Images
    std::shared_ptr<FrameOfReference> dataFrame = nullptr;
    std::shared_ptr<Transformation> mainTransformation = nullptr;
    std::vector<std::shared_ptr<FrameOfReference>> additionalFrames;
    std::vector<std::shared_ptr<Transformation>> additionalTransformations;
};
 
}
 
#endif //IMAGE_COMPONENT_H