summaryrefslogtreecommitdiff
path: root/test/core/model/TestDocument.hpp
blob: a4a8f7cd6eff79585ab469bdbefdcb29072786c5 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
/*
    Ousía
    Copyright (C) 2014, 2015  Benjamin Paaßen, Andreas Stöckel

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation, either version 3 of the License, or
    (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.

    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#ifndef _MODEL_TEST_DOCUMENT_HPP_
#define _MODEL_TEST_DOCUMENT_HPP_

#include <core/model/Document.hpp>
#include <core/model/Ontology.hpp>
#include <core/model/Typesystem.hpp>

#include "TestDocumentBuilder.hpp"

namespace ousia {

/**
 * This constructs a fairly simple test document for the "book" ontology. The
 * structure of the document can be seen in the Code below.
 */
static Rooted<Document> constructBookDocument(Manager &mgr, Logger &logger,
                                              Rooted<Ontology> bookOntology)
{
	// Start with the (empty) document.
	Rooted<Document> doc{new Document(mgr, "myDoc.oxd")};
	doc->referenceOntology(bookOntology);

	// Add the root.
	Rooted<StructuredEntity> root =
	    buildRootStructuredEntity(doc, logger, {"book"});
	if (root.isNull()) {
		return {nullptr};
	}

	// Add a paragraph.
	Rooted<StructuredEntity> foreword =
	    buildStructuredEntity(doc, logger, root, {"paragraph"});
	if (foreword.isNull()) {
		return {nullptr};
	}
	// Add its text.
	Rooted<StructuredEntity> foreword_text =
	    buildStructuredEntity(doc, logger, foreword, {"text"});
	if (foreword_text.isNull()) {
		return {nullptr};
	}
	// And its primitive content
	Variant text{"Some introductory text"};
	Rooted<DocumentPrimitive> foreword_primitive{
	    new DocumentPrimitive(mgr, foreword_text, text, DEFAULT_FIELD_NAME)};
	// Add a section.
	Rooted<StructuredEntity> section =
	    buildStructuredEntity(doc, logger, root, {"section"});
	// Add a paragraph for it.
	Rooted<StructuredEntity> main =
	    buildStructuredEntity(doc, logger, section, {"paragraph"});
	if (main.isNull()) {
		return {nullptr};
	}
	// Add its text.
	Rooted<StructuredEntity> main_text =
	    buildStructuredEntity(doc, logger, main, {"text"});
	if (main_text.isNull()) {
		return {nullptr};
	}
	// And its primitive content
	text = Variant{"Some actual text"};
	Rooted<DocumentPrimitive> main_primitive{
	    new DocumentPrimitive(mgr, main_text, text, DEFAULT_FIELD_NAME)};
	if (main_primitive.isNull()) {
		return {nullptr};
	}

	return doc;
}
}

#endif /* _TEST_DOCUMENT_HPP_ */