plane.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_PLANE_H
#define VTKWRAP_PLANE_H

#include <string>

#include "vtkPlaneSource.h"
#include "vtkPolyDataMapper.h"
#include "vtkActor.h"
#include "vtkProperty.h"

#include "colors.h"

class Plane
{
    friend std::ostream & operator<< (std::ostream & os, Plane const & hh);
public:
    Plane(double Xc, double Yc, double Zc,
          double nx, double ny, double nz, char const * Color=NULL, double Opacity=1.0)
    {
        // Color
        std::string * color;
        if (Color==NULL) color = new std::string ("red");
        else             color = new std::string (Color);

        // Create object
        _plane        = vtkPlaneSource    ::New();
        _plane_mapper = vtkPolyDataMapper ::New();
        _plane_actor  = vtkActor          ::New();
        _plane        -> SetCenter                 (Xc,Yc,Zc);
        _plane        -> SetNormal                 (nx,ny,nz);
        _plane_mapper -> SetInputConnection        (_plane->GetOutputPort());
        _plane_actor  -> SetMapper                 (_plane_mapper);
        _plane_actor  -> GetProperty()->SetColor   (CLR[color->c_str()].C); 
        _plane_actor  -> GetProperty()->SetOpacity (Opacity);

        // Clean up
        delete color;
    }
    Plane(double Xo, double Yo, double Zo,
          double X1, double Y1, double Z1,
          double X2, double Y2, double Z2,
          double nx, double ny, double nz, char const * Color=NULL, double Opacity=1.0)
    {
        // Color
        std::string * color;
        if (Color==NULL) color = new std::string ("red");
        else             color = new std::string (Color);

        // Create object
        _plane        = vtkPlaneSource    ::New();
        _plane_mapper = vtkPolyDataMapper ::New();
        _plane_actor  = vtkActor          ::New();
        _plane        -> SetOrigin                 (Xo,Yo,Zo);
        _plane        -> SetNormal                 (nx,ny,nz);
        _plane        -> SetPoint1                 (X1,Y1,Z1);
        _plane        -> SetPoint2                 (X2,Y2,Z2);
        _plane_mapper -> SetInputConnection        (_plane->GetOutputPort());
        _plane_actor  -> SetMapper                 (_plane_mapper);
        _plane_actor  -> GetProperty()->SetColor   (CLR[color->c_str()].C); 
        _plane_actor  -> GetProperty()->SetOpacity (Opacity);

        // Clean up
        delete color;
    }
    ~Plane()
    {
        _plane        -> Delete();
        _plane_mapper -> Delete();
        _plane_actor  -> Delete();
    }
    vtkActor           * GetActor()      { return _plane_actor;            }
    vtkAlgorithmOutput * GetOutputPort() { return _plane->GetOutputPort(); }
    vtkPlaneSource     * GetObject()     { return _plane;                  }
private:
    vtkPlaneSource    * _plane;
    vtkPolyDataMapper * _plane_mapper;
    vtkActor          * _plane_actor;
}; // class Plane

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

#endif // VTKWRAP_PLANE_H

// vim:fdm=marker

---