blender-archive/source/blender/blenkernel/intern/editmesh.c

/*
 * This program is free software; you can redistribute it and/or
 * modify it under the terms of the GNU General Public License
 * as published by the Free Software Foundation; either version 2
 * of the License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 *
 * The Original Code is Copyright (C) 2005 Blender Foundation.
 * All rights reserved.
 */

/** \file
 * \ingroup bke
 */

#include "MEM_guardedalloc.h"

#include "DNA_listBase.h"
#include "DNA_object_types.h"
#include "DNA_mesh_types.h"

#include "BLI_math.h"

#include "BKE_editmesh.h"
#include "BKE_cdderivedmesh.h"
#include "BKE_library.h"

BMEditMesh *BKE_editmesh_create(BMesh *bm, const bool do_tessellate)
{
  BMEditMesh *em = MEM_callocN(sizeof(BMEditMesh), __func__);

  em->bm = bm;
  if (do_tessellate) {
    BKE_editmesh_tessface_calc(em);
  }

  return em;
}

BMEditMesh *BKE_editmesh_copy(BMEditMesh *em)
{
  BMEditMesh *em_copy = MEM_callocN(sizeof(BMEditMesh), __func__);
  *em_copy = *em;

  em_copy->mesh_eval_cage = em_copy->mesh_eval_final = NULL;

  em_copy->derivedVertColor = NULL;
  em_copy->derivedVertColorLen = 0;
  em_copy->derivedFaceColor = NULL;
  em_copy->derivedFaceColorLen = 0;

  em_copy->bm = BM_mesh_copy(em->bm);

  /* The tessellation is NOT calculated on the copy here,
   * because currently all the callers of this function use
   * it to make a backup copy of the BMEditMesh to restore
   * it in the case of errors in an operation. For perf
   * reasons, in that case it makes more sense to do the
   * tessellation only when/if that copy ends up getting
   * used.*/
  em_copy->looptris = NULL;

  return em_copy;
}

/**
 * \brief Return the BMEditMesh for a given object
 *
 * \note this function assumes this is a mesh object,
 * don't add NULL data check here. caller must do that
 */
BMEditMesh *BKE_editmesh_from_object(Object *ob)
{
  BLI_assert(ob->type == OB_MESH);
  /* sanity check */
#if 0 /* disable in mutlti-object edit. */
#  ifndef NDEBUG
  if (((Mesh *)ob->data)->edit_mesh) {
    BLI_assert(((Mesh *)ob->data)->edit_mesh->ob == ob);
  }
#  endif
#endif
  return ((Mesh *)ob->data)->edit_mesh;
}

static void editmesh_tessface_calc_intern(BMEditMesh *em)
{
  /* allocating space before calculating the tessellation */

  BMesh *bm = em->bm;

  /* this assumes all faces can be scan-filled, which isn't always true,
   * worst case we over alloc a little which is acceptable */
  const int looptris_tot = poly_to_tri_count(bm->totface, bm->totloop);
  const int looptris_tot_prev_alloc = em->looptris ?
                                          (MEM_allocN_len(em->looptris) / sizeof(*em->looptris)) :
                                          0;

  BMLoop *(*looptris)[3];

  /* this means no reallocs for quad dominant models, for */
  if ((em->looptris != NULL) &&
      /* (*em->tottri >= looptris_tot)) */
      /* check against alloc'd size incase we over alloc'd a little */
      ((looptris_tot_prev_alloc >= looptris_tot) &&
       (looptris_tot_prev_alloc <= looptris_tot * 2))) {
    looptris = em->looptris;
  }
  else {
    if (em->looptris) {
      MEM_freeN(em->looptris);
    }
    looptris = MEM_mallocN(sizeof(*looptris) * looptris_tot, __func__);
  }

  em->looptris = looptris;

  /* after allocating the em->looptris, we're ready to tessellate */
  BM_mesh_calc_tessellation(em->bm, em->looptris, &em->tottri);
}

void BKE_editmesh_tessface_calc(BMEditMesh *em)
{
  editmesh_tessface_calc_intern(em);

  /* commented because editbmesh_build_data() ensures we get tessfaces */
#if 0
  if (em->mesh_eval_final && em->mesh_eval_final == em->mesh_eval_cage) {
    BKE_mesh_runtime_looptri_ensure(em->mesh_eval_final);
  }
  else if (em->mesh_eval_final) {
    BKE_mesh_runtime_looptri_ensure(em->mesh_eval_final);
    BKE_mesh_runtime_looptri_ensure(em->mesh_eval_cage);
  }
#endif
}

void BKE_editmesh_free_derivedmesh(BMEditMesh *em)
{
  if (em->mesh_eval_cage) {
    BKE_id_free(NULL, em->mesh_eval_cage);
  }
  if (em->mesh_eval_final && em->mesh_eval_final != em->mesh_eval_cage) {
    BKE_id_free(NULL, em->mesh_eval_final);
  }
  em->mesh_eval_cage = em->mesh_eval_final = NULL;
}

/*does not free the BMEditMesh struct itself*/
void BKE_editmesh_free(BMEditMesh *em)
{
  BKE_editmesh_free_derivedmesh(em);

  BKE_editmesh_color_free(em);

  if (em->looptris) {
    MEM_freeN(em->looptris);
  }

  if (em->bm) {
    BM_mesh_free(em->bm);
  }
}

void BKE_editmesh_color_free(BMEditMesh *em)
{
  if (em->derivedVertColor) {
    MEM_freeN(em->derivedVertColor);
  }
  if (em->derivedFaceColor) {
    MEM_freeN(em->derivedFaceColor);
  }
  em->derivedVertColor = NULL;
  em->derivedFaceColor = NULL;

  em->derivedVertColorLen = 0;
  em->derivedFaceColorLen = 0;
}

void BKE_editmesh_color_ensure(BMEditMesh *em, const char htype)
{
  switch (htype) {
    case BM_VERT:
      if (em->derivedVertColorLen != em->bm->totvert) {
        BKE_editmesh_color_free(em);
        em->derivedVertColor = MEM_mallocN(sizeof(*em->derivedVertColor) * em->bm->totvert,
                                           __func__);
        em->derivedVertColorLen = em->bm->totvert;
      }
      break;
    case BM_FACE:
      if (em->derivedFaceColorLen != em->bm->totface) {
        BKE_editmesh_color_free(em);
        em->derivedFaceColor = MEM_mallocN(sizeof(*em->derivedFaceColor) * em->bm->totface,
                                           __func__);
        em->derivedFaceColorLen = em->bm->totface;
      }
      break;
    default:
      BLI_assert(0);
      break;
  }
}

float (*BKE_editmesh_vertexCos_get_orco(BMEditMesh *em, int *r_numVerts))[3]
{
  BMIter iter;
  BMVert *eve;
  float(*orco)[3];
  int i;

  orco = MEM_mallocN(em->bm->totvert * sizeof(*orco), __func__);

  BM_ITER_MESH_INDEX (eve, &iter, em->bm, BM_VERTS_OF_MESH, i) {
    copy_v3_v3(orco[i], eve->co);
  }

  *r_numVerts = em->bm->totvert;

  return orco;
}

void BKE_editmesh_lnorspace_update(BMEditMesh *em)
{
  BMesh *bm = em->bm;

  /* We need to create clnors data if none exist yet, otherwise there is no way to edit them.
   * Similar code to MESH_OT_customdata_custom_splitnormals_add operator,
   * we want to keep same shading in case we were using autosmooth so far.
   * Note: there is a problem here, which is that if someone starts a normal editing operation on
   * previously autosmooth-ed mesh, and cancel that operation, generated clnors data remain,
   * with related sharp edges (and hence autosmooth is 'lost').
   * Not sure how critical this is, and how to fix that issue? */
  if (!CustomData_has_layer(&bm->ldata, CD_CUSTOMLOOPNORMAL)) {
    Mesh *me = em->ob->data;
    if (me->flag & ME_AUTOSMOOTH) {
      BM_edges_sharp_from_angle_set(bm, me->smoothresh);
    }
  }

  BM_lnorspace_update(bm);
}

/* If autosmooth not already set, set it */
void BKE_editmesh_ensure_autosmooth(BMEditMesh *em)
{
  Mesh *me = em->ob->data;
  if (!(me->flag & ME_AUTOSMOOTH)) {
    me->flag |= ME_AUTOSMOOTH;
    BKE_editmesh_lnorspace_update (em);
  }
}