Added bestK barplot function. #63

Author: ODiogoSilva

structure_threader/plotter/structplot.py

@@ -38,7 +38,7 @@
 
 # Create color pallete
 c = cl.scales["12"]["qual"]["Set3"]
-
+plt.style.use("ggplot")
 
 class PlotK:
     """
@@ -848,8 +848,6 @@ def plotk_static(self, kval, output_dir, bw=False, use_ind=False):
         with the --ind option, use those labels instead of population labels
         """
 
-        plt.style.use("ggplot")
-
         numinds = self.number_indv
 
         clist = [[i / 255. for i in x] for x in cl.to_numeric(c)]
@@ -944,6 +942,23 @@ def plotk_static(self, kval, output_dir, bw=False, use_ind=False):
         plt.close()
 
 
+def plot_normalization(norm_dict, outdir):
+
+    plt.clf()
+
+    keys = ["norm_mean", "lower_limit", "upper_limit"]
+
+    data = [(k, [vals[x] for x in keys]) for k, vals in norm_dict.items()]
+
+    means = [x[1][0] for x in data]
+    lower_limit = [x[1][0] - x[1][1] for x in data]
+    upper_limit = [x[1][2] - x[1][0] for x in data]
+
+    plt.bar(range(len(means)), means, yerr=[lower_limit, upper_limit])
+
+    plt.savefig(join(outdir, "bestK", "bestk_evidence.svg"), dpi=200)
+
+
 def main(result_files, fmt, outdir, bestk=None, popfile=None, indfile=None,
          filter_k=None, bw=False, use_ind=False):
     """

structure_threader/wrappers/maverick_wrapper.py

@@ -26,9 +26,10 @@
 
 try:
     import colorer.colorer as colorer
+    from plotter.structplot import plot_normalization
 except ImportError:
     import structure_threader.colorer.colorer as colorer
-
+    from structure_threader.plotter.structplot import plot_normalization
 
 def mav_cli_generator(arg, k_val, mav_params):
     """
@@ -145,8 +146,9 @@ def ti_test(outdir, norm_evidence, ti_in_use):
     bestk_file.write(output_text)
     bestk_file.close()
 
+    plot_normalization(criteria, outdir)
+
     return [int(bestk)]
-    # TODO: Draw a plot!
 
 
 def maverick_merger(outdir, k_list, mav_params, no_tests):