GameAILab1/Boid.pde

/// In this file, you will have to implement seek and waypoint-following
/// The relevant locations are marked with "TODO"

class Crumb
{
  PVector position;
  Crumb(PVector position)
  {
    this.position = position;
  }
  void draw()
  {
    fill(255);
    noStroke();
    circle(this.position.x, this.position.y, CRUMB_SIZE);
  }
}

class Boid
{
  Crumb[] crumbs = {};
  int last_crumb;
  float acceleration;
  float rotational_acceleration;
  KinematicMovement kinematic;
  PVector target;
  PVector direction;
  ArrayList<PVector> waypoints;
  boolean stillInRadius = true;
  int currentTarget = 0;

  Boid(PVector position, float heading, float max_speed, float max_rotational_speed, float acceleration, float rotational_acceleration)
  {
    this.kinematic = new KinematicMovement(position, heading, max_speed, max_rotational_speed);
    this.last_crumb = millis();
    this.acceleration = acceleration;
    this.rotational_acceleration = rotational_acceleration;
  }

  void update(float dt)
  {
    if (target != null)
    {
      // TODO: Implement seek here

      //This makes a vector with the direction our boid needs to go to
      PVector direction = PVector.sub(target, kinematic.position);

      //atan2(direction.y, direction.x) will return the direction we need to go in radians

      //print direction we need to go and the direction we are facing right now
      //println(atan2(direction.y, direction.x) + " " + normalize_angle_left_right(kinematic.getHeading()));

      float directionalThreshold = .1;
      //You have to normalize this too or the boid goes the wrong way sometimes
      float angleToTarget = normalize_angle_left_right(atan2(direction.y, direction.x) - normalize_angle_left_right(kinematic.getHeading()));
      float arrivalThreshold = 150.0;



      //This just draws a circle for visual debugging purposes
      circle(target.x, target.y, 3);

      //prints the angle to the target
      //println(angleToTarget);

      //if the angle is larger than the threshold in the positive direction, rotate counterclockwise
      if (angleToTarget > directionalThreshold) {
        kinematic.increaseSpeed(0.0, +1);

        //if the angle is smaller than the threshold in the negative direction, rotate clockwise
      } else if (angleToTarget < -directionalThreshold) {
        kinematic.increaseSpeed(0.0, -1);

        //if the angle is within our threshold, stop our rotational velocity by rotating opposite
      } else if (directionalThreshold > angleToTarget) {

        if (kinematic.getRotationalVelocity() > 0) {
          kinematic.increaseSpeed(0.0, -kinematic.getRotationalVelocity());
        } else if (kinematic.getRotationalVelocity() < 0) {
          kinematic.increaseSpeed(0.0, kinematic.getRotationalVelocity());
        }
      }


      //if the target is outside its arrival threshold, accelerate.
      //if the target is inside its arrival threshold, accelerate backwards until the speed is 0.
      if (direction.mag() > arrivalThreshold) {
        kinematic.increaseSpeed(1, 0);
      } else if (direction.mag() < arrivalThreshold) {


        //Need more specific code here to handle arrivals correctly
        //TODO: change this to slow down less / not at all if the angle to the next target is not large

        //This handles starting / stopping if there are more targets

        //This ensures that we don't crash because waypoints is null
        if (waypoints != null) {

          //this checks if there's another target to go to
          if (currentTarget + 1 < waypoints.size()) {



            //if so, change the speed depending on the angle to the next target

            //We can calculate the angle to the next target with the use of two vectors: one from our location to current target, one from current target to next target
            //use the dot product of those two vectors; this gives us the angle between them, and we can use that to calculate how much we should slow down

            //direction is our boid to target vector
            //this is at direction
            //current target to next target is here:
            PVector currentTargetToNext = PVector.sub(waypoints.get(currentTarget+1), target);

            //i'm not sure this is the best way to do this, it might be simpler to calculate the angle, but this should work too

            //holds dot product of targets
            float dotProductOfTargets = PVector.dot(currentTargetToNext, direction);

            //Dividing by both their magnitudes will normalize our result between -1 and 1
            dotProductOfTargets = dotProductOfTargets / (currentTargetToNext.mag() * direction.mag());

            //Add 1, divide by 2, this will cause our result to be between 0 and 1
            //If this is closer to 0, slow down more. If it's closer to 1, slow down less.
            dotProductOfTargets = (dotProductOfTargets + 1) / 2;

            //use an ideal speed for our boid, to tell it to either speed up or slow down whether it's going faster than this or not
            float idealSpeed = (dotProductOfTargets) * 80 + 15;

            float maxSpeed = 100 * pow(((PI - abs(angleToTarget)) / PI), 10);
            //println(maxSpeed);

            if (idealSpeed > maxSpeed) {
              idealSpeed = maxSpeed;
            }

            if (kinematic.getSpeed() < idealSpeed) {
              kinematic.increaseSpeed(1, 0);
            } else if (kinematic.getSpeed() > idealSpeed) {
              kinematic.increaseSpeed(-1, 0);
            }
          } else {

            //if no more targets to check, do the normal calculation

            //kinematic.getSpeed() is how fast we're moving, direction.mag() is how far are we from target
            //Ideal speed here should be 80 at dist 85, and reduce linearly from there, hitting 0 at 5 units?
            //This can be changed later if it isn't good
            
            
            float idealSpeed = (1 * direction.mag() - 5);
            
            if (idealSpeed < 0) {
              idealSpeed = 0;
            }

            if (kinematic.getSpeed() < idealSpeed) {
              kinematic.increaseSpeed(1, 0);
            } else if (kinematic.getSpeed() > idealSpeed) {
              kinematic.increaseSpeed(-1, 0);
            }
          }
        } else {

          //if waypoints is null, do normal things
          println("waypoints is null");

          //This code should trigger if there's only one target left

          //kinematic.getSpeed() is how fast we're moving, direction.mag() is how far are we from target
          //Ideal speed here should be 80 at dist 85, and reduce linearly from there, hitting 0 at 5 units?
          //This can be changed later if it isn't good

          float idealSpeed = 1 * direction.mag() + 10;

          //if idealSpeed is "negative" we should just set it to 0
          if (idealSpeed < 0) {
            idealSpeed = 0;
          }

          println(idealSpeed);

          //use this to know how off the target speed we are, and slow down accordingly
          //This will be positive if the ideal speed is higher than current speed, negative if ideal speed is lower.
          float speedOffset = (idealSpeed - kinematic.getSpeed());

          if (abs(speedOffset) < 1) {
            kinematic.increaseSpeed(speedOffset, 0);
          } else if (idealSpeed < speedOffset) {
            kinematic.increaseSpeed(1, 0);
          } else if (idealSpeed > speedOffset) {
            kinematic.increaseSpeed(-1, 0);
          }
        }
      }



      //drawing a line for testing purposes
      //line(kinematic.position.x, kinematic.position.y, kinematic.position.x + direction.x, kinematic.position.y + direction.y);

      //handling going to multiple targets

      //initial check exists because waypoints will be null for a single target
      if (waypoints != null) {
        //If within 5 units, move to next target
        if (direction.mag() < 5) {
          //This ensures that the same target can't trigger moving to the next target twice
          if (stillInRadius == false) {
            //this ensures that waypoints get cleared after finishing checking all targets
            if (currentTarget < waypoints.size() - 1) {
              currentTarget++;
            } else {
              currentTarget = 0;
              waypoints = null;
            }
          }
          stillInRadius = true;
          if (waypoints != null) {
            seek(waypoints.get(currentTarget));
          }
        } else {
          stillInRadius = false;
        }
      }
    }

    // place crumbs, do not change
    if (LEAVE_CRUMBS && (millis() - this.last_crumb > CRUMB_INTERVAL))
    {
      this.last_crumb = millis();
      this.crumbs = (Crumb[])append(this.crumbs, new Crumb(this.kinematic.position));
      if (this.crumbs.length > MAX_CRUMBS)
        this.crumbs = (Crumb[])subset(this.crumbs, 1);
    }

    // do not change
    this.kinematic.update(dt);

    draw();
  }

  void draw()
  {
    for (Crumb c : this.crumbs)
    {
      c.draw();
    }

    fill(255);
    noStroke();
    float x = kinematic.position.x;
    float y = kinematic.position.y;
    float r = kinematic.heading;
    circle(x, y, BOID_SIZE);
    // front
    float xp = x + BOID_SIZE*cos(r);
    float yp = y + BOID_SIZE*sin(r);

    // left
    float x1p = x - (BOID_SIZE/2)*sin(r);
    float y1p = y + (BOID_SIZE/2)*cos(r);

    // right
    float x2p = x + (BOID_SIZE/2)*sin(r);
    float y2p = y - (BOID_SIZE/2)*cos(r);
    triangle(xp, yp, x1p, y1p, x2p, y2p);
  }

  void seek(PVector target)
  {
    this.target = target;
  }

  void follow(ArrayList<PVector> waypoints)
  {

    this.waypoints = waypoints;

    seek(waypoints.get(0));
  }
}