vtkwin.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_VTKWIN_H
#define VTKWRAP_VTKWIN_H

#include "vtkCamera.h"
#include "vtkRenderer.h"
#include "vtkRenderWindow.h"
#include "vtkRenderWindowInteractor.h"
#include "vtkInteractorStyleSwitch.h"
#include "vtkActor.h"
#include "vtkActor2D.h"
#include "vtkWindowToImageFilter.h"
#include "vtkPNGWriter.h"

class VTKWin
{
public:
    // Constructor
    VTKWin(int Width=300, int Height=300);

    // Alternative constructor
    VTKWin(vtkRenderWindowInteractor * vtkRWI, int Width=300, int Height=300);

    // Destructor
    ~VTKWin();

    // Methods
    void AddActor        (vtkActor   * TheActor) { _renderer -> AddActor      (TheActor); }
    void AddActor2D      (vtkActor2D * TheActor) { _renderer -> AddActor2D    (TheActor); }
    void DelActor        (vtkActor   * TheActor) { _renderer -> RemoveActor   (TheActor); }
    void DelActor2D      (vtkActor2D * TheActor) { _renderer -> RemoveActor2D (TheActor); }
    void AddLight        (vtkLight   * Light)    { _renderer -> AddLight      (Light);    }
    void Show            ();
    void SetViewDefault  ();
    void SetViewPIplane  ();
    void WritePNG        (char const * Filename);
    void ResetCamera     ();

    // Access methods
    vtkRenderer               * GetRen    () { return _renderer;   }
    vtkRenderWindowInteractor * GetIRen   () { return _interactor; }
    vtkCamera                 * GetCamera () { return _camera;     }

private:
    // Data
    vtkCamera                 * _camera;
    vtkRenderer               * _renderer;
    vtkRenderWindow           * _ren_win;
    vtkRenderWindowInteractor * _interactor;
    vtkInteractorStyleSwitch  * _int_switch;

}; // class VTKWin




inline VTKWin::VTKWin(int Width, int Height) // {{{
{
    _camera = vtkCamera::New();
    SetViewDefault();

    _renderer = vtkRenderer::New();
    _ren_win  = vtkRenderWindow::New();
    _ren_win->AddRenderer(_renderer);
    _ren_win->SetSize(Width,Height);

    _interactor = vtkRenderWindowInteractor::New();
    _int_switch = vtkInteractorStyleSwitch::New();
    _interactor -> SetRenderWindow(_ren_win);
    _interactor -> SetInteractorStyle(_int_switch);
    _int_switch -> SetCurrentStyleToTrackballCamera();

    _renderer -> SetActiveCamera(_camera);
    _renderer -> SetBackground(1,1,1);

} // }}}

inline VTKWin::VTKWin(vtkRenderWindowInteractor * vtkRWI, int Width, int Height) // {{{
{
    _camera = vtkCamera::New();
    SetViewDefault();

    _renderer = vtkRenderer::New();
    _ren_win  = vtkRenderWindow::New();
    _ren_win->AddRenderer(_renderer);
    _ren_win->SetSize(Width,Height);

    _interactor = vtkRWI;
    _int_switch = vtkInteractorStyleSwitch::New();
    _interactor -> SetRenderWindow(_ren_win);
    _interactor -> SetInteractorStyle(_int_switch);
    _int_switch -> SetCurrentStyleToTrackballCamera();

    _renderer -> SetActiveCamera(_camera);
    _renderer -> SetBackground(1,1,1);

} // }}}

inline VTKWin::~VTKWin() // {{{
{
    _camera     -> Delete();
    _renderer   -> Delete();
    _ren_win    -> Delete();
    _interactor -> Delete();
    _int_switch -> Delete();

} // }}}

inline void VTKWin::Show() // {{{
{
    _ren_win    -> Render();
    _interactor -> Start();

} // }}}

inline void VTKWin::SetViewDefault() // {{{
{
    _camera->SetViewUp     (0,0,1);
    _camera->SetPosition   (2,1,1);
    _camera->SetFocalPoint (0,0,0);

} // }}}

inline void VTKWin::SetViewPIplane() // {{{
{
    _camera->SetViewUp     (0,0,1);
    _camera->SetPosition   (1,1,1);
    _camera->SetFocalPoint (0,0,0);

} // }}}

inline void VTKWin::WritePNG(char const * Filename) // {{{
{
    // Window to image filter
    vtkWindowToImageFilter * win_to_img = vtkWindowToImageFilter::New();
    win_to_img -> SetInput(_ren_win);
    win_to_img -> Update();

    // PNG writer
    vtkPNGWriter * writer = vtkPNGWriter::New();
    writer -> SetInput(win_to_img->GetOutput());
    writer -> SetFileName(Filename);
    writer -> Write();

    // Clean up
    win_to_img -> Delete();
    writer     -> Delete();

} // }}}

inline void VTKWin::ResetCamera() // {{{
{
    SetViewDefault();
    _renderer -> SetActiveCamera(_camera);
    _renderer -> ResetCamera();
    _renderer -> SetBackground(1,1,1);

} // }}}

#endif // VTKWRAP_VTKWIN_H

// vim:fdm=marker

---