refactoring: use add_ally instead of manually crating the structure all over the code.

This commit is contained in:
Enno Rehling 2014-10-31 15:13:05 +01:00
parent c8d5d52412
commit f27a77d288
6 changed files with 143 additions and 151 deletions

View File

@ -430,10 +430,8 @@ void set_alliance(faction * a, faction * b, int status)
sfp = &sf->next;
}
if (*sfp == NULL) {
ally *sf = *sfp = malloc(sizeof(ally));
sf->next = NULL;
ally *sf = ally_add(sfp, b);
sf->status = status;
sf->faction = b;
return;
}
(*sfp)->status |= status;

View File

@ -1,7 +1,7 @@
/*
Copyright (c) 1998-2010, Enno Rehling <enno@eressea.de>
Katja Zedel <katze@felidae.kn-bremen.de
Christian Schlittchen <corwin@amber.kn-bremen.de>
Katja Zedel <katze@felidae.kn-bremen.de
Christian Schlittchen <corwin@amber.kn-bremen.de>
Permission to use, copy, modify, and/or distribute this software for any
purpose with or without fee is hereby granted, provided that the above
@ -50,203 +50,200 @@ static int maxgid;
static group *new_group(faction * f, const char *name, int gid)
{
group **gp = &f->groups;
int index = gid % GMAXHASH;
group *g = calloc(sizeof(group), 1);
group **gp = &f->groups;
int index = gid % GMAXHASH;
group *g = calloc(sizeof(group), 1);
while (*gp)
gp = &(*gp)->next;
*gp = g;
while (*gp)
gp = &(*gp)->next;
*gp = g;
maxgid = _max(gid, maxgid);
g->name = _strdup(name);
g->gid = gid;
maxgid = _max(gid, maxgid);
g->name = _strdup(name);
g->gid = gid;
g->nexthash = ghash[index];
return ghash[index] = g;
g->nexthash = ghash[index];
return ghash[index] = g;
}
static void init_group(faction * f, group * g)
{
ally *a, **an;
ally *a, **an;
an = &g->allies;
for (a = f->allies; a; a = a->next)
if (a->faction) {
ally *ga = calloc(sizeof(ally), 1);
*ga = *a;
*an = ga;
an = &ga->next;
}
an = &g->allies;
for (a = f->allies; a; a = a->next)
if (a->faction) {
ally *ga = ally_add(an, a->faction);
ga->status = a->status;
an = &ga->next;
}
}
static group *find_groupbyname(group * g, const char *name)
{
while (g && unicode_utf8_strcasecmp(name, g->name) != 0)
g = g->next;
return g;
while (g && unicode_utf8_strcasecmp(name, g->name) != 0)
g = g->next;
return g;
}
static group *find_group(int gid)
{
int index = gid % GMAXHASH;
group *g = ghash[index];
while (g && g->gid != gid)
g = g->nexthash;
return g;
int index = gid % GMAXHASH;
group *g = ghash[index];
while (g && g->gid != gid)
g = g->nexthash;
return g;
}
static int read_group(attrib * a, void *owner, struct storage *store)
{
group *g;
int gid;
group *g;
int gid;
READ_INT(store, &gid);
a->data.v = g = find_group(gid);
if (g != 0) {
g->members++;
return AT_READ_OK;
}
return AT_READ_FAIL;
READ_INT(store, &gid);
a->data.v = g = find_group(gid);
if (g != 0) {
g->members++;
return AT_READ_OK;
}
return AT_READ_FAIL;
}
static void
write_group(const attrib * a, const void *owner, struct storage *store)
{
group *g = (group *) a->data.v;
WRITE_INT(store, g->gid);
group *g = (group *)a->data.v;
WRITE_INT(store, g->gid);
}
attrib_type at_group = { /* attribute for units assigned to a group */
"grp",
"grp",
DEFAULT_INIT,
DEFAULT_FINALIZE, DEFAULT_AGE, write_group, read_group, ATF_UNIQUE};
DEFAULT_FINALIZE, DEFAULT_AGE, write_group, read_group, ATF_UNIQUE };
void free_group(group * g)
{
int index = g->gid % GMAXHASH;
group **g_ptr = ghash + index;
while (*g_ptr && (*g_ptr)->gid != g->gid)
g_ptr = &(*g_ptr)->nexthash;
assert(*g_ptr == g);
*g_ptr = g->nexthash;
int index = g->gid % GMAXHASH;
group **g_ptr = ghash + index;
while (*g_ptr && (*g_ptr)->gid != g->gid)
g_ptr = &(*g_ptr)->nexthash;
assert(*g_ptr == g);
*g_ptr = g->nexthash;
while (g->allies) {
ally *a = g->allies;
g->allies = a->next;
free(a);
}
free(g->name);
free(g);
while (g->allies) {
ally *a = g->allies;
g->allies = a->next;
free(a);
}
free(g->name);
free(g);
}
group * get_group(const struct unit *u)
{
if (fval(u, UFL_GROUP)) {
attrib * a = a_find(u->attribs, &at_group);
if (a) {
return (group *) a->data.v;
if (fval(u, UFL_GROUP)) {
attrib * a = a_find(u->attribs, &at_group);
if (a) {
return (group *)a->data.v;
}
}
}
return 0;
return 0;
}
void set_group(struct unit *u, struct group *g)
{
attrib *a = NULL;
attrib *a = NULL;
if (fval(u, UFL_GROUP)) {
a = a_find(u->attribs, &at_group);
}
if (a) {
group *og = (group *) a->data.v;
if (og == g)
return;
--og->members;
}
if (g) {
if (!a) {
a = a_add(&u->attribs, a_new(&at_group));
fset(u, UFL_GROUP);
if (fval(u, UFL_GROUP)) {
a = a_find(u->attribs, &at_group);
}
if (a) {
group *og = (group *)a->data.v;
if (og == g)
return;
--og->members;
}
if (g) {
if (!a) {
a = a_add(&u->attribs, a_new(&at_group));
fset(u, UFL_GROUP);
}
a->data.v = g;
g->members++;
}
else if (a) {
a_remove(&u->attribs, a);
freset(u, UFL_GROUP);
}
a->data.v = g;
g->members++;
} else if (a) {
a_remove(&u->attribs, a);
freset(u, UFL_GROUP);
}
}
bool join_group(unit * u, const char *name)
{
group *g = NULL;
group *g = NULL;
if (name && name[0]) {
g = find_groupbyname(u->faction->groups, name);
if (g == NULL) {
g = new_group(u->faction, name, ++maxgid);
init_group(u->faction, g);
if (name && name[0]) {
g = find_groupbyname(u->faction->groups, name);
if (g == NULL) {
g = new_group(u->faction, name, ++maxgid);
init_group(u->faction, g);
}
}
}
set_group(u, g);
return true;
set_group(u, g);
return true;
}
void write_groups(struct storage *store, group * g)
{
while (g) {
ally *a;
WRITE_INT(store, g->gid);
WRITE_STR(store, g->name);
for (a = g->allies; a; a = a->next) {
if (a->faction) {
write_faction_reference(a->faction, store);
WRITE_INT(store, a->status);
}
while (g) {
ally *a;
WRITE_INT(store, g->gid);
WRITE_STR(store, g->name);
for (a = g->allies; a; a = a->next) {
if (a->faction) {
write_faction_reference(a->faction, store);
WRITE_INT(store, a->status);
}
}
WRITE_INT(store, 0);
a_write(store, g->attribs, g);
WRITE_SECTION(store);
g = g->next;
}
WRITE_INT(store, 0);
a_write(store, g->attribs, g);
WRITE_SECTION(store);
g = g->next;
}
WRITE_INT(store, 0);
}
void read_groups(struct storage *store, faction * f)
{
for (;;) {
ally **pa;
group *g;
int gid;
char buf[1024];
READ_INT(store, &gid);
if (gid == 0)
break;
READ_STR(store, buf, sizeof(buf));
g = new_group(f, buf, gid);
pa = &g->allies;
for (;;) {
ally *a;
variant fid;
READ_INT(store, &fid.i);
if (fid.i <= 0)
break;
if (global.data_version < STORAGE_VERSION && fid.i == 0)
break;
a = malloc(sizeof(ally));
*pa = a;
pa = &a->next;
READ_INT(store, &a->status);
ally **pa;
group *g;
int gid;
char buf[1024];
a->faction = findfaction(fid.i);
if (!a->faction)
ur_add(fid, &a->faction, resolve_faction);
READ_INT(store, &gid);
if (gid == 0)
break;
READ_STR(store, buf, sizeof(buf));
g = new_group(f, buf, gid);
pa = &g->allies;
for (;;) {
ally *a;
variant fid;
READ_INT(store, &fid.i);
if (fid.i <= 0)
break;
if (global.data_version < STORAGE_VERSION && fid.i == 0)
break;
a = ally_add(pa, findfaction(fid.i));
READ_INT(store, &a->status);
if (!a->faction)
ur_add(fid, &a->faction, resolve_faction);
}
*pa = 0;
a_read(store, &g->attribs, g);
}
*pa = 0;
a_read(store, &g->attribs, g);
}
}

View File

@ -19,6 +19,7 @@ void test_pool(CuTest *tc) {
faction *f;
region *r;
struct resource_type *rtype;
ally *al;
test_cleanup();
test_create_world();
@ -45,11 +46,10 @@ void test_pool(CuTest *tc) {
CuAssertIntEquals(tc, 100, get_pooled(u1, rtype, GET_POOLED_SLACK, INT_MAX));
CuAssertIntEquals(tc, 200, get_pooled(u1, rtype, GET_POOLED_SLACK | GET_POOLED_RESERVE, INT_MAX));
u3->faction->allies = calloc(1, sizeof(ally));
u3->faction->allies->faction = f;
u3->faction->allies->status = HELP_GUARD;
al = ally_add(&u3->faction->allies, f);
al->status = HELP_GUARD;
CuAssertIntEquals(tc, 0, get_pooled(u1, rtype, GET_ALLIED_SLACK | GET_ALLIED_RESERVE, INT_MAX));
u3->faction->allies->status = HELP_MONEY;
al->status = HELP_MONEY;
CuAssertIntEquals(tc, 200, get_pooled(u1, rtype, GET_ALLIED_SLACK, INT_MAX));
CuAssertIntEquals(tc, 200, get_pooled(u1, rtype, GET_ALLIED_RESERVE, INT_MAX));
CuAssertIntEquals(tc, 400, get_pooled(u1, rtype, GET_ALLIED_SLACK | GET_ALLIED_RESERVE, INT_MAX));

View File

@ -1119,8 +1119,7 @@ static ally **addally(const faction * f, ally ** sfp, int aid, int state)
if (state == 0)
return sfp;
sf = calloc(1, sizeof(ally));
sf->faction = af;
sf = ally_add(sfp, af);
if (!sf->faction) {
variant id;
id.i = aid;

View File

@ -1348,10 +1348,8 @@ int ally_cmd(unit * u, struct order *ord)
return 0;
}
else {
sf = calloc(1, sizeof(ally));
sf->faction = f;
sf = ally_add(sfp, f);
sf->status = 0;
addlist(sfp, sf);
}
}
switch (keyword) {

View File

@ -93,6 +93,7 @@ static void test_contact(CuTest * tc)
unit *u1, *u2, *u3;
building *b;
building_type *btype;
ally *al;
test_cleanup();
test_create_world();
@ -110,9 +111,8 @@ static void test_contact(CuTest * tc)
u_set_building(u1, b);
CuAssertIntEquals(tc, 0, can_contact(r, u1, u2));
u1->faction->allies = calloc(1, sizeof(ally));
u1->faction->allies->faction = u2->faction;
u1->faction->allies->status = HELP_ALL;
al = ally_add(&u1->faction->allies, u2->faction);
al->status = HELP_ALL;
CuAssertIntEquals(tc, HELP_GIVE, can_contact(r, u1, u2));
u_set_building(u2, b);
CuAssertIntEquals(tc, 1, can_contact(r, u1, u2));