sphere.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_SPHERE_H
#define VTKWRAP_SPHERE_H

#include <string>

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

#include "colors.h"

class Sphere
{
    friend std::ostream & operator<< (std::ostream & os, Sphere const & hh);
public:
    Sphere(double Xc, double Yc, double Zc, double Radius, char const * Color=NULL, double Opacity=1.0, int ThetaRes=20, int PhiRes=20)
    {
        // Color
        std::string * color;
        if (Color==NULL) color = new std::string ("red");
        else             color = new std::string (Color);

        // Create object
        _sphere        = vtkSphereSource   ::New();
        _sphere_mapper = vtkPolyDataMapper ::New();
        _sphere_actor  = vtkActor          ::New();
        _sphere        -> SetCenter                 (Xc,Yc,Zc);
        _sphere        -> SetRadius                 (Radius);
        _sphere        -> SetThetaResolution        (ThetaRes);
        _sphere        -> SetPhiResolution          (PhiRes);
        _sphere_mapper -> SetInputConnection        (_sphere->GetOutputPort());
        _sphere_actor  -> SetMapper                 (_sphere_mapper);
        _sphere_actor  -> GetProperty()->SetColor   (CLR[color->c_str()].C);    
        _sphere_actor  -> GetProperty()->SetOpacity (Opacity);

        // Clean up
        delete color;
    }
    ~Sphere()
    {
        _sphere        -> Delete();
        _sphere_mapper -> Delete();
        _sphere_actor  -> Delete();
    }
    vtkActor           * GetActor()      { return _sphere_actor;            }
    vtkAlgorithmOutput * GetOutputPort() { return _sphere->GetOutputPort(); }
    vtkSphereSource    * GetObject()     { return _sphere;                  }
private:
    vtkSphereSource   * _sphere;
    vtkPolyDataMapper * _sphere_mapper;
    vtkActor          * _sphere_actor;
}; // class Sphere

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

#endif // VTKWRAP_SPHERE_H

// vim:fdm=marker

---