@Override 
   public void update() { 
      numInState.put("alive", 0); 
      numInState.put("dead", 0); 
      for (int i = 0; i < getWidth(); i++) { 
         for (int j = 0; j < getHeight(); j++) { 
            updateCell(i, j); 
         } 
      } 
      setStates(); 
   } 

   @Override 
   public void update() { 
      numInState.put("satisfied", 0); 
      numInState.put("unsatisfied", 0); 
      satisfied = 0; 
      for (int i = 0; i < getWidth(); i++) { 
         for (int j = 0; j < getHeight(); j++) { 
            updateCell(i, j); 
         } 
      } 
      setStates(); 
   } 

   @Override 
   protected void setNeighbors(int xCoord, int yCoord) { 
      for (int i = -1; i <= 1; i++) { 
         for (int j = -1; j <= 1; j++) { 
            Cell neighbor = edgeManager.getCell(xCoord + i, yCoord + j); 
            if (neighbor != null) { 
               edgeManager.getCell(xCoord, yCoord).initializeNeighbor( 
                     neighbor); 
            } 
         } 
      } 
   } 

   @Override 
   protected void setNeighbors(int xCoord, int yCoord) { 
      for (int i = -1; i <= 1; i++) { 
         for (int j = -1; j <= 1; j++) { 
            Cell neighbor = edgeManager.getCell(xCoord + i, yCoord + j); 
            if (neighbor != null && neighbor != edgeManager.getCell(xCoord, yCoord)) { 
               edgeManager.getCell(xCoord, yCoord).initializeNeighbor( 
                     neighbor); 
            } 
         } 
      } 
   } 

   public PredatorModel(double width, double height, double turns, 
         double percentSharks, double percentFish) { 
      super(width, height); 
      numInState = new HashMap<String, Integer>(); 
      numInState.put("fish", 0); 
      numInState.put("shark", 0); 
      numInState.put("empty", 0); 
      edgeManager = new FiniteEdgeManager(myCells); 
      shapeManager = new SquareManager(myCells, edgeManager); 

   public SegregationModel(double double1, double double2, double threshold, 
         double percentRed, double percentBlue) { 
      super(double1, double2); 
      numInState = new HashMap<String, Integer>(); 
      numInState.put("satisfied", 0); 
      numInState.put("unsatisfied", 0); 
      edgeManager = new FiniteEdgeManager(myCells); 
      shapeManager = new SquareManager(myCells, edgeManager); 

   private void setColor(){ 
      if (myState == 0){ 
         myColor = Color.BLUE; 
      } 
      if (myState == 1){ 
         myColor = Color.GREEN; 
      } 
      if (myState == 2){ 
         myColor = Color.CYAN; 
      } 
   } 

   private void setColor(){ 
      if (myState == 0){ 
         myColor = Color.WHITE; 
      } 
      if (myState == 1){ 
         myColor = Color.RED; 
      } 
      if (myState == 2){ 
         myColor = Color.BLUE; 
      } 
   } 

   private void createSegregationModel() { 
      currentModel = new SegregationModel(currentParameters.get("gridWidth"), currentParameters.get("gridHeight"), currentParameters.get("threshold"), currentParameters.get("percentType1"), currentParameters.get("percentType2"));       
   } 

   private void createPredatorModel() { 
      currentModel = new PredatorModel(currentParameters.get("gridWidth"), currentParameters.get("gridHeight"), currentParameters.get("turnsAlive"), currentParameters.get("percentType1"), currentParameters.get("percentType2")); 
       
   } 

   @Override 
   public Color displayCell(int xCoord, int yCoord) 
         throws NullPointerException { 
      if (getCell(xCoord, yCoord).getCurrentState() == 0) { 
         return Color.BLUE; 
      } 
      if (getCell(xCoord, yCoord).getCurrentState() == 1) { 
         return Color.GREEN; 
      } 
      if (getCell(xCoord, yCoord).getCurrentState() == 2) { 
         return Color.RED; 
      } 
      return Color.BLACK; 
   } 

   @Override 
   public Color displayCell(int xCoord, int yCoord) 
         throws NullPointerException { 
      if (getCell(xCoord, yCoord).getCurrentState() == 0) { 
         return Color.WHITE; 
      } 
      if (getCell(xCoord, yCoord).getCurrentState() == 1) { 
         return Color.RED; 
      } 
      if (getCell(xCoord, yCoord).getCurrentState() == 2) { 
         return Color.BLUE; 
      } 
      return Color.BLACK; 
   } 

   public PredatorCell getNeighborInState(int state){ 
      int inState = neighborsInDoubleState(state); 
      if (inState > 0){ 
         int rand = (int) Math.floor(Math.random()*inState); 
         for (Cell neighbor: myNeighbors){ 
            if (neighbor.getCurrentState() == state 
                  && neighbor.getNextState() == state){ 
               if (rand == 0){ 
                  return (PredatorCell) neighbor; 
               } 
               rand--; 
            } 
         } 
      } 
      return null; 
   } 

   public SegregationCell getNeighborInState(int state){ 
      int inState = neighborsInDoubleState(state); 
      if (inState > 0){ 
         int rand = (int) Math.floor(Math.random()*inState); 
         for (Cell neighbor: myNeighbors){ 
            if (neighbor.getCurrentState() == state 
                  && neighbor.getNextState() == state){ 
               if (rand == 0){ 
                  return (SegregationCell) neighbor; 
               } 
               rand--; 
            } 
         } 
      } 
      return null; 
   } 

   public int neighborsInState(int state) { 
      int numNear = 0; 
      for (Cell neighbor : myNeighbors) { 
         if (neighbor.getCurrentState() == state) { 
            numNear++; 
         } 
      } 
      return numNear; 
   } 

   public int neighborsInState(int state) { 
      int numNear = 0; 
      for (Cell neighbor : myNeighbors) { 
         if (neighbor.getCurrentState() == state) { 
            numNear++; 
         } 
      } 
      return numNear; 
   } 

   public int neighborsInState(int state) { 
      int numNear = 0; 
      for (Cell neighbor : myNeighbors) { 
         if (neighbor.getCurrentState() == state) { 
            numNear++; 
         } 
      } 
      return numNear; 
   } 

      for (int i = 0; i < getWidth(); i++) { 
         for (int j = 0; j < getHeight(); j++) { 
            initializeCell(i, j, states, numInState); 
         } 
      } 

      for (int i = 0; i < getWidth(); i++) { 
         for (int j = 0; j < getHeight(); j++) { 
            initializeCell(i, j, states, numInState); 

   public int neighborsInDoubleState(int state) { 
      int numNear = 0; 
      for (Cell neighbor : myNeighbors) { 
         if (neighbor.getCurrentState() == state 
               && neighbor.getNextState() == state) { 
            numNear++; 
         } 
      } 
      return numNear; 
   } 

   public int neighborsInDoubleState(int state) { 
      int numNear = 0; 
      for (Cell neighbor : myNeighbors) { 
         if (neighbor.getCurrentState() == state 
               && neighbor.getNextState() == state) { 
            numNear++; 
         } 
      } 
      return numNear; 
   } 

      ArrayList<Integer> cells = new ArrayList<Integer>(); 
      for (int i = 0; i < getWidth() * getHeight(); i++) { 
         cells.add(i); 
      } 
      Collections.shuffle(cells); 
      for (int i = 0; i < getWidth() * getHeight(); i++) { 

      ArrayList<Integer> cells = new ArrayList<>(); 
      for (int i = 0; i < getWidth() * getHeight(); i++) { 
         cells.add(i); 
      } 
      Collections.shuffle(cells); 
      for (int i = 0; i < getWidth() * getHeight(); i++) { 

   private void initializeCell(int xCoord, int yCoord, List<Integer> states, 
         List<Integer> numInState) { 
      int index = (int) Math.floor(Math.random() * states.size()); 
      myCells[xCoord][yCoord] = new SegregationCell(xCoord, yCoord, 
            states.get(index)); 
      numInState.set(index, numInState.get(index) - 1); 
      if (numInState.get(index) == 0) { 
         numInState.remove(index); 
         states.remove(index); 
      } 
   } 

   private void initializeCell(int xCoord, int yCoord, List<Integer> states, 
         List<Integer> numInState) { 
      int index = (int) Math.floor(Math.random() * states.size()); 
      myCells[xCoord][yCoord] = new GameOfLifeCell(xCoord, yCoord, states.get(index)); 
      numInState.set(index, numInState.get(index) - 1); 
      if (numInState.get(index) == 0) { 
         numInState.remove(index); 
         states.remove(index); 
      } 
   } 

   private void initializeCells(int numRed, int numBlue, int numEmpty) { 
      ArrayList<Integer> states = new ArrayList<>(Arrays.asList(0, 1, 2)); 
      ArrayList<Integer> numInState = new ArrayList<>(Arrays.asList(numEmpty, 
            numRed, numBlue)); 

   private void initializeCells(int numSharks, int numFish, int numEmpty) { 
      ArrayList<Integer> states = new ArrayList<>(Arrays.asList(0, 1, 2)); 
      ArrayList<Integer> numInState = new ArrayList<>(Arrays.asList(numEmpty, 
            numFish, numSharks)); 

   public void initializeNeighbors() { 
      for (int i = 0; i < getWidth(); i++) { 
         for (int j = 0; j < getHeight(); j++) { 

   private void updateFishes() { 
      for (int i = 0; i < getWidth(); i++) { 
         for (int j = 0; j < getHeight(); j++) { 

    protected void setStates() { 
      for (int i = 0; i < getWidth(); i++) { 
         for (int j = 0; j < getHeight(); j++) { 

   private void updateSharks() { 
      for (int i = 0; i < getWidth(); i++) { 
         for (int j = 0; j < getHeight(); j++) { 

      int xCoord = i % getWidth(); 
      int yCoord = i / getWidth(); 

      int xCoord = i % getWidth(); 
      int yCoord = i / getWidth(); 

      current = moveAnimal(xCoord, yCoord, 1); 
      current.increaseLife(); 
      if (willBreed(getCell(current.getX(), current.getY()))) { 
         produceAnimal(current.getX(), current.getY(), 1); 
      } 

         current = moveAnimal(xCoord, yCoord, 2); 
      } 
      current.increaseLife(); 
      if (willBreed(getCell(current.getX(), current.getY()))) { 
         produceAnimal(current.getX(), current.getY(), 2); 
      } 

   public FireCell(int x, int y, int state) { 
      super(x, y); 
      currentState = new FireState(state); 
      nextState = new FireState(state); 
   } 

   public PredatorCell(int x, int y, int state) { 
      super(x, y); 
      currentState = new PredatorState(state); 
      nextState = new PredatorState(state); 
   } 

   public SegregationCell(int x, int y, int state) { 
      super(x, y); 
      currentState = new SegregationState(state); 
      nextState = new SegregationState(state); 
   } 

   public GameOfLifeCell(int x, int y, int state) { 
      super(x, y); 
      currentState = new GameOfLifeState(state); 
      nextState = new GameOfLifeState(state); 
   } 

   @Override 
   public void update() { 
      numInState.put("tree", 0); 
      numInState.put("burning", 0); 
      numInState.put("dead", 0); 

   @Override 
   public void update() { 
      numInState.put("fish", 0); 
      numInState.put("shark", 0); 
      numInState.put("empty", 0); 

            if (getCell(i, j).getCurrentState() == 1) { 
               numInState.put("fish", numInState.get("fish") + 1); 
            } 

            if (getCell(i, j).getCurrentState() == 0) { 
               numInState.put("empty", numInState.get("empty") + 1); 
            } 

      ArrayList<Integer> states = new ArrayList<>(Arrays.asList(0, 2)); 
      ArrayList<Integer> numInState = new ArrayList<>(Arrays.asList(numTrees, 
            numEmpty)); 

      ArrayList<Integer> states = new ArrayList<>(Arrays.asList(0, 1)); 
      ArrayList<Integer> numInState = new ArrayList<>(Arrays.asList(numDead, 
            numAlive)); 

      int numTrees = (int) (getWidth() * getHeight() * treeDensity); 
      int numEmpty = (getWidth() * getHeight()) - numTrees; 

      int numAlive = (int) (getWidth() * getHeight() * percentAlive); 
      int numDead = (getWidth() * getHeight()) - numAlive;