com.opensmile.maven.openSMILEclassifier.java Source code

Introduction

Here is the source code for com.opensmile.maven.openSMILEclassifier.java

Source

/***
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
    
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.
    
You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.
***/

package com.opensmile.maven;

import java.io.BufferedReader;
import java.io.File;
import java.io.IOException;
import java.io.InputStreamReader;

import org.json.JSONObject;

public class openSMILEclassifier extends Classifier {

    @Override
    public boolean configure(JSONObject jo) {
        configFile = jo.getString("config");
        entry = jo.getString("entry");
        intype = jo.getString("intype");
        return true;
    }

    @Override
    public JSONObject classify(String audioFile) {
        String line = "";
        String fullLine = "";
        try {
            String str = paths.SMILExtractDir + "SMILExtract " + " -C " + paths.SMILExtractDir + configFile + " -I "
                    + audioFile + " -l 1";
            Process p = Runtime.getRuntime().exec(str, null, new File(paths.SMILExtractDir));
            logger_instance.write(1, "opensmile command: " + str);
            p.waitFor();
            logger_instance.write(1, "opensmile command performed");
            BufferedReader in = new BufferedReader(new InputStreamReader(p.getInputStream()));
            while ((line = in.readLine()) != null)
                fullLine = fullLine + line;
            in = new BufferedReader(new InputStreamReader(p.getErrorStream()));
            while ((line = in.readLine()) != null)
                fullLine = fullLine + line;
            logger_instance.write(1, "opensmile output:" + fullLine);
        } catch (IOException | InterruptedException e) {
            e.printStackTrace();
        }
        JSONObject jo = new JSONObject(fullLine);

        JSONObject jop = new JSONObject();
        switch ((String) jo.get("TYPE")) {
        case "regression":
            double out = jo.getDouble("VALUE");
            double conf = (double) ((JSONObject) jo.getJSONArray("PROB").get(0)).get("CONFIDENCE");
            jop = createJsonEntry(entry, out);
            break;
        case "classification":
            jop = createJsonEntry(entry, jo.getJSONArray("PROB"));
            break;
        }
        return jop;
    }
}