Walker hero;
//arrays
BasicBullet[] bullets;
BugBase[] bugs;
PlantBase[] plants;
Explosion[] explosions;
PlantBullet[] plantBullets;
int bulletsCount;
int bugsCount;
int plantsCount;
int plantBulletsCount;
int explosionsCount;

boolean keyUp, keyDown, keyLeft, keyRight;
int weaponType = 0; //0 flamer, 1 machinegun...

void setup()
{
  int num = 5;
  //String windowWidthString = getParameter("WINDOWWIDTH");
  //if (windowWidthString != null) {
  //  num = Integer.parseInt(windowWidthString);
  //} 
  size(500, 400);
  frameRate(40);
  hero = new Walker(100, 100, 3);

  bullets = new BasicBullet[120];
  bugs = new BugBase[1801];
  explosions = new Explosion[120];
  plants = new PlantBase[40];
  plantBullets = new PlantBullet[80];
  //int num = 50;
  
  for (int i=0; i<num*3; i++)
  {  
    bugs[bugsCount] = new BugAgressive(new Vector2D(100, 200), 9, 3, 0);
    bugsCount ++;
  }
  for (int i=0; i<num*2; i++)
  {  
    bugs[bugsCount] = new BugAfraid(new Vector2D(200, 200), 6, 2, 1);
    bugsCount ++;
  }
  for (int i=0; i<num*3; i++)
  {  
    bugs[bugsCount] = new BugAgressive(new Vector2D(200, 300), 12, 1, 0);
    bugsCount ++;
  }
  for (int i=0; i<num; i++)
  {  
    bugs[bugsCount] = new BugAgressive(new Vector2D(200, 300), 18, 3, 0);
    bugsCount ++;
  }
  for (int i=0; i<num; i++)
  {  
    bugs[bugsCount] = new BugAfraid(new Vector2D(300, 200), 20, 3, 1);
    bugsCount ++;
  }
  
  plants[plantsCount] = new PlantBase(new Vector2D(120, 120), 45, 0.09);
  plantsCount ++;
  plants[plantsCount] = new PlantBase(new Vector2D(300, 120), 35, 0.08);
  plantsCount ++;
//  plants[plantsCount] = new PlantBase(new Vector2D(120, 280), 60, 0.04);
//  plantsCount ++;
  plants[plantsCount] = new PlantBase(new Vector2D(300, 280), 55, 0.067);
  plantsCount ++;
//  plants[plantsCount] = new PlantBase(new Vector2D(60, 219), 30, 0.092);
//  plantsCount ++;
  plants[plantsCount] = new PlantBase(new Vector2D(60, 350), 35, 0.088);
  plantsCount ++;
  plants[plantsCount] = new PlantBase(new Vector2D(410, 144), 42, 0.012);
  plantsCount ++;
  plants[plantsCount] = new PlantBase(new Vector2D(390, 320), 30, 0.033);
  plantsCount ++;

 
}
 
void draw()
{
  background(20, 10, 20);
  //noStroke();
  //lightFalloff(1.0, 0.1, 0.0);
  //pointLight(102, 153, 204, mouseX, mouseY, 30);    
  //fill(123);
  //rect(10, 10, 30, 30);

  hero.update(weaponType);
  hero.draw();
  for (int i=0; i<bulletsCount; i++) 
  {
    bullets[i].update();
    bullets[i].draw();
    Vector2D pos = new Vector2D(bullets[i].position);
    int removed = bullets[i].clean(i);
    if (weaponType == 3)
    {
      if (removed == 1)
      {
        explosions[explosionsCount] = new Explosion(pos, 3, 2);
        explosionsCount ++;
      }
    }
    i -= removed;
  }
  for (int i=0; i<bugsCount; i++) 
  {
    bugs[i].update(i);
    bugs[i].draw();
    float bugSize = (bugs[i].size);
    Vector2D bugPos = new Vector2D(bugs[i].position);
    int removed = bugs[i].clean(i);
    if (removed == 1)
    {
      explosions[explosionsCount] = new Explosion(bugPos, int(bugSize), 1);
      explosionsCount ++;
    }
    i -= removed;
  }
  for (int i=0; i<plantsCount; i++) 
  {
    plants[i].update(i);
    plants[i].draw();
    float plantSize = (plants[i].size);
    Vector2D plantPos = new Vector2D(plants[i].position);
    int removed = plants[i].clean(i);
    if (removed == 1)
    {
      explosions[explosionsCount] = new Explosion(plantPos, int(plantSize), 1);
      explosionsCount ++;
    }
    i -= removed;
  }
  
  for (int i=0; i<plantBulletsCount; i++) 
  {
    plantBullets[i].update();
    plantBullets[i].draw();
    Vector2D pos = new Vector2D(plantBullets[i].position);
    int removed = plantBullets[i].clean(i);
    if (weaponType == 3)
    {
      if (removed == 1)
      {
        explosions[explosionsCount] = new Explosion(pos, 3, 2);
        explosionsCount ++;
      }
    }
    i -= removed;
  }
  for (int i=0; i<explosionsCount; i++) 
  {
    explosions[i].draw(i);
    i -= explosions[i].clean(i);
  }

}

//this is here so we detect keys as we should

void keyReleased()
{
  if (key == 'w' || key == 'W')
  {
    keyUp = false;
  }
  if (key == 's' || key == 'S')
  {
    keyDown = false;
  }
  if (key == 'a' || key == 'A')
  {
    keyLeft = false;
  }
  if (key == 'd' || key == 'D')
  {
    keyRight = false;
  }
} 

void detectKeys()
{
  if(keyPressed)
  {
    if (key == 'w' || key == 'W'){ keyUp = true; }
    if (key == 's' || key == 'S'){keyDown = true; }
    if (key == 'a' || key == 'A'){keyLeft = true;}
    if (key == 'd' || key == 'D'){keyRight = true;}
    if (key == '1'){weaponType = 0;}
    if (key == '2'){weaponType = 1;}
    if (key == '3'){weaponType = 2;}
    if (key == '4'){weaponType = 3;}
    
  } 
}