summaryrefslogtreecommitdiffstats
path: root/fbreader/src/formats/tcr/PPLBookReader.cpp
blob: 9b7d271d57b3c2c05bbaaf1776f3a3baa597663a (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
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
/*
 * Copyright (C) 2004-2012 Geometer Plus <contact@geometerplus.com>
 *
 * 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 2 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, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
 * 02110-1301, USA.
 */

#include <cstring>
#include <cctype>

#include "PPLBookReader.h"
#include <ZLInputStream.h>

static const std::size_t BUFFER_SIZE = 2048;

PPLBookReader::PPLBookReader(BookModel &model, const std::string &encoding) : EncodedTextReader(encoding), myModelReader(model) {
	myBuffer = new char[BUFFER_SIZE + 1];
}

PPLBookReader::~PPLBookReader() {
	delete[] myBuffer;
}

bool PPLBookReader::currentParagraphIsEmpty() const {
	const char *ptr = myCurrentParagraph.data();
	const char *end = ptr + myCurrentParagraph.length();
	for (; ptr < end; ++ptr) {
		if (!std::isspace((unsigned char)*ptr)) {
			return false;
		}
	}
	return true;
}

void PPLBookReader::addParagraph() {
	static const std::string END_OF_TEXT = "<* >";
	if (!myCurrentParagraph.empty()) {
		if (currentParagraphIsEmpty()) {
			++myEmptyLineCounter;
			if (myEmptyLineCounter >= 2) {
				myModelReader.beginParagraph(ZLTextParagraph::EMPTY_LINE_PARAGRAPH);
				myModelReader.endParagraph();
			}
		} else if (myEmptyLineCounter < 2) {
			myModelReader.beginParagraph();
			myModelReader.addControl(TITLE, true);
			myModelReader.addData(myCurrentParagraph);
			myModelReader.endParagraph();
		} else if (myCurrentParagraph[0] == 9) {
			myModelReader.beginParagraph();
			myModelReader.addData(myCurrentParagraph);
			myModelReader.endParagraph();
		} else if ((myCurrentParagraph.length() >= 2) &&
				(myCurrentParagraph[0] == '*') &&
				(myCurrentParagraph[1] == ' ')) {
			myCurrentParagraph.erase(0, 2);
			myModelReader.insertEndOfSectionParagraph();
			myModelReader.beginContentsParagraph();
			myModelReader.addContentsData(myCurrentParagraph);
			myModelReader.endContentsParagraph();
			myModelReader.beginParagraph();
			myModelReader.addControl(SECTION_TITLE, true);
			myModelReader.addData(myCurrentParagraph);
			myModelReader.endParagraph();
		} else if (myCurrentParagraph.substr(0, 4) != END_OF_TEXT) {
			myModelReader.beginParagraph();
			myModelReader.addControl(SUBTITLE, true);
			myModelReader.addData(myCurrentParagraph);
			myModelReader.endParagraph();
		}
		myCurrentParagraph.erase();
	}
}

bool PPLBookReader::readDocument(ZLInputStream &stream) {
	if (!stream.open()) {
		return false;
	}

	myModelReader.setMainTextModel();
	myModelReader.pushKind(REGULAR);
	myCurrentParagraph.erase();
	myEmptyLineCounter = 0;

	// "PPL\r\n"
	stream.seek(5, false);

	std::size_t size;
	do {
		size = stream.read(myBuffer, BUFFER_SIZE);
		myBuffer[size] = '\0';

		const char *start = myBuffer;
		const char *end = myBuffer + size;
		const char *eol;
		do {
			eol = std::strchr(start, '\n');
			if (eol != 0) {
				if (start < eol) {
					myConverter->convert(myCurrentParagraph, start, eol);
				}
				addParagraph();
				start = eol + 1;
			} else {
				if (start < end) {
					myConverter->convert(myCurrentParagraph, start, end);
				}
			}
		} while (eol != 0);
	} while (size == BUFFER_SIZE);

	addParagraph();

	stream.close();

	return true;
}