diff options
Diffstat (limited to 'Examples/main/textWriter.cpp')
| -rw-r--r-- | Examples/main/textWriter.cpp | 210 |
1 files changed, 210 insertions, 0 deletions
diff --git a/Examples/main/textWriter.cpp b/Examples/main/textWriter.cpp new file mode 100644 index 0000000..6bb5ac4 --- /dev/null +++ b/Examples/main/textWriter.cpp @@ -0,0 +1,210 @@ +#include "textWriter.h" +#include "../../ComLightLib/comLightClient.h" +#include <array> +#define WIN32_LEAN_AND_MEAN +#include <pathcch.h> +#include <atlstr.h> +#include <atlfile.h> +#pragma comment(lib, "Pathcch.lib") + +namespace +{ + HRESULT replaceExtension( CString& path, LPCTSTR inputPath, LPCTSTR ext ) + { + path = inputPath; + + const size_t len = (size_t)path.GetLength() + 4; + wchar_t* buffer = path.GetBufferSetLength( (int)len ); + const HRESULT hr = PathCchRenameExtension( buffer, len, ext ); + path.ReleaseBuffer(); + return hr; + } + + // Abstract base class for text writers + class Writer + { + protected: + CAtlFile file; + virtual HRESULT impl( const Whisper::sSegment* const segments, const size_t length ) = 0; + + public: + HRESULT write( Whisper::iContext* context, LPCTSTR audioPath, LPCTSTR ext ) + { + CString path; + CHECK( replaceExtension( path, audioPath, ext ) ); + CHECK( file.Create( path, GENERIC_WRITE, 0, CREATE_ALWAYS ) ); + + using namespace Whisper; + + const eResultFlags resultFlags = eResultFlags::Timestamps | eResultFlags::Tokens; + ComLight::CComPtr<iTranscribeResult> result; + CHECK( context->getResults( resultFlags, &result ) ); + + sTranscribeLength len; + CHECK( result->getSize( len ) ); + const sSegment* const segments = result->getSegments(); + + return impl( segments, len.countSegments ); + } + }; + + HRESULT writeUtf8Bom( CAtlFile& file ) + { + const std::array<uint8_t, 3> bom = { 0xEF, 0xBB, 0xBF }; + return file.Write( bom.data(), 3 ); + } + + void printTime( CStringA& rdi, Whisper::sTimeSpan time, bool comma ) + { + Whisper::sTimeSpanFields fields = time; + const char separator = comma ? ',' : '.'; + rdi.AppendFormat( "%02d:%02d:%02d%c%03d", + (int)fields.hours, + (int)fields.minutes, + (int)fields.seconds, + separator, + fields.ticks / 10'000 ); + } + + void printTimeStamp( CStringA& rdi, Whisper::sTimeSpan ts ) + { + Whisper::sTimeSpanFields fields = ts; + uint32_t msec = fields.ticks / 10'000; + uint32_t hr = fields.days * 24 + fields.hours; + uint32_t min = fields.minutes; + uint32_t sec = fields.seconds; + rdi.AppendFormat( "%02d:%02d:%02d.%03d", hr, min, sec, msec ); + } + + const char* skipBlank( const char* rsi ) + { + while( true ) + { + const char c = *rsi; + if( c == ' ' || c == '\t' ) + { + rsi++; + continue; + } + return rsi; + } + } + + inline const char* cstr( const CStringA& s ) { return s; } + + HRESULT writeString( CAtlFile& file, const CStringA& line ) + { + if( line.GetLength() > 0 ) + CHECK( file.Write( cstr( line ), (DWORD)line.GetLength() ) ); + return S_OK; + } + + // Writer for UTF-8 text files + class TextWriter : public Writer + { + const bool timestamps; + + HRESULT impl( const Whisper::sSegment* const segments, const size_t length ) override final + { + CHECK( writeUtf8Bom( file ) ); + using namespace Whisper; + + CStringA line; + for( size_t i = 0; i < length; i++ ) + { + 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( writeString( file, line ) ); + } + return S_OK; + } + public: + TextWriter( bool tt ) : timestamps( tt ) { } + }; + + // Writer for SubRip format: https://en.wikipedia.org/wiki/SubRip#SubRip_file_format + class SubRipWriter : public Writer + { + HRESULT impl( const Whisper::sSegment* const segments, const size_t length ) override final + { + CHECK( writeUtf8Bom( file ) ); + using namespace Whisper; + + CStringA line; + for( size_t i = 0; i < length; i++ ) + { + const sSegment& seg = segments[ i ]; + + line.Format( "%zu\r\n", i + 1 ); + printTime( line, seg.time.begin, true ); + line += " --> "; + printTime( line, seg.time.end, true ); + line += "\r\n"; + line += skipBlank( seg.text ); + line += "\r\n\r\n"; + CHECK( writeString( file, line ) ); + } + return S_OK; + } + }; + + // Writer for WebVTT format: https://en.wikipedia.org/wiki/WebVTT + class VttWriter : public Writer + { + HRESULT impl( const Whisper::sSegment* const segments, const size_t length ) override final + { + CHECK( writeUtf8Bom( file ) ); + using namespace Whisper; + + CStringA line; + line = "WEBVTT\r\n\r\n"; + CHECK( writeString( file, line ) ); + + for( size_t i = 0; i < length; i++ ) + { + const sSegment& seg = segments[ i ]; + line = ""; + + printTime( line, seg.time.begin, false ); + line += " --> "; + printTime( line, seg.time.end, false ); + line += "\r\n"; + line += skipBlank( seg.text ); + line += "\r\n\r\n"; + CHECK( writeString( file, line ) ); + } + return S_OK; + } + }; +} + +HRESULT writeText( Whisper::iContext* context, LPCTSTR audioPath, bool timestamps ) +{ + TextWriter writer{ timestamps }; + return writer.write( context, audioPath, L".txt" ); +} + +HRESULT writeSubRip( Whisper::iContext* context, LPCTSTR audioPath ) +{ + SubRipWriter writer; + return writer.write( context, audioPath, L".srt" ); +} + +HRESULT writeWebVTT( Whisper::iContext* context, LPCTSTR audioPath ) +{ + VttWriter writer; + return writer.write( context, audioPath, L".vtt" ); +}
\ No newline at end of file |