/**************************************************************************
 *
 * Copyright 2011 Jose Fonseca
 * All Rights Reserved.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 * THE SOFTWARE.
 *
 **************************************************************************/


#include <string.h>
#include <iostream>
#include <fstream>

#include "os_string.hpp"
#include "image.hpp"
#include "retrace.hpp"
#include "glproc.hpp"
#include "glstate.hpp"
#include "glretrace.hpp"

#include "intel_stats.h"


namespace glretrace {

bool double_buffer = true;
unsigned long long profile = 0;
char *output = NULL;
unsigned long long profile_start = 0;
unsigned long long profile_end = 0;
bool insideGlBeginEnd = false;
trace::Parser parser;
glws::Profile defaultProfile = glws::PROFILE_COMPAT;
glws::Visual *visual = NULL;
glws::Drawable *drawable = NULL;
glws::Context *context = NULL;

unsigned frame = 0;
long long startTime = 0;
bool wait = false;

bool benchmark = false;
const char *compare_prefix = NULL;
const char *snapshot_prefix = NULL;
enum frequency snapshot_frequency = FREQUENCY_NEVER;

unsigned dump_state = ~0;

void
checkGlError(trace::Call &call) {
    GLenum error = glGetError();
    if (error == GL_NO_ERROR) {
        return;
    }

    std::ostream & os = retrace::warning(call);

    os << "glGetError(";
    os << call.name();
    os << ") = ";

    switch (error) {
    case GL_INVALID_ENUM:
        os << "GL_INVALID_ENUM";
        break;
    case GL_INVALID_VALUE:
        os << "GL_INVALID_VALUE";
        break;
    case GL_INVALID_OPERATION:
        os << "GL_INVALID_OPERATION";
        break;
    case GL_STACK_OVERFLOW:
        os << "GL_STACK_OVERFLOW";
        break;
    case GL_STACK_UNDERFLOW:
        os << "GL_STACK_UNDERFLOW";
        break;
    case GL_OUT_OF_MEMORY:
        os << "GL_OUT_OF_MEMORY";
        break;
    case GL_INVALID_FRAMEBUFFER_OPERATION:
        os << "GL_INVALID_FRAMEBUFFER_OPERATION";
        break;
    case GL_TABLE_TOO_LARGE:
        os << "GL_TABLE_TOO_LARGE";
        break;
    default:
        os << error;
        break;
    }
    os << "\n";
}

/**
 * Grow the current drawble.
 *
 * We need to infer the drawable size from GL calls because the drawable sizes
 * are specified by OS specific calls which we do not trace.
 */
void
updateDrawable(int width, int height) {
    if (!drawable) {
        return;
    }

    if (drawable->visible &&
        width  <= drawable->width &&
        height <= drawable->height) {
        return;
    }

    // Check for bound framebuffer last, as this may have a performance impact.
    GLint draw_framebuffer = 0;
    glGetIntegerv(GL_DRAW_FRAMEBUFFER_BINDING, &draw_framebuffer);
    if (draw_framebuffer != 0) {
        return;
    }

    drawable->resize(width, height);
    drawable->show();

    glScissor(0, 0, width, height);
}


void snapshot(unsigned call_no) {
    if (!drawable ||
        (!snapshot_prefix && !compare_prefix)) {
        return;
    }

    image::Image *ref = NULL;

    if (compare_prefix) {
        os::String filename = os::String::format("%s%010u.png", compare_prefix, call_no);
        ref = image::readPNG(filename);
        if (!ref) {
            return;
        }
        if (retrace::verbosity >= 0) {
            std::cout << "Read " << filename << "\n";
        }
    }

    image::Image *src = glstate::getDrawBufferImage(GL_RGBA);
    if (!src) {
        return;
    }

    if (snapshot_prefix) {
        if (snapshot_prefix[0] == '-' && snapshot_prefix[1] == 0) {
            char comment[21];
            snprintf(comment, sizeof comment, "%u", call_no);
            src->writePNM(std::cout, comment);
        } else {
            os::String filename = os::String::format("%s%010u.png", snapshot_prefix, call_no);
            if (src->writePNG(filename) && retrace::verbosity >= 0) {
                std::cout << "Wrote " << filename << "\n";
            }
        }
    }

    if (ref) {
        std::cout << "Snapshot " << call_no << " average precision of " << src->compare(*ref) << " bits\n";
        delete ref;
    }

    delete src;
}


void frame_complete(trace::Call &call) {
    ++frame;

    if (!drawable) {
        return;
    }

    if (!drawable->visible) {
        retrace::warning(call) << "could not infer drawable size (glViewport never called)\n";
    }

    if (snapshot_frequency == FREQUENCY_FRAME ||
        snapshot_frequency == FREQUENCY_FRAMEBUFFER) {
        snapshot(call.no);
    }
}


static void display(void) {
    int call_count = 0;
    retrace::Retracer retracer;

    retracer.addCallbacks(gl_callbacks);
    retracer.addCallbacks(glx_callbacks);
    retracer.addCallbacks(wgl_callbacks);
    retracer.addCallbacks(cgl_callbacks);
    retracer.addCallbacks(egl_callbacks);

    startTime = os::getTime();
    trace::Call *call;

    std::ofstream output_s;
    if (output) {
        output_s.open(output, std::ios::out);
        if (!output_s) {
            std::cerr << "Unable to open output file " << output;
            exit(1);
        }
        std::cout << "Will output to " << output << std::endl;
    }

    std::cout << "Starting at " << profile_start << std::endl;

    while ((call = parser.parse_call())) {
        retracer.retrace(*call);
        if (profile) {
            if (call->no == profile_start) {
                m_reset();
                std::cout << "Profiling: starting at frame " << call->no << std::endl;
            }
            if (profile_end > 0 && call->no == profile_end) {
                std::cout << "Profiling: finished at frame " << call->no << std::endl;
            }
            /* Do we have a starting frame */
            if ( (profile_start == 0 || call->no >= profile_start) && (profile_end == 0 || call->no <= profile_end)) {
                if (call_count == profile || (profile_end > 0 && call->no == profile_end)) {
                    Stats stat = m_count();
                    unsigned long long elapsed = stat.time.tv_sec * 1000000 + stat.time.tv_usec;
                    unsigned long long user = stat.user.tv_sec * 1000000 + stat.user.tv_usec;
                    float userp = user * 100. / elapsed;
                    float gpup = (stat.gpu)?100 - stat.gpuidle * 100. / stat.gpu:0;

                    ((output_s) ? output_s: std::cout) \
                             << "Call-No: " << call->no \
                             << "\tTime(us): " << elapsed \
                             << "\tGPU-%: " << gpup \
                             << "\tCPU-%: " << userp \
                             << "\tCPU-time(us): " << user \
                             << std::endl;

                    output_s.flush();

                    m_reset();

                    call_count = 0;
                }
                call_count++;
            }
        }

        if (!insideGlBeginEnd &&
            drawable && context &&
            call->no >= dump_state) {
            glstate::dumpCurrentContext(std::cout);
            exit(0);
        }

        delete call;
    }

    // Reached the end of trace
    glFlush();

    long long endTime = os::getTime();
    float timeInterval = (endTime - startTime) * 1.0E-6;

    if (retrace::verbosity >= -1) { 
        std::cout << 
            "Rendered " << frame << " frames"
            " in " <<  timeInterval << " secs,"
            " average of " << (frame/timeInterval) << " fps\n";
    }

    if (wait) {
        while (glws::processEvents()) {}
    } else {
        exit(0);
    }
}


static void usage(void) {
    std::cout << 
        "Usage: glretrace [OPTION] TRACE\n"
        "Replay TRACE.\n"
        "\n"
        "  -b           benchmark mode (no error checking or warning messages)\n"
        "  -c PREFIX    compare against snapshots\n"
        "  -core        use core profile\n"
        "  -db          use a double buffer visual (default)\n"
        "  -sb          use a single buffer visual\n"
        "  -s PREFIX    take snapshots; `-` for PNM stdout output\n"
        "  -S FREQUENCY snapshot frequency: frame (default), framebuffer, or draw\n"
        "  -v           verbose output\n"
        "  -D CALLNO    dump state at specific call no\n"
        "  -w           wait on final frame\n"
        "  -o FILE      save profile results to FILE\n"
        "  -p FRAMES    sample GPU/CPU usage once each FRAMES\n"
        "  -ps FRAME    start profiling on frame FRAME\n"
        "  -pe FRAME    end profile on frame FRAME\n";
}

extern "C"
int main(int argc, char **argv)
{

    int i;
    for (i = 1; i < argc; ++i) {
        const char *arg = argv[i];

        if (arg[0] != '-') {
            break;
        }

        if (!strcmp(arg, "--")) {
            break;
        } else if (!strcmp(arg, "-b")) {
            benchmark = true;
            retrace::verbosity = -1;
            glws::debug = false;
        } else if (!strcmp(arg, "-c")) {
            compare_prefix = argv[++i];
            if (snapshot_frequency == FREQUENCY_NEVER) {
                snapshot_frequency = FREQUENCY_FRAME;
            }
        } else if (!strcmp(arg, "-D")) {
            dump_state = atoi(argv[++i]);
            retrace::verbosity = -2;
        } else if (!strcmp(arg, "-core")) {
            defaultProfile = glws::PROFILE_CORE;
        } else if (!strcmp(arg, "-o")) {
            output = strdup(argv[++i]);
        } else if (!strcmp(arg, "-p")) {
            profile = atoll(argv[++i]);
        } else if (!strcmp(arg, "-ps")) {
            profile_start = atoll(argv[++i]);
        } else if (!strcmp(arg, "-pe")) {
            profile_end = atoll(argv[++i]);
        } else if (!strcmp(arg, "-db")) {
            double_buffer = true;
        } else if (!strcmp(arg, "-sb")) {
            double_buffer = false;
        } else if (!strcmp(arg, "--help")) {
            usage();
            return 0;
        } else if (!strcmp(arg, "-s")) {
            snapshot_prefix = argv[++i];
            if (snapshot_frequency == FREQUENCY_NEVER) {
                snapshot_frequency = FREQUENCY_FRAME;
            }
            if (snapshot_prefix[0] == '-' && snapshot_prefix[1] == 0) {
                retrace::verbosity = -2;
            }
        } else if (!strcmp(arg, "-S")) {
            arg = argv[++i];
            if (!strcmp(arg, "frame")) {
                snapshot_frequency = FREQUENCY_FRAME;
            } else if (!strcmp(arg, "framebuffer")) {
                snapshot_frequency = FREQUENCY_FRAMEBUFFER;
            } else if (!strcmp(arg, "draw")) {
                snapshot_frequency = FREQUENCY_DRAW;
            } else {
                std::cerr << "error: unknown frequency " << arg << "\n";
                usage();
                return 1;
            }
            if (snapshot_prefix == NULL) {
                snapshot_prefix = "";
            }
        } else if (!strcmp(arg, "-v")) {
            ++retrace::verbosity;
        } else if (!strcmp(arg, "-w")) {
            wait = true;
        } else {
            std::cerr << "error: unknown option " << arg << "\n";
            usage();
            return 1;
        }
    }

    if (profile)
        m_init();

    glws::init();
    visual = glws::createVisual(double_buffer);

    for ( ; i < argc; ++i) {
        if (!parser.open(argv[i])) {
            std::cerr << "error: failed to open " << argv[i] << "\n";
            return 1;
        }

        display();

        parser.close();
    }
    
    delete visual;
    glws::cleanup();

    return 0;
}

} /* namespace glretrace */