multires_bake.cc File Reference

#include <cstring>
#include "MEM_guardedalloc.h"
#include "DNA_modifier_types.h"
#include "DNA_scene_types.h"
#include "BLI_listbase.h"
#include "BLI_math_color.h"
#include "BLI_math_geom.h"
#include "BLI_math_matrix.h"
#include "BLI_threads.h"
#include "BKE_attribute.hh"
#include "BKE_ccg.hh"
#include "BKE_customdata.hh"
#include "BKE_global.hh"
#include "BKE_image.hh"
#include "BKE_lib_id.hh"
#include "BKE_mesh.hh"
#include "BKE_mesh_legacy_derived_mesh.hh"
#include "BKE_mesh_tangent.hh"
#include "BKE_multires.hh"
#include "BKE_subsurf.hh"
#include "DEG_depsgraph.hh"
#include "RE_multires_bake.h"
#include "RE_pipeline.h"
#include "RE_texture_margin.h"
#include "IMB_imbuf.hh"
#include "IMB_imbuf_types.hh"

Go to the source code of this file.

Classes
struct	MultiresBakeResult
struct	MResolvePixelData
struct	MBakeRast
struct	MHeightBakeData
struct	MNormalBakeData
struct	BakeImBufuserData
struct	MultiresBakeQueue
struct	MultiresBakeThread

Typedefs
using	MPassKnownData
using	MInitBakeData = void *(*)(MultiresBakeRender *bkr, ImBuf *ibuf)
using	MFreeBakeData = void (*)(void *bake_data)
using	MFlushPixel = void (*)(const MResolvePixelData *data, const int x, const int y)

Functions
static void	multiresbake_get_normal (const MResolvePixelData *data, const int tri_num, const int vert_index, float r_normal[3])
static void	init_bake_rast (MBakeRast *bake_rast, const ImBuf *ibuf, const MResolvePixelData *data, MFlushPixel flush_pixel, bool *do_update)
static void	flush_pixel (const MResolvePixelData *data, const int x, const int y)
static void	set_rast_triangle (const MBakeRast *bake_rast, const int x, const int y)
static void	rasterize_half (const MBakeRast *bake_rast, const float s0_s, const float t0_s, const float s1_s, const float t1_s, const float s0_l, const float t0_l, const float s1_l, const float t1_l, const int y0_in, const int y1_in, const int is_mid_right)
static void	bake_rasterize (const MBakeRast *bake_rast, const float st0_in[2], const float st1_in[2], const float st2_in[2])
static int	multiresbake_test_break (MultiresBakeRender *bkr)
static int	multires_bake_queue_next_tri (MultiresBakeQueue *queue)
static void *	do_multires_bake_thread (void *data_v)
static void	init_ccgdm_arrays (DerivedMesh *dm)
static void	do_multires_bake (MultiresBakeRender *bkr, Image *ima, ImageTile *tile, ImBuf *ibuf, bool require_tangent, MPassKnownData passKnownData, MInitBakeData initBakeData, MFreeBakeData freeBakeData, MultiresBakeResult *result)
static void	interp_bilinear_grid (const CCGKey &key, CCGElem *grid, float crn_x, float crn_y, int mode, float res[3])
static void	get_ccgdm_data (const blender::OffsetIndices< int > lores_polys, DerivedMesh *hidm, const int *index_mp_to_orig, const int lvl, const int face_index, const float u, const float v, float co[3], float n[3])
static void	interp_bilinear_mpoly (const blender::Span< blender::float3 > vert_positions, const blender::Span< blender::float3 > vert_normals, const blender::Span< int > corner_verts, const blender::IndexRange face, const float u, const float v, const int mode, float res[3])
static void	interp_barycentric_corner_tri (const blender::Span< blender::float3 > vert_positions, const blender::Span< blender::float3 > vert_normals, const blender::Span< int > corner_verts, const blender::int3 &corner_tri, const float u, const float v, const int mode, float res[3])
static void *	init_heights_data (MultiresBakeRender *bkr, ImBuf *ibuf)
static void	free_heights_data (void *bake_data)
static void	apply_heights_callback (const blender::Span< blender::float3 > vert_positions, const blender::Span< blender::float3 > vert_normals, const blender::OffsetIndices< int > faces, const blender::Span< int > corner_verts, const blender::Span< blender::int3 > corner_tris, const blender::Span< int > tri_faces, const blender::Span< blender::float2 > uv_map, DerivedMesh *hires_dm, void *thread_data_v, void *bake_data, ImBuf *ibuf, const int tri_index, const int lvl, const float st[2], float[3][3], const int x, const int y)
static void *	init_normal_data (MultiresBakeRender *bkr, ImBuf *)
static void	free_normal_data (void *bake_data)
static void	apply_tangmat_callback (const blender::Span< blender::float3 >, const blender::Span< blender::float3 >, const blender::OffsetIndices< int > faces, const blender::Span< int >, const blender::Span< blender::int3 > corner_tris, const blender::Span< int > tri_faces, const blender::Span< blender::float2 > uv_map, DerivedMesh *hires_dm, void *, void *bake_data, ImBuf *ibuf, const int tri_index, const int lvl, const float st[2], float tangmat[3][3], const int x, const int y)
static void	bake_ibuf_filter (ImBuf *ibuf, char *mask, const int margin, const char margin_type, DerivedMesh *dm, const float uv_offset[2])
static void	bake_ibuf_normalize_displacement (ImBuf *ibuf, const float *displacement, const char *mask, float displacement_min, float displacement_max)
static void	count_images (MultiresBakeRender *bkr)
static void	bake_images (MultiresBakeRender *bkr, MultiresBakeResult *result)
static void	finish_images (MultiresBakeRender *bkr, MultiresBakeResult *result)
void	RE_multires_bake_images (MultiresBakeRender *bkr)

Typedef Documentation

MFlushPixel

using MFlushPixel = void (*)(const MResolvePixelData *data, const int x, const int y)

Definition at line 100 of file multires_bake.cc.

MFreeBakeData

using MFreeBakeData = void (*)(void *bake_data)

Definition at line 62 of file multires_bake.cc.

MInitBakeData

using MInitBakeData = void *(*)(MultiresBakeRender *bkr, ImBuf *ibuf)

Definition at line 61 of file multires_bake.cc.

MPassKnownData

using MPassKnownData

Initial value:

 void (*)(blender::Span<blender::float3> vert_positions,
                                blender::Span<blender::float3> vert_normals,
                                blender::OffsetIndices<int> faces,
                                blender::Span<int> corner_verts,
                                blender::Span<blender::int3> corner_tris,
                                blender::Span<int> tri_faces,
                                blender::Span<blender::float2> uv_map,
                                DerivedMesh *hires_dm,
                                void *thread_data,
                                void *bake_data,
                                ImBuf *ibuf,
                                const int face_index,
                                const int lvl,
                                const float st[2],
                                float tangmat[3][3],
                                const int x,
                                const int y)

Definition at line 43 of file multires_bake.cc.

Function Documentation

apply_heights_callback()

void apply_heights_callback ( const blender::Span< blender::float3 > vert_positions, const blender::Span< blender::float3 > vert_normals, const blender::OffsetIndices< int > faces, const blender::Span< int > corner_verts, const blender::Span< blender::int3 > corner_tris, const blender::Span< int > tri_faces, const blender::Span< blender::float2 > uv_map, DerivedMesh * hires_dm, void * thread_data_v, void * bake_data, ImBuf * ibuf, const int tri_index, const int lvl, const float st[2], float [3][3], const int x, const int y )

static

Definition at line 867 of file multires_bake.cc.

References ImBuf::byte_buffer, clamp_v2(), ImBufByteBuffer::data, ImBufFloatBuffer::data, dot_v3v3(), faces, ImBuf::float_buffer, get_ccgdm_data(), MultiresBakeThread::height_max, MultiresBakeThread::height_min, MHeightBakeData::heights, interp_barycentric_corner_tri(), interp_bilinear_mpoly(), len, max_ff(), min_ff(), MHeightBakeData::orig_index_mp_to_orig, resolve_quad_uv_v2(), resolve_tri_uv_v2(), blender::IndexRange::size(), MHeightBakeData::ssdm, sub_v3_v3v3(), unit_float_to_uchar_clamp(), ImBuf::x, x, and y.

Referenced by bake_images().

apply_tangmat_callback()

void apply_tangmat_callback ( const blender::Span< blender::float3 > , const blender::Span< blender::float3 > , const blender::OffsetIndices< int > faces, const blender::Span< int > , const blender::Span< blender::int3 > corner_tris, const blender::Span< int > tri_faces, const blender::Span< blender::float2 > uv_map, DerivedMesh * hires_dm, void * , void * bake_data, ImBuf * ibuf, const int tri_index, const int lvl, const float st[2], float tangmat[3][3], const int x, const int y )

static

MultiresBake callback for normals' baking.

General idea:

• Find coord and normal of point with specified UV in hi-res mesh.
• Multiply it by tangmat.
• Vector in color space would be norm(vec) / 2 + (0.5, 0.5, 0.5).

Definition at line 990 of file multires_bake.cc.

References add_v3_v3(), ImBuf::byte_buffer, clamp_v2(), ImBufByteBuffer::data, ImBufFloatBuffer::data, faces, ImBuf::float_buffer, get_ccgdm_data(), mul_v3_m3v3(), normalize_v3_length(), MNormalBakeData::orig_index_mp_to_orig, resolve_quad_uv_v2(), resolve_tri_uv_v2(), rgb_float_to_uchar(), blender::IndexRange::size(), ImBuf::x, x, and y.

Referenced by bake_images().

bake_ibuf_filter()

void bake_ibuf_filter ( ImBuf * ibuf, char * mask, const int margin, const char margin_type, DerivedMesh * dm, const float uv_offset[2] )

static

Definition at line 1379 of file multires_bake.cc.

References BKE_imbuf_alpha_test(), IMB_filter_extend(), IMB_rectfill_alpha(), mask(), ImBuf::planes, R_BAKE_ADJACENT_FACES, R_BAKE_EXTEND, R_IMF_PLANES_RGBA, and RE_generate_texturemargin_adjacentfaces_dm().

Referenced by finish_images().

bake_ibuf_normalize_displacement()

void bake_ibuf_normalize_displacement ( ImBuf * ibuf, const float * displacement, const char * mask, float displacement_min, float displacement_max )

static

Definition at line 1414 of file multires_bake.cc.

References ImBuf::byte_buffer, ImBufByteBuffer::data, ImBufFloatBuffer::data, fabsf, FILTER_MASK_USED, ImBuf::float_buffer, i, IMB_get_pixel_count(), mask(), max_ff(), and unit_float_to_uchar_clamp().

Referenced by finish_images().

bake_images()

void bake_images ( MultiresBakeRender * bkr, MultiresBakeResult * result )

static

Definition at line 1491 of file multires_bake.cc.

References apply_heights_callback(), apply_tangmat_callback(), BKE_image_acquire_ibuf(), BKE_image_release_ibuf(), BKE_imageuser_default(), do_multires_bake(), FLT_MAX, free_heights_data(), free_normal_data(), Image::id, ID_TAG_DOIT, MultiresBakeRender::image, init_heights_data(), init_normal_data(), LISTBASE_FOREACH, BakeImBufuserData::mask_buffer, MEM_calloc_arrayN(), MEM_callocN(), MultiresBakeRender::mode, RE_BAKE_AO, RE_BAKE_DISPLACEMENT, RE_BAKE_NORMALS, result, ID::tag, ImageUser::tile, tile, Image::tiles, ImBuf::userdata, ImBuf::x, and ImBuf::y.

Referenced by RE_multires_bake_images().

bake_rasterize()

void bake_rasterize ( const MBakeRast * bake_rast, const float st0_in[2], const float st1_in[2], const float st2_in[2] )

static

Definition at line 294 of file multires_bake.cc.

References ceilf, MBakeRast::h, rasterize_half(), MBakeRast::w, and w().

Referenced by do_multires_bake_thread().

count_images()

void count_images ( MultiresBakeRender * bkr )

static

Definition at line 1459 of file multires_bake.cc.

References MultiresBakeRender::array, BLI_addtail(), BLI_genericNodeN(), BLI_listbase_clear(), data, i, Image::id, ID_TAG_DOIT, MultiresBakeRender::image, MultiresBakeRender::len, MultiresBakeRender::ob_image, ID::tag, and MultiresBakeRender::tot_image.

Referenced by RE_multires_bake_images().

do_multires_bake()

void do_multires_bake ( MultiresBakeRender * bkr, Image * ima, ImageTile * tile, ImBuf * ibuf, bool require_tangent, MPassKnownData passKnownData, MInitBakeData initBakeData, MFreeBakeData freeBakeData, MultiresBakeResult * result )

static

Definition at line 470 of file multires_bake.cc.

References blender::bke::MutableAttributeAccessor::add(), MResolvePixelData::bake_data, MultiresBakeThread::bake_rast, BKE_id_free(), BKE_image_get_tile_uv(), BKE_mesh_calc_loop_tangent_ex(), BKE_mesh_new_nomain(), MultiresBakeThread::bkr, BLI_listbase_count(), BLI_spin_end(), BLI_spin_init(), BLI_system_thread_count(), BLI_threadpool_end(), BLI_threadpool_init(), BLI_threadpool_insert(), CD_ORCO, CD_PROP_BOOL, CD_PROP_FLOAT2, CD_PROP_INT32, CD_TANGENT, MResolvePixelData::corner_tris, MResolvePixelData::corner_verts, Mesh::corners_num, MultiresBakeQueue::cur_tri, CustomData_get_layer_index(), CustomData_get_layer_named(), MultiresBakeThread::data, DM_get_loop_data_layer(), do_multires_bake_thread(), MultiresBakeRender::do_update, blender::bke::Edge, DerivedMesh::edgeData, Mesh::edges_num, blender::bke::Face, MResolvePixelData::face_normals, faces, MResolvePixelData::faces, Mesh::faces_num, FLT_MAX, flush_pixel(), VArray< T >::ForSpan(), DerivedMesh::getCornerEdgeArray, DerivedMesh::getCornerVertArray, DerivedMesh::getEdgeArray, DerivedMesh::getLoopDataArray, DerivedMesh::getNumEdges, DerivedMesh::getNumLoops, DerivedMesh::getNumPolys, DerivedMesh::getNumVerts, DerivedMesh::getPolyArray, DerivedMesh::getVertArray, DerivedMesh::getVertDataArray, MResolvePixelData::h, handles, MultiresBakeThread::height_max, MultiresBakeThread::height_min, MResolvePixelData::hires_dm, MultiresBakeRender::hires_dm, i, MResolvePixelData::ibuf, MultiresBakeThread::image, init_bake_rast(), init_ccgdm_arrays(), DerivedMesh::loopData, MultiresBakeRender::lores_dm, MResolvePixelData::lvl, MultiresBakeRender::lvl, MResolvePixelData::material_indices, max_ff(), min_ff(), MultiresBakeThread::num_total_faces, MResolvePixelData::pass_data, DerivedMesh::polyData, MResolvePixelData::pvtangent, MultiresBakeThread::queue, result, MResolvePixelData::sharp_faces, Span< T >::size(), MultiresBakeQueue::spin, DerivedMesh::tangent_mask, MResolvePixelData::thread_data, MultiresBakeRender::threads, tile, Image::tiles, MultiresBakeQueue::tot_tri, MResolvePixelData::tri_faces, MResolvePixelData::uv_map, MResolvePixelData::uv_offset, MResolvePixelData::vert_normals, MResolvePixelData::vert_positions, Mesh::verts_num, MResolvePixelData::w, ImBuf::x, and ImBuf::y.

Referenced by bake_images().

do_multires_bake_thread()

void * do_multires_bake_thread ( void * data_v )

static

Definition at line 392 of file multires_bake.cc.

References MultiresBakeRender::array, MultiresBakeThread::bake_rast, bake_rasterize(), MultiresBakeRender::baked_faces, MultiresBakeRender::baked_objects, MultiresBakeThread::bkr, BLI_spin_lock(), BLI_spin_unlock(), data, MultiresBakeThread::data, MultiresBakeRender::do_update, IB_DISPLAY_BUFFER_INVALID, IB_RECT_INVALID, MultiresBakeThread::image, MultiresBakeRender::len, multires_bake_queue_next_tri(), multiresbake_test_break(), MultiresBakeThread::num_total_faces, MultiresBakeRender::ob_image, MultiresBakeRender::progress, MultiresBakeThread::queue, MultiresBakeQueue::spin, sub_v2_v2v2(), and MultiresBakeRender::tot_obj.

Referenced by do_multires_bake().

finish_images()

void finish_images ( MultiresBakeRender * bkr, MultiresBakeResult * result )

static

Definition at line 1560 of file multires_bake.cc.

References bake_ibuf_filter(), bake_ibuf_normalize_displacement(), MultiresBakeRender::bake_margin, MultiresBakeRender::bake_margin_type, BKE_image_acquire_ibuf(), BKE_image_get_tile_uv(), BKE_image_mark_dirty(), BKE_image_release_ibuf(), BKE_imageuser_default(), ImBufFloatBuffer::data, DEG_id_tag_update(), BakeImBufuserData::displacement_buffer, ImBuf::float_buffer, IB_DISPLAY_BUFFER_INVALID, IB_MIPMAP_INVALID, IB_RECT_INVALID, Image::id, MultiresBakeRender::image, IMB_free_mipmaps(), LISTBASE_FOREACH, MultiresBakeRender::lores_dm, BakeImBufuserData::mask_buffer, MEM_freeN(), ImBuf::mipmap, MultiresBakeRender::mode, RE_BAKE_DISPLACEMENT, result, ImageUser::tile, tile, Image::tiles, ImBuf::userdata, ImBuf::userflags, ImBuf::x, and ImBuf::y.

Referenced by RE_multires_bake_images().

flush_pixel()

void flush_pixel ( const MResolvePixelData * data, const int x, const int y )

static

Definition at line 160 of file multires_bake.cc.

References cross_v3_v3v3(), data, invert_m3_m3(), mul_v3_fl(), multiresbake_get_normal(), resolve_tri_uv_v2(), sign(), v, w(), x, y, and zero_m3().

Referenced by do_multires_bake(), init_bake_rast(), and set_rast_triangle().

free_heights_data()

void free_heights_data ( void * bake_data )

static

Definition at line 850 of file multires_bake.cc.

References MEM_freeN(), DerivedMesh::release, and MHeightBakeData::ssdm.

Referenced by bake_images().

free_normal_data()

void free_normal_data ( void * bake_data )

static

Definition at line 975 of file multires_bake.cc.

References MEM_freeN().

Referenced by bake_images().

get_ccgdm_data()

void get_ccgdm_data ( const blender::OffsetIndices< int > lores_polys, DerivedMesh * hidm, const int * index_mp_to_orig, const int lvl, const int face_index, const float u, const float v, float co[3], float n[3] )

static

Definition at line 687 of file multires_bake.cc.

References CLAMP, col, DerivedMesh::getGridData, DerivedMesh::getGridKey, DerivedMesh::getGridOffset, DerivedMesh::getGridSize, interp_bilinear_grid(), mdisp_rot_face_to_crn(), size(), and v.

Referenced by apply_heights_callback(), and apply_tangmat_callback().

init_bake_rast()

void init_bake_rast ( MBakeRast * bake_rast, const ImBuf * ibuf, const MResolvePixelData * data, MFlushPixel flush_pixel, bool * do_update )

static

Definition at line 142 of file multires_bake.cc.

References data, MBakeRast::data, do_update, MBakeRast::do_update, flush_pixel(), MBakeRast::flush_pixel, MBakeRast::h, BakeImBufuserData::mask_buffer, MBakeRast::texels, ImBuf::userdata, MBakeRast::w, ImBuf::x, and ImBuf::y.

Referenced by do_multires_bake().

init_ccgdm_arrays()

void init_ccgdm_arrays ( DerivedMesh * dm )

static

Definition at line 453 of file multires_bake.cc.

References DerivedMesh::getGridData, DerivedMesh::getGridKey, DerivedMesh::getGridOffset, and DerivedMesh::getGridSize.

Referenced by do_multires_bake(), and init_heights_data().

init_heights_data()

void * init_heights_data ( MultiresBakeRender * bkr, ImBuf * ibuf )

static

Definition at line 812 of file multires_bake.cc.

References CD_ORIGINDEX, CLAMP, BakeImBufuserData::displacement_buffer, DerivedMesh::getPolyDataArray, MHeightBakeData::heights, IMB_get_pixel_count(), init_ccgdm_arrays(), SubsurfModifierData::levels, MultiresBakeRender::lores_dm, MultiresBakeRender::lvl, MEM_calloc_arrayN(), MEM_callocN(), MHeightBakeData::orig_index_mp_to_orig, SubsurfModifierData::quality, SubsurfModifierData::renderLevels, MultiresBakeRender::scene, MHeightBakeData::ssdm, subsurf_make_derived_from_derived(), SUBSURF_UV_SMOOTH_PRESERVE_BOUNDARIES, MultiresBakeRender::tot_lvl, MultiresBakeRender::use_lores_mesh, ImBuf::userdata, and SubsurfModifierData::uv_smooth.

Referenced by bake_images().

init_normal_data()

void * init_normal_data ( MultiresBakeRender * bkr, ImBuf *  )

static

Definition at line 962 of file multires_bake.cc.

References CD_ORIGINDEX, DerivedMesh::getPolyDataArray, MultiresBakeRender::lores_dm, MEM_callocN(), and MNormalBakeData::orig_index_mp_to_orig.

Referenced by bake_images().

interp_barycentric_corner_tri()

void interp_barycentric_corner_tri ( const blender::Span< blender::float3 > vert_positions, const blender::Span< blender::float3 > vert_normals, const blender::Span< int > corner_verts, const blender::int3 & corner_tri, const float u, const float v, const int mode, float res[3] )

static

Definition at line 785 of file multires_bake.cc.

References copy_v3_v3(), data, interp_barycentric_tri_v3(), and v.

Referenced by apply_heights_callback().

interp_bilinear_grid()

void interp_bilinear_grid ( const CCGKey & key, CCGElem * grid, float crn_x, float crn_y, int mode, float res[3] )

static

Definition at line 655 of file multires_bake.cc.

References CCG_grid_elem_co(), CCG_grid_elem_no(), copy_v3_v3(), data, CCGKey::grid_size, interp_bilinear_quad_v3(), and v.

Referenced by get_ccgdm_data().

interp_bilinear_mpoly()

void interp_bilinear_mpoly ( const blender::Span< blender::float3 > vert_positions, const blender::Span< blender::float3 > vert_normals, const blender::Span< int > corner_verts, const blender::IndexRange face, const float u, const float v, const int mode, float res[3] )

static

Definition at line 758 of file multires_bake.cc.

References copy_v3_v3(), data, interp_bilinear_quad_v3(), and v.

Referenced by apply_heights_callback().

multires_bake_queue_next_tri()

int multires_bake_queue_next_tri ( MultiresBakeQueue * queue )

static

Definition at line 374 of file multires_bake.cc.

References BLI_spin_lock(), BLI_spin_unlock(), MultiresBakeQueue::cur_tri, MultiresBakeQueue::spin, and MultiresBakeQueue::tot_tri.

Referenced by do_multires_bake_thread().

multiresbake_get_normal()

void multiresbake_get_normal ( const MResolvePixelData * data, const int tri_num, const int vert_index, float r_normal[3] )

static

Definition at line 125 of file multires_bake.cc.

References copy_v3_v3(), and data.

Referenced by flush_pixel().

multiresbake_test_break()

int multiresbake_test_break ( MultiresBakeRender * bkr )

static

Definition at line 340 of file multires_bake.cc.

References G, and MultiresBakeRender::stop.

Referenced by do_multires_bake_thread().

rasterize_half()

void rasterize_half ( const MBakeRast * bake_rast, const float s0_s, const float t0_s, const float s1_s, const float t1_s, const float s0_l, const float t0_l, const float s1_l, const float t1_l, const int y0_in, const int y1_in, const int is_mid_right )

static

Definition at line 244 of file multires_bake.cc.

References ceilf, fabsf, MBakeRast::h, set_rast_triangle(), MBakeRast::w, w(), x, and y.

Referenced by bake_rasterize().

RE_multires_bake_images()

void RE_multires_bake_images

(

MultiresBakeRender *

bkr

)

Definition at line 1625 of file multires_bake.cc.

References bake_images(), count_images(), finish_images(), and result.

Referenced by blender::ed::object::multiresbake_image_exec_locked(), and blender::ed::object::multiresbake_startjob().

set_rast_triangle()

void set_rast_triangle ( const MBakeRast * bake_rast, const int x, const int y )

static

Definition at line 228 of file multires_bake.cc.

References MBakeRast::data, MBakeRast::do_update, FILTER_MASK_USED, flush_pixel(), MBakeRast::h, MBakeRast::texels, MBakeRast::w, w(), x, and y.

Referenced by rasterize_half().