Mercurial > hg > mpdl-group
comparison software/eXist/mpdl-modules/src/org/exist/xquery/modules/mpdltext/GetLemmasByLuceneQuery.java @ 0:408254cf2f1d
Erstellung
author | Josef Willenborg <jwillenborg@mpiwg-berlin.mpg.de> |
---|---|
date | Wed, 24 Nov 2010 17:24:23 +0100 |
parents | |
children |
comparison
equal
deleted
inserted
replaced
-1:000000000000 | 0:408254cf2f1d |
---|---|
1 /* | |
2 * eXist Open Source Native XML Database: Extension module | |
3 * Copyright (C) 2008 Josef Willenborg | |
4 * jwillenborg@mpiwg-berlin.mpg.de | |
5 * http://www.mpiwg-berlin.mpg.de | |
6 * | |
7 * This program is free software; you can redistribute it and/or | |
8 * modify it under the terms of the GNU Lesser General Public License | |
9 * as published by the Free Software Foundation; either version 2 | |
10 * of the License, or (at your option) any later version. | |
11 * | |
12 * This program is distributed in the hope that it will be useful, | |
13 * but WITHOUT ANY WARRANTY; without even the implied warranty of | |
14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
15 * GNU Lesser General Public License for more details. | |
16 * | |
17 * You should have received a copy of the GNU Lesser General Public License | |
18 * along with this program; if not, write to the Free Software | |
19 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA. | |
20 * | |
21 * $Id: TextModule.java $ | |
22 */ | |
23 package org.exist.xquery.modules.mpdltext; | |
24 | |
25 import java.util.ArrayList; | |
26 | |
27 import org.exist.dom.QName; | |
28 import org.exist.memtree.DocumentImpl; | |
29 import org.exist.memtree.MemTreeBuilder; | |
30 import org.exist.xquery.BasicFunction; | |
31 import org.exist.xquery.Cardinality; | |
32 import org.exist.xquery.FunctionSignature; | |
33 import org.exist.xquery.XPathException; | |
34 import org.exist.xquery.XQueryContext; | |
35 import org.exist.xquery.value.Sequence; | |
36 import org.exist.xquery.value.SequenceType; | |
37 import org.exist.xquery.value.Type; | |
38 | |
39 import de.mpg.mpiwg.berlin.mpdl.exception.ApplicationException; | |
40 import de.mpg.mpiwg.berlin.mpdl.lt.morph.app.Form; | |
41 import de.mpg.mpiwg.berlin.mpdl.lt.morph.app.Lemma; | |
42 import de.mpg.mpiwg.berlin.mpdl.lt.morph.app.MorphologyCache; | |
43 | |
44 /** | |
45 * @author Josef Willenborg (jwillenborg@mpiwg-berlin.mpg.de) | |
46 */ | |
47 public class GetLemmasByLuceneQuery extends BasicFunction { | |
48 | |
49 public final static FunctionSignature signature = | |
50 new FunctionSignature( | |
51 new QName("get-lemmas-by-lucene-query", MPDLTextModule.NAMESPACE_URI, MPDLTextModule.PREFIX), | |
52 "A function which delivers the lemma of a given lucene query and language " + | |
53 "by the MPDL language technology", | |
54 new SequenceType[] { new SequenceType(Type.STRING, Cardinality.ZERO_OR_MORE), | |
55 new SequenceType(Type.STRING, Cardinality.ZERO_OR_MORE) }, | |
56 new SequenceType(Type.NODE, Cardinality.EXACTLY_ONE)); | |
57 | |
58 public GetLemmasByLuceneQuery(XQueryContext context) { | |
59 super(context, signature); | |
60 } | |
61 | |
62 public Sequence eval(Sequence[] args, Sequence contextSequence) throws XPathException { | |
63 Sequence langSeq = args[0]; | |
64 Sequence luceneQuerySeq = args[1]; | |
65 String language = ""; | |
66 String luceneQueryStr = ""; | |
67 if (langSeq.isEmpty() || luceneQuerySeq.isEmpty()) | |
68 return Sequence.EMPTY_SEQUENCE; | |
69 language = langSeq.getStringValue(); | |
70 luceneQueryStr = luceneQuerySeq.getStringValue(); | |
71 ArrayList<Lemma> lemmas = null; | |
72 try { | |
73 MorphologyCache morphologyCache = MorphologyCache.getInstance(); | |
74 lemmas = morphologyCache.getLemmasByLuceneQuery(language, luceneQueryStr, true); | |
75 } catch (ApplicationException e) { | |
76 throw new XPathException(e); | |
77 } | |
78 DocumentImpl doc = null; | |
79 if (lemmas != null) { | |
80 MemTreeBuilder builder = context.getDocumentBuilder(); | |
81 builder.startElement("", "lemmas", "lemmas", null); | |
82 for (int i=0; i<lemmas.size(); i++) { | |
83 Lemma lemma = lemmas.get(i); | |
84 builder.startElement("", "lemma", "lemma", null); | |
85 builder.startElement("", "provider", "provider", null); | |
86 builder.characters(lemma.getProvider()); | |
87 builder.endElement(); | |
88 builder.startElement("", "language", "language", null); | |
89 builder.characters(lemma.getLanguage()); | |
90 builder.endElement(); | |
91 builder.startElement("", "lemma-name", "lemma-name", null); | |
92 builder.characters(lemma.getLemmaName()); | |
93 builder.endElement(); | |
94 ArrayList<Form> forms = lemma.getFormsList(); | |
95 builder.startElement("", "forms-size", "forms-size", null); | |
96 builder.characters(String.valueOf(forms.size())); | |
97 builder.endElement(); | |
98 builder.startElement("", "forms", "forms", null); | |
99 for (int j=0; j<forms.size(); j++) { | |
100 Form form = forms.get(j); | |
101 builder.startElement("", "form", "form", null); | |
102 builder.startElement("", "provider", "provider", null); | |
103 builder.characters(form.getProvider()); | |
104 builder.endElement(); | |
105 builder.startElement("", "language", "language", null); | |
106 builder.characters(form.getLanguage()); | |
107 builder.endElement(); | |
108 builder.startElement("", "form-name", "form-name", null); | |
109 builder.characters(form.getFormName()); | |
110 builder.endElement(); | |
111 builder.startElement("", "lemma-name", "lemma-name", null); | |
112 builder.characters(lemma.getLemmaName()); | |
113 builder.endElement(); | |
114 builder.endElement(); | |
115 } | |
116 builder.endElement(); | |
117 builder.endElement(); | |
118 } | |
119 builder.endElement(); | |
120 doc = ((DocumentImpl)builder.getDocument()); | |
121 } else { | |
122 return Sequence.EMPTY_SEQUENCE; | |
123 } | |
124 return doc; | |
125 } | |
126 } |