File: printfqn.c

package info (click to toggle)
netcdf-parallel 1%3A4.9.0-1
  • links: PTS, VCS
  • area: main
  • in suites: bookworm
  • size: 113,164 kB
  • sloc: ansic: 267,893; sh: 12,869; cpp: 5,822; yacc: 2,613; makefile: 1,813; lex: 1,216; xml: 173; awk: 2
file content (206 lines) | stat: -rw-r--r-- 4,927 bytes parent folder | download | duplicates (4)
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
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <assert.h>

#ifdef HAVE_GETOPT_H
#include <getopt.h>
#endif

#if defined(_WIN32) && ! defined(__MINGW32__)
#include "XGetopt.h"
#endif

#include <netcdf.h>

#define CHECK(err) {if(err) report(err,__LINE__);}

/* Command line options */
struct Options {
    int debug;
    enum What {NONE=0, DIM=1, TYPE=2} what;
    char file[4096];
    char var[NC_MAX_NAME+1];
    char object[NC_MAX_NAME+1];
} options;

/**************************************************/


static void
report(int err, int lineno)
{
    fprintf(stderr,"Error: %d: %s\n", lineno, nc_strerror(err));
    exit(1);
}

void
usage(void)
{
    fprintf(stderr,"usage: printfqn [-D] [-V] -t|-d -v <varname> -f <filename> \n");
    exit(0);
}

int
get_id_parent(int ncid, int id, int* parentp, enum What what)
{
    int stat = NC_NOERR;
    int i;
    int nids;
    int ids[4096];

    /* Does this group have the id we are searching for? */
    if(what == TYPE) {
        if((stat=nc_inq_typeids(ncid,&nids,ids))) goto done;
    } else if(what == DIM) {
        if((stat=nc_inq_dimids(ncid,&nids,ids,0))) goto done;
    } else
        abort();
    assert(nids < 4096);

    /* Search for this id */
    for(i=0;i<nids;i++) {
	if(ids[i] == id) {
	    if(parentp) *parentp = ncid;
	    goto done;
	}
    }

    /* Else Search subgroups. */
    if((stat=nc_inq_grps(ncid,&nids,ids))) goto done;
    assert(nids < 4096);

    /* Recurse on each subgroup */
    for(i=0;i<nids;i++) {
	switch (stat = get_id_parent(ids[i],id,parentp,what)) {
	case NC_ENOTFOUND: break; /* not found; keep looking */
	case NC_NOERR: goto done; /* found it */
	default: goto done; /* some other error */
	}
    }
    stat = NC_ENOTFOUND; /* Not found */

done:
    return stat;

}

int
get_variable_info(int ncid, const char* name, int* gidp, int* vidp, int* tidp, int* ndimsp, int* dimids)
{
    int stat = NC_NOERR;
    int i;
    int nids;
    int ids[4096];
    char varname[NC_MAX_NAME];

    /* Assume only one occurrence of the variable in dataset */

    /* Does this group have the variable we are searching for? */
    if((stat=nc_inq_varids(ncid,&nids,ids))) goto done;
    assert(nids < 4096);

    /* Search for this variable name */
    for(i=0;i<nids;i++) {
	if((stat = nc_inq_varname(ncid,ids[i],varname))) goto done;
	if(strcmp(name,varname)==0) {
	    if(gidp) *gidp = ncid;
	    if(vidp) *vidp = ids[i];
	    if((stat = nc_inq_vartype(ncid,ids[i],tidp))) goto done;
    	    if((stat = nc_inq_varndims(ncid,ids[i],ndimsp))) goto done;
       	    if((stat = nc_inq_vardimid(ncid,ids[i],dimids))) goto done;
	    goto done;
	}
    }

    /* Else Search subgroups. */
    if((stat=nc_inq_grps(ncid,&nids,ids))) goto done;
    assert(nids < 4096);

    /* Recurse on each subgroup */
    for(i=0;i<nids;i++) {
	switch (stat = get_variable_info(ids[i],name,gidp,vidp,tidp,ndimsp,dimids)) {
	case NC_ENOTVAR: break; /* not found; keep looking */
	case NC_NOERR: goto done; /* found it */
	default: goto done; /* some other error */
	}
    }
    stat = NC_ENOTVAR; /* Not found */

done:
    return stat;
}

int
main(int argc, char** argv)
{
    int ncid, varid, gid, tid;
    size_t fqnlen, namelen;
    char fqn[4096];
    char name[NC_MAX_NAME];
    int ndims;
    int dimids[NC_MAX_VAR_DIMS];
    int c;

    memset((void*)&options,0,sizeof(options));

    while ((c = getopt(argc, argv, "DVdtv:f:")) != EOF) {
	switch(c) {
	case 'D': 
	    options.debug = 1;	    
	    break;
	case 'V': 
	    usage();
	    break;
	case 'd':
	    options.what = DIM;
	    break;
	case 't':
	    options.what = TYPE;
	    break;
	case 'v':
	    strcpy(options.var,optarg);
	    break;
	case 'f':
	    strcpy(options.file,optarg);
	    break;
	case ':':
	   fprintf(stderr,"option has no argument: %c\n",c);
	    usage();
	case '?':
	   fprintf(stderr,"unknown option\n");
	    usage();
	}
    }

    CHECK(nc_open(options.file,NC_NETCDF4,&ncid));

    /* Locate the parent group for the variable */
    CHECK(get_variable_info(ncid,options.var,&gid,&varid,&tid,&ndims,dimids));

    if(options.what == TYPE) {
        /* Get the simple type name from the variable */
        CHECK(nc_inq_type(ncid,tid,name,&namelen));
	/* Get the containing group id for the type (might not be same as ncid) */
        CHECK(get_id_parent(ncid,tid,&gid,TYPE));
    } else if(options.what == DIM) {
	/* get name of dimids[0] from the variable */
        CHECK(nc_inq_dimname(ncid,dimids[0],name));
        /* Get the containing group id for the type */
        CHECK(get_id_parent(ncid,dimids[0],&gid,DIM));
    }

    /* Get the FQN name for the containing group of the id */
    CHECK(nc_inq_grpname_full(gid,&fqnlen,fqn));
    assert(fqnlen < sizeof(fqn));

    if(strcmp(fqn,"/")==0) fqn[0] = '\0';

    /* report result  with no newline */
    printf("%s/%s",fqn,name);

    return 0;
}