#include <cstring>
#include "MEM_guardedalloc.h"
#include "DNA_brush_types.h"
#include "DNA_object_types.h"
#include "DNA_scene_types.h"
#include "DNA_space_types.h"
#include "BLI_bitmap.h"
#include "BLI_listbase.h"
#include "BLI_math_color_blend.h"
#include "BLI_stack.h"
#include "BLI_task.h"
#include "BKE_brush.hh"
#include "BKE_colorband.hh"
#include "BKE_context.hh"
#include "BKE_image.hh"
#include "BKE_paint.hh"
#include "BKE_report.hh"
#include "DEG_depsgraph.hh"
#include "ED_paint.hh"
#include "ED_screen.hh"
#include "IMB_colormanagement.hh"
#include "IMB_imbuf.hh"
#include "IMB_imbuf_types.hh"
#include "WM_api.hh"
#include "WM_types.hh"
#include "UI_view2d.hh"
#include "paint_intern.hh"

Classes
struct	BrushPainterCache

struct	BrushPainter

struct	ImagePaintRegion

struct	ImagePaintTile

struct	ImagePaintState

struct	Paint2DForeachData

Enumerations
enum	ImagePaintTileState { PAINT2D_TILE_UNINITIALIZED = 0 , PAINT2D_TILE_MISSING , PAINT2D_TILE_READY }

Functions
static BrushPainter *	brush_painter_2d_new (Scene scene, const Paint paint, Brush *brush, bool invert)

static void	brush_painter_2d_require_imbuf (Brush brush, ImagePaintTile tile, bool use_float, bool use_color_correction, bool invert)

static void	brush_painter_cache_2d_free (BrushPainterCache *cache)

static void	brush_imbuf_tex_co (const rctf *mapping, int x, int y, float texco[3])

static ushort *	brush_painter_mask_ibuf_new (BrushPainter *painter, const int size)

static void	brush_painter_mask_imbuf_update (BrushPainter painter, ImagePaintTile tile, const ushort *tex_mask_old, int origx, int origy, int w, int h, int xt, int yt, const int diameter)

static void	brush_painter_mask_imbuf_partial_update (BrushPainter painter, ImagePaintTile tile, const float pos[2], const int diameter)

static ImBuf *	brush_painter_imbuf_new (BrushPainter painter, ImagePaintTile tile, const int size, float pressure, float distance)

static void	brush_painter_imbuf_update (BrushPainter painter, ImagePaintTile tile, ImBuf *oldtexibuf, int origx, int origy, int w, int h, int xt, int yt)

static void	brush_painter_imbuf_partial_update (BrushPainter painter, ImagePaintTile tile, const float pos[2], const int diameter)

static void	brush_painter_2d_tex_mapping (ImagePaintState s, ImagePaintTile tile, const int diameter, const float pos[2], const float mouse[2], int mapmode, rctf *r_mapping)

static void	brush_painter_2d_refresh_cache (ImagePaintState s, BrushPainter painter, ImagePaintTile *tile, const float pos[2], const float mouse[2], float pressure, float distance, float size)

static bool	paint_2d_ensure_tile_canvas (ImagePaintState *s, int i)

static void	paint_2d_ibuf_rgb_get (ImBuf *ibuf, int x, int y, float r_rgb[4])

static void	paint_2d_ibuf_rgb_set (ImBuf *ibuf, int x, int y, const bool is_torus, const float rgb[4])

static void	paint_2d_ibuf_tile_convert (ImBuf ibuf, int x, int *y, short paint_tile)

static float	paint_2d_ibuf_add_if (ImBuf ibuf, int x, int y, float outrgb, short paint_tile, float w)

static void	paint_2d_lift_soften (ImagePaintState s, ImagePaintTile tile, ImBuf ibuf, ImBuf ibufb, const int *pos, const short paint_tile)

static void	paint_2d_set_region (ImagePaintRegion *region, int destx, int desty, int srcx, int srcy, int width, int height)

static int	paint_2d_torus_split_region (ImagePaintRegion region[4], ImBuf dbuf, ImBuf sbuf, short paint_tile)

static void	paint_2d_lift_smear (ImBuf ibuf, ImBuf ibufb, int *pos, short paint_tile)

static ImBuf *	paint_2d_lift_clone (ImBuf ibuf, ImBuf ibufb, const int *pos)

static void	paint_2d_convert_brushco (ImBuf *ibufb, const float pos[2], int ipos[2])

static void	paint_2d_do_making_brush (ImagePaintState s, ImagePaintTile tile, ImagePaintRegion region, ImBuf frombuf, float mask_max, short blend, int tilex, int tiley, int tilew, int tileh)

static void	paint_2d_op_foreach_do (void __restrict data_v, const int iter, const TaskParallelTLS __restrict)

static int	paint_2d_op (void state, ImagePaintTile tile, const float lastpos[2], const float pos[2])

static int	paint_2d_canvas_set (ImagePaintState *s)

static void	paint_2d_canvas_free (ImagePaintState *s)

static void	paint_2d_transform_mouse (View2D *v2d, const float in[2], float out[2])

static bool	is_inside_tile (const int size[2], const float pos[2], const float brush[2])

static void	paint_2d_uv_to_coord (ImagePaintTile *tile, const float uv[2], float coord[2])

void	paint_2d_stroke (void *ps, const float prev_mval[2], const float mval[2], const bool eraser, float pressure, float distance, float base_size)

void *	paint_2d_new_stroke (bContext C, wmOperator op, int mode)

void	paint_2d_redraw (const bContext C, void ps, bool final)

void	paint_2d_stroke_done (void *ps)

static void	paint_2d_fill_add_pixel_byte (const int x_px, const int y_px, ImBuf ibuf, BLI_Stack stack, BLI_bitmap *touched, const float color[4], float threshold_sq)

static void	paint_2d_fill_add_pixel_float (const int x_px, const int y_px, ImBuf ibuf, BLI_Stack stack, BLI_bitmap *touched, const float color[4], float threshold_sq)

static ImageUser *	paint_2d_get_tile_iuser (ImagePaintState *s, int tile_number)

void	paint_2d_bucket_fill (const bContext C, const float color[3], Brush br, const float mouse_init[2], const float mouse_final[2], void *ps)

void	paint_2d_gradient_fill (const bContext C, Brush br, const float mouse_init[2], const float mouse_final[2], void *ps)

Enumeration Type Documentation

◆ ImagePaintTileState

enum ImagePaintTileState

Enumerator
PAINT2D_TILE_UNINITIALIZED
PAINT2D_TILE_MISSING
PAINT2D_TILE_READY

Definition at line 96 of file paint_image_2d.cc.

Function Documentation

◆ brush_imbuf_tex_co()

static void brush_imbuf_tex_co	(	const rctf *	mapping,
		int	x,
		int	y,
		float	texco[3] )

static

Definition at line 201 of file paint_image_2d.cc.

Referenced by brush_painter_imbuf_new(), brush_painter_imbuf_update(), brush_painter_mask_ibuf_new(), and brush_painter_mask_imbuf_update().

◆ brush_painter_2d_new()

static BrushPainter * brush_painter_2d_new	(	Scene *	scene,
		const Paint *	paint,
		Brush *	brush,
		bool	invert )

static

Definition at line 138 of file paint_image_2d.cc.

References BrushPainter::brush, BrushPainter::cache_invert, BrushPainter::firsttouch, invert(), BrushPainter::paint, BrushPainter::scene, and scene.

Referenced by paint_2d_new_stroke().

◆ brush_painter_2d_refresh_cache()

static void brush_painter_2d_refresh_cache	(	ImagePaintState *	s,
		BrushPainter *	painter,
		ImagePaintTile *	tile,
		const float	pos[2],
		const float	mouse[2],
		float	pressure,
		float	distance,
		float	size )

static

Definition at line 697 of file paint_image_2d.cc.

Referenced by paint_2d_stroke().

◆ brush_painter_2d_require_imbuf()

static void brush_painter_2d_require_imbuf	(	Brush *	brush,
		ImagePaintTile *	tile,
		bool	use_float,
		bool	use_color_correction,
		bool	invert )

static

◆ brush_painter_2d_tex_mapping()

static void brush_painter_2d_tex_mapping	(	ImagePaintState *	s,
		ImagePaintTile *	tile,
		const int	diameter,
		const float	pos[2],
		const float	mouse[2],
		int	mapmode,
		rctf *	r_mapping )

static

Definition at line 644 of file paint_image_2d.cc.

References ELEM, float, floorf, int, MTEX_MAP_MODE_3D, MTEX_MAP_MODE_RANDOM, MTEX_MAP_MODE_STENCIL, MTEX_MAP_MODE_VIEW, pos, tile, UI_view2d_view_to_region_fl(), KernelWorkTile::x, rctf::xmax, rctf::xmin, KernelWorkTile::y, rctf::ymax, and rctf::ymin.

Referenced by brush_painter_2d_refresh_cache().

◆ brush_painter_cache_2d_free()

static void brush_painter_cache_2d_free ( BrushPainterCache * cache )

static

Definition at line 184 of file paint_image_2d.cc.

References BrushPainterCache::curve_mask_cache, BrushPainterCache::ibuf, IMB_freeImBuf(), MEM_freeN(), paint_curve_mask_cache_free_data(), BrushPainterCache::tex_mask, BrushPainterCache::tex_mask_old, and BrushPainterCache::texibuf.

Referenced by paint_2d_stroke_done().

◆ brush_painter_imbuf_new()

static ImBuf * brush_painter_imbuf_new	(	BrushPainter *	painter,
		ImagePaintTile *	tile,
		const int	size,
		float	pressure,
		float	distance )

static

◆ brush_painter_imbuf_partial_update()

static void brush_painter_imbuf_partial_update	(	BrushPainter *	painter,
		ImagePaintTile *	tile,
		const float	pos[2],
		const int	diameter )

static

Definition at line 580 of file paint_image_2d.cc.

References brush_painter_imbuf_update(), floorf, IB_rect, IB_rectfloat, BrushPainterCache::ibuf, IMB_allocImBuf(), IMB_freeImBuf(), IMB_rectclip(), int, min_ii(), pos, BrushPainterCache::texibuf, tile, BrushPainterCache::use_float, w(), ImBuf::x, and ImBuf::y.

Referenced by brush_painter_2d_refresh_cache().

◆ brush_painter_imbuf_update()

static void brush_painter_imbuf_update	(	BrushPainter *	painter,
		ImagePaintTile *	tile,
		ImBuf *	oldtexibuf,
		int	origx,
		int	origy,
		int	w,
		int	h,
		int	xt,
		int	yt )

static

Definition at line 456 of file paint_image_2d.cc.

References b, BKE_brush_sample_tex_3d(), BrushPainter::brush, brush_imbuf_tex_co(), ImBuf::byte_buffer, copy_v3_v3(), copy_v4_v4(), ImBufByteBuffer::data, ImBufFloatBuffer::data, ImBuf::float_buffer, BrushPainterCache::ibuf, Brush::image_brush_type, IMAGE_PAINT_BRUSH_TYPE_DRAW, IMB_colormanagement_display_get_named(), IMB_colormanagement_scene_linear_to_display_v3(), BrushPainterCache::invert, BrushPainterCache::is_texbrush, Brush::mtex, mul_v3_v3(), mul_v3_v3fl(), ot, BrushPainter::paint, paint_brush_color_get(), BrushPainter::pool, rgba_float_to_uchar(), BrushPainter::scene, BrushPainter::tex_mapping, BrushPainterCache::texibuf, tile, BrushPainterCache::use_color_correction, BrushPainterCache::use_float, w(), ImBuf::x, x, and y.

Referenced by brush_painter_imbuf_partial_update().

◆ brush_painter_mask_ibuf_new()

static ushort * brush_painter_mask_ibuf_new	(	BrushPainter *	painter,
		const int	size )

static

Definition at line 209 of file paint_image_2d.cc.

References BKE_brush_sample_masktex(), BrushPainter::brush, brush_imbuf_tex_co(), mask(), BrushPainter::mask_mapping, MEM_mallocN, BrushPainter::pool, BrushPainter::scene, size(), x, and y.

Referenced by brush_painter_2d_refresh_cache().

◆ brush_painter_mask_imbuf_partial_update()

static void brush_painter_mask_imbuf_partial_update	(	BrushPainter *	painter,
		ImagePaintTile *	tile,
		const float	pos[2],
		const int	diameter )

static

Update the brush mask image by trying to reuse the cached texture result. This can be considerably faster for brushes that change size due to pressure or textures that stick to the surface where only part of the pixels are new

Definition at line 294 of file paint_image_2d.cc.

References brush_painter_mask_imbuf_update(), floorf, IMB_rectclip(), int, MEM_freeN(), MEM_mallocN, min_ii(), pos, BrushPainterCache::tex_mask, BrushPainterCache::tex_mask_old, BrushPainterCache::tex_mask_old_h, BrushPainterCache::tex_mask_old_w, tile, w(), ImBuf::x, and ImBuf::y.

Referenced by brush_painter_2d_refresh_cache().

◆ brush_painter_mask_imbuf_update()

static void brush_painter_mask_imbuf_update	(	BrushPainter *	painter,
		ImagePaintTile *	tile,
		const ushort *	tex_mask_old,
		int	origx,
		int	origy,
		int	w,
		int	h,
		int	xt,
		int	yt,
		const int	diameter )

static

Definition at line 236 of file paint_image_2d.cc.

References b, BKE_brush_sample_masktex(), BrushPainter::brush, brush_imbuf_tex_co(), BrushPainter::mask_mapping, BrushPainter::pool, BrushPainter::scene, BrushPainterCache::tex_mask, BrushPainterCache::tex_mask_old, BrushPainterCache::tex_mask_old_w, tile, w(), x, and y.

Referenced by brush_painter_mask_imbuf_partial_update().

◆ is_inside_tile()

static bool is_inside_tile	(	const int	size[2],
		const float	pos[2],
		const float	brush[2] )

static

Definition at line 1470 of file paint_image_2d.cc.

References pos.

Referenced by paint_2d_stroke().

◆ paint_2d_bucket_fill()

void paint_2d_bucket_fill	(	const bContext *	C,
		const float	color[3],
		Brush *	br,
		const float	mouse_init[2],
		const float	mouse_final[2],
		void *	ps )

This function expects linear space color values.

Definition at line 1791 of file paint_image_2d.cc.

References BKE_brush_alpha_get(), BKE_image_acquire_ibuf(), BKE_image_get_tile_from_pos(), BKE_image_release_ibuf(), BKE_imageuser_default(), Brush::blend, blend_color_mix_byte(), blend_color_mix_float(), BLI_BITMAP_NEW, BLI_BITMAP_SET, BLI_stack_free(), BLI_stack_is_empty(), BLI_stack_new, BLI_stack_pop(), BLI_stack_push(), ImBuf::byte_buffer, copy_v3_v3(), copy_v4_v4(), CTX_wm_region(), CTX_wm_space_image(), ImBufByteBuffer::data, ImBufFloatBuffer::data, ED_imapaint_clear_partial_redraw(), ED_imapaint_dirty_region(), Brush::fill_threshold, ImBuf::float_buffer, SpaceImage::image, image_init(), imapaint_image_update(), IMB_blend_color_byte(), IMB_blend_color_float(), linearrgb_to_srgb_uchar3(), MEM_freeN(), NA_EDITED, NC_IMAGE, paint_2d_fill_add_pixel_byte(), paint_2d_fill_add_pixel_float(), paint_2d_get_tile_iuser(), paint_2d_transform_mouse(), rgba_uchar_to_float(), straight_to_premul_v4(), ImageUser::tile, ARegion::v2d, WM_event_add_notifier(), ImBuf::x, and ImBuf::y.

Referenced by ED_imapaint_bucket_fill(), and blender::ed::sculpt_paint::image::ops::paint::ImagePaintMode::paint_bucket_fill().

◆ paint_2d_canvas_free()

static void paint_2d_canvas_free ( ImagePaintState * s )

static

Definition at line 1452 of file paint_image_2d.cc.

References BKE_image_release_ibuf(), and paint_delete_blur_kernel().

Referenced by paint_2d_stroke_done().

◆ paint_2d_canvas_set()

static int paint_2d_canvas_set ( ImagePaintState * s )

static

Definition at line 1423 of file paint_image_2d.cc.

References BKE_image_acquire_ibuf(), BKE_image_release_ibuf(), ImBuf::byte_buffer, ImBufByteBuffer::data, ImBufFloatBuffer::data, ImBuf::float_buffer, IMAGE_PAINT_BRUSH_TYPE_CLONE, IMB_float_from_rect(), IMB_rect_from_float(), and paint_use_opacity_masking().

Referenced by paint_2d_new_stroke().

◆ paint_2d_convert_brushco()

static void paint_2d_convert_brushco	(	ImBuf *	ibufb,
		const float	pos[2],
		int	ipos[2] )

static

Definition at line 1203 of file paint_image_2d.cc.

References floorf, int, pos, ImBuf::x, and ImBuf::y.

Referenced by paint_2d_op().

◆ paint_2d_do_making_brush()

static void paint_2d_do_making_brush	(	ImagePaintState *	s,
		ImagePaintTile *	tile,
		ImagePaintRegion *	region,
		ImBuf *	frombuf,
		float	mask_max,
		short	blend,
		int	tilex,
		int	tiley,
		int	tilew,
		int	tileh )

static

Definition at line 1209 of file paint_image_2d.cc.

References blend, BRUSH_ACCUMULATE, ED_image_paint_tile_find(), ED_image_paint_tile_map_get(), ED_IMAGE_UNDO_TILE_SIZE, IB_DO_NOT_TAKE_OWNERSHIP, IMB_assign_byte_buffer(), IMB_assign_float_buffer(), IMB_initImBuf(), IMB_rectblend(), mask(), and tile.

Referenced by paint_2d_op(), and paint_2d_op_foreach_do().

◆ paint_2d_ensure_tile_canvas()

static bool paint_2d_ensure_tile_canvas	(	ImagePaintState *	s,
		int	i )

static

Definition at line 817 of file paint_image_2d.cc.

References BKE_image_acquire_ibuf(), BKE_image_release_ibuf(), ImBuf::byte_buffer, ImBuf::channels, ImBufByteBuffer::data, ImBufFloatBuffer::data, ImBuf::float_buffer, PAINT2D_TILE_MISSING, PAINT2D_TILE_READY, sqrtf, ImBuf::x, and ImBuf::y.

Referenced by paint_2d_get_tile_iuser(), and paint_2d_stroke().

◆ paint_2d_fill_add_pixel_byte()

static void paint_2d_fill_add_pixel_byte	(	const int	x_px,
		const int	y_px,
		ImBuf *	ibuf,
		BLI_Stack *	stack,
		BLI_bitmap *	touched,
		const float	color[4],
		float	threshold_sq )

static

Definition at line 1722 of file paint_image_2d.cc.

References BLI_BITMAP_SET, BLI_BITMAP_TEST, BLI_stack_push(), ImBuf::byte_buffer, ImBufByteBuffer::data, len_squared_v4v4(), rgba_uchar_to_float(), straight_to_premul_v4(), ImBuf::x, and ImBuf::y.

Referenced by paint_2d_bucket_fill().

◆ paint_2d_fill_add_pixel_float()

static void paint_2d_fill_add_pixel_float	(	const int	x_px,
		const int	y_px,
		ImBuf *	ibuf,
		BLI_Stack *	stack,
		BLI_bitmap *	touched,
		const float	color[4],
		float	threshold_sq )