Refactor/enhance BKE_particlesettings_make_local().

Now using modern features from libquery/libremap areas.

Provides same kind of fixes/improvements as for BKE_object_make_local() (see rBd1a4ae3f395a6).
This commit is contained in:
Bastien Montagne 2016-07-10 15:13:01 +02:00
parent bb8bfdd4b6
commit c8c00636bf
3 changed files with 31 additions and 45 deletions

@ -324,7 +324,7 @@ struct ModifierData *object_add_particle_system(struct Scene *scene, struct Obje
void object_remove_particle_system(struct Scene *scene, struct Object *ob);
struct ParticleSettings *psys_new_settings(const char *name, struct Main *main);
struct ParticleSettings *BKE_particlesettings_copy(struct Main *bmain, struct ParticleSettings *part);
void BKE_particlesettings_make_local(struct ParticleSettings *part);
void BKE_particlesettings_make_local(struct Main *bmain, struct ParticleSettings *part);
void psys_reset(struct ParticleSystem *psys, int mode);

@ -327,7 +327,7 @@ bool id_make_local(Main *bmain, ID *id, bool test)
if (!test) BKE_brush_make_local((Brush *)id);
return true;
case ID_PA:
if (!test) BKE_particlesettings_make_local((ParticleSettings *)id);
if (!test) BKE_particlesettings_make_local(bmain, (ParticleSettings *)id);
return true;
case ID_WM:
return false; /* can't be linked */

@ -76,6 +76,8 @@
#include "BKE_material.h"
#include "BKE_key.h"
#include "BKE_library.h"
#include "BKE_library_query.h"
#include "BKE_library_remap.h"
#include "BKE_depsgraph.h"
#include "BKE_modifier.h"
#include "BKE_mesh.h"
@ -3339,20 +3341,23 @@ ParticleSettings *BKE_particlesettings_copy(Main *bmain, ParticleSettings *part)
return partn;
}
static void expand_local_particlesettings(ParticleSettings *part)
static int extern_local_particlesettings_callback(
void *UNUSED(user_data), struct ID *UNUSED(id_self), struct ID **id_pointer, int cd_flag)
{
int i;
id_lib_extern((ID *)part->dup_group);
for (i = 0; i < MAX_MTEX; i++) {
if (part->mtex[i]) id_lib_extern((ID *)part->mtex[i]->tex);
/* We only tag usercounted ID usages as extern... Why? */
if ((cd_flag & IDWALK_USER) && *id_pointer) {
id_lib_extern(*id_pointer);
}
return IDWALK_RET_NOP;
}
void BKE_particlesettings_make_local(ParticleSettings *part)
static void expand_local_particlesettings(ParticleSettings *part)
{
BKE_library_foreach_ID_link(&part->id, extern_local_particlesettings_callback, NULL, 0);
}
void BKE_particlesettings_make_local(Main *bmain, ParticleSettings *part)
{
Main *bmain = G.main;
Object *ob;
bool is_local = false, is_lib = false;
/* - only lib users: do nothing
@ -3360,45 +3365,26 @@ void BKE_particlesettings_make_local(ParticleSettings *part)
* - mixed: make copy
*/
if (!ID_IS_LINKED_DATABLOCK(part)) return;
if (part->id.us == 1) {
id_clear_lib_data(bmain, &part->id);
expand_local_particlesettings(part);
if (!ID_IS_LINKED_DATABLOCK(part)) {
return;
}
/* test objects */
for (ob = bmain->object.first; ob && ELEM(false, is_lib, is_local); ob = ob->id.next) {
ParticleSystem *psys = ob->particlesystem.first;
for (; psys; psys = psys->next) {
if (psys->part == part) {
if (ID_IS_LINKED_DATABLOCK(ob)) is_lib = true;
else is_local = true;
}
}
}
BKE_library_ID_test_usages(bmain, part, &is_local, &is_lib);
if (is_local && is_lib == false) {
if (is_local) {
if (!is_lib) {
id_clear_lib_data(bmain, &part->id);
expand_local_particlesettings(part);
}
else if (is_local && is_lib) {
else {
ParticleSettings *part_new = BKE_particlesettings_copy(bmain, part);
part_new->id.us = 0;
/* Remap paths of new ID using old library as base. */
BKE_id_lib_local_paths(bmain, part->id.lib, &part_new->id);
/* do objects */
for (ob = bmain->object.first; ob; ob = ob->id.next) {
ParticleSystem *psys;
for (psys = ob->particlesystem.first; psys; psys = psys->next) {
if (psys->part == part && !ID_IS_LINKED_DATABLOCK(ob)) {
psys->part = part_new;
id_us_plus(&part_new->id);
id_us_min(&part->id);
}
}
BKE_libblock_remap(bmain, part, part_new, ID_REMAP_SKIP_INDIRECT_USAGE);
}
}
}