/* $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; }