git.sesse.net Git - mlt/blob - src/modules/opengl/filter_glsl_manager.cpp

   1 /*
   2  * filter_glsl_manager.cpp
   3  * Copyright (C) 2011-2012 Christophe Thommeret <hftom@free.fr>
   4  * Copyright (C) 2013 Dan Dennedy <dan@dennedy.org>
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software Foundation,
  18  * Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  19  */
  20
  21 #include <stdlib.h>
  22 #include <string>
  23 #include "glsl_manager.h"
  24 #include <movit/init.h>
  25 #include <movit/util.h>
  26 #include <movit/effect_chain.h>
  27 #include "mlt_movit_input.h"
  28 #include "mlt_flip_effect.h"
  29 #include <mlt++/MltEvent.h>
  30 #include <mlt++/MltProducer.h>
  31
  32 extern "C" {
  33 #include <framework/mlt_factory.h>
  34 }
  35
  36 #if defined(__DARWIN__)
  37 #include <OpenGL/OpenGL.h>
  38 #elif defined(WIN32)
  39 #include <wingdi.h>
  40 #else
  41 #include <GL/glx.h>
  42 #endif
  43
  44 void deleteManager(GlslManager *p)
  45 {
  46         delete p;
  47 }
  48
  49 GlslManager::GlslManager()
  50         : Mlt::Filter( mlt_filter_new() )
  51         , pbo(0)
  52         , initEvent(0)
  53         , closeEvent(0)
  54 {
  55         mlt_filter filter = get_filter();
  56         if ( filter ) {
  57                 // Set the mlt_filter child in case we choose to override virtual functions.
  58                 filter->child = this;
  59                 mlt_properties_set_data(mlt_global_properties(), "glslManager", this, 0,
  60                         (mlt_destructor) deleteManager, NULL);
  61
  62                 mlt_events_register( get_properties(), "init glsl", NULL );
  63                 mlt_events_register( get_properties(), "close glsl", NULL );
  64                 initEvent = listen("init glsl", this, (mlt_listener) GlslManager::onInit);
  65                 closeEvent = listen("close glsl", this, (mlt_listener) GlslManager::onClose);
  66         }
  67 }
  68
  69 GlslManager::~GlslManager()
  70 {
  71         mlt_log_debug(get_service(), "%s\n", __FUNCTION__);
  72         cleanupContext();
  73         delete initEvent;
  74         delete closeEvent;
  75 }
  76
  77 GlslManager* GlslManager::get_instance()
  78 {
  79         return (GlslManager*) mlt_properties_get_data(mlt_global_properties(), "glslManager", 0);
  80 }
  81
  82 glsl_fbo GlslManager::get_fbo(int width, int height)
  83 {
  84 #if defined(__DARWIN__)
  85         CGLContextObj context = CGLGetCurrentContext();
  86 #elif defined(WIN32)
  87         HGLRC context = wglGetCurrentContext();
  88 #else
  89         GLXContext context = glXGetCurrentContext();
  90 #endif
  91
  92         lock();
  93         for (int i = 0; i < fbo_list.count(); ++i) {
  94                 glsl_fbo fbo = (glsl_fbo) fbo_list.peek(i);
  95                 if (!fbo->used && (fbo->width == width) && (fbo->height == height) && (fbo->context == context)) {
  96                         fbo->used = 1;
  97                         unlock();
  98                         return fbo;
  99                 }
 100         }
 101         unlock();
 102
 103         GLuint fb = 0;
 104         glGenFramebuffers(1, &fb);
 105         if (!fb)
 106                 return NULL;
 107
 108         glsl_fbo fbo = new glsl_fbo_s;
 109         if (!fbo) {
 110                 glDeleteFramebuffers(1, &fb);
 111                 return NULL;
 112         }
 113         fbo->fbo = fb;
 114         fbo->width = width;
 115         fbo->height = height;
 116         fbo->used = 1;
 117         fbo->context = context;
 118         lock();
 119         fbo_list.push_back(fbo);
 120         unlock();
 121         return fbo;
 122 }
 123
 124 void GlslManager::release_fbo(glsl_fbo fbo)
 125 {
 126         fbo->used = 0;
 127 }
 128
 129 glsl_texture GlslManager::get_texture(int width, int height, GLint internal_format)
 130 {
 131         lock();
 132         for (int i = 0; i < texture_list.count(); ++i) {
 133                 glsl_texture tex = (glsl_texture) texture_list.peek(i);
 134                 if (!tex->used && (tex->width == width) && (tex->height == height) && (tex->internal_format == internal_format)) {
 135                         glBindTexture(GL_TEXTURE_2D, tex->texture);
 136                         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST);
 137                         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST);
 138                         glBindTexture( GL_TEXTURE_2D, 0);
 139                         tex->used = 1;
 140                         unlock();
 141                         return tex;
 142                 }
 143         }
 144         unlock();
 145
 146         GLuint tex = 0;
 147         glGenTextures(1, &tex);
 148         if (!tex)
 149                 return NULL;
 150
 151         glsl_texture gtex = new glsl_texture_s;
 152         if (!gtex) {
 153                 glDeleteTextures(1, &tex);
 154                 return NULL;
 155         }
 156         glBindTexture( GL_TEXTURE_2D, tex );
 157         glTexImage2D( GL_TEXTURE_2D, 0, internal_format, width, height, 0, internal_format, GL_UNSIGNED_BYTE, NULL );
 158     glTexParameterf( GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE );
 159     glTexParameterf( GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE );
 160     glTexParameteri( GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST );
 161     glTexParameteri( GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST );
 162     glBindTexture( GL_TEXTURE_2D, 0 );
 163
 164         gtex->texture = tex;
 165         gtex->width = width;
 166         gtex->height = height;
 167         gtex->internal_format = internal_format;
 168         gtex->used = 1;
 169         lock();
 170         texture_list.push_back(gtex);
 171         unlock();
 172         return gtex;
 173 }
 174
 175 void GlslManager::release_texture(glsl_texture texture)
 176 {
 177         texture->used = 0;
 178 }
 179
 180 glsl_pbo GlslManager::get_pbo(int size)
 181 {
 182         lock();
 183         if (!pbo) {
 184                 GLuint pb = 0;
 185                 glGenBuffers(1, &pb);
 186                 if (!pb) {
 187                         unlock();
 188                         return NULL;
 189                 }
 190
 191                 pbo = new glsl_pbo_s;
 192                 if (!pbo) {
 193                         glDeleteBuffers(1, &pb);
 194                         unlock();
 195                         return NULL;
 196                 }
 197                 pbo->pbo = pb;
 198                 pbo->size = 0;
 199         }
 200         if (size > pbo->size) {
 201                 glBindBuffer(GL_PIXEL_UNPACK_BUFFER_ARB, pbo->pbo);
 202                 glBufferData(GL_PIXEL_UNPACK_BUFFER_ARB, size, NULL, GL_STREAM_DRAW);
 203                 glBindBuffer(GL_PIXEL_UNPACK_BUFFER_ARB, 0);
 204                 pbo->size = size;
 205         }
 206         unlock();
 207         return pbo;
 208 }
 209
 210 void GlslManager::cleanupContext()
 211 {
 212         lock();
 213         while (fbo_list.peek_back()) {
 214                 glsl_fbo fbo = (glsl_fbo) fbo_list.pop_back();
 215                 glDeleteFramebuffers(1, &fbo->fbo);
 216                 delete fbo;
 217         }
 218         while (texture_list.peek_back()) {
 219                 glsl_texture texture = (glsl_texture) texture_list.pop_back();
 220                 glDeleteTextures(1, &texture->texture);
 221                 delete texture;
 222         }
 223         if (pbo) {
 224                 glDeleteBuffers(1, &pbo->pbo);
 225                 delete pbo;
 226                 pbo = 0;
 227         }
 228         unlock();
 229 }
 230
 231 void GlslManager::onInit( mlt_properties owner, GlslManager* filter )
 232 {
 233         mlt_log_debug( filter->get_service(), "%s\n", __FUNCTION__ );
 234 #ifdef WIN32
 235         std::string path = std::string(mlt_environment("MLT_APPDIR")).append("\\share\\movit");
 236 #elif defined(__DARWIN__) && defined(RELOCATABLE)
 237         std::string path = std::string(mlt_environment("MLT_APPDIR")).append("/share/movit");
 238 #else
 239         std::string path = std::string(getenv("MLT_MOVIT_PATH") ? getenv("MLT_MOVIT_PATH") : SHADERDIR);
 240 #endif
 241         ::init_movit( path, mlt_log_get_level() == MLT_LOG_DEBUG? MOVIT_DEBUG_ON : MOVIT_DEBUG_OFF );
 242         filter->set( "glsl_supported", movit_initialized );
 243 }
 244
 245 void GlslManager::onClose( mlt_properties owner, GlslManager *filter )
 246 {
 247         filter->cleanupContext();
 248 }
 249
 250 void GlslManager::onServiceChanged( mlt_properties owner, mlt_service aservice )
 251 {
 252         Mlt::Service service( aservice );
 253         service.lock();
 254         service.set( "movit chain", NULL, 0 );
 255         service.set( "movit input", NULL, 0 );
 256         // Destroy the effect list.
 257         GlslManager::get_instance()->set( service.get( "_unique_id" ), NULL, 0 );
 258         service.unlock();
 259 }
 260
 261 void GlslManager::onPropertyChanged( mlt_properties owner, mlt_service service, const char* property )
 262 {
 263         if ( property && std::string( property ) == "disable" )
 264                 onServiceChanged( owner, service );
 265 }
 266
 267 extern "C" {
 268
 269 mlt_filter filter_glsl_manager_init( mlt_profile profile, mlt_service_type type, const char *id, char *arg )
 270 {
 271         GlslManager* g = GlslManager::get_instance();
 272         if (g)
 273                 g->inc_ref();
 274         else
 275                 g = new GlslManager();
 276         return g->get_filter();
 277 }
 278
 279 } // extern "C"
 280
 281 Mlt::Properties GlslManager::effect_list( Mlt::Service& service )
 282 {
 283         char *unique_id =  service.get( "_unique_id" );
 284         mlt_properties properties = (mlt_properties) get_data( unique_id );
 285         if ( !properties ) {
 286                 properties = mlt_properties_new();
 287                 set( unique_id, properties, 0, (mlt_destructor) mlt_properties_close );
 288         }
 289         Mlt::Properties p( properties );
 290         return p;
 291 }
 292
 293 static void deleteChain( EffectChain* chain )
 294 {
 295         delete chain;
 296 }
 297
 298 bool GlslManager::init_chain( mlt_service aservice )
 299 {
 300         bool error = true;
 301         Mlt::Service service( aservice );
 302         EffectChain* chain = (EffectChain*) service.get_data( "movit chain" );
 303         if ( !chain ) {
 304                 mlt_profile profile = mlt_service_profile( aservice );
 305                 Input* input = new MltInput( profile->width, profile->height );
 306                 chain = new EffectChain( profile->display_aspect_num, profile->display_aspect_den );
 307                 chain->add_input( input );
 308                 service.set( "movit chain", chain, 0, (mlt_destructor) deleteChain );
 309                 service.set( "movit input", input, 0 );
 310                 service.set( "_movit finalized", 0 );
 311                 service.listen( "service-changed", aservice, (mlt_listener) GlslManager::onServiceChanged );
 312                 service.listen( "property-changed", aservice, (mlt_listener) GlslManager::onPropertyChanged );
 313                 error = false;
 314         }
 315         return error;
 316 }
 317
 318 EffectChain* GlslManager::get_chain( mlt_service service )
 319 {
 320         return (EffectChain*) mlt_properties_get_data( MLT_SERVICE_PROPERTIES(service), "movit chain", NULL );
 321 }
 322
 323 MltInput *GlslManager::get_input( mlt_service service )
 324 {
 325         return (MltInput*) mlt_properties_get_data( MLT_SERVICE_PROPERTIES(service), "movit input", NULL );
 326 }
 327
 328 void GlslManager::reset_finalized( mlt_service service )
 329 {
 330         mlt_properties_set_int( MLT_SERVICE_PROPERTIES(service), "_movit finalized", 0 );
 331 }
 332
 333 Effect* GlslManager::get_effect( mlt_filter filter, mlt_frame frame )
 334 {
 335         Mlt::Producer producer( mlt_producer_cut_parent( mlt_frame_get_original_producer( frame ) ) );
 336         char *unique_id = mlt_properties_get( MLT_FILTER_PROPERTIES(filter), "_unique_id" );
 337         return (Effect*) GlslManager::get_instance()->effect_list( producer ).get_data( unique_id );
 338 }
 339
 340 Effect* GlslManager::add_effect( mlt_filter filter, mlt_frame frame, Effect* effect )
 341 {
 342         Mlt::Producer producer( mlt_producer_cut_parent( mlt_frame_get_original_producer( frame ) ) );
 343         EffectChain* chain = (EffectChain*) producer.get_data( "movit chain" );
 344         chain->add_effect( effect );
 345         char *unique_id = mlt_properties_get( MLT_FILTER_PROPERTIES(filter), "_unique_id" );
 346         GlslManager::get_instance()->effect_list( producer ).set( unique_id, effect, 0 );
 347         return effect;
 348 }
 349
 350 Effect* GlslManager::add_effect( mlt_filter filter, mlt_frame frame, Effect* effect, Effect* input_b )
 351 {
 352         Mlt::Producer producer( mlt_producer_cut_parent( mlt_frame_get_original_producer( frame ) ) );
 353         EffectChain* chain = (EffectChain*) producer.get_data( "movit chain" );
 354         chain->add_effect( effect, chain->last_added_effect(),
 355                 input_b? input_b : chain->last_added_effect() );
 356         char *unique_id = mlt_properties_get( MLT_FILTER_PROPERTIES(filter), "_unique_id" );
 357         GlslManager::get_instance()->effect_list( producer ).set( unique_id, effect, 0 );
 358         return effect;
 359 }
 360
 361 void GlslManager::render_fbo( mlt_service service, void* chain, GLuint fbo, int width, int height )
 362 {
 363         EffectChain* effect_chain = (EffectChain*) chain;
 364         mlt_properties properties = MLT_SERVICE_PROPERTIES( service );
 365         if ( !mlt_properties_get_int( properties, "_movit finalized" ) ) {
 366                 mlt_properties_set_int( properties, "_movit finalized", 1 );
 367                 effect_chain->add_effect( new Mlt::VerticalFlip() );
 368                 effect_chain->finalize();
 369         }
 370         effect_chain->render_to_fbo( fbo, width, height );
 371 }
 372
 373 int GlslManager::render_frame_texture(mlt_service service, mlt_frame frame, int width, int height, uint8_t **image)
 374 {
 375         EffectChain* chain = get_chain( service );
 376         if (!chain) return 1;
 377         glsl_fbo fbo = get_fbo( width, height );
 378         if (!fbo) return 1;
 379         glsl_texture texture = get_texture( width, height, GL_RGBA );
 380         if (!texture) {
 381                 release_fbo( fbo );
 382                 return 1;
 383         }
 384
 385         glBindFramebuffer( GL_FRAMEBUFFER, fbo->fbo );
 386         check_error();
 387         glFramebufferTexture2D( GL_FRAMEBUFFER, GL_COLOR_ATTACHMENT0, GL_TEXTURE_2D, texture->texture, 0 );
 388         check_error();
 389         glBindFramebuffer( GL_FRAMEBUFFER, 0 );
 390         check_error();
 391
 392         render_fbo( service, chain, fbo->fbo, width, height );
 393
 394         glFinish();
 395         check_error();
 396         glBindFramebuffer( GL_FRAMEBUFFER, 0 );
 397         check_error();
 398         release_fbo( fbo );
 399
 400         *image = (uint8_t*) &texture->texture;
 401         mlt_frame_set_image( frame, *image, 0, NULL );
 402         mlt_properties_set_data( MLT_FRAME_PROPERTIES(frame), "movit.convert.texture", texture, 0,
 403                 (mlt_destructor) GlslManager::release_texture, NULL );
 404
 405         return 0;
 406 }
 407
 408 int GlslManager::render_frame_rgba(mlt_service service, mlt_frame frame, int width, int height, uint8_t **image)
 409 {
 410         EffectChain* chain = get_chain( service );
 411         if (!chain) return 1;
 412         glsl_fbo fbo = get_fbo( width, height );
 413         if (!fbo) return 1;
 414         glsl_texture texture = get_texture( width, height, GL_RGBA );
 415         if (!texture) {
 416                 release_fbo( fbo );
 417                 return 1;
 418         }
 419
 420         // Use a PBO to hold the data we read back with glReadPixels().
 421         // (Intel/DRI goes into a slow path if we don't read to PBO.)
 422         int img_size = width * height * 4;
 423         glsl_pbo pbo = get_pbo( img_size );
 424         if (!pbo) {
 425                 release_fbo( fbo );
 426                 release_texture(texture);
 427                 return 1;
 428         }
 429
 430         // Set the FBO
 431         check_error();
 432         glBindFramebuffer( GL_FRAMEBUFFER, fbo->fbo );
 433         check_error();
 434         glFramebufferTexture2D( GL_FRAMEBUFFER, GL_COLOR_ATTACHMENT0, GL_TEXTURE_2D, texture->texture, 0 );
 435         check_error();
 436         glBindFramebuffer( GL_FRAMEBUFFER, 0 );
 437         check_error();
 438
 439         render_fbo( service, chain, fbo->fbo, width, height );
 440
 441         // Read FBO into PBO
 442         glBindFramebuffer( GL_FRAMEBUFFER, fbo->fbo );
 443         check_error();
 444         glBindBuffer( GL_PIXEL_PACK_BUFFER_ARB, pbo->pbo );
 445         check_error();
 446         glBufferData( GL_PIXEL_PACK_BUFFER_ARB, img_size, NULL, GL_STREAM_READ );
 447         check_error();
 448         glReadPixels( 0, 0, width, height, GL_BGRA, GL_UNSIGNED_BYTE, BUFFER_OFFSET(0) );
 449         check_error();
 450
 451         // Copy from PBO
 452         uint8_t* buf = (uint8_t*) glMapBuffer( GL_PIXEL_PACK_BUFFER_ARB, GL_READ_ONLY );
 453         check_error();
 454         *image = (uint8_t*) mlt_pool_alloc( img_size );
 455         mlt_frame_set_image( frame, *image, img_size, mlt_pool_release );
 456         memcpy( *image, buf, img_size );
 457
 458         // Convert BGRA to RGBA
 459         register uint8_t *p = *image;
 460         register int n = width * height + 1;
 461         while ( --n ) {
 462                 uint8_t b = p[0];
 463                 *p = p[2]; p += 2;
 464                 *p = b; p += 2;
 465         }
 466
 467         // Release PBO and FBO
 468         glUnmapBuffer( GL_PIXEL_PACK_BUFFER_ARB );
 469         check_error();
 470         glBindBuffer( GL_PIXEL_PACK_BUFFER_ARB, 0 );
 471         check_error();
 472         glBindFramebuffer( GL_FRAMEBUFFER, 0 );
 473         check_error();
 474         glBindTexture( GL_TEXTURE_2D, 0 );
 475         check_error();
 476         mlt_properties_set_data( MLT_FRAME_PROPERTIES(frame), "movit.convert.texture", texture, 0,
 477                 (mlt_destructor) GlslManager::release_texture, NULL);
 478         release_fbo( fbo );
 479
 480         return 0;
 481 }
 482
 483 void GlslManager::lock_service( mlt_frame frame )
 484 {
 485         Mlt::Producer producer( mlt_producer_cut_parent( mlt_frame_get_original_producer( frame ) ) );
 486         producer.lock();
 487 }
 488
 489 void GlslManager::unlock_service( mlt_frame frame )
 490 {
 491         Mlt::Producer producer( mlt_producer_cut_parent( mlt_frame_get_original_producer( frame ) ) );
 492         producer.unlock();
 493 }