voice_recognition_alg.h

Go to the documentation of this file.

// Copyright (C) 2010-2011 Institut de Robotica i Informatica Industrial, CSIC-UPC.
// Author 
// All rights reserved.
//
// This file is part of iri-ros-pkg
// iri-ros-pkg is free software: you can redistribute it and/or modify
// it under the terms of the GNU Lesser 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 Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with this program.  If not, see <http://www.gnu.org/licenses/>.
// 
// IMPORTANT NOTE: This code has been generated through a script from the 
// iri_ros_scripts. Please do NOT delete any comments to guarantee the correctness
// of the scripts. ROS topics can be easly add by using those scripts. Please
// refer to the IRI wiki page for more information:
// http://wikiri.upc.es/index.php/Robotics_Lab

#ifndef _voice_recognition_alg_h_
#define _voice_recognition_alg_h_

#include <iri_voice_recognition/VoiceRecognitionConfig.h>
#include "mutex.h"

//include voice_recognition_alg main library
extern "C"
{
  #include <iatros-speech/version.h>
  #include <iatros/version.h>
  #include <prhlt/args.h>
  #include <iatros/lex.h>
  #include <iatros/hmm.h>
  #include <iatros/grammar.h>
  #include <iatros/viterbi.h>
  #include <iatros/heap.h>
  #include <iatros/lattice.h>
  #include <prhlt/trace.h>
  #include <prhlt/gzip.h>
  #include <prhlt/constants.h>
  #include <iatros-speech/libAudio.h>
  #include <iatros-speech/acquisitor.h>
  #include <time.h>
}

#define NONE 0
#define AUTORES 1
#define HABILIDADES 2
#define IDENTIFICACION 3
#define LUGAR 4
#define OTROS_PROYECTOS 5
#define OTROS_ROBOTS 6
#define ZONA 7
#define EDIF 8
#define AUL 9
#define NCLASSES 10

class VoiceRecognitionAlgorithm
{
  protected:
    CMutex alg_mutex_;

    // private attributes and methods
    std::string filePath_;
    
    args_t *args;
    decoder_t *decoder;
    acquisitor_t *acquisitor;
    
    std::string answer_type(std::string answerfn);
    std::string answer_type_location(std::string answerfn, std::string loc);
    std::string generate_answer(double *probs, char **sentences_str, int n);
    std::string outputs(const args_t *args, lattice_t *lattice);

  public:
    typedef iri_voice_recognition::VoiceRecognitionConfig Config;

    Config config_;

    VoiceRecognitionAlgorithm(void);

    void lock(void) { alg_mutex_.enter(); };

    void unlock(void) { alg_mutex_.exit(); };

    bool try_enter(void) { return alg_mutex_.try_enter(); };

    void config_update(Config& new_cfg, uint32_t level=0);

    // here define all voice_recognition_alg interface methods to retrieve and set
    // the driver parameters
    void initialize();
    std::string iteration();  
    void finalize();
    void setFilePath(std::string path);
    
    int answerCode_;

    ~VoiceRecognitionAlgorithm(void);
};

#endif