MediaInfo.cpp

Go to the documentation of this file.

#include <dcl/Config.h>
 
#undef _UNICODE
#undef UNICODE
 
#define _UNICODE
#define UNICODE
 
#ifdef __GNUC__
#pragma GCC diagnostic push
#pragma GCC diagnostic ignored "-Wunused-function"
#pragma GCC diagnostic ignored "-Wunused-variable"
#endif
 
#include "MediaInfoDLL.h"
 
#ifdef __GNUC__
#pragma GCC diagnostic pop
#endif
 
#ifdef __WINNT__
    #include <windows.h>
#else
    #include <dcl/_dlfcn.h>
#endif
 
#include <dcl/Dll.h>
#include <dcl/Numeric.h>
#include "MediaInfo.h"
 
#define __TRACE_THIS    0
#if __TRACE_THIS
#define __DCL_TRACE1_N  __DCL_TRACE1
#define __DCL_TRACE2_N  __DCL_TRACE2
#define __DCL_TRACE3_N  __DCL_TRACE3
#define __DCL_TRACE4_N  __DCL_TRACE4
#else
#define __DCL_TRACE1_N(fmt, arg)
#define __DCL_TRACE2_N(fmt, arg1, arg2)
#define __DCL_TRACE3_N(fmt, arg1, arg2, arg3)
#define __DCL_TRACE4_N(fmt, arg1, arg2, arg3, arg4)
#endif
 
__DCL_BEGIN_NAMESPACE
 
#if __DCL_HAVE_THIS_FILE__
#undef __THIS_FILE__
static const char_t __THIS_FILE__[] = __T("media/MediaInfo.cpp");
#endif
 
void MediaInfo::initialize()
    __DCL_THROWS1(DllException*)
{
    if (!MediaInfo_Module) {
        MediaInfoDLL_Load();
    }
    if (!MediaInfo_Module) {
#ifdef __WINNT__
        throw new DllException(MEDIAINFODLL_NAME, WINAPI_ERROR(GetLastError()));
#else
        String msg;
        __DCL_VERIFY(__dlerror(msg) == 0);
        throw new DllException(L"" MEDIAINFODLL_NAME, msg);
#endif
    };
}
void MediaInfo::cleanup()
{
    if (MediaInfo_Module) {
        MediaInfoDLL_UnLoad();
    }
}
 
MediaInfo::MediaInfo(const String& _filename)
        __DCL_THROWS2(DllException*, IOException*)
{
    if (!MediaInfo_Module) {
        MediaInfo::initialize();
    }
 
    __handle = MediaInfo_New();
    size_t r = MediaInfo_Open(__handle, _filename);
    if (r == 0) {
        throw new IOException(_filename, L"MediaInfo Open Failed");
    }
}
 
MediaInfo::~MediaInfo()
{
    MediaInfo_Close(__handle);
    MediaInfo_Delete(__handle);
}
 
String MediaInfo::inform() const
{
    String r = MediaInfo_Inform(__handle, 0);
    return r;
}
 
String MediaInfo::inform_r() const
{
    return inform().replace_r(L"[ \t]{2,}", L" ", true)
        .replace_r(L" :", L":", true)
        .replace_r(L"\r\n", L"\n", true)
        .replace_r(L"Complete name[^\n]+\n", L"", true)
        .trim();
}
 
String MediaInfo::option(const wchar_t* _option, const wchar_t* _value) const
{
    __DCL_ASSERT(_option != NULL);
    String r = MediaInfo_Option(__handle, _option, _value ? _value : L"");
    return r;
}
 
String MediaInfo::format() const
{
    return MediaInfo_Get(__handle, MediaInfo_Stream_General, 0, L"Format",
        MediaInfo_Info_Text, MediaInfo_Info_Name);
}
 
uint64_t MediaInfo::fileSize() const
{
    return UInt64::parse(
        MediaInfo_Get(__handle, MediaInfo_Stream_General, 0, L"FileSize",
            MediaInfo_Info_Text, MediaInfo_Info_Name),
        10, 0LLU
    );
}
 
unsigned MediaInfo::videoCount() const
{
    return UInt32::parse(
        MediaInfo_Get(__handle, MediaInfo_Stream_General, 0, L"VideoCount",
            MediaInfo_Info_Text, MediaInfo_Info_Name),
        10, 0LU
    );
}
 
unsigned MediaInfo::audioCount() const
{
    return UInt32::parse(
        MediaInfo_Get(__handle, MediaInfo_Stream_General, 0, L"AudioCount",
            MediaInfo_Info_Text, MediaInfo_Info_Name),
        10, 0LU
    );
}
 
unsigned MediaInfo::imageCount() const
{
    return UInt32::parse(
        MediaInfo_Get(__handle, MediaInfo_Stream_General, 0, L"ImageCount",
            MediaInfo_Info_Text, MediaInfo_Info_Name),
        10, 0LU
    );
}
 
String MediaInfo::videoFormat(size_t streamNumber) const
{
    return MediaInfo_Get(__handle, MediaInfo_Stream_Video, streamNumber, L"Format",
            MediaInfo_Info_Text, MediaInfo_Info_Name);
}
 
unsigned MediaInfo::videoWidth(size_t streamNumber) const
{
    return UInt32::parse(
        MediaInfo_Get(__handle, MediaInfo_Stream_Video, streamNumber, L"Width",
            MediaInfo_Info_Text, MediaInfo_Info_Name),
        10, 0LLU
    );
}
 
unsigned MediaInfo::videoHeight(size_t streamNumber) const
{
    return UInt32::parse(
        MediaInfo_Get(__handle, MediaInfo_Stream_Video, streamNumber, L"Height",
            MediaInfo_Info_Text, MediaInfo_Info_Name),
        10, 0LLU
    );
}
 
// Milliseconds
uint64_t MediaInfo::videoDuration(size_t streamNumber) const
{
    return UInt64::parse(
        MediaInfo_Get(__handle, MediaInfo_Stream_Video, streamNumber, L"Duration",
            MediaInfo_Info_Text, MediaInfo_Info_Name),
        10, 0LLU
    );
}
 
uint32_t MediaInfo::videoDurationAsSeconds(size_t streamNumber) const
{
    return (uint32_t) (videoDuration(streamNumber) / 1000);
}
 
unsigned MediaInfo::videoBitRate(size_t streamNumber) const
{
    return UInt32::parse(
        MediaInfo_Get(__handle, MediaInfo_Stream_Video, streamNumber, L"BitRate",
            MediaInfo_Info_Text, MediaInfo_Info_Name),
        10, 0LLU
    );
}
 
unsigned MediaInfo::videoBitRateMinimum(size_t streamNumber) const
{
    return UInt32::parse(
        MediaInfo_Get(__handle, MediaInfo_Stream_Video, streamNumber, L"BitRate_Minimum",
            MediaInfo_Info_Text, MediaInfo_Info_Name),
        10, 0LLU
    );
}
 
unsigned MediaInfo::videoBitRateMaximum(size_t streamNumber) const
{
    return UInt32::parse(
        MediaInfo_Get(__handle, MediaInfo_Stream_Video, streamNumber, L"BitRate_Maxmum",
            MediaInfo_Info_Text, MediaInfo_Info_Name),
        10, 0LLU
    );
}
 
String MediaInfo::audioFormat(size_t streamNumber) const
{
    return MediaInfo_Get(__handle, MediaInfo_Stream_Audio, streamNumber, L"Format",
        MediaInfo_Info_Text, MediaInfo_Info_Name);
}
 
// Milliseconds
uint64_t MediaInfo::audioDuration(size_t streamNumber) const
{
    return UInt64::parse(
        MediaInfo_Get(__handle, MediaInfo_Stream_Audio, streamNumber, L"Duration",
            MediaInfo_Info_Text, MediaInfo_Info_Name),
        10, 0LLU
    );
}
 
uint32_t MediaInfo::audioDurationAsSeconds(size_t streamNumber) const
{
    return (uint32_t) (audioDuration(streamNumber) / 1000);
}
 
unsigned MediaInfo::audioBitRate(size_t streamNumber) const
{
    return UInt32::parse(
        MediaInfo_Get(__handle, MediaInfo_Stream_Audio, streamNumber, L"BitRate",
            MediaInfo_Info_Text, MediaInfo_Info_Name),
        10, 0LLU
    );
}
 
unsigned MediaInfo::audioBitRateMinimum(size_t streamNumber) const
{
    return UInt32::parse(
        MediaInfo_Get(__handle, MediaInfo_Stream_Audio, streamNumber, L"BitRate_Minimum",
            MediaInfo_Info_Text, MediaInfo_Info_Name),
        10, 0LLU
    );
}
 
unsigned MediaInfo::audioBitRateMaximum(size_t streamNumber) const
{
    return UInt32::parse(
        MediaInfo_Get(__handle, MediaInfo_Stream_Audio, streamNumber, L"BitRate_Maxmum",
            MediaInfo_Info_Text, MediaInfo_Info_Name),
        10, 0LLU
    );
}
 
String MediaInfo::imageFormat(size_t streamNumber) const
{
    return MediaInfo_Get(__handle, MediaInfo_Stream_Image, streamNumber, L"Format",
        MediaInfo_Info_Text, MediaInfo_Info_Name);
}
 
unsigned MediaInfo::imageWidth(size_t streamNumber) const
{
    return UInt32::parse(
        MediaInfo_Get(__handle, MediaInfo_Stream_Image, streamNumber, L"Width",
            MediaInfo_Info_Text, MediaInfo_Info_Name),
        10, 0LLU
    );
}
 
unsigned MediaInfo::imageHeight(size_t streamNumber) const
{
    return UInt32::parse(
        MediaInfo_Get(__handle, MediaInfo_Stream_Image, streamNumber, L"Height",
            MediaInfo_Info_Text, MediaInfo_Info_Name),
        10, 0LLU
    );
}
 
static const wchar_t* __formatToType__[] = {
    L"ShockWave",       L"x-shockwave-flash"
    , L"Flash Video",   L"x-flv"
    , L"AVI",           L"avi"
    , L"MPEG Audio",    L"mpeg"
    , L"MPEG Video",    L"mpeg"
    , L"CDXA/MPEG-PS",  L"mp2p"
    , L"MPEG-PS",       L"mp2p"
    , L"MPEG-4",        L"mp4"
    , L"OGG",           L"ogg"
    , L"WebM",          L"webm"
};
 
String MediaInfo::contentType() const
{
    if (imageCount() > 0) {
        return L"image/" + format().toLowerCase();
    }
 
    String format = this->format();
    const wchar_t* type = NULL;
    for (size_t i = 0; i < __countof(__formatToType__, wchar_t*);
        i++, i++) {
        if (format.compareNoCase(__formatToType__[i]) == 0) {
            type = __formatToType__[i + 1];
            break;
        }
    }
 
    StringBuilder r;
    if (type != NULL) {
        if (videoCount() > 0) {
            r = L"video/";
        }
        else if (audioCount() > 0) {
            r = L"audio/";
        }
        else {
            r = L"application/";
        }
        r += type;
    }
    else {
        r = L"application/octet-stream";
    }
    return r;
}
 
__DCL_END_NAMESPACE