diff options
Diffstat (limited to 'source/blender/compositor/operations/COM_ImageOperation.cc')
| -rw-r--r-- | source/blender/compositor/operations/COM_ImageOperation.cc | 205 |
1 files changed, 205 insertions, 0 deletions
diff --git a/source/blender/compositor/operations/COM_ImageOperation.cc b/source/blender/compositor/operations/COM_ImageOperation.cc new file mode 100644 index 00000000000..ae5b7293a8c --- /dev/null +++ b/source/blender/compositor/operations/COM_ImageOperation.cc @@ -0,0 +1,205 @@ +/* + * 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. + * + * Copyright 2011, Blender Foundation. + */ + +#include "COM_ImageOperation.h" + +#include "BKE_image.h" +#include "BKE_scene.h" +#include "BLI_listbase.h" +#include "BLI_math.h" +#include "DNA_image_types.h" + +#include "IMB_colormanagement.h" +#include "IMB_imbuf.h" +#include "IMB_imbuf_types.h" + +#include "RE_pipeline.h" +#include "RE_texture.h" + +BaseImageOperation::BaseImageOperation() +{ + this->m_image = nullptr; + this->m_buffer = nullptr; + this->m_imageFloatBuffer = nullptr; + this->m_imageByteBuffer = nullptr; + this->m_imageUser = nullptr; + this->m_imagewidth = 0; + this->m_imageheight = 0; + this->m_framenumber = 0; + this->m_depthBuffer = nullptr; + this->m_numberOfChannels = 0; + this->m_rd = nullptr; + this->m_viewName = nullptr; +} +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; + ImageUser iuser = *this->m_imageUser; + + if (this->m_image == nullptr) { + return nullptr; + } + + /* local changes to the original ImageUser */ + if (BKE_image_is_multilayer(this->m_image) == false) { + iuser.multi_index = BKE_scene_multiview_view_id_get(this->m_rd, this->m_viewName); + } + + ibuf = BKE_image_acquire_ibuf(this->m_image, &iuser, nullptr); + if (ibuf == nullptr || (ibuf->rect == nullptr && ibuf->rect_float == nullptr)) { + BKE_image_release_ibuf(this->m_image, ibuf, nullptr); + return nullptr; + } + return ibuf; +} + +void BaseImageOperation::initExecution() +{ + ImBuf *stackbuf = getImBuf(); + this->m_buffer = stackbuf; + if (stackbuf) { + this->m_imageFloatBuffer = stackbuf->rect_float; + this->m_imageByteBuffer = stackbuf->rect; + this->m_depthBuffer = stackbuf->zbuf_float; + this->m_imagewidth = stackbuf->x; + this->m_imageheight = stackbuf->y; + this->m_numberOfChannels = stackbuf->channels; + } +} + +void BaseImageOperation::deinitExecution() +{ + this->m_imageFloatBuffer = nullptr; + this->m_imageByteBuffer = nullptr; + BKE_image_release_ibuf(this->m_image, this->m_buffer, nullptr); +} + +void BaseImageOperation::determineResolution(unsigned int resolution[2], + unsigned int /*preferredResolution*/[2]) +{ + ImBuf *stackbuf = getImBuf(); + + resolution[0] = 0; + resolution[1] = 0; + + if (stackbuf) { + resolution[0] = stackbuf->x; + resolution[1] = stackbuf->y; + } + + BKE_image_release_ibuf(this->m_image, stackbuf, nullptr); +} + +static void sampleImageAtLocation( + ImBuf *ibuf, float x, float y, PixelSampler sampler, bool make_linear_rgb, float color[4]) +{ + if (ibuf->rect_float) { + switch (sampler) { + case COM_PS_NEAREST: + nearest_interpolation_color(ibuf, nullptr, color, x, y); + break; + case COM_PS_BILINEAR: + bilinear_interpolation_color(ibuf, nullptr, color, x, y); + break; + case COM_PS_BICUBIC: + bicubic_interpolation_color(ibuf, nullptr, color, x, y); + break; + } + } + else { + unsigned char byte_color[4]; + switch (sampler) { + case COM_PS_NEAREST: + nearest_interpolation_color(ibuf, byte_color, nullptr, x, y); + break; + case COM_PS_BILINEAR: + bilinear_interpolation_color(ibuf, byte_color, nullptr, x, y); + break; + case COM_PS_BICUBIC: + bicubic_interpolation_color(ibuf, byte_color, nullptr, x, y); + break; + } + rgba_uchar_to_float(color, byte_color); + if (make_linear_rgb) { + IMB_colormanagement_colorspace_to_scene_linear_v4(color, false, ibuf->rect_colorspace); + } + } +} + +void ImageOperation::executePixelSampled(float output[4], float x, float y, PixelSampler sampler) +{ + int ix = x, iy = y; + if (this->m_imageFloatBuffer == nullptr && this->m_imageByteBuffer == nullptr) { + zero_v4(output); + } + else if (ix < 0 || iy < 0 || ix >= this->m_buffer->x || iy >= this->m_buffer->y) { + zero_v4(output); + } + else { + sampleImageAtLocation(this->m_buffer, x, y, sampler, true, output); + } +} + +void ImageAlphaOperation::executePixelSampled(float output[4], + float x, + float y, + PixelSampler sampler) +{ + float tempcolor[4]; + + if (this->m_imageFloatBuffer == nullptr && this->m_imageByteBuffer == nullptr) { + output[0] = 0.0f; + } + else { + tempcolor[3] = 1.0f; + sampleImageAtLocation(this->m_buffer, x, y, sampler, false, tempcolor); + output[0] = tempcolor[3]; + } +} + +void ImageDepthOperation::executePixelSampled(float output[4], + float x, + float y, + PixelSampler /*sampler*/) +{ + if (this->m_depthBuffer == nullptr) { + output[0] = 0.0f; + } + else { + if (x < 0 || y < 0 || x >= this->getWidth() || y >= this->getHeight()) { + output[0] = 0.0f; + } + else { + int offset = y * this->m_width + x; + output[0] = this->m_depthBuffer[offset]; + } + } +} |