X-Git-Url: http://git.indexdata.com/?p=yaz-moved-to-github.git;a=blobdiff_plain;f=src%2Fsortspec.c;h=7f272f5f9b32f7aab05291127e413e7870bd8c46;hp=76e0f2fa6f549d65aba63adf35a327aff7887632;hb=b161284b288325f5eb5557152d4da21d18dc8ded;hpb=9b2f4955aee1fcee3702bcd832dbaf7fca7243ad diff --git a/src/sortspec.c b/src/sortspec.c index 76e0f2f..7f272f5 100644 --- a/src/sortspec.c +++ b/src/sortspec.c @@ -1,5 +1,5 @@ /* This file is part of the YAZ toolkit. - * Copyright (C) 1995-2011 Index Data + * Copyright (C) 1995-2012 Index Data * See the file LICENSE for details. */ /** @@ -13,6 +13,7 @@ #include #include #include +#include #include #include @@ -23,12 +24,12 @@ Z_SortKeySpecList *yaz_sort_spec(ODR out, const char *arg) { char sort_string_buf[64], sort_flags[64]; Z_SortKeySpecList *sksl = (Z_SortKeySpecList *) - odr_malloc (out, sizeof(*sksl)); + odr_malloc(out, sizeof(*sksl)); int off; - + sksl->num_specs = 0; sksl->specs = (Z_SortKeySpec **)odr_malloc(out, sizeof(sksl->specs) * 20); - + while ((sscanf(arg, "%63s %63s%n", sort_string_buf, sort_flags, &off)) == 2 && off > 1) { @@ -37,14 +38,14 @@ Z_SortKeySpecList *yaz_sort_spec(ODR out, const char *arg) char *sort_string = sort_string_buf; Z_SortKeySpec *sks = (Z_SortKeySpec *) odr_malloc(out, sizeof(*sks)); Z_SortKey *sk = (Z_SortKey *) odr_malloc(out, sizeof(*sk)); - + arg += off; sksl->specs[sksl->num_specs++] = sks; sks->sortElement = (Z_SortElement *) - odr_malloc (out, sizeof(*sks->sortElement)); + odr_malloc(out, sizeof(*sks->sortElement)); sks->sortElement->which = Z_SortElement_generic; sks->sortElement->u.generic = sk; - + if ((sort_string_sep = strchr(sort_string, '='))) { int i = 0; @@ -55,12 +56,12 @@ Z_SortKeySpecList *yaz_sort_spec(ODR out, const char *arg) sk->u.sortAttributes->list = (Z_AttributeList *) odr_malloc(out, sizeof(*sk->u.sortAttributes->list)); sk->u.sortAttributes->list->attributes = (Z_AttributeElement **) - odr_malloc(out, 10 * + odr_malloc(out, 10 * sizeof(*sk->u.sortAttributes->list->attributes)); while (i < 10 && sort_string && sort_string_sep) { Z_AttributeElement *el = (Z_AttributeElement *) - odr_malloc (out, sizeof(*el)); + odr_malloc(out, sizeof(*el)); sk->u.sortAttributes->list->attributes[i] = el; el->attributeSet = 0; el->attributeType = odr_intdup(out, atoi(sort_string)); @@ -87,7 +88,7 @@ Z_SortKeySpecList *yaz_sort_spec(ODR out, const char *arg) sks->which = Z_SortKeySpec_null; sks->u.null = odr_nullval (); - + for (i = 0; sort_flags[i]; i++) { switch (sort_flags[i]) @@ -123,7 +124,8 @@ Z_SortKeySpecList *yaz_sort_spec(ODR out, const char *arg) sks->u.missingValueData->size = sks->u.missingValueData->len; sks->u.missingValueData->buf = (unsigned char*) odr_strdup(out, sort_flags+i); - i += strlen(sort_flags+i); + i += strlen(sort_flags+i) - 1; + break; } } } @@ -181,6 +183,7 @@ int yaz_sort_spec_to_cql(Z_SortKeySpecList *sksl, WRBUF w) wrbuf_puts(w, "/missingValue="); wrbuf_write(w, (const char *) sks->u.missingValueData->buf, sks->u.missingValueData->len); + break; } } return 0; @@ -233,6 +236,210 @@ int yaz_sort_spec_to_type7(Z_SortKeySpecList *sksl, WRBUF pqf) return 0; } +int yaz_sort_spec_to_srw_sortkeys(Z_SortKeySpecList *sksl, WRBUF w) +{ + int i; + for (i = 0; i < sksl->num_specs; i++) + { + Z_SortKeySpec *sks = sksl->specs[i]; + Z_SortKey *sk; + + if (sks->sortElement->which != Z_SortElement_generic) + return -1; + + sk = sks->sortElement->u.generic; + + if (i) + wrbuf_puts(w, " "); + + if (sk->which == Z_SortKey_sortAttributes) + return -1; + else if (sk->which == Z_SortKey_sortField) + { + wrbuf_puts(w, sk->u.sortField); + } + wrbuf_puts(w, ",,"); /* path is absent */ + switch (*sks->sortRelation) + { + case Z_SortKeySpec_ascending: + wrbuf_puts(w, "1"); + break; + case Z_SortKeySpec_descending: + wrbuf_puts(w, "0"); + break; + } + wrbuf_puts(w, ","); + switch (*sks->caseSensitivity) + { + case Z_SortKeySpec_caseSensitive: + wrbuf_puts(w, "1"); + break; + case Z_SortKeySpec_caseInsensitive: + wrbuf_puts(w, "0"); + break; + } + wrbuf_puts(w, ","); + switch (sks->which) + { + case Z_SortKeySpec_null: + wrbuf_puts(w, "highValue"); + break; + case Z_SortKeySpec_abort: + wrbuf_puts(w, "abort"); + break; + case Z_SortKeySpec_missingValueData: + wrbuf_write(w, (const char *) sks->u.missingValueData->buf, + sks->u.missingValueData->len); + break; + } + } + return 0; +} + +int yaz_sort_spec_to_solr_sortkeys(Z_SortKeySpecList *sksl, WRBUF w) +{ + int i; + for (i = 0; i < sksl->num_specs; i++) + { + Z_SortKeySpec *sks = sksl->specs[i]; + Z_SortKey *sk; + + if (sks->sortElement->which != Z_SortElement_generic) + return -1; + + sk = sks->sortElement->u.generic; + + if (i) + wrbuf_puts(w, ","); + + if (sk->which == Z_SortKey_sortAttributes) + return -1; + else if (sk->which == Z_SortKey_sortField) + { + wrbuf_puts(w, sk->u.sortField); + } + switch (*sks->sortRelation) + { + case Z_SortKeySpec_ascending: + wrbuf_puts(w, " asc"); + break; + case Z_SortKeySpec_descending: + wrbuf_puts(w, " desc"); + break; + } + } + return 0; +} + + +int yaz_srw_sortkeys_to_sort_spec(const char *srw_sortkeys, WRBUF w) +{ + /* sru sortkey layout: path,schema,ascending,caseSensitive,missingValue */ + /* see cql_sortby_to_sortkeys of YAZ. */ + char **sortspec; + int num_sortspec = 0; + int i; + NMEM nmem = nmem_create(); + + if (srw_sortkeys) + nmem_strsplit_blank(nmem, srw_sortkeys, &sortspec, &num_sortspec); + if (num_sortspec > 0) + { + for (i = 0; i < num_sortspec; i++) + { + char **arg; + int num_arg; + int ascending = 1; + int case_sensitive = 0; + const char *missing = 0; + nmem_strsplitx(nmem, ",", sortspec[i], &arg, &num_arg, 0); + + if (num_arg > 2 && arg[2][0]) + ascending = atoi(arg[2]); + if (num_arg > 3 && arg[3][0]) + case_sensitive = atoi(arg[3]); + if (num_arg > 4 && arg[4][0]) + missing = arg[4]; + + if (i) + wrbuf_puts(w, " "); + + wrbuf_puts(w, arg[0]); /* field */ + wrbuf_puts(w, " "); + + wrbuf_puts(w, ascending ? "a" : "d"); + wrbuf_puts(w, case_sensitive ? "s" : "i"); + if (missing) + { + if (!strcmp(missing, "omit")) { + ; + } + else if (!strcmp(missing, "abort")) + wrbuf_puts(w, "!"); + else if (!strcmp(missing, "lowValue")) { + ; + } + else if (!strcmp(missing, "highValue")) { + ; + } + else + { + wrbuf_puts(w, "="); + wrbuf_puts(w, missing); + } + } + } + } + nmem_destroy(nmem); + return 0; +} + +int yaz_solr_sortkeys_to_sort_spec(const char *solr_sortkeys, WRBUF w) +{ + /* Solr sortkey layout: field order[, field order] */ + /* see cql_sortby_to_sortkeys of YAZ. */ + char **sortspec; + int num_sortspec = 0; + int i; + NMEM nmem = nmem_create(); + + if (solr_sortkeys) + nmem_strsplit(nmem, ",", solr_sortkeys, &sortspec, &num_sortspec); + if (num_sortspec > 0) + { + for (i = 0; i < num_sortspec; i++) + { + char **arg; + int num_arg; + char order = 'a'; + int case_sensitive = 0; + nmem_strsplitx(nmem, " ", sortspec[i], &arg, &num_arg, 0); + + if (num_arg != 2) + return -1; + + if (yaz_matchstr(arg[1], "asc") && + yaz_matchstr(arg[1], "desc")) + return -1; + + if (arg[1][0]) { + order = tolower(arg[1][0]); + } + if (i) + wrbuf_puts(w, " "); + + wrbuf_puts(w, arg[0]); /* field */ + wrbuf_puts(w, " "); + + wrbuf_putc(w, order); + // Always in-sensitive + wrbuf_puts(w, case_sensitive ? "s" : "i"); + } + } + nmem_destroy(nmem); + return 0; +} + /* * Local variables: