Merge pull request #3248 from Atlante45/dde_face_tracker

Dde face tracker

interface/src/Application.cpp

@@ -1502,10 +1502,11 @@ glm::vec3 Application::getMouseVoxelWorldCoordinates(const VoxelDetail& mouseVox
 }
 
 FaceTracker* Application::getActiveFaceTracker() {
-    return _cara.isActive() ? static_cast<FaceTracker*>(&_cara) : 
-        (_faceshift.isActive() ? static_cast<FaceTracker*>(&_faceshift) :
-        (_faceplus.isActive() ? static_cast<FaceTracker*>(&_faceplus) :
-            (_visage.isActive() ? static_cast<FaceTracker*>(&_visage) : NULL)));
+    return (_dde.isActive() ? static_cast<FaceTracker*>(&_dde) :
+            (_cara.isActive() ? static_cast<FaceTracker*>(&_cara) :
+             (_faceshift.isActive() ? static_cast<FaceTracker*>(&_faceshift) :
+              (_faceplus.isActive() ? static_cast<FaceTracker*>(&_faceplus) :
+               (_visage.isActive() ? static_cast<FaceTracker*>(&_visage) : NULL)))));
 }
 
 struct SendVoxelsOperationArgs {
@@ -1937,13 +1938,21 @@ void Application::updateVisage() {
     _visage.update();
 }
 
+void Application::updateDDE() {
+    bool showWarnings = Menu::getInstance()->isOptionChecked(MenuOption::PipelineWarnings);
+    PerformanceWarning warn(showWarnings, "Application::updateDDE()");
+    
+    //  Update Cara
+    _dde.update();
+}
+
 void Application::updateCara() {
     bool showWarnings = Menu::getInstance()->isOptionChecked(MenuOption::PipelineWarnings);
     PerformanceWarning warn(showWarnings, "Application::updateCara()");
-
+    
     //  Update Cara
     _cara.update();
-
+    
     //  Copy angular velocity if measured by cara, to the head
     if (_cara.isActive()) {
         _myAvatar->getHead()->setAngularVelocity(_cara.getHeadAngularVelocity());
@@ -3265,6 +3274,7 @@ void Application::resetSensors() {
     _faceplus.reset();
     _faceshift.reset();
     _visage.reset();
+    _dde.reset();
 
     OculusManager::reset();
 

interface/src/Application.h

@@ -64,6 +64,7 @@
 #include "devices/SixenseManager.h"
 #include "devices/Visage.h"
 #include "devices/CaraFaceTracker.h"
+#include "devices/DdeFaceTracker.h"
 #include "models/ModelTreeRenderer.h"
 #include "particles/ParticleTreeRenderer.h"
 #include "renderer/AmbientOcclusionEffect.h"
@@ -214,6 +215,7 @@ public:
     Faceplus* getFaceplus() { return &_faceplus; }
     Faceshift* getFaceshift() { return &_faceshift; }
     Visage* getVisage() { return &_visage; }
+    DdeFaceTracker* getDDE() { return &_dde; }
     CaraFaceTracker* getCara() { return &_cara; }
     FaceTracker* getActiveFaceTracker();
     SixenseManager* getSixenseManager() { return &_sixenseManager; }
@@ -393,6 +395,7 @@ private:
     void updateFaceplus();
     void updateFaceshift();
     void updateVisage();
+    void updateDDE();
     void updateCara();
     void updateMyAvatarLookAtPosition();
     void updateThreads(float deltaTime);
@@ -493,6 +496,7 @@ private:
     Faceshift _faceshift;
     Visage _visage;
     CaraFaceTracker _cara;
+    DdeFaceTracker _dde;
 
     SixenseManager _sixenseManager;
     PrioVR _prioVR;

interface/src/avatar/Head.cpp

@@ -68,6 +68,9 @@ void Head::simulate(float deltaTime, bool isMine, bool billboard) {
         if ((_isFaceshiftConnected = faceTracker)) {
             _blendshapeCoefficients = faceTracker->getBlendshapeCoefficients();
             _isFaceshiftConnected = true;   
+        } else if (Application::getInstance()->getDDE()->isActive()) {
+            faceTracker = Application::getInstance()->getDDE();
+            _blendshapeCoefficients = faceTracker->getBlendshapeCoefficients();
         }
     }
     

interface/src/devices/CaraFaceTracker.h

@@ -9,8 +9,8 @@
 //  See the accompanying file LICENSE or http://www.apache.org/licenses/LICENSE-2.0.html
 //
 
-#ifndef hi_fi_CaraFaceTracker_h
-#define hi_fi_CaraFaceTracker_h
+#ifndef hifi_CaraFaceTracker_h
+#define hifi_CaraFaceTracker_h
 
 #include <QUdpSocket>
 
@@ -121,4 +121,4 @@ private:
     int _jawOpenIndex;
 };
 
-#endif //endif hi_fi_CaraFaceTracker_h
+#endif //endif hifi_CaraFaceTracker_h

interface/src/devices/DdeFaceTracker.cpp

@@ -0,0 +1,245 @@
+//
+//  DdeFaceTracker.cpp
+//
+//
+//  Created by Clement on 8/2/14.
+//  Copyright 2014 High Fidelity, Inc.
+//
+//  Distributed under the Apache License, Version 2.0.
+//  See the accompanying file LICENSE or http://www.apache.org/licenses/LICENSE-2.0.html
+//
+
+#include <SharedUtil.h>
+
+#include <QJsonDocument>
+#include <QJsonArray>
+#include <QJsonObject>
+#include <QElapsedTimer>
+
+#include "DdeFaceTracker.h"
+
+static const QHostAddress DDE_FEATURE_POINT_SERVER_ADDR("127.0.0.1");
+static const quint16 DDE_FEATURE_POINT_SERVER_PORT = 5555;
+
+static const int NUM_EXPRESSION = 46;
+static const int MIN_PACKET_SIZE = (8 + NUM_EXPRESSION) * sizeof(float) + sizeof(int);
+static const int MAX_NAME_SIZE = 31;
+
+struct Packet{
+    //roughly in mm
+    float focal_length[1];
+    float translation[3];
+    
+    //quaternion
+    float rotation[4];
+    
+    //blendshape coefficients ranging between -0.2 and 1.5
+    float expressions[NUM_EXPRESSION];
+    
+    //avatar id selected on the UI
+    int avatar_id;
+    
+    //client name, arbitrary length
+    char name[MAX_NAME_SIZE + 1];
+};
+
+DdeFaceTracker::DdeFaceTracker() :
+_lastReceiveTimestamp(0),
+_reset(false),
+_leftBlinkIndex(0), // see http://support.faceshift.com/support/articles/35129-export-of-blendshapes
+_rightBlinkIndex(1),
+_leftEyeOpenIndex(8),
+_rightEyeOpenIndex(9),
+_browDownLeftIndex(14),
+_browDownRightIndex(15),
+_browUpCenterIndex(16),
+_browUpLeftIndex(17),
+_browUpRightIndex(18),
+_mouthSmileLeftIndex(28),
+_mouthSmileRightIndex(29),
+_jawOpenIndex(21)
+{
+    _blendshapeCoefficients.resize(NUM_EXPRESSION);
+    
+    connect(&_udpSocket, SIGNAL(readyRead()), SLOT(readPendingDatagrams()));
+    connect(&_udpSocket, SIGNAL(error(QAbstractSocket::SocketError)), SLOT(socketErrorOccurred(QAbstractSocket::SocketError)));
+    connect(&_udpSocket, SIGNAL(stateChanged(QAbstractSocket::SocketState)), SLOT(socketStateChanged(QAbstractSocket::SocketState)));
+    
+    bindTo(DDE_FEATURE_POINT_SERVER_PORT);
+}
+
+DdeFaceTracker::DdeFaceTracker(const QHostAddress& host, quint16 port) :
+_lastReceiveTimestamp(0),
+_reset(false),
+_leftBlinkIndex(0), // see http://support.faceshift.com/support/articles/35129-export-of-blendshapes
+_rightBlinkIndex(1),
+_leftEyeOpenIndex(8),
+_rightEyeOpenIndex(9),
+_browDownLeftIndex(14),
+_browDownRightIndex(15),
+_browUpCenterIndex(16),
+_browUpLeftIndex(17),
+_browUpRightIndex(18),
+_mouthSmileLeftIndex(28),
+_mouthSmileRightIndex(29),
+_jawOpenIndex(21)
+{
+    _blendshapeCoefficients.resize(NUM_EXPRESSION);
+    
+    connect(&_udpSocket, SIGNAL(readyRead()), SLOT(readPendingDatagrams()));
+    connect(&_udpSocket, SIGNAL(error(QAbstractSocket::SocketError)), SLOT(socketErrorOccurred(QAbstractSocket::SocketError)));
+    connect(&_udpSocket, SIGNAL(stateChanged(QAbstractSocket::SocketState)), SIGNAL(socketStateChanged(QAbstractSocket::SocketState)));
+    
+    bindTo(host, port);
+}
+
+DdeFaceTracker::~DdeFaceTracker() {
+    if(_udpSocket.isOpen())
+        _udpSocket.close();
+}
+
+void DdeFaceTracker::init() {
+    
+}
+
+void DdeFaceTracker::reset() {
+    _reset  = true;
+}
+
+void DdeFaceTracker::update() {
+    
+}
+
+void DdeFaceTracker::bindTo(quint16 port) {
+    bindTo(QHostAddress::Any, port);
+}
+
+void DdeFaceTracker::bindTo(const QHostAddress& host, quint16 port) {
+    if(_udpSocket.isOpen()) {
+        _udpSocket.close();
+    }
+    _udpSocket.bind(host, port);
+}
+
+bool DdeFaceTracker::isActive() const {
+    static const int ACTIVE_TIMEOUT_USECS = 3000000; //3 secs
+    return (usecTimestampNow() - _lastReceiveTimestamp < ACTIVE_TIMEOUT_USECS);
+}
+
+//private slots and methods
+void DdeFaceTracker::socketErrorOccurred(QAbstractSocket::SocketError socketError) {
+    qDebug() << "[Error] DDE Face Tracker Socket Error: " << _udpSocket.errorString();
+}
+
+void DdeFaceTracker::socketStateChanged(QAbstractSocket::SocketState socketState) {
+    QString state;
+    switch(socketState) {
+        case QAbstractSocket::BoundState:
+            state = "Bounded";
+            break;
+        case QAbstractSocket::ClosingState:
+            state = "Closing";
+            break;
+        case QAbstractSocket::ConnectedState:
+            state = "Connected";
+            break;
+        case QAbstractSocket::ConnectingState:
+            state = "Connecting";
+            break;
+        case QAbstractSocket::HostLookupState:
+            state = "Host Lookup";
+            break;
+        case QAbstractSocket::ListeningState:
+            state = "Listening";
+            break;
+        case QAbstractSocket::UnconnectedState:
+            state = "Unconnected";
+            break;
+    }
+    qDebug() << "[Info] DDE Face Tracker Socket: " << socketState;
+}
+
+void DdeFaceTracker::readPendingDatagrams() {
+    QByteArray buffer;
+    while (_udpSocket.hasPendingDatagrams()) {
+        buffer.resize(_udpSocket.pendingDatagramSize());
+        _udpSocket.readDatagram(buffer.data(), buffer.size());
+    }
+    decodePacket(buffer);
+}
+
+float DdeFaceTracker::getBlendshapeCoefficient(int index) const {
+    return (index >= 0 && index < (int)_blendshapeCoefficients.size()) ? _blendshapeCoefficients[index] : 0.0f;
+}
+
+static const float DDE_MIN_RANGE = -0.2;
+static const float DDE_MAX_RANGE = 1.5;
+float rescaleCoef(float ddeCoef) {
+    return (ddeCoef - DDE_MIN_RANGE) / (DDE_MAX_RANGE - DDE_MIN_RANGE);
+}
+
+void DdeFaceTracker::decodePacket(const QByteArray& buffer) {
+    if(buffer.size() > MIN_PACKET_SIZE) {
+        Packet packet;
+        int bytesToCopy = glm::min((int)sizeof(packet), buffer.size());
+        memset(&packet.name, '\n', MAX_NAME_SIZE + 1);
+        memcpy(&packet, buffer.data(), bytesToCopy);
+        
+        glm::vec3 translation;
+        memcpy(&translation, packet.translation, sizeof(packet.translation));
+        glm::quat rotation;
+        memcpy(&rotation, &packet.rotation, sizeof(packet.rotation));
+        if (_reset) {
+            memcpy(&_referenceTranslation, &translation, sizeof(glm::vec3));
+            memcpy(&_referenceRotation, &rotation, sizeof(glm::quat));
+            _reset = false;
+        }
+
+        // Compute relative translation
+        float LEAN_DAMPING_FACTOR = 40;
+        translation -= _referenceTranslation;
+        translation /= LEAN_DAMPING_FACTOR;
+        translation.x *= -1;
+        
+        // Compute relative rotation
+        rotation = glm::inverse(_referenceRotation) * rotation;
+        
+        // copy values
+        _headTranslation = translation;
+        _headRotation = rotation;
+        
+        // Set blendshapes
+        float BLINK_MAGNIFIER = 2.0f;
+        _blendshapeCoefficients[_leftBlinkIndex] = rescaleCoef(packet.expressions[1]) * BLINK_MAGNIFIER;
+        _blendshapeCoefficients[_rightBlinkIndex] = rescaleCoef(packet.expressions[0]) * BLINK_MAGNIFIER;
+        
+        float leftBrow = 1.0f - rescaleCoef(packet.expressions[14]);
+        if (leftBrow < 0.5f) {
+            _blendshapeCoefficients[_browDownLeftIndex] = 1.0f - 2.0f * leftBrow;
+            _blendshapeCoefficients[_browUpLeftIndex] = 0.0f;
+        } else {
+            _blendshapeCoefficients[_browDownLeftIndex] = 0.0f;
+            _blendshapeCoefficients[_browUpLeftIndex] = 2.0f * (leftBrow - 0.5f);
+        }
+        float rightBrow = 1.0f - rescaleCoef(packet.expressions[15]);
+        if (rightBrow < 0.5f) {
+            _blendshapeCoefficients[_browDownRightIndex] = 1.0f - 2.0f * rightBrow;
+            _blendshapeCoefficients[_browUpRightIndex] = 0.0f;
+        } else {
+            _blendshapeCoefficients[_browDownRightIndex] = 0.0f;
+            _blendshapeCoefficients[_browUpRightIndex] = 2.0f * (rightBrow - 0.5f);
+        }
+        
+        float JAW_OPEN_MAGNIFIER = 1.4f;
+        _blendshapeCoefficients[_jawOpenIndex] = rescaleCoef(packet.expressions[21]) * JAW_OPEN_MAGNIFIER;
+        
+        
+        _blendshapeCoefficients[_mouthSmileLeftIndex] = rescaleCoef(packet.expressions[24]);
+        _blendshapeCoefficients[_mouthSmileRightIndex] = rescaleCoef(packet.expressions[23]);
+        
+        
+    } else {
+        qDebug() << "[Error] DDE Face Tracker Decode Error";
+    }
+    _lastReceiveTimestamp = usecTimestampNow();
+}

interface/src/devices/DdeFaceTracker.h

@@ -0,0 +1,89 @@
+//
+//  DdeFaceTracker.h
+//
+//
+//  Created by Clement on 8/2/14.
+//  Copyright 2014 High Fidelity, Inc.
+//
+//  Distributed under the Apache License, Version 2.0.
+//  See the accompanying file LICENSE or http://www.apache.org/licenses/LICENSE-2.0.html
+//
+
+#ifndef hifi_DdeFaceTracker_h
+#define hifi_DdeFaceTracker_h
+
+#include <QUdpSocket>
+
+#include "FaceTracker.h"
+
+class DdeFaceTracker : public FaceTracker {
+    Q_OBJECT
+    
+public:
+    DdeFaceTracker();
+    DdeFaceTracker(const QHostAddress& host, quint16 port);
+    ~DdeFaceTracker();
+    
+    //initialization
+    void init();
+    void reset();
+    void update();
+    
+    //sockets
+    void bindTo(quint16 port);
+    void bindTo(const QHostAddress& host, quint16 port);
+    bool isActive() const;
+    
+    float getLeftBlink() const { return getBlendshapeCoefficient(_leftBlinkIndex); }
+    float getRightBlink() const { return getBlendshapeCoefficient(_rightBlinkIndex); }
+    float getLeftEyeOpen() const { return getBlendshapeCoefficient(_leftEyeOpenIndex); }
+    float getRightEyeOpen() const { return getBlendshapeCoefficient(_rightEyeOpenIndex); }
+    
+    float getBrowDownLeft() const { return getBlendshapeCoefficient(_browDownLeftIndex); }
+    float getBrowDownRight() const { return getBlendshapeCoefficient(_browDownRightIndex); }
+    float getBrowUpCenter() const { return getBlendshapeCoefficient(_browUpCenterIndex); }
+    float getBrowUpLeft() const { return getBlendshapeCoefficient(_browUpLeftIndex); }
+    float getBrowUpRight() const { return getBlendshapeCoefficient(_browUpRightIndex); }
+    
+    float getMouthSize() const { return getBlendshapeCoefficient(_jawOpenIndex); }
+    float getMouthSmileLeft() const { return getBlendshapeCoefficient(_mouthSmileLeftIndex); }
+    float getMouthSmileRight() const { return getBlendshapeCoefficient(_mouthSmileRightIndex); }
+    
+private slots:
+    
+    //sockets
+    void socketErrorOccurred(QAbstractSocket::SocketError socketError);
+    void readPendingDatagrams();
+    void socketStateChanged(QAbstractSocket::SocketState socketState);
+    
+private:
+    float getBlendshapeCoefficient(int index) const;
+    void decodePacket(const QByteArray& buffer);
+    
+    // sockets
+    QUdpSocket _udpSocket;
+    quint64 _lastReceiveTimestamp;
+    
+    bool _reset;
+    glm::vec3 _referenceTranslation;
+    glm::quat _referenceRotation;
+    
+    int _leftBlinkIndex;
+    int _rightBlinkIndex;
+    int _leftEyeOpenIndex;
+    int _rightEyeOpenIndex;
+    
+    // Brows
+    int _browDownLeftIndex;
+    int _browDownRightIndex;
+    int _browUpCenterIndex;
+    int _browUpLeftIndex;
+    int _browUpRightIndex;
+    
+    int _mouthSmileLeftIndex;
+    int _mouthSmileRightIndex;
+    
+    int _jawOpenIndex;
+};
+
+#endif // hifi_DdeFaceTracker_h