HStickMeshManager.cs

Go to the documentation of this file.

using UnityEngine;
using System.Collections;
using System.Collections.Generic;
using System.Linq;
using UI;
using UnityEngine.Rendering;


public class HStickMeshManager : GenericManager {

    public GameObject[] hstick_meshes = new GameObject[0];
    public List<int> texturesToUpdate;
    public Material[] materialsSticks;
    public Texture2D[] texturesSticks;
    public Dictionary<int,KeyValuePair<int,int> > coordStickTexture;
    public static Texture texture;
    public static bool resetBrightness = false;
    public static bool resetShrink = false;
    public static bool resetScale = false;
    public static float scaleSticks = 1.0f;
    public static bool resetRadii = false;
    public static bool resetColors = false;
    public static bool resetTexture = false;
    
    public static float shininess = 0.0f;
    public static bool resetShininess = false;

    HStickMeshManager() {
        hstick_meshes = new GameObject[0];
    }

    public override void Init () {
        enabled = true;
        texturesToUpdate = new List<int>();
        materialsSticks = new Material[hstick_meshes.Length];
        texturesSticks = new Texture2D[hstick_meshes.Length];
        for(int i=0;i<hstick_meshes.Length;i++){
            materialsSticks[i] = hstick_meshes[i].GetComponent<Renderer>().material;
            texturesSticks[i] = (Texture2D)materialsSticks[i].mainTexture;
        }
        isInit = true;
        UnityMolMain.setCurrentBondManager(this);
    }

    public override void DestroyAll() {
        for(int i=0;i<hstick_meshes.Length;i++)
            GameObject.Destroy(hstick_meshes[i]);
        hstick_meshes = new GameObject[0];
        isInit = false;

    }

    public override void ToggleDistanceCueing(bool enabling) {
        float attenuation;
        attenuation = enabling? 1f : 0f;
        for (int i=0; i<hstick_meshes.Length; i++)
            hstick_meshes[i].GetComponent<Renderer>().material.SetFloat("_Attenuation", attenuation);
    }

    public override void SetColor(Color col, List<string> atom, string residue = "All", string chain = "All"){
    }


    public override void SetColor(Color col, int atomNum) {
    }

    public override void SetRadii(List<string> atom, string residue = "All", string chain = "All"){

    }
    public override void SetRadii(int id){

    }

    public override GameObject GetBall(int id){
        return null;
    }
    
    public override void ResetPositions(){

        KeyValuePair<int,int> res = new KeyValuePair<int,int>();
        float[] atomOne;
        float[] atomTwo;

        
        for(int i=0;i<Molecule.Model.MoleculeModel.bondEPList.Count;i++){
            int idAtom1 = Molecule.Model.MoleculeModel.bondEPList[i][0];
            int idAtom2 = Molecule.Model.MoleculeModel.bondEPList[i][1];
            if(coordStickTexture.TryGetValue(i,out res)){
                atomOne = Molecule.Model.MoleculeModel.atomsLocationlist[idAtom1];
                atomTwo = Molecule.Model.MoleculeModel.atomsLocationlist[idAtom2];

                texturesSticks[res.Key].SetPixel(4,res.Value,EncodeFloatToColor.EncodeFloatRGBA(atomOne[0]));
                texturesSticks[res.Key].SetPixel(5,res.Value,EncodeFloatToColor.EncodeFloatRGBA(atomOne[1]));
                texturesSticks[res.Key].SetPixel(6,res.Value,EncodeFloatToColor.EncodeFloatRGBA(atomOne[2]));

                texturesSticks[res.Key].SetPixel(7,res.Value,EncodeFloatToColor.EncodeFloatRGBA(atomTwo[0]));
                texturesSticks[res.Key].SetPixel(8,res.Value,EncodeFloatToColor.EncodeFloatRGBA(atomTwo[1]));
                texturesSticks[res.Key].SetPixel(9,res.Value,EncodeFloatToColor.EncodeFloatRGBA(atomTwo[2]));
                texturesToUpdate.Add(res.Key);
            }
        }
        ApplyTextures();

    }

    public override void ResetIMDSimulationPositions(){

        KeyValuePair<int,int> res = new KeyValuePair<int,int>();
        Vector3 atomOne = Vector3.zero;
        Vector3 atomTwo = Vector3.zero;

        
        for(int i=0;i<Molecule.Model.MoleculeModel.bondEPList.Count;i++){
            int idAtom1 = Molecule.Model.MoleculeModel.bondEPList[i][0];
            int idAtom2 = Molecule.Model.MoleculeModel.bondEPList[i][1];
            if(coordStickTexture.TryGetValue(i,out res)){
                atomOne = Molecule.Model.MoleculeModel.atomsIMDSimulationLocationlist[idAtom1];
                atomTwo = Molecule.Model.MoleculeModel.atomsIMDSimulationLocationlist[idAtom2];

                texturesSticks[res.Key].SetPixel(4,res.Value,EncodeFloatToColor.EncodeFloatRGBA(atomOne.x));
                texturesSticks[res.Key].SetPixel(5,res.Value,EncodeFloatToColor.EncodeFloatRGBA(atomOne.y));
                texturesSticks[res.Key].SetPixel(6,res.Value,EncodeFloatToColor.EncodeFloatRGBA(atomOne.z));

                texturesSticks[res.Key].SetPixel(7,res.Value,EncodeFloatToColor.EncodeFloatRGBA(atomTwo.x));
                texturesSticks[res.Key].SetPixel(8,res.Value,EncodeFloatToColor.EncodeFloatRGBA(atomTwo.y));
                texturesSticks[res.Key].SetPixel(9,res.Value,EncodeFloatToColor.EncodeFloatRGBA(atomTwo.z));
                texturesToUpdate.Add(res.Key);
            }
        }
        ApplyTextures();

    }



    public override void DisableRenderers() {
        for (int i=0; i< hstick_meshes.Length; i++)
            hstick_meshes[i].GetComponent<Renderer>().enabled = false;
    }


    public override void EnableRenderers(){
        UnityMolMain.setCurrentBondManager(this);
        for (int i=0; i< hstick_meshes.Length; i++)
            hstick_meshes[i].GetComponent<Renderer>().enabled = true;

    }

    public override void EnableShadows(){
        for (int i=0; i< hstick_meshes.Length; i++){
            hstick_meshes[i].GetComponent<Renderer>().shadowCastingMode = ShadowCastingMode.On;
            hstick_meshes[i].GetComponent<Renderer>().receiveShadows = true;
            hstick_meshes[i].GetComponent<Renderer>().material.shader = Shader.Find("FvNano/Sticks HyperBalls Shadow Merged");
        }
    }
    public override void DisableShadows(){
        for (int i=0; i< hstick_meshes.Length; i++){
            hstick_meshes[i].GetComponent<Renderer>().shadowCastingMode = ShadowCastingMode.Off;
            hstick_meshes[i].GetComponent<Renderer>().receiveShadows = false;
            hstick_meshes[i].GetComponent<Renderer>().material.shader = Shader.Find("FvNano/Sticks HyperBalls Merged");
        }
    }
    public override void showHydrogens(bool hide){


        KeyValuePair<int,int> res = new KeyValuePair<int,int>();

        
        for(int i=0;i<Molecule.Model.MoleculeModel.bondEPList.Count;i++){
            int idAtom1 = Molecule.Model.MoleculeModel.bondEPList[i][0];
            int idAtom2 = Molecule.Model.MoleculeModel.bondEPList[i][1];
            if(Molecule.Model.MoleculeModel.atomsTypelist[idAtom1].type == "H" ||
                       Molecule.Model.MoleculeModel.atomsTypelist[idAtom2].type == "H") {
                if(coordStickTexture.TryGetValue(i,out res)){
                    if(hide)
                        texturesSticks[res.Key].SetPixel(10,res.Value,Vector4.zero);
                    else
                        texturesSticks[res.Key].SetPixel(10,res.Value,Vector4.one);

                    texturesToUpdate.Add(res.Key);
                }
            }
        }
        ApplyTextures();

    }

    public void ResetBrightness(){
        for(int i=0;i<materialsSticks.Length;i++)
            materialsSticks[i].SetFloat("_Brightness",HBallManager.brightness);
        resetBrightness = false;
    }
    public void ResetTexture(){
        for(int i=0;i<materialsSticks.Length;i++){
            materialsSticks[i].SetTexture("_MatCap",texture);
            materialsSticks[i].SetTexture("_MatCap2",texture);

        }
        resetTexture = false;
    }
    public void ResetShrink(){
        for(int i=0;i<materialsSticks.Length;i++)
            materialsSticks[i].SetFloat("_Shrink",RepresentationOldGUI.shrink);
        resetShrink = false;
    }

    public void ResetScale(){
        for(int i=0;i<materialsSticks.Length;i++)
            materialsSticks[i].SetFloat("_Scale",RepresentationOldGUI.linkScale);
        resetScale = false;
    }
    
    public override void ResetRadii(){
        KeyValuePair<int,int> res = new KeyValuePair<int,int>();

        
        for(int i=0;i<Molecule.Model.MoleculeModel.bondEPList.Count;i++){
            int idAtom1 = Molecule.Model.MoleculeModel.bondEPList[i][0];
            int idAtom2 = Molecule.Model.MoleculeModel.bondEPList[i][1];
            if(coordStickTexture.TryGetValue(i,out res)){
                float oriRadius = Molecule.Model.MoleculeModel.atomsTypelist[idAtom1].radius;
                texturesSticks[res.Key].SetPixel(0,res.Value,EncodeFloatToColor.EncodeFloatRGBA(oriRadius
                                                    * RepresentationOldGUI.globalRadius
                                                    * Molecule.Model.MoleculeModel.atomsLocalScaleList[idAtom1]/100));

                oriRadius = Molecule.Model.MoleculeModel.atomsTypelist[idAtom2].radius;
                texturesSticks[res.Key].SetPixel(1,res.Value,EncodeFloatToColor.EncodeFloatRGBA(oriRadius
                                                    * RepresentationOldGUI.globalRadius
                                                    * Molecule.Model.MoleculeModel.atomsLocalScaleList[idAtom2]/100));
                texturesToUpdate.Add(res.Key);
            }
        }

        ApplyTextures();
        resetRadii = false;
    }

    public void ResetColors(){
        KeyValuePair<int,int> res = new KeyValuePair<int,int>();

        
        for(int i=0;i<Molecule.Model.MoleculeModel.bondEPList.Count;i++){
            int idAtom1 = Molecule.Model.MoleculeModel.bondEPList[i][0];
            int idAtom2 = Molecule.Model.MoleculeModel.bondEPList[i][1];
            Color colAtom1 = Molecule.Model.MoleculeModel.atomsColorList[idAtom1];
            Color colAtom2 = Molecule.Model.MoleculeModel.atomsColorList[idAtom2];
            if(coordStickTexture.TryGetValue(i,out res)){
                texturesSticks[res.Key].SetPixel(2,res.Value,colAtom1);
                texturesSticks[res.Key].SetPixel(3,res.Value,colAtom2);
                texturesToUpdate.Add(res.Key);
            }
        }
        resetColors = false;
        ApplyTextures();
    }

    public void ResetShininess(){
        for(int i=0;i<materialsSticks.Length;i++)
            materialsSticks[i].SetFloat("_Shininess",shininess);
        resetShininess = false;
    }


    public void ApplyTextures(){
        List<int> idtex = texturesToUpdate.Distinct().ToList();
        foreach(int id in idtex){
            if(id != -1)
                texturesSticks[id].Apply(false,false);
        }

        texturesToUpdate.Clear();
    }

    public void hideBondForAtom(int idAtom,bool show){
        KeyValuePair<int,int> res = new KeyValuePair<int,int>();

        
        for(int i=0;i<Molecule.Model.MoleculeModel.bondEPList.Count;i++){
            int idAtom1 = Molecule.Model.MoleculeModel.bondEPList[i][0];
            int idAtom2 = Molecule.Model.MoleculeModel.bondEPList[i][1];
            if(idAtom1 == idAtom || idAtom2 == idAtom) {
                if(coordStickTexture.TryGetValue(i,out res)){
                    if(show)
                        texturesSticks[res.Key].SetPixel(10,res.Value,Vector4.one);
                    else
                        texturesSticks[res.Key].SetPixel(10,res.Value,Vector4.zero);

                    texturesToUpdate.Add(res.Key);  
                }
            }
        }
    }
    public void ResetVisibility(){
        KeyValuePair<int,int> res = new KeyValuePair<int,int>();
        for(int i=0;i<Molecule.Model.MoleculeModel.bondEPList.Count;i++){
            if(coordStickTexture.TryGetValue(i,out res)){
                texturesSticks[res.Key].SetPixel(10,res.Value,Vector4.one);
                texturesToUpdate.Add(res.Key);  
            }
        }
        ApplyTextures();
        

    }

    //Dirty but mandatory
    void Update(){
        // if(UIData.bondtype != UIData.BondType.optihs) {
        //  enabled = false;
        //  return;
        // }
        if(resetRadii)
            ResetRadii();
        if(resetBrightness)
            ResetBrightness();
        if(resetShrink)
            ResetShrink();
        if(resetScale)
            ResetScale();
        if(resetColors)
            ResetColors();
        if(resetTexture)
            ResetTexture();
        if(resetShininess)
            ResetShininess();
    }
}