VolumetricFields.cs

Go to the documentation of this file.

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

public class VolumetricFields : Volumetric {
    private float maxCharge = - float.MaxValue;
    private float minCharge = float.MaxValue;
    private float chargeAmplitude;
    private bool init = false;
    
    private ReadDX readDx;

    public override void Init() {
        if(init == false){
            // We get the ReadDX object and from it, the values we need.
            readDx = ElectrostaticManager.readdx;
            density = readDx._grid;
            delta = readDx.GetDelta();
            origin = readDx.GetOrigin();
            
            // This is needed to correctly place the particles.
            origin.x = -origin.x;
            
            // Getting the bounds and amplitude of the electrostatics field.
            foreach(float f in density) {
                if(f > maxCharge)
                    maxCharge = f;
                if(f < minCharge)
                    minCharge = f;
            }
            
            // ChargeAmplitude = largest absolute value
            if ( (-minCharge) > maxCharge)
                chargeAmplitude = -minCharge;
            else
                chargeAmplitude = maxCharge;
            
            Debug.Log("Amplitude: "+chargeAmplitude);
            
            // 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;
                    
                    float charge = density[x,y,z];
                    Color c;
                    if(charge > 0)
                        c = Color.blue;
                    else
                        c = Color.red;
                    c.a = Mathf.Abs((charge / chargeAmplitude)); // Somewhat arbitrary, but guarantees 0 <= c.a <= 1.
                    if(raise) {
                        c.a = c.a * ALPHA_FACTOR;
                        if (c.a > 1f)
                            c.a = 1f;
                    }
                    
                    // Here, we cull.
                    float size = 0.8f * particleScale / delta.x;
                    if (c.a > ALPHA_THRESHOLD) {
                        ParticleSystem.Particle particle = new ParticleSystem.Particle();
                        Vector3 p = new Vector3(0f,0f,0f);
                        p = origin + Vector3.Scale(indices, delta);
                        particle.position = p;
                        particle.startColor = c;
                        //Debug.Log(particle.color);
                        //pSystem.renderer.material.color = c;
                        //particle.size = particleScale;
                        particle.startSize = size;
                        dynPoints.Add(particle);    
                    }
                }
            }
        }
    } // End of CreatePoints
    
}