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
|
/*
* Copyright (C) 2004-2012 Geometer Plus <[email protected]>
*
* 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 <ZLFile.h>
#include <ZLInputStream.h>
#include <ZLEncodingConverter.h>
#include <ZLStringUtil.h>
#include <ZLLanguageUtil.h>
#include <ZLFileImage.h>
#include "PdbPlugin.h"
#include "EReaderStream.h"
#include "PmlBookReader.h"
#include "../../library/Book.h"
bool EReaderPlugin::providesMetaInfo() const {
return true;
}
bool EReaderPlugin::acceptsFile(const ZLFile &file) const {
return PdbPlugin::fileType(file) == "PNRdPPrs";
}
void EReaderPlugin::readDocumentInternal(const ZLFile &file, BookModel &model, const PlainTextFormat &format, const std::string &encoding, ZLInputStream &stream) const {
if (!stream.open()) {
//TODO maybe anything else opens stream
return;
}
BookReader bookReader(model);
PmlBookReader pmlBookReader(bookReader, format, encoding);
bookReader.setMainTextModel();
pmlBookReader.readDocument(stream);
EReaderStream &estream = (EReaderStream&)stream;
const std::map<std::string, EReaderStream::ImageInfo>& imageIds = estream.images();
for(std::map<std::string, EReaderStream::ImageInfo>::const_iterator it = imageIds.begin(); it != imageIds.end(); ++it) {
const std::string id = it->first;
bookReader.addImage(id, new ZLFileImage(ZLFile(file.path(), it->second.Type), it->second.Offset, it->second.Size));
}
const std::map<std::string, unsigned short>& footnoteIds = estream.footnotes();
for(std::map<std::string, unsigned short>::const_iterator it = footnoteIds.begin(); it != footnoteIds.end(); ++it) {
const std::string id = it->first;
if (estream.switchStreamDestination(EReaderStream::FOOTNOTE, id)) {
bookReader.setFootnoteTextModel(id);
bookReader.addHyperlinkLabel(id);
pmlBookReader.readDocument(estream);
}
}
stream.close();
}
shared_ptr<ZLInputStream> EReaderPlugin::createStream(const ZLFile &file) const {
return new EReaderStream(file);
}
const std::string &EReaderPlugin::tryOpen(const ZLFile &file) const {
EReaderStream stream(file);
stream.open();
return stream.error();
}
bool EReaderPlugin::readMetaInfo(Book &book) const {
shared_ptr<ZLInputStream> stream = book.file().inputStream();
if (stream.isNull() || ! stream->open()) {
return false;
}
PdbHeader header;
if (!header.read(stream)) {
return false;
}
stream->seek(header.Offsets[0] + 46, true);
unsigned short metaInfoOffset;
PdbUtil::readUnsignedShort(*stream, metaInfoOffset);
if (metaInfoOffset == 0 || metaInfoOffset >= header.Offsets.size()) {
return false;
}
std::size_t currentOffset = header.Offsets[metaInfoOffset];
std::size_t nextOffset =
(metaInfoOffset + 1 < (unsigned short)header.Offsets.size()) ?
header.Offsets[metaInfoOffset + 1] : stream->sizeOfOpened();
if (nextOffset <= currentOffset) {
return false;
}
std::size_t length = nextOffset - currentOffset;
char* metaInfoBuffer = new char[length];
stream->seek(currentOffset, true);
stream->read(metaInfoBuffer, length);
std::string metaInfoStr(metaInfoBuffer, length);
delete[] metaInfoBuffer;
std::string metaInfoData[5]; // Title; Author; Rights; Publisher; isbn;
for (std::size_t i = 0; i < 5; ++i) {
const std::size_t index = metaInfoStr.find('\0');
metaInfoData[i] = metaInfoStr.substr(0,index);
metaInfoStr = metaInfoStr.substr(index + 1);
}
if (!metaInfoData[0].empty()) {
book.setTitle(metaInfoData[0]);
}
if (!metaInfoData[1].empty()) {
book.addAuthor(metaInfoData[1]);
}
stream->close();
return SimplePdbPlugin::readMetaInfo(book);
}
|
