Skip to content

Commit 466d5b5

Browse files
committed
vad : remove vad_mapping_table_initialized flag
1 parent 47a5dc3 commit 466d5b5

File tree

1 file changed

+3
-7
lines changed

1 file changed

+3
-7
lines changed

src/whisper.cpp

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -971,7 +971,6 @@ struct whisper_state {
971971
bool has_vad_segments = false;
972972

973973
std::vector<vad_time_mapping> vad_mapping_table;
974-
bool vad_mapping_table_initialized = false;
975974
};
976975

977976
struct whisper_context {
@@ -6638,7 +6637,7 @@ static bool whisper_vad(
66386637

66396638
// Clear any existing mapping table
66406639
state->vad_mapping_table.clear();
6641-
state->vad_mapping_table_initialized = false;
6640+
state->has_vad_segments = false;
66426641

66436642
if (state->vad_context == nullptr) {
66446643
struct whisper_vad_context_params vad_ctx_params = whisper_vad_default_context_params();
@@ -6663,7 +6662,6 @@ static bool whisper_vad(
66636662
// Initialize the time mapping table
66646663
state->vad_mapping_table.clear();
66656664
state->vad_mapping_table.reserve(vad_segments->data.size() * 4);
6666-
state->vad_mapping_table_initialized = true;
66676665

66686666
WHISPER_LOG_INFO("%s: detected %d speech segments\n", __func__, (int)vad_segments->data.size());
66696667
float overlap_seconds = vad_params.samples_overlap;
@@ -7937,8 +7935,7 @@ static int64_t map_processed_to_original_time(int64_t processed_time, const std:
79377935
// Function to get the starting timestamp of a segment
79387936
int64_t whisper_full_get_segment_t0_from_state(struct whisper_state* state, int i_segment) {
79397937
// If VAD wasn't used, return the original timestamp
7940-
if (!state->has_vad_segments || !state->vad_mapping_table_initialized ||
7941-
state->vad_mapping_table.empty()) {
7938+
if (!state->has_vad_segments || state->vad_mapping_table.empty()) {
79427939
return state->result_all[i_segment].t0;
79437940
}
79447941

@@ -7952,8 +7949,7 @@ int64_t whisper_full_get_segment_t0_from_state(struct whisper_state* state, int
79527949
// Function to get the ending timestamp of a segment
79537950
int64_t whisper_full_get_segment_t1_from_state(struct whisper_state* state, int i_segment) {
79547951
// If VAD wasn't used, return the original timestamp
7955-
if (!state->has_vad_segments || !state->vad_mapping_table_initialized ||
7956-
state->vad_mapping_table.empty()) {
7952+
if (!state->has_vad_segments || state->vad_mapping_table.empty()) {
79577953
return state->result_all[i_segment].t1;
79587954
}
79597955

0 commit comments

Comments
 (0)