netgen/libsrc/visualization/vssolution.hpp
2019-07-09 10:39:16 +02:00

458 lines
11 KiB
C++

#ifndef FILE_VSSOLUTION
#define FILE_VSSOLUTION
typedef void * ClientData;
struct Tcl_Interp;
namespace netgen
{
DLL_HEADER extern void ImportSolution (const char * filename);
class FieldLineCalc;
extern int Ng_Vis_Set (ClientData clientData,
Tcl_Interp * interp,
int argc, const char *argv[]);
class DLL_HEADER VisualSceneSolution : public VisualScene
{
friend class FieldLineCalc;
class ClipPlaneTrig
{
public:
struct ps
{
int pnr, locpnr;
};
ps points[3];
ElementIndex elnr;
};
class ClipPlanePoint
{
public:
ElementIndex elnr;
Point<3> lami;
Point<3> p;
};
#ifndef WIN32
// use OpenGL vertex buffers from OpenGL 2.x
// not supported by some drivers on windows
// try on your own responsibility
#define USE_BUFFERS
#endif
#ifdef USE_BUFFERS
bool has_surfel_vbo = false;
GLuint surfel_vbo[4]; //
size_t surfel_vbo_size;
#endif
int surfellist;
int linelist;
int element1dlist;
int clipplanelist_scal;
int clipplanelist_vec;
int isolinelist;
int clipplane_isolinelist;
int surface_vector_list;
// int cone_list;
int isosurface_list;
int pointcurvelist;
bool draw_fieldlines;
bool drawpointcurves;
bool draw_isosurface;
int num_fieldlines;
bool fieldlines_randomstart;
int fieldlineslist;
int num_fieldlineslists;
int fieldlines_startarea;
NgArray<double> fieldlines_startarea_parameter;
int fieldlines_startface;
string fieldlines_filename;
double fieldlines_reltolerance;
int fieldlines_rktype;
double fieldlines_rellength;
double fieldlines_relthickness;
int fieldlines_vecfunction;
bool fieldlines_fixedphase;
float fieldlines_phase;
int fieldlines_maxpoints;
int surfeltimestamp, clipplanetimestamp, solutiontimestamp;
int surfellinetimestamp;
int fieldlinestimestamp, surface_vector_timestamp;
int pointcurve_timestamp;
int isosurface_timestamp;
int subdivision_timestamp;
int timetimestamp;
double minval, maxval;
NgLock *lock;
#ifdef PARALLELGL
NgArray<int> par_linelists;
NgArray<int> par_surfellists;
#endif
NgArray<UserVisualizationObject*> user_vis;
public:
enum EvalFunc {
FUNC_ABS = 1,
FUNC_ABS_TENSOR = 2,
FUNC_MISES = 3,
FUNC_MAIN = 4
};
int evalfunc;
enum SolType
{
SOL_NODAL = 1,
SOL_ELEMENT = 2,
SOL_SURFACE_ELEMENT = 3,
SOL_NONCONTINUOUS = 4,
SOL_SURFACE_NONCONTINUOUS = 5,
SOL_VIRTUALFUNCTION = 6,
SOL_MARKED_ELEMENTS = 10,
SOL_ELEMENT_ORDER = 11,
};
class SolData
{
public:
SolData ();
~SolData ();
string name;
double * data;
int components;
int dist;
int order;
bool iscomplex;
bool draw_volume;
bool draw_surface;
SolType soltype;
SolutionData * solclass;
// internal variables:
int size;
};
NgArray<SolData*> soldata;
int usetexture; // 0..no, 1..1D texture (standard), 2..2D-texture (complex)
int clipsolution; // 0..no, 1..scal, 2..vec
int scalfunction, scalcomp, vecfunction;
int gridsize;
double xoffset, yoffset;
int autoscale, logscale;
double mminval, mmaxval;
int numisolines;
int subdivisions;
bool showclipsolution;
bool showsurfacesolution;
bool lineartexture;
int numtexturecols;
int multidimcomponent;
// bool fieldlineplot;
double time;
int deform;
double scaledeform;
bool imag_part;
private:
void BuildFieldLinesFromFile(NgArray<Point3d> & startpoints);
void BuildFieldLinesFromFace(NgArray<Point3d> & startpoints);
void BuildFieldLinesFromBox(NgArray<Point3d> & startpoints);
void BuildFieldLinesFromLine(NgArray<Point3d> & startpoints);
// weak_ptr<Mesh> wp_mesh;
public:
VisualSceneSolution ();
virtual ~VisualSceneSolution ();
virtual void BuildScene (int zoomall = 0);
virtual void DrawScene ();
virtual void MouseDblClick (int px, int py);
// void SetMesh (shared_ptr<Mesh> amesh);
// shared_ptr<Mesh> GetMesh () { return shared_ptr<Mesh>(wp_mesh); }
shared_ptr<Mesh> GetMesh () const { return shared_ptr<Mesh>(global_mesh); }
void BuildFieldLinesPlot ();
void AddSolutionData (SolData * soldata);
void ClearSolutionData ();
void UpdateSolutionTimeStamp ();
SolData * GetSolData (int i);
int GetNSolData () { return soldata.Size(); }
void SaveSolutionData (const char * filename);
/*
static void RealVec3d (const double * values, Vec3d & v,
bool iscomplex, bool imag);
*/
static Vec<3> RealVec3d (const double * values,
bool iscomplex, bool imag);
static void RealVec3d (const double * values, Vec3d & v,
bool iscomplex, double phaser, double phasei);
void SetSubdivision (int sd)
{
subdivisions = sd;
subdivision_timestamp = solutiontimestamp = NextTimeStamp();
}
void GetMinMax (int funcnr, int comp, double & minv, double & maxv) const;
void AddUserVisualizationObject (UserVisualizationObject * vis)
{
user_vis.Append (vis);
}
private:
void GetClippingPlaneTrigs (NgArray<ClipPlaneTrig> & trigs, NgArray<ClipPlanePoint> & pts);
void GetClippingPlaneGrid (NgArray<ClipPlanePoint> & pts);
void DrawCone (const Point<3> & p1, const Point<3> & p2, double r);
void DrawCylinder (const Point<3> & p1, const Point<3> & p2, double r);
// Get Function Value, local coordinates lam1, lam2, lam3,
bool GetValue (const SolData * data, ElementIndex elnr,
double lam1, double lam2, double lam3,
int comp, double & val) const;
bool GetValue (const SolData * data, ElementIndex elnr,
const double xref[], const double x[], const double dxdxref[],
int comp, double & val) const;
bool GetValueComplex (const SolData * data, ElementIndex elnr,
double lam1, double lam2, double lam3,
int comp, complex<double> & val) const;
bool GetValues (const SolData * data, ElementIndex elnr,
double lam1, double lam2, double lam3,
double * values) const;
bool GetValues (const SolData * data, ElementIndex elnr,
const double xref[], const double x[], const double dxdxref[],
double * values) const;
bool GetMultiValues (const SolData * data, ElementIndex elnr, int facetnr, int npt,
const double * xref, int sxref,
const double * x, int sx,
const double * dxdxref, int sdxdxref,
double * val, int sval) const;
bool GetSurfValue (const SolData * data, SurfaceElementIndex elnr, int facetnr,
double lam1, double lam2,
int comp, double & val) const;
bool GetSurfValue (const SolData * data, SurfaceElementIndex elnr, int facetnr,
const double xref[], const double x[], const double dxdxref[],
int comp, double & val) const;
bool GetSurfValueComplex (const SolData * data, SurfaceElementIndex elnr, int facetnr,
double lam1, double lam2,
int comp, complex<double> & val) const;
bool GetSurfValues (const SolData * data, SurfaceElementIndex elnr, int facetnr,
double lam1, double lam2,
double * values) const;
bool GetSurfValues (const SolData * data, SurfaceElementIndex elnr, int facetnr,
const double xref[], const double x[], const double dxdxref[],
double * values) const;
bool GetMultiSurfValues (const SolData * data, SurfaceElementIndex elnr, int facetnr,
int npt,
const double * xref, int sxref,
const double * x, int sx,
const double * dxdxref, int sdxdxref,
double * val, int sval) const;
double ExtractValue (const SolData * data, int comp, double * values) const;
complex<double> ExtractValueComplex (const SolData * data, int comp, double * values) const;
Vec<3> GetDeformation (ElementIndex elnr, const Point<3> & p) const;
Vec<3> GetSurfDeformation (SurfaceElementIndex selnr, int facetnr, double lam1, double lam2) const;
void GetPointDeformation (int pnum, Point<3> & p, SurfaceElementIndex elnr = -1) const;
public:
/// draw elements (build lists)
void DrawSurfaceElements ();
void DrawSurfaceElementLines ();
void Draw1DElements();
void DrawSurfaceVectors ();
void DrawTrigSurfaceVectors(const NgArray< Point<3> > & lp, const Point<3> & pmin, const Point<3> & pmax,
const int sei, const SolData * vsol);
void DrawIsoSurface(const SolData * sol, const SolData * grad, int comp);
void DrawIsoLines (const Point<3> & p1,
const Point<3> & p2,
const Point<3> & p3,
double val1, double val2, double val3);
// draw isolines between lines (p1,p2) and (p3,p4)
void DrawIsoLines2 (const Point<3> & p1,
const Point<3> & p2,
const Point<3> & p3,
const Point<3> & p4,
double val1, double val2, double val3, double val4);
void DrawClipPlaneTrigs (); // const SolData * sol, int comp);
void SetOpenGlColor(double val);
// 0 .. non, 1 .. scalar, 2 .. complex
void SetTextureMode (int texturemode) const;
friend int Ng_Vis_Set (ClientData clientData,
Tcl_Interp * interp,
int argc, const char *argv[]);
#ifdef PARALLELGL
void Broadcast ();
#endif
};
class RKStepper
{
private:
NgArray<double> c,b;
TABLE<double> *a;
int steps;
int order;
double tolerance;
NgArray<Vec3d> K;
int stepcount;
double h;
double startt;
double startt_bak;
Point3d startval;
Point3d startval_bak;
bool adaptive;
int adrun;
Point3d valh;
int notrestarted;
public:
~RKStepper();
RKStepper(int type = 0);
void SetTolerance(const double tol){tolerance = tol;}
void StartNextValCalc(const Point3d & astartval, const double astartt, const double ah, const bool aadaptive = false);
bool GetNextData(Point3d & val, double & t, double & ah);
bool FeedNextF(const Vec3d & f);
};
class FieldLineCalc
{
private:
const Mesh & mesh;
VisualSceneSolution & vss;
const VisualSceneSolution::SolData * vsol;
RKStepper stepper;
double maxlength;
int maxpoints;
int direction;
Point3d pmin, pmax;
double rad;
double phaser, phasei;
double critical_value;
bool randomized;
double thickness;
public:
FieldLineCalc(const Mesh & amesh, VisualSceneSolution & avss, const VisualSceneSolution::SolData * solution,
const double rel_length, const int amaxpoints = -1,
const double rel_thickness = -1, const double rel_tolerance = -1, const int rk_type = 0, const int adirection = 0);
void SetPhase(const double real, const double imag) { phaser = real; phasei = imag; }
void SetCriticalValue(const double val) { critical_value = val; }
void Randomized(void) { randomized = true; }
void NotRandomized(void) { randomized = false; }
void Calc(const Point3d & startpoint, NgArray<Point3d> & points, NgArray<double> & vals, NgArray<bool> & drawelems, NgArray<int> & dirstart);
void GenerateFieldLines(NgArray<Point3d> & potential_startpoints, const int numlines, const int gllist,
const double minval, const double maxval, const int logscale, double phaser, double phasei);
};
// DLL_HEADER extern VisualSceneSolution vssolution;
DLL_HEADER extern VisualSceneSolution & GetVSSolution();
}
#endif