X-Git-Url: https://git.ucc.asn.au/?p=ipdf%2Fcode.git;a=blobdiff_plain;f=src%2Fdocument.cpp;h=310562987cccae3483f584b47ebff3307505fc19;hp=76a534903b05accbd92df9f4f3adcbcea6c1a72b;hb=ac518ad085eb5bc84647e4aebe0c5ff06640cd0d;hpb=2db1211c9a24a0cd22da1d3ddcddf564f814fc0a diff --git a/src/document.cpp b/src/document.cpp index 76a5349..3105629 100644 --- a/src/document.cpp +++ b/src/document.cpp @@ -1,6 +1,11 @@ #include "document.h" #include "bezier.h" #include +#include + +#include "../contrib/pugixml-1.4/src/pugixml.cpp" + +#include "stb_truetype.h" using namespace IPDF; using namespace std; @@ -91,6 +96,79 @@ 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); +} + +QuadTreeIndex Document::GenQuadChild(QuadTreeIndex parent, QuadTreeNodeChildren type) +{ + QuadTreeIndex new_index = m_quadtree.nodes.size(); + m_quadtree.nodes.push_back(QuadTreeNode{QUADTREE_EMPTY, QUADTREE_EMPTY, QUADTREE_EMPTY, QUADTREE_EMPTY, parent, type, 0, 0}); + + 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)) + { + 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]); + m_count++; + } + } + m_quadtree.nodes[new_index].object_end = m_objects.bounds.size(); + switch (type) + { + case QTC_TOP_LEFT: + m_quadtree.nodes[parent].top_left = new_index; + break; + case QTC_TOP_RIGHT: + m_quadtree.nodes[parent].top_right = new_index; + break; + case QTC_BOTTOM_LEFT: + m_quadtree.nodes[parent].bottom_left = new_index; + break; + case QTC_BOTTOM_RIGHT: + m_quadtree.nodes[parent].bottom_right = new_index; + break; + default: + Fatal("Tried to add a QuadTree child of invalid type!"); + } + return new_index; +} + +// Reparent a quadtree node, making it the "type" child of a new node. +QuadTreeIndex Document::GenQuadParent(QuadTreeIndex child, QuadTreeNodeChildren type) +{ + QuadTreeIndex new_index = m_quadtree.nodes.size(); + m_quadtree.nodes.push_back(QuadTreeNode{QUADTREE_EMPTY, QUADTREE_EMPTY, QUADTREE_EMPTY, QUADTREE_EMPTY, -1, QTC_UNKNOWN, 0, 0}); + + m_quadtree.nodes[new_index].object_begin = m_objects.bounds.size(); + for (unsigned i = m_quadtree.nodes[child].object_begin; i < m_quadtree.nodes[child].object_end; ++i) + { + m_objects.bounds.push_back(TransformFromQuadChild(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]); + m_count++; + } + m_quadtree.nodes[new_index].object_end = m_objects.bounds.size(); + switch (type) + { + case QTC_TOP_LEFT: + m_quadtree.nodes[new_index].top_left = child; + break; + case QTC_TOP_RIGHT: + m_quadtree.nodes[new_index].top_right = child; + break; + case QTC_BOTTOM_LEFT: + m_quadtree.nodes[new_index].bottom_left = child; + break; + case QTC_BOTTOM_RIGHT: + m_quadtree.nodes[new_index].bottom_right = child; + break; + default: + Fatal("Tried to add a QuadTree child of invalid type!"); + } + return new_index; } #endif @@ -181,3 +259,471 @@ bool Document::operator==(const Document & equ) const && memcmp(m_objects.data_indices.data(), equ.m_objects.data_indices.data(), ObjectCount() * sizeof(unsigned)) == 0 && memcmp(m_objects.beziers.data(), equ.m_objects.beziers.data(), m_objects.beziers.size() * sizeof(Bezier)) == 0); } + + + +// Behold my amazing tokenizing abilities +static string & GetToken(const string & d, string & token, unsigned & i, const string & delims = "()[],{}<>;:=") +{ + token.clear(); + while (i < d.size() && iswspace(d[i])) + { + ++i; + } + + 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(""); + x = strtod(GetToken(d, token, i, delims).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, 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; + GetToken(s, command, i); + Debug("Token is \"%s\"", command.c_str()); + + SVGMatrix delta = {1,0,0,1,0,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.b = (strtod(token.c_str(), NULL)); + } + else + { + delta.b = delta.a; + } + + } + 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()); + + + 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); + } + + 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] = child.attribute(attrib_names[i]).as_float(); + + 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 = 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 + + + Rect rect(x,y,w,h); + Add(CIRCLE_FILLED, rect,0); + Debug("Added Circle %s", rect.Str().c_str()); + } + else if (strcmp(child.name(), "text") == 0) + { + 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); + } + } +} + +/** + * 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(); + + SVGMatrix transform = {bounds.w, 0,bounds.x, bounds.h,0,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::ParseSVGPathData(const string & d, const SVGMatrix & transform) +{ + Real x[4] = {0,0,0,0}; + Real y[4] = {0,0,0,0}; + + string token(""); + string command("m"); + + Real x0(0); + Real y0(0); + + unsigned i = 0; + unsigned prev_i = 0; + + bool start = false; + + static string delims("()[],{}<>;:=LlmMqQzZcC"); + + while (i < d.size() && GetToken(d, token, i).size() > 0) + { + if (isalpha(token[0])) + command = token; + else + { + i = prev_i; // hax + if(command == "") + command = "L"; + } + + bool relative = islower(command[0]); + + if (command == "m" || command == "M") + { + //Debug("Construct moveto command"); + Real dx = Real(strtod(GetToken(d,token,i,delims).c_str(),NULL)) * transform.a; + assert(GetToken(d,token,i,delims) == ","); + Real dy = Real(strtod(GetToken(d,token,i,delims).c_str(),NULL)) * transform.d; + + x[0] = (relative) ? x[0] + dx : dx + transform.e; + y[0] = (relative) ? y[0] + dy : dy + transform.f; + + + + //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 = Real(strtod(GetToken(d,token,i,delims).c_str(),NULL)) * transform.a; + assert(GetToken(d,token,i,delims) == ","); + Real dy = Real(strtod(GetToken(d,token,i,delims).c_str(),NULL))*transform.d; + + x[1] = (relative) ? x[0] + dx : dx + transform.e; + y[1] = (relative) ? y[0] + dy : dy + transform.f; + + dx = Real(strtod(GetToken(d,token,i,delims).c_str(),NULL)) *transform.a; + assert(GetToken(d,token,i,delims) == ","); + dy = Real(strtod(GetToken(d,token,i,delims).c_str(),NULL)) *transform.d; + + x[2] = (relative) ? x[0] + dx : dx + transform.e; + y[2] = (relative) ? y[0] + dy : dy + transform.f; + + if (command != "q" && command != "Q") + { + dx = Real(strtod(GetToken(d,token,i,delims).c_str(),NULL)) *transform.a; + assert(GetToken(d,token,i,delims) == ","); + dy = Real(strtod(GetToken(d,token,i,delims).c_str(),NULL)) *transform.d; + x[3] = (relative) ? x[0] + dx : dx + transform.e; + y[3] = (relative) ? y[0] + dy : dy + transform.f; + } + else + { + 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); + } + + 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])); + + x[0] = x[3]; + y[0] = y[3]; + + + } + else if (command == "l" || command == "L") + { + //Debug("Construct lineto command"); + + Real dx = Real(strtod(GetToken(d,token,i,delims).c_str(),NULL)) *transform.a; + assert(GetToken(d,token,i,delims) == ","); + Real dy = Real(strtod(GetToken(d,token,i,delims).c_str(),NULL)) *transform.d; + + x[1] = (relative) ? x[0] + dx : dx + transform.e; + y[1] = (relative) ? y[0] + dy : dy + transform.f; + + x[2] = x[1]; + y[2] = y[1]; + + x[3] = x[1]; + y[3] = y[1]; + + 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] 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]; + + } + else if (command == "z" || command == "Z") + { + //Debug("Construct returnto command"); + x[1] = x0; + y[1] = y0; + x[2] = x0; + y[2] = y0; + x[3] = x0; + y[3] = y0; + + 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])); + + x[0] = x[3]; + y[0] = y[3]; + command = "m"; + } + else + { + Warn("Unrecognised command \"%s\", set to \"m\"", command.c_str()); + command = "m"; + } + + if (!start) + { + x0 = x[0]; + y0 = y[0]; + start = true; + } + 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); + 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); + + // 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); +}