mirror of
https://github.com/ggerganov/whisper.cpp
synced 2026-03-08 07:59:36 +01:00
aa1bc0d1a6
* ruby : Bump version to 1.3.6 * Fix code in example * Add sample code to transcribe from MemoryView * Define GetVADContext macro * Use GetVADContext * Extract parse_full_args function * Use parse_full_args in ruby_whisper_full_parallel * Free samples after use * Check return value of parse_full_args() * Define GetVADParams macro * Add VAD::Context#segments_from_samples * Add tests for VAD::Context#segments_from_samples * Add signature for VAD::Context#segments_from_samples * Add sample code for VAD::Context#segments_from_samples * Add test for Whisper::Context#transcribe with Pathname * Make Whisper::Context#transcribe and Whisper::VAD::Context#detect accept Pathname * Update signature of Whisper::Context#transcribe * Fix variable name * Don't free memory view * Make parse_full_args return struct * Fallback when failed to get MemoryView * Add num of samples when too long * Check members of MemoryView * Fix a typo * Remove unnecessary include * Fix a typo * Fix a typo * Care the case of MemoryView doesn't fit spec * Add TODO comment * Add optimazation option to compiler flags * Use ALLOC_N instead of malloc * Add description to sample code * Rename and change args: parse_full_args -> parse_samples * Free samples when exception raised * Assign type check result to a variable * Define wrapper function of whisper_full * Change signature of parse_samples for rb_ensure * Ensure release MemoryView * Extract fill_samples function * Free samples memory when filling it failed * Free samples memory when transcription failed * Prepare transcription in wrapper funciton * Change function name * Simplify function boundary
106 lines
2.6 KiB
C
106 lines
2.6 KiB
C
#include "ruby_whisper.h"
|
|
|
|
extern ID id___method__;
|
|
extern ID id_to_enum;
|
|
|
|
extern VALUE cVADSegments;
|
|
|
|
extern VALUE rb_whisper_vad_segment_s_new(VALUE segments, int index);
|
|
|
|
static size_t
|
|
ruby_whisper_vad_segments_memsize(const void *p)
|
|
{
|
|
const ruby_whisper_vad_segments *rwvss = p;
|
|
size_t size = sizeof(rwvss);
|
|
if (!rwvss) {
|
|
return 0;
|
|
}
|
|
if (rwvss->segments) {
|
|
size += sizeof(rwvss->segments);
|
|
}
|
|
return size;
|
|
}
|
|
|
|
static void
|
|
ruby_whisper_vad_segments_free(void *p)
|
|
{
|
|
ruby_whisper_vad_segments *rwvss = (ruby_whisper_vad_segments *)p;
|
|
if (rwvss->segments) {
|
|
whisper_vad_free_segments(rwvss->segments);
|
|
rwvss->segments = NULL;
|
|
}
|
|
xfree(rwvss);
|
|
}
|
|
|
|
const rb_data_type_t ruby_whisper_vad_segments_type = {
|
|
"ruby_whisper_vad_segments",
|
|
{0, ruby_whisper_vad_segments_free, ruby_whisper_vad_segments_memsize,},
|
|
0, 0,
|
|
0
|
|
};
|
|
|
|
static VALUE
|
|
ruby_whisper_vad_segments_s_allocate(VALUE klass)
|
|
{
|
|
ruby_whisper_vad_segments *rwvss;
|
|
VALUE obj = TypedData_Make_Struct(klass, ruby_whisper_vad_segments, &ruby_whisper_vad_segments_type, rwvss);
|
|
rwvss->segments = NULL;
|
|
return obj;
|
|
}
|
|
|
|
VALUE
|
|
ruby_whisper_vad_segments_s_init(struct whisper_vad_segments *segments)
|
|
{
|
|
VALUE rb_segments;
|
|
ruby_whisper_vad_segments *rwvss;
|
|
|
|
rb_segments = ruby_whisper_vad_segments_s_allocate(cVADSegments);
|
|
TypedData_Get_Struct(rb_segments, ruby_whisper_vad_segments, &ruby_whisper_vad_segments_type, rwvss);
|
|
rwvss->segments = segments;
|
|
|
|
return rb_segments;
|
|
}
|
|
|
|
static VALUE
|
|
ruby_whisper_vad_segments_each(VALUE self)
|
|
{
|
|
ruby_whisper_vad_segments *rwvss;
|
|
VALUE method_name;
|
|
int n_segments, i;
|
|
|
|
if (!rb_block_given_p()) {
|
|
method_name = rb_funcall(self, id___method__, 0);
|
|
return rb_funcall(self, id_to_enum, 1, method_name);
|
|
}
|
|
|
|
GetVADSegments(self, rwvss);
|
|
n_segments = whisper_vad_segments_n_segments(rwvss->segments);
|
|
for (i = 0; i < n_segments; ++i) {
|
|
rb_yield(rb_whisper_vad_segment_s_new(self, i));
|
|
}
|
|
|
|
return self;
|
|
}
|
|
|
|
static VALUE
|
|
ruby_whisper_vad_segments_get_length(VALUE self)
|
|
{
|
|
ruby_whisper_vad_segments *rwvss;
|
|
int n_segments;
|
|
|
|
GetVADSegments(self, rwvss);
|
|
n_segments = whisper_vad_segments_n_segments(rwvss->segments);
|
|
|
|
return INT2NUM(n_segments);
|
|
}
|
|
|
|
void
|
|
init_ruby_whisper_vad_segments(VALUE *mVAD)
|
|
{
|
|
cVADSegments = rb_define_class_under(*mVAD, "Segments", rb_cObject);
|
|
rb_define_alloc_func(cVADSegments, ruby_whisper_vad_segments_s_allocate);
|
|
rb_define_method(cVADSegments, "each", ruby_whisper_vad_segments_each, 0);
|
|
rb_define_method(cVADSegments, "length", ruby_whisper_vad_segments_get_length, 0);
|
|
rb_include_module(cVADSegments, rb_path2class("Enumerable"));
|
|
}
|