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