mirror of
https://github.com/ggml-org/whisper.cpp.git
synced 2025-09-15 13:28:35 +08:00
7ffcd05267
* Fix type signature for Whisper.log_set
* Use cache file for model when offline
* Extract ruby_whisper_transcribe() into a file
* Extract Whisper::Error
* Use FileList for ext/*.{c,cpp,h}
* Extract Whisper::Segment
* Extract Whisper::Model
* Extract Whisper::Params
* Extract Whisper::Context
* Extract log_callback function
* Write base code in C rather than C++
* Use chdir instead of Dir.chdir in Rakefile
* Define alloc func for Whisper::Model
* Define Whisper::Params' calback and user data reader
* Add test for Whisper::Params.new with keyword arguments
* Make Whisper::Params.new accept keyword arguments
* Update type signatures
* Update README
* Update CLEAN targets
* Fix document comment for Whisper::Params#new_segment_callback=
* Use macro to define params
* Fix dependency of build task
* Set Whisper.finalize_log_callback visibility to private
* Make Whisper::Context#full and full_parallel return self
* Add test for Whisper::Context#full_get_segment
* Add Whisper::Context#full_get_segment
* Update signatures
* Update README
* Fix signature
* Resplace #initialize with .new in signature file [skip ci]
* Fix potential overflow
124 lines
3.3 KiB
C
124 lines
3.3 KiB
C
#include <ruby.h>
|
|
#include "ruby_whisper.h"
|
|
|
|
extern VALUE cSegment;
|
|
|
|
static void
|
|
rb_whisper_segment_mark(ruby_whisper_segment *rws)
|
|
{
|
|
rb_gc_mark(rws->context);
|
|
}
|
|
|
|
VALUE
|
|
ruby_whisper_segment_allocate(VALUE klass)
|
|
{
|
|
ruby_whisper_segment *rws;
|
|
rws = ALLOC(ruby_whisper_segment);
|
|
return Data_Wrap_Struct(klass, rb_whisper_segment_mark, RUBY_DEFAULT_FREE, rws);
|
|
}
|
|
|
|
VALUE
|
|
rb_whisper_segment_initialize(VALUE context, int index)
|
|
{
|
|
ruby_whisper_segment *rws;
|
|
const VALUE segment = ruby_whisper_segment_allocate(cSegment);
|
|
Data_Get_Struct(segment, ruby_whisper_segment, rws);
|
|
rws->context = context;
|
|
rws->index = index;
|
|
return segment;
|
|
};
|
|
|
|
/*
|
|
* Start time in milliseconds.
|
|
*
|
|
* call-seq:
|
|
* start_time -> Integer
|
|
*/
|
|
static VALUE
|
|
ruby_whisper_segment_get_start_time(VALUE self)
|
|
{
|
|
ruby_whisper_segment *rws;
|
|
Data_Get_Struct(self, ruby_whisper_segment, rws);
|
|
ruby_whisper *rw;
|
|
Data_Get_Struct(rws->context, ruby_whisper, rw);
|
|
const int64_t t0 = whisper_full_get_segment_t0(rw->context, rws->index);
|
|
// able to multiply 10 without overflow because to_timestamp() in whisper.cpp does it
|
|
return INT2NUM(t0 * 10);
|
|
}
|
|
|
|
/*
|
|
* End time in milliseconds.
|
|
*
|
|
* call-seq:
|
|
* end_time -> Integer
|
|
*/
|
|
static VALUE
|
|
ruby_whisper_segment_get_end_time(VALUE self)
|
|
{
|
|
ruby_whisper_segment *rws;
|
|
Data_Get_Struct(self, ruby_whisper_segment, rws);
|
|
ruby_whisper *rw;
|
|
Data_Get_Struct(rws->context, ruby_whisper, rw);
|
|
const int64_t t1 = whisper_full_get_segment_t1(rw->context, rws->index);
|
|
// able to multiply 10 without overflow because to_timestamp() in whisper.cpp does it
|
|
return INT2NUM(t1 * 10);
|
|
}
|
|
|
|
/*
|
|
* Whether the next segment is predicted as a speaker turn.
|
|
*
|
|
* call-seq:
|
|
* speaker_turn_next? -> bool
|
|
*/
|
|
static VALUE
|
|
ruby_whisper_segment_get_speaker_turn_next(VALUE self)
|
|
{
|
|
ruby_whisper_segment *rws;
|
|
Data_Get_Struct(self, ruby_whisper_segment, rws);
|
|
ruby_whisper *rw;
|
|
Data_Get_Struct(rws->context, ruby_whisper, rw);
|
|
return whisper_full_get_segment_speaker_turn_next(rw->context, rws->index) ? Qtrue : Qfalse;
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* text -> String
|
|
*/
|
|
static VALUE
|
|
ruby_whisper_segment_get_text(VALUE self)
|
|
{
|
|
ruby_whisper_segment *rws;
|
|
Data_Get_Struct(self, ruby_whisper_segment, rws);
|
|
ruby_whisper *rw;
|
|
Data_Get_Struct(rws->context, ruby_whisper, rw);
|
|
const char * text = whisper_full_get_segment_text(rw->context, rws->index);
|
|
return rb_str_new2(text);
|
|
}
|
|
|
|
/*
|
|
* call-seq:
|
|
* no_speech_prob -> Float
|
|
*/
|
|
static VALUE
|
|
ruby_whisper_segment_get_no_speech_prob(VALUE self)
|
|
{
|
|
ruby_whisper_segment *rws;
|
|
Data_Get_Struct(self, ruby_whisper_segment, rws);
|
|
ruby_whisper *rw;
|
|
Data_Get_Struct(rws->context, ruby_whisper, rw);
|
|
return DBL2NUM(whisper_full_get_segment_no_speech_prob(rw->context, rws->index));
|
|
}
|
|
|
|
void
|
|
init_ruby_whisper_segment(VALUE *mWhisper, VALUE *cContext)
|
|
{
|
|
cSegment = rb_define_class_under(*mWhisper, "Segment", rb_cObject);
|
|
|
|
rb_define_alloc_func(cSegment, ruby_whisper_segment_allocate);
|
|
rb_define_method(cSegment, "start_time", ruby_whisper_segment_get_start_time, 0);
|
|
rb_define_method(cSegment, "end_time", ruby_whisper_segment_get_end_time, 0);
|
|
rb_define_method(cSegment, "speaker_next_turn?", ruby_whisper_segment_get_speaker_turn_next, 0);
|
|
rb_define_method(cSegment, "text", ruby_whisper_segment_get_text, 0);
|
|
rb_define_method(cSegment, "no_speech_prob", ruby_whisper_segment_get_no_speech_prob, 0);
|
|
}
|