ShurikenParticleManager.cs

Go to the documentation of this file.

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


public class ShurikenParticleManager : GenericManager {
    public ParticleSystem pSystem;
    private Particle[] sourceParticles;
    private static Vector3 LITTLE_OFFSET = new Vector3(0f, 0f, -1f); // not sure why, but this is needed
    private ParticleSystem.Particle[] particles; // not exactly sure what the difference between Particle and ParticleSystem.Particle is
    public static int pNumber;
    //public bool test;
    public static float radiusFactor;   
    
    public override void Init () {
        Debug.Log("Shuriken Initializing");
        sourceParticles = MoleculeModel.p;
        pNumber = sourceParticles.Length;
        Debug.Log("Found: " + pNumber + " particles");
        particles = new ParticleSystem.Particle[pNumber];
        
        for(int i=0; i<pNumber; i++) {
            particles[i].position = sourceParticles[i].position + LITTLE_OFFSET;
            particles[i].startColor = sourceParticles[i].color;
            particles[i].startSize = sourceParticles[i].size * 0.5f * BallUpdate.radiusFactor;
        }
        
        //We draw the molecule HERE!
        pSystem.SetParticles(particles, pNumber);
        
        //CheckParticles();
        Renderer rd = pSystem.GetComponent<Renderer>();
        rd.enabled = true;

        enabled = true;
        Debug.Log("Shuriken finished initializing");
        isInit = true;

    }
    
    public override void DestroyAll() {
        Debug.Log("Destroy Particles");
        pSystem.Clear();
        isInit = false;
        
    }
    
    public override void SetColor(Color col, List<string> atoms, string residue = "All", string chain = "All"){
        
    }
    public override void SetColor(Color col, int atomNum){
        
    }
    public override void SetRadii(List<string> atoms, string residue = "All", string chain = "All"){}
    public override void SetRadii(int atomNum){}
    
    public override GameObject GetBall(int id){
        return null;
    }
    
    public override void ToggleDistanceCueing(bool enabling) {
        
    }
    
    private void CheckParticles() {
        ParticleSystem.Particle[] currentParticles = new ParticleSystem.Particle[pNumber];
        pSystem.GetParticles(currentParticles);
        foreach(ParticleSystem.Particle p in currentParticles) {
            Debug.Log(p.position.ToString());
            Debug.Log(p.startColor.ToString());
        }
        Debug.Log(pSystem.GetComponent<Renderer>().enabled.ToString());
    }
    
/*  private void ForDebugging() {
        radiusFactor = BallUpdate.radiusFactor;
        test = UIData.isParticlesInitialized;
        //Debug.Log(UIData.atomtype.ToString());
    }
*/
    
    public override void ResetRadii() {
        ParticleSystem.Particle[] currentParticles = new ParticleSystem.Particle[pNumber];
        pSystem.GetParticles(currentParticles);
        for(int i=0; i<pNumber; i++)
            currentParticles[i].startSize = currentParticles[i].startSize * BallUpdate.radiusFactor / BallUpdate.oldRadiusFactor;

        pSystem.SetParticles(currentParticles, pNumber);
        // Debug.Log("Resetting Particles radii");
        BallUpdate.oldRadiusFactor = BallUpdate.radiusFactor;
    }
    
    private void ResetColors() {
        if(UIData.atomtype == UIData.AtomType.particleball) {
            // Can't manage to change the color of the good atom. Particles and atomsColorList aren't organized the same way
//          for(int i=0; i<pNumber; i++)
//              currentParticles[i].color = MoleculeModel.atomsColorList[i];
            
            
            BallUpdate.resetColors = false;
            BallUpdate.bondsReadyToBeReset = true;
        }
    }
    
    public override void EnableRenderers() {
        pSystem.GetComponent<Renderer>().enabled = true;
        enabled = true;
        if(UIData.guided){
            ParticleSystem.Particle[] currentParticles = new ParticleSystem.Particle[pNumber];
            pSystem.GetParticles(currentParticles);
            for(int i=0; i<pNumber; i++)
                currentParticles[i].position = new Vector3(MoleculeModel.atomsLocationlist[i][0], MoleculeModel.atomsLocationlist[i][1], MoleculeModel.atomsLocationlist[i][2]);
            pSystem.SetParticles(currentParticles, pNumber);
        }
    }
    
    public override void DisableRenderers() {
        pSystem.GetComponent<Renderer>().enabled = false;
        enabled = false;
    }
    public override void EnableShadows(){
        pSystem.GetComponent<Renderer>().shadowCastingMode = ShadowCastingMode.On;
        pSystem.GetComponent<Renderer>().receiveShadows = true;
    }
    public override void DisableShadows(){
        pSystem.GetComponent<Renderer>().shadowCastingMode = ShadowCastingMode.Off;
        pSystem.GetComponent<Renderer>().receiveShadows = false;
    }


    
    void Update () {
        //ForDebugging();
        if(UIData.atomtype != UIData.AtomType.particleball ){
            DisableRenderers();
            return;
        }
        if(BallUpdate.radiusFactor != BallUpdate.oldRadiusFactor && UIData.isFileLoaded)
            ResetRadii();
        
        //if(BallUpdate.resetColors)
        //  ResetColors();
    }
    
    public override void ResetPositions(){
        for (int j=0; j<particles.Length; j++){
            particles[j].position = new Vector3(Molecule.Model.MoleculeModel.atomsLocationlist[j][0], 
                                                           Molecule.Model.MoleculeModel.atomsLocationlist[j][1],
                                                           Molecule.Model.MoleculeModel.atomsLocationlist[j][2]);
        }
        pSystem.SetParticles(particles, pNumber);
    }
    
    public override void ResetIMDSimulationPositions(){
//      for (int j=0; j<particles.Length; j++){
//          particles[j].position = new Vector3(Molecule.Model.MoleculeModel.atomsIMDSimulationLocationlist[(j*3)], 
//                                                    Molecule.Model.MoleculeModel.atomsIMDSimulationLocationlist[(j*3)+1],
//                                                    Molecule.Model.MoleculeModel.atomsIMDSimulationLocationlist[(j*3)+2]);
//      }
//      pSystem.SetParticles(particles, pNumber);
    }
    public override void showHydrogens(bool hide){
        ParticleSystem.Particle[] currentParticles = new ParticleSystem.Particle[pNumber];
        pSystem.GetParticles(currentParticles);
        //Not working for now
        for (int j=0; j<currentParticles.Length; j++){

            if(MoleculeModel.atomsTypelist[j].type == "H"){
                if(hide)
                    currentParticles[j].lifetime = -1f;

                else
                    currentParticles[j].lifetime = Mathf.Infinity;

            }

        }

        pSystem.SetParticles(currentParticles, pNumber);


        
    }
}