(file) Return to update_sharp_keys.c CVS log (file) (dir) Up to [Development] / JSOC / proj / sharp / apps

  1 mbobra 1.1 /*                                                                      
  2             *      MODULE NAME: update_sharp_keys.c
  3             *
  4             *	DESCRIPTION: This module recalculates SHARP keywords. This is accomplished by 
  5             *      cloning a record, recalculating keywords of choice (i.e., user input), 
  6             *      and pointing to the same segments as the old record. Associated error keys
  7             *      are computed by default. 
  8             * 
  9             *      This module accounts for versioning by appending to the keyword CODEVER7 and HISTORY:
 10             *      CODEVER7 will contain multiple lines: the production build of sharp.c, the 
 11             *      production build of include file sw_functions.c, and the production build
 12             *      of update_sharp_keys.c. HISTORY will include a human-readable sentence about which
 13             *      keywords were updated. 
 14             * 
 15 mbobra 1.10  *      N.B.: This module calculates the keyword specified in keylist and then does a 
 16              *      drms_copykeys() to copy over all the remaining keywords. However, if [1] a keyword
 17              *      does not exist in the .jsd files of the input series, and [2] said keyword X is not
 18              *      specified in keylist, then [3] drms_copykeys() will copy DRMS_MISSING_VALUE into 
 19              *      keyword X for all the output series.
 20              *
 21 mbobra 1.1   *      This module does not produce segments.
 22              *
 23              *      INPUTS     : -- DRMS SHARP series
 24              *                   -- DRMS SHARP CEA series
 25              *                   -- HARPNUM
 26              *                   -- comma separated list of keywords to recalculate
 27 mbobra 1.9   *                   -- DEBUG flag (use like this debug=debug)
 28 mbobra 1.1   *
 29              *	AUTHOR     : Monica Bobra
 30              *
 31              *	Version    :   v0.0	Jun 14 2013
 32 mbobra 1.6   *                     v0.1     Feb 11 2014 Added different input and output series
 33 mbobra 1.1   *
 34              *	EXAMPLE    :
 35 mbobra 1.6   *      update_sharp_keys sharpseriesin=hmi.sharp_720s sharpceaseriesin=hmi.sharp_cea_720s //
 36 mbobra 1.9   *      HARPNUM=1 sharpseriesout=hmi.sharp_720s sharpceaseriesout=hmi.sharp_cea_720s keylist=USFLUX,TOTPOT debug=debug
 37 mbobra 1.1   *
 38              */
 39             
 40             /* Include files */
 41             #include <stdio.h>
 42             #include <stdlib.h>
 43             #include <time.h>
 44             #include <sys/time.h>
 45             #include <math.h>
 46             #include <string.h>
 47             #include "jsoc_main.h"
 48             #include "astro.h"
 49             #include "fstats.h"
 50             #include "cartography.c"
 51             #include "fresize.h"
 52             #include "finterpolate.h"
 53             #include "img2helioVector.c"
 54             #include "copy_me_keys.c"
 55             #include "errorprop.c"
 56             #include "sw_functions.c"
 57             #include <mkl_blas.h>
 58 mbobra 1.1  #include <mkl_service.h>
 59             #include <mkl_lapack.h>
 60             #include <mkl_vml_functions.h>
 61             #include <omp.h>
 62              
 63             /* Define values */                     
 64             #define PI              (M_PI)
 65             #define RADSINDEG       (PI/180.)
 66             #define ARRLENGTH(ARR) (sizeof(ARR) / sizeof(ARR[0]))
 67             #define DIE(msg) {fflush(stdout); fprintf(stderr,"%s, status=%d\n", msg, status); return(status);}
 68             #define SHOW(msg) {printf("%s", msg); fflush(stdout);}
 69             
 70             char *module_name = "update_sharp_keys";  /* Module name */
 71 mbobra 1.6  char *version_id  = "2014 Feb 12";        /* Version number */
 72 mbobra 1.1  
 73             ModuleArgs_t module_args[] =
 74             {
 75 mbobra 1.9  	{ARG_STRING, "sharpseriesin",      NULL, "Input Sharp dataseries"},
 76             	{ARG_STRING, "sharpceaseriesin",   NULL, "Input Sharp CEA dataseries"},
 77 mbobra 1.6  	{ARG_STRING, "sharpseriesout",     NULL, "Output Sharp dataseries"},
 78             	{ARG_STRING, "sharpceaseriesout",  NULL, "Output Sharp CEA dataseries"},
 79 mbobra 1.9  	{ARG_INT,    "HARPNUM",            NULL, "HARP number"},
 80             	{ARG_STRING, "keylist",            NULL, "comma separated list of keywords to update"},
 81             	{ARG_STRING, "debug",              NULL, "debug mode functionality. use like this: debug=debug"},
 82 mbobra 1.1  	{ARG_END}
 83             };
 84             
 85             
 86             int DoIt(void)                    
 87             {
 88                     int errbufstat=setvbuf(stderr, NULL, _IONBF, BUFSIZ);
 89                     int outbufstat=setvbuf(stdout, NULL, _IONBF, BUFSIZ);
 90             
 91             	int status = DRMS_SUCCESS;
 92             	int nrecs, irec;
 93             
 94                     /* Keywords */
 95             	float mean_vf; 
 96             	float absFlux;
 97                     float count_mask;
 98             	float mean_hf;
 99             	float mean_gamma;
100             	float mean_derivative_btotal;
101             	float mean_derivative_bh; 
102             	float mean_derivative_bz;
103 mbobra 1.1  	float mean_jz;
104             	float us_i;
105             	float mean_alpha;
106             	float mean_ih;
107             	float total_us_ih;
108             	float total_abs_ih;
109             	float totaljz;
110             	float totpot;
111             	float meanpot;
112             	float area_w_shear_gt_45;
113             	float meanshear_angle;
114             	float area_w_shear_gt_45h;
115             	float meanshear_angleh; 
116                     float mean_derivative_btotal_err;
117                     float mean_vf_err;
118                     float mean_gamma_err;
119                     float mean_derivative_bh_err;
120                     float mean_derivative_bz_err;
121                     float mean_jz_err;
122                     float us_i_err; 
123                     float mean_alpha_err;
124 mbobra 1.1          float mean_ih_err;
125                     float total_us_ih_err;
126                     float total_abs_ih_err;
127                     float totaljz_err;
128                     float meanpot_err;
129                     float totpot_err;
130 mbobra 1.6          float Rparam;
131 mbobra 1.1          float meanshear_angle_err;
132 mbobra 1.8          float totfx;
133                     float totfy;
134                     float totfz;
135                     float totbsq;
136                     float epsx;
137                     float epsy;
138                     float epsz;
139             
140 mbobra 1.6  	char *sharpseriesin = (char *) params_get_str(&cmdparams, "sharpseriesin");
141             	char *sharpceaseriesin = (char *) params_get_str(&cmdparams, "sharpceaseriesin");
142             	char *sharpseriesout = (char *) params_get_str(&cmdparams, "sharpseriesout");
143             	char *sharpceaseriesout = (char *) params_get_str(&cmdparams, "sharpceaseriesout");
144             
145                     int sameflag = strcmp(sharpseriesin, sharpseriesout) == 0;
146                     int testflag = strcmp(sharpceaseriesin, sharpceaseriesout) == 0;
147                     if (testflag != sameflag)
148                       DIE("Either both outputs must be the same as their inputs, or both be different.");
149 mbobra 1.1  
150             	int harpnum = params_get_int(&cmdparams, "HARPNUM");
151 mbobra 1.6  
152 mbobra 1.1  	char sharpquery[256];
153             	char sharpceaquery[256];
154 mbobra 1.6  	sprintf(sharpceaquery, "%s[%d]\n", sharpceaseriesin,harpnum);
155             	printf(sharpceaquery, "%s[%d]\n", sharpceaseriesin,harpnum);
156 mbobra 1.1  
157 mbobra 1.6  	sprintf(sharpquery, "%s[%d]\n", sharpseriesin,harpnum);
158             	sprintf(sharpquery, "%s[%d]\n", sharpseriesin,harpnum);
159 mbobra 1.1  
160 mbobra 1.6  	DRMS_RecordSet_t *sharpinrecset= drms_open_records(drms_env, sharpquery, &status);
161 mbobra 1.1  	if (status != DRMS_SUCCESS)
162             		DIE("Problem opening sharp recordset.");
163 mbobra 1.6  	nrecs=sharpinrecset->n;
164             	if (nrecs == 0)
165 mbobra 1.1  		DIE("Sharp recordset contains no records.");
166 mbobra 1.6          printf("nrecs=%d\n",nrecs);
167 mbobra 1.1  
168             	DRMS_RecordSet_t *sharpceainrecset = drms_open_records(drms_env, sharpceaquery, &status);
169             	if (status != DRMS_SUCCESS)
170             		DIE("Problem opening sharp cea recordset.");
171 mbobra 1.6  	if (sharpceainrecset->n != nrecs)
172             		DIE("Sharp cea recordset differs from sharp recordset.");
173             
174                     DRMS_RecordSet_t *sharpoutrecset, *sharpceaoutrecset;
175                     if (sameflag)
176                     {
177             	sharpoutrecset = drms_clone_records(sharpinrecset, DRMS_PERMANENT, DRMS_SHARE_SEGMENTS, &status);
178             	if (status != DRMS_SUCCESS)
179             		DIE("Problem cloning sharp records.");
180             	sharpceaoutrecset = drms_clone_records(sharpceainrecset, DRMS_PERMANENT, DRMS_SHARE_SEGMENTS, &status);
181 mbobra 1.1  	if (status != DRMS_SUCCESS)
182             		DIE("Problem cloning sharp cea records.");
183 mbobra 1.6          }
184                     else
185                     {
186             	sharpoutrecset = drms_create_records(drms_env, nrecs, sharpseriesout, DRMS_PERMANENT, &status);
187             	if (status != DRMS_SUCCESS)
188             		DIE("Problem creating sharp records.");
189             	sharpceaoutrecset = drms_create_records(drms_env, nrecs, sharpceaseriesout, DRMS_PERMANENT, &status);
190             	if (status != DRMS_SUCCESS)
191             		DIE("Problem creating sharp cea records.");
192                     }
193             
194 mbobra 1.1  
195             	char *keylist = (char *) params_get_str(&cmdparams, "keylist");
196 mbobra 1.9  	char *debug   = (char *) params_get_str(&cmdparams, "debug");
197 mbobra 1.1  
198                     // Flags to indicate which keyword will be recalculated
199             	int meanvfflag  = (strstr(keylist,"USFLUX")  != NULL);  // generalize so that lowercase is acceptable
200             	int totpotflag  = (strstr(keylist,"TOTPOT")  != NULL);  
201                     int meangamflag = (strstr(keylist,"MEANGAM") != NULL);
202                     int meangbtflag = (strstr(keylist,"MEANGBT") != NULL);
203                     int meangbzflag = (strstr(keylist,"MEANGBZ") != NULL);
204                     int meangbhflag = (strstr(keylist,"MEANGBH") != NULL);
205                     int meanjzdflag = (strstr(keylist,"MEANJZD") != NULL);
206                     int totusjzflag = (strstr(keylist,"TOTUSJZ") != NULL);
207                     int meanalpflag = (strstr(keylist,"MEANALP") != NULL);
208                     int meanjzhflag = (strstr(keylist,"MEANJZH") != NULL);
209                     int totusjhflag = (strstr(keylist,"TOTUSJH") != NULL);
210                     int absnjzhflag = (strstr(keylist,"ABSNJZH") != NULL);
211                     int savncppflag = (strstr(keylist,"SAVNCPP") != NULL);
212                     int meanpotflag = (strstr(keylist,"MEANPOT") != NULL);
213                     int meanshrflag = (strstr(keylist,"MEANSHR") != NULL);
214                     int shrgt45flag = (strstr(keylist,"SHRGT45") != NULL);
215 mbobra 1.6          int r_valueflag = (strstr(keylist,"R_VALUE") != NULL);
216 mbobra 1.8          int totbsqflag  = (strstr(keylist,"TOTBSQ")  != NULL);
217                     int totfxflag   = (strstr(keylist,"TOTFX")   != NULL);
218                     int totfyflag   = (strstr(keylist,"TOTFY")   != NULL);
219                     int totfzflag   = (strstr(keylist,"TOTFZ")   != NULL);
220                     int epsxflag    = (strstr(keylist,"EPSX")    != NULL);
221                     int epsyflag    = (strstr(keylist,"EPSY")    != NULL);
222                     int epszflag    = (strstr(keylist,"EPSZ")    != NULL);
223 mbobra 1.9          int debugflag   = (strstr(debug,"debug")     != NULL);
224 mbobra 1.8  
225 mbobra 1.11     
226             	for (irec=0;irec<nrecs;irec++)
227             	{
228             
229             
230             	DRMS_Record_t *sharpinrec = sharpinrecset->records[irec];
231             	DRMS_Record_t *sharpceainrec = sharpceainrecset->records[irec];
232 mbobra 1.1  	DRMS_Segment_t *inseg = drms_segment_lookup(sharpceainrec, "Br");
233             	int nx = inseg->axis[0];
234             	int ny = inseg->axis[1];
235             	int nxny = nx * ny;
236             	int dims[2] = {nx, ny};
237 mbobra 1.6   
238 mbobra 1.1  	// Temp arrays 	
239             	float *bh      = (float *) (malloc(nxny * sizeof(float)));
240             	float *bt      = (float *) (malloc(nxny * sizeof(float)));
241             	float *jz      = (float *) (malloc(nxny * sizeof(float)));
242             	float *bpx     = (float *) (malloc(nxny * sizeof(float)));
243             	float *bpy     = (float *) (malloc(nxny * sizeof(float)));
244             	float *bpz     = (float *) (malloc(nxny * sizeof(float)));
245             	float *derx    = (float *) (malloc(nxny * sizeof(float)));
246             	float *dery    = (float *) (malloc(nxny * sizeof(float)));
247             	float *derx_bt = (float *) (malloc(nxny * sizeof(float)));
248             	float *dery_bt = (float *) (malloc(nxny * sizeof(float)));
249             	float *derx_bh = (float *) (malloc(nxny * sizeof(float)));
250             	float *dery_bh = (float *) (malloc(nxny * sizeof(float)));
251             	float *derx_bz = (float *) (malloc(nxny * sizeof(float)));
252             	float *dery_bz = (float *) (malloc(nxny * sizeof(float)));
253             	float *bt_err  = (float *) (malloc(nxny * sizeof(float)));
254             	float *bh_err  = (float *) (malloc(nxny * sizeof(float)));
255                     float *jz_err  = (float *) (malloc(nxny * sizeof(float)));
256                     float *jz_err_squared = (float *) (malloc(nxny * sizeof(float)));
257                     float *jz_rms_err = (float *) (malloc(nxny * sizeof(float)));
258                     float *jz_err_squared_smooth = (float *) (malloc(nxny * sizeof(float)));
259 mbobra 1.1  	float *jz_smooth = (float *) (malloc(nxny * sizeof(float)));
260 mbobra 1.11 	float *err_term1  = (float *) (malloc(nxny * sizeof(float)));
261             	float *err_term2  = (float *) (malloc(nxny * sizeof(float)));
262 mbobra 1.1  
263                     // ephemeris variables
264             	float  cdelt1_orig, cdelt1, dsun_obs, imcrpix1, imcrpix2, crpix1, crpix2;
265             	double rsun_ref, rsun_obs;
266             
267                     // outrecords 
268 mbobra 1.6  	DRMS_Record_t *sharpoutrec, *sharpceaoutrec, *testrec;
269 mbobra 1.1  
270                     // prepare to set CODEVER7 (CVS Version of the SHARP module)
271             	char *cvsinfo0;
272             	char *history0;
273 mbobra 1.11 	char *cvsinfo1 = strdup("$Id: update_sharp_keys.c,v 1.10 2014/06/09 21:07:41 mbobra Exp $");
274 mbobra 1.1  	char *cvsinfo2 = sw_functions_version();
275             	char *cvsinfoall = (char *)malloc(2048);
276                     char historyofthemodule[2048];
277                     cvsinfo0    = drms_getkey_string(sharpinrec, "CODEVER7", &status);
278                     strcpy(cvsinfoall,cvsinfo0);
279                     strcat(cvsinfoall,"\n");
280                     strcat(cvsinfoall,cvsinfo2);
281             
282                     // prepare to set HISTORY (History of the data)
283             	char timebuf[1024];
284             	float UNIX_epoch = -220924792.000; 	
285             	sprint_time(timebuf, (double)time(NULL) + UNIX_epoch, "ISO", 0);
286                     sprintf(historyofthemodule,"The following keywords were re-computed on %s: %s.",timebuf,keylist); 
287             	printf("historyofthemodule=%s\n",historyofthemodule);
288             	history0    = drms_getkey_string(sharpinrec, "HISTORY", &status);
289             	strcat(historyofthemodule,"\n");
290                     strcat(historyofthemodule,history0);
291              
292 mbobra 1.6          // grab one record such that we can malloc the R-parameter calculation arrays outside of the loop
293                     // assume scale = round(2./cdelt1) does not change for any given HARP
294                     testrec  = sharpceainrecset->records[nrecs/2];
295                     dsun_obs = drms_getkey_float(testrec, "DSUN_OBS",   &status); 
296                     rsun_ref = drms_getkey_double(testrec, "RSUN_REF", &status);
297                     cdelt1_orig = drms_getkey_float(testrec, "CDELT1",   &status);
298                     cdelt1      = (atan((rsun_ref*cdelt1_orig*RADSINDEG)/(dsun_obs)))*(1/RADSINDEG)*(3600.);
299                     int scale = round(2.0/cdelt1);
300                     int nx1 = nx/scale;
301                     int ny1 = ny/scale;
302 mbobra 1.7          int nxp = nx1+40;
303                     int nyp = ny1+40;
304 mbobra 1.6  
305 mbobra 1.10         // check to make sure that the dimensions passed into fsample() are adequate
306 mbobra 1.6  	if (nx1 > floor((nx-1)/scale + 1) )
307             		DIE("X-dimension of output array in fsample() is too large.");
308             	if (ny1 > floor((ny-1)/scale + 1) )
309 mbobra 1.10         	DIE("Y-dimension of output array in fsample() is too large.");
310 mbobra 1.6   
311                     // malloc some arrays for the R parameter calculation
312 mbobra 1.7          float *rim     = (float *)malloc(nx1*ny1*sizeof(float));
313                     float *p1p0    = (float *)malloc(nx1*ny1*sizeof(float));
314                     float *p1n0    = (float *)malloc(nx1*ny1*sizeof(float));
315                     float *p1p     = (float *)malloc(nx1*ny1*sizeof(float));
316                     float *p1n     = (float *)malloc(nx1*ny1*sizeof(float));
317                     float *p1      = (float *)malloc(nx1*ny1*sizeof(float));
318                     float *pmap    = (float *)malloc(nxp*nyp*sizeof(float));
319                     float *p1pad   = (float *)malloc(nxp*nyp*sizeof(float));
320                     float *pmapn   = (float *)malloc(nx1*ny1*sizeof(float));
321 mbobra 1.8  
322                     // define some arrays for the lorentz force calculation
323                     float *fx = (float *) (malloc(nxny * sizeof(float)));
324                     float *fy = (float *) (malloc(nxny * sizeof(float)));
325                     float *fz = (float *) (malloc(nxny * sizeof(float)));
326                 
327 mbobra 1.1     	   // Get emphemeris
328 mbobra 1.7    	   sharpinrec    = sharpinrecset->records[irec];
329 mbobra 1.6    	   sharpceainrec = sharpceainrecset->records[irec];
330 mbobra 1.7  	   cdelt1_orig   = drms_getkey_float(sharpceainrec, "CDELT1",   &status);
331             	   dsun_obs      = drms_getkey_float(sharpinrec, "DSUN_OBS",   &status);
332             	   rsun_ref      = drms_getkey_double(sharpinrec, "RSUN_REF", &status);
333             	   rsun_obs      = drms_getkey_double(sharpinrec, "RSUN_OBS", &status);
334             	   imcrpix1      = drms_getkey_float(sharpinrec, "IMCRPIX1", &status);
335             	   imcrpix2      = drms_getkey_float(sharpinrec, "IMCRPIX2", &status);
336             	   crpix1        = drms_getkey_float(sharpinrec, "CRPIX1", &status);
337             	   crpix2        = drms_getkey_float(sharpinrec, "CRPIX2", &status);
338 mbobra 1.6  
339 mbobra 1.1    	   sharpoutrec = sharpoutrecset->records[irec];
340               	   sharpceaoutrec = sharpceaoutrecset->records[irec];
341 mbobra 1.6             if (!sameflag)
342                        {
343                          drms_copykeys(sharpoutrec, sharpinrec, 0, kDRMS_KeyClass_Explicit);
344                          drms_copykeys(sharpceaoutrec, sharpceainrec, 0, kDRMS_KeyClass_Explicit);
345                        }
346             
347 mbobra 1.11            TIME trec, tnow; /* 1970.01.01_00:00:00_UTC */
348 mbobra 1.6             tnow = (double)time(NULL);
349                        tnow += UNIX_epoch;
350 mbobra 1.1  
351 mbobra 1.9             if (debugflag)
352                        {
353                           printf("i=%d\n",irec);
354                        }
355             
356 mbobra 1.6  	   // set CODEVER7 and HISTORY and DATE
357 mbobra 1.1  	   drms_setkey_string(sharpoutrec, "CODEVER7", cvsinfoall);
358                        drms_setkey_string(sharpoutrec,"HISTORY",historyofthemodule);           
359             	   drms_setkey_string(sharpceaoutrec, "CODEVER7", cvsinfoall);
360                        drms_setkey_string(sharpceaoutrec,"HISTORY",historyofthemodule);
361 mbobra 1.6             drms_setkey_time(sharpoutrec,"DATE",tnow);  
362                        drms_setkey_time(sharpceaoutrec,"DATE",tnow);
363 mbobra 1.1  
364             	   // Get bx, by, bz, mask	
365             	   // Use HARP (Turmon) bitmap as a threshold on spaceweather quantities
366 mbobra 1.6  	   DRMS_Segment_t *bitmaskSeg = drms_segment_lookup(sharpceainrec, "bitmap");
367 mbobra 1.1  	   DRMS_Array_t *bitmaskArray = drms_segment_read(bitmaskSeg, DRMS_TYPE_INT, &status);
368             	   int *bitmask = (int *) bitmaskArray->data;	 // get the previously made mask array	
369             
370             	   //Use conf_disambig map as a threshold on spaceweather quantities 
371 mbobra 1.6  	   DRMS_Segment_t *maskSeg = drms_segment_lookup(sharpceainrec, "conf_disambig");       
372 mbobra 1.1  	   DRMS_Array_t *maskArray = drms_segment_read(maskSeg, DRMS_TYPE_INT, &status);	
373                        if (status != DRMS_SUCCESS)
374                           DIE("No CONF_DISAMBIG segment.");
375             	   int *mask = (int *) maskArray->data;		// get the previously made mask array	
376             
377 mbobra 1.6   	   //Use magnetogram map to compute R
378             	   DRMS_Segment_t *losSeg = drms_segment_lookup(sharpceainrec, "magnetogram");
379             	   DRMS_Array_t *losArray = drms_segment_read(losSeg, DRMS_TYPE_FLOAT, &status);
380             	   float *los = (float *) losArray->data;		// los
381             
382             	   DRMS_Segment_t *bxSeg = drms_segment_lookup(sharpceainrec, "Bp");
383 mbobra 1.1  	   DRMS_Array_t *bxArray = drms_segment_read(bxSeg, DRMS_TYPE_FLOAT, &status);
384             	   float *bx = (float *) bxArray->data;		// bx
385             	
386 mbobra 1.6  	   DRMS_Segment_t *bySeg = drms_segment_lookup(sharpceainrec, "Bt");
387 mbobra 1.1  	   DRMS_Array_t *byArray = drms_segment_read(bySeg, DRMS_TYPE_FLOAT, &status);
388             	   float *by = (float *) byArray->data;		// by
389             	   for (int i = 0; i < nxny; i++) by[i] *= -1;
390             	
391 mbobra 1.6  	   DRMS_Segment_t *bzSeg = drms_segment_lookup(sharpceainrec, "Br");
392 mbobra 1.1  	   DRMS_Array_t *bzArray = drms_segment_read(bzSeg, DRMS_TYPE_FLOAT, &status);
393             	   float *bz = (float *) bzArray->data;		// bz
394             
395 mbobra 1.6     	   DRMS_Segment_t *bz_errSeg = drms_segment_lookup(sharpceainrec, "Br_err");
396 mbobra 1.1  	   DRMS_Array_t *bz_errArray = drms_segment_read(bz_errSeg, DRMS_TYPE_FLOAT, &status);
397             	   float *bz_err = (float *) bz_errArray->data;		// bz_err
398             
399 mbobra 1.6  	   DRMS_Segment_t *by_errSeg = drms_segment_lookup(sharpceainrec, "Bt_err");
400 mbobra 1.1  	   DRMS_Array_t *by_errArray = drms_segment_read(by_errSeg, DRMS_TYPE_FLOAT, &status);
401             	   float *by_err = (float *) by_errArray->data;		// by_err
402             
403 mbobra 1.6  	   DRMS_Segment_t *bx_errSeg = drms_segment_lookup(sharpceainrec, "Bp_err");
404 mbobra 1.1  	   DRMS_Array_t *bx_errArray = drms_segment_read(bx_errSeg, DRMS_TYPE_FLOAT, &status);
405             	   float *bx_err = (float *) bx_errArray->data;		// bx_err
406             
407 mbobra 1.6             /***** USFLUX, Example Function 1 *************************************/ 
408 mbobra 1.1  	   if (meanvfflag)
409             	   {
410                           // Compute unsigned flux 
411                           if (computeAbsFlux(bz_err, bz , dims, &absFlux, &mean_vf,  &mean_vf_err, 
412                                              &count_mask, mask, bitmask, cdelt1, rsun_ref, rsun_obs))
413                           {
414             	         mean_vf           = DRMS_MISSING_FLOAT;
415                              mean_vf_err       = DRMS_MISSING_FLOAT;
416                              count_mask        = DRMS_MISSING_INT;
417             	      }
418             
419                        drms_setkey_float(sharpoutrec, "USFLUX",  mean_vf);
420                        drms_setkey_float(sharpoutrec, "CMASK",   count_mask); 	
421                        drms_setkey_float(sharpoutrec, "ERRVF",   mean_vf_err);
422             
423                        drms_setkey_float(sharpceaoutrec, "USFLUX",  mean_vf);
424                        drms_setkey_float(sharpceaoutrec, "CMASK",   count_mask); 	
425                        drms_setkey_float(sharpceaoutrec, "ERRVF",   mean_vf_err);
426                        }
427 mbobra 1.6  
428                         /***** RPARAM, Example Function 14 *************************************/ 
429             	   if (r_valueflag)
430             	   {
431                           if (computeR(bz_err, los , dims, &Rparam, cdelt1, rim, p1p0, p1n0,
432 mbobra 1.7                             p1p, p1n, p1, pmap, nx1, ny1, scale, p1pad, nxp, nyp, pmapn))
433             
434 mbobra 1.6                {
435             		 Rparam            = DRMS_MISSING_FLOAT;
436             	      }
437             
438                        drms_setkey_float(sharpoutrec, "R_VALUE",  Rparam);
439                        drms_setkey_float(sharpceaoutrec, "R_VALUE", Rparam);
440             
441                        }
442               
443 mbobra 1.1             /***** MEANPOT and TOTPOT, Example Function 13  (Requires Keiji's code) **/
444                    
445                	   if (totpotflag || meanpotflag)
446             	   {
447                           // First compute potential field	
448                           for (int i = 0; i < nxny; i++) bpz[i] = bz[i];
449             	      greenpot(bpx, bpy, bpz, nx, ny);
450              
451                           // Then compute energy   
452             	      if (computeFreeEnergy(bx_err, by_err, bx, by, bpx, bpy, dims, 
453                                                 &meanpot, &meanpot_err, &totpot, &totpot_err, 
454             			            mask, bitmask, cdelt1, rsun_ref, rsun_obs)) 
455                           {
456             	         meanpot           = DRMS_MISSING_FLOAT; 
457             	         totpot            = DRMS_MISSING_FLOAT;
458                              meanpot_err       = DRMS_MISSING_FLOAT;
459                              totpot_err        = DRMS_MISSING_FLOAT;
460             	      }
461             	   
462                           // Set sharp keys          
463                           drms_setkey_float(sharpoutrec, "MEANPOT",  meanpot); 
464 mbobra 1.1                drms_setkey_float(sharpoutrec, "TOTPOT",   totpot); 	
465                           drms_setkey_float(sharpoutrec, "ERRMPOT",  meanpot_err); 
466                           drms_setkey_float(sharpoutrec, "ERRTPOT",  totpot_err);  
467             
468                           // Set sharp cea keys
469                           drms_setkey_float(sharpceaoutrec, "MEANPOT",  meanpot); 
470                           drms_setkey_float(sharpceaoutrec, "TOTPOT",   totpot); 	
471                           drms_setkey_float(sharpceaoutrec, "ERRMPOT",  meanpot_err); 
472                           drms_setkey_float(sharpceaoutrec, "ERRTPOT",  totpot_err);
473                        }
474                    
475                        /***** MEANGAM, Example Function 3 ************************************/
476                
477                        if (meangamflag)
478                        {
479                           // First compute horizontal field
480                           computeBh(bx_err, by_err, bh_err, bx, by, bz, bh, dims, &mean_hf, mask, bitmask);
481             
482                           // Then compute inclination angle, gamma
483                           if (computeGamma(bz_err, bh_err, bx, by, bz, bh, dims, &mean_gamma, &mean_gamma_err,mask, bitmask))
484             	      {	
485 mbobra 1.1                   mean_gamma                 =  DRMS_MISSING_FLOAT;
486                              mean_gamma_err             =  DRMS_MISSING_FLOAT;
487                           }
488             
489                           // Set sharp keys    
490                           drms_setkey_float(sharpoutrec, "MEANGAM",  mean_gamma); 
491                           drms_setkey_float(sharpoutrec, "ERRGAM",   mean_gamma_err); 	
492             
493                           // Set sharp cea keys
494                           drms_setkey_float(sharpceaoutrec, "MEANGAM", mean_gamma); 
495                           drms_setkey_float(sharpceaoutrec, "ERRGAM",  mean_gamma_err); 	
496                        }
497             
498                        /***** MEANGBT, Example Function 5 (Requires Function 4) *************/
499             
500                        if (meangbtflag)
501                        {
502                           // First, compute Bt 
503                           computeB_total(bx_err, by_err, bz_err, bt_err, bx, by, bz, bt, dims, mask, bitmask);
504             
505                           // Then take the derivative of Bt
506 mbobra 1.1  	      if (computeBtotalderivative(bt, dims, &mean_derivative_btotal, mask, bitmask, derx_bt, dery_bt, bt_err, 
507                                                       &mean_derivative_btotal_err))
508                           {
509             	         mean_derivative_btotal     = DRMS_MISSING_FLOAT;
510             	         mean_derivative_btotal_err = DRMS_MISSING_FLOAT;
511                           }
512             
513                           // Set sharp keys    
514                           drms_setkey_float(sharpoutrec, "MEANGBT",  mean_derivative_btotal); 
515                           drms_setkey_float(sharpoutrec, "ERRBT",  mean_derivative_btotal_err); 
516             	
517                           // Set sharp cea keys  
518                           drms_setkey_float(sharpceaoutrec, "MEANGBT", mean_derivative_btotal); 
519                           drms_setkey_float(sharpceaoutrec, "ERRBT", mean_derivative_btotal_err); 	
520                        }
521                
522                        /***** MEANGBH, Example Function 6 (Requires Function 2) *************/
523             
524                        if (meangbhflag)
525                        {
526                           // First, compute Bh
527 mbobra 1.1     	      computeBh(bx_err, by_err, bh_err, bx, by, bz, bh, dims, &mean_hf, mask, bitmask);
528             
529                           // Then take the derivative of Bh
530             	      if (computeBhderivative(bh, bh_err, dims, &mean_derivative_bh, &mean_derivative_bh_err, mask, bitmask, derx_bh, dery_bh))
531                           {
532             	         mean_derivative_bh       = DRMS_MISSING_FLOAT;
533                              mean_derivative_bh_err   = DRMS_MISSING_FLOAT;
534             	      }
535             
536                           // Set sharp keys   
537                           drms_setkey_float(sharpoutrec, "MEANGBH", mean_derivative_bh); 
538                           drms_setkey_float(sharpoutrec, "ERRBH", mean_derivative_bh_err); 
539             	
540                           // Set sharp cea keys  
541                           drms_setkey_float(sharpceaoutrec, "MEANGBH", mean_derivative_bh); 
542                           drms_setkey_float(sharpceaoutrec, "ERRBH", mean_derivative_bh_err); 	
543                        }
544             
545                        /***** MEANGBZ, Example Function 7 ************************************/
546             
547                        if (meangbzflag)
548 mbobra 1.1             {
549                           // Compute Bz derivative
550                           if (computeBzderivative(bz, bz_err, dims, &mean_derivative_bz, &mean_derivative_bz_err, mask, bitmask, derx_bz, dery_bz))
551                           {
552             	         mean_derivative_bz     = DRMS_MISSING_FLOAT; 
553                              mean_derivative_bz_err = DRMS_MISSING_FLOAT; 
554                           }
555             	
556                           // Set sharp keys   
557                           drms_setkey_float(sharpoutrec, "MEANGBZ",  mean_derivative_bz); 
558                           drms_setkey_float(sharpoutrec, "ERRBZ",  mean_derivative_bz_err); 
559             	
560                           // Set sharp cea keys   
561                           drms_setkey_float(sharpceaoutrec, "MEANGBZ", mean_derivative_bz); 
562                           drms_setkey_float(sharpceaoutrec, "ERRBZ", mean_derivative_bz_err); 	
563                        }
564             
565                        /***** MEANJZD and TOTUSJZ, Example Function 9 (Requires Function 8) ***/
566             
567                        if (meanjzdflag || totusjzflag)
568                        {
569 mbobra 1.1                // First, compute Jz on all pixels
570             	      computeJz(bx_err, by_err, bx, by, dims, jz, jz_err, jz_err_squared, mask, bitmask, cdelt1, rsun_ref, rsun_obs, 
571 mbobra 1.11                      derx, dery, err_term1, err_term2);
572 mbobra 1.1     
573                           // Then take the sums of the appropriate pixels
574                           if(computeJzsmooth(bx, by, dims, jz, jz_smooth, jz_err, jz_rms_err, jz_err_squared_smooth, &mean_jz,
575                                    &mean_jz_err, &us_i, &us_i_err, mask, bitmask, cdelt1,
576                                    rsun_ref, rsun_obs, derx, dery))
577             
578             
579                           {
580                              mean_jz                = DRMS_MISSING_FLOAT;
581             	         us_i                   = DRMS_MISSING_FLOAT;
582                              mean_jz_err            = DRMS_MISSING_FLOAT;
583                              us_i_err               = DRMS_MISSING_FLOAT;
584             	      }
585             
586                           // Set sharp keys   
587                           drms_setkey_float(sharpoutrec, "MEANJZD",  mean_jz); 
588                           drms_setkey_float(sharpoutrec, "TOTUSJZ",  us_i); 	
589                           drms_setkey_float(sharpoutrec, "ERRJZ",  mean_jz_err); 
590                           drms_setkey_float(sharpoutrec, "ERRUSI",  us_i_err); 	
591             
592                           // Set sharp cea keys   
593 mbobra 1.1                drms_setkey_float(sharpceaoutrec, "MEANJZD", mean_jz); 
594                           drms_setkey_float(sharpceaoutrec, "TOTUSJZ", us_i); 	
595                           drms_setkey_float(sharpceaoutrec, "ERRJZ", mean_jz_err); 
596                           drms_setkey_float(sharpceaoutrec, "ERRUSI", us_i_err); 	
597                        }
598             
599                        /***** MEANALP, Example Function 10 (Requires Function 8)*********/
600             
601                        if (meanalpflag)
602                        {
603                           // First, compute Jz on all pixels
604                	      computeJz(bx_err, by_err, bx, by, dims, jz, jz_err, jz_err_squared, mask, bitmask, cdelt1, rsun_ref, rsun_obs, 
605 mbobra 1.11                         derx, dery, err_term1, err_term2);
606 mbobra 1.1  
607                           // Then compute alpha quantities on select pixels
608             	      if (computeAlpha(jz_err, bz_err, bz, dims, jz, jz_smooth, &mean_alpha, &mean_alpha_err, 
609                                            mask, bitmask, cdelt1, rsun_ref, rsun_obs))
610                           {
611             	         mean_alpha             = DRMS_MISSING_FLOAT;
612                              mean_alpha_err         = DRMS_MISSING_FLOAT;
613                           }
614             
615                           // Set sharp keys   	
616                           drms_setkey_float(sharpoutrec, "MEANALP", mean_alpha); 
617                           drms_setkey_float(sharpoutrec, "ERRALP", mean_alpha_err); 	
618             
619                           // Set sharp cea keys   
620                           drms_setkey_float(sharpceaoutrec, "MEANALP", mean_alpha); 
621                           drms_setkey_float(sharpceaoutrec, "ERRALP", mean_alpha_err); 	
622             
623                        }
624             
625                        /***** MEANJZH, TOTUSJH, ABSNJZH, Example Function 11 (Requires Function 8) ***/
626             
627 mbobra 1.1             if (meanjzhflag || totusjhflag || absnjzhflag)
628                        {
629                           // First, compute Jz on all pixels
630                 	      computeJz(bx_err, by_err, bx, by, dims, jz, jz_err, jz_err_squared, mask, bitmask, cdelt1, rsun_ref, rsun_obs, 
631 mbobra 1.11                         derx, dery, err_term1, err_term2);
632 mbobra 1.1  
633                           // Then compute helicity quantities on select pixels
634                        if (computeHelicity(jz_err, jz_rms_err, bz_err, bz, dims, jz, &mean_ih, &mean_ih_err, &total_us_ih, 
635                                            &total_abs_ih, &total_us_ih_err, &total_abs_ih_err, mask, bitmask, cdelt1, rsun_ref, rsun_obs))
636                           {  
637             	         mean_ih                = DRMS_MISSING_FLOAT; 
638             	         total_us_ih            = DRMS_MISSING_FLOAT;
639               	         total_abs_ih           = DRMS_MISSING_FLOAT;
640                              mean_ih_err            = DRMS_MISSING_FLOAT;
641                              total_us_ih_err        = DRMS_MISSING_FLOAT;
642                              total_abs_ih_err       = DRMS_MISSING_FLOAT;
643             	      }
644             
645                           // Set sharp keys 
646                           drms_setkey_float(sharpoutrec, "MEANJZH",  mean_ih); 
647                           drms_setkey_float(sharpoutrec, "TOTUSJH",  total_us_ih); 	
648                           drms_setkey_float(sharpoutrec, "ABSNJZH",  total_abs_ih); 
649                           drms_setkey_float(sharpoutrec, "ERRMIH",  mean_ih_err); 	
650                           drms_setkey_float(sharpoutrec, "ERRTUI",  total_us_ih_err); 
651                           drms_setkey_float(sharpoutrec, "ERRTAI",  total_abs_ih_err); 	
652             
653 mbobra 1.1                // Set sharp cea keys 
654                           drms_setkey_float(sharpceaoutrec, "MEANJZH", mean_ih); 
655                           drms_setkey_float(sharpceaoutrec, "TOTUSJH", total_us_ih); 	
656                           drms_setkey_float(sharpceaoutrec, "ABSNJZH", total_abs_ih); 
657                           drms_setkey_float(sharpceaoutrec, "ERRMIH", mean_ih_err); 	
658                           drms_setkey_float(sharpceaoutrec, "ERRTUI", total_us_ih_err); 
659                           drms_setkey_float(sharpceaoutrec, "ERRTAI", total_abs_ih_err); 	
660                        }
661             
662                        /***** SAVNCPP, Example Function 12 (Requires Function 8) *******************/
663                
664                        if (savncppflag)
665                        {
666                           // First, compute Jz on all pixels
667               	      computeJz(bx_err, by_err, bx, by, dims, jz, jz_err, jz_err_squared, mask, bitmask, cdelt1, rsun_ref, rsun_obs, 
668 mbobra 1.11                         derx, dery, err_term1, err_term2);
669 mbobra 1.1  
670                           // Then compute sums of Jz on select pixels
671               	      if (computeSumAbsPerPolarity(jz_err, bz_err, bz, jz, dims, &totaljz, &totaljz_err, 
672             				           mask, bitmask, cdelt1, rsun_ref, rsun_obs))
673                           {
674             	         totaljz                = DRMS_MISSING_FLOAT;
675                              totaljz_err            = DRMS_MISSING_FLOAT;
676             	      }
677             
678                           // Set sharp keys 	
679                           drms_setkey_float(sharpoutrec, "SAVNCPP",  totaljz); 
680                           drms_setkey_float(sharpoutrec, "ERRJHT",  totaljz_err); 	
681             
682                           // Set sharp cea keys 
683                           drms_setkey_float(sharpceaoutrec, "SAVNCPP", totaljz); 
684                           drms_setkey_float(sharpceaoutrec, "ERRJHT", totaljz_err); 	
685                        }
686             
687                        /***** MEANSHR and SHRGT45, Example Function 14 (Requires Keiji's code) **********/
688             
689                        if (meanshrflag || shrgt45flag)
690 mbobra 1.1             {
691                           // First compute potential field
692             	      for (int i = 0; i < nxny; i++) bpz[i] = bz[i];
693             	      greenpot(bpx, bpy, bpz, nx, ny); 
694             
695                           // Then compute shear angles
696             	      if (computeShearAngle(bx_err, by_err, bh_err, bx, by, bz, bpx, bpy, bpz, dims, 
697             		   		    &meanshear_angle, &meanshear_angle_err, &area_w_shear_gt_45,  
698             				    mask, bitmask)) 
699                           {
700             	          meanshear_angle    = DRMS_MISSING_FLOAT; // If fail, fill in NaN
701             		  area_w_shear_gt_45 = DRMS_MISSING_FLOAT;
702                               meanshear_angle_err= DRMS_MISSING_FLOAT;
703             	      }
704             
705                           // Set sharp keys 	
706                           drms_setkey_float(sharpoutrec, "MEANSHR",  meanshear_angle); 
707                           drms_setkey_float(sharpoutrec, "SHRGT45",  area_w_shear_gt_45); 	
708                           drms_setkey_float(sharpoutrec, "ERRMSHA",  meanshear_angle_err); 	
709             
710                           // Set sharp cea keys 
711 mbobra 1.1                drms_setkey_float(sharpceaoutrec, "MEANSHR", meanshear_angle); 
712                           drms_setkey_float(sharpceaoutrec, "SHRGT45", area_w_shear_gt_45); 	
713                           drms_setkey_float(sharpceaoutrec, "ERRMSHA", meanshear_angle_err); 	  
714                        }
715             
716 mbobra 1.8            /***** TOTFX, TOTFY, TOTFX, TOTBSQ, EPSX, EPSY, EPSZ , Example Function 16 (Lorentz forces) **********/
717             
718                        if (totfxflag || totfyflag || totfzflag || totbsqflag || epsxflag || epsyflag || epszflag)
719                        {
720             
721                             // Compute Lorentz forces
722                     	if (computeLorentz(bx, by, bz, fx, fy, fz, dims, &totfx, &totfy, &totfz, &totbsq,
723                                 &epsx, &epsy, &epsz, mask, bitmask, cdelt1, rsun_ref, rsun_obs))
724                             {  
725             		   totfx             = DRMS_MISSING_FLOAT;
726                                totfy             = DRMS_MISSING_FLOAT;
727             		   totfz             = DRMS_MISSING_FLOAT;
728             		   totbsq            = DRMS_MISSING_FLOAT;
729                                epsx              = DRMS_MISSING_FLOAT;
730             		   epsy              = DRMS_MISSING_FLOAT;
731                                epsz              = DRMS_MISSING_FLOAT;
732                      	}
733             
734                           // Set sharp keys 	
735                           drms_setkey_float(sharpoutrec, "TOTFX", totfx); 
736                           drms_setkey_float(sharpoutrec, "TOTFY", totfy); 	
737 mbobra 1.8                drms_setkey_float(sharpoutrec, "TOTFZ", totfz); 	
738                           drms_setkey_float(sharpoutrec, "TOTBSQ",totbsq); 
739                           drms_setkey_float(sharpoutrec, "EPSX",  epsx); 	
740                           drms_setkey_float(sharpoutrec, "EPSY",  epsy); 	
741                           drms_setkey_float(sharpoutrec, "EPSZ",  epsz); 	
742             
743                           // Set sharp cea keys 
744                           drms_setkey_float(sharpceaoutrec, "TOTFX", totfx); 
745                           drms_setkey_float(sharpceaoutrec, "TOTFY", totfy); 	
746                           drms_setkey_float(sharpceaoutrec, "TOTFZ", totfz); 	
747                           drms_setkey_float(sharpceaoutrec, "TOTBSQ",totbsq); 
748                           drms_setkey_float(sharpceaoutrec, "EPSX",  epsx); 	
749                           drms_setkey_float(sharpceaoutrec, "EPSY",  epsy); 	
750                           drms_setkey_float(sharpceaoutrec, "EPSZ",  epsz); 	
751                           
752                        }
753             
754 mbobra 1.1             /******************************* END FLAGS **********************************/
755             
756 mbobra 1.10 
757 mbobra 1.1  	   drms_free_array(bitmaskArray);		
758             	   drms_free_array(maskArray);
759             	   drms_free_array(bxArray);           
760             	   drms_free_array(byArray);
761             	   drms_free_array(bzArray);
762             	   drms_free_array(bx_errArray);           
763             	   drms_free_array(by_errArray);
764             	   drms_free_array(bz_errArray);
765 mbobra 1.6  	   drms_free_array(losArray);
766 mbobra 1.10 
767 mbobra 1.11         // free the arrays that are related to the lorentz calculation
768                     free(fx); free(fy); free(fz);
769 mbobra 1.1          //used for select calculations
770             	free(bh); free(bt); free(jz); 
771             	free(bpx); free(bpy); free(bpz); 
772             	free(derx); free(dery);	 
773             	free(derx_bt); free(dery_bt); 	
774             	free(derx_bz); free(dery_bz);	 
775             	free(derx_bh); free(dery_bh); 
776             	free(bt_err); free(bh_err);  free(jz_err); 
777                     free(jz_err_squared); free(jz_rms_err);
778             	free(cvsinfoall);
779 mbobra 1.7  
780 mbobra 1.6          free(rim);
781                     free(p1p0);
782                     free(p1n0);
783                     free(p1p);
784                     free(p1n);
785                     free(p1);
786                     free(pmap);
787 mbobra 1.7          free(p1pad);
788                     free(pmapn);
789 mbobra 1.11         free(err_term2);
790                     free(err_term1);
791             
792             
793             	} //endfor
794             
795              	drms_close_records(sharpinrecset, DRMS_FREE_RECORD);
796             	drms_close_records(sharpceainrecset, DRMS_FREE_RECORD);
797             
798             	drms_close_records(sharpoutrecset, DRMS_INSERT_RECORD);
799             	drms_close_records(sharpceaoutrecset, DRMS_INSERT_RECORD);
800 mbobra 1.6  
801 mbobra 1.9  	printf("Success=%d\n",sameflag);
802 arta   1.3  return 0;    
803 mbobra 1.1  }	// DoIt

Karen Tian
Powered by
ViewCVS 0.9.4