sgridisosurf.h

/***********************************************************************************
 * VTKwrap - Simple VTK wrappers                                                   *
 * Copyright (C) 2005 Dorival de Moraes Pedroso <dorival.pedroso at gmail.com>     *
 *                                                                                 *
 * This file is part of VTKwrap.                                                   *
 *                                                                                 *
 * VTKwrap 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.                                                                        *
 *                                                                                 *
 * VTKwrap 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    *
 * VTKwrap; if not, write to the Free Software Foundation, Inc., 51 Franklin       *
 * Street, Fifth Floor, Boston, MA 02110-1301, USA                                 *
 ***********************************************************************************/

#ifndef VTKWRAP_SGRIDISOSURF_H
#define VTKWRAP_SGRIDISOSURF_H

#include "vtkStructuredGrid.h"
#include "vtkMarchingContourFilter.h"
#include "vtkPolyDataMapper.h"
#include "vtkActor.h"
#include "vtkProperty.h"
#include "vtkLookupTable.h"

class SGridIsoSurf
{
    friend std::ostream & operator<< (std::ostream & os, SGridIsoSurf const & sgiso);
public:
    SGridIsoSurf(vtkStructuredGrid * SGrid, double f, vtkLookupTable * lt=NULL) // F==level value
    {
        _isosurf        = vtkMarchingContourFilter ::New();
        _isosurf_mapper = vtkPolyDataMapper        ::New();
        _isosurf_actor  = vtkActor                 ::New();
        _isosurf        -> SetInput                (SGrid);
        _isosurf        -> SetValue                (0,f);
        _isosurf_mapper -> SetInputConnection      (_isosurf->GetOutputPort()); if (lt!=NULL)
        _isosurf_mapper -> SetLookupTable          (lt);
        _isosurf_actor  -> SetMapper               (_isosurf_mapper);

    }
    SGridIsoSurf(vtkStructuredGrid * SGrid, int nSurfs, double fMin, double fMax, vtkLookupTable * lt=NULL) // nSurfs between fMin and fMax
    {
        _isosurf        = vtkMarchingContourFilter ::New();
        _isosurf_mapper = vtkPolyDataMapper        ::New();
        _isosurf_actor  = vtkActor                 ::New();
        _isosurf        -> SetInput                (SGrid);
        _isosurf        -> GenerateValues          (nSurfs,fMin,fMax);
        _isosurf_mapper -> SetInputConnection      (_isosurf->GetOutputPort()); if (lt!=NULL)
        _isosurf_mapper -> SetLookupTable          (lt);
        _isosurf_actor  -> SetMapper               (_isosurf_mapper);
    }
    ~SGridIsoSurf()
    {
        _isosurf        -> Delete();
        _isosurf_mapper -> Delete();
        _isosurf_actor  -> Delete();
    }
    vtkActor * GetActor() { return _isosurf_actor; }
    vtkMarchingContourFilter * GetIsoSurf() { return _isosurf; }
private:
    vtkMarchingContourFilter * _isosurf;
    vtkPolyDataMapper        * _isosurf_mapper;
    vtkActor                 * _isosurf_actor;
}; // class SGridIsoSurf

std::ostream & operator<< (std::ostream & os, SGridIsoSurf const & sgiso)
{
    sgiso._isosurf->Print(os);
    return os;
}

#endif // VTKWRAP_SGRIDISOSURF_H

// vim:fdm=marker

---