//----------------------------------------------------------------------------
// name: face-relay.ck
// desc: relay FaceOSC messages to Wekinator
//
// get FaceOSC here (and also see the OSC message it sends)
//     https://github.com/kylemcdonald/ofxFaceTracker/releases
//
// author: Ge Wang (https://ccrma.stanford.edu/~ge/)
// date: Winter 2023
//----------------------------------------------------------------------------

// our OSC receiver (from FaceOSC)
OscIn oin;
// incoming port (from FaceOSC)
8338 => oin.port;
// our OSC message shuttle
OscMsg msg;
// listen for all message
oin.listenAll();

// destination host name
"localhost" => string hostname;
// destination port number: 6448 is Wekinator default
6448 => int port;
// our OSC sender (to Wekinator)
OscOut xmit;
// aim the transmitter at destination
xmit.dest( hostname, port );

// just two of the many parameters
float MOUTH_WIDTH;
float MOUTH_HEIGHT;
float EYEBROW_LEFT;
float EYEBROW_RIGHT;
float SCALE;
float JAW;
float POSITION_X;
float POSITION_Y;
float ORIENTATION_X;
float ORIENTATION_Y;
float ORIENTATION_Z;
float EYE_LEFT;
float EYE_RIGHT;

// print
cherr <= "listening for messages on port " <= oin.port()
      <= "..." <= IO.newline();
      
// spork the listener
spork ~ incoming();

// main shred loop
while( true )
{
    // can do things here at a different rate
    // for now, do nothing

    // advance time
    1::second => now;
}

// listener
fun void incoming()
{
    // infinite event loop
    while( true )
    {
        // wait for event to arrive
        oin => now;
        
        // grab the next message from the queue. 
        while( oin.recv(msg) )
        {         
            // print message type
            cherr <= "RECEIVED: \"" <= msg.address <= "\": ";        
            // print arguments
            printArgs( msg );
            
            // handle message
            if( msg.address == "/gesture/mouth/width" )
            {            
                // save
                msg.getFloat(0) => MOUTH_WIDTH;
                // (optional) could normalize here
                // Math.remap( MOUTH_WIDTH, MOUTH_WIDTH_MIN, MOUTH_WIDTH_MAX, 0, 1 ) => MOUTH_WIDTH;
            }
            else if( msg.address == "/gesture/mouth/height" )
            {
                // save
                msg.getFloat(0) => MOUTH_HEIGHT;
                // (optional) could normalize here
                // Math.remap( MOUTH_HEIGHT, MOUTH_HEIGHT_MIN, MOUTH_HEIGHT_MAX, 0, 1 ) => MOUTH_HEIGHT;
            }
            else if( msg.address == "/gesture/eyebrow/left" )
            {
                // save
                msg.getFloat(0) => EYEBROW_LEFT;
                // (optional) could normalize here
                // Math.remap( MOUTH_HEIGHT, MOUTH_HEIGHT_MIN, MOUTH_HEIGHT_MAX, 0, 1 ) => MOUTH_HEIGHT;
            }
            else if( msg.address == "/gesture/eyebrow/right" )
            {
                // save
                msg.getFloat(0) => EYEBROW_RIGHT;
                // (optional) could normalize here
                // Math.remap( MOUTH_HEIGHT, MOUTH_HEIGHT_MIN, MOUTH_HEIGHT_MAX, 0, 1 ) => MOUTH_HEIGHT;
            }
            else if( msg.address == "/pose/scale" )
            {
                // save
                msg.getFloat(0) => SCALE;
                // (optional) could normalize here
                // Math.remap( MOUTH_HEIGHT, MOUTH_HEIGHT_MIN, MOUTH_HEIGHT_MAX, 0, 1 ) => MOUTH_HEIGHT;
            }
            else if( msg.address == "/gesture/jaw" )
            {
                // save
                msg.getFloat(0) => JAW;
                // (optional) could normalize here
                // Math.remap( MOUTH_HEIGHT, MOUTH_HEIGHT_MIN, MOUTH_HEIGHT_MAX, 0, 1 ) => MOUTH_HEIGHT;
            }
            else if( msg.address == "/pose/position" )
            {
                // save
                msg.getFloat(0) => POSITION_X;
                // (optional) could normalize here
                // Math.remap( MOUTH_HEIGHT, MOUTH_HEIGHT_MIN, MOUTH_HEIGHT_MAX, 0, 1 ) => MOUTH_HEIGHT;
            }
            else if( msg.address == "/pose/position" )
            {
                // save
                msg.getFloat(1) => POSITION_Y;
                // (optional) could normalize here
                // Math.remap( MOUTH_HEIGHT, MOUTH_HEIGHT_MIN, MOUTH_HEIGHT_MAX, 0, 1 ) => MOUTH_HEIGHT;
            }
            else if( msg.address == "/pose/orientation" )
            {
                // save
                msg.getFloat(0) => ORIENTATION_X;
                // (optional) could normalize here
                // Math.remap( MOUTH_HEIGHT, MOUTH_HEIGHT_MIN, MOUTH_HEIGHT_MAX, 0, 1 ) => MOUTH_HEIGHT;
            }
            else if( msg.address == "/pose/orientation" )
            {
                // save
                msg.getFloat(1) => ORIENTATION_Y;
                // (optional) could normalize here
                // Math.remap( MOUTH_HEIGHT, MOUTH_HEIGHT_MIN, MOUTH_HEIGHT_MAX, 0, 1 ) => MOUTH_HEIGHT;
            }
            else if( msg.address == "/pose/orientation" )
            {
                // save
                msg.getFloat(2) => ORIENTATION_Z;
                // (optional) could normalize here
                // Math.remap( MOUTH_HEIGHT, MOUTH_HEIGHT_MIN, MOUTH_HEIGHT_MAX, 0, 1 ) => MOUTH_HEIGHT;
            }
            else if( msg.address == "/gesture/eye/left" )
            {
                // save
                msg.getFloat(1) => EYE_LEFT;
                // (optional) could normalize here
                // Math.remap( MOUTH_HEIGHT, MOUTH_HEIGHT_MIN, MOUTH_HEIGHT_MAX, 0, 1 ) => MOUTH_HEIGHT;
            }
            else if( msg.address == "/gesture/eye/right" )
            {
                // save
                msg.getFloat(1) => EYE_RIGHT;
                // (optional) could normalize here
                // Math.remap( MOUTH_HEIGHT, MOUTH_HEIGHT_MIN, MOUTH_HEIGHT_MAX, 0, 1 ) => MOUTH_HEIGHT;
            }
        }
        
        // reformat and relay message to Wekinator
        send2wek();
    }
}

// reformat and send what we want to Wekinator
fun void send2wek()
{
    // start the message...
    xmit.start( "/wek/inputs" );
    
    // print
    cherr <= "  *** SENDING: \"/wek/inputs/\": "
          <= MOUTH_WIDTH <= " " <= MOUTH_HEIGHT <= " " <= EYEBROW_LEFT <= " " <= EYEBROW_RIGHT <= " " <= SCALE <= " " <= JAW <= " " <= POSITION_X <= " " <= POSITION_Y <= " " 
          <= ORIENTATION_X <= " " <= ORIENTATION_Y <= " " <= ORIENTATION_Z <= " " <= EYE_LEFT <= " " <= EYE_RIGHT <= " " <= IO.newline();

    // add each for sending
    MOUTH_WIDTH => xmit.add;
    MOUTH_HEIGHT => xmit.add;
    EYEBROW_LEFT => xmit.add;
    EYEBROW_RIGHT => xmit.add;
    JAW => xmit.add;
    SCALE => xmit.add;
    POSITION_X => xmit.add;
    POSITION_Y => xmit.add;
    ORIENTATION_X => xmit.add;
    ORIENTATION_Y => xmit.add;
    ORIENTATION_Z => xmit.add;
    EYE_LEFT => xmit.add;
    EYE_RIGHT => xmit.add;
    
    // send it
    xmit.send();
}

// print argument
fun void printArgs( OscMsg msg )
{
    // iterate over
    for( int i; i < msg.numArgs(); i++ )
    {
        if( msg.typetag.charAt(i) == 'f' ) // float
        {
            cherr <= msg.getFloat(i) <= " ";
        }
        else if( msg.typetag.charAt(i) == 'i' ) // int
        {
            cherr <= msg.getInt(i) <= " ";
        }
        else if( msg.typetag.charAt(i) == 's' ) // string
        {
            cherr <= msg.getString(i) <= " ";
        }            
    }       
    
    // new line
    cherr <= IO.newline();
}