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)
