added text search functions, fixed the bug when initializing TagName, and some minor...
[SXSI/XMLTree.git] / XMLTree.cpp
index a3fefea..ea12fad 100644 (file)
@@ -1,6 +1,6 @@
 \r
 #include "XMLTree.h"\r
-#include <cstring>\r
+\r
 // functions to convert tag positions to the corresponding tree node and viceversa. \r
 // These are implemented in order to be able to change the tree and Tags representations, \r
 // without affecting the code so much.\r
@@ -51,7 +51,7 @@ void XMLTree::Save(unsigned char *filename)
     Tags->save(fp);\r
 \r
     // stores the texts   \r
-    //Text->Save(fp);\r
+    Text->Save(fp);\r
 \r
     fclose(fp);\r
 \r
@@ -105,7 +105,7 @@ XMLTree *XMLTree::Load(unsigned char *filename, int sample_rate_text)
     XML_Tree->Tags = static_sequence_wvtree::load(fp);\r
 \r
     // loads the texts   \r
-    //XML_Tree->Text->Load(fp,sample_rate_text);\r
+    XML_Tree->Text->Load(fp,sample_rate_text);\r
 \r
     fclose(fp);\r
     \r
@@ -136,9 +136,9 @@ XMLTree::~XMLTree()
     delete Tags;\r
     Tags = NULL;\r
 \r
-    //Text->~TextCollection();\r
-   // delete Text;\r
-   // Text = NULL;\r
+    Text->~TextCollection();\r
+    delete Text;\r
+    Text = NULL;\r
 \r
     initialized = false;\r
     finished = false;\r
@@ -147,18 +147,31 @@ XMLTree::~XMLTree()
 // root(): returns the tree root.\r
 treeNode XMLTree::Root() \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
     return root_node(Par);\r
  }\r
 \r
 // SubtreeSize(x): the number of nodes (and attributes) in the subtree of node x.\r
 int XMLTree::SubtreeSize(treeNode x) \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
     return subtree_size(Par, x);\r
  }\r
 \r
 // SubtreeTags(x,tag): the number of occurrences of tag within the subtree of node x.\r
 int XMLTree::SubtreeTags(treeNode x, TagType tag) \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
+\r
     int s = x + 2*subtree_size(Par, x) - 1;\r
  \r
     return Tags->rank(tag, s) - Tags->rank(tag, node2tagpos(x)-1);\r
@@ -174,12 +187,22 @@ bool XMLTree::IsLeaf(treeNode x)
 // IsAncestor(x,y): returns whether node x is ancestor of node y.\r
 bool XMLTree::IsAncestor(treeNode x, treeNode y) \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
+\r
     return is_ancestor(Par, x, y);\r
  }\r
 \r
 // IsChild(x,y): returns whether node x is parent of node y.\r
 bool XMLTree::IsChild(treeNode x, treeNode y) \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
+\r
     if (!is_ancestor(Par, x, y)) return false;\r
     return depth(Par, x) == (depth(Par, y) + 1);\r
  }\r
@@ -188,18 +211,33 @@ bool XMLTree::IsChild(treeNode x, treeNode y)
 // of Sadakane.\r
 int XMLTree::NumChildren(treeNode x) \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
+\r
     return degree(Par, x);\r
  }\r
 \r
 // ChildNumber(x): returns i if node x is the i-th children of its parent.\r
 int XMLTree::ChildNumber(treeNode x) \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
+\r
     return child_rank(Par, x);\r
  }\r
 \r
 // Depth(x): depth of node x, a simple binary rank on the parentheses sequence.\r
 int XMLTree::Depth(treeNode x) \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
+\r
     return depth(Par, x);\r
  }\r
 \r
@@ -207,6 +245,11 @@ int XMLTree::Depth(treeNode x)
 // nodes (i.e., tags, it disregards the texts in the tree).\r
 int XMLTree::Preorder(treeNode x) \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
+\r
     return preorder_rank(Par, x);\r
  }\r
 \r
@@ -214,12 +257,22 @@ int XMLTree::Preorder(treeNode x)
 // nodes (i.e., tags, it disregards the texts in the tree).\r
 int XMLTree::Postorder(treeNode x) \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
+\r
     return postorder_rank(Par, x);\r
  }\r
 \r
 // Tag(x): returns the tag identifier of node x.\r
 TagType XMLTree::Tag(treeNode x) \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
+\r
     return Tags->access(node2tagpos(x));\r
  }\r
 \r
@@ -227,6 +280,11 @@ TagType XMLTree::Tag(treeNode x)
 // returns {NULLT, NULLT} when there are no texts descending from x.\r
 range XMLTree::DocIds(treeNode x) \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
+\r
     range r;\r
     if (indexing_empty_texts) { // faster, no rank needed\r
        r.min = x;\r
@@ -250,12 +308,22 @@ range XMLTree::DocIds(treeNode x)
 // Parent(x): returns the parent node of node x.\r
 treeNode XMLTree::Parent(treeNode x) \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
+\r
     return parent(Par, x);\r
  }\r
 \r
 // Child(x,i): returns the i-th child of node x, assuming it exists.\r
 treeNode XMLTree::Child(treeNode x, int i) \r
- {\r
+{\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
+\r
     if (i <= OPTD) return naive_child(Par, x, i);\r
     else return child(Par, x, i);\r
  }\r
@@ -263,18 +331,33 @@ treeNode XMLTree::Child(treeNode x, int i)
 // FirstChild(x): returns the first child of node x, assuming it exists. Very fast in BP.\r
 treeNode XMLTree::FirstChild(treeNode x) \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
+\r
     return first_child(Par, x);\r
  }\r
 \r
 // NextSibling(x): returns the next sibling of node x, assuming it exists.\r
 treeNode XMLTree::NextSibling(treeNode x) \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
+\r
     return next_sibling(Par, x);\r
  }\r
 \r
 // PrevSibling(x): returns the previous sibling of node x, assuming it exists.\r
 treeNode XMLTree::PrevSibling(treeNode x) \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
+\r
     return prev_sibling(Par, x);\r
  }\r
 \r
@@ -283,6 +366,11 @@ treeNode XMLTree::PrevSibling(treeNode x)
 // efficiently, just iterating among the children of node x until finding the desired child.\r
 treeNode XMLTree::TaggedChild(treeNode x, int i, TagType tag) \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
+\r
     treeNode child;\r
    \r
     child = first_child(Par, x); // starts at first child of node x\r
@@ -301,6 +389,11 @@ treeNode XMLTree::TaggedChild(treeNode x, int i, TagType tag)
 // the subtree of x. Returns NULLT if there is none.\r
 treeNode XMLTree::TaggedDesc(treeNode x, TagType tag) \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
+\r
     int r, s;\r
     treeNode y;\r
     r = (int) Tags->rank(tag, node2tagpos(x));\r
@@ -315,6 +408,11 @@ treeNode XMLTree::TaggedDesc(treeNode x, TagType tag)
 // ancestor of x. Returns NULLT if there is none.\r
 treeNode XMLTree::TaggedPrec(treeNode x, TagType tag) \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
+    \r
     int r, s;\r
     treeNode node_s, root;\r
     r = (int)Tags->rank(tag, node2tagpos(x)-1);\r
@@ -335,6 +433,11 @@ treeNode XMLTree::TaggedPrec(treeNode x, TagType tag)
 // the subtree of x. Returns NULLT if there is none.\r
 treeNode XMLTree::TaggedFoll(treeNode x, TagType tag) \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
+\r
     int r, s;\r
     r = (int) Tags->rank(tag, node2tagpos(next_sibling(Par, x))-1);\r
     s = (int) Tags->select(tag, r+1);  // select returns -1 in case that there is no r+1-th tag.\r
@@ -347,6 +450,11 @@ treeNode XMLTree::TaggedFoll(treeNode x, TagType tag)
 // Assumes Doc ids start from 0.\r
 DocID XMLTree::PrevText(treeNode x) \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
+\r
     if (x == Root()) return NULLT;\r
     if (indexing_empty_texts)  // faster, no rank needed\r
        return (DocID)x-1;\r
@@ -362,6 +470,11 @@ DocID XMLTree::PrevText(treeNode x)
 // of node x, or NULLT if x is the root node. Assumes Doc ids start from 0.\r
 DocID XMLTree::NextText(treeNode x) \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
+\r
     if (x == Root()) return NULLT;\r
     if (indexing_empty_texts)  // faster, no rank needed\r
        return (DocID)x+2*subtree_size(Par, x)-1;\r
@@ -379,6 +492,11 @@ DocID XMLTree::NextText(treeNode x)
 // ids start from 0.\r
 DocID XMLTree::MyText(treeNode x) \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
+\r
     if (!IsLeaf(x)) return NULLT;\r
     if (indexing_empty_texts) // faster, no rank needed\r
        return (DocID)x;\r
@@ -394,6 +512,11 @@ DocID XMLTree::MyText(treeNode x)
 // all tree nodes and all text nodes. Assumes that the tree root has preorder 1.\r
 int XMLTree::TextXMLId(DocID d) \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
+\r
     if (indexing_empty_texts) \r
        return d + rank_open(Par, d)+1; // +1 because root has preorder 1\r
     else { // slower, needs rank and select\r
@@ -407,6 +530,11 @@ int XMLTree::TextXMLId(DocID d)
 // preorder 0;\r
 int XMLTree::NodeXMLId(treeNode x) \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
+\r
     if (indexing_empty_texts)\r
        return x - 1 + rank_open(Par, x);\r
     else {\r
@@ -419,6 +547,11 @@ int XMLTree::NodeXMLId(treeNode x)
 // ParentNode(d): returns the parent node of document identifier d.\r
 treeNode XMLTree::ParentNode(DocID d) \r
  {\r
+    if (!finished) {\r
+       fprintf(stderr, "Error: data structure has not been constructed properly\n");\r
+       exit(1);\r
+    }\r
+\r
     int s;\r
     if (indexing_empty_texts) s = d;\r
     else s = EBVector->select1(d);\r
@@ -457,6 +590,8 @@ int XMLTree::OpenDocument(bool empty_texts, int sample_rate_text)
        return NULLT;\r
     }\r
     \r
+    TagName = NULL;\r
+\r
     if (!indexing_empty_texts) {\r
        empty_texts_aux = (unsigned int *)malloc(sizeof(unsigned int));\r
        if (!empty_texts_aux) {\r
@@ -465,7 +600,7 @@ int XMLTree::OpenDocument(bool empty_texts, int sample_rate_text)
        }\r
     }\r
     \r
-    //Text = TextCollection::InitTextCollection(sample_rate_text);\r
+    Text = TextCollection::InitTextCollection((unsigned)sample_rate_text);\r
     \r
     return 1;  // indicates success in the initialization of the data structure\r
  }\r
@@ -500,7 +635,7 @@ int XMLTree::CloseDocument()
     Tags = new static_sequence_wvtree((uint *) tags_aux, (uint) npar-1, wtc, bmb, am);\r
 \r
     // makes the text collection static\r
-    //Text->MakeStatic();\r
+    Text->MakeStatic();\r
     \r
     // creates the data structure marking the non-empty texts (just in the case it is necessary)\r
     if (!indexing_empty_texts) \r
@@ -600,8 +735,9 @@ int XMLTree::NewClosingTag(unsigned char *tagname)
        }\r
        \r
        ntagnames++;\r
-       TagName[i] = (unsigned char *)malloc(sizeof(char)*(strlen((const char *)tagname)+1));\r
-       strcpy((char *)TagName[i], (const char *)tagname);\r
+       TagName[i] = (unsigned char *)malloc(sizeof(char)*(strlen((const char *)tagname)+2));\r
+       TagName[i][0] = '/';\r
+       strcpy((char *)&(TagName[i][1]), (const char *)tagname);\r
     } \r
 \r
     tags_aux = (TagType *)realloc(tags_aux, sizeof(TagType)*(npar + 1));\r
@@ -640,7 +776,7 @@ int XMLTree::NewText(unsigned char *s)
        bitset(empty_texts_aux, npar-1);  // marks the non-empty text with a 1 in the bit vector\r
     }\r
     \r
-    //Text->InsertText(s);\r
+    Text->InsertText(s);\r
     \r
     return 1; // success\r
  }\r
@@ -667,7 +803,7 @@ int XMLTree::NewEmptyText()
        \r
        bitclean(empty_texts_aux, npar-1);  // marks the empty text with a 0 in the bit vector\r
     }\r
-   // else Text->InsertText(&c); // we insert the empty text just in case we index all the texts\r
+    else Text->InsertText(&c); // we insert the empty text just in case we index all the texts\r
     \r
     return 1; // success    \r
  }\r
@@ -698,3 +834,5 @@ unsigned char *XMLTree::GetTagName(TagType tagid)
     return s;\r
  }\r
 \r
+\r
+\r