source/blender/editors/util/undo.c

   1 /**
   2  * $Id:
   3  *
   4  * ***** BEGIN GPL LICENSE BLOCK *****
   5  *
   6  * This program is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU General Public License
   8  * as published by the Free Software Foundation; either version 2
   9  * of the License, or (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with this program; if not, write to the Free Software Foundation,
  18  * Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  19  *
  20  * The Original Code is Copyright (C) 2004 Blender Foundation
  21  * All rights reserved.
  22  *
  23  * The Original Code is: all of this file.
  24  *
  25  * Contributor(s): none yet.
  26  *
  27  * ***** END GPL LICENSE BLOCK *****
  28  */
  29
  30
  31 #include <stdlib.h>
  32 #include <string.h>
  33 #include <math.h>
  34
  35 #include "MEM_guardedalloc.h"
  36
  37 #include "DNA_mesh_types.h"
  38 #include "DNA_meshdata_types.h"
  39 #include "DNA_object_types.h"
  40 #include "DNA_screen_types.h"
  41 #include "DNA_scene_types.h"
  42 #include "DNA_space_types.h"
  43 #include "DNA_userdef_types.h"
  44
  45 #include "BKE_blender.h"
  46 #include "BKE_context.h"
  47 #include "BKE_depsgraph.h"
  48 #include "BKE_global.h"
  49 #include "BKE_object.h"
  50 #include "BKE_text.h"
  51
  52 #include "BLI_blenlib.h"
  53 #include "BLI_editVert.h"
  54 #include "BLI_dynstr.h"
  55
  56 #include "BKE_utildefines.h"
  57
  58 #include "ED_armature.h"
  59 #include "ED_particle.h"
  60 #include "ED_curve.h"
  61 #include "ED_mball.h"
  62 #include "ED_mesh.h"
  63 #include "ED_object.h"
  64 #include "ED_screen.h"
  65 #include "ED_sculpt.h"
  66 #include "ED_util.h"
  67 #include "ED_text.h"
  68
  69 #include "WM_api.h"
  70 #include "WM_types.h"
  71
  72 #include "UI_interface.h"
  73 #include "UI_resources.h"
  74
  75 #include "util_intern.h"
  76
  77 /* ***************** generic undo system ********************* */
  78
  79 void ED_undo_push(bContext *C, char *str)
  80 {
  81         wmWindowManager *wm= CTX_wm_manager(C);
  82         Object *obedit= CTX_data_edit_object(C);
  83         Object *obact= CTX_data_active_object(C);
  84
  85         if(obedit) {
  86                 if (U.undosteps == 0) return;
  87
  88                 if(obedit->type==OB_MESH)
  89                         undo_push_mesh(C, str);
  90                 else if ELEM(obedit->type, OB_CURVE, OB_SURF)
  91                         undo_push_curve(C, str);
  92                 else if (obedit->type==OB_FONT)
  93                         undo_push_font(C, str);
  94                 else if (obedit->type==OB_MBALL)
  95                         undo_push_mball(C, str);
  96                 else if (obedit->type==OB_LATTICE)
  97                         undo_push_lattice(C, str);
  98                 else if (obedit->type==OB_ARMATURE)
  99                         undo_push_armature(C, str);
 100         }
 101         else if(obact && obact->mode & OB_MODE_PARTICLE_EDIT) {
 102                 if (U.undosteps == 0) return;
 103
 104                 PE_undo_push(CTX_data_scene(C), str);
 105         }
 106         else {
 107                 if(U.uiflag & USER_GLOBALUNDO)
 108                         BKE_write_undo(C, str);
 109         }
 110
 111         if(wm->file_saved) {
 112                 wm->file_saved= 0;
 113                 WM_event_add_notifier(C, NC_WM|ND_DATACHANGED, NULL);
 114         }
 115 }
 116
 117 static int ed_undo_step(bContext *C, int step, const char *undoname)
 118 {
 119         Object *obedit= CTX_data_edit_object(C);
 120         Object *obact= CTX_data_active_object(C);
 121         ScrArea *sa= CTX_wm_area(C);
 122
 123         if(sa && sa->spacetype==SPACE_IMAGE) {
 124                 SpaceImage *sima= (SpaceImage *)sa->spacedata.first;
 125
 126                 if((obact && obact->mode & OB_MODE_TEXTURE_PAINT) || sima->flag & SI_DRAWTOOL) {
 127                         ED_undo_paint_step(C, UNDO_PAINT_IMAGE, step);
 128
 129                         WM_event_add_notifier(C, NC_WINDOW, NULL);
 130                         return OPERATOR_FINISHED;
 131                 }
 132         }
 133
 134         if(sa && sa->spacetype==SPACE_TEXT) {
 135                 ED_text_undo_step(C, step);
 136         }
 137         else if(obedit) {
 138                 if ELEM7(obedit->type, OB_MESH, OB_FONT, OB_CURVE, OB_SURF, OB_MBALL, OB_LATTICE, OB_ARMATURE) {
 139                         if(undoname)
 140                                 undo_editmode_name(C, undoname);
 141                         else
 142                                 undo_editmode_step(C, step);
 143                 }
 144         }
 145         else {
 146                 int do_glob_undo= 0;
 147
 148                 if(obact && obact->mode & OB_MODE_TEXTURE_PAINT)
 149                         ED_undo_paint_step(C, UNDO_PAINT_IMAGE, step);
 150                 else if(obact && obact->mode & OB_MODE_SCULPT)
 151                         ED_undo_paint_step(C, UNDO_PAINT_MESH, step);
 152                 else if(obact && obact->mode & OB_MODE_PARTICLE_EDIT) {
 153                         if(step==1)
 154                                 PE_undo(CTX_data_scene(C));
 155                         else
 156                                 PE_redo(CTX_data_scene(C));
 157                 }
 158                 else {
 159                         do_glob_undo= 1;
 160                 }
 161
 162                 if(do_glob_undo) {
 163                         if(U.uiflag & USER_GLOBALUNDO) {
 164 #ifndef DISABLE_PYTHON
 165                                 // XXX          BPY_scripts_clear_pyobjects();
 166 #endif
 167                                 if(undoname)
 168                                         BKE_undo_name(C, undoname);
 169                                 else
 170                                         BKE_undo_step(C, step);
 171                         }
 172
 173                 }
 174         }
 175
 176         WM_event_add_notifier(C, NC_WINDOW, NULL);
 177
 178         return OPERATOR_FINISHED;
 179 }
 180
 181 void ED_undo_pop(bContext *C)
 182 {
 183         ed_undo_step(C, 1, NULL);
 184 }
 185 void ED_undo_redo(bContext *C)
 186 {
 187         ed_undo_step(C, -1, NULL);
 188 }
 189
 190 void ED_undo_push_op(bContext *C, wmOperator *op)
 191 {
 192         /* in future, get undo string info? */
 193         ED_undo_push(C, op->type->name);
 194 }
 195
 196 void ED_undo_pop_op(bContext *C, wmOperator *op)
 197 {
 198         /* search back a couple of undo's, in case something else added pushes */
 199         ed_undo_step(C, 0, op->type->name);
 200 }
 201
 202 static int ed_undo_exec(bContext *C, wmOperator *op)
 203 {
 204         /* "last operator" should disappear, later we can tie ths with undo stack nicer */
 205         WM_operator_stack_clear(C);
 206         return ed_undo_step(C, 1, NULL);
 207 }
 208
 209 static int ed_redo_exec(bContext *C, wmOperator *op)
 210 {
 211         return ed_undo_step(C, -1, NULL);
 212 }
 213
 214 void ED_undo_menu(bContext *C)
 215 {
 216         Object *obedit= CTX_data_edit_object(C);
 217         Object *obact= CTX_data_active_object(C);
 218
 219         if(obedit) {
 220                 //if ELEM7(obedit->type, OB_MESH, OB_FONT, OB_CURVE, OB_SURF, OB_MBALL, OB_LATTICE, OB_ARMATURE)
 221                 //      undo_editmode_menu();
 222         }
 223         else {
 224                 if(obact && obact->mode & OB_MODE_PARTICLE_EDIT)
 225                         PE_undo_menu(CTX_data_scene(C), CTX_data_active_object(C));
 226                 else if(U.uiflag & USER_GLOBALUNDO) {
 227                         char *menu= BKE_undo_menu_string();
 228                         if(menu) {
 229                                 short event= 0; // XXX pupmenu_col(menu, 20);
 230                                 MEM_freeN(menu);
 231                                 if(event>0) {
 232                                         BKE_undo_number(C, event);
 233                                 }
 234                         }
 235                 }
 236         }
 237 }
 238
 239 /* ********************** */
 240
 241 void ED_OT_undo(wmOperatorType *ot)
 242 {
 243         /* identifiers */
 244         ot->name= "Undo";
 245     ot->description= "Undo previous action.";
 246         ot->idname= "ED_OT_undo";
 247
 248         /* api callbacks */
 249         ot->exec= ed_undo_exec;
 250         ot->poll= ED_operator_screenactive;
 251 }
 252
 253 void ED_OT_redo(wmOperatorType *ot)
 254 {
 255         /* identifiers */
 256         ot->name= "Redo";
 257     ot->description= "Redo previous action.";
 258         ot->idname= "ED_OT_redo";
 259
 260         /* api callbacks */
 261         ot->exec= ed_redo_exec;
 262         ot->poll= ED_operator_screenactive;
 263 }
 264
 265