tanu/LSHTM_analysis
1
0
Fork 0
Code Issues Pull requests Releases Wiki Activity

allow position_annotation to specify colours

Browse source
This commit is contained in:
Tanushree Tunstall 2022-08-25 19:09:56 +01:00
parent 2cbc460f87
commit 79b251047d
1 changed files with 25 additions and 21 deletions
Download patch file Download diff file Expand all files Collapse all files

46
scripts/functions/position_annotation.R
View file

@ -8,7 +8,11 @@ position_annotation=function(plot_df,
active_aa_pos=1:100, active_aa_pos=1:100,
aa_pos_lig1=1:100, aa_pos_lig1=1:100,
aa_pos_lig2=1:100, aa_pos_lig2=1:100,
aa_pos_lig3=1:100 aa_pos_lig3=1:100,
drug_colour='green',
lig1_colour='slategrey',
lig2_colour='navyblue',
lig3_colour='purple'
) )
{ {
x_ats = 12 x_ats = 12
@ -33,20 +37,20 @@ position_annotation=function(plot_df,
plot_df$bg_all = plot_df$col_aa plot_df$bg_all = plot_df$col_aa
plot_df$bg_all = ifelse(plot_df[["position"]]%in%aa_pos_drug, plot_df$bg_all = ifelse(plot_df[["position"]]%in%aa_pos_drug,
"green", plot_df$bg_all) "drug", plot_df$bg_all)
plot_df$col_bg1 = plot_df$bg_all plot_df$col_bg1 = plot_df$bg_all
plot_df$col_bg1 = ifelse(plot_df[["position"]]%in%aa_pos_lig1, plot_df$col_bg1 = ifelse(plot_df[["position"]]%in%aa_pos_lig1,
"slategrey", plot_df$col_bg1) "lig1", plot_df$col_bg1)
plot_df$col_bg2 = plot_df$col_bg1 plot_df$col_bg2 = plot_df$col_bg1
plot_df$col_bg2 = ifelse(plot_df[["position"]]%in%aa_pos_lig2, plot_df$col_bg2 = ifelse(plot_df[["position"]]%in%aa_pos_lig2,
"navyblue", plot_df$col_bg2) "lig2", plot_df$col_bg2)
plot_df$col_bg3 = plot_df$col_bg2 plot_df$col_bg3 = plot_df$col_bg2
plot_df$col_bg3 = ifelse(plot_df[["position"]]%in%aa_pos_lig3 plot_df$col_bg3 = ifelse(plot_df[["position"]]%in%aa_pos_lig3
, "purple", plot_df$col_bg3) , "lig3", plot_df$col_bg3)
# the call to generate_distance_colour_map should probably be # the call to generate_distance_colour_map should probably be
# wherever the outer DF is built, and not here. # wherever the outer DF is built, and not here.
@ -78,21 +82,21 @@ position_annotation=function(plot_df,
#scale_x_discrete("Position", labels=factor(plot_df$position)) + #scale_x_discrete("Position", labels=factor(plot_df$position)) +
scale_color_manual(values = c( scale_color_manual(values = c(
"brown"="brown", "brown"="brown",
"green"="#00ff00", "drug"=drug_colour,
"transparent"="transparent", "transparent"="transparent",
"slategrey"="#2f4f4f", "lig1"=lig1_colour,
"navyblue"="#000080", "lig2"=lig2_colour,
"purple"="#a020f0" "lig3"=lig3_colour
), ),
#expand=c(0,0) #expand=c(0,0)
) + ) +
scale_fill_manual(values = c( scale_fill_manual(values = c(
"brown"="brown", "brown"="brown",
"green"="#00ff00", "drug"=drug_colour,
"transparent"="transparent", "transparent"="transparent",
"slategrey"="#2f4f4f", "lig1"=lig1_colour,
"navyblue"="#000080", "lig2"=lig2_colour,
"purple"="#a020f0" "lig3"=lig3_colour
), ),
#expand=c(0,0) #expand=c(0,0)
) + ) +
@ -117,21 +121,21 @@ position_annotation=function(plot_df,
#scale_x_discrete("Position", labels=factor(plot_df$position)) + #scale_x_discrete("Position", labels=factor(plot_df$position)) +
scale_color_manual(values = c( scale_color_manual(values = c(
"brown"="brown", "brown"="brown",
"green"="#00ff00", "drug"=drug_colour,
"transparent"="transparent", "transparent"="transparent",
"slategrey"="#2f4f4f", "lig1"=lig1_colour,
"navyblue"="#000080", "lig2"=lig2_colour,
"purple"="#a020f0" "lig3"=lig3_colour
), ),
#expand=c(0,0) #expand=c(0,0)
) + ) +
scale_fill_manual(values = c( scale_fill_manual(values = c(
"brown"="brown", "brown"="brown",
"green"="#00ff00", "drug"=drug_colour,
"transparent"="transparent", "transparent"="transparent",
"slategrey"="#2f4f4f", "lig1"=lig1_colour,
"navyblue"="#000080", "lig2"=lig2_colour,
"purple"="#a020f0" "lig3"=lig3_colour
), ),
#expand=c(0,0) #expand=c(0,0)
) + ) +