d2/vise.h

   1 // Copyright 2004 David Hilvert <dhilvert@auricle.dyndns.org>,
   2 //                              <dhilvert@ugcs.caltech.edu>
   3
   4 /*  This file is part of the Anti-Lamenessing Engine.
   5
   6     The Anti-Lamenessing Engine is free software; you can redistribute it and/or modify
   7     it under the terms of the GNU General Public License as published by
   8     the Free Software Foundation; either version 2 of the License, or
   9     (at your option) any later version.
  10
  11     The Anti-Lamenessing Engine is distributed in the hope that it will be useful,
  12     but WITHOUT ANY WARRANTY; without even the implied warranty of
  13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14     GNU General Public License for more details.
  15
  16     You should have received a copy of the GNU General Public License
  17     along with the Anti-Lamenessing Engine; if not, write to the Free Software
  18     Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19 */
  20
  21 /*
  22  * vise.h: A superclass for all video stabilization engine classes.
  23  */
  24
  25 #ifndef __vise_h__
  26 #define __vise_h__
  27
  28 #include "transformation.h"
  29 #include "image.h"
  30 #include "point.h"
  31
  32 /*
  33  * Class vise accepts images from vise_core.  It is assumed, when an image is
  34  * received, that all transformations required for stabilization are available.
  35  * This class is abstract, and must be subclassed to be instantiated.
  36  */
  37
  38 class vise {
  39         const char *prefix;
  40         const char *suffix;
  41 protected:
  42         ale_real scale_factor;
  43
  44         render *r;
  45
  46         /*
  47          * Write a frame.  This function does not support sequences having
  48          * more than 100,000,000 frames (starting count with frame zero).
  49          */
  50         void write_frame(const image *im, unsigned int frame_number) {
  51                 if (frame_number > 99999999) {
  52                         fprintf(stderr, "\n\n *** Frame count too high for d2::vise::write_frame() ***\n\n\n");
  53                         exit(1);
  54                 }
  55
  56                 int length = strlen(prefix) + strlen(suffix) + 8 + 1;
  57                 char *filename_string = (char *) malloc(length * sizeof(char));
  58
  59                 snprintf(filename_string, length, "%s%08d%s", prefix, frame_number, suffix);
  60
  61                 image_rw::write_image(filename_string, im, &image_rw::exp());
  62
  63                 free(filename_string);
  64         }
  65 public:
  66         vise(render *r, const char *prefix, const char *suffix, ale_real scale_factor) {
  67                 this->prefix = prefix;
  68                 this->suffix = suffix;
  69                 this->r = r;
  70                 this->scale_factor = scale_factor;
  71         }
  72
  73         /*
  74          * Accept an image for rendering.
  75          */
  76
  77         virtual void render_frame(unsigned int frame_number) = 0;
  78
  79         /*
  80          * Report the frame lag for this stabilizer.
  81          */
  82
  83         virtual unsigned int lag() = 0;
  84
  85         virtual ~vise() {
  86         }
  87
  88 };
  89
  90 #endif