1 /**************************************
4 * A Test Ocaml Driver which calls the C++ methods and
5 * adds a C wrapper interface with OCaml code.
11 /* OCaml memory managment */
13 #include <caml/mlvalues.h>
14 #include <caml/alloc.h>
15 #include <caml/memory.h>
16 #include <caml/callback.h>
17 #include <caml/fail.h>
20 #include "TextCollection/TextCollection.h"
21 #include "XMLDocShredder.h"
25 #define CAMLRAISECPP(e) (caml_failwith( ((e).what())))
26 #define NOT_IMPLEMENTED(s) (caml_failwith(s))
27 #define XMLTREE(x) ((XMLTree *)(x))
28 #define TEXTCOLLECTION(x)
29 #define TREENODEVAL(i) ((treeNode) (Int_val(i)))
31 extern "C" CAMLprim value caml_call_shredder_uri(value uri){
34 char *fn = String_val(uri);
36 XMLDocShredder shredder(fn);
37 shredder.processStartDocument(fn);
39 shredder.processEndDocument();
40 doc = (value) shredder.storageIfc_->returnDocument();
44 catch (const std::exception& e){
50 extern "C" CAMLprim value caml_call_shredder_string(value data){
53 unsigned int ln = string_length(data);
54 unsigned char *fn = (unsigned char*) String_val(data);
57 XMLDocShredder shredder(fn,ln);
58 shredder.processStartDocument("");
60 shredder.processEndDocument();
61 doc = (value) shredder.storageIfc_->returnDocument();
65 catch (const std::exception& e) {
70 void traversal_rec(XMLTree* tree, treeNode id){
74 int tag = tree->Tag(id);
76 tid = tree->PrevText(id);
77 char * data = (char *) (tree->getTextCollection())->GetText(tid);
78 if (tree->IsLeaf(id)){
79 tid = tree->MyText(id);
81 data = (char*) (tree->getTextCollection())->GetText(tid);
84 if (tree->NextSibling(id) == NULLT){
85 tid = tree->NextText(id);
86 data = (char*) (tree->getTextCollection())->GetText(tid);
89 traversal_rec(tree,tree->FirstChild(id));
90 traversal_rec(tree,tree->NextSibling(id));
94 extern "C" CAMLprim value caml_cpp_traversal(value tree){
96 traversal_rec(XMLTREE(tree),XMLTREE(tree)->Root());
101 extern "C" CAMLprim value caml_text_collection_get_text(value tc, value id){
104 const char* txt = (const char*) ((TextCollection*) tc)->GetText((DocID) Int_val(id));
105 CAMLreturn (caml_copy_string(txt));
107 extern "C" CAMLprim value caml_text_collection_empty_text(value tc,value id){
109 CAMLreturn ( Val_int(((TextCollection*) tc)->EmptyText((DocID) Int_val(id))));
112 extern "C" CAMLprim value caml_text_collection_is_contains(value tc,value str){
114 uchar * cstr = (uchar *) String_val(str);
115 CAMLreturn ( Val_bool((int) ((TextCollection*) tc)->IsContains(cstr)));
118 extern "C" CAMLprim value caml_text_collection_count_contains(value tc,value str){
120 uchar * cstr = (uchar *) String_val(str);
121 CAMLreturn ( Val_int(((TextCollection*) tc)->CountContains(cstr)));
125 extern "C" CAMLprim value caml_text_collection_contains(value tc,value str){
127 CAMLlocal1(resarray);
128 uchar * cstr = (uchar *) String_val(str);
129 std::vector<DocID> results;
130 results = ((TextCollection*) tc)->Contains(cstr);
132 resarray = caml_alloc_tuple(results.size());
134 for (int i=0; i<results.size();i++){
135 caml_initialize(&Field(resarray,i),Val_int(results[i]));
137 CAMLreturn (resarray);
141 extern "C" CAMLprim value caml_xml_tree_root(value tree){
143 CAMLreturn (TREENODEVAL(XMLTREE(tree)->Root()));
145 extern "C" CAMLprim value caml_xml_tree_text_collection(value tree){
147 CAMLreturn((value) XMLTREE(tree)->getTextCollection());
149 extern "C" CAMLprim value caml_xml_tree_parent(value tree, value id){
151 CAMLreturn(Val_int (XMLTREE(tree)->Parent(TREENODEVAL(id))));
153 extern "C" CAMLprim value caml_xml_tree_parent_doc(value tree, value id){
155 CAMLreturn(Val_int (XMLTREE(tree)->ParentNode(TREENODEVAL(id))));
158 extern "C" CAMLprim value caml_xml_tree_is_ancestor(value tree,value id1, value id2) {
159 CAMLparam3(tree,id1,id2);
160 CAMLreturn(Val_bool (XMLTREE(tree)->IsAncestor(TREENODEVAL(id1),TREENODEVAL(id2))));
163 extern "C" CAMLprim value caml_xml_tree_serialize(value tree, value filename){
164 CAMLparam2(tree,filename);
165 NOT_IMPLEMENTED("caml_xml_tree_serialize");
166 CAMLreturn(Val_unit);
169 extern "C" CAMLprim value caml_xml_tree_unserialize(value filename){
170 CAMLparam1(filename);
171 NOT_IMPLEMENTED("caml_xml_tree_unserialize");
172 CAMLreturn(Val_unit);
176 extern "C" CAMLprim value caml_xml_tree_first_child(value tree, value id){
178 CAMLreturn(Val_int (XMLTREE(tree)->FirstChild(TREENODEVAL(id))));
181 extern "C" CAMLprim value caml_xml_tree_is_leaf(value tree, value id){
183 CAMLreturn(Val_bool (XMLTREE(tree)->IsLeaf(TREENODEVAL(id))));
186 extern "C" CAMLprim value caml_xml_tree_next_sibling(value tree, value id){
188 CAMLreturn(Val_int (XMLTREE(tree)->NextSibling(TREENODEVAL(id))));
191 extern "C" CAMLprim value caml_xml_tree_prev_text(value tree, value id){
194 CAMLreturn(Val_int((XMLTREE(tree)->PrevText(TREENODEVAL(id)))));
197 extern "C" CAMLprim value caml_xml_tree_next_text(value tree, value id){
199 CAMLreturn(Val_int((XMLTREE(tree)->NextText(TREENODEVAL(id)))));
201 extern "C" CAMLprim value caml_xml_tree_my_text(value tree, value id){
203 CAMLreturn(Val_int((XMLTREE(tree)->MyText(TREENODEVAL(id)))));
206 extern "C" CAMLprim value caml_xml_tree_text_xml_id(value tree, value id){
208 CAMLreturn(Val_int((XMLTREE(tree)->TextXMLId(TREENODEVAL(id)))));
210 extern "C" CAMLprim value caml_xml_tree_node_xml_id(value tree, value id){
212 CAMLreturn(Val_int((XMLTREE(tree)->NodeXMLId(TREENODEVAL(id)))));
214 extern "C" CAMLprim value caml_xml_tree_tag(value tree, value id){
217 tag =(const char*) XMLTREE(tree)->GetTagName(XMLTREE(tree)->Tag(TREENODEVAL(id)));
219 CAMLreturn (caml_copy_string(tag));
221 extern "C" CAMLprim value caml_xml_tree_tag_id(value tree,value id){
224 int tag =XMLTREE(tree)->Tag(TREENODEVAL(id));
226 CAMLreturn (Val_unit);
228 extern "C" CAMLprim value caml_xml_tree_nullt(value unit){