geomplot.hpp

Go to the documentation of this file.

/* Copyright (c) 2005-2012 Taneli Kalvas. All rights reserved.
 *
 * You can redistribute this software 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.
 * 
 * This library 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 this library (file "COPYING" included in the package);
 * if not, write to the Free Software Foundation, Inc., 51 Franklin
 * Street, Fifth Floor, Boston, MA 02110-1301 USA
 * 
 * If you have questions about your rights to use or distribute this
 * software, please contact Berkeley Lab's Technology Transfer
 * Department at TTD@lbl.gov. Other questions, comments and bug
 * reports should be sent directly to the author via email at
 * taneli.kalvas@jyu.fi.
 * 
 * NOTICE. This software was developed under partial funding from the
 * U.S.  Department of Energy.  As such, the U.S. Government has been
 * granted for itself and others acting on its behalf a paid-up,
 * nonexclusive, irrevocable, worldwide license in the Software to
 * reproduce, prepare derivative works, and perform publicly and
 * display publicly.  Beginning five (5) years after the date
 * permission to assert copyright is obtained from the U.S. Department
 * of Energy, and subject to any subsequent five (5) year renewals,
 * the U.S. Government is granted for itself and others acting on its
 * behalf a paid-up, nonexclusive, irrevocable, worldwide license in
 * the Software to reproduce, prepare derivative works, distribute
 * copies to the public, perform publicly and display publicly, and to
 * permit others to do so.
 */

#ifndef GEOMPLOT_HPP
#define GEOMPLOT_HPP 1


#include "types.hpp"
#include "frame.hpp"
#include "geometry.hpp"
#include "meshscalarfield.hpp"
#include "vectorfield.hpp"
#include "epot_field.hpp"
#include "particledatabase.hpp"
#include "solidgraph.hpp"
#include "eqpotgraph.hpp"
#include "particlegraph.hpp"
#include "meshgraph.hpp"
#include "fieldgraph.hpp"




class GeomPlot {

    Frame                   *_frame;

    const Geometry          &_geom;
    const EpotField         *_epot;
    const MeshScalarField   *_scharge;
    const MeshScalarField   *_tdens;
    const VectorField       *_bfield;
    const VectorField       *_efield;
    const ParticleDataBase  *_pdb;

    FieldGraph              *_fieldgraph;

    SolidGraph              *_solidgraph;
    EqPotGraph              *_eqpotgraph;
    ParticleGraph           *_particlegraph;
    MeshGraph               *_meshgraph;

    view_e                  _view;
    int                     _level;
    int                     _vb[3];

    uint32_t                _eqlines_auto;
    std::vector<double>     _eqlines_manual;
    uint32_t                _particle_div;
    uint32_t                _particle_offset;
    bool                    _scharge_field;
    bool                    _qm_discretation;
    bool                    _mesh;

    bool                    _cache;

    void reset_graphs( void );

public:

    GeomPlot( Frame &frame, const Geometry &geom );

    ~GeomPlot();

    void build_plot( void );

    void disable_cache( void );

    void set_epot( const EpotField *epot );

    void set_eqlines_manual( const std::vector<double> &pot );

    std::vector<double> get_eqlines_manual( void ) const {
        return( _eqlines_manual );
    }

    void set_eqlines_auto( uint32_t N );

    uint32_t get_eqlines_auto( void ) const {
        return( _eqlines_auto );
    }

    void set_bfield( const VectorField *bfield );

    const VectorField *get_bfield( void ) const {
        return( _bfield );
    }

    void set_efield( const VectorField *efield );

    const VectorField *get_efield( void ) const {
        return( _efield );
    }

    void set_trajdens( const MeshScalarField *tdens );

    const MeshScalarField *get_trajdens( void ) const {
        return( _tdens );
    }

    void set_scharge( const MeshScalarField *scharge );

    const MeshScalarField *get_scharge( void ) const {
        return( _scharge );
    }

    void set_fieldgraph_plot( field_type_e fieldplot );

    const FieldGraph *fieldgraph( void ) const {
        return( _fieldgraph );
    }

    FieldGraph *fieldgraph( void ) {
        return( _fieldgraph );
    }

    void enable_colormap_legend( bool enable ) {
        _frame->enable_colormap_legend( enable );
    }

    void set_particle_database( const ParticleDataBase *pdb ) {
        set_particledatabase( pdb );
    }

    void set_particledatabase( const ParticleDataBase *pdb );

    void set_particle_div( uint32_t particle_div, uint32_t particle_offset = 0 );

    uint32_t get_particle_div( void ) const {
        return( _particle_div );
    }

    uint32_t get_particle_offset( void ) const {
        return( _particle_offset );
    }

    void set_qm_discretation( bool enable );

    bool get_qm_discretation( void ) const {
        return( _qm_discretation );
    }

    void set_mesh( bool enable );

    bool get_mesh( void ) const {
        return( _mesh );
    }

    void set_view( view_e view, int level = -1 );

    void set_view_si( view_e view, double level );

    view_e get_view( void ) const {
        return( _view );
    }

    int get_level( void ) const {
        return( _level );
    }

    double get_level_si( void ) const {
        return( _geom.origo(_vb[2])+_level*_geom.h() );
    }

    int vb( int i ) const {
        return( _vb[i] );
    }

    void get_vb( int vb[3] ) const {
        vb[0] = _vb[0];
        vb[1] = _vb[1];
        vb[2] = _vb[2];
    }
    
};


#endif