Search in sources :

Example 21 with FieldDocument

use of de.ids_mannheim.korap.index.FieldDocument in project Krill by KorAP.

the class TestHighlight method checkHighlights.

@Test
public void checkHighlights() throws IOException, QueryException {
    KrillIndex ki = new KrillIndex();
    String json = new String("{" + "  \"fields\" : [" + "    { " + "      \"primaryData\" : \"abc\"" + "    }," + "    {" + "      \"name\" : \"tokens\"," + "      \"data\" : [" + "         [ \"s:a\", \"i:a\", \"_0#0-1\", \"-:t$<i>3\"]," + "         [ \"s:b\", \"i:b\", \"_1#1-2\" ]," + "         [ \"s:c\", \"i:c\", \"_2#2-3\" ]" + "      ]" + "    }" + "  ]" + "}");
    FieldDocument fd = ki.addDoc(json);
    ki.commit();
    QueryBuilder kq = new QueryBuilder("tokens");
    Result kr = ki.search((SpanQuery) kq.seq(kq.nr(1, kq.seg("s:b"))).toQuery());
    Match km = kr.getMatch(0);
    assertEquals(km.getStartPos(), 1);
    assertEquals(km.getEndPos(), 2);
    assertEquals(km.getStartPos(1), 1);
    assertEquals(km.getEndPos(1), 2);
    assertEquals("<span class=\"context-left\">a</span><span class=\"match\"><mark><mark class=\"class-1 level-0\">b</mark></mark></span><span class=\"context-right\">c</span>", km.getSnippetHTML());
    kr = ki.search((SpanQuery) kq.seq(kq.nr(1, kq.seg("s:b"))).append(kq.nr(2, kq.seg("s:c"))).toQuery());
    km = kr.getMatch(0);
    assertEquals(km.getStartPos(), 1);
    assertEquals(km.getEndPos(), 3);
    assertEquals(km.getStartPos(1), 1);
    assertEquals(km.getEndPos(1), 2);
    assertEquals(km.getStartPos(2), 2);
    assertEquals(km.getEndPos(2), 3);
    assertEquals("<span class=\"context-left\">a</span><span class=\"match\"><mark><mark class=\"class-1 level-0\">b</mark><mark class=\"class-2 level-0\">c</mark></mark></span><span class=\"context-right\"></span>", km.getSnippetHTML());
    kr = ki.search((SpanQuery) kq.seq(kq.nr(1, kq.seq(kq.seg("s:a")).append(kq.seg("s:b")))).append(kq.nr(2, kq.seg("s:c"))).toQuery());
    km = kr.getMatch(0);
    assertEquals(km.getStartPos(), 0);
    assertEquals(km.getEndPos(), 3);
    assertEquals(km.getStartPos(1), 0);
    assertEquals(km.getEndPos(1), 2);
    assertEquals(km.getStartPos(2), 2);
    assertEquals(km.getEndPos(2), 3);
    assertEquals("<span class=\"context-left\"></span><span class=\"match\"><mark><mark class=\"class-1 level-0\">ab</mark><mark class=\"class-2 level-0\">c</mark></mark></span><span class=\"context-right\"></span>", km.getSnippetHTML());
    kr = ki.search((SpanQuery) kq.nr(3, kq.seq(kq.nr(1, kq.seq(kq.seg("s:a")).append(kq.seg("s:b")))).append(kq.nr(2, kq.seg("s:c")))).toQuery());
    km = kr.getMatch(0);
    assertEquals(km.getStartPos(), 0);
    assertEquals(km.getEndPos(), 3);
    assertEquals(km.getStartPos(1), 0);
    assertEquals(km.getEndPos(1), 2);
    assertEquals(km.getStartPos(2), 2);
    assertEquals(km.getEndPos(2), 3);
    assertEquals(km.getStartPos(3), 0);
    assertEquals(km.getEndPos(3), 3);
    assertEquals("<span class=\"context-left\"></span><span class=\"match\"><mark><mark class=\"class-3 level-0\"><mark class=\"class-1 level-1\">ab</mark><mark class=\"class-2 level-1\">c</mark></mark></mark></span><span class=\"context-right\"></span>", km.getSnippetHTML());
}
Also used : QueryBuilder(de.ids_mannheim.korap.query.QueryBuilder) FieldDocument(de.ids_mannheim.korap.index.FieldDocument) KrillIndex(de.ids_mannheim.korap.KrillIndex) Result(de.ids_mannheim.korap.response.Result) Match(de.ids_mannheim.korap.response.Match) SpanQuery(org.apache.lucene.search.spans.SpanQuery) Test(org.junit.Test)

Example 22 with FieldDocument

use of de.ids_mannheim.korap.index.FieldDocument in project Krill by KorAP.

the class TestHighlight method highlightEscapes.

@Test
public void highlightEscapes() throws IOException, QueryException {
    KrillIndex ki = new KrillIndex();
    FieldDocument fd = new FieldDocument();
    fd.addString("ID", "doc-1");
    fd.addString("UID", "1");
    fd.addString("textSigle", "c1/d1/1");
    // Make this clean for HTML and Brackets!
    fd.addTV("base", "Mit \"Mann\" & {Ma\\us}", "[(0-3)s:Mit|i:mit|_0#0-3|-:t$<i>4|<>:base/t:t$<b>64<i>0<i>20<i>4<b>0]" + "[(4-10)s:\"Mann\"|i:\"mann\"|base/l:\"Mann\"|_1#4-10]" + "[(11-12)s:&|i:&|base/l:&|_2#11-12]" + "[(13-20)s:{Ma\\us}|i:{ma\\us}|_3#13-20]");
    ki.addDoc(fd);
    // Commit!
    ki.commit();
    QueryBuilder kq = new QueryBuilder("base");
    SpanQuery q = (SpanQuery) kq.tag("base/t:t").toQuery();
    Krill qs = new Krill(q);
    qs.getMeta().getContext().left.setToken(true).setLength((short) 0);
    qs.getMeta().getContext().right.setToken(true).setLength((short) 0);
    Result kr = ki.search(qs);
    assertEquals((long) 1, kr.getTotalResults());
    assertEquals("[[Mit \"Mann\" & \\{Ma\\\\us\\}]]", kr.getMatch(0).getSnippetBrackets());
    assertEquals("<span class=\"context-left\"></span><span class=\"match\"><mark>Mit &quot;Mann&quot; &amp; {Ma\\us}</mark></span><span class=\"context-right\"></span>", kr.getMatch(0).getSnippetHTML());
    assertEquals("match-c1/d1/1-p0-4", kr.getMatch(0).getID());
    Match km = ki.getMatchInfo("match-c1/d1/1-p0-4", "base", true, (ArrayList) null, (ArrayList) null, true, true, false);
    assertEquals(0, km.getStartPos());
    assertEquals("<span class=\"context-left\"></span>" + "<span class=\"match\"><mark><span title=\"base/t:t\">" + "Mit " + "<span title=\"base/l:&quot;Mann&quot;\">" + "&quot;Mann&quot;" + "</span>" + " " + "<span title=\"base/l:&amp;\">&amp;</span>" + " " + "{Ma\\us}" + "</span>" + "</mark></span>" + "<span class=\"context-right\"></span>", km.getSnippetHTML());
}
Also used : Krill(de.ids_mannheim.korap.Krill) QueryBuilder(de.ids_mannheim.korap.query.QueryBuilder) FieldDocument(de.ids_mannheim.korap.index.FieldDocument) KrillIndex(de.ids_mannheim.korap.KrillIndex) SpanQuery(org.apache.lucene.search.spans.SpanQuery) Result(de.ids_mannheim.korap.response.Result) Match(de.ids_mannheim.korap.response.Match) Test(org.junit.Test)

Example 23 with FieldDocument

use of de.ids_mannheim.korap.index.FieldDocument in project Krill by KorAP.

the class TestHighlight method highlightMissingBug.

@Test
public void highlightMissingBug() throws IOException, QueryException {
    KrillIndex ki = new KrillIndex();
    FieldDocument fd = new FieldDocument();
    fd.addString("ID", "doc-1");
    fd.addString("UID", "1");
    fd.addTV("base", "abab", "[(0-1)s:a|i:a|_0#0-1|-:t$<i>4]" + "[(1-2)s:b|i:b|_1#1-2]" + "[(2-3)s:a|i:c|_2#2-3]" + "[(3-4)s:b|i:a|_3#3-4]");
    ki.addDoc(fd);
    fd = new FieldDocument();
    fd.addString("ID", "doc-2");
    fd.addString("UID", "2");
    fd.addTV("base", "aba", "[(0-1)s:a|i:a|_0#0-1|-:t$<i>3]" + "[(1-2)s:b|i:b|_1#1-2]" + "[(2-3)s:a|i:c|_2#2-3]");
    ki.addDoc(fd);
    // Commit!
    ki.commit();
    fd = new FieldDocument();
    fd.addString("ID", "doc-3");
    fd.addString("UID", "3");
    fd.addTV("base", "abab", "[(0-1)s:a|i:a|_0#0-1|-:t$<i>4]" + "[(1-2)s:b|i:b|_1#1-2]" + "[(2-3)s:a|i:c|_2#2-3]" + "[(3-4)s:b|i:a|_3#3-4]");
    ki.addDoc(fd);
    // Commit!
    ki.commit();
    fd = new FieldDocument();
    fd.addString("ID", "doc-4");
    fd.addString("UID", "4");
    fd.addTV("base", "aba", "[(0-1)s:a|i:a|_0#0-1|-:t$<i>3]" + "[(1-2)s:b|i:b|_1#1-2]" + "[(2-3)s:a|i:c|_2#2-3]");
    ki.addDoc(fd);
    // Commit!
    ki.commit();
    QueryBuilder kq = new QueryBuilder("base");
    SpanQuery q = (SpanQuery) kq.or(kq.nr(1, kq.seg("s:a"))).or(kq.nr(2, kq.seg("s:b"))).toQuery();
    Result kr = ki.search(q);
    assertEquals((long) 14, kr.getTotalResults());
    assertEquals("[[{1:a}]]bab", kr.getMatch(0).getSnippetBrackets());
    assertEquals("a[[{2:b}]]ab", kr.getMatch(1).getSnippetBrackets());
    assertEquals("ab[[{1:a}]]b", kr.getMatch(2).getSnippetBrackets());
    assertEquals("aba[[{2:b}]]", kr.getMatch(3).getSnippetBrackets());
    assertEquals("[[{1:a}]]ba", kr.getMatch(4).getSnippetBrackets());
    assertEquals("a[[{2:b}]]a", kr.getMatch(5).getSnippetBrackets());
    assertEquals("ab[[{1:a}]]", kr.getMatch(6).getSnippetBrackets());
    assertEquals("[[{1:a}]]bab", kr.getMatch(7).getSnippetBrackets());
    assertEquals("a[[{2:b}]]ab", kr.getMatch(8).getSnippetBrackets());
    assertEquals("ab[[{1:a}]]b", kr.getMatch(9).getSnippetBrackets());
    assertEquals("aba[[{2:b}]]", kr.getMatch(10).getSnippetBrackets());
    assertEquals("[[{1:a}]]ba", kr.getMatch(11).getSnippetBrackets());
    assertEquals("a[[{2:b}]]a", kr.getMatch(12).getSnippetBrackets());
    assertEquals("ab[[{1:a}]]", kr.getMatch(13).getSnippetBrackets());
    kq = new QueryBuilder("base");
    q = (SpanQuery) kq.or(kq.nr(1, kq.seg("i:a"))).or(kq.nr(2, kq.seg("i:c"))).toQuery();
    Krill qs = new Krill(q);
    qs.getMeta().getContext().left.setToken(true).setLength((short) 1);
    qs.getMeta().getContext().right.setToken(true).setLength((short) 1);
    kr = ki.search(qs);
    assertEquals((long) 10, kr.getTotalResults());
    assertEquals("[[{1:a}]]b ...", kr.getMatch(0).getSnippetBrackets());
    assertEquals("... b[[{2:a}]]b", kr.getMatch(1).getSnippetBrackets());
    assertEquals("... a[[{1:b}]]", kr.getMatch(2).getSnippetBrackets());
    assertEquals("[[{1:a}]]b ...", kr.getMatch(3).getSnippetBrackets());
    assertEquals("... b[[{2:a}]]", kr.getMatch(4).getSnippetBrackets());
    assertEquals("[[{1:a}]]b ...", kr.getMatch(5).getSnippetBrackets());
    assertEquals("... b[[{2:a}]]b", kr.getMatch(6).getSnippetBrackets());
    assertEquals("... a[[{1:b}]]", kr.getMatch(7).getSnippetBrackets());
    assertEquals("[[{1:a}]]b ...", kr.getMatch(8).getSnippetBrackets());
    assertEquals("... b[[{2:a}]]", kr.getMatch(9).getSnippetBrackets());
    qs.getMeta().getContext().left.setToken(true).setLength((short) 0);
    qs.getMeta().getContext().right.setToken(true).setLength((short) 0);
    kr = ki.search(qs);
    assertEquals((long) 10, kr.getTotalResults());
    assertEquals("[[{1:a}]] ...", kr.getMatch(0).getSnippetBrackets());
    assertEquals("... [[{2:a}]] ...", kr.getMatch(1).getSnippetBrackets());
    assertEquals("... [[{1:b}]]", kr.getMatch(2).getSnippetBrackets());
    assertEquals("[[{1:a}]] ...", kr.getMatch(3).getSnippetBrackets());
    assertEquals("... [[{2:a}]]", kr.getMatch(4).getSnippetBrackets());
    assertEquals("[[{1:a}]] ...", kr.getMatch(5).getSnippetBrackets());
    assertEquals("... [[{2:a}]] ...", kr.getMatch(6).getSnippetBrackets());
    assertEquals("... [[{1:b}]]", kr.getMatch(7).getSnippetBrackets());
    assertEquals("[[{1:a}]] ...", kr.getMatch(8).getSnippetBrackets());
    assertEquals("... [[{2:a}]]", kr.getMatch(9).getSnippetBrackets());
    q = (SpanQuery) kq.nr(3, kq.or(kq.nr(1, kq.seg("i:a"))).or(kq.nr(2, kq.seg("i:c")))).toQuery();
    qs = new Krill(q);
    qs.getMeta().getContext().left.setToken(true).setLength((short) 0);
    qs.getMeta().getContext().right.setToken(true).setLength((short) 0);
    kr = ki.search(qs);
    assertEquals((long) 10, kr.getTotalResults());
    assertEquals("[[{1:{3:a}}]] ...", kr.getMatch(0).getSnippetBrackets());
    assertEquals("... [[{2:{3:a}}]] ...", kr.getMatch(1).getSnippetBrackets());
    assertEquals("... [[{1:{3:b}}]]", kr.getMatch(2).getSnippetBrackets());
    assertEquals("[[{1:{3:a}}]] ...", kr.getMatch(3).getSnippetBrackets());
    assertEquals("... [[{2:{3:a}}]]", kr.getMatch(4).getSnippetBrackets());
    assertEquals("[[{1:{3:a}}]] ...", kr.getMatch(5).getSnippetBrackets());
    assertEquals("... [[{2:{3:a}}]] ...", kr.getMatch(6).getSnippetBrackets());
    assertEquals("... [[{1:{3:b}}]]", kr.getMatch(7).getSnippetBrackets());
    assertEquals("[[{1:{3:a}}]] ...", kr.getMatch(8).getSnippetBrackets());
    assertEquals("... [[{2:{3:a}}]]", kr.getMatch(9).getSnippetBrackets());
}
Also used : Krill(de.ids_mannheim.korap.Krill) QueryBuilder(de.ids_mannheim.korap.query.QueryBuilder) FieldDocument(de.ids_mannheim.korap.index.FieldDocument) KrillIndex(de.ids_mannheim.korap.KrillIndex) SpanQuery(org.apache.lucene.search.spans.SpanQuery) Result(de.ids_mannheim.korap.response.Result) Test(org.junit.Test)

Example 24 with FieldDocument

use of de.ids_mannheim.korap.index.FieldDocument in project Krill by KorAP.

the class TestKrillCollectionIndex method createDoc2.

private FieldDocument createDoc2() {
    FieldDocument fd = new FieldDocument();
    fd.addString("ID", "doc-2");
    fd.addString("author", "Peter");
    fd.addKeyword("textClass", "Kultur Reisen");
    fd.addInt("pubDate", 20051207);
    fd.addText("text", "Der junge Mann hatte keine andere Wahl");
    return fd;
}
Also used : FieldDocument(de.ids_mannheim.korap.index.FieldDocument)

Example 25 with FieldDocument

use of de.ids_mannheim.korap.index.FieldDocument in project Krill by KorAP.

the class TestKrillCollectionIndex method uidCollectionWithDeletions.

@Test
public void uidCollectionWithDeletions() throws IOException {
    // Construct index
    KrillIndex ki = new KrillIndex();
    // Indexing test files
    int uid = 1;
    for (String i : new String[] { "00001", "00002", "00003", "00004", "00005", "00006", "02439" }) {
        FieldDocument fd = ki.addDoc(uid++, getClass().getResourceAsStream("/wiki/" + i + ".json.gz"), true);
    }
    ;
    ki.commit();
    assertEquals("Documents", 7, ki.numberOf("documents"));
    assertEquals("Paragraphs", 174, ki.numberOf("paragraphs"));
    assertEquals("Sentences", 281, ki.numberOf("sentences"));
    assertEquals("Tokens", 2661, ki.numberOf("tokens"));
    assertTrue(ki.delDoc(3));
    ki.commit();
    assertEquals("Documents", 6, ki.numberOf("documents"));
    assertEquals("Paragraphs", 146, ki.numberOf("paragraphs"));
    assertEquals("Sentences", 212, ki.numberOf("sentences"));
    assertEquals("Tokens", 2019, ki.numberOf("tokens"));
    assertTrue(ki.delDoc(2));
    assertTrue(ki.delDoc(3));
    assertTrue(ki.delDoc(4));
    assertTrue(ki.delDoc(5));
    assertTrue(ki.delDoc(6));
    assertTrue(ki.delDoc(7));
    ki.commit();
    assertEquals("Documents", 1, ki.numberOf("documents"));
    assertEquals("Paragraphs", 75, ki.numberOf("paragraphs"));
}
Also used : FieldDocument(de.ids_mannheim.korap.index.FieldDocument) KrillIndex(de.ids_mannheim.korap.KrillIndex) Test(org.junit.Test)

Aggregations

FieldDocument (de.ids_mannheim.korap.index.FieldDocument)40 KrillIndex (de.ids_mannheim.korap.KrillIndex)28 Test (org.junit.Test)28 Result (de.ids_mannheim.korap.response.Result)20 Krill (de.ids_mannheim.korap.Krill)15 QueryBuilder (de.ids_mannheim.korap.query.QueryBuilder)14 JsonNode (com.fasterxml.jackson.databind.JsonNode)7 Match (de.ids_mannheim.korap.response.Match)7 SpanQuery (org.apache.lucene.search.spans.SpanQuery)7 ObjectMapper (com.fasterxml.jackson.databind.ObjectMapper)6 KrillCollection (de.ids_mannheim.korap.KrillCollection)4 Test (de.ids_mannheim.korap.Test)4 TestSimple.getJsonString (de.ids_mannheim.korap.TestSimple.getJsonString)2 CollectionBuilder (de.ids_mannheim.korap.collection.CollectionBuilder)2 MultiTermTokenStream (de.ids_mannheim.korap.index.MultiTermTokenStream)1 Response (de.ids_mannheim.korap.response.Response)1 SearchContext (de.ids_mannheim.korap.response.SearchContext)1 Consumes (javax.ws.rs.Consumes)1 PUT (javax.ws.rs.PUT)1 Path (javax.ws.rs.Path)1