Differential D2889 Diff 9435 source/blender/modifiers/intern/MOD_gpencilcolor.c

Changeset View

Standalone View

source/blender/modifiers/intern/MOD_gpencilcolor.c

This file was added.

				/*
				* *** BEGIN GPL LICENSE BLOCK ***
				*
				* 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.
				*
				* The Original Code is Copyright (C) 2005 by the Blender Foundation.
				* All rights reserved.
				*
				* Contributor(s): Antonio Vazquez
				*
				* *** END GPL LICENSE BLOCK ***
				*
				*/

				/** \file blender/modifiers/intern/MOD_gpencilcolor.c
				* \ingroup modifiers
				*/

				#include <stdio.h>

				#include "DNA_scene_types.h"
				#include "DNA_object_types.h"
				#include "DNA_gpencil_types.h"

				#include "BLI_utildefines.h"
				#include "BLI_ghash.h"

				#include "BKE_global.h"
				#include "BKE_DerivedMesh.h"
				#include "BKE_gpencil.h"
				#include "BKE_paint.h"

				#include "MOD_modifiertypes.h"

				static void initData(ModifierData *md)
				{
				GpencilColorModifierData gpmd = (GpencilColorModifierData )md;
				gpmd->passindex = 0;
				ARRAY_SET_ITEMS(gpmd->hsv, 1.0f, 1.0f, 1.0f);
				gpmd->layername[0] = '\0';
				gpmd->flag \|= GP_COLOR_CREATE_COLORS;

				BKE_gpencil_batch_cache_alldirty();
				}

				static void copyData(ModifierData md, ModifierData target)
				{
				modifier_copyData_generic(md, target);
				}

				static DerivedMesh applyModifier(ModifierData md, const struct EvaluationContext UNUSED(eval_ctx), Object ob,
				DerivedMesh *UNUSED(dm),
				ModifierApplyFlag UNUSED(flag))
				{
				GpencilColorModifierData mmd = (GpencilColorModifierData )md;
				bGPdata *gpd;
				Palette *newpalette = NULL;
				if ((!ob) \|\| (!ob->gpd)) {
				return NULL;
				}
				gpd = ob->gpd;
				GHash *gh_layer = BLI_ghash_str_new("GP_Color Layer modifier");
				GHash *gh_color;

				for (bGPDlayer *gpl = gpd->layers.first; gpl; gpl = gpl->next) {
				for (bGPDframe *gpf = gpl->frames.first; gpf; gpf = gpf->next) {
				for (bGPDstroke *gps = gpf->strokes.first; gps; gps = gps->next) {
				/* look for palette */
				gh_color = (GHash *)BLI_ghash_lookup(gh_layer, gps->palette->id.name);
				if (gh_color == NULL) {
				gh_color = BLI_ghash_str_new("GP_Color Correction modifier");
				BLI_ghash_insert(gh_layer, gps->palette->id.name, gh_color);
				}

				/* look for color */
				PaletteColor newpalcolor = (PaletteColor )BLI_ghash_lookup(gh_color, gps->palcolor->info);
				if (newpalcolor == NULL) {
				if (mmd->flag & GP_COLOR_CREATE_COLORS) {
				if (!newpalette) {
				newpalette = BKE_palette_add(G.main, "Palette");
				}
				newpalcolor = BKE_palette_color_copy(newpalette, gps->palcolor);
				BLI_strncpy(gps->colorname, newpalcolor->info, sizeof(gps->colorname));
				gps->palcolor = newpalcolor;
				}
				else {
				newpalcolor = gps->palcolor;
				}
				BLI_ghash_insert(gh_color, gps->palcolor->info, newpalcolor);
				BKE_gpencil_color_modifier(-1, (GpencilColorModifierData *)md, ob, gpl, gps);
				}
				else {
				gps->palcolor = newpalcolor;
				}
				}
				}
				}
				/* free hash buffers */
				GHashIterator *ihash = BLI_ghashIterator_new(gh_layer);
				while (!BLI_ghashIterator_done(ihash)) {
				GHash *gh = BLI_ghashIterator_getValue(ihash);
				if (gh) {
				BLI_ghash_free(gh, NULL, NULL);
				gh = NULL;
				}
				BLI_ghashIterator_step(ihash);
				}
				BLI_ghashIterator_free(ihash);

				if (gh_layer) {
				BLI_ghash_free(gh_layer, NULL, NULL);
				gh_layer = NULL;
				}

				return NULL;
				}

				ModifierTypeInfo modifierType_GpencilColor = {
				/* name */ "Hue/Saturation",
				/* structName */ "GpencilColorModifierData",
				/* structSize */ sizeof(GpencilColorModifierData),
				/* type */ eModifierTypeType_Gpencil,
				/* flags */ eModifierTypeFlag_GpencilMod \| eModifierTypeFlag_SupportsEditmode,

				/* copyData */ copyData,
				/* deformVerts */ NULL,
				/* deformMatrices */ NULL,
				/* deformVertsEM */ NULL,
				/* deformMatricesEM */ NULL,
				/* applyModifier */ applyModifier,
				/* applyModifierEM */ NULL,
				/* initData */ initData,
				/* requiredDataMask */ NULL,
				/* freeData */ NULL,
				/* isDisabled */ NULL,
				/* updateDepsgraph */ NULL,
				/* dependsOnTime */ NULL,
				/* dependsOnNormals */ NULL,
				/* foreachObjectLink */ NULL,
				/* foreachIDLink */ NULL,
				/* foreachTexLink */ NULL,
				};