#include "pyconfig.h" // Py_GIL_DISABLED

#ifndef Py_GIL_DISABLED

# define Py_LIMITED_API 0x030d0000

#include <Python.h>

#include <windows.h>

#include <mmsystem.h>

PyDoc_STRVAR(sound_module_doc,

"SND_NOSTOP - Do not interrupt any sounds currently playing\n" // Raising RuntimeError if needed

"SND_NOWAIT - Return immediately if the sound driver is busy\n" // Without any errors

"MessageBeep(type) - Call Windows MessageBeep.");

#include "clinic/winsound.c.h"

static PyObject *

winsound_PlaySound_impl(PyObject *module, PyObject *sound, int flags)

{

int ok;

wchar_t *wsound;

Py_buffer view = {NULL, NULL};

if (sound == Py_None) {

wsound = NULL;

} else if (flags & SND_MEMORY) {

if (flags & SND_ASYNC) {

/* Sidestep reference counting headache; unfortunately this also

PyErr_SetString(PyExc_RuntimeError,

"Cannot play asynchronously from memory");

return NULL;

}

if (PyObject_GetBuffer(sound, &view, PyBUF_SIMPLE) < 0) {

return NULL;

}

wsound = (wchar_t *)view.buf;

} else if (PyBytes_Check(sound)) {

PyObject *type_name = PyType_GetQualName(Py_TYPE(sound));

if (type_name != NULL) {

PyErr_Format(PyExc_TypeError,

"'sound' must be str, os.PathLike, or None, not %S",

type_name);

Py_DECREF(type_name);

}

return NULL;

} else {

PyObject *obj = PyOS_FSPath(sound);

// Either <obj> is unicode/bytes/NULL, or a helpful message

// has been surfaced to the user about how they gave a non-path.

if (obj == NULL) return NULL;

if (PyBytes_Check(obj)) {

PyErr_Format(PyExc_TypeError,

"'sound' must resolve to str, not bytes");

Py_DECREF(obj);

return NULL;

}

wsound = PyUnicode_AsWideCharString(obj, NULL);

Py_DECREF(obj);

if (wsound == NULL) return NULL;

}

Py_BEGIN_ALLOW_THREADS

ok = PlaySoundW(wsound, NULL, flags);

Py_END_ALLOW_THREADS

if (view.obj) {

PyBuffer_Release(&view);

} else if (sound != Py_None) {

PyMem_Free(wsound);

}

if (!ok) {

PyErr_SetString(PyExc_RuntimeError, "Failed to play sound");

return NULL;

}

Py_RETURN_NONE;

}

static PyObject *

winsound_Beep_impl(PyObject *module, int frequency, int duration)

{

BOOL ok;

if (frequency < 37 || frequency > 32767) {

PyErr_SetString(PyExc_ValueError,

"frequency must be in 37 thru 32767");

return NULL;

}

Py_BEGIN_ALLOW_THREADS

ok = Beep(frequency, duration);

Py_END_ALLOW_THREADS

if (!ok) {

PyErr_SetString(PyExc_RuntimeError,"Failed to beep");

return NULL;

}

Py_RETURN_NONE;

}

static PyObject *

winsound_MessageBeep_impl(PyObject *module, int type)

{

BOOL ok;

Py_BEGIN_ALLOW_THREADS

ok = MessageBeep(type);

Py_END_ALLOW_THREADS

if (!ok) {

PyErr_SetExcFromWindowsErr(PyExc_RuntimeError, 0);

return NULL;

}

Py_RETURN_NONE;

}

static struct PyMethodDef sound_methods[] =

{

WINSOUND_PLAYSOUND_METHODDEF

WINSOUND_BEEP_METHODDEF

WINSOUND_MESSAGEBEEP_METHODDEF

{NULL, NULL}

};

#define ADD_DEFINE(CONST) do { \

if (PyModule_AddIntConstant(module, #CONST, CONST) < 0) { \

return -1; \

} \

} while (0)

static int

exec_module(PyObject *module)

{

ADD_DEFINE(SND_ASYNC);

ADD_DEFINE(SND_NODEFAULT);

ADD_DEFINE(SND_NOSTOP);

ADD_DEFINE(SND_NOWAIT);

ADD_DEFINE(SND_ALIAS);

ADD_DEFINE(SND_FILENAME);

ADD_DEFINE(SND_MEMORY);

ADD_DEFINE(SND_PURGE);

ADD_DEFINE(SND_LOOP);

ADD_DEFINE(SND_APPLICATION);

ADD_DEFINE(MB_OK);

ADD_DEFINE(MB_ICONASTERISK);

ADD_DEFINE(MB_ICONEXCLAMATION);

ADD_DEFINE(MB_ICONHAND);

ADD_DEFINE(MB_ICONQUESTION);

#undef ADD_DEFINE

return 0;

}

static PyModuleDef_Slot sound_slots[] = {

{Py_mod_exec, exec_module},

{Py_mod_multiple_interpreters, Py_MOD_PER_INTERPRETER_GIL_SUPPORTED},

{Py_mod_gil, Py_MOD_GIL_NOT_USED},

{0, NULL}

};

static struct PyModuleDef winsoundmodule = {

.m_base = PyModuleDef_HEAD_INIT,

.m_name = "winsound",

.m_doc = sound_module_doc,

.m_methods = sound_methods,

.m_slots = sound_slots,

};

PyInit_winsound(void)

{

return PyModuleDef_Init(&winsoundmodule);

}