Skip to content

Commit 5e350b1

Browse files
committed
Add Context#full_get_setgment_text
1 parent 0672e6f commit 5e350b1

File tree

1 file changed

+9
-0
lines changed

1 file changed

+9
-0
lines changed

bindings/ruby/ext/ruby_whisper.cpp

+9
Original file line numberDiff line numberDiff line change
@@ -315,6 +315,14 @@ static VALUE ruby_whisper_full_get_segment_speaker_turn_next(VALUE self, VALUE i
315315
return speaker_turn_next ? Qtrue : Qfalse;
316316
}
317317

318+
static VALUE ruby_whisper_full_get_segment_text(VALUE self, VALUE i_segment) {
319+
ruby_whisper *rw;
320+
Data_Get_Struct(self, ruby_whisper, rw);
321+
const int c_i_segment = ruby_whisper_full_check_segment_index(rw, i_segment);
322+
const char * text = whisper_full_get_segment_text(rw->context, c_i_segment);
323+
return rb_str_new2(text);
324+
}
325+
318326
/*
319327
* params.language = "auto" | "en", etc...
320328
*/
@@ -483,6 +491,7 @@ void Init_whisper() {
483491
rb_define_method(cContext, "full_get_segment_t0", ruby_whisper_full_get_segment_t0, 1);
484492
rb_define_method(cContext, "full_get_segment_t1", ruby_whisper_full_get_segment_t1, 1);
485493
rb_define_method(cContext, "full_get_segment_speaker_turn_next", ruby_whisper_full_get_segment_speaker_turn_next, 1);
494+
rb_define_method(cContext, "full_get_segment_text", ruby_whisper_full_get_segment_text, 1);
486495

487496
rb_define_alloc_func(cParams, ruby_whisper_params_allocate);
488497

0 commit comments

Comments
 (0)