e410b7c73741077b04e0ee2bd6930fe04b5d02ad
[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.27 2002-08-28 19:34:50 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 /*
149  * Locate some data under this node. This routine should handle variants
150  * prettily.
151  */
152 static char *get_data(data1_node *n, int *len)
153 {
154     char *r;
155
156     while (n)
157     {
158         if (n->which == DATA1N_data)
159         {
160             int i;
161             *len = n->u.data.len;
162
163             for (i = 0; i<*len; i++)
164                 if (!d1_isspace(n->u.data.data[i]))
165                     break;
166             while (*len && d1_isspace(n->u.data.data[*len - 1]))
167                 (*len)--;
168             *len = *len - i;
169             if (*len > 0)
170                 return n->u.data.data + i;
171         }
172         if (n->which == DATA1N_tag)
173             n = n->child;
174         else if (n->which == DATA1N_data)
175             n = n->next;
176         else
177             break;      
178     }
179     r = "";
180     *len = strlen(r);
181     return r;
182 }
183
184 static void memint (char *p, int val, int len)
185 {
186     char buf[10];
187
188     if (len == 1)
189         *p = val + '0';
190     else
191     {
192         sprintf (buf, "%08d", val);
193         memcpy (p, buf+8-len, len);
194     }
195 }
196
197 static int is_indicator (data1_marctab *p, data1_node *subf)
198 {
199 #if 1
200     if (p->indicator_length != 2 ||
201         (subf && subf->which == DATA1N_tag && strlen(subf->u.tag.tag) == 2))
202         return 1;
203 #else
204     if (subf && subf->which == DATA1N_tag && subf->child->which == DATA1N_tag)
205         return 1;
206 #endif
207     return 0;
208 }
209
210 static int nodetomarc(data1_handle dh,
211                       data1_marctab *p, data1_node *n, int selected,
212                       char **buf, int *size)
213 {
214     int len = 26;
215     int dlen;
216     int base_address = 25;
217     int entry_p, data_p;
218     char *op;
219     data1_node *field, *subf;
220
221     yaz_log (LOG_DEBUG, "nodetomarc");
222
223     for (field = n->child; field; field = field->next)
224     {
225         int is00X = 0;
226
227         if (field->which != DATA1N_tag)
228         {
229             yaz_log(LOG_WARN, "Malformed field composition for marc output.");
230             return -1;
231         }
232         if (selected && !field->u.tag.node_selected)
233             continue;
234
235         subf = field->child;
236         if (!subf)
237             continue;
238
239         len += 4 + p->length_data_entry + p->length_starting
240             + p->length_implementation;
241         base_address += 3 + p->length_data_entry + p->length_starting
242             + p->length_implementation;
243
244         if (subf->which == DATA1N_data)
245             is00X = 1;
246         if (!data1_is_xmlmode(dh))
247         {
248             if (subf->which == DATA1N_tag && !strcmp(subf->u.tag.tag, "@"))
249                 is00X = 1;
250         }
251             
252         
253         if (!is00X)
254             len += p->indicator_length;  
255         /*  we'll allow no indicator if length is not 2 */
256         if (is_indicator (p, subf))
257             subf = subf->child;
258
259         for (; subf; subf = subf->next)
260         {
261             if (!is00X)
262                 len += p->identifier_length;
263             get_data(subf, &dlen);
264             len += dlen;
265         }
266     }
267
268     if (!*buf)
269         *buf = (char *)xmalloc(*size = len);
270     else if (*size <= len)
271         *buf = (char *)xrealloc(*buf, *size = len);
272         
273     op = *buf;
274     memint (op, len, 5);
275     memcpy (op+5, p->record_status, 1);
276     memcpy (op+6, p->implementation_codes, 4);
277     memint (op+10, p->indicator_length, 1);
278     memint (op+11, p->identifier_length, 1);
279     memint (op+12, base_address, 5);
280     memcpy (op+17, p->user_systems, 3);
281     memint (op+20, p->length_data_entry, 1);
282     memint (op+21, p->length_starting, 1);
283     memint (op+22, p->length_implementation, 1);
284     memcpy (op+23, p->future_use, 1);
285     
286     entry_p = 24;
287     data_p = base_address;
288
289     for (field = n->child; field; field = field->next)
290     {
291         int is00X = 0;
292
293         int data_0 = data_p;
294         char *indicator_data = "    ";
295         if (selected && !field->u.tag.node_selected)
296             continue;
297
298         subf = field->child;
299         if (!subf)
300             continue;
301
302         if (subf->which == DATA1N_data)
303             is00X = 1;
304         if (!data1_is_xmlmode(dh))
305         {
306             if (subf->which == DATA1N_tag && !strcmp(subf->u.tag.tag, "@"))
307                 is00X = 1;
308         }
309
310         if (is_indicator (p, subf))
311         {
312             indicator_data = subf->u.tag.tag;
313             subf = subf->child;
314         }
315         if (!is00X)
316         {
317             memcpy (op + data_p, indicator_data, p->indicator_length);
318             data_p += p->indicator_length;
319         }
320         for (; subf; subf = subf->next)
321         {
322             char *data;
323
324             if (!is00X)
325             {
326                 const char *identifier = "a";
327                 if (subf->which != DATA1N_tag)
328                     yaz_log(LOG_WARN, "Malformed fields for marc output.");
329                 else
330                     identifier = subf->u.tag.tag;
331                 op[data_p] = ISO2709_IDFS;
332                 memcpy (op + data_p+1, identifier, p->identifier_length-1);
333                 data_p += p->identifier_length;
334             }
335             data = get_data(subf, &dlen);
336             memcpy (op + data_p, data, dlen);
337             data_p += dlen;
338         }
339         op[data_p++] = ISO2709_FS;
340
341         memcpy (op + entry_p, field->u.tag.tag, 3);
342         entry_p += 3;
343         memint (op + entry_p, data_p - data_0, p->length_data_entry);
344         entry_p += p->length_data_entry;
345         memint (op + entry_p, data_0 - base_address, p->length_starting);
346         entry_p += p->length_starting;
347         entry_p += p->length_implementation;
348     }
349     op[entry_p++] = ISO2709_FS;
350     assert (entry_p == base_address);
351     op[data_p++] = ISO2709_RS;
352     assert (data_p == len);
353     return len;
354 }
355
356 char *data1_nodetomarc(data1_handle dh, data1_marctab *p, data1_node *n,
357                        int selected, int *len)
358 {
359     int *size;
360     char **buf = data1_get_map_buf (dh, &size);
361
362     n = data1_get_root_tag (dh, n);
363     if (!n)
364         return 0;
365     *len = nodetomarc(dh, p, n, selected, buf, size);
366     return *buf;
367 }