ASCOT5
Loading...
Searching...
No Matches
hdf5_mhd.c
Go to the documentation of this file.
1
6#include <stdlib.h>
7#include <hdf5.h>
8#include <hdf5_hl.h>
9#include "../ascot5.h"
10#include "../mhd.h"
11#include "../mhd/mhd_stat.h"
12#include "../mhd/mhd_nonstat.h"
13#include "hdf5_helpers.h"
14#include "hdf5_mhd.h"
15
16#define MHDPATH
17
18int hdf5_mhd_read_stat(hid_t f, mhd_stat_data* data, char* qid);
19int hdf5_mhd_read_nonstat(hid_t f, mhd_nonstat_data* data, char* qid);
20
31int hdf5_mhd_init(hid_t f, mhd_data* data, char* qid) {
32
33 char path[256];
34 int err = 1;
35
36 /* Read data the QID corresponds to */
37 hdf5_gen_path("/mhd/MHD_STAT_XXXXXXXXXX", qid, path);
38 if( !hdf5_find_group(f, path) ) {
39 data->type = mhd_type_stat;
40 err = hdf5_mhd_read_stat(f, &data->stat, qid);
41 }
42 hdf5_gen_path("/mhd/MHD_NONSTAT_XXXXXXXXXX", qid, path);
43 if( !hdf5_find_group(f, path) ) {
44 data->type = mhd_type_nonstat;
45 err = hdf5_mhd_read_nonstat(f, &data->nonstat, qid);
46 }
47 return err;
48}
49
59int hdf5_mhd_read_stat(hid_t f, mhd_stat_data* data, char* qid) {
60 #undef MHDPATH
61 #define MHDPATH "/mhd/MHD_STAT_XXXXXXXXXX/"
62
63 int nmode, nrho;
64 real rhomin, rhomax;
65 if( hdf5_read_int( MHDPATH "nmode", &nmode,
66 f, qid, __FILE__, __LINE__) ) {return 1;}
67 if( hdf5_read_int( MHDPATH "nrho", &nrho,
68 f, qid, __FILE__, __LINE__) ) {return 1;}
69 if( hdf5_read_double(MHDPATH "rhomin", &rhomin,
70 f, qid, __FILE__, __LINE__) ) {return 1;}
71 if( hdf5_read_double(MHDPATH "rhomax", &rhomax,
72 f, qid, __FILE__, __LINE__) ) {return 1;}
73
74 int* moden = (int*) malloc( nmode * sizeof(int) );
75 int* modem = (int*) malloc( nmode * sizeof(int) );
76 real* omega_nm = (real*) malloc( nmode * sizeof(real) );
77 real* phase_nm = (real*) malloc( nmode * sizeof(real) );
78 real* amplitude_nm = (real*)malloc( nmode * sizeof(real) );
79 if( hdf5_read_int( MHDPATH "nmodes", moden,
80 f, qid, __FILE__, __LINE__) ) {return 1;}
81 if( hdf5_read_int( MHDPATH "mmodes", modem,
82 f, qid, __FILE__, __LINE__) ) {return 1;}
83 if( hdf5_read_double(MHDPATH "omega", omega_nm,
84 f, qid, __FILE__, __LINE__) ) {return 1;}
85 if( hdf5_read_double(MHDPATH "phase", phase_nm,
86 f, qid, __FILE__, __LINE__) ) {return 1;}
87 if( hdf5_read_double(MHDPATH "amplitude", amplitude_nm,
88 f, qid, __FILE__, __LINE__) ) {return 1;}
89
90 /* Allocate offload array */
91 real* phi = (real*) malloc( nrho * nmode * sizeof(real) );
92 real* alpha = (real*) malloc( nrho * nmode * sizeof(real) );
93 if( hdf5_read_double(MHDPATH "alpha", alpha,
94 f, qid, __FILE__, __LINE__) ) {return 1;}
95 if( hdf5_read_double(MHDPATH "phi", phi,
96 f, qid, __FILE__, __LINE__) ) {return 1;}
97
98 int err = mhd_stat_init(data, nmode, nrho, rhomin, rhomax, moden, modem,
99 amplitude_nm, omega_nm, phase_nm, alpha, phi);
100 free(moden);
101 free(modem);
102 free(omega_nm);
103 free(phase_nm);
104 free(amplitude_nm);
105 free(phi);
106 free(alpha);
107 return err;
108}
109
119int hdf5_mhd_read_nonstat(hid_t f, mhd_nonstat_data* data, char* qid) {
120 #undef MHDPATH
121 #define MHDPATH "/mhd/MHD_NONSTAT_XXXXXXXXXX/"
122
123 int nmode, nrho, ntime;
124 real rhomin, rhomax, tmin, tmax;
125 if( hdf5_read_int( MHDPATH "nmode", &nmode,
126 f, qid, __FILE__, __LINE__) ) {return 1;}
127 if( hdf5_read_int( MHDPATH "nrho", &nrho,
128 f, qid, __FILE__, __LINE__) ) {return 1;}
129 if( hdf5_read_double(MHDPATH "rhomin", &rhomin,
130 f, qid, __FILE__, __LINE__) ) {return 1;}
131 if( hdf5_read_double(MHDPATH "rhomax", &rhomax,
132 f, qid, __FILE__, __LINE__) ) {return 1;}
133 if( hdf5_read_int( MHDPATH "ntime", &ntime,
134 f, qid, __FILE__, __LINE__) ) {return 1;}
135 if( hdf5_read_double(MHDPATH "tmin", &tmin,
136 f, qid, __FILE__, __LINE__) ) {return 1;}
137 if( hdf5_read_double(MHDPATH "tmax", &tmax,
138 f, qid, __FILE__, __LINE__) ) {return 1;}
139
140 int* moden = (int*) malloc( nmode * sizeof(int) );
141 int* modem = (int*) malloc( nmode * sizeof(int) );
142 real* omega_nm = (real*) malloc( nmode * sizeof(real) );
143 real* phase_nm = (real*) malloc( nmode * sizeof(real) );
144 real* amplitude_nm = (real*)malloc( nmode * sizeof(real) );
145 if( hdf5_read_int( MHDPATH "nmodes", moden,
146 f, qid, __FILE__, __LINE__) ) {return 1;}
147 if( hdf5_read_int( MHDPATH "mmodes", modem,
148 f, qid, __FILE__, __LINE__) ) {return 1;}
149 if( hdf5_read_double(MHDPATH "omega", omega_nm,
150 f, qid, __FILE__, __LINE__) ) {return 1;}
151 if( hdf5_read_double(MHDPATH "phase", phase_nm,
152 f, qid, __FILE__, __LINE__) ) {return 1;}
153 if( hdf5_read_double(MHDPATH "amplitude", amplitude_nm,
154 f, qid, __FILE__, __LINE__) ) {return 1;}
155
156 /* Allocate offload array */
157 real* phi = (real*) malloc( nrho * ntime * nmode * sizeof(real) );
158 real* alpha = (real*) malloc( nrho * ntime * nmode * sizeof(real) );
159 if( hdf5_read_double(MHDPATH "alpha", alpha,
160 f, qid, __FILE__, __LINE__) ) {return 1;}
161 if( hdf5_read_double(MHDPATH "phi", phi,
162 f, qid, __FILE__, __LINE__) ) {return 1;}
163
164 int err = mhd_nonstat_init(data, nmode, nrho, ntime, rhomin, rhomax,
165 tmin, tmax, moden, modem, amplitude_nm,
166 omega_nm, phase_nm, alpha, phi);
167 free(moden);
168 free(modem);
169 free(omega_nm);
170 free(phase_nm);
171 free(amplitude_nm);
172 free(phi);
173 free(alpha);
174 return err;
175}
ascot5.h
Main header file for ASCOT5.
real
double real
Definition ascot5.h:85
hdf5_find_group
herr_t hdf5_find_group(hid_t loc, const char *path)
Checks if given group exists within given hdf5 file. Negative value is returned if the group doesn't ...
Definition hdf5_helpers.c:100
hdf5_read_double
int hdf5_read_double(const char *var, real *ptr, hid_t file, char *qid, const char *errfile, int errline)
Read double-valued data from ASCOT5 HDF5 file.
Definition hdf5_helpers.c:166
hdf5_gen_path
char * hdf5_gen_path(const char *original, char *qid, char *path)
Generate a valid path from a given template and qid.
Definition hdf5_helpers.c:131
hdf5_read_int
int hdf5_read_int(const char *var, int *ptr, hid_t file, char *qid, const char *errfile, int errline)
Read int-valued data from ASCOT5 HDF5 file.
Definition hdf5_helpers.c:201
hdf5_helpers.h
Header file for hdf5_helpers.h.
hdf5_mhd_init
int hdf5_mhd_init(hid_t f, mhd_data *data, char *qid)
Initialize MHD data from HDF5 file.
Definition hdf5_mhd.c:31
hdf5_mhd_read_nonstat
int hdf5_mhd_read_nonstat(hid_t f, mhd_nonstat_data *data, char *qid)
Read nonstationary MHD data from HDF5 file.
Definition hdf5_mhd.c:119
MHDPATH
#define MHDPATH
Definition hdf5_mhd.c:16
hdf5_mhd_read_stat
int hdf5_mhd_read_stat(hid_t f, mhd_stat_data *data, char *qid)
Read stationary MHD data from HDF5 file.
Definition hdf5_mhd.c:59
hdf5_mhd.h
Header file for hdf5_mhd.c.
mhd.h
Header file for mhd.c.
mhd_type_stat
@ mhd_type_stat
Definition mhd.h:25
mhd_type_nonstat
@ mhd_type_nonstat
Definition mhd.h:26
mhd_nonstat_init
int mhd_nonstat_init(mhd_nonstat_data *data, int nmode, int nrho, int ntime, real rhomin, real rhomax, real tmin, real tmax, int *moden, int *modem, real *amplitude_nm, real *omega_nm, real *phase_nm, real *alpha, real *phi)
Load MHD data.
Definition mhd_nonstat.c:23
mhd_nonstat.h
Header file for mhd_nonstat.c.
mhd_stat_init
int mhd_stat_init(mhd_stat_data *data, int nmode, int nrho, real rhomin, real rhomax, int *moden, int *modem, real *amplitude_nm, real *omega_nm, real *phase_nm, real *alpha, real *phi)
Load MHD data.
Definition mhd_stat.c:23
mhd_stat.h
Header file for mhd_stat.c.
mhd_data
MHD simulation data.
Definition mhd.h:35
mhd_data::type
mhd_type type
Definition mhd.h:36
mhd_data::stat
mhd_stat_data stat
Definition mhd.h:37
mhd_data::nonstat
mhd_nonstat_data nonstat
Definition mhd.h:38
mhd_nonstat_data
MHD parameters.
Definition mhd_nonstat.h:18
mhd_stat_data
MHD stat parameters on the target.
Definition mhd_stat.h:18
Generated on Fri Feb 7 2025 13:39:30 for ASCOT5 by doxygen 1.13.2