sems/plug-in/ivr/IvrAudio.cpp

#include "IvrAudio.h"
#include "AmAudio.h"
#include "AmSession.h"

#include "log.h"

#ifdef IVR_WITH_TTS

#define TTS_CACHE_PATH "/tmp/"
extern "C" cst_voice *register_cmu_us_kal();

#endif //ivr_with_tts


static PyObject* IvrAudioFile_new(PyTypeObject *type, PyObject *args, PyObject *kwds)
{
    DBG("---------- IvrAudioFile_alloc -----------\n");
    IvrAudioFile *self;

    self = (IvrAudioFile *)type->tp_alloc(type, 0);
	
    if (self != NULL) {

	self->af = new AmAudioFile();
	if(!self->af){
	    Py_DECREF(self);
	    return NULL;
	}

#ifdef IVR_WITH_TTS
	flite_init();
	self->tts_voice = register_cmu_us_kal();
#endif

    }

    return (PyObject *)self;
}

static void IvrAudioFile_dealloc(IvrAudioFile* self)
{
    DBG("---------- IvrAudioFile_dealloc -----------\n");
    delete self->af;
    self->af = NULL;
    self->ob_type->tp_free((PyObject*)self);
}

static PyObject* IvrAudioFile_open(IvrAudioFile* self, PyObject* args)
{
    int ivr_open_mode;
    AmAudioFile::OpenMode open_mode;
    char* filename;

    if(!PyArg_ParseTuple(args,"si",&filename,&ivr_open_mode))
	return NULL;

    switch(ivr_open_mode){
    case AUDIO_READ:
	open_mode = AmAudioFile::Read;
	break;
    case AUDIO_WRITE:
	open_mode = AmAudioFile::Write;
	break;
    default:
	PyErr_SetString(PyExc_TypeError,"Unknown open mode");
	return NULL;
	break;
    }

    if(self->af->open(filename,open_mode)){
	PyErr_SetString(PyExc_IOError,"Could not open file");
	return NULL;
    }

    Py_INCREF(Py_None);
    return Py_None;
}

#ifdef IVR_WITH_TTS
static PyObject* IvrAudioFile_tts(IvrAudioFile* self, PyObject* args)
{
    char* text;
    if(!PyArg_ParseTuple(args,"s",&text))
	return NULL;
    
    string filename = string(TTS_CACHE_PATH) + AmSession::getNewId() + string(".wav");
    flite_text_to_speech(text,self->tts_voice,filename.c_str());
    
    if(self->af->open(filename.c_str(),AmAudioFile::Read)){
	PyErr_SetString(PyExc_IOError,"Could not open file");
	return NULL;
    }

    Py_INCREF(Py_None);
    return Py_None;
}
#endif
    
static PyObject* IvrAudioFile_close(IvrAudioFile* self, PyObject*)
{
    self->af->close();
    Py_INCREF(Py_None);
    return Py_None;
}


static PyMethodDef IvrAudioFile_methods[] = {
    {"open", (PyCFunction)IvrAudioFile_open, METH_VARARGS,
     "open the audio file"
    },
    {"close", (PyCFunction)IvrAudioFile_close, METH_NOARGS,
     "close the audio file"
    },
#ifdef IVR_WITH_TTS
    {"tts", (PyCFunction)IvrAudioFile_tts, METH_VARARGS,
     "text to speech"
    },
#endif
    {NULL}  /* Sentinel */
};


static PyObject* IvrAudioFile_getloop(IvrAudioFile* self, void*)
{
    PyObject* loop = self->af->loop.get() ? Py_True : Py_False;
    Py_INCREF(loop);
    return loop;
}

static int IvrAudioFile_setloop(IvrAudioFile* self, PyObject* value, void*)
{
    if (value == NULL) {
	PyErr_SetString(PyExc_TypeError, "Cannot delete the first attribute");
	return -1;
    }
  
    if(value == Py_True)
	self->af->loop.set(true);

    else if(value == Py_False)
	self->af->loop.set(false);

    else {
	PyErr_SetString(PyExc_TypeError, 
			"The first attribute value must be a boolean");
	return -1;
    }

    return 0;
}

static PyGetSetDef IvrAudioFile_getseters[] = {
    {"loop", 
     (getter)IvrAudioFile_getloop, (setter)IvrAudioFile_setloop,
     "repeat mode",
     NULL},
    {NULL}  /* Sentinel */
};
    
PyTypeObject IvrAudioFileType = {
	
    PyObject_HEAD_INIT(NULL)
    0,                         /*ob_size*/
    "ivr.IvrAudioFile",        /*tp_name*/
    sizeof(IvrAudioFile),      /*tp_basicsize*/
    0,                         /*tp_itemsize*/
    (destructor)IvrAudioFile_dealloc, /*tp_dealloc*/
    0,                         /*tp_print*/
    0,                         /*tp_getattr*/
    0,                         /*tp_setattr*/
    0,                         /*tp_compare*/
    0,                         /*tp_repr*/
    0,                         /*tp_as_number*/
    0,                         /*tp_as_sequence*/
    0,                         /*tp_as_mapping*/
    0,                         /*tp_hash */
    0,                         /*tp_call*/
    0,                         /*tp_str*/
    0,                         /*tp_getattro*/
    0,                         /*tp_setattro*/
    0,                         /*tp_as_buffer*/
    Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /*tp_flags*/
    "An audio file",           /*tp_doc*/
    0,		               /* tp_traverse */
    0,		               /* tp_clear */
    0,		               /* tp_richcompare */
    0,		               /* tp_weaklistoffset */
    0,		               /* tp_iter */
    0,		               /* tp_iternext */
    IvrAudioFile_methods,      /* tp_methods */
    0,                         /* tp_members */
    IvrAudioFile_getseters,    /* tp_getset */
    0,                         /* tp_base */
    0,                         /* tp_dict */
    0,                         /* tp_descr_get */
    0,                         /* tp_descr_set */
    0,                         /* tp_dictoffset */
    0,                         /* tp_init */
    0,                         /* tp_alloc */
    IvrAudioFile_new,          /* tp_new */
};