Examples with Struct com.yahoo.document.datatypes.Struct used on opensource projects

Example 36 with Struct

use of com.yahoo.document.datatypes.Struct in project vespa by vespa-engine.

the class Bug4475379TestCase method annotate.

public void annotate(DocumentTypeManager manager, Document document) {
    AnnotationTypeRegistry registry = manager.getAnnotationTypeRegistry();
    AnnotationType company = registry.getType("company");
    AnnotationType industry = registry.getType("industry");
    AnnotationType person = registry.getType("person");
    AnnotationType location = registry.getType("location");
    Annotation compAn1;
    Annotation personAn1;
    Annotation locAn1;
    Annotation indAn1;
    Annotation compAn2;
    Annotation personAn2;
    Annotation locAn2;
    Annotation indAn2;
    {
        Struct companyValue1 = (Struct) company.getDataType().createFieldValue();
        companyValue1.setFieldValue("name", new StringFieldValue("Sun"));
        companyValue1.setFieldValue("ceo", new StringFieldValue("Scott Mcnealy"));
        companyValue1.setFieldValue("lat", new FloatFieldValue(37.7f));
        companyValue1.setFieldValue("lon", new FloatFieldValue(-122.44f));
        companyValue1.setFieldValue("alt", 60.456);
        companyValue1.setFieldValue("vertical", new StringFieldValue("software"));
        compAn1 = new Annotation(company, companyValue1);
    }
    {
        Struct personValue1 = new Struct(manager.getDataType("annotation.person"));
        personValue1.setFieldValue("name", new StringFieldValue("Richard Bair"));
        personAn1 = new Annotation(person, personValue1);
    }
    {
        Struct locValue1 = new Struct(manager.getDataType("annotation.location"));
        locValue1.setFieldValue("name", new StringFieldValue("Prinsens Gate"));
        locAn1 = new Annotation(location, locValue1);
    }
    {
        Struct indValue1 = new Struct(manager.getDataType("annotation.industry"));
        indValue1.setFieldValue("vertical", new StringFieldValue("Software Services"));
        indAn1 = new Annotation(industry, indValue1);
    }
    {
        Struct companyValue2 = (Struct) company.getDataType().createFieldValue();
        companyValue2.setFieldValue("name", new StringFieldValue("Yahoo"));
        companyValue2.setFieldValue("ceo", new StringFieldValue("Carol Bartz"));
        companyValue2.setFieldValue("lat", new FloatFieldValue(32.1f));
        companyValue2.setFieldValue("lon", new FloatFieldValue(-48.44f));
        companyValue2.setFieldValue("alt", 33.56);
        companyValue2.setFieldValue("vertical", new StringFieldValue("Research"));
        compAn2 = new Annotation(company, companyValue2);
    }
    {
        Struct personValue2 = new Struct(manager.getDataType("annotation.person"));
        personValue2.setFieldValue("name", new StringFieldValue("Kim Johansen"));
        personAn2 = new Annotation(person, personValue2);
    }
    {
        Struct locValue2 = new Struct(manager.getDataType("annotation.location"));
        locValue2.setFieldValue("name", new StringFieldValue("RT Nagar"));
        locAn2 = new Annotation(location, locValue2);
    }
    {
        Struct indValue2 = new Struct(manager.getDataType("annotation.industry"));
        indValue2.setFieldValue("vertical", new StringFieldValue("Software Consulting"));
        indAn2 = new Annotation(industry, indValue2);
    }
    SpanTree tree = new SpanTree("test");
    SpanList root = (SpanList) tree.getRoot();
    AlternateSpanList branch = new AlternateSpanList();
    SpanNode span1 = new Span(0, 3);
    SpanNode span2 = new Span(1, 9);
    SpanNode span3 = new Span(12, 10);
    SpanNode span11 = new Span(0, 3);
    SpanNode span22 = new Span(1, 9);
    SpanNode span33 = new Span(12, 10);
    SpanList alternate1 = new SpanList();
    alternate1.add(span3);
    alternate1.add(span2);
    alternate1.add(span1);
    SpanList alternate2 = new SpanList();
    alternate2.add(span11);
    alternate2.add(span22);
    alternate2.add(span33);
    tree.annotate(span1, compAn1);
    tree.annotate(span2, personAn1);
    tree.annotate(span3, locAn1);
    tree.annotate(span1, indAn1);
    tree.annotate(span11, compAn2);
    tree.annotate(span22, personAn2);
    tree.annotate(span33, locAn2);
    tree.annotate(span11, indAn2);
    List<SpanNode> subtreeList1 = new ArrayList<>();
    subtreeList1.add(alternate1);
    List<SpanNode> subtreeList2 = new ArrayList<>();
    subtreeList2.add(alternate2);
    branch.addChildren(1, subtreeList1, 20.0d);
    branch.addChildren(2, subtreeList2, 50.0d);
    root.add(branch);
    StringFieldValue body = (StringFieldValue) document.getFieldValue(document.getDataType().getField("body"));
    body.setSpanTree(tree);
    document.setFieldValue(document.getDataType().getField("body"), body);
}

Example 37 with Struct

use of com.yahoo.document.datatypes.Struct in project vespa by vespa-engine.

the class Bug6425939TestCase method createSpanTree.

private SpanTree createSpanTree() {
    SpanList root = new SpanList();
    SpanTree tree = new SpanTree("SpanTree1", root);
    SpanNode node = new Span(0, 0);
    Struct ps = new Struct(person);
    ps.setFieldValue("foo", "epic badger");
    ps.setFieldValue("bar", 54321);
    tree.annotate(node, new Annotation(personA, ps));
    root.add(node);
    return tree;
}

Example 38 with Struct

use of com.yahoo.document.datatypes.Struct in project vespa by vespa-engine.

the class StructDataTypeTestCase method testCompatibleWith.

public void testCompatibleWith() {
    StructDataType personType = new StructDataType("person");
    Field firstName = new Field("firstname", DataType.STRING);
    Field lastName = new Field("lastname", DataType.STRING);
    personType.addField(firstName);
    personType.addField(lastName);
    StructDataType employeeType = new StructDataType("employee");
    Field empId = new Field("employeeid", DataType.INT);
    employeeType.addField(empId);
    employeeType.inherit(personType);
    Struct person = new Struct(personType);
    Struct employee = new Struct(employeeType);
    assertTrue(personType.isValueCompatible(person));
    assertTrue(personType.isValueCompatible(employee));
    assertTrue(employeeType.isValueCompatible(employee));
    assertFalse(employeeType.isValueCompatible(person));
    StructDataType containerType = new StructDataType("containerstruct");
    Field structPolymorphic = new Field("structpolymorphic", personType);
    containerType.addField(structPolymorphic);
    Struct container = new Struct(containerType);
    container.setFieldValue(structPolymorphic, person);
    container.setFieldValue(structPolymorphic, employee);
}