1 files changed, 22 insertions, 11 deletions

diff --git a/src/docs/class.puml b/src/docs/class.puml
index 869a707..6a527ae 100644
--- a/src/docs/class.puml
+++ b/src/docs/class.puml
@@ -12,20 +12,23 @@ package utils {
 
 package viewer {
   class Viewer {
-    Game
-    Stage
+    final Game
+    Viewer(List<Board>)
     void main(String[])
     void eventLoop(ApplicationContext)
-    void renderFrame(Graphics2D,ApplicationContext)
+    void renderFrame(Graphics2D,ApplicationContext,List<Event>)
   }
 }
 
 package context {
   class Context {
-    Context(Stage,List<Event>,GraphicsContext)
-    final Stage
+    Context(Game,List<Event>,GraphicsContext)
+    final Game
     final List<Event>
     final GraphicsContext
+    Game getGame()
+    List<Event> getEvents()
+    GraphicsContext getGraphicsContext()
   }
 
   class GraphicsContext {
@@ -49,10 +52,18 @@ package context {
   }
 
   class Game {
-    final GameController
-    int index
+    Stage
+    final List<Controller>
+    int indexBoard
     final List<Board>
-    Game(GameController,List<Board>)
+    bool over
+    Game(List<Board>)
+    Stage getStage()
+    bool isOver()
+    void setOver()
+    void nextStage()
+    void retryStage()
+    List<Event> update(Context context)
   }
 }
 
@@ -77,6 +88,8 @@ package event {
   }
 
   class ConfirmEvent implements InputEvent
+  class GameOverEvent implements Event
+
 
   class ExplosionEvent implements GameEvent {
     Block source
@@ -157,8 +170,8 @@ package model {
   class RobotBlock extends Block
 
   class Stage {
-    StageController
     List<Block>
+    Board
     Stage(Board)
     List<Block> getBlocks()
     List<Event> update(Context)
@@ -179,9 +192,7 @@ package controller {
   class BlockControllerFactory {
     BlockController build(Block)
   }
-
   class GameController implements Controller
-  class StageController implements Controller
 
   abstract class PhysicsController extends BlockController
   abstract class DisplayController extends BlockController