ColorPicker.cs

Go to the documentation of this file.

using UnityEngine;
using System;
using System.Collections;
using System.Collections.Generic;
using Molecule.Model;



public class ColorPicker{
    
    private Texture2D m_aTexture;
    private Rect m_text_area = new Rect(5, 45, 228, 228);
    private Rect m_close_area = new Rect(5, 20, 50, 20);
    private Rect m_r_area = new Rect(60, 20, 50, 20);
    private Rect m_g_area = new Rect(115, 20, 50, 20);
    private Rect m_b_area = new Rect(170, 20, 50, 20);
    private int m_r_value = 255;
    private int m_g_value = 255;
    private int m_b_value = 255;
    private Rect m_activeArea;
    private Rect m_rect;
    private string m_title;
    private bool m_enabled;
    private int callId;
    public bool changed = false;
    
    private List<string> m_atoms = new List<string>();
    private string m_residue = "All";
    private string m_chain = "All";
    private Camera mainCamera = null;

    public bool enabled
    {
        get
        {
            return m_enabled;
        }
        set
        {
            m_enabled = value;
        }
    }
    //We use a ColorObject to reference a variable from outside
    //Impossible to do with UnityEngine.Color as it is a Struct
    private ColorObject m_colorObj;
    public ColorObject color
    {
        get
        {
            return m_colorObj;
        }
        set
        {
            m_colorObj = value;
        }
    }
    
    public ColorPicker(int xstart,int ystart,int width, int height, 
        ColorObject colorObj, List<string> atoms, string residue = "All", string chain = "All", 
        string title="Color Picker", int caller=-1)
    {
        m_colorObj = colorObj;
        m_atoms = atoms;
        m_residue = residue;
        m_chain = chain;
        m_r_value = (int)(colorObj.color.r * 255f);
        m_g_value = (int)(colorObj.color.g * 255f);
        m_b_value = (int)(colorObj.color.b * 255f);
        m_rect = new Rect(xstart,ystart,width,height);
        m_aTexture = (Texture2D)Resources.Load("ImprovedColorPicker");
        m_title=title;
        this.enabled = true;
        m_activeArea = new Rect(m_rect.x + m_text_area.x, m_rect.y + m_text_area.y, m_text_area.width, m_text_area.height);
        callId = caller;
    }   

    public ColorPicker(Rect r, 
        ColorObject colorObj, List<string> atoms, string residue = "All", string chain = "All", 
        string title="Color Picker", int caller=-1)
    {
        m_colorObj = colorObj;
        m_atoms = atoms;
        m_residue = residue;
        m_chain = chain;
        m_r_value = (int)(colorObj.color.r * 255f);
        m_g_value = (int)(colorObj.color.g * 255f);
        m_b_value = (int)(colorObj.color.b * 255f);
        m_rect = r;
        m_aTexture = (Texture2D)Resources.Load("ImprovedColorPicker");
        m_title=title;
        this.enabled = true;
        m_activeArea = new Rect(m_rect.x + m_text_area.x, m_rect.y + m_text_area.y, m_text_area.width, m_text_area.height);
        callId = caller;
    }

    public void SwitchEnable(int caller)
    {
        if(caller == callId)
            enabled = !enabled;
    }

    public void OnGUI()
    {
        if(enabled)
        {
            
            if(mainCamera == null)
                mainCamera = Camera.main;
            GUI.Window(60, m_rect, loadColor, m_title);

            Vector3 mousePos = Input.mousePosition;
            mousePos.y = Screen.height - mousePos.y;
            if(m_activeArea.Contains(mousePos) && Input.GetMouseButton(0) && GUIUtility.hotControl == 0)
            {       
                int X = (int)((mousePos.x - m_activeArea.x) / m_activeArea.width * m_aTexture.width);
                int Y = (int)((mousePos.y - m_activeArea.y) / m_activeArea.height * m_aTexture.height);
//              Debug.Log(X + " " + Y + " " + mousePos.x + " " + mousePos.y);
                m_r_value = (int)(m_aTexture.GetPixel(X,-Y).r * 255);
                m_g_value = (int)(m_aTexture.GetPixel(X,-Y).g * 255);
                m_b_value = (int)(m_aTexture.GetPixel(X,-Y).b * 255);
//              Debug.Log("Color: "+m_color);
                changed = true;
            }
            else
                changed = false;
            if(m_r_value < 0)
                m_r_value = 0;
            if(m_r_value > 255)
                m_r_value = 255;
            if(m_g_value < 0)
                m_g_value = 0;
            if(m_g_value > 255)
                m_g_value = 255;
            if(m_b_value < 0)
                m_b_value = 0;
            if(m_b_value > 255)
                m_b_value = 255;
            m_colorObj.color = new Color((float)m_r_value/255f, (float)m_g_value/255f, (float)m_b_value/255f, 1f);
            
            if(m_atoms != null){
                GenericManager manager = UnityMolMain.getCurrentAtomManager();
                if(!UI.GUIMoleculeController.toggle_NA_CLICK){
                    manager.SetColor(m_colorObj.color, m_atoms, m_residue, m_chain);
                }
                else{  //Atom selection 

                    //TODO : Do something smarter like a list of atom numbers
                    if(UI.UIData.atomtype == UI.UIData.AtomType.optihb){
                        foreach(GameObject obj in mainCamera.GetComponent<ClickAtom>().objList){
                            string toparse = obj.name.Split(new string[] {"Collider_Atom_"}, StringSplitOptions.None)[1];
                            int atomnumber = int.Parse(toparse);

                            manager.SetColor(m_colorObj.color, atomnumber);

                        }
                    }
                    else{
                        foreach(GameObject obj in mainCamera.GetComponent<ClickAtom>().objList)
                            manager.SetColor(m_colorObj.color, (int)obj.GetComponent<BallUpdate>().number);
                    }

                }
            }
            
        }
    }
    
    public void loadColor(int a){

        if(GUI.Button(m_close_area, "Close"))
            UI.GUIMoleculeController.m_colorPicker = null;
        string temp;
        temp = GUI.TextField(m_r_area, m_r_value.ToString());
        int.TryParse(temp, out m_r_value);
        temp = GUI.TextField(m_g_area, m_g_value.ToString());
        int.TryParse(temp, out m_g_value);
        temp = GUI.TextField(m_b_area, m_b_value.ToString());
        int.TryParse(temp, out m_b_value);
        GUI.DrawTexture(m_text_area, m_aTexture, ScaleMode.ScaleToFit, true, 0F);
        if (Event.current.type == EventType.Repaint)
            MoleculeModel.newtooltip = GUI.tooltip;
        GUI.DragWindow();
        
    }

}