IBSimu 1.0.4
particlegraph.hpp
Go to the documentation of this file.
00001 
00005 /* Copyright (c) 2005-2010 Taneli Kalvas. All rights reserved.
00006  *
00007  * You can redistribute this software and/or modify it under the terms
00008  * of the GNU General Public License as published by the Free Software
00009  * Foundation; either version 2 of the License, or (at your option)
00010  * any later version.
00011  * 
00012  * This library is distributed in the hope that it will be useful, but
00013  * WITHOUT ANY WARRANTY; without even the implied warranty of
00014  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
00015  * General Public License for more details.
00016  * 
00017  * You should have received a copy of the GNU General Public License
00018  * along with this library (file "COPYING" included in the package);
00019  * if not, write to the Free Software Foundation, Inc., 51 Franklin
00020  * Street, Fifth Floor, Boston, MA 02110-1301 USA
00021  * 
00022  * If you have questions about your rights to use or distribute this
00023  * software, please contact Berkeley Lab's Technology Transfer
00024  * Department at TTD@lbl.gov. Other questions, comments and bug
00025  * reports should be sent directly to the author via email at
00026  * taneli.kalvas@jyu.fi.
00027  * 
00028  * NOTICE. This software was developed under partial funding from the
00029  * U.S.  Department of Energy.  As such, the U.S. Government has been
00030  * granted for itself and others acting on its behalf a paid-up,
00031  * nonexclusive, irrevocable, worldwide license in the Software to
00032  * reproduce, prepare derivative works, and perform publicly and
00033  * display publicly.  Beginning five (5) years after the date
00034  * permission to assert copyright is obtained from the U.S. Department
00035  * of Energy, and subject to any subsequent five (5) year renewals,
00036  * the U.S. Government is granted for itself and others acting on its
00037  * behalf a paid-up, nonexclusive, irrevocable, worldwide license in
00038  * the Software to reproduce, prepare derivative works, distribute
00039  * copies to the public, perform publicly and display publicly, and to
00040  * permit others to do so.
00041  */
00042 
00043 #ifndef PARTICLEGRAPH_HPP
00044 #define PARTICLEGRAPH_HPP 1
00045 
00046 
00047 #include <cairo.h>
00048 #include <vector>
00049 #include "color.hpp"
00050 #include "graph3d.hpp"
00051 #include "coordmapper.hpp"
00052 #include "geometry.hpp"
00053 #include "particledatabase.hpp"
00054 #include "lineclip.hpp"
00055 
00056 
00068 class ParticleGraph : public Graph3D {
00069     
00070     const Geometry         &_g;          
00071     const ParticleDataBase &_pdb;        
00072     int                     _particlediv;
00074     std::vector<Color>      _color;      
00076     double                  _ox[5];      
00077     size_t                  _coordsize;  
00078     double                 *_coord;      
00079     bool                    _qm_discr;   
00081     void get_point( const Coordmapper *cm, double *coord, double s, 
00082                     double Ax, double Bx, double Cx, double Dx, 
00083                     double Ay, double By, double Cy, double Dy ) const;
00084 
00085     void draw_linear( const Coordmapper *cm, LineClip &lc, 
00086                       double x[5], bool first ) const;
00087 
00088     void draw_curve( const Coordmapper *cm, LineClip &lc, 
00089                      double x[5], bool first );
00090 
00091 public:
00092 
00095     ParticleGraph( const Geometry &g, const ParticleDataBase &pdb,
00096                   int particlediv = 10, bool qm_discr = true );
00097 
00100     virtual ~ParticleGraph();
00101 
00108     void set_particle_div( size_t particle_div );
00109 
00115     void set_qm_discretation( bool qm_discr );
00116 
00123     virtual void plot( cairo_t *cairo, const Coordmapper *cm, const double range[4] );
00124 
00130     virtual void get_bbox( double bbox[4] );
00131 
00134     void add_color( const Color &color );
00135 
00138     void clear_colors( void );
00139 
00140 };
00141 
00142 
00143 
00144 #endif
00145 
00146 
00147 
00148 
00149 
00150 
00151 
00152 
00153 
00154 
00155 
00156 
00157 
00158 
00159 
00160 
00161 
00162