vtkwrap_test03.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 "numerical/meshgrid.h"

#include "vtkwrap/structgrid.h"
#include "vtkwrap/hedgehog.h"
#include "vtkwrap/vtkwin.h"
#include "vtkwrap/sgridoutline.h"
#include "vtkwrap/sgridisosurf.h"
#include "vtkwrap/axes.h"
#include "vtkwrap/colors.h"

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

int main(int argc, char **argv)
{
    if (argc!=4)
    {
        cout << "Usage:\n";
        cout << "      " << argv[0] << " NP NQ NT\n";
        cout << endl;
        cout << "ex.:  " << argv[0] << " 20 20 20\n";
        cout << endl;
        return 1;
    }

    MeshGrid mg(0.0,1.0,atoi(argv[1]), 0.0,1.0,atoi(argv[2]), 0.0,1.0,atoi(argv[3]));
    //cout << mg << endl;
    
    double * F = new double [mg.Length()];
    double c[] = {0.5,0.5,0.5}; // center of sphere
    double   r = 0.25;          // radius of sphere
    for (int i=0; i<mg.Length(); ++i)
        F[i] = pow(mg.X(i)-c[0],2.0) + pow(mg.Y(i)-c[1],2.0) + pow(mg.Z(i)-c[2],2.0) - r*r;


    StructGrid::VectorTuple * V = new StructGrid::VectorTuple [mg.Length()];
    for (int i=0; i<mg.Length(); ++i)
    {
        V[i].vx = mg.X(i);
        V[i].vy = mg.Y(i);
        V[i].vz = mg.Z(i);
    }

    StructGrid sg(mg.X(),mg.nX(), mg.Y(),mg.nY(), mg.Z(),mg.nZ(), F, V);
    cout << sg << endl;
    sg.WriteFile("tmp.structgrid.vtk");

    HedgeHog hh(sg.GetGrid());
    cout << hh << endl;

    SGridOutline sgo(sg.GetGrid());
    cout << sgo << endl;

    vtkLookupTable * lt = vtkLookupTable::New();
    lt->SetNumberOfColors(1);
    lt->Build();
    lt->SetTableValue(0,CLR["blue"].C);
    SGridIsoSurf sgiso(sg.GetGrid(),0.0, lt);
    
    //ContrastTable ct;
    //SGridIsoSurf sgiso(sg.GetGrid(),4, -r*r, 0.18, ct.GetTable());
    sgiso.GetActor()->GetProperty()->SetOpacity(0.3);
    cout << sgiso << endl;

    Axes ax(1.2, true);
    cout << ax << endl;

    VTKWin win;
    win.AddActor(hh.GetActor());
    win.AddActor(sgo.GetActor());
    win.AddActor(sgiso.GetActor());
    win.AddActor(ax.GetActor());
    win.AddActor(ax.GetXLabelActor());
    win.AddActor(ax.GetYLabelActor());
    win.AddActor(ax.GetZLabelActor());
    win.Show();

    delete [] F;
    delete [] V;

    lt -> Delete();

    return 0;
}

---