source/blender/blenkernel/intern/pointcache.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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  19  *
  20  * The Original Code is Copyright (C) 2001-2002 by NaN Holding BV.
  21  * All rights reserved.
  22  *
  23  * Contributor(s): Campbell Barton <ideasman42@gmail.com>
  24  *
  25  * ***** END GPL LICENSE BLOCK *****
  26  */
  27
  28 /** \file blender/blenkernel/intern/pointcache.c
  29  *  \ingroup bke
  30  */
  31
  32
  33 #include <stdlib.h>
  34 #include <stdio.h>
  35 #include <string.h>
  36 #include <sys/stat.h>
  37 #include <sys/types.h>
  38
  39 #include "MEM_guardedalloc.h"
  40
  41 #include "DNA_ID.h"
  42 #include "DNA_cloth_types.h"
  43 #include "DNA_modifier_types.h"
  44 #include "DNA_object_types.h"
  45 #include "DNA_object_force.h"
  46 #include "DNA_particle_types.h"
  47 #include "DNA_scene_types.h"
  48 #include "DNA_smoke_types.h"
  49
  50 #include "BLI_blenlib.h"
  51 #include "BLI_threads.h"
  52 #include "BLI_math.h"
  53 #include "BLI_utildefines.h"
  54
  55 #include "PIL_time.h"
  56
  57 #include "WM_api.h"
  58
  59 #include "BKE_anim.h"
  60 #include "BKE_blender.h"
  61 #include "BKE_cloth.h"
  62 #include "BKE_depsgraph.h"
  63 #include "BKE_global.h"
  64 #include "BKE_library.h"
  65 #include "BKE_main.h"
  66 #include "BKE_object.h"
  67 #include "BKE_particle.h"
  68 #include "BKE_pointcache.h"
  69 #include "BKE_scene.h"
  70 #include "BKE_smoke.h"
  71 #include "BKE_softbody.h"
  72 #include "BKE_utildefines.h"
  73
  74 #include "BIK_api.h"
  75
  76 /* both in intern */
  77 #include "smoke_API.h"
  78
  79 #ifdef WITH_LZO
  80 #include "minilzo.h"
  81 #else
  82 /* used for non-lzo cases */
  83 #define LZO_OUT_LEN(size)     ((size) + (size) / 16 + 64 + 3)
  84 #endif
  85
  86 #ifdef WITH_LZMA
  87 #include "LzmaLib.h"
  88 #endif
  89
  90 /* needed for directory lookup */
  91 /* untitled blend's need getpid for a unique name */
  92 #ifndef WIN32
  93   #include <dirent.h>
  94 #include <unistd.h>
  95 #else
  96 #include <process.h>
  97   #include "BLI_winstuff.h"
  98 #endif
  99
 100 #define PTCACHE_DATA_FROM(data, type, from)             if(data[type]) { memcpy(data[type], from, ptcache_data_size[type]); }
 101 #define PTCACHE_DATA_TO(data, type, index, to)  if(data[type]) { memcpy(to, (char*)data[type] + (index ? index * ptcache_data_size[type] : 0), ptcache_data_size[type]); }
 102
 103 /* could be made into a pointcache option */
 104 #define DURIAN_POINTCACHE_LIB_OK 1
 105
 106 static int ptcache_data_size[] = {
 107                 sizeof(unsigned int), // BPHYS_DATA_INDEX
 108                 3 * sizeof(float), // BPHYS_DATA_LOCATION
 109                 3 * sizeof(float), // BPHYS_DATA_VELOCITY
 110                 4 * sizeof(float), // BPHYS_DATA_ROTATION
 111                 3 * sizeof(float), // BPHYS_DATA_AVELOCITY / BPHYS_DATA_XCONST
 112                 sizeof(float), // BPHYS_DATA_SIZE
 113                 3 * sizeof(float), // BPHYS_DATA_TIMES
 114                 sizeof(BoidData) // case BPHYS_DATA_BOIDS
 115 };
 116
 117 static int ptcache_extra_datasize[] = {
 118         0,
 119         sizeof(ParticleSpring)
 120 };
 121
 122 /* forward declerations */
 123 static int ptcache_file_compressed_read(PTCacheFile *pf, unsigned char *result, unsigned int len);
 124 static int ptcache_file_compressed_write(PTCacheFile *pf, unsigned char *in, unsigned int in_len, unsigned char *out, int mode);
 125 static int ptcache_file_write(PTCacheFile *pf, void *f, unsigned int tot, unsigned int size);
 126 static int ptcache_file_read(PTCacheFile *pf, void *f, unsigned int tot, unsigned int size);
 127
 128 /* Common functions */
 129 static int ptcache_basic_header_read(PTCacheFile *pf)
 130 {
 131         int error=0;
 132
 133         /* Custom functions should read these basic elements too! */
 134         if(!error && !fread(&pf->totpoint, sizeof(unsigned int), 1, pf->fp))
 135                 error = 1;
 136
 137         if(!error && !fread(&pf->data_types, sizeof(unsigned int), 1, pf->fp))
 138                 error = 1;
 139
 140         return !error;
 141 }
 142 static int ptcache_basic_header_write(PTCacheFile *pf)
 143 {
 144         /* Custom functions should write these basic elements too! */
 145         if(!fwrite(&pf->totpoint, sizeof(unsigned int), 1, pf->fp))
 146                 return 0;
 147
 148         if(!fwrite(&pf->data_types, sizeof(unsigned int), 1, pf->fp))
 149                 return 0;
 150
 151         return 1;
 152 }
 153 /* Softbody functions */
 154 static int  ptcache_softbody_write(int index, void *soft_v, void **data, int UNUSED(cfra))
 155 {
 156         SoftBody *soft= soft_v;
 157         BodyPoint *bp = soft->bpoint + index;
 158
 159         PTCACHE_DATA_FROM(data, BPHYS_DATA_LOCATION, bp->pos);
 160         PTCACHE_DATA_FROM(data, BPHYS_DATA_VELOCITY, bp->vec);
 161
 162         return 1;
 163 }
 164 static void ptcache_softbody_read(int index, void *soft_v, void **data, float UNUSED(cfra), float *old_data)
 165 {
 166         SoftBody *soft= soft_v;
 167         BodyPoint *bp = soft->bpoint + index;
 168
 169         if(old_data) {
 170                 memcpy(bp->pos, data, 3 * sizeof(float));
 171                 memcpy(bp->vec, data + 3, 3 * sizeof(float));
 172         }
 173         else {
 174                 PTCACHE_DATA_TO(data, BPHYS_DATA_LOCATION, 0, bp->pos);
 175                 PTCACHE_DATA_TO(data, BPHYS_DATA_VELOCITY, 0, bp->vec);
 176         }
 177 }
 178 static void ptcache_softbody_interpolate(int index, void *soft_v, void **data, float cfra, float cfra1, float cfra2, float *old_data)
 179 {
 180         SoftBody *soft= soft_v;
 181         BodyPoint *bp = soft->bpoint + index;
 182         ParticleKey keys[4];
 183         float dfra;
 184
 185         if(cfra1 == cfra2)
 186                 return;
 187
 188         VECCOPY(keys[1].co, bp->pos);
 189         VECCOPY(keys[1].vel, bp->vec);
 190
 191         if(old_data) {
 192                 memcpy(keys[2].co, old_data, 3 * sizeof(float));
 193                 memcpy(keys[2].vel, old_data + 3, 3 * sizeof(float));
 194         }
 195         else
 196                 BKE_ptcache_make_particle_key(keys+2, 0, data, cfra2);
 197
 198         dfra = cfra2 - cfra1;
 199
 200         mul_v3_fl(keys[1].vel, dfra);
 201         mul_v3_fl(keys[2].vel, dfra);
 202
 203         psys_interpolate_particle(-1, keys, (cfra - cfra1) / dfra, keys, 1);
 204
 205         mul_v3_fl(keys->vel, 1.0f / dfra);
 206
 207         VECCOPY(bp->pos, keys->co);
 208         VECCOPY(bp->vec, keys->vel);
 209 }
 210 static int  ptcache_softbody_totpoint(void *soft_v, int UNUSED(cfra))
 211 {
 212         SoftBody *soft= soft_v;
 213         return soft->totpoint;
 214 }
 215 /* Particle functions */
 216 void BKE_ptcache_make_particle_key(ParticleKey *key, int index, void **data, float time)
 217 {
 218         PTCACHE_DATA_TO(data, BPHYS_DATA_LOCATION, index, key->co);
 219         PTCACHE_DATA_TO(data, BPHYS_DATA_VELOCITY, index, key->vel);
 220
 221         /* no rotation info, so make something nice up */
 222         if(data[BPHYS_DATA_ROTATION]==NULL) {
 223                 vec_to_quat( key->rot, key->vel, OB_NEGX, OB_POSZ);
 224         }
 225         else {
 226                 PTCACHE_DATA_TO(data, BPHYS_DATA_ROTATION, index, key->rot);
 227         }
 228
 229         PTCACHE_DATA_TO(data, BPHYS_DATA_AVELOCITY, index, key->ave);
 230         key->time = time;
 231 }
 232 static int  ptcache_particle_write(int index, void *psys_v, void **data, int cfra)
 233 {
 234         ParticleSystem *psys= psys_v;
 235         ParticleData *pa = psys->particles + index;
 236         BoidParticle *boid = (psys->part->phystype == PART_PHYS_BOIDS) ? pa->boid : NULL;
 237         float times[3];
 238         int step = psys->pointcache->step;
 239
 240         /* No need to store unborn or died particles outside cache step bounds */
 241         if(data[BPHYS_DATA_INDEX] && (cfra < pa->time - step || cfra > pa->dietime + step))
 242                 return 0;
 243
 244         times[0]= pa->time;
 245         times[1]= pa->dietime;
 246         times[2]= pa->lifetime;
 247
 248         PTCACHE_DATA_FROM(data, BPHYS_DATA_INDEX, &index);
 249         PTCACHE_DATA_FROM(data, BPHYS_DATA_LOCATION, pa->state.co);
 250         PTCACHE_DATA_FROM(data, BPHYS_DATA_VELOCITY, pa->state.vel);
 251         PTCACHE_DATA_FROM(data, BPHYS_DATA_ROTATION, pa->state.rot);
 252         PTCACHE_DATA_FROM(data, BPHYS_DATA_AVELOCITY, pa->state.ave);
 253         PTCACHE_DATA_FROM(data, BPHYS_DATA_SIZE, &pa->size);
 254         PTCACHE_DATA_FROM(data, BPHYS_DATA_TIMES, times);
 255
 256         if(boid)
 257                 PTCACHE_DATA_FROM(data, BPHYS_DATA_BOIDS, &boid->data);
 258
 259         /* return flag 1+1=2 for newly born particles to copy exact birth location to previously cached frame */
 260         return 1 + (pa->state.time >= pa->time && pa->prev_state.time <= pa->time);
 261 }
 262 static void ptcache_particle_read(int index, void *psys_v, void **data, float cfra, float *old_data)
 263 {
 264         ParticleSystem *psys= psys_v;
 265         ParticleData *pa;
 266         BoidParticle *boid;
 267         float timestep = 0.04f*psys->part->timetweak;
 268
 269         if(index >= psys->totpart)
 270                 return;
 271
 272         pa = psys->particles + index;
 273         boid = (psys->part->phystype == PART_PHYS_BOIDS) ? pa->boid : NULL;
 274
 275         if(cfra > pa->state.time)
 276                 memcpy(&pa->prev_state, &pa->state, sizeof(ParticleKey));
 277
 278         if(old_data){
 279                 /* old format cache */
 280                 memcpy(&pa->state, old_data, sizeof(ParticleKey));
 281                 return;
 282         }
 283
 284         BKE_ptcache_make_particle_key(&pa->state, 0, data, cfra);
 285
 286         /* set frames cached before birth to birth time */
 287         if(cfra < pa->time)
 288                 pa->state.time = pa->time;
 289         else if(cfra > pa->dietime)
 290                 pa->state.time = pa->dietime;
 291
 292         if(data[BPHYS_DATA_SIZE])
 293                 PTCACHE_DATA_TO(data, BPHYS_DATA_SIZE, 0, &pa->size);
 294
 295         if(data[BPHYS_DATA_TIMES]) {
 296                 float times[3];
 297                 PTCACHE_DATA_TO(data, BPHYS_DATA_TIMES, 0, &times);
 298                 pa->time = times[0];
 299                 pa->dietime = times[1];
 300                 pa->lifetime = times[2];
 301         }
 302
 303         if(boid)
 304                 PTCACHE_DATA_TO(data, BPHYS_DATA_BOIDS, 0, &boid->data);
 305
 306         /* determine velocity from previous location */
 307         if(data[BPHYS_DATA_LOCATION] && !data[BPHYS_DATA_VELOCITY]) {
 308                 if(cfra > pa->prev_state.time) {
 309                         sub_v3_v3v3(pa->state.vel, pa->state.co, pa->prev_state.co);
 310                         mul_v3_fl(pa->state.vel, (cfra - pa->prev_state.time) * timestep);
 311                 }
 312                 else {
 313                         sub_v3_v3v3(pa->state.vel, pa->prev_state.co, pa->state.co);
 314                         mul_v3_fl(pa->state.vel, (pa->prev_state.time - cfra) * timestep);
 315                 }
 316         }
 317
 318         /* determine rotation from velocity */
 319         if(data[BPHYS_DATA_LOCATION] && !data[BPHYS_DATA_ROTATION]) {
 320                 vec_to_quat( pa->state.rot,pa->state.vel, OB_NEGX, OB_POSZ);
 321         }
 322 }
 323 static void ptcache_particle_interpolate(int index, void *psys_v, void **data, float cfra, float cfra1, float cfra2, float *old_data)
 324 {
 325         ParticleSystem *psys= psys_v;
 326         ParticleData *pa;
 327         ParticleKey keys[4];
 328         float dfra, timestep = 0.04f*psys->part->timetweak;
 329
 330         if(index >= psys->totpart)
 331                 return;
 332
 333         pa = psys->particles + index;
 334
 335         /* particle wasn't read from first cache so can't interpolate */
 336         if((int)cfra1 < pa->time - psys->pointcache->step || (int)cfra1 > pa->dietime + psys->pointcache->step)
 337                 return;
 338
 339         cfra = MIN2(cfra, pa->dietime);
 340         cfra1 = MIN2(cfra1, pa->dietime);
 341         cfra2 = MIN2(cfra2, pa->dietime);
 342
 343         if(cfra1 == cfra2)
 344                 return;
 345
 346         memcpy(keys+1, &pa->state, sizeof(ParticleKey));
 347         if(old_data)
 348                 memcpy(keys+2, old_data, sizeof(ParticleKey));
 349         else
 350                 BKE_ptcache_make_particle_key(keys+2, 0, data, cfra2);
 351
 352         /* determine velocity from previous location */
 353         if(data[BPHYS_DATA_LOCATION] && !data[BPHYS_DATA_VELOCITY]) {
 354                 if(keys[1].time > keys[2].time) {
 355                         sub_v3_v3v3(keys[2].vel, keys[1].co, keys[2].co);
 356                         mul_v3_fl(keys[2].vel, (keys[1].time - keys[2].time) * timestep);
 357                 }
 358                 else {
 359                         sub_v3_v3v3(keys[2].vel, keys[2].co, keys[1].co);
 360                         mul_v3_fl(keys[2].vel, (keys[2].time - keys[1].time) * timestep);
 361                 }
 362         }
 363
 364         /* determine rotation from velocity */
 365         if(data[BPHYS_DATA_LOCATION] && !data[BPHYS_DATA_ROTATION]) {
 366                 vec_to_quat( keys[2].rot,keys[2].vel, OB_NEGX, OB_POSZ);
 367         }
 368
 369         if(cfra > pa->time)
 370                 cfra1 = MAX2(cfra1, pa->time);
 371
 372         dfra = cfra2 - cfra1;
 373
 374         mul_v3_fl(keys[1].vel, dfra * timestep);
 375         mul_v3_fl(keys[2].vel, dfra * timestep);
 376
 377         psys_interpolate_particle(-1, keys, (cfra - cfra1) / dfra, &pa->state, 1);
 378         interp_qt_qtqt(pa->state.rot, keys[1].rot, keys[2].rot, (cfra - cfra1) / dfra);
 379
 380         mul_v3_fl(pa->state.vel, 1.f / (dfra * timestep));
 381
 382         pa->state.time = cfra;
 383 }
 384
 385 static int  ptcache_particle_totpoint(void *psys_v, int UNUSED(cfra))
 386 {
 387         ParticleSystem *psys = psys_v;
 388         return psys->totpart;
 389 }
 390 static int  ptcache_particle_totwrite(void *psys_v, int cfra)
 391 {
 392         ParticleSystem *psys = psys_v;
 393         ParticleData *pa= psys->particles;
 394         int p, step = psys->pointcache->step;
 395         int totwrite = 0;
 396
 397         if(cfra == 0)
 398                 return psys->totpart;
 399
 400         for(p=0; p<psys->totpart; p++,pa++)
 401                 totwrite += (cfra >= pa->time - step && cfra <= pa->dietime + step);
 402
 403         return totwrite;
 404 }
 405
 406 static void ptcache_particle_extra_write(void *psys_v, PTCacheMem *pm, int UNUSED(cfra))
 407 {
 408         ParticleSystem *psys = psys_v;
 409         PTCacheExtra *extra = NULL;
 410
 411         if(psys->part->phystype == PART_PHYS_FLUID &&
 412                 psys->part->fluid && psys->part->fluid->flag & SPH_VISCOELASTIC_SPRINGS &&
 413                 psys->tot_fluidsprings && psys->fluid_springs) {
 414
 415                 extra = MEM_callocN(sizeof(PTCacheExtra), "Point cache: fluid extra data");
 416
 417                 extra->type = BPHYS_EXTRA_FLUID_SPRINGS;
 418                 extra->totdata = psys->tot_fluidsprings;
 419
 420                 extra->data = MEM_callocN(extra->totdata * ptcache_extra_datasize[extra->type], "Point cache: extra data");
 421                 memcpy(extra->data, psys->fluid_springs, extra->totdata * ptcache_extra_datasize[extra->type]);
 422
 423                 BLI_addtail(&pm->extradata, extra);
 424         }
 425 }
 426
 427 static void ptcache_particle_extra_read(void *psys_v, PTCacheMem *pm, float UNUSED(cfra))
 428 {
 429         ParticleSystem *psys = psys_v;
 430         PTCacheExtra *extra = pm->extradata.first;
 431
 432         for(; extra; extra=extra->next) {
 433                 switch(extra->type) {
 434                         case BPHYS_EXTRA_FLUID_SPRINGS:
 435                         {
 436                                 if(psys->fluid_springs)
 437                                         MEM_freeN(psys->fluid_springs);
 438
 439                                 psys->fluid_springs = MEM_dupallocN(extra->data);
 440                                 psys->tot_fluidsprings = psys->alloc_fluidsprings = extra->totdata;
 441                                 break;
 442                         }
 443                 }
 444         }
 445 }
 446
 447 /* Cloth functions */
 448 static int  ptcache_cloth_write(int index, void *cloth_v, void **data, int UNUSED(cfra))
 449 {
 450         ClothModifierData *clmd= cloth_v;
 451         Cloth *cloth= clmd->clothObject;
 452         ClothVertex *vert = cloth->verts + index;
 453
 454         PTCACHE_DATA_FROM(data, BPHYS_DATA_LOCATION, vert->x);
 455         PTCACHE_DATA_FROM(data, BPHYS_DATA_VELOCITY, vert->v);
 456         PTCACHE_DATA_FROM(data, BPHYS_DATA_XCONST, vert->xconst);
 457
 458         return 1;
 459 }
 460 static void ptcache_cloth_read(int index, void *cloth_v, void **data, float UNUSED(cfra), float *old_data)
 461 {
 462         ClothModifierData *clmd= cloth_v;
 463         Cloth *cloth= clmd->clothObject;
 464         ClothVertex *vert = cloth->verts + index;
 465
 466         if(old_data) {
 467                 memcpy(vert->x, data, 3 * sizeof(float));
 468                 memcpy(vert->xconst, data + 3, 3 * sizeof(float));
 469                 memcpy(vert->v, data + 6, 3 * sizeof(float));
 470         }
 471         else {
 472                 PTCACHE_DATA_TO(data, BPHYS_DATA_LOCATION, 0, vert->x);
 473                 PTCACHE_DATA_TO(data, BPHYS_DATA_VELOCITY, 0, vert->v);
 474                 PTCACHE_DATA_TO(data, BPHYS_DATA_XCONST, 0, vert->xconst);
 475         }
 476 }
 477 static void ptcache_cloth_interpolate(int index, void *cloth_v, void **data, float cfra, float cfra1, float cfra2, float *old_data)
 478 {
 479         ClothModifierData *clmd= cloth_v;
 480         Cloth *cloth= clmd->clothObject;
 481         ClothVertex *vert = cloth->verts + index;
 482         ParticleKey keys[4];
 483         float dfra;
 484
 485         if(cfra1 == cfra2)
 486                 return;
 487
 488         VECCOPY(keys[1].co, vert->x);
 489         VECCOPY(keys[1].vel, vert->v);
 490
 491         if(old_data) {
 492                 memcpy(keys[2].co, old_data, 3 * sizeof(float));
 493                 memcpy(keys[2].vel, old_data + 6, 3 * sizeof(float));
 494         }
 495         else
 496                 BKE_ptcache_make_particle_key(keys+2, 0, data, cfra2);
 497
 498         dfra = cfra2 - cfra1;
 499
 500         mul_v3_fl(keys[1].vel, dfra);
 501         mul_v3_fl(keys[2].vel, dfra);
 502
 503         psys_interpolate_particle(-1, keys, (cfra - cfra1) / dfra, keys, 1);
 504
 505         mul_v3_fl(keys->vel, 1.0f / dfra);
 506
 507         VECCOPY(vert->x, keys->co);
 508         VECCOPY(vert->v, keys->vel);
 509
 510         /* should vert->xconst be interpolated somehow too? - jahka */
 511 }
 512
 513 static int  ptcache_cloth_totpoint(void *cloth_v, int UNUSED(cfra))
 514 {
 515         ClothModifierData *clmd= cloth_v;
 516         return clmd->clothObject ? clmd->clothObject->numverts : 0;
 517 }
 518
 519 /* Smoke functions */
 520 static int  ptcache_smoke_totpoint(void *smoke_v, int UNUSED(cfra))
 521 {
 522         SmokeModifierData *smd= (SmokeModifierData *)smoke_v;
 523         SmokeDomainSettings *sds = smd->domain;
 524
 525         if(sds->fluid) {
 526                 return sds->res[0]*sds->res[1]*sds->res[2];
 527         }
 528         else
 529                 return 0;
 530 }
 531 static int  ptcache_smoke_write(PTCacheFile *pf, void *smoke_v)
 532 {
 533         SmokeModifierData *smd= (SmokeModifierData *)smoke_v;
 534         SmokeDomainSettings *sds = smd->domain;
 535         int ret = 0;
 536
 537         if(sds->fluid) {
 538                 size_t res = sds->res[0]*sds->res[1]*sds->res[2];
 539                 float dt, dx, *dens, *densold, *heat, *heatold, *vx, *vy, *vz, *vxold, *vyold, *vzold;
 540                 unsigned char *obstacles;
 541                 unsigned int in_len = sizeof(float)*(unsigned int)res;
 542                 unsigned char *out = (unsigned char *)MEM_callocN(LZO_OUT_LEN(in_len)*4, "pointcache_lzo_buffer");
 543                 //int mode = res >= 1000000 ? 2 : 1;
 544                 int mode=1;             // light
 545                 if (sds->cache_comp == SM_CACHE_HEAVY) mode=2;  // heavy
 546
 547                 smoke_export(sds->fluid, &dt, &dx, &dens, &densold, &heat, &heatold, &vx, &vy, &vz, &vxold, &vyold, &vzold, &obstacles);
 548
 549                 ptcache_file_compressed_write(pf, (unsigned char *)sds->shadow, in_len, out, mode);
 550                 ptcache_file_compressed_write(pf, (unsigned char *)dens, in_len, out, mode);
 551                 ptcache_file_compressed_write(pf, (unsigned char *)densold, in_len, out, mode);
 552                 ptcache_file_compressed_write(pf, (unsigned char *)heat, in_len, out, mode);
 553                 ptcache_file_compressed_write(pf, (unsigned char *)heatold, in_len, out, mode);
 554                 ptcache_file_compressed_write(pf, (unsigned char *)vx, in_len, out, mode);
 555                 ptcache_file_compressed_write(pf, (unsigned char *)vy, in_len, out, mode);
 556                 ptcache_file_compressed_write(pf, (unsigned char *)vz, in_len, out, mode);
 557                 ptcache_file_compressed_write(pf, (unsigned char *)vxold, in_len, out, mode);
 558                 ptcache_file_compressed_write(pf, (unsigned char *)vyold, in_len, out, mode);
 559                 ptcache_file_compressed_write(pf, (unsigned char *)vzold, in_len, out, mode);
 560                 ptcache_file_compressed_write(pf, (unsigned char *)obstacles, (unsigned int)res, out, mode);
 561                 ptcache_file_write(pf, &dt, 1, sizeof(float));
 562                 ptcache_file_write(pf, &dx, 1, sizeof(float));
 563
 564                 MEM_freeN(out);
 565
 566                 ret = 1;
 567         }
 568
 569         if(sds->wt) {
 570                 int res_big_array[3];
 571                 int res_big;
 572                 int res = sds->res[0]*sds->res[1]*sds->res[2];
 573                 float *dens, *densold, *tcu, *tcv, *tcw;
 574                 unsigned int in_len = sizeof(float)*(unsigned int)res;
 575                 unsigned int in_len_big;
 576                 unsigned char *out;
 577                 int mode;
 578
 579                 smoke_turbulence_get_res(sds->wt, res_big_array);
 580                 res_big = res_big_array[0]*res_big_array[1]*res_big_array[2];
 581                 //mode =  res_big >= 1000000 ? 2 : 1;
 582                 mode = 1;       // light
 583                 if (sds->cache_high_comp == SM_CACHE_HEAVY) mode=2;     // heavy
 584
 585                 in_len_big = sizeof(float) * (unsigned int)res_big;
 586
 587                 smoke_turbulence_export(sds->wt, &dens, &densold, &tcu, &tcv, &tcw);
 588
 589                 out = (unsigned char *)MEM_callocN(LZO_OUT_LEN(in_len_big), "pointcache_lzo_buffer");
 590                 ptcache_file_compressed_write(pf, (unsigned char *)dens, in_len_big, out, mode);
 591                 ptcache_file_compressed_write(pf, (unsigned char *)densold, in_len_big, out, mode);
 592                 MEM_freeN(out);
 593
 594                 out = (unsigned char *)MEM_callocN(LZO_OUT_LEN(in_len), "pointcache_lzo_buffer");
 595                 ptcache_file_compressed_write(pf, (unsigned char *)tcu, in_len, out, mode);
 596                 ptcache_file_compressed_write(pf, (unsigned char *)tcv, in_len, out, mode);
 597                 ptcache_file_compressed_write(pf, (unsigned char *)tcw, in_len, out, mode);
 598                 MEM_freeN(out);
 599
 600                 ret = 1;
 601         }
 602
 603         return ret;
 604 }
 605 static void ptcache_smoke_read(PTCacheFile *pf, void *smoke_v)
 606 {
 607         SmokeModifierData *smd= (SmokeModifierData *)smoke_v;
 608         SmokeDomainSettings *sds = smd->domain;
 609
 610         if(sds->fluid) {
 611                 size_t res = sds->res[0]*sds->res[1]*sds->res[2];
 612                 float dt, dx, *dens, *densold, *heat, *heatold, *vx, *vy, *vz, *vxold, *vyold, *vzold;
 613                 unsigned char *obstacles;
 614                 unsigned int out_len = (unsigned int)res * sizeof(float);
 615
 616                 smoke_export(sds->fluid, &dt, &dx, &dens, &densold, &heat, &heatold, &vx, &vy, &vz, &vxold, &vyold, &vzold, &obstacles);
 617
 618                 ptcache_file_compressed_read(pf, (unsigned char *)sds->shadow, out_len);
 619                 ptcache_file_compressed_read(pf, (unsigned char*)dens, out_len);
 620                 ptcache_file_compressed_read(pf, (unsigned char*)densold, out_len);
 621                 ptcache_file_compressed_read(pf, (unsigned char*)heat, out_len);
 622                 ptcache_file_compressed_read(pf, (unsigned char*)heatold, out_len);
 623                 ptcache_file_compressed_read(pf, (unsigned char*)vx, out_len);
 624                 ptcache_file_compressed_read(pf, (unsigned char*)vy, out_len);
 625                 ptcache_file_compressed_read(pf, (unsigned char*)vz, out_len);
 626                 ptcache_file_compressed_read(pf, (unsigned char*)vxold, out_len);
 627                 ptcache_file_compressed_read(pf, (unsigned char*)vyold, out_len);
 628                 ptcache_file_compressed_read(pf, (unsigned char*)vzold, out_len);
 629                 ptcache_file_compressed_read(pf, (unsigned char*)obstacles, (unsigned int)res);
 630                 ptcache_file_read(pf, &dt, 1, sizeof(float));
 631                 ptcache_file_read(pf, &dx, 1, sizeof(float));
 632
 633                 if(pf->data_types & (1<<BPHYS_DATA_SMOKE_HIGH) && sds->wt) {
 634                         int res = sds->res[0]*sds->res[1]*sds->res[2];
 635                         int res_big, res_big_array[3];
 636                         float *dens, *densold, *tcu, *tcv, *tcw;
 637                         unsigned int out_len = sizeof(float)*(unsigned int)res;
 638                         unsigned int out_len_big;
 639
 640                         smoke_turbulence_get_res(sds->wt, res_big_array);
 641                         res_big = res_big_array[0]*res_big_array[1]*res_big_array[2];
 642                         out_len_big = sizeof(float) * (unsigned int)res_big;
 643
 644                         smoke_turbulence_export(sds->wt, &dens, &densold, &tcu, &tcv, &tcw);
 645
 646                         ptcache_file_compressed_read(pf, (unsigned char*)dens, out_len_big);
 647                         ptcache_file_compressed_read(pf, (unsigned char*)densold, out_len_big);
 648
 649                         ptcache_file_compressed_read(pf, (unsigned char*)tcu, out_len);
 650                         ptcache_file_compressed_read(pf, (unsigned char*)tcv, out_len);
 651                         ptcache_file_compressed_read(pf, (unsigned char*)tcw, out_len);
 652                 }
 653         }
 654 }
 655
 656 /* Creating ID's */
 657 void BKE_ptcache_id_from_softbody(PTCacheID *pid, Object *ob, SoftBody *sb)
 658 {
 659         memset(pid, 0, sizeof(PTCacheID));
 660
 661         pid->ob= ob;
 662         pid->calldata= sb;
 663         pid->type= PTCACHE_TYPE_SOFTBODY;
 664         pid->cache= sb->pointcache;
 665         pid->cache_ptr= &sb->pointcache;
 666         pid->ptcaches= &sb->ptcaches;
 667         pid->totpoint= pid->totwrite= ptcache_softbody_totpoint;
 668
 669         pid->write_point                        = ptcache_softbody_write;
 670         pid->read_point                         = ptcache_softbody_read;
 671         pid->interpolate_point          = ptcache_softbody_interpolate;
 672
 673         pid->write_stream                       = NULL;
 674         pid->read_stream                        = NULL;
 675
 676         pid->write_extra_data           = NULL;
 677         pid->read_extra_data            = NULL;
 678         pid->interpolate_extra_data     = NULL;
 679
 680         pid->write_header                       = ptcache_basic_header_write;
 681         pid->read_header                        = ptcache_basic_header_read;
 682
 683         pid->data_types= (1<<BPHYS_DATA_LOCATION) | (1<<BPHYS_DATA_VELOCITY);
 684         pid->info_types= 0;
 685
 686         pid->stack_index = pid->cache->index;
 687 }
 688 void BKE_ptcache_id_from_particles(PTCacheID *pid, Object *ob, ParticleSystem *psys)
 689 {
 690         memset(pid, 0, sizeof(PTCacheID));
 691
 692         pid->ob= ob;
 693         pid->calldata= psys;
 694         pid->type= PTCACHE_TYPE_PARTICLES;
 695         pid->stack_index= psys->pointcache->index;
 696         pid->cache= psys->pointcache;
 697         pid->cache_ptr= &psys->pointcache;
 698         pid->ptcaches= &psys->ptcaches;
 699
 700         if(psys->part->type != PART_HAIR)
 701                 pid->flag |= PTCACHE_VEL_PER_SEC;
 702
 703         pid->totpoint                           = ptcache_particle_totpoint;
 704         pid->totwrite                           = ptcache_particle_totwrite;
 705
 706         pid->write_point                                = ptcache_particle_write;
 707         pid->read_point                         = ptcache_particle_read;
 708         pid->interpolate_point          = ptcache_particle_interpolate;
 709
 710         pid->write_stream                       = NULL;
 711         pid->read_stream                        = NULL;
 712
 713         pid->write_extra_data           = NULL;
 714         pid->read_extra_data            = NULL;
 715         pid->interpolate_extra_data     = NULL;
 716
 717         pid->write_header                       = ptcache_basic_header_write;
 718         pid->read_header                        = ptcache_basic_header_read;
 719
 720         pid->data_types = (1<<BPHYS_DATA_LOCATION) | (1<<BPHYS_DATA_VELOCITY) | (1<<BPHYS_DATA_INDEX);
 721
 722         if(psys->part->phystype == PART_PHYS_BOIDS)
 723                 pid->data_types|= (1<<BPHYS_DATA_AVELOCITY) | (1<<BPHYS_DATA_ROTATION) | (1<<BPHYS_DATA_BOIDS);
 724         else if(psys->part->phystype == PART_PHYS_FLUID && psys->part->fluid && psys->part->fluid->flag & SPH_VISCOELASTIC_SPRINGS) {
 725                 pid->write_extra_data = ptcache_particle_extra_write;
 726                 pid->read_extra_data = ptcache_particle_extra_read;
 727         }
 728
 729         if(psys->part->rotmode!=PART_ROT_VEL
 730                 || psys->part->avemode!=PART_AVE_SPIN || psys->part->avefac!=0.0f)
 731                 pid->data_types|= (1<<BPHYS_DATA_AVELOCITY) | (1<<BPHYS_DATA_ROTATION);
 732
 733         if(psys->part->flag & PART_ROT_DYN)
 734                 pid->data_types|= (1<<BPHYS_DATA_ROTATION);
 735
 736         pid->info_types= (1<<BPHYS_DATA_TIMES);
 737 }
 738 void BKE_ptcache_id_from_cloth(PTCacheID *pid, Object *ob, ClothModifierData *clmd)
 739 {
 740         memset(pid, 0, sizeof(PTCacheID));
 741
 742         pid->ob= ob;
 743         pid->calldata= clmd;
 744         pid->type= PTCACHE_TYPE_CLOTH;
 745         pid->stack_index= clmd->point_cache->index;
 746         pid->cache= clmd->point_cache;
 747         pid->cache_ptr= &clmd->point_cache;
 748         pid->ptcaches= &clmd->ptcaches;
 749         pid->totpoint= pid->totwrite= ptcache_cloth_totpoint;
 750
 751         pid->write_point                        = ptcache_cloth_write;
 752         pid->read_point                         = ptcache_cloth_read;
 753         pid->interpolate_point          = ptcache_cloth_interpolate;
 754
 755         pid->write_stream                       = NULL;
 756         pid->read_stream                        = NULL;
 757
 758         pid->write_extra_data           = NULL;
 759         pid->read_extra_data            = NULL;
 760         pid->interpolate_extra_data     = NULL;
 761
 762         pid->write_header                       = ptcache_basic_header_write;
 763         pid->read_header                        = ptcache_basic_header_read;
 764
 765         pid->data_types= (1<<BPHYS_DATA_LOCATION) | (1<<BPHYS_DATA_VELOCITY) | (1<<BPHYS_DATA_XCONST);
 766         pid->info_types= 0;
 767 }
 768 void BKE_ptcache_id_from_smoke(PTCacheID *pid, struct Object *ob, struct SmokeModifierData *smd)
 769 {
 770         SmokeDomainSettings *sds = smd->domain;
 771
 772         memset(pid, 0, sizeof(PTCacheID));
 773
 774         pid->ob= ob;
 775         pid->calldata= smd;
 776
 777         pid->type= PTCACHE_TYPE_SMOKE_DOMAIN;
 778         pid->stack_index= sds->point_cache[0]->index;
 779
 780         pid->cache= sds->point_cache[0];
 781         pid->cache_ptr= &(sds->point_cache[0]);
 782         pid->ptcaches= &(sds->ptcaches[0]);
 783
 784         pid->totpoint= pid->totwrite= ptcache_smoke_totpoint;
 785
 786         pid->write_point                        = NULL;
 787         pid->read_point                         = NULL;
 788         pid->interpolate_point          = NULL;
 789
 790         pid->read_stream                        = ptcache_smoke_read;
 791         pid->write_stream                       = ptcache_smoke_write;
 792
 793         pid->write_extra_data           = NULL;
 794         pid->read_extra_data            = NULL;
 795         pid->interpolate_extra_data     = NULL;
 796
 797         pid->write_header                       = ptcache_basic_header_write;
 798         pid->read_header                        = ptcache_basic_header_read;
 799
 800         pid->data_types= 0;
 801         pid->info_types= 0;
 802
 803         if(sds->fluid)
 804                 pid->data_types |= (1<<BPHYS_DATA_SMOKE_LOW);
 805         if(sds->wt)
 806                 pid->data_types |= (1<<BPHYS_DATA_SMOKE_HIGH);
 807 }
 808 void BKE_ptcache_ids_from_object(ListBase *lb, Object *ob, Scene *scene, int duplis)
 809 {
 810         PTCacheID *pid;
 811         ParticleSystem *psys;
 812         ModifierData *md;
 813
 814         lb->first= lb->last= NULL;
 815
 816         if(ob->soft) {
 817                 pid= MEM_callocN(sizeof(PTCacheID), "PTCacheID");
 818                 BKE_ptcache_id_from_softbody(pid, ob, ob->soft);
 819                 BLI_addtail(lb, pid);
 820         }
 821
 822         for(psys=ob->particlesystem.first; psys; psys=psys->next) {
 823                 if(psys->part==NULL)
 824                         continue;
 825
 826                 /* check to make sure point cache is actually used by the particles */
 827                 if(ELEM(psys->part->phystype, PART_PHYS_NO, PART_PHYS_KEYED))
 828                         continue;
 829
 830                 /* hair needs to be included in id-list for cache edit mode to work */
 831                 /* if(psys->part->type == PART_HAIR && (psys->flag & PSYS_HAIR_DYNAMICS)==0) */
 832                 /*      continue; */
 833
 834                 if(psys->part->type == PART_FLUID)
 835                         continue;
 836
 837                 pid= MEM_callocN(sizeof(PTCacheID), "PTCacheID");
 838                 BKE_ptcache_id_from_particles(pid, ob, psys);
 839                 BLI_addtail(lb, pid);
 840         }
 841
 842         for(md=ob->modifiers.first; md; md=md->next) {
 843                 if(md->type == eModifierType_Cloth) {
 844                         pid= MEM_callocN(sizeof(PTCacheID), "PTCacheID");
 845                         BKE_ptcache_id_from_cloth(pid, ob, (ClothModifierData*)md);
 846                         BLI_addtail(lb, pid);
 847                 }
 848                 if(md->type == eModifierType_Smoke) {
 849                         SmokeModifierData *smd = (SmokeModifierData *)md;
 850                         if(smd->type & MOD_SMOKE_TYPE_DOMAIN)
 851                         {
 852                                 pid= MEM_callocN(sizeof(PTCacheID), "PTCacheID");
 853                                 BKE_ptcache_id_from_smoke(pid, ob, (SmokeModifierData*)md);
 854                                 BLI_addtail(lb, pid);
 855                         }
 856                 }
 857         }
 858
 859         if(scene && (duplis-- > 0) && (ob->transflag & OB_DUPLI)) {
 860                 ListBase *lb_dupli_ob;
 861
 862                 if((lb_dupli_ob=object_duplilist(scene, ob))) {
 863                         DupliObject *dob;
 864                         for(dob= lb_dupli_ob->first; dob; dob= dob->next) {
 865                                 if(dob->ob != ob) { /* avoids recursive loops with dupliframes: bug 22988 */
 866                                         ListBase lb_dupli_pid;
 867                                         BKE_ptcache_ids_from_object(&lb_dupli_pid, dob->ob, scene, duplis);
 868                                         BLI_movelisttolist(lb, &lb_dupli_pid);
 869                                         if(lb_dupli_pid.first)
 870                                                 printf("Adding Dupli\n");
 871                                 }
 872                         }
 873
 874                         free_object_duplilist(lb_dupli_ob);     /* does restore */
 875                 }
 876         }
 877 }
 878
 879 /* File handling */
 880
 881 /*      Takes an Object ID and returns a unique name
 882         - id: object id
 883         - cfra: frame for the cache, can be negative
 884         - stack_index: index in the modifier stack. we can have cache for more then one stack_index
 885 */
 886
 887 #define MAX_PTCACHE_PATH FILE_MAX
 888 #define MAX_PTCACHE_FILE ((FILE_MAXDIR+FILE_MAXFILE)*2)
 889
 890 static int ptcache_path(PTCacheID *pid, char *filename)
 891 {
 892         Library *lib= (pid->ob)? pid->ob->id.lib: NULL;
 893         const char *blendfilename= (lib && (pid->cache->flag & PTCACHE_IGNORE_LIBPATH)==0) ? lib->filepath: G.main->name;
 894         size_t i;
 895
 896         if(pid->cache->flag & PTCACHE_EXTERNAL) {
 897                 strcpy(filename, pid->cache->path);
 898
 899                 if(strncmp(filename, "//", 2)==0)
 900                         BLI_path_abs(filename, blendfilename);
 901
 902                 return BLI_add_slash(filename); /* new strlen() */
 903         }
 904         else if (G.relbase_valid || lib) {
 905                 char file[MAX_PTCACHE_PATH]; /* we dont want the dir, only the file */
 906
 907                 BLI_split_dirfile(blendfilename, NULL, file);
 908                 i = strlen(file);
 909
 910                 /* remove .blend */
 911                 if (i > 6)
 912                         file[i-6] = '\0';
 913
 914                 snprintf(filename, MAX_PTCACHE_PATH, "//"PTCACHE_PATH"%s", file); /* add blend file name to pointcache dir */
 915                 BLI_path_abs(filename, blendfilename);
 916                 return BLI_add_slash(filename); /* new strlen() */
 917         }
 918
 919         /* use the temp path. this is weak but better then not using point cache at all */
 920         /* btempdir is assumed to exist and ALWAYS has a trailing slash */
 921         snprintf(filename, MAX_PTCACHE_PATH, "%s"PTCACHE_PATH"%d", btempdir, abs(getpid()));
 922
 923         return BLI_add_slash(filename); /* new strlen() */
 924 }
 925
 926 static int ptcache_filename(PTCacheID *pid, char *filename, int cfra, short do_path, short do_ext)
 927 {
 928         int len=0;
 929         char *idname;
 930         char *newname;
 931         filename[0] = '\0';
 932         newname = filename;
 933
 934         if (!G.relbase_valid && (pid->cache->flag & PTCACHE_EXTERNAL)==0) return 0; /* save blend file before using disk pointcache */
 935
 936         /* start with temp dir */
 937         if (do_path) {
 938                 len = ptcache_path(pid, filename);
 939                 newname += len;
 940         }
 941         if(pid->cache->name[0] == '\0' && (pid->cache->flag & PTCACHE_EXTERNAL)==0) {
 942                 idname = (pid->ob->id.name+2);
 943                 /* convert chars to hex so they are always a valid filename */
 944                 while('\0' != *idname) {
 945                         snprintf(newname, MAX_PTCACHE_FILE, "%02X", (char)(*idname++));
 946                         newname+=2;
 947                         len += 2;
 948                 }
 949         }
 950         else {
 951                 int temp = (int)strlen(pid->cache->name);
 952                 strcpy(newname, pid->cache->name);
 953                 newname+=temp;
 954                 len += temp;
 955         }
 956
 957         if (do_ext) {
 958
 959                 if(pid->cache->index < 0)
 960                         pid->cache->index =  pid->stack_index = object_insert_ptcache(pid->ob);
 961
 962                 if(pid->cache->flag & PTCACHE_EXTERNAL) {
 963                         if(pid->cache->index >= 0)
 964                                 snprintf(newname, MAX_PTCACHE_FILE, "_%06d_%02d"PTCACHE_EXT, cfra, pid->stack_index); /* always 6 chars */
 965                         else
 966                                 snprintf(newname, MAX_PTCACHE_FILE, "_%06d"PTCACHE_EXT, cfra); /* always 6 chars */
 967                 }
 968                 else {
 969                         snprintf(newname, MAX_PTCACHE_FILE, "_%06d_%02d"PTCACHE_EXT, cfra, pid->stack_index); /* always 6 chars */
 970                 }
 971                 len += 16;
 972         }
 973
 974         return len; /* make sure the above string is always 16 chars */
 975 }
 976
 977 /* youll need to close yourself after! */
 978 static PTCacheFile *ptcache_file_open(PTCacheID *pid, int mode, int cfra)
 979 {
 980         PTCacheFile *pf;
 981         FILE *fp = NULL;
 982         char filename[(FILE_MAXDIR+FILE_MAXFILE)*2];
 983
 984 #ifndef DURIAN_POINTCACHE_LIB_OK
 985         /* don't allow writing for linked objects */
 986         if(pid->ob->id.lib && mode == PTCACHE_FILE_WRITE)
 987                 return NULL;
 988 #endif
 989         if (!G.relbase_valid && (pid->cache->flag & PTCACHE_EXTERNAL)==0) return NULL; /* save blend file before using disk pointcache */
 990
 991         ptcache_filename(pid, filename, cfra, 1, 1);
 992
 993         if (mode==PTCACHE_FILE_READ) {
 994                 if (!BLI_exists(filename)) {
 995                         return NULL;
 996                 }
 997                  fp = fopen(filename, "rb");
 998         } else if (mode==PTCACHE_FILE_WRITE) {
 999                 BLI_make_existing_file(filename); /* will create the dir if needs be, same as //textures is created */
1000                 fp = fopen(filename, "wb");
1001         } else if (mode==PTCACHE_FILE_UPDATE) {
1002                 BLI_make_existing_file(filename);
1003                 fp = fopen(filename, "rb+");
1004         }
1005
1006         if (!fp)
1007                 return NULL;
1008
1009         pf= MEM_mallocN(sizeof(PTCacheFile), "PTCacheFile");
1010         pf->fp= fp;
1011         pf->old_format = 0;
1012         pf->frame = cfra;
1013
1014         return pf;
1015 }
1016 static void ptcache_file_close(PTCacheFile *pf)
1017 {
1018         if(pf) {
1019                 fclose(pf->fp);
1020                 MEM_freeN(pf);
1021         }
1022 }
1023
1024 static int ptcache_file_compressed_read(PTCacheFile *pf, unsigned char *result, unsigned int len)
1025 {
1026         int r = 0;
1027         unsigned char compressed = 0;
1028         size_t in_len;
1029 #ifdef WITH_LZO
1030         size_t out_len = len;
1031 #endif
1032         unsigned char *in;
1033         unsigned char *props = MEM_callocN(16*sizeof(char), "tmp");
1034
1035         ptcache_file_read(pf, &compressed, 1, sizeof(unsigned char));
1036         if(compressed) {
1037                 unsigned int size;
1038                 ptcache_file_read(pf, &size, 1, sizeof(unsigned int));
1039                 in_len = (size_t)size;
1040                 if(in_len==0) {
1041                         /* do nothing */
1042                 }
1043                 else {
1044                         in = (unsigned char *)MEM_callocN(sizeof(unsigned char)*in_len, "pointcache_compressed_buffer");
1045                         ptcache_file_read(pf, in, in_len, sizeof(unsigned char));
1046 #ifdef WITH_LZO
1047                         if(compressed == 1)
1048                                 r = lzo1x_decompress_safe(in, (lzo_uint)in_len, result, (lzo_uint *)&out_len, NULL);
1049 #endif
1050 #ifdef WITH_LZMA
1051                         if(compressed == 2)
1052                         {
1053                                 size_t sizeOfIt;
1054                                 size_t leni = in_len, leno = out_len;
1055                                 ptcache_file_read(pf, &size, 1, sizeof(unsigned int));
1056                                 sizeOfIt = (size_t)size;
1057                                 ptcache_file_read(pf, props, sizeOfIt, sizeof(unsigned char));
1058                                 r = LzmaUncompress(result, &leno, in, &leni, props, sizeOfIt);
1059                         }
1060 #endif
1061                         MEM_freeN(in);
1062                 }
1063         }
1064         else {
1065                 ptcache_file_read(pf, result, len, sizeof(unsigned char));
1066         }
1067
1068         MEM_freeN(props);
1069
1070         return r;
1071 }
1072 static int ptcache_file_compressed_write(PTCacheFile *pf, unsigned char *in, unsigned int in_len, unsigned char *out, int mode)
1073 {
1074         int r = 0;
1075         unsigned char compressed = 0;
1076         size_t out_len= 0;
1077         unsigned char *props = MEM_callocN(16*sizeof(char), "tmp");
1078         size_t sizeOfIt = 5;
1079
1080         (void)mode; /* unused when building w/o compression */
1081
1082 #ifdef WITH_LZO
1083         out_len= LZO_OUT_LEN(in_len);
1084         if(mode == 1) {
1085                 LZO_HEAP_ALLOC(wrkmem, LZO1X_MEM_COMPRESS);
1086
1087                 r = lzo1x_1_compress(in, (lzo_uint)in_len, out, (lzo_uint *)&out_len, wrkmem);
1088                 if (!(r == LZO_E_OK) || (out_len >= in_len))
1089                         compressed = 0;
1090                 else
1091                         compressed = 1;
1092         }
1093 #endif
1094 #ifdef WITH_LZMA
1095         if(mode == 2) {
1096
1097                 r = LzmaCompress(out, &out_len, in, in_len,//assume sizeof(char)==1....
1098                                                 props, &sizeOfIt, 5, 1 << 24, 3, 0, 2, 32, 2);
1099
1100                 if(!(r == SZ_OK) || (out_len >= in_len))
1101                         compressed = 0;
1102                 else
1103                         compressed = 2;
1104         }
1105 #endif
1106
1107         ptcache_file_write(pf, &compressed, 1, sizeof(unsigned char));
1108         if(compressed) {
1109                 unsigned int size = out_len;
1110                 ptcache_file_write(pf, &size, 1, sizeof(unsigned int));
1111                 ptcache_file_write(pf, out, out_len, sizeof(unsigned char));
1112         }
1113         else
1114                 ptcache_file_write(pf, in, in_len, sizeof(unsigned char));
1115
1116         if(compressed == 2)
1117         {
1118                 unsigned int size = sizeOfIt;
1119                 ptcache_file_write(pf, &sizeOfIt, 1, sizeof(unsigned int));
1120                 ptcache_file_write(pf, props, size, sizeof(unsigned char));
1121         }
1122
1123         MEM_freeN(props);
1124
1125         return r;
1126 }
1127 static int ptcache_file_read(PTCacheFile *pf, void *f, unsigned int tot, unsigned int size)
1128 {
1129         return (fread(f, size, tot, pf->fp) == tot);
1130 }
1131 static int ptcache_file_write(PTCacheFile *pf, void *f, unsigned int tot, unsigned int size)
1132 {
1133         return (fwrite(f, size, tot, pf->fp) == tot);
1134 }
1135 static int ptcache_file_data_read(PTCacheFile *pf)
1136 {
1137         int i;
1138
1139         for(i=0; i<BPHYS_TOT_DATA; i++) {
1140                 if((pf->data_types & (1<<i)) && !ptcache_file_read(pf, pf->cur[i], 1, ptcache_data_size[i]))
1141                         return 0;
1142         }
1143
1144         return 1;
1145 }
1146 static int ptcache_file_data_write(PTCacheFile *pf)
1147 {
1148         int i;
1149
1150         for(i=0; i<BPHYS_TOT_DATA; i++) {
1151                 if((pf->data_types & (1<<i)) && !ptcache_file_write(pf, pf->cur[i], 1, ptcache_data_size[i]))
1152                         return 0;
1153         }
1154
1155         return 1;
1156 }
1157 static int ptcache_file_header_begin_read(PTCacheFile *pf)
1158 {
1159         unsigned int typeflag=0;
1160         int error=0;
1161         char bphysics[8];
1162
1163         pf->data_types = 0;
1164
1165         if(fread(bphysics, sizeof(char), 8, pf->fp) != 8)
1166                 error = 1;
1167
1168         if(!error && strncmp(bphysics, "BPHYSICS", 8))
1169                 error = 1;
1170
1171         if(!error && !fread(&typeflag, sizeof(unsigned int), 1, pf->fp))
1172                 error = 1;
1173
1174         pf->type = (typeflag & PTCACHE_TYPEFLAG_TYPEMASK);
1175         pf->flag = (typeflag & PTCACHE_TYPEFLAG_FLAGMASK);
1176
1177         /* if there was an error set file as it was */
1178         if(error)
1179                 fseek(pf->fp, 0, SEEK_SET);
1180
1181         return !error;
1182 }
1183 static int ptcache_file_header_begin_write(PTCacheFile *pf)
1184 {
1185         const char *bphysics = "BPHYSICS";
1186         unsigned int typeflag = pf->type + pf->flag;
1187
1188         if(fwrite(bphysics, sizeof(char), 8, pf->fp) != 8)
1189                 return 0;
1190
1191         if(!fwrite(&typeflag, sizeof(unsigned int), 1, pf->fp))
1192                 return 0;
1193
1194         return 1;
1195 }
1196
1197 /* Data pointer handling */
1198 int BKE_ptcache_data_size(int data_type)
1199 {
1200         return ptcache_data_size[data_type];
1201 }
1202
1203 static void ptcache_file_pointers_init(PTCacheFile *pf)
1204 {
1205         int data_types = pf->data_types;
1206
1207         pf->cur[BPHYS_DATA_INDEX] =             (data_types & (1<<BPHYS_DATA_INDEX))    ?               &pf->data.index : NULL;
1208         pf->cur[BPHYS_DATA_LOCATION] =  (data_types & (1<<BPHYS_DATA_LOCATION)) ?               &pf->data.loc   : NULL;
1209         pf->cur[BPHYS_DATA_VELOCITY] =  (data_types & (1<<BPHYS_DATA_VELOCITY)) ?               &pf->data.vel   : NULL;
1210         pf->cur[BPHYS_DATA_ROTATION] =  (data_types & (1<<BPHYS_DATA_ROTATION)) ?               &pf->data.rot   : NULL;
1211         pf->cur[BPHYS_DATA_AVELOCITY] = (data_types & (1<<BPHYS_DATA_AVELOCITY))?               &pf->data.ave   : NULL;
1212         pf->cur[BPHYS_DATA_SIZE] =              (data_types & (1<<BPHYS_DATA_SIZE))             ?               &pf->data.size  : NULL;
1213         pf->cur[BPHYS_DATA_TIMES] =             (data_types & (1<<BPHYS_DATA_TIMES))    ?               &pf->data.times : NULL;
1214         pf->cur[BPHYS_DATA_BOIDS] =             (data_types & (1<<BPHYS_DATA_BOIDS))    ?               &pf->data.boids : NULL;
1215 }
1216
1217 /* Check to see if point number "index" is in pm, uses binary search for index data. */
1218 int BKE_ptcache_mem_index_find(PTCacheMem *pm, unsigned int index)
1219 {
1220         if(pm->data[BPHYS_DATA_INDEX]) {
1221                 unsigned int *data = pm->data[BPHYS_DATA_INDEX];
1222                 unsigned int mid, low = 0, high = pm->totpoint - 1;
1223
1224                 if(index < *data || index > *(data+high))
1225                         return -1;
1226
1227                 /* check simple case for continuous indexes first */
1228                 if(index-*data < high && data[index-*data] == index)
1229                         return index-*data;
1230
1231                 while(low <= high) {
1232                         mid= (low + high)/2;
1233
1234                         if(data[mid] > index)
1235                                 high = mid - 1;
1236                         else if(data[mid] < index)
1237                                 low = mid + 1;
1238                         else
1239                                 return mid;
1240                 }
1241
1242                 return -1;
1243         }
1244         else {
1245                 return (index < pm->totpoint ? index : -1);
1246         }
1247 }
1248
1249 void BKE_ptcache_mem_pointers_init(PTCacheMem *pm)
1250 {
1251         int data_types = pm->data_types;
1252         int i;
1253
1254         for(i=0; i<BPHYS_TOT_DATA; i++)
1255                 pm->cur[i] = ((data_types & (1<<i)) ? pm->data[i] : NULL);
1256 }
1257
1258 void BKE_ptcache_mem_pointers_incr(PTCacheMem *pm)
1259 {
1260         int i;
1261
1262         for(i=0; i<BPHYS_TOT_DATA; i++) {
1263                 if(pm->cur[i])
1264                         pm->cur[i] = (char*)pm->cur[i] + ptcache_data_size[i];
1265         }
1266 }
1267 int  BKE_ptcache_mem_pointers_seek(int point_index, PTCacheMem *pm)
1268 {
1269         int data_types = pm->data_types;
1270         int i, index = BKE_ptcache_mem_index_find(pm, point_index);
1271
1272         if(index < 0) {
1273                 /* Can't give proper location without reallocation, so don't give any location.
1274                  * Some points will be cached improperly, but this only happens with simulation
1275                  * steps bigger than cache->step, so the cache has to be recalculated anyways
1276                  * at some point.
1277                  */
1278                 return 0;
1279         }
1280
1281         for(i=0; i<BPHYS_TOT_DATA; i++)
1282                 pm->cur[i] = data_types & (1<<i) ? (char*)pm->data[i] + index * ptcache_data_size[i] : NULL;
1283
1284         return 1;
1285 }
1286 static void ptcache_data_alloc(PTCacheMem *pm)
1287 {
1288         int data_types = pm->data_types;
1289         int totpoint = pm->totpoint;
1290         int i;
1291
1292         for(i=0; i<BPHYS_TOT_DATA; i++) {
1293                 if(data_types & (1<<i))
1294                         pm->data[i] = MEM_callocN(totpoint * ptcache_data_size[i], "PTCache Data");
1295         }
1296 }
1297 static void ptcache_data_free(PTCacheMem *pm)
1298 {
1299         void **data = pm->data;
1300         int i;
1301
1302         for(i=0; i<BPHYS_TOT_DATA; i++) {
1303                 if(data[i])
1304                         MEM_freeN(data[i]);
1305         }
1306 }
1307 static void ptcache_data_copy(void *from[], void *to[])
1308 {
1309         int i;
1310         for(i=0; i<BPHYS_TOT_DATA; i++) {
1311         /* note, durian file 03.4b_comp crashes if to[i] is not tested
1312          * its NULL, not sure if this should be fixed elsewhere but for now its needed */
1313                 if(from[i] && to[i])
1314                         memcpy(to[i], from[i], ptcache_data_size[i]);
1315         }
1316 }
1317
1318 static void ptcache_extra_free(PTCacheMem *pm)
1319 {
1320         PTCacheExtra *extra = pm->extradata.first;
1321
1322         if(extra) {
1323                 for(; extra; extra=extra->next) {
1324                         if(extra->data)
1325                                 MEM_freeN(extra->data);
1326                 }
1327
1328                 BLI_freelistN(&pm->extradata);
1329         }
1330 }
1331 static int ptcache_old_elemsize(PTCacheID *pid)
1332 {
1333         if(pid->type==PTCACHE_TYPE_SOFTBODY)
1334                 return 6 * sizeof(float);
1335         else if(pid->type==PTCACHE_TYPE_PARTICLES)
1336                 return sizeof(ParticleKey);
1337         else if(pid->type==PTCACHE_TYPE_CLOTH)
1338                 return 9 * sizeof(float);
1339
1340         return 0;
1341 }
1342
1343 static void ptcache_find_frames_around(PTCacheID *pid, unsigned int frame, int *fra1, int *fra2)
1344 {
1345         if(pid->cache->flag & PTCACHE_DISK_CACHE) {
1346                 int cfra1=frame-1, cfra2=frame+1;
1347
1348                 while(cfra1 >= pid->cache->startframe && !BKE_ptcache_id_exist(pid, cfra1))
1349                         cfra1--;
1350
1351                 if(cfra1 < pid->cache->startframe)
1352                         cfra1 = 0;
1353
1354                 while(cfra2 <= pid->cache->endframe && !BKE_ptcache_id_exist(pid, cfra2))
1355                         cfra2++;
1356
1357                 if(cfra2 > pid->cache->endframe)
1358                         cfra2 = 0;
1359
1360                 if(cfra1 && !cfra2) {
1361                         *fra1 = 0;
1362                         *fra2 = cfra1;
1363                 }
1364                 else {
1365                         *fra1 = cfra1;
1366                         *fra2 = cfra2;
1367                 }
1368         }
1369         else if(pid->cache->mem_cache.first) {
1370                 PTCacheMem *pm = pid->cache->mem_cache.first;
1371                 PTCacheMem *pm2 = pid->cache->mem_cache.last;
1372
1373                 while(pm->next && pm->next->frame < frame)
1374                         pm= pm->next;
1375
1376                 if(pm2->frame < frame) {
1377                         pm2 = NULL;
1378                 }
1379                 else {
1380                         while(pm2->prev && pm2->prev->frame > frame) {
1381                                 pm2= pm2->prev;
1382                         }
1383                 }
1384
1385                 if(!pm2) {
1386                         *fra1 = 0;
1387                         *fra2 = pm->frame;
1388                 }
1389                 else {
1390                         *fra1 = pm->frame;
1391                         *fra2 = pm2->frame;
1392                 }
1393         }
1394 }
1395
1396 static PTCacheMem *ptcache_disk_frame_to_mem(PTCacheID *pid, int cfra)
1397 {
1398         PTCacheFile *pf = ptcache_file_open(pid, PTCACHE_FILE_READ, cfra);
1399         PTCacheMem *pm = NULL;
1400         unsigned int i, error = 0;
1401
1402         if(pf == NULL)
1403                 return NULL;
1404
1405         if(!ptcache_file_header_begin_read(pf))
1406                 error = 1;
1407
1408         if(!error && (pf->type != pid->type || !pid->read_header(pf)))
1409                 error = 1;
1410
1411         if(!error) {
1412                 pm = MEM_callocN(sizeof(PTCacheMem), "Pointcache mem");
1413
1414                 pm->totpoint = pf->totpoint;
1415                 pm->data_types = pf->data_types;
1416                 pm->frame = pf->frame;
1417
1418                 ptcache_data_alloc(pm);
1419
1420                 if(pf->flag & PTCACHE_TYPEFLAG_COMPRESS) {
1421                         for(i=0; i<BPHYS_TOT_DATA; i++) {
1422                                 unsigned int out_len = pm->totpoint*ptcache_data_size[i];
1423                                 if(pf->data_types & (1<<i))
1424                                         ptcache_file_compressed_read(pf, (unsigned char*)(pm->data[i]), out_len);
1425                         }
1426                 }
1427                 else {
1428                         BKE_ptcache_mem_pointers_init(pm);
1429                         ptcache_file_pointers_init(pf);
1430
1431                         for(i=0; i<pm->totpoint; i++) {
1432                                 if(!ptcache_file_data_read(pf)) {
1433                                         error = 1;
1434                                         break;
1435                                 }
1436                                 ptcache_data_copy(pf->cur, pm->cur);
1437                                 BKE_ptcache_mem_pointers_incr(pm);
1438                         }
1439                 }
1440         }
1441
1442         if(!error && pf->flag & PTCACHE_TYPEFLAG_EXTRADATA) {
1443                 unsigned int extratype = 0;
1444
1445                 while(ptcache_file_read(pf, &extratype, 1, sizeof(unsigned int))) {
1446                         PTCacheExtra *extra = MEM_callocN(sizeof(PTCacheExtra), "Pointcache extradata");
1447
1448                         extra->type = extratype;
1449
1450                         ptcache_file_read(pf, &extra->totdata, 1, sizeof(unsigned int));
1451
1452                         extra->data = MEM_callocN(extra->totdata * ptcache_extra_datasize[extra->type], "Pointcache extradata->data");
1453
1454                         if(pf->flag & PTCACHE_TYPEFLAG_COMPRESS)
1455                                 ptcache_file_compressed_read(pf, (unsigned char*)(extra->data), extra->totdata*ptcache_extra_datasize[extra->type]);
1456                         else
1457                                 ptcache_file_read(pf, extra->data, extra->totdata, ptcache_extra_datasize[extra->type]);
1458
1459                         BLI_addtail(&pm->extradata, extra);
1460                 }
1461         }
1462
1463         if(error && pm) {
1464                 ptcache_data_free(pm);
1465                 ptcache_extra_free(pm);
1466                 MEM_freeN(pm);
1467                 pm = NULL;
1468         }
1469
1470         ptcache_file_close(pf);
1471
1472         if (error && G.f & G_DEBUG)
1473                 printf("Error reading from disk cache\n");
1474
1475         return pm;
1476 }
1477 static int ptcache_mem_frame_to_disk(PTCacheID *pid, PTCacheMem *pm)
1478 {
1479         PTCacheFile *pf = NULL;
1480         unsigned int i, error = 0;
1481
1482         BKE_ptcache_id_clear(pid, PTCACHE_CLEAR_FRAME, pm->frame);
1483
1484         pf = ptcache_file_open(pid, PTCACHE_FILE_WRITE, pm->frame);
1485
1486         if(pf==NULL) {
1487                 if (G.f & G_DEBUG)
1488                         printf("Error opening disk cache file for writing\n");
1489                 return 0;
1490         }
1491
1492         pf->data_types = pm->data_types;
1493         pf->totpoint = pm->totpoint;
1494         pf->type = pid->type;
1495         pf->flag = 0;
1496
1497         if(pm->extradata.first)
1498                 pf->flag |= PTCACHE_TYPEFLAG_EXTRADATA;
1499
1500         if(pid->cache->compression)
1501                 pf->flag |= PTCACHE_TYPEFLAG_COMPRESS;
1502
1503         if(!ptcache_file_header_begin_write(pf) || !pid->write_header(pf))
1504                 error = 1;
1505
1506         if(!error) {
1507                 if(pid->cache->compression) {
1508                         for(i=0; i<BPHYS_TOT_DATA; i++) {
1509                                 if(pm->data[i]) {
1510                                         unsigned int in_len = pm->totpoint*ptcache_data_size[i];
1511                                         unsigned char *out = (unsigned char *)MEM_callocN(LZO_OUT_LEN(in_len)*4, "pointcache_lzo_buffer");
1512                                         ptcache_file_compressed_write(pf, (unsigned char*)(pm->data[i]), in_len, out, pid->cache->compression);
1513                                         MEM_freeN(out);
1514                                 }
1515                         }
1516                 }
1517                 else {
1518                         BKE_ptcache_mem_pointers_init(pm);
1519                         ptcache_file_pointers_init(pf);
1520
1521                         for(i=0; i<pm->totpoint; i++) {
1522                                 ptcache_data_copy(pm->cur, pf->cur);
1523                                 if(!ptcache_file_data_write(pf)) {
1524                                         error = 1;
1525                                         break;
1526                                 }
1527                                 BKE_ptcache_mem_pointers_incr(pm);
1528                         }
1529                 }
1530         }
1531
1532         if(!error && pm->extradata.first) {
1533                 PTCacheExtra *extra = pm->extradata.first;
1534
1535                 for(; extra; extra=extra->next) {
1536                         if(extra->data == NULL || extra->totdata == 0)
1537                                 continue;
1538
1539                         ptcache_file_write(pf, &extra->type, 1, sizeof(unsigned int));
1540                         ptcache_file_write(pf, &extra->totdata, 1, sizeof(unsigned int));
1541
1542                         if(pid->cache->compression) {
1543                                 unsigned int in_len = extra->totdata * ptcache_extra_datasize[extra->type];
1544                                 unsigned char *out = (unsigned char *)MEM_callocN(LZO_OUT_LEN(in_len)*4, "pointcache_lzo_buffer");
1545                                 ptcache_file_compressed_write(pf, (unsigned char*)(extra->data), in_len, out, pid->cache->compression);
1546                                 MEM_freeN(out);
1547                         }
1548                         else {
1549                                 ptcache_file_write(pf, extra->data, extra->totdata, ptcache_extra_datasize[extra->type]);
1550                         }
1551                 }
1552         }
1553
1554         ptcache_file_close(pf);
1555
1556         if (error && G.f & G_DEBUG)
1557                 printf("Error writing to disk cache\n");
1558
1559         return error==0;
1560 }
1561
1562 static int ptcache_read_stream(PTCacheID *pid, int cfra)
1563 {
1564         PTCacheFile *pf = ptcache_file_open(pid, PTCACHE_FILE_READ, cfra);
1565         int error = 0;
1566
1567         if(pid->read_stream == NULL)
1568                 return 0;
1569
1570         if(pf == NULL) {
1571                 if (G.f & G_DEBUG)
1572                         printf("Error opening disk cache file for reading\n");
1573                 return 0;
1574         }
1575
1576         if(!ptcache_file_header_begin_read(pf))
1577                 error = 1;
1578
1579         if(!error && (pf->type != pid->type || !pid->read_header(pf)))
1580                 error = 1;
1581
1582         if(!error && pf->totpoint != pid->totpoint(pid->calldata, cfra))
1583                 error = 1;
1584
1585         if(!error) {
1586                 ptcache_file_pointers_init(pf);
1587
1588                 // we have stream reading here
1589                 pid->read_stream(pf, pid->calldata);
1590         }
1591
1592         ptcache_file_close(pf);
1593
1594         return error == 0;
1595 }
1596 static int ptcache_read(PTCacheID *pid, int cfra)
1597 {
1598         PTCacheMem *pm = NULL;
1599         int i;
1600         int *index = &i;
1601
1602         /* get a memory cache to read from */
1603         if(pid->cache->flag & PTCACHE_DISK_CACHE) {
1604                 pm = ptcache_disk_frame_to_mem(pid, cfra);
1605         }
1606         else {
1607                 pm = pid->cache->mem_cache.first;
1608
1609                 while(pm && pm->frame != cfra)
1610                         pm = pm->next;
1611         }
1612
1613         /* read the cache */
1614         if(pm) {
1615                 int totpoint = pm->totpoint;
1616
1617                 if((pid->data_types & (1<<BPHYS_DATA_INDEX)) == 0)
1618                         totpoint = MIN2(totpoint, pid->totpoint(pid->calldata, cfra));
1619
1620                 BKE_ptcache_mem_pointers_init(pm);
1621
1622                 for(i=0; i<totpoint; i++) {
1623                         if(pm->data_types & (1<<BPHYS_DATA_INDEX))
1624                                 index = pm->cur[BPHYS_DATA_INDEX];
1625
1626                         pid->read_point(*index, pid->calldata, pm->cur, (float)pm->frame, NULL);
1627
1628                         BKE_ptcache_mem_pointers_incr(pm);
1629                 }
1630
1631                 if(pid->read_extra_data && pm->extradata.first)
1632                         pid->read_extra_data(pid->calldata, pm, (float)pm->frame);
1633
1634                 /* clean up temporary memory cache */
1635                 if(pid->cache->flag & PTCACHE_DISK_CACHE) {
1636                         ptcache_data_free(pm);
1637                         ptcache_extra_free(pm);
1638                         MEM_freeN(pm);
1639                 }
1640         }
1641
1642         return 1;
1643 }
1644 static int ptcache_interpolate(PTCacheID *pid, float cfra, int cfra1, int cfra2)
1645 {
1646         PTCacheMem *pm = NULL;
1647         int i;
1648         int *index = &i;
1649
1650         /* get a memory cache to read from */
1651         if(pid->cache->flag & PTCACHE_DISK_CACHE) {
1652                 pm = ptcache_disk_frame_to_mem(pid, cfra2);
1653         }
1654         else {
1655                 pm = pid->cache->mem_cache.first;
1656
1657                 while(pm && pm->frame != cfra2)
1658                         pm = pm->next;
1659         }
1660
1661         /* read the cache */
1662         if(pm) {
1663                 int totpoint = pm->totpoint;
1664
1665                 if((pid->data_types & (1<<BPHYS_DATA_INDEX)) == 0)
1666                         totpoint = MIN2(totpoint, pid->totpoint(pid->calldata, (int)cfra));
1667
1668                 BKE_ptcache_mem_pointers_init(pm);
1669
1670                 for(i=0; i<totpoint; i++) {
1671                         if(pm->data_types & (1<<BPHYS_DATA_INDEX))
1672                                 index = pm->cur[BPHYS_DATA_INDEX];
1673
1674                         pid->interpolate_point(*index, pid->calldata, pm->cur, cfra, (float)cfra1, (float)cfra2, NULL);
1675                         BKE_ptcache_mem_pointers_incr(pm);
1676                 }
1677
1678                 if(pid->interpolate_extra_data && pm->extradata.first)
1679                         pid->interpolate_extra_data(pid->calldata, pm, cfra, (float)cfra1, (float)cfra2);
1680
1681                 /* clean up temporary memory cache */
1682                 if(pid->cache->flag & PTCACHE_DISK_CACHE) {
1683                         ptcache_data_free(pm);
1684                         ptcache_extra_free(pm);
1685                         MEM_freeN(pm);
1686                 }
1687         }
1688
1689         return 1;
1690 }
1691 /* reads cache from disk or memory */
1692 /* possible to get old or interpolated result */
1693 int BKE_ptcache_read(PTCacheID *pid, float cfra)
1694 {
1695         int cfrai = (int)cfra, cfra1=0, cfra2=0;
1696         int ret = 0;
1697
1698         /* nothing to read to */
1699         if(pid->totpoint(pid->calldata, cfrai) == 0)
1700                 return 0;
1701
1702         if(pid->cache->flag & PTCACHE_READ_INFO) {
1703                 pid->cache->flag &= ~PTCACHE_READ_INFO;
1704                 ptcache_read(pid, 0);
1705         }
1706
1707         /* first check if we have the actual frame cached */
1708         if(cfra == (float)cfrai && BKE_ptcache_id_exist(pid, cfrai))
1709                 cfra1 = cfrai;
1710
1711         /* no exact cache frame found so try to find cached frames around cfra */
1712         if(cfra1 == 0)
1713                 ptcache_find_frames_around(pid, cfrai, &cfra1, &cfra2);
1714
1715         if(cfra1 == 0 && cfra2 == 0)
1716                 return 0;
1717
1718         /* don't read old cache if already simulated past cached frame */
1719         if(cfra1 == 0 && cfra2 && cfra2 <= pid->cache->simframe)
1720                 return 0;
1721         if(cfra1 && cfra1 == cfra2)
1722                 return 0;
1723
1724         if(cfra1) {
1725                 if(pid->read_stream)
1726                         ptcache_read_stream(pid, cfra1);
1727                 else if(pid->read_point)
1728                         ptcache_read(pid, cfra1);
1729         }
1730
1731         if(cfra2) {
1732                 if(pid->read_stream)
1733                         ptcache_read_stream(pid, cfra2);
1734                 else if(pid->read_point) {
1735                         if(cfra1 && cfra2 && pid->interpolate_point)
1736                                 ptcache_interpolate(pid, cfra, cfra1, cfra2);
1737                         else
1738                                 ptcache_read(pid, cfra2);
1739                 }
1740         }
1741
1742         if(cfra1)
1743                 ret = (cfra2 ? PTCACHE_READ_INTERPOLATED : PTCACHE_READ_EXACT);
1744         else if(cfra2) {
1745                 ret = PTCACHE_READ_OLD;
1746                 pid->cache->simframe = cfra2;
1747         }
1748
1749         if((pid->cache->flag & PTCACHE_QUICK_CACHE)==0) {
1750                 cfrai = (int)cfra;
1751                 /* clear invalid cache frames so that better stuff can be simulated */
1752                 if(pid->cache->flag & PTCACHE_OUTDATED) {
1753                         BKE_ptcache_id_clear(pid, PTCACHE_CLEAR_AFTER, cfrai);
1754                 }
1755                 else if(pid->cache->flag & PTCACHE_FRAMES_SKIPPED) {
1756                         if(cfra <= pid->cache->last_exact)
1757                                 pid->cache->flag &= ~PTCACHE_FRAMES_SKIPPED;
1758
1759                         BKE_ptcache_id_clear(pid, PTCACHE_CLEAR_AFTER, MAX2(cfrai, pid->cache->last_exact));
1760                 }
1761         }
1762
1763         return ret;
1764 }
1765 static int ptcache_write_stream(PTCacheID *pid, int cfra, int totpoint)
1766 {
1767         PTCacheFile *pf = NULL;
1768         int error = 0;
1769
1770         BKE_ptcache_id_clear(pid, PTCACHE_CLEAR_FRAME, cfra);
1771
1772         pf = ptcache_file_open(pid, PTCACHE_FILE_WRITE, cfra);
1773
1774         if(pf==NULL) {
1775                 if (G.f & G_DEBUG)
1776                         printf("Error opening disk cache file for writing\n");
1777                 return 0;
1778         }
1779
1780         pf->data_types = pid->data_types;
1781         pf->totpoint = totpoint;
1782         pf->type = pid->type;
1783         pf->flag = 0;
1784
1785         if(!error && (!ptcache_file_header_begin_write(pf) || !pid->write_header(pf)))
1786                 error = 1;
1787
1788         if(!error && pid->write_stream)
1789                 pid->write_stream(pf, pid->calldata);
1790
1791         ptcache_file_close(pf);
1792
1793         if (error && G.f & G_DEBUG)
1794                 printf("Error writing to disk cache\n");
1795
1796         return error == 0;
1797 }
1798 static int ptcache_write(PTCacheID *pid, int cfra, int overwrite)
1799 {
1800         PointCache *cache = pid->cache;
1801         PTCacheMem *pm=NULL, *pm2=NULL;
1802         int totpoint = pid->totpoint(pid->calldata, cfra);
1803         int i, error = 0;
1804
1805         pm = MEM_callocN(sizeof(PTCacheMem), "Pointcache mem");
1806
1807         pm->totpoint = pid->totwrite(pid->calldata, cfra);
1808         pm->data_types = cfra ? pid->data_types : pid->info_types;
1809
1810         ptcache_data_alloc(pm);
1811         BKE_ptcache_mem_pointers_init(pm);
1812
1813         if(overwrite) {
1814                 if(cache->flag & PTCACHE_DISK_CACHE) {
1815                         int fra = cfra-1;
1816
1817                         while(fra >= cache->startframe && !BKE_ptcache_id_exist(pid, fra))
1818                                 fra--;
1819
1820                         pm2 = ptcache_disk_frame_to_mem(pid, fra);
1821                 }
1822                 else
1823                         pm2 = cache->mem_cache.last;
1824         }
1825
1826         if(pid->write_point) {
1827                 for(i=0; i<totpoint; i++) {
1828                         int write = pid->write_point(i, pid->calldata, pm->cur, cfra);
1829                         if(write) {
1830                                 BKE_ptcache_mem_pointers_incr(pm);
1831
1832                                 /* newly born particles have to be copied to previous cached frame */
1833                                 if(overwrite && write == 2 && pm2 && BKE_ptcache_mem_pointers_seek(i, pm2))
1834                                         pid->write_point(i, pid->calldata, pm2->cur, cfra);
1835                         }
1836                 }
1837         }
1838
1839         if(pid->write_extra_data)
1840                 pid->write_extra_data(pid->calldata, pm, cfra);
1841
1842         pm->frame = cfra;
1843
1844         if(cache->flag & PTCACHE_DISK_CACHE) {
1845                 error += !ptcache_mem_frame_to_disk(pid, pm);
1846
1847                 // if(pm) /* pm is always set */
1848                 {
1849                         ptcache_data_free(pm);
1850                         ptcache_extra_free(pm);
1851                         MEM_freeN(pm);
1852                 }
1853
1854                 if(pm2) {
1855                         error += !ptcache_mem_frame_to_disk(pid, pm2);
1856                         ptcache_data_free(pm2);
1857                         ptcache_extra_free(pm2);
1858                         MEM_freeN(pm2);
1859                 }
1860         }
1861         else {
1862                 BLI_addtail(&cache->mem_cache, pm);
1863         }
1864
1865         return error;
1866 }
1867 static int ptcache_write_needed(PTCacheID *pid, int cfra, int *overwrite)
1868 {
1869         PointCache *cache = pid->cache;
1870         int ofra = 0, efra = cache->endframe;
1871
1872         /* allways start from scratch on the first frame */
1873         if(cfra && cfra == cache->startframe) {
1874                 BKE_ptcache_id_clear(pid, PTCACHE_CLEAR_ALL, cfra);
1875                 cache->flag &= ~PTCACHE_REDO_NEEDED;
1876                 return 1;
1877         }
1878
1879         if(pid->cache->flag & PTCACHE_DISK_CACHE) {
1880                 if(cfra==0 && cache->startframe > 0)
1881                         return 1;
1882
1883                                 /* find last cached frame */
1884                 while(efra > cache->startframe && !BKE_ptcache_id_exist(pid, efra))
1885                         efra--;
1886
1887                 /* find second last cached frame */
1888                 ofra = efra-1;
1889                 while(ofra > cache->startframe && !BKE_ptcache_id_exist(pid, ofra))
1890                         ofra--;
1891         }
1892         else {
1893                 PTCacheMem *pm = cache->mem_cache.last;
1894                 /* don't write info file in memory */
1895                 if(cfra == 0)
1896                         return 0;
1897
1898                 if(pm == NULL)
1899                         return 1;
1900
1901                 efra = pm->frame;
1902                 ofra = (pm->prev ? pm->prev->frame : efra - cache->step);
1903         }
1904
1905         if(efra >= cache->startframe && cfra > efra) {
1906                 if(ofra >= cache->startframe && efra - ofra < cache->step) {
1907                         /* overwrite previous frame */
1908                         BKE_ptcache_id_clear(pid, PTCACHE_CLEAR_FRAME, efra);
1909                         *overwrite = 1;
1910                 }
1911                 return 1;
1912         }
1913
1914         return 0;
1915 }
1916 /* writes cache to disk or memory */
1917 int BKE_ptcache_write(PTCacheID *pid, unsigned int cfra)
1918 {
1919         PointCache *cache = pid->cache;
1920         int totpoint = pid->totpoint(pid->calldata, cfra);
1921         int overwrite = 0, error = 0;
1922
1923         if(totpoint == 0 || (cfra ? pid->data_types == 0 : pid->info_types == 0))
1924                 return 0;
1925
1926         if(ptcache_write_needed(pid, cfra, &overwrite)==0)
1927                 return 0;
1928
1929         if(pid->write_stream) {
1930                 ptcache_write_stream(pid, cfra, totpoint);
1931         }
1932         else if(pid->write_point) {
1933                 error += ptcache_write(pid, cfra, overwrite);
1934         }
1935
1936         /* Mark frames skipped if more than 1 frame forwards since last non-skipped frame. */
1937         if(cfra - cache->last_exact == 1 || cfra == cache->startframe) {
1938                 cache->last_exact = cfra;
1939                 cache->flag &= ~PTCACHE_FRAMES_SKIPPED;
1940         }
1941         /* Don't mark skipped when writing info file (frame 0) */
1942         else if(cfra)
1943                 cache->flag |= PTCACHE_FRAMES_SKIPPED;
1944
1945         /* Update timeline cache display */
1946         if(cfra && cache->cached_frames)
1947                 cache->cached_frames[cfra-cache->startframe] = 1;
1948
1949         BKE_ptcache_update_info(pid);
1950
1951         return !error;
1952 }
1953 /* youll need to close yourself after!
1954  * mode - PTCACHE_CLEAR_ALL,
1955
1956 */
1957 /* Clears & resets */
1958 void BKE_ptcache_id_clear(PTCacheID *pid, int mode, unsigned int cfra)
1959 {
1960         unsigned int len; /* store the length of the string */
1961         unsigned int sta, end;
1962
1963         /* mode is same as fopen's modes */
1964         DIR *dir;
1965         struct dirent *de;
1966         char path[MAX_PTCACHE_PATH];
1967         char filename[MAX_PTCACHE_FILE];
1968         char path_full[MAX_PTCACHE_FILE];
1969         char ext[MAX_PTCACHE_PATH];
1970
1971         if(!pid || !pid->cache || pid->cache->flag & PTCACHE_BAKED)
1972                 return;
1973
1974         sta = pid->cache->startframe;
1975         end = pid->cache->endframe;
1976
1977 #ifndef DURIAN_POINTCACHE_LIB_OK
1978         /* don't allow clearing for linked objects */
1979         if(pid->ob->id.lib)
1980                 return;
1981 #endif
1982
1983         /*if (!G.relbase_valid) return; *//* save blend file before using pointcache */
1984
1985         /* clear all files in the temp dir with the prefix of the ID and the ".bphys" suffix */
1986         switch (mode) {
1987         case PTCACHE_CLEAR_ALL:
1988         case PTCACHE_CLEAR_BEFORE:
1989         case PTCACHE_CLEAR_AFTER:
1990                 if(pid->cache->flag & PTCACHE_DISK_CACHE) {
1991                         ptcache_path(pid, path);
1992
1993                         len = ptcache_filename(pid, filename, cfra, 0, 0); /* no path */
1994
1995                         dir = opendir(path);
1996                         if (dir==NULL)
1997                                 return;
1998
1999                         snprintf(ext, sizeof(ext), "_%02d"PTCACHE_EXT, pid->stack_index);
2000
2001                         while ((de = readdir(dir)) != NULL) {
2002                                 if (strstr(de->d_name, ext)) { /* do we have the right extension?*/
2003                                         if (strncmp(filename, de->d_name, len ) == 0) { /* do we have the right prefix */
2004                                                 if (mode == PTCACHE_CLEAR_ALL) {
2005                                                         pid->cache->last_exact = MIN2(pid->cache->startframe, 0);
2006                                                         BLI_join_dirfile(path_full, sizeof(path_full), path, de->d_name);
2007                                                         BLI_delete(path_full, 0, 0);
2008                                                 } else {
2009                                                         /* read the number of the file */
2010                                                         unsigned int frame, len2 = (int)strlen(de->d_name);
2011                                                         char num[7];
2012
2013                                                         if (len2 > 15) { /* could crash if trying to copy a string out of this range*/
2014                                                                 BLI_strncpy(num, de->d_name + (strlen(de->d_name) - 15), sizeof(num));
2015                                                                 frame = atoi(num);
2016
2017                                                                 if((mode==PTCACHE_CLEAR_BEFORE && frame < cfra) ||
2018                                                                 (mode==PTCACHE_CLEAR_AFTER && frame > cfra)     ) {
2019
2020                                                                         BLI_join_dirfile(path_full, sizeof(path_full), path, de->d_name);
2021                                                                         BLI_delete(path_full, 0, 0);
2022                                                                         if(pid->cache->cached_frames && frame >=sta && frame <= end)
2023                                                                                 pid->cache->cached_frames[frame-sta] = 0;
2024                                                                 }
2025                                                         }
2026                                                 }
2027                                         }
2028                                 }
2029                         }
2030                         closedir(dir);
2031
2032                         if(mode == PTCACHE_CLEAR_ALL && pid->cache->cached_frames)
2033                                 memset(pid->cache->cached_frames, 0, MEM_allocN_len(pid->cache->cached_frames));
2034                 }
2035                 else {
2036                         PTCacheMem *pm= pid->cache->mem_cache.first;
2037                         PTCacheMem *link= NULL;
2038
2039                         if(mode == PTCACHE_CLEAR_ALL) {
2040                                 /*we want startframe if the cache starts before zero*/
2041                                 pid->cache->last_exact = MIN2(pid->cache->startframe, 0);
2042                                 for(; pm; pm=pm->next) {
2043                                         ptcache_data_free(pm);
2044                                         ptcache_extra_free(pm);
2045                                 }
2046                                 BLI_freelistN(&pid->cache->mem_cache);
2047
2048                                 if(pid->cache->cached_frames)
2049                                         memset(pid->cache->cached_frames, 0, MEM_allocN_len(pid->cache->cached_frames));
2050                         } else {
2051                                 while(pm) {
2052                                         if((mode==PTCACHE_CLEAR_BEFORE && pm->frame < cfra)     ||
2053                                         (mode==PTCACHE_CLEAR_AFTER && pm->frame > cfra) ) {
2054                                                 link = pm;
2055                                                 if(pid->cache->cached_frames && pm->frame >=sta && pm->frame <= end)
2056                                                         pid->cache->cached_frames[pm->frame-sta] = 0;
2057                                                 ptcache_data_free(pm);
2058                                                 ptcache_extra_free(pm);
2059                                                 pm = pm->next;
2060                                                 BLI_freelinkN(&pid->cache->mem_cache, link);
2061                                         }
2062                                         else
2063                                                 pm = pm->next;
2064                                 }
2065                         }
2066                 }
2067                 break;
2068
2069         case PTCACHE_CLEAR_FRAME:
2070                 if(pid->cache->flag & PTCACHE_DISK_CACHE) {
2071                         if(BKE_ptcache_id_exist(pid, cfra)) {
2072                                 ptcache_filename(pid, filename, cfra, 1, 1); /* no path */
2073                                 BLI_delete(filename, 0, 0);
2074                         }
2075                 }
2076                 else {
2077                         PTCacheMem *pm = pid->cache->mem_cache.first;
2078
2079                         for(; pm; pm=pm->next) {
2080                                 if(pm->frame == cfra) {
2081                                         ptcache_data_free(pm);
2082                                         ptcache_extra_free(pm);
2083                                         BLI_freelinkN(&pid->cache->mem_cache, pm);
2084                                         break;
2085                                 }
2086                         }
2087                 }
2088                 if(pid->cache->cached_frames && cfra>=sta && cfra<=end)
2089                         pid->cache->cached_frames[cfra-sta] = 0;
2090                 break;
2091         }
2092
2093         BKE_ptcache_update_info(pid);
2094 }
2095 int  BKE_ptcache_id_exist(PTCacheID *pid, int cfra)
2096 {
2097         if(!pid->cache)
2098                 return 0;
2099
2100         if(cfra<pid->cache->startframe || cfra > pid->cache->endframe)
2101                 return 0;
2102
2103         if(pid->cache->cached_frames && pid->cache->cached_frames[cfra-pid->cache->startframe]==0)
2104                 return 0;
2105
2106         if(pid->cache->flag & PTCACHE_DISK_CACHE) {
2107                 char filename[MAX_PTCACHE_FILE];
2108
2109                 ptcache_filename(pid, filename, cfra, 1, 1);
2110
2111                 return BLI_exists(filename);
2112         }
2113         else {
2114                 PTCacheMem *pm = pid->cache->mem_cache.first;
2115
2116                 for(; pm; pm=pm->next) {
2117                         if(pm->frame==cfra)
2118                                 return 1;
2119                 }
2120                 return 0;
2121         }
2122 }
2123 void BKE_ptcache_id_time(PTCacheID *pid, Scene *scene, float cfra, int *startframe, int *endframe, float *timescale)
2124 {
2125         Object *ob;
2126         PointCache *cache;
2127         /* float offset; unused for now */
2128         float time, nexttime;
2129
2130         /* TODO: this has to be sorter out once bsystem_time gets redone, */
2131         /*       now caches can handle interpolating etc. too - jahka */
2132
2133         /* time handling for point cache:
2134          * - simulation time is scaled by result of bsystem_time
2135          * - for offsetting time only time offset is taken into account, since
2136          *   that's always the same and can't be animated. a timeoffset which
2137          *   varies over time is not simpe to support.
2138          * - field and motion blur offsets are currently ignored, proper solution
2139          *   is probably to interpolate results from two frames for that ..
2140          */
2141
2142         ob= pid->ob;
2143         cache= pid->cache;
2144
2145         if(timescale) {
2146                 time= bsystem_time(scene, ob, cfra, 0.0f);
2147                 nexttime= bsystem_time(scene, ob, cfra+1.0f, 0.0f);
2148
2149                 *timescale= MAX2(nexttime - time, 0.0f);
2150         }
2151
2152         if(startframe && endframe) {
2153                 *startframe= cache->startframe;
2154                 *endframe= cache->endframe;
2155
2156                 /* TODO: time handling with object offsets and simulated vs. cached
2157                  * particles isn't particularly easy, so for now what you see is what
2158                  * you get. In the future point cache could handle the whole particle
2159                  * system timing. */
2160 #if 0
2161                 if ((ob->partype & PARSLOW)==0) {
2162                         offset= give_timeoffset(ob);
2163
2164                         *startframe += (int)(offset+0.5f);
2165                         *endframe += (int)(offset+0.5f);
2166                 }
2167 #endif
2168         }
2169
2170         /* verify cached_frames array is up to date */
2171         if(cache->cached_frames) {
2172                 if(MEM_allocN_len(cache->cached_frames) != sizeof(char) * (cache->endframe-cache->startframe+1)) {
2173                         MEM_freeN(cache->cached_frames);
2174                         cache->cached_frames = NULL;
2175                 }
2176         }
2177
2178         if(cache->cached_frames==NULL && cache->endframe > cache->startframe) {
2179                 unsigned int sta=cache->startframe;
2180                 unsigned int end=cache->endframe;
2181
2182                 cache->cached_frames = MEM_callocN(sizeof(char) * (cache->endframe-cache->startframe+1), "cached frames array");
2183
2184                 if(pid->cache->flag & PTCACHE_DISK_CACHE) {
2185                         /* mode is same as fopen's modes */
2186                         DIR *dir;
2187                         struct dirent *de;
2188                         char path[MAX_PTCACHE_PATH];
2189                         char filename[MAX_PTCACHE_FILE];
2190                         char ext[MAX_PTCACHE_PATH];
2191                         unsigned int len; /* store the length of the string */
2192
2193                         ptcache_path(pid, path);
2194
2195                         len = ptcache_filename(pid, filename, (int)cfra, 0, 0); /* no path */
2196
2197                         dir = opendir(path);
2198                         if (dir==NULL)
2199                                 return;
2200
2201                         snprintf(ext, sizeof(ext), "_%02d"PTCACHE_EXT, pid->stack_index);
2202
2203                         while ((de = readdir(dir)) != NULL) {
2204                                 if (strstr(de->d_name, ext)) { /* do we have the right extension?*/
2205                                         if (strncmp(filename, de->d_name, len ) == 0) { /* do we have the right prefix */
2206                                                 /* read the number of the file */
2207                                                 unsigned int frame, len2 = (int)strlen(de->d_name);
2208                                                 char num[7];
2209
2210                                                 if (len2 > 15) { /* could crash if trying to copy a string out of this range*/
2211                                                         BLI_strncpy(num, de->d_name + (strlen(de->d_name) - 15), sizeof(num));
2212                                                         frame = atoi(num);
2213
2214                                                         if(frame >= sta && frame <= end)
2215                                                                 cache->cached_frames[frame-sta] = 1;
2216                                                 }
2217                                         }
2218                                 }
2219                         }
2220                         closedir(dir);
2221                 }
2222                 else {
2223                         PTCacheMem *pm= pid->cache->mem_cache.first;
2224
2225                         while(pm) {
2226                                 if(pm->frame >= sta && pm->frame <= end)
2227                                         cache->cached_frames[pm->frame-sta] = 1;
2228                                 pm = pm->next;
2229                         }
2230                 }
2231         }
2232 }
2233 int  BKE_ptcache_id_reset(Scene *scene, PTCacheID *pid, int mode)
2234 {
2235         PointCache *cache;
2236         int reset, clear, after;
2237
2238         if(!pid->cache)
2239                 return 0;
2240
2241         cache= pid->cache;
2242         reset= 0;
2243         clear= 0;
2244         after= 0;
2245
2246         if(mode == PTCACHE_RESET_DEPSGRAPH) {
2247                 if(!(cache->flag & PTCACHE_BAKED) && !BKE_ptcache_get_continue_physics()) {
2248                         if(cache->flag & PTCACHE_QUICK_CACHE)
2249                                 clear= 1;
2250
2251                         after= 1;
2252                 }
2253
2254                 cache->flag |= PTCACHE_OUTDATED;
2255         }
2256         else if(mode == PTCACHE_RESET_BAKED) {
2257                 if(!BKE_ptcache_get_continue_physics()) {
2258                         reset= 1;
2259                         clear= 1;
2260                 }
2261                 else
2262                         cache->flag |= PTCACHE_OUTDATED;
2263         }
2264         else if(mode == PTCACHE_RESET_OUTDATED) {
2265                 reset = 1;
2266
2267                 if(cache->flag & PTCACHE_OUTDATED && !(cache->flag & PTCACHE_BAKED)) {
2268                         clear= 1;
2269                         cache->flag &= ~PTCACHE_OUTDATED;
2270                 }
2271         }
2272
2273         if(reset) {
2274                 BKE_ptcache_invalidate(cache);
2275                 cache->flag &= ~PTCACHE_REDO_NEEDED;
2276
2277                 if(pid->type == PTCACHE_TYPE_CLOTH)
2278                         cloth_free_modifier(pid->calldata);
2279                 else if(pid->type == PTCACHE_TYPE_SOFTBODY)
2280                         sbFreeSimulation(pid->calldata);
2281                 else if(pid->type == PTCACHE_TYPE_PARTICLES)
2282                         psys_reset(pid->calldata, PSYS_RESET_DEPSGRAPH);
2283                 else if(pid->type == PTCACHE_TYPE_SMOKE_DOMAIN)
2284                         smokeModifier_reset(pid->calldata);
2285                 else if(pid->type == PTCACHE_TYPE_SMOKE_HIGHRES)
2286                         smokeModifier_reset_turbulence(pid->calldata);
2287         }
2288         if(clear)
2289                 BKE_ptcache_id_clear(pid, PTCACHE_CLEAR_ALL, 0);
2290         else if(after)
2291                 BKE_ptcache_id_clear(pid, PTCACHE_CLEAR_AFTER, CFRA);
2292
2293         return (reset || clear || after);
2294 }
2295 int  BKE_ptcache_object_reset(Scene *scene, Object *ob, int mode)
2296 {
2297         PTCacheID pid;
2298         ParticleSystem *psys;
2299         ModifierData *md;
2300         int reset, skip;
2301
2302         reset= 0;
2303         skip= 0;
2304
2305         if(ob->soft) {
2306                 BKE_ptcache_id_from_softbody(&pid, ob, ob->soft);
2307                 reset |= BKE_ptcache_id_reset(scene, &pid, mode);
2308         }
2309
2310         for(psys=ob->particlesystem.first; psys; psys=psys->next) {
2311                 /* children or just redo can be calculated without reseting anything */
2312                 if(psys->recalc & PSYS_RECALC_REDO || psys->recalc & PSYS_RECALC_CHILD)
2313                         skip = 1;
2314                 /* Baked cloth hair has to be checked too, because we don't want to reset */
2315                 /* particles or cloth in that case -jahka */
2316                 else if(psys->clmd) {
2317                         BKE_ptcache_id_from_cloth(&pid, ob, psys->clmd);
2318                         if(mode == PSYS_RESET_ALL || !(psys->part->type == PART_HAIR && (pid.cache->flag & PTCACHE_BAKED)))
2319                                 reset |= BKE_ptcache_id_reset(scene, &pid, mode);
2320                         else
2321                                 skip = 1;
2322                 }
2323
2324                 if(skip == 0 && psys->part) {
2325                         BKE_ptcache_id_from_particles(&pid, ob, psys);
2326                         reset |= BKE_ptcache_id_reset(scene, &pid, mode);
2327                 }
2328         }
2329
2330         for(md=ob->modifiers.first; md; md=md->next) {
2331                 if(md->type == eModifierType_Cloth) {
2332                         BKE_ptcache_id_from_cloth(&pid, ob, (ClothModifierData*)md);
2333                         reset |= BKE_ptcache_id_reset(scene, &pid, mode);
2334                 }
2335                 if(md->type == eModifierType_Smoke) {
2336                         SmokeModifierData *smd = (SmokeModifierData *)md;
2337                         if(smd->type & MOD_SMOKE_TYPE_DOMAIN)
2338                         {
2339                                 BKE_ptcache_id_from_smoke(&pid, ob, (SmokeModifierData*)md);
2340                                 reset |= BKE_ptcache_id_reset(scene, &pid, mode);
2341                         }
2342                 }
2343         }
2344
2345         if (ob->type == OB_ARMATURE)
2346                 BIK_clear_cache(ob->pose);
2347
2348         return reset;
2349 }
2350
2351 /* Use this when quitting blender, with unsaved files */
2352 void BKE_ptcache_remove(void)
2353 {
2354         char path[MAX_PTCACHE_PATH];
2355         char path_full[MAX_PTCACHE_PATH];
2356         int rmdir = 1;
2357
2358         ptcache_path(NULL, path);
2359
2360         if (BLI_exist(path)) {
2361                 /* The pointcache dir exists? - remove all pointcache */
2362
2363                 DIR *dir;
2364                 struct dirent *de;
2365
2366                 dir = opendir(path);
2367                 if (dir==NULL)
2368                         return;
2369
2370                 while ((de = readdir(dir)) != NULL) {
2371                         if( strcmp(de->d_name, ".")==0 || strcmp(de->d_name, "..")==0) {
2372                                 /* do nothing */
2373                         } else if (strstr(de->d_name, PTCACHE_EXT)) { /* do we have the right extension?*/
2374                                 BLI_join_dirfile(path_full, sizeof(path_full), path, de->d_name);
2375                                 BLI_delete(path_full, 0, 0);
2376                         } else {
2377                                 rmdir = 0; /* unknown file, dont remove the dir */
2378                         }
2379                 }
2380
2381                 closedir(dir);
2382         } else {
2383                 rmdir = 0; /* path dosnt exist  */
2384         }
2385
2386         if (rmdir) {
2387                 BLI_delete(path, 1, 0);
2388         }
2389 }
2390
2391 /* Continuous Interaction */
2392
2393 static int CONTINUE_PHYSICS = 0;
2394
2395 void BKE_ptcache_set_continue_physics(Main *bmain, Scene *scene, int enable)
2396 {
2397         Object *ob;
2398
2399         if(CONTINUE_PHYSICS != enable) {
2400                 CONTINUE_PHYSICS = enable;
2401
2402                 if(CONTINUE_PHYSICS == 0) {
2403                         for(ob=bmain->object.first; ob; ob=ob->id.next)
2404                                 if(BKE_ptcache_object_reset(scene, ob, PTCACHE_RESET_OUTDATED))
2405                                         DAG_id_tag_update(&ob->id, OB_RECALC_DATA);
2406                 }
2407         }
2408 }
2409
2410 int  BKE_ptcache_get_continue_physics(void)
2411 {
2412         return CONTINUE_PHYSICS;
2413 }
2414
2415 /* Point Cache handling */
2416
2417 PointCache *BKE_ptcache_add(ListBase *ptcaches)
2418 {
2419         PointCache *cache;
2420
2421         cache= MEM_callocN(sizeof(PointCache), "PointCache");
2422         cache->startframe= 1;
2423         cache->endframe= 250;
2424         cache->step= 10;
2425         cache->index = -1;
2426
2427         BLI_addtail(ptcaches, cache);
2428
2429         return cache;
2430 }
2431
2432 void BKE_ptcache_free_mem(ListBase *mem_cache)
2433 {
2434         PTCacheMem *pm = mem_cache->first;
2435
2436         if(pm) {
2437                 for(; pm; pm=pm->next) {
2438                         ptcache_data_free(pm);
2439                         ptcache_extra_free(pm);
2440                 }
2441
2442                 BLI_freelistN(mem_cache);
2443         }
2444 }
2445 void BKE_ptcache_free(PointCache *cache)
2446 {
2447         BKE_ptcache_free_mem(&cache->mem_cache);
2448         if(cache->edit && cache->free_edit)
2449                 cache->free_edit(cache->edit);
2450         if(cache->cached_frames)
2451                 MEM_freeN(cache->cached_frames);
2452         MEM_freeN(cache);
2453 }
2454 void BKE_ptcache_free_list(ListBase *ptcaches)
2455 {
2456         PointCache *cache = ptcaches->first;
2457
2458         while(cache) {
2459                 BLI_remlink(ptcaches, cache);
2460                 BKE_ptcache_free(cache);
2461                 cache = ptcaches->first;
2462         }
2463 }
2464
2465 static PointCache *ptcache_copy(PointCache *cache)
2466 {
2467         PointCache *ncache;
2468
2469         ncache= MEM_dupallocN(cache);
2470
2471         /* hmm, should these be copied over instead? */
2472         ncache->mem_cache.first = NULL;
2473         ncache->mem_cache.last = NULL;
2474         ncache->cached_frames = NULL;
2475         ncache->edit = NULL;
2476
2477         ncache->flag= 0;
2478         ncache->simframe= 0;
2479
2480         return ncache;
2481 }
2482 /* returns first point cache */
2483 PointCache *BKE_ptcache_copy_list(ListBase *ptcaches_new, ListBase *ptcaches_old)
2484 {
2485         PointCache *cache = ptcaches_old->first;
2486
2487         ptcaches_new->first = ptcaches_new->last = NULL;
2488
2489         for(; cache; cache=cache->next)
2490                 BLI_addtail(ptcaches_new, ptcache_copy(cache));
2491
2492         return ptcaches_new->first;
2493 }
2494
2495
2496 /* Baking */
2497 void BKE_ptcache_quick_cache_all(Main *bmain, Scene *scene)
2498 {
2499         PTCacheBaker baker;
2500
2501         baker.bake=0;
2502         baker.break_data=NULL;
2503         baker.break_test=NULL;
2504         baker.pid=NULL;
2505         baker.progressbar=NULL;
2506         baker.progressend=NULL;
2507         baker.progresscontext=NULL;
2508         baker.render=0;
2509         baker.anim_init = 0;
2510         baker.main=bmain;
2511         baker.scene=scene;
2512         baker.quick_step=scene->physics_settings.quick_cache_step;
2513
2514         BKE_ptcache_bake(&baker);
2515 }
2516
2517 /* Simulation thread, no need for interlocks as data written in both threads
2518  are only unitary integers (I/O assumed to be atomic for them) */
2519 typedef struct {
2520         int break_operation;
2521         int thread_ended;
2522         int endframe;
2523         int step;
2524         int *cfra_ptr;
2525         Main *main;
2526         Scene *scene;
2527 } ptcache_bake_data;
2528
2529 static void ptcache_dt_to_str(char *str, double dtime)
2530 {
2531         if(dtime > 60.0) {
2532                 if(dtime > 3600.0)
2533                         sprintf(str, "%ih %im %is", (int)(dtime/3600), ((int)(dtime/60))%60, ((int)dtime) % 60);
2534                 else
2535                         sprintf(str, "%im %is", ((int)(dtime/60))%60, ((int)dtime) % 60);
2536         }
2537         else
2538                 sprintf(str, "%is", ((int)dtime) % 60);
2539 }
2540
2541 static void *ptcache_bake_thread(void *ptr) {
2542         int usetimer = 0, sfra, efra;
2543         double stime, ptime, ctime, fetd;
2544         char run[32], cur[32], etd[32];
2545
2546         ptcache_bake_data *data = (ptcache_bake_data*)ptr;
2547
2548         stime = ptime = PIL_check_seconds_timer();
2549         sfra = *data->cfra_ptr;
2550         efra = data->endframe;
2551
2552         for(; (*data->cfra_ptr <= data->endframe) && !data->break_operation; *data->cfra_ptr+=data->step) {
2553                 scene_update_for_newframe(data->main, data->scene, data->scene->lay);
2554                 if(G.background) {
2555                         printf("bake: frame %d :: %d\n", (int)*data->cfra_ptr, data->endframe);
2556                 }
2557                 else {
2558                         ctime = PIL_check_seconds_timer();
2559
2560                         fetd = (ctime-ptime)*(efra-*data->cfra_ptr)/data->step;
2561
2562                         if(usetimer || fetd > 60.0) {
2563                                 usetimer = 1;
2564
2565                                 ptcache_dt_to_str(cur, ctime-ptime);
2566                                 ptcache_dt_to_str(run, ctime-stime);
2567                                 ptcache_dt_to_str(etd, fetd);
2568
2569                                 printf("Baked for %s, current frame: %i/%i (%.3fs), ETC: %s          \r", run, *data->cfra_ptr-sfra+1, efra-sfra+1, ctime-ptime, etd);
2570                         }
2571                         ptime = ctime;
2572                 }
2573         }
2574
2575         if(usetimer) {
2576                 ptcache_dt_to_str(run, PIL_check_seconds_timer()-stime);
2577                 printf("Bake %s %s (%i frames simulated).                       \n", (data->break_operation ? "canceled after" : "finished in"), run, *data->cfra_ptr-sfra);
2578         }
2579
2580         data->thread_ended = TRUE;
2581         return NULL;
2582 }
2583
2584 /* if bake is not given run simulations to current frame */
2585 void BKE_ptcache_bake(PTCacheBaker* baker)
2586 {
2587         Main *bmain = baker->main;
2588         Scene *scene = baker->scene;
2589         Scene *sce_iter; /* SETLOOPER macro only */
2590         Base *base;
2591         ListBase pidlist;
2592         PTCacheID *pid = baker->pid;
2593         PointCache *cache = NULL;
2594         float frameleno = scene->r.framelen;
2595         int cfrao = CFRA;
2596         int startframe = MAXFRAME;
2597         int bake = baker->bake;
2598         int render = baker->render;
2599         ListBase threads;
2600         ptcache_bake_data thread_data;
2601         int progress, old_progress;
2602
2603         thread_data.endframe = baker->anim_init ? scene->r.sfra : CFRA;
2604         thread_data.step = baker->quick_step;
2605         thread_data.cfra_ptr = &CFRA;
2606         thread_data.scene = baker->scene;
2607         thread_data.main = baker->main;
2608
2609         G.afbreek = 0;
2610
2611         /* set caches to baking mode and figure out start frame */
2612         if(pid) {
2613                 /* cache/bake a single object */
2614                 cache = pid->cache;
2615                 if((cache->flag & PTCACHE_BAKED)==0) {
2616                         if(pid->type==PTCACHE_TYPE_PARTICLES) {
2617                                 ParticleSystem *psys= pid->calldata;
2618
2619                                 /* a bit confusing, could make this work better in the UI */
2620                                 if(psys->part->type == PART_EMITTER)
2621                                         psys_get_pointcache_start_end(scene, pid->calldata, &cache->startframe, &cache->endframe);
2622                         }
2623                         else if(pid->type == PTCACHE_TYPE_SMOKE_HIGHRES) {
2624                                 /* get all pids from the object and search for smoke low res */
2625                                 ListBase pidlist2;
2626                                 PTCacheID *pid2;
2627                                 BKE_ptcache_ids_from_object(&pidlist2, pid->ob, scene, MAX_DUPLI_RECUR);
2628                                 for(pid2=pidlist2.first; pid2; pid2=pid2->next) {
2629                                         if(pid2->type == PTCACHE_TYPE_SMOKE_DOMAIN)
2630                                         {
2631                                                 if(pid2->cache && !(pid2->cache->flag & PTCACHE_BAKED)) {
2632                                                         if(bake || pid2->cache->flag & PTCACHE_REDO_NEEDED)
2633                                                                 BKE_ptcache_id_clear(pid2, PTCACHE_CLEAR_ALL, 0);
2634                                                         if(bake) {
2635                                                                 pid2->cache->flag |= PTCACHE_BAKING;
2636                                                                 pid2->cache->flag &= ~PTCACHE_BAKED;
2637                                                         }
2638                                                 }
2639                                         }
2640                                 }
2641                                 BLI_freelistN(&pidlist2);
2642                         }
2643
2644                         if(bake || cache->flag & PTCACHE_REDO_NEEDED)
2645                                 BKE_ptcache_id_clear(pid, PTCACHE_CLEAR_ALL, 0);
2646
2647                         startframe = MAX2(cache->last_exact, cache->startframe);
2648
2649                         if(bake) {
2650                                 thread_data.endframe = cache->endframe;
2651                                 cache->flag |= PTCACHE_BAKING;
2652                         }
2653                         else {
2654                                 thread_data.endframe = MIN2(thread_data.endframe, cache->endframe);
2655                         }
2656
2657                         cache->flag &= ~PTCACHE_BAKED;
2658                 }
2659         }
2660         else for(SETLOOPER(scene, sce_iter, base)) {
2661                 /* cache/bake everything in the scene */
2662                 BKE_ptcache_ids_from_object(&pidlist, base->object, scene, MAX_DUPLI_RECUR);
2663
2664                 for(pid=pidlist.first; pid; pid=pid->next) {
2665                         cache = pid->cache;
2666                         if((cache->flag & PTCACHE_BAKED)==0) {
2667                                 if(pid->type==PTCACHE_TYPE_PARTICLES) {
2668                                         ParticleSystem *psys = (ParticleSystem*)pid->calldata;
2669                                         /* skip hair & keyed particles */
2670                                         if(psys->part->type == PART_HAIR || psys->part->phystype == PART_PHYS_KEYED)
2671                                                 continue;
2672
2673                                         psys_get_pointcache_start_end(scene, pid->calldata, &cache->startframe, &cache->endframe);
2674                                 }
2675
2676                                 if((cache->flag & PTCACHE_REDO_NEEDED || (cache->flag & PTCACHE_SIMULATION_VALID)==0)
2677                                         && ((cache->flag & PTCACHE_QUICK_CACHE)==0 || render || bake))
2678                                         BKE_ptcache_id_clear(pid, PTCACHE_CLEAR_ALL, 0);
2679
2680                                 startframe = MIN2(startframe, cache->startframe);
2681
2682                                 if(bake || render) {
2683                                         cache->flag |= PTCACHE_BAKING;
2684
2685                                         if(bake)
2686                                                 thread_data.endframe = MAX2(thread_data.endframe, cache->endframe);
2687                                 }
2688
2689                                 cache->flag &= ~PTCACHE_BAKED;
2690
2691                         }
2692                 }
2693                 BLI_freelistN(&pidlist);
2694         }
2695
2696         CFRA = startframe;
2697         scene->r.framelen = 1.0;
2698         thread_data.break_operation = FALSE;
2699         thread_data.thread_ended = FALSE;
2700         old_progress = -1;
2701
2702         WM_cursor_wait(1);
2703
2704         if(G.background) {
2705                 ptcache_bake_thread((void*)&thread_data);
2706         }
2707         else {
2708                 BLI_init_threads(&threads, ptcache_bake_thread, 1);
2709                 BLI_insert_thread(&threads, (void*)&thread_data);
2710
2711                 while (thread_data.thread_ended == FALSE) {
2712
2713                         if(bake)
2714                                 progress = (int)(100.0f * (float)(CFRA - startframe)/(float)(thread_data.endframe-startframe));
2715                         else
2716                                 progress = CFRA;
2717
2718                         /* NOTE: baking should not redraw whole ui as this slows things down */
2719                         if ((baker->progressbar) && (progress != old_progress)) {
2720                                 baker->progressbar(baker->progresscontext, progress);
2721                                 old_progress = progress;
2722                         }
2723
2724                         /* Delay to lessen CPU load from UI thread */
2725                         PIL_sleep_ms(200);
2726
2727                         /* NOTE: breaking baking should leave calculated frames in cache, not clear it */
2728                         if(blender_test_break() && !thread_data.break_operation) {
2729                                 thread_data.break_operation = TRUE;
2730                                 if (baker->progressend)
2731                                         baker->progressend(baker->progresscontext);
2732                                 WM_cursor_wait(1);
2733                         }
2734                 }
2735
2736         BLI_end_threads(&threads);
2737         }
2738         /* clear baking flag */
2739         if(pid) {
2740                 cache->flag &= ~(PTCACHE_BAKING|PTCACHE_REDO_NEEDED);
2741                 cache->flag |= PTCACHE_SIMULATION_VALID;
2742                 if(bake) {
2743                         cache->flag |= PTCACHE_BAKED;
2744                         /* write info file */
2745                         if(cache->flag & PTCACHE_DISK_CACHE)
2746                                 BKE_ptcache_write(pid, 0);
2747                 }
2748         }
2749         else for(SETLOOPER(scene, sce_iter, base)) {
2750                 BKE_ptcache_ids_from_object(&pidlist, base->object, scene, MAX_DUPLI_RECUR);
2751
2752                 for(pid=pidlist.first; pid; pid=pid->next) {
2753                         /* skip hair particles */
2754                         if(pid->type==PTCACHE_TYPE_PARTICLES && ((ParticleSystem*)pid->calldata)->part->type == PART_HAIR)
2755                                 continue;
2756
2757                         cache = pid->cache;
2758
2759                         if(thread_data.step > 1)
2760                                 cache->flag &= ~(PTCACHE_BAKING|PTCACHE_OUTDATED);
2761                         else
2762                                 cache->flag &= ~(PTCACHE_BAKING|PTCACHE_REDO_NEEDED);
2763
2764                         cache->flag |= PTCACHE_SIMULATION_VALID;
2765
2766                         if(bake) {
2767                                 cache->flag |= PTCACHE_BAKED;
2768                                 if(cache->flag & PTCACHE_DISK_CACHE)
2769                                         BKE_ptcache_write(pid, 0);
2770                         }
2771                 }
2772                 BLI_freelistN(&pidlist);
2773         }
2774
2775         scene->r.framelen = frameleno;
2776         CFRA = cfrao;
2777
2778         if(bake) /* already on cfra unless baking */
2779                 scene_update_for_newframe(bmain, scene, scene->lay);
2780
2781         if (thread_data.break_operation)
2782                 WM_cursor_wait(0);
2783         else if (baker->progressend)
2784                 baker->progressend(baker->progresscontext);
2785
2786         WM_cursor_wait(0);
2787
2788         /* TODO: call redraw all windows somehow */
2789 }
2790 /* Helpers */
2791 void BKE_ptcache_disk_to_mem(PTCacheID *pid)
2792 {
2793         PointCache *cache = pid->cache;
2794         PTCacheMem *pm = NULL;
2795         int baked = cache->flag & PTCACHE_BAKED;
2796         int cfra, sfra = cache->startframe, efra = cache->endframe;
2797
2798         /* Remove possible bake flag to allow clear */
2799         cache->flag &= ~PTCACHE_BAKED;
2800
2801         /* PTCACHE_DISK_CACHE flag was cleared already */
2802         BKE_ptcache_id_clear(pid, PTCACHE_CLEAR_ALL, 0);
2803
2804         /* restore possible bake flag */
2805         cache->flag |= baked;
2806
2807         for(cfra=sfra; cfra <= efra; cfra++) {
2808                 pm = ptcache_disk_frame_to_mem(pid, cfra);
2809
2810                 if(pm)
2811                         BLI_addtail(&pid->cache->mem_cache, pm);
2812         }
2813 }
2814 void BKE_ptcache_mem_to_disk(PTCacheID *pid)
2815 {
2816         PointCache *cache = pid->cache;
2817         PTCacheMem *pm = cache->mem_cache.first;
2818         int baked = cache->flag & PTCACHE_BAKED;
2819
2820         /* Remove possible bake flag to allow clear */
2821         cache->flag &= ~PTCACHE_BAKED;
2822
2823         /* PTCACHE_DISK_CACHE flag was set already */
2824         BKE_ptcache_id_clear(pid, PTCACHE_CLEAR_ALL, 0);
2825
2826         /* restore possible bake flag */
2827         cache->flag |= baked;
2828
2829         for(; pm; pm=pm->next) {
2830                 if(ptcache_mem_frame_to_disk(pid, pm)==0) {
2831                         cache->flag &= ~PTCACHE_DISK_CACHE;
2832                         break;
2833                 }
2834         }
2835
2836         /* write info file */
2837         if(cache->flag & PTCACHE_BAKED)
2838                 BKE_ptcache_write(pid, 0);
2839 }
2840 void BKE_ptcache_toggle_disk_cache(PTCacheID *pid)
2841 {
2842         PointCache *cache = pid->cache;
2843         int last_exact = cache->last_exact;
2844
2845         if (!G.relbase_valid){
2846                 cache->flag &= ~PTCACHE_DISK_CACHE;
2847                 if (G.f & G_DEBUG)
2848                         printf("File must be saved before using disk cache!\n");
2849                 return;
2850         }
2851
2852         if(cache->cached_frames) {
2853                 MEM_freeN(cache->cached_frames);
2854                 cache->cached_frames=NULL;
2855         }
2856
2857         if(cache->flag & PTCACHE_DISK_CACHE)
2858                 BKE_ptcache_mem_to_disk(pid);
2859         else
2860                 BKE_ptcache_disk_to_mem(pid);
2861
2862         cache->flag ^= PTCACHE_DISK_CACHE;
2863         BKE_ptcache_id_clear(pid, PTCACHE_CLEAR_ALL, 0);
2864         cache->flag ^= PTCACHE_DISK_CACHE;
2865
2866         cache->last_exact = last_exact;
2867
2868         BKE_ptcache_id_time(pid, NULL, 0.0f, NULL, NULL, NULL);
2869
2870         BKE_ptcache_update_info(pid);
2871 }
2872
2873 void BKE_ptcache_disk_cache_rename(PTCacheID *pid, char *from, char *to)
2874 {
2875         char old_name[80];
2876         int len; /* store the length of the string */
2877         /* mode is same as fopen's modes */
2878         DIR *dir;
2879         struct dirent *de;
2880         char path[MAX_PTCACHE_PATH];
2881         char old_filename[MAX_PTCACHE_FILE];
2882         char new_path_full[MAX_PTCACHE_FILE];
2883         char old_path_full[MAX_PTCACHE_FILE];
2884         char ext[MAX_PTCACHE_PATH];
2885
2886         /* save old name */
2887         strcpy(old_name, pid->cache->name);
2888
2889         /* get "from" filename */
2890         strcpy(pid->cache->name, from);
2891
2892         len = ptcache_filename(pid, old_filename, 0, 0, 0); /* no path */
2893
2894         ptcache_path(pid, path);
2895         dir = opendir(path);
2896         if(dir==NULL) {
2897                 strcpy(pid->cache->name, old_name);
2898                 return;
2899         }
2900
2901         snprintf(ext, sizeof(ext), "_%02d"PTCACHE_EXT, pid->stack_index);
2902
2903         /* put new name into cache */
2904         strcpy(pid->cache->name, to);
2905
2906         while ((de = readdir(dir)) != NULL) {
2907                 if (strstr(de->d_name, ext)) { /* do we have the right extension?*/
2908                         if (strncmp(old_filename, de->d_name, len ) == 0) { /* do we have the right prefix */
2909                                 /* read the number of the file */
2910                                 int frame, len2 = (int)strlen(de->d_name);
2911                                 char num[7];
2912
2913                                 if (len2 > 15) { /* could crash if trying to copy a string out of this range*/
2914                                         BLI_strncpy(num, de->d_name + (strlen(de->d_name) - 15), sizeof(num));
2915                                         frame = atoi(num);
2916
2917                                         BLI_join_dirfile(old_path_full, sizeof(old_path_full), path, de->d_name);
2918                                         ptcache_filename(pid, new_path_full, frame, 1, 1);
2919                                         BLI_rename(old_path_full, new_path_full);
2920                                 }
2921                         }
2922                 }
2923         }
2924         closedir(dir);
2925
2926         strcpy(pid->cache->name, old_name);
2927 }
2928
2929 void BKE_ptcache_load_external(PTCacheID *pid)
2930 {
2931         /*todo*/
2932         PointCache *cache = pid->cache;
2933         int len; /* store the length of the string */
2934         int info = 0;
2935         int start = MAXFRAME;
2936         int end = -1;
2937
2938         /* mode is same as fopen's modes */
2939         DIR *dir;
2940         struct dirent *de;
2941         char path[MAX_PTCACHE_PATH];
2942         char filename[MAX_PTCACHE_FILE];
2943         char ext[MAX_PTCACHE_PATH];
2944
2945         if(!cache)
2946                 return;
2947
2948         ptcache_path(pid, path);
2949
2950         len = ptcache_filename(pid, filename, 1, 0, 0); /* no path */
2951
2952         dir = opendir(path);
2953         if (dir==NULL)
2954                 return;
2955
2956         if(cache->index >= 0)
2957                 snprintf(ext, sizeof(ext), "_%02d"PTCACHE_EXT, cache->index);
2958         else
2959                 strcpy(ext, PTCACHE_EXT);
2960
2961         while ((de = readdir(dir)) != NULL) {
2962                 if (strstr(de->d_name, ext)) { /* do we have the right extension?*/
2963                         if (strncmp(filename, de->d_name, len ) == 0) { /* do we have the right prefix */
2964                                 /* read the number of the file */
2965                                 int frame, len2 = (int)strlen(de->d_name);
2966                                 char num[7];
2967
2968                                 if (len2 > 15) { /* could crash if trying to copy a string out of this range*/
2969                                         BLI_strncpy(num, de->d_name + (strlen(de->d_name) - 15), sizeof(num));
2970                                         frame = atoi(num);
2971
2972                                         if(frame) {
2973                                                 start = MIN2(start, frame);
2974                                                 end = MAX2(end, frame);
2975                                         }
2976                                         else
2977                                                 info = 1;
2978                                 }
2979                         }
2980                 }
2981         }
2982         closedir(dir);
2983
2984         if(start != MAXFRAME) {
2985                 PTCacheFile *pf;
2986
2987                 cache->startframe = start;
2988                 cache->endframe = end;
2989                 cache->totpoint = 0;
2990
2991                 if(pid->type == PTCACHE_TYPE_SMOKE_DOMAIN)
2992                         ; /*necessary info in every file*/
2993                 /* read totpoint from info file (frame 0) */
2994                 else if(info) {
2995                         pf= ptcache_file_open(pid, PTCACHE_FILE_READ, 0);
2996
2997                         if(pf) {
2998                                 if(ptcache_file_header_begin_read(pf)) {
2999                                         if(pf->type == pid->type && pid->read_header(pf)) {
3000                                                 cache->totpoint = pf->totpoint;
3001                                                 cache->flag |= PTCACHE_READ_INFO;
3002                                         }
3003                                         else {
3004                                                 cache->totpoint = 0;
3005                                         }
3006                                 }
3007                                 ptcache_file_close(pf);
3008                         }
3009                 }
3010                 /* or from any old format cache file */
3011                 else {
3012                         float old_data[14];
3013                         int elemsize = ptcache_old_elemsize(pid);
3014                         pf= ptcache_file_open(pid, PTCACHE_FILE_READ, cache->startframe);
3015
3016                         if(pf) {
3017                                 while(ptcache_file_read(pf, old_data, 1, elemsize))
3018                                         cache->totpoint++;
3019
3020                                 ptcache_file_close(pf);
3021                         }
3022                 }
3023                 cache->flag |= (PTCACHE_BAKED|PTCACHE_DISK_CACHE|PTCACHE_SIMULATION_VALID);
3024                 cache->flag &= ~(PTCACHE_OUTDATED|PTCACHE_FRAMES_SKIPPED);
3025         }
3026
3027         BKE_ptcache_update_info(pid);
3028 }
3029
3030 void BKE_ptcache_update_info(PTCacheID *pid)
3031 {
3032         PointCache *cache = pid->cache;
3033         PTCacheExtra *extra = NULL;
3034         int totframes = 0;
3035         char mem_info[64];
3036
3037         if(cache->flag & PTCACHE_EXTERNAL) {
3038                 int cfra = cache->startframe;
3039
3040                 for(; cfra<=cache->endframe; cfra++) {
3041                         if(BKE_ptcache_id_exist(pid, cfra))
3042                                 totframes++;
3043                 }
3044
3045                 /* smoke doesn't use frame 0 as info frame so can't check based on totpoint */
3046                 if(pid->type == PTCACHE_TYPE_SMOKE_DOMAIN && totframes)
3047                         sprintf(cache->info, "%i frames found!", totframes);
3048                 else if(totframes && cache->totpoint)
3049                         sprintf(cache->info, "%i points found!", cache->totpoint);
3050                 else
3051                         sprintf(cache->info, "No valid data to read!");
3052                 return;
3053         }
3054
3055         if(cache->flag & PTCACHE_DISK_CACHE) {
3056                 if(pid->type == PTCACHE_TYPE_SMOKE_DOMAIN)
3057                 {
3058                         int totpoint = pid->totpoint(pid->calldata, 0);
3059
3060                         if(cache->totpoint > totpoint)
3061                                 sprintf(mem_info, "%i cells + High Resolution cached", totpoint);
3062                         else
3063                                 sprintf(mem_info, "%i cells cached", totpoint);
3064                 }
3065                 else {
3066                         int cfra = cache->startframe;
3067
3068                         for(; cfra<=cache->endframe; cfra++) {
3069                                 if(BKE_ptcache_id_exist(pid, cfra))
3070                                         totframes++;
3071                         }
3072
3073                         sprintf(mem_info, "%i frames on disk", totframes);
3074                 }
3075         }
3076         else {
3077                 PTCacheMem *pm = cache->mem_cache.first;
3078                 float bytes = 0.0f;
3079                 int i, mb;
3080
3081                 for(; pm; pm=pm->next) {
3082                         for(i=0; i<BPHYS_TOT_DATA; i++)
3083                                 bytes += MEM_allocN_len(pm->data[i]);
3084
3085                         for(extra=pm->extradata.first; extra; extra=extra->next) {
3086                                 bytes += MEM_allocN_len(extra->data);
3087                                 bytes += sizeof(PTCacheExtra);
3088                         }
3089
3090                         bytes += sizeof(PTCacheMem);
3091
3092                         totframes++;
3093                 }
3094
3095                 mb = (bytes > 1024.0f * 1024.0f);
3096
3097                 sprintf(mem_info, "%i frames in memory (%.1f %s)",
3098                         totframes,
3099                         bytes / (mb ? 1024.0f * 1024.0f : 1024.0f),
3100                         mb ? "Mb" : "kb");
3101         }
3102
3103         if(cache->flag & PTCACHE_OUTDATED) {
3104                 sprintf(cache->info, "%s, cache is outdated!", mem_info);
3105         }
3106         else if(cache->flag & PTCACHE_FRAMES_SKIPPED) {
3107                 sprintf(cache->info, "%s, not exact since frame %i.", mem_info, cache->last_exact);
3108         }
3109         else
3110                 sprintf(cache->info, "%s.", mem_info);
3111 }
3112
3113 void BKE_ptcache_validate(PointCache *cache, int framenr)
3114 {
3115         if(cache) {
3116                 cache->flag |= PTCACHE_SIMULATION_VALID;
3117                 cache->simframe = framenr;
3118         }
3119 }
3120 void BKE_ptcache_invalidate(PointCache *cache)
3121 {
3122         if(cache) {
3123                 cache->flag &= ~PTCACHE_SIMULATION_VALID;
3124                 cache->simframe = 0;
3125                 cache->last_exact = MIN2(cache->startframe, 0);
3126         }
3127 }
3128