/* * 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_IDMaskOperation.h" IDMaskOperation::IDMaskOperation() : NodeOperation() { this->addInputSocket(COM_DT_VALUE); this->addOutputSocket(COM_DT_VALUE); this->m_inputProgram = NULL; } void IDMaskOperation::initExecution() { this->m_inputProgram = this->getInputSocketReader(0); } void IDMaskOperation::executePixelSampled(float output[4], float x, float y, PixelSampler sampler) { float inputValue[4]; this->m_inputProgram->readSampled(inputValue, x, y, sampler); /* 'round' since sampling may adjust value slightly. * ID-mask input are originally integers too. */ const float a = (roundf(inputValue[0]) == this->m_objectIndex) ? 1.0f : 0.0f; output[0] = a; } void IDMaskOperation::deinitExecution() { this->m_inputProgram = NULL; }