vtkfigVectorView.h

Go to the documentation of this file.

 
#ifndef VTKFIG_QUIVER_H
#define VTKFIG_QUIVER_H
 
#include <vtkPolyData.h>
#include <vtkDoubleArray.h>
 
#include "vtkfigFigure.h"
#include "vtkfigTools.h"
 
 
 
namespace vtkfig
{
 
 
  class VectorView: public Figure
    {
    public:
 
 
      static std::shared_ptr<VectorView> New();
      
      virtual std::string SubClassName() override final {return std::string("VectorView");}
    
 
      void ShowQuiver(bool b) { state.show_quiver=b;}
 
      void SetQuiverArrowScale(double scale) { state.quiver_arrowscale_user=scale;}
 
      void ShowQuiverColorbar(bool b) { state.show_quiver_colorbar=b;}
 
      void SetQuiverGrid(int nx, int ny);
 
      void SetQuiverGrid(int nx, int ny, int nz);
      
      template <class V>
        void SetQuiverPoints( const V&p);
      
 
 
      template <class V> 
        void  SetQuiverRGBTable(const V & tab, int lutsize);
      
      void SetQuiverRGBTable(RGBTable & tab, int lutsize);
      
 
      void ShowStreamLines(bool b) { state.show_stream=b;}
 
      void ShowStreamLineColorbar(bool b) { state.show_stream_colorbar=b;}
 
      void SetStreamLineMaximumPropagation(double l){ state.stream_maximum_propagation=l;}
 
      void SetStreamLineMaximumNumberOfSteps(int l){ state.stream_maximum_number_of_steps=l;}
 
      void SetStreamLineInitialIntegrationStep(double l){ state.stream_initial_integration_step=l;}
      
      void SetStreamLineMaximumIntegrationStep(double l){ state.stream_maximum_integration_step=l;}
      
      void SetStreamLineWidth(double w){state.stream_ribbonwidth=w;}
 
      
      template <class V>
        void SetStreamLineSeedPoints( const V&p);
 
 
      template <class V> 
        void  SetStreamLineRGBTable(const V & tab, int lutsize);
      
      void SetStreamLineRGBTable(RGBTable & tab, int lutsize);
      
 
    protected:
      VectorView();
      ~VectorView(){};
 
    private:
 
      void RTBuildVTKPipeline() override final;
      void ServerMPSend(vtkSmartPointer<internals::Communicator> communicator) override final;
      void ClientMPReceive(vtkSmartPointer<internals::Communicator> communicator) override final;
      
 
      template <class DATA> void  RTBuildVTKPipeline0();
 
        
      vtkSmartPointer<vtkPolyData> probePolyData;
      vtkSmartPointer<vtkPolyData> seedPolyData;
    };  
  
 
  template <class V>
    inline
    void VectorView::SetQuiverPoints( const V&p)
  {
 
    auto probePoints =  vtkSmartPointer<vtkPoints>::New();
    if (state.spacedim==2)
      for (int i=0;i<p.size();i+=2)
        probePoints->InsertNextPoint (p[i+0],p[i+1],0);
    
    if (state.spacedim==3)
      for (int i=0;i<p.size();i+=3)
        probePoints->InsertNextPoint (p[i+0],p[i+1],p[i+2]);
    probePolyData =vtkSmartPointer<vtkPolyData>::New();
    probePolyData->SetPoints(probePoints);
  };
  
 
 
  template <class V> 
    inline
    void  VectorView::SetQuiverRGBTable(const V & tab, int lutsize)
  {
    RGBTable rgbtab;
    rgbtab.resize(tab.size()/4);
    for (int i=0,j=0; i<tab.size(); i+=4,j++)
    {
      rgbtab[j].x=tab[i+0];
      rgbtab[j].r=tab[i+1];
      rgbtab[j].g=tab[i+2];
      rgbtab[j].b=tab[i+3];
    }
    SetQuiverRGBTable(rgbtab, lutsize);
  }
 
  template <class V> 
    inline
    void  VectorView::SetStreamLineRGBTable(const V & tab, int lutsize)
  {
    RGBTable rgbtab;
    rgbtab.resize(tab.size()/4);
    for (int i=0,j=0; i<tab.size(); i+=4,j++)
    {
      rgbtab[j].x=tab[i+0];
      rgbtab[j].r=tab[i+1];
      rgbtab[j].g=tab[i+2];
      rgbtab[j].b=tab[i+3];
    }
    SetStreamLineRGBTable(rgbtab, lutsize);
  }
 
 
  template <class V>
    inline
    void VectorView::SetStreamLineSeedPoints( const V&p)
  {
 
    auto seedPoints =  vtkSmartPointer<vtkPoints>::New();
    if (state.spacedim==2)
      for (size_t i=0;i<p.size();i+=2)
        seedPoints->InsertNextPoint (p[i+0]*coordinate_scale_factor,p[i+1]*coordinate_scale_factor,0);
    
    if (state.spacedim==3)
      for (size_t i=0;i<p.size();i+=3)
        seedPoints->InsertNextPoint (p[i+0]*coordinate_scale_factor,
                                     p[i+1]*coordinate_scale_factor,
                                     p[i+2]*coordinate_scale_factor);
    seedPolyData =vtkSmartPointer<vtkPolyData>::New();
    seedPolyData->SetPoints(seedPoints);
  };
  
  
}
 
 
#endif