transform_mode_tosphere.c

Go to the documentation of this file.

/*
 * 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) 2001-2002 by NaN Holding BV.
 * All rights reserved.
 */
 
#include <stdlib.h>
 
#include "BLI_math.h"
#include "BLI_string.h"
 
#include "MEM_guardedalloc.h"
 
#include "BKE_context.h"
#include "BKE_unit.h"
 
#include "ED_screen.h"
 
#include "UI_interface.h"
 
#include "BLT_translation.h"
 
#include "transform.h"
#include "transform_mode.h"
#include "transform_snap.h"
 
/* -------------------------------------------------------------------- */
struct ToSphereInfo {
  float prop_size_prev;
  float radius;
};
 
static void to_sphere_radius_update(TransInfo *t)
{
  struct ToSphereInfo *data = t->custom.mode.data;
  float radius = 0.0f;
 
  const bool is_local_center = transdata_check_local_center(t, t->around);
 
  if (t->flag & T_PROP_EDIT_ALL) {
    int factor_accum = 0.0f;
    FOREACH_TRANS_DATA_CONTAINER (t, tc) {
      TransData *td = tc->data;
      for (int i = 0; i < tc->data_len; i++, td++) {
        if (td->factor == 0.0f) {
          continue;
        }
        const float *center = is_local_center ? td->center : tc->center_local;
        radius += td->factor * len_v3v3(center, td->iloc);
        factor_accum += td->factor;
      }
    }
    if (factor_accum != 0.0f) {
      radius /= factor_accum;
    }
  }
  else {
    FOREACH_TRANS_DATA_CONTAINER (t, tc) {
      TransData *td = tc->data;
      for (int i = 0; i < tc->data_len; i++, td++) {
        const float *center = is_local_center ? td->center : tc->center_local;
        radius += len_v3v3(center, td->iloc);
      }
    }
    radius /= (float)t->data_len_all;
  }
 
  data->prop_size_prev = t->prop_size;
  data->radius = radius;
}
 
/* -------------------------------------------------------------------- */
static void applyToSphere(TransInfo *t, const int UNUSED(mval[2]))
{
  const bool is_local_center = transdata_check_local_center(t, t->around);
 
  float vec[3];
  float ratio, radius;
  int i;
  char str[UI_MAX_DRAW_STR];
 
  ratio = t->values[0];
 
  transform_snap_increment(t, &ratio);
 
  applyNumInput(&t->num, &ratio);
 
  CLAMP(ratio, 0.0f, 1.0f);
 
  t->values_final[0] = ratio;
 
  /* header print for NumInput */
  if (hasNumInput(&t->num)) {
    char c[NUM_STR_REP_LEN];
 
    outputNumInput(&(t->num), c, &t->scene->unit);
 
    BLI_snprintf(str, sizeof(str), TIP_("To Sphere: %s %s"), c, t->proptext);
  }
  else {
    /* default header print */
    BLI_snprintf(str, sizeof(str), TIP_("To Sphere: %.4f %s"), ratio, t->proptext);
  }
 
  const struct ToSphereInfo *data = t->custom.mode.data;
  if (data->prop_size_prev != t->prop_size) {
    to_sphere_radius_update(t);
  }
 
  FOREACH_TRANS_DATA_CONTAINER (t, tc) {
    TransData *td = tc->data;
    for (i = 0; i < tc->data_len; i++, td++) {
      float tratio;
      if (td->flag & TD_SKIP) {
        continue;
      }
 
      const float *center = is_local_center ? td->center : tc->center_local;
 
      sub_v3_v3v3(vec, td->iloc, center);
 
      radius = normalize_v3(vec);
 
      tratio = ratio * td->factor;
 
      mul_v3_fl(vec, radius * (1.0f - tratio) + data->radius * tratio);
 
      add_v3_v3v3(td->loc, center, vec);
    }
  }
 
  recalcData(t);
 
  ED_area_status_text(t->area, str);
}
 
void initToSphere(TransInfo *t)
{
  t->mode = TFM_TOSPHERE;
  t->transform = applyToSphere;
 
  initMouseInputMode(t, &t->mouse, INPUT_HORIZONTAL_RATIO);
 
  t->idx_max = 0;
  t->num.idx_max = 0;
  t->snap[0] = 0.1f;
  t->snap[1] = t->snap[0] * 0.1f;
 
  copy_v3_fl(t->num.val_inc, t->snap[0]);
  t->num.unit_sys = t->scene->unit.system;
  t->num.unit_type[0] = B_UNIT_NONE;
 
  t->num.val_flag[0] |= NUM_NULL_ONE | NUM_NO_NEGATIVE;
  t->flag |= T_NO_CONSTRAINT;
 
  struct ToSphereInfo *data = MEM_callocN(sizeof(*data), __func__);
  t->custom.mode.data = data;
  t->custom.mode.use_free = true;
 
  to_sphere_radius_update(t);
}