eqpotgraph.hpp

Go to the documentation of this file.

/* Copyright (c) 2005-2011 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 EQPOTGRAPH_HPP
#define EQPOTGRAPH_HPP 1


#include <vector>
#include "geometry.hpp"
#include "scalarfield.hpp"
#include "graph3d.hpp"
#include "color.hpp"


class EqPotGraph : public Graph3D {

    struct Line {
        double x[4];              /* Coordinates of line segment (x1,y1,x2,y2) */
        
        Line( double _x1, double _y1, double _x2, double _y2 ) { 
            x[0] = _x1; 
            x[1] = _y1;
            x[2] = _x2; 
            x[3] = _y2;
        }

        double &operator[]( int i ) { return( x[i] ); }
        const double &operator[]( int i ) const { return( x[i] ); }
    };

    struct EqPotLines {
        double            pot;    /* Potential value for equipotential line. */
        std::vector<Line> x;      /* Line segment coordinates. */
        
        EqPotLines( double pot ) : pot(pot) {}
    };

    Color                               _color;
    const ScalarField                   _epot;
    const Geometry                     &_geom;
    bool                                _data_built;

    std::vector<double>                 _eqlines_manual;
    size_t                              _eqlines_auto;
    std::vector<EqPotLines *>           _lines;

    view_e                              _oview;
    double                              _olevel;

    bool                                _cache;


    bool eqline_exists( double pot1, signed char sol1, 
                        double pot2, signed char sol2, 
                        double pot ) const;
    void build_data( void );

public:

    EqPotGraph( const ScalarField &epot, const Geometry &geom );

    virtual ~EqPotGraph();

    void disable_cache( void );

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

    void set_eqlines_auto( size_t N );

    virtual void plot( cairo_t *cairo, const Coordmapper *cm, const double range[4] );

    virtual void plot_sample( cairo_t *cairo, double x, double y, double width, double height );

    virtual void get_bbox( double bbox[4] );
};


#endif