X-Git-Url: https://git.ucc.asn.au/?p=ipdf%2Fcode.git;a=blobdiff_plain;f=src%2Fdocument.cpp;h=6038b198501798ed66243bb8794ba9b92cad63a4;hp=71f3ed4103a7b492545bc4c3460208b84a4d72ea;hb=22430ba7a7e9981661efcd1e80d34ba7bb9e71a5;hpb=d69b8de94411bee43edc9e16f33bfa0d5d1d6b3b diff --git a/src/document.cpp b/src/document.cpp index 71f3ed4..6038b19 100644 --- a/src/document.cpp +++ b/src/document.cpp @@ -3,6 +3,10 @@ #include #include +#include "../contrib/pugixml-1.4/src/pugixml.cpp" + +#include "stb_truetype.h" + using namespace IPDF; using namespace std; @@ -93,7 +97,6 @@ void Document::GenBaseQuadtree() m_quadtree.nodes.push_back(QuadTreeNode{QUADTREE_EMPTY, QUADTREE_EMPTY, QUADTREE_EMPTY, QUADTREE_EMPTY, QUADTREE_EMPTY, QTC_UNKNOWN, 0, ObjectCount()}); m_quadtree.root_id = 0; GenQuadChild(0, QTC_TOP_LEFT); - GenQuadParent(0, QTC_BOTTOM_RIGHT); } QuadTreeIndex Document::GenQuadChild(QuadTreeIndex parent, QuadTreeNodeChildren type) @@ -104,8 +107,9 @@ QuadTreeIndex Document::GenQuadChild(QuadTreeIndex parent, QuadTreeNodeChildren m_quadtree.nodes[new_index].object_begin = m_objects.bounds.size(); for (unsigned i = m_quadtree.nodes[parent].object_begin; i < m_quadtree.nodes[parent].object_end; ++i) { - if (ContainedInQuadChild(m_objects.bounds[i], type)) + if (IntersectsQuadChild(m_objects.bounds[i], type)) { + Debug("Adding %s -> %s", m_objects.bounds[i].Str().c_str(), TransformToQuadChild(m_objects.bounds[i], type).Str().c_str()); m_objects.bounds.push_back(TransformToQuadChild(m_objects.bounds[i], type)); m_objects.types.push_back(m_objects.types[i]); m_objects.data_indices.push_back(m_objects.data_indices[i]); @@ -258,111 +262,233 @@ 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) +// Behold my amazing tokenizing abilities +static string & GetToken(const string & d, string & token, unsigned & i, const string & delims = "()[],{}<>;:=") { - using namespace pugi; + token.clear(); + while (i < d.size() && iswspace(d[i])) + { + ++i; + } - xml_document doc_xml; - ifstream input(filename.c_str(), ios_base::in); - xml_parse_result result = doc_xml.load(input); + while (i < d.size()) + { + if (iswspace(d[i]) || strchr(delims.c_str(),d[i]) != NULL) + { + if (token.size() == 0 && !iswspace(d[i])) + { + token += d[i++]; + } + break; + } + token += d[i++]; + } + //Debug("Got token \"%s\"", token.c_str()); + return token; +} + +static void GetXYPair(const string & d, Real & x, Real & y, unsigned & i,const string & delims = "()[],{}<>;:=") +{ + string token(""); + while (GetToken(d, token, i, delims) == ","); + x = strtod(token.c_str(),NULL); + if (GetToken(d, token, i, delims) != ",") + { + Fatal("Expected \",\" seperating x,y pair"); + } + y = strtod(GetToken(d, token, i, delims).c_str(),NULL); +} + +static void TransformXYPair(Real & x, Real & y, const SVGMatrix & transform) +{ + Real x0(x); + x = transform.a * x + transform.c * y + transform.e; + y = transform.b * x0 + transform.d * y + transform.f; +} + +void Document::ParseSVGTransform(const string & s, SVGMatrix & transform) +{ + Debug("Parsing transform %s", s.c_str()); + string token; + string command; + unsigned i = 0; - if (!result) - Fatal("Couldn't load \"%s\" - %s", filename.c_str(), result.description()); - - Debug("Loaded XML - %s", result.description()); + while (i < s.size()) + { + GetToken(s, command, i); + if (command == "," || command == "" || command == ":") + { + if (i < s.size()) + GetToken(s, command, i); + else + return; + } + Debug("Token is \"%s\"", command.c_str()); - input.close(); + SVGMatrix delta = {1,0,0,0,1,0}; + + + assert(GetToken(s,token, i) == "("); + if (command == "translate") + { + GetXYPair(s, delta.e, delta.f, i); + assert(GetToken(s,token, i) == ")"); + } + else if (command == "matrix") + { + GetXYPair(s, delta.a, delta.b,i); + GetXYPair(s, delta.c, delta.d,i); + GetXYPair(s, delta.e, delta.f,i); + assert(GetToken(s,token, i) == ")"); + } + else if (command == "scale") + { + delta.a = (strtod(GetToken(s,token,i).c_str(), NULL)); + GetToken(s, token, i); + if (token == ",") + { + delta.d = (strtod(GetToken(s,token,i).c_str(), NULL)); + assert(GetToken(s, token, i) == ")"); + } + else + { + delta.d = delta.a; + assert(token == ")"); + } + + } + else + { + Warn("Unrecognised transform \"%s\", using identity", command.c_str()); + } + + Debug("Old transform is {%f,%f,%f,%f,%f,%f}", transform.a, transform.b, transform.c, transform.d,transform.e,transform.f); + Debug("Delta transform is {%f,%f,%f,%f,%f,%f}", delta.a, delta.b, delta.c, delta.d,delta.e,delta.f); + + SVGMatrix old(transform); + transform.a = old.a * delta.a + old.c * delta.b; + transform.c = old.a * delta.c + old.c * delta.d; + transform.e = old.a * delta.e + old.c * delta.f + old.e; + + transform.b = old.b * delta.a + old.d * delta.b; + transform.d = old.b * delta.c + old.d * delta.d; + transform.f = old.b * delta.e + old.d * delta.f + old.f; + + Debug("New transform is {%f,%f,%f,%f,%f,%f}", transform.a, transform.b, transform.c, transform.d,transform.e,transform.f); + } +} + +void Document::ParseSVGNode(pugi::xml_node & root, SVGMatrix & parent_transform) +{ + Debug("Parse node <%s>", root.name()); - // Combine all SVG tags into one thing because lazy - for (xml_node svg : doc_xml.children("svg")) - { - Real width = svg.attribute("width").as_float() * bounds.w; - Real height = svg.attribute("width").as_float() * bounds.h; + for (pugi::xml_node child = root.first_child(); child; child = child.next_sibling()) + { + SVGMatrix transform(parent_transform); + pugi::xml_attribute attrib_trans = child.attribute("transform"); + if (!attrib_trans.empty()) + { + ParseSVGTransform(attrib_trans.as_string(), transform); + } - // 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) { + + ParseSVGNode(child, transform); + 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, transform); + } + else if (strcmp(child.name(), "line") == 0) + { + Real x0(child.attribute("x1").as_float()); + Real y0(child.attribute("y1").as_float()); + Real x1(child.attribute("x2").as_float()); + Real y1(child.attribute("y2").as_float()); + TransformXYPair(x0,y0,transform); + TransformXYPair(x1,y1,transform); + 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")); - 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")) + Real x2(coords[0]+coords[2]); + Real y2(coords[1]+coords[3]); + TransformXYPair(coords[0],coords[1],transform); // x, y, transform + TransformXYPair(x2,y2,transform); + coords[2] = x2 - coords[0]; + coords[3] = y2 - coords[1]; + + bool outline = !(child.attribute("fill") && strcmp(child.attribute("fill").as_string(),"none") != 0); + Add(outline?RECT_OUTLINE:RECT_FILLED, Rect(coords[0], coords[1], coords[2], coords[3]),0); + } + 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); + Real y = (cy - r); + TransformXYPair(x,y,transform); + Real w = Real(2)*r*transform.a; // width scales + Real h = Real(2)*r*transform.d; // height scales - Real x = (cx - r)/width + bounds.x; - Real y = (cy - r)/height + bounds.y; - Real w = 2*r/width; - Real h = 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(); + Real y = child.attribute("y").as_float(); + TransformXYPair(x,y,transform); + Debug("Add text \"%s\"", child.child_value()); + AddText(child.child_value(), 0.05, x, y); } } - - //Fatal("Done"); - - - } -// Behold my amazing tokenizing abilities -static string & GetToken(const string & d, string & token, unsigned & i) +/** + * Load an SVG into a rectangle + */ +void Document::LoadSVG(const string & filename, const Rect & bounds) { - token.clear(); - while (i < d.size() && iswspace(d[i])) - { - ++i; - } + using namespace pugi; - while (i < d.size()) - { - if (d[i] == ',' || (isalpha(d[i]) && d[i] != 'e') || iswspace(d[i])) - { - if (token.size() == 0 && !iswspace(d[i])) - { - token += d[i++]; - } - break; - } - token += d[i++]; - } - Debug("Got token \"%s\"", token.c_str()); - return token; + 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(); + // a c e, b d f + SVGMatrix transform = {bounds.w, 0,bounds.x, 0,bounds.h,bounds.y}; + ParseSVGNode(doc_xml, transform); } + // 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 SVGMatrix & transform) { Real x[4] = {0,0,0,0}; Real y[4] = {0,0,0,0}; @@ -378,6 +504,8 @@ void Document::AddPathFromString(const string & d, const Rect & bounds) bool start = false; + static string delims("()[],{}<>;:=LlmMqQzZcC"); + while (i < d.size() && GetToken(d, token, i).size() > 0) { if (isalpha(token[0])) @@ -393,41 +521,39 @@ 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; - assert(GetToken(d,token,i) == ","); - Real dy = strtod(GetToken(d,token,i).c_str(),NULL) / bounds.h; + //Debug("Construct moveto command"); + Real dx = Real(strtod(GetToken(d,token,i,delims).c_str(),NULL)); + assert(GetToken(d,token,i,delims) == ","); + Real dy = Real(strtod(GetToken(d,token,i,delims).c_str(),NULL)); x[0] = (relative) ? x[0] + dx : dx; y[0] = (relative) ? y[0] + dy : dy; - - - Debug("mmoveto %f,%f", Float(x[0]),Float(y[0])); + //Debug("mmoveto %f,%f", Float(x[0]),Float(y[0])); command = (command == "m") ? "l" : "L"; } 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; - assert(GetToken(d,token,i) == ","); - Real dy = strtod(GetToken(d,token,i).c_str(),NULL)/bounds.h; + //Debug("Construct curveto command"); + Real dx = Real(strtod(GetToken(d,token,i,delims).c_str(),NULL)); + assert(GetToken(d,token,i,delims) == ","); + Real dy = Real(strtod(GetToken(d,token,i,delims).c_str(),NULL)); 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; - assert(GetToken(d,token,i) == ","); - dy = strtod(GetToken(d,token,i).c_str(),NULL) / bounds.h; + dx = Real(strtod(GetToken(d,token,i,delims).c_str(),NULL)); + assert(GetToken(d,token,i,delims) == ","); + dy = Real(strtod(GetToken(d,token,i,delims).c_str(),NULL)); 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; - assert(GetToken(d,token,i) == ","); - dy = strtod(GetToken(d,token,i).c_str(),NULL) / bounds.h; + dx = Real(strtod(GetToken(d,token,i,delims).c_str(),NULL)); + assert(GetToken(d,token,i,delims) == ","); + dy = Real(strtod(GetToken(d,token,i,delims).c_str(),NULL)); x[3] = (relative) ? x[0] + dx : dx; y[3] = (relative) ? y[0] + dy : dy; } @@ -435,48 +561,59 @@ void Document::AddPathFromString(const string & d, const Rect & bounds) { x[3] = x[2]; y[3] = y[2]; + Real old_x1(x[1]), old_y1(y[1]); + x[1] = x[0] + Real(2) * (old_x1 - x[0])/ Real(3); + y[1] = y[0] + Real(2) * (old_y1 - y[0])/ Real(3); + x[2] = x[3] + Real(2) * (old_x1 - x[3])/ Real(3); + y[2] = y[3] + Real(2) * (old_y1 - y[3])/ Real(3); } + Real x3(x[3]); + Real y3(y[3]); + for (int j = 0; j < 4; ++j) + TransformXYPair(x[j],y[j], transform); + unsigned index = AddBezierData(Bezier(x[0],y[0],x[1],y[1],x[2],y[2],x[3],y[3])); Add(BEZIER,Rect(0,0,1,1),index); - Debug("[%u] curveto %f,%f %f,%f %f,%f", index, Float(x[1]),Float(y[1]),Float(x[2]),Float(y[2]),Float(x[3]),Float(y[3])); + //Debug("[%u] curveto %f,%f %f,%f %f,%f", index, Float(x[1]),Float(y[1]),Float(x[2]),Float(y[2]),Float(x[3]),Float(y[3])); - x[0] = x[3]; - y[0] = y[3]; + x[0] = x3; + y[0] = y3; } else if (command == "l" || command == "L") { - Debug("Construct lineto command"); + Debug("Construct lineto command, relative %d", relative); - Real dx = 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 dx = Real(strtod(GetToken(d,token,i,delims).c_str(),NULL)); + assert(GetToken(d,token,i,delims) == ","); + Real dy = Real(strtod(GetToken(d,token,i,delims).c_str(),NULL)); - x[1] = (relative) ? x0 + dx : dx; - y[1] = (relative) ? y0 + dy : dy; + x[1] = (relative) ? x[0] + dx : dx; + y[1] = (relative) ? y[0] + dy : dy; - x[2] = x[1]; - y[2] = y[1]; + Real x1(x[1]); + Real y1(y[1]); - x[3] = x[1]; - y[3] = y[1]; + TransformXYPair(x[0],y[0],transform); + TransformXYPair(x[1],y[1],transform); - unsigned index = AddBezierData(Bezier(x[0],y[0],x[1],y[1],x[2],y[2],x[3],y[3])); + + unsigned index = AddBezierData(Bezier(x[0],y[0],x[1],y[1],x[1],y[1],x[1],y[1])); Add(BEZIER,Rect(0,0,1,1),index); - Debug("[%u] lineto %f,%f %f,%f", index, Float(x[0]),Float(y[0]),Float(x[1]),Float(y[1])); + //Debug("[%u] lineto %f,%f %f,%f", index, Float(x[0]),Float(y[0]),Float(x[1]),Float(y[1])); - x[0] = x[3]; - y[0] = y[3]; + x[0] = x1; + y[0] = y1; } else if (command == "z" || command == "Z") { - Debug("Construct returnto command"); + //Debug("Construct returnto command"); x[1] = x0; y[1] = y0; x[2] = x0; @@ -484,13 +621,18 @@ void Document::AddPathFromString(const string & d, const Rect & bounds) x[3] = x0; y[3] = y0; + Real x3(x[3]); + Real y3(y[3]); + for (int j = 0; j < 4; ++j) + TransformXYPair(x[j],y[j], transform); + unsigned index = AddBezierData(Bezier(x[0],y[0],x[1],y[1],x[2],y[2],x[3],y[3])); Add(BEZIER,Rect(0,0,1,1),index); - Debug("[%u] returnto %f,%f %f,%f", index, Float(x[0]),Float(y[0]),Float(x[1]),Float(y[1])); + //Debug("[%u] returnto %f,%f %f,%f", index, Float(x[0]),Float(y[0]),Float(x[1]),Float(y[1])); - x[0] = x[3]; - y[0] = y[3]; + x[0] = x3; + y[0] = y3; command = "m"; } else @@ -508,3 +650,114 @@ void Document::AddPathFromString(const string & d, const Rect & bounds) prev_i = i; } } + +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); + int ascent = 0, descent = 0, line_gap = 0; + stbtt_GetFontVMetrics(&m_font, &ascent, &descent, &line_gap); + Real y_advance = Real(font_scale) * Real(ascent - descent + line_gap); + for (unsigned i = 0; i < text.size(); ++i) + { + if (text[i] == '\n') + { + y += y_advance; + x = x0; + } + if (!isprint(text[i])) + continue; + + int advance_width = 0, left_side_bearing = 0, kerning = 0; + stbtt_GetCodepointHMetrics(&m_font, text[i], &advance_width, &left_side_bearing); + if (i > 1) + { + kerning = stbtt_GetCodepointKernAdvance(&m_font, text[i-1], text[i]); + } + x += Real(font_scale) * Real(left_side_bearing + kerning); + AddFontGlyphAtPoint(&m_font, text[i], font_scale, x, y); + x += Real(font_scale) * Real(advance_width); + } +} + +void Document::AddFontGlyphAtPoint(stbtt_fontinfo *font, int character, Real scale, Real x, Real y) +{ + int glyph_index = stbtt_FindGlyphIndex(font, character); + + // Check if there is actully a glyph to render. + if (stbtt_IsGlyphEmpty(font, glyph_index)) + { + return; + } + + stbtt_vertex *instructions; + int num_instructions = stbtt_GetGlyphShape(font, glyph_index, &instructions); + + Real current_x(0), current_y(0); + + for (int i = 0; i < num_instructions; ++i) + { + // TTF uses 16-bit signed ints for coordinates: + // with the y-axis inverted compared to us. + // Convert and scale any data. + Real inst_x = Real(instructions[i].x)*scale; + Real inst_y = Real(instructions[i].y)*-scale; + Real inst_cx = Real(instructions[i].cx)*scale; + Real inst_cy = Real(instructions[i].cy)*-scale; + Real old_x(current_x), old_y(current_y); + current_x = inst_x; + current_y = inst_y; + unsigned bezier_index; + switch(instructions[i].type) + { + // Move To + case STBTT_vmove: + break; + // Line To + case STBTT_vline: + bezier_index = AddBezierData(Bezier(old_x + x, old_y + y, old_x + x, old_y + y, current_x + x, current_y + y, current_x + x, current_y + y)); + Add(BEZIER,Rect(0,0,1,1),bezier_index); + break; + // Quadratic Bezier To: + case STBTT_vcurve: + // Quadratic -> Cubic: + // - Endpoints are the same. + // - cubic1 = quad0+(2/3)*(quad1-quad0) + // - cubic2 = quad2+(2/3)*(quad1-quad2) + bezier_index = AddBezierData(Bezier(old_x + x, old_y + y, old_x + Real(2)*(inst_cx-old_x)/Real(3) + x, old_y + Real(2)*(inst_cy-old_y)/Real(3) + y, + current_x + Real(2)*(inst_cx-current_x)/Real(3) + x, current_y + Real(2)*(inst_cy-current_y)/Real(3) + y, current_x + x, current_y + y)); + Add(BEZIER,Rect(0,0,1,1),bezier_index); + break; + } + } + + stbtt_FreeShape(font, instructions); +}