DepthCueing.cs

Go to the documentation of this file.

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


public class DepthCueing {
    private ReadDX readDx;
    private float[,,] depth;
    private static Vector3 invDelta;
    private static Vector3 origin = MoleculeModel.MinValue;
    private BallUpdate[] balls;
    private int xDim, yDim, zDim;
    private float depthAmplitude;
    
    private static float OFFSET = 1.0f;
    private static float MAX_DEPTH = -10.0f;
    private static List<Color> originalColors;
    
    public static bool isEnabled = false;
    public static bool reset = false;
    public static float POWER = 2;
    
    
    public DepthCueing() {
        // We get the ReadDX object and from it, the values we need.
        readDx = ElectrostaticManager.readdx;
        depth = readDx._grid;
        xDim = depth.GetLength(0);
        yDim = depth.GetLength(1);
        zDim = depth.GetLength(2);
        
        Debug.Log("Dimensions:");
        Debug.Log(xDim.ToString());
        Debug.Log(yDim.ToString());
        Debug.Log(zDim.ToString());
        
        Vector3 delta = readDx.GetDelta();
        invDelta = delta;
        
        invDelta.x = 1f/delta.x;
        invDelta.y = 1f/delta.y;
        invDelta.z = 1f/delta.z;
        
        origin = readDx.GetOrigin();
        
        // This is needed to correctly localize things. Unity's convention is different from our input data.
        origin.x = -origin.x;
        
        balls = GameObject.FindObjectsOfType(typeof(BallUpdate)) as BallUpdate[];
        
        BuildColorList();
        DepthCueing.reset = false;
    }
    
    private void BuildColorList() {
        originalColors = new List<Color>(MoleculeModel.atomsColorList);
    }
    
    
    private bool InBounds(int x, int y, int z){
        if(x<0 || y<0 || z<0)
            return false;
        else
            return true;
    }
    
    public void Darken() {
        int x, y, z;
        float pointDepth;
        float depthPercentage;
        float colorFactor;
        Vector3 position = Vector3.zero;
        Vector3 indices = Vector3.zero;
        foreach(BallUpdate bu in balls) {
            position = bu.transform.position;
            indices = Vector3.Scale(invDelta, (position - origin));
            x = (int)indices.x;
            x = -x; // Again, the X axis must be flipped.
            y = (int)indices.y;
            z = (int)indices.z;
            
            pointDepth = depth[x,y,z];
            
            if(pointDepth>=0f)
                continue; // The values that are inside the molecule are negative.
            
            pointDepth = pointDepth + OFFSET;
            
            if(pointDepth <= MAX_DEPTH)
                bu.atomcolor = Color.black;
            else {
                depthPercentage = Mathf.Abs(pointDepth/MAX_DEPTH);
                colorFactor = 1f - depthPercentage;
                colorFactor = Mathf.Pow(colorFactor, 2);
                //bu.atomcolor = bu.atomcolor * colorFactor * (1.25f + POWER/10f);
                MoleculeModel.atomsColorList[(int)bu.number] *= colorFactor * (1.25f + POWER/10f);
            }
        }
        BallUpdate.resetColors = true;
        isEnabled = true;
    }
    
    public void Revert() {
        Debug.Log("DepthCueing: Reverting to original colors");
        MoleculeModel.atomsColorList = originalColors;
        BallUpdate.resetColors = true;
        isEnabled = false;
    }
    
}