17#include "core_tries.h"
18#include "base_itries.h"
26void init_itries(
void);
27static YAP_Bool p_itrie_open(
void);
28static YAP_Bool p_itrie_close(
void);
29static YAP_Bool p_itrie_close_all(
void);
30static YAP_Bool p_itrie_mode(
void);
31static YAP_Bool p_itrie_timestamp(
void);
32static YAP_Bool p_itrie_put_entry(
void);
33static YAP_Bool p_itrie_update_entry(
void);
34static YAP_Bool p_itrie_check_entry(
void);
35static YAP_Bool p_itrie_get_entry(
void);
36static YAP_Bool p_itrie_get_data(
void);
37static YAP_Bool p_itrie_traverse_init(
void);
38static YAP_Bool p_itrie_traverse_cont(
void);
39static YAP_Bool p_itrie_remove_entry(
void);
40static YAP_Bool p_itrie_remove_subtree(
void);
41static YAP_Bool p_itrie_add(
void);
42static YAP_Bool p_itrie_subtract(
void);
43static YAP_Bool p_itrie_join(
void);
44static YAP_Bool p_itrie_intersect(
void);
45static YAP_Bool p_itrie_count_join(
void);
46static YAP_Bool p_itrie_count_intersect(
void);
47static YAP_Bool p_itrie_save(
void);
48static YAP_Bool p_itrie_save_as_trie(
void);
49static YAP_Bool p_itrie_load(
void);
50static YAP_Bool p_itrie_save2stream(
void);
51static YAP_Bool p_itrie_loadFromStream(
void);
52static YAP_Bool p_itrie_stats(
void);
53static YAP_Bool p_itrie_max_stats(
void);
54static YAP_Bool p_itrie_usage(
void);
55static YAP_Bool p_itrie_print(
void);
63void init_itries(
void) {
66 YAP_UserCPredicate(
"itrie_open", p_itrie_open, 1);
67 YAP_UserCPredicate(
"itrie_close", p_itrie_close, 1);
68 YAP_UserCPredicate(
"itrie_close_all", p_itrie_close_all, 0);
69 YAP_UserCPredicate(
"itrie_mode", p_itrie_mode, 2);
70 YAP_UserCPredicate(
"itrie_timestamp", p_itrie_timestamp, 2);
71 YAP_UserCPredicate(
"itrie_put_entry", p_itrie_put_entry, 2);
72 YAP_UserCPredicate(
"itrie_update_entry", p_itrie_update_entry, 2);
73 YAP_UserCPredicate(
"itrie_check_entry", p_itrie_check_entry, 3);
74 YAP_UserCPredicate(
"itrie_get_entry", p_itrie_get_entry, 2);
75 YAP_UserCPredicate(
"itrie_get_data", p_itrie_get_data, 2);
76 YAP_UserBackCPredicate(
"itrie_traverse", p_itrie_traverse_init, p_itrie_traverse_cont, 2, 0);
77 YAP_UserCPredicate(
"itrie_remove_entry", p_itrie_remove_entry, 1);
78 YAP_UserCPredicate(
"itrie_remove_subtree", p_itrie_remove_subtree, 1);
79 YAP_UserCPredicate(
"itrie_add", p_itrie_add, 2);
80 YAP_UserCPredicate(
"itrie_subtract", p_itrie_subtract, 2);
81 YAP_UserCPredicate(
"itrie_join", p_itrie_join, 2);
82 YAP_UserCPredicate(
"itrie_intersect", p_itrie_intersect, 2);
83 YAP_UserCPredicate(
"itrie_count_join", p_itrie_count_join, 3);
84 YAP_UserCPredicate(
"itrie_count_intersect", p_itrie_count_intersect, 3);
85 YAP_UserCPredicate(
"itrie_save", p_itrie_save, 2);
86 YAP_UserCPredicate(
"itrie_save_as_trie", p_itrie_save_as_trie, 2);
87 YAP_UserCPredicate(
"itrie_load", p_itrie_load, 2);
88 YAP_UserCPredicate(
"itrie_save2stream", p_itrie_save2stream, 2);
89 YAP_UserCPredicate(
"itrie_loadFromstream", p_itrie_loadFromStream, 2);
90 YAP_UserCPredicate(
"itrie_stats", p_itrie_stats, 4);
91 YAP_UserCPredicate(
"itrie_max_stats", p_itrie_max_stats, 4);
92 YAP_UserCPredicate(
"itrie_usage", p_itrie_usage, 4);
93 YAP_UserCPredicate(
"itrie_print", p_itrie_print, 1);
104#define arg_itrie YAP_ARG1
105static YAP_Bool p_itrie_open(
void) {
109 if (!YAP_IsVarTerm(arg_itrie))
113 itrie = itrie_open();
114 return YAP_Unify(arg_itrie, YAP_MkIntTerm((YAP_Int) itrie));
120#define arg_itrie YAP_ARG1
121static YAP_Bool p_itrie_close(
void) {
123 if (!YAP_IsIntTerm(arg_itrie))
127 itrie_close((
TrEntry) YAP_IntOfTerm(arg_itrie));
134static YAP_Bool p_itrie_close_all(
void) {
141#define arg_itrie YAP_ARG1
142#define arg_mode YAP_ARG2
143static YAP_Bool p_itrie_mode(
void) {
145 const char *mode_str;
149 if (!YAP_IsIntTerm(arg_itrie))
153 if (YAP_IsVarTerm(arg_mode)) {
154 mode = itrie_get_mode((
TrEntry) YAP_IntOfTerm(arg_itrie));
155 if (mode == ITRIES_MODE_INC_POS)
156 mode_term = YAP_MkAtomTerm(YAP_LookupAtom(
"inc_pos"));
157 else if (mode == ITRIES_MODE_DEC_POS)
158 mode_term = YAP_MkAtomTerm(YAP_LookupAtom(
"dec_pos"));
159 else if (mode == ITRIES_MODE_INC_NEG)
160 mode_term = YAP_MkAtomTerm(YAP_LookupAtom(
"inc_neg"));
161 else if (mode == ITRIES_MODE_DEC_NEG)
162 mode_term = YAP_MkAtomTerm(YAP_LookupAtom(
"dec_neg"));
163 else if (mode == ITRIES_MODE_NONE)
164 mode_term = YAP_MkAtomTerm(YAP_LookupAtom(
"none"));
167 return YAP_Unify(arg_mode, mode_term);
171 mode_str = YAP_AtomName(YAP_AtomOfTerm(arg_mode));
172 if (!strcmp(mode_str,
"inc_pos"))
173 mode = ITRIES_MODE_INC_POS;
174 else if (!strcmp(mode_str,
"dec_pos"))
175 mode = ITRIES_MODE_DEC_POS;
176 else if (!strcmp(mode_str,
"inc_neg"))
177 mode = ITRIES_MODE_INC_NEG;
178 else if (!strcmp(mode_str,
"dec_neg"))
179 mode = ITRIES_MODE_DEC_NEG;
180 else if (!strcmp(mode_str,
"none"))
181 mode = ITRIES_MODE_NONE;
184 itrie_set_mode((
TrEntry) YAP_IntOfTerm(arg_itrie), mode);
192#define arg_itrie YAP_ARG1
193#define arg_time YAP_ARG2
194static YAP_Bool p_itrie_timestamp(
void) {
198 if (!YAP_IsIntTerm(arg_itrie))
202 if (YAP_IsVarTerm(arg_time)) {
203 time = itrie_get_timestamp((
TrEntry) YAP_IntOfTerm(arg_itrie));
204 return YAP_Unify(arg_time, YAP_MkIntTerm(time));
208 if (YAP_IsIntTerm(arg_time)) {
209 time = YAP_IntOfTerm(arg_time);
210 itrie_set_timestamp((
TrEntry) YAP_IntOfTerm(arg_itrie), time);
221#define arg_itrie YAP_ARG1
222#define arg_entry YAP_ARG2
223static YAP_Bool p_itrie_put_entry(
void) {
225 if (!YAP_IsIntTerm(arg_itrie))
229 itrie_put_entry((
TrEntry) YAP_IntOfTerm(arg_itrie), arg_entry);
237#define arg_itrie YAP_ARG1
238#define arg_entry YAP_ARG2
239static YAP_Bool p_itrie_update_entry(
void) {
241 if (!YAP_IsIntTerm(arg_itrie))
245 itrie_update_entry((
TrEntry) YAP_IntOfTerm(arg_itrie), arg_entry);
253#define arg_itrie YAP_ARG1
254#define arg_entry YAP_ARG2
255#define arg_ref YAP_ARG3
256static YAP_Bool p_itrie_check_entry(
void) {
260 if (!YAP_IsIntTerm(arg_itrie))
264 if (!(data = itrie_check_entry((
TrEntry) YAP_IntOfTerm(arg_itrie), arg_entry)))
266 return YAP_Unify(arg_ref, YAP_MkIntTerm((YAP_Int) data));
274#define arg_ref YAP_ARG1
275#define arg_entry YAP_ARG2
276static YAP_Bool p_itrie_get_entry(
void) {
280 if (!YAP_IsIntTerm(arg_ref))
284 entry = itrie_get_entry((
TrData) YAP_IntOfTerm(arg_ref));
285 return YAP_Unify(arg_entry, entry);
292#define arg_ref YAP_ARG1
293#define arg_data YAP_ARG2
294static YAP_Bool p_itrie_get_data(
void) {
298 YAP_Int pos, neg, time;
301 if (!YAP_IsIntTerm(arg_ref))
305 itrie_get_data((
TrData) YAP_IntOfTerm(arg_ref), &pos, &neg, &time);
306 list = YAP_MkAtomTerm(YAP_LookupAtom(
"[]"));
307 f = YAP_MkFunctor(YAP_LookupAtom(
"timestamp"), 1);
308 item = YAP_MkIntTerm(time);
309 item = YAP_MkApplTerm(f, 1, &item);
310 list = YAP_MkPairTerm(item, list);
311 f = YAP_MkFunctor(YAP_LookupAtom(
"neg"), 1);
312 item = YAP_MkIntTerm(neg);
313 item = YAP_MkApplTerm(f, 1, &item);
314 list = YAP_MkPairTerm(item, list);
315 f = YAP_MkFunctor(YAP_LookupAtom(
"pos"), 1);
316 item = YAP_MkIntTerm(pos);
317 item = YAP_MkApplTerm(f, 1, &item);
318 list = YAP_MkPairTerm(item, list);
319 return YAP_Unify(arg_data, list);
326#define arg_itrie YAP_ARG1
327#define arg_ref YAP_ARG2
328static YAP_Bool p_itrie_traverse_init(
void) {
332 if (!YAP_IsIntTerm(arg_itrie))
336 if (!(data = itrie_traverse_init((
TrEntry) YAP_IntOfTerm(arg_itrie)))) {
340 return YAP_Unify(arg_ref, YAP_MkIntTerm((YAP_Int) data));
347#define arg_itrie YAP_ARG1
348#define arg_ref YAP_ARG2
349static YAP_Bool p_itrie_traverse_cont(
void) {
353 if (!(data = itrie_traverse_cont((
TrEntry) YAP_IntOfTerm(arg_itrie)))) {
357 return YAP_Unify(arg_ref, YAP_MkIntTerm((YAP_Int) data));
364#define arg_ref YAP_ARG1
365static YAP_Bool p_itrie_remove_entry(
void) {
367 if (!YAP_IsIntTerm(arg_ref))
371 itrie_remove_entry((
TrData) YAP_IntOfTerm(arg_ref));
378#define arg_ref YAP_ARG1
379static YAP_Bool p_itrie_remove_subtree(
void) {
381 if (!YAP_IsIntTerm(arg_ref))
385 itrie_remove_subtree((
TrData) YAP_IntOfTerm(arg_ref));
392#define arg_itrie_dest YAP_ARG1
393#define arg_itrie_source YAP_ARG2
394static YAP_Bool p_itrie_add(
void) {
396 if (!YAP_IsIntTerm(arg_itrie_dest))
398 if (!YAP_IsIntTerm(arg_itrie_source))
402 itrie_add((
TrEntry) YAP_IntOfTerm(arg_itrie_dest), (
TrEntry) YAP_IntOfTerm(arg_itrie_source));
406#undef arg_itrie_source
410#define arg_itrie_dest YAP_ARG1
411#define arg_itrie_source YAP_ARG2
412static YAP_Bool p_itrie_subtract(
void) {
414 if (!YAP_IsIntTerm(arg_itrie_dest))
416 if (!YAP_IsIntTerm(arg_itrie_source))
420 itrie_subtract((
TrEntry) YAP_IntOfTerm(arg_itrie_dest), (
TrEntry) YAP_IntOfTerm(arg_itrie_source));
424#undef arg_itrie_source
428#define arg_itrie_dest YAP_ARG1
429#define arg_itrie_source YAP_ARG2
430static YAP_Bool p_itrie_join(
void) {
432 if (!YAP_IsIntTerm(arg_itrie_dest))
434 if (!YAP_IsIntTerm(arg_itrie_source))
438 itrie_join((
TrEntry) YAP_IntOfTerm(arg_itrie_dest), (
TrEntry) YAP_IntOfTerm(arg_itrie_source));
442#undef arg_itrie_source
446#define arg_itrie_dest YAP_ARG1
447#define arg_itrie_source YAP_ARG2
448static YAP_Bool p_itrie_intersect(
void) {
450 if (!YAP_IsIntTerm(arg_itrie_dest))
452 if (!YAP_IsIntTerm(arg_itrie_source))
456 itrie_intersect((
TrEntry) YAP_IntOfTerm(arg_itrie_dest), (
TrEntry) YAP_IntOfTerm(arg_itrie_source));
460#undef arg_itrie_source
464#define arg_itrie1 YAP_ARG1
465#define arg_itrie2 YAP_ARG2
466#define arg_entries YAP_ARG3
467static YAP_Bool p_itrie_count_join(
void) {
471 if (!YAP_IsIntTerm(arg_itrie1))
473 if (!YAP_IsIntTerm(arg_itrie2))
477 entries = itrie_count_join((
TrEntry) YAP_IntOfTerm(arg_itrie1), (
TrEntry) YAP_IntOfTerm(arg_itrie2));
478 return YAP_Unify(arg_entries, YAP_MkIntTerm(entries));
486#define arg_itrie1 YAP_ARG1
487#define arg_itrie2 YAP_ARG2
488#define arg_entries YAP_ARG3
489static YAP_Bool p_itrie_count_intersect(
void) {
493 if (!YAP_IsIntTerm(arg_itrie1))
495 if (!YAP_IsIntTerm(arg_itrie2))
499 entries = itrie_count_intersect((
TrEntry) YAP_IntOfTerm(arg_itrie1), (
TrEntry) YAP_IntOfTerm(arg_itrie2));
500 return YAP_Unify(arg_entries, YAP_MkIntTerm(entries));
508#define arg_itrie YAP_ARG1
509#define arg_file YAP_ARG2
510static YAP_Bool p_itrie_save(
void) {
511 const char *file_str;
515 if (!YAP_IsIntTerm(arg_itrie))
517 if (!YAP_IsAtomTerm(arg_file))
521 file_str = YAP_AtomName(YAP_AtomOfTerm(arg_file));
522 if (!(file = fopen(file_str,
"w")))
526 itrie_save((
TrEntry) YAP_IntOfTerm(arg_itrie), file);
536#define arg_itrie YAP_ARG1
537#define arg_file YAP_ARG2
538static YAP_Bool p_itrie_save_as_trie(
void) {
539 const char *file_str;
543 if (!YAP_IsIntTerm(arg_itrie))
545 if (!YAP_IsAtomTerm(arg_file))
549 file_str = YAP_AtomName(YAP_AtomOfTerm(arg_file));
550 if (!(file = fopen(file_str,
"w")))
554 itrie_save_as_trie((
TrEntry) YAP_IntOfTerm(arg_itrie), file);
564#define arg_itrie YAP_ARG1
565#define arg_file YAP_ARG2
566static YAP_Bool p_itrie_load(
void) {
568 const char *file_str;
572 if (!YAP_IsVarTerm(arg_itrie))
574 if (!YAP_IsAtomTerm(arg_file))
578 file_str = YAP_AtomName(YAP_AtomOfTerm(arg_file));
579 if (!(file = fopen(file_str,
"r")))
583 if (!(itrie = itrie_load(file)))
587 return YAP_Unify(arg_itrie, YAP_MkIntTerm((YAP_Int) itrie));
594#define arg_itrie YAP_ARG1
595#define arg_stream YAP_ARG2
596static YAP_Bool p_itrie_save2stream(
void) {
600 if (!YAP_IsIntTerm(arg_itrie))
602 if ((file = (FILE*) YAP_FileDescriptorFromStream(arg_stream)) == NULL)
606 itrie_save((
TrEntry) YAP_IntOfTerm(arg_itrie), file);
614#define arg_itrie YAP_ARG1
615#define arg_stream YAP_ARG2
616static YAP_Bool p_itrie_loadFromStream(
void) {
621 if (!YAP_IsVarTerm(arg_itrie))
623 if (!(file = (FILE*) YAP_FileDescriptorFromStream(arg_stream)))
627 if (!(itrie = itrie_load(file)))
629 return YAP_Unify(arg_itrie, YAP_MkIntTerm((YAP_Int) itrie));
636#define arg_memory YAP_ARG1
637#define arg_tries YAP_ARG2
638#define arg_entries YAP_ARG3
639#define arg_nodes YAP_ARG4
640static YAP_Bool p_itrie_stats(
void) {
641 YAP_Int memory, tries, entries, nodes;
644 itrie_stats(&memory, &tries, &entries, &nodes);
645 if (!YAP_Unify(arg_memory, YAP_MkIntTerm(memory)))
647 if (!YAP_Unify(arg_tries, YAP_MkIntTerm(tries)))
649 if (!YAP_Unify(arg_entries, YAP_MkIntTerm(entries)))
651 if (!YAP_Unify(arg_nodes, YAP_MkIntTerm(nodes)))
662#define arg_memory YAP_ARG1
663#define arg_tries YAP_ARG2
664#define arg_entries YAP_ARG3
665#define arg_nodes YAP_ARG4
666static YAP_Bool p_itrie_max_stats(
void) {
667 YAP_Int memory, tries, entries, nodes;
670 itrie_max_stats(&memory, &tries, &entries, &nodes);
671 if (!YAP_Unify(arg_memory, YAP_MkIntTerm(memory)))
673 if (!YAP_Unify(arg_tries, YAP_MkIntTerm(tries)))
675 if (!YAP_Unify(arg_entries, YAP_MkIntTerm(entries)))
677 if (!YAP_Unify(arg_nodes, YAP_MkIntTerm(nodes)))
688#define arg_itrie YAP_ARG1
689#define arg_entries YAP_ARG2
690#define arg_nodes YAP_ARG3
691#define arg_virtualnodes YAP_ARG4
692static YAP_Bool p_itrie_usage(
void) {
693 YAP_Int entries, nodes, virtualnodes;
696 if (!YAP_IsIntTerm(arg_itrie))
700 itrie_usage((
TrEntry) YAP_IntOfTerm(arg_itrie), &entries, &nodes, &virtualnodes);
701 if (!YAP_Unify(arg_entries, YAP_MkIntTerm(entries)))
703 if (!YAP_Unify(arg_nodes, YAP_MkIntTerm(nodes)))
705 if (!YAP_Unify(arg_virtualnodes, YAP_MkIntTerm(virtualnodes)))
712#undef arg_virtualnodes
716#define arg_itrie YAP_ARG1
717static YAP_Bool p_itrie_print(
void) {
719 if (!YAP_IsIntTerm(arg_itrie))
723 itrie_print((
TrEntry) YAP_IntOfTerm(arg_itrie));