/*
Copyright (C) 1996-1997 Id Software, Inc.

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., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

*/

#include <ctype.h>
#include <stdio.h>

#include "client.h"
#include "cmd.h"
#include "console.h"
#include "host.h"
#include "net.h"
#include "protocol.h"
#include "quakedef.h"
#include "sys.h"
#include "zone.h"

static void CL_FinishTimeDemo(void);

/*
==============================================================================

DEMO CODE

When a demo is playing back, all NET_SendMessages are skipped, and
NET_GetMessages are read from the demo file.

Whenever cl.time gets past the last received message, another message is
read from the demo file.
==============================================================================
*/

/*
==============
CL_StopPlayback

Called when a demo file runs out, or the user starts a game
==============
*/
void
CL_StopPlayback(void)
{
    if (!cls.demoplayback)
	return;

    fclose(cls.demofile);
    cls.demoplayback = false;
    cls.demofile = NULL;
    cls.state = ca_disconnected;

    if (cls.timedemo)
	CL_FinishTimeDemo();
}

/*
====================
CL_WriteDemoMessage

Dumps the current net message, prefixed by the length and view angles
====================
*/
static void
CL_WriteDemoMessage(void)
{
    int len;
    int i;
    float f;

    len = LittleLong(net_message.cursize);
    fwrite(&len, 4, 1, cls.demofile);
    for (i = 0; i < 3; i++) {
	f = LittleFloat(cl.viewangles[i]);
	fwrite(&f, 4, 1, cls.demofile);
    }
    fwrite(net_message.data, net_message.cursize, 1, cls.demofile);
    fflush(cls.demofile);
}

/*
====================
CL_GetMessage

Handles recording and playback of demos, on top of NET_ code
====================
*/
int
CL_GetMessage(void)
{
    int r, i;
    float f;

    if (cls.demoplayback) {
	// decide if it is time to grab the next message
	// always grab until fully connected
	if (cls.state == ca_active) {
	    if (cls.timedemo) {
		if (host_framecount == cls.td_lastframe)
		    return 0;	// already read this frame's message

		cls.td_lastframe = host_framecount;

		// if this is the second frame, grab the real td_starttime
		// so the bogus time on the first frame doesn't count
		if (host_framecount == cls.td_startframe + 1)
		    cls.td_starttime = realtime;
	    } else if (cl.time <= cl.mtime[0]) {
		// don't need another message yet
		return 0;
	    }
	}
	// get the next message
	fread(&net_message.cursize, 4, 1, cls.demofile);
	VectorCopy(cl.mviewangles[0], cl.mviewangles[1]);
	for (i = 0; i < 3; i++) {
	    fread(&f, 4, 1, cls.demofile);
	    cl.mviewangles[0][i] = LittleFloat(f);
	}

	net_message.cursize = LittleLong(net_message.cursize);
	if (net_message.cursize > MAX_MSGLEN)
	    Sys_Error("Demo message > MAX_MSGLEN");
	r = fread(net_message.data, net_message.cursize, 1, cls.demofile);
	if (r != 1) {
	    CL_StopPlayback();
	    return 0;
	}

	return 1;
    }

    while (1) {
	r = NET_GetMessage(cls.netcon);

	if (r != 1 && r != 2)
	    return r;

	// discard nop keepalive message
	if (net_message.cursize == 1 && net_message.data[0] == svc_nop)
	    Con_Printf("<-- server to client keepalive\n");
	else
	    break;
    }

    if (cls.demorecording)
	CL_WriteDemoMessage();

    return r;
}


/*
====================
CL_Stop_f

stop recording a demo
====================
*/
void
CL_Stop_f(void)
{
    if (cmd_source != src_command)
	return;

    if (!cls.demorecording) {
	Con_Printf("Not recording a demo.\n");
	return;
    }
// write a disconnect message to the demo file
    SZ_Clear(&net_message);
    MSG_WriteByte(&net_message, svc_disconnect);
    CL_WriteDemoMessage();

// finish up
    fclose(cls.demofile);
    cls.demofile = NULL;
    cls.demorecording = false;
    Con_Printf("Completed demo\n");
}

/*
====================
CL_Record_f

record <demoname> <map> [cd track]
====================
*/
void
CL_Record_f(void)
{
    int c, track, length, err;
    char name[MAX_OSPATH];

    if (cmd_source != src_command)
	return;

    c = Cmd_Argc();
    if (c != 2 && c != 3 && c != 4) {
	Con_Printf("record <demoname> [<map> [cd track]]\n");
	return;
    }

    if (strstr(Cmd_Argv(1), "..")) {
	Con_Printf("Relative pathnames are not allowed.\n");
	return;
    }

    if (c == 2 && cls.state >= ca_connected) {
	Con_Printf("Can not record - already connected to server\n"
		   "Client demo recording must be started before"
		   " connecting\n");
	return;
    }
// write the forced cd track number, or -1
    if (c == 4) {
	track = atoi(Cmd_Argv(3));
	Con_Printf("Forcing CD track to %i\n", cls.forcetrack);
    } else
	track = -1;

    /* Grab the filename before our cmd args disappear */
    length = qsnprintf(name, sizeof(name), "%s/%s", com_gamedir, Cmd_Argv(1));
    err = COM_DefaultExtension(name, ".dem", name, sizeof(name));
    if (length >= sizeof(name) || err) {
	Con_Printf("Error: demo path name too long.\n");
	return;
    }

    /* start the map up */
    if (c > 2) {
	Cmd_ExecuteString(va("map %s", Cmd_Argv(2)), src_command);
	if (cls.state != ca_connected)
	    return;
    }

    /* open the demo file */
    Con_Printf("recording to %s.\n", name);
    cls.demofile = fopen(name, "wb");
    if (!cls.demofile) {
	Con_Printf("ERROR: couldn't open.\n");
	return;
    }

    cls.forcetrack = track;
    fprintf(cls.demofile, "%i\n", cls.forcetrack);

    cls.demorecording = true;
}

/*
====================
CL_PlayDemo_f

play [demoname]
====================
*/
void
CL_PlayDemo_f(void)
{
    char name[MAX_QPATH], forcetrack[12];
    int i, c, err;

    if (cmd_source != src_command)
	return;

    if (Cmd_Argc() != 2) {
	Con_Printf("play <demoname> : plays a demo\n");
	return;
    }
//
// disconnect from server
//
    CL_Disconnect();

//
// open the demo file
//
    err = COM_DefaultExtension(Cmd_Argv(1), ".dem", name, sizeof(name));
    if (err) {
	Con_Printf("Error: demo filename too long\n");
	cls.demonum = -1;	/* stop demo loop */
	return;
    }

    Con_Printf("Playing demo from %s.\n", name);
    COM_FOpenFile(name, &cls.demofile);
    if (!cls.demofile) {
	Con_Printf("ERROR: couldn't open.\n");
	cls.demonum = -1;	/* stop demo loop */
	return;
    }

    /*
     * Parse an integer for the demo CD track.  Pull the string out
     * manually because fscanf can chew up a space following '\n'
     * which can mess up the rest of the demo playback.  If we didn't
     * find a newline within the first 12 characters, assume the demo
     * file is invalid.
     */
    cls.forcetrack = 0;
    for (i = 0; i < 12; i++) {
	c = getc(cls.demofile);
	forcetrack[i] = c & 127;
	if (c == '\n')
	    break;
    }
    if (c != '\n') {
	Con_Printf("ERROR: demo \"%s\" is invalid\n", name);
	goto error_out;
    }

    /* Terminate before the '\n' and parse the track number */
    forcetrack[i] = 0;
    i = sscanf(forcetrack, "%d", &cls.forcetrack);
    if (i != 1) {
	Con_Printf("Error: invalid cd track '%s' in demo %s\n",
		   forcetrack, name);
	goto error_out;
    }

    /* Continue with demo playback */
    cls.demoplayback = true;
    cls.state = ca_connected;
    return;

 error_out:
    fclose(cls.demofile);
    cls.demofile = NULL;
    cls.demonum = -1;
}

void
CL_Demo_Arg_f(struct stree_root *root, int argnum)
{
    if (argnum != 1)
        return;

    COM_ScanDir(root, "", Cmd_Argv(1), ".dem", 0);
}

/*
====================
CL_FinishTimeDemo

====================
*/
static void
CL_FinishTimeDemo(void)
{
    int frames;
    float time;

    cls.timedemo = false;

// the first frame didn't count
    frames = (host_framecount - cls.td_startframe) - 1;
    time = realtime - cls.td_starttime;
    if (!time)
	time = 1;
    Con_Printf("%i frames %5.1f seconds %5.1f fps\n", frames, time,
	       frames / time);
}

/*
====================
CL_TimeDemo_f

timedemo [demoname]
====================
*/
void
CL_TimeDemo_f(void)
{
    if (cmd_source != src_command)
	return;

    if (Cmd_Argc() != 2) {
	Con_Printf("timedemo <demoname> : gets demo speeds\n");
	return;
    }

    CL_PlayDemo_f();
    if (!cls.demofile)
	return;

// cls.td_starttime will be grabbed at the second frame of the demo, so
// all the loading time doesn't get counted

    cls.timedemo = true;
    cls.td_startframe = host_framecount;
    cls.td_lastframe = -1;	// get a new message this frame
}
