SpeechTranscription_nwc_yarp.cpp

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: 2023-2023 Istituto Italiano di Tecnologia (IIT)
 * SPDX-License-Identifier: BSD-3-Clause
 */
 
#include "SpeechTranscription_nwc_yarp.h"
 
#include <yarp/os/LogComponent.h>
#include <yarp/os/LogStream.h>
#include <yarp/os/Os.h>
 
namespace {
YARP_LOG_COMPONENT(SPEECHTR_NWC, "yarp.devices.speechTranscription_nwc_yarp")
}
 
SpeechTranscription_nwc_yarp::~SpeechTranscription_nwc_yarp()
{
    closeMain();
}
 
bool SpeechTranscription_nwc_yarp::close()
{
    return closeMain();
}
 
bool SpeechTranscription_nwc_yarp::open(yarp::os::Searchable& config)
{
    if (!parseParams(config)) { return false; }
 
    std::string thriftClientPortName = m_local + "/thrift:c";
    std::string thriftServerPortName = m_remote + "/rpc";
 
    if(!m_thriftClientPort.open(thriftClientPortName))
    {
        yCError(SPEECHTR_NWC) << "open() failed: could not open" << thriftClientPortName <<"check the network";
 
        return false;
    }
 
    if(!yarp::os::Network::connect(thriftClientPortName, thriftServerPortName))
    {
        yCError(SPEECHTR_NWC) << "Network::connect() failed: could not connect" << thriftClientPortName << "with" << thriftServerPortName;
 
        return false;
    }
 
    if(!m_thriftClient.yarp().attachAsClient(m_thriftClientPort))
    {
        yCError(SPEECHTR_NWC) << "Error! Cannot attach the m_thriftClientPort (" << thriftClientPortName << ") port as a client";
 
        return false;
    }
 
    //Check the protocol version
    if (!m_thriftClient.checkProtocolVersion()) { return false; }
 
    yCInfo(SPEECHTR_NWC) << "Opening of NWC successful";
    return true;
}
 
bool SpeechTranscription_nwc_yarp::closeMain()
{
    // Close thrift client port
    m_thriftClientPort.close();
 
    return true;
}
 
yarp::dev::ReturnValue SpeechTranscription_nwc_yarp::setLanguage(const std::string& language)
{
    return_set_language result = m_thriftClient.set_language(language);
    if(!result.ret)
    {
        yCError(SPEECHTR_NWC) << "Error while setting language to" << language;
    }
    else
    {
    }
    return result.ret;
}
 
yarp::dev::ReturnValue SpeechTranscription_nwc_yarp::getLanguage(std::string& language)
{
    return_get_language result = m_thriftClient.get_language();
    if(!result.ret)
    {
        yCError(SPEECHTR_NWC) << "Error while retrieving language";
    }
    else
    {
        language = result.language;
    }
    return result.ret;
}
 
yarp::dev::ReturnValue SpeechTranscription_nwc_yarp::transcribe(const yarp::sig::Sound& sound, std::string& transcription, double& score)
{
    YARP_UNUSED(sound);
    YARP_UNUSED(transcription);
    YARP_UNUSED(score);
 
    return_transcribe result = m_thriftClient.transcribe(sound);
 
    if(!result.ret)
    {
        yCError(SPEECHTR_NWC) << "Error while transcribing the audio signal";
    }
    else
    {
        transcription = result.transcription;
        score = result.score;
    }
    return result.ret;
}