Message ID | 20190517042151.lhjrqi6lm2yayx43@manj |
---|---|
State | Withdrawn |
Headers | show |
On Thu, May 16, 2019 at 9:21 PM Andriy Gelman <andriy.gelman@gmail.com> wrote: > On Thu, 16. May 18:28, Jun Li wrote: > > On Thu, May 16, 2019 at 12:54 PM Andriy Gelman <andriy.gelman@gmail.com> > > wrote: > > > > > Hi Jun, > > > > > > On Thu, 16. May 00:12, Jun Li wrote: > > > > Fix #6945 > > > > Current implementaion for autorotate works fine for stream > > > > level rotataion but no support for frame level operation > > > > and frame flip. This patch is for adding flip support and > > > > per frame operations. > > > > --- > > > > fftools/cmdutils.c | 9 ++--- > > > > fftools/cmdutils.h | 2 +- > > > > fftools/ffmpeg.c | 21 ++++++++++- > > > > fftools/ffmpeg.h | 2 + > > > > fftools/ffmpeg_filter.c | 81 > ++++++++++++++++++++++++++++++++++++++--- > > > > fftools/ffplay.c | 28 +++++++++++--- > > > > 6 files changed, 126 insertions(+), 17 deletions(-) > > > > > > > > diff --git a/fftools/cmdutils.c b/fftools/cmdutils.c > > > > index 9cfbc45c2b..b8bb904419 100644 > > > > --- a/fftools/cmdutils.c > > > > +++ b/fftools/cmdutils.c > > > > @@ -2172,13 +2172,12 @@ void *grow_array(void *array, int elem_size, > int > > > *size, int new_size) > > > > return array; > > > > } > > > > > > > > -double get_rotation(AVStream *st) > > > > +double get_rotation_hflip(int32_t display_matrix[9], int* hflip) > > > > { > > > > - uint8_t* displaymatrix = av_stream_get_side_data(st, > > > > - > > > AV_PKT_DATA_DISPLAYMATRIX, NULL); > > > > double theta = 0; > > > > - if (displaymatrix) > > > > - theta = -av_display_rotation_get((int32_t*) displaymatrix); > > > > + > > > > + if (display_matrix) > > > > + theta = -av_display_rotation_hflip_get(display_matrix, > hflip); > > > > > > > > theta -= 360*floor(theta/360 + 0.9/360); > > > > > > > > diff --git a/fftools/cmdutils.h b/fftools/cmdutils.h > > > > index 6e2e0a2acb..dce44ed6e1 100644 > > > > --- a/fftools/cmdutils.h > > > > +++ b/fftools/cmdutils.h > > > > @@ -643,6 +643,6 @@ void *grow_array(void *array, int elem_size, int > > > *size, int new_size); > > > > char name[128];\ > > > > av_get_channel_layout_string(name, sizeof(name), 0, ch_layout); > > > > > > > > -double get_rotation(AVStream *st); > > > > +double get_rotation_hflip(int32_t display_matrix[9], int* hflip); > > > > > > > > #endif /* FFTOOLS_CMDUTILS_H */ > > > > diff --git a/fftools/ffmpeg.c b/fftools/ffmpeg.c > > > > index 01f04103cf..9ea1aaa930 100644 > > > > --- a/fftools/ffmpeg.c > > > > +++ b/fftools/ffmpeg.c > > > > @@ -2126,6 +2126,24 @@ static int > > > ifilter_has_all_input_formats(FilterGraph *fg) > > > > return 1; > > > > } > > > > > > > > +static int ifilter_display_matrix_need_from_frame(InputFilter > *ifilter, > > > AVFrame* frame) > > > > +{ > > > > + int32_t* stream_matrix = > > > (int32_t*)av_stream_get_side_data(ifilter->ist->st, > > > > + > > > AV_PKT_DATA_DISPLAYMATRIX, NULL); > > > > + // if we already have display matrix from stream, use it > instead of > > > extracting > > > > + // from frame. > > > > + if (stream_matrix) { > > > > + memcpy(ifilter->display_matrix, stream_matrix, 4 * 9); > > > > + return 0; > > > > + } > > > > + > > > > + // cleanup the display matrix, may be from last frame > > > > + memset(ifilter->display_matrix, 0, 4 * 9); > > > > + av_display_rotation_set(ifilter->display_matrix, 0); > > > > + > > > > + return !!av_dict_get(frame->metadata, "Orientation", NULL, 0); > > > > +} > > > > + > > > > static int ifilter_send_frame(InputFilter *ifilter, AVFrame *frame) > > > > { > > > > FilterGraph *fg = ifilter->graph; > > > > @@ -2141,7 +2159,8 @@ static int ifilter_send_frame(InputFilter > > > *ifilter, AVFrame *frame) > > > > ifilter->channel_layout != > frame->channel_layout; > > > > break; > > > > case AVMEDIA_TYPE_VIDEO: > > > > - need_reinit |= ifilter->width != frame->width || > > > > + need_reinit |= > ifilter_display_matrix_need_from_frame(ifilter, > > > frame) || > > > > + ifilter->width != frame->width || > > > > ifilter->height != frame->height; > > > > break; > > > > } > > > > diff --git a/fftools/ffmpeg.h b/fftools/ffmpeg.h > > > > index eb1eaf6363..8331720663 100644 > > > > --- a/fftools/ffmpeg.h > > > > +++ b/fftools/ffmpeg.h > > > > @@ -251,6 +251,8 @@ typedef struct InputFilter { > > > > int channels; > > > > uint64_t channel_layout; > > > > > > > > + int32_t display_matrix[9]; > > > > + > > > > AVBufferRef *hw_frames_ctx; > > > > > > > > int eof; > > > > diff --git a/fftools/ffmpeg_filter.c b/fftools/ffmpeg_filter.c > > > > index 72838de1e2..1894f30561 100644 > > > > --- a/fftools/ffmpeg_filter.c > > > > +++ b/fftools/ffmpeg_filter.c > > > > @@ -328,6 +328,7 @@ static void init_input_filter(FilterGraph *fg, > > > AVFilterInOut *in) > > > > fg->inputs[fg->nb_inputs - 1]->format = -1; > > > > fg->inputs[fg->nb_inputs - 1]->type = > ist->st->codecpar->codec_type; > > > > fg->inputs[fg->nb_inputs - 1]->name = describe_filter_link(fg, > in, > > > 1); > > > > + av_display_rotation_set(fg->inputs[fg->nb_inputs - > > > 1]->display_matrix, 0); > > > > > > > > fg->inputs[fg->nb_inputs - 1]->frame_queue = av_fifo_alloc(8 * > > > sizeof(AVFrame*)); > > > > if (!fg->inputs[fg->nb_inputs - 1]->frame_queue) > > > > @@ -807,22 +808,43 @@ static int > > > configure_input_video_filter(FilterGraph *fg, InputFilter *ifilter, > > > > last_filter = ifilter->filter; > > > > > > > > if (ist->autorotate) { > > > > - double theta = get_rotation(ist->st); > > > > + int hflip = 0; > > > > + double theta = get_rotation_hflip(ifilter->display_matrix, > > > &hflip); > > > > > > > > - if (fabs(theta - 90) < 1.0) { > > > > + if (fabs(theta) < 1.0) { > > > > + if (hflip) > > > > + ret = insert_filter(&last_filter, &pad_idx, "hflip", > > > NULL); > > > > + } else if (fabs(theta - 90) < 1.0) { > > > > ret = insert_filter(&last_filter, &pad_idx, "transpose", > > > "clock"); > > > > - } else if (fabs(theta - 180) < 1.0) { > > > > - ret = insert_filter(&last_filter, &pad_idx, "hflip", > NULL); > > > > if (ret < 0) > > > > return ret; > > > > - ret = insert_filter(&last_filter, &pad_idx, "vflip", > NULL); > > > > + if (hflip) > > > > + ret = insert_filter(&last_filter, &pad_idx, "hflip", > > > NULL); > > > > + } else if (fabs(theta - 180) < 1.0) { > > > > + if (hflip) { // rotate 180 and hflip equals vflip > > > > + ret = insert_filter(&last_filter, &pad_idx, "vflip", > > > NULL); > > > > + } else { > > > > + ret = insert_filter(&last_filter, &pad_idx, "hflip", > > > NULL); > > > > + if (ret < 0) > > > > + return ret; > > > > + ret = insert_filter(&last_filter, &pad_idx, "vflip", > > > NULL); > > > > + } > > > > } else if (fabs(theta - 270) < 1.0) { > > > > ret = insert_filter(&last_filter, &pad_idx, "transpose", > > > "cclock"); > > > > + if (ret < 0) > > > > + return ret; > > > > + if (hflip) > > > > + ret = insert_filter(&last_filter, &pad_idx, "hflip", > > > NULL); > > > > } else if (fabs(theta) > 1.0) { > > > > char rotate_buf[64]; > > > > snprintf(rotate_buf, sizeof(rotate_buf), "%f*PI/180", > > > theta); > > > > ret = insert_filter(&last_filter, &pad_idx, "rotate", > > > rotate_buf); > > > > + if (ret < 0) > > > > + return ret; > > > > + if (hflip) > > > > + ret = insert_filter(&last_filter, &pad_idx, "hflip", > > > NULL); > > > > } > > > > > > Just a thought, if you model your transformation matrix as M = RH, > where H > > > is > > > the horizontal flip and R is the rotation, is it possible to simplify > the > > > logic > > > here to: > > > > > > //your H transformation > > > if (hflip) > > > ret = insert_filter(&last_filter, &pad_idx, "hflip", NULL); > > > > > > //the R transformation > > > same code as before.. > > > > > > > Sorry Andriy, but I donot quite get it. Do you mean put this code > > //your H transformation > > if (hflip) > > ret = insert_filter(&last_filter, &pad_idx, "hflip", NULL); > > outside of the if check ? > > Sorry I wasn't clear. Before your patch, this part of the code was just > applying a rotation. > > With your patch, you are proposing a horizontal mirror operation and a > rotation. > These are two separate operations. So, I do not see a reason to add extra > logic > to the rotation part of the code. > > I attach a diff applied to your patch set V2 to show what I mean. Note > that I > had to change a couple of things on how you define transformation for case > 5 and > 7 (since mirror horizontal is second argument). I tested on the examples > from ticket #6945 and this worked. > > P.s. the diff is applied to your patch set V2 and not V3. I had errors > applying > V3 to master branch, but didn't have time to look into it. > > Andriy > _______________________________________________ > ffmpeg-devel mailing list > ffmpeg-devel@ffmpeg.org > https://ffmpeg.org/mailman/listinfo/ffmpeg-devel > > To unsubscribe, visit link above, or email > ffmpeg-devel-request@ffmpeg.org with subject "unsubscribe". Hi Andriy, case 5: // mirror horizontal and rotate 270 CW av_display_rotation_set(m, 270.0); - av_display_matrix_flip(m, 0, 1); + av_display_matrix_flip(m, 1, 0); //mirror horizontal is the first arg? break; I believe hlip + rotate_270 = rotate_270 + vlip ? But I may be wrong on it. And I am going to abandon this patch regarding discussion with Nicolas. Thanks for the review. Best Regards, Jun
diff --git a/fftools/ffmpeg_filter.c b/fftools/ffmpeg_filter.c index 1894f30561..0b7f6a6089 100644 --- a/fftools/ffmpeg_filter.c +++ b/fftools/ffmpeg_filter.c @@ -811,40 +811,25 @@ static int configure_input_video_filter(FilterGraph *fg, InputFilter *ifilter, int hflip = 0; double theta = get_rotation_hflip(ifilter->display_matrix, &hflip); - if (fabs(theta) < 1.0) { - if (hflip) - ret = insert_filter(&last_filter, &pad_idx, "hflip", NULL); - } else if (fabs(theta - 90) < 1.0) { + //mirror part + if (hflip) + ret = insert_filter(&last_filter, &pad_idx, "hflip", NULL); + + //rotation part + if (fabs(theta - 90) < 1.0) { ret = insert_filter(&last_filter, &pad_idx, "transpose", "clock"); + } else if (fabs(theta - 180) < 1.0) { + ret = insert_filter(&last_filter, &pad_idx, "hflip", NULL); if (ret < 0) return ret; - if (hflip) - ret = insert_filter(&last_filter, &pad_idx, "hflip", NULL); - } else if (fabs(theta - 180) < 1.0) { - if (hflip) { // rotate 180 and hflip equals vflip - ret = insert_filter(&last_filter, &pad_idx, "vflip", NULL); - } else { - ret = insert_filter(&last_filter, &pad_idx, "hflip", NULL); - if (ret < 0) - return ret; - ret = insert_filter(&last_filter, &pad_idx, "vflip", NULL); - } + ret = insert_filter(&last_filter, &pad_idx, "vflip", NULL); } else if (fabs(theta - 270) < 1.0) { ret = insert_filter(&last_filter, &pad_idx, "transpose", "cclock"); - if (ret < 0) - return ret; - if (hflip) - ret = insert_filter(&last_filter, &pad_idx, "hflip", NULL); } else if (fabs(theta) > 1.0) { char rotate_buf[64]; snprintf(rotate_buf, sizeof(rotate_buf), "%f*PI/180", theta); ret = insert_filter(&last_filter, &pad_idx, "rotate", rotate_buf); - if (ret < 0) - return ret; - if (hflip) - ret = insert_filter(&last_filter, &pad_idx, "hflip", NULL); } - if (ret < 0) return ret; } @@ -1232,14 +1217,14 @@ static void set_display_matrix_from_frame(const AVFrame *frame, int32_t m[9]) break; case 5: // mirror horizontal and rotate 270 CW av_display_rotation_set(m, 270.0); - av_display_matrix_flip(m, 0, 1); + av_display_matrix_flip(m, 1, 0); //mirror horizontal is the first arg? break; case 6: // rotate 90 CW av_display_rotation_set(m, 90.0); break; case 7: // mirror horizontal and rotate 90 CW av_display_rotation_set(m, 90.0); - av_display_matrix_flip(m, 0, 1); + av_display_matrix_flip(m, 1, 0); break; case 8: // rotate 270 CW av_display_rotation_set(m, 270.0);