Merge pull request #11 from CPqD/feature/KASR-2124

Suporte aos classificadore

Author: fzaruch

recognizer/src/main/java/br/com/cpqd/asr/recognizer/model/RecognitionResult.java

@@ -17,6 +17,7 @@
 
 import java.util.ArrayList;
 import java.util.List;
+import br.com.cpqd.asr.recognizer.model.classifier.ClassifierResult;
 
 /**
  * Represents the final result of the recognition process.
@@ -43,6 +44,7 @@ public class RecognitionResult {
 
 	/** the list of recognition result alternative sentences. */
 	private List<RecognitionAlternative> alternatives = new ArrayList<>();
+	private ClassifierResult classifierResult = new ClassifierResult();
 
 	public RecognitionResultCode getResultCode() {
 		return resultCode;
@@ -92,6 +94,14 @@ public void setAlternatives(List<RecognitionAlternative> alternatives) {
 		this.alternatives = alternatives;
 	}
 
+	public ClassifierResult getClassifierResult() {
+		return classifierResult;
+	}
+
+	public void setClassifierResult(ClassifierResult classifierResult) {
+		this.classifierResult = classifierResult;
+	}
+
 	@Override
 	public String toString() {
 		return "[code=" + resultCode + ", alternatives=" + alternatives + "]";

recognizer/src/main/java/br/com/cpqd/asr/recognizer/model/classifier/AgeResult.java

@@ -0,0 +1,51 @@
+package br.com.cpqd.asr.recognizer.model.classifier;
+
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.Map;
+
+public class AgeResult {
+    private String event = "";
+    private Integer age = 0;
+    private Map<String, Double> p = new HashMap<>();
+    private String confidence = "";
+    public AgeResult() {
+    }
+
+    public void setEvent(String event) {
+        this.event = event;
+    }
+
+    public String getEvent() {
+        return this.event;
+    }
+
+    public void setAge(Integer age) {
+        this.age = age;
+    }
+
+    public Integer getAge() {
+        return this.age;
+    }
+
+    public void setConfidence(String confidence) {
+        this.event = confidence;
+    }
+
+    public String getConfidence() {
+        return this.confidence;
+    }
+
+    public void setP(Map<String, Double> p) {
+        this.p.clear();
+        this.p.putAll(p);
+    }
+
+    public Map<String, Double> getP() {
+        return this.p;
+    }
+
+    public String toString() {
+        return "[ event=" + event + ", age= " + age + ", p=" + p + ", confidence=" + confidence + " ]";
+    }
+}

recognizer/src/main/java/br/com/cpqd/asr/recognizer/model/classifier/ClassifierResult.java

@@ -0,0 +1,40 @@
+package br.com.cpqd.asr.recognizer.model.classifier;
+
+public class ClassifierResult {
+    private GenderResult genderResult;
+    private AgeResult ageResult;
+    private EmotionResult emotionResult;
+
+    public void setGenderResult(GenderResult genderResult) {
+        this.genderResult = genderResult;
+    }
+    public GenderResult getGenderResult() {
+        return genderResult;
+    }
+
+    public void setAgeResult(AgeResult ageResult) {
+        this.ageResult = ageResult;
+    }
+    public AgeResult getAgeResult() {
+        return ageResult;
+    }
+
+    public void setEmotionResult(EmotionResult emotionResult) {
+        this.emotionResult = emotionResult;
+    }
+    public EmotionResult getEmotionResult() {
+        return emotionResult;
+    }
+
+    @Override
+    public String toString() {
+        String str = "";
+        if (ageResult != null)
+            str += ageResult.toString();
+        if (emotionResult != null)
+            str += " " + emotionResult.toString();
+        if (genderResult != null)
+            str += " " + genderResult.toString();
+        return str;
+    }
+}

recognizer/src/main/java/br/com/cpqd/asr/recognizer/model/classifier/EmotionResult.java

@@ -0,0 +1,59 @@
+package br.com.cpqd.asr.recognizer.model.classifier;
+
+import java.util.HashMap;
+import java.util.Map;
+
+public class EmotionResult {
+    private String event = "";
+    private String emotion = "";
+    private String group = "";
+    private Map<String, Double> p = new HashMap<>();
+    private Map<String, Double> p_groups = new HashMap<>();
+    public EmotionResult() {
+    }
+    public void setEvent(String event) {
+        this.event = event;
+    }
+
+    public String getEvent() {
+        return this.event;
+    }
+
+    public void setEmotion(String emotion) {
+        this.emotion = emotion;
+    }
+
+    public String getEmotion() {
+        return this.emotion;
+    }
+
+    public void setGroup(String group) {
+        this.event = group;
+    }
+
+    public String getGroup() {
+        return this.group;
+    }
+
+    public void setP(Map<String, Double> p) {
+        this.p.clear();
+        this.p.putAll(p);
+    }
+
+    public Map<String, Double> getP() {
+        return this.p;
+    }
+
+    public void setP_groups(Map<String, Double> p) {
+        this.p_groups.clear();
+        this.p_groups.putAll(p);
+    }
+
+    public Map<String, Double> getP_groups() {
+        return this.p_groups;
+    }
+
+    public String toString() {
+        return "[ event=" + event + ", emotion= " + emotion + ", group= " + group + ", p=" + p + ", p_groups=" + p_groups + " ]";
+    }
+}

recognizer/src/main/java/br/com/cpqd/asr/recognizer/model/classifier/GenderResult.java

@@ -0,0 +1,38 @@
+package br.com.cpqd.asr.recognizer.model.classifier;
+
+import java.util.ArrayList;
+import java.util.List;
+
+public class GenderResult {
+    private String event = "";
+    private String gender = "";
+    private List<Double> p = new ArrayList<>();
+
+    public void setEvent(String event) {
+        this.event = event;
+    }
+
+    public String getEvent() {
+        return this.event;
+    }
+
+    public void setGender(String gender) {
+        this.gender = gender;
+    }
+
+    public String getGender() {
+        return this.gender;
+    }
+
+    public void setP (List<Double> p) {
+        this.p.clear();
+        this.p.addAll(p);
+    }
+    public List<Double> getP() {
+        return this.p;
+    }
+
+    public String toString() {
+        return "[ event=" + event + ", gender= " + gender + ", p=" + p + " ]";
+    }
+}