def plot_BAF_heatmap(args): if PLT_AVAIL == True: pass else: print "matplotlib.pyplot not available, skip plotting..." sys.stdout.flush() sys.exit(-1) BAF_heatmap = BAFHeatMap() BAF_heatmap.read_heatmap(args.filename_base) BAF_heatmap.get_color_max() segments = Segments() inseg_file_name = args.filename_base + ".PyLOH.segments" segments.read_segfile(inseg_file_name) seg_num = segments.num outheatmap_dir_name = args.filename_base + ".PyLOH.heatmap.plot" if os.path.exists(outheatmap_dir_name) == False: os.mkdir(outheatmap_dir_name) for j in range(0, seg_num): BAF_counts_j = BAF_heatmap.BAF_counts[j] seg_name_j = segments[j][0] print "Plotting segment {0}...".format(seg_name_j) sys.stdout.flush() plot_BAF_heatmap_by_segment(BAF_counts_j, seg_name_j, outheatmap_dir_name)
def plot_BAF_heatmap(args): if PLT_AVAIL == True: pass else: print "matplotlib.pyplot not available, skip plotting..." sys.stdout.flush() sys.exit(-1) BAF_heatmap = BAFHeatMap() BAF_heatmap.read_heatmap(args.filename_base) BAF_heatmap.get_color_max() segments = Segments() inseg_file_name = args.filename_base + '.PyLOH.segments' segments.read_segfile(inseg_file_name) seg_num = segments.num outheatmap_dir_name = args.filename_base + '.PyLOH.heatmap.plot' if os.path.exists(outheatmap_dir_name) == False: os.mkdir(outheatmap_dir_name) for j in range(0, seg_num): BAF_counts_j = BAF_heatmap.BAF_counts[j] seg_name_j = segments[j][0] print 'Plotting segment {0}...'.format(seg_name_j) sys.stdout.flush() plot_BAF_heatmap_by_segment(BAF_counts_j, seg_name_j, outheatmap_dir_name)