Aegisub/core/subtitle_provider_libass.cpp

180 lines
5.3 KiB
C++

// -*- c-basic-offset: 8; indent-tabs-mode: t -*-
// Copyright (c) 2006, Evgeniy Stepanov
// All rights reserved.
//
// Redistribution and use in source and binary forms, with or without
// modification, are permitted provided that the following conditions are met:
//
// * Redistributions of source code must retain the above copyright notice,
// this list of conditions and the following disclaimer.
// * Redistributions in binary form must reproduce the above copyright notice,
// this list of conditions and the following disclaimer in the documentation
// and/or other materials provided with the distribution.
// * Neither the name of the Aegisub Group nor the names of its contributors
// may be used to endorse or promote products derived from this software
// without specific prior written permission.
//
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
// POSSIBILITY OF SUCH DAMAGE.
//
// -----------------------------------------------------------------------------
//
// AEGISUB
//
// Website: http://aegisub.cellosoft.com
// Contact: mailto:zeratul@cellosoft.com
//
///////////
// Headers
#ifdef HAVE_LIBASS
#include <wx/wxprec.h>
#include <wx/image.h>
#include "subtitle_provider.h"
#include "video_provider.h"
#include "ass_file.h"
extern "C" {
#include <ass/ass.h>
}
class SubtitleProviderLibASS : public SubtitleProvider, SubtitleProvider::Overlay {
private:
class MyClass : public Class {
public:
MyClass() : Class(L"libass") {
ass_library = ass_library_init();
if (!ass_library) throw _T("ass_library_init failed");
ass_set_fonts_dir(ass_library, "");
ass_set_extract_fonts(ass_library, 0);
ass_set_style_overrides(ass_library, NULL);
};
virtual SubtitleProvider *Get(AssFile *subs) { return new SubtitleProviderLibASS(subs); };
};
static MyClass me;
static ass_library_t* ass_library;
ass_renderer_t* ass_renderer;
ass_track_t* ass_track;
wxString last_sub;
AssFile *subs;
VideoProvider *vpro;
public:
SubtitleProviderLibASS(AssFile *_subs);
virtual ~SubtitleProviderLibASS();
virtual void Bind(VideoProvider *_vpro);
virtual void SetParams(int width, int height);
virtual void Render(wxImage &frame, int ms);
virtual void Unbind();
};
SubtitleProviderLibASS::MyClass SubtitleProviderLibASS::me;
ass_library_t* SubtitleProviderLibASS::ass_library;
SubtitleProviderLibASS::SubtitleProviderLibASS(AssFile *_subs) : ass_track(NULL)
{
subs = _subs;
wxString text = subs->GetString();
ass_renderer = ass_renderer_init(ass_library);
if (!ass_renderer) throw _T("ass_renderer_init failed");
//ass_set_margins(ass_renderer, 0, 0, 0, 0);
//ass_set_use_margins(ass_renderer, 0);
ass_set_font_scale(ass_renderer, 1.);
ass_set_fonts(ass_renderer, NULL, "Sans");
};
SubtitleProviderLibASS::~SubtitleProviderLibASS()
{
Unbind();
//asa_close(inst);
};
void SubtitleProviderLibASS::Bind(VideoProvider *_vpro)
{
vpro = _vpro;
vpro->AttachOverlay(this);
}
void SubtitleProviderLibASS::SetParams(int width, int height)
{
ass_set_frame_size(ass_renderer, width, height);
}
#define _r(c) ((c)>>24)
#define _g(c) (((c)>>16)&0xFF)
#define _b(c) (((c)>>8)&0xFF)
#define _a(c) ((c)&0xFF)
static void blend_single(wxImage &frame, ass_image_t* img) {
unsigned char opacity = 255 - _a(img->color);
unsigned char r = _r(img->color);
unsigned char g = _g(img->color);
unsigned char b = _b(img->color);
int dst_stride = frame.GetWidth() * 3;
unsigned char* src;
unsigned char* dst;
src = img->bitmap;
dst = frame.GetData() + img->dst_y * dst_stride + img->dst_x * 3;
for (int y = 0; y < img->h; ++y) {
for (int x = 0; x < img->w; ++x) {
unsigned k = ((unsigned)src[x]) * opacity / 255;
// possible endianness problems
dst[x*3] = (k*r + (255-k)*dst[x*3]) / 255;
dst[x*3+1] = (k*g + (255-k)*dst[x*3+1]) / 255;
dst[x*3+2] = (k*b + (255-k)*dst[x*3+2]) / 255;
}
src += img->stride;
dst += dst_stride;
}
}
static void blend(wxImage &frame, ass_image_t* img) {
int cnt = 0;
while (img) {
blend_single(frame, img);
++cnt;
img = img->next;
}
}
void SubtitleProviderLibASS::Render(wxImage &frame, int ms)
{
wxString text = subs->GetString();
if (text != last_sub) { // almost always true because of "Video Position" header
if (ass_track)
ass_free_track(ass_track);
ass_track = ass_read_memory(ass_library, ((char*)text.GetData()) + 4, sizeof(wchar_t) * text.Length() - 4, "UCS-4LE");
}
if (!ass_track)
throw _T("libass parse error");
ass_image_t* img = ass_render_frame(ass_renderer, ass_track, ms);
blend(frame, img);
}
void SubtitleProviderLibASS::Unbind()
{
if (vpro)
vpro->AttachOverlay(NULL);
vpro = NULL;
}
#endif /* HAVE_LIBASS */