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 "filter_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         , prev_sync(NULL)
  55 {
  56         mlt_filter filter = get_filter();
  57         if ( filter ) {
  58                 // Set the mlt_filter child in case we choose to override virtual functions.
  59                 filter->child = this;
  60                 mlt_properties_set_data(mlt_global_properties(), "glslManager", this, 0,
  61                         (mlt_destructor) deleteManager, NULL);
  62
  63                 mlt_events_register( get_properties(), "init glsl", NULL );
  64                 mlt_events_register( get_properties(), "close glsl", NULL );
  65                 initEvent = listen("init glsl", this, (mlt_listener) GlslManager::onInit);
  66                 closeEvent = listen("close glsl", this, (mlt_listener) GlslManager::onClose);
  67         }
  68 }
  69
  70 GlslManager::~GlslManager()
  71 {
  72         mlt_log_debug(get_service(), "%s\n", __FUNCTION__);
  73         cleanupContext();
  74 // XXX If there is still a frame holding a reference to a texture after this
  75 // destructor is called, then it will crash in release_texture().
  76 //      while (texture_list.peek_back())
  77 //              delete (glsl_texture) texture_list.pop_back();
  78         delete initEvent;
  79         delete closeEvent;
  80         if (prev_sync != NULL) {
  81                 glDeleteSync( prev_sync );
  82         }
  83 }
  84
  85 GlslManager* GlslManager::get_instance()
  86 {
  87         return (GlslManager*) mlt_properties_get_data(mlt_global_properties(), "glslManager", 0);
  88 }
  89
  90 glsl_fbo GlslManager::get_fbo(int width, int height)
  91 {
  92 #if defined(__DARWIN__)
  93         CGLContextObj context = CGLGetCurrentContext();
  94 #elif defined(WIN32)
  95         HGLRC context = wglGetCurrentContext();
  96 #else
  97         GLXContext context = glXGetCurrentContext();
  98 #endif
  99
 100         lock();
 101         for (int i = 0; i < fbo_list.count(); ++i) {
 102                 glsl_fbo fbo = (glsl_fbo) fbo_list.peek(i);
 103                 if (!fbo->used && (fbo->width == width) && (fbo->height == height) && (fbo->context == context)) {
 104                         fbo->used = 1;
 105                         unlock();
 106                         return fbo;
 107                 }
 108         }
 109         unlock();
 110
 111         GLuint fb = 0;
 112         glGenFramebuffers(1, &fb);
 113         if (!fb)
 114                 return NULL;
 115
 116         glsl_fbo fbo = new glsl_fbo_s;
 117         if (!fbo) {
 118                 glDeleteFramebuffers(1, &fb);
 119                 return NULL;
 120         }
 121         fbo->fbo = fb;
 122         fbo->width = width;
 123         fbo->height = height;
 124         fbo->used = 1;
 125         fbo->context = context;
 126         lock();
 127         fbo_list.push_back(fbo);
 128         unlock();
 129         return fbo;
 130 }
 131
 132 void GlslManager::release_fbo(glsl_fbo fbo)
 133 {
 134         fbo->used = 0;
 135 }
 136
 137 glsl_texture GlslManager::get_texture(int width, int height, GLint internal_format)
 138 {
 139         lock();
 140         for (int i = 0; i < texture_list.count(); ++i) {
 141                 glsl_texture tex = (glsl_texture) texture_list.peek(i);
 142                 if (!tex->used && (tex->width == width) && (tex->height == height) && (tex->internal_format == internal_format)) {
 143                         glBindTexture(GL_TEXTURE_2D, tex->texture);
 144                         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST);
 145                         glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST);
 146                         glBindTexture( GL_TEXTURE_2D, 0);
 147                         tex->used = 1;
 148                         unlock();
 149                         return tex;
 150                 }
 151         }
 152
 153         // Recycle a glsl_texture with deleted glTexture.
 154         glsl_texture gtex = 0;
 155         for (int i = 0; i < texture_list.count(); ++i) {
 156                 glsl_texture tex = (glsl_texture) texture_list.peek(i);
 157                 if (!tex->used && !tex->width && !tex->height) {
 158                         gtex = tex;
 159                         break;
 160                 }
 161         }
 162         unlock();
 163
 164         GLuint tex = 0;
 165         glGenTextures(1, &tex);
 166         if (!tex) {
 167                 glDeleteTextures(1, &tex);
 168                 return NULL;
 169         }
 170
 171         if (!gtex) {
 172                 gtex = new glsl_texture_s;
 173                 if (!gtex)
 174                         return NULL;
 175         }
 176
 177         glBindTexture( GL_TEXTURE_2D, tex );
 178         glTexImage2D( GL_TEXTURE_2D, 0, internal_format, width, height, 0, internal_format, GL_UNSIGNED_BYTE, NULL );
 179     glTexParameterf( GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP_TO_EDGE );
 180     glTexParameterf( GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP_TO_EDGE );
 181     glTexParameteri( GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST );
 182     glTexParameteri( GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_NEAREST );
 183     glBindTexture( GL_TEXTURE_2D, 0 );
 184
 185         gtex->texture = tex;
 186         gtex->width = width;
 187         gtex->height = height;
 188         gtex->internal_format = internal_format;
 189         gtex->used = 1;
 190         lock();
 191         texture_list.push_back(gtex);
 192         unlock();
 193         return gtex;
 194 }
 195
 196 void GlslManager::release_texture(glsl_texture texture)
 197 {
 198         texture->used = 0;
 199 }
 200
 201 glsl_pbo GlslManager::get_pbo(int size)
 202 {
 203         lock();
 204         if (!pbo) {
 205                 GLuint pb = 0;
 206                 glGenBuffers(1, &pb);
 207                 if (!pb) {
 208                         unlock();
 209                         return NULL;
 210                 }
 211
 212                 pbo = new glsl_pbo_s;
 213                 if (!pbo) {
 214                         glDeleteBuffers(1, &pb);
 215                         unlock();
 216                         return NULL;
 217                 }
 218                 pbo->pbo = pb;
 219                 pbo->size = 0;
 220         }
 221         if (size > pbo->size) {
 222                 glBindBuffer(GL_PIXEL_UNPACK_BUFFER_ARB, pbo->pbo);
 223                 glBufferData(GL_PIXEL_UNPACK_BUFFER_ARB, size, NULL, GL_STREAM_DRAW);
 224                 glBindBuffer(GL_PIXEL_UNPACK_BUFFER_ARB, 0);
 225                 pbo->size = size;
 226         }
 227         unlock();
 228         return pbo;
 229 }
 230
 231 void GlslManager::cleanupContext()
 232 {
 233         lock();
 234         while (fbo_list.peek_back()) {
 235                 glsl_fbo fbo = (glsl_fbo) fbo_list.pop_back();
 236                 glDeleteFramebuffers(1, &fbo->fbo);
 237                 delete fbo;
 238         }
 239         for (int i = 0; i < texture_list.count(); ++i) {
 240                 glsl_texture texture = (glsl_texture) texture_list.peek(i);
 241                 glDeleteTextures(1, &texture->texture);
 242                 texture->used = 0;
 243                 texture->width = 0;
 244                 texture->height = 0;
 245         }
 246         if (pbo) {
 247                 glDeleteBuffers(1, &pbo->pbo);
 248                 delete pbo;
 249                 pbo = 0;
 250         }
 251         unlock();
 252 }
 253
 254 void GlslManager::onInit( mlt_properties owner, GlslManager* filter )
 255 {
 256         mlt_log_debug( filter->get_service(), "%s\n", __FUNCTION__ );
 257 #ifdef WIN32
 258         std::string path = std::string(mlt_environment("MLT_APPDIR")).append("\\share\\movit");
 259 #elif defined(__DARWIN__) && defined(RELOCATABLE)
 260         std::string path = std::string(mlt_environment("MLT_APPDIR")).append("/share/movit");
 261 #else
 262         std::string path = std::string(getenv("MLT_MOVIT_PATH") ? getenv("MLT_MOVIT_PATH") : SHADERDIR);
 263 #endif
 264         ::init_movit( path, mlt_log_get_level() == MLT_LOG_DEBUG? MOVIT_DEBUG_ON : MOVIT_DEBUG_OFF );
 265         filter->set( "glsl_supported", movit_initialized );
 266 }
 267
 268 void GlslManager::onClose( mlt_properties owner, GlslManager *filter )
 269 {
 270         filter->cleanupContext();
 271 }
 272
 273 void GlslManager::onServiceChanged( mlt_properties owner, mlt_service aservice )
 274 {
 275         Mlt::Service service( aservice );
 276         service.lock();
 277         service.set( "movit chain", NULL, 0 );
 278         service.set( "movit input", NULL, 0 );
 279         // Destroy the effect list.
 280         GlslManager::get_instance()->set( service.get( "_unique_id" ), NULL, 0 );
 281         service.unlock();
 282 }
 283
 284 void GlslManager::onPropertyChanged( mlt_properties owner, mlt_service service, const char* property )
 285 {
 286         if ( property && std::string( property ) == "disable" )
 287                 onServiceChanged( owner, service );
 288 }
 289
 290 extern "C" {
 291
 292 mlt_filter filter_glsl_manager_init( mlt_profile profile, mlt_service_type type, const char *id, char *arg )
 293 {
 294         GlslManager* g = GlslManager::get_instance();
 295         if (g)
 296                 g->inc_ref();
 297         else
 298                 g = new GlslManager();
 299         return g->get_filter();
 300 }
 301
 302 } // extern "C"
 303
 304 Mlt::Properties GlslManager::effect_list( Mlt::Service& service )
 305 {
 306         char *unique_id =  service.get( "_unique_id" );
 307         mlt_properties properties = (mlt_properties) get_data( unique_id );
 308         if ( !properties ) {
 309                 properties = mlt_properties_new();
 310                 set( unique_id, properties, 0, (mlt_destructor) mlt_properties_close );
 311         }
 312         Mlt::Properties p( properties );
 313         return p;
 314 }
 315
 316 static void deleteChain( EffectChain* chain )
 317 {
 318         delete chain;
 319 }
 320
 321 bool GlslManager::init_chain( mlt_service aservice )
 322 {
 323         bool error = true;
 324         Mlt::Service service( aservice );
 325         EffectChain* chain = (EffectChain*) service.get_data( "movit chain" );
 326         if ( !chain ) {
 327                 mlt_profile profile = mlt_service_profile( aservice );
 328                 Input* input = new MltInput( profile->width, profile->height );
 329                 chain = new EffectChain( profile->display_aspect_num, profile->display_aspect_den );
 330                 chain->add_input( input );
 331                 service.set( "movit chain", chain, 0, (mlt_destructor) deleteChain );
 332                 service.set( "movit input", input, 0 );
 333                 service.set( "_movit finalized", 0 );
 334                 service.listen( "service-changed", aservice, (mlt_listener) GlslManager::onServiceChanged );
 335                 service.listen( "property-changed", aservice, (mlt_listener) GlslManager::onPropertyChanged );
 336                 error = false;
 337         }
 338         return error;
 339 }
 340
 341 EffectChain* GlslManager::get_chain( mlt_service service )
 342 {
 343         return (EffectChain*) mlt_properties_get_data( MLT_SERVICE_PROPERTIES(service), "movit chain", NULL );
 344 }
 345
 346 MltInput *GlslManager::get_input( mlt_service service )
 347 {
 348         return (MltInput*) mlt_properties_get_data( MLT_SERVICE_PROPERTIES(service), "movit input", NULL );
 349 }
 350
 351 void GlslManager::reset_finalized( mlt_service service )
 352 {
 353         mlt_properties_set_int( MLT_SERVICE_PROPERTIES(service), "_movit finalized", 0 );
 354 }
 355
 356 Effect* GlslManager::get_effect( mlt_service service, mlt_frame frame )
 357 {
 358         Mlt::Producer producer( mlt_producer_cut_parent( mlt_frame_get_original_producer( frame ) ) );
 359         char *unique_id = mlt_properties_get( MLT_SERVICE_PROPERTIES(service), "_unique_id" );
 360         return (Effect*) GlslManager::get_instance()->effect_list( producer ).get_data( unique_id );
 361 }
 362
 363 Effect* GlslManager::add_effect( mlt_service service, mlt_frame frame, Effect* effect )
 364 {
 365         Mlt::Producer producer( mlt_producer_cut_parent( mlt_frame_get_original_producer( frame ) ) );
 366         EffectChain* chain = (EffectChain*) producer.get_data( "movit chain" );
 367         chain->add_effect( effect );
 368         char *unique_id = mlt_properties_get( MLT_SERVICE_PROPERTIES(service), "_unique_id" );
 369         GlslManager::get_instance()->effect_list( producer ).set( unique_id, effect, 0 );
 370         return effect;
 371 }
 372
 373 Effect* GlslManager::add_effect( mlt_service service, mlt_frame frame, Effect* effect, Effect* input_b )
 374 {
 375         Mlt::Producer producer( mlt_producer_cut_parent( mlt_frame_get_original_producer( frame ) ) );
 376         EffectChain* chain = (EffectChain*) producer.get_data( "movit chain" );
 377         chain->add_effect( effect, chain->last_added_effect(),
 378                 input_b? input_b : chain->last_added_effect() );
 379         char *unique_id = mlt_properties_get( MLT_SERVICE_PROPERTIES(service), "_unique_id" );
 380         GlslManager::get_instance()->effect_list( producer ).set( unique_id, effect, 0 );
 381         return effect;
 382 }
 383
 384 void GlslManager::render_fbo( mlt_service service, void* chain, GLuint fbo, int width, int height )
 385 {
 386         EffectChain* effect_chain = (EffectChain*) chain;
 387         mlt_properties properties = MLT_SERVICE_PROPERTIES( service );
 388         if ( !mlt_properties_get_int( properties, "_movit finalized" ) ) {
 389                 mlt_properties_set_int( properties, "_movit finalized", 1 );
 390                 effect_chain->add_effect( new Mlt::VerticalFlip() );
 391                 effect_chain->finalize();
 392         }
 393         effect_chain->render_to_fbo( fbo, width, height );
 394 }
 395
 396 int GlslManager::render_frame_texture(mlt_service service, mlt_frame frame, int width, int height, uint8_t **image)
 397 {
 398         EffectChain* chain = get_chain( service );
 399         if (!chain) return 1;
 400         glsl_fbo fbo = get_fbo( width, height );
 401         if (!fbo) return 1;
 402         glsl_texture texture = get_texture( width, height, GL_RGBA );
 403         if (!texture) {
 404                 release_fbo( fbo );
 405                 return 1;
 406         }
 407
 408         glBindFramebuffer( GL_FRAMEBUFFER, fbo->fbo );
 409         check_error();
 410         glFramebufferTexture2D( GL_FRAMEBUFFER, GL_COLOR_ATTACHMENT0, GL_TEXTURE_2D, texture->texture, 0 );
 411         check_error();
 412         glBindFramebuffer( GL_FRAMEBUFFER, 0 );
 413         check_error();
 414
 415         // Make sure we never have more than one frame pending at any time.
 416         // This ensures we do not swamp the GPU with so much work
 417         // that we cannot actually display the frames we generate.
 418         if (prev_sync != NULL) {
 419                 glFlush();
 420                 glClientWaitSync( prev_sync, 0, GL_TIMEOUT_IGNORED );
 421                 glDeleteSync( prev_sync );
 422         }
 423         render_fbo( service, chain, fbo->fbo, width, height );
 424         prev_sync = glFenceSync( GL_SYNC_GPU_COMMANDS_COMPLETE, 0 );
 425         GLsync sync = glFenceSync( GL_SYNC_GPU_COMMANDS_COMPLETE, 0 );
 426
 427         check_error();
 428         glBindFramebuffer( GL_FRAMEBUFFER, 0 );
 429         check_error();
 430         release_fbo( fbo );
 431
 432         *image = (uint8_t*) &texture->texture;
 433         mlt_frame_set_image( frame, *image, 0, NULL );
 434         mlt_properties_set_data( MLT_FRAME_PROPERTIES(frame), "movit.convert.texture", texture, 0,
 435                 (mlt_destructor) GlslManager::release_texture, NULL );
 436         mlt_properties_set_data( MLT_FRAME_PROPERTIES(frame), "movit.convert.fence", sync, 0, NULL, NULL );
 437
 438         return 0;
 439 }
 440
 441 int GlslManager::render_frame_rgba(mlt_service service, mlt_frame frame, int width, int height, uint8_t **image)
 442 {
 443         EffectChain* chain = get_chain( service );
 444         if (!chain) return 1;
 445         glsl_fbo fbo = get_fbo( width, height );
 446         if (!fbo) return 1;
 447         glsl_texture texture = get_texture( width, height, GL_RGBA );
 448         if (!texture) {
 449                 release_fbo( fbo );
 450                 return 1;
 451         }
 452
 453         // Use a PBO to hold the data we read back with glReadPixels().
 454         // (Intel/DRI goes into a slow path if we don't read to PBO.)
 455         int img_size = width * height * 4;
 456         glsl_pbo pbo = get_pbo( img_size );
 457         if (!pbo) {
 458                 release_fbo( fbo );
 459                 release_texture(texture);
 460                 return 1;
 461         }
 462
 463         // Set the FBO
 464         check_error();
 465         glBindFramebuffer( GL_FRAMEBUFFER, fbo->fbo );
 466         check_error();
 467         glFramebufferTexture2D( GL_FRAMEBUFFER, GL_COLOR_ATTACHMENT0, GL_TEXTURE_2D, texture->texture, 0 );
 468         check_error();
 469         glBindFramebuffer( GL_FRAMEBUFFER, 0 );
 470         check_error();
 471
 472         render_fbo( service, chain, fbo->fbo, width, height );
 473
 474         // Read FBO into PBO
 475         glBindFramebuffer( GL_FRAMEBUFFER, fbo->fbo );
 476         check_error();
 477         glBindBuffer( GL_PIXEL_PACK_BUFFER_ARB, pbo->pbo );
 478         check_error();
 479         glBufferData( GL_PIXEL_PACK_BUFFER_ARB, img_size, NULL, GL_STREAM_READ );
 480         check_error();
 481         glReadPixels( 0, 0, width, height, GL_BGRA, GL_UNSIGNED_BYTE, BUFFER_OFFSET(0) );
 482         check_error();
 483
 484         // Copy from PBO
 485         uint8_t* buf = (uint8_t*) glMapBuffer( GL_PIXEL_PACK_BUFFER_ARB, GL_READ_ONLY );
 486         check_error();
 487         *image = (uint8_t*) mlt_pool_alloc( img_size );
 488         mlt_frame_set_image( frame, *image, img_size, mlt_pool_release );
 489         memcpy( *image, buf, img_size );
 490
 491         // Convert BGRA to RGBA
 492         register uint8_t *p = *image;
 493         register int n = width * height + 1;
 494         while ( --n ) {
 495                 uint8_t b = p[0];
 496                 *p = p[2]; p += 2;
 497                 *p = b; p += 2;
 498         }
 499
 500         // Release PBO and FBO
 501         glUnmapBuffer( GL_PIXEL_PACK_BUFFER_ARB );
 502         check_error();
 503         glBindBuffer( GL_PIXEL_PACK_BUFFER_ARB, 0 );
 504         check_error();
 505         glBindFramebuffer( GL_FRAMEBUFFER, 0 );
 506         check_error();
 507         glBindTexture( GL_TEXTURE_2D, 0 );
 508         check_error();
 509         mlt_properties_set_data( MLT_FRAME_PROPERTIES(frame), "movit.convert.texture", texture, 0,
 510                 (mlt_destructor) GlslManager::release_texture, NULL);
 511         release_fbo( fbo );
 512
 513         return 0;
 514 }
 515
 516 void GlslManager::lock_service( mlt_frame frame )
 517 {
 518         Mlt::Producer producer( mlt_producer_cut_parent( mlt_frame_get_original_producer( frame ) ) );
 519         producer.lock();
 520 }
 521
 522 void GlslManager::unlock_service( mlt_frame frame )
 523 {
 524         Mlt::Producer producer( mlt_producer_cut_parent( mlt_frame_get_original_producer( frame ) ) );
 525         producer.unlock();
 526 }