Slice.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 CANONICAL_SLICE_H
#define CANONICAL_SLICE_H

// -- Core stuff
#include "CamiTKAPI.h"
#include "InterfaceBitMap.h"

// -- vtk stuff
#include <vtkSmartPointer.h>
#include <vtkWindowLevelLookupTable.h>
#include <vtkImageMapToColors.h>
#include <vtkActor.h>
#include <vtkImageActor.h>
#include <vtkImageChangeInformation.h>
#include <vtkImageReslice.h>
#include <vtkMatrix4x4.h>
#include <vtkPolyDataMapper.h>
#include <vtkPlaneSource.h>
#include <vtkTransform.h>
#include <vtkPixel.h>
#include <vtkUnstructuredGrid.h>
#include <vtkDataSetMapper.h>
#include <vtkLine.h>
#include <vtkPolygon.h>
#include <vtkTextProperty.h>
#include <vtkTextMapper.h>
#include <vtkAxesActor.h>
#include <vtkTransformPolyDataFilter.h>

namespace camitk {
class CAMITK_API Slice : public InterfaceBitMap {
public:
    /* -------------------------------------------------------------------- */
    /* -------------------------------------------------------------------- */

    enum SliceOrientation {
        AXIAL,
        CORONAL,
        SAGITTAL,
        AXIAL_NEURO,
        ARBITRARY
    };

    Slice(vtkSmartPointer<vtkImageData> volume, SliceOrientation AXIAL_ORIENTATION, vtkSmartPointer<vtkWindowLevelLookupTable> lookupTable = nullptr);

    ~Slice() override;

    /* -------------------------------------------------------------------- */
    /* -------------------------------------------------------------------- */

    void setOriginalVolume(vtkSmartPointer<vtkImageData> img) override;

    void setImageWorldTransform(vtkSmartPointer<vtkTransform>) override;

    vtkSmartPointer<vtkImageActor> get2DImageActor() const override;

    vtkSmartPointer<vtkImageActor> get3DImageActor() const override;

    vtkSmartPointer<vtkActor> getPickPlaneActor() const override;

    vtkSmartPointer<vtkActor> getPixelActor() override;

    virtual vtkSmartPointer<vtkAxesActor> get2DAxesActor();

    void pixelPicked(double, double, double) override;

    void reslicedToVolumeCoords(const double* ijk, double* xyz);

    void volumeToReslicedCoords(const double* xyz, double* ijk);

    void updatePickPlane() override;

    int getNumberOfSlices() const override;

    int getSlice() const override;

    void setSlice(int s) override;

    void setSlice(double x, double y, double z) override;

    int getNumberOfColors() const override;

    void setPixelRealPosition(double, double, double) override;

    vtkSmartPointer<vtkImageData> getImageData() const override;

    // BEGIN TODO : put all of this into a dedicated interface
    QMap<QString, vtkSmartPointer<vtkProp> > extraProp;

    vtkSmartPointer<vtkProp> getProp(const QString&) override;

    unsigned int getNumberOfProp() const override;

    vtkSmartPointer<vtkProp> getProp(unsigned int) override;

    bool addProp(const QString&,  vtkSmartPointer<vtkProp>) override;

    bool removeProp(const QString&) override;
    // END TODO


protected:
    /* -------------------------------------------------------------------- */
    /* -------------------------------------------------------------------- */

    virtual void init();

    virtual void initActors();

    /* -------------------------------------------------------------------- */
    /* ---------------------------------------------------------------------*/

    SliceOrientation sliceOrientation;

    vtkSmartPointer<vtkImageData> originalVolume;

    int extent[6];

    int currentSliceIndex;

    vtkSmartPointer<vtkWindowLevelLookupTable>  lut;

    int originalDimensions[3];

    double originalSpacing[3];

    double originalSize[3];

    vtkSmartPointer<vtkImageMapToColors>    imgToMapFilter;

    vtkSmartPointer<vtkImageActor>        image3DActor;

    vtkSmartPointer<vtkImageActor>              image2DActor;
    vtkSmartPointer<vtkPlaneSource> pickPlane;

    vtkSmartPointer<vtkPolyDataMapper> pickPlaneMapper;

    vtkSmartPointer<vtkActor> pickPlaneActor;


    void initPixelActor();

    void updatePixelActorPosition(double x, double y, double z);


    vtkSmartPointer<vtkActor> pixelActor;


    void update2DAxesActorPosition();
    void init2DAxesActor();
    vtkSmartPointer<vtkAxesActor> axes2DActor;


};

}

#endif // CANONICAL_SLICE_H