X-Git-Url: https://git.ucc.asn.au/?p=ipdf%2Fcode.git;a=blobdiff_plain;f=src%2Fdocument.cpp;h=f23b4068960f7652474c728cb2ae13cae7234d8a;hp=a4b20d08f091ecaa4585377343a9f97f7fccf9af;hb=4166b2c124b0bc652e9f8e5245488cdd2b86ebf0;hpb=46fa3e99a49b47c18836d44d806971192907c304 diff --git a/src/document.cpp b/src/document.cpp index a4b20d0..f23b406 100644 --- a/src/document.cpp +++ b/src/document.cpp @@ -3,6 +3,8 @@ #include #include +#include "../contrib/pugixml-1.4/src/pugixml.cpp" + #include "stb_truetype.h" using namespace IPDF; @@ -260,80 +262,99 @@ bool Document::operator==(const Document & equ) const } -#include "../contrib/pugixml-1.4/src/pugixml.hpp" -#include "../contrib/pugixml-1.4/src/pugixml.cpp" -/** - * Load an SVG into a rectangle - */ -void Document::LoadSVG(const string & filename, const Rect & bounds) -{ - using namespace pugi; - - xml_document doc_xml; - ifstream input(filename.c_str(), ios_base::in); - xml_parse_result result = doc_xml.load(input); - - if (!result) - Fatal("Couldn't load \"%s\" - %s", filename.c_str(), result.description()); - - Debug("Loaded XML - %s", result.description()); - - input.close(); - // Combine all SVG tags into one thing because lazy - for (xml_node svg : doc_xml.children("svg")) +void Document::ParseSVGNode(pugi::xml_node & root, const Rect & bounds, Real & width, Real & height) +{ + Debug("Parse node <%s>", root.name()); + pugi::xml_attribute attrib_w = root.attribute("width"); + pugi::xml_attribute attrib_h = root.attribute("height"); + if (!attrib_w.empty()) + width = attrib_w.as_float() * bounds.w; + if (!attrib_h.empty()) + height = attrib_h.as_float() * bounds.h; + + for (pugi::xml_node child = root.first_child(); child; child = child.next_sibling()) { - Real width = svg.attribute("width").as_float() * bounds.w; - Real height = svg.attribute("width").as_float() * bounds.h; - + - // Rectangles - Real coords[4]; - const char * attrib_names[] = {"x", "y", "width", "height"}; - for (pugi::xml_node rect : svg.children("rect")) + if (strcmp(child.name(), "svg") == 0 || strcmp(child.name(),"g") == 0 + || strcmp(child.name(), "group") == 0) + { + //TODO: Handle translates etc here + ParseSVGNode(child, bounds, width, height); + continue; + } + else if (strcmp(child.name(), "path") == 0) + { + string d = child.attribute("d").as_string(); + Debug("Path data attribute is \"%s\"", d.c_str()); + ParseSVGPathData(d, Rect(bounds.x,bounds.y,width,height)); + } + else if (strcmp(child.name(), "line") == 0) { + Real x0(child.attribute("x1").as_float()/width + bounds.x); + Real y0(child.attribute("y1").as_float()/height + bounds.y); + Real x1(child.attribute("x2").as_float()/width + bounds.x); + Real y1(child.attribute("y2").as_float()/height + bounds.y); + unsigned index = AddBezierData(Bezier(x0,y0,x1,y1,x1,y1,x1,y1)); + Add(BEZIER, Rect(0,0,1,1), index); + } + else if (strcmp(child.name(), "rect") == 0) + { + Real coords[4]; + const char * attrib_names[] = {"x", "y", "width", "height"}; for (size_t i = 0; i < 4; ++i) - coords[i] = rect.attribute(attrib_names[i]).as_float(); + coords[i] = child.attribute(attrib_names[i]).as_float(); - bool outline = !(rect.attribute("fill")); + bool outline = !(child.attribute("fill")); Add(outline?RECT_OUTLINE:RECT_FILLED, Rect(coords[0]/width + bounds.x, coords[1]/height + bounds.y, coords[2]/width, coords[3]/height),0); - Debug("Added rectangle"); - } - - // Circles - for (pugi::xml_node circle : svg.children("circle")) + } + else if (strcmp(child.name(), "circle") == 0) { - Real cx = circle.attribute("cx").as_float(); - Real cy = circle.attribute("cy").as_float(); - Real r = circle.attribute("r").as_float(); + Real cx = child.attribute("cx").as_float(); + Real cy = child.attribute("cy").as_float(); + Real r = child.attribute("r").as_float(); Real x = (cx - r)/width + bounds.x; Real y = (cy - r)/height + bounds.y; - Real w = 2*r/width; - Real h = 2*r/height; + Real w = Real(2)*r/width; + Real h = Real(2)*r/height; Rect rect(x,y,w,h); Add(CIRCLE_FILLED, rect,0); - Debug("Added Circle %s", rect.Str().c_str()); - - } - - // paths - for (pugi::xml_node path : svg.children("path")) + Debug("Added Circle %s", rect.Str().c_str()); + } + else if (strcmp(child.name(), "text") == 0) { - - string d = path.attribute("d").as_string(); - Debug("Path data attribute is \"%s\"", d.c_str()); - AddPathFromString(d, Rect(bounds.x,bounds.y,width,height)); - + Real x = child.attribute("x").as_float()/width + bounds.x; + Real y = child.attribute("y").as_float()/height + bounds.y; + Debug("Add text \"%s\"", child.child_value()); + AddText(child.child_value(), 0.05, x, y); } } +} + +/** + * Load an SVG into a rectangle + */ +void Document::LoadSVG(const string & filename, const Rect & bounds) +{ + using namespace pugi; - //Fatal("Done"); + xml_document doc_xml; + ifstream input(filename.c_str(), ios_base::in); + xml_parse_result result = doc_xml.load(input); + if (!result) + Fatal("Couldn't load \"%s\" - %s", filename.c_str(), result.description()); + + Debug("Loaded XML - %s", result.description()); - + input.close(); + Real width(1); + Real height(1); + ParseSVGNode(doc_xml, bounds,width,height); } // Behold my amazing tokenizing abilities @@ -357,14 +378,14 @@ static string & GetToken(const string & d, string & token, unsigned & i) } token += d[i++]; } - Debug("Got token \"%s\"", token.c_str()); + //Debug("Got token \"%s\"", token.c_str()); return token; } // Fear the wrath of the tokenizing svg data // Seriously this isn't really very DOM-like at all is it? -void Document::AddPathFromString(const string & d, const Rect & bounds) +void Document::ParseSVGPathData(const string & d, const Rect & bounds) { Real x[4] = {0,0,0,0}; Real y[4] = {0,0,0,0}; @@ -396,9 +417,9 @@ void Document::AddPathFromString(const string & d, const Rect & bounds) if (command == "m" || command == "M") { Debug("Construct moveto command"); - Real dx = strtod(GetToken(d,token,i).c_str(),NULL) / bounds.w; + Real dx = Real(strtod(GetToken(d,token,i).c_str(),NULL)) / bounds.w; assert(GetToken(d,token,i) == ","); - Real dy = strtod(GetToken(d,token,i).c_str(),NULL) / bounds.h; + Real dy = Real(strtod(GetToken(d,token,i).c_str(),NULL)) / bounds.h; x[0] = (relative) ? x[0] + dx : dx; y[0] = (relative) ? y[0] + dy : dy; @@ -411,25 +432,25 @@ void Document::AddPathFromString(const string & d, const Rect & bounds) else if (command == "c" || command == "C" || command == "q" || command == "Q") { Debug("Construct curveto command"); - Real dx = strtod(GetToken(d,token,i).c_str(),NULL)/bounds.w; + Real dx = Real(strtod(GetToken(d,token,i).c_str(),NULL))/bounds.w; assert(GetToken(d,token,i) == ","); - Real dy = strtod(GetToken(d,token,i).c_str(),NULL)/bounds.h; + Real dy = Real(strtod(GetToken(d,token,i).c_str(),NULL))/bounds.h; x[1] = (relative) ? x[0] + dx : dx; y[1] = (relative) ? y[0] + dy : dy; - dx = strtod(GetToken(d,token,i).c_str(),NULL) / bounds.w; + dx = Real(strtod(GetToken(d,token,i).c_str(),NULL)) / bounds.w; assert(GetToken(d,token,i) == ","); - dy = strtod(GetToken(d,token,i).c_str(),NULL) / bounds.h; + dy = Real(strtod(GetToken(d,token,i).c_str(),NULL)) / bounds.h; x[2] = (relative) ? x[0] + dx : dx; y[2] = (relative) ? y[0] + dy : dy; if (command != "q" && command != "Q") { - dx = strtod(GetToken(d,token,i).c_str(),NULL) / bounds.w; + dx = Real(strtod(GetToken(d,token,i).c_str(),NULL)) / bounds.w; assert(GetToken(d,token,i) == ","); - dy = strtod(GetToken(d,token,i).c_str(),NULL) / bounds.h; + dy = Real(strtod(GetToken(d,token,i).c_str(),NULL)) / bounds.h; x[3] = (relative) ? x[0] + dx : dx; y[3] = (relative) ? y[0] + dy : dy; } @@ -459,9 +480,9 @@ void Document::AddPathFromString(const string & d, const Rect & bounds) { Debug("Construct lineto command"); - Real dx = strtod(GetToken(d,token,i).c_str(),NULL) / bounds.w; + Real dx = Real(strtod(GetToken(d,token,i).c_str(),NULL)) / bounds.w; assert(GetToken(d,token,i) == ","); - Real dy = strtod(GetToken(d,token,i).c_str(),NULL) / bounds.h; + Real dy = Real(strtod(GetToken(d,token,i).c_str(),NULL)) / bounds.h; x[1] = (relative) ? x[0] + dx : dx; y[1] = (relative) ? y[0] + dy : dy; @@ -516,6 +537,53 @@ void Document::AddPathFromString(const string & d, const Rect & bounds) } } +void Document::SetFont(const string & font_filename) +{ + if (m_font_data != NULL) + { + free(m_font_data); + } + + FILE *font_file = fopen("DejaVuSansMono.ttf", "rb"); + fseek(font_file, 0, SEEK_END); + size_t font_file_size = ftell(font_file); + fseek(font_file, 0, SEEK_SET); + m_font_data = (unsigned char*)malloc(font_file_size); + size_t read = fread(m_font_data, 1, font_file_size, font_file); + if (read != font_file_size) + { + Fatal("Failed to read font data from \"%s\" - Read %u bytes expected %u - %s", font_filename.c_str(), read, font_file_size, strerror(errno)); + } + fclose(font_file); + stbtt_InitFont(&m_font, m_font_data, 0); +} + +void Document::AddText(const string & text, Real scale, Real x, Real y) +{ + if (m_font_data == NULL) + { + Warn("No font loaded"); + return; + } + + float font_scale = stbtt_ScaleForPixelHeight(&m_font, scale); + Real x0(x); + //Real y0(y); + for (unsigned i = 0; i < text.size(); ++i) + { + if (text[i] == '\n') + { + y += 0.5*scale; + x = x0; + } + if (!isprint(text[i])) + continue; + + AddFontGlyphAtPoint(&m_font, text[i], font_scale, x, y); + x += 0.5*scale; + } +} + void Document::AddFontGlyphAtPoint(stbtt_fontinfo *font, int character, Real scale, Real x, Real y) { int glyph_index = stbtt_FindGlyphIndex(font, character);