Added -P to create a data file w/ Q, D and C per line

[blktrace.git] / btt / devs.c

diff --git a/btt/devs.c b/btt/devs.c
index 6c9870ff3f0bfdf398abf97f3a1c7eb02e7fd419..e5304773087b413c0e5307a16b94caa9c4f0f40c 100644 (file)
--- a/btt/devs.c
+++ b/btt/devs.c
@@ -93,6 +93,8 @@ void dip_exit(void)
                unplug_hist_exit(dip->unplug_hist_handle);
                if (output_all_data)
                        q2d_release(dip->q2d_priv);
+               if (dip->pit_fp)
+                       fclose(dip->pit_fp);
                free(dip);
        }
 }
@@ -106,6 +108,16 @@ static inline char *mkhandle(char *str, __u32 device, char *post)
        return str;
 }
 
+static inline FILE *open_pit(char *str)
+{
+       FILE *fp = fopen(str, "w");
+
+       if (fp == NULL)
+               perror(str);
+
+       return fp;
+}
+
 struct d_info *dip_add(__u32 device, struct io *iop)
 {
        struct d_info *dip = __dip_find(device);
@@ -139,6 +151,9 @@ struct d_info *dip_add(__u32 device, struct io *iop)
                if (output_all_data)
                        dip->q2d_priv = q2d_init();
                n_devs++;
+               if (per_io_trees)
+                       dip->pit_fp = open_pit(mkhandle(per_io_trees,
+                                                         device, "_pit.dat"));
        }
 
        if (dip->pre_culling) {