107 lines
3.3 KiB
Java
107 lines
3.3 KiB
Java
package evironment.jumpingDino;
|
|
|
|
import core.Environment;
|
|
import core.State;
|
|
import core.StepResultEnvironment;
|
|
import core.gui.Visualizable;
|
|
import evironment.jumpingDino.gui.DinoWorldComponent;
|
|
import lombok.Getter;
|
|
|
|
import javax.swing.*;
|
|
import java.awt.*;
|
|
|
|
@Getter
|
|
public class DinoWorld implements Environment<DinoAction>, Visualizable {
|
|
private Dino dino;
|
|
private Obstacle currentObstacle;
|
|
private boolean randomObstacleSpeed;
|
|
private boolean randomObstacleDistance;
|
|
|
|
public DinoWorld(boolean randomObstacleSpeed, boolean randomObstacleDistance){
|
|
this.randomObstacleSpeed = randomObstacleSpeed;
|
|
this.randomObstacleDistance = randomObstacleDistance;
|
|
dino = new Dino(Config.DINO_SIZE, Config.DINO_STARTING_X, Config.FRAME_HEIGHT - Config.GROUND_Y - Config.DINO_SIZE, 0, 0, Color.GREEN);
|
|
spawnNewObstacle();
|
|
}
|
|
|
|
public DinoWorld(){
|
|
this(false, false);
|
|
}
|
|
|
|
private boolean ranIntoObstacle(){
|
|
Obstacle o = currentObstacle;
|
|
Dino p = dino;
|
|
|
|
boolean xAxis = (o.getX() <= p.getX() && p.getX() < o.getX() + Config.OBSTACLE_SIZE)
|
|
|| (o.getX() <= p.getX() + Config.DINO_SIZE && p.getX() + Config.DINO_SIZE < o.getX() + Config.OBSTACLE_SIZE);
|
|
|
|
boolean yAxis = (o.getY() <= p.getY() && p.getY() < o.getY() + Config.OBSTACLE_SIZE)
|
|
|| (o.getY() <= p.getY() + Config.DINO_SIZE && p.getY() + Config.DINO_SIZE < o.getY() + Config.OBSTACLE_SIZE);
|
|
|
|
return xAxis && yAxis;
|
|
}
|
|
|
|
private int getDistanceToObstacle(){
|
|
return currentObstacle.getX() - dino.getX() + Config.DINO_SIZE;
|
|
}
|
|
|
|
@Override
|
|
public StepResultEnvironment step(DinoAction action) {
|
|
boolean done = false;
|
|
int reward = 1;
|
|
|
|
if(action == DinoAction.JUMP){
|
|
dino.jump();
|
|
}
|
|
|
|
dino.tick();
|
|
currentObstacle.tick();
|
|
if(currentObstacle.getX() < -Config.OBSTACLE_SIZE){
|
|
spawnNewObstacle();
|
|
}
|
|
|
|
if(ranIntoObstacle()){
|
|
done = true;
|
|
}
|
|
|
|
return new StepResultEnvironment(new DinoState(getDistanceToObstacle()), reward, done, "");
|
|
}
|
|
|
|
|
|
private void spawnNewObstacle(){
|
|
int dx;
|
|
int xSpawn;
|
|
|
|
if(randomObstacleSpeed){
|
|
dx = -(int)((Math.random() + 0.5) * Config.OBSTACLE_SPEED);
|
|
}else{
|
|
dx = -Config.OBSTACLE_SPEED;
|
|
}
|
|
|
|
if(randomObstacleDistance){
|
|
// randomly spawning more right outside of the screen
|
|
xSpawn = (int)(Math.random() + 0.5 * Config.FRAME_WIDTH + Config.FRAME_WIDTH + Config.OBSTACLE_SIZE);
|
|
}else{
|
|
// instantly respawning on the left screen border
|
|
xSpawn = Config.FRAME_WIDTH + Config.OBSTACLE_SIZE;
|
|
}
|
|
|
|
currentObstacle = new Obstacle(Config.OBSTACLE_SIZE, xSpawn, Config.FRAME_HEIGHT - Config.GROUND_Y - Config.OBSTACLE_SIZE, dx, 0, Color.BLACK);
|
|
}
|
|
|
|
private void spawnDino(){
|
|
dino = new Dino(Config.DINO_SIZE, Config.DINO_STARTING_X, Config.FRAME_HEIGHT - Config.GROUND_Y - Config.DINO_SIZE, 0, 0, Color.GREEN);
|
|
}
|
|
@Override
|
|
public State reset() {
|
|
spawnDino();
|
|
spawnNewObstacle();
|
|
return new DinoState(getDistanceToObstacle());
|
|
}
|
|
|
|
@Override
|
|
public JComponent visualize() {
|
|
return new DinoWorldComponent(this);
|
|
}
|
|
}
|