ugridlabels.h

/*************************************************************************************
 * MechSys - A C++ library to simulate (Continuum) Mechanical Systems                *
 * Copyright (C) 2005 Dorival de Moraes Pedroso <dorival.pedroso at gmail.com>       *
 * Copyright (C) 2005 Raul Dario Durand Farfan  <raul.durand at gmail.com>           *
 *                                                                                   *
 * This file is part of MechSys.                                                     *
 *                                                                                   *
 * MechSys 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.                                                                          *
 *                                                                                   *
 * MechSys 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      *
 * MechSys; if not, write to the Free Software Foundation, Inc., 51 Franklin Street, *
 * Fifth Floor, Boston, MA 02110-1301, USA                                           *
 *************************************************************************************/

#ifndef MECHSYS_UGRIDLABELS_H
#define MECHSYS_UGRIDLABELS_H

// VTK
#include "vtkRenderer.h"
#include "vtkUnstructuredGrid.h"
#include "vtkIdFilter.h"
#include "vtkCellCenters.h"
#include "vtkSelectVisiblePoints.h"
#include "vtkBorderRepresentation.h"
#include "vtkBorderWidget.h"
#include "vtkLabeledDataMapper.h"
#include "vtkTextProperty.h"
#include "vtkActor2D.h"
#include "vtkProperty.h"
#include "vtkProperty2D.h"

// MechSys
#include "vtkwrap/selectcallback.h"
#include "vtkwrap/colors.h"
#include "vtkwrap/vtkwin.h"

class UGridLabels
{
public:
    // Constructor
    UGridLabels(VTKWin & Win, vtkUnstructuredGrid * Dataset);
    UGridLabels(VTKWin & Win, vtkAlgorithmOutput * OutputPort);

    // Destructor
    ~UGridLabels();

    // Methods
    void AddActorsTo   (VTKWin & Win);
    void DelActorsFrom (VTKWin & Win);
    void EnableWidget  (int Enable=1) { _wid->SetEnabled(Enable); }

private:
    // Data
    vtkIdFilter             * _ids;
    vtkCellCenters          * _cen;
    vtkSelectVisiblePoints  * _nodes_vis_pts;
    vtkSelectVisiblePoints  * _cells_vis_pts;
    vtkBorderRepresentation * _rep;
    vtkBorderWidget         * _wid;
    SelectCallBack          * _sel_cback;
    vtkTextProperty         * _tnodes;
    vtkTextProperty         * _tcells;
    vtkLabeledDataMapper    * _nodes_ldm;
    vtkLabeledDataMapper    * _cells_ldm;
    vtkActor2D              * _lnodes_actor;
    vtkActor2D              * _lcells_actor;

    // Private methods
    void _init_for_given_ids_and_cen(VTKWin & Win);

}; // class UGridLabels

inline UGridLabels::UGridLabels(VTKWin & Win, vtkUnstructuredGrid * Dataset) // {{{
{
    // IdFilter and CellCenters
    _ids = vtkIdFilter    ::New();
    _cen = vtkCellCenters ::New();
    _ids -> SetInput           (Dataset);
    _ids -> PointIdsOn         ();
    _ids -> CellIdsOn          ();
    _ids -> FieldDataOn        ();
    _cen -> SetInputConnection (_ids->GetOutputPort());

    // Init
    _init_for_given_ids_and_cen(Win);

} // }}}

inline UGridLabels::UGridLabels(VTKWin & Win, vtkAlgorithmOutput * OutputPort) // {{{
{
    // IdFilter and CellCenters
    _ids = vtkIdFilter    ::New();
    _cen = vtkCellCenters ::New();
    _ids -> SetInputConnection (OutputPort);
    _ids -> PointIdsOn         ();
    _ids -> CellIdsOn          ();
    _ids -> FieldDataOn        ();
    _cen -> SetInputConnection (_ids->GetOutputPort());

    // Init
    _init_for_given_ids_and_cen(Win);

} // }}}

inline void UGridLabels::_init_for_given_ids_and_cen(VTKWin & Win) // {{{
{
    // Select visible points
    _nodes_vis_pts = vtkSelectVisiblePoints::New();
    _nodes_vis_pts -> SetInputConnection (_ids->GetOutputPort());
    _nodes_vis_pts -> SetRenderer        (Win.GetRen());
    _nodes_vis_pts -> SelectionWindowOn  ();

    // Select visible points
    _cells_vis_pts = vtkSelectVisiblePoints::New();
    _cells_vis_pts -> SetInputConnection (_cen->GetOutputPort());
    _cells_vis_pts -> SetRenderer        (Win.GetRen());
    _cells_vis_pts -> SelectionWindowOn  ();

    // Border widget
    _rep       = vtkBorderRepresentation ::New();
    _wid       = vtkBorderWidget         ::New();
    _sel_cback = SelectCallBack          ::New();
    _rep       -> SetPosition       (0.1,0.1);
    _rep       -> SetPosition2      (0.8,0.8);
    _rep       -> SetShowBorderToOn ();
    _rep       -> GetBorderProperty ()->SetColor(CLR["blue"].C);
    _wid       -> SetInteractor     (Win.GetIRen());
    _wid       -> SetRepresentation (_rep);
    _wid       -> SelectableOn      ();
    _wid       -> AddObserver       (vtkCommand::EnableEvent     , _sel_cback);
    _wid       -> AddObserver       (vtkCommand::InteractionEvent, _sel_cback);
    _sel_cback -> Setup             (Win.GetRen(), _rep, _nodes_vis_pts, _cells_vis_pts);

    // Nodes labels text properties
    _tnodes = vtkTextProperty::New();
    _tnodes -> SetFontSize                (18);
    _tnodes -> SetFontFamilyToArial       ();
    _tnodes -> SetJustificationToCentered ();
    _tnodes -> BoldOn                     ();
    _tnodes -> ItalicOn                   ();
    _tnodes -> ShadowOn                   ();
    _tnodes -> SetColor                   (CLR["blue"].C);

    // Cells labels text properties
    _tcells = vtkTextProperty::New();
    _tcells -> SetFontSize                (18);
    _tcells -> SetFontFamilyToArial       ();
    _tcells -> SetJustificationToCentered ();
    _tcells -> BoldOn                     ();
    _tcells -> ItalicOn                   ();
    _tcells -> ShadowOn                   ();
    _tcells -> SetColor                   (CLR["red"].C);

    // Nodes Labels
    _nodes_ldm    = vtkLabeledDataMapper ::New();
    _lnodes_actor = vtkActor2D           ::New();
    _nodes_ldm    -> SetInputConnection           (_nodes_vis_pts->GetOutputPort());
    _nodes_ldm    -> SetLabelFormat               ("%g");
    _nodes_ldm    -> SetLabelModeToLabelFieldData ();
    _nodes_ldm    -> SetLabelTextProperty         (_tnodes);
    _lnodes_actor -> SetMapper                    (_nodes_ldm);

    // Cells Labels
    _cells_ldm    = vtkLabeledDataMapper ::New();
    _lcells_actor = vtkActor2D           ::New();
    _cells_ldm    -> SetInputConnection           (_cells_vis_pts->GetOutputPort());
    _cells_ldm    -> SetLabelFormat               ("%g");
    _cells_ldm    -> SetLabelModeToLabelFieldData ();
    _cells_ldm    -> SetLabelTextProperty         (_tcells);
    _lcells_actor -> SetMapper                    (_cells_ldm);
} // }}}

inline UGridLabels::~UGridLabels() // {{{
{
    _ids           -> Delete();
    _cen           -> Delete();
    _nodes_vis_pts -> Delete();
    _cells_vis_pts -> Delete();
    _rep           -> Delete();
    _wid           -> Delete();
    _sel_cback     -> Delete();
    _tnodes        -> Delete();
    _tcells        -> Delete();
    _nodes_ldm     -> Delete();
    _cells_ldm     -> Delete();
    _lnodes_actor  -> Delete();
    _lcells_actor  -> Delete();
} // }}}

inline void UGridLabels::AddActorsTo(VTKWin & Win) // {{{
{
    Win.AddActor2D(_lnodes_actor);
    Win.AddActor2D(_lcells_actor);
} // }}}

inline void UGridLabels::DelActorsFrom(VTKWin & Win) // {{{
{
    Win.DelActor2D(_lnodes_actor);
    Win.DelActor2D(_lcells_actor);
} // }}}

#endif // MECHSYS_UGRIDLABELS_H

// vim:fdm=marker

---