vtkwrap_test10.cpp

/***********************************************************************************
 * 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                                 *
 ***********************************************************************************/

#include <iostream>

// VTK
#include "vtkPoints.h"
#include "vtkPolyData.h"
#include "vtkDelaunay3D.h"
#include "vtkShrinkFilter.h"
#include "vtkDataSetMapper.h"
#include "vtkOutlineFilter.h"
#include "vtkProperty.h"
#include "vtkPolyDataMapper.h"
#include "vtkGeometryFilter.h"
#include "vtkUnstructuredGridGeometryFilter.h"
#include "vtkUnstructuredGridWriter.h"
#include "vtkPolyDataWriter.h"

// MechCys
#include "util/exception.h"
#include "numerical/meshgrid.h"
#include "vtkwrap/vtkwin.h"

using std::cout;
using std::endl;

int main(int argc, char **argv) try
{
    if (argc!=2)
    {
        cout << "Usage:\n";
        cout << "      " << argv[0] << " NPTS\n";
        cout << endl;
        cout << "ex.:  " << argv[0] << " 6\n";
        cout << endl;
        return 1;
    }

    // Create a meshgrid
    int  N = atoi(argv[1]); if (N<2) throw new Fatal(_("NPTS must be greater than 1"));
    REAL L = 10.0;
    MeshGrid mg(0.0,L,atoi(argv[1]),  // X
                0.0,L,atoi(argv[1]),  // Y
                0.0,L,atoi(argv[1])); // Z

    // Create VTK points
    vtkPoints * points = vtkPoints::New();
    points->Allocate(mg.Length());
    for (int i=0; i<mg.Length(); ++i)
    {
        double P[3] = {mg.X(i), mg.Y(i), mg.Z(i)};
        points->InsertPoint(i,P);
    }

    // Create a 3D triangulation
    //   - The Tolerance is the distance that nearly coincident points are merged together.
    //   - Delaunay does better if points are well spaced.
    //   - The alpha value is the radius of circumcircles, circumspheres.
    //     Any mesh entity whose circumcircle is smaller than this value is output.
    vtkPolyData   * vertices = vtkPolyData::New();
    vtkDelaunay3D * delaunay = vtkDelaunay3D::New();
    vertices -> SetPoints(points);
    delaunay -> SetInput(vertices);
    delaunay -> SetTolerance(0.01);
    //delaunay -> SetAlpha(L);
    delaunay -> SetAlpha(0); 
    delaunay -> BoundingTriangulationOff();

    // Extract surface
    vtkGeometryFilter * geofil = vtkGeometryFilter::New();
    //vtkUnstructuredGridGeometryFilter * geofil = vtkUnstructuredGridGeometryFilter::New();
    geofil -> SetInputConnection(delaunay->GetOutputPort());

    // Shrink the result to help see it better.
    vtkShrinkFilter * shrink = vtkShrinkFilter::New();
    //shrink -> SetInputConnection(delaunay->GetOutputPort());
    shrink -> SetInputConnection(geofil->GetOutputPort());
    shrink -> SetShrinkFactor(0.85);

    // Mapper and actor
    vtkDataSetMapper * mapper = vtkDataSetMapper::New();
    vtkActor         * actor  = vtkActor::New();
    mapper -> SetInputConnection (shrink->GetOutputPort());
    actor  -> SetMapper          (mapper);
    actor  -> GetProperty        () -> SetColor(0,1,0);

    // Create a box around the points (Outline)
    vtkOutlineFilter  * outline        = vtkOutlineFilter::New();
    vtkPolyDataMapper * outline_mapper = vtkPolyDataMapper::New();
    vtkActor          * outline_actor  = vtkActor::New();
    outline        -> SetInputConnection(delaunay->GetOutputPort());
    outline_mapper -> SetInput    (outline->GetOutput());
    outline_actor  -> SetMapper   (outline_mapper);
    outline_actor  -> GetProperty () -> SetColor(0,0,0);

    // Window
    VTKWin win;
    win.AddActor(actor);
    win.AddActor(outline_actor);
    win.Show();

    // Write elements
    vtkUnstructuredGridWriter * writer = vtkUnstructuredGridWriter::New();
    writer -> SetInputConnection(delaunay->GetOutputPort());
    writer -> SetFileName("test10.elements.vtk");
    writer -> Write();

    // Write faces
    vtkPolyDataWriter * fwriter = vtkPolyDataWriter::New();
    //vtkUnstructuredGridWriter * fwriter = vtkUnstructuredGridWriter::New();
    fwriter -> SetInputConnection(geofil->GetOutputPort());
    fwriter -> SetFileName("test10.faces.vtk");
    fwriter -> Write();

    // Clean up
    points         -> Delete();
    vertices       -> Delete();
    delaunay       -> Delete();
    shrink         -> Delete();
    mapper         -> Delete();
    actor          -> Delete();
    outline        -> Delete();
    outline_mapper -> Delete();
    outline_actor  -> Delete();
    writer         -> Delete();
    fwriter        -> Delete();

    return 0;
}
catch (Exception * e) //{{{ 
{
    e->Cout();
    if (e->IsFatal()) {delete e; exit(1);}
    delete e;
}
catch (char const * m)
{
    std::cout << "Fatal: " << m << std::endl;
    exit (1);
}
catch (...)
{
    std::cout << "Some exception (...) ocurred\n";
} //}}} 

// vim:fdm=marker

---