Consistent cancellation API across the library: S_OK = continue, S_FALSE = stop

5 files changed, 13 insertions, 3 deletions

diff --git a/Whisper/API/MfStructs.h b/Whisper/API/MfStructs.h
index cd27659..39255de 100644
--- a/Whisper/API/MfStructs.h
+++ b/Whisper/API/MfStructs.h
@@ -40,8 +40,11 @@ namespace Whisper
 		Stalled = 0x80,
 	};
 
+	// Return S_OK to continue, or S_FALSE to stop the capture session
 	using pfnShouldCancel = HRESULT( __stdcall* )( void* pv ) noexcept;
+
 	using pfnCaptureStatus = HRESULT( __stdcall* )( void* pv, eCaptureStatus status ) noexcept;
+
 	struct sCaptureCallbacks
 	{
 		pfnShouldCancel shouldCancel;
diff --git a/Whisper/API/sFullParams.h b/Whisper/API/sFullParams.h
index 0a1d352..42d48a4 100644
--- a/Whisper/API/sFullParams.h
+++ b/Whisper/API/sFullParams.h
@@ -14,6 +14,8 @@ namespace Whisper
 	};
 
 	using pfnNewSegment = HRESULT( __cdecl* )( iContext* ctx, uint32_t n_new, void* user_data ) noexcept;
+
+	// Return S_OK to proceed, or S_FALSE to stop the process and return S_OK from runFull / runStreamed method
 	using pfnEncoderBegin = HRESULT( __cdecl* )( iContext* ctx, void* user_data ) noexcept;
 
 	enum struct eFullParamsFlags : uint32_t
diff --git a/Whisper/API/sLoadModelCallbacks.h b/Whisper/API/sLoadModelCallbacks.h
index f5248c6..8f274c6 100644
--- a/Whisper/API/sLoadModelCallbacks.h
+++ b/Whisper/API/sLoadModelCallbacks.h
@@ -3,6 +3,7 @@
 namespace Whisper
 {
 	using pfnLoadProgress = HRESULT( __stdcall* )( double val, void* pv ) noexcept;
+	// Return S_OK to continue, or S_FALSE to fail with "The operation was canceled by the user" status code
 	using pfnCancel = HRESULT( __stdcall* )( void* pv ) noexcept;
 	
 	struct sLoadModelCallbacks
diff --git a/Whisper/Whisper/ContextImpl.capture.cpp b/Whisper/Whisper/ContextImpl.capture.cpp
index 617c1f3..0062c2a 100644
--- a/Whisper/Whisper/ContextImpl.capture.cpp
+++ b/Whisper/Whisper/ContextImpl.capture.cpp
@@ -171,7 +171,7 @@ namespace
 		HRESULT checkCancel() noexcept
 		{
 			if( nullptr == callbacks.shouldCancel )
-				return S_FALSE;
+				return S_OK;
 			return callbacks.shouldCancel( callbacks.pv );
 		}
 
@@ -415,7 +415,7 @@ HRESULT COMLIGHTCALL ContextImpl::runCapture( const sFullParams& params, const s
 	{
 		HRESULT hr = capture.checkCancel();
 		CHECK( hr );
-		if( hr == S_OK )
+		if( hr != S_OK )
 			return S_OK;
 		CHECK( capture.run() );
 	}
diff --git a/Whisper/Whisper/ContextImpl.cpp b/Whisper/Whisper/ContextImpl.cpp
index 175af74..59d644b 100644
--- a/Whisper/Whisper/ContextImpl.cpp
+++ b/Whisper/Whisper/ContextImpl.cpp
@@ -257,6 +257,7 @@ HRESULT COMLIGHTCALL ContextImpl::runFullImpl( const sFullParams& params, const
 	int seek = seek_start;
 	// Start measuring "Run" profiler value, both CPU and GPU times
 	auto prof = context.completeProfiler();
+	bool stoppedPrematurely = false;
 	while( true )
 	{
 		if( nullptr != progress.pfn )
@@ -278,7 +279,10 @@ HRESULT COMLIGHTCALL ContextImpl::runFullImpl( const sFullParams& params, const
 			if( FAILED( hr ) )
 				return hr;
 			if( hr != S_OK )
+			{
+				stoppedPrematurely = true;
 				break;
+			}
 		}
 
 		// encode audio features starting at offset seek
@@ -511,7 +515,7 @@ HRESULT COMLIGHTCALL ContextImpl::runFullImpl( const sFullParams& params, const
 		seek += seek_delta;
 	}
 
-	if( nullptr != progress.pfn )
+	if( nullptr != progress.pfn && !stoppedPrematurely )
 	{
 		auto cb = profiler.cpuBlock( eCpuBlock::Callbacks );
 		CHECK( progress.pfn( 1.0, this, progress.pv ) );