/*CopyrightHere*/
package repast.simphony.demo.mousetrap;

import repast.simphony.context.Context;
import repast.simphony.engine.controller.NullAbstractControllerAction;
import repast.simphony.engine.environment.RunEnvironmentBuilder;
import repast.simphony.engine.environment.RunState;
import repast.simphony.engine.environment.GUIRegistryType;
import repast.simphony.engine.schedule.IAction;
import repast.simphony.engine.schedule.ISchedule;
import repast.simphony.engine.schedule.NonModelAction;
import repast.simphony.engine.schedule.ScheduleParameters;
import repast.simphony.parameter.Parameters;
import repast.simphony.parameter.SimpleParameterSetter;
import repast.simphony.random.RandomHelper;
import repast.simphony.scenario.ModelInitializer;
import repast.simphony.scenario.Scenario;
import repast.simphony.visualization.IDisplay;

/**
 * @author Nick Collier
 */
public class TrapInitializer implements ModelInitializer {

  private IDisplay display;

  @NonModelAction
  static class DisplayUpdater implements IAction {

    private IDisplay display;

    public DisplayUpdater(IDisplay display) {
      this.display = display;
    }

    public void execute() {
      display.update();
    }
  }

  /**
   * This is ran after the model has been loaded. This is only ran once, but the settings set
   * through the {@link repast.simphony..scenario.Scenario} will apply to every run of the simulation.
   *
   * @param scen the {@link repast.simphony..scenario.Scenario} object that hold settings for the run
   */
  public void initialize(Scenario scen, RunEnvironmentBuilder builder) {
    scen.addMasterControllerAction(new NullAbstractControllerAction() {
      @Override
      public void runInitialize(RunState runState, final Context context, Parameters params) {

        if (context.getTypeID().equals("All Traps")) {
          ISchedule schedule = repast.simphony.engine.environment.RunEnvironment.getInstance()
                  .getCurrentSchedule();
          schedule.schedule(ScheduleParameters.createOneTime(1.0), new IAction() {
            public void execute() {
              MouseTrap trap = (MouseTrap) context.getRandomObject();
              trap.trigger();
            }
          });

        }
      }

      public String toString() {
        return "Schedule a single trap to trigger at 1.0";
      }
    });


    scen.addMasterControllerAction(new NullAbstractControllerAction() {
      @Override
      public void runInitialize(RunState runState, Context context, Parameters runParams) {
        display = new CustomDisplay(context);
        runState.getGUIRegistry().addDisplay("Trigger Count", GUIRegistryType.OTHER, display);
        runState.getScheduleRegistry().getModelSchedule().schedule(ScheduleParameters.createRepeating(1, 1,
                ScheduleParameters.END), new DisplayUpdater(display));

      }

      @Override
      public void runCleanup(RunState runState, Context context) {
        display.destroy();
        display = null;
      }

      public String toString() {
        return "Create a custom display";
      }
    });



  }

}