bitmap.h

Go to the documentation of this file.

/*
 * RVE: Realtime Video Effects
 * Copyright (C) 2002 Jonas Jensen <jbj@knef.dk>
 * 
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 */

#ifndef _BITMAP_H
#define _BITMAP_H

#include <stdlib.h>
#include <string.h>

#ifndef MIN
# define MAX(a, b) ((a) > (b) ? (a) : (b))
# define MIN(a, b) ((a) < (b) ? (a) : (b))
#endif

class Bitmap {
public:
    enum ColorSpace { CS_RGB, CS_RGBA, CS_BGR, CS_ABGR, CS_YV12, CS_YV12A };

    Bitmap(int width = 0, int height = 0): 
        _width(height), _height(height), _buf(NULL) {
    }
    virtual ~Bitmap() { free(_buf); }
    
    int w() const { return _width; }
    int h() const { return _height; }
    int get_space() const { return _width * _height * _bpp / 8; }
    int get_space(int width, int height) const {
        return width * height * _bpp / 8;
    }
    unsigned char *get_buf() const { return _buf; }
    
    virtual Bitmap *create() const = 0;
    virtual Bitmap *clone() const;

    virtual void set_size(int width, int height);

    virtual void clear();

    virtual enum ColorSpace get_color_space() const = 0;
    virtual void convert(Bitmap &target) const;
    //TODO
    virtual Bitmap *unify_csp(Bitmap &bitmap) const;

    void resize(int left = 0, int right = 0, int top = 0, int bottom = 0);
    Bitmap *resize_copy(int left=0, int right=0, int top=0, int bottom=0) const;

    virtual void blend(const Bitmap &source, 
            int xoff = 0, int yoff = 0, int alpha = 0);

    virtual void put(const Bitmap &source, int xoff = 0, int yoff = 0);

protected:
    void alloc(int width, int height);
    virtual void resize_into(int left, int right, int top, int bottom, 
            Bitmap *target) const = 0;

    virtual int  channel_count() const { return 3; }
    virtual int  get_pixel_alpha(int pixel) const = 0;
    virtual void set_pixel_alpha(int pixel, int value) = 0;
    virtual int  get_pixel_val(int pixel, int channel) const = 0;
    virtual void set_pixel_val(int pixel, int channel, int value) = 0;

    int _bpp;

private:
    int _width;
    int _height;
    unsigned char *_buf;
};

class Packed: public Bitmap {
protected:
    virtual void resize_into(int left, int right, int top, int bottom, 
            Bitmap *target) const;

    virtual int  get_pixel_val(int pixel, int channel) const {
        return get_buf()[pixel*_bpp/8 + channel]; }
    virtual void set_pixel_val(int pixel, int channel, int value) {
        get_buf()[pixel*_bpp/8 + channel] = value; }
};

class RGB: public Packed {
public:
    RGB(int width = 0, int height = 0) { _bpp = 24; set_size(width, height); }
    virtual enum ColorSpace get_color_space() const { return CS_RGB; }
    virtual Bitmap *create() const { return new RGB; }

protected:
    virtual int  get_pixel_alpha(int pixel) const { return 255; }
    virtual void set_pixel_alpha(int pixel, int value) {}
};

class RGBA: public Packed {
public:
    RGBA(int width = 0, int height = 0) { _bpp = 32; set_size(width, height); }
    virtual enum ColorSpace get_color_space() const { return CS_RGBA; }
    virtual Bitmap *create() const { return new RGBA; }
    virtual void blend(const Bitmap &source, 
            int xoff = 0, int yoff = 0, int alpha = 0);
    virtual void put(const Bitmap &source, int xoff = 0, int yoff = 0);

protected:
    virtual int  get_pixel_alpha(int pixel) const {
        return get_buf()[pixel*4+3]; }
    virtual void set_pixel_alpha(int pixel, int value) { 
        get_buf()[pixel*4+3] = value; }
};

class ABGR: public Packed {
public:
    ABGR(int width = 0, int height = 0) { _bpp = 32; set_size(width, height); }
    virtual enum ColorSpace get_color_space() const { return CS_ABGR; }
    virtual Bitmap *create() const { return new ABGR; }

protected:
    virtual int get_pixel_alpha(int pixel) const { return get_buf()[pixel*4+0];}
    virtual void set_pixel_alpha(int pixel, int value) { 
        get_buf()[pixel*4+0] = value; };
};

#endif

---