annotate viz_graphs.py @ 27:a6f94856a282 draft

Uploaded
author glogobyte
date Sun, 05 Dec 2021 13:29:56 +0000
parents
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
27
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
1 # Import FPDF class
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
2 from fpdf import FPDF
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
3
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
4 # Import glob module to find all the files matching a pattern
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
5 import glob
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
6
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
7 #############################################################################################################################################################3
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
8
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
9 def pdf_after_DE(analysis,top,font_path,iso_star_fl,non_star_fl):
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
10
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
11 # Image extensions
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
12 if analysis=="2":
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
13 image_extensions = ("tem.png","a2.png","non.png")
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
14 else:
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
15 image_extensions = ("tem.png","a2.png")
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
16
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
17 # This list will hold the images file names
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
18 images = []
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
19
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
20 # Build the image list by merging the glob results (a list of files)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
21 # for each extension. We are taking images from current folder.
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
22 for extension in image_extensions:
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
23 images.extend(glob.glob(extension))
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
24
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
25 # Create instance of FPDF class
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
26 pdf = FPDF('P', 'in', 'letter')
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
27 pdf.add_font('uni-arial', '', font_path+"/arial-unicode-ms.ttf", uni=True)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
28 # Add new page. Without this you cannot create the document.
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
29 pdf.add_page()
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
30 # Set font to Arial, 'B'old, 16 pts
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
31 pdf.set_font('Arial', 'B', 16.0)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
32
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
33 # Page header
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
34 pdf.cell(pdf.w-0.5, 0.5, 'Differential expression of miRNAs and isomiRs',align='C')
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
35 #pdf.ln(0.25)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
36
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
37 pdf.ln(0.7)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
38 pdf.set_font('Arial','B', 12.0)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
39 if "tem.png" in images:
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
40 pdf.cell(pdf.w-0.5, 0.5, 'Top '+top+' differentially expressed miRNA and templated isoforms',align='C')
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
41 # Smaller font for image captions
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
42 pdf.set_font('Arial', '', 10.0)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
43 # Image caption
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
44 pdf.ln(0.4)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
45 pdf.image(images[images.index("tem.png")],x=0.8, w=7, h=8)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
46 pdf.ln(0.3)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
47 if iso_star_fl==1:
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
48 pdf.set_font('uni-arial', '', 9.0)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
49 pdf.cell(0.2)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
50 pdf.cell(3.0, 0.0, " ★ IsomiRs potentially generated from multiple loci")
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
51 #pdf.set_font('Arial','B', 12.0)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
52 else:
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
53 print("WARNING: There aren't miRNAs which fullfiled these criteria" )
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
54 pdf.set_font('Arial','B', 12.0)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
55 if "non.png" in images and analysis=="2":
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
56 if "tem.png" in images: pdf.add_page()
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
57 pdf.ln(0.7)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
58 pdf.cell(pdf.w-0.5, 0.5, 'Top '+top+' differentially expressed non-templated isomiRs',align='C')
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
59 pdf.ln(0.4)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
60 pdf.image(images[images.index("non.png")],x=0.8, w=7, h=8)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
61 pdf.ln(0.3)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
62 if non_star_fl==1:
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
63 pdf.set_font('uni-arial', '', 9.0)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
64 pdf.cell(0.2)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
65 pdf.cell(3.0, 0.0, " ★ IsomiRs potentially generated from multiple loci")
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
66
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
67 #pdf.image(images[images.index("non.png")],x=0.5, w=7.5, h=6.5)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
68 else:
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
69 print("WARNING: There aren't non-template miRNAs which fullfiled these criteria" )
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
70
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
71 pdf.set_font('Arial','B', 12.0)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
72 if "a2.png" in images:
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
73 if len(images)>=2: pdf.add_page()
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
74 pdf.ln(0.5)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
75 pdf.cell(pdf.w-0.5, 0.5, 'Top differentially expressed miRNAs and isomiRs grouped by arm',align='C')
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
76 pdf.ln(0.4)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
77 pdf.image(images[images.index("a2.png")],x=0.8, w=7, h=8)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
78 pdf.ln(0.3)
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
79 else:
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
80 print("WARNING: There aren't non-template miRNAs which fullfiled these criteria" )
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
81
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
82
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
83 pdf.output('report2.pdf', 'F')
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
84
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
85
a6f94856a282 Uploaded
glogobyte
parents:
diff changeset
86