IBSimu 1.0.4
|
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