ugridclipper.h

/*************************************************************************************
 * MechSys - A C++ library to simulate (Continuum) Mechanical Systems                *
 * Copyright (C) 2005 Dorival de Moraes Pedroso <dorival.pedroso at gmail.com>       *
 * Copyright (C) 2005 Raul Dario Durand Farfan  <raul.durand at gmail.com>           *
 *                                                                                   *
 * This file is part of MechSys.                                                     *
 *                                                                                   *
 * MechSys is free software; you can redistribute it and/or modify it under the      *
 * terms of the GNU General Public License as published by the Free Software         *
 * Foundation; either version 2 of the License, or (at your option) any later        *
 * version.                                                                          *
 *                                                                                   *
 * MechSys 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 General Public License for more details.          *
 *                                                                                   *
 * You should have received a copy of the GNU General Public License along with      *
 * MechSys; if not, write to the Free Software Foundation, Inc., 51 Franklin Street, *
 * Fifth Floor, Boston, MA 02110-1301, USA                                           *
 *************************************************************************************/

#ifndef MECHSYS_UGRIDCLIPPER_H
#define MECHSYS_UGRIDCLIPPER_H

#ifdef HAVE_CONFIG_H
  #include "config.h"
#else
  #ifndef REAL
    #define REAL double
  #endif
#endif

// VTK
#include "vtkRenderer.h"
#include "vtkUnstructuredGrid.h"
#include "vtkPlane.h"
#include "vtkClipDataSet.h"
#include "vtkDataSetMapper.h"
#include "vtkActor.h"
#include "vtkImplicitPlaneWidget2.h"
#include "vtkImplicitPlaneRepresentation.h"
#include "vtkProperty.h"
#include "vtkAlgorithmOutput.h"

// MechSys
#include "vtkwrap/vtkwin.h"
#include "vtkwrap/colors.h"
#include "vtkwrap/updateplanecallback.h"

class UGridClipper
{
public:
    // Constructor
    UGridClipper(VTKWin & Win, vtkUnstructuredGrid * Dataset, vtkAlgorithmOutput * OutputPort=NULL,
                 REAL x0=0, REAL y0=0, REAL z0=0,
                 REAL nx=1, REAL ny=1, REAL nz=1);

    // Destructor
    ~UGridClipper();
    
    // Methods
    void AddActorsTo   (VTKWin & Win);
    void DelActorsFrom (VTKWin & Win);
    void EnableWidget  (int Enable=1) { _widget->SetEnabled(Enable); }
    vtkAlgorithmOutput * GetOutputPort() { if (_clipper!=NULL) return _clipper->GetOutputPort(); else return NULL; }

private:
    // Data
    vtkPlane                       * _plane;
    vtkClipDataSet                 * _clipper;
    vtkDataSetMapper               * _mapper;
    vtkActor                       * _actor;
    UpdatePlaneCallBack            * _update_cback;
    vtkImplicitPlaneWidget2        * _widget;
    vtkImplicitPlaneRepresentation * _rep;
    
}; // class UGridClipper




inline UGridClipper::UGridClipper(VTKWin & Win, vtkUnstructuredGrid * Dataset, vtkAlgorithmOutput * AlgoOut, // {{{
                                  REAL x0, REAL y0, REAL z0,
                                  REAL nx, REAL ny, REAL nz)
{
    // Clipper
    _plane   = vtkPlane         ::New();
    _clipper = vtkClipDataSet   ::New();
    _mapper  = vtkDataSetMapper ::New();
    _actor   = vtkActor         ::New();
    _plane   -> SetOrigin          (x0,y0,z0);
    _plane   -> SetNormal          (nx,ny,nz); if (AlgoOut==NULL) {
    _clipper -> SetInput           (Dataset);  } else {
    _clipper -> SetInputConnection (AlgoOut);  }
    _clipper -> SetClipFunction    (_plane);
    _clipper -> InsideOutOn        ();
    _mapper  -> SetInputConnection (_clipper->GetOutputPort());
    _actor   -> SetMapper          (_mapper);

    // Widget
    _update_cback = UpdatePlaneCallBack            ::New();
    _rep          = vtkImplicitPlaneRepresentation ::New();
    _widget       = vtkImplicitPlaneWidget2        ::New();
    _rep          -> SetPlaceFactor     (1.25);
    _rep          -> PlaceWidget        (Dataset->GetBounds());
    _rep          -> DrawPlaneOff       ();
    _rep          -> GetNormalProperty  ()->SetColor(CLR["peacock"].C);
    _rep          -> GetOutlineProperty ()->SetColor(CLR["black"].C);
    _rep          -> GetEdgesProperty   ()->SetColor(CLR["salmon"].C);
    _rep          -> SetOrigin          (x0,y0,z0);
    _rep          -> SetNormal          (nx,ny,nz);
    _update_cback -> SetPlane           (_plane);
    _widget       -> SetInteractor      (Win.GetIRen());
    _widget       -> SetRepresentation  (_rep);;
    _widget       -> AddObserver        (vtkCommand::InteractionEvent, _update_cback);
    //_widget       -> AddObserver    (vtkCommand::EnableEvent     , _update_cback);

} // }}}

inline UGridClipper::~UGridClipper() // {{{
{
    _plane        -> Delete();
    _clipper      -> Delete();
    _mapper       -> Delete();
    _actor        -> Delete();
    _update_cback -> Delete();
    _widget       -> Delete();
} // }}}

inline void UGridClipper::AddActorsTo(VTKWin & Win) // {{{
{
    Win.AddActor(_actor);
} // }}}

inline void UGridClipper::DelActorsFrom(VTKWin & Win) // {{{
{
    Win.DelActor(_actor);
} // }}}

#endif // MECHSYS_UGRIDCLIPPER_H

// vim:fdm=marker

---