diarization_c_api.h

/* Copyright 2022 ID R&D Inc. All Rights Reserved. */
 
#pragma once
 
#include <voicesdk/core/core_c_api.h>
#include <voicesdk/core/config.h>
 
#ifdef __cplusplus
extern "C" {
#endif
 
DECL_TYPED_ARRAY(VoiceSdkAudioIntervalArray, VoiceSdkAudioInterval)
 
typedef struct VoiceSdkTimestamp {
    size_t speaker_id;
    VoiceSdkAudioIntervalArray audio_intervals;
} VoiceSdkTimestamp;
 
typedef struct VoiceSdkTimestampArray {
    size_t size;
    VoiceSdkTimestamp* data;
} VoiceSdkTimestampArray;
 
VOICE_SDK_API
void VoiceSdkTimestampArrayRelease(VoiceSdkTimestampArray* array);
 
typedef struct VoiceSdkDiarizationEngine VoiceSdkDiarizationEngine;
 
VOICE_SDK_API
VoiceSdkDiarizationEngine* VoiceSdkDiarizationEngineCreate(const char* init_path, char** error_msg);
 
VOICE_SDK_API
void VoiceSdkDiarizationEngineRelease(VoiceSdkDiarizationEngine* engine);
 
VOICE_SDK_API
VoiceSdkTimestampArray* VoiceSdkDiarizationEngineGetSegmentationWavFile(const VoiceSdkDiarizationEngine* engine,
                                                                        const char* wav_path, size_t num_speakers,
                                                                        char** error_msg);
 
VOICE_SDK_API
VoiceSdkTimestampArray* VoiceSdkDiarizationEngineGetSegmentationAudioFile(const VoiceSdkDiarizationEngine* engine,
                                                                          const char* audio_path, size_t num_speakers,
                                                                          char** error_msg);
 
VOICE_SDK_API
VoiceSdkTimestampArray* VoiceSdkDiarizationEngineGetSegmentationFloatSamples(const VoiceSdkDiarizationEngine* engine,
                                                                             const float* float_samples,
                                                                             size_t num_samples, size_t sample_rate,
                                                                             size_t num_speakers, char** error_msg);
 
VOICE_SDK_API
VoiceSdkTimestampArray* VoiceSdkDiarizationEngineGetSegmentationPcm16Samples(const VoiceSdkDiarizationEngine* engine,
                                                                             const int16_t* samples, size_t num_samples,
                                                                             size_t sample_rate, size_t num_speakers,
                                                                             char** error_msg);
 
VOICE_SDK_API
VoiceSdkTimestampArray* VoiceSdkDiarizationEngineGetSegmentationPcm16Bytes(const VoiceSdkDiarizationEngine* engine,
                                                                           const uint8_t* bytes, size_t num_bytes,
                                                                           size_t sample_rate, size_t num_speakers,
                                                                           char** error_msg);
 
#ifdef __cplusplus
}
#endif