fac835a414ab815e1fc0454d1904402bb16c7728
[blender.git] / source / blender / editors / physics / rigidbody_constraint.c
1 /*
2  * ***** BEGIN GPL LICENSE BLOCK *****
3  *
4  * This program is free software; you can redistribute it and/or
5  * modify it under the terms of the GNU General Public License
6  * as published by the Free Software Foundation; either version 2
7  * of the License, or (at your option) any later version.
8  *
9  * This program is distributed in the hope that it will be useful,
10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  * GNU General Public License for more details.
13  *
14  * You should have received a copy of the GNU General Public License
15  * along with this program; if not, write to the Free Software Foundation,
16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17  *
18  * The Original Code is Copyright (C) 2013 Blender Foundation
19  * All rights reserved.
20  *
21  * The Original Code is: all of this file.
22  *
23  * Contributor(s): Sergej Reich
24  *
25  * ***** END GPL LICENSE BLOCK *****
26  */
27
28 /** \file rigidbody_constraint.c
29  *  \ingroup editor_physics
30  *  \brief Rigid Body constraint editing operators
31  */
32
33 #include <stdlib.h>
34 #include <string.h>
35
36 #include "MEM_guardedalloc.h"
37
38 #include "DNA_group_types.h"
39 #include "DNA_object_types.h"
40 #include "DNA_rigidbody_types.h"
41 #include "DNA_scene_types.h"
42
43 #include "BLI_blenlib.h"
44 #include "BLI_math.h"
45
46 #include "BKE_context.h"
47 #include "BKE_depsgraph.h"
48 #include "BKE_group.h"
49 #include "BKE_object.h"
50 #include "BKE_report.h"
51 #include "BKE_rigidbody.h"
52
53 #include "RNA_access.h"
54 #include "RNA_define.h"
55 #include "RNA_enum_types.h"
56
57 #include "WM_api.h"
58 #include "WM_types.h"
59
60 #include "ED_physics.h"
61 #include "ED_screen.h"
62
63 #include "physics_intern.h"
64
65 /* ********************************************** */
66 /* Helper API's for RigidBody Constraint Editing */
67
68 static int ED_operator_rigidbody_con_active_poll(bContext *C)
69 {
70         if (ED_operator_object_active_editable(C)) {
71                 Object *ob = CTX_data_active_object(C);
72                 return (ob && ob->rigidbody_constraint);
73         }
74         else
75                 return 0;
76 }
77
78
79 void ED_rigidbody_con_add(wmOperator *op, Scene *scene, Object *ob, int type)
80 {
81         RigidBodyWorld *rbw = BKE_rigidbody_get_world(scene);
82
83         /* check that object doesn't already have a constraint */
84         if (ob->rigidbody_constraint) {
85                 BKE_reportf(op->reports, RPT_INFO, "Object '%s' already has a Rigid Body Constraint", ob->id.name + 2);
86                 return;
87         }
88         /* create constraint group if it doesn't already exits */
89         if (rbw->constraints == NULL) {
90                 rbw->constraints = add_group("RigidBodyConstraints");
91         }
92         /* make rigidbody constraint settings */
93         ob->rigidbody_constraint = BKE_rigidbody_create_constraint(scene, ob, type);
94         ob->rigidbody_constraint->flag |= RBC_FLAG_NEEDS_VALIDATE;
95
96         /* add constraint to rigid body constraint group */
97         add_to_group(rbw->constraints, ob, scene, NULL);
98 }
99
100 void ED_rigidbody_con_remove(Scene *scene, Object *ob)
101 {
102         RigidBodyWorld *rbw = BKE_rigidbody_get_world(scene);
103
104         BKE_rigidbody_remove_constraint(scene, ob);
105         if (rbw)
106                 rem_from_group(rbw->constraints, ob, scene, NULL);
107
108         DAG_id_tag_update(&ob->id, OB_RECALC_OB);
109 }
110
111 /* ********************************************** */
112 /* Active Object Add/Remove Operators */
113
114 /* ************ Add Rigid Body Constraint ************** */
115
116 static int rigidbody_con_add_exec(bContext *C, wmOperator *op)
117 {
118         Scene *scene = CTX_data_scene(C);
119         RigidBodyWorld *rbw = BKE_rigidbody_get_world(scene);
120         Object *ob = (scene) ? OBACT : NULL;
121         int type = RNA_enum_get(op->ptr, "type");
122
123         /* sanity checks */
124         if (ELEM(NULL, scene, rbw)) {
125                 BKE_report(op->reports, RPT_ERROR, "No Rigid Body World to add Rigid Body Constraint to");
126                 return OPERATOR_CANCELLED;
127         }
128         /* apply to active object */
129         ED_rigidbody_con_add(op, scene, ob, type);
130
131         /* send updates */
132         DAG_ids_flush_update(CTX_data_main(C), 0);
133
134         WM_event_add_notifier(C, NC_OBJECT | ND_TRANSFORM, NULL);
135         WM_event_add_notifier(C, NC_GROUP | NA_EDITED, NULL);
136
137         /* done */
138         return OPERATOR_FINISHED;
139 }
140
141 void RIGIDBODY_OT_constraint_add(wmOperatorType *ot)
142 {
143         /* identifiers */
144         ot->idname = "RIGIDBODY_OT_constraint_add";
145         ot->name = "Add Rigid Body Constraint";
146         ot->description = "Add Rigid Body Constraint to active object";
147
148         /* callbacks */
149         ot->exec = rigidbody_con_add_exec;
150         ot->poll = ED_operator_object_active_editable;
151
152         /* flags */
153         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
154
155         /* properties */
156         ot->prop = RNA_def_enum(ot->srna, "type", rigidbody_con_type_items, RBC_TYPE_FIXED, "Rigid Body Constraint Type", "");
157 }
158
159 /* ************ Remove Rigid Body Constraint ************** */
160
161 static int rigidbody_con_remove_exec(bContext *C, wmOperator *op)
162 {
163         Scene *scene = CTX_data_scene(C);
164         Object *ob = (scene) ? OBACT : NULL;
165
166         /* sanity checks */
167         if (scene == NULL)
168                 return OPERATOR_CANCELLED;
169
170         /* apply to active object */
171         if (ELEM(NULL, ob, ob->rigidbody_constraint)) {
172                 BKE_report(op->reports, RPT_ERROR, "Object has no Rigid Body Constraint to remove");
173                 return OPERATOR_CANCELLED;
174         }
175         else {
176                 ED_rigidbody_con_remove(scene, ob);
177         }
178
179         /* send updates */
180         DAG_ids_flush_update(CTX_data_main(C), 0);
181
182         WM_event_add_notifier(C, NC_OBJECT | ND_TRANSFORM, NULL);
183         WM_event_add_notifier(C, NC_GROUP | NA_EDITED, NULL);
184
185         /* done */
186         return OPERATOR_FINISHED;
187 }
188
189 void RIGIDBODY_OT_constraint_remove(wmOperatorType *ot)
190 {
191         /* identifiers */
192         ot->idname = "RIGIDBODY_OT_constraint_remove";
193         ot->name = "Remove Rigid Body Constraint";
194         ot->description = "Remove Rigid Body Constraint from Object";
195
196         /* callbacks */
197         ot->exec = rigidbody_con_remove_exec;
198         ot->poll = ED_operator_rigidbody_con_active_poll;
199
200         /* flags */
201         ot->flag = OPTYPE_REGISTER | OPTYPE_UNDO;
202 }