annotate viz_graphs.py @ 8:edec16f9566a draft

Uploaded
author glogobyte
date Fri, 16 Oct 2020 18:16:06 +0000
parents
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
8
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
1 import sys
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
2 def pdf_before_DE(analysis):
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
3
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
4 # Import FPDF class
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
5 from fpdf import FPDF, fpdf
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
6
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
7 # Import glob module to find all the files matching a pattern
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
8 import glob
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
9
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
10 # Image extensions
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
11 if analysis=="2":
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
12 image_extensions = ("c_hist_red.png","t_hist_red.png","pie_non.png","spider_red.png","spider_non_red.png","c_logo.png","t_logo.png","c_bar.png","t_bar.png")
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
13 else:
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
14 image_extensions = ("c_hist_red.png","t_hist_red.png","pie_tem.png","spider_red.png","spider_non_red.png")
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
15 # This list will hold the images file names
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
16 images = []
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
17
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
18 # Build the image list by merging the glob results (a list of files)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
19 # for each extension. We are taking images from current folder.
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
20 for extension in image_extensions:
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
21 images.extend(glob.glob(extension))
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
22 #sys.exit(images)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
23 # Create instance of FPDF class
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
24 pdf = FPDF('P', 'in', 'A4')
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
25 # Add new page. Without this you cannot create the document.
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
26 pdf.add_page()
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
27 # Set font to Arial, 'B'old, 16 pts
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
28 pdf.set_font('Arial', 'B', 20.0)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
29
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
30 # Page header
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
31 pdf.cell(pdf.w-0.5, 0.5, 'IsomiR Profile Report',align='C')
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
32 pdf.ln(0.7)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
33 pdf.set_font('Arial','', 16.0)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
34 pdf.cell(pdf.w-0.5, 0.5, 'sRNA Length Distribution',align='C')
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
35
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
36 # Smaller font for image captions
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
37 pdf.set_font('Arial', '', 11.0)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
38
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
39 # Image caption
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
40 pdf.ln(0.5)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
41
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
42 yh=FPDF.get_y(pdf)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
43 pdf.image(images[0],x=0.3,w=4, h=3)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
44 pdf.image(images[1],x=4,y=yh, w=4, h=3)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
45 pdf.ln(0.3)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
46
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
47 # Image caption
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
48 pdf.cell(0.2)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
49 pdf.cell(3.0, 0.0, " Mapped and unmapped reads to custom precussor arm reference DB (5p and 3p arms) in Control (left)")
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
50 pdf.ln(0.2)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
51 pdf.cell(0.2)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
52 pdf.cell(3.0, 0.0, " and Treated (right) groups")
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
53
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
54
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
55 pdf.ln(0.5)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
56 h1=FPDF.get_y(pdf)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
57 pdf.image(images[2],x=1, w=6.5, h=5)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
58 h2=FPDF.get_y(pdf)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
59 FPDF.set_y(pdf,h1+0.2)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
60 pdf.set_font('Arial','', 14.0)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
61 pdf.cell(pdf.w-0.5, 0.5, 'Template and non-template IsomiRs',align='C')
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
62 pdf.set_font('Arial', '', 11.0)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
63 FPDF.set_y(pdf,h2)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
64 FPDF.set_y(pdf,9.5)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
65 # Image caption
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
66 pdf.cell(0.2)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
67 if analysis=="2":
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
68 pdf.cell(3.0, 0.0, " Template, non-template, miRNA reference and unmapped sequences as percentage of total sRNA")
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
69 else:
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
70 pdf.cell(3.0, 0.0, " Template, miRNA reference and unmapped sequences as percentage of total sRNA")
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
71 pdf.ln(0.2)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
72 pdf.cell(0.2)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
73 pdf.cell(3.0, 0.0, " reads in Control (left) and treated (right) groups")
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
74
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
75
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
76
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
77 pdf.add_page()
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
78 pdf.set_font('Arial', 'B', 16.0)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
79 pdf.cell(pdf.w-0.5, 0.5, "Reference form and isomiR among total miRNA reads",align='C')
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
80 pdf.ln(0.7)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
81 pdf.set_font('Arial', 'B', 12.0)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
82 pdf.cell(pdf.w-0.5, 0.5, "Template isomiR profile (redundant)",align='C')
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
83 pdf.ln(0.5)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
84 pdf.image(images[3],x=1.5, w=5.5, h=4)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
85 pdf.ln(0.6)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
86 pdf.cell(pdf.w-0.5, 0.0, "Template isomiR profile (non-redundant)",align='C')
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
87 pdf.set_font('Arial', '', 12.0)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
88 pdf.ln(0.2)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
89 pdf.image(images[4],x=1.5, w=5.5, h=4)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
90 pdf.ln(0.3)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
91 pdf.set_font('Arial', '', 11.0)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
92 pdf.cell(0.2)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
93 pdf.cell(3.0, 0.0, " * IsomiRs potentialy initiated from multiple loci")
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
94
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
95
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
96 if analysis=="2":
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
97 pdf.add_page('L')
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
98
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
99 pdf.set_font('Arial', 'B', 16.0)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
100 pdf.cell(pdf.w-0.5, 0.5, "Non-template IsomiRs",align='C')
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
101 pdf.ln(0.5)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
102 pdf.set_font('Arial', 'B', 12.0)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
103 pdf.cell(pdf.w-0.5, 0.5, "3' Additions of reference of isomiR sequence",align='C')
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
104 pdf.ln(0.7)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
105
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
106 yh=FPDF.get_y(pdf)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
107 pdf.image(images[5],x=1.5,w=3.65, h=2.65)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
108 pdf.image(images[7],x=6.5,y=yh, w=3.65, h=2.65)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
109 pdf.ln(0.5)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
110 yh=FPDF.get_y(pdf)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
111 pdf.image(images[6],x=1.5,w=3.65, h=2.65)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
112 pdf.image(images[8],x=6.5,y=yh, w=3.65, h=2.65)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
113
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
114 pdf.close()
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
115 pdf.output('report1.pdf','F')
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
116
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
117
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
118
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
119
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
120 #############################################################################################################################################################3
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
121
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
122 def pdf_after_DE(analysis,top):
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
123
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
124 # Import FPDF class
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
125 from fpdf import FPDF
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
126
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
127 # Import glob module to find all the files matching a pattern
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
128 import glob
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
129
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
130 # Image extensions
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
131 if analysis=="2":
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
132 image_extensions = ("tem.png","a2.png","non.png")
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
133 else:
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
134 image_extensions = ("tem.png","a2.png")
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
135
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
136 # This list will hold the images file names
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
137 images = []
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
138
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
139 # Build the image list by merging the glob results (a list of files)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
140 # for each extension. We are taking images from current folder.
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
141 for extension in image_extensions:
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
142 images.extend(glob.glob(extension))
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
143 #sys.exit(images)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
144 # Create instance of FPDF class
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
145 pdf = FPDF('P', 'in', 'letter')
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
146 # Add new page. Without this you cannot create the document.
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
147 pdf.add_page()
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
148 # Set font to Arial, 'B'old, 16 pts
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
149 pdf.set_font('Arial', 'B', 16.0)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
150
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
151 # Page header
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
152 pdf.cell(pdf.w-0.5, 0.5, 'Differential expression of miRNAs and Isoforms',align='C')
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
153 #pdf.ln(0.25)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
154
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
155 pdf.ln(0.7)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
156 pdf.set_font('Arial','B', 12.0)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
157 if "tem.png" in images:
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
158 pdf.cell(pdf.w-0.5, 0.5, 'Top '+top+' most differentially expressed miRNA and template isoforms',align='C')
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
159 # Smaller font for image captions
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
160 pdf.set_font('Arial', '', 10.0)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
161 # Image caption
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
162 pdf.ln(0.4)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
163 pdf.image(images[images.index("tem.png")],x=0.8, w=7, h=8)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
164 pdf.ln(0.3)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
165 pdf.set_font('Arial','B', 12.0)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
166 else:
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
167 print("WARNING: There aren't miRNAs which fullfiled these criteria" )
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
168
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
169 if "non.png" in images and analysis=="2":
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
170 if "tem.png" in images: pdf.add_page()
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
171 pdf.ln(0.7)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
172 pdf.cell(pdf.w-0.5, 0.5, 'Top '+top+' most differentially expressed non-template isomiRs',align='C')
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
173 pdf.ln(0.4)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
174 pdf.image(images[images.index("non.png")],x=0.5, w=7.5, h=6.5)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
175 else:
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
176 print("WARNING: There aren't non-template miRNAs which fullfiled these criteria" )
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
177
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
178
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
179 if "a2.png" in images:
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
180 if len(images)>=2: pdf.add_page()
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
181 pdf.ln(0.5)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
182 pdf.cell(pdf.w-0.5, 0.5, 'Top '+top+' most differentially expressed miRNAs and isomiRs grouped by arm',align='C')
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
183 pdf.ln(0.4)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
184 pdf.image(images[images.index("a2.png")],x=0.8, w=7, h=8)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
185 pdf.ln(0.3)
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
186 else:
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
187 print("WARNING: There aren't non-template miRNAs which fullfiled these criteria" )
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
188
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
189
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
190 pdf.output('report2.pdf', 'F')
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
191
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
192
edec16f9566a Uploaded
glogobyte
parents:
diff changeset
193