4 * Copyright (C) 2008 Adam Williams <broadcast at earthling dot net>
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
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.
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
18 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
22 #include "bcsignals.h"
27 #include "undostack.h"
30 UndoStack::UndoStack() : List<UndoStackItem>()
35 UndoStack::~UndoStack()
39 UndoStackItem *UndoStack::get_current_undo()
41 UndoStackItem *item = current;
42 if( item && (number_of(item) & 1) ) item = item->previous;
43 if( item && !(number_of(item) & 1) ) item = item->previous;
47 UndoStackItem *UndoStack::get_current_redo()
49 UndoStackItem *item = current ? current : first;
50 if( item && (number_of(item) & 1) ) item = item->next;
51 if( item && !(number_of(item) & 1) ) item = item->next;
56 UndoStackItem* UndoStack::push()
58 // current is only 0 if before first undo
60 current = insert_after(current);
62 current = insert_before(first);
64 // delete future undos if necessary
65 if( current && current->next ) {
66 while( current->next ) remove(last);
69 // delete oldest 2 undos if necessary
70 if( total() > UNDOLEVELS ) {
71 for( int i=0; i<2; ++i ) {
72 UndoStackItem *second = first->next;
76 if( !second->is_key() ) {
77 temp_data = second->get_data();
81 // Convert new first to key buffer.
82 if( !second->is_key() ) {
83 second->set_data(temp_data);
92 UndoStackItem* UndoStack::pull_next()
94 // use first entry if none
98 // use next entry if there is a next entry
101 // don't change current if there is no next entry
109 void UndoStack::dump(FILE *fp)
111 fprintf(fp,"UndoStack::dump\n");
112 UndoStackItem *current = last;
114 for( int i=0; current && i<32; ++i,current=PREVIOUS ) {
115 fprintf(fp,"%c %2d %14p %-24s %8d %04jx %c\n", current->is_key() ? 'k' : ' ',
116 i, current, current->get_description(), current->get_size(),
117 current->get_flags(), current == this->current ? '*' : ' ');
118 //char fn[BCSTRLEN]; sprintf(fn,"/tmp/undo%d", i); FILE *fp = fopen(fn,"w");
119 //if( fp ) { char *cp=current->get_data(); fwrite(cp,strlen(cp),1,fp); fclose(fp); delete [] cp; }
123 // undo can be incremental or direct, from key to current
124 //#define UNDO_INCREMENTAL
133 static unsigned char* apply_difference(unsigned char *before, int before_len,
134 unsigned char *patch, int patch_len, int *result_len)
136 XMLBuffer xbfr((const char *)patch, patch_len, 0);
138 xbfr.read((char *)&len, sizeof(len));
139 char *result = new char[len];
140 char *rp = result, *bp = (char*)before;
142 while( xbfr.itell() < patch_len ) {
144 xbfr.read((char*)&urecd, sizeof(urecd));
145 int ofs = rp - result;
146 if( urecd.new_ofs > ofs ) {
147 int sz = urecd.new_ofs - ofs;
151 if( urecd.new_len > 0 )
152 xbfr.read(rp, urecd.new_len);
153 bp += urecd.old_len; rp += urecd.new_len;
156 // All data from before_ptr to end of result buffer is identical
157 int sz = bp - (char *)before;
158 if( (before_len-=sz) > 0 ) {
159 memcpy(rp, bp, before_len);
163 *result_len = rp - result;
164 return (unsigned char *)result;
167 UndoStackItem::UndoStackItem()
168 : ListItem<UndoStackItem>()
170 description = data = 0;
174 session_filename = 0;
177 UndoStackItem::~UndoStackItem()
179 delete [] description;
181 delete [] session_filename;
184 void UndoStackItem::set_description(char *description)
186 delete [] this->description;
187 this->description= 0;
188 this->description = new char[strlen(description) + 1];
189 strcpy(this->description, description);
192 void UndoStackItem::set_filename(char *filename)
194 delete [] session_filename;
195 this->session_filename = cstrdup(filename);
198 char* UndoStackItem::get_filename()
200 return session_filename;
204 const char* UndoStackItem::get_description()
214 UndoHash::UndoHash(char *txt, int len, UndoHash *nxt)
219 line[va] = line[vb] = -1;
220 occurs[va] = occurs[vb] = 0;
222 UndoHash::~UndoHash()
226 UndoHashTable::UndoHashTable()
228 for( int i=0; i<hash_sz; ++i ) table[i] = 0;
229 bof = new UndoHash(0, 0, 0);
230 eof = new UndoHash(0, 0, 0);
232 UndoHashTable::~UndoHashTable()
236 for( int i=0; i<hash_sz; ++i ) {
237 for( UndoHash *nxt=0, *hp=table[i]; hp; hp=nxt ) {
238 nxt = hp->nxt; delete hp;
243 UndoHash *UndoHashTable::add(char *txt, int len)
245 uint8_t *bp = (uint8_t *)txt;
247 for( int i=len; --i>=0; ++bp ) {
249 v = (v + (v>>hash_sz2)) & hash_sz1;
251 UndoHash *hp = table[v];
252 while( hp && strncmp(hp->txt,txt,len) ) hp = hp->nxt;
254 hp = new UndoHash(txt, len, table[v]);
260 UndoLine::UndoLine(UndoHash *hash)
262 this->txt = 0; this->len = 0;
264 hash->occurs[va] = hash->occurs[vb] = 1;
267 UndoLine::UndoLine(UndoHashTable *hash, char *txt, int len)
269 this->txt = txt; this->len = len;
270 this->hash = hash->add(txt, len);
272 UndoLine::~UndoLine()
276 int UndoLine::eq(UndoLine *ln)
278 return hash == ln->hash ? 1 : 0;
281 void UndoVersion::scan_lines(UndoHashTable *hash, char *sp, char *ep)
283 append(new UndoLine(hash->bof));
284 for( int line=1; sp<ep; ++line ) {
286 while( sp<ep && *sp++ != '\n' );
287 UndoLine *ln = new UndoLine(hash, txt, sp-txt);
288 ln->hash->line[ver] = line;
289 ++ln->hash->occurs[ver];
292 append(new UndoLine(hash->eof));
296 void UndoStackItem::set_data(char *data)
298 delete [] this->data; this->data = 0;
299 this->data_size = 0; this->key = 0;
300 int data_size = strlen(data)+1;
302 UndoStackItem *current = this;
303 // Search for key buffer within interval
304 for( int i=UNDO_KEY_INTERVAL; --i>=0 && current && !current->key; current=PREVIOUS );
306 int need_key = current ? 0 : 1;
307 // setting need_key = 1 forces all undo data items to key (not diffd)
309 char *prev = 0; int prev_size = 0;
311 #ifdef UNDO_INCREMENTAL
312 prev = previous->get_data();
313 prev_size = prev ? strlen(prev)+1 : 0;
315 prev = current->data;
316 prev_size = current->data_size;
318 int min_size = data_size/16;
319 // if the previous is a sizable incremental diff
320 if( previous && !previous->key &&
321 previous->data_size > min_size ) {
322 // and this minimum diff distance is a lot different, need key
323 int dist = abs(data_size - prev_size);
324 if( abs(dist - previous->data_size) > min_size )
329 int bfr_size = BCTEXTLEN;
330 char *bfr = new char[bfr_size];
331 XMLBuffer xbfr(bfr_size, bfr, 1);
332 xbfr.write((char*)&data_size, sizeof(data_size));
334 UndoVersion alines(va), blines(vb);
335 char *asp = data, *aep = asp + data_size-1;
336 char *bsp = prev, *bep = bsp + prev_size-1;
337 alines.scan_lines(&hash, asp, aep);
338 blines.scan_lines(&hash, bsp, bep);
340 int asz = alines.size(), bsz = blines.size();
341 while( asz > 0 && bsz > 0 && alines[asz-1]->eq(blines[bsz-1]) ) {
346 while( ai < asz || bi < bsz ) {
347 // skip running match
348 if( alines[ai]->eq(blines[bi]) ) { ++ai; ++bi; continue; }
349 char *adp = alines[ai]->txt, *bdp = blines[bi]->txt;
350 // find best unique match
351 int ma = asz, mb = bsz;
352 int mx = ma + mb + 1;
353 for( int ia=ai; ia<asz && ia-ai<mx; ++ia ) {
354 UndoHash *ah = alines[ia]->hash;
355 if( ah->occurs[va] != 1 || ah->occurs[vb] != 1 ) continue;
356 int m = (ah->line[va] - ai) + (ah->line[vb] - bi);
357 if( m >= mx ) continue;
358 ma = ah->line[va]; mb = ah->line[vb]; mx = m;
361 while( ma > 0 && mb > 0 && alines[ma-1]->eq(blines[mb-1]) ) { --ma; --mb; }
362 char *ap = alines[ma]->txt, *bp = blines[mb]->txt;
365 urecd.new_ofs = adp - data;
366 urecd.new_len = ap - adp;
367 urecd.old_len = bp - bdp;
368 xbfr.write((const char*)&urecd, sizeof(urecd));
369 xbfr.write((const char*)adp, urecd.new_len);
372 int64_t pos = xbfr.otell();
373 if( abs(data_size-prev_size) < pos/2 ) {
374 this->data = new char[this->data_size = pos];
376 xbfr.read(this->data, pos);
379 char *test_data = (char*)apply_difference(
380 (unsigned char *)prev, prev_size,
381 (unsigned char *)this->data, this->data_size,
383 if( test_size != data_size || memcmp(test_data, data, test_size) ) {
384 printf("UndoStackItem::set_data: *** incremental undo failed!\n");
385 delete [] this->data; this->data = 0; this->data_size = 0;
395 #ifdef UNDO_INCREMENTAL
400 this->data = new char[this->data_size = data_size];
401 memcpy(this->data, data, this->data_size);
405 char* UndoStackItem::get_data()
407 // Find latest key buffer
408 UndoStackItem *current = this;
409 while( current && !current->key )
412 printf("UndoStackItem::get_data: no key buffer found!\n");
416 // This is the key buffer
417 if( current == this ) {
418 char *result = new char[data_size];
419 memcpy(result, data, data_size);
422 #ifdef UNDO_INCREMENTAL
424 char *current_data = current->get_data();
425 int current_size = current->data_size;
428 // Do incremental updates
430 char *new_data = (char*)apply_difference(
431 (unsigned char*)current_data, current_size,
432 (unsigned char*)current->data, current->data_size,
434 delete [] current_data;
436 if( current == this )
438 current_data = new_data;
439 current_size = new_size;
442 // Never hit this object.
443 delete [] current_data;
444 printf("UndoStackItem::get_data: lost starting object!\n");
448 char *new_data = (char*)apply_difference(
449 (unsigned char*)current->data, current->data_size,
450 (unsigned char*)this->data, this->data_size, &new_size);
455 int UndoStackItem::is_key()
460 int UndoStackItem::get_size()
465 void UndoStackItem::set_flags(uint64_t flags)
467 this->load_flags = flags;
470 uint64_t UndoStackItem::get_flags()
475 void UndoStackItem::set_creator(void *creator)
477 this->creator = creator;
480 void* UndoStackItem::get_creator()
485 void UndoStackItem::save(FILE *fp)
487 fwrite(&key,1,sizeof(key),fp);
488 fwrite(&load_flags,1,sizeof(load_flags),fp);
489 fwrite(&data_size,1,sizeof(data_size),fp);
490 fwrite(data,1,data_size,fp);
491 for( char *bp=session_filename; *bp; ++bp ) fputc(*bp, fp);
493 for( char *bp=description; *bp; ++bp ) fputc(*bp, fp);
497 void UndoStackItem::load(FILE *fp)
499 fread(&key,1,sizeof(key),fp);
500 fread(&load_flags,1,sizeof(load_flags),fp);
501 fread(&data_size,1,sizeof(data_size),fp);
502 fread(data=new char[data_size],1,data_size,fp);
503 char filename[BCTEXTLEN], descr[BCTEXTLEN];
504 char *bp = filename, *ep = bp+sizeof(filename)-1;
505 for( int ch; bp<ep && (ch=fgetc(fp))>0; ++bp ) *bp = ch;
507 session_filename = cstrdup(filename);
508 bp = descr; ep = bp+sizeof(descr)-1;
509 for( int ch; bp<ep && (ch=fgetc(fp))>0; ++bp ) *bp = ch;
511 description = cstrdup(descr);
512 //printf("read undo key=%d,flags=%jx,data_size=%d,data=%p,file=%s,descr=%s\n",
513 // key, load_flags, data_size, data, session_filename, description);
516 void UndoStack::save(FILE *fp)
518 for( UndoStackItem *item=first; item; item=item->next ) {
519 int is_current = item == current ? 1 : 0;
520 fwrite(&is_current,1,sizeof(is_current),fp);
522 // if( item == current ) break; // stop at current
526 void UndoStack::load(FILE *fp)
528 while( last ) delete last;
530 UndoStackItem *current_item = 0;
532 while( fread(&is_current,1,sizeof(is_current),fp) == sizeof(is_current) ) {
533 UndoStackItem *item = push();
535 if( is_current ) current_item = item;
537 if( current_item ) current = current_item;