Differential D13523 Diff 46102 source/blender/nodes/composite/nodes/node_composite_colorbalance.cc

Changeset View

Standalone View

source/blender/nodes/composite/nodes/node_composite_colorbalance.cc

Show All 15 Lines
* The Original Code is Copyright (C) 2006 Blender Foundation.		* The Original Code is Copyright (C) 2006 Blender Foundation.
* All rights reserved.		* All rights reserved.
*/		*/

/** \file		/** \file
* \ingroup cmpnodes		* \ingroup cmpnodes
*/		*/

		#include "RNA_access.h"

		#include "UI_interface.h"
		#include "UI_resources.h"

#include "node_composite_util.hh"		#include "node_composite_util.hh"

/* ***************** Color Balance ******************************* */		/* ***************** Color Balance ******************************* */

namespace blender::nodes {		namespace blender::nodes {

static void cmp_node_colorbalance_declare(NodeDeclarationBuilder &b)		static void cmp_node_colorbalance_declare(NodeDeclarationBuilder &b)
{		{
▲ Show 20 Lines • Show All 42 Lines • ▼ Show 20 Lines	static void node_composit_init_colorbalance(bNodeTree UNUSED(ntree), bNode node)
n->gain[0] = n->gain[1] = n->gain[2] = 1.0f;		n->gain[0] = n->gain[1] = n->gain[2] = 1.0f;

n->slope[0] = n->slope[1] = n->slope[2] = 1.0f;		n->slope[0] = n->slope[1] = n->slope[2] = 1.0f;
n->offset[0] = n->offset[1] = n->offset[2] = 0.0f;		n->offset[0] = n->offset[1] = n->offset[2] = 0.0f;
n->power[0] = n->power[1] = n->power[2] = 1.0f;		n->power[0] = n->power[1] = n->power[2] = 1.0f;
node->storage = n;		node->storage = n;
}		}

		static void node_composit_buts_colorbalance(uiLayout layout, bContext UNUSED(C), PointerRNA *ptr)
		{
		uiLayout split, col, *row;

		uiItemR(layout, ptr, "correction_method", UI_ITEM_R_SPLIT_EMPTY_NAME, nullptr, ICON_NONE);

		if (RNA_enum_get(ptr, "correction_method") == 0) {

		split = uiLayoutSplit(layout, 0.0f, false);
		col = uiLayoutColumn(split, false);
		uiTemplateColorPicker(col, ptr, "lift", true, true, false, true);
		row = uiLayoutRow(col, false);
		uiItemR(row, ptr, "lift", UI_ITEM_R_SPLIT_EMPTY_NAME, nullptr, ICON_NONE);

		col = uiLayoutColumn(split, false);
		uiTemplateColorPicker(col, ptr, "gamma", true, true, true, true);
		row = uiLayoutRow(col, false);
		uiItemR(row, ptr, "gamma", UI_ITEM_R_SPLIT_EMPTY_NAME, nullptr, ICON_NONE);

		col = uiLayoutColumn(split, false);
		uiTemplateColorPicker(col, ptr, "gain", true, true, true, true);
		row = uiLayoutRow(col, false);
		uiItemR(row, ptr, "gain", UI_ITEM_R_SPLIT_EMPTY_NAME, nullptr, ICON_NONE);
		}
		else {

		split = uiLayoutSplit(layout, 0.0f, false);
		col = uiLayoutColumn(split, false);
		uiTemplateColorPicker(col, ptr, "offset", true, true, false, true);
		row = uiLayoutRow(col, false);
		uiItemR(row, ptr, "offset", UI_ITEM_R_SPLIT_EMPTY_NAME, nullptr, ICON_NONE);
		uiItemR(col, ptr, "offset_basis", UI_ITEM_R_SPLIT_EMPTY_NAME, nullptr, ICON_NONE);

		col = uiLayoutColumn(split, false);
		uiTemplateColorPicker(col, ptr, "power", true, true, false, true);
		row = uiLayoutRow(col, false);
		uiItemR(row, ptr, "power", UI_ITEM_R_SPLIT_EMPTY_NAME, nullptr, ICON_NONE);

		col = uiLayoutColumn(split, false);
		uiTemplateColorPicker(col, ptr, "slope", true, true, false, true);
		row = uiLayoutRow(col, false);
		uiItemR(row, ptr, "slope", UI_ITEM_R_SPLIT_EMPTY_NAME, nullptr, ICON_NONE);
		}
		}

		static void node_composit_buts_colorbalance_ex(uiLayout *layout,
		bContext *UNUSED(C),
		PointerRNA *ptr)
		{
		uiItemR(layout, ptr, "correction_method", UI_ITEM_R_SPLIT_EMPTY_NAME, nullptr, ICON_NONE);

		if (RNA_enum_get(ptr, "correction_method") == 0) {

		uiTemplateColorPicker(layout, ptr, "lift", true, true, false, true);
		uiItemR(layout, ptr, "lift", UI_ITEM_R_SPLIT_EMPTY_NAME, nullptr, ICON_NONE);

		uiTemplateColorPicker(layout, ptr, "gamma", true, true, true, true);
		uiItemR(layout, ptr, "gamma", UI_ITEM_R_SPLIT_EMPTY_NAME, nullptr, ICON_NONE);

		uiTemplateColorPicker(layout, ptr, "gain", true, true, true, true);
		uiItemR(layout, ptr, "gain", UI_ITEM_R_SPLIT_EMPTY_NAME, nullptr, ICON_NONE);
		}
		else {
		uiTemplateColorPicker(layout, ptr, "offset", true, true, false, true);
		uiItemR(layout, ptr, "offset", UI_ITEM_R_SPLIT_EMPTY_NAME, nullptr, ICON_NONE);

		uiTemplateColorPicker(layout, ptr, "power", true, true, false, true);
		uiItemR(layout, ptr, "power", UI_ITEM_R_SPLIT_EMPTY_NAME, nullptr, ICON_NONE);

		uiTemplateColorPicker(layout, ptr, "slope", true, true, false, true);
		uiItemR(layout, ptr, "slope", UI_ITEM_R_SPLIT_EMPTY_NAME, nullptr, ICON_NONE);
		}
		}

void register_node_type_cmp_colorbalance()		void register_node_type_cmp_colorbalance()
{		{
static bNodeType ntype;		static bNodeType ntype;

cmp_node_type_base(&ntype, CMP_NODE_COLORBALANCE, "Color Balance", NODE_CLASS_OP_COLOR, 0);		cmp_node_type_base(&ntype, CMP_NODE_COLORBALANCE, "Color Balance", NODE_CLASS_OP_COLOR, 0);
ntype.declare = blender::nodes::cmp_node_colorbalance_declare;		ntype.declare = blender::nodes::cmp_node_colorbalance_declare;
		ntype.draw_buttons = node_composit_buts_colorbalance;
		ntype.draw_buttons_ex = node_composit_buts_colorbalance_ex;
node_type_size(&ntype, 400, 200, 400);		node_type_size(&ntype, 400, 200, 400);
node_type_init(&ntype, node_composit_init_colorbalance);		node_type_init(&ntype, node_composit_init_colorbalance);
node_type_storage(		node_type_storage(
&ntype, "NodeColorBalance", node_free_standard_storage, node_copy_standard_storage);		&ntype, "NodeColorBalance", node_free_standard_storage, node_copy_standard_storage);

nodeRegisterType(&ntype);		nodeRegisterType(&ntype);
}		}