diff options
Diffstat (limited to 'source/blender/compositor/operations/COM_ImageOperation.cpp')
-rw-r--r-- | source/blender/compositor/operations/COM_ImageOperation.cpp | 149 |
1 files changed, 149 insertions, 0 deletions
diff --git a/source/blender/compositor/operations/COM_ImageOperation.cpp b/source/blender/compositor/operations/COM_ImageOperation.cpp new file mode 100644 index 00000000000..52c2e44f7d4 --- /dev/null +++ b/source/blender/compositor/operations/COM_ImageOperation.cpp @@ -0,0 +1,149 @@ +/* + * Copyright 2011, Blender Foundation. + * + * 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. + * + * Contributor: + * Jeroen Bakker + * Monique Dewanchand + */ + +#include "COM_ImageOperation.h" + +#include "BLI_listbase.h" +#include "DNA_scene_types.h" +#include "DNA_image_types.h" +#include "BKE_image.h" +#include "BLI_math.h" + +extern "C" { + #include "RE_pipeline.h" + #include "RE_shader_ext.h" + #include "RE_render_ext.h" + #include "IMB_imbuf.h" + #include "IMB_imbuf_types.h" +} + +BaseImageOperation::BaseImageOperation(): NodeOperation() { + this->image = NULL; + this->buffer = NULL; + this->imageBuffer = NULL; + this->imageUser = NULL; + this->imagewidth = 0; + this->imageheight = 0; + this->framenumber = 0; + this->depthBuffer = NULL; + this->numberOfChannels = 0; +} +ImageOperation::ImageOperation(): BaseImageOperation() { + this->addOutputSocket(COM_DT_COLOR); +} +ImageAlphaOperation::ImageAlphaOperation(): BaseImageOperation() { + this->addOutputSocket(COM_DT_VALUE); +} +ImageDepthOperation::ImageDepthOperation(): BaseImageOperation() { + this->addOutputSocket(COM_DT_VALUE); +} + +ImBuf* BaseImageOperation::getImBuf() { + ImBuf *ibuf; + + ibuf= BKE_image_get_ibuf(this->image, this->imageUser); + if(ibuf==NULL || (ibuf->rect==NULL && ibuf->rect_float==NULL)) { + return NULL; + } + + if (ibuf->rect_float == NULL) { + IMB_float_from_rect(ibuf); + } + return ibuf; +} + + +void BaseImageOperation::initExecution() { + ImBuf *stackbuf= getImBuf(); + this->buffer = stackbuf; + if (stackbuf) { + this->imageBuffer = stackbuf->rect_float; + this->depthBuffer = stackbuf->zbuf_float; + this->imagewidth = stackbuf->x; + this->imageheight = stackbuf->y; + this->numberOfChannels = stackbuf->channels; + } +} + +void BaseImageOperation::deinitExecution() { + this->imageBuffer= NULL; +} + +void BaseImageOperation::determineResolution(unsigned int resolution[], unsigned int preferredResolution[]) { + ImBuf *stackbuf= getImBuf(); + if (stackbuf) { + resolution[0] = stackbuf->x; + resolution[1] = stackbuf->y; + } +} + +void ImageOperation::executePixel(float *color, float x, float y, PixelSampler sampler, MemoryBuffer *inputBuffers[]) { + if (this->imageBuffer == NULL || x < 0 || y < 0 || x >= this->getWidth() || y >= this->getHeight() ) { + color[0] = 0.0f; + color[1] = 0.0f; + color[2] = 0.0f; + color[3] = 0.0f; + } else { + switch (sampler) { + case COM_PS_NEAREST: + neareast_interpolation_color(this->buffer, NULL, color, x, y); + break; + case COM_PS_BILINEAR: + bilinear_interpolation_color(this->buffer, NULL, color, x, y); + break; + case COM_PS_BICUBIC: + bicubic_interpolation_color(this->buffer, NULL, color, x, y); + break; + } + } +} + +void ImageAlphaOperation::executePixel(float *color, float x, float y, PixelSampler sampler, MemoryBuffer *inputBuffers[]) { + float tempcolor[4]; + + if (this->imageBuffer == NULL || x < 0 || y < 0 || x >= this->getWidth() || y >= this->getHeight() ) { + color[0] = 0.0f; + } else { + tempcolor[3] = 1.0f; + switch (sampler) { + case COM_PS_NEAREST: + neareast_interpolation_color(this->buffer, NULL, tempcolor, x, y); + break; + case COM_PS_BILINEAR: + bilinear_interpolation_color(this->buffer, NULL, tempcolor, x, y); + break; + case COM_PS_BICUBIC: + bicubic_interpolation_color(this->buffer, NULL, tempcolor, x, y); + break; + } + color[0] = tempcolor[3]; + } +} + +void ImageDepthOperation::executePixel(float *color, float x, float y, PixelSampler sampler, MemoryBuffer *inputBuffers[]) { + if (this->depthBuffer == NULL || x < 0 || y < 0 || x >= this->getWidth() || y >= this->getHeight() ) { + color[0] = 0.0f; + } else { + int offset = y * width + x; + color[0] = this->depthBuffer[offset]; + } +} |