/*
 *  Empire - A multi-player, client/server Internet based war game.
 *  Copyright (C) 1986-2006, Dave Pare, Jeff Bailey, Thomas Ruschak,
 *                           Ken Stevens, Steve McClure
 *
 *  This program is free software; you can redistribute it and/or modify
 *  it under the terms of the GNU General Public License as published by
 *  the Free Software Foundation; either version 2 of the License, or
 *  (at your option) any later version.
 *
 *  This program is distributed in the hope that it will be useful,
 *  but WITHOUT ANY WARRANTY; without even the implied warranty of
 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *  GNU General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public License
 *  along with this program; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 *  ---
 *
 *  See files README, COPYING and CREDITS in the root of the source
 *  tree for related information and legal notices.  It is expected
 *  that future projects/authors will amend these files as needed.
 *
 *  ---
 *
 *  powe.c: Do a power report
 * 
 *  Known contributors to this file:
 *     Dave Pare
 *     Ken Stevens, 1995
 *     Steve McClure, 1998-2000
 */

#include <config.h>

#include <string.h>
#include "misc.h"
#include "player.h"
#include "sect.h"
#include "nat.h"
#include "ship.h"
#include "land.h"
#include "power.h"
#include "item.h"
#include "plane.h"
#include "xy.h"
#include "nsc.h"
#include "file.h"
#include "optlist.h"
#include "commands.h"


#include <fcntl.h>

struct powsort {
    float powval;
    natid cnum;
};

static void addtopow(short *vec, struct powstr *pow);
static void gen_power(void);
static void out5(double value, int round_val, int round_flag);
static int powcmp(const void *, const void *);
static int set_target(char *, int *);

int
powe(void)
{
    struct natstr *natp;
    struct natstr *natp2;
    int round_flag;
    time_t pow_time;
    struct nstr_item ni;
    struct powstr pow;
    int num;
    int power_generated = 0;
    int targets[MAXNOC];
    int use_targets = 0;
    int got_targets = 0;
    int no_numbers = 0;
    s_char *p;

    memset(targets, 0, sizeof(targets));
    natp = getnatp(player->cnum);
    num = MAXNOC;
    if (player->argp[1] && player->argp[1][0] == 'n') {
	if (natp->nat_btu < 1)
	    pr("\n  Insufficient BTUs, using the last report.\n\n");
	else {
	    gen_power();
	    power_generated = 1;
	    if (player->argp[2] && (num = atoi(player->argp[2])) < 0) {
		num = MAXNOC;
		if (player->god)
		    no_numbers = 1;
		else
		    return RET_SYN;
	    }
	}
    } else if (player->argp[1] && player->argp[1][0] == 'c') {
	if (!player->argp[2])
	    return RET_SYN;
	if (strchr(player->argp[2], '/')) {
	    p = strtok(player->argp[2], "/");
	    do {
		got_targets |= set_target(p, targets);
	    } while (NULL != (p = strtok(0, "/")));
	} else {
	    got_targets |= set_target(player->argp[2], targets);
	}
	use_targets = 1;
    } else if (player->argp[1] && (num = atoi(player->argp[1])) < 0) {
	num = -(num);
	if (num > MAXNOC || num < 0)
	    num = MAXNOC;
	if (player->god)
	    no_numbers = 1;
	else
	    return RET_SYN;
    }

    if (use_targets && !got_targets)
	return RET_FAIL;

    if (!power_generated) {
	snxtitem_all(&ni, EF_POWER);
	if (!nxtitem(&ni, &pow)) {
	    pr("Power for this game has not been built yet.  Type 'power new' to build it.\n");
	    return RET_FAIL;
	}
    }
    pr("     - = [   Empire Power Report   ] = -\n");
    pow_time = ef_mtime(EF_POWER);
    pr("      as of %s\n         sects  eff civ", ctime(&pow_time));
    pr("  mil  shell gun pet  iron dust oil  pln ship unit money\n");
    snxtitem_all(&ni, EF_POWER);
    while ((nxtitem(&ni, &pow)) && num > 0) {
	if (pow.p_nation == 0 || pow.p_power <= 0.0)
	    continue;
	if (opt_HIDDEN) {
	    if (!player->god && pow.p_nation != player->cnum)
		continue;
	}
	natp2 = getnatp(pow.p_nation);
	if (natp2->nat_stat == STAT_GOD)
	    continue;
	if (use_targets && !targets[pow.p_nation])
	    continue;
	if (pow.p_nation != player->cnum && !player->god)
	    round_flag = 1;
	else
	    round_flag = 0;
	num--;
	pr("%9.9s", cname(pow.p_nation));
	out5(pow.p_sects, 5, round_flag);
	if (pow.p_sects)
	    pr("%4.0f%%", pow.p_effic / pow.p_sects);
	else
	    pr("   0%%");
	out5(pow.p_civil, 50, round_flag);
	out5(pow.p_milit, 50, round_flag);
	out5(pow.p_shell, 25, round_flag);
	out5(pow.p_guns, 5, round_flag);
	out5(pow.p_petrol, 50, round_flag);
	out5(pow.p_iron, 50, round_flag);
	out5(pow.p_dust, 50, round_flag);
	out5(pow.p_oil, 50, round_flag);
	out5(pow.p_planes, 10, round_flag);
	out5(pow.p_ships, 10, round_flag);
	out5(pow.p_units, 10, round_flag);
	out5(pow.p_money, 5000, round_flag);
	pr("\n");
	if (player->god && !no_numbers)
	    pr("%9.2f\n", pow.p_power);
    }
    if (!opt_HIDDEN || player->god) {
	pr("          ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ---- ----\n");
	getpower(0, &pow);
	pr("worldwide");
	out5(pow.p_sects, 5, !player->god);
	pr("%4.0f%%", pow.p_effic / (pow.p_sects + 0.1));
	out5(pow.p_civil, 50, !player->god);
	out5(pow.p_milit, 50, !player->god);
	out5(pow.p_shell, 25, !player->god);
	out5(pow.p_guns, 5, !player->god);
	out5(pow.p_petrol, 50, !player->god);
	out5(pow.p_iron, 50, !player->god);
	out5(pow.p_dust, 50, !player->god);
	out5(pow.p_oil, 50, !player->god);
	out5(pow.p_planes, 10, !player->god);
	out5(pow.p_ships, 10, !player->god);
	out5(pow.p_units, 10, !player->god);
	out5(pow.p_money, 5000, !player->god);
	pr("\n");
    }
    return RET_OK;
}

static void
out5(double value, int round_val, int round_flag)
{
    if (value > round_val && round_flag)
	value = (int)(value / round_val + 0.5) * round_val;
    if (value < -995000.)
	pr("%4.0fM", value / 1e6);
    else if (value < -9950.)
	pr("%4.0fK", value / 1000.);
    else if (value < -999.)
	pr("%4.1fK", value / 1000.);
    else if (value < 1000.)
	pr("%4.0f ", value);
    else if (value < 10000.)
	pr("%4.1fK", value / 1000.);
    else if (value < 1e6)
	pr("%4.0fK", value / 1000.);
    else if (value < 1e7)
	pr("%4.1fM", value / 1e6);
    else if (value >= 1e9)
	pr("%4.0fG", value / 1e9);
    else
	pr("%4.0fM", value / 1e6);
}

static void
gen_power(void)
{
    float *f_ptr;
    float *f_pt2;
    struct powstr *pow;
    int i;
    struct sctstr sect;
    struct plnstr plane;
    struct shpstr ship;
    struct lndstr land;
    struct powstr powbuf[MAXNOC];
    struct nstr_item ni;
    struct nstr_sect ns;
    struct powsort order[MAXNOC];
    struct natstr *natp;
    float f;

    player->btused += 10;
    memset(powbuf, 0, sizeof(powbuf));
    snxtsct_all(&ns);
    while (nxtsct(&ns, &sect)) {
	if (sect.sct_own == 0)
	    continue;
	pow = &powbuf[sect.sct_own];
	pow->p_sects += 1.0;
	pow->p_effic += sect.sct_effic;
	addtopow(sect.sct_item, pow);
    }
    snxtitem_all(&ni, EF_LAND);
    while (nxtitem(&ni, &land)) {
	if (land.lnd_own == 0)
	    continue;
	pow = &powbuf[land.lnd_own];
	addtopow(land.lnd_item, pow);
	f = (lchr[(int)land.lnd_type].l_lcm / 10.0) * (land.lnd_effic / 100.0);
	f += (lchr[(int)land.lnd_type].l_hcm / 10.0) * (land.lnd_effic / 100.0);
	pow->p_power += f * 2;
	if (!(lchr[(int)land.lnd_type].l_flags & L_SPY))
	    pow->p_units += 1.0;
    }
    snxtitem_all(&ni, EF_SHIP);
    while (nxtitem(&ni, &ship)) {
	if (ship.shp_own == 0)
	    continue;
	pow = &powbuf[ship.shp_own];
	addtopow(ship.shp_item, pow);
	f = (mchr[(int)ship.shp_type].m_lcm / 10.0) * (ship.shp_effic / 100.0);
	f += (mchr[(int)ship.shp_type].m_hcm / 10.0) * (ship.shp_effic / 100.0);
	pow->p_power += f * 2;
	pow->p_ships += 1.0;
    }
    snxtitem_all(&ni, EF_PLANE);
    while (nxtitem(&ni, &plane)) {
	if (plane.pln_own == 0)
	    continue;
	pow = &powbuf[plane.pln_own];
	pow->p_planes += 1.0;
	natp = getnatp(plane.pln_own);
	pow->p_power += 20 * (plane.pln_effic / 100.0) *
	    (natp->nat_level[NAT_TLEV] / 500.0);
    }
    for (i = 1; NULL != (natp = getnatp(i)); i++) {
	pow = &powbuf[i];
	pow->p_nation = i;
	if (natp->nat_stat != STAT_ACTIVE) {
	    pow->p_power = 0.;
	    continue;
	}
	pow->p_money = natp->nat_money;
	pow->p_power += pow->p_money / 100.;

	pow->p_power += pow->p_petrol / 500.0;

	pow->p_power += (pow->p_civil + pow->p_milit) / 10.0;
	pow->p_power += pow->p_shell / 12.5;
	pow->p_power += pow->p_iron / 100.0;
	pow->p_power += pow->p_dust / 5 + pow->p_oil / 10 + pow->p_bars;
	pow->p_power += pow->p_guns / 2.5;
	if (pow->p_sects > 0)
	    pow->p_power += (pow->p_sects
			     * ((pow->p_effic / pow->p_sects) / 100.0))
		* 10.0;
	if (natp->nat_level[NAT_TLEV] > 0.0)
	    pow->p_power = pow->p_power *
		(((float)natp->nat_level[NAT_TLEV]) / 500.0);
	else
	    pow->p_power = pow->p_power * (1.0 / 500.0);
	/* ack.  add this vec to the "world power" element */
	f_pt2 = &powbuf[0].p_sects;
	f_ptr = &pow->p_sects;
	while (f_ptr <= &pow->p_power) {
	    *f_pt2 += *f_ptr;
	    f_pt2++;
	    f_ptr++;
	}
    }
    for (i = 1; i < MAXNOC; i++) {
	struct natstr *np;
	int maxpop;

	if (opt_RES_POP) {
	    np = getnatp(i);
	    maxpop = max_population(np->nat_level[NAT_RLEV], SCT_MINE, 0);
	    powbuf[i].p_power *= 1.0 + maxpop / 10000.0;
	}
	order[i].powval = powbuf[i].p_power;
	order[i].cnum = i;
    }
    qsort(&order[1], MAXNOC - 1, sizeof(*order), powcmp);
    putpower(0, &powbuf[0]);
    for (i = 1; i < MAXNOC; i++) {
	putpower(i, &powbuf[order[i].cnum]);
    }
#ifdef _WIN32
    /*
     * At least some versions of Windows fail to update mtime on
     * write().  Bad, because `power' displays that time.  Attempt to
     * force an update.
     */
    _commit(empfile[EF_POWER].fd);
#endif
}

static int
powcmp(const void *a, const void *b)
{
    const struct powsort *p1 = a;
    const struct powsort *p2 = b;

    if (p1->powval > p2->powval)
	return -1;
    if (p1->powval < p2->powval)
	return 1;
    return 0;
}

static void
addtopow(short *vec, struct powstr *pow)
{
    pow->p_civil += vec[I_CIVIL];
    pow->p_milit += vec[I_MILIT];
    pow->p_shell += vec[I_SHELL];
    pow->p_guns += vec[I_GUN];
    pow->p_petrol += vec[I_PETROL];
    pow->p_iron += vec[I_IRON];
    pow->p_dust += vec[I_DUST];
    pow->p_food += vec[I_FOOD];
    pow->p_oil += vec[I_OIL];
    pow->p_bars += vec[I_BAR];
    pow->p_power += vec[I_LCM] / 10.0;
    pow->p_power += vec[I_HCM] / 5.0;
}

static int
set_target(char *p, int *targets)
{
    int target;
    struct natstr *natp;

    if (!p)
	return 0;
    target = natarg(p, NULL);
    if (target < 0)
	return 0;
    natp = getnatp(target);
    if (natp->nat_stat != STAT_ACTIVE) {
	pr("Country '%s' is not a normal country\n", p);
	return 0;
    }

    targets[target] = 1;
    return 1;
}