206 lines
6.4 KiB
C
206 lines
6.4 KiB
C
/*
|
|
* data-types.c
|
|
* Copyright (C) 2016 Kovid Goyal <kovid at kovidgoyal.net>
|
|
*
|
|
* Distributed under terms of the GPL3 license.
|
|
*/
|
|
|
|
#include "data-types.h"
|
|
#include "modes.h"
|
|
#include <stddef.h>
|
|
#include <termios.h>
|
|
#ifdef WITH_PROFILER
|
|
#include <gperftools/profiler.h>
|
|
#endif
|
|
|
|
/* To millisecond (10^-3) */
|
|
#define SEC_TO_MS 1000
|
|
|
|
/* To microseconds (10^-6) */
|
|
#define MS_TO_US 1000
|
|
#define SEC_TO_US (SEC_TO_MS * MS_TO_US)
|
|
|
|
/* To nanoseconds (10^-9) */
|
|
#define US_TO_NS 1000
|
|
#define MS_TO_NS (MS_TO_US * US_TO_NS)
|
|
#define SEC_TO_NS (SEC_TO_MS * MS_TO_NS)
|
|
|
|
/* Conversion from nanoseconds */
|
|
#define NS_TO_MS (1000 * 1000)
|
|
#define NS_TO_US (1000)
|
|
|
|
#ifdef __APPLE__
|
|
#include <mach/mach_time.h>
|
|
static mach_timebase_info_data_t timebase = {0};
|
|
static inline double monotonic_() {
|
|
return ((double)(mach_absolute_time() * timebase.numer) / timebase.denom)/SEC_TO_NS;
|
|
}
|
|
#else
|
|
#include <time.h>
|
|
static inline double monotonic_() {
|
|
struct timespec ts = {0};
|
|
#ifdef CLOCK_HIGHRES
|
|
clock_gettime(CLOCK_HIGHRES, &ts);
|
|
#elif CLOCK_MONOTONIC_RAW
|
|
clock_gettime(CLOCK_MONOTONIC_RAW, &ts);
|
|
#else
|
|
clock_gettime(CLOCK_MONOTONIC, &ts);
|
|
#endif
|
|
return (((double)ts.tv_nsec) / SEC_TO_NS) + (double)ts.tv_sec;
|
|
}
|
|
#endif
|
|
|
|
double monotonic() { return monotonic_(); }
|
|
|
|
static PyObject*
|
|
wcwidth_wrap(PyObject UNUSED *self, PyObject *chr) {
|
|
return PyLong_FromUnsignedLong(safe_wcwidth(PyLong_AsLong(chr)));
|
|
}
|
|
|
|
static PyObject*
|
|
change_wcwidth_wrap(PyObject UNUSED *self, PyObject *use9) {
|
|
change_wcwidth(PyObject_IsTrue(use9));
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject*
|
|
redirect_std_streams(PyObject UNUSED *self, PyObject *args) {
|
|
char *devnull = NULL;
|
|
if (!PyArg_ParseTuple(args, "s", &devnull)) return NULL;
|
|
if (freopen(devnull, "r", stdin) == NULL) return PyErr_SetFromErrno(PyExc_EnvironmentError);
|
|
if (freopen(devnull, "w", stdout) == NULL) return PyErr_SetFromErrno(PyExc_EnvironmentError);
|
|
if (freopen(devnull, "w", stderr) == NULL) return PyErr_SetFromErrno(PyExc_EnvironmentError);
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject*
|
|
pyset_iutf8(PyObject UNUSED *self, PyObject *args) {
|
|
int fd, on;
|
|
if (!PyArg_ParseTuple(args, "ip", &fd, &on)) return NULL;
|
|
if (!set_iutf8(fd, on & 1)) return PyErr_SetFromErrno(PyExc_OSError);
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
#ifdef WITH_PROFILER
|
|
static PyObject*
|
|
start_profiler(PyObject UNUSED *self, PyObject *args) {
|
|
char *path;
|
|
if (!PyArg_ParseTuple(args, "s", &path)) return NULL;
|
|
ProfilerStart(path);
|
|
Py_RETURN_NONE;
|
|
}
|
|
|
|
static PyObject*
|
|
stop_profiler(PyObject UNUSED *self) {
|
|
ProfilerStop();
|
|
Py_RETURN_NONE;
|
|
}
|
|
#endif
|
|
|
|
static PyMethodDef module_methods[] = {
|
|
{"set_iutf8", (PyCFunction)pyset_iutf8, METH_VARARGS, ""},
|
|
{"thread_write", (PyCFunction)cm_thread_write, METH_VARARGS, ""},
|
|
{"parse_bytes", (PyCFunction)parse_bytes, METH_VARARGS, ""},
|
|
{"parse_bytes_dump", (PyCFunction)parse_bytes_dump, METH_VARARGS, ""},
|
|
{"redirect_std_streams", (PyCFunction)redirect_std_streams, METH_VARARGS, ""},
|
|
{"wcwidth", (PyCFunction)wcwidth_wrap, METH_O, ""},
|
|
{"change_wcwidth", (PyCFunction)change_wcwidth_wrap, METH_O, ""},
|
|
#ifdef WITH_PROFILER
|
|
{"start_profiler", (PyCFunction)start_profiler, METH_VARARGS, ""},
|
|
{"stop_profiler", (PyCFunction)stop_profiler, METH_NOARGS, ""},
|
|
#endif
|
|
{NULL, NULL, 0, NULL} /* Sentinel */
|
|
};
|
|
|
|
|
|
static struct PyModuleDef module = {
|
|
.m_base = PyModuleDef_HEAD_INIT,
|
|
.m_name = "fast_data_types", /* name of module */
|
|
.m_doc = NULL,
|
|
.m_size = -1,
|
|
.m_methods = module_methods
|
|
};
|
|
|
|
|
|
extern int init_LineBuf(PyObject *);
|
|
extern int init_HistoryBuf(PyObject *);
|
|
extern int init_Cursor(PyObject *);
|
|
extern int init_ChildMonitor(PyObject *);
|
|
extern int init_Line(PyObject *);
|
|
extern int init_ColorProfile(PyObject *);
|
|
extern int init_Screen(PyObject *);
|
|
extern int init_Face(PyObject *);
|
|
extern bool init_freetype_library(PyObject*);
|
|
extern bool init_fontconfig_library(PyObject*);
|
|
extern bool init_glfw(PyObject *m);
|
|
extern bool init_sprites(PyObject *module);
|
|
extern bool init_state(PyObject *module);
|
|
extern bool init_keys(PyObject *module);
|
|
extern bool init_shaders(PyObject *module);
|
|
extern bool init_shaders_debug(PyObject *module);
|
|
#ifdef __APPLE__
|
|
extern int init_CoreText(PyObject *);
|
|
extern bool init_cocoa(PyObject *module);
|
|
#endif
|
|
|
|
|
|
EXPORTED PyMODINIT_FUNC
|
|
PyInit_fast_data_types(void) {
|
|
PyObject *m;
|
|
|
|
m = PyModule_Create(&module);
|
|
if (m == NULL) return NULL;
|
|
#ifdef __APPLE__
|
|
mach_timebase_info(&timebase);
|
|
#endif
|
|
|
|
if (m != NULL) {
|
|
if (!init_LineBuf(m)) return NULL;
|
|
if (!init_HistoryBuf(m)) return NULL;
|
|
if (!init_Line(m)) return NULL;
|
|
if (!init_Cursor(m)) return NULL;
|
|
if (!init_ChildMonitor(m)) return NULL;
|
|
if (!init_ColorProfile(m)) return NULL;
|
|
if (!init_Screen(m)) return NULL;
|
|
if (!init_glfw(m)) return NULL;
|
|
if (!init_sprites(m)) return NULL;
|
|
if (!init_state(m)) return NULL;
|
|
if (!init_keys(m)) return NULL;
|
|
if (PySys_GetObject("debug_gl") == Py_True) {
|
|
if (!init_shaders_debug(m)) return NULL;
|
|
} else {
|
|
if (!init_shaders(m)) return NULL;
|
|
}
|
|
#ifdef __APPLE__
|
|
if (!init_CoreText(m)) return NULL;
|
|
if (!init_cocoa(m)) return NULL;
|
|
#else
|
|
if (!init_Face(m)) return NULL;
|
|
if (!init_freetype_library(m)) return NULL;
|
|
if (!init_fontconfig_library(m)) return NULL;
|
|
#endif
|
|
|
|
#define OOF(n) #n, offsetof(Cell, n)
|
|
if (PyModule_AddObject(m, "CELL", Py_BuildValue("{sI sI sI sI sI sI sI sI sI}",
|
|
OOF(ch), OOF(fg), OOF(bg), OOF(decoration_fg), OOF(cc), OOF(sprite_x), OOF(sprite_y), OOF(sprite_z), "size", sizeof(Cell))) != 0) return NULL;
|
|
#undef OOF
|
|
PyModule_AddIntConstant(m, "BOLD", BOLD_SHIFT);
|
|
PyModule_AddIntConstant(m, "ITALIC", ITALIC_SHIFT);
|
|
PyModule_AddIntConstant(m, "REVERSE", REVERSE_SHIFT);
|
|
PyModule_AddIntConstant(m, "STRIKETHROUGH", STRIKE_SHIFT);
|
|
PyModule_AddIntConstant(m, "DECORATION", DECORATION_SHIFT);
|
|
PyModule_AddStringMacro(m, BRACKETED_PASTE_START);
|
|
PyModule_AddStringMacro(m, BRACKETED_PASTE_END);
|
|
PyModule_AddStringMacro(m, ERROR_PREFIX);
|
|
PyModule_AddIntMacro(m, CURSOR_BLOCK);
|
|
PyModule_AddIntMacro(m, CURSOR_BEAM);
|
|
PyModule_AddIntMacro(m, CURSOR_UNDERLINE);
|
|
PyModule_AddIntMacro(m, DECAWM);
|
|
PyModule_AddIntMacro(m, DECCOLM);
|
|
PyModule_AddIntMacro(m, DECOM);
|
|
PyModule_AddIntMacro(m, IRM);
|
|
}
|
|
|
|
return m;
|
|
}
|