FrameBuffer.h

/* 
 * File:   FrameBuffer.h
 * Author: zoizoi
 *
 * Created on 30 January 2011, 20:29
 */

#ifndef FRAMEBUFFER_H
#define	FRAMEBUFFER_H
#include "vtkEssentials.h"

class FrameBuffer {
public:
    FrameBuffer(vtkSmartPointer<vtkRenderWindow> renderWindow);
    FrameBuffer(const FrameBuffer& orig);

    virtual ~FrameBuffer() {
       // delete data;
    };

    void grabData() {
        size = renderWindow->GetSize();
        len = size[0] * size[1];
        data = renderWindow->GetRGBAPixelData(0, 0, (size[0] - 1), (size[1] - 1), 0);
    }
    float* copyData(){
        float* dataCopy=new float[len]();
        for(int i=0;i<len;i++){
            dataCopy[i]=data[i];
        }
        return dataCopy;
    }
    float* getData() {
        return data;
    }

    int getLen() {
        return len;
    }

    void delData(){
        delete data;
    }

private:
    vtkSmartPointer<vtkRenderWindow> renderWindow;
    float* data;
    int* size;
    int len;
};

#endif	/* FRAMEBUFFER_H */