1 files changed, 10 insertions, 13 deletions

diff --git a/mbest.h b/mbest.h
index fb7664f..2851fb9 100644
--- a/mbest.h
+++ b/mbest.h
@@ -33,27 +33,24 @@
 #define MBEST_STAGES 4
 
 struct MBEST_LIST {
-    int   index[MBEST_STAGES];    /* index of each stage that lead us to this error */
-    float error;
+  int index[MBEST_STAGES]; /* index of each stage that lead us to this error */
+  float error;
 };
 
 struct MBEST {
-    int                entries;   /* number of entries in mbest list   */
-    struct MBEST_LIST *list;
+  int entries; /* number of entries in mbest list   */
+  struct MBEST_LIST *list;
 };
 
 struct MBEST *mbest_create(int entries);
 void mbest_destroy(struct MBEST *mbest);
+void mbest_precompute_weight(float cb[], float w[], int k, int m);
 void mbest_insert(struct MBEST *mbest, int index[], float error);
-void mbest_search(const float  *cb, float vec[], float w[], int k, int m, struct MBEST *mbest, int index[]);
-void mbest_search450(const float  *cb, float vec[], float w[], int k,int shorterK, int m, struct MBEST *mbest, int index[]);
-
-// #define MBEST_PRINT_OUT
-#ifdef MBEST_PRINT_OUT
- #define MBEST_PRINT(a,b) mbest_print((a),(b))
-#else
- #define MBEST_PRINT(a,b) 
-#endif
+void mbest_search(const float *cb, float vec[], int k, int m,
+                  struct MBEST *mbest, int index[]);
+void mbest_search450(const float *cb, float vec[], float w[], int k,
+                     int shorterK, int m, struct MBEST *mbest, int index[]);
 
+void mbest_print(char title[], struct MBEST *mbest);
 
 #endif