libmpcodecs/vf_tile.c

   1 /*
   2  * filter to tile a serie of image in a single, bigger, image
   3  *
   4  * The parameters are:
   5  *
   6  *    xtile:  number of tile on the x axis (5)
   7  *    ytile:  number of tile on the y axis (5)
   8  *    xytile: when write the image, it can be different then xtile * ytile
   9  *            (for example you can write 8 * 7 tile, writing the file every
  10  *            50 frame, to have one image every 2 seconds @ 25 fps ).
  11  *    start:  pixel at the start (x/y), default 2
  12  *    delta:  pixel between 2 tile, (x/y), default 4
  13  *
  14  * For example a valid command line is:
  15  *    ... -vf tile=10:5:-1:4:8 ...
  16  * that make images of 10 * 5 tiles, with 4 pixel at the beginning and
  17  * 8 pixel between tiles.
  18  *
  19  * The default command is:
  20  *    ... -vf tile=5:5:25:2:4
  21  *
  22  * If you omit a parameter or put a value less then 0, the default is used.
  23  *    ... -vf tile=10:5::-1:10
  24  *
  25  * You can also stop when you're ok
  26  *    ... -vf tile=10:5
  27  * (and this is probably the option you will use more often ...)
  28  *
  29  * Probably is good to put the scale filter before the tile :-)
  30  *
  31  * copyright (c) 2003 Daniele Forghieri ( guru@digitalfantasy.it )
  32  *
  33  * This file is part of MPlayer.
  34  *
  35  * MPlayer is free software; you can redistribute it and/or modify
  36  * it under the terms of the GNU General Public License as published by
  37  * the Free Software Foundation; either version 2 of the License, or
  38  * (at your option) any later version.
  39  *
  40  * MPlayer is distributed in the hope that it will be useful,
  41  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  42  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  43  * GNU General Public License for more details.
  44  *
  45  * You should have received a copy of the GNU General Public License along
  46  * with MPlayer; if not, write to the Free Software Foundation, Inc.,
  47  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  48  */
  49
  50 // strtoi memcpy_pic
  51
  52 #include <stdio.h>
  53 #include <stdlib.h>
  54 #include <string.h>
  55
  56 #include "config.h"
  57 #include "mp_msg.h"
  58 #include "cpudetect.h"
  59
  60 #include "img_format.h"
  61 #include "mp_image.h"
  62 #include "vf.h"
  63
  64 #include "libvo/fastmemcpy.h"
  65
  66 /* private data */
  67 struct vf_priv_s {
  68     /* configuration data */
  69     /* Number on hor/ver tiles */
  70     int  xtile;
  71     int  ytile;
  72     /* When write the whole frame (default = xtile * ytile) */
  73     int  xytile;
  74     /* pixel at start / end (default = 4) */
  75     int  start;
  76     /* pixel between image (default = 2) */
  77     int  delta;
  78 //    /* Background color, in destination format */
  79 //    int  bkgSet;
  80
  81     /* Work data */
  82     int  frame_cur;
  83     double start_pts;
  84 };
  85
  86
  87 static int config(struct vf_instance *vf,
  88                   int width, int height, int d_width, int d_height,
  89                   unsigned int flags, unsigned int outfmt){
  90
  91     struct vf_priv_s  *priv;
  92     int               xw;
  93     int               yh;
  94
  95     /* Calculate new destination size */
  96     priv = vf->priv;
  97     xw = priv->start * 2 +
  98         priv->xtile * width +
  99         (priv->xtile - 1) * priv->delta;
 100     yh = priv->start * 2 +
 101         priv->ytile * height +
 102         (priv->ytile - 1) * priv->delta;
 103
 104     mp_msg(MSGT_VFILTER,MSGL_V,"vf_tile:config size set to %d * %d\n", xw, yh);
 105
 106     return vf_next_config(vf, xw, yh, xw, yh, flags, outfmt);
 107 }
 108
 109 /* Filter handler */
 110 static int put_image(struct vf_instance *vf, mp_image_t *mpi, double pts)
 111 {
 112     mp_image_t        *dmpi;
 113     struct vf_priv_s  *priv;
 114     int               t;
 115     int               xw;
 116     int               yh;
 117     int               xi;
 118     int               yi;
 119     int               by;
 120     int               dw;
 121
 122     /* Calculate new size */
 123     priv = vf->priv;
 124     xw = priv->start * 2 +
 125         priv->xtile * mpi->w +
 126         (priv->xtile - 1) * priv->delta;
 127     yh = priv->start * 2 +
 128         priv->ytile * mpi->h+
 129         (priv->ytile - 1) * priv->delta;
 130
 131     /* Get the big image! */
 132     dmpi=vf_get_image(vf->next, mpi->imgfmt,
 133                       MP_IMGTYPE_STATIC, MP_IMGFLAG_ACCEPT_STRIDE,
 134                       xw, yh);
 135
 136     /* bytes x pixel & bytes x line */
 137     if (mpi->flags & MP_IMGFLAG_PLANAR) {
 138         by = 1;
 139         dw = mpi->w;
 140     }
 141     else {
 142         by = (mpi->bpp + 7) / 8;
 143         dw = mpi->w * by;
 144     }
 145     /* Index position */
 146     t = priv->frame_cur % priv->xytile;
 147 //    if ((t == 0) && (bkg != 0)) {
 148 //        /* First frame, delete the background */
 149 //
 150 //    }
 151     if (t == 0)
 152         priv->start_pts = pts;
 153
 154     /* Position of image */
 155     xi = priv->start + (mpi->w + priv->delta) * (t % priv->xtile);
 156     yi = priv->start + (mpi->h + priv->delta) * (t / priv->xtile);
 157
 158     /* Copy first (or only) plane */
 159     memcpy_pic( dmpi->planes[0] + xi * by + yi * dmpi->stride[0],
 160                 mpi->planes[0],
 161                 dw,
 162                 mpi->h,
 163                 dmpi->stride[0],
 164                 mpi->stride[0]);
 165
 166     if (mpi->flags & MP_IMGFLAG_PLANAR) {
 167         /* Copy the other 2 planes */
 168         memcpy_pic( dmpi->planes[1] + (xi >> mpi->chroma_x_shift) + (yi >> mpi->chroma_y_shift) * dmpi->stride[1],
 169                     mpi->planes[1],
 170                     mpi->chroma_width,
 171                     mpi->chroma_height,
 172                     dmpi->stride[1],
 173                     mpi->stride[1]);
 174          memcpy_pic( dmpi->planes[2] + (xi >> mpi->chroma_x_shift) + (yi >> mpi->chroma_y_shift) * dmpi->stride[2],
 175                     mpi->planes[2],
 176                     mpi->chroma_width,
 177                     mpi->chroma_height,
 178                     dmpi->stride[2],
 179                     mpi->stride[2]);
 180     }
 181
 182     /* Increment current frame */
 183     ++priv->frame_cur;
 184
 185     if (t == priv->xytile - 1)  {
 186         /* Display the composition */
 187         dmpi->width  = xw;
 188         dmpi->height = yh;
 189         return vf_next_put_image(vf, dmpi, priv->start_pts);
 190     }
 191     else {
 192         /* Skip the frame */
 193         return 0;
 194     }
 195 }
 196
 197 static void uninit(struct vf_instance *vf)
 198 {
 199     /* free local data */
 200     free(vf->priv);
 201 }
 202
 203 /* rgb/bgr 12->32 supported & some Yxxx */
 204 static int query_format(struct vf_instance *vf, unsigned int fmt)
 205 {
 206         switch (fmt) {
 207         /* rgb 12...32 bit */
 208         case IMGFMT_RGB12:
 209         case IMGFMT_RGB15:
 210         case IMGFMT_RGB16:
 211         case IMGFMT_RGB24:
 212         case IMGFMT_RGB32:
 213         /* bgr 12...32 bit */
 214         case IMGFMT_BGR12:
 215         case IMGFMT_BGR15:
 216         case IMGFMT_BGR16:
 217         case IMGFMT_BGR24:
 218         case IMGFMT_BGR32:
 219         /* Various Yxxx Formats */
 220         case IMGFMT_444P:
 221         case IMGFMT_422P:
 222         case IMGFMT_411P:
 223         case IMGFMT_YUY2:
 224         case IMGFMT_YV12:
 225         case IMGFMT_I420:
 226         case IMGFMT_YVU9:
 227         case IMGFMT_IF09:
 228         case IMGFMT_IYUV:
 229                 return vf_next_query_format(vf, fmt);
 230         }
 231         return 0;
 232 }
 233
 234 /* Get an integer from the string pointed by s, adjusting s.
 235  * If the value is less then 0 def_val is used.
 236  * Return 0 for ok
 237  *
 238  * Look below ( in vf_open(...) ) for a use ...
 239  */
 240 static int parse_int(char **s, int *rt, int def_val)
 241 {
 242
 243     int     t = 0;
 244
 245     if (**s) {
 246         /* Get value (dec, hex or octal) */
 247         t = strtol( *s, s, 0 );
 248
 249         /* Use default */
 250         if (t < 0) {
 251             t = def_val;
 252         }
 253
 254         if (**s == ':') {
 255             /* Point to next character (problably a digit) */
 256             ++(*s);
 257         }
 258         else if (**s != '\0') {
 259             /* Error, we got some wrong char */
 260             return 1;
 261         }
 262     }
 263     else {
 264         t = def_val;
 265     }
 266
 267     *rt = t;
 268     return 0;
 269
 270 }
 271
 272 /* Main entry funct for the filter */
 273 static int vf_open(vf_instance_t *vf, char *args)
 274 {
 275     struct vf_priv_s *p;
 276     int              er;
 277
 278     vf->put_image    = put_image;
 279     vf->query_format = query_format;
 280     vf->config       = config;
 281     vf->uninit       = uninit;
 282     vf->default_reqs = VFCAP_ACCEPT_STRIDE;
 283     /* Private data */
 284     vf->priv = p = calloc(1, sizeof(struct vf_priv_s));
 285     if (p == NULL) {
 286         return 0;
 287     }
 288
 289     if (args == NULL) {
 290         /* Use the default */
 291         args = "";
 292     }
 293     /* Parse all the arguments */
 294     er =  parse_int( &args, &p->xtile,  5 );
 295     er |= parse_int( &args, &p->ytile,  5 );
 296     er |= parse_int( &args, &p->xytile, 0 );
 297     er |= parse_int( &args, &p->start,  2 );
 298     er |= parse_int( &args, &p->delta,  4 );
 299 //    er |= parse_int( &args, &p->bkgSet, 0 );
 300
 301     if (er) {
 302         mp_tmsg(MSGT_VFILTER, MSGL_ERR, "[VF_FRAMESTEP] Error parsing argument.\n");
 303         return 0;
 304     }
 305     /* Load some default */
 306     if ((p->xytile <= 0) || (p->xytile > p->xtile * p->ytile)) {
 307         p->xytile = p->xtile * p->ytile;
 308     }
 309
 310     /* Say what happen: use mp_msg(...)? */
 311     if ( mp_msg_test(MSGT_VFILTER,MSGL_V) ) {
 312         printf("vf_tile: tiling %d * %d, output every %d frames\n",
 313                p->xtile,
 314                p->ytile,
 315                p->xytile);
 316         printf("vf_tile: start pixel %d, delta pixel %d\n",
 317                p->start,
 318                p->delta);
 319 //      printf("vf_tile: background 0x%x\n",
 320 //             p->bkgSet);
 321     }
 322     return 1;
 323 }
 324
 325 const vf_info_t vf_info_tile = {
 326     "Make a single image tiling x/y images",
 327     "tile",
 328     "Daniele Forghieri",
 329     "",
 330     vf_open,
 331     NULL
 332 };