research.m1stereo.tv/gitweb/ - melted/blob - src/modules/valerie/consumer_valerie.c

   1 /*
   2  * consumer_westley.c -- a libxml2 serialiser of mlt service networks
   3  * Copyright (C) 2003-2004 Ushodaya Enterprises Limited
   4  * Author: 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 "consumer_valerie.h"
  22 #include <valerie/valerie.h>
  23 #include <valerie/valerie_remote.h>
  24 #include <framework/mlt.h>
  25 #include <stdio.h>
  26 #include <stdlib.h>
  27 #include <string.h>
  28 #include <pthread.h>
  29 #include <unistd.h>
  30
  31 static int consumer_is_stopped( mlt_consumer this );
  32 static int consumer_start( mlt_consumer this );
  33
  34 /** This is what will be called by the factory
  35 */
  36
  37 mlt_consumer consumer_valerie_init( char *arg )
  38 {
  39         // Create the consumer object
  40         mlt_consumer this = calloc( sizeof( struct mlt_consumer_s ), 1 );
  41
  42         // If no malloc'd and consumer init ok
  43         if ( this != NULL && mlt_consumer_init( this, NULL ) == 0 )
  44         {
  45                 if ( arg != NULL && strchr( arg, ':' ) )
  46                 {
  47                         char *temp = NULL;
  48                         int port = atoi( strchr( arg, ':' ) + 1 );
  49                         mlt_properties_set( mlt_consumer_properties( this ), "server", arg );
  50                         temp = mlt_properties_get( mlt_consumer_properties( this ), "server" );
  51                         *( strchr( temp, ':' ) ) = '\0';
  52                         mlt_properties_set_int( mlt_consumer_properties( this ), "port", port );
  53                 }
  54                 else
  55                 {
  56                         mlt_properties_set( mlt_consumer_properties( this ), "server", arg == NULL ? "localhost" : arg );
  57                         mlt_properties_set_int( mlt_consumer_properties( this ), "port", 5250 );
  58                 }
  59
  60                 mlt_properties_set_int( mlt_consumer_properties( this ), "unit", 0 );
  61                 mlt_properties_set( mlt_consumer_properties( this ), "command", "append" );
  62
  63                 // Allow thread to be started/stopped
  64                 this->start = consumer_start;
  65                 this->is_stopped = consumer_is_stopped;
  66
  67                 // Return the consumer produced
  68                 return this;
  69         }
  70
  71         // malloc or consumer init failed
  72         free( this );
  73
  74         // Indicate failure
  75         return NULL;
  76 }
  77
  78 static int consumer_start( mlt_consumer this )
  79 {
  80         // Get the producer service
  81         mlt_service service = mlt_service_producer( mlt_consumer_service( this ) );
  82
  83         // Get the properties object
  84         mlt_properties properties = mlt_consumer_properties( this );
  85
  86         // Get all the properties now
  87         char *server = mlt_properties_get( properties, "server" );
  88         int port = mlt_properties_get_int( properties, "port" );
  89         char *command = mlt_properties_get( properties, "command" );
  90         int unit = mlt_properties_get_int( properties, "unit" );
  91         char *title = mlt_properties_get( properties, "title" );
  92
  93         // If this is a reuse, then a valerie object will exist
  94         valerie connection = mlt_properties_get_data( properties, "connection", NULL );
  95
  96         // Special case - we can get a doc too...
  97         char *doc = mlt_properties_get( properties, "westley" );
  98
  99         if ( service != NULL || doc != NULL )
 100         {
 101                 // Initiate the connection if required
 102                 if ( connection == NULL )
 103                 {
 104                         valerie_parser parser = valerie_parser_init_remote( server, port );
 105                         connection = valerie_init( parser );
 106                         if ( valerie_connect( connection ) == valerie_ok )
 107                         {
 108                                 mlt_properties_set_data( properties, "connection", connection, 0, ( mlt_destructor )valerie_close, NULL );
 109                                 mlt_properties_set_data( properties, "parser", parser, 0, ( mlt_destructor )valerie_parser_close, NULL );
 110                         }
 111                         else
 112                         {
 113                                 fprintf( stderr, "Unable to connect to the server at %s:%d\n", server, port );
 114                                 mlt_properties_set_int( properties, "_error", 1 );
 115                                 valerie_close( connection );
 116                                 valerie_parser_close( parser );
 117                                 connection = NULL;
 118                         }
 119                 }
 120
 121                 // If we have connection, push the service over
 122                 if ( connection != NULL )
 123                 {
 124                         if ( doc == NULL )
 125                         {
 126                                 int error;
 127
 128                                 // Set the title if provided
 129                                 if ( title != NULL )
 130                                         mlt_properties_set( mlt_service_properties( service ), "title", title );
 131                                 else if ( mlt_properties_get( mlt_service_properties( service ), "title" ) == NULL )
 132                                         mlt_properties_set( mlt_service_properties( service ), "title", "Anonymous Submission" );
 133
 134                                 // Push the service
 135                                 error = valerie_unit_push( connection, unit, command, service );
 136
 137                                 // Report error
 138                                 if ( error != valerie_ok )
 139                                         fprintf( stderr, "Push failed on %s:%d %s u%d (%d)\n", server, port, command, unit, error );
 140                         }
 141                         else
 142                         {
 143                                 // Push the service
 144                                 int error = valerie_unit_receive( connection, unit, command, doc );
 145
 146                                 // Report error
 147                                 if ( error != valerie_ok )
 148                                         fprintf( stderr, "Send failed on %s:%d %s u%d (%d)\n", server, port, command, unit, error );
 149                         }
 150                 }
 151         }
 152
 153         mlt_consumer_stop( this );
 154         mlt_consumer_stopped( this );
 155
 156         return 0;
 157 }
 158
 159 static int consumer_is_stopped( mlt_consumer this )
 160 {
 161         return 1;
 162 }