--- /dev/null
+#include <errno.h>
+#include <fcntl.h>
+#include <stdlib.h>
+#include <string.h>
+#include <limits.h>
+
+#include "libzmpeg3.h"
+
+static uint8_t def_palette[] = { /* better than nothing */
+ 0x00, 0x80, 0x80, 0x00, /* black #000000 */
+ 0xff, 0x80, 0x80, 0x00, /* white #ffffff */
+ 0x4c, 0x54, 0xff, 0x00, /* red #ff0000 */
+ 0x96, 0x2b, 0x15, 0x00, /* green #00ff00 */
+ 0x1d, 0xff, 0x6b, 0x00, /* blue #0000ff */
+ 0xe2, 0x00, 0x94, 0x00, /* yellow #ffff00 */
+ 0x69, 0xd4, 0xea, 0x00, /* purple #ff00ff */
+ 0xb3, 0xab, 0x00, 0x00, /* cyan #00ffff */
+ 0x80, 0x80, 0x80, 0x00, /* grey #808080 */
+ 0xa6, 0x6a, 0xbf, 0x00, /* pink #ff8080 */
+ 0xcb, 0x55, 0x4a, 0x00, /* lime #80ff80 */
+ 0x8e, 0xbf, 0x75, 0x00, /* lavndr #8080ff */
+ 0xd9, 0x95, 0x40, 0x00, /* brt cyan #80ffff */
+ 0xb4, 0xaa, 0xb5, 0x00, /* brt purple #ff80ff */
+ 0xf1, 0x40, 0x8a, 0x00, /* brt yellow #ffff80 */
+ 0xff, 0x80, 0x80, 0x00, /* white #ffffff */
+};
+
+int zmpeg3_t::
+init(const char *title_path)
+{
+ cpus = 1;
+ vts_title = 0;
+ total_vts_titles = 1;
+ interleave = angle = 0;
+ total_interleaves = 1;
+ bitfont_t::init_fonts();
+ demuxer = new demuxer_t(this, 0, 0, -1);
+ seekable = 1; log_errs = getenv("LOG_ERRS") ? 1 : 0;
+ memcpy(&palette[0], &def_palette[0], sizeof(def_palette));
+ char *cp = getenv("PTS_PADDING");
+ pts_padding = cp ? atoi(cp) : 1;
+ index_bytes = 0x300000;
+ cell_time = -1.;
+ recd_fd = -1;
+ iopened = fs->is_open;
+ int ret = !iopened ? fs->open_file() : 0;
+ if( ret == 0 ) {
+ int toc_atracks = INT_MAX;
+ int toc_vtracks = INT_MAX;
+ ret = get_file_type(&toc_atracks, &toc_vtracks, title_path);
+ if( ret == 0 ) {
+ /* Start from scratch */
+ if( !demuxer->total_titles )
+ demuxer->create_title();
+ /* Generate tracks */
+ if( is_transport_stream() || is_program_stream() ) {
+ int i, n;
+ /* Create video tracks */
+ for( i=n=0; i<MAX_STREAMZ && total_vtracks<toc_vtracks; ++i ) {
+ if( demuxer->vstream_table[i] ) {
+ vtrack[total_vtracks] = new_vtrack_t(i, demuxer, n++);
+ if( vtrack[total_vtracks] ) ++total_vtracks;
+ }
+ }
+ /* Create audio tracks */
+ for( i=n=0; i<MAX_STREAMZ && total_atracks<toc_atracks; ++i ) {
+ int format = demuxer->astream_table[i];
+ if( format > 0 ) {
+ atrack[total_atracks] = new_atrack_t(i, format, demuxer, n++);
+ if( atrack[total_atracks] ) ++total_atracks;
+ }
+ }
+ /* Create subtile tracks */
+ for( i=0; i<MAX_STREAMZ; ++i ) {
+ if( demuxer->sstream_table[i] ) {
+ strack[total_stracks] = new strack_t(i);
+ if( strack[total_stracks] ) ++total_stracks;
+ }
+ }
+ }
+ else if( is_video_stream() ) {
+ /* Create video tracks */
+ vtrack[0] = new_vtrack_t(-1, demuxer, 0);
+ if( vtrack[0] ) ++total_vtracks;
+ }
+ else if( is_audio_stream() ) {
+ /* Create audio tracks */
+ atrack[0] = new_atrack_t(-1, afmt_UNKNOWN, demuxer, 0);
+ if( atrack[0] ) ++total_atracks;
+ }
+ }
+ if( (fs->access() & ZIO_SEQUENTIAL) != 0 )
+ set_pts_padding(-1);
+ demuxer->seek_byte(0);
+ restart();
+ if( !iopened )
+ fs->close_file();
+ }
+ allocate_slice_decoders();
+ return ret;
+}
+
+void zmpeg3_t::
+restart()
+{
+ int i;
+
+ for( i=0; i<total_vtracks; ++i ) {
+ vtrack_t *vtrk = vtrack[i];
+ if( vtrk && vtrk->video ) {
+ vtrk->current_position = 0;
+ vtrk->reset_pts();
+ vtrk->frame_cache->clear();
+ vtrk->video->rewind_video(0);
+ }
+ }
+
+ for( i=0; i<total_atracks; ++i ) {
+ atrack_t *atrk = atrack[i];
+ if( atrk && atrk->audio ) {
+ atrk->current_position = 0;
+ atrk->reset_pts();
+ atrk->audio->output_size = 0;
+ atrk->audio->output_position = 0;
+ atrk->audio->rewind_audio();
+ }
+ }
+
+ fs->restart();
+}
+
+zmpeg3_t::
+zmpeg3_t(const char *path)
+#ifdef ZDVB
+ : dvb(this)
+#endif
+{
+ fs = new fs_t(this, path);
+ cpus = 1; log_errs = getenv("LOG_ERRS") ? 1 : 0;
+ demuxer = new demuxer_t(this, 0, 0, -1);
+// for toc generation
+ seekable = 0;
+ memcpy(&palette[0], &def_palette[0], sizeof(def_palette));
+ char *cp = getenv("PTS_PADDING");
+ pts_padding = cp ? atoi(cp) : 1;
+ index_bytes = 0x300000;
+ cell_time = -1.;
+ recd_fd = -1;
+}
+
+zmpeg3_t::
+zmpeg3_t(const char *path, int &ret, int atype, const char *title_path)
+#ifdef ZDVB
+ : dvb(this)
+#endif
+{
+ fs = new fs_t(this, path, atype);
+ ret = init(title_path);
+}
+
+zmpeg3_t::
+zmpeg3_t(int fd, int &ret, int atype)
+#ifdef ZDVB
+ : dvb(this)
+#endif
+{
+ fs = new fs_t(this, fd, atype);
+ ret = init();
+}
+
+zmpeg3_t::
+zmpeg3_t(FILE *fp, int &ret, int atype)
+#ifdef ZDVB
+ : dvb(this)
+#endif
+{
+ fs = new fs_t(this, fp, atype);
+ ret = init();
+}
+
+zmpeg3_t::
+~zmpeg3_t()
+{
+ int i;
+ int debug = 0;
+
+ if( debug ) zmsg("0\n");
+ stop_record();
+ delete_slice_decoders();
+
+ if( debug ) zmsg("1\n");
+ for( i=0; i<total_stracks; ++i )
+ delete strack[i];
+
+ if( debug ) zmsg("2\n");
+ for( i=0; i<total_atracks; ++i )
+ delete atrack[i];
+
+ if( debug ) zmsg("3\n");
+ for( i=0; i<total_vtracks; ++i )
+ delete vtrack[i];
+
+ if( debug ) zmsg("4\n");
+ delete demuxer;
+ delete fs;
+ if( playinfo ) delete playinfo;
+
+ if( debug ) zmsg("5\n");
+ if( frame_offsets ) {
+ for( i=0; i<total_vtracks; ++i ) {
+ delete [] frame_offsets[i];
+ delete [] keyframe_numbers[i];
+ }
+
+ if( debug ) zmsg("6\n");
+ delete [] frame_offsets;
+ delete [] keyframe_numbers;
+ delete [] total_frame_offsets;
+ delete [] total_keyframe_numbers;
+ delete [] video_eof;
+ }
+
+ if( debug ) zmsg("7\n");
+ if( sample_offsets ) {
+ for( i=0; i<total_atracks; ++i )
+ delete [] sample_offsets[i];
+
+ delete [] sample_offsets;
+ delete [] total_sample_offsets;
+ }
+
+ if( debug ) zmsg("8\n");
+ delete [] channel_counts;
+ delete [] nudging;
+ delete [] total_samples;
+ delete [] audio_eof;
+
+ if( debug ) zmsg("9\n");
+ if( indexes ) {
+ for( i=0; i<total_indexes; ++i )
+ delete indexes[i];
+
+ if( debug ) zmsg("10\n");
+ delete [] indexes;
+ }
+ bitfont_t::destroy_fonts();
+}
+
+
+zmpeg3_t:: index_t::
+index_t()
+{
+ index_zoom = 1;
+}
+
+zmpeg3_t:: index_t::
+~index_t()
+{
+ for( int i=0;i<index_channels; ++i )
+ delete [] index_data[i];
+ delete [] index_data;
+}
+
+// Return number of tracks in the table of contents
+int zmpeg3_t::
+index_tracks()
+{
+ return total_indexes;
+}
+
+// Return number of channels in track
+int zmpeg3_t::
+index_channels(int track)
+{
+ if( track<0 || track>=total_indexes ) return 0;
+ return indexes[track]->index_channels;
+}
+
+// Return zoom factor of index
+int zmpeg3_t::
+index_zoom()
+{
+ return !total_indexes ? 0 : indexes[0]->index_zoom;
+}
+
+// Number of high/low pairs in a channel of the track
+int zmpeg3_t::
+index_size(int track)
+{
+ if( track<0 || track>=total_indexes ) return 0;
+ int size = indexes[track]->index_size;
+ if( track < total_atracks && atrack[track]->nudge > 0 )
+ size -= atrack[track]->nudge / indexes[0]->index_zoom;
+ return size;
+}
+
+// Get data for one index channel
+float* zmpeg3_t::
+index_data(int track, int channel)
+{
+ if( track<0 || track>=total_indexes ) return 0;
+ float *data = indexes[track]->index_data[channel];
+ if( track < total_atracks && atrack[track]->nudge > 0 )
+ data += (atrack[track]->nudge / indexes[0]->index_zoom) * 2;
+ return data;
+}
+
+zstrack_t* zmpeg3_t::
+get_strack_id(int id, video_t *vid)
+{
+ for( int i=0; i<total_stracks; ++i ) {
+ strack_t *strk = strack[i];
+ if( strk->video && strk->video != vid ) continue;
+ if( strk->id == id ) return strack[i];
+ }
+ return 0;
+}
+
+zstrack_t* zmpeg3_t::
+get_strack(int number)
+{
+ if( number >= total_stracks || number < 0 ) return 0;
+ return strack[number];
+}
+
+zstrack_t* zmpeg3_t::
+create_strack(int id, zvideo_t *vid)
+{
+ int i, j;
+ strack_t *strk = 0;
+
+ if( !(strk=get_strack_id(id, vid)) ) {
+ strk = new strack_t(id, vid);
+ for( i=0; i<total_stracks; ++i ) {
+ /* Search for first ID > current id */
+ if( strack[i]->id > id ) {
+ /* Shift back 1 */
+ for( j=total_stracks; --j>=i; )
+ strack[j+1] = strack[j];
+ break;
+ }
+ }
+
+/* Store in table */
+ strack[i] = strk;
+ ++total_stracks;
+ }
+
+ return strk;
+}
+
+int zmpeg3_t::
+show_subtitle(int stream, int strk)
+{
+ int result = -1;
+ if( stream>=0 && stream<total_vtracks ) {
+ vtrack_t *vtrk = vtrack[stream];
+ if( vtrk && vtrk->video )
+ result = vtrk->video->show_subtitle(strk);
+ }
+ return result;
+}
+
+static inline int is_toc_(uint32_t bits)
+{
+ return (bits == zmpeg3_t::TOC_PREFIX);
+}
+
+
+static inline int is_ifo_(uint32_t bits)
+{
+ return (bits == zmpeg3_t::IFO_PREFIX);
+}
+
+static inline int is_transport_(uint32_t bits)
+{
+ return (((bits >> 24) & 0xff) == zmpeg3_t::SYNC_BYTE);
+}
+
+static inline int is_bd_(uint32_t bits1, uint32_t bits2, char *ext)
+{
+ return (((bits2 >> 24) & 0xff) == zmpeg3_t::SYNC_BYTE) &&
+ ( !ext || (ext &&
+ (!strncasecmp(ext, ".m2ts", 5) ||
+ !strncasecmp(ext, ".mts", 4))) );
+}
+
+static inline int is_program_(uint32_t bits)
+{
+ return (bits == zmpeg3_t::PACK_START_CODE);
+}
+
+static inline int is_mpeg_audio_(uint32_t bits)
+{
+ return (bits & 0xfff00000) == 0xfff00000 ||
+ (bits & 0xffff0000) == 0xffe30000 ||
+ ((bits >> 8) == zmpeg3_t::ID3_PREFIX) ||
+ (bits == zmpeg3_t::RIFF_CODE);
+}
+
+static inline int is_mpeg_video_(uint32_t bits)
+{
+ return (bits == zmpeg3_t::SEQUENCE_START_CODE ||
+ bits == zmpeg3_t::PICTURE_START_CODE);
+}
+
+static inline int is_ac3_(uint32_t bits)
+{
+ return (((bits & 0xffff0000) >> 16) == zmpeg3_t::AC3_START_CODE);
+}
+
+int zmpeg3_t::
+check_sig(char *path)
+{
+ int result = 0;
+ fs_t *fs = new fs_t(0, path);
+
+ if( !fs->open_file() ) { /* File found */
+ char *ext = strrchr(path, '.');
+ uint32_t bits = fs->read_uint32();
+ uint32_t bits2 = fs->read_uint32();
+
+ /* pre-approved suffixes */
+ if( ext && !strncasecmp(ext, ".mp3", 4) ) {
+ result = 1;
+ }
+ /* Test header */
+ else if( bits == TOC_PREFIX ) {
+ result = 1;
+ }
+ /* Blu-Ray or AVC-HD*/
+ else if( is_bd_(bits, bits2, ext) ) {
+ result = 1;
+ }
+ else if( (((bits >> 24) & 0xff) == zmpeg3_t::SYNC_BYTE ) ||
+ (bits == zmpeg3_t::PACK_START_CODE) ||
+ ((bits & 0xfff00000) == 0xfff00000) ||
+ ((bits & 0xffff0000) == 0xffe30000) ||
+ (bits == zmpeg3_t::SEQUENCE_START_CODE) ||
+ (bits == zmpeg3_t::PICTURE_START_CODE) ||
+ (((bits & 0xffff0000) >> 16) == zmpeg3_t::AC3_START_CODE) ||
+ ((bits >> 8) == zmpeg3_t::ID3_PREFIX) ||
+ (bits == zmpeg3_t::RIFF_CODE) ||
+ (bits == zmpeg3_t::IFO_PREFIX)) {
+ result = 1;
+
+ /* Test file extension. */
+ if( ext ) { ++ext;
+ if( strcasecmp(ext, "ifo") &&
+ strcasecmp(ext, "mp2") &&
+ strcasecmp(ext, "mp3") &&
+ strcasecmp(ext, "m1v") &&
+ strcasecmp(ext, "m2v") &&
+ strcasecmp(ext, "m2s") &&
+ strcasecmp(ext, "mpg") &&
+ strcasecmp(ext, "vob") &&
+ strcasecmp(ext, "ts") &&
+ strcasecmp(ext, "vts") &&
+ strcasecmp(ext, "mpeg") &&
+ strcasecmp(ext, "m2t") &&
+ strcasecmp(ext, "ac3") )
+ result = 0;
+ }
+ }
+ fs->close_file();
+ }
+
+ delete fs;
+ return result;
+}
+
+int zmpeg3_t::
+calculate_packet_size()
+{
+ if( is_transport_stream() )
+ return is_bd() ? BD_PACKET_SIZE : TS_PACKET_SIZE;
+ if( is_program_stream() )
+ return 0;
+ if( is_audio_stream() || is_video_stream() )
+ return DVD_PACKET_SIZE;
+ zerr("undefined stream type.\n");
+ return -1;
+}
+
+int zmpeg3_t::
+get_file_type(int *toc_atracks, int *toc_vtracks, const char *title_path)
+{
+ int result = 0;
+ file_type = 0;
+ uint32_t bits = fs->read_uint32();
+ uint32_t bits2 = fs->read_uint32();
+
+ /* TOC */
+ if( is_toc_(bits) ) {
+ /* Table of contents for another title set */
+ result = toc_atracks && toc_vtracks ?
+ read_toc(toc_atracks, toc_vtracks, title_path) : 1;
+ if( result ) fs->close_file();
+ }
+ else if( is_ifo_(bits) ) {
+ /* IFO file */
+ if( !read_ifo() )
+ file_type = FT_PROGRAM_STREAM | FT_IFO_FILE;
+ fs->close_file();
+ }
+ else if( is_bd_(bits,bits2,0) ) {
+ file_type = FT_TRANSPORT_STREAM | FT_BD_FILE;
+ }
+ /* Transport stream */
+ else if( is_transport_(bits) ) {
+ file_type = FT_TRANSPORT_STREAM;
+ }
+ /* Program stream */
+ /* Determine packet size empirically */
+ else if( is_program_(bits) ) {
+ file_type = FT_PROGRAM_STREAM;
+ }
+ /* MPEG Audio only */
+ else if( is_mpeg_audio_(bits) ) {
+ file_type = FT_AUDIO_STREAM;
+ }
+ /* Video only */
+ else if( is_mpeg_video_(bits) ) {
+ file_type = FT_VIDEO_STREAM;
+ }
+ /* AC3 Audio only */
+ else if( is_ac3_(bits) ) {
+ file_type = FT_AUDIO_STREAM;
+ }
+ else {
+ const char *ext = strrchr(fs->path, '.');
+ if( ext ) {
+ if( !strncasecmp(ext, ".mp3", 4) ) {
+ file_type = FT_AUDIO_STREAM;
+ }
+ }
+ }
+//zmsgs("2 %x\n", file_type);
+ if( !file_type ) {
+ zerr("not a readable stream.\n");
+ if( !result ) result = ERR_UNDEFINED_ERROR;
+ }
+ else
+ packet_size = calculate_packet_size();
+ return result;
+}
+
+int zmpeg3_t::
+set_pts_padding(int v)
+{
+ int ret = pts_padding;
+ pts_padding = v;
+ if( pts_padding >= 0 ) {
+ for( int vtk=0; vtk<total_vtracks; ++vtk )
+ vtrack[vtk]->pts_starttime = -1.;
+ for( int atk=0; atk<total_atracks; ++atk )
+ atrack[atk]->pts_starttime = -1.;
+ }
+ else {
+ for( int vtk=0; vtk<total_vtracks; ++vtk )
+ vtrack[vtk]->pts_starttime = vtrack[vtk]->pts_offset = 0.;
+ for( int atk=0; atk<total_atracks; ++atk )
+ atrack[atk]->pts_starttime = atrack[atk]->pts_offset = 0.;
+ }
+ return ret;
+}
+
+int zmpeg3_t::
+set_cpus(int cpus)
+{
+ this->cpus = cpus;
+ int i;
+ for( i=0; i<total_vtracks; ++i )
+ vtrack[i]->video->set_cpus(cpus);
+ return 0;
+}
+
+int zmpeg3_t::
+has_audio()
+{
+ return total_atracks > 0;
+}
+
+int zmpeg3_t::
+total_astreams()
+{
+ return total_atracks;
+}
+
+int zmpeg3_t::
+audio_channels(int stream)
+{
+ return stream>=0 && stream<total_atracks ?
+ atrack[stream]->channels : -1;
+}
+
+double zmpeg3_t::
+sample_rate(int stream)
+{
+ return stream>=0 && stream<total_atracks ?
+ (double)atrack[stream]->sample_rate : -1.;
+}
+
+long zmpeg3_t::
+get_sample(int stream)
+{
+ return stream>=0 && stream<total_atracks ?
+ atrack[stream]->current_position : -1;
+}
+
+int zmpeg3_t::
+set_sample(long sample, int stream)
+{
+ int result = 1;
+ if( stream>=0 && stream<total_atracks ) {
+ atrack_t *atrk = atrack[stream];
+ if( atrk && atrk->audio ) {
+//zmsgs("1 %d %ld\n", atrk->current_position, sample);
+ if( atrk->current_position != sample ) {
+ atrk->current_position = sample;
+ atrk->audio->seek_sample(sample);
+ }
+ result = 0;
+ }
+ }
+ return result;
+}
+
+long zmpeg3_t::
+audio_nudge(int stream)
+{
+ long result = 0;
+ if( stream>=0 && stream<total_atracks ) {
+ atrack_t *atrk = atrack[stream];
+ if( atrk ) result = atrk->nudge;
+ }
+ return result;
+}
+
+long zmpeg3_t::
+audio_samples(int stream)
+{
+ long result = -1;
+ if( stream>=0 && stream<total_atracks ) {
+ atrack_t *atrk = atrack[stream];
+ if( atrk ) result = atrk->total_samples;
+ }
+ return result;
+}
+
+const char* zmpeg3_t::
+audio_format(int stream)
+{
+ const char *result = 0;
+ if( stream>=0 && stream<total_atracks ) {
+ switch( atrack[stream]->format ) {
+ case afmt_IGNORE: result = "Ignore"; break;
+ case afmt_UNKNOWN: result = "Unknown"; break;
+ case afmt_MPEG: result = "MPEG"; break;
+ case afmt_AC3: result = "AC3"; break;
+ case afmt_PCM: result = "PCM"; break;
+ case afmt_AAC: result = "AAC"; break;
+ case afmt_JESUS: result = "Vorbis"; break;
+ default: result = "Undef"; break;
+ }
+ }
+ return result;
+}
+
+int zmpeg3_t::
+has_video()
+{
+ return total_vtracks > 0;
+}
+
+int zmpeg3_t::
+total_vstreams()
+{
+ return total_vtracks;
+}
+
+int zmpeg3_t::
+video_width(int stream)
+{
+ int result = -1;
+ if( stream>=0 && stream<total_vtracks ) {
+ vtrack_t *vtrk = vtrack[stream];
+ if( vtrk ) result = vtrk->width;
+ }
+ return result;
+}
+
+int zmpeg3_t::
+video_height(int stream)
+{
+ int result = -1;
+ if( stream>=0 && stream<total_vtracks ) {
+ vtrack_t *vtrk = vtrack[stream];
+ if( vtrk ) result = vtrk->height;
+ }
+ return result;
+}
+
+int zmpeg3_t::
+coded_width(int stream)
+{
+ int result = -1;
+ if( stream>=0 && stream<total_vtracks ) {
+ vtrack_t *vtrk = vtrack[stream];
+ if( vtrk )
+ result = vtrk->video->coded_picture_width;
+ }
+ return result;
+}
+
+int zmpeg3_t::
+coded_height(int stream)
+{
+ int result = -1;
+ if( stream>=0 && stream<total_vtracks ) {
+ vtrack_t *vtrk = vtrack[stream];
+ if( vtrk )
+ result = vtrk->video->coded_picture_height;
+ }
+ return result;
+}
+
+int zmpeg3_t::
+video_pid(int stream)
+{
+ int result = -1;
+ if( stream>=0 && stream<total_vtracks ) {
+ vtrack_t *vtrk = vtrack[stream];
+ if( vtrk ) result = vtrk->pid;
+ }
+ return result;
+}
+
+float zmpeg3_t::
+aspect_ratio(int stream)
+{
+ float result = 0.0;
+ if( stream>=0 && stream<total_vtracks ) {
+ vtrack_t *vtrk = vtrack[stream];
+ if( vtrk ) result = vtrk->aspect_ratio;
+ }
+ return result;
+}
+
+double zmpeg3_t::
+frame_rate(int stream)
+{
+ double result = -1.0;
+ if( stream>=0 && stream<total_vtracks ) {
+ vtrack_t *vtrk = vtrack[stream];
+ if( vtrk ) result = vtrk->frame_rate;
+ }
+ return result;
+}
+
+long zmpeg3_t::
+video_frames(int stream)
+{
+ long result = -1;
+ if( stream>=0 && stream<total_vtracks ) {
+ vtrack_t *vtrk = vtrack[stream];
+ if( vtrk ) result = vtrk->total_frames;
+ }
+ return result;
+}
+
+long zmpeg3_t::
+get_frame(int stream)
+{
+ long result = -1;
+ if( stream>=0 && stream<total_vtracks ) {
+ vtrack_t *vtrk = vtrack[stream];
+ if( vtrk ) result = vtrk->current_position;
+ }
+ return result;
+}
+
+int zmpeg3_t::
+set_frame(long frame, int stream)
+{
+ int result = -1;
+ if( stream>=0 && stream<total_vtracks ) {
+ vtrack_t *vtrk = vtrack[stream];
+ if( vtrk && vtrk->video ) {
+ if( vtrk->current_position != frame ) {
+ vtrk->current_position = frame;
+ vtrk->video->seek_frame(frame);
+ }
+ result = 0;
+ }
+ }
+ return result;
+}
+
+int zmpeg3_t::
+seek_byte(int64_t byte)
+{
+ int i;
+
+ for( i=0; i<total_vtracks; ++i ) {
+ vtrack_t *vtrk = vtrack[i];
+ if( vtrk && vtrk->video ) {
+ vtrk->current_position = 0;
+ if( byte )
+ vtrk->video->seek_byte(byte);
+ else
+ vtrk->video->rewind_video();
+ }
+ }
+
+ for( i=0; i<total_atracks; ++i ) {
+ atrack_t *atrk = atrack[i];
+ if( atrk && atrk->audio ) {
+ atrk->current_position = 0;
+ atrk->audio->seek_byte(byte);
+ }
+ }
+
+ return 0;
+}
+
+int zmpeg3_t::
+previous_frame(int stream)
+{
+ int result = 1;
+
+ if( stream>=0 && stream<total_vtracks ) {
+ vtrack_t *vtrk = vtrack[stream];
+ if( vtrk && vtrk->video ) {
+ last_type_read = 2;
+ last_stream_read = stream;
+ result = vtrk->video->previous_frame();
+ }
+ }
+ return result;
+}
+
+int64_t zmpeg3_t::
+tell_byte()
+{
+ int64_t result = -1;
+ if( last_type_read == 1 )
+ result = atrack[last_stream_read]->demuxer->tell_byte();
+ else if( last_type_read == 2 )
+ result =vtrack[last_stream_read]->demuxer->tell_byte();
+ return result;
+}
+
+int64_t zmpeg3_t::
+get_bytes()
+{
+ return demuxer->movie_size();
+}
+
+double zmpeg3_t::
+get_audio_time(int stream)
+{
+ double atime = -1.0;
+ if( stream >= 0 && stream < total_atracks ) {
+ double samplerate = sample_rate(stream);
+ if( samplerate > 0. ) {
+ int64_t sample = get_sample(stream);
+ if( !pts_padding || is_audio_stream() || is_video_stream() ) {
+ atime = sample / samplerate;
+ }
+ else {
+ atime = atrack[stream]->audio_time +
+ (sample - atrack[stream]->pts_position) / samplerate;
+ }
+ }
+ }
+ return atime;
+}
+
+double zmpeg3_t::
+get_video_time(int stream)
+{
+ double vtime = -1.0;
+ if( stream >= 0 && stream < total_vtracks ) {
+ double framerate = frame_rate(stream);
+ if( !pts_padding || is_audio_stream() || is_video_stream() ) {
+ if( framerate > 0. )
+ vtime = (double)get_frame(stream) / framerate;
+ }
+ else {
+ vtime = vtrack[stream]->video_time;
+ if( vtime >= 0. && framerate > 0. )
+ vtime += (get_frame(stream) - vtrack[stream]->pts_position) / framerate;
+ }
+ }
+ return vtime;
+}
+
+/* pts timecode available only in transport stream */
+double zmpeg3_t::
+get_time()
+{
+ double time = 0;
+ if( last_type_read == 1 )
+ time = get_audio_time(last_stream_read);
+ else if( last_type_read == 2 )
+ time = get_video_time(last_stream_read);
+ return time;
+}
+
+int zmpeg3_t::
+get_total_vts_titles()
+{
+ return total_vts_titles;
+}
+
+int zmpeg3_t::
+set_vts_title(int title)
+{
+ int result = vts_title;
+ if( title >= 0 )
+ vts_title = title;
+ return result;
+}
+
+int zmpeg3_t::
+get_total_interleaves()
+{
+ return total_interleaves;
+}
+
+int zmpeg3_t::
+set_interleave(int inlv)
+{
+ int result = interleave;
+ if( inlv >= 0 )
+ interleave = inlv;
+ return result;
+}
+
+int zmpeg3_t::
+set_angle(int a)
+{
+ int result = angle;
+ if( a >= 0 )
+ angle = a;
+ return result;
+}
+
+int zmpeg3_t::
+set_program(int no)
+{
+ int result = set_vts_title()*100 + set_angle()*10 + set_interleave();
+ if( no >= 0 ) {
+ set_vts_title(no / 100);
+ set_angle((no/10) % 10);
+ set_interleave(no % 10);
+ }
+ return result;
+}
+
+int zmpeg3_t::
+get_cell_time(int no, double &time)
+{
+ int result = 1, i = 0;
+ zcell_t *cell = 0;
+ while( no >= 0 ) {
+ result = demuxer->get_cell(i, cell);
+ if( result < 0 ) return result;
+ if( result > 0 ) i = cell->cell_no;
+ ++i; --no;
+ }
+ time = cell->cell_time;
+ return 0;
+}
+
+int zmpeg3_t::
+end_of_audio(int stream)
+{
+ int result = 1;
+ if( stream>=0 && stream<total_atracks ) {
+ atrack_t *atrk = atrack[stream];
+ if( atrk && atrk->demuxer )
+ result = atrk->demuxer->eof();
+ }
+ return result;
+}
+
+int zmpeg3_t::
+end_of_video(int stream)
+{
+ int result = 1;
+ if( stream>=0 && stream<total_vtracks ) {
+ vtrack_t *vtrk = vtrack[stream];
+ if( vtrk && vtrk->demuxer )
+ result = vtrk->demuxer->eof();
+ }
+ return result;
+}
+
+int zmpeg3_t::
+drop_frames(long frames, int stream)
+{
+ int result = -1;
+ if( stream>=0 && stream<total_vtracks ) {
+ vtrack_t *vtrk = vtrack[stream];
+ if( vtrk && vtrk->video ) {
+ result = vtrk->video->drop_frames(frames, 0);
+ if( frames > 0 )
+ vtrk->current_position += frames;
+ last_type_read = 2;
+ last_stream_read = stream;
+ }
+ }
+ return result;
+}
+
+int zmpeg3_t::
+colormodel(int stream)
+{
+ int result = -1;
+ if( stream>=0 && stream<total_vtracks ) {
+ vtrack_t *vtrk = vtrack[stream];
+ if( vtrk && vtrk->video )
+ result = vtrk->video->colormodel();
+ }
+ return result;
+}
+
+int zmpeg3_t::
+set_rowspan(int bytes, int stream)
+{
+ int result = 1;
+ if( stream>=0 && stream<total_vtracks ) {
+ vtrack_t *vtrk = vtrack[stream];
+ if( vtrk && vtrk->video ) {
+ vtrk->video->row_span = bytes;
+ result = 0;
+ }
+ }
+ return result;
+}
+
+
+int zmpeg3_t::
+read_frame(uint8_t **output_rows,
+ int in_x, int in_y, int in_w, int in_h,
+ int out_w, int out_h, int color_model, int stream)
+{
+ int result = -1;
+//zmsgs("1 %d\n", vtrack[stream]->current_position);
+ if( stream>=0 && stream<total_vtracks ) {
+ vtrack_t *vtrk = vtrack[stream];
+ if( vtrk && vtrk->video ) {
+ result = vtrk->video->read_frame(output_rows,
+ in_x, in_y, in_w, in_h, out_w, out_h, color_model);
+//zmsg(" 2\n");
+ last_type_read = 2;
+ last_stream_read = stream;
+ vtrk->current_position++;
+ }
+ }
+//zmsgs("2 %d\n", vtrack[stream]->current_position);
+ return result;
+}
+
+int zmpeg3_t::
+read_yuvframe(char *y_output, char *u_output, char *v_output,
+ int in_x, int in_y, int in_w, int in_h, int stream)
+{
+ int result = -1;
+//zmsg("1\n");
+ if( stream>=0 && stream<total_vtracks ) {
+ vtrack_t *vtrk = vtrack[stream];
+ if( vtrk && vtrk->video ) {
+ result = vtrk->video->read_yuvframe(y_output, u_output, v_output,
+ in_x, in_y, in_w, in_h);
+ last_type_read = 2;
+ last_stream_read = stream;
+ vtrk->current_position++;
+ }
+ }
+//zmsg("100\n");
+ return result;
+}
+
+int zmpeg3_t::
+read_yuvframe_ptr(char **y_output, char **u_output, char **v_output,
+ int stream)
+{
+ int result = -1;
+ if( stream>=0 && stream<total_vtracks ) {
+ vtrack_t *vtrk = vtrack[stream];
+ if( vtrk && vtrk->video ) {
+ result = vtrk->video->read_yuvframe_ptr(y_output, u_output, v_output);
+ last_type_read = 2;
+ last_stream_read = stream;
+ vtrk->current_position++;
+ }
+ }
+ return result;
+}
+
+int zmpeg3_t::
+read_audio(void *output, int type, int channel, long samples, int stream)
+{
+ int result = -1;
+ if( stream>=0 && stream<total_atracks ) {
+ atrack_t *atrk = atrack[stream];
+ if( atrk && atrk->audio ) {
+ atrk->audio->update_audio_history();
+ result = atrk->audio->decode_audio(output, type, channel, samples);
+ last_type_read = 1;
+ last_stream_read = stream;
+ atrk->current_position += samples;
+ }
+ }
+ return result;
+}
+
+int zmpeg3_t::
+read_audio(double *output_d, int channel, long samples, int stream)
+{
+ return read_audio((void *)output_d, atyp_DOUBLE, channel, samples, stream);
+}
+
+int zmpeg3_t::
+read_audio(float *output_f, int channel, long samples, int stream)
+{
+ return read_audio((void *)output_f, atyp_FLOAT, channel, samples, stream);
+}
+
+int zmpeg3_t::
+read_audio(int *output_i, int channel, long samples, int stream)
+{
+ return read_audio((void *)output_i, atyp_INT, channel, samples, stream);
+}
+
+int zmpeg3_t::
+read_audio(short *output_s, int channel, long samples, int stream)
+{
+ return read_audio((void *)output_s, atyp_SHORT, channel, samples, stream);
+}
+
+int zmpeg3_t::
+reread_audio(void *output, int atyp, int channel, long samples, int stream)
+{
+ int result = -1;
+ if( stream>=0 && stream<total_atracks ) {
+ atrack_t *atrk = atrack[stream];
+ if( atrk && atrk->audio ) {
+ set_sample( atrk->current_position-samples, stream);
+ result = atrk->audio->decode_audio(output, atyp, channel, samples);
+ last_type_read = 1;
+ last_stream_read = stream;
+ atrk->current_position += samples;
+ }
+ }
+ return result;
+}
+
+int zmpeg3_t::
+reread_audio(double *output_d, int channel, long samples, int stream)
+{
+ return reread_audio((void *)output_d, atyp_DOUBLE, channel, samples, stream);
+}
+
+int zmpeg3_t::
+reread_audio(float *output_f, int channel, long samples, int stream)
+{
+ return reread_audio((void *)output_f, atyp_FLOAT, channel, samples, stream);
+}
+
+int zmpeg3_t::
+reread_audio(int *output_i, int channel, long samples, int stream)
+{
+ return reread_audio((void *)output_i, atyp_INT, channel, samples, stream);
+}
+
+int zmpeg3_t::
+reread_audio(short *output_s, int channel, long samples, int stream)
+{
+ return reread_audio((void *)output_s, atyp_SHORT, channel, samples, stream);
+}
+
+
+int zmpeg3_t::
+read_audio_chunk(uint8_t *output, long *size, long max_size, int stream)
+{
+ int result = -1;
+ if( stream>=0 && stream<total_atracks ) {
+ atrack_t *atrk = atrack[stream];
+ if( atrk && atrk->audio ) {
+ result = atrk->audio->read_raw(output, size, max_size);
+ last_type_read = 1;
+ last_stream_read = stream;
+ }
+ }
+ return result;
+}
+
+int zmpeg3_t::
+read_video_chunk( uint8_t *output, long *size, long max_size, int stream)
+{
+ int result = -1;
+ if( stream>=0 && stream<total_vtracks ) {
+ vtrack_t *vtrk = vtrack[stream];
+ if( vtrk && vtrk->video ) {
+ result = vtrk->video->read_raw(output, size, max_size);
+ last_type_read = 2;
+ last_stream_read = stream;
+ }
+ }
+ return result;
+}
+
+int64_t zmpeg3_t::
+memory_usage()
+{
+ int64_t result = 0;
+ for( int i=0; i<total_vtracks; ++i )
+ if( vtrack[i]->frame_cache )
+ result += vtrack[i]->frame_cache->memory_usage();
+ return result;
+}
+
+int zmpeg3_t::
+start_record(int fd, int bsz)
+{
+ if( recd_fd >= 0 ) return 1;
+ struct stat64 st;
+ memset(&st, 0, sizeof(st));
+ if( fstat64(fd, &st) < 0 ) return 1;
+ recd_fd = fd;
+ recd_pos = st.st_size;
+ if (fs->start_record(bsz)) return recd_fd = -1;
+ return 0;
+}
+int zmpeg3_t::
+stop_record()
+{
+ if( fs->stop_record() ) return 1;
+ recd_fd = -1;
+ return 0;
+}
+
+void zmpeg3_t::
+write_record(uint8_t *data, int len)
+{
+ if( recd_fd < 0 ) return;
+ if( write(recd_fd, data, len) < 0 ) {
+ zmsgs("write err: %s\n",strerror(errno));
+ return;
+ }
+ recd_pos += len;
+}
+
+int zmpeg3_t::
+pause_reader(int v)
+{
+ return fs->pause_reader(v);
+}
+
+int zmpeg3_t::
+get_thumbnail(int trk, int64_t &frn, uint8_t *&t, int &w, int &h)
+{
+ int result = -1;
+ if( trk>=0 && trk<total_vtracks ) {
+ vtrack_t *vtrk = vtrack[trk];
+ if( vtrk ) {
+ video_t *vid = vtrk->video;
+ frn = vid->framenum;
+ t = vid->tdat;
+ w = 2*vid->mb_width;
+ h = 2*vid->mb_height;
+ result = 0;
+ }
+ }
+ return result;
+}
+
+int zmpeg3_t::
+set_thumbnail_callback(int trk, int skim, int thumb,
+ zthumbnail_cb fn, void *p)
+{
+ int result = 1;
+ // all tracks, toc build
+ if( trk == -1 )
+ result = 0;
+ // one track, commercials verify
+ else if( trk >= 0 && trk<total_vtracks ) {
+ vtrack_t *vtrk = vtrack[trk];
+ if( vtrk ) {
+ video_t *vid = vtrk->video;
+ if( !(result=vid->seek_video()) ) {
+ vid->skim = skim;
+ vid->thumb = thumb;
+ vid->ithumb = 0;
+ result = 0;
+ }
+ }
+ }
+ if( !result ) {
+ thumbnail_priv = p;
+ thumbnail_fn = fn;
+ }
+ return result;
+}
+
+int zmpeg3_t::
+set_cc_text_callback(int trk, zcc_text_cb fn)
+{
+ int result = -1;
+ if( trk>=0 && trk<total_vtracks ) {
+ vtrack_t *vtrk = vtrack[trk];
+ if( vtrk ) {
+ video_t *vid = vtrk->video;
+ vid->get_cc()->text_cb = fn;
+ result = 0;
+ }
+ }
+ return result;
+}
+
+int64_t zmpeg3_t::
+calculate_source_date(char *path)
+{
+ struct stat64 ostat;
+ memset(&ostat,0,sizeof(struct stat64));
+ return stat64(path, &ostat) < 0 ? 0 : ostat.st_mtime;
+}
+
+int64_t zmpeg3_t::
+calculate_source_date(int fd)
+{
+ struct stat64 ostat;
+ memset(&ostat,0,sizeof(struct stat64));
+ return fstat64(fd, &ostat) < 0 ? 0 : ostat.st_mtime;
+}
+
+void zdebug(int n)
+{
+ zmsgs("%d\n",n);
+}
+
+
projects / goodguy / history.git / blobdiff