schemeId = {
return NGL.ColormakerRegistry.addScheme(function (params) {
this.atomColor = function (atom) {
if(selected_sites[atom.residue.resno]) {
console.log("# Site: " + atom.residue.resno)
let site_color = "gray";
console.log("Class: " + selected_sites[atom.residue.resno].class)
if (selected_sites[atom.residue.resno].class == "Purifying") {
site_color = "red";
}
if (selected_sites[atom.residue.resno].class == "Diversifying") {
site_color = "blue";
}
if (selected_sites[atom.residue.resno].class == "Invariable") {
site_color = "yellow";
}
return color_hash[site_color];
}
else {
return color_hash["gray"]
}
}
})
}