247 lines
7.4 KiB
C++
247 lines
7.4 KiB
C++
|
|
|
|
#include "avicodecrestrictions.h"
|
|
#include "tconvert.h"
|
|
#include <windows.h>
|
|
#include <vfw.h>
|
|
|
|
namespace {
|
|
|
|
HIC getCodec(const std::wstring &codecName, int &bpp) {
|
|
HIC hic = 0;
|
|
ICINFO icinfo;
|
|
memset(&icinfo, 0, sizeof(ICINFO));
|
|
bool found = false;
|
|
|
|
char descr[2048], name[2048];
|
|
DWORD fccType = 0;
|
|
|
|
BITMAPINFO inFmt;
|
|
memset(&inFmt, 0, sizeof(BITMAPINFO));
|
|
|
|
inFmt.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
|
|
inFmt.bmiHeader.biWidth = inFmt.bmiHeader.biHeight = 100;
|
|
inFmt.bmiHeader.biPlanes = 1;
|
|
inFmt.bmiHeader.biCompression = BI_RGB;
|
|
for (bpp = 32; (bpp >= 24) && !found; bpp -= 8) {
|
|
// find the codec.
|
|
inFmt.bmiHeader.biBitCount = bpp;
|
|
for (int i = 0; ICInfo(fccType, i, &icinfo); i++) {
|
|
hic = ICOpen(icinfo.fccType, icinfo.fccHandler, ICMODE_COMPRESS);
|
|
|
|
ICGetInfo(hic, &icinfo, sizeof(ICINFO)); // Find out the compressor name
|
|
WideCharToMultiByte(CP_ACP, 0, icinfo.szDescription, -1, descr,
|
|
sizeof(descr), 0, 0);
|
|
WideCharToMultiByte(CP_ACP, 0, icinfo.szName, -1, name, sizeof(name), 0,
|
|
0);
|
|
|
|
std::string compressorName;
|
|
compressorName = std::string(name) + " '" + std::to_string(bpp) + "' " +
|
|
std::string(descr);
|
|
|
|
if (hic) {
|
|
if (ICCompressQuery(hic, &inFmt, NULL) != ICERR_OK) {
|
|
ICClose(hic);
|
|
continue; // Skip this compressor if it can't handle the format.
|
|
}
|
|
if (::to_wstring(compressorName) == codecName) {
|
|
found = true;
|
|
break;
|
|
}
|
|
ICClose(hic);
|
|
}
|
|
}
|
|
if (found) break;
|
|
}
|
|
return hic;
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
bool canWork(const HIC &hic, const TDimension &resolution, int bpp) {
|
|
int lx = resolution.lx, ly = resolution.ly;
|
|
|
|
BITMAPINFO bi;
|
|
bi.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
|
|
bi.bmiHeader.biPlanes = 1;
|
|
bi.bmiHeader.biCompression = BI_RGB;
|
|
bi.bmiHeader.biXPelsPerMeter = 80;
|
|
bi.bmiHeader.biYPelsPerMeter = 72;
|
|
bi.bmiHeader.biClrUsed = 0;
|
|
bi.bmiHeader.biClrImportant = 0;
|
|
bi.bmiHeader.biBitCount = bpp;
|
|
bi.bmiHeader.biWidth = lx;
|
|
bi.bmiHeader.biHeight = ly;
|
|
|
|
return ICERR_OK == ICCompressQuery(hic, &bi.bmiHeader, NULL);
|
|
}
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
void AviCodecRestrictions::getRestrictions(const std::wstring &codecName,
|
|
QString &restrictions) {
|
|
restrictions.clear();
|
|
if (codecName == L"Uncompressed") {
|
|
restrictions = QObject::tr("No restrictions for uncompressed avi video");
|
|
return;
|
|
}
|
|
// find the codec
|
|
int bpp;
|
|
HIC hic = getCodec(codecName, bpp);
|
|
if (!hic) {
|
|
restrictions = QObject::tr(
|
|
"It is not possible to communicate with the codec.\n Probably the "
|
|
"codec cannot work correctly.");
|
|
return;
|
|
}
|
|
|
|
BITMAPINFO bi;
|
|
bi.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
|
|
bi.bmiHeader.biPlanes = 1;
|
|
bi.bmiHeader.biCompression = BI_RGB;
|
|
bi.bmiHeader.biXPelsPerMeter = 80;
|
|
bi.bmiHeader.biYPelsPerMeter = 72;
|
|
bi.bmiHeader.biClrUsed = 0;
|
|
bi.bmiHeader.biClrImportant = 0;
|
|
|
|
int lx = 640, ly = 480;
|
|
bi.bmiHeader.biWidth = lx;
|
|
bi.bmiHeader.biHeight = ly;
|
|
|
|
// Loop until we can find a width, height, and depth that works!
|
|
int i;
|
|
|
|
// check the x lenght
|
|
bi.bmiHeader.biBitCount = bpp;
|
|
for (i = 3; i >= 0; i--) {
|
|
bi.bmiHeader.biWidth = lx + (1 << i);
|
|
bi.bmiHeader.biSizeImage =
|
|
((bi.bmiHeader.biWidth * bi.bmiHeader.biBitCount + 31) / 32) * 4 * ly;
|
|
|
|
if (ICERR_OK != ICCompressQuery(hic, &bi.bmiHeader, NULL)) break;
|
|
}
|
|
if (i >= 0)
|
|
restrictions = QObject::tr("video width must be a multiple of %1")
|
|
.arg(QString::number(1 << (i + 1)));
|
|
|
|
// check the y lenght
|
|
bi.bmiHeader.biWidth = 640;
|
|
for (i = 3; i >= 0; i--) {
|
|
bi.bmiHeader.biHeight = ly + (1 << i);
|
|
bi.bmiHeader.biSizeImage =
|
|
((lx * bi.bmiHeader.biBitCount + 31) / 32) * 4 * bi.bmiHeader.biHeight;
|
|
|
|
if (ICERR_OK != ICCompressQuery(hic, &bi.bmiHeader, NULL)) break;
|
|
}
|
|
if (i >= 0)
|
|
restrictions = restrictions + "\n" +
|
|
QObject::tr("video lenght must be a multiple of %1")
|
|
.arg(QString::number(1 << (i + 1)));
|
|
|
|
ICClose(hic);
|
|
|
|
if (restrictions.isEmpty())
|
|
restrictions = QObject::tr("No restrictions for this codec");
|
|
else
|
|
restrictions.prepend(QObject::tr("Resolution restrictions:") + "\n");
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
bool AviCodecRestrictions::canWriteMovie(const std::wstring &codecName,
|
|
const TDimension &resolution) {
|
|
if (codecName == L"Uncompressed") {
|
|
return true;
|
|
}
|
|
// find the codec
|
|
int bpp;
|
|
HIC hic = getCodec(codecName, bpp);
|
|
if (!hic) return false;
|
|
|
|
bool test = canWork(hic, resolution, bpp);
|
|
|
|
ICClose(hic);
|
|
return test;
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
bool AviCodecRestrictions::canBeConfigured(const std::wstring &codecName) {
|
|
if (codecName == L"Uncompressed") return false;
|
|
|
|
// find the codec
|
|
int bpp;
|
|
HIC hic = getCodec(codecName, bpp);
|
|
if (!hic) return false;
|
|
|
|
bool test = ICQueryConfigure(hic);
|
|
ICClose(hic);
|
|
return test;
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
void AviCodecRestrictions::openConfiguration(const std::wstring &codecName,
|
|
void *winId) {
|
|
if (codecName == L"Uncompressed") return;
|
|
|
|
// find the codec
|
|
int bpp;
|
|
HIC hic = getCodec(codecName, bpp);
|
|
if (!hic) return;
|
|
|
|
ICConfigure(hic, winId);
|
|
ICClose(hic);
|
|
}
|
|
|
|
//-----------------------------------------------------------------------------
|
|
|
|
QMap<std::wstring, bool> AviCodecRestrictions::getUsableCodecs(
|
|
const TDimension &resolution) {
|
|
QMap<std::wstring, bool> codecs;
|
|
|
|
HIC hic = 0;
|
|
ICINFO icinfo;
|
|
memset(&icinfo, 0, sizeof(ICINFO));
|
|
|
|
char descr[2048], name[2048];
|
|
DWORD fccType = 0;
|
|
|
|
BITMAPINFO inFmt;
|
|
memset(&inFmt, 0, sizeof(BITMAPINFO));
|
|
|
|
inFmt.bmiHeader.biSize = sizeof(BITMAPINFOHEADER);
|
|
inFmt.bmiHeader.biWidth = inFmt.bmiHeader.biHeight = 100;
|
|
inFmt.bmiHeader.biPlanes = 1;
|
|
inFmt.bmiHeader.biCompression = BI_RGB;
|
|
int bpp;
|
|
for (bpp = 32; (bpp >= 24); bpp -= 8) {
|
|
// find the codec.
|
|
inFmt.bmiHeader.biBitCount = bpp;
|
|
for (int i = 0; ICInfo(fccType, i, &icinfo); i++) {
|
|
hic = ICOpen(icinfo.fccType, icinfo.fccHandler, ICMODE_COMPRESS);
|
|
|
|
ICGetInfo(hic, &icinfo, sizeof(ICINFO)); // Find out the compressor name
|
|
WideCharToMultiByte(CP_ACP, 0, icinfo.szDescription, -1, descr,
|
|
sizeof(descr), 0, 0);
|
|
WideCharToMultiByte(CP_ACP, 0, icinfo.szName, -1, name, sizeof(name), 0,
|
|
0);
|
|
|
|
std::wstring compressorName;
|
|
compressorName =
|
|
::to_wstring(std::string(name) + " '" + std::to_string(bpp) + "' " +
|
|
std::string(descr));
|
|
|
|
if (hic) {
|
|
if (ICCompressQuery(hic, &inFmt, NULL) != ICERR_OK) {
|
|
ICClose(hic);
|
|
continue; // Skip this compressor if it can't handle the format.
|
|
}
|
|
codecs[compressorName] = canWork(hic, resolution, bpp);
|
|
ICClose(hic);
|
|
}
|
|
}
|
|
}
|
|
return codecs;
|
|
}
|