4585 lines
		
	
	
		
			138 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			4585 lines
		
	
	
		
			138 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/*
 | 
						|
 * ***** 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) 2017 by Blender Foundation.
 | 
						|
 * All rights reserved.
 | 
						|
 *
 | 
						|
 * Contributor(s): Blender Foundation, Mike Erwin, Dalai Felinto
 | 
						|
 *
 | 
						|
 * ***** END GPL LICENSE BLOCK *****
 | 
						|
 */
 | 
						|
 | 
						|
/** \file draw_cache_impl_mesh.c
 | 
						|
 *  \ingroup draw
 | 
						|
 *
 | 
						|
 * \brief Mesh API for render engines
 | 
						|
 */
 | 
						|
 | 
						|
#include "MEM_guardedalloc.h"
 | 
						|
 | 
						|
#include "BLI_utildefines.h"
 | 
						|
#include "BLI_math_vector.h"
 | 
						|
#include "BLI_math_bits.h"
 | 
						|
#include "BLI_math_color.h"
 | 
						|
#include "BLI_string.h"
 | 
						|
#include "BLI_alloca.h"
 | 
						|
#include "BLI_edgehash.h"
 | 
						|
 | 
						|
#include "DNA_mesh_types.h"
 | 
						|
#include "DNA_meshdata_types.h"
 | 
						|
#include "DNA_object_types.h"
 | 
						|
 | 
						|
#include "BKE_customdata.h"
 | 
						|
#include "BKE_deform.h"
 | 
						|
#include "BKE_editmesh.h"
 | 
						|
#include "BKE_editmesh_tangent.h"
 | 
						|
#include "BKE_mesh.h"
 | 
						|
#include "BKE_mesh_tangent.h"
 | 
						|
#include "BKE_colorband.h"
 | 
						|
 | 
						|
#include "bmesh.h"
 | 
						|
 | 
						|
#include "GPU_batch.h"
 | 
						|
#include "GPU_draw.h"
 | 
						|
#include "GPU_material.h"
 | 
						|
#include "GPU_texture.h"
 | 
						|
 | 
						|
#include "DRW_render.h"
 | 
						|
 | 
						|
#include "draw_cache_impl.h"  /* own include */
 | 
						|
 | 
						|
static void mesh_batch_cache_clear(Mesh *me);
 | 
						|
 | 
						|
/* ---------------------------------------------------------------------- */
 | 
						|
 | 
						|
/** \name Mesh/BMesh Interface (direct access to basic data).
 | 
						|
 * \{ */
 | 
						|
 | 
						|
static int mesh_render_verts_len_get(Mesh *me)
 | 
						|
{
 | 
						|
	return me->edit_btmesh ? me->edit_btmesh->bm->totvert : me->totvert;
 | 
						|
}
 | 
						|
 | 
						|
static int mesh_render_edges_len_get(Mesh *me)
 | 
						|
{
 | 
						|
	return me->edit_btmesh ? me->edit_btmesh->bm->totedge : me->totedge;
 | 
						|
}
 | 
						|
 | 
						|
static int mesh_render_looptri_len_get(Mesh *me)
 | 
						|
{
 | 
						|
	return me->edit_btmesh ? me->edit_btmesh->tottri : poly_to_tri_count(me->totpoly, me->totloop);
 | 
						|
}
 | 
						|
 | 
						|
static int mesh_render_polys_len_get(Mesh *me)
 | 
						|
{
 | 
						|
	return me->edit_btmesh ? me->edit_btmesh->bm->totface : me->totpoly;
 | 
						|
}
 | 
						|
 | 
						|
static int mesh_render_mat_len_get(Mesh *me)
 | 
						|
{
 | 
						|
	return MAX2(1, me->totcol);
 | 
						|
}
 | 
						|
 | 
						|
static int UNUSED_FUNCTION(mesh_render_loops_len_get)(Mesh *me)
 | 
						|
{
 | 
						|
	return me->edit_btmesh ? me->edit_btmesh->bm->totloop : me->totloop;
 | 
						|
}
 | 
						|
 | 
						|
/** \} */
 | 
						|
 | 
						|
 | 
						|
/* ---------------------------------------------------------------------- */
 | 
						|
 | 
						|
/** \name Mesh/BMesh Interface (indirect, partially cached access to complex data).
 | 
						|
 * \{ */
 | 
						|
 | 
						|
typedef struct EdgeAdjacentPolys {
 | 
						|
	int count;
 | 
						|
	int face_index[2];
 | 
						|
} EdgeAdjacentPolys;
 | 
						|
 | 
						|
typedef struct EdgeAdjacentVerts {
 | 
						|
	int vert_index[2]; /* -1 if none */
 | 
						|
} EdgeAdjacentVerts;
 | 
						|
 | 
						|
typedef struct EdgeDrawAttr {
 | 
						|
	uchar v_flag;
 | 
						|
	uchar e_flag;
 | 
						|
	uchar crease;
 | 
						|
	uchar bweight;
 | 
						|
} EdgeDrawAttr;
 | 
						|
 | 
						|
typedef struct MeshRenderData {
 | 
						|
	int types;
 | 
						|
 | 
						|
	int vert_len;
 | 
						|
	int edge_len;
 | 
						|
	int tri_len;
 | 
						|
	int loop_len;
 | 
						|
	int poly_len;
 | 
						|
	int mat_len;
 | 
						|
	int loose_vert_len;
 | 
						|
	int loose_edge_len;
 | 
						|
 | 
						|
	BMEditMesh *edit_bmesh;
 | 
						|
	struct EditMeshData *edit_data;
 | 
						|
 | 
						|
	MVert *mvert;
 | 
						|
	MEdge *medge;
 | 
						|
	MLoop *mloop;
 | 
						|
	MPoly *mpoly;
 | 
						|
	float (*orco)[3];  /* vertex coordinates normalized to bounding box */
 | 
						|
	bool is_orco_allocated;
 | 
						|
	MDeformVert *dvert;
 | 
						|
	MLoopUV *mloopuv;
 | 
						|
	MLoopCol *mloopcol;
 | 
						|
	float (*loop_normals)[3];
 | 
						|
 | 
						|
	/* CustomData 'cd' cache for efficient access. */
 | 
						|
	struct {
 | 
						|
		struct {
 | 
						|
			MLoopUV **uv;
 | 
						|
			int       uv_len;
 | 
						|
			int       uv_active;
 | 
						|
 | 
						|
			MLoopCol **vcol;
 | 
						|
			int        vcol_len;
 | 
						|
			int        vcol_active;
 | 
						|
 | 
						|
			float (**tangent)[4];
 | 
						|
			int      tangent_len;
 | 
						|
			int      tangent_active;
 | 
						|
 | 
						|
			bool *auto_vcol;
 | 
						|
		} layers;
 | 
						|
 | 
						|
		/* Custom-data offsets (only needed for BMesh access) */
 | 
						|
		struct {
 | 
						|
			int crease;
 | 
						|
			int bweight;
 | 
						|
			int *uv;
 | 
						|
			int *vcol;
 | 
						|
		} offset;
 | 
						|
 | 
						|
		struct {
 | 
						|
			char (*auto_mix)[32];
 | 
						|
			char (*uv)[32];
 | 
						|
			char (*vcol)[32];
 | 
						|
			char (*tangent)[32];
 | 
						|
		} uuid;
 | 
						|
 | 
						|
		/* for certain cases we need an output loop-data storage (bmesh tangents) */
 | 
						|
		struct {
 | 
						|
			CustomData ldata;
 | 
						|
			/* grr, special case variable (use in place of 'dm->tangent_mask') */
 | 
						|
			short tangent_mask;
 | 
						|
		} output;
 | 
						|
	} cd;
 | 
						|
 | 
						|
	BMVert *eve_act;
 | 
						|
	BMEdge *eed_act;
 | 
						|
	BMFace *efa_act;
 | 
						|
 | 
						|
	/* Data created on-demand (usually not for bmesh-based data). */
 | 
						|
	EdgeAdjacentPolys *edges_adjacent_polys;
 | 
						|
	MLoopTri *mlooptri;
 | 
						|
	int *loose_edges;
 | 
						|
	int *loose_verts;
 | 
						|
 | 
						|
	float (*poly_normals)[3];
 | 
						|
	float (*vert_weight_color)[3];
 | 
						|
	char (*vert_color)[3];
 | 
						|
	GPUPackedNormal *poly_normals_pack;
 | 
						|
	GPUPackedNormal *vert_normals_pack;
 | 
						|
	bool *edge_select_bool;
 | 
						|
} MeshRenderData;
 | 
						|
 | 
						|
enum {
 | 
						|
	MR_DATATYPE_VERT       = 1 << 0,
 | 
						|
	MR_DATATYPE_EDGE       = 1 << 1,
 | 
						|
	MR_DATATYPE_LOOPTRI    = 1 << 2,
 | 
						|
	MR_DATATYPE_LOOP       = 1 << 3,
 | 
						|
	MR_DATATYPE_POLY       = 1 << 4,
 | 
						|
	MR_DATATYPE_OVERLAY    = 1 << 5,
 | 
						|
	MR_DATATYPE_SHADING    = 1 << 6,
 | 
						|
	MR_DATATYPE_DVERT      = 1 << 7,
 | 
						|
	MR_DATATYPE_LOOPCOL    = 1 << 8,
 | 
						|
	MR_DATATYPE_LOOPUV     = 1 << 9,
 | 
						|
};
 | 
						|
 | 
						|
/**
 | 
						|
 * These functions look like they would be slow but they will typically return true on the first iteration.
 | 
						|
 * Only false when all attached elements are hidden.
 | 
						|
 */
 | 
						|
static bool bm_vert_has_visible_edge(const BMVert *v)
 | 
						|
{
 | 
						|
	const BMEdge *e_iter, *e_first;
 | 
						|
 | 
						|
	e_iter = e_first = v->e;
 | 
						|
	do {
 | 
						|
		if (!BM_elem_flag_test(e_iter, BM_ELEM_HIDDEN)) {
 | 
						|
			return true;
 | 
						|
		}
 | 
						|
	} while ((e_iter = BM_DISK_EDGE_NEXT(e_iter, v)) != e_first);
 | 
						|
	return false;
 | 
						|
}
 | 
						|
 | 
						|
static bool bm_edge_has_visible_face(const BMEdge *e)
 | 
						|
{
 | 
						|
	const BMLoop *l_iter, *l_first;
 | 
						|
	l_iter = l_first = e->l;
 | 
						|
	do {
 | 
						|
		if (!BM_elem_flag_test(l_iter->f, BM_ELEM_HIDDEN)) {
 | 
						|
			return true;
 | 
						|
		}
 | 
						|
	} while ((l_iter = l_iter->radial_next) != l_first);
 | 
						|
	return false;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
static void mesh_cd_calc_used_gpu_layers(
 | 
						|
        CustomData *UNUSED(cd_vdata), uchar cd_vused[CD_NUMTYPES],
 | 
						|
        CustomData *cd_ldata, ushort cd_lused[CD_NUMTYPES],
 | 
						|
        struct GPUMaterial **gpumat_array, int gpumat_array_len)
 | 
						|
{
 | 
						|
	/* See: DM_vertex_attributes_from_gpu for similar logic */
 | 
						|
	GPUVertexAttribs gattribs = {{{0}}};
 | 
						|
 | 
						|
	for (int i = 0; i < gpumat_array_len; i++) {
 | 
						|
		GPUMaterial *gpumat = gpumat_array[i];
 | 
						|
		if (gpumat) {
 | 
						|
			GPU_material_vertex_attributes(gpumat, &gattribs);
 | 
						|
			for (int j = 0; j < gattribs.totlayer; j++) {
 | 
						|
				const char *name = gattribs.layer[j].name;
 | 
						|
				int type = gattribs.layer[j].type;
 | 
						|
				int layer = -1;
 | 
						|
 | 
						|
				if (type == CD_AUTO_FROM_NAME) {
 | 
						|
					/* We need to deduct what exact layer is used.
 | 
						|
					 *
 | 
						|
					 * We do it based on the specified name.
 | 
						|
					 */
 | 
						|
					if (name[0] != '\0') {
 | 
						|
						layer = CustomData_get_named_layer(cd_ldata, CD_MLOOPUV, name);
 | 
						|
						type = CD_MTFACE;
 | 
						|
 | 
						|
						if (layer == -1) {
 | 
						|
							layer = CustomData_get_named_layer(cd_ldata, CD_MLOOPCOL, name);
 | 
						|
							type = CD_MCOL;
 | 
						|
						}
 | 
						|
#if 0					/* Tangents are always from UV's - this will never happen. */
 | 
						|
						if (layer == -1) {
 | 
						|
							layer = CustomData_get_named_layer(cd_ldata, CD_TANGENT, name);
 | 
						|
							type = CD_TANGENT;
 | 
						|
						}
 | 
						|
#endif
 | 
						|
						if (layer == -1) {
 | 
						|
							continue;
 | 
						|
						}
 | 
						|
					}
 | 
						|
					else {
 | 
						|
						/* Fall back to the UV layer, which matches old behavior. */
 | 
						|
						type = CD_MTFACE;
 | 
						|
					}
 | 
						|
				}
 | 
						|
 | 
						|
				switch (type) {
 | 
						|
					case CD_MTFACE:
 | 
						|
					{
 | 
						|
						if (layer == -1) {
 | 
						|
							layer = (name[0] != '\0') ?
 | 
						|
							        CustomData_get_named_layer(cd_ldata, CD_MLOOPUV, name) :
 | 
						|
							        CustomData_get_active_layer(cd_ldata, CD_MLOOPUV);
 | 
						|
						}
 | 
						|
						if (layer != -1) {
 | 
						|
							cd_lused[CD_MLOOPUV] |= (1 << layer);
 | 
						|
						}
 | 
						|
						break;
 | 
						|
					}
 | 
						|
					case CD_TANGENT:
 | 
						|
					{
 | 
						|
						if (layer == -1) {
 | 
						|
							layer = (name[0] != '\0') ?
 | 
						|
							        CustomData_get_named_layer(cd_ldata, CD_MLOOPUV, name) :
 | 
						|
							        CustomData_get_active_layer(cd_ldata, CD_MLOOPUV);
 | 
						|
						}
 | 
						|
						if (layer != -1) {
 | 
						|
							cd_lused[CD_TANGENT] |= (1 << layer);
 | 
						|
						}
 | 
						|
						else {
 | 
						|
							/* no UV layers at all => requesting orco */
 | 
						|
							cd_lused[CD_TANGENT] |= DM_TANGENT_MASK_ORCO;
 | 
						|
							cd_vused[CD_ORCO] |= 1;
 | 
						|
						}
 | 
						|
						break;
 | 
						|
					}
 | 
						|
					case CD_MCOL:
 | 
						|
					{
 | 
						|
						if (layer == -1) {
 | 
						|
							layer = (name[0] != '\0') ?
 | 
						|
							        CustomData_get_named_layer(cd_ldata, CD_MLOOPCOL, name) :
 | 
						|
							        CustomData_get_active_layer(cd_ldata, CD_MLOOPCOL);
 | 
						|
						}
 | 
						|
						if (layer != -1) {
 | 
						|
							cd_lused[CD_MLOOPCOL] |= (1 << layer);
 | 
						|
						}
 | 
						|
						break;
 | 
						|
					}
 | 
						|
					case CD_ORCO:
 | 
						|
					{
 | 
						|
						cd_vused[CD_ORCO] |= 1;
 | 
						|
						break;
 | 
						|
					}
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
static void mesh_render_calc_normals_loop_and_poly(const Mesh *me, const float split_angle, MeshRenderData *rdata)
 | 
						|
{
 | 
						|
	BLI_assert((me->flag & ME_AUTOSMOOTH) != 0);
 | 
						|
 | 
						|
	int totloop = me->totloop;
 | 
						|
	int totpoly = me->totpoly;
 | 
						|
	float (*loop_normals)[3] = MEM_mallocN(sizeof(*loop_normals) * totloop, __func__);
 | 
						|
	float (*poly_normals)[3] = MEM_mallocN(sizeof(*poly_normals) * totpoly, __func__);
 | 
						|
	short (*clnors)[2] = CustomData_get_layer(&me->ldata, CD_CUSTOMLOOPNORMAL);
 | 
						|
 | 
						|
	BKE_mesh_calc_normals_poly(
 | 
						|
	        me->mvert, NULL, me->totvert,
 | 
						|
	        me->mloop, me->mpoly, totloop, totpoly, poly_normals, false);
 | 
						|
 | 
						|
	BKE_mesh_normals_loop_split(
 | 
						|
	        me->mvert, me->totvert, me->medge, me->totedge,
 | 
						|
	        me->mloop, loop_normals, totloop, me->mpoly, poly_normals, totpoly,
 | 
						|
	        true, split_angle, NULL, clnors, NULL);
 | 
						|
 | 
						|
	rdata->loop_len = totloop;
 | 
						|
	rdata->poly_len = totpoly;
 | 
						|
	rdata->loop_normals = loop_normals;
 | 
						|
	rdata->poly_normals = poly_normals;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/**
 | 
						|
 * TODO(campbell): 'gpumat_array' may include materials linked to the object.
 | 
						|
 * While not default, object materials should be supported.
 | 
						|
 * Although this only impacts the data that's generated, not the materials that display.
 | 
						|
 */
 | 
						|
static MeshRenderData *mesh_render_data_create_ex(
 | 
						|
        Mesh *me, const int types,
 | 
						|
        struct GPUMaterial **gpumat_array, uint gpumat_array_len)
 | 
						|
{
 | 
						|
	MeshRenderData *rdata = MEM_callocN(sizeof(*rdata), __func__);
 | 
						|
	rdata->types = types;
 | 
						|
	rdata->mat_len = mesh_render_mat_len_get(me);
 | 
						|
 | 
						|
	CustomData_reset(&rdata->cd.output.ldata);
 | 
						|
 | 
						|
	const bool is_auto_smooth = (me->flag & ME_AUTOSMOOTH) != 0;
 | 
						|
	const float split_angle = is_auto_smooth ? me->smoothresh : (float)M_PI;
 | 
						|
 | 
						|
	if (me->edit_btmesh) {
 | 
						|
		BMEditMesh *embm = me->edit_btmesh;
 | 
						|
		BMesh *bm = embm->bm;
 | 
						|
 | 
						|
		rdata->edit_bmesh = embm;
 | 
						|
		rdata->edit_data = me->runtime.edit_data;
 | 
						|
 | 
						|
		int bm_ensure_types = 0;
 | 
						|
		if (types & MR_DATATYPE_VERT) {
 | 
						|
			rdata->vert_len = bm->totvert;
 | 
						|
			bm_ensure_types |= BM_VERT;
 | 
						|
		}
 | 
						|
		if (types & MR_DATATYPE_EDGE) {
 | 
						|
			rdata->edge_len = bm->totedge;
 | 
						|
			bm_ensure_types |= BM_EDGE;
 | 
						|
		}
 | 
						|
		if (types & MR_DATATYPE_LOOPTRI) {
 | 
						|
			BKE_editmesh_tessface_calc(embm);
 | 
						|
			int tottri = embm->tottri;
 | 
						|
			rdata->mlooptri = MEM_mallocN(sizeof(*rdata->mlooptri) * embm->tottri, __func__);
 | 
						|
			for (int index = 0; index < tottri ; index ++ ) {
 | 
						|
				BMLoop **bmtri = embm->looptris[index];
 | 
						|
				MLoopTri *mtri = &rdata->mlooptri[index];
 | 
						|
				mtri->tri[0] = BM_elem_index_get(bmtri[0]);
 | 
						|
				mtri->tri[1] = BM_elem_index_get(bmtri[1]);
 | 
						|
				mtri->tri[2] = BM_elem_index_get(bmtri[2]);
 | 
						|
			}
 | 
						|
			rdata->tri_len = tottri;
 | 
						|
		}
 | 
						|
		if (types & MR_DATATYPE_LOOP) {
 | 
						|
			int totloop = bm->totloop;
 | 
						|
			if (is_auto_smooth) {
 | 
						|
				rdata->loop_normals = MEM_mallocN(sizeof(*rdata->loop_normals) * totloop, __func__);
 | 
						|
				int cd_loop_clnors_offset = CustomData_get_layer_index(&bm->ldata, CD_CUSTOMLOOPNORMAL);
 | 
						|
				BM_loops_calc_normal_vcos(
 | 
						|
				        bm, NULL, NULL, NULL, true, split_angle, rdata->loop_normals, NULL, NULL,
 | 
						|
				        cd_loop_clnors_offset, false);
 | 
						|
			}
 | 
						|
			rdata->loop_len = totloop;
 | 
						|
			bm_ensure_types |= BM_LOOP;
 | 
						|
		}
 | 
						|
		if (types & MR_DATATYPE_POLY) {
 | 
						|
			rdata->poly_len = bm->totface;
 | 
						|
			bm_ensure_types |= BM_FACE;
 | 
						|
		}
 | 
						|
		if (types & MR_DATATYPE_OVERLAY) {
 | 
						|
			rdata->efa_act = BM_mesh_active_face_get(bm, false, true);
 | 
						|
			rdata->eed_act = BM_mesh_active_edge_get(bm);
 | 
						|
			rdata->eve_act = BM_mesh_active_vert_get(bm);
 | 
						|
			rdata->cd.offset.crease = CustomData_get_offset(&bm->edata, CD_CREASE);
 | 
						|
			rdata->cd.offset.bweight = CustomData_get_offset(&bm->edata, CD_BWEIGHT);
 | 
						|
		}
 | 
						|
		if (types & (MR_DATATYPE_DVERT)) {
 | 
						|
			bm_ensure_types |= BM_VERT;
 | 
						|
		}
 | 
						|
		if (rdata->edit_data != NULL) {
 | 
						|
			bm_ensure_types |= BM_VERT;
 | 
						|
		}
 | 
						|
 | 
						|
		BM_mesh_elem_index_ensure(bm, bm_ensure_types);
 | 
						|
		BM_mesh_elem_table_ensure(bm, bm_ensure_types & ~BM_LOOP);
 | 
						|
		if (types & MR_DATATYPE_OVERLAY) {
 | 
						|
			rdata->loose_vert_len = rdata->loose_edge_len = 0;
 | 
						|
 | 
						|
			int *lverts = rdata->loose_verts = MEM_mallocN(rdata->vert_len * sizeof(int), "Loose Vert");
 | 
						|
			int *ledges = rdata->loose_edges = MEM_mallocN(rdata->edge_len * sizeof(int), "Loose Edges");
 | 
						|
 | 
						|
			{
 | 
						|
				BLI_assert((bm->elem_table_dirty & BM_VERT) == 0);
 | 
						|
				BMVert **vtable = bm->vtable;
 | 
						|
				for (int i = 0; i < bm->totvert; i++) {
 | 
						|
					const BMVert *eve = vtable[i];
 | 
						|
					if (!BM_elem_flag_test(eve, BM_ELEM_HIDDEN)) {
 | 
						|
						/* Loose vert */
 | 
						|
						if (eve->e == NULL || !bm_vert_has_visible_edge(eve)) {
 | 
						|
							lverts[rdata->loose_vert_len++] = i;
 | 
						|
						}
 | 
						|
					}
 | 
						|
				}
 | 
						|
			}
 | 
						|
 | 
						|
			{
 | 
						|
				BLI_assert((bm->elem_table_dirty & BM_EDGE) == 0);
 | 
						|
				BMEdge **etable = bm->etable;
 | 
						|
				for (int i = 0; i < bm->totedge; i++) {
 | 
						|
					const BMEdge *eed = etable[i];
 | 
						|
					if (!BM_elem_flag_test(eed, BM_ELEM_HIDDEN)) {
 | 
						|
						/* Loose edge */
 | 
						|
						if (eed->l == NULL || !bm_edge_has_visible_face(eed)) {
 | 
						|
							ledges[rdata->loose_edge_len++] = i;
 | 
						|
						}
 | 
						|
					}
 | 
						|
				}
 | 
						|
			}
 | 
						|
 | 
						|
			rdata->loose_verts = MEM_reallocN(rdata->loose_verts, rdata->loose_vert_len * sizeof(int));
 | 
						|
			rdata->loose_edges = MEM_reallocN(rdata->loose_edges, rdata->loose_edge_len * sizeof(int));
 | 
						|
		}
 | 
						|
	}
 | 
						|
	else {
 | 
						|
		if (types & (MR_DATATYPE_VERT)) {
 | 
						|
			rdata->vert_len = me->totvert;
 | 
						|
			rdata->mvert = CustomData_get_layer(&me->vdata, CD_MVERT);
 | 
						|
		}
 | 
						|
		if (types & (MR_DATATYPE_EDGE)) {
 | 
						|
			rdata->edge_len = me->totedge;
 | 
						|
			rdata->medge = CustomData_get_layer(&me->edata, CD_MEDGE);
 | 
						|
		}
 | 
						|
		if (types & MR_DATATYPE_LOOPTRI) {
 | 
						|
			const int tri_len = rdata->tri_len = poly_to_tri_count(me->totpoly, me->totloop);
 | 
						|
			rdata->mlooptri = MEM_mallocN(sizeof(*rdata->mlooptri) * tri_len, __func__);
 | 
						|
			BKE_mesh_recalc_looptri(me->mloop, me->mpoly, me->mvert, me->totloop, me->totpoly, rdata->mlooptri);
 | 
						|
		}
 | 
						|
		if (types & MR_DATATYPE_LOOP) {
 | 
						|
			rdata->loop_len = me->totloop;
 | 
						|
			rdata->mloop = CustomData_get_layer(&me->ldata, CD_MLOOP);
 | 
						|
 | 
						|
			if (is_auto_smooth) {
 | 
						|
				mesh_render_calc_normals_loop_and_poly(me, split_angle, rdata);
 | 
						|
			}
 | 
						|
		}
 | 
						|
		if (types & MR_DATATYPE_POLY) {
 | 
						|
			rdata->poly_len = me->totpoly;
 | 
						|
			rdata->mpoly = CustomData_get_layer(&me->pdata, CD_MPOLY);
 | 
						|
		}
 | 
						|
		if (types & MR_DATATYPE_DVERT) {
 | 
						|
			rdata->vert_len = me->totvert;
 | 
						|
			rdata->dvert = CustomData_get_layer(&me->vdata, CD_MDEFORMVERT);
 | 
						|
		}
 | 
						|
		if (types & MR_DATATYPE_LOOPCOL) {
 | 
						|
			rdata->loop_len = me->totloop;
 | 
						|
			rdata->mloopcol = CustomData_get_layer(&me->ldata, CD_MLOOPCOL);
 | 
						|
		}
 | 
						|
		if (types & MR_DATATYPE_LOOPUV) {
 | 
						|
			rdata->loop_len = me->totloop;
 | 
						|
			rdata->mloopuv = CustomData_get_layer(&me->ldata, CD_MLOOPUV);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	if (types & MR_DATATYPE_SHADING) {
 | 
						|
		CustomData *cd_vdata, *cd_ldata;
 | 
						|
 | 
						|
		if (me->edit_btmesh) {
 | 
						|
			BMesh *bm = me->edit_btmesh->bm;
 | 
						|
			cd_vdata = &bm->vdata;
 | 
						|
			cd_ldata = &bm->ldata;
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			cd_vdata = &me->vdata;
 | 
						|
			cd_ldata = &me->ldata;
 | 
						|
		}
 | 
						|
 | 
						|
		/* Add edge/poly if we need them */
 | 
						|
		uchar cd_vused[CD_NUMTYPES] = {0};
 | 
						|
		ushort cd_lused[CD_NUMTYPES] = {0};
 | 
						|
 | 
						|
		mesh_cd_calc_used_gpu_layers(
 | 
						|
		        cd_vdata, cd_vused,
 | 
						|
		        cd_ldata, cd_lused,
 | 
						|
		        gpumat_array, gpumat_array_len);
 | 
						|
 | 
						|
 | 
						|
		rdata->cd.layers.uv_active = CustomData_get_active_layer(cd_ldata, CD_MLOOPUV);
 | 
						|
		rdata->cd.layers.vcol_active = CustomData_get_active_layer(cd_ldata, CD_MLOOPCOL);
 | 
						|
		rdata->cd.layers.tangent_active = rdata->cd.layers.uv_active;
 | 
						|
 | 
						|
#define CD_VALIDATE_ACTIVE_LAYER(active_index, used) \
 | 
						|
		if ((active_index != -1) && (used & (1 << active_index)) == 0) { \
 | 
						|
			active_index = -1; \
 | 
						|
		} ((void)0)
 | 
						|
 | 
						|
		CD_VALIDATE_ACTIVE_LAYER(rdata->cd.layers.uv_active, cd_lused[CD_MLOOPUV]);
 | 
						|
		CD_VALIDATE_ACTIVE_LAYER(rdata->cd.layers.tangent_active, cd_lused[CD_TANGENT]);
 | 
						|
		CD_VALIDATE_ACTIVE_LAYER(rdata->cd.layers.vcol_active, cd_lused[CD_MLOOPCOL]);
 | 
						|
 | 
						|
#undef CD_VALIDATE_ACTIVE_LAYER
 | 
						|
 | 
						|
		rdata->is_orco_allocated = false;
 | 
						|
		if (cd_vused[CD_ORCO] & 1) {
 | 
						|
			rdata->orco = CustomData_get_layer(cd_vdata, CD_ORCO);
 | 
						|
			/* If orco is not available compute it ourselves */
 | 
						|
			if (!rdata->orco) {
 | 
						|
				rdata->is_orco_allocated = true;
 | 
						|
				if (me->edit_btmesh) {
 | 
						|
					BMesh *bm = me->edit_btmesh->bm;
 | 
						|
					rdata->orco = MEM_mallocN(sizeof(*rdata->orco) * rdata->vert_len, "orco mesh");
 | 
						|
					BLI_assert((bm->elem_table_dirty & BM_VERT) == 0);
 | 
						|
					BMVert **vtable = bm->vtable;
 | 
						|
					for (int i = 0; i < bm->totvert; i++) {
 | 
						|
						copy_v3_v3(rdata->orco[i], vtable[i]->co);
 | 
						|
					}
 | 
						|
					BKE_mesh_orco_verts_transform(me, rdata->orco, rdata->vert_len, 0);
 | 
						|
				}
 | 
						|
				else {
 | 
						|
					rdata->orco = MEM_mallocN(sizeof(*rdata->orco) * rdata->vert_len, "orco mesh");
 | 
						|
					MVert *mvert = rdata->mvert;
 | 
						|
					for (int a = 0; a < rdata->vert_len; a++, mvert++) {
 | 
						|
						copy_v3_v3(rdata->orco[a], mvert->co);
 | 
						|
					}
 | 
						|
					BKE_mesh_orco_verts_transform(me, rdata->orco, rdata->vert_len, 0);
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			rdata->orco = NULL;
 | 
						|
		}
 | 
						|
 | 
						|
		/* don't access mesh directly, instead use vars taken from BMesh or Mesh */
 | 
						|
#define me DONT_USE_THIS
 | 
						|
#ifdef  me /* quiet warning */
 | 
						|
#endif
 | 
						|
		struct {
 | 
						|
			uint uv_len;
 | 
						|
			uint vcol_len;
 | 
						|
		} cd_layers_src = {
 | 
						|
			.uv_len = CustomData_number_of_layers(cd_ldata, CD_MLOOPUV),
 | 
						|
			.vcol_len = CustomData_number_of_layers(cd_ldata, CD_MLOOPCOL),
 | 
						|
		};
 | 
						|
 | 
						|
		rdata->cd.layers.uv_len = count_bits_i(cd_lused[CD_MLOOPUV]);
 | 
						|
		rdata->cd.layers.tangent_len = count_bits_i(cd_lused[CD_TANGENT]);
 | 
						|
		rdata->cd.layers.vcol_len = count_bits_i(cd_lused[CD_MLOOPCOL]);
 | 
						|
 | 
						|
		rdata->cd.layers.uv = MEM_mallocN(sizeof(*rdata->cd.layers.uv) * rdata->cd.layers.uv_len, __func__);
 | 
						|
		rdata->cd.layers.vcol = MEM_mallocN(sizeof(*rdata->cd.layers.vcol) * rdata->cd.layers.vcol_len, __func__);
 | 
						|
		rdata->cd.layers.tangent = MEM_mallocN(sizeof(*rdata->cd.layers.tangent) * rdata->cd.layers.tangent_len, __func__);
 | 
						|
 | 
						|
		rdata->cd.uuid.uv = MEM_mallocN(sizeof(*rdata->cd.uuid.uv) * rdata->cd.layers.uv_len, __func__);
 | 
						|
		rdata->cd.uuid.vcol = MEM_mallocN(sizeof(*rdata->cd.uuid.vcol) * rdata->cd.layers.vcol_len, __func__);
 | 
						|
		rdata->cd.uuid.tangent = MEM_mallocN(sizeof(*rdata->cd.uuid.tangent) * rdata->cd.layers.tangent_len, __func__);
 | 
						|
 | 
						|
		rdata->cd.offset.uv = MEM_mallocN(sizeof(*rdata->cd.offset.uv) * rdata->cd.layers.uv_len, __func__);
 | 
						|
		rdata->cd.offset.vcol = MEM_mallocN(sizeof(*rdata->cd.offset.vcol) * rdata->cd.layers.vcol_len, __func__);
 | 
						|
 | 
						|
		/* Allocate max */
 | 
						|
		rdata->cd.layers.auto_vcol = MEM_callocN(
 | 
						|
		        sizeof(*rdata->cd.layers.auto_vcol) * rdata->cd.layers.vcol_len, __func__);
 | 
						|
		rdata->cd.uuid.auto_mix = MEM_mallocN(
 | 
						|
		        sizeof(*rdata->cd.uuid.auto_mix) * (rdata->cd.layers.vcol_len + rdata->cd.layers.uv_len), __func__);
 | 
						|
 | 
						|
		/* XXX FIXME XXX */
 | 
						|
		/* We use a hash to identify each data layer based on its name.
 | 
						|
		 * Gawain then search for this name in the current shader and bind if it exists.
 | 
						|
		 * NOTE : This is prone to hash collision.
 | 
						|
		 * One solution to hash collision would be to format the cd layer name
 | 
						|
		 * to a safe glsl var name, but without name clash.
 | 
						|
		 * NOTE 2 : Replicate changes to code_generate_vertex_new() in gpu_codegen.c */
 | 
						|
		if (rdata->cd.layers.vcol_len != 0) {
 | 
						|
			for (int i_src = 0, i_dst = 0; i_src < cd_layers_src.vcol_len; i_src++, i_dst++) {
 | 
						|
				if ((cd_lused[CD_MLOOPCOL] & (1 << i_src)) == 0) {
 | 
						|
					i_dst--;
 | 
						|
					if (rdata->cd.layers.vcol_active >= i_src) {
 | 
						|
						rdata->cd.layers.vcol_active--;
 | 
						|
					}
 | 
						|
				}
 | 
						|
				else {
 | 
						|
					const char *name = CustomData_get_layer_name(cd_ldata, CD_MLOOPCOL, i_src);
 | 
						|
					uint hash = BLI_ghashutil_strhash_p(name);
 | 
						|
					BLI_snprintf(rdata->cd.uuid.vcol[i_dst], sizeof(*rdata->cd.uuid.vcol), "c%u", hash);
 | 
						|
					rdata->cd.layers.vcol[i_dst] = CustomData_get_layer_n(cd_ldata, CD_MLOOPCOL, i_src);
 | 
						|
					if (rdata->edit_bmesh) {
 | 
						|
						rdata->cd.offset.vcol[i_dst] = CustomData_get_n_offset(
 | 
						|
						        &rdata->edit_bmesh->bm->ldata, CD_MLOOPCOL, i_src);
 | 
						|
					}
 | 
						|
 | 
						|
					/* Gather number of auto layers. */
 | 
						|
					/* We only do vcols that are not overridden by uvs */
 | 
						|
					if (CustomData_get_named_layer_index(cd_ldata, CD_MLOOPUV, name) == -1) {
 | 
						|
						BLI_snprintf(
 | 
						|
						        rdata->cd.uuid.auto_mix[rdata->cd.layers.uv_len + i_dst],
 | 
						|
						        sizeof(*rdata->cd.uuid.auto_mix), "a%u", hash);
 | 
						|
						rdata->cd.layers.auto_vcol[i_dst] = true;
 | 
						|
					}
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		/* Start Fresh */
 | 
						|
		CustomData_free_layers(cd_ldata, CD_TANGENT, rdata->loop_len);
 | 
						|
		CustomData_free_layers(cd_ldata, CD_MLOOPTANGENT, rdata->loop_len);
 | 
						|
 | 
						|
		if (rdata->cd.layers.uv_len != 0) {
 | 
						|
			for (int i_src = 0, i_dst = 0; i_src < cd_layers_src.uv_len; i_src++, i_dst++) {
 | 
						|
				if ((cd_lused[CD_MLOOPUV] & (1 << i_src)) == 0) {
 | 
						|
					i_dst--;
 | 
						|
					if (rdata->cd.layers.uv_active >= i_src) {
 | 
						|
						rdata->cd.layers.uv_active--;
 | 
						|
					}
 | 
						|
				}
 | 
						|
				else {
 | 
						|
					const char *name = CustomData_get_layer_name(cd_ldata, CD_MLOOPUV, i_src);
 | 
						|
					uint hash = BLI_ghashutil_strhash_p(name);
 | 
						|
 | 
						|
					BLI_snprintf(rdata->cd.uuid.uv[i_dst], sizeof(*rdata->cd.uuid.uv), "u%u", hash);
 | 
						|
					rdata->cd.layers.uv[i_dst] = CustomData_get_layer_n(cd_ldata, CD_MLOOPUV, i_src);
 | 
						|
					if (rdata->edit_bmesh) {
 | 
						|
						rdata->cd.offset.uv[i_dst] = CustomData_get_n_offset(
 | 
						|
						        &rdata->edit_bmesh->bm->ldata, CD_MLOOPUV, i_src);
 | 
						|
					}
 | 
						|
					BLI_snprintf(rdata->cd.uuid.auto_mix[i_dst], sizeof(*rdata->cd.uuid.auto_mix), "a%u", hash);
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		if (rdata->cd.layers.tangent_len != 0) {
 | 
						|
 | 
						|
			/* -------------------------------------------------------------------- */
 | 
						|
			/* Pre-calculate tangents into 'rdata->cd.output.ldata' */
 | 
						|
 | 
						|
			BLI_assert(!CustomData_has_layer(&rdata->cd.output.ldata, CD_TANGENT));
 | 
						|
 | 
						|
			/* Tangent Names */
 | 
						|
			char tangent_names[MAX_MTFACE][MAX_NAME];
 | 
						|
			for (int i_src = 0, i_dst = 0; i_src < cd_layers_src.uv_len; i_src++, i_dst++) {
 | 
						|
				if ((cd_lused[CD_TANGENT] & (1 << i_src)) == 0) {
 | 
						|
					i_dst--;
 | 
						|
				}
 | 
						|
				else {
 | 
						|
					BLI_strncpy(
 | 
						|
					        tangent_names[i_dst],
 | 
						|
					        CustomData_get_layer_name(cd_ldata, CD_MLOOPUV, i_src), MAX_NAME);
 | 
						|
				}
 | 
						|
			}
 | 
						|
 | 
						|
			/* If tangent from orco is requested, decrement tangent_len */
 | 
						|
			int actual_tangent_len = (cd_lused[CD_TANGENT] & DM_TANGENT_MASK_ORCO) ?
 | 
						|
			        rdata->cd.layers.tangent_len - 1 : rdata->cd.layers.tangent_len;
 | 
						|
			if (rdata->edit_bmesh) {
 | 
						|
				BMEditMesh *em = rdata->edit_bmesh;
 | 
						|
				BMesh *bm = em->bm;
 | 
						|
 | 
						|
				if (is_auto_smooth && rdata->loop_normals == NULL) {
 | 
						|
					/* Should we store the previous array of `loop_normals` in somewhere? */
 | 
						|
					rdata->loop_len = bm->totloop;
 | 
						|
					rdata->loop_normals = MEM_mallocN(sizeof(*rdata->loop_normals) * rdata->loop_len, __func__);
 | 
						|
					BM_loops_calc_normal_vcos(bm, NULL, NULL, NULL, true, split_angle, rdata->loop_normals, NULL, NULL, -1, false);
 | 
						|
				}
 | 
						|
 | 
						|
				bool calc_active_tangent = false;
 | 
						|
 | 
						|
				BKE_editmesh_loop_tangent_calc(
 | 
						|
				        em, calc_active_tangent,
 | 
						|
				        tangent_names, actual_tangent_len,
 | 
						|
				        rdata->poly_normals, rdata->loop_normals,
 | 
						|
				        rdata->orco,
 | 
						|
				        &rdata->cd.output.ldata, bm->totloop,
 | 
						|
				        &rdata->cd.output.tangent_mask);
 | 
						|
			}
 | 
						|
			else {
 | 
						|
#undef me
 | 
						|
 | 
						|
				if (is_auto_smooth && rdata->loop_normals == NULL) {
 | 
						|
					/* Should we store the previous array of `loop_normals` in CustomData? */
 | 
						|
					mesh_render_calc_normals_loop_and_poly(me, split_angle, rdata);
 | 
						|
				}
 | 
						|
 | 
						|
				bool calc_active_tangent = false;
 | 
						|
 | 
						|
				BKE_mesh_calc_loop_tangent_ex(
 | 
						|
				        me->mvert,
 | 
						|
				        me->mpoly, me->totpoly,
 | 
						|
				        me->mloop,
 | 
						|
				        rdata->mlooptri, rdata->tri_len,
 | 
						|
				        cd_ldata,
 | 
						|
				        calc_active_tangent,
 | 
						|
				        tangent_names, actual_tangent_len,
 | 
						|
				        rdata->poly_normals, rdata->loop_normals,
 | 
						|
				        rdata->orco,
 | 
						|
				        &rdata->cd.output.ldata, me->totloop,
 | 
						|
				        &rdata->cd.output.tangent_mask);
 | 
						|
 | 
						|
					/* If we store tangents in the mesh, set temporary. */
 | 
						|
#if 0
 | 
						|
				CustomData_set_layer_flag(cd_ldata, CD_TANGENT, CD_FLAG_TEMPORARY);
 | 
						|
#endif
 | 
						|
 | 
						|
#define me DONT_USE_THIS
 | 
						|
#ifdef  me /* quiet warning */
 | 
						|
#endif
 | 
						|
			}
 | 
						|
 | 
						|
			/* End tangent calculation */
 | 
						|
			/* -------------------------------------------------------------------- */
 | 
						|
 | 
						|
			BLI_assert(CustomData_number_of_layers(&rdata->cd.output.ldata, CD_TANGENT) == rdata->cd.layers.tangent_len);
 | 
						|
 | 
						|
			int i_dst = 0;
 | 
						|
			for (int i_src = 0; i_src < cd_layers_src.uv_len; i_src++, i_dst++) {
 | 
						|
				if ((cd_lused[CD_TANGENT] & (1 << i_src)) == 0) {
 | 
						|
					i_dst--;
 | 
						|
					if (rdata->cd.layers.tangent_active >= i_src) {
 | 
						|
						rdata->cd.layers.tangent_active--;
 | 
						|
					}
 | 
						|
				}
 | 
						|
				else {
 | 
						|
					const char *name = CustomData_get_layer_name(cd_ldata, CD_MLOOPUV, i_src);
 | 
						|
					uint hash = BLI_ghashutil_strhash_p(name);
 | 
						|
 | 
						|
					BLI_snprintf(rdata->cd.uuid.tangent[i_dst], sizeof(*rdata->cd.uuid.tangent), "t%u", hash);
 | 
						|
 | 
						|
					/* Done adding tangents. */
 | 
						|
 | 
						|
					/* note: BKE_editmesh_loop_tangent_calc calculates 'CD_TANGENT',
 | 
						|
					 * not 'CD_MLOOPTANGENT' (as done below). It's OK, they're compatible. */
 | 
						|
 | 
						|
					/* note: normally we'd use 'i_src' here, but 'i_dst' is in sync with 'rdata->cd.output' */
 | 
						|
					rdata->cd.layers.tangent[i_dst] = CustomData_get_layer_n(&rdata->cd.output.ldata, CD_TANGENT, i_dst);
 | 
						|
					if (rdata->tri_len != 0) {
 | 
						|
						BLI_assert(rdata->cd.layers.tangent[i_dst] != NULL);
 | 
						|
					}
 | 
						|
				}
 | 
						|
			}
 | 
						|
			if (cd_lused[CD_TANGENT] & DM_TANGENT_MASK_ORCO) {
 | 
						|
				const char *name = CustomData_get_layer_name(&rdata->cd.output.ldata, CD_TANGENT, i_dst);
 | 
						|
				uint hash = BLI_ghashutil_strhash_p(name);
 | 
						|
				BLI_snprintf(rdata->cd.uuid.tangent[i_dst], sizeof(*rdata->cd.uuid.tangent), "t%u", hash);
 | 
						|
 | 
						|
				rdata->cd.layers.tangent[i_dst] = CustomData_get_layer_n(&rdata->cd.output.ldata, CD_TANGENT, i_dst);
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
#undef me
 | 
						|
	}
 | 
						|
 | 
						|
	return rdata;
 | 
						|
}
 | 
						|
 | 
						|
static void mesh_render_data_free(MeshRenderData *rdata)
 | 
						|
{
 | 
						|
	if (rdata->is_orco_allocated) {
 | 
						|
		MEM_SAFE_FREE(rdata->orco);
 | 
						|
	}
 | 
						|
	MEM_SAFE_FREE(rdata->cd.offset.uv);
 | 
						|
	MEM_SAFE_FREE(rdata->cd.offset.vcol);
 | 
						|
	MEM_SAFE_FREE(rdata->cd.uuid.auto_mix);
 | 
						|
	MEM_SAFE_FREE(rdata->cd.uuid.uv);
 | 
						|
	MEM_SAFE_FREE(rdata->cd.uuid.vcol);
 | 
						|
	MEM_SAFE_FREE(rdata->cd.uuid.tangent);
 | 
						|
	MEM_SAFE_FREE(rdata->cd.layers.uv);
 | 
						|
	MEM_SAFE_FREE(rdata->cd.layers.vcol);
 | 
						|
	MEM_SAFE_FREE(rdata->cd.layers.tangent);
 | 
						|
	MEM_SAFE_FREE(rdata->cd.layers.auto_vcol);
 | 
						|
	MEM_SAFE_FREE(rdata->loose_verts);
 | 
						|
	MEM_SAFE_FREE(rdata->loose_edges);
 | 
						|
	MEM_SAFE_FREE(rdata->edges_adjacent_polys);
 | 
						|
	MEM_SAFE_FREE(rdata->mlooptri);
 | 
						|
	MEM_SAFE_FREE(rdata->loop_normals);
 | 
						|
	MEM_SAFE_FREE(rdata->poly_normals);
 | 
						|
	MEM_SAFE_FREE(rdata->poly_normals_pack);
 | 
						|
	MEM_SAFE_FREE(rdata->vert_normals_pack);
 | 
						|
	MEM_SAFE_FREE(rdata->vert_weight_color);
 | 
						|
	MEM_SAFE_FREE(rdata->edge_select_bool);
 | 
						|
	MEM_SAFE_FREE(rdata->vert_color);
 | 
						|
 | 
						|
	CustomData_free(&rdata->cd.output.ldata, rdata->loop_len);
 | 
						|
 | 
						|
	MEM_freeN(rdata);
 | 
						|
}
 | 
						|
 | 
						|
static MeshRenderData *mesh_render_data_create(Mesh *me, const int types)
 | 
						|
{
 | 
						|
	return mesh_render_data_create_ex(me, types, NULL, 0);
 | 
						|
}
 | 
						|
 | 
						|
/** \} */
 | 
						|
 | 
						|
 | 
						|
/* ---------------------------------------------------------------------- */
 | 
						|
 | 
						|
/** \name Accessor Functions
 | 
						|
 * \{ */
 | 
						|
 | 
						|
static const char *mesh_render_data_uv_auto_layer_uuid_get(const MeshRenderData *rdata, int layer)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & MR_DATATYPE_SHADING);
 | 
						|
	return rdata->cd.uuid.auto_mix[layer];
 | 
						|
}
 | 
						|
 | 
						|
static const char *mesh_render_data_vcol_auto_layer_uuid_get(const MeshRenderData *rdata, int layer)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & MR_DATATYPE_SHADING);
 | 
						|
	return rdata->cd.uuid.auto_mix[rdata->cd.layers.uv_len + layer];
 | 
						|
}
 | 
						|
 | 
						|
static const char *mesh_render_data_uv_layer_uuid_get(const MeshRenderData *rdata, int layer)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & MR_DATATYPE_SHADING);
 | 
						|
	return rdata->cd.uuid.uv[layer];
 | 
						|
}
 | 
						|
 | 
						|
static const char *mesh_render_data_vcol_layer_uuid_get(const MeshRenderData *rdata, int layer)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & MR_DATATYPE_SHADING);
 | 
						|
	return rdata->cd.uuid.vcol[layer];
 | 
						|
}
 | 
						|
 | 
						|
static const char *mesh_render_data_tangent_layer_uuid_get(const MeshRenderData *rdata, int layer)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & MR_DATATYPE_SHADING);
 | 
						|
	return rdata->cd.uuid.tangent[layer];
 | 
						|
}
 | 
						|
 | 
						|
static int mesh_render_data_verts_len_get(const MeshRenderData *rdata)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & MR_DATATYPE_VERT);
 | 
						|
	return rdata->vert_len;
 | 
						|
}
 | 
						|
 | 
						|
static int mesh_render_data_loose_verts_len_get(const MeshRenderData *rdata)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & MR_DATATYPE_OVERLAY);
 | 
						|
	return rdata->loose_vert_len;
 | 
						|
}
 | 
						|
 | 
						|
static int mesh_render_data_edges_len_get(const MeshRenderData *rdata)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & MR_DATATYPE_EDGE);
 | 
						|
	return rdata->edge_len;
 | 
						|
}
 | 
						|
 | 
						|
static int mesh_render_data_loose_edges_len_get(const MeshRenderData *rdata)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & MR_DATATYPE_OVERLAY);
 | 
						|
	return rdata->loose_edge_len;
 | 
						|
}
 | 
						|
 | 
						|
static int mesh_render_data_looptri_len_get(const MeshRenderData *rdata)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & MR_DATATYPE_LOOPTRI);
 | 
						|
	return rdata->tri_len;
 | 
						|
}
 | 
						|
 | 
						|
static int mesh_render_data_mat_len_get(const MeshRenderData *rdata)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & MR_DATATYPE_POLY);
 | 
						|
	return rdata->mat_len;
 | 
						|
}
 | 
						|
 | 
						|
static int UNUSED_FUNCTION(mesh_render_data_loops_len_get)(const MeshRenderData *rdata)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & MR_DATATYPE_LOOP);
 | 
						|
	return rdata->loop_len;
 | 
						|
}
 | 
						|
 | 
						|
static int mesh_render_data_polys_len_get(const MeshRenderData *rdata)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & MR_DATATYPE_POLY);
 | 
						|
	return rdata->poly_len;
 | 
						|
}
 | 
						|
 | 
						|
/** \} */
 | 
						|
 | 
						|
 | 
						|
/* ---------------------------------------------------------------------- */
 | 
						|
 | 
						|
/** \name Internal Cache (Lazy Initialization)
 | 
						|
 * \{ */
 | 
						|
 | 
						|
/** Ensure #MeshRenderData.poly_normals_pack */
 | 
						|
static void mesh_render_data_ensure_poly_normals_pack(MeshRenderData *rdata)
 | 
						|
{
 | 
						|
	GPUPackedNormal *pnors_pack = rdata->poly_normals_pack;
 | 
						|
	if (pnors_pack == NULL) {
 | 
						|
		if (rdata->edit_bmesh) {
 | 
						|
			BMesh *bm = rdata->edit_bmesh->bm;
 | 
						|
			BMIter fiter;
 | 
						|
			BMFace *efa;
 | 
						|
			int i;
 | 
						|
 | 
						|
			pnors_pack = rdata->poly_normals_pack = MEM_mallocN(sizeof(*pnors_pack) * rdata->poly_len, __func__);
 | 
						|
			BM_ITER_MESH_INDEX(efa, &fiter, bm, BM_FACES_OF_MESH, i) {
 | 
						|
				pnors_pack[i] = GPU_normal_convert_i10_v3(efa->no);
 | 
						|
			}
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			float (*pnors)[3] = rdata->poly_normals;
 | 
						|
 | 
						|
			if (!pnors) {
 | 
						|
				pnors = rdata->poly_normals = MEM_mallocN(sizeof(*pnors) * rdata->poly_len, __func__);
 | 
						|
				BKE_mesh_calc_normals_poly(
 | 
						|
				        rdata->mvert, NULL, rdata->vert_len,
 | 
						|
				        rdata->mloop, rdata->mpoly, rdata->loop_len, rdata->poly_len, pnors, true);
 | 
						|
			}
 | 
						|
 | 
						|
			pnors_pack = rdata->poly_normals_pack = MEM_mallocN(sizeof(*pnors_pack) * rdata->poly_len, __func__);
 | 
						|
			for (int i = 0; i < rdata->poly_len; i++) {
 | 
						|
				pnors_pack[i] = GPU_normal_convert_i10_v3(pnors[i]);
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/** Ensure #MeshRenderData.vert_normals_pack */
 | 
						|
static void mesh_render_data_ensure_vert_normals_pack(MeshRenderData *rdata)
 | 
						|
{
 | 
						|
	GPUPackedNormal *vnors_pack = rdata->vert_normals_pack;
 | 
						|
	if (vnors_pack == NULL) {
 | 
						|
		if (rdata->edit_bmesh) {
 | 
						|
			BMesh *bm = rdata->edit_bmesh->bm;
 | 
						|
			BMIter viter;
 | 
						|
			BMVert *eve;
 | 
						|
			int i;
 | 
						|
 | 
						|
			vnors_pack = rdata->vert_normals_pack = MEM_mallocN(sizeof(*vnors_pack) * rdata->vert_len, __func__);
 | 
						|
			BM_ITER_MESH_INDEX(eve, &viter, bm, BM_VERT, i) {
 | 
						|
				vnors_pack[i] = GPU_normal_convert_i10_v3(eve->no);
 | 
						|
			}
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			/* data from mesh used directly */
 | 
						|
			BLI_assert(0);
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
/** Ensure #MeshRenderData.vert_color */
 | 
						|
static void mesh_render_data_ensure_vert_color(MeshRenderData *rdata)
 | 
						|
{
 | 
						|
	char (*vcol)[3] = rdata->vert_color;
 | 
						|
	if (vcol == NULL) {
 | 
						|
		if (rdata->edit_bmesh) {
 | 
						|
			BMesh *bm = rdata->edit_bmesh->bm;
 | 
						|
			const int cd_loop_color_offset = CustomData_get_offset(&bm->ldata, CD_MLOOPCOL);
 | 
						|
			if (cd_loop_color_offset == -1) {
 | 
						|
				goto fallback;
 | 
						|
			}
 | 
						|
 | 
						|
			vcol = rdata->vert_color = MEM_mallocN(sizeof(*vcol) * rdata->loop_len, __func__);
 | 
						|
 | 
						|
			BMIter fiter;
 | 
						|
			BMFace *efa;
 | 
						|
			int i = 0;
 | 
						|
 | 
						|
			BM_ITER_MESH(efa, &fiter, bm, BM_FACES_OF_MESH) {
 | 
						|
				BMLoop *l_iter, *l_first;
 | 
						|
				l_iter = l_first = BM_FACE_FIRST_LOOP(efa);
 | 
						|
				do {
 | 
						|
					const MLoopCol *lcol = BM_ELEM_CD_GET_VOID_P(l_iter, cd_loop_color_offset);
 | 
						|
					vcol[i][0] = lcol->r;
 | 
						|
					vcol[i][1] = lcol->g;
 | 
						|
					vcol[i][2] = lcol->b;
 | 
						|
					i += 1;
 | 
						|
				} while ((l_iter = l_iter->next) != l_first);
 | 
						|
			}
 | 
						|
			BLI_assert(i == rdata->loop_len);
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			if (rdata->mloopcol == NULL) {
 | 
						|
				goto fallback;
 | 
						|
			}
 | 
						|
 | 
						|
			vcol = rdata->vert_color = MEM_mallocN(sizeof(*vcol) * rdata->loop_len, __func__);
 | 
						|
 | 
						|
			for (int i = 0; i < rdata->loop_len; i++) {
 | 
						|
				vcol[i][0] = rdata->mloopcol[i].r;
 | 
						|
				vcol[i][1] = rdata->mloopcol[i].g;
 | 
						|
				vcol[i][2] = rdata->mloopcol[i].b;
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
	return;
 | 
						|
 | 
						|
fallback:
 | 
						|
	vcol = rdata->vert_color = MEM_mallocN(sizeof(*vcol) * rdata->loop_len, __func__);
 | 
						|
 | 
						|
	for (int i = 0; i < rdata->loop_len; i++) {
 | 
						|
		vcol[i][0] = 255;
 | 
						|
		vcol[i][1] = 255;
 | 
						|
		vcol[i][2] = 255;
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/* TODO, move into shader? */
 | 
						|
static void rgb_from_weight(float r_rgb[3], const float weight)
 | 
						|
{
 | 
						|
	const float blend = ((weight / 2.0f) + 0.5f);
 | 
						|
 | 
						|
	if (weight <= 0.25f) {    /* blue->cyan */
 | 
						|
		r_rgb[0] = 0.0f;
 | 
						|
		r_rgb[1] = blend * weight * 4.0f;
 | 
						|
		r_rgb[2] = blend;
 | 
						|
	}
 | 
						|
	else if (weight <= 0.50f) {  /* cyan->green */
 | 
						|
		r_rgb[0] = 0.0f;
 | 
						|
		r_rgb[1] = blend;
 | 
						|
		r_rgb[2] = blend * (1.0f - ((weight - 0.25f) * 4.0f));
 | 
						|
	}
 | 
						|
	else if (weight <= 0.75f) {  /* green->yellow */
 | 
						|
		r_rgb[0] = blend * ((weight - 0.50f) * 4.0f);
 | 
						|
		r_rgb[1] = blend;
 | 
						|
		r_rgb[2] = 0.0f;
 | 
						|
	}
 | 
						|
	else if (weight <= 1.0f) {  /* yellow->red */
 | 
						|
		r_rgb[0] = blend;
 | 
						|
		r_rgb[1] = blend * (1.0f - ((weight - 0.75f) * 4.0f));
 | 
						|
		r_rgb[2] = 0.0f;
 | 
						|
	}
 | 
						|
	else {
 | 
						|
		/* exceptional value, unclamped or nan,
 | 
						|
		 * avoid uninitialized memory use */
 | 
						|
		r_rgb[0] = 1.0f;
 | 
						|
		r_rgb[1] = 0.0f;
 | 
						|
		r_rgb[2] = 1.0f;
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static void vertex_weight_color(float vweight[3], float weight, bool show_alert_color)
 | 
						|
{
 | 
						|
	CLAMP(weight, 0.0f, 1.0f);
 | 
						|
 | 
						|
	if (show_alert_color) {
 | 
						|
		bTheme *theme = U.themes.first;
 | 
						|
 | 
						|
		rgb_uchar_to_float(vweight, (uchar *)theme->tv3d.vertex_unreferenced);
 | 
						|
	}
 | 
						|
	else if (U.flag & USER_CUSTOM_RANGE) {
 | 
						|
		BKE_colorband_evaluate(&U.coba_weight, weight, vweight);
 | 
						|
	}
 | 
						|
	else {
 | 
						|
		rgb_from_weight(vweight, weight);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static void evaluate_vertex_weight(float vweight[3], const MDeformVert *dvert, const struct DRW_MeshWeightState *wstate)
 | 
						|
{
 | 
						|
	float input = 0.0f;
 | 
						|
	bool show_alert_color = false;
 | 
						|
 | 
						|
	if (wstate->flags & DRW_MESH_WEIGHT_STATE_MULTIPAINT) {
 | 
						|
		/* Multi-Paint feature */
 | 
						|
		input = BKE_defvert_multipaint_collective_weight(
 | 
						|
		        dvert, wstate->defgroup_len, wstate->defgroup_sel, wstate->defgroup_sel_count,
 | 
						|
		        (wstate->flags & DRW_MESH_WEIGHT_STATE_AUTO_NORMALIZE) != 0);
 | 
						|
 | 
						|
		/* make it black if the selected groups have no weight on a vertex */
 | 
						|
		if (input == 0.0f) {
 | 
						|
			show_alert_color = true;
 | 
						|
		}
 | 
						|
	}
 | 
						|
	else {
 | 
						|
		/* default, non tricky behavior */
 | 
						|
		input = defvert_find_weight(dvert, wstate->defgroup_active);
 | 
						|
 | 
						|
		if (input == 0.0f) {
 | 
						|
			switch (wstate->alert_mode) {
 | 
						|
				case OB_DRAW_GROUPUSER_ACTIVE:
 | 
						|
					show_alert_color = true;
 | 
						|
					break;
 | 
						|
 | 
						|
				case OB_DRAW_GROUPUSER_ALL:
 | 
						|
					show_alert_color = defvert_is_weight_zero(dvert, wstate->defgroup_len);
 | 
						|
					break;
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	vertex_weight_color(vweight, input, show_alert_color);
 | 
						|
}
 | 
						|
 | 
						|
/* color-code for missing data (full brightness isn't easy on the eye). */
 | 
						|
static const unsigned char missing_weight_color[3] = { 0xa0, 0x00, 0xa0 };
 | 
						|
 | 
						|
/** Ensure #MeshRenderData.vert_weight_color */
 | 
						|
static void mesh_render_data_ensure_vert_weight_color(MeshRenderData *rdata, const struct DRW_MeshWeightState *wstate)
 | 
						|
{
 | 
						|
	float (*vweight)[3] = rdata->vert_weight_color;
 | 
						|
	if (vweight == NULL) {
 | 
						|
		if (wstate->defgroup_active == -1) {
 | 
						|
			goto fallback;
 | 
						|
		}
 | 
						|
 | 
						|
		if (rdata->edit_bmesh) {
 | 
						|
			BMesh *bm = rdata->edit_bmesh->bm;
 | 
						|
			const int cd_dvert_offset = CustomData_get_offset(&bm->vdata, CD_MDEFORMVERT);
 | 
						|
			if (cd_dvert_offset == -1) {
 | 
						|
				goto fallback;
 | 
						|
			}
 | 
						|
 | 
						|
			BMIter viter;
 | 
						|
			BMVert *eve;
 | 
						|
			int i;
 | 
						|
 | 
						|
			vweight = rdata->vert_weight_color = MEM_mallocN(sizeof(*vweight) * rdata->vert_len, __func__);
 | 
						|
			BM_ITER_MESH_INDEX(eve, &viter, bm, BM_VERT, i) {
 | 
						|
				const MDeformVert *dvert = BM_ELEM_CD_GET_VOID_P(eve, cd_dvert_offset);
 | 
						|
				evaluate_vertex_weight(vweight[i], dvert, wstate);
 | 
						|
			}
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			if (rdata->dvert == NULL) {
 | 
						|
				goto fallback;
 | 
						|
			}
 | 
						|
 | 
						|
			vweight = rdata->vert_weight_color = MEM_mallocN(sizeof(*vweight) * rdata->vert_len, __func__);
 | 
						|
			for (int i = 0; i < rdata->vert_len; i++) {
 | 
						|
				evaluate_vertex_weight(vweight[i], &rdata->dvert[i], wstate);
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
	return;
 | 
						|
 | 
						|
fallback:
 | 
						|
	vweight = rdata->vert_weight_color = MEM_mallocN(sizeof(*vweight) * rdata->vert_len, __func__);
 | 
						|
 | 
						|
	float error_color[3];
 | 
						|
 | 
						|
	if ((wstate->defgroup_active < 0) && (wstate->defgroup_len > 0)) {
 | 
						|
		rgb_uchar_to_float(error_color, missing_weight_color);
 | 
						|
	}
 | 
						|
	else {
 | 
						|
		vertex_weight_color(error_color, 0.0f, wstate->alert_mode != OB_DRAW_GROUPUSER_NONE);
 | 
						|
	}
 | 
						|
 | 
						|
	for (int i = 0; i < rdata->vert_len; i++) {
 | 
						|
		copy_v3_v3(vweight[i], error_color);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/** Ensure #MeshRenderData.edge_select_bool */
 | 
						|
static void mesh_render_data_ensure_edge_select_bool(MeshRenderData *rdata, bool use_wire)
 | 
						|
{
 | 
						|
	bool *edge_select_bool = rdata->edge_select_bool;
 | 
						|
	if (edge_select_bool == NULL) {
 | 
						|
		edge_select_bool = rdata->edge_select_bool =
 | 
						|
		        MEM_callocN(sizeof(*edge_select_bool) * rdata->edge_len, __func__);
 | 
						|
 | 
						|
		for (int i = 0; i < rdata->poly_len; i++) {
 | 
						|
			MPoly *poly = &rdata->mpoly[i];
 | 
						|
 | 
						|
			if (poly->flag & ME_FACE_SEL) {
 | 
						|
				for (int j = 0; j < poly->totloop; j++) {
 | 
						|
					MLoop *loop = &rdata->mloop[poly->loopstart + j];
 | 
						|
					if (use_wire) {
 | 
						|
						edge_select_bool[loop->e] = true;
 | 
						|
					}
 | 
						|
					else {
 | 
						|
						/* Not totally correct, will cause problems for edges with 3x faces. */
 | 
						|
						edge_select_bool[loop->e] = !edge_select_bool[loop->e];
 | 
						|
					}
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/** \} */
 | 
						|
 | 
						|
/* ---------------------------------------------------------------------- */
 | 
						|
 | 
						|
/** \name Internal Cache Generation
 | 
						|
 * \{ */
 | 
						|
 | 
						|
static bool mesh_render_data_pnors_pcenter_select_get(
 | 
						|
        MeshRenderData *rdata, const int poly,
 | 
						|
        float r_pnors[3], float r_center[3], bool *r_selected)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & (MR_DATATYPE_VERT | MR_DATATYPE_LOOP | MR_DATATYPE_POLY));
 | 
						|
 | 
						|
	if (rdata->edit_bmesh) {
 | 
						|
		const BMFace *efa = BM_face_at_index(rdata->edit_bmesh->bm, poly);
 | 
						|
		if (BM_elem_flag_test(efa, BM_ELEM_HIDDEN)) {
 | 
						|
			return false;
 | 
						|
		}
 | 
						|
		BM_face_calc_center_mean(efa, r_center);
 | 
						|
		copy_v3_v3(r_pnors, efa->no);
 | 
						|
		*r_selected = (BM_elem_flag_test(efa, BM_ELEM_SELECT) != 0) ? true : false;
 | 
						|
	}
 | 
						|
	else {
 | 
						|
		MVert *mvert = rdata->mvert;
 | 
						|
		const MPoly *mpoly = rdata->mpoly + poly;
 | 
						|
		const MLoop *mloop = rdata->mloop + mpoly->loopstart;
 | 
						|
 | 
						|
		BKE_mesh_calc_poly_center(mpoly, mloop, mvert, r_center);
 | 
						|
		BKE_mesh_calc_poly_normal(mpoly, mloop, mvert, r_pnors);
 | 
						|
 | 
						|
		*r_selected = false; /* No selection if not in edit mode */
 | 
						|
	}
 | 
						|
 | 
						|
	return true;
 | 
						|
}
 | 
						|
 | 
						|
static bool mesh_render_data_edge_vcos_manifold_pnors(
 | 
						|
        MeshRenderData *rdata, const int edge_index,
 | 
						|
        float **r_vco1, float **r_vco2, float **r_pnor1, float **r_pnor2, bool *r_is_manifold)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & (MR_DATATYPE_VERT | MR_DATATYPE_EDGE | MR_DATATYPE_LOOP | MR_DATATYPE_POLY));
 | 
						|
 | 
						|
	if (rdata->edit_bmesh) {
 | 
						|
		BMesh *bm = rdata->edit_bmesh->bm;
 | 
						|
		BMEdge *eed = BM_edge_at_index(bm, edge_index);
 | 
						|
		if (BM_elem_flag_test(eed, BM_ELEM_HIDDEN)) {
 | 
						|
			return false;
 | 
						|
		}
 | 
						|
		*r_vco1 = eed->v1->co;
 | 
						|
		*r_vco2 = eed->v2->co;
 | 
						|
		if (BM_edge_is_manifold(eed)) {
 | 
						|
			*r_pnor1 = eed->l->f->no;
 | 
						|
			*r_pnor2 = eed->l->radial_next->f->no;
 | 
						|
			*r_is_manifold = true;
 | 
						|
		}
 | 
						|
		else if (eed->l != NULL) {
 | 
						|
			*r_pnor1 = eed->l->f->no;
 | 
						|
			*r_pnor2 = eed->l->f->no;
 | 
						|
			*r_is_manifold = false;
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			*r_pnor1 = eed->v1->no;
 | 
						|
			*r_pnor2 = eed->v1->no;
 | 
						|
			*r_is_manifold = false;
 | 
						|
		}
 | 
						|
	}
 | 
						|
	else {
 | 
						|
		MVert *mvert = rdata->mvert;
 | 
						|
		MEdge *medge = rdata->medge;
 | 
						|
		EdgeAdjacentPolys *eap = rdata->edges_adjacent_polys;
 | 
						|
		float (*pnors)[3] = rdata->poly_normals;
 | 
						|
 | 
						|
		if (!eap) {
 | 
						|
			const MLoop *mloop = rdata->mloop;
 | 
						|
			const MPoly *mpoly = rdata->mpoly;
 | 
						|
			const int poly_len = rdata->poly_len;
 | 
						|
			const bool do_pnors = (poly_len != 0 && pnors == NULL);
 | 
						|
 | 
						|
			eap = rdata->edges_adjacent_polys = MEM_mallocN(sizeof(*eap) * rdata->edge_len, __func__);
 | 
						|
			for (int i = 0; i < rdata->edge_len; i++) {
 | 
						|
				eap[i].count = 0;
 | 
						|
				eap[i].face_index[0] = -1;
 | 
						|
				eap[i].face_index[1] = -1;
 | 
						|
			}
 | 
						|
			if (do_pnors) {
 | 
						|
				pnors = rdata->poly_normals = MEM_mallocN(sizeof(*pnors) * poly_len, __func__);
 | 
						|
			}
 | 
						|
 | 
						|
			for (int i = 0; i < poly_len; i++, mpoly++) {
 | 
						|
				if (do_pnors) {
 | 
						|
					BKE_mesh_calc_poly_normal(mpoly, mloop + mpoly->loopstart, mvert, pnors[i]);
 | 
						|
				}
 | 
						|
 | 
						|
				const int loopend = mpoly->loopstart + mpoly->totloop;
 | 
						|
				for (int j = mpoly->loopstart; j < loopend; j++) {
 | 
						|
					const int edge_idx = mloop[j].e;
 | 
						|
					if (eap[edge_idx].count < 2) {
 | 
						|
						eap[edge_idx].face_index[eap[edge_idx].count] = i;
 | 
						|
					}
 | 
						|
					eap[edge_idx].count++;
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		BLI_assert(eap && (rdata->poly_len == 0 || pnors != NULL));
 | 
						|
 | 
						|
		*r_vco1 = mvert[medge[edge_index].v1].co;
 | 
						|
		*r_vco2 = mvert[medge[edge_index].v2].co;
 | 
						|
		if (eap[edge_index].face_index[0] == -1) {
 | 
						|
			/* Edge has no poly... */
 | 
						|
			*r_pnor1 = *r_pnor2 = mvert[medge[edge_index].v1].co; /* XXX mvert.no are shorts... :( */
 | 
						|
			*r_is_manifold = false;
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			*r_pnor1 = pnors[eap[edge_index].face_index[0]];
 | 
						|
 | 
						|
			float nor[3], v1[3], v2[3], r_center[3];
 | 
						|
			const MPoly *mpoly = rdata->mpoly + eap[edge_index].face_index[0];
 | 
						|
			const MLoop *mloop = rdata->mloop + mpoly->loopstart;
 | 
						|
 | 
						|
			BKE_mesh_calc_poly_center(mpoly, mloop, mvert, r_center);
 | 
						|
			sub_v3_v3v3(v1, *r_vco2, *r_vco1);
 | 
						|
			sub_v3_v3v3(v2, r_center, *r_vco1);
 | 
						|
			cross_v3_v3v3(nor, v1, v2);
 | 
						|
 | 
						|
			if (dot_v3v3(nor, *r_pnor1) < 0.0) {
 | 
						|
				SWAP(float *, *r_vco1, *r_vco2);
 | 
						|
			}
 | 
						|
 | 
						|
			if (eap[edge_index].count == 2) {
 | 
						|
				BLI_assert(eap[edge_index].face_index[1] >= 0);
 | 
						|
				*r_pnor2 = pnors[eap[edge_index].face_index[1]];
 | 
						|
				*r_is_manifold = true;
 | 
						|
			}
 | 
						|
			else {
 | 
						|
				*r_pnor2 = pnors[eap[edge_index].face_index[0]];
 | 
						|
				*r_is_manifold = false;
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	return true;
 | 
						|
}
 | 
						|
 | 
						|
static uchar mesh_render_data_looptri_flag(MeshRenderData *rdata, const BMFace *efa)
 | 
						|
{
 | 
						|
	uchar fflag = 0;
 | 
						|
 | 
						|
	if (efa == rdata->efa_act)
 | 
						|
		fflag |= VFLAG_FACE_ACTIVE;
 | 
						|
 | 
						|
	if (BM_elem_flag_test(efa, BM_ELEM_SELECT))
 | 
						|
		fflag |= VFLAG_FACE_SELECTED;
 | 
						|
 | 
						|
#ifdef WITH_FREESTYLE
 | 
						|
	BMesh *bm = rdata->edit_bmesh->bm;
 | 
						|
	if (CustomData_has_layer(&bm->pdata, CD_FREESTYLE_FACE)) {
 | 
						|
		FreestyleFace *ffa = CustomData_bmesh_get(&bm->pdata, efa->head.data, CD_FREESTYLE_FACE);
 | 
						|
 | 
						|
		if (ffa->flag & FREESTYLE_FACE_MARK)
 | 
						|
			fflag |= VFLAG_FACE_FREESTYLE;
 | 
						|
	}
 | 
						|
#endif
 | 
						|
 | 
						|
	return fflag;
 | 
						|
}
 | 
						|
 | 
						|
static void mesh_render_data_edge_flag(
 | 
						|
        const MeshRenderData *rdata, const BMEdge *eed,
 | 
						|
        EdgeDrawAttr *eattr)
 | 
						|
{
 | 
						|
	eattr->e_flag |= VFLAG_EDGE_EXISTS;
 | 
						|
 | 
						|
	if (eed == rdata->eed_act)
 | 
						|
		eattr->e_flag |= VFLAG_EDGE_ACTIVE;
 | 
						|
 | 
						|
	if (BM_elem_flag_test(eed, BM_ELEM_SELECT))
 | 
						|
		eattr->e_flag |= VFLAG_EDGE_SELECTED;
 | 
						|
 | 
						|
	if (BM_elem_flag_test(eed, BM_ELEM_SEAM))
 | 
						|
		eattr->e_flag |= VFLAG_EDGE_SEAM;
 | 
						|
 | 
						|
	if (!BM_elem_flag_test(eed, BM_ELEM_SMOOTH))
 | 
						|
		eattr->e_flag |= VFLAG_EDGE_SHARP;
 | 
						|
 | 
						|
#ifdef WITH_FREESTYLE
 | 
						|
	BMesh *bm = rdata->edit_bmesh->bm;
 | 
						|
	if (CustomData_has_layer(&bm->edata, CD_FREESTYLE_EDGE)) {
 | 
						|
		FreestyleEdge *fed = CustomData_bmesh_get(&bm->edata, eed->head.data, CD_FREESTYLE_EDGE);
 | 
						|
 | 
						|
		if (fed->flag & FREESTYLE_EDGE_MARK)
 | 
						|
			eattr->e_flag |= VFLAG_EDGE_FREESTYLE;
 | 
						|
	}
 | 
						|
#endif
 | 
						|
 | 
						|
	/* Use a byte for value range */
 | 
						|
	if (rdata->cd.offset.crease != -1) {
 | 
						|
		float crease = BM_ELEM_CD_GET_FLOAT(eed, rdata->cd.offset.crease);
 | 
						|
		if (crease > 0) {
 | 
						|
			eattr->crease = (uchar)(crease * 255.0f);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	/* Use a byte for value range */
 | 
						|
	if (rdata->cd.offset.bweight != -1) {
 | 
						|
		float bweight = BM_ELEM_CD_GET_FLOAT(eed, rdata->cd.offset.bweight);
 | 
						|
		if (bweight > 0) {
 | 
						|
			eattr->bweight = (uchar)(bweight * 255.0f);
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static uchar mesh_render_data_vertex_flag(MeshRenderData *rdata, const BMVert *eve)
 | 
						|
{
 | 
						|
	uchar vflag = 0;
 | 
						|
 | 
						|
	/* Current vertex */
 | 
						|
	if (eve == rdata->eve_act)
 | 
						|
		vflag |= VFLAG_VERTEX_ACTIVE;
 | 
						|
 | 
						|
	if (BM_elem_flag_test(eve, BM_ELEM_SELECT))
 | 
						|
		vflag |= VFLAG_VERTEX_SELECTED;
 | 
						|
 | 
						|
	return vflag;
 | 
						|
}
 | 
						|
 | 
						|
static void add_overlay_tri(
 | 
						|
        MeshRenderData *rdata, GPUVertBuf *vbo_pos, GPUVertBuf *vbo_nor, GPUVertBuf *vbo_data,
 | 
						|
        const uint pos_id, const uint vnor_id, const uint lnor_id, const uint data_id,
 | 
						|
        const BMLoop **bm_looptri, const int base_vert_idx)
 | 
						|
{
 | 
						|
	uchar fflag;
 | 
						|
	uchar vflag;
 | 
						|
 | 
						|
	if (vbo_pos) {
 | 
						|
		/* TODO(sybren): deduplicate this and all the other places it's pasted to in this file. */
 | 
						|
		if (rdata->edit_data && rdata->edit_data->vertexCos) {
 | 
						|
			for (uint i = 0; i < 3; i++) {
 | 
						|
				int vidx = BM_elem_index_get(bm_looptri[i]->v);
 | 
						|
				const float *pos = rdata->edit_data->vertexCos[vidx];
 | 
						|
				GPU_vertbuf_attr_set(vbo_pos, pos_id, base_vert_idx + i, pos);
 | 
						|
			}
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			for (uint i = 0; i < 3; i++) {
 | 
						|
				const float *pos = bm_looptri[i]->v->co;
 | 
						|
				GPU_vertbuf_attr_set(vbo_pos, pos_id, base_vert_idx + i, pos);
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	if (vbo_nor) {
 | 
						|
		/* TODO real loop normal */
 | 
						|
		GPUPackedNormal lnor = GPU_normal_convert_i10_v3(bm_looptri[0]->f->no);
 | 
						|
		for (uint i = 0; i < 3; i++) {
 | 
						|
			GPUPackedNormal vnor = GPU_normal_convert_i10_v3(bm_looptri[i]->v->no);
 | 
						|
			GPU_vertbuf_attr_set(vbo_nor, vnor_id, base_vert_idx + i, &vnor);
 | 
						|
			GPU_vertbuf_attr_set(vbo_nor, lnor_id, base_vert_idx + i, &lnor);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	if (vbo_data) {
 | 
						|
		fflag = mesh_render_data_looptri_flag(rdata, bm_looptri[0]->f);
 | 
						|
		uint i_prev = 1, i = 2;
 | 
						|
		for (uint i_next = 0; i_next < 3; i_next++) {
 | 
						|
			vflag = mesh_render_data_vertex_flag(rdata, bm_looptri[i]->v);
 | 
						|
			EdgeDrawAttr eattr = {0};
 | 
						|
			if (bm_looptri[i_next] == bm_looptri[i_prev]->prev) {
 | 
						|
				mesh_render_data_edge_flag(rdata, bm_looptri[i_next]->e, &eattr);
 | 
						|
			}
 | 
						|
			eattr.v_flag = fflag | vflag;
 | 
						|
			GPU_vertbuf_attr_set(vbo_data, data_id, base_vert_idx + i, &eattr);
 | 
						|
 | 
						|
			i_prev = i;
 | 
						|
			i = i_next;
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static void add_overlay_loose_edge(
 | 
						|
        MeshRenderData *rdata, GPUVertBuf *vbo_pos, GPUVertBuf *vbo_nor, GPUVertBuf *vbo_data,
 | 
						|
        const uint pos_id, const uint vnor_id, const uint data_id,
 | 
						|
        const BMEdge *eed, const int base_vert_idx)
 | 
						|
{
 | 
						|
	if (vbo_pos) {
 | 
						|
		/* TODO(sybren): deduplicate this and all the other places it's pasted to in this file. */
 | 
						|
		if (rdata->edit_data && rdata->edit_data->vertexCos) {
 | 
						|
			for (uint i = 0; i < 2; i++) {
 | 
						|
				int vidx = BM_elem_index_get((&eed->v1)[i]);
 | 
						|
				const float *pos = rdata->edit_data->vertexCos[vidx];
 | 
						|
				GPU_vertbuf_attr_set(vbo_pos, pos_id, base_vert_idx + i, pos);
 | 
						|
			}
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			for (int i = 0; i < 2; ++i) {
 | 
						|
				const float *pos = (&eed->v1)[i]->co;
 | 
						|
				GPU_vertbuf_attr_set(vbo_pos, pos_id, base_vert_idx + i, pos);
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	if (vbo_nor) {
 | 
						|
		for (int i = 0; i < 2; ++i) {
 | 
						|
			GPUPackedNormal vnor = GPU_normal_convert_i10_v3((&eed->v1)[i]->no);
 | 
						|
			GPU_vertbuf_attr_set(vbo_nor, vnor_id, base_vert_idx + i, &vnor);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	if (vbo_data) {
 | 
						|
		EdgeDrawAttr eattr = {0};
 | 
						|
		mesh_render_data_edge_flag(rdata, eed, &eattr);
 | 
						|
		for (int i = 0; i < 2; ++i) {
 | 
						|
			eattr.v_flag = mesh_render_data_vertex_flag(rdata, (&eed->v1)[i]);
 | 
						|
			GPU_vertbuf_attr_set(vbo_data, data_id, base_vert_idx + i, &eattr);
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static void add_overlay_loose_vert(
 | 
						|
        MeshRenderData *rdata, GPUVertBuf *vbo_pos, GPUVertBuf *vbo_nor, GPUVertBuf *vbo_data,
 | 
						|
        const uint pos_id, const uint vnor_id, const uint data_id,
 | 
						|
        const BMVert *eve, const int base_vert_idx)
 | 
						|
{
 | 
						|
	if (vbo_pos) {
 | 
						|
		/* TODO(sybren): deduplicate this and all the other places it's pasted to in this file. */
 | 
						|
		if (rdata->edit_data && rdata->edit_data->vertexCos) {
 | 
						|
			int vidx = BM_elem_index_get(eve);
 | 
						|
			const float *pos = rdata->edit_data->vertexCos[vidx];
 | 
						|
			GPU_vertbuf_attr_set(vbo_pos, pos_id, base_vert_idx, pos);
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			const float *pos = eve->co;
 | 
						|
			GPU_vertbuf_attr_set(vbo_pos, pos_id, base_vert_idx, pos);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	if (vbo_nor) {
 | 
						|
		GPUPackedNormal vnor = GPU_normal_convert_i10_v3(eve->no);
 | 
						|
		GPU_vertbuf_attr_set(vbo_nor, vnor_id, base_vert_idx, &vnor);
 | 
						|
	}
 | 
						|
 | 
						|
	if (vbo_data) {
 | 
						|
		uchar vflag[4] = {0, 0, 0, 0};
 | 
						|
		vflag[0] = mesh_render_data_vertex_flag(rdata, eve);
 | 
						|
		GPU_vertbuf_attr_set(vbo_data, data_id, base_vert_idx, vflag);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/** \} */
 | 
						|
 | 
						|
/* ---------------------------------------------------------------------- */
 | 
						|
 | 
						|
/** \name Vertex Group Selection
 | 
						|
 * \{ */
 | 
						|
 | 
						|
/** Reset the selection structure, deallocating heap memory as appropriate. */
 | 
						|
void DRW_mesh_weight_state_clear(struct DRW_MeshWeightState *wstate)
 | 
						|
{
 | 
						|
	MEM_SAFE_FREE(wstate->defgroup_sel);
 | 
						|
 | 
						|
	memset(wstate, 0, sizeof(*wstate));
 | 
						|
 | 
						|
	wstate->defgroup_active = -1;
 | 
						|
}
 | 
						|
 | 
						|
/** Copy selection data from one structure to another, including heap memory. */
 | 
						|
void DRW_mesh_weight_state_copy(struct DRW_MeshWeightState *wstate_dst, const struct DRW_MeshWeightState *wstate_src)
 | 
						|
{
 | 
						|
	MEM_SAFE_FREE(wstate_dst->defgroup_sel);
 | 
						|
 | 
						|
	memcpy(wstate_dst, wstate_src, sizeof(*wstate_dst));
 | 
						|
 | 
						|
	if (wstate_src->defgroup_sel) {
 | 
						|
		wstate_dst->defgroup_sel = MEM_dupallocN(wstate_src->defgroup_sel);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/** Compare two selection structures. */
 | 
						|
bool DRW_mesh_weight_state_compare(const struct DRW_MeshWeightState *a, const struct DRW_MeshWeightState *b)
 | 
						|
{
 | 
						|
	return a->defgroup_active == b->defgroup_active &&
 | 
						|
	       a->defgroup_len == b->defgroup_len &&
 | 
						|
	       a->flags == b->flags &&
 | 
						|
	       a->alert_mode == b->alert_mode &&
 | 
						|
	       a->defgroup_sel_count == b->defgroup_sel_count &&
 | 
						|
	       ((!a->defgroup_sel && !b->defgroup_sel) ||
 | 
						|
	        (a->defgroup_sel && b->defgroup_sel &&
 | 
						|
	         memcmp(a->defgroup_sel, b->defgroup_sel, a->defgroup_len * sizeof(bool)) == 0));
 | 
						|
}
 | 
						|
 | 
						|
/** \} */
 | 
						|
 | 
						|
/* ---------------------------------------------------------------------- */
 | 
						|
 | 
						|
/** \name Mesh GPUBatch Cache
 | 
						|
 * \{ */
 | 
						|
 | 
						|
typedef struct MeshBatchCache {
 | 
						|
	GPUVertBuf *pos_in_order;
 | 
						|
	GPUIndexBuf *edges_in_order;
 | 
						|
	GPUIndexBuf *edges_adjacency; /* Store edges with adjacent vertices. */
 | 
						|
	GPUIndexBuf *triangles_in_order;
 | 
						|
	GPUIndexBuf *ledges_in_order;
 | 
						|
 | 
						|
	GPUTexture *pos_in_order_tx; /* Depending on pos_in_order */
 | 
						|
 | 
						|
	GPUBatch *all_verts;
 | 
						|
	GPUBatch *all_edges;
 | 
						|
	GPUBatch *all_triangles;
 | 
						|
 | 
						|
	GPUVertBuf *pos_with_normals;
 | 
						|
	GPUVertBuf *tri_aligned_uv;  /* Active UV layer (mloopuv) */
 | 
						|
 | 
						|
	/**
 | 
						|
	 * Other uses are all positions or loose elements.
 | 
						|
	 * This stores all visible elements, needed for selection.
 | 
						|
	 */
 | 
						|
	GPUVertBuf *ed_fcenter_pos_with_nor_and_sel;
 | 
						|
	GPUVertBuf *ed_edge_pos;
 | 
						|
	GPUVertBuf *ed_vert_pos;
 | 
						|
 | 
						|
	GPUBatch *triangles_with_normals;
 | 
						|
	GPUBatch *ledges_with_normals;
 | 
						|
 | 
						|
	/* Skip hidden (depending on paint select mode) */
 | 
						|
	GPUBatch *triangles_with_weights;
 | 
						|
	GPUBatch *triangles_with_vert_colors;
 | 
						|
	/* Always skip hidden */
 | 
						|
	GPUBatch *triangles_with_select_mask;
 | 
						|
	GPUBatch *triangles_with_select_id;
 | 
						|
	uint       triangles_with_select_id_offset;
 | 
						|
 | 
						|
	GPUBatch *facedot_with_select_id;  /* shares vbo with 'overlay_facedots' */
 | 
						|
	GPUBatch *edges_with_select_id;
 | 
						|
	GPUBatch *verts_with_select_id;
 | 
						|
 | 
						|
	uint facedot_with_select_id_offset;
 | 
						|
	uint edges_with_select_id_offset;
 | 
						|
	uint verts_with_select_id_offset;
 | 
						|
 | 
						|
	GPUBatch *points_with_normals;
 | 
						|
	GPUBatch *fancy_edges; /* owns its vertex buffer (not shared) */
 | 
						|
 | 
						|
	GPUBatch *edge_detection;
 | 
						|
 | 
						|
	GPUVertBuf *edges_face_overlay;
 | 
						|
	GPUTexture *edges_face_overlay_tx;
 | 
						|
	int edges_face_overlay_tri_count; /* Number of tri in edges_face_overlay(_adj)_tx */
 | 
						|
 | 
						|
	/* Maybe have shaded_triangles_data split into pos_nor and uv_tangent
 | 
						|
	 * to minimize data transfer for skinned mesh. */
 | 
						|
	GPUVertFormat shaded_triangles_format;
 | 
						|
	GPUVertBuf *shaded_triangles_data;
 | 
						|
	GPUIndexBuf **shaded_triangles_in_order;
 | 
						|
	GPUBatch **shaded_triangles;
 | 
						|
 | 
						|
	/* Texture Paint.*/
 | 
						|
	/* per-texture batch */
 | 
						|
	GPUBatch **texpaint_triangles;
 | 
						|
	GPUBatch  *texpaint_triangles_single;
 | 
						|
 | 
						|
	/* Edit Cage Mesh buffers */
 | 
						|
	GPUVertBuf *ed_tri_pos;
 | 
						|
	GPUVertBuf *ed_tri_nor; /* LoopNor, VertNor */
 | 
						|
	GPUVertBuf *ed_tri_data;
 | 
						|
 | 
						|
	GPUVertBuf *ed_ledge_pos;
 | 
						|
	GPUVertBuf *ed_ledge_nor; /* VertNor */
 | 
						|
	GPUVertBuf *ed_ledge_data;
 | 
						|
 | 
						|
	GPUVertBuf *ed_lvert_pos;
 | 
						|
	GPUVertBuf *ed_lvert_nor; /* VertNor */
 | 
						|
	GPUVertBuf *ed_lvert_data;
 | 
						|
 | 
						|
	GPUBatch *overlay_triangles;
 | 
						|
	GPUBatch *overlay_triangles_nor; /* GPU_PRIM_POINTS */
 | 
						|
	GPUBatch *overlay_loose_edges;
 | 
						|
	GPUBatch *overlay_loose_edges_nor; /* GPU_PRIM_POINTS */
 | 
						|
	GPUBatch *overlay_loose_verts;
 | 
						|
	GPUBatch *overlay_facedots;
 | 
						|
 | 
						|
	GPUBatch *overlay_weight_faces;
 | 
						|
	GPUBatch *overlay_weight_verts;
 | 
						|
	GPUBatch *overlay_paint_edges;
 | 
						|
 | 
						|
	/* arrays of bool uniform names (and value) that will be use to
 | 
						|
	 * set srgb conversion for auto attribs.*/
 | 
						|
	char *auto_layer_names;
 | 
						|
	int *auto_layer_is_srgb;
 | 
						|
	int auto_layer_len;
 | 
						|
 | 
						|
	/* settings to determine if cache is invalid */
 | 
						|
	bool is_maybe_dirty;
 | 
						|
	bool is_dirty; /* Instantly invalidates cache, skipping mesh check */
 | 
						|
	int edge_len;
 | 
						|
	int tri_len;
 | 
						|
	int poly_len;
 | 
						|
	int vert_len;
 | 
						|
	int mat_len;
 | 
						|
	bool is_editmode;
 | 
						|
 | 
						|
	struct DRW_MeshWeightState weight_state;
 | 
						|
 | 
						|
	/* XXX, only keep for as long as sculpt mode uses shaded drawing. */
 | 
						|
	bool is_sculpt_points_tag;
 | 
						|
 | 
						|
	/* Valid only if edges_adjacency is up to date. */
 | 
						|
	bool is_manifold;
 | 
						|
} MeshBatchCache;
 | 
						|
 | 
						|
/* GPUBatch cache management. */
 | 
						|
 | 
						|
static bool mesh_batch_cache_valid(Mesh *me)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = me->runtime.batch_cache;
 | 
						|
 | 
						|
	if (cache == NULL) {
 | 
						|
		return false;
 | 
						|
	}
 | 
						|
 | 
						|
	/* XXX find another place for this */
 | 
						|
	if (cache->mat_len != mesh_render_mat_len_get(me)) {
 | 
						|
		cache->is_maybe_dirty = true;
 | 
						|
	}
 | 
						|
 | 
						|
	if (cache->is_editmode != (me->edit_btmesh != NULL)) {
 | 
						|
		return false;
 | 
						|
	}
 | 
						|
 | 
						|
	if (cache->is_dirty) {
 | 
						|
		return false;
 | 
						|
	}
 | 
						|
 | 
						|
	if (cache->is_maybe_dirty == false) {
 | 
						|
		return true;
 | 
						|
	}
 | 
						|
	else {
 | 
						|
		if (cache->is_editmode) {
 | 
						|
			return false;
 | 
						|
		}
 | 
						|
		else if ((cache->vert_len != mesh_render_verts_len_get(me)) ||
 | 
						|
		         (cache->edge_len != mesh_render_edges_len_get(me)) ||
 | 
						|
		         (cache->tri_len  != mesh_render_looptri_len_get(me)) ||
 | 
						|
		         (cache->poly_len != mesh_render_polys_len_get(me)) ||
 | 
						|
		         (cache->mat_len   != mesh_render_mat_len_get(me)))
 | 
						|
		{
 | 
						|
			return false;
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	return true;
 | 
						|
}
 | 
						|
 | 
						|
static void mesh_batch_cache_init(Mesh *me)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = me->runtime.batch_cache;
 | 
						|
 | 
						|
	if (!cache) {
 | 
						|
		cache = me->runtime.batch_cache = MEM_callocN(sizeof(*cache), __func__);
 | 
						|
	}
 | 
						|
	else {
 | 
						|
		memset(cache, 0, sizeof(*cache));
 | 
						|
	}
 | 
						|
 | 
						|
	cache->is_editmode = me->edit_btmesh != NULL;
 | 
						|
 | 
						|
	if (cache->is_editmode == false) {
 | 
						|
		cache->edge_len = mesh_render_edges_len_get(me);
 | 
						|
		cache->tri_len = mesh_render_looptri_len_get(me);
 | 
						|
		cache->poly_len = mesh_render_polys_len_get(me);
 | 
						|
		cache->vert_len = mesh_render_verts_len_get(me);
 | 
						|
	}
 | 
						|
 | 
						|
	cache->mat_len = mesh_render_mat_len_get(me);
 | 
						|
 | 
						|
	cache->is_maybe_dirty = false;
 | 
						|
	cache->is_dirty = false;
 | 
						|
 | 
						|
	DRW_mesh_weight_state_clear(&cache->weight_state);
 | 
						|
}
 | 
						|
 | 
						|
static MeshBatchCache *mesh_batch_cache_get(Mesh *me)
 | 
						|
{
 | 
						|
	if (!mesh_batch_cache_valid(me)) {
 | 
						|
		mesh_batch_cache_clear(me);
 | 
						|
		mesh_batch_cache_init(me);
 | 
						|
	}
 | 
						|
	return me->runtime.batch_cache;
 | 
						|
}
 | 
						|
 | 
						|
static void mesh_batch_cache_check_vertex_group(MeshBatchCache *cache, const struct DRW_MeshWeightState *wstate)
 | 
						|
{
 | 
						|
	if (!DRW_mesh_weight_state_compare(&cache->weight_state, wstate)) {
 | 
						|
		GPU_BATCH_DISCARD_SAFE(cache->triangles_with_weights);
 | 
						|
 | 
						|
		DRW_mesh_weight_state_clear(&cache->weight_state);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static void mesh_batch_cache_discard_shaded_tri(MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	GPU_VERTBUF_DISCARD_SAFE(cache->shaded_triangles_data);
 | 
						|
	if (cache->shaded_triangles_in_order) {
 | 
						|
		for (int i = 0; i < cache->mat_len; ++i) {
 | 
						|
			GPU_INDEXBUF_DISCARD_SAFE(cache->shaded_triangles_in_order[i]);
 | 
						|
		}
 | 
						|
	}
 | 
						|
	if (cache->shaded_triangles) {
 | 
						|
		for (int i = 0; i < cache->mat_len; ++i) {
 | 
						|
			GPU_BATCH_DISCARD_SAFE(cache->shaded_triangles[i]);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	MEM_SAFE_FREE(cache->shaded_triangles_in_order);
 | 
						|
	MEM_SAFE_FREE(cache->shaded_triangles);
 | 
						|
 | 
						|
	MEM_SAFE_FREE(cache->auto_layer_names);
 | 
						|
	MEM_SAFE_FREE(cache->auto_layer_is_srgb);
 | 
						|
}
 | 
						|
 | 
						|
void DRW_mesh_batch_cache_dirty_tag(Mesh *me, int mode)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = me->runtime.batch_cache;
 | 
						|
	if (cache == NULL) {
 | 
						|
		return;
 | 
						|
	}
 | 
						|
	switch (mode) {
 | 
						|
		case BKE_MESH_BATCH_DIRTY_MAYBE_ALL:
 | 
						|
			cache->is_maybe_dirty = true;
 | 
						|
			break;
 | 
						|
		case BKE_MESH_BATCH_DIRTY_SELECT:
 | 
						|
			GPU_VERTBUF_DISCARD_SAFE(cache->ed_tri_data);
 | 
						|
			GPU_VERTBUF_DISCARD_SAFE(cache->ed_ledge_data);
 | 
						|
			GPU_VERTBUF_DISCARD_SAFE(cache->ed_lvert_data);
 | 
						|
			GPU_VERTBUF_DISCARD_SAFE(cache->ed_fcenter_pos_with_nor_and_sel); /* Contains select flag */
 | 
						|
			GPU_VERTBUF_DISCARD_SAFE(cache->ed_edge_pos);
 | 
						|
			GPU_VERTBUF_DISCARD_SAFE(cache->ed_vert_pos);
 | 
						|
 | 
						|
			GPU_BATCH_DISCARD_SAFE(cache->overlay_triangles);
 | 
						|
			GPU_BATCH_DISCARD_SAFE(cache->overlay_loose_verts);
 | 
						|
			GPU_BATCH_DISCARD_SAFE(cache->overlay_loose_edges);
 | 
						|
			GPU_BATCH_DISCARD_SAFE(cache->overlay_facedots);
 | 
						|
			/* Edit mode selection. */
 | 
						|
			GPU_BATCH_DISCARD_SAFE(cache->facedot_with_select_id);
 | 
						|
			GPU_BATCH_DISCARD_SAFE(cache->edges_with_select_id);
 | 
						|
			GPU_BATCH_DISCARD_SAFE(cache->verts_with_select_id);
 | 
						|
			break;
 | 
						|
		case BKE_MESH_BATCH_DIRTY_ALL:
 | 
						|
			cache->is_dirty = true;
 | 
						|
			break;
 | 
						|
		case BKE_MESH_BATCH_DIRTY_SHADING:
 | 
						|
			mesh_batch_cache_discard_shaded_tri(cache);
 | 
						|
			break;
 | 
						|
		case BKE_MESH_BATCH_DIRTY_SCULPT_COORDS:
 | 
						|
			cache->is_sculpt_points_tag = true;
 | 
						|
			break;
 | 
						|
		default:
 | 
						|
			BLI_assert(0);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * This only clear the batches associated to the given vertex buffer.
 | 
						|
 **/
 | 
						|
static void mesh_batch_cache_clear_selective(Mesh *me, GPUVertBuf *vert)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = me->runtime.batch_cache;
 | 
						|
	if (!cache) {
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	BLI_assert(vert != NULL);
 | 
						|
 | 
						|
	if (cache->pos_with_normals == vert) {
 | 
						|
		GPU_BATCH_DISCARD_SAFE(cache->triangles_with_normals);
 | 
						|
		GPU_BATCH_DISCARD_SAFE(cache->triangles_with_weights);
 | 
						|
		GPU_BATCH_DISCARD_SAFE(cache->triangles_with_vert_colors);
 | 
						|
		GPU_BATCH_DISCARD_SAFE(cache->triangles_with_select_id);
 | 
						|
		GPU_BATCH_DISCARD_SAFE(cache->triangles_with_select_mask);
 | 
						|
		GPU_BATCH_DISCARD_SAFE(cache->points_with_normals);
 | 
						|
		GPU_BATCH_DISCARD_SAFE(cache->ledges_with_normals);
 | 
						|
		if (cache->shaded_triangles) {
 | 
						|
			for (int i = 0; i < cache->mat_len; ++i) {
 | 
						|
				GPU_BATCH_DISCARD_SAFE(cache->shaded_triangles[i]);
 | 
						|
			}
 | 
						|
		}
 | 
						|
		MEM_SAFE_FREE(cache->shaded_triangles);
 | 
						|
		if (cache->texpaint_triangles) {
 | 
						|
			for (int i = 0; i < cache->mat_len; ++i) {
 | 
						|
				GPU_BATCH_DISCARD_SAFE(cache->texpaint_triangles[i]);
 | 
						|
			}
 | 
						|
		}
 | 
						|
		MEM_SAFE_FREE(cache->texpaint_triangles);
 | 
						|
		GPU_BATCH_DISCARD_SAFE(cache->texpaint_triangles_single);
 | 
						|
	}
 | 
						|
	/* TODO: add the other ones if needed. */
 | 
						|
	else {
 | 
						|
		/* Does not match any vertbuf in the batch cache! */
 | 
						|
		BLI_assert(0);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static void mesh_batch_cache_clear(Mesh *me)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = me->runtime.batch_cache;
 | 
						|
	if (!cache) {
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	GPU_BATCH_DISCARD_SAFE(cache->all_verts);
 | 
						|
	GPU_BATCH_DISCARD_SAFE(cache->all_edges);
 | 
						|
	GPU_BATCH_DISCARD_SAFE(cache->all_triangles);
 | 
						|
 | 
						|
	GPU_VERTBUF_DISCARD_SAFE(cache->pos_in_order);
 | 
						|
	DRW_TEXTURE_FREE_SAFE(cache->pos_in_order_tx);
 | 
						|
	GPU_INDEXBUF_DISCARD_SAFE(cache->edges_in_order);
 | 
						|
	GPU_INDEXBUF_DISCARD_SAFE(cache->triangles_in_order);
 | 
						|
	GPU_INDEXBUF_DISCARD_SAFE(cache->ledges_in_order);
 | 
						|
 | 
						|
	GPU_VERTBUF_DISCARD_SAFE(cache->ed_tri_pos);
 | 
						|
	GPU_VERTBUF_DISCARD_SAFE(cache->ed_tri_nor);
 | 
						|
	GPU_VERTBUF_DISCARD_SAFE(cache->ed_tri_data);
 | 
						|
	GPU_VERTBUF_DISCARD_SAFE(cache->ed_ledge_pos);
 | 
						|
	GPU_VERTBUF_DISCARD_SAFE(cache->ed_ledge_nor);
 | 
						|
	GPU_VERTBUF_DISCARD_SAFE(cache->ed_ledge_data);
 | 
						|
	GPU_VERTBUF_DISCARD_SAFE(cache->ed_lvert_pos);
 | 
						|
	GPU_VERTBUF_DISCARD_SAFE(cache->ed_lvert_nor);
 | 
						|
	GPU_VERTBUF_DISCARD_SAFE(cache->ed_lvert_data);
 | 
						|
	GPU_BATCH_DISCARD_SAFE(cache->overlay_triangles);
 | 
						|
	GPU_BATCH_DISCARD_SAFE(cache->overlay_triangles_nor);
 | 
						|
	GPU_BATCH_DISCARD_SAFE(cache->overlay_loose_verts);
 | 
						|
	GPU_BATCH_DISCARD_SAFE(cache->overlay_loose_edges);
 | 
						|
	GPU_BATCH_DISCARD_SAFE(cache->overlay_loose_edges_nor);
 | 
						|
 | 
						|
	GPU_BATCH_DISCARD_SAFE(cache->overlay_weight_faces);
 | 
						|
	GPU_BATCH_DISCARD_SAFE(cache->overlay_weight_verts);
 | 
						|
	GPU_BATCH_DISCARD_SAFE(cache->overlay_paint_edges);
 | 
						|
	GPU_BATCH_DISCARD_SAFE(cache->overlay_facedots);
 | 
						|
 | 
						|
	GPU_BATCH_DISCARD_SAFE(cache->triangles_with_normals);
 | 
						|
	GPU_BATCH_DISCARD_SAFE(cache->points_with_normals);
 | 
						|
	GPU_BATCH_DISCARD_SAFE(cache->ledges_with_normals);
 | 
						|
	GPU_VERTBUF_DISCARD_SAFE(cache->pos_with_normals);
 | 
						|
	GPU_BATCH_DISCARD_SAFE(cache->triangles_with_weights);
 | 
						|
	GPU_BATCH_DISCARD_SAFE(cache->triangles_with_vert_colors);
 | 
						|
	GPU_VERTBUF_DISCARD_SAFE(cache->tri_aligned_uv);
 | 
						|
	GPU_VERTBUF_DISCARD_SAFE(cache->ed_fcenter_pos_with_nor_and_sel);
 | 
						|
	GPU_VERTBUF_DISCARD_SAFE(cache->ed_edge_pos);
 | 
						|
	GPU_VERTBUF_DISCARD_SAFE(cache->ed_vert_pos);
 | 
						|
	GPU_BATCH_DISCARD_SAFE(cache->triangles_with_select_mask);
 | 
						|
	GPU_BATCH_DISCARD_SAFE(cache->triangles_with_select_id);
 | 
						|
	GPU_BATCH_DISCARD_SAFE(cache->facedot_with_select_id);
 | 
						|
	GPU_BATCH_DISCARD_SAFE(cache->edges_with_select_id);
 | 
						|
	GPU_BATCH_DISCARD_SAFE(cache->verts_with_select_id);
 | 
						|
 | 
						|
	GPU_BATCH_DISCARD_SAFE(cache->fancy_edges);
 | 
						|
 | 
						|
	GPU_INDEXBUF_DISCARD_SAFE(cache->edges_adjacency);
 | 
						|
	GPU_BATCH_DISCARD_SAFE(cache->edge_detection);
 | 
						|
 | 
						|
	GPU_VERTBUF_DISCARD_SAFE(cache->edges_face_overlay);
 | 
						|
	DRW_TEXTURE_FREE_SAFE(cache->edges_face_overlay_tx);
 | 
						|
 | 
						|
	mesh_batch_cache_discard_shaded_tri(cache);
 | 
						|
 | 
						|
	if (cache->texpaint_triangles) {
 | 
						|
		for (int i = 0; i < cache->mat_len; ++i) {
 | 
						|
			GPU_BATCH_DISCARD_SAFE(cache->texpaint_triangles[i]);
 | 
						|
		}
 | 
						|
	}
 | 
						|
	MEM_SAFE_FREE(cache->texpaint_triangles);
 | 
						|
 | 
						|
	GPU_BATCH_DISCARD_SAFE(cache->texpaint_triangles_single);
 | 
						|
 | 
						|
	DRW_mesh_weight_state_clear(&cache->weight_state);
 | 
						|
}
 | 
						|
 | 
						|
void DRW_mesh_batch_cache_free(Mesh *me)
 | 
						|
{
 | 
						|
	mesh_batch_cache_clear(me);
 | 
						|
	MEM_SAFE_FREE(me->runtime.batch_cache);
 | 
						|
}
 | 
						|
 | 
						|
/* GPUBatch cache usage. */
 | 
						|
 | 
						|
static GPUVertBuf *mesh_batch_cache_get_tri_shading_data(MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & (MR_DATATYPE_VERT | MR_DATATYPE_LOOPTRI | MR_DATATYPE_LOOP | MR_DATATYPE_POLY));
 | 
						|
#define USE_COMP_MESH_DATA
 | 
						|
 | 
						|
	if (cache->shaded_triangles_data == NULL) {
 | 
						|
		const uint uv_len = rdata->cd.layers.uv_len;
 | 
						|
		const uint tangent_len = rdata->cd.layers.tangent_len;
 | 
						|
		const uint vcol_len = rdata->cd.layers.vcol_len;
 | 
						|
		const uint layers_combined_len = uv_len + vcol_len + tangent_len;
 | 
						|
		cache->auto_layer_len = 0;
 | 
						|
 | 
						|
		if (layers_combined_len == 0) {
 | 
						|
			return NULL;
 | 
						|
		}
 | 
						|
 | 
						|
		GPUVertFormat *format = &cache->shaded_triangles_format;
 | 
						|
 | 
						|
		GPU_vertformat_clear(format);
 | 
						|
 | 
						|
		/* initialize vertex format */
 | 
						|
		uint *layers_combined_id = BLI_array_alloca(layers_combined_id, layers_combined_len);
 | 
						|
		uint *uv_id = layers_combined_id;
 | 
						|
		uint *tangent_id = uv_id + uv_len;
 | 
						|
		uint *vcol_id = tangent_id + tangent_len;
 | 
						|
 | 
						|
		/* Not needed, just for sanity. */
 | 
						|
		if (uv_len == 0) { uv_id = NULL; }
 | 
						|
		if (tangent_len == 0) { tangent_id = NULL; }
 | 
						|
		if (vcol_len == 0) { vcol_id = NULL; }
 | 
						|
 | 
						|
		/* Count number of auto layer and allocate big enough name buffer. */
 | 
						|
		uint auto_names_len = 0;
 | 
						|
		uint auto_ofs = 0;
 | 
						|
		uint auto_id = 0;
 | 
						|
		for (uint i = 0; i < uv_len; i++) {
 | 
						|
			const char *attrib_name = mesh_render_data_uv_auto_layer_uuid_get(rdata, i);
 | 
						|
			auto_names_len += strlen(attrib_name) + 2; /* include null terminator and b prefix. */
 | 
						|
			cache->auto_layer_len++;
 | 
						|
		}
 | 
						|
		for (uint i = 0; i < vcol_len; i++) {
 | 
						|
			if (rdata->cd.layers.auto_vcol[i]) {
 | 
						|
				const char *attrib_name = mesh_render_data_vcol_auto_layer_uuid_get(rdata, i);
 | 
						|
				auto_names_len += strlen(attrib_name) + 2; /* include null terminator and b prefix. */
 | 
						|
				cache->auto_layer_len++;
 | 
						|
			}
 | 
						|
		}
 | 
						|
		auto_names_len += 1; /* add an ultimate '\0' terminator */
 | 
						|
		cache->auto_layer_names = MEM_callocN(auto_names_len * sizeof(char), "Auto layer name buf");
 | 
						|
		cache->auto_layer_is_srgb = MEM_mallocN(cache->auto_layer_len * sizeof(int), "Auto layer value buf");
 | 
						|
 | 
						|
		for (uint i = 0; i < uv_len; i++) {
 | 
						|
			/* UV */
 | 
						|
			const char *attrib_name = mesh_render_data_uv_layer_uuid_get(rdata, i);
 | 
						|
#if defined(USE_COMP_MESH_DATA) && 0 /* these are clamped. Maybe use them as an option in the future */
 | 
						|
			uv_id[i] = GPU_vertformat_attr_add(format, attrib_name, GPU_COMP_I16, 2, GPU_FETCH_INT_TO_FLOAT_UNIT);
 | 
						|
#else
 | 
						|
			uv_id[i] = GPU_vertformat_attr_add(format, attrib_name, GPU_COMP_F32, 2, GPU_FETCH_FLOAT);
 | 
						|
#endif
 | 
						|
 | 
						|
			/* Auto Name */
 | 
						|
			attrib_name = mesh_render_data_uv_auto_layer_uuid_get(rdata, i);
 | 
						|
			GPU_vertformat_alias_add(format, attrib_name);
 | 
						|
 | 
						|
			/* +1 include null terminator. */
 | 
						|
			auto_ofs += 1 + BLI_snprintf_rlen(
 | 
						|
			        cache->auto_layer_names + auto_ofs, auto_names_len - auto_ofs, "b%s", attrib_name);
 | 
						|
			cache->auto_layer_is_srgb[auto_id++] = 0; /* tag as not srgb */
 | 
						|
 | 
						|
			if (i == rdata->cd.layers.uv_active) {
 | 
						|
				GPU_vertformat_alias_add(format, "u");
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		for (uint i = 0; i < tangent_len; i++) {
 | 
						|
			const char *attrib_name = mesh_render_data_tangent_layer_uuid_get(rdata, i);
 | 
						|
			/* WATCH IT : only specifying 3 component instead of 4 (4th is sign).
 | 
						|
			 * That may cause some problem but I could not make it to fail (fclem) */
 | 
						|
#ifdef USE_COMP_MESH_DATA
 | 
						|
			/* Tangents need more precision than 10_10_10 */
 | 
						|
			tangent_id[i] = GPU_vertformat_attr_add(format, attrib_name, GPU_COMP_I16, 3, GPU_FETCH_INT_TO_FLOAT_UNIT);
 | 
						|
#else
 | 
						|
			tangent_id[i] = GPU_vertformat_attr_add(format, attrib_name, GPU_COMP_F32, 3, GPU_FETCH_FLOAT);
 | 
						|
#endif
 | 
						|
 | 
						|
			if (i == rdata->cd.layers.tangent_active) {
 | 
						|
				GPU_vertformat_alias_add(format, "t");
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		for (uint i = 0; i < vcol_len; i++) {
 | 
						|
			const char *attrib_name = mesh_render_data_vcol_layer_uuid_get(rdata, i);
 | 
						|
			vcol_id[i] = GPU_vertformat_attr_add(format, attrib_name, GPU_COMP_U8, 3, GPU_FETCH_INT_TO_FLOAT_UNIT);
 | 
						|
 | 
						|
			/* Auto layer */
 | 
						|
			if (rdata->cd.layers.auto_vcol[i]) {
 | 
						|
				attrib_name = mesh_render_data_vcol_auto_layer_uuid_get(rdata, i);
 | 
						|
 | 
						|
				GPU_vertformat_alias_add(format, attrib_name);
 | 
						|
 | 
						|
				/* +1 include null terminator. */
 | 
						|
				auto_ofs += 1 + BLI_snprintf_rlen(
 | 
						|
				        cache->auto_layer_names + auto_ofs, auto_names_len - auto_ofs, "b%s", attrib_name);
 | 
						|
				cache->auto_layer_is_srgb[auto_id++] = 1; /* tag as srgb */
 | 
						|
			}
 | 
						|
 | 
						|
			if (i == rdata->cd.layers.vcol_active) {
 | 
						|
				GPU_vertformat_alias_add(format, "c");
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		const uint tri_len = mesh_render_data_looptri_len_get(rdata);
 | 
						|
 | 
						|
		GPUVertBuf *vbo = cache->shaded_triangles_data = GPU_vertbuf_create_with_format(format);
 | 
						|
 | 
						|
		const int vbo_len_capacity = tri_len * 3;
 | 
						|
		int vbo_len_used = 0;
 | 
						|
		GPU_vertbuf_data_alloc(vbo, vbo_len_capacity);
 | 
						|
 | 
						|
		GPUVertBufRaw *layers_combined_step = BLI_array_alloca(layers_combined_step, layers_combined_len);
 | 
						|
 | 
						|
		GPUVertBufRaw *uv_step      = layers_combined_step;
 | 
						|
		GPUVertBufRaw *tangent_step = uv_step + uv_len;
 | 
						|
		GPUVertBufRaw *vcol_step    = tangent_step + tangent_len;
 | 
						|
 | 
						|
		/* Not needed, just for sanity. */
 | 
						|
		if (uv_len == 0) { uv_step = NULL; }
 | 
						|
		if (tangent_len == 0) { tangent_step = NULL; }
 | 
						|
		if (vcol_len == 0) { vcol_step = NULL; }
 | 
						|
 | 
						|
		for (uint i = 0; i < uv_len; i++) {
 | 
						|
			GPU_vertbuf_attr_get_raw_data(vbo, uv_id[i], &uv_step[i]);
 | 
						|
		}
 | 
						|
		for (uint i = 0; i < tangent_len; i++) {
 | 
						|
			GPU_vertbuf_attr_get_raw_data(vbo, tangent_id[i], &tangent_step[i]);
 | 
						|
		}
 | 
						|
		for (uint i = 0; i < vcol_len; i++) {
 | 
						|
			GPU_vertbuf_attr_get_raw_data(vbo, vcol_id[i], &vcol_step[i]);
 | 
						|
		}
 | 
						|
 | 
						|
		/* TODO deduplicate all verts and make use of GPUIndexBuf in
 | 
						|
		 * mesh_batch_cache_get_triangles_in_order_split_by_material. */
 | 
						|
		if (rdata->edit_bmesh) {
 | 
						|
			for (uint i = 0; i < tri_len; i++) {
 | 
						|
				const BMLoop **bm_looptri = (const BMLoop **)rdata->edit_bmesh->looptris[i];
 | 
						|
				if (BM_elem_flag_test(bm_looptri[0]->f, BM_ELEM_HIDDEN)) {
 | 
						|
					continue;
 | 
						|
				}
 | 
						|
				/* UVs */
 | 
						|
				for (uint j = 0; j < uv_len; j++) {
 | 
						|
					const uint layer_offset = rdata->cd.offset.uv[j];
 | 
						|
					for (uint t = 0; t < 3; t++) {
 | 
						|
						const float *elem = ((MLoopUV *)BM_ELEM_CD_GET_VOID_P(bm_looptri[t], layer_offset))->uv;
 | 
						|
						copy_v2_v2(GPU_vertbuf_raw_step(&uv_step[j]), elem);
 | 
						|
					}
 | 
						|
				}
 | 
						|
				/* TANGENTs */
 | 
						|
				for (uint j = 0; j < tangent_len; j++) {
 | 
						|
					float (*layer_data)[4] = rdata->cd.layers.tangent[j];
 | 
						|
					for (uint t = 0; t < 3; t++) {
 | 
						|
						const float *elem = layer_data[BM_elem_index_get(bm_looptri[t])];
 | 
						|
						normal_float_to_short_v3(GPU_vertbuf_raw_step(&tangent_step[j]), elem);
 | 
						|
					}
 | 
						|
				}
 | 
						|
				/* VCOLs */
 | 
						|
				for (uint j = 0; j < vcol_len; j++) {
 | 
						|
					const uint layer_offset = rdata->cd.offset.vcol[j];
 | 
						|
					for (uint t = 0; t < 3; t++) {
 | 
						|
						const uchar *elem = &((MLoopCol *)BM_ELEM_CD_GET_VOID_P(bm_looptri[t], layer_offset))->r;
 | 
						|
						copy_v3_v3_uchar(GPU_vertbuf_raw_step(&vcol_step[j]), elem);
 | 
						|
					}
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			for (uint i = 0; i < tri_len; i++) {
 | 
						|
				const MLoopTri *mlt = &rdata->mlooptri[i];
 | 
						|
 | 
						|
				/* UVs */
 | 
						|
				for (uint j = 0; j < uv_len; j++) {
 | 
						|
					const MLoopUV *layer_data = rdata->cd.layers.uv[j];
 | 
						|
					for (uint t = 0; t < 3; t++) {
 | 
						|
						const float *elem = layer_data[mlt->tri[t]].uv;
 | 
						|
						copy_v2_v2(GPU_vertbuf_raw_step(&uv_step[j]), elem);
 | 
						|
					}
 | 
						|
				}
 | 
						|
				/* TANGENTs */
 | 
						|
				for (uint j = 0; j < tangent_len; j++) {
 | 
						|
					float (*layer_data)[4] = rdata->cd.layers.tangent[j];
 | 
						|
					for (uint t = 0; t < 3; t++) {
 | 
						|
						const float *elem = layer_data[mlt->tri[t]];
 | 
						|
#ifdef USE_COMP_MESH_DATA
 | 
						|
						normal_float_to_short_v3(GPU_vertbuf_raw_step(&tangent_step[j]), elem);
 | 
						|
#else
 | 
						|
						copy_v3_v3(GPU_vertbuf_raw_step(&tangent_step[j]), elem);
 | 
						|
#endif
 | 
						|
					}
 | 
						|
				}
 | 
						|
				/* VCOLs */
 | 
						|
				for (uint j = 0; j < vcol_len; j++) {
 | 
						|
					const MLoopCol *layer_data = rdata->cd.layers.vcol[j];
 | 
						|
					for (uint t = 0; t < 3; t++) {
 | 
						|
						const uchar *elem = &layer_data[mlt->tri[t]].r;
 | 
						|
						copy_v3_v3_uchar(GPU_vertbuf_raw_step(&vcol_step[j]), elem);
 | 
						|
					}
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		vbo_len_used = GPU_vertbuf_raw_used(&layers_combined_step[0]);
 | 
						|
 | 
						|
#ifndef NDEBUG
 | 
						|
		/* Check all layers are write aligned. */
 | 
						|
		if (layers_combined_len > 1) {
 | 
						|
			for (uint i = 1; i < layers_combined_len; i++) {
 | 
						|
				BLI_assert(vbo_len_used == GPU_vertbuf_raw_used(&layers_combined_step[i]));
 | 
						|
			}
 | 
						|
		}
 | 
						|
#endif
 | 
						|
 | 
						|
		if (vbo_len_capacity != vbo_len_used) {
 | 
						|
			GPU_vertbuf_data_resize(vbo, vbo_len_used);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
#undef USE_COMP_MESH_DATA
 | 
						|
 | 
						|
	return cache->shaded_triangles_data;
 | 
						|
}
 | 
						|
 | 
						|
static GPUVertBuf *mesh_batch_cache_get_tri_uv_active(
 | 
						|
        MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & (MR_DATATYPE_VERT | MR_DATATYPE_LOOPTRI | MR_DATATYPE_LOOP | MR_DATATYPE_LOOPUV));
 | 
						|
 | 
						|
 | 
						|
	if (cache->tri_aligned_uv == NULL) {
 | 
						|
		const MLoopUV *mloopuv = rdata->mloopuv;
 | 
						|
		if (mloopuv == NULL) {
 | 
						|
			return NULL;
 | 
						|
		}
 | 
						|
 | 
						|
		uint vidx = 0;
 | 
						|
 | 
						|
		static GPUVertFormat format = { 0 };
 | 
						|
		static struct { uint uv; } attr_id;
 | 
						|
		if (format.attr_len == 0) {
 | 
						|
			attr_id.uv = GPU_vertformat_attr_add(&format, "uv", GPU_COMP_F32, 2, GPU_FETCH_FLOAT);
 | 
						|
		}
 | 
						|
 | 
						|
		const int tri_len = mesh_render_data_looptri_len_get(rdata);
 | 
						|
 | 
						|
		GPUVertBuf *vbo = cache->tri_aligned_uv = GPU_vertbuf_create_with_format(&format);
 | 
						|
 | 
						|
		const int vbo_len_capacity = tri_len * 3;
 | 
						|
		int vbo_len_used = 0;
 | 
						|
		GPU_vertbuf_data_alloc(vbo, vbo_len_capacity);
 | 
						|
 | 
						|
 | 
						|
		BMEditMesh *embm = rdata->edit_bmesh;
 | 
						|
		/* get uv's from active UVMap */
 | 
						|
		if (rdata->edit_bmesh) {
 | 
						|
			/* edit mode */
 | 
						|
			BMesh *bm = embm->bm;
 | 
						|
 | 
						|
			const int layer_offset = CustomData_get_offset(&bm->ldata, CD_MLOOPUV);
 | 
						|
			for (uint i = 0; i < tri_len; i++) {
 | 
						|
				const BMLoop **bm_looptri = (const BMLoop **)embm->looptris[i];
 | 
						|
				if (BM_elem_flag_test(bm_looptri[0]->f, BM_ELEM_HIDDEN)) {
 | 
						|
					continue;
 | 
						|
				}
 | 
						|
				for (uint t = 0; t < 3; t++) {
 | 
						|
					const BMLoop *loop = bm_looptri[t];
 | 
						|
					const int index = BM_elem_index_get(loop);
 | 
						|
					if (index != -1) {
 | 
						|
						const float *elem = ((MLoopUV *)BM_ELEM_CD_GET_VOID_P(loop, layer_offset))->uv;
 | 
						|
						GPU_vertbuf_attr_set(vbo, attr_id.uv, vidx++, elem);
 | 
						|
					}
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			/* object mode */
 | 
						|
			for (int i = 0; i < tri_len; i++) {
 | 
						|
				const MLoopTri *mlt = &rdata->mlooptri[i];
 | 
						|
				GPU_vertbuf_attr_set(vbo, attr_id.uv, vidx++, mloopuv[mlt->tri[0]].uv);
 | 
						|
				GPU_vertbuf_attr_set(vbo, attr_id.uv, vidx++, mloopuv[mlt->tri[1]].uv);
 | 
						|
				GPU_vertbuf_attr_set(vbo, attr_id.uv, vidx++, mloopuv[mlt->tri[2]].uv);
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		vbo_len_used = vidx;
 | 
						|
 | 
						|
		BLI_assert(vbo_len_capacity == vbo_len_used);
 | 
						|
		UNUSED_VARS_NDEBUG(vbo_len_used);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->tri_aligned_uv;
 | 
						|
}
 | 
						|
 | 
						|
static GPUVertBuf *mesh_batch_cache_get_tri_pos_and_normals_ex(
 | 
						|
        MeshRenderData *rdata, const bool use_hide,
 | 
						|
        GPUVertBuf **r_vbo)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & (MR_DATATYPE_VERT | MR_DATATYPE_LOOPTRI | MR_DATATYPE_LOOP | MR_DATATYPE_POLY));
 | 
						|
 | 
						|
	if (*r_vbo == NULL) {
 | 
						|
		static GPUVertFormat format = { 0 };
 | 
						|
		static struct { uint pos, nor; } attr_id;
 | 
						|
		if (format.attr_len == 0) {
 | 
						|
			attr_id.pos = GPU_vertformat_attr_add(&format, "pos", GPU_COMP_F32, 3, GPU_FETCH_FLOAT);
 | 
						|
			attr_id.nor = GPU_vertformat_attr_add(&format, "nor", GPU_COMP_I10, 3, GPU_FETCH_INT_TO_FLOAT_UNIT);
 | 
						|
		}
 | 
						|
 | 
						|
		const int tri_len = mesh_render_data_looptri_len_get(rdata);
 | 
						|
 | 
						|
		GPUVertBuf *vbo = *r_vbo = GPU_vertbuf_create_with_format(&format);
 | 
						|
 | 
						|
		const int vbo_len_capacity = tri_len * 3;
 | 
						|
		int vbo_len_used = 0;
 | 
						|
		GPU_vertbuf_data_alloc(vbo, vbo_len_capacity);
 | 
						|
 | 
						|
		GPUVertBufRaw pos_step, nor_step;
 | 
						|
		GPU_vertbuf_attr_get_raw_data(vbo, attr_id.pos, &pos_step);
 | 
						|
		GPU_vertbuf_attr_get_raw_data(vbo, attr_id.nor, &nor_step);
 | 
						|
 | 
						|
		float (*lnors)[3] = rdata->loop_normals;
 | 
						|
 | 
						|
		if (rdata->edit_bmesh) {
 | 
						|
			GPUPackedNormal *pnors_pack, *vnors_pack;
 | 
						|
 | 
						|
			if (lnors == NULL) {
 | 
						|
				mesh_render_data_ensure_poly_normals_pack(rdata);
 | 
						|
				mesh_render_data_ensure_vert_normals_pack(rdata);
 | 
						|
 | 
						|
				pnors_pack = rdata->poly_normals_pack;
 | 
						|
				vnors_pack = rdata->vert_normals_pack;
 | 
						|
			}
 | 
						|
 | 
						|
			for (int i = 0; i < tri_len; i++) {
 | 
						|
				const BMLoop **bm_looptri = (const BMLoop **)rdata->edit_bmesh->looptris[i];
 | 
						|
				const BMFace *bm_face = bm_looptri[0]->f;
 | 
						|
 | 
						|
				/* use_hide always for edit-mode */
 | 
						|
				if (BM_elem_flag_test(bm_face, BM_ELEM_HIDDEN)) {
 | 
						|
					continue;
 | 
						|
				}
 | 
						|
 | 
						|
				if (lnors) {
 | 
						|
					for (uint t = 0; t < 3; t++) {
 | 
						|
						const float *nor = lnors[BM_elem_index_get(bm_looptri[t])];
 | 
						|
						*((GPUPackedNormal *)GPU_vertbuf_raw_step(&nor_step)) = GPU_normal_convert_i10_v3(nor);
 | 
						|
					}
 | 
						|
				}
 | 
						|
				else if (BM_elem_flag_test(bm_face, BM_ELEM_SMOOTH)) {
 | 
						|
					for (uint t = 0; t < 3; t++) {
 | 
						|
						*((GPUPackedNormal *)GPU_vertbuf_raw_step(&nor_step)) = vnors_pack[BM_elem_index_get(bm_looptri[t]->v)];
 | 
						|
					}
 | 
						|
				}
 | 
						|
				else {
 | 
						|
					const GPUPackedNormal *snor_pack = &pnors_pack[BM_elem_index_get(bm_face)];
 | 
						|
					for (uint t = 0; t < 3; t++) {
 | 
						|
						*((GPUPackedNormal *)GPU_vertbuf_raw_step(&nor_step)) = *snor_pack;
 | 
						|
					}
 | 
						|
				}
 | 
						|
 | 
						|
				/* TODO(sybren): deduplicate this and all the other places it's pasted to in this file. */
 | 
						|
				if (rdata->edit_data && rdata->edit_data->vertexCos) {
 | 
						|
					for (uint t = 0; t < 3; t++) {
 | 
						|
						int vidx = BM_elem_index_get(bm_looptri[t]->v);
 | 
						|
						const float *pos = rdata->edit_data->vertexCos[vidx];
 | 
						|
						copy_v3_v3(GPU_vertbuf_raw_step(&pos_step), pos);
 | 
						|
					}
 | 
						|
				}
 | 
						|
				else {
 | 
						|
					for (uint t = 0; t < 3; t++) {
 | 
						|
						copy_v3_v3(GPU_vertbuf_raw_step(&pos_step), bm_looptri[t]->v->co);
 | 
						|
					}
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			if (lnors == NULL) {
 | 
						|
				/* Use normals from vertex. */
 | 
						|
				mesh_render_data_ensure_poly_normals_pack(rdata);
 | 
						|
			}
 | 
						|
 | 
						|
			for (int i = 0; i < tri_len; i++) {
 | 
						|
				const MLoopTri *mlt = &rdata->mlooptri[i];
 | 
						|
				const MPoly *mp = &rdata->mpoly[mlt->poly];
 | 
						|
 | 
						|
				if (use_hide && (mp->flag & ME_HIDE)) {
 | 
						|
					continue;
 | 
						|
				}
 | 
						|
 | 
						|
				const uint vtri[3] = {
 | 
						|
					rdata->mloop[mlt->tri[0]].v,
 | 
						|
					rdata->mloop[mlt->tri[1]].v,
 | 
						|
					rdata->mloop[mlt->tri[2]].v,
 | 
						|
				};
 | 
						|
 | 
						|
				if (lnors) {
 | 
						|
					for (uint t = 0; t < 3; t++) {
 | 
						|
						const float *nor = lnors[mlt->tri[t]];
 | 
						|
						*((GPUPackedNormal *)GPU_vertbuf_raw_step(&nor_step)) = GPU_normal_convert_i10_v3(nor);
 | 
						|
					}
 | 
						|
				}
 | 
						|
				else if (mp->flag & ME_SMOOTH) {
 | 
						|
					for (uint t = 0; t < 3; t++) {
 | 
						|
						const MVert *mv = &rdata->mvert[vtri[t]];
 | 
						|
						*((GPUPackedNormal *)GPU_vertbuf_raw_step(&nor_step)) = GPU_normal_convert_i10_s3(mv->no);
 | 
						|
					}
 | 
						|
				}
 | 
						|
				else {
 | 
						|
					const GPUPackedNormal *pnors_pack = &rdata->poly_normals_pack[mlt->poly];
 | 
						|
					for (uint t = 0; t < 3; t++) {
 | 
						|
						*((GPUPackedNormal *)GPU_vertbuf_raw_step(&nor_step)) = *pnors_pack;
 | 
						|
					}
 | 
						|
				}
 | 
						|
 | 
						|
				for (uint t = 0; t < 3; t++) {
 | 
						|
					const MVert *mv = &rdata->mvert[vtri[t]];
 | 
						|
					copy_v3_v3(GPU_vertbuf_raw_step(&pos_step), mv->co);
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		vbo_len_used = GPU_vertbuf_raw_used(&pos_step);
 | 
						|
		BLI_assert(vbo_len_used == GPU_vertbuf_raw_used(&nor_step));
 | 
						|
 | 
						|
		if (vbo_len_capacity != vbo_len_used) {
 | 
						|
			GPU_vertbuf_data_resize(vbo, vbo_len_used);
 | 
						|
		}
 | 
						|
	}
 | 
						|
	return *r_vbo;
 | 
						|
}
 | 
						|
 | 
						|
static GPUVertBuf *mesh_batch_cache_get_tri_pos_and_normals(
 | 
						|
        MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	return mesh_batch_cache_get_tri_pos_and_normals_ex(
 | 
						|
	        rdata, false,
 | 
						|
	        &cache->pos_with_normals);
 | 
						|
}
 | 
						|
static GPUVertBuf *mesh_create_tri_pos_and_normals_visible_only(
 | 
						|
        MeshRenderData *rdata)
 | 
						|
{
 | 
						|
	GPUVertBuf *vbo_dummy = NULL;
 | 
						|
	return mesh_batch_cache_get_tri_pos_and_normals_ex(
 | 
						|
	        rdata, true,
 | 
						|
	        &vbo_dummy);
 | 
						|
}
 | 
						|
 | 
						|
static GPUVertBuf *mesh_batch_cache_get_facedot_pos_with_normals_and_flag(
 | 
						|
        MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & (MR_DATATYPE_VERT | MR_DATATYPE_LOOP | MR_DATATYPE_POLY));
 | 
						|
 | 
						|
	if (cache->ed_fcenter_pos_with_nor_and_sel == NULL) {
 | 
						|
		static GPUVertFormat format = { 0 };
 | 
						|
		static struct { uint pos, data; } attr_id;
 | 
						|
		if (format.attr_len == 0) {
 | 
						|
			attr_id.pos = GPU_vertformat_attr_add(&format, "pos", GPU_COMP_F32, 3, GPU_FETCH_FLOAT);
 | 
						|
			attr_id.data = GPU_vertformat_attr_add(&format, "norAndFlag", GPU_COMP_I10, 4, GPU_FETCH_INT_TO_FLOAT_UNIT);
 | 
						|
		}
 | 
						|
 | 
						|
		const int vbo_len_capacity = mesh_render_data_polys_len_get(rdata);
 | 
						|
		int vidx = 0;
 | 
						|
 | 
						|
		GPUVertBuf *vbo = cache->ed_fcenter_pos_with_nor_and_sel = GPU_vertbuf_create_with_format(&format);
 | 
						|
		GPU_vertbuf_data_alloc(vbo, vbo_len_capacity);
 | 
						|
		for (int i = 0; i < vbo_len_capacity; ++i) {
 | 
						|
			float pcenter[3], pnor[3];
 | 
						|
			bool selected = false;
 | 
						|
 | 
						|
			if (mesh_render_data_pnors_pcenter_select_get(rdata, i, pnor, pcenter, &selected)) {
 | 
						|
 | 
						|
				GPUPackedNormal nor = { .x = 0, .y = 0, .z = -511 };
 | 
						|
				nor = GPU_normal_convert_i10_v3(pnor);
 | 
						|
				nor.w = selected ? 1 : 0;
 | 
						|
				GPU_vertbuf_attr_set(vbo, attr_id.data, vidx, &nor);
 | 
						|
 | 
						|
				GPU_vertbuf_attr_set(vbo, attr_id.pos, vidx, pcenter);
 | 
						|
 | 
						|
				vidx += 1;
 | 
						|
			}
 | 
						|
		}
 | 
						|
		const int vbo_len_used = vidx;
 | 
						|
		BLI_assert(vbo_len_used <= vbo_len_capacity);
 | 
						|
		if (vbo_len_used != vbo_len_capacity) {
 | 
						|
			GPU_vertbuf_data_resize(vbo, vbo_len_used);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->ed_fcenter_pos_with_nor_and_sel;
 | 
						|
}
 | 
						|
 | 
						|
static GPUVertBuf *mesh_batch_cache_get_edges_visible(
 | 
						|
        MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & (MR_DATATYPE_VERT | MR_DATATYPE_EDGE));
 | 
						|
 | 
						|
	if (cache->ed_edge_pos == NULL) {
 | 
						|
		static GPUVertFormat format = { 0 };
 | 
						|
		static struct { uint pos, data; } attr_id;
 | 
						|
		if (format.attr_len == 0) {
 | 
						|
			attr_id.pos = GPU_vertformat_attr_add(&format, "pos", GPU_COMP_F32, 3, GPU_FETCH_FLOAT);
 | 
						|
		}
 | 
						|
 | 
						|
		const int vbo_len_capacity = mesh_render_data_edges_len_get(rdata) * 2;
 | 
						|
		int vidx = 0;
 | 
						|
 | 
						|
		GPUVertBuf *vbo = cache->ed_edge_pos = GPU_vertbuf_create_with_format(&format);
 | 
						|
		GPU_vertbuf_data_alloc(vbo, vbo_len_capacity);
 | 
						|
		if (rdata->edit_bmesh) {
 | 
						|
			BMesh *bm = rdata->edit_bmesh->bm;
 | 
						|
			BMIter iter;
 | 
						|
			BMEdge *eed;
 | 
						|
 | 
						|
			BM_ITER_MESH (eed, &iter, bm, BM_EDGES_OF_MESH) {
 | 
						|
				if (!BM_elem_flag_test(eed, BM_ELEM_HIDDEN)) {
 | 
						|
					GPU_vertbuf_attr_set(vbo, attr_id.pos, vidx, eed->v1->co);
 | 
						|
					vidx += 1;
 | 
						|
					GPU_vertbuf_attr_set(vbo, attr_id.pos, vidx, eed->v2->co);
 | 
						|
					vidx += 1;
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			/* not yet done! */
 | 
						|
			BLI_assert(0);
 | 
						|
		}
 | 
						|
		const int vbo_len_used = vidx;
 | 
						|
		if (vbo_len_used != vbo_len_capacity) {
 | 
						|
			GPU_vertbuf_data_resize(vbo, vbo_len_used);
 | 
						|
		}
 | 
						|
		UNUSED_VARS_NDEBUG(vbo_len_used);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->ed_edge_pos;
 | 
						|
}
 | 
						|
 | 
						|
static GPUVertBuf *mesh_batch_cache_get_verts_visible(
 | 
						|
        MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & MR_DATATYPE_VERT);
 | 
						|
 | 
						|
	if (cache->ed_vert_pos == NULL) {
 | 
						|
		static GPUVertFormat format = { 0 };
 | 
						|
		static struct { uint pos, data; } attr_id;
 | 
						|
		if (format.attr_len == 0) {
 | 
						|
			attr_id.pos = GPU_vertformat_attr_add(&format, "pos", GPU_COMP_F32, 3, GPU_FETCH_FLOAT);
 | 
						|
		}
 | 
						|
 | 
						|
		const int vbo_len_capacity = mesh_render_data_verts_len_get(rdata);
 | 
						|
		uint vidx = 0;
 | 
						|
 | 
						|
		GPUVertBuf *vbo = cache->ed_vert_pos = GPU_vertbuf_create_with_format(&format);
 | 
						|
		GPU_vertbuf_data_alloc(vbo, vbo_len_capacity);
 | 
						|
		if (rdata->edit_bmesh) {
 | 
						|
			BMesh *bm = rdata->edit_bmesh->bm;
 | 
						|
			BMIter iter;
 | 
						|
			BMVert *eve;
 | 
						|
 | 
						|
			BM_ITER_MESH (eve, &iter, bm, BM_VERTS_OF_MESH) {
 | 
						|
				if (!BM_elem_flag_test(eve, BM_ELEM_HIDDEN)) {
 | 
						|
					GPU_vertbuf_attr_set(vbo, attr_id.pos, vidx, eve->co);
 | 
						|
					vidx += 1;
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			for (int i = 0; i < vbo_len_capacity; i++) {
 | 
						|
				const MVert *mv = &rdata->mvert[i];
 | 
						|
				if (!(mv->flag & ME_HIDE)) {
 | 
						|
					GPU_vertbuf_attr_set(vbo, attr_id.pos, vidx, mv->co);
 | 
						|
					vidx += 1;
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		const uint vbo_len_used = vidx;
 | 
						|
		if (vbo_len_used != vbo_len_capacity) {
 | 
						|
			GPU_vertbuf_data_resize(vbo, vbo_len_used);
 | 
						|
		}
 | 
						|
 | 
						|
		UNUSED_VARS_NDEBUG(vbo_len_used);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->ed_vert_pos;
 | 
						|
}
 | 
						|
 | 
						|
static GPUVertBuf *mesh_create_facedot_select_id(
 | 
						|
        MeshRenderData *rdata, uint select_id_offset)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & (MR_DATATYPE_VERT | MR_DATATYPE_LOOP | MR_DATATYPE_POLY));
 | 
						|
 | 
						|
	GPUVertBuf *vbo;
 | 
						|
	{
 | 
						|
		static GPUVertFormat format = { 0 };
 | 
						|
		static struct { uint pos, col; } attr_id;
 | 
						|
		if (format.attr_len == 0) {
 | 
						|
			attr_id.col = GPU_vertformat_attr_add(&format, "color", GPU_COMP_I32, 1, GPU_FETCH_INT);
 | 
						|
		}
 | 
						|
 | 
						|
		const int vbo_len_capacity = mesh_render_data_polys_len_get(rdata);
 | 
						|
		int vidx = 0;
 | 
						|
 | 
						|
		vbo = GPU_vertbuf_create_with_format(&format);
 | 
						|
		GPU_vertbuf_data_alloc(vbo, vbo_len_capacity);
 | 
						|
		uint select_index = select_id_offset;
 | 
						|
 | 
						|
		if (rdata->edit_bmesh) {
 | 
						|
			BMesh *bm = rdata->edit_bmesh->bm;
 | 
						|
			BMIter iter;
 | 
						|
			BMEdge *efa;
 | 
						|
 | 
						|
			BM_ITER_MESH (efa, &iter, bm, BM_FACES_OF_MESH) {
 | 
						|
				if (!BM_elem_flag_test(efa, BM_ELEM_HIDDEN)) {
 | 
						|
					int select_id;
 | 
						|
					GPU_select_index_get(select_index, &select_id);
 | 
						|
					GPU_vertbuf_attr_set(vbo, attr_id.col, vidx, &select_id);
 | 
						|
					vidx += 1;
 | 
						|
				}
 | 
						|
				select_index += 1;
 | 
						|
			}
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			/* not yet done! */
 | 
						|
			BLI_assert(0);
 | 
						|
		}
 | 
						|
		const int vbo_len_used = vidx;
 | 
						|
		if (vbo_len_used != vbo_len_capacity) {
 | 
						|
			GPU_vertbuf_data_resize(vbo, vbo_len_used);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	return vbo;
 | 
						|
}
 | 
						|
 | 
						|
static GPUVertBuf *mesh_create_edges_select_id(
 | 
						|
        MeshRenderData *rdata, uint select_id_offset)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & (MR_DATATYPE_VERT | MR_DATATYPE_EDGE));
 | 
						|
 | 
						|
	GPUVertBuf *vbo;
 | 
						|
	{
 | 
						|
		static GPUVertFormat format = { 0 };
 | 
						|
		static struct { uint pos, col; } attr_id;
 | 
						|
		if (format.attr_len == 0) {
 | 
						|
			attr_id.col = GPU_vertformat_attr_add(&format, "color", GPU_COMP_I32, 1, GPU_FETCH_INT);
 | 
						|
		}
 | 
						|
 | 
						|
		const int vbo_len_capacity = mesh_render_data_edges_len_get(rdata) * 2;
 | 
						|
		int vidx = 0;
 | 
						|
 | 
						|
		vbo = GPU_vertbuf_create_with_format(&format);
 | 
						|
		GPU_vertbuf_data_alloc(vbo, vbo_len_capacity);
 | 
						|
		uint select_index = select_id_offset;
 | 
						|
 | 
						|
		if (rdata->edit_bmesh) {
 | 
						|
			BMesh *bm = rdata->edit_bmesh->bm;
 | 
						|
			BMIter iter;
 | 
						|
			BMEdge *eed;
 | 
						|
 | 
						|
			BM_ITER_MESH (eed, &iter, bm, BM_EDGES_OF_MESH) {
 | 
						|
				if (!BM_elem_flag_test(eed, BM_ELEM_HIDDEN)) {
 | 
						|
					int select_id;
 | 
						|
					GPU_select_index_get(select_index, &select_id);
 | 
						|
					GPU_vertbuf_attr_set(vbo, attr_id.col, vidx, &select_id);
 | 
						|
					vidx += 1;
 | 
						|
					GPU_vertbuf_attr_set(vbo, attr_id.col, vidx, &select_id);
 | 
						|
					vidx += 1;
 | 
						|
				}
 | 
						|
				select_index += 1;
 | 
						|
			}
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			/* not yet done! */
 | 
						|
			BLI_assert(0);
 | 
						|
		}
 | 
						|
		const int vbo_len_used = vidx;
 | 
						|
		if (vbo_len_used != vbo_len_capacity) {
 | 
						|
			GPU_vertbuf_data_resize(vbo, vbo_len_used);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	return vbo;
 | 
						|
}
 | 
						|
 | 
						|
static GPUVertBuf *mesh_create_verts_select_id(
 | 
						|
        MeshRenderData *rdata, uint select_id_offset)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & (MR_DATATYPE_VERT | MR_DATATYPE_LOOP | MR_DATATYPE_POLY));
 | 
						|
 | 
						|
	GPUVertBuf *vbo;
 | 
						|
	{
 | 
						|
		static GPUVertFormat format = { 0 };
 | 
						|
		static struct { uint pos, col; } attr_id;
 | 
						|
		if (format.attr_len == 0) {
 | 
						|
			attr_id.col = GPU_vertformat_attr_add(&format, "color", GPU_COMP_I32, 1, GPU_FETCH_INT);
 | 
						|
		}
 | 
						|
 | 
						|
		const int vbo_len_capacity = mesh_render_data_verts_len_get(rdata);
 | 
						|
		int vidx = 0;
 | 
						|
 | 
						|
		vbo = GPU_vertbuf_create_with_format(&format);
 | 
						|
		GPU_vertbuf_data_alloc(vbo, vbo_len_capacity);
 | 
						|
		uint select_index = select_id_offset;
 | 
						|
 | 
						|
		if (rdata->edit_bmesh) {
 | 
						|
			BMesh *bm = rdata->edit_bmesh->bm;
 | 
						|
			BMIter iter;
 | 
						|
			BMVert *eve;
 | 
						|
 | 
						|
			BM_ITER_MESH (eve, &iter, bm, BM_VERTS_OF_MESH) {
 | 
						|
				if (!BM_elem_flag_test(eve, BM_ELEM_HIDDEN)) {
 | 
						|
					int select_id;
 | 
						|
					GPU_select_index_get(select_index, &select_id);
 | 
						|
					GPU_vertbuf_attr_set(vbo, attr_id.col, vidx, &select_id);
 | 
						|
					vidx += 1;
 | 
						|
				}
 | 
						|
				select_index += 1;
 | 
						|
			}
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			for (int i = 0; i < vbo_len_capacity; i++) {
 | 
						|
				const MVert *mv = &rdata->mvert[i];
 | 
						|
				if (!(mv->flag & ME_HIDE)) {
 | 
						|
					int select_id;
 | 
						|
					GPU_select_index_get(select_index, &select_id);
 | 
						|
					GPU_vertbuf_attr_set(vbo, attr_id.col, vidx, &select_id);
 | 
						|
					vidx += 1;
 | 
						|
				}
 | 
						|
				select_index += 1;
 | 
						|
			}
 | 
						|
		}
 | 
						|
		const int vbo_len_used = vidx;
 | 
						|
		if (vbo_len_used != vbo_len_capacity) {
 | 
						|
			GPU_vertbuf_data_resize(vbo, vbo_len_used);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	return vbo;
 | 
						|
}
 | 
						|
 | 
						|
static GPUVertBuf *mesh_create_tri_weights(
 | 
						|
        MeshRenderData *rdata, bool use_hide, const struct DRW_MeshWeightState *wstate)
 | 
						|
{
 | 
						|
	BLI_assert(
 | 
						|
	        rdata->types &
 | 
						|
	        (MR_DATATYPE_VERT | MR_DATATYPE_LOOPTRI | MR_DATATYPE_LOOP | MR_DATATYPE_POLY | MR_DATATYPE_DVERT));
 | 
						|
 | 
						|
	GPUVertBuf *vbo;
 | 
						|
	{
 | 
						|
		uint cidx = 0;
 | 
						|
 | 
						|
		static GPUVertFormat format = { 0 };
 | 
						|
		static struct { uint col; } attr_id;
 | 
						|
		if (format.attr_len == 0) {
 | 
						|
			attr_id.col = GPU_vertformat_attr_add(&format, "color", GPU_COMP_F32, 3, GPU_FETCH_FLOAT);
 | 
						|
		}
 | 
						|
 | 
						|
		vbo = GPU_vertbuf_create_with_format(&format);
 | 
						|
 | 
						|
		const int tri_len = mesh_render_data_looptri_len_get(rdata);
 | 
						|
		const int vbo_len_capacity = tri_len * 3;
 | 
						|
		int vbo_len_used = 0;
 | 
						|
		GPU_vertbuf_data_alloc(vbo, vbo_len_capacity);
 | 
						|
 | 
						|
		mesh_render_data_ensure_vert_weight_color(rdata, wstate);
 | 
						|
		const float (*vert_weight_color)[3] = rdata->vert_weight_color;
 | 
						|
 | 
						|
		if (rdata->edit_bmesh) {
 | 
						|
			for (int i = 0; i < tri_len; i++) {
 | 
						|
				const BMLoop **ltri = (const BMLoop **)rdata->edit_bmesh->looptris[i];
 | 
						|
				/* Assume 'use_hide' */
 | 
						|
				if (!BM_elem_flag_test(ltri[0]->f, BM_ELEM_HIDDEN)) {
 | 
						|
					for (uint tri_corner = 0; tri_corner < 3; tri_corner++) {
 | 
						|
						const int v_index = BM_elem_index_get(ltri[tri_corner]->v);
 | 
						|
						GPU_vertbuf_attr_set(vbo, attr_id.col, cidx++, vert_weight_color[v_index]);
 | 
						|
					}
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			for (int i = 0; i < tri_len; i++) {
 | 
						|
				const MLoopTri *mlt = &rdata->mlooptri[i];
 | 
						|
				if (!(use_hide && (rdata->mpoly[mlt->poly].flag & ME_HIDE))) {
 | 
						|
					for (uint tri_corner = 0; tri_corner < 3; tri_corner++) {
 | 
						|
						const uint v_index = rdata->mloop[mlt->tri[tri_corner]].v;
 | 
						|
						GPU_vertbuf_attr_set(vbo, attr_id.col, cidx++, vert_weight_color[v_index]);
 | 
						|
					}
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		vbo_len_used = cidx;
 | 
						|
 | 
						|
		if (vbo_len_capacity != vbo_len_used) {
 | 
						|
			GPU_vertbuf_data_resize(vbo, vbo_len_used);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	return vbo;
 | 
						|
}
 | 
						|
 | 
						|
static GPUVertBuf *mesh_create_tri_vert_colors(
 | 
						|
        MeshRenderData *rdata, bool use_hide)
 | 
						|
{
 | 
						|
	BLI_assert(
 | 
						|
	        rdata->types &
 | 
						|
	        (MR_DATATYPE_VERT | MR_DATATYPE_LOOPTRI | MR_DATATYPE_LOOP | MR_DATATYPE_POLY | MR_DATATYPE_LOOPCOL));
 | 
						|
 | 
						|
	GPUVertBuf *vbo;
 | 
						|
	{
 | 
						|
		uint cidx = 0;
 | 
						|
 | 
						|
		static GPUVertFormat format = { 0 };
 | 
						|
		static struct { uint col; } attr_id;
 | 
						|
		if (format.attr_len == 0) {
 | 
						|
			attr_id.col = GPU_vertformat_attr_add(&format, "color", GPU_COMP_U8, 3, GPU_FETCH_INT_TO_FLOAT_UNIT);
 | 
						|
		}
 | 
						|
 | 
						|
		const int tri_len = mesh_render_data_looptri_len_get(rdata);
 | 
						|
 | 
						|
		vbo = GPU_vertbuf_create_with_format(&format);
 | 
						|
 | 
						|
		const uint vbo_len_capacity = tri_len * 3;
 | 
						|
		GPU_vertbuf_data_alloc(vbo, vbo_len_capacity);
 | 
						|
 | 
						|
		mesh_render_data_ensure_vert_color(rdata);
 | 
						|
		const char (*vert_color)[3] = rdata->vert_color;
 | 
						|
 | 
						|
		if (rdata->edit_bmesh) {
 | 
						|
			for (int i = 0; i < tri_len; i++) {
 | 
						|
				const BMLoop **ltri = (const BMLoop **)rdata->edit_bmesh->looptris[i];
 | 
						|
				/* Assume 'use_hide' */
 | 
						|
				if (!BM_elem_flag_test(ltri[0]->f, BM_ELEM_HIDDEN)) {
 | 
						|
					for (uint tri_corner = 0; tri_corner < 3; tri_corner++) {
 | 
						|
						const int l_index = BM_elem_index_get(ltri[tri_corner]);
 | 
						|
						GPU_vertbuf_attr_set(vbo, attr_id.col, cidx++, vert_color[l_index]);
 | 
						|
					}
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			for (int i = 0; i < tri_len; i++) {
 | 
						|
				const MLoopTri *mlt = &rdata->mlooptri[i];
 | 
						|
				if (!(use_hide && (rdata->mpoly[mlt->poly].flag & ME_HIDE))) {
 | 
						|
					for (uint tri_corner = 0; tri_corner < 3; tri_corner++) {
 | 
						|
						const uint l_index = mlt->tri[tri_corner];
 | 
						|
						GPU_vertbuf_attr_set(vbo, attr_id.col, cidx++, vert_color[l_index]);
 | 
						|
					}
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		const uint vbo_len_used = cidx;
 | 
						|
 | 
						|
		if (vbo_len_capacity != vbo_len_used) {
 | 
						|
			GPU_vertbuf_data_resize(vbo, vbo_len_used);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	return vbo;
 | 
						|
}
 | 
						|
 | 
						|
static GPUVertBuf *mesh_create_tri_select_id(
 | 
						|
        MeshRenderData *rdata, bool use_hide, uint select_id_offset)
 | 
						|
{
 | 
						|
	BLI_assert(
 | 
						|
	        rdata->types &
 | 
						|
	        (MR_DATATYPE_VERT | MR_DATATYPE_LOOPTRI | MR_DATATYPE_LOOP | MR_DATATYPE_POLY));
 | 
						|
 | 
						|
	GPUVertBuf *vbo;
 | 
						|
	{
 | 
						|
		uint cidx = 0;
 | 
						|
 | 
						|
		static GPUVertFormat format = { 0 };
 | 
						|
		static struct { uint col; } attr_id;
 | 
						|
		if (format.attr_len == 0) {
 | 
						|
			attr_id.col = GPU_vertformat_attr_add(&format, "color", GPU_COMP_I32, 1, GPU_FETCH_INT);
 | 
						|
		}
 | 
						|
 | 
						|
		const int tri_len = mesh_render_data_looptri_len_get(rdata);
 | 
						|
 | 
						|
		vbo = GPU_vertbuf_create_with_format(&format);
 | 
						|
 | 
						|
		const int vbo_len_capacity = tri_len * 3;
 | 
						|
		int vbo_len_used = 0;
 | 
						|
		GPU_vertbuf_data_alloc(vbo, vbo_len_capacity);
 | 
						|
 | 
						|
		if (rdata->edit_bmesh) {
 | 
						|
			for (int i = 0; i < tri_len; i++) {
 | 
						|
				const BMLoop **ltri = (const BMLoop **)rdata->edit_bmesh->looptris[i];
 | 
						|
				/* Assume 'use_hide' */
 | 
						|
				if (!BM_elem_flag_test(ltri[0]->f, BM_ELEM_HIDDEN)) {
 | 
						|
					const int poly_index = BM_elem_index_get(ltri[0]->f);
 | 
						|
					int select_id;
 | 
						|
					GPU_select_index_get(poly_index + select_id_offset, &select_id);
 | 
						|
					for (uint tri_corner = 0; tri_corner < 3; tri_corner++) {
 | 
						|
						GPU_vertbuf_attr_set(vbo, attr_id.col, cidx++, &select_id);
 | 
						|
					}
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			for (int i = 0; i < tri_len; i++) {
 | 
						|
				const MLoopTri *mlt = &rdata->mlooptri[i];
 | 
						|
				const int poly_index = mlt->poly;
 | 
						|
				if (!(use_hide && (rdata->mpoly[poly_index].flag & ME_HIDE))) {
 | 
						|
					int select_id;
 | 
						|
					GPU_select_index_get(poly_index + select_id_offset, &select_id);
 | 
						|
					for (uint tri_corner = 0; tri_corner < 3; tri_corner++) {
 | 
						|
						GPU_vertbuf_attr_set(vbo, attr_id.col, cidx++, &select_id);
 | 
						|
					}
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		vbo_len_used = cidx;
 | 
						|
 | 
						|
		if (vbo_len_capacity != vbo_len_used) {
 | 
						|
			GPU_vertbuf_data_resize(vbo, vbo_len_used);
 | 
						|
		}
 | 
						|
	}
 | 
						|
	return vbo;
 | 
						|
}
 | 
						|
 | 
						|
static GPUVertBuf *mesh_batch_cache_get_vert_pos_and_nor_in_order(
 | 
						|
        MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & MR_DATATYPE_VERT);
 | 
						|
 | 
						|
	if (cache->pos_in_order == NULL) {
 | 
						|
		static GPUVertFormat format = { 0 };
 | 
						|
		static struct { uint pos, nor; } attr_id;
 | 
						|
		if (format.attr_len == 0) {
 | 
						|
			/* Normal is padded so that the vbo can be used as a buffer texture */
 | 
						|
			attr_id.pos = GPU_vertformat_attr_add(&format, "pos", GPU_COMP_F32, 3, GPU_FETCH_FLOAT);
 | 
						|
			attr_id.nor = GPU_vertformat_attr_add(&format, "nor", GPU_COMP_I16, 4, GPU_FETCH_INT_TO_FLOAT_UNIT);
 | 
						|
		}
 | 
						|
 | 
						|
		GPUVertBuf *vbo = cache->pos_in_order = GPU_vertbuf_create_with_format(&format);
 | 
						|
		const int vbo_len_capacity = mesh_render_data_verts_len_get(rdata);
 | 
						|
		GPU_vertbuf_data_alloc(vbo, vbo_len_capacity);
 | 
						|
 | 
						|
		if (rdata->edit_bmesh) {
 | 
						|
			BMesh *bm = rdata->edit_bmesh->bm;
 | 
						|
			BMIter iter;
 | 
						|
			BMVert *eve;
 | 
						|
			uint i;
 | 
						|
 | 
						|
			BM_ITER_MESH_INDEX (eve, &iter, bm, BM_VERTS_OF_MESH, i) {
 | 
						|
				static short no_short[4];
 | 
						|
				normal_float_to_short_v3(no_short, eve->no);
 | 
						|
 | 
						|
				GPU_vertbuf_attr_set(vbo, attr_id.pos, i, eve->co);
 | 
						|
				GPU_vertbuf_attr_set(vbo, attr_id.nor, i, no_short);
 | 
						|
			}
 | 
						|
			BLI_assert(i == vbo_len_capacity);
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			for (int i = 0; i < vbo_len_capacity; ++i) {
 | 
						|
				GPU_vertbuf_attr_set(vbo, attr_id.pos, i, rdata->mvert[i].co);
 | 
						|
				GPU_vertbuf_attr_set(vbo, attr_id.nor, i, rdata->mvert[i].no); /* XXX actually reading 4 shorts */
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->pos_in_order;
 | 
						|
}
 | 
						|
 | 
						|
static GPUVertFormat *edit_mesh_overlay_pos_format(uint *r_pos_id)
 | 
						|
{
 | 
						|
	static GPUVertFormat format_pos = { 0 };
 | 
						|
	static uint pos_id;
 | 
						|
	if (format_pos.attr_len == 0) {
 | 
						|
		pos_id = GPU_vertformat_attr_add(&format_pos, "pos", GPU_COMP_F32, 3, GPU_FETCH_FLOAT);
 | 
						|
	}
 | 
						|
	*r_pos_id = pos_id;
 | 
						|
	return &format_pos;
 | 
						|
}
 | 
						|
 | 
						|
static GPUVertFormat *edit_mesh_overlay_nor_format(uint *r_vnor_id, uint *r_lnor_id)
 | 
						|
{
 | 
						|
	static GPUVertFormat format_nor = { 0 };
 | 
						|
	static GPUVertFormat format_nor_loop = { 0 };
 | 
						|
	static uint vnor_id, vnor_loop_id, lnor_id;
 | 
						|
	if (format_nor.attr_len == 0) {
 | 
						|
		vnor_id = GPU_vertformat_attr_add(&format_nor, "vnor", GPU_COMP_I10, 3, GPU_FETCH_INT_TO_FLOAT_UNIT);
 | 
						|
		vnor_loop_id = GPU_vertformat_attr_add(&format_nor_loop, "vnor", GPU_COMP_I10, 3, GPU_FETCH_INT_TO_FLOAT_UNIT);
 | 
						|
		lnor_id = GPU_vertformat_attr_add(&format_nor_loop, "lnor", GPU_COMP_I10, 3, GPU_FETCH_INT_TO_FLOAT_UNIT);
 | 
						|
	}
 | 
						|
	if (r_lnor_id) {
 | 
						|
		*r_vnor_id = vnor_loop_id;
 | 
						|
		*r_lnor_id = lnor_id;
 | 
						|
		return &format_nor_loop;
 | 
						|
	}
 | 
						|
	else {
 | 
						|
		*r_vnor_id = vnor_id;
 | 
						|
		return &format_nor;
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static GPUVertFormat *edit_mesh_overlay_data_format(uint *r_data_id)
 | 
						|
{
 | 
						|
	static GPUVertFormat format_flag = { 0 };
 | 
						|
	static uint data_id;
 | 
						|
	if (format_flag.attr_len == 0) {
 | 
						|
		data_id = GPU_vertformat_attr_add(&format_flag, "data", GPU_COMP_U8, 4, GPU_FETCH_INT);
 | 
						|
	}
 | 
						|
	*r_data_id = data_id;
 | 
						|
	return &format_flag;
 | 
						|
}
 | 
						|
 | 
						|
static void mesh_batch_cache_create_overlay_tri_buffers(
 | 
						|
        MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & (MR_DATATYPE_VERT | MR_DATATYPE_LOOPTRI));
 | 
						|
 | 
						|
	const int tri_len = mesh_render_data_looptri_len_get(rdata);
 | 
						|
 | 
						|
	const int vbo_len_capacity = tri_len * 3;
 | 
						|
	int vbo_len_used = 0;
 | 
						|
 | 
						|
	/* Positions */
 | 
						|
	GPUVertBuf *vbo_pos = NULL;
 | 
						|
	static struct { uint pos, vnor, lnor, data; } attr_id;
 | 
						|
	if (cache->ed_tri_pos == NULL) {
 | 
						|
		vbo_pos = cache->ed_tri_pos =
 | 
						|
		        GPU_vertbuf_create_with_format(edit_mesh_overlay_pos_format(&attr_id.pos));
 | 
						|
		GPU_vertbuf_data_alloc(vbo_pos, vbo_len_capacity);
 | 
						|
	}
 | 
						|
 | 
						|
	/* Normals */
 | 
						|
	GPUVertBuf *vbo_nor = NULL;
 | 
						|
	if (cache->ed_tri_nor == NULL) {
 | 
						|
		vbo_nor = cache->ed_tri_nor =
 | 
						|
		        GPU_vertbuf_create_with_format(edit_mesh_overlay_nor_format(&attr_id.vnor, &attr_id.lnor));
 | 
						|
		GPU_vertbuf_data_alloc(vbo_nor, vbo_len_capacity);
 | 
						|
	}
 | 
						|
 | 
						|
	/* Data */
 | 
						|
	GPUVertBuf *vbo_data = NULL;
 | 
						|
	if (cache->ed_tri_data == NULL) {
 | 
						|
		vbo_data = cache->ed_tri_data =
 | 
						|
		        GPU_vertbuf_create_with_format(edit_mesh_overlay_data_format(&attr_id.data));
 | 
						|
		GPU_vertbuf_data_alloc(vbo_data, vbo_len_capacity);
 | 
						|
	}
 | 
						|
 | 
						|
	for (int i = 0; i < tri_len; i++) {
 | 
						|
		const BMLoop **bm_looptri = (const BMLoop **)rdata->edit_bmesh->looptris[i];
 | 
						|
		if (!BM_elem_flag_test(bm_looptri[0]->f, BM_ELEM_HIDDEN)) {
 | 
						|
			add_overlay_tri(
 | 
						|
			        rdata, vbo_pos, vbo_nor, vbo_data,
 | 
						|
			        attr_id.pos, attr_id.vnor, attr_id.lnor, attr_id.data,
 | 
						|
			        bm_looptri, vbo_len_used);
 | 
						|
 | 
						|
			vbo_len_used += 3;
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	/* Finish */
 | 
						|
	if (vbo_len_used != vbo_len_capacity) {
 | 
						|
		if (vbo_pos != NULL) {
 | 
						|
			GPU_vertbuf_data_resize(vbo_pos, vbo_len_used);
 | 
						|
		}
 | 
						|
		if (vbo_nor != NULL) {
 | 
						|
			GPU_vertbuf_data_resize(vbo_nor, vbo_len_used);
 | 
						|
		}
 | 
						|
		if (vbo_data != NULL) {
 | 
						|
			GPU_vertbuf_data_resize(vbo_data, vbo_len_used);
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static void mesh_batch_cache_create_overlay_ledge_buffers(
 | 
						|
        MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & (MR_DATATYPE_VERT | MR_DATATYPE_LOOPTRI));
 | 
						|
 | 
						|
	const int ledge_len = mesh_render_data_loose_edges_len_get(rdata);
 | 
						|
 | 
						|
	const int vbo_len_capacity = ledge_len * 2;
 | 
						|
	int vbo_len_used = 0;
 | 
						|
 | 
						|
	/* Positions */
 | 
						|
	GPUVertBuf *vbo_pos = NULL;
 | 
						|
	static struct { uint pos, vnor, data; } attr_id;
 | 
						|
	if (cache->ed_ledge_pos == NULL) {
 | 
						|
		vbo_pos = cache->ed_ledge_pos =
 | 
						|
		        GPU_vertbuf_create_with_format(edit_mesh_overlay_pos_format(&attr_id.pos));
 | 
						|
		GPU_vertbuf_data_alloc(vbo_pos, vbo_len_capacity);
 | 
						|
	}
 | 
						|
 | 
						|
	/* Normals */
 | 
						|
	GPUVertBuf *vbo_nor = NULL;
 | 
						|
	if (cache->ed_ledge_nor == NULL) {
 | 
						|
		vbo_nor = cache->ed_ledge_nor =
 | 
						|
		        GPU_vertbuf_create_with_format(edit_mesh_overlay_nor_format(&attr_id.vnor, NULL));
 | 
						|
		GPU_vertbuf_data_alloc(vbo_nor, vbo_len_capacity);
 | 
						|
	}
 | 
						|
 | 
						|
	/* Data */
 | 
						|
	GPUVertBuf *vbo_data = NULL;
 | 
						|
	if (cache->ed_ledge_data == NULL) {
 | 
						|
		vbo_data = cache->ed_ledge_data =
 | 
						|
		        GPU_vertbuf_create_with_format(edit_mesh_overlay_data_format(&attr_id.data));
 | 
						|
		GPU_vertbuf_data_alloc(vbo_data, vbo_len_capacity);
 | 
						|
	}
 | 
						|
 | 
						|
	if (rdata->edit_bmesh) {
 | 
						|
		BMesh *bm = rdata->edit_bmesh->bm;
 | 
						|
		for (uint i = 0; i < ledge_len; i++) {
 | 
						|
			const BMEdge *eed = BM_edge_at_index(bm, rdata->loose_edges[i]);
 | 
						|
			if (!BM_elem_flag_test(eed, BM_ELEM_HIDDEN)) {
 | 
						|
				add_overlay_loose_edge(
 | 
						|
				        rdata, vbo_pos, vbo_nor, vbo_data,
 | 
						|
				        attr_id.pos, attr_id.vnor, attr_id.data,
 | 
						|
				        eed, vbo_len_used);
 | 
						|
				vbo_len_used += 2;
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	/* Finish */
 | 
						|
	if (vbo_len_used != vbo_len_capacity) {
 | 
						|
		if (vbo_pos != NULL) {
 | 
						|
			GPU_vertbuf_data_resize(vbo_pos, vbo_len_used);
 | 
						|
		}
 | 
						|
		if (vbo_nor != NULL) {
 | 
						|
			GPU_vertbuf_data_resize(vbo_nor, vbo_len_used);
 | 
						|
		}
 | 
						|
		if (vbo_data != NULL) {
 | 
						|
			GPU_vertbuf_data_resize(vbo_data, vbo_len_used);
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
static void mesh_batch_cache_create_overlay_lvert_buffers(
 | 
						|
        MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & (MR_DATATYPE_VERT | MR_DATATYPE_LOOPTRI));
 | 
						|
 | 
						|
	BMesh *bm = rdata->edit_bmesh->bm;
 | 
						|
	const int lvert_len = mesh_render_data_loose_verts_len_get(rdata);
 | 
						|
 | 
						|
	const int vbo_len_capacity = lvert_len;
 | 
						|
	int vbo_len_used = 0;
 | 
						|
 | 
						|
	static struct { uint pos, vnor, data; } attr_id;
 | 
						|
 | 
						|
	/* Positions */
 | 
						|
	GPUVertBuf *vbo_pos = NULL;
 | 
						|
	if (cache->ed_lvert_pos == NULL) {
 | 
						|
		vbo_pos = cache->ed_lvert_pos =
 | 
						|
		        GPU_vertbuf_create_with_format(edit_mesh_overlay_pos_format(&attr_id.pos));
 | 
						|
		GPU_vertbuf_data_alloc(vbo_pos, vbo_len_capacity);
 | 
						|
	}
 | 
						|
 | 
						|
	/* Normals */
 | 
						|
	GPUVertBuf *vbo_nor = NULL;
 | 
						|
	if (cache->ed_lvert_nor == NULL) {
 | 
						|
		vbo_nor = cache->ed_lvert_nor =
 | 
						|
		        GPU_vertbuf_create_with_format(edit_mesh_overlay_nor_format(&attr_id.vnor, NULL));
 | 
						|
		GPU_vertbuf_data_alloc(vbo_nor, vbo_len_capacity);
 | 
						|
	}
 | 
						|
 | 
						|
	/* Data */
 | 
						|
	GPUVertBuf *vbo_data = NULL;
 | 
						|
	if (cache->ed_lvert_data == NULL) {
 | 
						|
		vbo_data = cache->ed_lvert_data =
 | 
						|
		        GPU_vertbuf_create_with_format(edit_mesh_overlay_data_format(&attr_id.data));
 | 
						|
		GPU_vertbuf_data_alloc(vbo_data, vbo_len_capacity);
 | 
						|
	}
 | 
						|
 | 
						|
	for (uint i = 0; i < lvert_len; i++) {
 | 
						|
		BMVert *eve = BM_vert_at_index(bm, rdata->loose_verts[i]);
 | 
						|
		add_overlay_loose_vert(
 | 
						|
		        rdata, vbo_pos, vbo_nor, vbo_data,
 | 
						|
		        attr_id.pos, attr_id.vnor, attr_id.data,
 | 
						|
		        eve, vbo_len_used);
 | 
						|
		vbo_len_used += 1;
 | 
						|
	}
 | 
						|
 | 
						|
	/* Finish */
 | 
						|
	if (vbo_len_used != vbo_len_capacity) {
 | 
						|
		if (vbo_pos != NULL) {
 | 
						|
			GPU_vertbuf_data_resize(vbo_pos, vbo_len_used);
 | 
						|
		}
 | 
						|
		if (vbo_nor != NULL) {
 | 
						|
			GPU_vertbuf_data_resize(vbo_nor, vbo_len_used);
 | 
						|
		}
 | 
						|
		if (vbo_data != NULL) {
 | 
						|
			GPU_vertbuf_data_resize(vbo_data, vbo_len_used);
 | 
						|
		}
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/* Position */
 | 
						|
static GPUVertBuf *mesh_batch_cache_get_edit_tri_pos(
 | 
						|
        MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & MR_DATATYPE_VERT);
 | 
						|
 | 
						|
	if (cache->ed_tri_pos == NULL) {
 | 
						|
		mesh_batch_cache_create_overlay_tri_buffers(rdata, cache);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->ed_tri_pos;
 | 
						|
}
 | 
						|
 | 
						|
static GPUVertBuf *mesh_batch_cache_get_edit_ledge_pos(
 | 
						|
        MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & MR_DATATYPE_VERT);
 | 
						|
 | 
						|
	if (cache->ed_ledge_pos == NULL) {
 | 
						|
		mesh_batch_cache_create_overlay_ledge_buffers(rdata, cache);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->ed_ledge_pos;
 | 
						|
}
 | 
						|
 | 
						|
static GPUVertBuf *mesh_batch_cache_get_edit_lvert_pos(
 | 
						|
        MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & MR_DATATYPE_VERT);
 | 
						|
 | 
						|
	if (cache->ed_lvert_pos == NULL) {
 | 
						|
		mesh_batch_cache_create_overlay_lvert_buffers(rdata, cache);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->ed_lvert_pos;
 | 
						|
}
 | 
						|
 | 
						|
/* Normal */
 | 
						|
static GPUVertBuf *mesh_batch_cache_get_edit_tri_nor(
 | 
						|
        MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & MR_DATATYPE_VERT);
 | 
						|
 | 
						|
	if (cache->ed_tri_nor == NULL) {
 | 
						|
		mesh_batch_cache_create_overlay_tri_buffers(rdata, cache);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->ed_tri_nor;
 | 
						|
}
 | 
						|
 | 
						|
static GPUVertBuf *mesh_batch_cache_get_edit_ledge_nor(
 | 
						|
        MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & MR_DATATYPE_VERT);
 | 
						|
 | 
						|
	if (cache->ed_ledge_nor == NULL) {
 | 
						|
		mesh_batch_cache_create_overlay_ledge_buffers(rdata, cache);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->ed_ledge_nor;
 | 
						|
}
 | 
						|
 | 
						|
static GPUVertBuf *mesh_batch_cache_get_edit_lvert_nor(
 | 
						|
        MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & MR_DATATYPE_VERT);
 | 
						|
 | 
						|
	if (cache->ed_lvert_nor == NULL) {
 | 
						|
		mesh_batch_cache_create_overlay_lvert_buffers(rdata, cache);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->ed_lvert_nor;
 | 
						|
}
 | 
						|
 | 
						|
/* Data */
 | 
						|
static GPUVertBuf *mesh_batch_cache_get_edit_tri_data(
 | 
						|
        MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & MR_DATATYPE_VERT);
 | 
						|
 | 
						|
	if (cache->ed_tri_data == NULL) {
 | 
						|
		mesh_batch_cache_create_overlay_tri_buffers(rdata, cache);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->ed_tri_data;
 | 
						|
}
 | 
						|
 | 
						|
static GPUVertBuf *mesh_batch_cache_get_edit_ledge_data(
 | 
						|
        MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & MR_DATATYPE_VERT);
 | 
						|
 | 
						|
	if (cache->ed_ledge_data == NULL) {
 | 
						|
		mesh_batch_cache_create_overlay_ledge_buffers(rdata, cache);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->ed_ledge_data;
 | 
						|
}
 | 
						|
 | 
						|
static GPUVertBuf *mesh_batch_cache_get_edit_lvert_data(
 | 
						|
        MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & MR_DATATYPE_VERT);
 | 
						|
 | 
						|
	if (cache->ed_lvert_data == NULL) {
 | 
						|
		mesh_batch_cache_create_overlay_lvert_buffers(rdata, cache);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->ed_lvert_data;
 | 
						|
}
 | 
						|
 | 
						|
static GPUIndexBuf *mesh_batch_cache_get_edges_in_order(MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & (MR_DATATYPE_VERT | MR_DATATYPE_EDGE));
 | 
						|
 | 
						|
	if (cache->edges_in_order == NULL) {
 | 
						|
		const int vert_len = mesh_render_data_verts_len_get(rdata);
 | 
						|
		const int edge_len = mesh_render_data_edges_len_get(rdata);
 | 
						|
 | 
						|
		GPUIndexBufBuilder elb;
 | 
						|
		GPU_indexbuf_init(&elb, GPU_PRIM_LINES, edge_len, vert_len);
 | 
						|
 | 
						|
		BLI_assert(rdata->types & MR_DATATYPE_EDGE);
 | 
						|
 | 
						|
		if (rdata->edit_bmesh) {
 | 
						|
			BMesh *bm = rdata->edit_bmesh->bm;
 | 
						|
			BMIter eiter;
 | 
						|
			BMEdge *eed;
 | 
						|
			BM_ITER_MESH(eed, &eiter, bm, BM_EDGES_OF_MESH) {
 | 
						|
				if (!BM_elem_flag_test(eed, BM_ELEM_HIDDEN)) {
 | 
						|
					GPU_indexbuf_add_line_verts(&elb, BM_elem_index_get(eed->v1),  BM_elem_index_get(eed->v2));
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			const MEdge *ed = rdata->medge;
 | 
						|
			for (int i = 0; i < edge_len; i++, ed++) {
 | 
						|
				GPU_indexbuf_add_line_verts(&elb, ed->v1, ed->v2);
 | 
						|
			}
 | 
						|
		}
 | 
						|
		cache->edges_in_order = GPU_indexbuf_build(&elb);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->edges_in_order;
 | 
						|
}
 | 
						|
 | 
						|
#define NO_EDGE INT_MAX
 | 
						|
static GPUIndexBuf *mesh_batch_cache_get_edges_adjacency(MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & (MR_DATATYPE_VERT | MR_DATATYPE_LOOP | MR_DATATYPE_LOOPTRI));
 | 
						|
 | 
						|
	if (cache->edges_adjacency == NULL) {
 | 
						|
		const int vert_len = mesh_render_data_verts_len_get(rdata);
 | 
						|
		const int tri_len = mesh_render_data_looptri_len_get(rdata);
 | 
						|
 | 
						|
		cache->is_manifold = true;
 | 
						|
 | 
						|
		/* Allocate max but only used indices are sent to GPU. */
 | 
						|
		GPUIndexBufBuilder elb;
 | 
						|
		GPU_indexbuf_init(&elb, GPU_PRIM_LINES_ADJ, tri_len * 3, vert_len);
 | 
						|
 | 
						|
		EdgeHash *eh = BLI_edgehash_new_ex(__func__, tri_len * 3);
 | 
						|
		/* Create edges for each pair of triangles sharing an edge. */
 | 
						|
		for (int i = 0; i < tri_len; i++) {
 | 
						|
			for (int e = 0; e < 3; ++e) {
 | 
						|
				uint v0, v1, v2;
 | 
						|
				if (rdata->edit_bmesh) {
 | 
						|
					const BMLoop **bm_looptri = (const BMLoop **)rdata->edit_bmesh->looptris[i];
 | 
						|
					if (BM_elem_flag_test(bm_looptri[0]->f, BM_ELEM_HIDDEN)) {
 | 
						|
						break;
 | 
						|
					}
 | 
						|
					v0 = BM_elem_index_get(bm_looptri[e]->v);
 | 
						|
					v1 = BM_elem_index_get(bm_looptri[(e + 1) % 3]->v);
 | 
						|
					v2 = BM_elem_index_get(bm_looptri[(e + 2) % 3]->v);
 | 
						|
				}
 | 
						|
				else {
 | 
						|
					MLoop *mloop = rdata->mloop;
 | 
						|
					MLoopTri *mlt = rdata->mlooptri + i;
 | 
						|
					v0 = mloop[mlt->tri[e]].v;
 | 
						|
					v1 = mloop[mlt->tri[(e + 1) % 3]].v;
 | 
						|
					v2 = mloop[mlt->tri[(e + 2) % 3]].v;
 | 
						|
				}
 | 
						|
				bool inv_indices = (v1 > v2);
 | 
						|
				void **pval;
 | 
						|
				bool value_is_init = BLI_edgehash_ensure_p(eh, v1, v2, &pval);
 | 
						|
				int v_data = POINTER_AS_INT(*pval);
 | 
						|
				if (!value_is_init || v_data == NO_EDGE) {
 | 
						|
					/* Save the winding order inside the sign bit. Because the
 | 
						|
					 * edgehash sort the keys and we need to compare winding later. */
 | 
						|
					int value = (int)v0 + 1; /* Int 0 cannot be signed */
 | 
						|
					*pval = POINTER_FROM_INT((inv_indices) ? -value : value);
 | 
						|
				}
 | 
						|
				else {
 | 
						|
					/* HACK Tag as not used. Prevent overhead of BLI_edgehash_remove. */
 | 
						|
					*pval = POINTER_FROM_INT(NO_EDGE);
 | 
						|
					bool inv_opposite = (v_data < 0);
 | 
						|
					uint v_opposite = (uint)abs(v_data) - 1;
 | 
						|
 | 
						|
					if (inv_opposite == inv_indices) {
 | 
						|
						/* Don't share edge if triangles have non matching winding. */
 | 
						|
						GPU_indexbuf_add_line_adj_verts(&elb, v0, v1, v2, v0);
 | 
						|
						GPU_indexbuf_add_line_adj_verts(&elb, v_opposite, v1, v2, v_opposite);
 | 
						|
						cache->is_manifold = false;
 | 
						|
					}
 | 
						|
					else {
 | 
						|
						GPU_indexbuf_add_line_adj_verts(&elb, v0, v1, v2, v_opposite);
 | 
						|
					}
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		/* Create edges for remaning non manifold edges. */
 | 
						|
		EdgeHashIterator *ehi;
 | 
						|
		for (ehi = BLI_edgehashIterator_new(eh);
 | 
						|
		     BLI_edgehashIterator_isDone(ehi) == false;
 | 
						|
		     BLI_edgehashIterator_step(ehi))
 | 
						|
		{
 | 
						|
			uint v1, v2;
 | 
						|
			int v_data = POINTER_AS_INT(BLI_edgehashIterator_getValue(ehi));
 | 
						|
			if (v_data == NO_EDGE) {
 | 
						|
				continue;
 | 
						|
			}
 | 
						|
			BLI_edgehashIterator_getKey(ehi, &v1, &v2);
 | 
						|
			uint v0 = (uint)abs(v_data) - 1;
 | 
						|
			if (v_data < 0) { /* inv_opposite  */
 | 
						|
				SWAP(uint, v1, v2);
 | 
						|
			}
 | 
						|
			GPU_indexbuf_add_line_adj_verts(&elb, v0, v1, v2, v0);
 | 
						|
			cache->is_manifold = false;
 | 
						|
		}
 | 
						|
		BLI_edgehashIterator_free(ehi);
 | 
						|
		BLI_edgehash_free(eh, NULL);
 | 
						|
 | 
						|
		cache->edges_adjacency = GPU_indexbuf_build(&elb);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->edges_adjacency;
 | 
						|
}
 | 
						|
#undef NO_EDGE
 | 
						|
 | 
						|
static EdgeHash *create_looptri_edge_adjacency_hash(MeshRenderData *rdata)
 | 
						|
{
 | 
						|
	const int tri_len = mesh_render_data_looptri_len_get(rdata);
 | 
						|
	/* Create adjacency info in looptri */
 | 
						|
	EdgeHash *eh = BLI_edgehash_new_ex(__func__, tri_len * 3);
 | 
						|
	/* Create edges for each pair of triangles sharing an edge. */
 | 
						|
	for (int i = 0; i < tri_len; i++) {
 | 
						|
		for (int e = 0; e < 3; ++e) {
 | 
						|
			uint v0, v1, v2;
 | 
						|
			if (rdata->edit_bmesh) {
 | 
						|
				const BMLoop **bm_looptri = (const BMLoop **)rdata->edit_bmesh->looptris[i];
 | 
						|
				if (BM_elem_flag_test(bm_looptri[0]->f, BM_ELEM_HIDDEN)) {
 | 
						|
					break;
 | 
						|
				}
 | 
						|
				v0 = BM_elem_index_get(bm_looptri[e]->v);
 | 
						|
				v1 = BM_elem_index_get(bm_looptri[(e + 1) % 3]->v);
 | 
						|
				v2 = BM_elem_index_get(bm_looptri[(e + 2) % 3]->v);
 | 
						|
			}
 | 
						|
			else {
 | 
						|
				MLoop *mloop = rdata->mloop;
 | 
						|
				MLoopTri *mlt = rdata->mlooptri + i;
 | 
						|
				v0 = mloop[mlt->tri[e]].v;
 | 
						|
				v1 = mloop[mlt->tri[(e + 1) % 3]].v;
 | 
						|
				v2 = mloop[mlt->tri[(e + 2) % 3]].v;
 | 
						|
			}
 | 
						|
 | 
						|
			EdgeAdjacentVerts **eav;
 | 
						|
			bool value_is_init = BLI_edgehash_ensure_p(eh, v1, v2, (void ***)&eav);
 | 
						|
			if (!value_is_init) {
 | 
						|
				*eav = MEM_mallocN(sizeof(**eav), "EdgeAdjacentVerts");
 | 
						|
				(*eav)->vert_index[0] = v0;
 | 
						|
				(*eav)->vert_index[1] = -1;
 | 
						|
			}
 | 
						|
			else {
 | 
						|
				if ((*eav)->vert_index[1] == -1) {
 | 
						|
					(*eav)->vert_index[1] = v0;
 | 
						|
				}
 | 
						|
				else {
 | 
						|
					/* Not a manifold edge. */
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
	}
 | 
						|
	return eh;
 | 
						|
}
 | 
						|
 | 
						|
static GPUVertBuf *mesh_batch_cache_create_edges_overlay_texture_buf(MeshRenderData *rdata)
 | 
						|
{
 | 
						|
	const int tri_len = mesh_render_data_looptri_len_get(rdata);
 | 
						|
 | 
						|
	GPUVertFormat format = {0};
 | 
						|
	uint index_id = GPU_vertformat_attr_add(&format, "index", GPU_COMP_U32, 1, GPU_FETCH_INT);
 | 
						|
	GPUVertBuf *vbo = GPU_vertbuf_create_with_format(&format);
 | 
						|
 | 
						|
	int vbo_len_capacity = tri_len * 3;
 | 
						|
	GPU_vertbuf_data_alloc(vbo, vbo_len_capacity);
 | 
						|
 | 
						|
	int vidx = 0;
 | 
						|
	EdgeHash *eh = NULL;
 | 
						|
	eh = create_looptri_edge_adjacency_hash(rdata);
 | 
						|
 | 
						|
	for (int i = 0; i < tri_len; i++) {
 | 
						|
		bool edge_is_real[3] = {false, false, false};
 | 
						|
 | 
						|
		MEdge *medge = rdata->medge;
 | 
						|
		MLoop *mloop = rdata->mloop;
 | 
						|
		MLoopTri *mlt = rdata->mlooptri + i;
 | 
						|
 | 
						|
		int j, j_next;
 | 
						|
		for (j = 2, j_next = 0; j_next < 3; j = j_next++) {
 | 
						|
			MEdge *ed = &medge[mloop[mlt->tri[j]].e];
 | 
						|
			uint tri_edge[2]  = {mloop[mlt->tri[j]].v, mloop[mlt->tri[j_next]].v};
 | 
						|
 | 
						|
			if (((ed->v1 == tri_edge[0]) && (ed->v2 == tri_edge[1])) ||
 | 
						|
			    ((ed->v1 == tri_edge[1]) && (ed->v2 == tri_edge[0])))
 | 
						|
			{
 | 
						|
				edge_is_real[j] = true;
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		for (int e = 0; e < 3; ++e) {
 | 
						|
			int v0 = mloop[mlt->tri[e]].v;
 | 
						|
			int v1 = mloop[mlt->tri[(e + 1) % 3]].v;
 | 
						|
			EdgeAdjacentVerts *eav = BLI_edgehash_lookup(eh, v0, v1);
 | 
						|
			uint value = (uint)v0;
 | 
						|
			/* Real edge */
 | 
						|
			if (edge_is_real[e]) {
 | 
						|
				value |= (1 << 30);
 | 
						|
			}
 | 
						|
			/* Non-manifold edge */
 | 
						|
			if (eav->vert_index[1] == -1) {
 | 
						|
				value |= (1 << 31);
 | 
						|
			}
 | 
						|
			GPU_vertbuf_attr_set(vbo, index_id, vidx++, &value);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	BLI_edgehash_free(eh, MEM_freeN);
 | 
						|
 | 
						|
	int vbo_len_used = vidx;
 | 
						|
 | 
						|
	if (vbo_len_capacity != vbo_len_used) {
 | 
						|
		GPU_vertbuf_data_resize(vbo, vbo_len_used);
 | 
						|
	}
 | 
						|
 | 
						|
	return vbo;
 | 
						|
}
 | 
						|
 | 
						|
static GPUTexture *mesh_batch_cache_get_edges_overlay_texture_buf(MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & (MR_DATATYPE_VERT | MR_DATATYPE_EDGE | MR_DATATYPE_LOOP | MR_DATATYPE_LOOPTRI));
 | 
						|
 | 
						|
	BLI_assert(rdata->edit_bmesh == NULL); /* Not supported in edit mode */
 | 
						|
 | 
						|
	if (cache->edges_face_overlay_tx != NULL) {
 | 
						|
		return cache->edges_face_overlay_tx;
 | 
						|
	}
 | 
						|
 | 
						|
	GPUVertBuf *vbo = cache->edges_face_overlay = mesh_batch_cache_create_edges_overlay_texture_buf(rdata);
 | 
						|
 | 
						|
	/* Upload data early because we need to create the texture for it. */
 | 
						|
	GPU_vertbuf_use(vbo);
 | 
						|
	cache->edges_face_overlay_tx = GPU_texture_create_from_vertbuf(vbo);
 | 
						|
	cache->edges_face_overlay_tri_count = vbo->vertex_alloc / 3;
 | 
						|
 | 
						|
	return cache->edges_face_overlay_tx;
 | 
						|
}
 | 
						|
 | 
						|
static GPUTexture *mesh_batch_cache_get_vert_pos_and_nor_in_order_buf(MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & (MR_DATATYPE_VERT | MR_DATATYPE_EDGE | MR_DATATYPE_LOOP | MR_DATATYPE_LOOPTRI));
 | 
						|
 | 
						|
	if (cache->pos_in_order_tx == NULL) {
 | 
						|
		GPUVertBuf *pos_in_order = mesh_batch_cache_get_vert_pos_and_nor_in_order(rdata, cache);
 | 
						|
		GPU_vertbuf_use(pos_in_order); /* Upload early for buffer texture creation. */
 | 
						|
		cache->pos_in_order_tx = GPU_texture_create_buffer(GPU_R32F, pos_in_order->vbo_id);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->pos_in_order_tx;
 | 
						|
}
 | 
						|
 | 
						|
static GPUIndexBuf *mesh_batch_cache_get_triangles_in_order(MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & (MR_DATATYPE_VERT | MR_DATATYPE_LOOPTRI));
 | 
						|
 | 
						|
	if (cache->triangles_in_order == NULL) {
 | 
						|
		const int vert_len = mesh_render_data_verts_len_get(rdata);
 | 
						|
		const int tri_len = mesh_render_data_looptri_len_get(rdata);
 | 
						|
 | 
						|
		GPUIndexBufBuilder elb;
 | 
						|
		GPU_indexbuf_init(&elb, GPU_PRIM_TRIS, tri_len, vert_len);
 | 
						|
 | 
						|
		if (rdata->edit_bmesh) {
 | 
						|
			for (int i = 0; i < tri_len; ++i) {
 | 
						|
				const BMLoop **ltri = (const BMLoop **)rdata->edit_bmesh->looptris[i];
 | 
						|
				if (!BM_elem_flag_test(ltri[0]->f, BM_ELEM_HIDDEN)) {
 | 
						|
					for (uint tri_corner = 0; tri_corner < 3; tri_corner++) {
 | 
						|
						GPU_indexbuf_add_generic_vert(&elb, BM_elem_index_get(ltri[tri_corner]->v));
 | 
						|
					}
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			for (int i = 0; i < tri_len; ++i) {
 | 
						|
				const MLoopTri *mlt = &rdata->mlooptri[i];
 | 
						|
				for (uint tri_corner = 0; tri_corner < 3; tri_corner++) {
 | 
						|
					GPU_indexbuf_add_generic_vert(&elb, mlt->tri[tri_corner]);
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		cache->triangles_in_order = GPU_indexbuf_build(&elb);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->triangles_in_order;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
static GPUIndexBuf *mesh_batch_cache_get_loose_edges(MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & (MR_DATATYPE_VERT | MR_DATATYPE_LOOPTRI));
 | 
						|
 | 
						|
	if (cache->ledges_in_order == NULL) {
 | 
						|
		const int vert_len = mesh_render_data_verts_len_get(rdata);
 | 
						|
		const int edge_len = mesh_render_data_edges_len_get(rdata);
 | 
						|
 | 
						|
		/* Alloc max (edge_len) and upload only needed range. */
 | 
						|
		GPUIndexBufBuilder elb;
 | 
						|
		GPU_indexbuf_init(&elb, GPU_PRIM_LINES, edge_len, vert_len);
 | 
						|
 | 
						|
		if (rdata->edit_bmesh) {
 | 
						|
			/* No need to support since edit mesh already draw them.
 | 
						|
			 * But some engines may want them ... */
 | 
						|
			BMesh *bm = rdata->edit_bmesh->bm;
 | 
						|
			BMIter eiter;
 | 
						|
			BMEdge *eed;
 | 
						|
			BM_ITER_MESH(eed, &eiter, bm, BM_EDGES_OF_MESH) {
 | 
						|
				if (!BM_elem_flag_test(eed, BM_ELEM_HIDDEN) && BM_edge_is_wire(eed)) {
 | 
						|
					GPU_indexbuf_add_line_verts(&elb, BM_elem_index_get(eed->v1),  BM_elem_index_get(eed->v2));
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			for (int i = 0; i < edge_len; ++i) {
 | 
						|
				const MEdge *medge = &rdata->medge[i];
 | 
						|
				if (medge->flag & ME_LOOSEEDGE) {
 | 
						|
					GPU_indexbuf_add_line_verts(&elb, medge->v1, medge->v2);
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		cache->ledges_in_order = GPU_indexbuf_build(&elb);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->ledges_in_order;
 | 
						|
}
 | 
						|
 | 
						|
static GPUIndexBuf **mesh_batch_cache_get_triangles_in_order_split_by_material(
 | 
						|
        MeshRenderData *rdata, MeshBatchCache *cache)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & (MR_DATATYPE_VERT | MR_DATATYPE_POLY));
 | 
						|
 | 
						|
	if (cache->shaded_triangles_in_order == NULL) {
 | 
						|
		const int poly_len = mesh_render_data_polys_len_get(rdata);
 | 
						|
		const int tri_len = mesh_render_data_looptri_len_get(rdata);
 | 
						|
		const int mat_len = mesh_render_data_mat_len_get(rdata);
 | 
						|
 | 
						|
		int *mat_tri_len = MEM_callocN(sizeof(*mat_tri_len) * mat_len, __func__);
 | 
						|
		cache->shaded_triangles_in_order = MEM_callocN(sizeof(*cache->shaded_triangles) * mat_len, __func__);
 | 
						|
		GPUIndexBufBuilder *elb = MEM_callocN(sizeof(*elb) * mat_len, __func__);
 | 
						|
 | 
						|
		/* Note that polygons (not triangles) are used here.
 | 
						|
		 * This OK because result is _guaranteed_ to be the same. */
 | 
						|
		if (rdata->edit_bmesh) {
 | 
						|
			BMesh *bm = rdata->edit_bmesh->bm;
 | 
						|
			BMIter fiter;
 | 
						|
			BMFace *efa;
 | 
						|
 | 
						|
			BM_ITER_MESH(efa, &fiter, bm, BM_FACES_OF_MESH) {
 | 
						|
				if (!BM_elem_flag_test(efa, BM_ELEM_HIDDEN)) {
 | 
						|
					const short ma_id = efa->mat_nr < mat_len ? efa->mat_nr : 0;
 | 
						|
					mat_tri_len[ma_id] += (efa->len - 2);
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			for (uint i = 0; i < poly_len; i++) {
 | 
						|
				const MPoly *mp = &rdata->mpoly[i]; ;
 | 
						|
				const short ma_id = mp->mat_nr < mat_len ? mp->mat_nr : 0;
 | 
						|
				mat_tri_len[ma_id] += (mp->totloop - 2);
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		/* Init ELBs. */
 | 
						|
		for (int i = 0; i < mat_len; ++i) {
 | 
						|
			GPU_indexbuf_init(&elb[i], GPU_PRIM_TRIS, mat_tri_len[i], tri_len * 3);
 | 
						|
		}
 | 
						|
 | 
						|
		/* Populate ELBs. */
 | 
						|
		uint nidx = 0;
 | 
						|
		if (rdata->edit_bmesh) {
 | 
						|
			BMesh *bm = rdata->edit_bmesh->bm;
 | 
						|
			BMIter fiter;
 | 
						|
			BMFace *efa;
 | 
						|
 | 
						|
			BM_ITER_MESH(efa, &fiter, bm, BM_FACES_OF_MESH) {
 | 
						|
				if (!BM_elem_flag_test(efa, BM_ELEM_HIDDEN)) {
 | 
						|
					const short ma_id = efa->mat_nr < mat_len ? efa->mat_nr : 0;
 | 
						|
					for (int j = 2; j < efa->len; j++) {
 | 
						|
						GPU_indexbuf_add_tri_verts(&elb[ma_id], nidx + 0, nidx + 1, nidx + 2);
 | 
						|
						nidx += 3;
 | 
						|
					}
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		else {
 | 
						|
			for (uint i = 0; i < poly_len; i++) {
 | 
						|
				const MPoly *mp = &rdata->mpoly[i]; ;
 | 
						|
				const short ma_id = mp->mat_nr < mat_len ? mp->mat_nr : 0;
 | 
						|
				for (int j = 2; j < mp->totloop; j++) {
 | 
						|
					GPU_indexbuf_add_tri_verts(&elb[ma_id], nidx + 0, nidx + 1, nidx + 2);
 | 
						|
					nidx += 3;
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		/* Build ELBs. */
 | 
						|
		for (int i = 0; i < mat_len; ++i) {
 | 
						|
			cache->shaded_triangles_in_order[i] = GPU_indexbuf_build(&elb[i]);
 | 
						|
		}
 | 
						|
 | 
						|
		MEM_freeN(mat_tri_len);
 | 
						|
		MEM_freeN(elb);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->shaded_triangles_in_order;
 | 
						|
}
 | 
						|
 | 
						|
static GPUVertBuf *mesh_create_edge_pos_with_sel(
 | 
						|
        MeshRenderData *rdata, bool use_wire, bool use_select_bool)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & (MR_DATATYPE_VERT | MR_DATATYPE_EDGE | MR_DATATYPE_POLY | MR_DATATYPE_LOOP));
 | 
						|
	BLI_assert(rdata->edit_bmesh == NULL);
 | 
						|
 | 
						|
	GPUVertBuf *vbo;
 | 
						|
	{
 | 
						|
		uint vidx = 0, cidx = 0;
 | 
						|
 | 
						|
		static GPUVertFormat format = { 0 };
 | 
						|
		static struct { uint pos, sel; } attr_id;
 | 
						|
		if (format.attr_len == 0) {
 | 
						|
			attr_id.pos = GPU_vertformat_attr_add(&format, "pos", GPU_COMP_F32, 3, GPU_FETCH_FLOAT);
 | 
						|
			attr_id.sel = GPU_vertformat_attr_add(&format, "select", GPU_COMP_U8, 1, GPU_FETCH_INT);
 | 
						|
		}
 | 
						|
 | 
						|
		const int edge_len = mesh_render_data_edges_len_get(rdata);
 | 
						|
 | 
						|
		vbo = GPU_vertbuf_create_with_format(&format);
 | 
						|
 | 
						|
		const int vbo_len_capacity = edge_len * 2;
 | 
						|
		int vbo_len_used = 0;
 | 
						|
		GPU_vertbuf_data_alloc(vbo, vbo_len_capacity);
 | 
						|
 | 
						|
		if (use_select_bool) {
 | 
						|
			mesh_render_data_ensure_edge_select_bool(rdata, use_wire);
 | 
						|
		}
 | 
						|
		bool *edge_select_bool = use_select_bool ? rdata->edge_select_bool : NULL;
 | 
						|
 | 
						|
		for (int i = 0; i < edge_len; i++) {
 | 
						|
			const MEdge *ed = &rdata->medge[i];
 | 
						|
 | 
						|
			uchar edge_vert_sel;
 | 
						|
			if (use_select_bool && edge_select_bool[i]) {
 | 
						|
				edge_vert_sel = true;
 | 
						|
			}
 | 
						|
			else if (use_wire) {
 | 
						|
				edge_vert_sel = false;
 | 
						|
			}
 | 
						|
			else {
 | 
						|
				continue;
 | 
						|
			}
 | 
						|
 | 
						|
			GPU_vertbuf_attr_set(vbo, attr_id.sel, cidx++, &edge_vert_sel);
 | 
						|
			GPU_vertbuf_attr_set(vbo, attr_id.sel, cidx++, &edge_vert_sel);
 | 
						|
 | 
						|
			GPU_vertbuf_attr_set(vbo, attr_id.pos, vidx++, rdata->mvert[ed->v1].co);
 | 
						|
			GPU_vertbuf_attr_set(vbo, attr_id.pos, vidx++, rdata->mvert[ed->v2].co);
 | 
						|
		}
 | 
						|
		vbo_len_used = vidx;
 | 
						|
 | 
						|
		if (vbo_len_capacity != vbo_len_used) {
 | 
						|
			GPU_vertbuf_data_resize(vbo, vbo_len_used);
 | 
						|
		}
 | 
						|
	}
 | 
						|
 | 
						|
	return vbo;
 | 
						|
}
 | 
						|
 | 
						|
static GPUIndexBuf *mesh_create_tri_overlay_weight_faces(
 | 
						|
        MeshRenderData *rdata)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & (MR_DATATYPE_VERT | MR_DATATYPE_LOOPTRI));
 | 
						|
 | 
						|
	{
 | 
						|
		const int vert_len = mesh_render_data_verts_len_get(rdata);
 | 
						|
		const int tri_len = mesh_render_data_looptri_len_get(rdata);
 | 
						|
 | 
						|
		GPUIndexBufBuilder elb;
 | 
						|
		GPU_indexbuf_init(&elb, GPU_PRIM_TRIS, tri_len, vert_len);
 | 
						|
 | 
						|
		for (int i = 0; i < tri_len; ++i) {
 | 
						|
			const MLoopTri *mlt = &rdata->mlooptri[i];
 | 
						|
			if (!(rdata->mpoly[mlt->poly].flag & (ME_FACE_SEL | ME_HIDE))) {
 | 
						|
				for (uint tri_corner = 0; tri_corner < 3; tri_corner++) {
 | 
						|
					GPU_indexbuf_add_generic_vert(&elb, rdata->mloop[mlt->tri[tri_corner]].v);
 | 
						|
				}
 | 
						|
			}
 | 
						|
		}
 | 
						|
		return GPU_indexbuf_build(&elb);
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Non-edit mode vertices (only used for weight-paint mode).
 | 
						|
 */
 | 
						|
static GPUVertBuf *mesh_create_vert_pos_with_overlay_data(
 | 
						|
        MeshRenderData *rdata)
 | 
						|
{
 | 
						|
	BLI_assert(rdata->types & (MR_DATATYPE_VERT));
 | 
						|
	BLI_assert(rdata->edit_bmesh == NULL);
 | 
						|
 | 
						|
	GPUVertBuf *vbo;
 | 
						|
	{
 | 
						|
		uint cidx = 0;
 | 
						|
 | 
						|
		static GPUVertFormat format = { 0 };
 | 
						|
		static struct { uint data; } attr_id;
 | 
						|
		if (format.attr_len == 0) {
 | 
						|
			attr_id.data = GPU_vertformat_attr_add(&format, "data", GPU_COMP_I8, 1, GPU_FETCH_INT);
 | 
						|
		}
 | 
						|
 | 
						|
		const int vert_len = mesh_render_data_verts_len_get(rdata);
 | 
						|
 | 
						|
		vbo = GPU_vertbuf_create_with_format(&format);
 | 
						|
 | 
						|
		const int vbo_len_capacity = vert_len;
 | 
						|
		int vbo_len_used = 0;
 | 
						|
		GPU_vertbuf_data_alloc(vbo, vbo_len_capacity);
 | 
						|
 | 
						|
		for (int i = 0; i < vert_len; i++) {
 | 
						|
			const MVert *mv = &rdata->mvert[i];
 | 
						|
			const char data = mv->flag & (SELECT | ME_HIDE);
 | 
						|
			GPU_vertbuf_attr_set(vbo, attr_id.data, cidx++, &data);
 | 
						|
		}
 | 
						|
		vbo_len_used = cidx;
 | 
						|
 | 
						|
		if (vbo_len_capacity != vbo_len_used) {
 | 
						|
			GPU_vertbuf_data_resize(vbo, vbo_len_used);
 | 
						|
		}
 | 
						|
	}
 | 
						|
	return vbo;
 | 
						|
}
 | 
						|
 | 
						|
/** \} */
 | 
						|
 | 
						|
 | 
						|
/* ---------------------------------------------------------------------- */
 | 
						|
 | 
						|
/** \name Public API
 | 
						|
 * \{ */
 | 
						|
 | 
						|
GPUBatch *DRW_mesh_batch_cache_get_all_edges(Mesh *me)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->all_edges == NULL) {
 | 
						|
		/* create batch from Mesh */
 | 
						|
		const int datatype = MR_DATATYPE_VERT | MR_DATATYPE_EDGE;
 | 
						|
		MeshRenderData *rdata = mesh_render_data_create(me, datatype);
 | 
						|
 | 
						|
		cache->all_edges = GPU_batch_create(
 | 
						|
		         GPU_PRIM_LINES, mesh_batch_cache_get_vert_pos_and_nor_in_order(rdata, cache),
 | 
						|
		                         mesh_batch_cache_get_edges_in_order(rdata, cache));
 | 
						|
 | 
						|
		mesh_render_data_free(rdata);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->all_edges;
 | 
						|
}
 | 
						|
 | 
						|
GPUBatch *DRW_mesh_batch_cache_get_all_triangles(Mesh *me)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->all_triangles == NULL) {
 | 
						|
		/* create batch from DM */
 | 
						|
		const int datatype = MR_DATATYPE_VERT | MR_DATATYPE_LOOPTRI;
 | 
						|
		MeshRenderData *rdata = mesh_render_data_create(me, datatype);
 | 
						|
 | 
						|
		cache->all_triangles = GPU_batch_create(
 | 
						|
		        GPU_PRIM_TRIS, mesh_batch_cache_get_vert_pos_and_nor_in_order(rdata, cache),
 | 
						|
		        mesh_batch_cache_get_triangles_in_order(rdata, cache));
 | 
						|
 | 
						|
		mesh_render_data_free(rdata);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->all_triangles;
 | 
						|
}
 | 
						|
 | 
						|
GPUBatch *DRW_mesh_batch_cache_get_triangles_with_normals(Mesh *me)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->triangles_with_normals == NULL) {
 | 
						|
		const int datatype = MR_DATATYPE_VERT | MR_DATATYPE_LOOPTRI | MR_DATATYPE_LOOP | MR_DATATYPE_POLY;
 | 
						|
		MeshRenderData *rdata = mesh_render_data_create(me, datatype);
 | 
						|
 | 
						|
		cache->triangles_with_normals = GPU_batch_create(
 | 
						|
		        GPU_PRIM_TRIS, mesh_batch_cache_get_tri_pos_and_normals(rdata, cache), NULL);
 | 
						|
 | 
						|
		mesh_render_data_free(rdata);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->triangles_with_normals;
 | 
						|
}
 | 
						|
 | 
						|
GPUBatch *DRW_mesh_batch_cache_get_loose_edges_with_normals(Mesh *me)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->ledges_with_normals == NULL) {
 | 
						|
		const int datatype = MR_DATATYPE_VERT | MR_DATATYPE_EDGE | MR_DATATYPE_LOOPTRI | MR_DATATYPE_LOOP | MR_DATATYPE_POLY;
 | 
						|
		MeshRenderData *rdata = mesh_render_data_create(me, datatype);
 | 
						|
 | 
						|
		cache->ledges_with_normals = GPU_batch_create(
 | 
						|
		        GPU_PRIM_LINES, mesh_batch_cache_get_vert_pos_and_nor_in_order(rdata, cache),
 | 
						|
		                        mesh_batch_cache_get_loose_edges(rdata, cache));
 | 
						|
 | 
						|
		mesh_render_data_free(rdata);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->ledges_with_normals;
 | 
						|
}
 | 
						|
 | 
						|
GPUBatch *DRW_mesh_batch_cache_get_triangles_with_normals_and_weights(
 | 
						|
        Mesh *me, const struct DRW_MeshWeightState *wstate)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	mesh_batch_cache_check_vertex_group(cache, wstate);
 | 
						|
 | 
						|
	if (cache->triangles_with_weights == NULL) {
 | 
						|
		const bool use_hide = (me->editflag & (ME_EDIT_PAINT_VERT_SEL | ME_EDIT_PAINT_FACE_SEL)) != 0;
 | 
						|
		const int datatype =
 | 
						|
		        MR_DATATYPE_VERT | MR_DATATYPE_LOOPTRI | MR_DATATYPE_LOOP | MR_DATATYPE_POLY | MR_DATATYPE_DVERT;
 | 
						|
		MeshRenderData *rdata = mesh_render_data_create(me, datatype);
 | 
						|
 | 
						|
		cache->triangles_with_weights = GPU_batch_create_ex(
 | 
						|
		        GPU_PRIM_TRIS, mesh_create_tri_weights(rdata, use_hide, wstate), NULL, GPU_BATCH_OWNS_VBO);
 | 
						|
 | 
						|
		DRW_mesh_weight_state_copy(&cache->weight_state, wstate);
 | 
						|
 | 
						|
		GPUVertBuf *vbo_tris = use_hide ?
 | 
						|
		        mesh_create_tri_pos_and_normals_visible_only(rdata) :
 | 
						|
		        mesh_batch_cache_get_tri_pos_and_normals(rdata, cache);
 | 
						|
 | 
						|
		GPU_batch_vertbuf_add_ex(cache->triangles_with_weights, vbo_tris, use_hide);
 | 
						|
 | 
						|
		mesh_render_data_free(rdata);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->triangles_with_weights;
 | 
						|
}
 | 
						|
 | 
						|
GPUBatch *DRW_mesh_batch_cache_get_triangles_with_normals_and_vert_colors(Mesh *me)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->triangles_with_vert_colors == NULL) {
 | 
						|
		const bool use_hide = (me->editflag & (ME_EDIT_PAINT_VERT_SEL | ME_EDIT_PAINT_FACE_SEL)) != 0;
 | 
						|
		const int datatype =
 | 
						|
		        MR_DATATYPE_VERT | MR_DATATYPE_LOOPTRI | MR_DATATYPE_LOOP | MR_DATATYPE_POLY | MR_DATATYPE_LOOPCOL;
 | 
						|
		MeshRenderData *rdata = mesh_render_data_create(me, datatype);
 | 
						|
 | 
						|
		cache->triangles_with_vert_colors = GPU_batch_create_ex(
 | 
						|
		        GPU_PRIM_TRIS, mesh_create_tri_vert_colors(rdata, use_hide), NULL, GPU_BATCH_OWNS_VBO);
 | 
						|
 | 
						|
		GPUVertBuf *vbo_tris = use_hide ?
 | 
						|
		        mesh_create_tri_pos_and_normals_visible_only(rdata) :
 | 
						|
		        mesh_batch_cache_get_tri_pos_and_normals(rdata, cache);
 | 
						|
		GPU_batch_vertbuf_add_ex(cache->triangles_with_vert_colors, vbo_tris, use_hide);
 | 
						|
 | 
						|
		mesh_render_data_free(rdata);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->triangles_with_vert_colors;
 | 
						|
}
 | 
						|
 | 
						|
 | 
						|
struct GPUBatch *DRW_mesh_batch_cache_get_triangles_with_select_id(
 | 
						|
        struct Mesh *me, bool use_hide, uint select_id_offset)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->triangles_with_select_id_offset != select_id_offset) {
 | 
						|
		cache->triangles_with_select_id_offset = select_id_offset;
 | 
						|
		GPU_BATCH_DISCARD_SAFE(cache->triangles_with_select_id);
 | 
						|
	}
 | 
						|
 | 
						|
	if (cache->triangles_with_select_id == NULL) {
 | 
						|
		const int datatype =
 | 
						|
		        MR_DATATYPE_VERT | MR_DATATYPE_LOOPTRI | MR_DATATYPE_LOOP | MR_DATATYPE_POLY;
 | 
						|
		MeshRenderData *rdata = mesh_render_data_create(me, datatype);
 | 
						|
 | 
						|
		cache->triangles_with_select_id = GPU_batch_create_ex(
 | 
						|
		        GPU_PRIM_TRIS, mesh_create_tri_select_id(rdata, use_hide, select_id_offset), NULL, GPU_BATCH_OWNS_VBO);
 | 
						|
 | 
						|
		GPUVertBuf *vbo_tris = use_hide ?
 | 
						|
		        mesh_create_tri_pos_and_normals_visible_only(rdata) :
 | 
						|
		        mesh_batch_cache_get_tri_pos_and_normals(rdata, cache);
 | 
						|
		GPU_batch_vertbuf_add_ex(cache->triangles_with_select_id, vbo_tris, use_hide);
 | 
						|
 | 
						|
		mesh_render_data_free(rdata);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->triangles_with_select_id;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Same as #DRW_mesh_batch_cache_get_triangles_with_select_id
 | 
						|
 * without the ID's, use to mask out geometry, eg - dont select face-dots behind other faces.
 | 
						|
 */
 | 
						|
struct GPUBatch *DRW_mesh_batch_cache_get_triangles_with_select_mask(struct Mesh *me, bool use_hide)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
	if (cache->triangles_with_select_mask == NULL) {
 | 
						|
		const int datatype =
 | 
						|
		        MR_DATATYPE_VERT | MR_DATATYPE_LOOPTRI | MR_DATATYPE_LOOP | MR_DATATYPE_POLY;
 | 
						|
		MeshRenderData *rdata = mesh_render_data_create(me, datatype);
 | 
						|
 | 
						|
		GPUVertBuf *vbo_tris = use_hide ?
 | 
						|
		        mesh_create_tri_pos_and_normals_visible_only(rdata) :
 | 
						|
		        mesh_batch_cache_get_tri_pos_and_normals(rdata, cache);
 | 
						|
 | 
						|
		cache->triangles_with_select_mask = GPU_batch_create_ex(
 | 
						|
		        GPU_PRIM_TRIS, vbo_tris, NULL, use_hide ? GPU_BATCH_OWNS_VBO : 0);
 | 
						|
 | 
						|
		mesh_render_data_free(rdata);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->triangles_with_select_mask;
 | 
						|
}
 | 
						|
 | 
						|
GPUBatch *DRW_mesh_batch_cache_get_points_with_normals(Mesh *me)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->points_with_normals == NULL) {
 | 
						|
		const int datatype = MR_DATATYPE_VERT | MR_DATATYPE_LOOPTRI | MR_DATATYPE_LOOP | MR_DATATYPE_POLY;
 | 
						|
		MeshRenderData *rdata = mesh_render_data_create(me, datatype);
 | 
						|
 | 
						|
		cache->points_with_normals = GPU_batch_create(
 | 
						|
		        GPU_PRIM_POINTS, mesh_batch_cache_get_tri_pos_and_normals(rdata, cache), NULL);
 | 
						|
 | 
						|
		mesh_render_data_free(rdata);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->points_with_normals;
 | 
						|
}
 | 
						|
 | 
						|
GPUBatch *DRW_mesh_batch_cache_get_all_verts(Mesh *me)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->all_verts == NULL) {
 | 
						|
		/* create batch from DM */
 | 
						|
		MeshRenderData *rdata = mesh_render_data_create(me, MR_DATATYPE_VERT);
 | 
						|
 | 
						|
		cache->all_verts = GPU_batch_create(
 | 
						|
		        GPU_PRIM_POINTS, mesh_batch_cache_get_vert_pos_and_nor_in_order(rdata, cache), NULL);
 | 
						|
 | 
						|
		mesh_render_data_free(rdata);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->all_verts;
 | 
						|
}
 | 
						|
 | 
						|
GPUBatch *DRW_mesh_batch_cache_get_fancy_edges(Mesh *me)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->fancy_edges == NULL) {
 | 
						|
		/* create batch from DM */
 | 
						|
		static GPUVertFormat format = { 0 };
 | 
						|
		static struct { uint pos, n1, n2; } attr_id;
 | 
						|
		if (format.attr_len == 0) {
 | 
						|
			attr_id.pos = GPU_vertformat_attr_add(&format, "pos", GPU_COMP_F32, 3, GPU_FETCH_FLOAT);
 | 
						|
 | 
						|
			attr_id.n1 = GPU_vertformat_attr_add(&format, "N1", GPU_COMP_I10, 3, GPU_FETCH_INT_TO_FLOAT_UNIT);
 | 
						|
			attr_id.n2 = GPU_vertformat_attr_add(&format, "N2", GPU_COMP_I10, 3, GPU_FETCH_INT_TO_FLOAT_UNIT);
 | 
						|
		}
 | 
						|
		GPUVertBuf *vbo = GPU_vertbuf_create_with_format(&format);
 | 
						|
 | 
						|
		MeshRenderData *rdata = mesh_render_data_create(
 | 
						|
		        me, MR_DATATYPE_VERT | MR_DATATYPE_EDGE | MR_DATATYPE_LOOP | MR_DATATYPE_POLY);
 | 
						|
 | 
						|
		const int edge_len = mesh_render_data_edges_len_get(rdata);
 | 
						|
 | 
						|
		const int vbo_len_capacity = edge_len * 2; /* these are PRIM_LINE verts, not mesh verts */
 | 
						|
		int vbo_len_used = 0;
 | 
						|
		GPU_vertbuf_data_alloc(vbo, vbo_len_capacity);
 | 
						|
		for (int i = 0; i < edge_len; ++i) {
 | 
						|
			float *vcos1, *vcos2;
 | 
						|
			float *pnor1 = NULL, *pnor2 = NULL;
 | 
						|
			bool is_manifold;
 | 
						|
 | 
						|
			if (mesh_render_data_edge_vcos_manifold_pnors(rdata, i, &vcos1, &vcos2, &pnor1, &pnor2, &is_manifold)) {
 | 
						|
 | 
						|
				GPUPackedNormal n1value = { .x = 0, .y = 0, .z = +511 };
 | 
						|
				GPUPackedNormal n2value = { .x = 0, .y = 0, .z = -511 };
 | 
						|
 | 
						|
				if (is_manifold) {
 | 
						|
					n1value = GPU_normal_convert_i10_v3(pnor1);
 | 
						|
					n2value = GPU_normal_convert_i10_v3(pnor2);
 | 
						|
				}
 | 
						|
 | 
						|
				const GPUPackedNormal *n1 = &n1value;
 | 
						|
				const GPUPackedNormal *n2 = &n2value;
 | 
						|
 | 
						|
				GPU_vertbuf_attr_set(vbo, attr_id.pos, 2 * i, vcos1);
 | 
						|
				GPU_vertbuf_attr_set(vbo, attr_id.n1, 2 * i, n1);
 | 
						|
				GPU_vertbuf_attr_set(vbo, attr_id.n2, 2 * i, n2);
 | 
						|
 | 
						|
				GPU_vertbuf_attr_set(vbo, attr_id.pos, 2 * i + 1, vcos2);
 | 
						|
				GPU_vertbuf_attr_set(vbo, attr_id.n1, 2 * i + 1, n1);
 | 
						|
				GPU_vertbuf_attr_set(vbo, attr_id.n2, 2 * i + 1, n2);
 | 
						|
 | 
						|
				vbo_len_used += 2;
 | 
						|
			}
 | 
						|
		}
 | 
						|
		if (vbo_len_used != vbo_len_capacity) {
 | 
						|
			GPU_vertbuf_data_resize(vbo, vbo_len_used);
 | 
						|
		}
 | 
						|
 | 
						|
		cache->fancy_edges = GPU_batch_create_ex(GPU_PRIM_LINES, vbo, NULL, GPU_BATCH_OWNS_VBO);
 | 
						|
 | 
						|
		mesh_render_data_free(rdata);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->fancy_edges;
 | 
						|
}
 | 
						|
 | 
						|
GPUBatch *DRW_mesh_batch_cache_get_edge_detection(Mesh *me, bool *r_is_manifold)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->edge_detection == NULL) {
 | 
						|
		const int options = MR_DATATYPE_VERT | MR_DATATYPE_LOOP | MR_DATATYPE_LOOPTRI;
 | 
						|
 | 
						|
		MeshRenderData *rdata = mesh_render_data_create(me, options);
 | 
						|
 | 
						|
		cache->edge_detection = GPU_batch_create_ex(
 | 
						|
		        GPU_PRIM_LINES_ADJ, mesh_batch_cache_get_vert_pos_and_nor_in_order(rdata, cache),
 | 
						|
		        mesh_batch_cache_get_edges_adjacency(rdata, cache), 0);
 | 
						|
 | 
						|
		mesh_render_data_free(rdata);
 | 
						|
	}
 | 
						|
 | 
						|
	if (r_is_manifold) {
 | 
						|
		*r_is_manifold = cache->is_manifold;
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->edge_detection;
 | 
						|
}
 | 
						|
 | 
						|
void DRW_mesh_batch_cache_get_wireframes_face_texbuf(
 | 
						|
        Mesh *me, GPUTexture **verts_data, GPUTexture **face_indices, int *tri_count)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->edges_face_overlay_tx == NULL || cache->pos_in_order_tx == NULL) {
 | 
						|
		const int options = MR_DATATYPE_VERT | MR_DATATYPE_EDGE | MR_DATATYPE_LOOP | MR_DATATYPE_LOOPTRI;
 | 
						|
 | 
						|
		MeshRenderData *rdata = mesh_render_data_create(me, options);
 | 
						|
 | 
						|
		mesh_batch_cache_get_edges_overlay_texture_buf(rdata, cache);
 | 
						|
		mesh_batch_cache_get_vert_pos_and_nor_in_order_buf(rdata, cache);
 | 
						|
 | 
						|
		mesh_render_data_free(rdata);
 | 
						|
	}
 | 
						|
 | 
						|
	*tri_count = cache->edges_face_overlay_tri_count;
 | 
						|
	*face_indices = cache->edges_face_overlay_tx;
 | 
						|
	*verts_data = cache->pos_in_order_tx;
 | 
						|
}
 | 
						|
 | 
						|
static void mesh_batch_cache_create_overlay_batches(Mesh *me)
 | 
						|
{
 | 
						|
	BLI_assert(me->edit_btmesh != NULL);
 | 
						|
 | 
						|
	/* Since MR_DATATYPE_OVERLAY is slow to generate, generate them all at once */
 | 
						|
	const int options =
 | 
						|
	        MR_DATATYPE_VERT | MR_DATATYPE_EDGE | MR_DATATYPE_LOOP | MR_DATATYPE_POLY |
 | 
						|
	        MR_DATATYPE_LOOPTRI | MR_DATATYPE_OVERLAY;
 | 
						|
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
	MeshRenderData *rdata = mesh_render_data_create(me, options);
 | 
						|
 | 
						|
	if (cache->overlay_triangles == NULL) {
 | 
						|
		cache->overlay_triangles = GPU_batch_create(
 | 
						|
		        GPU_PRIM_TRIS, mesh_batch_cache_get_edit_tri_pos(rdata, cache), NULL);
 | 
						|
		GPU_batch_vertbuf_add(cache->overlay_triangles, mesh_batch_cache_get_edit_tri_nor(rdata, cache));
 | 
						|
		GPU_batch_vertbuf_add(cache->overlay_triangles, mesh_batch_cache_get_edit_tri_data(rdata, cache));
 | 
						|
	}
 | 
						|
 | 
						|
	if (cache->overlay_loose_edges == NULL) {
 | 
						|
		cache->overlay_loose_edges = GPU_batch_create(
 | 
						|
		        GPU_PRIM_LINES, mesh_batch_cache_get_edit_ledge_pos(rdata, cache), NULL);
 | 
						|
		GPU_batch_vertbuf_add(cache->overlay_loose_edges, mesh_batch_cache_get_edit_ledge_nor(rdata, cache));
 | 
						|
		GPU_batch_vertbuf_add(cache->overlay_loose_edges, mesh_batch_cache_get_edit_ledge_data(rdata, cache));
 | 
						|
	}
 | 
						|
 | 
						|
	if (cache->overlay_loose_verts == NULL) {
 | 
						|
		cache->overlay_loose_verts = GPU_batch_create(
 | 
						|
		        GPU_PRIM_POINTS, mesh_batch_cache_get_edit_lvert_pos(rdata, cache), NULL);
 | 
						|
		GPU_batch_vertbuf_add(cache->overlay_loose_verts, mesh_batch_cache_get_edit_lvert_nor(rdata, cache));
 | 
						|
		GPU_batch_vertbuf_add(cache->overlay_loose_verts, mesh_batch_cache_get_edit_lvert_data(rdata, cache));
 | 
						|
	}
 | 
						|
 | 
						|
	if (cache->overlay_triangles_nor == NULL) {
 | 
						|
		cache->overlay_triangles_nor = GPU_batch_create(
 | 
						|
		        GPU_PRIM_POINTS, mesh_batch_cache_get_edit_tri_pos(rdata, cache), NULL);
 | 
						|
		GPU_batch_vertbuf_add(cache->overlay_triangles_nor, mesh_batch_cache_get_edit_tri_nor(rdata, cache));
 | 
						|
	}
 | 
						|
 | 
						|
	if (cache->overlay_loose_edges_nor == NULL) {
 | 
						|
		cache->overlay_loose_edges_nor = GPU_batch_create(
 | 
						|
		        GPU_PRIM_POINTS, mesh_batch_cache_get_edit_ledge_pos(rdata, cache), NULL);
 | 
						|
		GPU_batch_vertbuf_add(cache->overlay_loose_edges_nor, mesh_batch_cache_get_edit_ledge_nor(rdata, cache));
 | 
						|
	}
 | 
						|
 | 
						|
	mesh_render_data_free(rdata);
 | 
						|
}
 | 
						|
 | 
						|
GPUBatch *DRW_mesh_batch_cache_get_overlay_triangles(Mesh *me)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->overlay_triangles == NULL) {
 | 
						|
		mesh_batch_cache_create_overlay_batches(me);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->overlay_triangles;
 | 
						|
}
 | 
						|
 | 
						|
GPUBatch *DRW_mesh_batch_cache_get_overlay_loose_edges(Mesh *me)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->overlay_loose_edges == NULL) {
 | 
						|
		mesh_batch_cache_create_overlay_batches(me);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->overlay_loose_edges;
 | 
						|
}
 | 
						|
 | 
						|
GPUBatch *DRW_mesh_batch_cache_get_overlay_loose_verts(Mesh *me)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->overlay_loose_verts == NULL) {
 | 
						|
		mesh_batch_cache_create_overlay_batches(me);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->overlay_loose_verts;
 | 
						|
}
 | 
						|
 | 
						|
GPUBatch *DRW_mesh_batch_cache_get_overlay_triangles_nor(Mesh *me)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->overlay_triangles_nor == NULL) {
 | 
						|
		mesh_batch_cache_create_overlay_batches(me);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->overlay_triangles_nor;
 | 
						|
}
 | 
						|
 | 
						|
GPUBatch *DRW_mesh_batch_cache_get_overlay_loose_edges_nor(Mesh *me)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->overlay_loose_edges_nor == NULL) {
 | 
						|
		mesh_batch_cache_create_overlay_batches(me);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->overlay_loose_edges_nor;
 | 
						|
}
 | 
						|
 | 
						|
GPUBatch *DRW_mesh_batch_cache_get_overlay_facedots(Mesh *me)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->overlay_facedots == NULL) {
 | 
						|
		MeshRenderData *rdata = mesh_render_data_create(me, MR_DATATYPE_VERT | MR_DATATYPE_LOOP | MR_DATATYPE_POLY);
 | 
						|
 | 
						|
		cache->overlay_facedots = GPU_batch_create(
 | 
						|
		        GPU_PRIM_POINTS, mesh_batch_cache_get_facedot_pos_with_normals_and_flag(rdata, cache), NULL);
 | 
						|
 | 
						|
		mesh_render_data_free(rdata);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->overlay_facedots;
 | 
						|
}
 | 
						|
 | 
						|
GPUBatch *DRW_mesh_batch_cache_get_facedots_with_select_id(Mesh *me, uint select_id_offset)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->facedot_with_select_id_offset != select_id_offset) {
 | 
						|
		cache->facedot_with_select_id_offset = select_id_offset;
 | 
						|
		GPU_BATCH_DISCARD_SAFE(cache->edges_with_select_id);
 | 
						|
	}
 | 
						|
 | 
						|
	if (cache->facedot_with_select_id == NULL) {
 | 
						|
		MeshRenderData *rdata = mesh_render_data_create(me, MR_DATATYPE_VERT | MR_DATATYPE_LOOP | MR_DATATYPE_POLY);
 | 
						|
 | 
						|
		/* We only want the 'pos', not the normals or flag.
 | 
						|
		 * Use since this is almost certainly already created. */
 | 
						|
		cache->facedot_with_select_id = GPU_batch_create(
 | 
						|
		        GPU_PRIM_POINTS, mesh_batch_cache_get_facedot_pos_with_normals_and_flag(rdata, cache), NULL);
 | 
						|
 | 
						|
		GPU_batch_vertbuf_add_ex(
 | 
						|
		        cache->facedot_with_select_id,
 | 
						|
		        mesh_create_facedot_select_id(rdata, select_id_offset), true);
 | 
						|
 | 
						|
		mesh_render_data_free(rdata);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->facedot_with_select_id;
 | 
						|
}
 | 
						|
 | 
						|
GPUBatch *DRW_mesh_batch_cache_get_edges_with_select_id(Mesh *me, uint select_id_offset)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->edges_with_select_id_offset != select_id_offset) {
 | 
						|
		cache->edges_with_select_id_offset = select_id_offset;
 | 
						|
		GPU_BATCH_DISCARD_SAFE(cache->edges_with_select_id);
 | 
						|
	}
 | 
						|
 | 
						|
	if (cache->edges_with_select_id == NULL) {
 | 
						|
		MeshRenderData *rdata = mesh_render_data_create(me, MR_DATATYPE_VERT | MR_DATATYPE_EDGE);
 | 
						|
 | 
						|
		cache->edges_with_select_id = GPU_batch_create(
 | 
						|
		        GPU_PRIM_LINES, mesh_batch_cache_get_edges_visible(rdata, cache), NULL);
 | 
						|
 | 
						|
		GPU_batch_vertbuf_add_ex(
 | 
						|
		        cache->edges_with_select_id,
 | 
						|
		        mesh_create_edges_select_id(rdata, select_id_offset), true);
 | 
						|
 | 
						|
		mesh_render_data_free(rdata);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->edges_with_select_id;
 | 
						|
}
 | 
						|
 | 
						|
GPUBatch *DRW_mesh_batch_cache_get_verts_with_select_id(Mesh *me, uint select_id_offset)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->verts_with_select_id_offset != select_id_offset) {
 | 
						|
		cache->verts_with_select_id_offset = select_id_offset;
 | 
						|
		GPU_BATCH_DISCARD_SAFE(cache->verts_with_select_id);
 | 
						|
	}
 | 
						|
 | 
						|
	if (cache->verts_with_select_id == NULL) {
 | 
						|
		MeshRenderData *rdata = mesh_render_data_create(me, MR_DATATYPE_VERT);
 | 
						|
 | 
						|
		cache->verts_with_select_id = GPU_batch_create(
 | 
						|
		        GPU_PRIM_POINTS, mesh_batch_cache_get_verts_visible(rdata, cache), NULL);
 | 
						|
 | 
						|
		GPU_batch_vertbuf_add_ex(
 | 
						|
		        cache->verts_with_select_id,
 | 
						|
		        mesh_create_verts_select_id(rdata, select_id_offset), true);
 | 
						|
 | 
						|
		mesh_render_data_free(rdata);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->verts_with_select_id;
 | 
						|
}
 | 
						|
 | 
						|
GPUBatch **DRW_mesh_batch_cache_get_surface_shaded(
 | 
						|
        Mesh *me, struct GPUMaterial **gpumat_array, uint gpumat_array_len,
 | 
						|
        char **auto_layer_names, int **auto_layer_is_srgb, int *auto_layer_count)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->shaded_triangles == NULL) {
 | 
						|
		/* create batch from DM */
 | 
						|
		const int datatype =
 | 
						|
		        MR_DATATYPE_VERT | MR_DATATYPE_LOOP | MR_DATATYPE_LOOPTRI |
 | 
						|
		        MR_DATATYPE_POLY | MR_DATATYPE_SHADING;
 | 
						|
		MeshRenderData *rdata = mesh_render_data_create_ex(me, datatype, gpumat_array, gpumat_array_len);
 | 
						|
 | 
						|
		const int mat_len = mesh_render_data_mat_len_get(rdata);
 | 
						|
 | 
						|
		cache->shaded_triangles = MEM_callocN(sizeof(*cache->shaded_triangles) * mat_len, __func__);
 | 
						|
 | 
						|
		GPUIndexBuf **el = mesh_batch_cache_get_triangles_in_order_split_by_material(rdata, cache);
 | 
						|
 | 
						|
		GPUVertBuf *vbo = mesh_batch_cache_get_tri_pos_and_normals(rdata, cache);
 | 
						|
		GPUVertBuf *vbo_shading = mesh_batch_cache_get_tri_shading_data(rdata, cache);
 | 
						|
 | 
						|
		for (int i = 0; i < mat_len; ++i) {
 | 
						|
			cache->shaded_triangles[i] = GPU_batch_create(
 | 
						|
			        GPU_PRIM_TRIS, vbo, el[i]);
 | 
						|
			if (vbo_shading) {
 | 
						|
				GPU_batch_vertbuf_add(cache->shaded_triangles[i], vbo_shading);
 | 
						|
			}
 | 
						|
		}
 | 
						|
 | 
						|
		mesh_render_data_free(rdata);
 | 
						|
	}
 | 
						|
 | 
						|
	if (auto_layer_names) {
 | 
						|
		*auto_layer_names = cache->auto_layer_names;
 | 
						|
		*auto_layer_is_srgb = cache->auto_layer_is_srgb;
 | 
						|
		*auto_layer_count = cache->auto_layer_len;
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->shaded_triangles;
 | 
						|
}
 | 
						|
 | 
						|
GPUBatch **DRW_mesh_batch_cache_get_surface_texpaint(Mesh *me)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->texpaint_triangles == NULL) {
 | 
						|
		/* create batch from DM */
 | 
						|
		const int datatype =
 | 
						|
		        MR_DATATYPE_VERT | MR_DATATYPE_LOOP | MR_DATATYPE_POLY | MR_DATATYPE_LOOPTRI | MR_DATATYPE_LOOPUV;
 | 
						|
		MeshRenderData *rdata = mesh_render_data_create(me, datatype);
 | 
						|
 | 
						|
		const int mat_len = mesh_render_data_mat_len_get(rdata);
 | 
						|
 | 
						|
		cache->texpaint_triangles = MEM_callocN(sizeof(*cache->texpaint_triangles) * mat_len, __func__);
 | 
						|
 | 
						|
		GPUIndexBuf **el = mesh_batch_cache_get_triangles_in_order_split_by_material(rdata, cache);
 | 
						|
 | 
						|
		GPUVertBuf *vbo = mesh_batch_cache_get_tri_pos_and_normals(rdata, cache);
 | 
						|
		for (int i = 0; i < mat_len; ++i) {
 | 
						|
			cache->texpaint_triangles[i] = GPU_batch_create(
 | 
						|
			        GPU_PRIM_TRIS, vbo, el[i]);
 | 
						|
			GPUVertBuf *vbo_uv = mesh_batch_cache_get_tri_uv_active(rdata, cache);
 | 
						|
			if (vbo_uv) {
 | 
						|
				GPU_batch_vertbuf_add(cache->texpaint_triangles[i], vbo_uv);
 | 
						|
			}
 | 
						|
		}
 | 
						|
		mesh_render_data_free(rdata);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->texpaint_triangles;
 | 
						|
}
 | 
						|
 | 
						|
GPUBatch *DRW_mesh_batch_cache_get_surface_texpaint_single(Mesh *me)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->texpaint_triangles_single == NULL) {
 | 
						|
		/* create batch from DM */
 | 
						|
		const int datatype =
 | 
						|
		        MR_DATATYPE_VERT | MR_DATATYPE_LOOP | MR_DATATYPE_POLY | MR_DATATYPE_LOOPTRI | MR_DATATYPE_LOOPUV;
 | 
						|
		MeshRenderData *rdata = mesh_render_data_create(me, datatype);
 | 
						|
 | 
						|
		GPUVertBuf *vbo = mesh_batch_cache_get_tri_pos_and_normals(rdata, cache);
 | 
						|
 | 
						|
		cache->texpaint_triangles_single = GPU_batch_create(
 | 
						|
		        GPU_PRIM_TRIS, vbo, NULL);
 | 
						|
		GPUVertBuf *vbo_uv = mesh_batch_cache_get_tri_uv_active(rdata, cache);
 | 
						|
		if (vbo_uv) {
 | 
						|
			GPU_batch_vertbuf_add(cache->texpaint_triangles_single, vbo_uv);
 | 
						|
		}
 | 
						|
		mesh_render_data_free(rdata);
 | 
						|
	}
 | 
						|
	return cache->texpaint_triangles_single;
 | 
						|
}
 | 
						|
 | 
						|
GPUBatch *DRW_mesh_batch_cache_get_weight_overlay_edges(Mesh *me, bool use_wire, bool use_sel)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->overlay_paint_edges == NULL) {
 | 
						|
		/* create batch from Mesh */
 | 
						|
		const int datatype = MR_DATATYPE_VERT | MR_DATATYPE_EDGE | MR_DATATYPE_POLY | MR_DATATYPE_LOOP;
 | 
						|
		MeshRenderData *rdata = mesh_render_data_create(me, datatype);
 | 
						|
 | 
						|
		cache->overlay_paint_edges = GPU_batch_create_ex(
 | 
						|
		        GPU_PRIM_LINES, mesh_create_edge_pos_with_sel(rdata, use_wire, use_sel), NULL, GPU_BATCH_OWNS_VBO);
 | 
						|
 | 
						|
		mesh_render_data_free(rdata);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->overlay_paint_edges;
 | 
						|
}
 | 
						|
 | 
						|
GPUBatch *DRW_mesh_batch_cache_get_weight_overlay_faces(Mesh *me)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->overlay_weight_faces == NULL) {
 | 
						|
		/* create batch from Mesh */
 | 
						|
		const int datatype = MR_DATATYPE_VERT | MR_DATATYPE_POLY | MR_DATATYPE_LOOP | MR_DATATYPE_LOOPTRI;
 | 
						|
		MeshRenderData *rdata = mesh_render_data_create(me, datatype);
 | 
						|
 | 
						|
		cache->overlay_weight_faces = GPU_batch_create_ex(
 | 
						|
		        GPU_PRIM_TRIS, mesh_batch_cache_get_vert_pos_and_nor_in_order(rdata, cache),
 | 
						|
		        mesh_create_tri_overlay_weight_faces(rdata), GPU_BATCH_OWNS_INDEX);
 | 
						|
 | 
						|
		mesh_render_data_free(rdata);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->overlay_weight_faces;
 | 
						|
}
 | 
						|
 | 
						|
GPUBatch *DRW_mesh_batch_cache_get_weight_overlay_verts(Mesh *me)
 | 
						|
{
 | 
						|
	MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
 | 
						|
	if (cache->overlay_weight_verts == NULL) {
 | 
						|
		/* create batch from Mesh */
 | 
						|
		MeshRenderData *rdata = mesh_render_data_create(me, MR_DATATYPE_VERT);
 | 
						|
 | 
						|
		cache->overlay_weight_verts = GPU_batch_create(
 | 
						|
		        GPU_PRIM_POINTS, mesh_batch_cache_get_vert_pos_and_nor_in_order(rdata, cache), NULL);
 | 
						|
 | 
						|
		GPU_batch_vertbuf_add_ex(
 | 
						|
		        cache->overlay_weight_verts,
 | 
						|
		        mesh_create_vert_pos_with_overlay_data(rdata), true);
 | 
						|
		mesh_render_data_free(rdata);
 | 
						|
	}
 | 
						|
 | 
						|
	return cache->overlay_weight_verts;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Needed for when we draw with shaded data.
 | 
						|
 */
 | 
						|
void DRW_mesh_cache_sculpt_coords_ensure(Mesh *me)
 | 
						|
{
 | 
						|
	if (me->runtime.batch_cache) {
 | 
						|
		MeshBatchCache *cache = mesh_batch_cache_get(me);
 | 
						|
		if (cache && cache->pos_with_normals && cache->is_sculpt_points_tag) {
 | 
						|
			/* XXX Force update of all the batches that contains the pos_with_normals buffer.
 | 
						|
			 * TODO(fclem): Ideally, Gawain should provide a way to update a buffer without destroying it. */
 | 
						|
			mesh_batch_cache_clear_selective(me, cache->pos_with_normals);
 | 
						|
			GPU_VERTBUF_DISCARD_SAFE(cache->pos_with_normals);
 | 
						|
		}
 | 
						|
		cache->is_sculpt_points_tag = false;
 | 
						|
	}
 | 
						|
}
 | 
						|
 | 
						|
/** \} */
 |