virtual void start_element (
const unsigned long line_number,
const std::string& name,
const dlib::attribute_list& atts
)
{
//cout << "on line " << line_number << " we hit the <" << name << "> tag" << endl;
if( name == xmlLabel )
{
isItLabel = true;
}
// print all the tag's attributes
atts.reset();
while (atts.move_next())
{
//cout << "\tattribute: " << atts.element().key() << " = " << atts.element().value() << endl;
}
}
void BNetParser::start_element(const unsigned long line_number, const std::string& name, const dlib::attribute_list& atts)
{
const char *cname = name.c_str(); // Should be able to just use string == "" according to some guy at StackOverflow
if (strcmp(cname, "cpt") == 0)
{
// start of new node element
currentNode = new ParsedNode();
atts.reset();
atts.move_next();
std::string nodename = atts.element().value();
currentNode->name = nodename;
}
else if (strcmp(cname, "state") == 0)
{
// start of new state element
atts.reset();
atts.move_next();
std::string statename = atts.element().value();
currentNode->states.push_back(statename);
}
else if (strcmp(cname, "parents") == 0)
{
// start of new parents element
waitingForProbs = 0;
// Prepare to receive parents in "characters" function
}
else if (strcmp(cname, "probabilities") == 0)
{
// start of new probabilities element
waitingForProbs = 1;
// Prepare to receive probabilities in "characters" function
}
}
virtual void start_element (
const unsigned long,
const std::string& name,
const dlib::attribute_list& atts) override
{
if(name == "image"){
atts.reset();
if(atts.move_next()){
img_name_.emplace_back(atts.element().value());
location_.emplace_back();
}
}else if(name == "box"){
atts.reset();
dlib::rectangle rect;
long h = 0;
enum {height, left, top, width};
for(size_t index = 0; atts.move_next(); ++index){
switch(index){
case height :{
h = std::stol(atts.element().value()) - 1;
h = std::max<long>(0, h);
break;
}
case left :{
rect.set_left(std::stol(atts.element().value()));
break;
}
case top :{
rect.set_top(std::stol(atts.element().value()));
break;
}
case width:{
rect.set_right(rect.left() + std::stol(atts.element().value()) - 1);
rect.set_right(std::max<long>(0, rect.right()));
break;
}
default:{
break;
}
}
}
rect.set_bottom(rect.top() + h);
location_.back().emplace_back(rect);
}
}
virtual void start_element (
const unsigned long line_number,
const std::string& name,
const dlib::attribute_list& atts
)
{
try
{
if (ts.size() == 0)
{
if (name != "dataset")
{
std::ostringstream sout;
sout << "Invalid XML document. Root tag must be <dataset>. Found <" << name << "> instead.";
throw dlib::error(sout.str());
}
else
{
ts.push_back(name);
return;
}
}
if (name == "box")
{
if (atts.is_in_list("top")) temp_box.rect.top() = sa = atts["top"];
else throw dlib::error("<box> missing required attribute 'top'");
if (atts.is_in_list("left")) temp_box.rect.left() = sa = atts["left"];
else throw dlib::error("<box> missing required attribute 'left'");
if (atts.is_in_list("width")) temp_box.rect.right() = sa = atts["width"];
else throw dlib::error("<box> missing required attribute 'width'");
if (atts.is_in_list("height")) temp_box.rect.bottom() = sa = atts["height"];
else throw dlib::error("<box> missing required attribute 'height'");
if (atts.is_in_list("difficult")) temp_box.difficult = sa = atts["difficult"];
if (atts.is_in_list("truncated")) temp_box.truncated = sa = atts["truncated"];
if (atts.is_in_list("occluded")) temp_box.occluded = sa = atts["occluded"];
temp_box.rect.bottom() += temp_box.rect.top()-1;
temp_box.rect.right() += temp_box.rect.left()-1;
}
else if (name == "part" && ts.back() == "box")
{
point temp;
if (atts.is_in_list("x")) temp.x() = sa = atts["x"];
else throw dlib::error("<part> missing required attribute 'x'");
if (atts.is_in_list("y")) temp.y() = sa = atts["y"];
else throw dlib::error("<part> missing required attribute 'y'");
if (atts.is_in_list("name"))
{
if (temp_box.parts.count(atts["name"])==0)
{
temp_box.parts[atts["name"]] = temp;
}
else
{
throw dlib::error("<part> with name '" + atts["name"] + "' is defined more than one time in a single box.");
}
}
else
{
throw dlib::error("<part> missing required attribute 'name'");
}
}
else if (name == "image")
{
temp_image.boxes.clear();
if (atts.is_in_list("file")) temp_image.filename = atts["file"];
else throw dlib::error("<image> missing required attribute 'file'");
}
ts.push_back(name);
}
catch (error& e)
{
throw dlib::error("Error on line " + cast_to_string(line_number) + ": " + e.what());
}
}
virtual void start_element (
const unsigned long ,
const std::string& name,
const dlib::attribute_list& atts
)
{
if (ts.size() == 0)
{
if (name != "dataset")
{
std::ostringstream sout;
sout << "Invalid XML document. Root tag must be <dataset>. Found <" << name << "> instead.";
throw dlib::error(sout.str());
}
else
{
ts.push_back(name);
return;
}
}
if (name == "box")
{
if (atts.is_in_list("top")) temp_box.rect.top() = sa = atts["top"];
else throw dlib::error("<box> missing required attribute 'top'");
if (atts.is_in_list("left")) temp_box.rect.left() = sa = atts["left"];
else throw dlib::error("<box> missing required attribute 'left'");
if (atts.is_in_list("width")) temp_box.rect.right() = sa = atts["width"];
else throw dlib::error("<box> missing required attribute 'width'");
if (atts.is_in_list("height")) temp_box.rect.bottom() = sa = atts["height"];
else throw dlib::error("<box> missing required attribute 'height'");
if (atts.is_in_list("difficult")) temp_box.difficult = sa = atts["difficult"];
if (atts.is_in_list("truncated")) temp_box.truncated = sa = atts["truncated"];
if (atts.is_in_list("occluded")) temp_box.occluded = sa = atts["occluded"];
temp_box.rect.bottom() += temp_box.rect.top()-1;
temp_box.rect.right() += temp_box.rect.left()-1;
}
else if (name == "head" && ts.back() == "box")
{
if (atts.is_in_list("x")) temp_box.head.x() = sa = atts["x"];
else throw dlib::error("<head> missing required attribute 'x'");
if (atts.is_in_list("y")) temp_box.head.y() = sa = atts["y"];
else throw dlib::error("<head> missing required attribute 'y'");
}
else if (name == "image")
{
temp_image.boxes.clear();
if (atts.is_in_list("file")) temp_image.filename = atts["file"];
else throw dlib::error("<image> missing required attribute 'file'");
}
ts.push_back(name);
}
