proj/cookbook/smpl_07.c

/*
 *  smpl_07.c                                           $DRMS/proj/cookbook/
 *
 *  Populates a data series consisting of variable size images with random
 *    data from a selected distribution function
 *  Illustrates features of the DRMS_Array and DRMS_Segment structs, and
 *    writing to DRMS record segments
 *
 *  Usage:
 *    smpl_07 [ds= cols= rows= ]
 *
 *  Revision history is at end of file.
 */

char *module_name = "CookbookRecipe:07";
char *version_id = "1.0";

#include <jsoc_main.h>

ModuleArgs_t module_args[] = {
  {ARG_STRING,  "ds",   "drms.images", "name of data series"},
  {ARG_INT,     "cols", "0",  "number of columns in image (default (rows)"},
  {ARG_INT,     "rows", "0",  "number of rows in image (default: cols)"},
  {ARG_NUME,    "dist", "uniform", "distribution for pseudo-randoms",
      "uniform, dblexp, normal"},
  {ARG_INT,     "seed", "0",  "initial seed for pseudo-random number generator"},
  {ARG_FLAG, "v", "", "verbose flag"}, 
  {ARG_END}
};

enum dist_type {UNIFORM, DBL_EXP, NORMAL};

float next_rand (int distrib) {
  double val0, val1, rmod, gf;
  static double sav;
  static int avail = 0;

  switch (distrib) {
    case (UNIFORM):
      return 2 * drand48 () - 1.0;
    case (DBL_EXP):
      val0 = -log (drand48 ());
      val1 = (drand48 () < 0.5) ? -0.5 : +0.5;
      return val0 * val1;
    case (NORMAL):
      if (avail) {
        avail = 0;
        return sav;
      }
      rmod = 2.0;
      while (rmod >= 1.0) {
        val0 = 2 * drand48 () - 1.0;
        val1 = 2 * drand48 () - 1.0;
        rmod = val0 * val0 + val1 * val1;
      }
      gf = sqrt (-2.0 * log (rmod) / rmod);
      sav = gf * val0;
      avail = 1;
      return gf * val1;
    default:
      return 0.0;
  }
}

void rand_fill_array (float *data, int cols, int rows, int df) {
  double xr;
  int n, col, row, pr;
  int colsM1 = cols - 1;
  double kf1 = 1.0, kf2 = 0.5, kf3 = 1.0 / 3.0, kf4 = 0.25, kf5 = 0.2;

  n = row = 0;
  data[n++] = next_rand (df);
  for (col = 1; col < colsM1; col++, n++) {
    data[n] = kf1 * data[n-1] + next_rand (df);
  }
  data[n++] = kf2 * (data[n-1] + data[n+1-cols]) + next_rand (df);

  row++;
  for (; row < rows; row++) {
    pr = n - cols;
    data[n++] = kf3 * (data[n-1] + data[pr] + data[pr+1]) + next_rand (df);
    pr++;
    for (col = 1; col < colsM1; col++, n++, pr++) {
      data[n] = kf4 * (data[pr-1] + data[pr] + data[pr+1] + data[n-1]) +
          next_rand (df);
    }
    data[n++] = kf5 * (data[pr-1] + data[pr] + data[pr+1] + data[n-1] + data[n+1-cols]) +
        next_rand (df);
  }
}

int DoIt (void) {
  CmdParams_t *params = &cmdparams;
  DRMS_Record_t *record;
  DRMS_Segment_t *record_segment;
  DRMS_Array_t *data_array;
  float *data;
  int axes[2];
  int status = 0;

  char *series = strdup (params_get_str (params, "ds"));
  int cols = params_get_int (params, "cols");
  int rows = params_get_int (params, "rows");
  int distrib = params_get_int (params, "dist");
  long int seed = params_get_int (params, "seed");
  int verbose = params_isflagset (params, "v");
        /*  make sure that at least one of cols and rows has been specified  */
  if (cols <= 0) cols = rows;
  if (rows <= 0) rows = cols;
  if (rows <= 0) {
    fprintf (stderr, "Error: at least one of cols and rows must be specified\n");
    return 1;
  }
                                           /*  create the output data array  */
  axes[0] = cols;
  axes[1] = rows;
  data = (float *)malloc (cols * rows * sizeof (float));
  data_array = drms_array_create (DRMS_TYPE_FLOAT, 2, axes, (void *)data,
      &status);
                               /*  fill the output array (with random data)  */
  srand48 (seed);
  rand_fill_array (data, cols, rows, distrib);

  if (verbose) {
    int col, row, n;
    int mincol, maxcol, minrow, maxrow;
    float minval, maxval;

    minval = maxval = data[0];
    mincol = minrow = maxcol = maxrow = 0;
    for (n = 0, row = 0; row < rows; row++) {
      for (col = 0; col < cols; col++, n++) {
        if (data[n] < minval) {
          minval = data[n];
          mincol = col;
          minrow = row;
        }
        if (data[n] > maxval) {
          maxval = data[n];
          maxcol = col;
          maxrow = row;
        }
      }
    }
    printf ("min @ [%d, %d] : %f\n", mincol, minrow, minval);
    printf ("max @ [%d, %d] : %f\n", maxcol, maxrow, maxval);
  }

  record = drms_create_record (drms_env, series, DRMS_PERMANENT, &status);
  if (!record) {
    fprintf (stderr, "Error: Unable to create record in series %s\n", series);
    fprintf (stderr, "       Does series exist and is it writeable by you?\n");
    return 1;
  }
  record_segment = drms_segment_lookupnum (record, 0);
                        /*  set the output scaling parameters for the array  */
  data_array->bscale = 1.0;
  data_array->bzero = 0.0;
  status = drms_segment_write (record_segment, data_array, 0);
  if (status) {
    fprintf (stderr, "Error writing data to segment 0 of series %s\n", series);
    fprintf (stderr, "      series may not have appropriate structure\n");
    drms_free_array (data_array);
    drms_close_record (record, DRMS_FREE_RECORD);
    return 1;
  }
  drms_close_record (record, DRMS_INSERT_RECORD);
  return 0;
}
/*
 *  Revision History
 *
 *  09.08.02    file created by R Bogart
 */
Revision:	1.4
Committed:	Fri Nov 4 00:32:19 2011 UTC (11 years, 10 months ago) by rick
Content type:	text/plain
Branch:	MAIN
CVS Tags:	NetDRMS_Ver_6-4, NetDRMS_Ver_8-0, NetDRMS_Ver_6-2, NetDRMS_Ver_6-3, NetDRMS_Ver_6-0, NetDRMS_Ver_6-1, Ver_6-0, Ver_6-1, Ver_6-2, Ver_6-3, Ver_6-4, NetDRMS_Ver_8-8, Ver_8-5, NetDRMS_Ver_7-0, NetDRMS_Ver_8-1, Ver_7-0, Ver_LATEST, NetDRMS_Ver_LATEST, NetDRMS_Ver_8-12, NetDRMS_Ver_8-10, NetDRMS_Ver_8-11, NetDRMS_Ver_9-1, NetDRMS_Ver_9-0, NetDRMS_Ver_9-3, NetDRMS_Ver_9-2, NetDRMS_Ver_9-5, NetDRMS_Ver_9-4, NetDRMS_Ver_8-2, NetDRMS_Ver_8-3, NetDRMS_Ver_9-41, Ver_9-41, Ver_DRMSLATEST, NetDRMS_Ver_8-4, NetDRMS_Ver_8-5, NetDRMS_Ver_8-6, Ver_8-8, NetDRMS_Ver_8-7, Ver_8-2, Ver_9-3, Ver_8-0, Ver_8-1, Ver_8-6, Ver_8-7, Ver_8-4, Ver_8-11, Ver_7-1, Ver_9-1, Ver_8-3, NetDRMS_Ver_7-1, Ver_9-5, Ver_9-4, Ver_8-10, Ver_9-2, Ver_8-12, Ver_9-0, HEAD
Changes since 1.3:	+1 -1 lines
Log Message:	strdup return of params_get_str()
#	Content
1	/*
2	* smpl_07.c $DRMS/proj/cookbook/
3	*
4	* Populates a data series consisting of variable size images with random
5	* data from a selected distribution function
6	* Illustrates features of the DRMS_Array and DRMS_Segment structs, and
7	* writing to DRMS record segments
8	*
9	* Usage:
10	* smpl_07 [ds= cols= rows= ]
11	*
12	* Revision history is at end of file.
13	*/
14
15	char *module_name = "CookbookRecipe:07";
16	char *version_id = "1.0";
17
18	#include <jsoc_main.h>
19
20	ModuleArgs_t module_args[] = {
21	{ARG_STRING, "ds", "drms.images", "name of data series"},
22	{ARG_INT, "cols", "0", "number of columns in image (default (rows)"},
23	{ARG_INT, "rows", "0", "number of rows in image (default: cols)"},
24	{ARG_NUME, "dist", "uniform", "distribution for pseudo-randoms",
25	"uniform, dblexp, normal"},
26	{ARG_INT, "seed", "0", "initial seed for pseudo-random number generator"},
27	{ARG_FLAG, "v", "", "verbose flag"},
28	{ARG_END}
29	};
30
31	enum dist_type {UNIFORM, DBL_EXP, NORMAL};
32
33	float next_rand (int distrib) {
34	double val0, val1, rmod, gf;
35	static double sav;
36	static int avail = 0;
37
38	switch (distrib) {
39	case (UNIFORM):
40	return 2 * drand48 () - 1.0;
41	case (DBL_EXP):
42	val0 = -log (drand48 ());
43	val1 = (drand48 () < 0.5) ? -0.5 : +0.5;
44	return val0 * val1;
45	case (NORMAL):
46	if (avail) {
47	avail = 0;
48	return sav;
49	}
50	rmod = 2.0;
51	while (rmod >= 1.0) {
52	val0 = 2 * drand48 () - 1.0;
53	val1 = 2 * drand48 () - 1.0;
54	rmod = val0 * val0 + val1 * val1;
55	}
56	gf = sqrt (-2.0 * log (rmod) / rmod);
57	sav = gf * val0;
58	avail = 1;
59	return gf * val1;
60	default:
61	return 0.0;
62	}
63	}
64
65	void rand_fill_array (float *data, int cols, int rows, int df) {
66	double xr;
67	int n, col, row, pr;
68	int colsM1 = cols - 1;
69	double kf1 = 1.0, kf2 = 0.5, kf3 = 1.0 / 3.0, kf4 = 0.25, kf5 = 0.2;
70
71	n = row = 0;
72	data[n++] = next_rand (df);
73	for (col = 1; col < colsM1; col++, n++) {
74	data[n] = kf1 * data[n-1] + next_rand (df);
75	}
76	data[n++] = kf2 * (data[n-1] + data[n+1-cols]) + next_rand (df);
77
78	row++;
79	for (; row < rows; row++) {
80	pr = n - cols;
81	data[n++] = kf3 * (data[n-1] + data[pr] + data[pr+1]) + next_rand (df);
82	pr++;
83	for (col = 1; col < colsM1; col++, n++, pr++) {
84	data[n] = kf4 * (data[pr-1] + data[pr] + data[pr+1] + data[n-1]) +
85	next_rand (df);
86	}
87	data[n++] = kf5 * (data[pr-1] + data[pr] + data[pr+1] + data[n-1] + data[n+1-cols]) +
88	next_rand (df);
89	}
90	}
91
92	int DoIt (void) {
93	CmdParams_t *params = &cmdparams;
94	DRMS_Record_t *record;
95	DRMS_Segment_t *record_segment;
96	DRMS_Array_t *data_array;
97	float *data;
98	int axes[2];
99	int status = 0;
100
101	char *series = strdup (params_get_str (params, "ds"));
102	int cols = params_get_int (params, "cols");
103	int rows = params_get_int (params, "rows");
104	int distrib = params_get_int (params, "dist");
105	long int seed = params_get_int (params, "seed");
106	int verbose = params_isflagset (params, "v");
107	/* make sure that at least one of cols and rows has been specified */
108	if (cols <= 0) cols = rows;
109	if (rows <= 0) rows = cols;
110	if (rows <= 0) {
111	fprintf (stderr, "Error: at least one of cols and rows must be specified\n");
112	return 1;
113	}
114	/* create the output data array */
115	axes[0] = cols;
116	axes[1] = rows;
117	data = (float )malloc (cols rows * sizeof (float));
118	data_array = drms_array_create (DRMS_TYPE_FLOAT, 2, axes, (void *)data,
119	&status);
120	/* fill the output array (with random data) */
121	srand48 (seed);
122	rand_fill_array (data, cols, rows, distrib);
123
124	if (verbose) {
125	int col, row, n;
126	int mincol, maxcol, minrow, maxrow;
127	float minval, maxval;
128
129	minval = maxval = data[0];
130	mincol = minrow = maxcol = maxrow = 0;
131	for (n = 0, row = 0; row < rows; row++) {
132	for (col = 0; col < cols; col++, n++) {
133	if (data[n] < minval) {
134	minval = data[n];
135	mincol = col;
136	minrow = row;
137	}
138	if (data[n] > maxval) {
139	maxval = data[n];
140	maxcol = col;
141	maxrow = row;
142	}
143	}
144	}
145	printf ("min @ [%d, %d] : %f\n", mincol, minrow, minval);
146	printf ("max @ [%d, %d] : %f\n", maxcol, maxrow, maxval);
147	}
148
149	record = drms_create_record (drms_env, series, DRMS_PERMANENT, &status);
150	if (!record) {
151	fprintf (stderr, "Error: Unable to create record in series %s\n", series);
152	fprintf (stderr, " Does series exist and is it writeable by you?\n");
153	return 1;
154	}
155	record_segment = drms_segment_lookupnum (record, 0);
156	/* set the output scaling parameters for the array */
157	data_array->bscale = 1.0;
158	data_array->bzero = 0.0;
159	status = drms_segment_write (record_segment, data_array, 0);
160	if (status) {
161	fprintf (stderr, "Error writing data to segment 0 of series %s\n", series);
162	fprintf (stderr, " series may not have appropriate structure\n");
163	drms_free_array (data_array);
164	drms_close_record (record, DRMS_FREE_RECORD);
165	return 1;
166	}
167	drms_close_record (record, DRMS_INSERT_RECORD);
168	return 0;
169	}
170	/*
171	* Revision History
172	*
173	* 09.08.02 file created by R Bogart
174	*/