71 lines
1.5 KiB
Java
71 lines
1.5 KiB
Java
package com.rubenvandeven.emotionhero;
|
|
|
|
import android.graphics.Color;
|
|
|
|
import com.affectiva.android.affdex.sdk.detector.Face;
|
|
|
|
/**
|
|
* Created by ruben on 16/08/16.
|
|
*/
|
|
|
|
public enum Emotion {
|
|
ANGER,
|
|
CONTEMPT,
|
|
DISGUST,
|
|
FEAR,
|
|
JOY,
|
|
SADNESS,
|
|
SURPRISE;
|
|
|
|
public float getValueFromFace(Face face)
|
|
{
|
|
if(face == null)
|
|
return 0f;
|
|
|
|
switch (this) {
|
|
case ANGER:
|
|
return face.emotions.getAnger();
|
|
case CONTEMPT:
|
|
return face.emotions.getContempt();
|
|
case DISGUST:
|
|
return face.emotions.getDisgust();
|
|
case FEAR:
|
|
return face.emotions.getFear();
|
|
case JOY:
|
|
return face.emotions.getJoy();
|
|
case SADNESS:
|
|
return face.emotions.getSadness();
|
|
case SURPRISE:
|
|
return face.emotions.getSurprise();
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
public int getColor()
|
|
{
|
|
switch (this) {
|
|
case ANGER:
|
|
return Color.RED;
|
|
case CONTEMPT:
|
|
return Color.BLUE;
|
|
case DISGUST:
|
|
return Color.GREEN;
|
|
case FEAR:
|
|
return Color.MAGENTA;
|
|
case JOY:
|
|
return Color.YELLOW;
|
|
case SADNESS:
|
|
return Color.CYAN;
|
|
case SURPRISE:
|
|
return Color.LTGRAY;
|
|
}
|
|
|
|
return Color.BLACK;
|
|
}
|
|
|
|
|
|
public String getDbName() {
|
|
return this.name().toLowerCase();
|
|
}
|
|
}
|