be05840f601426f8afd20b1d74eaff7c87dfae5f
[blender.git] / source / blender / compositor / nodes / COM_MaskNode.cpp
1 /*
2  * Copyright 2012, Blender Foundation.
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * Contributor:
19  *              Jeroen Bakker
20  *              Monique Dewanchand
21  *              Sergey Sharybin
22  */
23
24 #include "COM_MaskNode.h"
25 #include "COM_ExecutionSystem.h"
26 #include "COM_MaskOperation.h"
27
28 extern "C" {
29 #  include "DNA_mask_types.h"
30 }
31
32 MaskNode::MaskNode(bNode *editorNode) : Node(editorNode)
33 {
34         /* pass */
35 }
36
37 void MaskNode::convertToOperations(NodeConverter &converter, const CompositorContext &context) const
38 {
39         const RenderData *rd = context.getRenderData();
40
41         NodeOutput *outputMask = this->getOutputSocket(0);
42
43         bNode *editorNode = this->getbNode();
44         NodeMask *data = (NodeMask *)editorNode->storage;
45         Mask *mask = (Mask *)editorNode->id;
46
47         // always connect the output image
48         MaskOperation *operation = new MaskOperation();
49
50         if (editorNode->custom1 & CMP_NODEFLAG_MASK_FIXED) {
51                 operation->setMaskWidth(data->size_x);
52                 operation->setMaskHeight(data->size_y);
53         }
54         else if (editorNode->custom1 & CMP_NODEFLAG_MASK_FIXED_SCENE) {
55                 operation->setMaskWidth(data->size_x * (rd->size / 100.0f));
56                 operation->setMaskHeight(data->size_y * (rd->size / 100.0f));
57         }
58         else {
59                 operation->setMaskWidth(rd->xsch * rd->size / 100.0f);
60                 operation->setMaskHeight(rd->ysch * rd->size / 100.0f);
61         }
62
63         operation->setMask(mask);
64         operation->setFramenumber(context.getFramenumber());
65         operation->setSmooth((bool)(editorNode->custom1 & CMP_NODEFLAG_MASK_AA) != 0);
66         operation->setFeather((bool)(editorNode->custom1 & CMP_NODEFLAG_MASK_NO_FEATHER) == 0);
67
68         if ((editorNode->custom1 & CMP_NODEFLAG_MASK_MOTION_BLUR) &&
69             (editorNode->custom2 > 1) &&
70             (editorNode->custom3 > FLT_EPSILON))
71         {
72                 operation->setMotionBlurSamples(editorNode->custom2);
73                 operation->setMotionBlurShutter(editorNode->custom3);
74         }
75
76         converter.addOperation(operation);
77         converter.mapOutputSocket(outputMask, operation->getOutputSocket());
78 }