a9a5625b0b761a4e8c890fb9f309a68723ca0296
[yaz-moved-to-github.git] / src / facet.c
1 /* This file is part of the YAZ toolkit.
2  * Copyright (C) 1995-2011 Index Data
3  * See the file LICENSE for details.
4  */
5
6 /** 
7  * \file facet.c
8  * \brief Facet utilities
9  */
10
11 #if HAVE_CONFIG_H
12 #include <config.h>
13 #endif
14
15 #include <yaz/facet.h>
16 #include <yaz/diagbib1.h>
17 #include <yaz/oid_db.h>
18 #include <yaz/oid_std.h>
19 #include <yaz/otherinfo.h>
20 #include <assert.h>
21
22 void yaz_oi_set_facetlist(
23     Z_OtherInformation **otherInformation, ODR odr,
24     Z_FacetList *facet_list)
25 {
26     int categoryValue = 1;
27     Z_External *z_external = 0;
28     Z_OtherInformationUnit *oi =
29         yaz_oi_update(otherInformation, odr, yaz_oid_userinfo_facet_1,
30                       categoryValue, 0);
31     if (!oi)
32         return;
33     oi->which = Z_OtherInfo_externallyDefinedInfo;
34     z_external = odr_malloc(odr, sizeof(*z_external));
35     z_external->which = Z_External_userFacets;
36     z_external->direct_reference = odr_oiddup(odr, yaz_oid_userinfo_facet_1);
37     z_external->indirect_reference = 0;
38     z_external->descriptor = 0;
39     z_external->u.facetList = facet_list;
40     oi->information.externallyDefinedInfo = z_external;
41 }
42
43 Z_FacetList *yaz_oi_get_facetlist(Z_OtherInformation **otherInformation)
44 {
45     int categoryValue = 1;
46     Z_External *z_external = 0;
47     Z_OtherInformationUnit *oi =
48         yaz_oi_update(otherInformation, 0, yaz_oid_userinfo_facet_1,
49                       categoryValue, 0);
50     if (!oi)
51         return 0;
52     z_external = oi->information.externallyDefinedInfo;
53
54     if (z_external && z_external->which == Z_External_userFacets)
55         return z_external->u.facetList;
56     return 0;
57 }
58
59 /* Little helper to extract a string attribute */
60 /* Gets the first string, there is usually only one */
61 /* in case of errors, returns null */
62
63 void yaz_facet_attr_init(struct yaz_facet_attr *attr_values)
64 {
65     attr_values->errcode   = 0;
66     attr_values->errstring = 0;
67     attr_values->relation  = 0;
68     attr_values->useattr   = 0;
69     attr_values->useattrbuff[0] = 0;
70     attr_values->limit     = 0;
71 }
72
73 static const char *stringattr(Z_ComplexAttribute *c)
74 {
75     int i;
76      Z_StringOrNumeric *son;
77     for (i = 0; i < c->num_list; i++)
78     {
79         son = c->list[i];
80         if ( son->which == Z_StringOrNumeric_string)
81             return son->u.string;
82     }
83     return 0;
84 }
85
86 /* Use attribute, @attr1, can be numeric or string */
87 static void useattr(Z_AttributeElement *ae, struct yaz_facet_attr *av)
88 {
89     const char *s;
90     if (ae->which == Z_AttributeValue_complex)
91     {
92         s = stringattr(ae->value.complex);
93         if (s)
94         {
95             if (!av->useattr)
96                 av->useattr = s;
97             else
98             { /* already seen one, can't have duplicates */
99                 av->errcode = YAZ_BIB1_UNSUPP_ATTRIBUTE_COMBI;
100                 av->errstring = "multiple use attributes";
101             }
102         }
103         else
104         { /* complex that did not return a string */
105             av->errcode = YAZ_BIB1_UNSUPP_ATTRIBUTE_COMBI;
106             av->errstring = "non-string complex attribute";
107         }
108     }
109     else
110     { /* numeric - could translate 4 to 'title' etc */
111         sprintf(av->useattrbuff, ODR_INT_PRINTF, *ae->value.numeric);
112         av->useattr = av->useattrbuff;
113     }
114 } /* useattr */
115
116
117 /* TODO rename to sortorder attr */
118 static void relationattr(Z_AttributeElement *ae, struct yaz_facet_attr *av)
119 {
120     if (ae->which == Z_AttributeValue_numeric)
121     {
122         if (*ae->value.numeric == 0)
123             av->relation = "desc";
124         else if (*ae->value.numeric == 1)
125                 av->relation = "asc";
126             else
127         if (*ae->value.numeric == 3) {
128             av->relation = "unknown/unordered";
129         } else {
130             av->errcode = YAZ_BIB1_UNSUPP_RELATION_ATTRIBUTE;
131             sprintf(av->useattrbuff, ODR_INT_PRINTF,
132                         *ae-> attributeType);
133             av->errstring = av->useattrbuff;
134         }
135     }
136     else
137     {
138         av->errcode = YAZ_BIB1_UNSUPP_RELATION_ATTRIBUTE;
139         av->errstring = "non-numeric relation attribute";
140     }
141 } /* relationattr */
142
143 static void limitattr(Z_AttributeElement *ae, struct yaz_facet_attr *av)
144 {
145     if (ae->which == Z_AttributeValue_numeric)
146     {
147         av->limit = *ae->value.numeric;
148     }
149     else
150     {
151         av->errcode = YAZ_BIB1_UNSUPP_ATTRIBUTE;
152         av->errstring = "non-numeric limit attribute";
153     }
154 } /* relationattr */
155
156 /* Get the index to be searched from the attributes.
157    @attr 1
158      can be either "string", or some well-known value like
159      4 for title
160    Returns a null and sets errors in rr,
161    emtpy string if no attr found,
162    or the string itself - always a pointer to the Z-structs,
163    so no need to free that string!
164 */
165
166 void yaz_facet_attr_get_z_attributes(const Z_AttributeList *attributes,
167                                      struct yaz_facet_attr *av)
168 {
169     int i;
170     Z_AttributeElement *ae;
171     for (i = 0; i < attributes->num_attributes; i++)
172     {
173         ae = attributes->attributes[i];
174         /* ignoring the attributeSet here */
175         if (*ae->attributeType == 1)
176         { /* use attribute */
177             useattr(ae, av);
178         }
179         else if (*ae->attributeType == 2)
180         { /* sortorder */
181             relationattr(ae, av);
182         }
183         else if (*ae->attributeType == 3)
184         { /* limit */
185             limitattr(ae, av);
186         }
187         else
188         { /* unknown attribute */
189             av->errcode = YAZ_BIB1_UNSUPP_ATTRIBUTE_TYPE;
190             sprintf(av->useattrbuff, ODR_INT_PRINTF,
191                         *ae-> attributeType);
192             av->errstring = av->useattrbuff;
193             yaz_log(YLOG_WARN, "Unsupported attribute type %s", av->useattrbuff);
194             /* would like to give a better message, but the standard */
195             /* tells me to return the attribute type */
196         }
197         if (av->errcode)
198             return; /* no need to dig deeper, return on first error */
199     }
200     return;
201 } /* facetattrs */
202
203 Z_Term *term_create(ODR odr, const char *cstr)
204 {
205     Z_Term *term = odr_malloc(odr, sizeof(*term));
206     term->which = Z_Term_characterString;
207     term->u.characterString = odr_strdup(odr, cstr);
208     return term;
209 }
210
211 Z_FacetTerm* facet_term_create(ODR odr, Z_Term *term, int freq)
212 {
213     Z_FacetTerm *facet_term = odr_malloc(odr, sizeof(*facet_term));
214     facet_term->count = odr_malloc(odr, sizeof(*facet_term->count));
215     facet_term->term = term;
216     *facet_term->count = freq;
217     return facet_term;
218 }
219
220 Z_FacetField* facet_field_create(ODR odr, Z_AttributeList *attributes,
221                                  int num_terms)
222 {
223     Z_FacetField *facet_field = odr_malloc(odr, sizeof(*facet_field));
224     facet_field->attributes = attributes;
225     facet_field->num_terms = num_terms;
226     facet_field->terms = odr_malloc(odr, num_terms * sizeof(*facet_field->terms));
227     return facet_field;
228 }
229
230 void facet_field_term_set(ODR odr, Z_FacetField *field,
231                           Z_FacetTerm *facet_term, int index)
232 {
233     assert(0 <= index && index < field->num_terms);
234     field->terms[index] = facet_term;
235 }
236
237 Z_FacetList* facet_list_create(ODR odr, int num_facets)
238 {
239     Z_FacetList *facet_list = odr_malloc(odr, sizeof(*facet_list));
240     facet_list->num = num_facets;
241     facet_list->elements =
242         odr_malloc(odr, facet_list->num * sizeof(*facet_list->elements));
243     return facet_list;
244 }
245
246 void facet_list_field_set(ODR odr, Z_FacetList *list, Z_FacetField *field,
247                           int index)
248 {
249     assert(0 <= index && index < list->num);
250     list->elements[index] = field;
251 }
252
253 /*
254  * Local variables:
255  * c-basic-offset: 4
256  * c-file-style: "Stroustrup"
257  * indent-tabs-mode: nil
258  * End:
259  * vim: shiftwidth=4 tabstop=8 expandtab
260  */
261