37 lines
1.0 KiB
Java
37 lines
1.0 KiB
Java
package evironment.antGame;
|
|
|
|
import core.StepResultEnvironment;
|
|
|
|
public class AntWorldContinuous extends AntWorld {
|
|
public AntWorldContinuous(int width, int height) {
|
|
super(width, height);
|
|
}
|
|
|
|
public AntWorldContinuous() {
|
|
super();
|
|
}
|
|
|
|
@Override
|
|
public StepResultEnvironment step(AntAction action) {
|
|
AntObservation observation;
|
|
Cell currentCell = grid.getCell(myAnt.getPos());
|
|
|
|
StepCalculation sc = processStep(action);
|
|
|
|
// flag is set to true if food gets dropped onto starts
|
|
if(sc.checkCompletion) {
|
|
grid.spawnNewFood();
|
|
}
|
|
// valid movement
|
|
if(!sc.stayOnCell) {
|
|
myAnt.getPos().setLocation(sc.potentialNextPos);
|
|
}
|
|
|
|
// get observation after action was computed
|
|
observation = new AntObservation(grid.getCell(myAnt.getPos()), myAnt.getPos(), myAnt.hasFood());
|
|
|
|
return new StepResultEnvironment(new AntState(grid.getGrid(), observation.getPos(), observation.hasFood()), sc.reward, false, sc.info);
|
|
}
|
|
|
|
}
|