mirror of
https://github.com/ggerganov/whisper.cpp
synced 2026-04-18 21:05:40 +02:00
d9b7613b34
* Add Whisper::VAD::Context * Add test for Whisper::VAD::Context * Add Whisper::VAD::Segment * Add Whisper::VAD::Segments * Add Whisper::VAD::Context#detect * Define Whisper::VAD::Segments#each * Define Whisper::VAD::Segment#start_time and #end_time * Define Whisper::VAD::Segment#deconstruct_keys * Add tests for Whisper::VAD family * Add signatures for VAD family * Add document on VAD in README * Define Whisper::VAD::Segments#length * Add test for Whisper::VAD::Segments#length * Add signature of Segments#length * Make vad_segments responsible to initialize VAD::Segments * Remove meaningless argument check * Check NULL of segments member * Add tests for Whisper::VAD::Segments * Initialize Whisper::VAD::Segment on .allocate * Add tests for Whisper::VAD::Segment * Check NULL of context member * Add test for Whisper::VAD::Context.allocate
76 lines
2.0 KiB
C
76 lines
2.0 KiB
C
#include <ruby.h>
|
|
#include "ruby_whisper.h"
|
|
|
|
extern ID id_to_s;
|
|
|
|
extern VALUE cVADContext;
|
|
|
|
extern VALUE ruby_whisper_vad_detect(VALUE self, VALUE file_path, VALUE params);
|
|
extern VALUE ruby_whisper_normalize_model_path(VALUE model_path);
|
|
|
|
static size_t
|
|
ruby_whisper_vad_context_memsize(const void *p)
|
|
{
|
|
const ruby_whisper_vad_context *rwvc = p;
|
|
size_t size = sizeof(rwvc);
|
|
if (!rwvc) {
|
|
return 0;
|
|
}
|
|
if (rwvc->context) {
|
|
size += sizeof(rwvc->context);
|
|
}
|
|
return size;
|
|
}
|
|
|
|
static void
|
|
ruby_whisper_vad_context_free(void *p)
|
|
{
|
|
ruby_whisper_vad_context *rwvc = (ruby_whisper_vad_context *)p;
|
|
if (rwvc->context) {
|
|
whisper_vad_free(rwvc->context);
|
|
rwvc->context = NULL;
|
|
}
|
|
xfree(rwvc);
|
|
}
|
|
|
|
const rb_data_type_t ruby_whisper_vad_context_type = {
|
|
"ruby_whisper_vad_context",
|
|
{0, ruby_whisper_vad_context_free, ruby_whisper_vad_context_memsize,},
|
|
0, 0,
|
|
0
|
|
};
|
|
|
|
static VALUE
|
|
ruby_whisper_vad_context_s_allocate(VALUE klass)
|
|
{
|
|
ruby_whisper_vad_context *rwvc;
|
|
VALUE obj = TypedData_Make_Struct(klass, ruby_whisper_vad_context, &ruby_whisper_vad_context_type, rwvc);
|
|
rwvc->context = NULL;
|
|
return obj;
|
|
}
|
|
|
|
static VALUE
|
|
ruby_whisper_vad_context_initialize(VALUE self, VALUE model_path)
|
|
{
|
|
ruby_whisper_vad_context *rwvc;
|
|
struct whisper_vad_context *context;
|
|
|
|
model_path = ruby_whisper_normalize_model_path(model_path);
|
|
context = whisper_vad_init_from_file_with_params(StringValueCStr(model_path), whisper_vad_default_context_params());
|
|
if (context == NULL) {
|
|
rb_raise(rb_eRuntimeError, "Failed to initialize whisper VAD context");
|
|
}
|
|
TypedData_Get_Struct(self, ruby_whisper_vad_context, &ruby_whisper_vad_context_type, rwvc);
|
|
rwvc->context = context;
|
|
|
|
return Qnil;
|
|
}
|
|
|
|
void init_ruby_whisper_vad_context(VALUE *mVAD)
|
|
{
|
|
cVADContext = rb_define_class_under(*mVAD, "Context", rb_cObject);
|
|
rb_define_alloc_func(cVADContext, ruby_whisper_vad_context_s_allocate);
|
|
rb_define_method(cVADContext, "initialize", ruby_whisper_vad_context_initialize, 1);
|
|
rb_define_method(cVADContext, "detect", ruby_whisper_vad_detect, 2);
|
|
}
|