fc0127ae35fb89c790570886b0626cab489ed2c4
[yaz-moved-to-github.git] / retrieval / d1_marc.c
1 /*
2  * Copyright (c) 1995-2002, Index Data.
3  * See the file LICENSE for details.
4  * Sebastian Hammer, Adam Dickmeiss
5  *
6  * $Id: d1_marc.c,v 1.20 2002-07-03 14:09:34 adam Exp $
7  */
8
9
10 #include <assert.h>
11 #include <stdlib.h>
12 #include <string.h>
13
14 #include <yaz/oid.h>
15 #include <yaz/log.h>
16 #include <yaz/marcdisp.h>
17 #include <yaz/readconf.h>
18 #include <yaz/xmalloc.h>
19 #include <yaz/data1.h>
20 #include <yaz/tpath.h>
21
22 data1_marctab *data1_read_marctab (data1_handle dh, const char *file)
23 {
24     FILE *f;
25     NMEM mem = data1_nmem_get (dh);
26     data1_marctab *res = (data1_marctab *)nmem_malloc(mem, sizeof(*res));
27     char line[512], *argv[50];
28     int lineno = 0;
29     int argc;
30     
31     if (!(f = data1_path_fopen(dh, file, "r")))
32     {
33         yaz_log(LOG_WARN|LOG_ERRNO, "%s", file);
34         return 0;
35     }
36
37     res->name = 0;
38     res->reference = VAL_NONE;
39     res->next = 0;
40     res->length_data_entry = 4;
41     res->length_starting = 5;
42     res->length_implementation = 0;
43     strcpy(res->future_use, "4");
44
45     strcpy(res->record_status, "n");
46     strcpy(res->implementation_codes, "    ");
47     res->indicator_length = 2;
48     res->identifier_length = 2;
49     res->force_indicator_length = -1;
50     res->force_identifier_length = -1;
51     strcpy(res->user_systems, "z  ");
52     
53     while ((argc = readconf_line(f, &lineno, line, 512, argv, 50)))
54         if (!strcmp(*argv, "name"))
55         {
56             if (argc != 2)
57             {
58                 yaz_log(LOG_WARN, "%s:%d:Missing arg for %s", file, lineno,
59                         *argv);
60                 continue;
61             }
62             res->name = nmem_strdup(mem, argv[1]);
63         }
64         else if (!strcmp(*argv, "reference"))
65         {
66             if (argc != 2)
67             {
68                 yaz_log(LOG_WARN, "%s:%d: Missing arg for %s", file, lineno,
69                         *argv);
70                 continue;
71             }
72             if ((res->reference = oid_getvalbyname(argv[1])) == VAL_NONE)
73             {
74                 yaz_log(LOG_WARN, "%s:%d: Unknown tagset reference '%s'",
75                         file, lineno, argv[1]);
76                 continue;
77             }
78         }
79         else if (!strcmp(*argv, "length-data-entry"))
80         {
81             if (argc != 2)
82             {
83                 yaz_log(LOG_WARN, "%s:%d: Missing arg for %s", file, lineno,
84                         *argv);
85                 continue;
86             }
87             res->length_data_entry = atoi(argv[1]);
88         }
89         else if (!strcmp(*argv, "length-starting"))
90         {
91             if (argc != 2)
92             {
93                 yaz_log(LOG_WARN, "%s:%d: Missing arg for %s", file, lineno,
94                         *argv);
95                 continue;
96             }
97             res->length_starting = atoi(argv[1]);
98         }
99         else if (!strcmp(*argv, "length-implementation"))
100         {
101             if (argc != 2)
102             {
103                 yaz_log(LOG_WARN, "%s:%d: Missing arg for %s", file, lineno,
104                         *argv);
105                 continue;
106             }
107             res->length_implementation = atoi(argv[1]);
108         }
109         else if (!strcmp(*argv, "future-use"))
110         {
111             if (argc != 2)
112             {
113                 yaz_log(LOG_WARN, "%s:%d: Missing arg for %s", file, lineno,
114                         *argv);
115                 continue;
116             }
117             strncpy(res->future_use, argv[1], 2);
118         }
119         else if (!strcmp(*argv, "force-indicator-length"))
120         {
121             if (argc != 2)
122             {
123                 yaz_log(LOG_WARN, "%s:%d: Missing arg for %s", file, lineno,
124                         *argv);
125                 continue;
126             }
127             res->force_indicator_length = atoi(argv[1]);
128         }
129         else if (!strcmp(*argv, "force-identifier-length"))
130         {
131             if (argc != 2)
132             {
133                 yaz_log(LOG_WARN, "%s:%d: Missing arg for %s", file, lineno,
134                         *argv);
135                 continue;
136             }
137             res->force_identifier_length = atoi(argv[1]);
138         }
139         else
140             yaz_log(LOG_WARN, "%s:%d: Unknown directive '%s'", file, lineno,
141                     *argv);
142
143     fclose(f);
144     return res;
145 }
146
147 /*
148  * Locate some data under this node. This routine should handle variants
149  * prettily.
150  */
151 static char *get_data(data1_node *n, int *len)
152 {
153     char *r;
154
155     while (n->which != DATA1N_data && n->child)
156         n = n->child;
157     if (n->which != DATA1N_data || 
158         (n->u.data.what != DATA1I_text && n->u.data.what != DATA1I_num))
159     {
160         r = "[Structured/included data]";
161         *len = strlen(r);
162         return r;
163     }
164
165     *len = n->u.data.len;
166     while (*len && d1_isspace(n->u.data.data[*len - 1]))
167         (*len)--;
168     return n->u.data.data;
169 }
170
171 static void memint (char *p, int val, int len)
172 {
173     char buf[10];
174
175     if (len == 1)
176         *p = val + '0';
177     else
178     {
179         sprintf (buf, "%08d", val);
180         memcpy (p, buf+8-len, len);
181     }
182 }
183
184 static int is_indicator (data1_marctab *p, data1_node *subf)
185 {
186 #if 1
187     if (p->indicator_length != 2 ||
188         (subf && subf->which == DATA1N_tag && strlen(subf->u.tag.tag) == 2))
189         return 1;
190 #else
191     if (subf && subf->which == DATA1N_tag && subf->child->which == DATA1N_tag)
192         return 1;
193 #endif
194     return 0;
195 }
196
197 static int nodetomarc(data1_marctab *p, data1_node *n, int selected,
198     char **buf, int *size)
199 {
200     int len = 26;
201     int dlen;
202     int base_address = 25;
203     int entry_p, data_p;
204     char *op;
205     data1_node *field, *subf;
206
207     yaz_log (LOG_DEBUG, "nodetomarc");
208
209     for (field = n->child; field; field = field->next)
210     {
211         int is00X = 0;
212
213         if (field->which != DATA1N_tag)
214         {
215             yaz_log(LOG_WARN, "Malformed field composition for marc output.");
216             return -1;
217         }
218         if (selected && !field->u.tag.node_selected)
219             continue;
220         len += 4 + p->length_data_entry + p->length_starting
221             + p->length_implementation;
222         base_address += 3 + p->length_data_entry + p->length_starting
223             + p->length_implementation;
224         subf = field->child;
225
226         if (subf->which == DATA1N_data)
227             is00X = 1;
228         
229         if (!is00X)
230             len += p->indicator_length;  
231         /*  we'll allow no indicator if length is not 2 */
232         if (is_indicator (p, subf))
233             subf = subf->child;
234
235         for (; subf; subf = subf->next)
236         {
237             if (!is00X)
238                 len += p->identifier_length;
239             get_data(subf, &dlen);
240             len += dlen;
241         }
242     }
243
244     if (!*buf)
245         *buf = (char *)xmalloc(*size = len);
246     else if (*size <= len)
247         *buf = (char *)xrealloc(*buf, *size = len);
248         
249     op = *buf;
250     memint (op, len, 5);
251     memcpy (op+5, p->record_status, 1);
252     memcpy (op+6, p->implementation_codes, 4);
253     memint (op+10, p->indicator_length, 1);
254     memint (op+11, p->identifier_length, 1);
255     memint (op+12, base_address, 5);
256     memcpy (op+17, p->user_systems, 3);
257     memint (op+20, p->length_data_entry, 1);
258     memint (op+21, p->length_starting, 1);
259     memint (op+22, p->length_implementation, 1);
260     memcpy (op+23, p->future_use, 1);
261     
262     entry_p = 24;
263     data_p = base_address;
264
265     for (field = n->child; field; field = field->next)
266     {
267         int is00X = 0;
268
269         int data_0 = data_p;
270         char *indicator_data = "    ";
271         if (selected && !field->u.tag.node_selected)
272             continue;
273
274         subf = field->child;
275
276         if (subf->which == DATA1N_data)
277             is00X = 1;
278
279         if (is_indicator (p, subf))
280         {
281             indicator_data = subf->u.tag.tag;
282             subf = subf->child;
283         }
284         if (!is00X)
285         {
286             memcpy (op + data_p, indicator_data, p->indicator_length);
287             data_p += p->indicator_length;
288         }
289         for (; subf; subf = subf->next)
290         {
291             char *data;
292
293             if (!is00X)
294             {
295                 const char *identifier = "a";
296                 if (subf->which != DATA1N_tag)
297                     yaz_log(LOG_WARN, "Malformed fields for marc output.");
298                 else
299                     identifier = subf->u.tag.tag;
300                 op[data_p] = ISO2709_IDFS;
301                 memcpy (op + data_p+1, identifier, p->identifier_length-1);
302                 data_p += p->identifier_length;
303             }
304             data = get_data(subf, &dlen);
305             memcpy (op + data_p, data, dlen);
306             data_p += dlen;
307         }
308         op[data_p++] = ISO2709_FS;
309
310         memcpy (op + entry_p, field->u.tag.tag, 3);
311         entry_p += 3;
312         memint (op + entry_p, data_p - data_0, p->length_data_entry);
313         entry_p += p->length_data_entry;
314         memint (op + entry_p, data_0 - base_address, p->length_starting);
315         entry_p += p->length_starting;
316         entry_p += p->length_implementation;
317     }
318     op[entry_p++] = ISO2709_FS;
319     assert (entry_p == base_address);
320     op[data_p++] = ISO2709_RS;
321     assert (data_p == len);
322     return len;
323 }
324
325 char *data1_nodetomarc(data1_handle dh, data1_marctab *p, data1_node *n,
326                        int selected, int *len)
327 {
328     int *size;
329     char **buf = data1_get_map_buf (dh, &size);
330
331     n = data1_get_root_tag (dh, n);
332     if (!n)
333         return 0;
334     *len = nodetomarc(p, n, selected, buf, size);
335     return *buf;
336 }