“Text with timestamps” output format option

1 files changed, 26 insertions, 9 deletions

diff --git a/Examples/WhisperDesktop/TranscribeDlg.cpp b/Examples/WhisperDesktop/TranscribeDlg.cpp
index 4f4944d..742c1d4 100644
--- a/Examples/WhisperDesktop/TranscribeDlg.cpp
+++ b/Examples/WhisperDesktop/TranscribeDlg.cpp
@@ -108,7 +108,8 @@ void TranscribeDlg::printModelDescription()
 void TranscribeDlg::populateOutputFormats()
 {
 	transcribeOutFormat.AddString( L"None" );
-	transcribeOutFormat.AddString( L"Text File" );
+	transcribeOutFormat.AddString( L"Text file" );
+	transcribeOutFormat.AddString( L"Text with timestamps" );
 	transcribeOutFormat.AddString( L"SubRip subtitles" );
 	transcribeOutFormat.AddString( L"WebVTT subtitles" );
 }
@@ -117,8 +118,9 @@ enum struct TranscribeDlg::eOutputFormat : uint8_t
 {
 	None = 0,
 	Text = 1,
-	SubRip = 2,
-	WebVTT = 3
+	TextTimestamps = 2,
+	SubRip = 3,
+	WebVTT = 4,
 };
 
 // CBN_SELCHANGE notification for IDC_OUTPUT_FORMAT combobox
@@ -163,10 +165,10 @@ void TranscribeDlg::onBrowseMedia()
 		setOutputPath( path );
 }
 
-static const LPCTSTR outputFilters = L"Text files (*.txt)\0*.txt\0SubRip subtitles (*.srt)\0*.srt\0WebVTT subtitles (*.vtt)\0*.vtt\0\0";
-static const std::array<LPCTSTR, 3> outputExtensions =
+static const LPCTSTR outputFilters = L"Text files (*.txt)\0*.txt\0Text with timestamps (*.txt)\0*.txt\0SubRip subtitles (*.srt)\0*.srt\0WebVTT subtitles (*.vtt)\0*.vtt\0\0";
+static const std::array<LPCTSTR, 4> outputExtensions =
 {
-	L".txt", L".srt", L".vtt"
+	L".txt", L".txt", L".srt", L".vtt"
 };
 
 void TranscribeDlg::setOutputPath( const CString& input )
@@ -414,7 +416,9 @@ HRESULT TranscribeDlg::transcribe()
 	switch( format )
 	{
 	case eOutputFormat::Text:
-		return writeTextFile( segments, len.countSegments, outputFile );
+		return writeTextFile( segments, len.countSegments, outputFile, false );
+	case eOutputFormat::TextTimestamps:
+		return writeTextFile( segments, len.countSegments, outputFile, true );
 	case eOutputFormat::SubRip:
 		return writeSubRip( segments, len.countSegments, outputFile );
 	case eOutputFormat::WebVTT:
@@ -479,14 +483,27 @@ namespace
 using Whisper::sSegment;
 
 
-HRESULT TranscribeDlg::writeTextFile( const sSegment* const segments, const size_t length, CAtlFile& file )
+HRESULT TranscribeDlg::writeTextFile( const sSegment* const segments, const size_t length, CAtlFile& file, bool timestamps )
 {
 	using namespace Whisper;
 	CHECK( writeUtf8Bom( file ) );
 	CStringA line;
 	for( size_t i = 0; i < length; i++ )
 	{
-		line = skipBlank( segments[ i ].text );
+		const sSegment& seg = segments[ i ];
+
+		if( timestamps )
+		{
+			line = "[";
+			printTimeStamp( line, seg.time.begin );
+			line += " --> ";
+			printTimeStamp( line, seg.time.end );
+			line += "]  ";
+		}
+		else
+			line = "";
+
+		line += skipBlank( seg.text );
 		line += "\r\n";
 		CHECK( write( file, line ) );
 	}