VolumetricDensity.cs

Go to the documentation of this file.

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

public class VolumetricDensity : Volumetric {   
    private GameObject pdb2den;
    private PDBtoDEN genDensity;
    private float maxDensity = float.MinValue;
    private float minDensity = float.MaxValue;
    private float densityAmplitude;
    
    private static Vector3 inverseDelta ;
    private static Vector3 fudgeFactor;
    
    public override void Init () {
        // First we use PDBtoDen to create a density grid.
        if(pdb2den == null){
            pdb2den = new GameObject();
            pdb2den.name = "pdb2den";
            pdb2den.AddComponent<PDBtoDEN>();
        }
        genDensity = pdb2den.GetComponent<PDBtoDEN>();
        
        float delta_scalar = 2.0f;
        delta = new Vector3(delta_scalar, delta_scalar, delta_scalar);
        genDensity.TranPDBtoDEN (delta_scalar);
        delta = PDBtoDEN.delta;
        fudgeFactor = PDBtoDEN.fudgeFactor;
        
        // We get that grid and its delta.
        density = PDBtoDEN.GridS;
        //delta = PDBtoDEN.delta;
        
        inverseDelta =  new Vector3(    1f/delta.x,
                                        1f/delta.y,
                                        1f/delta.z  );
        
        // We need to refresh the molecule's origin when it's not the first molecule for which we generate a volumetric density.
        origin = MoleculeModel.MinValue;
        
        // We determine the amplitude of the density.
        foreach(float f in density)
        {
            if(f > maxDensity)
                maxDensity = f;
            if(f < minDensity)
                minDensity = f;
        }
        densityAmplitude = maxDensity - minDensity ;
        
        // Creating the dynamic particle list, building the static particle array,
        // setting it to the particle system, and enabling the renderer.
        SetParticleSystem();
    } // End of Init
    
    
    public override void CreatePoints() {
        // Getting the size of the density grid.
        int dim0 = density.GetLength(0);
        int dim1 = density.GetLength(1);
        int dim2 = density.GetLength(2);

        Vector3 indices = Vector3.zero;

        for(int x=0; x<dim0; x++) {
            for(int y=0; y<dim1; y++) {
                for(int z=0; z<dim2; z++) {
                    indices.x = x;
                    indices.y = y;
                    indices.z = z;
                    
                    Color c = Color.blue; // arbitrary
                    c.a = (density[x,y,z] / densityAmplitude); // Somewhat arbitrary too, but guarantees 0 <= c.a <= 1.
                    
                    // Here, we cull.
                    float size = particleScale * inverseDelta.x; // Assumes delta.x == delta.y == delta.z. For now, that's true, but it might change.
                    if (c.a > ALPHA_THRESHOLD) {
                        ParticleSystem.Particle particle = new ParticleSystem.Particle();
                        Vector3 p = new Vector3(0f,0f,0f);
                        p = Vector3.Scale(inverseDelta,(indices - fudgeFactor)) + origin;
                        particle.position = p;
                        particle.startColor = c;
                        particle.startSize = size;
                        dynPoints.Add(particle);
                    }
                }
            }
        }
    } // End of CreatePoints
    
}