hvirtual/quicktime/libmjpeg.h

   1 /*
   2  * This library is free software; you can redistribute it and/or modify it
   3  * under the terms of the GNU Lesser General Public License as published
   4  * by the Free Software Foundation; either version 2 of the License, or
   5  * (at your option) any later version.
   6  *
   7  * This library is distributed in the hope that it will be useful, but
   8  * WITHOUT ANY WARRANTY; without even the implied warranty of
   9  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  10  * Lesser General Public License for more details.
  11  *
  12  * You should have received a copy of the GNU Lesser General Public
  13  * License along with this library; if not, write to the Free Software
  14  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307
  15  * USA
  16  */
  17
  18  #ifndef LIBMJPEG_H
  19 #define LIBMJPEG_H
  20
  21
  22 /* Motion JPEG library */
  23
  24
  25
  26 #ifdef __cplusplus
  27 extern "C" {
  28 #endif
  29
  30
  31
  32 #include <stdio.h>
  33 #include "jpeglib.h"
  34 #include <png.h>       /* Need setjmp.h included by png.h */
  35 #include "pthread.h"
  36
  37 #define MAXFIELDS 2
  38 #define QUICKTIME_MJPA_MARKSIZE 40
  39 #define QUICKTIME_JPEG_TAG 0x6d6a7067
  40
  41
  42 struct mjpeg_error_mgr {
  43   struct jpeg_error_mgr pub;    /* "public" fields */
  44   jmp_buf setjmp_buffer;        /* for return to caller */
  45 };
  46
  47 typedef struct mjpeg_error_mgr* mjpeg_error_ptr;
  48
  49 #ifndef __alpha__
  50 typedef struct
  51 {
  52 } mjpeg_lml_hdr;
  53
  54 typedef struct
  55 {
  56 } mjpeg_dc10_hdr;
  57 #endif
  58
  59
  60 // The compressor structure is shared between decompressors and compressors
  61 typedef struct
  62 {
  63         void *mjpeg;
  64         int instance;
  65         unsigned char *output_buffer;    /* Buffer for MJPEG output */
  66         long output_size;     /* Size of image stored in buffer */
  67         long output_allocated;    /* Allocated size of output buffer */
  68         struct jpeg_decompress_struct jpeg_decompress;
  69         struct jpeg_compress_struct jpeg_compress;
  70         struct mjpeg_error_mgr jpeg_error;
  71         pthread_t tid;   /* ID of thread */
  72         pthread_mutex_t input_lock, output_lock;
  73         int done;     /* Flag to end */
  74         int error;
  75 // Pointer to uncompressed YUV rows
  76 // [3 planes][downsampled rows][downsampled pixels]
  77         unsigned char **rows[3];
  78 /* Temp rows for each MCU */
  79         unsigned char **mcu_rows[3];
  80 /* Height of the field */
  81         int field_h;
  82         int coded_field_h;
  83 } mjpeg_compressor;
  84
  85 typedef struct
  86 {
  87 // Dimensions of user frame buffer
  88         int output_w;
  89         int output_h;
  90 // Dimensions for encoder frame buffer
  91         int coded_w, coded_h;
  92         int fields;
  93         int quality;
  94         int use_float;
  95         int kludge;
  96         int cpus;
  97 // Color model of user interface.
  98         int color_model;
  99 // Color model of compressed data.  Since MJPA streams use 4:2:0
 100     int jpeg_color_model;
 101 // Error in compression process
 102         int error;
 103
 104         mjpeg_compressor *compressors[MAXFIELDS];
 105         mjpeg_compressor *decompressors[MAXFIELDS];
 106
 107 // Temp frame for interlacing
 108 // [3 planes][downsampled rows][downsampled pixels]
 109         unsigned char *temp_data;
 110         unsigned char **temp_rows[3];
 111         unsigned char **row_argument, *y_argument, *u_argument, *v_argument;
 112
 113 // Buffer passed to user
 114 // This can contain one progressive field or two fields with headers
 115         unsigned char *output_data;
 116         long output_size;
 117         long output_allocated;
 118 // Offset to field2 in output_data
 119         long output_field2;
 120 // Buffer passed from user
 121         unsigned char *input_data;
 122         long input_size;
 123 // Offset to field2 in input_data
 124         long input_field2;
 125         int deinterlace;
 126         int rowspan;
 127
 128 // Workarounds for thread unsafe libraries
 129         pthread_mutex_t decompress_init;
 130         int decompress_initialized;
 131 } mjpeg_t;
 132
 133
 134
 135
 136
 137 // Entry points
 138 mjpeg_t* mjpeg_new(int w,
 139         int h,
 140         int fields);
 141 void mjpeg_delete(mjpeg_t *mjpeg);
 142
 143 void mjpeg_set_quality(mjpeg_t *mjpeg, int quality);
 144 void mjpeg_set_float(mjpeg_t *mjpeg, int use_float);
 145 // This is useful when producing realtime NTSC output for a JPEG board.
 146 void mjpeg_set_deinterlace(mjpeg_t *mjpeg, int value);
 147 void mjpeg_set_cpus(mjpeg_t *mjpeg, int cpus);
 148 void mjpeg_set_rowspan(mjpeg_t *mjpeg, int rowspan);
 149
 150
 151 int mjpeg_get_fields(mjpeg_t *mjpeg);
 152
 153 int mjpeg_decompress(mjpeg_t *mjpeg,
 154         unsigned char *buffer,
 155         long buffer_len,
 156         long input_field2,
 157         unsigned char **row_pointers,
 158         unsigned char *y_plane,
 159         unsigned char *u_plane,
 160         unsigned char *v_plane,
 161         int color_model,
 162         int cpus);
 163
 164 int mjpeg_compress(mjpeg_t *mjpeg,
 165         unsigned char **row_pointers,
 166         unsigned char *y_plane,
 167         unsigned char *u_plane,
 168         unsigned char *v_plane,
 169         int color_model,
 170         int cpus);
 171
 172 // Get buffer information after compressing
 173 unsigned char* mjpeg_output_buffer(mjpeg_t *mjpeg);
 174 long mjpeg_output_field2(mjpeg_t *mjpeg);
 175 long mjpeg_output_size(mjpeg_t *mjpeg);
 176
 177 // Retrieve width and height from a buffer of JPEG data
 178 void mjpeg_video_size(unsigned char *data, long data_size, int *w, int *h);
 179
 180
 181
 182 // Calculate marker contents and insert them into a buffer.
 183 // Reallocates the buffer if it isn't big enough so make sure it's big enough
 184 // when passing VFrames.
 185 // field2_offset is set to -1 if the markers already exist or the field offset
 186 // if markers don't already exist.
 187 void mjpeg_insert_quicktime_markers(unsigned char **buffer,
 188         long *buffer_size,
 189         long *buffer_allocated,
 190         int fields,
 191         long *field2_offset);
 192 void mjpeg_insert_avi_markers(unsigned char **buffer,
 193         long *buffer_size,
 194         long *buffer_allocated,
 195         int fields,
 196         long *field2_offset);
 197
 198 // Get the second field offset from the markers
 199 long mjpeg_get_buz_field2(unsigned char *buffer, long buffer_size);
 200 long mjpeg_get_lml33_field2(unsigned char *buffer, long buffer_size);
 201 long mjpeg_get_quicktime_field2(unsigned char *buffer, long buffer_size);
 202 // Field dominance is retrieved for the jpeg decoder.  AVI stores field
 203 // dominance in each field.
 204 long mjpeg_get_avi_field2(unsigned char *buffer,
 205         long buffer_size,
 206         int *field_dominance);
 207 long mjpeg_get_field2(unsigned char *buffer, long buffer_size);
 208
 209 #ifdef __cplusplus
 210 }
 211 #endif
 212
 213 #endif