hedgehog.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_HEDGEHOG_H
#define VTKWRAP_HEDGEHOG_H

#include "vtkStructuredGrid.h"
#include "vtkHedgeHog.h"
#include "vtkPolyDataMapper.h"
#include "vtkActor.h"
#include "vtkProperty.h"

class HedgeHog // Porco-espinho
{
    friend std::ostream & operator<< (std::ostream & os, HedgeHog const & hh);
public:
    HedgeHog(vtkStructuredGrid * SGrid, double ScaleFactor=0.1)
    {
        _hedgehog        = vtkHedgeHog         ::New();
        _hedgehog_mapper = vtkPolyDataMapper   ::New();
        _hedgehog_actor  = vtkActor            ::New();
        _hedgehog        -> SetInput           (SGrid);
        _hedgehog        -> SetScaleFactor     (ScaleFactor);
        _hedgehog_mapper -> SetInputConnection (_hedgehog->GetOutputPort());
        _hedgehog_actor  -> SetMapper          (_hedgehog_mapper);
        _hedgehog_actor  -> GetProperty        () -> SetColor(0,0,0);   
    }
    ~HedgeHog()
    {
        _hedgehog        -> Delete();
        _hedgehog_mapper -> Delete();
        _hedgehog_actor  -> Delete();
    }
    vtkActor * GetActor() { return _hedgehog_actor; }
private:
    vtkHedgeHog       * _hedgehog;
    vtkPolyDataMapper * _hedgehog_mapper;
    vtkActor          * _hedgehog_actor;
}; // class HedgeHog

std::ostream & operator<< (std::ostream & os, HedgeHog const & hh)
{
    hh._hedgehog->Print(os);
    return os;
}

#endif // VTKWRAP_HEDGEHOG_H

// vim:fdm=marker

---