cube.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_CUBE_H
#define VTKWRAP_CUBE_H

#include <string>

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

#include "colors.h"

class Cube
{
    friend std::ostream & operator<< (std::ostream & os, Cube const & hh);
public:
    Cube(double Xc, double Yc, double Zc, double Xlen, double Ylen, double Zlen, 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
        _cube        = vtkCubeSource     ::New();
        _cube_mapper = vtkPolyDataMapper ::New();
        _cube_actor  = vtkActor          ::New();
        _cube        -> SetCenter                 (Xc,Yc,Zc);
        _cube        -> SetXLength                (Xlen);
        _cube        -> SetYLength                (Ylen);
        _cube        -> SetZLength                (Zlen);
        _cube_mapper -> SetInputConnection        (_cube->GetOutputPort());
        _cube_actor  -> SetMapper                 (_cube_mapper);
        _cube_actor  -> GetProperty()->SetColor   (CLR[color->c_str()].C);  
        _cube_actor  -> GetProperty()->SetOpacity (Opacity);

        // Borders
        _wire_mapper = vtkPolyDataMapper ::New();
        _wire_actor  = vtkActor          ::New();
        _wire_mapper -> SetInput            (_cube->GetOutput());
        _wire_mapper -> ScalarVisibilityOff ();
        _wire_actor  -> SetMapper           (_wire_mapper);
        _wire_actor  -> GetProperty         ()->SetRepresentationToWireframe();

        _cube_mapper->SetResolveCoincidentTopologyPolygonOffsetParameters(0,1);
        _cube_mapper->SetResolveCoincidentTopologyToPolygonOffset();
        _wire_mapper->SetResolveCoincidentTopologyPolygonOffsetParameters(1,1);
        _wire_mapper->SetResolveCoincidentTopologyToPolygonOffset();

        // Same color for inside and outside edges
        _wire_mapper->ScalarVisibilityOff();
        _wire_actor->GetProperty()->SetAmbient(1.0);
        _wire_actor->GetProperty()->SetDiffuse(0.0);
        _wire_actor->GetProperty()->SetSpecular(0.0);

        // Clean up
        delete color;
    }
    ~Cube()
    {
        _cube        -> Delete();
        _cube_mapper -> Delete();
        _cube_actor  -> Delete();
        _wire_mapper -> Delete();
        _wire_actor  -> Delete();
    }
    vtkActor           * GetActor()      { return _cube_actor;            }
    vtkActor           * GetWireActor()  { return _wire_actor;            }
    vtkAlgorithmOutput * GetOutputPort() { return _cube->GetOutputPort(); }
    vtkCubeSource      * GetObject()     { return _cube;                  }
    Cube & WireWidth (int          Width) { _wire_actor->GetProperty()->SetLineWidth(Width);    return (*this); }
    Cube & WireColor (char const * Color) { _wire_actor->GetProperty()->SetColor(CLR[Color].C); return (*this); }
private:
    vtkCubeSource     * _cube;
    vtkPolyDataMapper * _cube_mapper;
    vtkActor          * _cube_actor;
    vtkPolyDataMapper * _wire_mapper;
    vtkActor          * _wire_actor;

}; // class Cube

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

#endif // VTKWRAP_CUBE_H

// vim:fdm=marker

---