kono
/
refo
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

render 5 frames for every RL step 

- temp. repainting JComponent in env.step()
This commit is contained in:
Jan Löwenstrom 2020-01-01 18:05:59 +01:00
parent ec86006a07
commit b7d991cc92
2 changed files with 16 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
src/main/java/evironment/jumpingDino/DinoWorld.java
View File

@ -16,12 +16,14 @@ public class DinoWorld implements Environment<DinoAction>, Visualizable {
private Obstacle currentObstacle; private Obstacle currentObstacle;
private boolean randomObstacleSpeed; private boolean randomObstacleSpeed;
private boolean randomObstacleDistance; private boolean randomObstacleDistance;
private DinoWorldComponent comp;
public DinoWorld(boolean randomObstacleSpeed, boolean randomObstacleDistance){ public DinoWorld(boolean randomObstacleSpeed, boolean randomObstacleDistance){
this.randomObstacleSpeed = randomObstacleSpeed; this.randomObstacleSpeed = randomObstacleSpeed;
this.randomObstacleDistance = randomObstacleDistance; 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); dino = new Dino(Config.DINO_SIZE, Config.DINO_STARTING_X, Config.FRAME_HEIGHT - Config.GROUND_Y - Config.DINO_SIZE, 0, 0, Color.GREEN);
spawnNewObstacle(); spawnNewObstacle();
comp = new DinoWorldComponent(this);
} }
public DinoWorld(){ public DinoWorld(){
@ -54,17 +56,19 @@ public class DinoWorld implements Environment<DinoAction>, Visualizable {
dino.jump(); dino.jump();
} }
dino.tick(); for(int i= 0; i < 5; ++i){
currentObstacle.tick(); dino.tick();
if(currentObstacle.getX() < -Config.OBSTACLE_SIZE){ currentObstacle.tick();
spawnNewObstacle(); if(currentObstacle.getX() < -Config.OBSTACLE_SIZE){
spawnNewObstacle();
}
comp.repaint();
if(ranIntoObstacle()){
done = true;
break;
}
} }
return new StepResultEnvironment(new DinoStateWithSpeed(getDistanceToObstacle(), getCurrentObstacle().getDx()), reward, done, "");
if(ranIntoObstacle()){
done = true;
}
return new StepResultEnvironment(new DinoState(getDistanceToObstacle()), reward, done, "");
} }
@ -101,6 +105,6 @@ public class DinoWorld implements Environment<DinoAction>, Visualizable {
@Override @Override
public JComponent visualize() { public JComponent visualize() {
return new DinoWorldComponent(this); return comp;
} }
} }