hlrc

/*

 * This file is part of hlrc_server

 *

 * Copyright(c) sschulz <AT> techfak.uni-bielefeld.de

 * http://opensource.cit-ec.de/projects/hlrc_server

 *

 * This file may be licensed under the terms of the

 * GNU General Public License Version 3 (the ``GPL''),

 * or (at your option) any later version.

 *

 * Software distributed under the License is distributed

 * on an ``AS IS'' basis, WITHOUT WARRANTY OF ANY KIND, either

 * express or implied. See the GPL for the specific language

 * governing rights and limitations.

 *

 * You should have received a copy of the GPL along with this

 * program. If not, go to http://www.gnu.org/licenses/gpl.html

 * or write to the Free Software Foundation, Inc.,

 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.

 *

 * The development of this software was supported by the

 * Excellence Cluster EXC 277 Cognitive Interaction Technology.

 * The Excellence Cluster EXC 277 is a grant of the Deutsche

 * Forschungsgemeinschaft (DFG) in the context of the German

 * Excellence Initiative.

 *

 */

#pragma once

#include "AudioData.h"

using namespace std;

using namespace hlrc_server;

#include "hlrc_server/phoneme.h"

#include "hlrc_server/soundchunk.h"

#include "hlrc_server/utterance.h"

#include "hlrc_server/ttsAction.h"

// converter from ros utterance

class UtteranceROS : public Utterance {

public:

        UtteranceROS(ttsResultConstPtr ros_utterance) {

                // set text:

                set_text(ros_utterance->utterance.text);

                // convert soundchunk to audio data:

                extract_audio_data(ros_utterance->utterance.audio);

                // convert phonemes:

                extract_phonemes(ros_utterance->utterance.phonemes);

        }

        ~UtteranceROS(){};

        void extract_audio_data(soundchunk sound_chunk) {

                // extract data:

                unsigned int audio_len = sound_chunk.data.size();

                char* audio_data_char = (char*)sound_chunk.data.data();

                audio_data->samples.resize(audio_len);

                audio_data->samples.assign(audio_data_char, audio_data_char + audio_len);

                printf("audio samplesize is %d bytes\n", (unsigned int)audio_data->samples.size());

                // extract format:

                audio_data->sample_signed = true;

                switch (sound_chunk.sample_type) {

                        case (soundchunk::SAMPLE_U8):

                                audio_data->sample_signed = false; // and fall through:

                        case (soundchunk::SAMPLE_S8):

                                audio_data->sample_bit = 8;

                                break;

                        case (soundchunk::SAMPLE_U16):

                                audio_data->sample_signed = false; // and fall through:

                        case (soundchunk::SAMPLE_S16):

                                audio_data->sample_bit = 16;

                                break;

                        case (soundchunk::SAMPLE_U24):

                                audio_data->sample_signed = false; // and fall through:

                        case (soundchunk::SAMPLE_S24):

                                audio_data->sample_bit = 24;

                                break;

                        default:

                                printf("> invalid sample type %d in ROS SoundChunk! ignoring request!\n", sound_chunk.sample_type);

                                throw runtime_error("UtteranceROS::convert_audio_data() unsupported sample type in ros SoundChunk");

                }

                // bitrate

                audio_data->sample_rate = sound_chunk.rate;

                // endianness

                if (sound_chunk.endianess == soundchunk::ENDIAN_LITTLE) {

                        audio_data->sample_big_endian = false;

                }

                else if (sound_chunk.endianess == soundchunk::ENDIAN_BIG) {

                        audio_data->sample_big_endian = true;

                }

                else {

                        printf("> invalid SoundChunk byte_format");

                        throw runtime_error("UtteranceROS::convert_audio_data() unsupported byte_format in ros SoundChunk");

                }

                // number of channels

                audio_data->sample_channels = sound_chunk.channels;

                printf("> new AudioData: %s\n", audio_data->to_string().c_str());

        }

        void extract_phonemes(utterance::_phonemes_type ros_phones) {

                // extract phoneme vector

                phonemes_vector.clear();

                for (auto it = ros_phones.begin(); it < ros_phones.end(); it++) {

                        Utterance::symbol_duration_pair_t phoneme = make_pair(it->symbol, it->duration);

                        phonemes_vector.push_back(phoneme);

                }

        }

};