void DjVuSource::ReadAnnotations(GP<ByteStream> pInclStream,
set<GUTF8String>& processed, GP<ByteStream> pAnnoStream)
{
// Look for shared annotations
GUTF8String strInclude;
char buf[1024];
int nLength;
while ((nLength = pInclStream->read(buf, 1024)))
strInclude += GUTF8String(buf, nLength);
// Eat '\n' in the beginning and at the end
while (strInclude.length() > 0 && strInclude[0] == '\n')
strInclude = strInclude.substr(1, static_cast<unsigned int>(-1));
while (strInclude.length() > 0 && strInclude[static_cast<int>(strInclude.length()) - 1] == '\n')
strInclude.setat(strInclude.length() - 1, 0);
if (strInclude.length() > 0 && processed.find(strInclude) == processed.end())
{
processed.insert(strInclude);
GURL urlInclude = m_pDjVuDoc->id_to_url(strInclude);
GP<DataPool> pool = m_pDjVuDoc->request_data(NULL, urlInclude);
GP<ByteStream> stream = pool->get_stream();
GP<IFFByteStream> iff(IFFByteStream::create(stream));
// Check file format
GUTF8String chkid;
if (!iff->get_chunk(chkid) ||
(chkid != "FORM:DJVI" && chkid != "FORM:DJVU" &&
chkid != "FORM:PM44" && chkid != "FORM:BM44"))
{
return;
}
// Find chunk with page info
while (iff->get_chunk(chkid) != 0)
{
GP<ByteStream> chunk_stream = iff->get_bytestream();
if (chkid == "INCL")
{
ReadAnnotations(pInclStream, processed, pAnnoStream);
}
else if (chkid == "FORM:ANNO")
{
pAnnoStream->copy(*chunk_stream);
}
else if (chkid == "ANTa" || chkid == "ANTz")
{
const GP<IFFByteStream> iffout = IFFByteStream::create(pAnnoStream);
iffout->put_chunk(chkid);
iffout->copy(*chunk_stream);
iffout->close_chunk();
}
iff->seek_close_chunk();
}
}
}
inline void ReadAnnotations(
const string& file,
vector<pair<string, vector<float> > >& samples,
const int key_point_count = 72,
const bool warnEmpty = false)
{
ReadAnnotations(file.c_str(), samples, key_point_count, warnEmpty);
}
void DBTreeCtrl::ReadNode(tinyxml2::XMLElement *lvRootNode, wxTreeItemId parent, DBCatalog *cat)
{
wxString str, str2;
long ival;
int level;
tinyxml2::XMLElement *child = lvRootNode->FirstChildElement();
while (child)
{
if (strcmp(child->Name(), "Group") == 0)
{
str = child->Attribute("Name");
wxTreeItemId item = AppendItem(parent, str, 1);
DBItemInfo* item_data = new DBItemInfo;
item_data->type = 1;//group;
SetItemData(item, item_data);
ReadNode(child, item, cat);
}
if (strcmp(child->Name(), "Data") == 0)
{
DBItemInfo* item_data = new DBItemInfo;
str = child->Attribute("Name");
str2 = child->Attribute("Description");
item_data->desc = str + wxT(" ") + str2;
item_data->desc.MakeLower();
item_data->name = str + wxT(" ") + str2;
if (cat) item_data->cat_name = cat->name;
wxTreeItemId item = AppendItem(parent, item_data->name, 0);
item_data->type = 2;//data;
str = child->Attribute("DataURL");
item_data->data_url = str;
str = child->Attribute("SampleID");
str.ToLong(&ival);
item_data->sample_id = ival;
str = child->Attribute("AnnotationURL");
item_data->annotation_url = str;
ReadAnnotations(str, item_data->annotations);
SetItemData(item, item_data);
if (cat) cat->items.push_back(item_data);
}
child = child->NextSiblingElement();
}
}
PageInfo DjVuSource::ReadPageInfo(int nPage, bool bNeedText, bool bNeedAnno)
{
ASSERT(nPage >= 0 && nPage < m_nPageCount);
PageInfo pageInfo;
pageInfo.szPage.cx = 100;
pageInfo.szPage.cy = 100;
pageInfo.nDPI = 100;
pageInfo.bDecoded = true;
GP<ByteStream> pAnnoStream;
if (bNeedAnno)
pAnnoStream = ByteStream::create();
GP<ByteStream> pTextStream;
if (bNeedText)
pTextStream = ByteStream::create();
try
{
// Get raw data from the document and decode only requested chunks
// DjVuFile is not used to ensure that we do not wait for a lock
// to be released and thus do not block the UI thread
GURL url = m_pDjVuDoc->page_to_url(nPage);
GP<DataPool> pool = m_pDjVuDoc->request_data(NULL, url);
GP<ByteStream> stream = pool->get_stream();
GP<IFFByteStream> iff(IFFByteStream::create(stream));
// Check file format
GUTF8String chkid;
if (!iff->get_chunk(chkid) ||
(chkid != "FORM:DJVI" && chkid != "FORM:DJVU" &&
chkid != "FORM:PM44" && chkid != "FORM:BM44"))
{
return pageInfo;
}
bool bHasIW44 = false;
// Find chunk with page info
while (iff->get_chunk(chkid) != 0)
{
GP<ByteStream> chunk_stream = iff->get_bytestream();
if (chkid == "INFO")
{
// Get page dimensions and resolution from info chunk
GP<DjVuInfo> pInfo = DjVuInfo::create();
pInfo->decode(*chunk_stream);
// Check data for consistency
pageInfo.szPage.cx = max(pInfo->width, 0);
pageInfo.szPage.cy = max(pInfo->height, 0);
pageInfo.nInitialRotate = pInfo->orientation;
pageInfo.nDPI = max(pInfo->dpi, 0);
if ((pInfo->orientation & 1) != 0)
swap(pageInfo.szPage.cx, pageInfo.szPage.cy);
}
else if (!bHasIW44 && (chkid == "PM44" || chkid == "BM44"))
{
bHasIW44 = true;
// Get image dimensions and resolution from bitmap chunk
UINT serial = chunk_stream->read8();
UINT slices = chunk_stream->read8();
UINT major = chunk_stream->read8();
UINT minor = chunk_stream->read8();
UINT xhi = chunk_stream->read8();
UINT xlo = chunk_stream->read8();
UINT yhi = chunk_stream->read8();
UINT ylo = chunk_stream->read8();
pageInfo.szPage.cx = (xhi << 8) | xlo;
pageInfo.szPage.cy = (yhi << 8) | ylo;
pageInfo.nDPI = 100;
}
else if (chkid == "TXTa" || chkid == "TXTz")
{
pageInfo.bHasText = true;
if (bNeedText)
{
const GP<IFFByteStream> iffout = IFFByteStream::create(pTextStream);
iffout->put_chunk(chkid);
iffout->copy(*chunk_stream);
iffout->close_chunk();
}
}
else if (bNeedAnno && chkid == "FORM:ANNO")
{
pAnnoStream->copy(*chunk_stream);
}
else if (bNeedAnno && (chkid == "ANTa" || chkid == "ANTz"))
{
const GP<IFFByteStream> iffout = IFFByteStream::create(pAnnoStream);
iffout->put_chunk(chkid);
iffout->copy(*chunk_stream);
iffout->close_chunk();
}
else if (bNeedAnno && chkid == "INCL")
{
set<GUTF8String> processed;
ReadAnnotations(chunk_stream, processed, pAnnoStream);
}
iff->seek_close_chunk();
}
if (bNeedText && pTextStream->tell())
pageInfo.DecodeText(pTextStream);
if (bNeedAnno && pAnnoStream->tell())
pageInfo.DecodeAnno(pAnnoStream);
}
catch (GException&)
{
}
catch (...)
{
if (pApplication != NULL)
pApplication->ReportFatalError();
}
return pageInfo;
}
