milxQtPlot.h

/*=========================================================================
  The Software is copyright (c) Commonwealth Scientific and Industrial Research Organisation (CSIRO)
  ABN 41 687 119 230.
  All rights reserved.

  Licensed under the CSIRO BSD 3-Clause License
  You may not use this file except in compliance with the License.
  You may obtain a copy of the License in the file LICENSE.md or at

  https://stash.csiro.au/projects/SMILI/repos/smili/browse/license.txt

  Unless required by applicable law or agreed to in writing, software
  distributed under the License is distributed on an "AS IS" BASIS,
  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  See the License for the specific language governing permissions and
  limitations under the License.
=========================================================================*/
#ifndef MILXQTPLOT_H
#define MILXQTPLOT_H

//VTK
#include <vtkChartXY.h>
#include <vtkSmartVolumeMapper.h>
#include <vtkVolume.h>

#include "milxQtModel.h"

class MILXQT_EXPORT milxQtPlot : public milxQtModel
{
    Q_OBJECT

public:
    milxQtPlot(QWidget *theParent = 0, bool contextSystem = true);
    virtual ~milxQtPlot();

    void setPlotType2D(const int indexX = 0, const int indexY = 1)
    {
        plotType2D = true;
        plotType3D = false;
        plotTypeSurface = false;
        plotTypeVolume = false;
        xIndex = indexX;
        yIndex = indexY;
    }
    void setPlotType3D(const int indexX = 0, const int indexY = 1, const int indexZ = 2)
    {
        plotType2D = false;
        plotType3D = true;
        plotTypeSurface = false;
        plotTypeVolume = false;
        xIndex = indexX;
        yIndex = indexY;
        zIndex = indexZ;
    }
    inline void setPlotTypeSurface(const int zAxis = 2)
    {
        plotTypeSurface = true;
        plotTypeVolume = false;
        plotType2D = false;
        plotType3D = false;
        displaceAxis = zAxis;
    }
    inline void setPlotTypeVolume()
    {
        plotTypeVolume = true;
        plotTypeSurface = false;
        plotType2D = false;
        plotType3D = false;
    }

    void SetSource(vtkSmartPointer<vtkImageData> img, const bool eightbit, const bool useGPU = false)
    {
        imageData = vtkSmartPointer<vtkImageData>::New();
            imageData->DeepCopy(img);
        sourceLoaded = true;
        sourceEightbit = eightbit;
        plotTypeGPU = useGPU;
    }
    void SetSource(vtkSmartPointer<vtkTable> tbl)
    {
        table->DeepCopy(tbl);
        printDebug("Source Table is " + QString::number(table->GetNumberOfRows()) + "x" + QString::number(table->GetNumberOfColumns()));
        sourceLoaded = true;
    }
    inline vtkSmartPointer<vtkTable> GetSource()
    {
        return table;
    }
    inline vtkSmartPointer<vtkImageData> GetSourceImage()
    {
        return imageData;
    }
    inline virtual vtkDataSet* GetDataSet()
    {
        if(plotTypeSurface)
            return model.Result();

        return imageData;
    }

    inline void setOpacityTransferValues(std::vector<float> values)
    {
        opacityTransferValues = values;
    }
    inline std::vector<float> getOpacityTransferValues()
    {
        return opacityTransferValues;
    }
    inline void setColourTransferValues(std::vector<float> values)
    {
        colourTransferValues = values;
    }
    inline std::vector<float> getColourTransferValues()
    {
        return colourTransferValues;
    }

public slots:
    inline void refresh()
    {
        milxQtRenderWindow::refresh();
    }
    virtual void createMenu(QMenu *menu);

    void generatePlot();

    void scatterPlot(vtkSmartPointer<vtkTable> table, const int xColumn, const int yColumn);
    void scatterPlot(vtkSmartPointer<vtkTable> table, const int xColumn, const int yColumn, const int zColumn);
    void surfacePlot(vtkSmartPointer<vtkImageData> img, const int zDirection = 2);
    void surfacePlot(vtkSmartPointer<vtkTable> table);
    void volumePlot(vtkSmartPointer<vtkImageData> img, const bool eightbit, const bool quiet = false);

    void renameAxes(QString xLabel = "", QString yLabel = "", QString zLabel = "");
    void renameTitle(QString newTitle = "");
    void points(const bool showIt = true);
    void logScale(const bool showIt = true);
    void legend(const bool showIt = true);
    void enableScale(QString title = "");
    void scaleDisplay();
    virtual void updateCoords(vtkObject *obj);
    virtual void updateLookupTable();
    virtual void updateVolumePlot(int value);

signals:

protected:
    bool sourceLoaded; 
    bool sourceEightbit; 
    bool plotType2D; 
    bool plotType3D; 
    bool plotTypeSurface; 
    bool plotTypeVolume; 
    bool plotTypeGPU; 
    bool plotted; 
    int xIndex; 
    int yIndex; 
    int zIndex; 
    int displaceAxis; 

    vtkSmartPointer<vtkTable> table; 
    vtkSmartPointer<vtkImageData> imageData; 
    vtkSmartPointer<vtkChartXY> templateChart; 
    vtkSmartPointer<vtkVolume> volume; 
    vtkSmartPointer<vtkSmartVolumeMapper> volumeFixedMapper; 

    //------------------
    //Context Menu
    QAction* xAxisName; 
    QAction* titleName; 
    QAction* pointsAct; 
    QAction* logScaleAct; 
    QAction* legendAct; 

    std::vector<float> opacityTransferValues; 
    std::vector<float> colourTransferValues; 

    void createActions();
    void createConnections();
    void contextMenuEvent(QContextMenuEvent *event);

private:
    size_t tickNumber; 
    float slideInterval; 

    QSlider *lowSldr;
    QSlider *highSldr;
};

#endif // MILXQTPLOT_H