1 files changed, 389 insertions, 0 deletions
diff --git a/source/blender/editors/object/editlattice.c b/source/blender/editors/object/editlattice.c
new file mode 100644
index 00000000000..3e30efd635a
--- /dev/null
+++ b/source/blender/editors/object/editlattice.c
@@ -0,0 +1,389 @@
+/**
+ * $Id$
+ *
+ * ***** 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., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
+ *
+ * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
+ * All rights reserved.
+ *
+ * Contributor(s): Blender Foundation
+ *
+ * ***** END GPL LICENSE BLOCK *****
+ */
+
+#include <stdlib.h>
+#include <string.h>
+#include <math.h>
+
+#include "MEM_guardedalloc.h"
+
+#include "BLI_blenlib.h"
+#include "BLI_arithb.h"
+
+#include "DNA_curve_types.h"
+#include "DNA_key_types.h"
+#include "DNA_lattice_types.h"
+#include "DNA_meshdata_types.h"
+#include "DNA_object_types.h"
+#include "DNA_scene_types.h"
+#include "DNA_view3d_types.h"
+
+#include "BKE_armature.h"
+#include "BKE_context.h"
+#include "BKE_depsgraph.h"
+#include "BKE_global.h"
+#include "BKE_key.h"
+#include "BKE_lattice.h"
+#include "BKE_mesh.h"
+#include "BKE_utildefines.h"
+
+#include "ED_object.h"
+#include "ED_screen.h"
+#include "ED_view3d.h"
+#include "ED_util.h"
+
+#include "WM_api.h"
+#include "WM_types.h"
+
+#include "object_intern.h"
+
+/********************** Load/Make/Free ********************/
+
+void free_editLatt(Object *ob)
+{
+	Lattice *lt= ob->data;
+	
+	if(lt->editlatt) {
+		if(lt->editlatt->def)
+			MEM_freeN(lt->editlatt->def);
+		if(lt->editlatt->dvert) 
+			free_dverts(lt->editlatt->dvert, lt->editlatt->pntsu*lt->editlatt->pntsv*lt->editlatt->pntsw);
+		
+		MEM_freeN(lt->editlatt);
+		lt->editlatt= NULL;
+	}
+}
+
+void make_editLatt(Object *obedit)
+{
+	Lattice *lt= obedit->data;
+	KeyBlock *actkey;
+	
+	free_editLatt(obedit);
+	
+	lt= obedit->data;
+
+	actkey= ob_get_keyblock(obedit);
+	if(actkey)
+		key_to_latt(actkey, lt);
+
+	lt->editlatt= MEM_dupallocN(lt);
+	lt->editlatt->def= MEM_dupallocN(lt->def);
+	
+	if(lt->dvert) {
+		int tot= lt->pntsu*lt->pntsv*lt->pntsw;
+		lt->editlatt->dvert = MEM_mallocN (sizeof (MDeformVert)*tot, "Lattice MDeformVert");
+		copy_dverts(lt->editlatt->dvert, lt->dvert, tot);
+	}
+}
+
+void load_editLatt(Object *obedit)
+{
+	Lattice *lt;
+	KeyBlock *actkey;
+	BPoint *bp;
+	float *fp;
+	int tot;
+	
+	lt= obedit->data;
+	
+	actkey= ob_get_keyblock(obedit);
+
+	if(actkey) {
+		/* active key: vertices */
+		tot= lt->editlatt->pntsu*lt->editlatt->pntsv*lt->editlatt->pntsw;
+		
+		if(actkey->data) MEM_freeN(actkey->data);
+		
+		fp=actkey->data= MEM_callocN(lt->key->elemsize*tot, "actkey->data");
+		actkey->totelem= tot;
+	
+		bp= lt->editlatt->def;
+		while(tot--) {
+			VECCOPY(fp, bp->vec);
+			fp+= 3;
+			bp++;
+		}
+	}
+	else {
+		MEM_freeN(lt->def);
+	
+		lt->def= MEM_dupallocN(lt->editlatt->def);
+
+		lt->flag= lt->editlatt->flag;
+
+		lt->pntsu= lt->editlatt->pntsu;
+		lt->pntsv= lt->editlatt->pntsv;
+		lt->pntsw= lt->editlatt->pntsw;
+		
+		lt->typeu= lt->editlatt->typeu;
+		lt->typev= lt->editlatt->typev;
+		lt->typew= lt->editlatt->typew;
+	}
+	
+	if(lt->dvert) {
+		free_dverts(lt->dvert, lt->pntsu*lt->pntsv*lt->pntsw);
+		lt->dvert= NULL;
+	}
+	
+	if(lt->editlatt->dvert) {
+		int tot= lt->pntsu*lt->pntsv*lt->pntsw;
+		
+		lt->dvert = MEM_mallocN (sizeof (MDeformVert)*tot, "Lattice MDeformVert");
+		copy_dverts(lt->dvert, lt->editlatt->dvert, tot);
+	}
+}
+
+/************************** Operators *************************/
+
+static void setflagsLatt(Object *obedit, int flag)
+{
+	Lattice *lt= obedit->data;
+	BPoint *bp;
+	int a;
+	
+	bp= lt->editlatt->def;
+	
+	a= lt->editlatt->pntsu*lt->editlatt->pntsv*lt->editlatt->pntsw;
+	
+	while(a--) {
+		if(bp->hide==0) {
+			bp->f1= flag;
+		}
+		bp++;
+	}
+}
+
+int de_select_all_exec(bContext *C, wmOperator *op)
+{
+	Object *obedit= CTX_data_edit_object(C);
+	Lattice *lt= obedit->data;
+	BPoint *bp;
+	int a, deselect= 0;
+	
+	bp= lt->editlatt->def;
+	a= lt->editlatt->pntsu*lt->editlatt->pntsv*lt->editlatt->pntsw;
+	
+	while(a--) {
+		if(bp->hide==0) {
+			if(bp->f1) {
+				deselect= 1;
+				break;
+			}
+		}
+		bp++;
+	}
+
+	if(deselect)
+		setflagsLatt(obedit, 0);
+	else
+		setflagsLatt(obedit, 1);
+	
+	WM_event_add_notifier(C, NC_OBJECT|ND_GEOM_SELECT, obedit);
+
+	return OPERATOR_FINISHED;
+}
+
+void LATTICE_OT_select_all_toggle(wmOperatorType *ot)
+{
+	/* identifiers */
+	ot->name= "Select or Deselect All";
+	ot->idname= "LATTICE_OT_select_all_toggle";
+	
+	/* api callbacks */
+	ot->exec= de_select_all_exec;
+	ot->poll= ED_operator_editlattice;
+	
+	/* flags */
+	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
+}
+
+int make_regular_poll(bContext *C)
+{
+	Object *ob;
+
+	if(ED_operator_editlattice(C)) return 1;
+
+	ob= CTX_data_active_object(C);
+	return (ob && ob->type==OB_LATTICE);
+}
+
+int make_regular_exec(bContext *C, wmOperator *op)
+{
+	Scene *scene= CTX_data_scene(C);
+	Object *ob= CTX_data_edit_object(C);
+	Lattice *lt;
+	
+	if(ob) {
+		lt= ob->data;
+		resizelattice(lt->editlatt, lt->pntsu, lt->pntsv, lt->pntsw, NULL);
+	}
+	else {
+		ob= CTX_data_active_object(C);
+		lt= ob->data;
+		resizelattice(lt, lt->pntsu, lt->pntsv, lt->pntsw, NULL);
+	}
+	
+	DAG_object_flush_update(scene, ob, OB_RECALC_DATA);
+	WM_event_add_notifier(C, NC_OBJECT|ND_GEOM_DATA, ob);
+
+	return OPERATOR_FINISHED;
+}
+
+void LATTICE_OT_make_regular(wmOperatorType *ot)
+{
+	/* identifiers */
+	ot->name= "Make Regular";
+	ot->idname= "LATTICE_OT_make_regular";
+	
+	/* api callbacks */
+	ot->exec= make_regular_exec;
+	ot->poll= make_regular_poll;
+	
+	/* flags */
+	ot->flag= OPTYPE_REGISTER|OPTYPE_UNDO;
+}
+
+/****************************** Mouse Selection *************************/
+
+static void findnearestLattvert__doClosest(void *userData, BPoint *bp, int x, int y)
+{
+	struct { BPoint *bp; short dist, select, mval[2]; } *data = userData;
+	float temp = abs(data->mval[0]-x) + abs(data->mval[1]-y);
+	
+	if((bp->f1 & SELECT)==data->select)
+		temp += 5;
+
+	if(temp<data->dist) {
+		data->dist = temp;
+
+		data->bp = bp;
+	}
+}
+
+static BPoint *findnearestLattvert(ViewContext *vc, short mval[2], int sel)
+{
+		/* sel==1: selected gets a disadvantage */
+		/* in nurb and bezt or bp the nearest is written */
+		/* return 0 1 2: handlepunt */
+	struct { BPoint *bp; short dist, select, mval[2]; } data = {0};
+
+	data.dist = 100;
+	data.select = sel;
+	data.mval[0]= mval[0];
+	data.mval[1]= mval[1];
+
+	lattice_foreachScreenVert(vc, findnearestLattvert__doClosest, &data);
+
+	return data.bp;
+}
+
+void mouse_lattice(bContext *C, short mval[2], int extend)
+{
+	ViewContext vc;
+	BPoint *bp= NULL;
+
+	view3d_set_viewcontext(C, &vc);
+	bp= findnearestLattvert(&vc, mval, 1);
+
+	if(bp) {
+		if(extend==0) {
+			setflagsLatt(vc.obedit, 0);
+			bp->f1 |= SELECT;
+		}
+		else
+			bp->f1 ^= SELECT; /* swap */
+
+		WM_event_add_notifier(C, NC_OBJECT|ND_GEOM_SELECT, vc.obedit);
+	}
+}
+
+/******************************** Undo *************************/
+
+typedef struct UndoLattice {
+	BPoint *def;
+	int pntsu, pntsv, pntsw;
+} UndoLattice;
+
+static void undoLatt_to_editLatt(void *data, void *edata)
+{
+	UndoLattice *ult= (UndoLattice*)data;
+	Lattice *editlatt= (Lattice *)edata;
+	int a= editlatt->pntsu*editlatt->pntsv*editlatt->pntsw;
+
+	memcpy(editlatt->def, ult->def, a*sizeof(BPoint));
+}
+
+static void *editLatt_to_undoLatt(void *edata)
+{
+	UndoLattice *ult= MEM_callocN(sizeof(UndoLattice), "UndoLattice");
+	Lattice *editlatt= (Lattice *)edata;
+	
+	ult->def= MEM_dupallocN(editlatt->def);
+	ult->pntsu= editlatt->pntsu;
+	ult->pntsv= editlatt->pntsv;
+	ult->pntsw= editlatt->pntsw;
+	
+	return ult;
+}
+
+static void free_undoLatt(void *data)
+{
+	UndoLattice *ult= (UndoLattice*)data;
+
+	if(ult->def) MEM_freeN(ult->def);
+	MEM_freeN(ult);
+}
+
+static int validate_undoLatt(void *data, void *edata)
+{
+	UndoLattice *ult= (UndoLattice*)data;
+	Lattice *editlatt= (Lattice *)edata;
+
+	return (ult->pntsu == editlatt->pntsu &&
+	        ult->pntsv == editlatt->pntsv &&
+	        ult->pntsw == editlatt->pntsw);
+}
+
+static void *get_editlatt(bContext *C)
+{
+	Object *obedit= CTX_data_edit_object(C);
+
+	if(obedit && obedit->type==OB_LATTICE) {
+		Lattice *lt= obedit->data;
+		return lt->editlatt;
+	}
+
+	return NULL;
+}
+
+/* and this is all the undo system needs to know */
+void undo_push_lattice(bContext *C, char *name)
+{
+	undo_editmode_push(C, name, get_editlatt, free_undoLatt, undoLatt_to_editLatt, editLatt_to_undoLatt, validate_undoLatt);
+}
+