X-Git-Url: http://git.nguyen.vg/gitweb/?a=blobdiff_plain;f=xml-tree-inc.hpp;h=18c91e3dad2564b2cd5b06cfd8158d107ff1ad37;hb=HEAD;hp=19745ed49d96f158af235592dc3692fd4e849677;hpb=9775b1833487a525901cf968d91a9e7f193395c5;p=SXSI%2FXMLTree.git

diff --git a/xml-tree-inc.hpp b/xml-tree-inc.hpp
index 19745ed..18c91e3 100644
--- a/xml-tree-inc.hpp
+++ b/xml-tree-inc.hpp
@@ -41,15 +41,12 @@ inline uint32_t xml_tree::subtree_elements(xml_tree::node_t x) const
 
   int32_t size = bp_subtree_size(par, x) - 1;
   if (size <= 0) return 0;
-  size -= subtree_tags(x, xml_tree::PCDATA_OPEN_TAG_ID);
-  size -= subtree_tags(x, xml_tree::ATTRIBUTE_OPEN_TAG_ID);
-  size -= subtree_tags(x, xml_tree::ATTRIBUTE_DATA_OPEN_TAG_ID);
-  if (size < 3) return (uint32_t) size;
-  std::unordered_set<xml_tree::tag_t>::iterator it;
-  for(it = this->attribute_ids->begin();
-      it != this->attribute_ids->end();
-      ++it)
-    size -= subtree_tags(x, *it);
+  uint32_t num_texts = subtree_tags(x, xml_tree::PCDATA_OPEN_TAG_ID);
+  uint32_t num_atts = subtree_tags(x, xml_tree::ATTRIBUTE_OPEN_TAG_ID);
+  uint32_t num_att_data = subtree_tags(x, xml_tree::ATTRIBUTE_DATA_OPEN_TAG_ID);
+  size -= num_texts;
+  size -= num_atts;
+  size -= 2*num_att_data;
   return (uint32_t) size;
 
 }
@@ -167,7 +164,7 @@ inline xml_tree::node_t xml_tree::tagged_next_close(xml_tree::node_t x,
 					      xml_tree::tag_t label) const
 {
   xml_tree::node_t i=x;
-  for(xml_tree::node_t i = x+1; i < std::min(x+100, this->par->n); i++)
+  for(i = x+1; i < std::min(x+100, this->par->n); i++)
     if (tag(i) == label)
       return i;