Gear_VideoTexture::~Gear_VideoTexture()
{
if (_file!=NULL)
mpeg3_close(_file);
pthread_mutex_destroy(_mutex);
for (int i=0;i<1024;i++)
free(_frame[i]);
}
EXPORT(sqInt) shutdownModule(void) {
sqInt i;
for (i = 1; i <= maximumNumberOfFilesToWatch; i += 1) {
if ((mpegFiles[i]) != 0) {
mpeg3_close(mpegFiles[i]);
mpegFiles[i] = 0;
}
}
return true;
}
int mpeg2enc(int argc, char *argv[])
{
stdin_fd = stdin;
verbose = 1;
/* Read command line */
readcmdline(argc, argv);
/* read quantization matrices */
readquantmat();
if(!strlen(out_path))
{
fprintf(stderr, "No output file given.\n");
}
/* open output file */
if(!(outfile = fopen(out_path, "wb")))
{
sprintf(errortext,"Couldn't create output file %s", out_path);
error(errortext);
}
init();
if(nframes < 0x7fffffff)
printf("Frame Completion Current bitrate Predicted file size\n");
putseq();
stop_slice_engines();
stop_motion_engines();
stop_transform_engines();
stop_itransform_engines();
fclose(outfile);
fclose(statfile);
if(qt_file) quicktime_close(qt_file);
if(qt_output) quicktime_close(qt_output);
if(mpeg_file) mpeg3_close(mpeg_file);
if(do_stdin)
{
fclose(stdin_fd);
}
pthread_mutex_destroy(&input_lock);
pthread_mutex_destroy(&output_lock);
return 0;
}
static void
IDirectFBVideoProvider_Libmpeg3_Destruct( IDirectFBVideoProvider *thiz )
{
IDirectFBVideoProvider_Libmpeg3_data *data;
data = (IDirectFBVideoProvider_Libmpeg3_data*)thiz->priv;
thiz->Stop( thiz );
mpeg3_close( data->file );
D_FREE( data->video.buffer );
D_FREE( data->rgb.lines );
pthread_mutex_destroy( &data->video.lock );
pthread_mutex_destroy( &data->audio.lock );
D_FREE( data->filename );
DIRECT_DEALLOCATE_INTERFACE( thiz );
}
EXPORT(sqInt) primitiveMPEG3Close(void) {
mpeg3_t ** index;
mpeg3_t * file;
sqInt fileHandle;
fileHandle = interpreterProxy->stackValue(0);
if (interpreterProxy->failed()) {
return null;
}
file = mpeg3tValueOf(fileHandle);
if (file == null) {
return null;
}
removeFileEntry(file); mpeg3_close(file);
index = ((mpeg3_t **) (interpreterProxy->firstIndexableField(fileHandle)));
*index = 0;
if (interpreterProxy->failed()) {
return null;
}
interpreterProxy->pop(1);
return null;
}
int mpeg3_generate_toc(FILE *output, char *path, int timecode_search, int print_streams)
{
mpeg3_t *file = mpeg3_open(path);
mpeg3_demuxer_t *demuxer;
int i;
if(file)
{
demuxer = mpeg3_new_demuxer(file, 0, 0, -1);
if(file->is_ifo_file)
{
int i;
mpeg3io_open_file(file->fs);
mpeg3demux_read_ifo(file, demuxer, 1);
mpeg3io_close_file(file->fs);
for(i = 0; i < demuxer->total_titles; i++)
{
fprintf(output, "TOCVERSION %d\n", TOCVERSION);
if(file->is_program_stream)
fprintf(output, "PROGRAM_STREAM\n");
else
if(file->is_transport_stream)
fprintf(output, "TRANSPORT_STREAM\n");
fprintf(output, "PATH: %s\n"
"SIZE: %ld\n"
"PACKETSIZE: %ld\n",
demuxer->titles[i]->fs->path,
demuxer->titles[i]->total_bytes,
file->packet_size);
/* Just print the first title's streams */
if(i == 0)
mpeg3demux_print_streams(demuxer, output);
mpeg3demux_print_timecodes(demuxer->titles[i], output);
}
return 0;
}
else
{
char complete_path[MPEG3_STRLEN];
mpeg3io_complete_path(complete_path, path);
mpeg3demux_create_title(demuxer, timecode_search, output);
fprintf(output, "TOCVERSION %d\n"
"PATH: %s\n", TOCVERSION, complete_path);
if(file->is_program_stream)
fprintf(output, "PROGRAM_STREAM\n");
else
if(file->is_transport_stream)
fprintf(output, "TRANSPORT_STREAM\n");
else
if(file->is_video_stream)
fprintf(output, "VIDEO_STREAM\n");
else
if(file->is_audio_stream)
fprintf(output, "AUDIO_STREAM\n");
/* Just print the first title's streams */
if(print_streams) mpeg3demux_print_streams(demuxer, output);
if(file->is_transport_stream || file->is_program_stream)
{
fprintf(output, "SIZE: %ld\n", demuxer->titles[demuxer->current_title]->total_bytes);
fprintf(output, "PACKETSIZE: %ld\n", file->packet_size);
}
mpeg3demux_print_timecodes(demuxer->titles[demuxer->current_title], output);
return 0;
}
mpeg3_delete_demuxer(demuxer);
mpeg3_close(file);
}
return 1;
}
void Gear_VideoTexture::onUpdateSettings()
{
// Timer timer;
// timer.reset();
char tempstr[1024];
// XXX todo : parameters
double power = 1.0;
int nEpochs = 3;
double alpha = 0.999;
pthread_mutex_lock(_mutex);
// Initialize (open) the movie.
strcpy(tempstr,_settings.get(SETTING_FILENAME)->valueStr().c_str());
std::cout << "opening movie : " << tempstr << std::endl;
if (_file!=NULL)
mpeg3_close(_file);
_file = mpeg3_open(tempstr);
if (_file==NULL)
{
std::cout << "error opening movie : " << tempstr << std::endl;
pthread_mutex_unlock(_mutex);
return;
}
_sizeX = mpeg3_video_width(_file, 0);
_sizeY = mpeg3_video_height(_file, 0);
//_nFrames = (int)CLAMP((long int)_settings.get(SETTING_NFRAMES)->valueInt(), 1l, mpeg3_video_frames(_file, 0));
_nFrames = (int)_settings.get(SETTING_NFRAMES)->valueInt();
ASSERT_ERROR(_nFrames >= 1);
std::cout << "movie size X : " << _sizeX << std::endl;
std::cout << "movie size Y : " << _sizeY << std::endl;
std::cout << "numframes : " << _nFrames << " / " << mpeg3_video_frames(_file, 0) << std::endl;
std::cout << "movie samplerate : " << mpeg3_sample_rate(_file,0) << std::endl;
for (int i=0;i<_sizeY-1;i++)
_frame[i] = (RGBA*) realloc(_frame[i], _sizeX * sizeof(RGBA));
//from the doc :
//You must allocate 4 extra bytes in the last output_row. This is scratch area for the MMX routines.
_frame[_sizeY-1] = (RGBA*) realloc(_frame[_sizeY-1], (_sizeX * sizeof(RGBA)) + 4);
// std::cout << "Time to initialize things: " << timer.getTime() << std::endl;
// timer.reset();
// Fill sequences and distance matrix.
_distances.resize(_nFrames, _nFrames);
_sequences.resize(_nFrames);
_size = _sizeX * _sizeY;
NOTICE("Filling distance and sequence matrix.");
for (int i=0; i<_nFrames; ++i)
{
Array2D<RGBA>& currImage = _sequences[i];
currImage.resize(_sizeX, _sizeY);
// Read current image.
mpeg3_read_frame(_file, (unsigned char**)_frame, 0, 0, _sizeX, _sizeY, _sizeX, _sizeY, MPEG3_RGBA8888, 0);
// Add image to sequences.
for(int y=0;y<_sizeY;y++)
memcpy(currImage.row(y), _frame[y], sizeof(RGBA) * _sizeX);
// Update distance matrix (this is the bottleneck of the whole algorithm).
_distances(i,i) = 0.0;
for (int j=0; j<i; ++j)
_distances(i,j) = _distances(j,i) = L2((unsigned char*)currImage.data(), (unsigned char*)_sequences[j].data(), (size_t)_size*SIZE_RGBA);
}
// std::cout << "Time to fill sequence and distance: " << timer.getTime() << std::endl;
// timer.reset();
#if DEBUG_NOTICE
std::cout << "Distances: " << std::endl;
for (int i=0; i<_nFrames; ++i)
{
for (int j=0; j<_nFrames; ++j)
std::cout << _distances(j,i) << " ";
std::cout << std::endl;
}
std::cout << std::endl;
#endif
// Add temporal coherence to distance matrix by smoothing the distance with linear interpolation.
NOTICE("Computing smoothed distances.");
int timeWindowLength = CLAMP(_settings.get(SETTING_TIMEWINDOWLENGTH)->valueInt(), 0, 2);
_smoothedDistances.resize(_nFrames, _nFrames);
switch (timeWindowLength)
{
case 0:
memcpy(_smoothedDistances.data(), _distances.data(), _distances.size());
break;
case 1:
for (int i=0; i<_nFrames; ++i)
{
for (int j=0; j<_nFrames; ++j)
{
int iPrev = MAX(i-1,0);
int jPrev = MAX(j-1,0);
_smoothedDistances(j,i) =
0.5 * _distances(jPrev,iPrev) +
0.5 * _distances(j,i);
}
}
break;
case 2:
for (int i=0; i<_nFrames; ++i)
{
for (int j=0; j<_nFrames; ++j)
{
int iPrev = MAX(i-1,0);
int jPrev = MAX(j-1,0);
int iNext = MIN(i+1,_nFrames-1);
int jNext = MIN(j+1,_nFrames-1);
int iPrev2 = MAX(i-2,0);
int jPrev2 = MAX(j-2,0);
_smoothedDistances(j,i) =
0.125 * _distances(jPrev2,iPrev2) +
0.375 * _distances(jPrev,iPrev) +
0.375 * _distances(j,i) +
0.125 * _distances(jNext,iNext);
}
}
break;
default:;
error("Wrong time window length specified, please check");
}
// std::cout << "Time to compute smoothed distances: " << timer.getTime() << std::endl;
// timer.reset();
#if DEBUG_NOTICE
std::cout << "Smoothed distances: " << std::endl;
for (int i=0; i<_nFrames; ++i)
{
for (int j=0; j<_nFrames; ++j)
std::cout << _smoothedDistances(i,j) << " ";
std::cout << std::endl;
}
std::cout << std::endl;
#endif
// Using Q-learning, recompute the matrix of distances.
NOTICE("Computing final set of distances using Q-learning.");
_minDistances.resize(_nFrames);
// Initialize distances.
for (int i=0; i<_nFrames; ++i)
for (int j=0; j<=i; ++j)
_distances(i,j) = _distances(j,i) = _smoothedDistances(i,j) = _smoothedDistances(j,i) = pow(_smoothedDistances(i,j), power);
// Q-learning.
for (int t=0; t<nEpochs; ++t)
{
#if DEBUG_NOTICE
std::cout << "Q-learn distances step " << t << " : " << std::endl;
for (int i=0; i<_nFrames; ++i)
{
for (int j=0; j<_nFrames; ++j)
std::cout << _distances(j,i) << " ";
std::cout << std::endl;
}
std::cout << std::endl;
#endif
// Init min distances.
for (int j=0; j<_nFrames; ++j)
_minDistances[j] = min(_distances.row(j), (size_t)_nFrames);
for (int i=_nFrames-1; i>=0; --i)
for (int j=0; j<_nFrames; ++j)
{
// Update distances.
_distances(j,i) = _smoothedDistances(j,i) + alpha * _minDistances[j];
// Update min distances.
_minDistances[j] = min(_distances.row(j), (size_t)_nFrames);
}
}
// std::cout << "Time to compute Q-learned distances: " << timer.getTime() << std::endl;
// timer.reset();
// Calculate the mean (smoothed) distance.
double meanDistance = sum(_distances.data(), _distances.size()) / (double)_distances.size();
// Compute cumulative probabilities.
_logCumProbs.resize(_nFrames, _nFrames);
for (int i=0; i<_nFrames; ++i)
{
// Compute logCumProbs.
_logCumProbs(0,i) = LOG_ZERO;
for (int j=1; j<_nFrames; ++j)
_logCumProbs(j,i) = logAdd(_logCumProbs(j-1,i), -_distances(j,i) / meanDistance);
// Normalize to make a true probability.
double norm = _logCumProbs(_nFrames-1,i);
for (int j=0; j<_nFrames; ++j)
_logCumProbs(j,i) -= norm;
}
// std::cout << "Time to compute probabilities: "<< timer.getTime() << std::endl;
#if DEBUG_NOTICE
std::cout << "Distances: " << std::endl;
for (int i=0; i<_nFrames; ++i)
{
for (int j=0; j<_nFrames; ++j)
std::cout << _distances(j,i) << " ";
std::cout << std::endl;
}
std::cout << std::endl;
std::cout << "Probabilities: " << std::endl;
for (int i=0; i<_nFrames; ++i)
{
for (int j=0; j<_nFrames; ++j)
std::cout << exp(_logCumProbs(j,i)) << " ";
std::cout << std::endl;
}
std::cout << std::endl;
#endif
_currentFrame = 0;
pthread_mutex_unlock(_mutex);
}
int main2()
{
mpeg3_t *file;
int i, result = 0;
unsigned char *output, **output_rows;
float *audio_output_f;
short *audio_output_i;
long total_samples = 0;
Rect sourceRect;
OSErr error;
PixMapHandle hPixmap;
Ptr gBaseLocation;
long targetRowBytes;
file = mpeg3_open("randomAlien.mpg");
if(file)
{
mpeg3_set_cpus(file, 1);
//audio_output_f = (float *) memoryAllocate(1,BUFSIZE * sizeof(float));
//audio_output_i = (short *) memoryAllocate(1,BUFSIZE * sizeof(short));
//mpeg3_set_sample(file, 11229518, 0);
//result = mpeg3_read_audio(file, audio_output_f, 0, 0, BUFSIZE, 0);
// result = mpeg3_read_audio(file, 0, audio_output_i, 1, BUFSIZE, 0);
// fwrite(audio_output_i, BUFSIZE, 1, stdout);
//mpeg3_set_frame(file, 1000, 0);
sourceRect.top = 0;
sourceRect.left = 0;
sourceRect.bottom = mpeg3_video_height(file, 0);
sourceRect.right = mpeg3_video_width(file, 0);
error = NewGWorld (&gpGWOffScreen, 32, &sourceRect, NULL, NULL, keepLocal);
if (error != noErr)
{
DebugStr ("\pUnable to allocate off screen image");
}
hPixmap = GetGWorldPixMap (gpGWOffScreen);
error = LockPixels (hPixmap);
gBaseLocation = GetPixBaseAddr(hPixmap);
targetRowBytes = ((**hPixmap).rowBytes & 0x3FFF)/4;
output_rows = (unsigned char **) memoryAllocate(1,sizeof(unsigned char*) * mpeg3_video_height(file, 0));
for(i = 0; i < mpeg3_video_height(file, 0); i++)
output_rows[i] = (unsigned char*) gBaseLocation + i * targetRowBytes*4;
for (i=0;i < mpeg3_video_frames(file, 0);i++) {
result = mpeg3_read_frame(file,
output_rows,
0,
0,
mpeg3_video_width(file, 0),
mpeg3_video_height(file, 0),
mpeg3_video_width(file, 0),
mpeg3_video_height(file, 0),
MPEG3_RGBAF8888,
0);
CopyBits (GetPortBitMapForCopyBits(gpGWOffScreen), GetPortBitMapForCopyBits(GetWindowPort(pWindow)), &sourceRect, &sourceRect, srcCopy, NULL);
}
UnlockPixels (hPixmap);
DisposeGWorld(gpGWOffScreen);
memoryFree(output_rows);
fprintf(stderr, "Audio streams: %d\n", mpeg3_total_astreams(file));
for(i = 0; i < mpeg3_total_astreams(file); i++)
{
fprintf(stderr, " Stream %d: channels %d sample rate %d total samples %ld\n",
i,
mpeg3_audio_channels(file, i),
mpeg3_sample_rate(file, i),
mpeg3_audio_samples(file, i));
}
fprintf(stderr, "Video streams: %d\n", mpeg3_total_vstreams(file));
for(i = 0; i < mpeg3_total_vstreams(file); i++)
{
fprintf(stderr, " Stream %d: width %d height %d frame rate %0.3f total frames %ld\n",
i,
mpeg3_video_width(file, i),
mpeg3_video_height(file, i),
mpeg3_frame_rate(file, i),
mpeg3_video_frames(file, i));
}
mpeg3_close(file);
}
int main(int argc, char *argv[])
{
int error = 0;
int frame_count = -1;
char *row_pointers[3];
int do_audio = 0;
int do_video = 0;
int channels = 0;
long afragment = 65536;
float **audio_output;
int layer = 0;
int astream = 0;
char input_path[1024];
char output_path[1024];
int i;
long current_frame = 0;
long current_sample = 0;
pthread_mutex_init(&mutex, NULL);
signal(SIGINT, trap_interrupt);
input_path[0] = 0;
output_path[0] = 0;
if(argc < 3)
{
printf("Usage: %s [-a] [-v] <mpeg file> <output movie> [frame count]\n", argv[0]);
exit(1);
}
for(i = 1; i < argc; i++)
{
if(!strcmp(argv[i], "-a"))
{
do_audio = 1;
}
else
if(!strcmp(argv[i], "-v"))
{
do_video = 1;
}
else
if(!input_path[0])
{
strcpy(input_path, argv[i]);
}
else
if(!output_path[0])
{
strcpy(output_path, argv[i]);
}
else
frame_count = atol(argv[i]);
}
//printf("main 1\n");
if(!(input = mpeg3_open(input_path, &error)))
{
exit(1);
}
//printf("main 1\n");
if(!(output = quicktime_open(output_path, 0, 1)))
{
exit(1);
}
//printf("main 1\n");
if(do_video)
{
if(!mpeg3_total_vstreams(input))
{
do_video = 0;
}
else
{
quicktime_set_video(output,
1,
mpeg3_video_width(input, layer),
mpeg3_video_height(input, layer),
mpeg3_frame_rate(input, layer),
VIDEO_CODEC);
quicktime_set_jpeg(output, 80, 0);
}
}
//printf("main 1\n");
if(do_audio)
{
if(!mpeg3_total_astreams(input))
{
do_audio = 0;
}
else
{
int i;
channels = mpeg3_audio_channels(input, astream);
quicktime_set_audio(output,
channels,
mpeg3_sample_rate(input, 0),
24,
AUDIO_CODEC);
audio_output = malloc(sizeof(float*) * channels);
for(i = 0; i < channels; i++)
audio_output[i] = malloc(sizeof(float) * afragment);
}
}
//printf("main 1\n");
// quicktime_set_jpeg(output, 100, 0);
// mpeg3_set_mmx(input, 0);
while((!(do_video && mpeg3_end_of_video(input, layer)) ||
!(do_audio && mpeg3_end_of_audio(input, astream))) &&
(current_frame < frame_count || frame_count < 0))
{
//printf("%d %d\n", mpeg3_end_of_video(input, layer), mpeg3_end_of_audio(input, astream));
if(do_audio)
{
if(!mpeg3_end_of_audio(input, astream))
{
int fragment = afragment;
int i, j, k;
i = astream;
k = 0;
for(j = 0; j < mpeg3_audio_channels(input, i); j++, k++)
{
if(j == 0)
mpeg3_read_audio(input,
audio_output[k], /* Pointer to pre-allocated buffer of floats */
0, /* Pointer to pre-allocated buffer of int16's */
j, /* Channel to decode */
fragment, /* Number of samples to decode */
i);
else
mpeg3_reread_audio(input,
audio_output[k], /* Pointer to pre-allocated buffer of floats */
0, /* Pointer to pre-allocated buffer of int16's */
j, /* Channel to decode */
fragment, /* Number of samples to decode */
i);
}
quicktime_encode_audio(output,
0,
audio_output,
fragment);
current_sample += fragment;
if(!do_video)
{
printf(" %d samples written\r", current_sample);
fflush(stdout);
}
}
else
current_sample += afragment;
}
if(do_video)
{
if(!mpeg3_end_of_video(input, layer))
{
int fragment;
if(do_audio)
fragment = (long)((double)current_sample /
mpeg3_sample_rate(input, 0) *
mpeg3_frame_rate(input, layer) -
current_frame);
else
fragment = 1;
for(i = 0; i < fragment && !mpeg3_end_of_video(input, layer); i++)
{
mpeg3_read_yuvframe_ptr(input,
&row_pointers[0],
&row_pointers[1],
&row_pointers[2],
layer);
switch(mpeg3_colormodel(input, layer))
{
case MPEG3_YUV420P:
quicktime_set_cmodel(output, BC_YUV420P);
break;
case MPEG3_YUV422P:
quicktime_set_cmodel(output, BC_YUV422P);
break;
}
quicktime_encode_video(output,
(unsigned char **)row_pointers,
0);
current_frame++;
printf(" %d frames written\r", current_frame);
fflush(stdout);
}
}
}
}
printf("main 2\n");
printf("\n");
quicktime_close(output);
mpeg3_close(input);
}