add isl_printer_print_schedule

Signed-off-by: Sven Verdoolaege <skimo@kotnet.org>

diff --git a/doc/user.pod b/doc/user.pod
index 899fb58..b885501 100644 (file)
--- a/doc/user.pod
+++ b/doc/user.pod
@@ -2919,6 +2919,12 @@ from an C<isl_schedule> using the following function.
        __isl_give isl_union_map *isl_schedule_get_map(
                __isl_keep isl_schedule *sched);
 
+A representation of the schedule can be printed using
+        
+       __isl_give isl_printer *isl_printer_print_schedule(
+               __isl_take isl_printer *p,
+               __isl_keep isl_schedule *schedule);
+
 A representation of the schedule as a forest of bands can be obtained
 using the following function.
 

diff --git a/include/isl/schedule.h b/include/isl/schedule.h
index 6899cfd..3aa99a2 100644 (file)
--- a/include/isl/schedule.h
+++ b/include/isl/schedule.h
@@ -26,6 +26,10 @@ int isl_schedule_n_band(__isl_keep isl_schedule *sched);
 __isl_give isl_union_map *isl_schedule_get_band(__isl_keep isl_schedule *sched,
        unsigned band);
 
+__isl_give isl_printer *isl_printer_print_schedule(__isl_take isl_printer *p,
+       __isl_keep isl_schedule *schedule);
+void isl_schedule_dump(__isl_keep isl_schedule *schedule);
+
 #if defined(__cplusplus)
 }
 #endif

diff --git a/isl_schedule.c b/isl_schedule.c
index 0b1ff56..bb747de 100644 (file)
--- a/isl_schedule.c
+++ b/isl_schedule.c
@@ -2621,3 +2621,72 @@ __isl_give isl_band_list *isl_schedule_get_band_forest(
                schedule->band_forest = construct_forest(schedule);
        return isl_band_list_copy(schedule->band_forest);
 }
+
+static __isl_give isl_printer *print_band_list(__isl_take isl_printer *p,
+       __isl_keep isl_band_list *list);
+
+static __isl_give isl_printer *print_band(__isl_take isl_printer *p,
+       __isl_keep isl_band *band)
+{
+       isl_band_list *children;
+
+       p = isl_printer_start_line(p);
+       p = isl_printer_print_union_map(p, band->map);
+       p = isl_printer_end_line(p);
+
+       if (!isl_band_has_children(band))
+               return p;
+
+       children = isl_band_get_children(band);
+
+       p = isl_printer_indent(p, 4);
+       p = print_band_list(p, children);
+       p = isl_printer_indent(p, -4);
+
+       isl_band_list_free(children);
+
+       return p;
+}
+
+static __isl_give isl_printer *print_band_list(__isl_take isl_printer *p,
+       __isl_keep isl_band_list *list)
+{
+       int i, n;
+
+       n = isl_band_list_n_band(list);
+       for (i = 0; i < n; ++i) {
+               isl_band *band;
+               band = isl_band_list_get_band(list, i);
+               p = print_band(p, band);
+               isl_band_free(band);
+       }
+
+       return p;
+}
+
+__isl_give isl_printer *isl_printer_print_schedule(__isl_take isl_printer *p,
+       __isl_keep isl_schedule *schedule)
+{
+       isl_band_list *forest;
+
+       forest = isl_schedule_get_band_forest(schedule);
+
+       p = print_band_list(p, forest);
+
+       isl_band_list_free(forest);
+
+       return p;
+}
+
+void isl_schedule_dump(__isl_keep isl_schedule *schedule)
+{
+       isl_printer *printer;
+
+       if (!schedule)
+               return;
+
+       printer = isl_printer_to_file(isl_schedule_get_ctx(schedule), stderr);
+       printer = isl_printer_print_schedule(printer, schedule);
+
+       isl_printer_free(printer);
+}