ModelPart.h

Go to the documentation of this file.

 
#ifndef VIEWER_MODELPART_H
#define VIEWER_MODELPART_H
 
#include <QString>
#include <QList>
#include <QVariant>
 
#include <vtkSmartPointer.h>
#include <vtkMapper.h>
#include <vtkActor.h>
#include <vtkSTLReader.h>
#include <vtkColor.h>
#include <vtkShrinkFilter.h>
#include <vtkClipDataSet.h>
#include <vtkPlane.h>
#include <vtkDataSetMapper.h>
#include <vtkDataSet.h>
 
class ModelPart {
public:
    ModelPart(const QList<QVariant>& data, ModelPart* parent = nullptr);
 
    ~ModelPart();
 
    void appendChild(ModelPart* item);
    ModelPart* child(int row);
 
    int childCount() const;   
    int columnCount() const;
 
    QVariant data(int column) const;
 
 
    void set( int column, const QVariant& value );
 
    ModelPart* parentItem();
 
    int row() const;
 
 
    void setColour(const unsigned char R, const unsigned char G, const unsigned char B);
 
    unsigned char getColourR();
 
    unsigned char getColourG();
 
    unsigned char getColourB();
 
    void setVisible(bool isVisible);
 
    bool visible();
    
    void loadSTL(QString fileName);
 
    vtkSmartPointer<vtkActor> getActor();
 
    vtkSmartPointer<vtkActor> getNewActor();
 
    bool removeChild(int row);
 
    // Shrink filter for VR rendering
    void setShrinkFilter(bool enabled, double factor);
 
    bool shrinkFilterEnabled() const;
 
    double shrinkFactor() const;
 
    void updatePipeline();
 
    void updateVRPipeline();
 
    // Clip filter for VR rendering
 
    void setClipFilter(bool enabled, int axis, double value, bool invert);
 
    bool clipFilterEnabled() const;
 
    int clipAxis() const;
 
    double clipValue() const;
 
    bool clipInvert() const;
 
private:
    QList<ModelPart*>                           m_childItems;       
    QList<QVariant>                             m_itemData;         
    ModelPart*                                  m_parentItem;       
    /* These are some typical properties that I think the part will need, you might
     * want to add you own.
     */
    bool                                        isVisible;          
    unsigned char colourR;   
    unsigned char colourG;   
    unsigned char colourB;   
    /* These are vtk properties that will be used to load/render a model of this part,
     * commented out for now but will be used later
     */
    vtkSmartPointer<vtkSTLReader>               file;               
    vtkSmartPointer<vtkMapper>                  mapper;             
    vtkSmartPointer<vtkActor>                   actor;              
    vtkSmartPointer<vtkActor>                   vrActor;            
    vtkColor3<unsigned char>                    colour;             
    // Shrink filter for VR rendering
    bool applyShrinkFilter;     
    double shrinkFilterFactor;  
    vtkSmartPointer<vtkShrinkFilter> shrinkFilter;
 
    // Clip filter for VR rendering
    bool applyClipFilter;    
    int clipFilterAxis;      
    double clipFilterValue;  
    bool invertClipFilter;   
    vtkSmartPointer<vtkClipDataSet> clipFilter;
    vtkSmartPointer<vtkPlane> clipPlane;
    vtkSmartPointer<vtkDataSetMapper> vrMapper;
    vtkSmartPointer<vtkDataSet> vrFilteredData;
 
};  
 
 
#endif