/** Copyright by Barry G. Becker, 2000-2011. Licensed under MIT License: http://www.opensource.org/licenses/MIT */
package com.barrybecker4.simulation.liquid;
import com.barrybecker4.common.util.FileUtil;
import com.barrybecker4.optimization.Optimizer;
import com.barrybecker4.optimization.parameter.NumericParameterArray;
import com.barrybecker4.optimization.parameter.ParameterArray;
import com.barrybecker4.optimization.parameter.types.Parameter;
import com.barrybecker4.optimization.strategy.OptimizationStrategyType;
import com.barrybecker4.simulation.common.ui.Simulator;
import com.barrybecker4.simulation.common.ui.SimulatorOptionsDialog;
import com.barrybecker4.simulation.liquid.config.ConfigurationEnum;
import com.barrybecker4.simulation.liquid.model.LiquidEnvironment;
import com.barrybecker4.simulation.liquid.rendering.EnvironmentRenderer;
import com.barrybecker4.simulation.liquid.rendering.RenderingOptions;
import com.barrybecker4.ui.util.GUIUtil;
import javax.swing.*;
import java.awt.*;
import java.awt.event.MouseEvent;
import java.awt.event.MouseListener;
/**
* Main class for particle liquid simulation.
*
* @author Barry Becker
*/
public class LiquidSimulator extends Simulator implements MouseListener {
private LiquidEnvironment environment_;
private EnvironmentRenderer envRenderer_;
/** These options can be changed while the simulation is running. */
private LiquidDynamicOptions dynamicOptions_;
/** The initial time step. It may adapt. */
private static final double INITIAL_TIME_STEP = 0.005;
private static final Color BG_COLOR = Color.white;
private static final int NUM_OPT_PARAMS = 3;
private boolean advectionOnly = false;
/**
* Constructor
*/
public LiquidSimulator() {
super("Liquid");
environment_ = new LiquidEnvironment( ConfigurationEnum.getDefaultValue().getFileName());
commonInit();
}
public void loadEnvironment(String configFile) {
environment_ = new LiquidEnvironment(configFile);
environment_.setAdvectionOnly(advectionOnly);
commonInit();
}
@Override
protected void reset() {
boolean oldPaused = this.isPaused();
setPaused(true);
environment_.reset();
commonInit();
setPaused(oldPaused);
}
private void commonInit() {
initCommonUI();
envRenderer_ = new EnvironmentRenderer(environment_);
int s = (int) envRenderer_.getScale();
setPreferredSize(new Dimension( environment_.getWidth() * s, environment_.getHeight() * s));
}
@Override
protected SimulatorOptionsDialog createOptionsDialog() {
return new LiquidOptionsDialog( frame_, this );
}
@Override
protected double getInitialTimeStep() {
return INITIAL_TIME_STEP;
}
/**
* @return a new recommended time step change.
*/
@Override
public double timeStep() {
if ( !isPaused() ) {
timeStep_ = environment_.stepForward( timeStep_);
}
return timeStep_;
}
public LiquidEnvironment getEnvironment() {
return environment_;
}
@Override
public void setScale( double scale ) {
envRenderer_.setScale(scale);
}
@Override
public double getScale() {
return envRenderer_.getScale();
}
public RenderingOptions getRenderingOptions() {
return envRenderer_.getRenderingOptions();
}
public boolean getSingleStepMode() {
return !isAnimating();
}
public void setSingleStepMode(boolean singleStep) {
setAnimating(!singleStep);
if (singleStep) {
addMouseListener(this);
}
else {
removeMouseListener(this);
}
}
public boolean getAdvectionOnly() {
return advectionOnly;
}
public void setAdvectionOnly(boolean advectOnly) {
advectionOnly = advectOnly;
environment_.setAdvectionOnly(advectOnly);
}
@Override
public JPanel createDynamicControls() {
dynamicOptions_ = new LiquidDynamicOptions(this);
return dynamicOptions_;
}
@Override
public void doOptimization() {
Optimizer optimizer;
if (GUIUtil.hasBasicService())
optimizer = new Optimizer( this );
else
optimizer = new Optimizer( this, FileUtil.getHomeDir()+ "performance/liquid/liquid_optimization.txt" );
Parameter[] params = new Parameter[3];
ParameterArray paramArray = new NumericParameterArray( params );
setPaused(false);
optimizer.doOptimization(OptimizationStrategyType.GENETIC_SEARCH, paramArray, 0.3);
}
/**
* *** implements the key method of the Optimizee interface
*
* evaluates the liquids fitness.
*/
@Override
public double evaluateFitness( ParameterArray params ) {
assert false : "not implemented yet";
return 0.0;
}
public double getOptimalFitness() {
return 0;
}
@Override
public Color getBackground() {
return BG_COLOR;
}
@Override
public void paint( Graphics g ) {
if (g==null) return;
Graphics2D g2 = (Graphics2D) g;
envRenderer_.render(g2, getWidth(), getHeight());
}
@Override
protected String getFileNameBase() {
return FileUtil.getHomeDir() + "temp/animations/simulation/" + "liquid/liquidFrame";
}
public void mouseClicked(MouseEvent e) {
//System.out.println("mclick timeStep="+ timeStep_ );
environment_.stepForward( timeStep_);
this.repaint();
}
public void mousePressed(MouseEvent e) {}
public void mouseReleased(MouseEvent e) {}
public void mouseEntered(MouseEvent e) {}
public void mouseExited(MouseEvent e) {}
}