mirror of
https://github.com/JGRennison/OpenTTD-patches.git
synced 2024-11-17 21:25:40 +00:00
132 lines
3.3 KiB
C++
132 lines
3.3 KiB
C++
/* $Id$ */
|
|
|
|
/** @file newgrf_townname.cpp
|
|
* Implementation of Action 0F "universal holder" structure and functions.
|
|
* This file implements a linked-lists of townname generators,
|
|
* holding everything that the newgrf action 0F will send over to OpenTTD.
|
|
*/
|
|
|
|
#include "stdafx.h"
|
|
#include "newgrf_townname.h"
|
|
#include "core/alloc_func.hpp"
|
|
#include "string_func.h"
|
|
|
|
static GRFTownName *_grf_townnames = NULL;
|
|
|
|
GRFTownName *GetGRFTownName(uint32 grfid)
|
|
{
|
|
GRFTownName *t = _grf_townnames;
|
|
for (; t != NULL; t = t->next) {
|
|
if (t->grfid == grfid) return t;
|
|
}
|
|
return NULL;
|
|
}
|
|
|
|
GRFTownName *AddGRFTownName(uint32 grfid)
|
|
{
|
|
GRFTownName *t = GetGRFTownName(grfid);
|
|
if (t == NULL) {
|
|
t = CallocT<GRFTownName>(1);
|
|
t->grfid = grfid;
|
|
t->next = _grf_townnames;
|
|
_grf_townnames = t;
|
|
}
|
|
return t;
|
|
}
|
|
|
|
void DelGRFTownName(uint32 grfid)
|
|
{
|
|
GRFTownName *t = _grf_townnames;
|
|
GRFTownName *p = NULL;
|
|
for (;t != NULL; p = t, t = t->next) if (t->grfid == grfid) break;
|
|
if (t != NULL) {
|
|
for (int i = 0; i < 128; i++) {
|
|
for (int j = 0; j < t->nbparts[i]; j++) {
|
|
for (int k = 0; k < t->partlist[i][j].partcount; k++) {
|
|
if (!HasBit(t->partlist[i][j].parts[k].prob, 7)) free(t->partlist[i][j].parts[k].data.text);
|
|
}
|
|
free(t->partlist[i][j].parts);
|
|
}
|
|
free(t->partlist[i]);
|
|
}
|
|
if (p != NULL) {
|
|
p->next = t->next;
|
|
} else {
|
|
_grf_townnames = t->next;
|
|
}
|
|
free(t);
|
|
}
|
|
}
|
|
|
|
static char *RandomPart(char *buf, GRFTownName *t, uint32 seed, byte id, const char *last)
|
|
{
|
|
assert(t != NULL);
|
|
for (int i = 0; i < t->nbparts[id]; i++) {
|
|
byte count = t->partlist[id][i].bitcount;
|
|
uint16 maxprob = t->partlist[id][i].maxprob;
|
|
uint32 r = (GB(seed, t->partlist[id][i].bitstart, count) * maxprob) >> count;
|
|
for (int j = 0; j < t->partlist[id][i].partcount; j++) {
|
|
byte prob = t->partlist[id][i].parts[j].prob;
|
|
maxprob -= GB(prob, 0, 7);
|
|
if (maxprob > r) continue;
|
|
if (HasBit(prob, 7)) {
|
|
buf = RandomPart(buf, t, seed, t->partlist[id][i].parts[j].data.id, last);
|
|
} else {
|
|
buf = strecat(buf, t->partlist[id][i].parts[j].data.text, last);
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
return buf;
|
|
}
|
|
|
|
char *GRFTownNameGenerate(char *buf, uint32 grfid, uint16 gen, uint32 seed, const char *last)
|
|
{
|
|
strecpy(buf, "", last);
|
|
for (GRFTownName *t = _grf_townnames; t != NULL; t = t->next) {
|
|
if (t->grfid == grfid) {
|
|
assert(gen < t->nb_gen);
|
|
buf = RandomPart(buf, t, seed, t->id[gen], last);
|
|
break;
|
|
}
|
|
}
|
|
return buf;
|
|
}
|
|
|
|
StringID *GetGRFTownNameList()
|
|
{
|
|
int nb_names = 0, n = 0;
|
|
for (GRFTownName *t = _grf_townnames; t != NULL; t = t->next) nb_names += t->nb_gen;
|
|
StringID *list = MallocT<StringID>(nb_names + 1);
|
|
for (GRFTownName *t = _grf_townnames; t != NULL; t = t->next) {
|
|
for (int j = 0; j < t->nb_gen; j++) list[n++] = t->name[j];
|
|
}
|
|
list[n] = INVALID_STRING_ID;
|
|
return list;
|
|
}
|
|
|
|
void CleanUpGRFTownNames()
|
|
{
|
|
while (_grf_townnames != NULL) DelGRFTownName(_grf_townnames->grfid);
|
|
}
|
|
|
|
uint32 GetGRFTownNameId(int gen)
|
|
{
|
|
for (GRFTownName *t = _grf_townnames; t != NULL; t = t->next) {
|
|
if (gen < t->nb_gen) return t->grfid;
|
|
gen -= t->nb_gen;
|
|
}
|
|
/* Fallback to no NewGRF */
|
|
return 0;
|
|
}
|
|
|
|
uint16 GetGRFTownNameType(int gen)
|
|
{
|
|
for (GRFTownName *t = _grf_townnames; t != NULL; t = t->next) {
|
|
if (gen < t->nb_gen) return gen;
|
|
gen -= t->nb_gen;
|
|
}
|
|
/* Fallback to english original */
|
|
return SPECSTR_TOWNNAME_ENGLISH;
|
|
}
|