1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
|
// Copyright (c) 2010-2026, Lawrence Livermore National Security, LLC. Produced
// at the Lawrence Livermore National Laboratory. All Rights reserved. See files
// LICENSE and NOTICE for details. LLNL-CODE-443271.
//
// This file is part of the GLVis visualization tool and library. For more
// information and source code availability see https://glvis.org.
//
// GLVis is free software; you can redistribute it and/or modify it under the
// terms of the BSD-3 license. We welcome feedback and contributions, see file
// CONTRIBUTING.md for details.
#ifndef GLVIS_VSVECTOR_HPP
#define GLVIS_VSVECTOR_HPP
#include <mfem.hpp>
#include "gl/types.hpp"
#include "vssolution.hpp"
class VisualizationSceneVector : public VisualizationSceneSolution
{
protected:
mfem::Vector *solx, *soly;
int drawdisp, drawvector;
gl3::GlDrawable vector_buf;
gl3::GlDrawable displine_buf;
mfem::GridFunction *VecGridF{};
void Init();
void NewMeshAndSolution(mfem::Mesh *new_m, mfem::Mesh *new_mc,
mfem::Vector *new_sol_x, mfem::Vector *new_sol_y,
mfem::GridFunction *new_u = nullptr);
void GetRefinedValues(const int i, const mfem::IntegrationRule &ir,
mfem::Vector &vals, mfem::DenseMatrix &tr,
const bool do_shrink = true) override;
int GetRefinedValuesAndNormals(int i, const mfem::IntegrationRule &ir,
mfem::Vector &vals,mfem::DenseMatrix &tr,
mfem::DenseMatrix &normals) override;
double (*Vec2Scalar)(double, double);
void DrawVector(double, double, double, double, double);
double maxlen;
mfem::Vector vc0;
mfem::IsoparametricTransformation T0;
int GetFunctionAutoRefineFactor() override;
public:
VisualizationSceneVector(Window &win_);
void NewMeshAndSolution(const DataState &s) override;
void FindNewBox(bool prepare) override;
void FindNewValueRange(bool prepare) override;
virtual ~VisualizationSceneVector();
std::string GetHelpString() const override;
void NPressed();
void PrepareDisplacedMesh();
void PrepareLines() override
{ VisualizationSceneSolution::PrepareLines(); PrepareDisplacedMesh(); }
void ToggleDrawElems() override;
virtual void PrepareVectorField();
void ToggleVectorField();
void ToggleDisplacements()
{
drawdisp = (drawdisp+1)%4;
if (drawdisp != 1)
{
PrepareDisplacedMesh();
}
}
gl3::SceneInfo GetSceneObjs() override;
void glTF_Export() override;
void EventUpdateColors() override { Prepare(); PrepareVectorField(); }
// refinement factor for the vectors
int RefineFactor;
double ArrowScale;
void CycleVec2Scalar(int print = 0);
};
#endif
|
