/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ /* * RTVideoRec Realtime video recoder and encoder for Linux * * Copyright (C) 2004 Bent Bisballe * Copyright (C) 2004 B. Stultiens * Copyright (C) 2004 Koen Otter and Glenn van der Meyden * * This program is free software; you can redistribute it and/or modify * it under the terms of the GNU General Public License as published by * the Free Software Foundation; either version 2 of the License, or * (at your option) any later version. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU General Public License for more details. * * You should have received a copy of the GNU General Public License * along with this program; if not, write to the Free Software * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA */ #include #ifdef USE_GUI #include #include #include #include #include #include "decoder.h" #include "debug.h" Decoder::Decoder(Error* err, sem_t *gencode_sem, sem_t *gplayer_sem, Queue *gencode_queue, Queue *gplayer_queue, pthread_mutex_t *gmutex, volatile int *grunning) { errobj = err; encode_sem = gencode_sem; player_sem = gplayer_sem; encode_queue = gencode_queue; player_queue = gplayer_queue; mutex = gmutex; running = grunning; AVCodec *dec_codec; // Find ffmpeg-dv-codec if(!(dec_codec = avcodec_find_decoder(CODEC_ID_DVVIDEO))) { errobj->pushError("Unable to find codec."); return; } // Initialize ffmpeg-dv-codec if(avcodec_open(&dvcodec, dec_codec) < 0) { errobj->pushError("Error while opening codec for input stream."); return; } } Decoder::~Decoder() { avcodec_close(&dvcodec); } static int raw_reader( raw1394handle_t handle, int channel, size_t length, quadlet_t *data ) { static char *framedata = NULL; // Only process packets with reasonable length. if ( length > 16 ) { unsigned char * p = ( unsigned char* ) & data[ 3 ]; int section_type = p[ 0 ] >> 5; // section type is in bits 5 - 7 int dif_sequence = p[ 1 ] >> 4; // dif sequence number is in bits 4 - 7 int dif_block = p[ 2 ]; if ( section_type == 0 && dif_sequence == 0 ) { if ( framedata != NULL ) { raw1394_set_userdata(handle, (void *)framedata); framedata = NULL; } } if(!framedata) { framedata = (char *)malloc(DVPACKAGE_SIZE); // dvframe.h if(!framedata) { // We're fucked exit(1); } } switch ( section_type ) { case 0: // 1 Header block // p[3] |= 0x80; // hack to force PAL data memcpy( framedata + dif_sequence * 150 * 80, p, 480 ); break; case 1: // 2 Subcode blocks memcpy( framedata + dif_sequence * 150 * 80 + ( 1 + dif_block ) * 80, p, 480 ); break; case 2: // 3 VAUX blocks memcpy( framedata + dif_sequence * 150 * 80 + ( 3 + dif_block ) * 80, p, 480 ); break; case 3: // 9 Audio blocks interleaved with video memcpy( framedata + dif_sequence * 150 * 80 + ( 6 + dif_block * 16 ) * 80, p, 480 ); break; case 4: // 135 Video blocks interleaved with audio memcpy( framedata + dif_sequence * 150 * 80 + ( 7 + ( dif_block / 15 ) + dif_block ) * 80, p, 480 ); break; default: // we can´t handle any other data break; } } return 0; } void Decoder::decode() { // FIXME: Read port and channel data from config. int port = 0; int channel = 63; int n_ports; struct raw1394_portinfo pinf[ 16 ]; raw1394handle_t handle; // Get handle to firewire channels handle = raw1394_new_handle(); if(!handle) { fprintf( stderr, "raw1394 - failed to get handle: %s.\n", strerror( errno ) ); exit( EXIT_FAILURE ); } // how many adapters are hook in? if((n_ports = raw1394_get_port_info(handle, pinf, 16)) < 0 ) { fprintf( stderr, "raw1394 - failed to get port info: %s.\n", strerror( errno ) ); raw1394_destroy_handle( handle ); exit( EXIT_FAILURE ); } // Tell raw1394 which host adapter to use if(raw1394_set_port(handle, port) < 0 ) { errobj->pushError("Error while opening codec for input stream."); fprintf( stderr, "raw1394 - failed to set set port: %s.\n", strerror( errno ) ); exit( EXIT_FAILURE ); } raw1394_set_iso_handler( handle, channel, raw_reader); // 63 is default channel... sucks. raw1394_set_userdata( handle, ( void* ) NULL); raw1394_start_iso_rcv( handle, channel); while(*running) { AVPacket pkt; uint8_t *ptr; int len; SDL_Event user_event; while(1) { raw1394_loop_iterate(handle); pkt.data = (uint8_t *)raw1394_get_userdata(handle); if(pkt.data) { raw1394_set_userdata(handle, NULL); break; } } printf("1"); fflush(stdout); len = pkt.size = DVPACKAGE_SIZE; ptr = pkt.data; pkt.stream_index = 0; // NOTE: we only decode video, we only need the data from stream_index 0 // (stream 0: video, stream 1: audio) // while(pkt.stream_index == 0 && len > 0) { int ret; int got_picture; // buf_t *buf = buf_alloc(); FFFrame *fff = new FFFrame(); ALLOC(fff, "FFFrame in decode"); memset(fff->frame, 0 , sizeof(AVFrame)); //DVFrame *dvf = new DVFrame(); ALLOC(dvf, "DVFrame in decode"); // dvf->type = DVF_VIDEO; // memcpy(dvf->frame, ptr, len); printf("2"); fflush(stdout); ret = avcodec_decode_video(&dvcodec, fff->frame, &got_picture, ptr, len); printf("3"); fflush(stdout); if(ret < 0) { errobj->pushError("Error while decoding stream."); return; } printf("4"); fflush(stdout); pthread_mutex_lock(mutex); // encode_queue->push(dvf); player_queue->push(fff); pthread_mutex_unlock(mutex); printf("5"); fflush(stdout); sem_post(encode_sem); user_event.type = SDL_USEREVENT; user_event.user.code = 0; user_event.user.data1 = NULL; user_event.user.data2 = NULL; printf("6"); fflush(stdout); SDL_PushEvent(&user_event); printf("7"); fflush(stdout); free(ptr); printf("8"); fflush(stdout); } printf("9"); fflush(stdout); /* Kick the others so they wake up with empty queues */ sem_post(encode_sem); pthread_exit(NULL); } void Decoder::run() { decode(); } #endif /*USE_GUI*/