Roku Developer Program

Join our online forum to talk to Roku developers and fellow channel creators. Ask questions, share tips with the community, and find helpful resources.
cancel
Showing results for 
Show  only  | Search instead for 
Did you mean: 
SolveLLC
Visitor

Proper support for Ioncannon segmenter

Would someone at Roku take a look at the Ioncannon segmenter (http://svn.assembla.com/svn/legend/segmenter/). We've done extensive testing with this tool, and it works properly on the iThings, but causes issues with the Roku, mainly that it doesn't enforce segmentation on I-frame boundaries. There are really two options for this, change what Roku's doing (probably not the best option) or let us know what the issue is so that we can work with the developer to correct the bug.

I realize that Roku is busy, but this tool, working on the Roku would be very useful to quite a few I would imagine. Our current options are to place a OS X server in the data center *just* to go segmenting or go with someone like Inlet, which has offered my first born as a trade. Both are expenses that don't really make sense for the simple thing that we are trying to do.
0 Kudos
6 REPLIES 6
retrotom
Visitor

Re: Proper support for Ioncannon segmenter

Here's the segmenter code we use for Gabby. We added the necessary "break on I-Frame"...uhh feature... as well as some other stuff. To be honest, the I-Frame boundary makes sense, without it you have to seek forwards/backwards until you hit one because you can't render P-Frames (or B-Frames) without them. I thought the IonCannon segmenter did break on I-Frames though. Anyways...here's the code. Keep in mind that it's part of a larger project so you may have to adapt it to whatever you're doing.


#pragma once

extern "C"{
#include "libavformat/avformat.h"
}

struct config_info
{
const char *input_filename;
int segment_length;
const char *temp_directory;
const char *filename_prefix;
int start_index;
double start_time;
char *segmentIndexFileLocation;
};

class CSegmenter
{
private:

FILE* segmentIndexFile;

AVStream *add_output_stream(AVFormatContext *output_format_context, AVStream *input_stream)
{
AVCodecContext *input_codec_context;
AVCodecContext *output_codec_context;
AVStream *output_stream;

output_stream = av_new_stream(output_format_context, 0);
if (!output_stream)
{
fprintf(stderr, "Segmenter error: Could not allocate stream\n");
__exit(1);
}

input_codec_context = input_stream->codec;
output_codec_context = output_stream->codec;

output_codec_context->codec_id = input_codec_context->codec_id;
output_codec_context->codec_type = input_codec_context->codec_type;
output_codec_context->codec_tag = input_codec_context->codec_tag;
output_codec_context->bit_rate = input_codec_context->bit_rate;
output_codec_context->extradata = input_codec_context->extradata;
output_codec_context->extradata_size = input_codec_context->extradata_size;

if(av_q2d(input_codec_context->time_base) * input_codec_context->ticks_per_frame > av_q2d(input_stream->time_base) && av_q2d(input_stream->time_base) < 1.0/1000)
{
output_codec_context->time_base = input_codec_context->time_base;
output_codec_context->time_base.num *= input_codec_context->ticks_per_frame;
}
else
{
output_codec_context->time_base = input_stream->time_base;
}

switch (input_codec_context->codec_type)
{
case CODEC_TYPE_AUDIO:
output_codec_context->channel_layout = input_codec_context->channel_layout;
output_codec_context->sample_rate = input_codec_context->sample_rate;
output_codec_context->channels = input_codec_context->channels;
output_codec_context->frame_size = input_codec_context->frame_size;
if ((input_codec_context->block_align == 1 && input_codec_context->codec_id == CODEC_ID_MP3) || input_codec_context->codec_id == CODEC_ID_AC3)
{
output_codec_context->block_align = 0;
}
else
{
output_codec_context->block_align = input_codec_context->block_align;
}
break;
case CODEC_TYPE_VIDEO:
output_codec_context->pix_fmt = input_codec_context->pix_fmt;
output_codec_context->width = input_codec_context->width;
output_codec_context->height = input_codec_context->height;
output_codec_context->has_b_frames = input_codec_context->has_b_frames;

if (output_format_context->oformat->flags & AVFMT_GLOBALHEADER)
{
output_codec_context->flags |= CODEC_FLAG_GLOBAL_HEADER;
}
break;
default:
break;
}

return output_stream;
}

void __exit(int exitCode)
{
if(segmentIndexFile)
{
_fclose_nolock(segmentIndexFile);
segmentIndexFile = NULL;
}

ExitThread(exitCode);
}

public:

CSegmenter(void)
{
segmentIndexFile = NULL;
}

~CSegmenter(void)
{
if(segmentIndexFile)
{
_fclose_nolock(segmentIndexFile);
segmentIndexFile = NULL;
}
}

int Run(int argc, char** argv)
{
if(argc < 4)
{
fprintf(stderr, "Usage: %s <input filename> <segment length> <output location> <filename prefix> <start segment> <start time>\n", "segmenter.exe");
return 1;
}

struct config_info config;

memset(&config, 0, sizeof(struct config_info));

config.input_filename = argv[0];
config.segment_length = atoi(argv[1]);
config.temp_directory = argv[2];
config.filename_prefix = argv[3];

if(argc > 5)
{
config.start_index = atoi(argv[4]);
config.start_time = atof(argv[5]);
}
else
{
config.start_index = 0;
config.start_time = 0;
}

//setup the segment index file
config.segmentIndexFileLocation = (char*)malloc(sizeof(char)*1024);
memset(config.segmentIndexFileLocation, 0, 1024);
snprintf(config.segmentIndexFileLocation, strlen(config.temp_directory) + 10, "%s\\%s", config.temp_directory, "index.xml");

char *output_filename = (char*)malloc(sizeof(char) * (strlen(config.temp_directory) + 1 + strlen(config.filename_prefix) + 10));
if (!output_filename)
{
fprintf(stderr, "Segmenter error: Could not allocate space for output filenames\n");
__exit(1);
}

segmentIndexFile = fopen(config.segmentIndexFileLocation, config.start_index ? "a" : "w");
if(segmentIndexFile == NULL)
{
fprintf(stderr, "Segmenter error: Could not find open index file for writing\n");
__exit(1);
}

AVInputFormat *input_format = av_find_input_format("mpegts");
if (!input_format)
{
fprintf(stderr, "Segmenter error: Could not find MPEG-TS demuxer\n");
__exit(1);
}

AVFormatContext *input_context = NULL;
int ret = av_open_input_file(&input_context, config.input_filename, input_format, 0, NULL);
if (ret != 0)
{
fprintf(stderr, "Segmenter error: Could not open input file, make sure it is an mpegts file: %d\n", ret);
__exit(1);
}

if (av_find_stream_info(input_context) < 0)
{
fprintf(stderr, "Segmenter error: Could not read stream information\n");
__exit(1);
}

#if LIBAVFORMAT_VERSION_MAJOR >= 52 && LIBAVFORMAT_VERSION_MINOR >= 45
AVOutputFormat *output_format = av_guess_format("mpegts", NULL, NULL);
#else
AVOutputFormat *output_format = guess_format("mpegts", NULL, NULL);
#endif
if (!output_format)
{
fprintf(stderr, "Segmenter error: Could not find MPEG-TS muxer\n");
__exit(1);
}

AVFormatContext *output_context = avformat_alloc_context();
if (!output_context)
{
fprintf(stderr, "Segmenter error: Could not allocated output context");
__exit(1);
}
output_context->oformat = output_format;

int video_index = -1;
int audio_index = -1;

AVStream *video_stream;
AVStream *audio_stream;

int i;

for (i = 0; i < input_context->nb_streams && (video_index < 0 || audio_index < 0); i++)
{
switch (input_context->streams[i]->codec->codec_type) {
case CODEC_TYPE_VIDEO:
video_index = i;
input_context->streams[i]->discard = AVDISCARD_NONE;
video_stream = add_output_stream(output_context, input_context->streams[i]);
break;
case CODEC_TYPE_AUDIO:
audio_index = i;
input_context->streams[i]->discard = AVDISCARD_NONE;
audio_stream = add_output_stream(output_context, input_context->streams[i]);
break;
default:
input_context->streams[i]->discard = AVDISCARD_ALL;
break;
}
}

if (av_set_parameters(output_context, NULL) < 0)
{
fprintf(stderr, "Segmenter error: Invalid output format parameters\n");
__exit(1);
}

dump_format(output_context, 0, config.filename_prefix, 1);

if(video_index >= 0)
{
AVCodec *codec = avcodec_find_decoder(video_stream->codec->codec_id);
if (!codec)
{
fprintf(stderr, "Segmenter error: Could not find video decoder, key frames will not be honored\n");
}

if (avcodec_open(video_stream->codec, codec) < 0)
{
fprintf(stderr, "Segmenter error: Could not open video decoder, key frames will not be honored\n");
}
}

unsigned int output_index = config.start_index ? config.start_time ? 99999 : config.start_index + 1 : 1;
snprintf(output_filename, strlen(config.temp_directory) + 1 + strlen(config.filename_prefix) + 10, "%s\\%s-%05u.ts", config.temp_directory, config.filename_prefix, output_index++);
if (url_fopen(&output_context->pb, output_filename, URL_WRONLY) < 0)
{
fprintf(stderr, "Segmenter error: Could not open '%s'\n", output_filename);
__exit(1);
}

if (av_write_header(output_context))
{
fprintf(stderr, "Segmenter error: Could not write mpegts header to first output file\n");
__exit(1);
}

unsigned int first_segment = 1;
unsigned int last_segment = 0;
unsigned int videoFrameCount = 0;
int64_t lastValidPts = -9223372036854775808;
int lastDuration = 0;
unsigned int first_video_packet = 1;
AVFrac old_video_pts;

double prev_segment_time = 0;
int decode_done;

do
{
double segment_time = 0;
AVPacket packet;

decode_done = av_read_frame(input_context, &packet);
if (decode_done < 0)
{
break;
}

if (av_dup_packet(&packet) < 0)
{
fprintf(stderr, "Segmenter error: Could not duplicate packet");
av_free_packet(&packet);
break;
}

if (packet.stream_index == video_index)
{
if((packet.flags & AV_PKT_FLAG_KEY))
{
if(video_stream->pts.val)
segment_time = (double)video_stream->pts.val * video_stream->time_base.num / video_stream->time_base.den;
}
else
{
segment_time = prev_segment_time;
}

if(packet.pts == -9223372036854775808)
{
if(lastValidPts != -9223372036854775808)
packet.pts = lastValidPts + lastDuration;
}
else
{
lastDuration = packet.duration;
lastValidPts = packet.pts;
}

first_video_packet = 0;
++videoFrameCount;
}
else if (video_index < 0)
{
segment_time = (double)audio_stream->pts.val * audio_stream->time_base.num / audio_stream->time_base.den;
}
else
{
segment_time = prev_segment_time;
}

//we need to skip a certain amount of time before creating the segment files
if(config.start_time)
{
//are we lower than skip time
if(segment_time < config.start_time)
{
av_interleaved_write_frame(output_context, &packet);
goto __free_av_packet; //just free the packet
}
else
{
config.start_time = 0; // set the skip time to zero; write packet and proceed normally
output_index = config.start_index - 1;
}
}

// done writing the current file?
if ((int)(segment_time - prev_segment_time) >= config.segment_length)
{
old_video_pts = video_stream->pts;

put_flush_packet(output_context->pb);
av_write_trailer(output_context);
url_fclose(output_context->pb);

//update the index
if(segmentIndexFile){
fprintf(segmentIndexFile, "<segment index=\"%d\" file=\"%s\" duration=\"%f\" />\n", output_index - 1, output_filename, (float)(segment_time - prev_segment_time));
fflush(segmentIndexFile);
}

snprintf(output_filename, strlen(config.temp_directory) + 1 + strlen(config.filename_prefix) + 10, "%s\\%s-%05u.ts", config.temp_directory, config.filename_prefix, output_index++);
if (url_fopen(&output_context->pb, output_filename, URL_WRONLY) < 0)
{
fprintf(stderr, "Segmenter error: Could not open '%s'\n", output_filename);
break;
}
else if (av_write_header(output_context))
{
fprintf(stderr, "Segmenter error: Could not write mpegts header to output file\n");
__exit(1);
}

prev_segment_time = segment_time;
}

ret = av_interleaved_write_frame(output_context, &packet);
if (ret < 0)
{
fprintf(stderr, "Segmenter error: Could not write frame of stream: %d\n", ret);
}
else if (ret > 0)
{
fprintf(stderr, "Segmenter info: End of stream requested\n");
av_free_packet(&packet);
break;
}

__free_av_packet:
av_free_packet(&packet);
} while (!decode_done);

av_write_trailer(output_context);

if (video_index >= 0)
{
avcodec_close(video_stream->codec);
}

for(i = 0; i < output_context->nb_streams; i++)
{
av_freep(&output_context->streams[i]->codec);
av_freep(&output_context->streams[i]);
}

url_fclose(output_context->pb);
av_free(output_context);

return 0;
}
};
0 Kudos
SolveLLC
Visitor

Re: Proper support for Ioncannon segmenter

Just wanted to "bump" this hoping that Kevin or Patrick could give some insight.
0 Kudos
RokuKevin
Visitor

Re: Proper support for Ioncannon segmenter

Yes. We require segmenting on I-Frame boundries, and last time I checked (about 3 months ago) the ion cannon segmenter did not do this properly.

The segmenting tools on the Mac have been most robust entry level solution I have found.

I suggest you try retrotom's segmenter and report back your results. He's using the same libs as ffmpeg, so it could be a good start to an open source segmenting solution if others would like to test and build on it.

--Kevin
0 Kudos
brandon15811
Visitor

Re: Proper support for Ioncannon segmenter

Heres another open source segmenter:
http://www.ioncannon.net/projects/http- ... stributor/
0 Kudos
SolveLLC
Visitor

Re: Proper support for Ioncannon segmenter

"retrotom" wrote:
Here's the segmenter code we use for Gabby. We added the necessary "break on I-Frame"...uhh feature... as well as some other stuff. To be honest, the I-Frame boundary makes sense, without it you have to seek forwards/backwards until you hit one because you can't render P-Frames (or B-Frames) without them. I thought the IonCannon segmenter did break on I-Frames though. Anyways...here's the code. Keep in mind that it's part of a larger project so you may have to adapt it to whatever you're doing.


This doesn't compile for me on Debian and I'm not much of a C++ coder. Any tips on getting it compiled?
0 Kudos
retrotom
Visitor

Re: Proper support for Ioncannon segmenter

"SolveLLC" wrote:

This doesn't compile for me on Debian and I'm not much of a C++ coder. Any tips on getting it compiled?


I have some code in there that's Windows specific -- which is why I said you'd need to adapt it to whatever you're doing. Most likely, in the the __exit function and the destructor, you need to change ExitThread to exit and _fclose_nolock to plain ol' fclose.

So the functions would look like this:

void __exit(int exitCode)
{
if(segmentIndexFile)
{
fclose(segmentIndexFile);
segmentIndexFile = NULL;
}

exit(exitCode);
}

~CSegmenter(void)
{
if(segmentIndexFile)
{
fclose(segmentIndexFile);
segmentIndexFile = NULL;
}
}


I believe those are the only windows-specific things that you'll need to change. Also, this file is a class...and not a program by itself. It could be adapted to run as such, but it's expected to be called by other code. You should check out some C++ tutorials in order to figure how to do that.
0 Kudos