/* Copyright 2004, Sam Reid */
package edu.colorado.phet.qm.tests;

import edu.colorado.phet.common.math.Vector2D;
import edu.colorado.phet.qm.model.ParticleUnits;
import edu.colorado.phet.qm.model.WaveDebugger;
import edu.colorado.phet.qm.model.WaveSetup;
import edu.colorado.phet.qm.model.Wavefunction;
import edu.colorado.phet.qm.model.math.Complex;
import edu.colorado.phet.qm.model.waves.GaussianWave2D;
import edu.colorado.phet.qm.view.complexcolormaps.VisualColorMap3;

import java.awt.geom.Point2D;
import java.text.DecimalFormat;

/**
 * User: Sam Reid
 * Date: Dec 21, 2005
 * Time: 4:20:17 PM
 * Copyright (c) Dec 21, 2005 by Sam Reid
 */

public class TestUnits2 {

    public static class NeutronUnits extends ParticleUnits {
//        private double scaleDown=0.00000001;
        private double scaleDown = 1;//0.0000000001;

        public NeutronUnits() {
            setHbar( new Value( 0.000658, 1, "eV ps" ) );
            setDx( new Value( 1.0, 0.1, "nm" ) );
            setDt( new Value( 1, 1, "ps" ) );

            setMass( new Value( 104800000, 1.0 / 10000.0, "eV fs^2/nm^2" ) );
            setMinVelocity( new Value( 5 * scaleDown, 0.1, "km/s" ) );
            setMaxVelocity( new Value( 20 * scaleDown, 0.1, "km/s" ) );
        }
    }

    public TestUnits2() {
        Wavefunction wavefunction = new Wavefunction( 50, 50 );
        NeutronUnits neutronUnits = new NeutronUnits();
        double momentumY = neutronUnits.getAverageVelocity() * neutronUnits.getMass().getValue();
        System.out.println( "momentumY = " + momentumY );
        GaussianWave2D gaussianWave2D = new GaussianWave2D( new Point2D.Double( 25, 25 ),
                                                            new Vector2D.Double( 0, momentumY ), 3.5 * 2 );
        WaveSetup waveSetup = new WaveSetup( gaussianWave2D );
        waveSetup.initialize( wavefunction );

        WaveDebugger waveDebugger = new WaveDebugger( "wave", wavefunction );
        waveDebugger.setComplexColorMap( new VisualColorMap3() );
        waveDebugger.setVisible( true );

        printWaveToScreen( wavefunction );
    }

    private void printWaveToScreen( Wavefunction wavefunction ) {
        DecimalFormat formatter = new DecimalFormat( "0.00" );
        for( int k = 0; k < wavefunction.getHeight(); k++ ) {
            for( int i = 0; i < wavefunction.getWidth(); i++ ) {
                Complex val = wavefunction.valueAt( i, k );
                String s = formatter.format( val.getReal() );
                if( s.equals( formatter.format( -0.000000001 ) ) ) {
                    s = formatter.format( 0 );
                }
                String spaces = "  ";
                if( s.startsWith( "-" ) ) {
                    spaces = " ";
                }
                System.out.print( spaces + s );

            }
            System.out.println( "" );
        }
    }

    public static void main( String[] args ) {
        new TestUnits2().start();
    }

    private void start() {
    }

    class GaussianWave1D {

        private double momentum;
        private double x0;
        private double dxLattice;

        public GaussianWave1D( double momentum, double x0, double dxLattice ) {
            this.momentum = momentum;
            this.x0 = x0;
            this.dxLattice = dxLattice;
        }

        public Complex getValue( double x ) {
            double norm = Math.pow( Math.PI * dxLattice * dxLattice, -1.0 / 4.0 );
            double space = Math.exp( -( x - x0 ) * ( x - x0 ) / 2 / dxLattice / dxLattice );
            Complex momentumTerm = Complex.exponentiateImaginary( momentum * ( x - x0 ) );
            momentumTerm.scale( norm * space );
            return momentumTerm;
        }
    }
}