// The Coding Train / Daniel Shiffman

// https://thecodingtrain.com/Courses/ml5-beginners-guide/7.1-posenet.html

// https://youtu.be/OIo-DIOkNVg

// https://editor.p5js.org/codingtrain/sketches/ULA97pJXR

​

let video;

let poseNet;

let pose;

let skeleton;

let loco= 0;

​

function setup() {

  createCanvas(640, 480);

  video = createCapture(VIDEO);

  video.hide();

  poseNet = ml5.poseNet(video, modelLoaded);

  poseNet.on('pose', gotPoses);

}

​

function gotPoses(poses) {

  //console.log(poses); 

  if (poses.length > 0) {

    pose = poses[0].pose;

    skeleton = poses[0].skeleton;

  }

}

​

​

function modelLoaded() {

  console.log('poseNet ready');

}

​

function draw() {

   if (!serialActive) {

    text("Press Space Bar to select Serial Port", 20, 30);

  } else {

    text("Connected", 20, 30);

  }

  image(video, 0, 0);

​

  if (pose) {

    fill(255, 0, 0);

    ellipse(pose.nose.x, pose.nose.y, 20);

    loco = int(pose.nose.x);

    val = int(map(loco, 0, 640, 0, 180));

​

    print(val)

  }

}

​

function keyPressed() {

  if (key == " ") {

    // important to have in order to start the serial connection!!

    setUpSerial();

  }

}

​

function readSerial(data) {

  ////////////////////////////////////

  //READ FROM ARDUINO HERE

  ////////////////////////////////////

​

  if (data != null) {

    // make sure there is actually a message

    // split the message

    let fromArduino = split(trim(data), ",");

    // if the right length, then proceed

    if (fromArduino.length == 2) {

      // only store values here

      // do everything with those values in the main draw loop

      print("nice");

      // We take the string we get from Arduino and explicitly

      // convert it to a number by using int()

      // e.g. "103" becomes 103

    }

​

    //////////////////////////////////

    //SEND TO ARDUINO HERE (handshake)

    //////////////////////////////////

    let sendToArduino = val + "\n";

    writeSerial(sendToArduino);

  }

}