vtkwrap_test07.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>
#include <cmath>
#include <vector>

#include "numerical/meshgrid.h"

#include "vtkwrap/vtkwin.h"
#include "vtkwrap/sgridoutline.h"
#include "vtkwrap/axes.h"
#include "vtkwrap/colors.h"
#include "vtkwrap/cutclip.h"
#include "vtkwrap/sphere.h"
#include "vtkwrap/cube.h"
#include "vtkwrap/structgrid.h"
#include "vtkwrap/sgridisosurf.h"

int main(int argc, char **argv)
{
    // Window
    VTKWin win;

    // Axes
    Axes ax(2, true);
    win.AddActor(ax.GetActor());

    // Generate grid
    const int n_pts = 20;
    MeshGrid mg(0,1,n_pts, 0,1,n_pts, 0,1,n_pts);
    int Size = mg.Length();

    // Calculate function values
    double * F = new double [Size];
    for (int i=0; i<Size; ++i)
        F[i] = pow(mg.X(i),2.0)+pow(mg.Y(i),2.0)+pow(mg.Z(i),2.0) - 1.0;

    // Create a VTK structured grid
    StructGrid sg(mg.X(),n_pts, mg.Y(),n_pts, mg.Z(),n_pts, F);

    // LookupTable of colors
    vtkLookupTable * lt = vtkLookupTable::New();
    lt->SetNumberOfColors(1);
    lt->Build();
    lt->SetTableValue(0,CLR["blue"].C);

    // Generate the Yield Surface
    SGridIsoSurf sgiso(sg.GetGrid(), /*F*/0.0, lt);
    win.AddActor(sgiso.GetActor());

    // Show
    win.Show();

    // Clean up
    delete [] F;
    lt->Delete();

    return 0;
}

---