Diff of /PhenPred/__init__.py [000000] .. [305123]

Switch to side-by-side view

--- a
+++ b/PhenPred/__init__.py
@@ -0,0 +1,134 @@
+#!/usr/bin/env python
+# Copyright (C) 2022 Emanuel Goncalves
+
+import matplotlib.pyplot as plt
+
+# Matplotlib config
+plt.rcParams["figure.figsize"] = [4, 4]
+plt.rcParams["figure.dpi"] = 300
+
+# Matplotlib set font to sans-times
+plt.rcParams["font.family"] = "sans-serif"
+plt.rcParams["font.sans-serif"] = "Arial"
+
+# Matplotlib set main axis font size
+plt.rcParams["axes.titlesize"] = 7
+
+# Matplotlib set legend font size
+plt.rcParams["legend.fontsize"] = 6
+
+# Matplotlib set legend title font size
+plt.rcParams["legend.title_fontsize"] = 6
+
+# Matplotlib set tick label font size
+plt.rcParams["axes.labelsize"] = 6
+
+# Matplotlib set tick label font size
+plt.rcParams["xtick.labelsize"] = 6
+plt.rcParams["ytick.labelsize"] = 6
+
+# Matplotlib set grid line width
+plt.rcParams["grid.linewidth"] = 0.5
+
+# Matplotlib ommit top and right spines
+plt.rcParams["axes.spines.top"] = False
+plt.rcParams["axes.spines.right"] = False
+
+# Matplotlib set grid line
+plt.rcParams["axes.grid"] = True
+
+# Matplotlib set grid line style
+plt.rcParams["grid.linestyle"] = "--"
+plt.rcParams["grid.linewidth"] = 0.15
+
+# Matplotlib set grid line color
+plt.rcParams["grid.color"] = "black"
+
+# Matplotlib set grid line alpha
+plt.rcParams["grid.alpha"] = 0.5
+
+# Matplotlib set legend frameon
+plt.rcParams["legend.frameon"] = False
+
+# Matplotlib set legend loc
+plt.rcParams["legend.loc"] = "best"
+
+# Matplotlib set axis below true
+plt.rcParams["axes.axisbelow"] = True
+
+# Matplotlib illustrator export
+plt.rcParams["pdf.fonttype"] = 42
+plt.rcParams["ps.fonttype"] = 42
+
+# Palette - tissue type
+PALETTE_TTYPE = {
+    "Lung": "#007fff",
+    "Prostate": "#665d1e",
+    "Stomach": "#ffbf00",
+    "Central Nervous System": "#fbceb1",
+    "Skin": "#ff033e",
+    "Bladder": "#ab274f",
+    "Haematopoietic and Lymphoid": "#d5e6f7",
+    "Kidney": "#7cb9e8",
+    "Thyroid": "#efdecd",
+    "Soft Tissue": "#8db600",
+    "Head and Neck": "#e9d66b",
+    "Ovary": "#b284be",
+    "Bone": "#b2beb5",
+    "Endometrium": "#10b36f",
+    "Breast": "#6e7f80",
+    "Pancreas": "#ff7e00",
+    "Peripheral Nervous System": "#87a96b",
+    "Cervix": "#c9ffe5",
+    "Large Intestine": "#9f2b68",
+    "Liver": "#00ffff",
+    "Vulva": "#008000",
+    "Esophagus": "#cd9575",
+    "Biliary Tract": "#72a0c1",
+    "Other tissue": "#a32638",
+    "Small Intestine": "#9966cc",
+    "Placenta": "#f19cbb",
+    "Testis": "#e32636",
+    "Adrenal Gland": "#3b7a57",
+    "Uterus": "#7a3b5e",
+    "Unknown": "#a32638",
+    "Eye": "#ff1493",
+}
+
+OMIC_PALETTE = {
+    "conditionals": "#4c72b0",
+    "copynumber": "#dd8452",
+    "drugresponse": "#55a868",
+    "metabolomics": "#c44e52",
+    "proteomics": "#8172b3",
+    "crisprcas9": "#937860",
+    "transcriptomics": "#da8bc3",
+    "methylation": "#8c8c8c",
+}
+
+OMIC_NAMES = dict(
+    copynumber="Copy number",
+    mutations="Mutations",
+    fusions="Fusions",
+    methylation="Methylation",
+    transcriptomics="Transcriptomics",
+    proteomics="Proteomics",
+    phosphoproteomics="Phosphoproteomics",
+    metabolomics="Metabolomics",
+    drugresponse="Drug response",
+    crisprcas9="CRISPR-Cas9",
+    growth="Growth",
+    conditionals="Conditionals",
+)
+
+
+def save_figure(path, extensions=["png", "pdf"], dpi=300, transparent=False, **kwargs):
+    for e in extensions:
+        plt.savefig(
+            path + "." + e,
+            bbox_inches="tight",
+            dpi=dpi,
+            transparent=transparent,
+            **kwargs,
+        )
+    plt.close("all")