particlegraph.hpp
Go to the documentation of this file.
1 
5 /* Copyright (c) 2005-2011 Taneli Kalvas. All rights reserved.
6  *
7  * You can redistribute this software and/or modify it under the terms
8  * of the GNU General Public License as published by the Free Software
9  * Foundation; either version 2 of the License, or (at your option)
10  * any later version.
11  *
12  * This library is distributed in the hope that it will be useful, but
13  * WITHOUT ANY WARRANTY; without even the implied warranty of
14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
15  * General Public License for more details.
16  *
17  * You should have received a copy of the GNU General Public License
18  * along with this library (file "COPYING" included in the package);
19  * if not, write to the Free Software Foundation, Inc., 51 Franklin
20  * Street, Fifth Floor, Boston, MA 02110-1301 USA
21  *
22  * If you have questions about your rights to use or distribute this
23  * software, please contact Berkeley Lab's Technology Transfer
24  * Department at TTD@lbl.gov. Other questions, comments and bug
25  * reports should be sent directly to the author via email at
26  * taneli.kalvas@jyu.fi.
27  *
28  * NOTICE. This software was developed under partial funding from the
29  * U.S. Department of Energy. As such, the U.S. Government has been
30  * granted for itself and others acting on its behalf a paid-up,
31  * nonexclusive, irrevocable, worldwide license in the Software to
32  * reproduce, prepare derivative works, and perform publicly and
33  * display publicly. Beginning five (5) years after the date
34  * permission to assert copyright is obtained from the U.S. Department
35  * of Energy, and subject to any subsequent five (5) year renewals,
36  * the U.S. Government is granted for itself and others acting on its
37  * behalf a paid-up, nonexclusive, irrevocable, worldwide license in
38  * the Software to reproduce, prepare derivative works, distribute
39  * copies to the public, perform publicly and display publicly, and to
40  * permit others to do so.
41  */
42 
43 #ifndef PARTICLEGRAPH_HPP
44 #define PARTICLEGRAPH_HPP 1
45 
46 
47 #include <cairo.h>
48 #include <vector>
49 #include "color.hpp"
50 #include "graph3d.hpp"
51 #include "coordmapper.hpp"
52 #include "geometry.hpp"
53 #include "particledatabase.hpp"
54 #include "lineclip.hpp"
55 
56 
68 class ParticleGraph : public Graph3D {
69 
70  const Geometry &_g;
71  const ParticleDataBase &_pdb;
72  int _particlediv;
74  std::vector<Color> _color;
76  double _ox[5];
77  size_t _coordsize;
78  double *_coord;
79  bool _qm_discr;
81  void get_point( const Coordmapper *cm, double *coord, double s,
82  double Ax, double Bx, double Cx, double Dx,
83  double Ay, double By, double Cy, double Dy ) const;
84 
85  void draw_linear( const Coordmapper *cm, LineClip &lc,
86  double x[5], bool first ) const;
87 
88  void draw_curve( const Coordmapper *cm, LineClip &lc,
89  double x[5], bool first );
90 
91 public:
92 
95  ParticleGraph( const Geometry &g, const ParticleDataBase &pdb,
96  int particlediv = 10, bool qm_discr = true );
97 
100  virtual ~ParticleGraph();
101 
108  void set_particle_div( size_t particle_div );
109 
115  void set_qm_discretation( bool qm_discr );
116 
126  virtual void plot( cairo_t *cairo, const Coordmapper *cm, const double range[4] );
127 
132  virtual void plot_sample( cairo_t *cairo, double x, double y, double width, double height );
133 
139  virtual void get_bbox( double bbox[4] );
140 
143  void add_color( const Color &color );
144 
147  void clear_colors( void );
148 
149 };
150 
151 
152 
153 #endif