let facemesh;

let video;

let facePredictions = [];

​

​

function preload() {

  facemesh = ml5.faceMesh({flipHorizontal: true})

}

​

function setup() {

  createCanvas(640, 480);

​

  video = createCapture(VIDEO, {flipped: true});

  video.size(width, height);

  facemesh.detectStart(video, results => { facePredictions = results; });

​

​

  video.hide();

}

​

function draw() {

  image(video, 0, 0, width, height);

    drawLaserRays();

}

​

​

function drawLaserRays() {

  if (facePredictions.length) {

    const rightEye = facePredictions[0].keypoints[159];

    const leftEye = facePredictions[0].keypoints[386];

    const nose1 = facePredictions[0].keypoints[6];

    const nose2 = facePredictions[0].keypoints[168];

let midPoint = createVector((nose1.x + nose2.x) / 2, (nose1.y + nose2.y) / 2);

​

    // Calculate head center

    let headCenter = createVector((leftEye.x + rightEye.x) / 2, (leftEye.y + rightEye.y) / 2);

​

    // Determine head orientation

    let headOrientation = createVector(midPoint.x - headCenter.x, (midPoint.y - headCenter.y));

    console.log(headOrientation.mag())

    headOrientation.normalize();

​

    // Adjust laser direction by reversing both x and y for correct yaw and pitch

    //let laserDirection = p5.Vector.fromAngle(atan2(headOrientation.y, headOrientation.x) );

let laserDirection = createVector(headOrientation.x, headOrientation.y);

    let leftLaserEnd = createVector(leftEye.x + laserDirection.x * 1000, leftEye.y + laserDirection.y * 1000);

    let rightLaserEnd = createVector(rightEye.x + laserDirection.x * 1000, rightEye.y + laserDirection.y * 1000);

​

    stroke(255, 0, 0);

    line(leftEye.x, leftEye.y, leftLaserEnd.x, leftLaserEnd.y);

    line(rightEye.x, rightEye.y, rightLaserEnd.x, rightLaserEnd.y);

    fill(0,0,255);

    circle(headCenter.x, headCenter.y, 5)

    fill(0,255,0);

    circle(midPoint.x, midPoint.y, 5);

  }

}