From 204b998abdc2df8bfeccdbd47a08d6a76c103369 Mon Sep 17 00:00:00 2001
From: Vitaliy Mysak <vmysak@bccfe.ca>
Date: Fri, 12 Jul 2024 15:04:06 -0700
Subject: [PATCH] proviral_landscape_plot: add colours for new error codes

---
 .../proviral_landscape_plot/proviral_landscape_plot.py    | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/alldata/bblab_site/tools/proviral_landscape_plot/proviral_landscape_plot.py b/alldata/bblab_site/tools/proviral_landscape_plot/proviral_landscape_plot.py
index 0f55e6f..c8b5b4e 100644
--- a/alldata/bblab_site/tools/proviral_landscape_plot/proviral_landscape_plot.py
+++ b/alldata/bblab_site/tools/proviral_landscape_plot/proviral_landscape_plot.py
@@ -15,12 +15,16 @@
                    'Premature Stop': "#CC6677",
                    'Chimera': "#AA4499",
                    'Scrambled': "#882255",
+                   'Large Insertion': "#DDCC77",
+                   'Frameshift': "#DDAA77",
                    }
+
 # colors are chosen from Paul Tol's muted color scheme, which is color-blind safe
 # if another defect color is needed, this one is recommended: #DDCC77
 HIGHLIGHT_COLORS = {'Defect Region': "black",
                     'Inverted Region': "#AFAFAF",
                     }
+
 DEFECT_TYPE = {'LargeDeletion': 'Large Deletion',
                'LongDeletion': 'Large Deletion',
                'InternalInversion': 'Inversion',
@@ -39,8 +43,8 @@
                'Inferred_PrematureStopORInframeDEL_GagNoATG': 'Premature Stop',
                'InternalStopInOrf': "Premature Stop",
                'DeletionInOrf': "Premature Stop",
-               'InsertionInOrf': "Premature Stop",
-               'FrameshiftInOrf': "Premature Stop",
+               'InsertionInOrf': "Large Insertion",
+               'FrameshiftInOrf': "Frameshift",
                '5DEFECT': "5' Defect",
                '5DFECT_IntoGag': "5' Defect",  # this is a typo in HIVSeqinR
                '5DEFECT_GagNoATGGagPassed': "5' Defect",