frame.h

Go to the documentation of this file.

/*
* frame.h -- utilities for process digital video frames
* Copyright (C) 2000 Arne Schirmacher <arne@schirmacher.de>
*
* 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.
*/

#ifndef _FRAME_H
#define _FRAME_H 1

#include "config.h"

#include <iostream>
using std::cerr;
using std::endl;

#include <time.h>
#include <string>
#include <stdio.h>
using std::string;

#include <samplerate.h>

#include "endian_types.h"

#ifdef WITH_LIBDV
#include <libdv/dv.h>
#include <libdv/dv_types.h>
#endif

#if defined(HAVE_LIBAVCODEC)
extern "C"
{
#   include <avcodec.h>
}
#endif

#define FRAME_MAX_WIDTH 720
#define FRAME_MAX_HEIGHT 576

typedef struct Pack
{
        unsigned char data[ 5 ];
}
Pack;

typedef struct TimeCode
{
        int hour;
        int min;
        int sec;
        int frame;
}
TimeCode;


typedef struct AudioInfo
{
        int frames;
        int frequency;
        int samples;
        int channels;
        int quantization;
}
AudioInfo;


class VideoInfo
{
public:
        int width;
        int height;
        bool isPAL;
        TimeCode timeCode;
        struct tm       recDate;

        VideoInfo();

        //    string GetTimeCodeString();
        //    string GetRecDateString();
}
;


class Frame
{
public:
        unsigned char data[ 144000 ];
        int bytesInFrame;

#if defined(HAVE_LIBAVCODEC)

        AVCodecContext libavcodec;
#endif

#ifdef WITH_LIBDV

        dv_decoder_t *decoder;
#endif

        int16_t *audio_buffers[ 4 ];

public:
        Frame();
        ~Frame();

        bool GetSSYBPack( int packNum, Pack &pack ) const;
        bool GetVAUXPack( int packNum, Pack &pack ) const;
        bool GetAAUXPack( int packNum, Pack &pack ) const;
        bool GetTimeCode( TimeCode &timeCode ) const;
        bool GetRecordingDate( struct tm &recDate ) const;
        string GetRecordingDate( void ) const;
        bool GetAudioInfo( AudioInfo &info ) const;
        bool GetVideoInfo( VideoInfo &info ) const;
        int GetFrameSize( void ) const;
        float GetFrameRate( void ) const;
        bool IsPAL( void ) const;
        bool IsNewRecording( void ) const;
        bool IsNormalSpeed( void ) const;
        bool IsComplete( void ) const;
        int ExtractAudio( void *sound ) const;
#ifdef WITH_LIBDV

        void SetPreferredQuality( );
        int ExtractAudio( int16_t **channels ) const;
        void ExtractHeader( void );
        void Deinterlace( uint8_t *pdst, uint8_t *psrc, int stride, int height );
        int ExtractRGB( void *rgb );
        int ExtractPreviewRGB( void *rgb );
        int ExtractYUV( void *yuv );
        int ExtractYUV420( uint8_t *yuv, uint8_t *output[ 3 ] );
        int ExtractPreviewYUV( void *yuv );
        bool IsWide( void ) const;
        int GetWidth();
        int GetHeight();
        void SetRecordingDate( time_t *datetime, int frame );
        void SetTimeCode( int frame );
        bool EncodeAudio( AudioInfo &info, int16_t **channels );
        void GetUpperField( void *image, int bpp );
        void GetLowerField( void *image, int bpp );
        static int CalculateNumberSamples( int frequency, int iteration );
#endif

private:
        static bool maps_initialized;
        static int palmap_ch1[ 2000 ];
        static int palmap_ch2[ 2000 ];
        static int palmap_2ch1[ 2000 ];
        static int palmap_2ch2[ 2000 ];
        static int ntscmap_ch1[ 2000 ];
        static int ntscmap_ch2[ 2000 ];
        static int ntscmap_2ch1[ 2000 ];
        static int ntscmap_2ch2[ 2000 ];
        static short compmap[ 4096 ];
        //FILE *libdv_log;
};

typedef enum {
    AUDIO_RESAMPLE_SRC_SINC_BEST_QUALITY = 0,
    AUDIO_RESAMPLE_SRC_SINC_MEDIUM_QUALITY = 1,
    AUDIO_RESAMPLE_SRC_SINC_FASTEST = 2,
    AUDIO_RESAMPLE_SRC_ZERO_ORDER_HOLD = 3,
    AUDIO_RESAMPLE_SRC_LINEAR = 4,
    AUDIO_RESAMPLE_INTERNAL = 5
}
AudioResampleType;

#define BUFFER_LEN 20480

template <class input_t, class output_t> class AudioResample
{
protected:
        int output_rate;
        input_t *input;

public:
        AudioResample( int rate ) : output_rate( rate )
        {
                input = new input_t[ BUFFER_LEN ];
                output = new output_t[ BUFFER_LEN ];
        }
        virtual ~AudioResample()
        {
                delete[] input;
                delete[] output;
        }
        virtual void Resample( input_t *samples, int input_rate, int channels, int samples_this_frame )
        { }
        void Resample( Frame &frame )
        {
                //cout << "Resample -----------------" << endl;
                if ( output_rate != 0 )
                {
                        frame.ExtractAudio( input );
                        AudioInfo info;
                        frame.GetAudioInfo( info );
                        /*
                        cout << "Audio info: " << endl
                             << "  dec-..->frequency: " 
                             << info.frequency << endl
                             << "  samples: " 
                             << info.samples << endl;
                        */
                        if ( info.frequency && output_rate != info.frequency )
                        {
                                Resample( input,
                                          info.frequency,
                                          info.channels,
                                          info.samples );
                        }
                        else
                        {
                                for (int i=0; i<info.channels*info.samples; i++)
                                        output[i] = input[i];

                                size = info.channels*info.samples*sizeof(output_t);
                        }
                }
                else
                {
                        size = 0;
                }
                // cout << "size: " << size << endl;
        }
        void SetOutputFrequency( int output_rate )
        {
                this->output_rate = output_rate;
        }
        int GetOutputFrequency( )
        {
                return this->output_rate;
        }

        output_t *output;
        int size;
};

template <class input_t, class output_t> class InternalAudioResample : public AudioResample<input_t, output_t>
{
public:
        InternalAudioResample( ) : AudioResample<input_t,output_t>( 0 )
        { }
        
        InternalAudioResample( int output_rate ) : AudioResample<input_t,output_t>( output_rate )
        { }
        virtual ~InternalAudioResample()
        { }
        void Resample( input_t *input, int input_rate, int channels, int samples )
        {
                float ratio = ( float ) this->output_rate / ( float ) input_rate;
                this->size = ( int ) ( ( float ) samples * ratio );

                int rounding = 1 << 15;
                unsigned int xfactor = ( samples << 16 ) / this->size;
                unsigned int xmax = xfactor * this->size;
                unsigned int i = 0;
                unsigned int o = 0;
                this->size *= sizeof(output_t) * channels;

                for ( unsigned int xft = 0; xft < xmax; xft += xfactor )
                {
                        i = ( ( xft + rounding ) >> 16 ) * channels;
                        for (int c=0; c < channels; c++)
                                this->output[o+c] = input[i+c];
                        o += channels;
                }

        }
};

template <class input_t, class output_t> class SrcAudioResample : public AudioResample<input_t, output_t>
{
public:
        SrcAudioResample( int converter ) : AudioResample<input_t,output_t>( 0 )
        {
                int srcError = 0;
                state = src_new( converter, 2, &srcError );
                if ( srcError != 0 )
                        cerr << "SRC: " << src_strerror( srcError ) << endl;
        }
        SrcAudioResample( int converter, int output_rate ) : AudioResample<input_t,output_t>( output_rate )
        {
                int srcError = 0;
                state = src_new( converter, 2, &srcError );
                if ( srcError != 0 )
                        cerr << "SRC: " << src_strerror( srcError ) << endl;
        }
        virtual ~SrcAudioResample()
        {
                src_delete( state );
        }
        void Resample( input_t *input, int input_rate, int channels, int samples )
        {
                SRC_DATA data;
                static float input_buffer[ BUFFER_LEN ];
                static float output_buffer[ BUFFER_LEN ];

                for ( int i = 0; i < samples * channels; ++i )
                        input_buffer[ i ] = ( float ) input[ i ] / 32768.0;

                data.data_in = input_buffer;
                data.data_out = output_buffer;
                data.src_ratio = ( float ) this->output_rate / ( float ) input_rate;
                data.input_frames = samples;
                data.output_frames = BUFFER_LEN / channels;
                data.end_of_input = 0;
                int result = src_process ( state, &data );
                if ( result != 0 )
                        cerr << "SRC: " << src_strerror( result ) << endl;
                this->size = data.output_frames_gen * channels * sizeof(output_t);
                for ( int i = 0; i < data.output_frames_gen * channels; ++i )
                {
                        float sample = output_buffer[ i ];
                        if ( sample > 1.0 )
                                sample = 1.0;
                        if ( sample < -1.0 )
                                sample = -1.0;
                        if ( sample >= 0 )
                                this->output[ i ] = ( long int )( 32767.0 * sample );
                        else
                                this->output[ i ] = ( long int )( 32768.0 * sample );
                }
        }

private:
        SRC_STATE *state;
};

template <class input_t, class output_t> class AudioResampleFactory
{
public:
        static AudioResample<input_t, output_t> *createAudioResample( AudioResampleType type, int output_rate = 0 )
        {
                switch ( type )
                {
                case AUDIO_RESAMPLE_SRC_SINC_BEST_QUALITY:
                        return new SrcAudioResample<input_t, output_t>( SRC_SINC_BEST_QUALITY, output_rate );
                case AUDIO_RESAMPLE_SRC_SINC_MEDIUM_QUALITY:
                        return new SrcAudioResample<input_t, output_t>( SRC_SINC_MEDIUM_QUALITY, output_rate );
                case AUDIO_RESAMPLE_SRC_SINC_FASTEST:
                        return new SrcAudioResample<input_t, output_t>( SRC_SINC_FASTEST, output_rate );
                case AUDIO_RESAMPLE_SRC_ZERO_ORDER_HOLD:
                        return new SrcAudioResample<input_t, output_t>( SRC_ZERO_ORDER_HOLD, output_rate );
                case AUDIO_RESAMPLE_SRC_LINEAR:
                        return new SrcAudioResample<input_t, output_t>( SRC_LINEAR, output_rate );
                default:
                        return new InternalAudioResample<input_t, output_t>( output_rate );
                }
        }
};

class FramePool
{
public:
        virtual Frame *GetFrame( ) = 0;
        virtual void DoneWithFrame( Frame * ) = 0;
};

extern FramePool *GetFramePool( );

#endif

---