95 lines
2.5 KiB
C
95 lines
2.5 KiB
C
#include "ruby_whisper.h"
|
|
|
|
#define ITERATE_ATTRS(ITERATOR) \
|
|
ITERATOR(start_time, t0, TIME) \
|
|
ITERATOR(end_time, t1, TIME) \
|
|
ITERATOR(text, text, STRING)
|
|
|
|
#define VAL_FROM_TIME(v) (LONG2NUM((v) * 10))
|
|
#define VAL_FROM_STRING(v) (rb_str_new2(v))
|
|
#define READER(type) VAL_FROM_##type
|
|
#define DEF_ATTR(rb_name, c_name, type) \
|
|
static VALUE \
|
|
ruby_whisper_parakeet_get_##rb_name(VALUE self) \
|
|
{ \
|
|
ruby_whisper_parakeet_segment *rwps; \
|
|
GetParakeetSegment(self, rwps); \
|
|
ruby_whisper_parakeet_context *rwpc; \
|
|
GetParakeetContext(rwps->context, rwpc); \
|
|
return READER(type)(parakeet_full_get_segment_##c_name(rwpc->context, rwps->index)); \
|
|
}
|
|
|
|
extern VALUE cParakeetSegment;
|
|
extern const rb_data_type_t ruby_whisper_parakeet_context_type;
|
|
|
|
static void
|
|
rb_whisper_parakeet_segment_mark(void *p)
|
|
{
|
|
ruby_whisper_parakeet_segment *rwps = (ruby_whisper_parakeet_segment *)p;
|
|
rb_gc_mark(rwps->context);
|
|
}
|
|
|
|
static size_t
|
|
ruby_whisper_parakeet_segment_memsize(const void *p)
|
|
{
|
|
const ruby_whisper_parakeet_segment *rwps = (const ruby_whisper_parakeet_segment *)p;
|
|
if (!rwps) {
|
|
return 0;
|
|
}
|
|
size_t size = sizeof(*rwps);
|
|
if (rwps->index) {
|
|
size += sizeof(rwps->index);
|
|
}
|
|
return size;
|
|
}
|
|
|
|
static const rb_data_type_t ruby_whisper_parakeet_segment_type = {
|
|
"ruby_whisper_segment",
|
|
{rb_whisper_parakeet_segment_mark, RUBY_DEFAULT_FREE, ruby_whisper_parakeet_segment_memsize,},
|
|
0, 0,
|
|
0
|
|
};
|
|
|
|
static VALUE
|
|
ruby_whisper_parakeet_segment_s_allocate(VALUE klass)
|
|
{
|
|
ruby_whisper_parakeet_segment *rwps;
|
|
return TypedData_Make_Struct(klass, ruby_whisper_parakeet_segment, &ruby_whisper_parakeet_segment_type, rwps);
|
|
}
|
|
|
|
VALUE
|
|
ruby_whisper_parakeet_segment_init(VALUE context, int index)
|
|
{
|
|
ruby_whisper_parakeet_segment *rwps;
|
|
|
|
const VALUE segment = ruby_whisper_parakeet_segment_s_allocate(cParakeetSegment);
|
|
TypedData_Get_Struct(segment, ruby_whisper_parakeet_segment, &ruby_whisper_parakeet_segment_type, rwps);
|
|
rwps->context = context;
|
|
rwps->index = index;
|
|
|
|
return segment;
|
|
}
|
|
|
|
ITERATE_ATTRS(DEF_ATTR)
|
|
|
|
void
|
|
init_ruby_whisper_parakeet_segment(VALUE *mParakeet)
|
|
{
|
|
cParakeetSegment = rb_define_class_under(*mParakeet, "Segment", rb_cObject);
|
|
|
|
rb_define_alloc_func(cParakeetSegment, ruby_whisper_parakeet_segment_s_allocate);
|
|
|
|
#define REGISTER_ATTR(rb_name, c_name, type) \
|
|
rb_define_method(cParakeetSegment, #rb_name, ruby_whisper_parakeet_get_##rb_name, 0);
|
|
|
|
ITERATE_ATTRS(REGISTER_ATTR)
|
|
|
|
#undef REGISTER_ATTR
|
|
}
|
|
|
|
#undef DEF_ATTR
|
|
#undef READER
|
|
#undef VAL_FROM_STRING
|
|
#undef VAL_FROM_TIME
|
|
#undef ITERATE_ATTRS
|