bedding-planes.cpp

/***********************************************************************************
 * YieldSurf - Simple generator of yield surface data                              *
 * Copyright (C) 2005 Dorival de Moraes Pedroso <dorival.pedroso at gmail.com>     *
 *                                                                                 *
 * This file is part of YieldSurf.                                                 *
 *                                                                                 *
 * YieldSurf 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.                                                                        *
 *                                                                                 *
 * YieldSurf 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    *
 * YieldSurf; if not, write to the Free Software Foundation, Inc., 51 Franklin     *
 * Street, Fifth Floor, Boston, MA 02110-1301, USA                                 *
 ***********************************************************************************/

// STL
#include <algorithm>
#include <iostream>

// VTK
#include "vtkLightKit.h"

// MechSys
#include "numerical/meshgrid.h"
#include "vtkwrap/vtkwin.h"
#include "vtkwrap/axes.h"
#include "vtkwrap/arrow.h"
#include "vtkwrap/cube.h"
#include "vtkwrap/plane.h"
#include "util/string.h"
#include "util/util.h"
#include "util/exception.h"

using Util::Sq3;

int main(int argc, char **argv) try
{
    // Input
    int cam_key = 1;
    if (argc>=2) cam_key = atoi(argv[1]);

    // Window
    //VTKWin win(1280,1024);
    VTKWin win;
    switch (cam_key)
    {
        case 1: { win.GetCamera()->SetPosition(1,-0.25,0.25);    break; }
        case 2: { win.GetCamera()->SetPosition(1,-1,0);          break; }
        case 3: { win.SetViewPIplane();                          break; }
    }

    // Constants
    double L = 10;
    double Lmax = 1.3*L;

    // Axes
    Axes ax(Lmax,1.5*Lmax,Lmax, "x","y","z", "peacock","black", 0.010,0.002, false);
    ax.AddActorsTo(win);

    // Arrows
    double l = 0.3*L;
    double xa = 0.8*L;
    double ya = 0.8*L;
    Arrow::Geometry geom(/*BodyRadius*/0.010*Lmax, /*TipRadius*/0.020*Lmax,  /*TipMultiplier*/4.0, /*Resolution*/60);
    Arrow n (0 , 0 , -l , "blue" , L/2 , L/2 , L+l , &geom);
    Arrow a (0 , 0 , l  , "red"  , xa , ya , L   , &geom);
    win.AddActor(n.GetActor());
    win.AddActor(a.GetActor());

    // Cube
    Cube cu1(0.5*L, 0.5*L, 0.5*L, L,L,L, "wheat" ,0.2);
    cu1.WireWidth(2).WireColor("black");
    win.AddActor(cu1.GetActor());
    win.AddActor(cu1.GetWireActor());

    // Planes
    char const color[] = "wheat";
    double        opac = 0.7;
    double          dx = L/10.0;
    Plane  pl1(0,0,   dx,  0,L,   dx,  L,0,   dx, 0,0,1, color, opac);
    Plane  pl2(0,0, 2*dx,  0,L, 2*dx,  L,0, 2*dx, 0,0,1, color, opac);
    Plane  pl3(0,0, 3*dx,  0,L, 3*dx,  L,0, 3*dx, 0,0,1, color, opac);
    Plane  pl4(0,0, 4*dx,  0,L, 4*dx,  L,0, 4*dx, 0,0,1, color, opac);
    Plane  pl5(0,0, 5*dx,  0,L, 5*dx,  L,0, 5*dx, 0,0,1, color, opac);
    Plane  pl6(0,0, 6*dx,  0,L, 6*dx,  L,0, 6*dx, 0,0,1, color, opac);
    Plane  pl7(0,0, 7*dx,  0,L, 7*dx,  L,0, 7*dx, 0,0,1, color, opac);
    Plane  pl8(0,0, 8*dx,  0,L, 8*dx,  L,0, 8*dx, 0,0,1, color, opac);
    Plane  pl9(0,0, 9*dx,  0,L, 9*dx,  L,0, 9*dx, 0,0,1, color, opac);
    Plane pl10(0,0,10*dx,  0,L,10*dx,  L,0,10*dx, 0,0,1, color, opac);
    win.AddActor( pl1.GetActor());
    win.AddActor( pl2.GetActor());
    win.AddActor( pl3.GetActor());
    win.AddActor( pl4.GetActor());
    win.AddActor( pl5.GetActor());
    win.AddActor( pl6.GetActor());
    win.AddActor( pl7.GetActor());
    win.AddActor( pl8.GetActor());
    win.AddActor( pl9.GetActor());
    win.AddActor(pl10.GetActor());

    // Lights
    vtkLightKit * lk = vtkLightKit::New();
    lk->SetKeyLightIntensity(1.0);
    lk->AddLightsToRenderer(win.GetRen());

    // Show
    win.Show();

    // Write file
    win.WritePNG("bedding-planes.png");
    
    // Clean up
    lk->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

---