File: NodalDataManager.h

package info (click to toggle)
madlib 1.3.0-5
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 8,196 kB
  • sloc: cpp: 39,851; sh: 10,041; makefile: 473
file content (102 lines) | stat: -rw-r--r-- 3,045 bytes parent folder | download | duplicates (6)
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
100
101
102
// -*- C++ -*-
// -------------------------------------------------------------------
// MAdLib - Copyright (C) 2008-2009 Universite catholique de Louvain
//
// See the Copyright.txt and License.txt files for license information. 
// You should have received a copy of these files along with MAdLib. 
// If not, see <http://www.madlib.be/license/>
//
// Please report all bugs and problems to <contrib@madlib.be>
//
// Authors: Gaetan Compere, Jean-Francois Remacle
// -------------------------------------------------------------------

#ifndef _H_NODALDATAMANAGER
#define _H_NODALDATAMANAGER

#include "MAdSingleton.h"
#include "MeshDataBaseInterface.h"

#include <vector>
#include <string>
#include <set>
#include <iostream>

namespace MAd {

  // -------------------------------------------------------------------

  class NodalDataManager {

  public:
  
    NodalDataManager():mesh(NULL),
                       knownDataNames(NULL), knownDataVecNames(NULL){};
    ~NodalDataManager() {};
  
    void initialize(pMesh m);
    void setMesh(pMesh m);
    void finalize();

    void removeAllData();

    void diagnostics (std::ostream& out) const;

    // will attach the datas to the nodes of the mesh
    // order in vector is the node iterator order
    void registerData  (std::string name, const std::vector<double>);
    void registerVData (std::string name, const std::vector<std::vector<double> >);

    // will fill the variable with the datas attached to the nodes
    // this function does not allocate memory
    // order in vector is the node iterator order
    void getMeshData  (std::string name, std::vector<double> *)          const;
    void getMeshVData (std::string name, std::vector<std::vector<double> > *) const;

    // get the data at a node
    void getData  (std::string name, pVertex pv, double* res)         const;
    void getVData (std::string name, pVertex pv, std::vector<double>& res) const;

    // will delete the data attached to the nodes
    void removeData  (std::string name);
    void removeVData (std::string name);

    // write the data in a postprocessing file
    void writeData  (std::string name, const char *fn);
    void writeVData (std::string name, const char *fn);

    // functions to keep track of the initial coordinates
    void storeCoordinates();
    void getStoredCoordinates(pVertex pv, std::vector<double>& res);
    void removeCoordinates();
    bool isCoordinates();
  
  private:

    pMesh mesh;

    // prefix for identification of the datas from this manager
    // ensures there is no clash with other datas
    std::string prefix;
    std::string prefixVec;

    // identifier for the coordinates
    std::string coordNameBase;

    // stored with the prefix
    std::set<std::string>* knownDataNames;
    std::set<std::string>* knownDataVecNames;

  private:

    bool nameExists(std::string name, int vec) const;

  };

  // -------------------------------------------------------------------

  typedef MAdSingleton<NodalDataManager> NodalDataManagerSgl;

}

#endif