forked from Unidata/netcdf-c
-
Notifications
You must be signed in to change notification settings - Fork 0
/
util.c
1459 lines (1347 loc) · 46.5 KB
/
util.c
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
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
395
396
397
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
750
751
752
753
754
755
756
757
758
759
760
761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
776
777
778
779
780
781
782
783
784
785
786
787
788
789
790
791
792
793
794
795
796
797
798
799
800
801
802
803
804
805
806
807
808
809
810
811
812
813
814
815
816
817
818
819
820
821
822
823
824
825
826
827
828
829
830
831
832
833
834
835
836
837
838
839
840
841
842
843
844
845
846
847
848
849
850
851
852
853
854
855
856
857
858
859
860
861
862
863
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
880
881
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
900
901
902
903
904
905
906
907
908
909
910
911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
963
964
965
966
967
968
969
970
971
972
973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
992
993
994
995
996
997
998
999
1000
/*********************************************************************
* Copyright 1996, UCAR/Unidata
* See netcdf/COPYRIGHT file for copying and redistribution conditions.
* $Id: util.c 2792 2014-10-27 06:02:59Z wkliao $
*********************************************************************/
#include <math.h> /* floor() */
#include "tests.h"
void
print_nok(int nok)
{
if (verbose || nfails > 0)
print("\n");
print(" %d good comparisons. ", nok);
}
/* Is value within external type range? */
int
inRange(const double value, const nc_type xtype)
{
switch (xtype) {
case NC_CHAR: return value >= X_CHAR_MIN && value <= X_CHAR_MAX;
case NC_BYTE: return value >= X_BYTE_MIN && value <= X_BYTE_MAX;
case NC_SHORT: return value >= X_SHORT_MIN && value <= X_SHORT_MAX;
case NC_INT: return value >= X_INT_MIN && value <= X_INT_MAX;
case NC_FLOAT: return value >= X_FLOAT_MIN && value <= X_FLOAT_MAX;
case NC_DOUBLE: return value >= X_DOUBLE_MIN && value <= X_DOUBLE_MAX;
case NC_UBYTE: return value >= 0 && value <= X_UCHAR_MAX;
case NC_USHORT: return value >= 0 && value <= X_USHORT_MAX;
case NC_UINT: return value >= 0 && value <= X_UINT_MAX;
case NC_INT64: return value >= X_INT64_MIN && value <= X_INT64_MAX;
case NC_UINT64: return value >= 0 && value <= X_UINT64_MAX;
default: assert(0);
return(0);
}
}
static int
inRange_uchar(const int cdf_format,
const double value,
const nc_type xtype)
{
/* check value of type xtype if within uchar range */
if (cdf_format < NC_FORMAT_CDF5 && xtype == NC_BYTE) {
/* netCDF specification make a special case for type conversion between
* uchar and scahr: do not check for range error. See
* http://www.unidata.ucar.edu/software/netcdf/docs/data_type.html#type_conversion
*/
return(value >= 0 && value <= 255);
/* this is to ensure value is within the range of uchar */
}
/* else */
return inRange(value, xtype);
}
static int
inRange_float(const double value, const nc_type xtype)
{
double min, max;
switch (xtype) {
case NC_CHAR: min = X_CHAR_MIN; max = X_CHAR_MAX; break;
case NC_BYTE: min = X_BYTE_MIN; max = X_BYTE_MAX; break;
case NC_SHORT: min = X_SHORT_MIN; max = X_SHORT_MAX; break;
case NC_INT: min = X_INT_MIN; max = X_INT_MAX; break;
case NC_FLOAT:
if(FLT_MAX < X_FLOAT_MAX) {
min = (-FLT_MAX);
max = FLT_MAX;
} else {
min = X_FLOAT_MIN;
max = X_FLOAT_MAX;
}
break;
case NC_DOUBLE:
if(FLT_MAX < X_DOUBLE_MAX) {
min = (-FLT_MAX);
max = FLT_MAX;
} else {
min = X_DOUBLE_MIN;
max = X_DOUBLE_MAX;
}
break;
case NC_UBYTE: min = 0; max = X_UCHAR_MAX; break;
case NC_USHORT: min = 0; max = X_USHORT_MAX; break;
case NC_UINT: min = 0; max = X_UINT_MAX; break;
case NC_INT64: min = X_INT64_MIN; max = X_INT64_MAX; break;
case NC_UINT64: min = 0; max = X_UINT64_MAX; break;
default: assert(0);
}
if(!( value >= min && value <= max)) {
#if 0 /* DEBUG */
if(xtype == NC_FLOAT) {
fprintf(stderr, "\n");
fprintf(stderr, "min % .17e\n", min);
fprintf(stderr, "value % .17e\n", value);
fprintf(stderr, "max % .17e\n", max);
}
#endif
return 0;
}
#if FLT_MANT_DIG != DBL_MANT_DIG
/* else */
{
const float fvalue = (float)value;
return fvalue >= min && fvalue <= max;
}
#else
return 1;
#endif
}
/* wrapper for inRange to handle special NC_BYTE/uchar adjustment */
/* this function checks whether "value" to be casted to type "itype" is
* within the range of external "xtype".
*/
int
inRange3(const int cdf_format,
const double value,
const nc_type xtype,
const nct_itype itype)
{
/* netCDF specification make a special case for type conversion between
* uchar and NC_BYTE: do not check for range error. See
* http://www.unidata.ucar.edu/software/netcdf/docs/data_type.html#type_conversion
* The _uchar and _schar functions were introduced in netCDF-3 to eliminate
* an ambiguity, and support both signed and unsigned byte data. In
* netCDF-2, whether the external NC_BYTE type represented signed or
* unsigned values was left up to the user. In netcdf-3, we treat NC_BYTE
* as signed for the purposes of conversion to short, int, long, float, or
* double. (Of course, no conversion takes place when the internal type is
* signed char.) In the _uchar functions, we treat NC_BYTE as if it were
* unsigned. Thus, no NC_ERANGE error can occur converting between NC_BYTE
* and unsigned char.
*/
switch (itype) {
case NCT_UCHAR:
return inRange_uchar(cdf_format, value, xtype);
case NCT_FLOAT:
return inRange_float(value, xtype);
default:
break;
}
return inRange(value, xtype);
}
/*
* Does x == y, where one is internal and other external (netCDF)?
* Use tolerant comparison based on IEEE FLT_EPSILON or DBL_EPSILON.
*/
int
equal(const double x,
const double y,
nc_type xtype, /* external data type */
nct_itype itype)
{
const double flt_epsilon = 1.19209290E-07;
const double dbl_epsilon = 2.2204460492503131E-16;
double epsilon;
epsilon = xtype == NC_FLOAT ||
itype == NCT_FLOAT ? flt_epsilon : dbl_epsilon;
if (xtype == NC_CHAR && itype == NCT_TEXT) {
/* because in-memory data type char can be signed or unsigned,
* type cast the value from external NC_CHAR before the comparison
*/
char x2 = (char) x;
char y2 = (char) y;
return ABS(x2-y2) <= epsilon * MAX( ABS(x2), ABS(y2));
}
return ABS(x-y) <= epsilon * MAX( ABS(x), ABS(y));
}
/* this function is for the APIs without itype, i.e. xtype == itype */
int
equal2(const double x,
const double y,
nc_type xtype) /* external data type */
{
const double flt_epsilon = 1.19209290E-07;
const double dbl_epsilon = 2.2204460492503131E-16;
double epsilon;
epsilon = xtype == NC_FLOAT ? flt_epsilon : dbl_epsilon;
if (xtype == NC_CHAR) {
/* because in-memory data type char can be signed or unsigned,
* type cast the value from external NC_CHAR before the comparison
*/
char x2 = (char) x;
char y2 = (char) y;
return ABS(x2-y2) <= epsilon * MAX( ABS(x2), ABS(y2));
}
return ABS(x-y) <= epsilon * MAX( ABS(x), ABS(y));
}
/* Test whether two int vectors are equal. If so return 1, else 0 */
int
int_vec_eq(const int *v1, const int *v2, const int n)
{
int i;
for (i= 0; i < n && v1[i] == v2[i]; i++)
;
return i == n;
}
/*
* Generate random integer from 0 to n-1
* Like throwing an n-sided dice marked 0, 1, 2, ..., n-1
*/
size_t roll( size_t n )
{
size_t r;
do
/*
* Compute a pseudo-random value between 0.0 and 1.0, multiply
* it by n-1, and then find the nearest integer.
*
* We don't use RAND_MAX here because not all compilation
* environments define it (e.g. gcc(1) under SunOS 4.1.4).
*/
r = (size_t)(((rand() % 32768) / 32767.0) * (n - 1) + 0.5);
while (r >= n);
return r;
}
/*
* Convert number to mixed base
*
* E.g. to convert 41 inches to yards, feet and inches:
* size_t base[] = {1, 3, 12};
* size_t result[3];
* status = toMixedBase(41, 3, base, result);
*
* Author: Harvey Davies, Unidata/UCAR, Boulder, Colorado
*/
int
toMixedBase(
size_t number, /* number to be converted to mixed base */
int length,
const size_t base[], /* dimensioned [length], base[0] ignored */
size_t result[]) /* dimensioned [length] */
{
int i;
if (length > 0) {
for (i = length - 1; i > 0; i--) {
if (base[i] == 0) return 1;
result[i] = number % base[i];
number = number / base[i];
}
result[0] = number;
}
return 0;
}
/*
* Convert number from mixed base
*
* E.g. to convert 1 yard, 0 feet, 5 inches to inches:
* size_t number[] = {1, 0, 5};
* size_t base[] = {1, 3, 12};
* inches = fromMixedBase(3, number, base);
*
* Author: Harvey Davies, Unidata/UCAR, Boulder, Colorado
*/
size_t
fromMixedBase(int length,
size_t number[], /* dimensioned [length] */
size_t base[]) /* dimensioned [length], base[0] ignored */
{
size_t i;
size_t result = 0;
for (i = 1; i < length; i++) {
result += number[i-1];
result *= base[i];
}
if (length > 0)
result += number[i-1];
return result;
}
/* Convert any nc_type to double */
int nc2dbl ( const nc_type xtype, const void *p, double *result)
{
if ( ! p ) return 2;
if ( ! result ) return 3;
switch (xtype) {
case NC_CHAR: *result = *((char *) p); break;
case NC_BYTE: *result = *((signed char *) p); break;
case NC_UBYTE: *result = *((unsigned char *) p); break;
case NC_SHORT: *result = *((short *) p); break;
case NC_USHORT: *result = *((unsigned short *) p); break;
case NC_INT:
#if INT_MAX >= X_INT_MAX
*result = *((int *) p); break;
#else
*result = *((long *) p); break;
#endif
case NC_UINT:
#if UINT_MAX >= X_UINT_MAX
*result = *((unsigned int *) p); break;
#else
*result = *((unsigned long *) p); break;
#endif
case NC_FLOAT: *result = *((float *) p); break;
case NC_DOUBLE: *result = *((double *) p); break;
case NC_INT64: *result = *((long long *) p); break;
case NC_UINT64: *result = *((unsigned long long *) p); break;
default: return 1;
}
return 0;
}
/* Convert double to any nc_type */
int dbl2nc ( const double d, const nc_type xtype, void *p)
{
double r; /* rounded value */
if (p == NULL) return 1;
switch (xtype) {
case NC_CHAR:
r = floor(0.5+d);
/* d is obtained from hash() which may be set to X_CHAR_MIN (0)
* or X_CHAR_MAX (255). When in-memory data type char is signed
* (i.e. ranged from -128 to 127), we should still allow a type
* cast a unsigned value > 127 to a signed char without
* reporting it as a range error.
*/
if ( r < X_CHAR_MIN || r > X_CHAR_MAX ) return 2;
#if defined(__CHAR_UNSIGNED__) && __CHAR_UNSIGNED__ != 0
*((signed char*) p) = (signed char)r;
#else
*((char *) p) = (char)r;
#endif
break;
case NC_BYTE:
r = floor(0.5+d);
if ( r < schar_min || r > schar_max ) return 2;
*((signed char *) p) = (signed char)r;
break;
case NC_UBYTE:
r = floor(0.5+d);
if ( r < 0.0 || r > uchar_max ) return 2;
*((unsigned char *) p) = (unsigned char)r;
break;
case NC_SHORT:
r = floor(0.5+d);
if ( r < short_min || r > short_max ) return 2;
*((short *) p) = (short)r;
break;
case NC_USHORT:
r = floor(0.5+d);
if ( r < 0.0 || r > ushort_max ) return 2;
*((unsigned short *) p) = (unsigned short)r;
break;
case NC_INT:
r = floor(0.5+d);
if ( r < long_min || r > long_max ) return 2;
#if INT_MAX >= X_INT_MAX
*((int *) p) = (int)r;
#else
*((long *) p) = (long)r;
#endif
break;
case NC_UINT:
r = floor(0.5+d);
if ( r < 0.0 || r > uint_max ) return 2;
#if UINT_MAX >= X_UINT_MAX
*((unsigned int *) p) = (unsigned int)r;
#else
*((unsigned long *) p) = (unsigned long)r;
#endif
break;
case NC_FLOAT:
if ( fabs(d) > float_max ) return 2;
*((float *) p) = (float)d;
break;
case NC_DOUBLE:
*((double *) p) = (double)d;
break;
case NC_INT64:
r = floor(0.5+d);
if ( r < int64_min || r > int64_max ) return 2;
*((long long *) p) = (long long)r;
break;
case NC_UINT64:
r = floor(0.5+d);
if ( r < 0.0 || r > uint64_max ) return 2;
*((unsigned long long *) p) = (unsigned long long)r;
break;
default:
return 1;
}
return 0;
}
#define FUZZ (1.19209290E-07)
#ifdef USE_EXTREME_NUMBERS
/* Generate data values as function of type, rank (-1 for attribute), index */
double
hash( const nc_type xtype, const int rank, const size_t *index )
{
double base;
double result;
int d; /* index of dimension */
/* If vector then elements 0 & 1 are min & max. Elements 2 & 3 are */
/* just < min & > max (except for NC_CHAR & NC_DOUBLE) */
if (abs(rank) == 1 && index[0] <= 3) {
switch (index[0]) {
case 0:
switch (xtype) {
case NC_CHAR: return X_CHAR_MIN;
case NC_BYTE: return X_BYTE_MIN;
case NC_SHORT: return X_SHORT_MIN;
case NC_INT: return X_INT_MIN;
case NC_FLOAT: return X_FLOAT_MIN;
case NC_DOUBLE: return X_DOUBLE_MIN;
case NC_UBYTE: return 0;
case NC_USHORT: return 0;
case NC_UINT: return 0;
case NC_INT64: return X_INT_MIN - 128.0; /* slight smaller
than INT_MIN */
case NC_UINT64: return 0;
default: assert(0);
}
case 1:
switch (xtype) {
case NC_CHAR: return X_CHAR_MAX;
case NC_BYTE: return X_BYTE_MAX;
case NC_SHORT: return X_SHORT_MAX;
case NC_INT: return X_INT_MAX;
case NC_FLOAT: return X_FLOAT_MAX;
case NC_DOUBLE: return X_DOUBLE_MAX;
case NC_UBYTE: return X_UCHAR_MAX;
case NC_USHORT: return X_USHORT_MAX;
case NC_UINT: return X_UINT_MAX;
case NC_INT64: return X_INT_MAX + 128.0;
/* slightly bigger than INT_MAX */
case NC_UINT64: return X_UINT_MAX + 128.0;
/* slightly bigger than UINT_MAX */
default: assert(0);
}
case 2:
switch (xtype) {
case NC_CHAR: return 'A';
case NC_BYTE: return X_BYTE_MIN-1.0;
case NC_SHORT: return X_SHORT_MIN-1.0;
case NC_INT: return X_INT_MIN-1.0;
case NC_FLOAT: return X_FLOAT_MIN * (1.0 + FUZZ);
case NC_DOUBLE: return -1.0;
case NC_UBYTE: return -1.0;
case NC_USHORT: return -1.0;
case NC_UINT: return -1.0;
case NC_INT64: return -1.0; /* skip test */
case NC_UINT64: return -1.0;
default: assert(0);
}
case 3:
switch (xtype) {
case NC_CHAR: return 'Z';
case NC_BYTE: return X_BYTE_MAX+1.0;
case NC_SHORT: return X_SHORT_MAX+1.0;
case NC_INT: return X_INT_MAX+1.0;
case NC_FLOAT: return X_FLOAT_MAX * (1.0 + FUZZ);
case NC_DOUBLE: return 1.0;
case NC_UBYTE: return X_UCHAR_MAX +1.0;
case NC_USHORT: return X_USHORT_MAX+1.0;
case NC_UINT: return X_UINT_MAX +1.0;
case NC_INT64: return 1.0; /* skip test */
case NC_UINT64: return 1.0; /* skip test */
default: assert(0);
}
}
} else {
switch (xtype) {
case NC_CHAR: base = 2; break;
case NC_BYTE: base = -2; break;
case NC_SHORT: base = -5; break;
case NC_INT: base = -20; break;
case NC_FLOAT: base = -9; break;
case NC_DOUBLE: base = -10; break;
/* not sure what right values are */
case NC_UBYTE: base = 2; break;
case NC_USHORT: base = 5; break;
case NC_UINT: base = 20; break;
case NC_INT64: base = -20; break;
case NC_UINT64: base = 20; break;
default: assert(0);
}
result = rank < 0 ? base * 7 : base * (rank + 1);
for (d = 0; d < abs(rank); d++)
result = base * (result + index[d]);
}
return result;
}
#else /* USE_EXTREME_NUMBERS */
#define SANE_SHORT 3333
#define SANE_INT 2222
#define SANE_FLOAT 300.0
#define SANE_DOUBLE 1000.0
/* Generate data values as function of type, rank (-1 for attribute), index */
double
hash( const nc_type xtype, const int rank, const size_t *index )
{
double base;
double result;
int d; /* index of dimension */
/* If vector then elements 0 & 1 are min & max. Elements 2 & 3 are */
/* just < min & > max (except for NC_CHAR & NC_DOUBLE) */
if (abs(rank) == 1 && index[0] <= 3) {
switch (index[0]) {
case 0:
switch (xtype) {
case NC_CHAR: return X_CHAR_MIN;
case NC_BYTE: return X_BYTE_MIN;
case NC_SHORT: return SANE_SHORT;
case NC_INT: return SANE_INT;
case NC_FLOAT: return SANE_FLOAT;
case NC_DOUBLE: return SANE_DOUBLE;
case NC_UBYTE: return 0;
case NC_USHORT: return 0;
case NC_UINT: return 0;
case NC_INT64: return X_INT_MIN - 128.0; /* slight smaller
than INT_MIN */
case NC_UINT64: return 0;
default: assert(0);
}
case 1:
switch (xtype) {
case NC_CHAR: return X_CHAR_MAX;
case NC_BYTE: return X_BYTE_MAX;
case NC_SHORT: return SANE_SHORT;
case NC_INT: return SANE_INT;
case NC_FLOAT: return SANE_FLOAT;
case NC_DOUBLE: return SANE_DOUBLE;
case NC_UBYTE: return X_UCHAR_MAX;
case NC_USHORT: return X_USHORT_MAX;
case NC_UINT: return X_UINT_MAX;
case NC_INT64: return X_INT_MAX + 128.0;
/* slightly bigger than INT_MAX */
case NC_UINT64: return X_UINT_MAX + 128.0;
/* slightly bigger than UINT_MAX */
default: assert(0);
}
case 2:
switch (xtype) {
case NC_CHAR: return 'A';
case NC_BYTE: return X_BYTE_MIN-1.0;
case NC_SHORT: return SANE_SHORT-1.0;
case NC_INT: return SANE_INT-1.0;
case NC_FLOAT: return SANE_FLOAT * (1.0 + FUZZ);
case NC_DOUBLE: return -1.0;
case NC_UBYTE: return -1.0;
case NC_USHORT: return -1.0;
case NC_UINT: return -1.0;
case NC_INT64: return -1.0; /* skip test */
case NC_UINT64: return -1.0;
default: assert(0);
}
case 3:
switch (xtype) {
case NC_CHAR: return 'Z';
case NC_BYTE: return X_BYTE_MAX+1.0;
case NC_SHORT: return SANE_SHORT+1.0;
case NC_INT: return SANE_INT+1.0;
case NC_FLOAT: return SANE_FLOAT * (1.0 + FUZZ);
case NC_DOUBLE: return 1.0;
case NC_UBYTE: return X_UCHAR_MAX +1.0;
case NC_USHORT: return X_USHORT_MAX+1.0;
case NC_UINT: return X_UINT_MAX +1.0;
case NC_INT64: return 1.0; /* skip test */
case NC_UINT64: return 1.0; /* skip test */
default: assert(0);
}
}
} else {
switch (xtype) {
case NC_CHAR: base = 2; break;
case NC_BYTE: base = -2; break;
case NC_SHORT: base = -5; break;
case NC_INT: base = -20; break;
case NC_FLOAT: base = -9; break;
case NC_DOUBLE: base = -10; break;
/* not sure what right values are */
case NC_UBYTE: base = 2; break;
case NC_USHORT: base = 5; break;
case NC_UINT: base = 20; break;
case NC_INT64: base = -20; break;
case NC_UINT64: base = 20; break;
default: assert(0);
}
result = rank < 0 ? base * 7 : base * (rank + 1);
for (d = 0; d < abs(rank); d++)
result = base * (result + index[d]);
}
return result;
}
#endif
/* wrapper for hash to handle special NC_BYTE/uchar adjustment */
double
hash4(const int cdf_format,
const nc_type xtype,
const int rank,
const size_t *index,
const nct_itype itype)
{
double result;
result = hash( xtype, rank, index );
/* netCDF specification make a special case for type conversion between
* uchar and NC_BYTE: do not check for range error. See
* http://www.unidata.ucar.edu/software/netcdf/docs/data_type.html#type_conversion
* The _uchar and _schar functions were introduced in netCDF-3 to eliminate
* an ambiguity, and support both signed and unsigned byte data. In
* netCDF-2, whether the external NC_BYTE type represented signed or
* unsigned values was left up to the user. In netcdf-3, we treat NC_BYTE
* as signed for the purposes of conversion to short, int, long, float, or
* double. (Of course, no conversion takes place when the internal type is
* signed char.) In the _uchar functions, we treat NC_BYTE as if it were
* unsigned. Thus, no NC_ERANGE error can occur converting between NC_BYTE
* and unsigned char.
*/
if (cdf_format < NC_FORMAT_CDF5 &&
itype == NCT_UCHAR && xtype == NC_BYTE &&
result >= -128 && result < 0)
result += 256;
return result;
}
static nc_type
char2type(char letter) {
switch (letter) {
case 'c': return NC_CHAR;
case 'b': return NC_BYTE;
case 's': return NC_SHORT;
case 'i': return NC_INT;
case 'f': return NC_FLOAT;
case 'd': return NC_DOUBLE;
case 'y': return NC_UBYTE;
case 't': return NC_USHORT;
case 'u': return NC_UINT;
case 'x': return NC_INT64;
case 'z': return NC_UINT64;
default: assert(0);
}
return NC_CHAR; /* Just to keep compiler happy */
}
static void
init_dims(const char *digit)
{
int dimid; /* index of dimension */
for (dimid = 0; dimid < NDIMS; dimid++)
{
dim_len[dimid] = dimid == 0 ? NRECS : dimid;
dim_name[dimid][0] = 'D';
dim_name[dimid][1] = digit[dimid];
dim_name[dimid][2] = '\0';
}
}
static void
init_gatts(const char *type_letter)
{
int attid;
for (attid = 0; attid < numGatts; attid++)
{
gatt_name[attid][0] = 'G';
gatt_name[attid][1] = type_letter[attid];
gatt_name[attid][2] = '\0';
gatt_len[attid] = 1 + attid;
gatt_type[attid] = char2type (type_letter[attid]);
}
}
static size_t
product(int nn, const size_t *sp)
{
size_t result = 1;
while(nn-- > 0)
result *= *sp++;
return result;
}
/*
define global variables:
dim_name, dim_len,
var_name, var_type, var_rank, var_shape, var_natts, var_dimid, var_nels
att_name, gatt_name, att_type, gatt_type, att_len, gatt_len
*/
void
init_gvars (void)
{
const size_t max_dim_len[MAX_RANK] = {
MAX_DIM_LEN +1,
MAX_DIM_LEN,
MAX_DIM_LEN
};
const char type_letter[] = "cbsifdytuxz";
/* c:char, b:byte, s:short, i:int, f:float, d:double, y:ubyte, t:ushort,
* u:uint, x:int64, z:uint64
*/
const char digit[] = "r123456789";
int rank;
int vn; /* var number */
int xtype; /* index of type */
int an; /* attribute number */
assert(sizeof(max_dim_len)/sizeof(max_dim_len[0]) >= MAX_RANK);
init_dims(digit);
for (vn=0; vn<numVars; vn++)
memset(var_name[vn], 0, 2+MAX_RANK);
for (rank = 0, vn = 0, xtype = 0, an = 0; rank <= MAX_RANK; rank++)
{
/* number variables of a type and rank */
const size_t nvars = product(rank, max_dim_len);
size_t jj;
for (jj = 0; jj < nvars; jj++)
{
/* number types of this shape */
const int ntypes = rank < 2 ? numTypes : 1;
int tc;
for (tc = 0; tc < ntypes;
tc++, vn++, xtype = (xtype + 1) % numTypes)
{
size_t tmp[MAX_RANK];
var_name[vn][0] = type_letter[xtype];
var_type[vn] = char2type (type_letter[xtype]);
var_rank[vn] = rank;
var_natts[vn] = rank == 0 ? vn % (MAX_NATTS + 1) : 0;
{
int ac;
for (ac = 0; ac < var_natts[vn]; ac++, an++)
{
att_name[vn][ac][0] = type_letter[an % numTypes];
att_name[vn][ac][1] = '\0';
att_len[vn][ac] = an;
att_type[vn][ac] = char2type (type_letter[an % numTypes]);
}
} /* ac block */
#ifndef NDEBUG
assert(toMixedBase (jj, rank, max_dim_len, tmp) == 0);
#else
(void) toMixedBase (jj, rank, max_dim_len, tmp);
#endif
{
int dn; /* dimension number */
for (dn = 0; dn < rank; dn++)
var_dimid[vn][dn] = (int)tmp[dn];
for (dn = 0, var_nels[vn] = 1; dn < rank; dn++)
{
var_dimid[vn][dn] += dn > 0;
assert (var_dimid[vn][dn] <= 9);
var_name[vn][dn + 1] = digit[var_dimid[vn][dn]];
var_shape[vn][dn] = var_dimid[vn][dn] ?
var_dimid[vn][dn] : NRECS;
var_nels[vn] *= var_shape[vn][dn];
}
} /* dn block */
}
}
}
init_gatts(type_letter);
}
/* define dims defined by global variables */
void
def_dims(int ncid)
{
int err; /* status */
int i;
int dimid; /* dimension id */
for (i = 0; i < NDIMS; i++) {
err = nc_def_dim(ncid, dim_name[i], i==0 ? NC_UNLIMITED : dim_len[i],
&dimid);
IF (err) error("nc_def_dim: %s", nc_strerror(err));
}
}
/* define vars defined by global variables */
void
def_vars(int ncid)
{
int err; /* status */
int i;
int var_id;
for (i = 0; i < numVars; i++) {
err = nc_def_var(ncid, var_name[i], var_type[i], var_rank[i],
var_dimid[i], &var_id);
IF (err) error("nc_def_var: %s", nc_strerror(err));
}
}
/* put attributes defined by global variables */
void
put_atts(int ncid)
{
int err; /* status */
int i;
size_t k;
int j; /* index of attribute */
int allInRange;
double att[MAX_NELS];
char catt[MAX_NELS];
for (i = -1; i < numVars; i++) {
for (j = 0; j < NATTS(i); j++) {
if (ATT_TYPE(i,j) == NC_CHAR) {
for (k = 0; k < ATT_LEN(i,j); k++) {
catt[k] = (char) hash(ATT_TYPE(i,j), -1, &k);
}
err = nc_put_att_text(ncid, i, ATT_NAME(i,j),
ATT_LEN(i,j), catt);
IF (err)
error("nc_put_att_text: %s", nc_strerror(err));
} else {
for (allInRange = 1, k = 0; k < ATT_LEN(i,j); k++) {
att[k] = hash(ATT_TYPE(i,j), -1, &k);
allInRange = allInRange && inRange(att[k], ATT_TYPE(i,j));
}
err = nc_put_att_double(ncid, i, ATT_NAME(i,j),
ATT_TYPE(i,j), ATT_LEN(i,j), att);
if (allInRange) {
IF (err)
error("nc_put_att_double: %s", nc_strerror(err));
} else {
IF (err != NC_ERANGE)
error("type-conversion range error: status = %d", err);
}
}
}
}
}
/* put variables defined by global variables */
void
put_vars(int ncid)
{
size_t start[MAX_RANK];
size_t index[MAX_RANK];
int err; /* status */
int i;
size_t j;
double value[MAX_NELS];
char text[MAX_NELS];
int allInRange;
for (j = 0; j < MAX_RANK; j++)
start[j] = 0;
for (i = 0; i < numVars; i++) {
for (allInRange = 1, j = 0; j < var_nels[i]; j++) {
err = toMixedBase(j, var_rank[i], var_shape[i], index);
IF (err) error("toMixedBase");
if (var_name[i][0] == 'c') {
text[j] = (char) hash(var_type[i], var_rank[i], index);
} else {
value[j] = hash(var_type[i], var_rank[i], index);
allInRange = allInRange && inRange(value[j], var_type[i]);
}
}
if (var_name[i][0] == 'c') {
err = nc_put_vara_text(ncid, i, start, var_shape[i], text);
IF (err)
error("nc_put_vara_text: %s", nc_strerror(err));
} else {
err = nc_put_vara_double(ncid, i, start, var_shape[i], value);
if (allInRange) {
IF (err)
error("nc_put_vara_double: %s", nc_strerror(err));
} else {
IF (err != NC_ERANGE)
error("type-conversion range error: status = %d", err);
}
}
}
}
/* Create & write all of specified file using global variables */
void
write_file(char *filename)
{
int ncid; /* netCDF id */
int err; /* status */
err = file_create(filename, NC_CLOBBER, &ncid);
IF (err)
error("nc_create: %s", nc_strerror(err));
def_dims(ncid);
def_vars(ncid);
put_atts(ncid);
err = nc_enddef(ncid);
IF (err)
error("nc_enddef: %s", nc_strerror(err));
#ifdef USE_PNETCDF
{ int i,format;
nc_inq_format_extended(ncid, &format, NULL);
if (format == NC_FORMATX_PNETCDF) {
for (i = 0; i < numVars; i++) {
err = nc_var_par_access(ncid, i, NC_COLLECTIVE);
IF (err) error("nc_var_par_access: %s", nc_strerror(err));
}
}}
#endif
put_vars(ncid);
err = nc_close (ncid);
IF (err)
error("nc_close: %s", nc_strerror(err));
}
/*
* check dimensions of specified file have expected name & length
*/
void
check_dims(int ncid)
{
char name[NC_MAX_NAME];
size_t length;
int i;
int err; /* status */
for (i = 0; i < NDIMS; i++) {
err = nc_inq_dim(ncid, i, name, &length);
IF (err)
error("nc_inq_dim: %s", nc_strerror(err));
IF (strcmp(name, dim_name[i]) != 0)
error("Unexpected name of dimension %d: '%s', expected: '%s'", i, name, dim_name[i]);
IF (length != dim_len[i])
error("Unexpected length %d of dimension %d, expected %zu", length, i, dim_len[i]);
}
}
/*
* check variables of specified file have expected name, type, shape & values
*/
void
check_vars(int ncid)
{
size_t index[MAX_RANK];
char text, name[NC_MAX_NAME];
int i, err; /* status */
size_t j;
int nok = 0; /* count of valid comparisons */
int isChar, ndims, dimids[MAX_RANK];
double value, expect;
nc_type xtype;
size_t length;
for (i = 0; i < numVars; i++) {
isChar = var_type[i] == NC_CHAR;
err = nc_inq_var(ncid, i, name, &xtype, &ndims, dimids, NULL);
IF (err)
error("nc_inq_var: %s", nc_strerror(err));
IF (strcmp(name, var_name[i]) != 0)
error("Unexpected var_name");
IF (xtype != var_type[i])
error("Unexpected type");
IF (ndims != var_rank[i])