diff options
Diffstat (limited to 'source/blender/compositor/operations/COM_CompositorOperation.cpp')
-rw-r--r-- | source/blender/compositor/operations/COM_CompositorOperation.cpp | 122 |
1 files changed, 122 insertions, 0 deletions
diff --git a/source/blender/compositor/operations/COM_CompositorOperation.cpp b/source/blender/compositor/operations/COM_CompositorOperation.cpp new file mode 100644 index 00000000000..7590b4be299 --- /dev/null +++ b/source/blender/compositor/operations/COM_CompositorOperation.cpp @@ -0,0 +1,122 @@ +/* + * 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_CompositorOperation.h" +#include "COM_SocketConnection.h" +#include "BLI_listbase.h" +#include "DNA_scene_types.h" +#include "BKE_image.h" + +extern "C" { + #include "RE_pipeline.h" + #include "RE_shader_ext.h" + #include "RE_render_ext.h" + #include "MEM_guardedalloc.h" +#include "render_types.h" +} +#include "PIL_time.h" + + +CompositorOperation::CompositorOperation() : NodeOperation() { + this->addInputSocket(COM_DT_COLOR); + this->addInputSocket(COM_DT_VALUE); + + this->setScene(NULL); + this->outputBuffer = NULL; + this->imageInput = NULL; + this->alphaInput = NULL; +} + +void CompositorOperation::initExecution() { + // When initializing the tree during initial load the width and height can be zero. + this->imageInput = getInputSocketReader(0); + this->alphaInput = getInputSocketReader(1); + if (this->getWidth() * this->getHeight() != 0) { + this->outputBuffer=(float*) MEM_callocN(this->getWidth()*this->getHeight()*4*sizeof(float), "CompositorOperation"); + } + const Scene * scene = this->scene; + Render* re= RE_GetRender(scene->id.name); + RenderResult *rr= RE_AcquireResultWrite(re); + if(rr) { + if(rr->rectf != NULL) { + MEM_freeN(rr->rectf); + } + rr->rectf= outputBuffer; + } + if (re) { + RE_ReleaseResult(re); + re = NULL; + } + +} + +void CompositorOperation::deinitExecution() { + this->outputBuffer = NULL; + this->imageInput = NULL; + this->alphaInput = NULL; +} + + +void CompositorOperation::executeRegion(rcti *rect, unsigned int tileNumber, MemoryBuffer** memoryBuffers) { + float color[8]; // 7 is enough + float* buffer = this->outputBuffer; + + if (!buffer) return; + int x1 = rect->xmin; + int y1 = rect->ymin; + int x2 = rect->xmax; + int y2 = rect->ymax; + int offset = (y1*this->getWidth() + x1 ) * 4; + int x; + int y; + bool breaked = false; + + for (y = y1 ; y < y2 && (!breaked); y++) { + for (x = x1 ; x < x2 && (!breaked) ; x++) { + imageInput->read(color, x, y, COM_PS_NEAREST, memoryBuffers); + if (alphaInput != NULL) { + alphaInput->read(&(color[3]), x, y, COM_PS_NEAREST, memoryBuffers); + } + buffer[offset] = color[0]; + buffer[offset+1] = color[1]; + buffer[offset+2] = color[2]; + buffer[offset+3] = color[3]; + offset +=4; + if (tree->test_break && tree->test_break(tree->tbh)) { + breaked = true; + } + } + offset += (this->getWidth()-(x2-x1))*4; + } +} + +void CompositorOperation::determineResolution(unsigned int resolution[], unsigned int preferredResolution[]) { + int width = this->scene->r.xsch*this->scene->r.size/100; + int height= this->scene->r.ysch*this->scene->r.size/100; + preferredResolution[0] = width; + preferredResolution[1] = height; + + NodeOperation::determineResolution(resolution, preferredResolution); + + resolution[0] = width; + resolution[1] = height; +} |