From 13ed9af395da12a6b25eb2f9e6805973cd4cdbd8 Mon Sep 17 00:00:00 2001
From: dvanbeek <d.vanbeek@vumc.nl>
Date: Thu, 6 Oct 2016 10:26:14 +0200
Subject: [PATCH] Added support for plotting only 13/18/21

---
 plot.py | 27 ++++++++++++++++++++-------
 1 file changed, 20 insertions(+), 7 deletions(-)

diff --git a/plot.py b/plot.py
index cd5dea8..9788999 100644
--- a/plot.py
+++ b/plot.py
@@ -48,13 +48,24 @@ def plotResults(sample, markedBins, kept, kept2, outputFile, zScoresDict,zSmooth
 	colorBlinds			= (0.85,0.85,0.85)
 	colorWaste			= (0.7,0.7,0.7)
 	
+	prefix = ""
 	chromsToProcess = range(1,23)
 	binScalar = 320
 	edgeSize = 0.15
+	figsize=(8, 6)
+	nrLegend = 4
+	textFromBottom = 0.06
+	textFromLeft = 0.05
 	
 	if limit:
 		chromsToProcess = [13,18,21]
 		binScalar = 480
+		prefix = "_filter"
+		figsize = (4.5, 3)
+		figsize = (4.5, 3)
+		nrLegend = 2
+		textFromBottom = 0.19
+		textFromLeft = 0.03
 
 	def printMarkers(chrom):
 		binWidth = 1/float(len(sample[str(chrom)]))*binScalar
@@ -99,12 +110,14 @@ def preparePlot(chrom):
 		printMarkers(str(chrom))
 	
 	print 'Plotting Z-Scores'
-	ax = plt.figure(2)
-	ax.text(0.5, 0.06, 'Chromosomal position in bins', ha='center', va='bottom')
-	ax.text(0.05, 0.5, 'Z-score', ha='left', va='center', rotation='vertical')
-	ax.text(0.5, 0.93, 'Sample ' + sampleName, ha='center', va='bottom')
-	
+	ax = plt.figure(2, figsize=figsize)
+	if limit:
+		ax.subplots_adjust(left=0.15, top=0.9, bottom=0.25)
+	ax.text(0.5, textFromBottom, 'Chromosomal position in bins', ha='center', va='bottom')
+	ax.text(textFromLeft, 0.5, 'Z-score', ha='left', va='center', rotation='vertical')
+	ax.suptitle('Sample ' + sampleName)
 
+	
 
 	for chrom in chromsToProcess:
 		preparePlot(chrom)
@@ -142,9 +155,9 @@ def preparePlot(chrom):
 	bLGrey	= plt.Rectangle((0, 0), 1, 1, fc=colorBlinds)
 	bGrey	= plt.Rectangle((0, 0), 1, 1, fc=colorWaste)
 	
-	ax.legend((lRed, lBlue, bLGreen, bDGreen, bPink, bPurple, bLGrey, bGrey), ('Z-score windowed method', 'Z-score individual bin method', 'Detected deviation by individual bin method', 'Called by individual bin method', 'Detected deviation by windowed method', 'Called by windowed method', 'Not enough reference bins available', 'Unmappable region'), 'lower left',prop={'size':6}, mode='expand', ncol=4)
+	ax.legend((lRed, lBlue, bLGreen, bDGreen, bPink, bPurple, bLGrey, bGrey), ('Z-score windowed method', 'Z-score individual bin method', 'Detected deviation by individual bin method', 'Called by individual bin method', 'Detected deviation by windowed method', 'Called by windowed method', 'Not enough reference bins available', 'Unmappable region'), 'lower left',prop={'size':6}, mode='expand', ncol=nrLegend)
 	
-	plt.savefig(outputFile+'.'+fileType,figsize=(11.7, 8.3),dpi=160)
+	plt.savefig(outputFile + prefix + '.' + fileType, figsize=figsize, dpi=160)
 
 
 if __name__ == "__main__":