/*
  Copyright (C) 2010-2011  Charles Pence <charles@charlespence.net>
  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 3 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, see <http://www.gnu.org/licenses/>.
*/
#include "pch.h"
#include "quakedef.h"
#include "filesystem/filesystem.h"
#include "filesystem/file.h"

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;
    
    cls.demofile.Close();
    cls.demoplayback = false;
    cls.state = ca_disconnected;

    if (cls.timedemo)
        CL_FinishTimeDemo ();
}

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

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

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

/*
====================
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
        if (cls.signon == SIGNONS)    // allways grab until fully connected
        {
            if (cls.timedemo)
            {
                if (host_framecount == cls.td_lastframe)
                    return 0;        // allready 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 > 0 && */ cl.time <= cl.mtime[0])
            {
                    return 0;        // don't need another message yet
            }
        }

    // get the next message
        cls.demofile.Read(&net_message.cursize, 4);
        VectorCopy (cl.mviewangles[0], cl.mviewangles[1]);
        for (i=0 ; i<3 ; i++)
        {
            cls.demofile.Read(&f, 4);
            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");
        if (!cls.demofile.Read(net_message.data, net_message.cursize))
        {
            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
    cls.demofile.Close();
    cls.demorecording = false;
    Con_Printf ("Completed demo\n");
}

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

record <demoname> <map> [cd track]
====================
*/
void CL_Record_f (void)
{
    int        c;
    std::string name;
    int        track;

    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 (c == 2 && cls.state == ca_connected)
    {
        Con_Printf("Can not record - already connected to server\nClient demo recording must be started before connecting\n");
        return;
    }

// write the forced cd track number, or -1
    if (c == 4)
    {
        track = Util::lexical_cast<int>(Cmd_Argv(3));
        Con_Printf ("Forcing CD track to %i\n", cls.forcetrack);
    }
    else
        track = -1;
    
    name = Cmd_Argv(1);

//
// start the map up
//
    if (c > 2)
        Cmd_ExecuteString (va("map %s", Cmd_Argv(2).c_str()), src_command);

//
// open the demo file
//
    name = filesystem::DefaultPathExtension(name, ".dem");

    Con_Printf ("recording to %s.\n", name.c_str());
    if (!cls.demofile.OpenWrite(name))
    {
        Con_Printf ("ERROR: couldn't open.\n");
        return;
    }

    cls.forcetrack = track;
    
    cls.demofile.Printf("%i\n", cls.forcetrack);
    cls.demorecording = true;
}


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

play [demoname]
====================
*/
void CL_PlayDemo_f (void)
{
    std::string name;

    if (cmd_source != src_command)
        return;

    if (Cmd_Argc() != 2)
    {
        Con_Printf ("play <demoname> : plays a demo\n");
        return;
    }
    
    if (cls.demofile.open())
        cls.demofile.Close();

//
// disconnect from server
//
    CL_Disconnect ();

//
// open the demo file
//
    name = Cmd_Argv(1);
    name = filesystem::DefaultPathExtension(name, ".dem");

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

    cls.demoplayback = true;
    cls.state = ca_connected;
    cls.forcetrack = 0;

    std::string track = cls.demofile.GetLine();
    cls.forcetrack = Util::lexical_cast<int>(track);
}

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

====================
*/
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 ();

// 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
}

