CVideoHandler.cpp 21 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699
  1. /*
  2. * CVideoHandler.cpp, part of VCMI engine
  3. *
  4. * Authors: listed in file AUTHORS in main folder
  5. *
  6. * License: GNU General Public License v2.0 or later
  7. * Full text of license available in license.txt file, in main folder
  8. *
  9. */
  10. #include "StdInc.h"
  11. #include "CVideoHandler.h"
  12. #ifndef DISABLE_VIDEO
  13. #include "ISoundPlayer.h"
  14. #include "../CGameInfo.h"
  15. #include "../CMT.h"
  16. #include "../eventsSDL/InputHandler.h"
  17. #include "../gui/CGuiHandler.h"
  18. #include "../render/Canvas.h"
  19. #include "../render/IScreenHandler.h"
  20. #include "../renderSDL/SDL_Extensions.h"
  21. #include "../../lib/filesystem/CInputStream.h"
  22. #include "../../lib/filesystem/Filesystem.h"
  23. #include "../../lib/texts/CGeneralTextHandler.h"
  24. #include "../../lib/texts/Languages.h"
  25. #include <SDL_render.h>
  26. extern "C" {
  27. #include <libavformat/avformat.h>
  28. #include <libavcodec/avcodec.h>
  29. #include <libavutil/imgutils.h>
  30. #include <libavutil/opt.h>
  31. #include <libswscale/swscale.h>
  32. #include <libswresample/swresample.h>
  33. }
  34. // Define a set of functions to read data
  35. static int lodRead(void * opaque, uint8_t * buf, int size)
  36. {
  37. auto * data = static_cast<CInputStream *>(opaque);
  38. auto bytesRead = data->read(buf, size);
  39. if(bytesRead == 0)
  40. return AVERROR_EOF;
  41. return bytesRead;
  42. }
  43. static si64 lodSeek(void * opaque, si64 pos, int whence)
  44. {
  45. auto * data = static_cast<CInputStream *>(opaque);
  46. if(whence & AVSEEK_SIZE)
  47. return data->getSize();
  48. return data->seek(pos);
  49. }
  50. [[noreturn]] static void throwFFmpegError(int errorCode)
  51. {
  52. std::array<char, AV_ERROR_MAX_STRING_SIZE> errorMessage{};
  53. av_strerror(errorCode, errorMessage.data(), errorMessage.size());
  54. throw std::runtime_error(errorMessage.data());
  55. }
  56. static std::unique_ptr<CInputStream> findVideoData(const VideoPath & videoToOpen)
  57. {
  58. if(CResourceHandler::get()->existsResource(videoToOpen))
  59. return CResourceHandler::get()->load(videoToOpen);
  60. auto highQualityVideoToOpenWithDir = videoToOpen.addPrefix("VIDEO/");
  61. auto lowQualityVideo = videoToOpen.toType<EResType::VIDEO_LOW_QUALITY>();
  62. auto lowQualityVideoWithDir = highQualityVideoToOpenWithDir.toType<EResType::VIDEO_LOW_QUALITY>();
  63. if(CResourceHandler::get()->existsResource(highQualityVideoToOpenWithDir))
  64. return CResourceHandler::get()->load(highQualityVideoToOpenWithDir);
  65. if(CResourceHandler::get()->existsResource(lowQualityVideo))
  66. return CResourceHandler::get()->load(lowQualityVideo);
  67. if(CResourceHandler::get()->existsResource(lowQualityVideoWithDir))
  68. return CResourceHandler::get()->load(lowQualityVideoWithDir);
  69. return nullptr;
  70. }
  71. bool FFMpegStream::openInput(const VideoPath & videoToOpen)
  72. {
  73. input = findVideoData(videoToOpen);
  74. return input != nullptr;
  75. }
  76. void FFMpegStream::openContext()
  77. {
  78. static const int BUFFER_SIZE = 4096;
  79. input->seek(0);
  80. auto * buffer = static_cast<unsigned char *>(av_malloc(BUFFER_SIZE)); // will be freed by ffmpeg
  81. context = avio_alloc_context(buffer, BUFFER_SIZE, 0, input.get(), lodRead, nullptr, lodSeek);
  82. formatContext = avformat_alloc_context();
  83. formatContext->pb = context;
  84. // filename is not needed - file was already open and stored in this->data;
  85. int avfopen = avformat_open_input(&formatContext, "dummyFilename", nullptr, nullptr);
  86. if(avfopen != 0)
  87. throwFFmpegError(avfopen);
  88. // Retrieve stream information
  89. int findStreamInfo = avformat_find_stream_info(formatContext, nullptr);
  90. if(avfopen < 0)
  91. throwFFmpegError(findStreamInfo);
  92. }
  93. void FFMpegStream::openCodec(int desiredStreamIndex)
  94. {
  95. streamIndex = desiredStreamIndex;
  96. // Find the decoder for the stream
  97. codec = avcodec_find_decoder(formatContext->streams[streamIndex]->codecpar->codec_id);
  98. if(codec == nullptr)
  99. throw std::runtime_error("Unsupported codec");
  100. codecContext = avcodec_alloc_context3(codec);
  101. if(codecContext == nullptr)
  102. throw std::runtime_error("Failed to create codec context");
  103. // Get a pointer to the codec context for the video stream
  104. int ret = avcodec_parameters_to_context(codecContext, formatContext->streams[streamIndex]->codecpar);
  105. if(ret < 0)
  106. {
  107. //We cannot get codec from parameters
  108. avcodec_free_context(&codecContext);
  109. throwFFmpegError(ret);
  110. }
  111. // Open codec
  112. ret = avcodec_open2(codecContext, codec, nullptr);
  113. if(ret < 0)
  114. {
  115. // Could not open codec
  116. codec = nullptr;
  117. throwFFmpegError(ret);
  118. }
  119. // Allocate video frame
  120. frame = av_frame_alloc();
  121. }
  122. const AVCodecParameters * FFMpegStream::getCodecParameters() const
  123. {
  124. return formatContext->streams[streamIndex]->codecpar;
  125. }
  126. const AVCodecContext * FFMpegStream::getCodecContext() const
  127. {
  128. return codecContext;
  129. }
  130. const AVFrame * FFMpegStream::getCurrentFrame() const
  131. {
  132. return frame;
  133. }
  134. void CVideoInstance::openVideo()
  135. {
  136. openContext();
  137. openCodec(findVideoStream());
  138. }
  139. void CVideoInstance::prepareOutput(float scaleFactor, bool useTextureOutput)
  140. {
  141. //setup scaling
  142. dimensions = Point(getCodecContext()->width * scaleFactor, getCodecContext()->height * scaleFactor) * GH.screenHandler().getScalingFactor();
  143. // Allocate a place to put our YUV image on that screen
  144. if (useTextureOutput)
  145. {
  146. std::array potentialFormats = {
  147. AV_PIX_FMT_YUV420P, // -> SDL_PIXELFORMAT_IYUV - most of H3 videos use YUV format, so it is preferred to save some space & conversion time
  148. AV_PIX_FMT_RGB32, // -> SDL_PIXELFORMAT_ARGB8888 - some .smk videos actually use palette, so RGB > YUV. This is also our screen texture format
  149. AV_PIX_FMT_NONE
  150. };
  151. auto preferredFormat = avcodec_find_best_pix_fmt_of_list(potentialFormats.data(), getCodecContext()->pix_fmt, false, nullptr);
  152. if (preferredFormat == AV_PIX_FMT_YUV420P)
  153. textureYUV = SDL_CreateTexture( mainRenderer, SDL_PIXELFORMAT_IYUV, SDL_TEXTUREACCESS_STREAMING, dimensions.x, dimensions.y);
  154. else
  155. textureRGB = SDL_CreateTexture( mainRenderer, SDL_PIXELFORMAT_ARGB8888, SDL_TEXTUREACCESS_STREAMING, dimensions.x, dimensions.y);
  156. sws = sws_getContext(getCodecContext()->width, getCodecContext()->height, getCodecContext()->pix_fmt,
  157. dimensions.x, dimensions.y, preferredFormat,
  158. SWS_BICUBIC, nullptr, nullptr, nullptr);
  159. }
  160. else
  161. {
  162. surface = CSDL_Ext::newSurface(dimensions);
  163. sws = sws_getContext(getCodecContext()->width, getCodecContext()->height, getCodecContext()->pix_fmt,
  164. dimensions.x, dimensions.y, AV_PIX_FMT_RGB32,
  165. SWS_BICUBIC, nullptr, nullptr, nullptr);
  166. }
  167. if (sws == nullptr)
  168. throw std::runtime_error("Failed to create sws");
  169. }
  170. void FFMpegStream::decodeNextFrame()
  171. {
  172. int rc = avcodec_receive_frame(codecContext, frame);
  173. // frame extracted - data that was sent to codecContext before was sufficient
  174. if (rc == 0)
  175. return;
  176. // returning AVERROR(EAGAIN) is legal - this indicates that codec requires more data from input stream to decode next frame
  177. if(rc != AVERROR(EAGAIN))
  178. throwFFmpegError(rc);
  179. for(;;)
  180. {
  181. AVPacket packet;
  182. // codecContext does not have enough input data - read next packet from input stream
  183. int ret = av_read_frame(formatContext, &packet);
  184. if(ret < 0)
  185. {
  186. if(ret == AVERROR_EOF)
  187. {
  188. av_packet_unref(&packet);
  189. av_frame_free(&frame);
  190. frame = nullptr;
  191. return;
  192. }
  193. throwFFmpegError(ret);
  194. }
  195. // Is this a packet from the stream that needs decoding?
  196. if(packet.stream_index == streamIndex)
  197. {
  198. // Decode read packet
  199. // Note: this method may return AVERROR(EAGAIN). However this should never happen with ffmpeg API
  200. // since there is guaranteed call to avcodec_receive_frame and ffmpeg API promises that *both* of these methods will never return AVERROR(EAGAIN).
  201. int rc = avcodec_send_packet(codecContext, &packet);
  202. if(rc < 0)
  203. throwFFmpegError(rc);
  204. rc = avcodec_receive_frame(codecContext, frame);
  205. if(rc == AVERROR(EAGAIN))
  206. {
  207. // still need more data - read next packet
  208. av_packet_unref(&packet);
  209. continue;
  210. }
  211. else if(rc < 0)
  212. {
  213. throwFFmpegError(rc);
  214. }
  215. else
  216. {
  217. // read succesful. Exit the loop
  218. av_packet_unref(&packet);
  219. return;
  220. }
  221. }
  222. av_packet_unref(&packet);
  223. }
  224. }
  225. bool CVideoInstance::loadNextFrame()
  226. {
  227. decodeNextFrame();
  228. const AVFrame * frame = getCurrentFrame();
  229. if(!frame)
  230. return false;
  231. uint8_t * data[4] = {};
  232. int linesize[4] = {};
  233. if(textureYUV)
  234. {
  235. av_image_alloc(data, linesize, dimensions.x, dimensions.y, AV_PIX_FMT_YUV420P, 1);
  236. sws_scale(sws, frame->data, frame->linesize, 0, getCodecContext()->height, data, linesize);
  237. SDL_UpdateYUVTexture(textureYUV, nullptr, data[0], linesize[0], data[1], linesize[1], data[2], linesize[2]);
  238. av_freep(&data[0]);
  239. }
  240. if(textureRGB)
  241. {
  242. av_image_alloc(data, linesize, dimensions.x, dimensions.y, AV_PIX_FMT_RGB32, 1);
  243. sws_scale(sws, frame->data, frame->linesize, 0, getCodecContext()->height, data, linesize);
  244. SDL_UpdateTexture(textureRGB, nullptr, data[0], linesize[0]);
  245. av_freep(&data[0]);
  246. }
  247. if(surface)
  248. {
  249. // Avoid buffer overflow caused by sws_scale():
  250. // http://trac.ffmpeg.org/ticket/9254
  251. size_t pic_bytes = surface->pitch * surface->h;
  252. size_t ffmped_pad = 1024; /* a few bytes of overflow will go here */
  253. void * for_sws = av_malloc(pic_bytes + ffmped_pad);
  254. data[0] = (ui8 *)for_sws;
  255. linesize[0] = surface->pitch;
  256. sws_scale(sws, frame->data, frame->linesize, 0, getCodecContext()->height, data, linesize);
  257. memcpy(surface->pixels, for_sws, pic_bytes);
  258. av_free(for_sws);
  259. }
  260. return true;
  261. }
  262. bool CVideoInstance::videoEnded()
  263. {
  264. return getCurrentFrame() == nullptr;
  265. }
  266. CVideoInstance::~CVideoInstance()
  267. {
  268. sws_freeContext(sws);
  269. SDL_DestroyTexture(textureYUV);
  270. SDL_DestroyTexture(textureRGB);
  271. SDL_FreeSurface(surface);
  272. }
  273. FFMpegStream::~FFMpegStream()
  274. {
  275. av_frame_free(&frame);
  276. #if (LIBAVCODEC_VERSION_MAJOR < 61 )
  277. // deprecated, apparently no longer necessary - avcodec_free_context should suffice
  278. avcodec_close(codecContext);
  279. #endif
  280. avcodec_free_context(&codecContext);
  281. avformat_close_input(&formatContext);
  282. av_free(context);
  283. }
  284. Point CVideoInstance::size()
  285. {
  286. return dimensions / GH.screenHandler().getScalingFactor();
  287. }
  288. void CVideoInstance::show(const Point & position, Canvas & canvas)
  289. {
  290. if(sws == nullptr)
  291. throw std::runtime_error("No video to show!");
  292. CSDL_Ext::blitSurface(surface, canvas.getInternalSurface(), position * GH.screenHandler().getScalingFactor());
  293. }
  294. double FFMpegStream::getCurrentFrameEndTime() const
  295. {
  296. #if(LIBAVUTIL_VERSION_MAJOR < 58)
  297. auto packet_duration = frame->pkt_duration;
  298. #else
  299. auto packet_duration = frame->duration;
  300. #endif
  301. return (frame->pts + packet_duration) * av_q2d(formatContext->streams[streamIndex]->time_base);
  302. }
  303. double FFMpegStream::getCurrentFrameDuration() const
  304. {
  305. #if(LIBAVUTIL_VERSION_MAJOR < 58)
  306. auto packet_duration = frame->pkt_duration;
  307. #else
  308. auto packet_duration = frame->duration;
  309. #endif
  310. return packet_duration * av_q2d(formatContext->streams[streamIndex]->time_base);
  311. }
  312. void CVideoInstance::tick(uint32_t msPassed)
  313. {
  314. if(sws == nullptr)
  315. throw std::runtime_error("No video to show!");
  316. if(videoEnded())
  317. throw std::runtime_error("Video already ended!");
  318. frameTime += msPassed / 1000.0;
  319. if(frameTime >= getCurrentFrameEndTime())
  320. loadNextFrame();
  321. }
  322. struct FFMpegFormatDescription
  323. {
  324. uint8_t sampleSizeBytes;
  325. uint8_t wavFormatID;
  326. bool isPlanar;
  327. };
  328. static FFMpegFormatDescription getAudioFormatProperties(int audioFormat)
  329. {
  330. switch (audioFormat)
  331. {
  332. case AV_SAMPLE_FMT_U8: return { 1, 1, false};
  333. case AV_SAMPLE_FMT_U8P: return { 1, 1, true};
  334. case AV_SAMPLE_FMT_S16: return { 2, 1, false};
  335. case AV_SAMPLE_FMT_S16P: return { 2, 1, true};
  336. case AV_SAMPLE_FMT_S32: return { 4, 1, false};
  337. case AV_SAMPLE_FMT_S32P: return { 4, 1, true};
  338. case AV_SAMPLE_FMT_S64: return { 8, 1, false};
  339. case AV_SAMPLE_FMT_S64P: return { 8, 1, true};
  340. case AV_SAMPLE_FMT_FLT: return { 4, 3, false};
  341. case AV_SAMPLE_FMT_FLTP: return { 4, 3, true};
  342. case AV_SAMPLE_FMT_DBL: return { 8, 3, false};
  343. case AV_SAMPLE_FMT_DBLP: return { 8, 3, true};
  344. }
  345. throw std::runtime_error("Invalid audio format");
  346. }
  347. int FFMpegStream::findAudioStream() const
  348. {
  349. std::vector<int> audioStreamIndices;
  350. for(int i = 0; i < formatContext->nb_streams; i++)
  351. if(formatContext->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_AUDIO)
  352. audioStreamIndices.push_back(i);
  353. if (audioStreamIndices.empty())
  354. return -1;
  355. if (audioStreamIndices.size() == 1)
  356. return audioStreamIndices.front();
  357. // multiple audio streams - try to pick best one based on language settings
  358. std::map<int, std::string> streamToLanguage;
  359. // Approach 1 - check if stream has language set in metadata
  360. for (auto const & index : audioStreamIndices)
  361. {
  362. const AVDictionaryEntry *e = av_dict_get(formatContext->streams[index]->metadata, "language", nullptr, 0);
  363. if (e)
  364. streamToLanguage[index] = e->value;
  365. }
  366. // Approach 2 - no metadata found. This may be video from Chronicles which have predefined (presumably hardcoded) list of languages
  367. if (streamToLanguage.empty())
  368. {
  369. if (audioStreamIndices.size() == 2)
  370. {
  371. streamToLanguage[audioStreamIndices[0]] = Languages::getLanguageOptions(Languages::ELanguages::ENGLISH).tagISO2;
  372. streamToLanguage[audioStreamIndices[1]] = Languages::getLanguageOptions(Languages::ELanguages::GERMAN).tagISO2;
  373. }
  374. if (audioStreamIndices.size() == 5)
  375. {
  376. streamToLanguage[audioStreamIndices[0]] = Languages::getLanguageOptions(Languages::ELanguages::ENGLISH).tagISO2;
  377. streamToLanguage[audioStreamIndices[1]] = Languages::getLanguageOptions(Languages::ELanguages::FRENCH).tagISO2;
  378. streamToLanguage[audioStreamIndices[2]] = Languages::getLanguageOptions(Languages::ELanguages::GERMAN).tagISO2;
  379. streamToLanguage[audioStreamIndices[3]] = Languages::getLanguageOptions(Languages::ELanguages::ITALIAN).tagISO2;
  380. streamToLanguage[audioStreamIndices[4]] = Languages::getLanguageOptions(Languages::ELanguages::SPANISH).tagISO2;
  381. }
  382. }
  383. std::string preferredLanguageName = CGI->generaltexth->getPreferredLanguage();
  384. std::string preferredTag = Languages::getLanguageOptions(preferredLanguageName).tagISO2;
  385. for (auto const & entry : streamToLanguage)
  386. if (entry.second == preferredTag)
  387. return entry.first;
  388. return audioStreamIndices.front();
  389. }
  390. int FFMpegStream::findVideoStream() const
  391. {
  392. for(int i = 0; i < formatContext->nb_streams; i++)
  393. if(formatContext->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_VIDEO)
  394. return i;
  395. return -1;
  396. }
  397. std::pair<std::unique_ptr<ui8 []>, si64> CAudioInstance::extractAudio(const VideoPath & videoToOpen)
  398. {
  399. if (!openInput(videoToOpen))
  400. return { nullptr, 0};
  401. openContext();
  402. int audioStreamIndex = findAudioStream();
  403. if (audioStreamIndex == -1)
  404. return { nullptr, 0};
  405. openCodec(audioStreamIndex);
  406. const auto * codecpar = getCodecParameters();
  407. std::vector<ui8> samples;
  408. auto formatProperties = getAudioFormatProperties(codecpar->format);
  409. #if(LIBAVUTIL_VERSION_MAJOR < 58)
  410. int numChannels = codecpar->channels;
  411. #else
  412. int numChannels = codecpar->ch_layout.nb_channels;
  413. #endif
  414. samples.reserve(44100 * 5); // arbitrary 5-second buffer to reduce reallocations
  415. if (formatProperties.isPlanar && numChannels > 1)
  416. {
  417. // Format is 'planar', which is not supported by wav / SDL
  418. // Use swresample part of ffmpeg to deplanarize audio into format supported by wav / SDL
  419. auto sourceFormat = static_cast<AVSampleFormat>(codecpar->format);
  420. auto targetFormat = av_get_alt_sample_fmt(sourceFormat, false);
  421. SwrContext * swr_ctx = swr_alloc();
  422. #if (LIBAVUTIL_VERSION_MAJOR < 58)
  423. av_opt_set_channel_layout(swr_ctx, "in_chlayout", codecpar->channel_layout, 0);
  424. av_opt_set_channel_layout(swr_ctx, "out_chlayout", codecpar->channel_layout, 0);
  425. #else
  426. av_opt_set_chlayout(swr_ctx, "in_chlayout", &codecpar->ch_layout, 0);
  427. av_opt_set_chlayout(swr_ctx, "out_chlayout", &codecpar->ch_layout, 0);
  428. #endif
  429. av_opt_set_int(swr_ctx, "in_sample_rate", codecpar->sample_rate, 0);
  430. av_opt_set_sample_fmt(swr_ctx, "in_sample_fmt", sourceFormat, 0);
  431. av_opt_set_int(swr_ctx, "out_sample_rate", codecpar->sample_rate, 0);
  432. av_opt_set_sample_fmt(swr_ctx, "out_sample_fmt", targetFormat, 0);
  433. int initResult = swr_init(swr_ctx);
  434. if (initResult < 0)
  435. throwFFmpegError(initResult);
  436. std::vector<uint8_t> frameSamplesBuffer;
  437. for (;;)
  438. {
  439. decodeNextFrame();
  440. const AVFrame * frame = getCurrentFrame();
  441. if (!frame)
  442. break;
  443. size_t samplesToRead = frame->nb_samples * numChannels;
  444. size_t bytesToRead = samplesToRead * formatProperties.sampleSizeBytes;
  445. frameSamplesBuffer.resize(std::max(frameSamplesBuffer.size(), bytesToRead));
  446. uint8_t * frameSamplesPtr = frameSamplesBuffer.data();
  447. int result = swr_convert(swr_ctx, &frameSamplesPtr, frame->nb_samples, const_cast<const uint8_t **>(frame->data), frame->nb_samples);
  448. if (result < 0)
  449. throwFFmpegError(result);
  450. size_t samplesToCopy = result * numChannels;
  451. size_t bytesToCopy = samplesToCopy * formatProperties.sampleSizeBytes;
  452. samples.insert(samples.end(), frameSamplesBuffer.begin(), frameSamplesBuffer.begin() + bytesToCopy);
  453. }
  454. swr_free(&swr_ctx);
  455. }
  456. else
  457. {
  458. for (;;)
  459. {
  460. decodeNextFrame();
  461. const AVFrame * frame = getCurrentFrame();
  462. if (!frame)
  463. break;
  464. size_t samplesToRead = frame->nb_samples * numChannels;
  465. size_t bytesToRead = samplesToRead * formatProperties.sampleSizeBytes;
  466. samples.insert(samples.end(), frame->data[0], frame->data[0] + bytesToRead);
  467. }
  468. }
  469. struct WavHeader {
  470. ui8 RIFF[4] = {'R', 'I', 'F', 'F'};
  471. ui32 ChunkSize;
  472. ui8 WAVE[4] = {'W', 'A', 'V', 'E'};
  473. ui8 fmt[4] = {'f', 'm', 't', ' '};
  474. ui32 Subchunk1Size = 16;
  475. ui16 AudioFormat = 1;
  476. ui16 NumOfChan = 2;
  477. ui32 SamplesPerSec = 22050;
  478. ui32 bytesPerSec = 22050 * 2;
  479. ui16 blockAlign = 1;
  480. ui16 bitsPerSample = 32;
  481. ui8 Subchunk2ID[4] = {'d', 'a', 't', 'a'};
  482. ui32 Subchunk2Size;
  483. };
  484. WavHeader wav;
  485. wav.ChunkSize = samples.size() + sizeof(WavHeader) - 8;
  486. wav.AudioFormat = formatProperties.wavFormatID; // 1 = PCM, 3 = IEEE float
  487. wav.NumOfChan = numChannels;
  488. wav.SamplesPerSec = codecpar->sample_rate;
  489. wav.bytesPerSec = codecpar->sample_rate * formatProperties.sampleSizeBytes;
  490. wav.bitsPerSample = formatProperties.sampleSizeBytes * 8;
  491. wav.Subchunk2Size = samples.size() + sizeof(WavHeader) - 44;
  492. auto * wavPtr = reinterpret_cast<ui8*>(&wav);
  493. auto dat = std::make_pair(std::make_unique<ui8[]>(samples.size() + sizeof(WavHeader)), samples.size() + sizeof(WavHeader));
  494. std::copy(wavPtr, wavPtr + sizeof(WavHeader), dat.first.get());
  495. std::copy(samples.begin(), samples.end(), dat.first.get() + sizeof(WavHeader));
  496. return dat;
  497. }
  498. bool CVideoPlayer::openAndPlayVideoImpl(const VideoPath & name, const Point & position, bool useOverlay, bool stopOnKey)
  499. {
  500. CVideoInstance instance;
  501. auto extractedAudio = getAudio(name);
  502. int audioHandle = CCS->soundh->playSound(extractedAudio);
  503. if (!instance.openInput(name))
  504. return true;
  505. instance.openVideo();
  506. instance.prepareOutput(1, true);
  507. auto lastTimePoint = boost::chrono::steady_clock::now();
  508. while(instance.loadNextFrame())
  509. {
  510. if(stopOnKey)
  511. {
  512. GH.input().fetchEvents();
  513. if(GH.input().ignoreEventsUntilInput())
  514. {
  515. CCS->soundh->stopSound(audioHandle);
  516. return false;
  517. }
  518. }
  519. SDL_Rect rect;
  520. rect.x = position.x;
  521. rect.y = position.y;
  522. rect.w = instance.dimensions.x;
  523. rect.h = instance.dimensions.y;
  524. SDL_RenderFillRect(mainRenderer, &rect);
  525. if(instance.textureYUV)
  526. SDL_RenderCopy(mainRenderer, instance.textureYUV, nullptr, &rect);
  527. else
  528. SDL_RenderCopy(mainRenderer, instance.textureRGB, nullptr, &rect);
  529. SDL_RenderPresent(mainRenderer);
  530. // Framerate delay
  531. double targetFrameTimeSeconds = instance.getCurrentFrameDuration();
  532. auto targetFrameTime = boost::chrono::milliseconds(static_cast<int>(1000 * targetFrameTimeSeconds));
  533. auto timePointAfterPresent = boost::chrono::steady_clock::now();
  534. auto timeSpentBusy = boost::chrono::duration_cast<boost::chrono::milliseconds>(timePointAfterPresent - lastTimePoint);
  535. if(targetFrameTime > timeSpentBusy)
  536. boost::this_thread::sleep_for(targetFrameTime - timeSpentBusy);
  537. lastTimePoint = boost::chrono::steady_clock::now();
  538. }
  539. return true;
  540. }
  541. void CVideoPlayer::playSpellbookAnimation(const VideoPath & name, const Point & position)
  542. {
  543. openAndPlayVideoImpl(name, position * GH.screenHandler().getScalingFactor(), false, false);
  544. }
  545. std::unique_ptr<IVideoInstance> CVideoPlayer::open(const VideoPath & name, float scaleFactor)
  546. {
  547. auto result = std::make_unique<CVideoInstance>();
  548. if (!result->openInput(name))
  549. return nullptr;
  550. result->openVideo();
  551. result->prepareOutput(scaleFactor, false);
  552. result->loadNextFrame(); // prepare 1st frame
  553. return result;
  554. }
  555. std::pair<std::unique_ptr<ui8[]>, si64> CVideoPlayer::getAudio(const VideoPath & videoToOpen)
  556. {
  557. AudioPath audioPath = videoToOpen.toType<EResType::SOUND>();
  558. AudioPath audioPathVideoDir = audioPath.addPrefix("VIDEO/");
  559. if(CResourceHandler::get()->existsResource(audioPath))
  560. return CResourceHandler::get()->load(audioPath)->readAll();
  561. if(CResourceHandler::get()->existsResource(audioPathVideoDir))
  562. return CResourceHandler::get()->load(audioPathVideoDir)->readAll();
  563. CAudioInstance audio;
  564. return audio.extractAudio(videoToOpen);
  565. }
  566. #endif