<commit_before>/*
 * Copyright (c) 2000-2004 The Regents of The University of Michigan
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met: redistributions of source code must retain the above copyright
 * notice, this list of conditions and the following disclaimer;
 * redistributions in binary form must reproduce the above copyright
 * notice, this list of conditions and the following disclaimer in the
 * documentation and/or other materials provided with the distribution;
 * neither the name of the copyright holders nor the names of its
 * contributors may be used to endorse or promote products derived from
 * this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

///
/// @file sim/main.cc
///
#include <sys/types.h>
#include <sys/stat.h>
#include <libgen.h>
#include <stdlib.h>
#include <signal.h>

#include <list>
#include <string>
#include <vector>

#include "base/copyright.hh"
#include "base/embedfile.hh"
#include "base/inifile.hh"
#include "base/misc.hh"
#include "base/output.hh"
#include "base/pollevent.hh"
#include "base/statistics.hh"
#include "base/str.hh"
#include "base/time.hh"
#include "cpu/base_cpu.hh"
#include "cpu/full_cpu/smt.hh"
#include "sim/async.hh"
#include "sim/builder.hh"
#include "sim/configfile.hh"
#include "sim/host.hh"
#include "sim/sim_events.hh"
#include "sim/sim_exit.hh"
#include "sim/sim_object.hh"
#include "sim/stat_control.hh"
#include "sim/stats.hh"
#include "sim/universe.hh"
#include "sim/pyconfig/pyconfig.hh"

using namespace std;

// See async.h.
volatile bool async_event = false;
volatile bool async_dump = false;
volatile bool async_dumpreset = false;
volatile bool async_exit = false;
volatile bool async_io = false;
volatile bool async_alarm = false;

/// Stats signal handler.
void
dumpStatsHandler(int sigtype)
{
    async_event = true;
    async_dump = true;
}

void
dumprstStatsHandler(int sigtype)
{
    async_event = true;
    async_dumpreset = true;
}

/// Exit signal handler.
void
exitNowHandler(int sigtype)
{
    async_event = true;
    async_exit = true;
}

/// Abort signal handler.
void
abortHandler(int sigtype)
{
    cerr << "Program aborted at cycle " << curTick << endl;

#if TRACING_ON
    // dump trace buffer, if there is one
    Trace::theLog.dump(cerr);
#endif
}

/// Simulator executable name
char *myProgName = "";

/// Show brief help message.
void
showBriefHelp(ostream &out)
{
    char *prog = basename(myProgName);

    ccprintf(out, "Usage:\n");
    ccprintf(out,
"%s [-d <dir>] [-E <var>[=<val>]] [-I <dir>] [-P <python>]\n"
"        [--<var>=<val>] <config file>\n"
"\n"
"   -d            set the output directory to <dir>\n"
"   -E            set the environment variable <var> to <val> (or 'True')\n"
"   -I            add the directory <dir> to python's path\n"
"   -P            execute <python> directly in the configuration\n"
"   --var=val     set the python variable <var> to '<val>'\n"
"   <configfile>  config file name (.py or .mpy)\n",
             prog);

    ccprintf(out, "%s -X\n    -X            extract embedded files\n", prog);
    ccprintf(out, "%s -h\n    -h            print long help\n", prog);
}

/// Show verbose help message.  Includes parameter listing from
/// showBriefHelp(), plus an exhaustive list of ini-file parameters
/// and SimObjects (with their parameters).
void
showLongHelp(ostream &out)
{
    showBriefHelp(out);

    out << endl
        << endl
        << "-----------------" << endl
        << "Global Parameters" << endl
        << "-----------------" << endl
        << endl;

    ParamContext::describeAllContexts(out);

    out << endl
        << endl
        << "-----------------" << endl
        << "Simulator Objects" << endl
        << "-----------------" << endl
        << endl;

    SimObjectClass::describeAllClasses(out);
}

/// Print welcome message.
void
sayHello(ostream &out)
{
    extern const char *compileDate;	// from date.cc

    ccprintf(out, "M5 Simulator System\n");
    // display copyright
    ccprintf(out, "%s\n", briefCopyright);
    ccprintf(out, "M5 compiled on %d\n", compileDate);

    char *host = getenv("HOSTNAME");
    if (!host)
        host = getenv("HOST");

    if (host)
        ccprintf(out, "M5 executing on %s\n", host);

    ccprintf(out, "M5 simulation started %s\n", Time::start);
}

///
/// Echo the command line for posterity in such a way that it can be
/// used to rerun the same simulation (given the same .ini files).
///
void
echoCommandLine(int argc, char **argv, ostream &out)
{
    out << "command line: " << argv[0];
    for (int i = 1; i < argc; i++) {
        string arg(argv[i]);

        out << ' ';

        // If the arg contains spaces, we need to quote it.
        // The rest of this is overkill to make it look purty.

        // print dashes first outside quotes
        int non_dash_pos = arg.find_first_not_of("-");
        out << arg.substr(0, non_dash_pos);	// print dashes
        string body = arg.substr(non_dash_pos);	// the rest

        // if it's an assignment, handle the lhs & rhs separately
        int eq_pos = body.find("=");
        if (eq_pos == string::npos) {
            out << quote(body);
        }
        else {
            string lhs(body.substr(0, eq_pos));
            string rhs(body.substr(eq_pos + 1));

            out << quote(lhs) << "=" << quote(rhs);
        }
    }
    out << endl << endl;
}

char *
getOptionString(int &index, int argc, char **argv)
{
    char *option = argv[index] + 2;
    if (*option != '\0')
        return option;

    // We didn't find an argument, it must be in the next variable.
    if (++index >= argc)
        panic("option string for option '%s' not found", argv[index - 1]);

    return argv[index];
}

int
main(int argc, char **argv)
{
    // Save off program name
    myProgName = argv[0];

    signal(SIGFPE, SIG_IGN);		// may occur on misspeculated paths
    signal(SIGTRAP, SIG_IGN);
    signal(SIGUSR1, dumpStatsHandler);		// dump intermediate stats
    signal(SIGUSR2, dumprstStatsHandler);	// dump and reset stats
    signal(SIGINT, exitNowHandler);		// dump final stats and exit
    signal(SIGABRT, abortHandler);

    sayHello(cerr);

    bool configfile_found = false;
    PythonConfig pyconfig;
    string outdir;

    // Parse command-line options.
    // Since most of the complex options are handled through the
    // config database, we don't mess with getopts, and just parse
    // manually.
    for (int i = 1; i < argc; ++i) {
        char *arg_str = argv[i];

        // if arg starts with '--', parse as a special python option
        // of the format --<python var>=<string value>, if the arg
        // starts with '-', it should be a simulator option with a
        // format similar to getopt.  In any other case, treat the
        // option as a configuration file name and load it.
        if (arg_str[0] == '-' && arg_str[1] == '-') {
            string str = &arg_str[2];
            string var, val;

            if (!split_first(str, var, val, '='))
                panic("Could not parse configuration argument '%s'\n"
                      "Expecting --<variable>=<value>\n", arg_str);

            pyconfig.setVariable(var, val);
        } else if (arg_str[0] == '-') {
            char *option;
            string var, val;

            // switch on second char
            switch (arg_str[1]) {
              case 'd':
                outdir = getOptionString(i, argc, argv);
                break;

              case 'h':
                showLongHelp(cerr);
                exit(1);

              case 'E':
                option = getOptionString(i, argc, argv);
                if (!split_first(option, var, val, '='))
                    val = "True";

                if (setenv(var.c_str(), val.c_str(), true) == -1)
                    panic("setenv: %s\n", strerror(errno));
                break;

              case 'I':
                option = getOptionString(i, argc, argv);
                pyconfig.addPath(option);
                break;

              case 'P':
                option = getOptionString(i, argc, argv);
                pyconfig.writeLine(option);
                break;

              case 'X': {
                  list<EmbedFile> lst;
                  EmbedMap::all(lst);
                  list<EmbedFile>::iterator i = lst.begin();
                  list<EmbedFile>::iterator end = lst.end();

                  while (i != end) {
                      cprintf("Embedded File: %s\n", i->name);
                      cout.write(i->data, i->length);
                      ++i;
                  }

                  return 0;
              }

              default:
                showBriefHelp(cerr);
                panic("invalid argument '%s'\n", arg_str);
            }
        } else {
            string file(arg_str);
            string base, ext;

            if (!split_last(file, base, ext, '.') ||
                ext != "py" && ext != "mpy")
                panic("Config file '%s' must end in '.py' or '.mpy'\n", file);

            pyconfig.load(file);
            configfile_found = true;
        }
    }

    if (outdir.empty()) {
        char *env = getenv("OUTPUT_DIR");
        outdir = env ? env : ".";
    }

    simout.setDirectory(outdir);

    char *env = getenv("CONFIG_OUTPUT");
    if (!env)
        env = "config.out";
    configStream = simout.find(env);

    if (!configfile_found)
        panic("no configuration file specified!");

    // The configuration database is now complete; start processing it.
    IniFile inifile;
    if (!pyconfig.output(inifile))
        panic("Error processing python code");

    // Initialize statistics database
    Stats::InitSimStats();

    // Now process the configuration hierarchy and create the SimObjects.
    ConfigHierarchy configHierarchy(inifile);
    configHierarchy.build();
    configHierarchy.createSimObjects();

    // Parse and check all non-config-hierarchy parameters.
    ParamContext::parseAllContexts(inifile);
    ParamContext::checkAllContexts();

    // Print hello message to stats file if it's actually a file.  If
    // it's not (i.e. it's cout or cerr) then we already did it above.
    if (simout.isFile(*outputStream))
        sayHello(*outputStream);

    // Echo command line and all parameter settings to stats file as well.
    echoCommandLine(argc, argv, *outputStream);
    ParamContext::showAllContexts(*configStream);

    // Do a second pass to finish initializing the sim objects
    SimObject::initAll();

    // Restore checkpointed state, if any.
    configHierarchy.unserializeSimObjects();

    // Done processing the configuration database.
    // Check for unreferenced entries.
    if (inifile.printUnreferenced())
        panic("unreferenced sections/entries in the intermediate ini file");

    SimObject::regAllStats();

    // uncomment the following to get PC-based execution-time profile
#ifdef DO_PROFILE
    init_profile((char *)&_init, (char *)&_fini);
#endif

    // Check to make sure that the stats package is properly initialized
    Stats::check();

    // Reset to put the stats in a consistent state.
    Stats::reset();

    // Nothing to simulate if we don't have at least one CPU somewhere.
    if (BaseCPU::numSimulatedCPUs() == 0) {
        cerr << "Fatal: no CPUs to simulate." << endl;
        exit(1);
    }

    warn("Entering event queue.  Starting simulation...\n");
    SimStartup();
    while (!mainEventQueue.empty()) {
        assert(curTick <= mainEventQueue.nextTick() &&
               "event scheduled in the past");

        // forward current cycle to the time of the first event on the
        // queue
        curTick = mainEventQueue.nextTick();
        mainEventQueue.serviceOne();

        if (async_event) {
            async_event = false;
            if (async_dump) {
                async_dump = false;

                using namespace Stats;
                SetupEvent(Dump, curTick);
            }

            if (async_dumpreset) {
                async_dumpreset = false;

                using namespace Stats;
                SetupEvent(Dump | Reset, curTick);
            }

            if (async_exit) {
                async_exit = false;
                new SimExitEvent("User requested STOP");
            }

            if (async_io || async_alarm) {
                async_io = false;
                async_alarm = false;
                pollQueue.service();
            }
        }
    }

    // This should never happen... every conceivable way for the
    // simulation to terminate (hit max cycles/insts, signal,
    // simulated system halts/exits) generates an exit event, so we
    // should never run out of events on the queue.
    exitNow("no events on event loop!  All CPUs must be idle.", 1);

    return 0;
}
<commit_msg>Include errno.h to fix compile errors in gcc 3.4<commit_after>/*
 * Copyright (c) 2000-2004 The Regents of The University of Michigan
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met: redistributions of source code must retain the above copyright
 * notice, this list of conditions and the following disclaimer;
 * redistributions in binary form must reproduce the above copyright
 * notice, this list of conditions and the following disclaimer in the
 * documentation and/or other materials provided with the distribution;
 * neither the name of the copyright holders nor the names of its
 * contributors may be used to endorse or promote products derived from
 * this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

///
/// @file sim/main.cc
///
#include <sys/types.h>
#include <sys/stat.h>
#include <errno.h>
#include <libgen.h>
#include <stdlib.h>
#include <signal.h>

#include <list>
#include <string>
#include <vector>

#include "base/copyright.hh"
#include "base/embedfile.hh"
#include "base/inifile.hh"
#include "base/misc.hh"
#include "base/output.hh"
#include "base/pollevent.hh"
#include "base/statistics.hh"
#include "base/str.hh"
#include "base/time.hh"
#include "cpu/base_cpu.hh"
#include "cpu/full_cpu/smt.hh"
#include "sim/async.hh"
#include "sim/builder.hh"
#include "sim/configfile.hh"
#include "sim/host.hh"
#include "sim/sim_events.hh"
#include "sim/sim_exit.hh"
#include "sim/sim_object.hh"
#include "sim/stat_control.hh"
#include "sim/stats.hh"
#include "sim/universe.hh"
#include "sim/pyconfig/pyconfig.hh"

using namespace std;

// See async.h.
volatile bool async_event = false;
volatile bool async_dump = false;
volatile bool async_dumpreset = false;
volatile bool async_exit = false;
volatile bool async_io = false;
volatile bool async_alarm = false;

/// Stats signal handler.
void
dumpStatsHandler(int sigtype)
{
    async_event = true;
    async_dump = true;
}

void
dumprstStatsHandler(int sigtype)
{
    async_event = true;
    async_dumpreset = true;
}

/// Exit signal handler.
void
exitNowHandler(int sigtype)
{
    async_event = true;
    async_exit = true;
}

/// Abort signal handler.
void
abortHandler(int sigtype)
{
    cerr << "Program aborted at cycle " << curTick << endl;

#if TRACING_ON
    // dump trace buffer, if there is one
    Trace::theLog.dump(cerr);
#endif
}

/// Simulator executable name
char *myProgName = "";

/// Show brief help message.
void
showBriefHelp(ostream &out)
{
    char *prog = basename(myProgName);

    ccprintf(out, "Usage:\n");
    ccprintf(out,
"%s [-d <dir>] [-E <var>[=<val>]] [-I <dir>] [-P <python>]\n"
"        [--<var>=<val>] <config file>\n"
"\n"
"   -d            set the output directory to <dir>\n"
"   -E            set the environment variable <var> to <val> (or 'True')\n"
"   -I            add the directory <dir> to python's path\n"
"   -P            execute <python> directly in the configuration\n"
"   --var=val     set the python variable <var> to '<val>'\n"
"   <configfile>  config file name (.py or .mpy)\n",
             prog);

    ccprintf(out, "%s -X\n    -X            extract embedded files\n", prog);
    ccprintf(out, "%s -h\n    -h            print long help\n", prog);
}

/// Show verbose help message.  Includes parameter listing from
/// showBriefHelp(), plus an exhaustive list of ini-file parameters
/// and SimObjects (with their parameters).
void
showLongHelp(ostream &out)
{
    showBriefHelp(out);

    out << endl
        << endl
        << "-----------------" << endl
        << "Global Parameters" << endl
        << "-----------------" << endl
        << endl;

    ParamContext::describeAllContexts(out);

    out << endl
        << endl
        << "-----------------" << endl
        << "Simulator Objects" << endl
        << "-----------------" << endl
        << endl;

    SimObjectClass::describeAllClasses(out);
}

/// Print welcome message.
void
sayHello(ostream &out)
{
    extern const char *compileDate;	// from date.cc

    ccprintf(out, "M5 Simulator System\n");
    // display copyright
    ccprintf(out, "%s\n", briefCopyright);
    ccprintf(out, "M5 compiled on %d\n", compileDate);

    char *host = getenv("HOSTNAME");
    if (!host)
        host = getenv("HOST");

    if (host)
        ccprintf(out, "M5 executing on %s\n", host);

    ccprintf(out, "M5 simulation started %s\n", Time::start);
}

///
/// Echo the command line for posterity in such a way that it can be
/// used to rerun the same simulation (given the same .ini files).
///
void
echoCommandLine(int argc, char **argv, ostream &out)
{
    out << "command line: " << argv[0];
    for (int i = 1; i < argc; i++) {
        string arg(argv[i]);

        out << ' ';

        // If the arg contains spaces, we need to quote it.
        // The rest of this is overkill to make it look purty.

        // print dashes first outside quotes
        int non_dash_pos = arg.find_first_not_of("-");
        out << arg.substr(0, non_dash_pos);	// print dashes
        string body = arg.substr(non_dash_pos);	// the rest

        // if it's an assignment, handle the lhs & rhs separately
        int eq_pos = body.find("=");
        if (eq_pos == string::npos) {
            out << quote(body);
        }
        else {
            string lhs(body.substr(0, eq_pos));
            string rhs(body.substr(eq_pos + 1));

            out << quote(lhs) << "=" << quote(rhs);
        }
    }
    out << endl << endl;
}

char *
getOptionString(int &index, int argc, char **argv)
{
    char *option = argv[index] + 2;
    if (*option != '\0')
        return option;

    // We didn't find an argument, it must be in the next variable.
    if (++index >= argc)
        panic("option string for option '%s' not found", argv[index - 1]);

    return argv[index];
}

int
main(int argc, char **argv)
{
    // Save off program name
    myProgName = argv[0];

    signal(SIGFPE, SIG_IGN);		// may occur on misspeculated paths
    signal(SIGTRAP, SIG_IGN);
    signal(SIGUSR1, dumpStatsHandler);		// dump intermediate stats
    signal(SIGUSR2, dumprstStatsHandler);	// dump and reset stats
    signal(SIGINT, exitNowHandler);		// dump final stats and exit
    signal(SIGABRT, abortHandler);

    sayHello(cerr);

    bool configfile_found = false;
    PythonConfig pyconfig;
    string outdir;

    // Parse command-line options.
    // Since most of the complex options are handled through the
    // config database, we don't mess with getopts, and just parse
    // manually.
    for (int i = 1; i < argc; ++i) {
        char *arg_str = argv[i];

        // if arg starts with '--', parse as a special python option
        // of the format --<python var>=<string value>, if the arg
        // starts with '-', it should be a simulator option with a
        // format similar to getopt.  In any other case, treat the
        // option as a configuration file name and load it.
        if (arg_str[0] == '-' && arg_str[1] == '-') {
            string str = &arg_str[2];
            string var, val;

            if (!split_first(str, var, val, '='))
                panic("Could not parse configuration argument '%s'\n"
                      "Expecting --<variable>=<value>\n", arg_str);

            pyconfig.setVariable(var, val);
        } else if (arg_str[0] == '-') {
            char *option;
            string var, val;

            // switch on second char
            switch (arg_str[1]) {
              case 'd':
                outdir = getOptionString(i, argc, argv);
                break;

              case 'h':
                showLongHelp(cerr);
                exit(1);

              case 'E':
                option = getOptionString(i, argc, argv);
                if (!split_first(option, var, val, '='))
                    val = "True";

                if (setenv(var.c_str(), val.c_str(), true) == -1)
                    panic("setenv: %s\n", strerror(errno));
                break;

              case 'I':
                option = getOptionString(i, argc, argv);
                pyconfig.addPath(option);
                break;

              case 'P':
                option = getOptionString(i, argc, argv);
                pyconfig.writeLine(option);
                break;

              case 'X': {
                  list<EmbedFile> lst;
                  EmbedMap::all(lst);
                  list<EmbedFile>::iterator i = lst.begin();
                  list<EmbedFile>::iterator end = lst.end();

                  while (i != end) {
                      cprintf("Embedded File: %s\n", i->name);
                      cout.write(i->data, i->length);
                      ++i;
                  }

                  return 0;
              }

              default:
                showBriefHelp(cerr);
                panic("invalid argument '%s'\n", arg_str);
            }
        } else {
            string file(arg_str);
            string base, ext;

            if (!split_last(file, base, ext, '.') ||
                ext != "py" && ext != "mpy")
                panic("Config file '%s' must end in '.py' or '.mpy'\n", file);

            pyconfig.load(file);
            configfile_found = true;
        }
    }

    if (outdir.empty()) {
        char *env = getenv("OUTPUT_DIR");
        outdir = env ? env : ".";
    }

    simout.setDirectory(outdir);

    char *env = getenv("CONFIG_OUTPUT");
    if (!env)
        env = "config.out";
    configStream = simout.find(env);

    if (!configfile_found)
        panic("no configuration file specified!");

    // The configuration database is now complete; start processing it.
    IniFile inifile;
    if (!pyconfig.output(inifile))
        panic("Error processing python code");

    // Initialize statistics database
    Stats::InitSimStats();

    // Now process the configuration hierarchy and create the SimObjects.
    ConfigHierarchy configHierarchy(inifile);
    configHierarchy.build();
    configHierarchy.createSimObjects();

    // Parse and check all non-config-hierarchy parameters.
    ParamContext::parseAllContexts(inifile);
    ParamContext::checkAllContexts();

    // Print hello message to stats file if it's actually a file.  If
    // it's not (i.e. it's cout or cerr) then we already did it above.
    if (simout.isFile(*outputStream))
        sayHello(*outputStream);

    // Echo command line and all parameter settings to stats file as well.
    echoCommandLine(argc, argv, *outputStream);
    ParamContext::showAllContexts(*configStream);

    // Do a second pass to finish initializing the sim objects
    SimObject::initAll();

    // Restore checkpointed state, if any.
    configHierarchy.unserializeSimObjects();

    // Done processing the configuration database.
    // Check for unreferenced entries.
    if (inifile.printUnreferenced())
        panic("unreferenced sections/entries in the intermediate ini file");

    SimObject::regAllStats();

    // uncomment the following to get PC-based execution-time profile
#ifdef DO_PROFILE
    init_profile((char *)&_init, (char *)&_fini);
#endif

    // Check to make sure that the stats package is properly initialized
    Stats::check();

    // Reset to put the stats in a consistent state.
    Stats::reset();

    // Nothing to simulate if we don't have at least one CPU somewhere.
    if (BaseCPU::numSimulatedCPUs() == 0) {
        cerr << "Fatal: no CPUs to simulate." << endl;
        exit(1);
    }

    warn("Entering event queue.  Starting simulation...\n");
    SimStartup();
    while (!mainEventQueue.empty()) {
        assert(curTick <= mainEventQueue.nextTick() &&
               "event scheduled in the past");

        // forward current cycle to the time of the first event on the
        // queue
        curTick = mainEventQueue.nextTick();
        mainEventQueue.serviceOne();

        if (async_event) {
            async_event = false;
            if (async_dump) {
                async_dump = false;

                using namespace Stats;
                SetupEvent(Dump, curTick);
            }

            if (async_dumpreset) {
                async_dumpreset = false;

                using namespace Stats;
                SetupEvent(Dump | Reset, curTick);
            }

            if (async_exit) {
                async_exit = false;
                new SimExitEvent("User requested STOP");
            }

            if (async_io || async_alarm) {
                async_io = false;
                async_alarm = false;
                pollQueue.service();
            }
        }
    }

    // This should never happen... every conceivable way for the
    // simulation to terminate (hit max cycles/insts, signal,
    // simulated system halts/exits) generates an exit event, so we
    // should never run out of events on the queue.
    exitNow("no events on event loop!  All CPUs must be idle.", 1);

    return 0;
}
<|endoftext|>


<commit_before>/*
 * Copyright (c) 2000-2004 The Regents of The University of Michigan
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met: redistributions of source code must retain the above copyright
 * notice, this list of conditions and the following disclaimer;
 * redistributions in binary form must reproduce the above copyright
 * notice, this list of conditions and the following disclaimer in the
 * documentation and/or other materials provided with the distribution;
 * neither the name of the copyright holders nor the names of its
 * contributors may be used to endorse or promote products derived from
 * this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

///
/// @file sim/main.cc
///
#include <sys/types.h>
#include <sys/stat.h>
#include <stdlib.h>
#include <signal.h>

#include <list>
#include <string>
#include <vector>

#include "base/copyright.hh"
#include "base/embedfile.hh"
#include "base/inifile.hh"
#include "base/misc.hh"
#include "base/pollevent.hh"
#include "base/statistics.hh"
#include "base/str.hh"
#include "base/time.hh"
#include "cpu/base_cpu.hh"
#include "cpu/full_cpu/smt.hh"
#include "sim/async.hh"
#include "sim/builder.hh"
#include "sim/configfile.hh"
#include "sim/host.hh"
#include "sim/sim_events.hh"
#include "sim/sim_exit.hh"
#include "sim/sim_object.hh"
#include "sim/stat_control.hh"
#include "sim/stats.hh"
#include "sim/universe.hh"
#include "sim/pyconfig/pyconfig.hh"

using namespace std;

// See async.h.
volatile bool async_event = false;
volatile bool async_dump = false;
volatile bool async_dumpreset = false;
volatile bool async_exit = false;
volatile bool async_io = false;
volatile bool async_alarm = false;

/// Stats signal handler.
void
dumpStatsHandler(int sigtype)
{
    async_event = true;
    async_dump = true;
}

void
dumprstStatsHandler(int sigtype)
{
    async_event = true;
    async_dumpreset = true;
}

/// Exit signal handler.
void
exitNowHandler(int sigtype)
{
    async_event = true;
    async_exit = true;
}

/// Abort signal handler.
void
abortHandler(int sigtype)
{
    cerr << "Program aborted at cycle " << curTick << endl;

#if TRACING_ON
    // dump trace buffer, if there is one
    Trace::theLog.dump(cerr);
#endif
}

/// Simulator executable name
const char *myProgName = "";

/// Show brief help message.
static void
showBriefHelp(ostream &out)
{
    out << "Usage: " << myProgName
        << " [-hnu] [-Dname[=def]] [-Uname] [-I[dir]] "
        << "<config-spec> [<config-spec> ...]\n"
        << "[] [<config file> ...]\n"
        << "   -h: print long help (including parameter listing)\n"
        << "   -u: don't quit on unreferenced parameters\n"
        << "   -D,-U,-I: passed to cpp for preprocessing .ini files\n"
        << "   <config-spec>: config file name (.ini or .py) or\n"
        << "                  single param (--<section>:<param>=<value>)"
        << endl;
}

/// Show verbose help message.  Includes parameter listing from
/// showBriefHelp(), plus an exhaustive list of ini-file parameters
/// and SimObjects (with their parameters).
static void
showLongHelp(ostream &out)
{
    showBriefHelp(out);

    out << endl
        << endl
        << "-----------------" << endl
        << "Global Parameters" << endl
        << "-----------------" << endl
        << endl;

    ParamContext::describeAllContexts(out);

    out << endl
        << endl
        << "-----------------" << endl
        << "Simulator Objects" << endl
        << "-----------------" << endl
        << endl;

    SimObjectClass::describeAllClasses(out);
}

/// Print welcome message.
static void
sayHello(ostream &out)
{
    extern const char *compileDate;	// from date.cc

    ccprintf(out, "M5 Simulator System\n");
    // display copyright
    ccprintf(out, "%s\n", briefCopyright);
    ccprintf(out, "M5 compiled on %d\n", compileDate);

    char *host = getenv("HOSTNAME");
    if (!host)
        host = getenv("HOST");

    if (host)
        ccprintf(out, "M5 executing on %s\n", host);

    ccprintf(out, "M5 simulation started %s\n", Time::start);
}

///
/// Echo the command line for posterity in such a way that it can be
/// used to rerun the same simulation (given the same .ini files).
///
static void
echoCommandLine(int argc, char **argv, ostream &out)
{
    out << "command line: " << argv[0];
    for (int i = 1; i < argc; i++) {
        string arg(argv[i]);

        out << ' ';

        // If the arg contains spaces, we need to quote it.
        // The rest of this is overkill to make it look purty.

        // print dashes first outside quotes
        int non_dash_pos = arg.find_first_not_of("-");
        out << arg.substr(0, non_dash_pos);	// print dashes
        string body = arg.substr(non_dash_pos);	// the rest

        // if it's an assignment, handle the lhs & rhs separately
        int eq_pos = body.find("=");
        if (eq_pos == string::npos) {
            out << quote(body);
        }
        else {
            string lhs(body.substr(0, eq_pos));
            string rhs(body.substr(eq_pos + 1));

            out << quote(lhs) << "=" << quote(rhs);
        }
    }
    out << endl << endl;
}

///
/// The simulator configuration database.  This is the union of all
/// specified .ini files.  This shouldn't need to be visible outside
/// this file, as it is passed as a parameter to all the param-parsing
/// routines.
///
static IniFile simConfigDB;

/// M5 entry point.
int
main(int argc, char **argv)
{
    // Save off program name
    myProgName = argv[0];

    signal(SIGFPE, SIG_IGN);		// may occur on misspeculated paths
    signal(SIGTRAP, SIG_IGN);
    signal(SIGUSR1, dumpStatsHandler);		// dump intermediate stats
    signal(SIGUSR2, dumprstStatsHandler);	// dump and reset stats
    signal(SIGINT, exitNowHandler);		// dump final stats and exit
    signal(SIGABRT, abortHandler);

    sayHello(cerr);

    // Initialize statistics database
    Stats::InitSimStats();

    vector<char *> cppArgs;

    // Should we quit if there are unreferenced parameters?  By
    // default, yes... it's a good way of catching typos in
    // section/parameter names (which otherwise go by silently).  Use
    // -u to override.
    bool quitOnUnreferenced = true;

    bool python_initialized = false;

    // Parse command-line options.
    // Since most of the complex options are handled through the
    // config database, we don't mess with getopts, and just parse
    // manually.
    for (int i = 1; i < argc; ++i) {
        char *arg_str = argv[i];

        // if arg starts with '-', parse as option,
        // else treat it as a configuration file name and load it
        if (arg_str[0] == '-') {

            // switch on second char
            switch (arg_str[1]) {
              case 'X': {
                  list<EmbedFile> lst;
                  EmbedMap::all(lst);
                  list<EmbedFile>::iterator i = lst.begin();
                  list<EmbedFile>::iterator end = lst.end();

                  while (i != end) {
                      cprintf("Embedded File: %s\n", i->name);
                      cout.write(i->data, i->length);
                      ++i;
                  }

                  return 0;
              }

              case 'h':
                // -h: show help
                showLongHelp(cerr);
                exit(1);

              case 'u':
                // -u: don't quit on unreferenced parameters
                quitOnUnreferenced = false;
                break;

              case 'D':
              case 'U':
                // cpp options: record & pass to cpp.  Note that these
                // cannot have spaces, i.e., '-Dname=val' is OK, but
                // '-D name=val' is not.  I don't consider this a
                // problem, since even though gnu cpp accepts the
                // latter, other cpp implementations do not (Tru64,
                // for one).
                cppArgs.push_back(arg_str);
                break;

              case 'I': {
                  // We push -I as an argument to cpp
                  cppArgs.push_back(arg_str);

                  string arg = arg_str + 2;
                  eat_white(arg);

                  // Send this as the python path
                  addPythonPath(arg);
              } break;

              case 'P':
                if (!python_initialized) {
                    initPythonConfig();
                    python_initialized = true;
                }
                writePythonString(arg_str + 2);
                writePythonString("\n");

              case 'E':
                if (putenv(arg_str + 2) == -1)
                    panic("putenv: %s\n", strerror(errno));
                break;

              case '-':
                // command-line configuration parameter:
                // '--<section>:<parameter>=<value>'
                if (!simConfigDB.add(arg_str + 2)) {
                    // parse error
                    ccprintf(cerr,
                             "Could not parse configuration argument '%s'\n"
                             "Expecting --<section>:<parameter>=<value>\n",
                             arg_str);
                    exit(0);
                }
                break;

              default:
                showBriefHelp(cerr);
                ccprintf(cerr, "Fatal: invalid argument '%s'\n", arg_str);
                exit(0);
            }
        }
        else {
            // no '-', treat as config file name

            // make STL string out of file name
            string filename(arg_str);

            int ext_loc = filename.rfind(".");

            string ext =
                (ext_loc != string::npos) ? filename.substr(ext_loc) : "";

            if (ext == ".ini") {
                if (!simConfigDB.loadCPP(filename, cppArgs)) {
                    cprintf("Error processing file %s\n", filename);
                    exit(1);
                }
            } else if (ext == ".py" || ext == ".mpy") {
                if (!python_initialized) {
                    initPythonConfig();
                    python_initialized = true;
                }
                loadPythonConfig(filename);
            }
            else {
                cprintf("Config file name '%s' must end in '.py' or '.ini'.\n",
                        filename);
                exit(1);
            }
        }
    }

    if (python_initialized && finishPythonConfig(simConfigDB)) {
        cprintf("Error processing python code\n");
        exit(1);
    }

    // The configuration database is now complete; start processing it.

    // Parse and check all non-config-hierarchy parameters.
    ParamContext::parseAllContexts(simConfigDB);
    ParamContext::checkAllContexts();

    // Print header info into stats file.  Can't do this sooner since
    // the stat file name is set via a .ini param... thus it just got
    // opened above during ParamContext::checkAllContexts().

    // Now process the configuration hierarchy and create the SimObjects.
    ConfigHierarchy configHierarchy(simConfigDB);
    configHierarchy.build();
    configHierarchy.createSimObjects();

    // Print hello message to stats file if it's actually a file.  If
    // it's not (i.e. it's cout or cerr) then we already did it above.
    if (outputStream != &cout && outputStream != &cerr)
        sayHello(*outputStream);

    // Echo command line and all parameter settings to stats file as well.
    echoCommandLine(argc, argv, *outputStream);
    ParamContext::showAllContexts(*configStream);

    // Do a second pass to finish initializing the sim objects
    SimObject::initAll();

    // Restore checkpointed state, if any.
    configHierarchy.unserializeSimObjects();

    // Done processing the configuration database.
    // Check for unreferenced entries.
    if (simConfigDB.printUnreferenced() && quitOnUnreferenced) {
        cerr << "Fatal: unreferenced .ini sections/entries." << endl
             << "If this is not an error, add 'unref_section_ok=y' or "
             << "'unref_entries_ok=y' to the appropriate sections "
             << "to suppress this message." << endl;
        exit(1);
    }

    SimObject::regAllStats();

    // uncomment the following to get PC-based execution-time profile
#ifdef DO_PROFILE
    init_profile((char *)&_init, (char *)&_fini);
#endif

    // Check to make sure that the stats package is properly initialized
    Stats::check();

    // Reset to put the stats in a consistent state.
    Stats::reset();

    // Nothing to simulate if we don't have at least one CPU somewhere.
    if (BaseCPU::numSimulatedCPUs() == 0) {
        cerr << "Fatal: no CPUs to simulate." << endl;
        exit(1);
    }

    warn("Entering event queue.  Starting simulation...\n");
    SimStartup();
    while (!mainEventQueue.empty()) {
        assert(curTick <= mainEventQueue.nextTick() &&
               "event scheduled in the past");

        // forward current cycle to the time of the first event on the
        // queue
        curTick = mainEventQueue.nextTick();
        mainEventQueue.serviceOne();

        if (async_event) {
            async_event = false;
            if (async_dump) {
                async_dump = false;

                using namespace Stats;
                SetupEvent(Dump, curTick);
            }

            if (async_dumpreset) {
                async_dumpreset = false;

                using namespace Stats;
                SetupEvent(Dump | Reset, curTick);
            }

            if (async_exit) {
                async_exit = false;
                new SimExitEvent("User requested STOP");
            }

            if (async_io || async_alarm) {
                async_io = false;
                async_alarm = false;
                pollQueue.service();
            }
        }
    }

    // This should never happen... every conceivable way for the
    // simulation to terminate (hit max cycles/insts, signal,
    // simulated system halts/exits) generates an exit event, so we
    // should never run out of events on the queue.
    exitNow("no events on event loop!  All CPUs must be idle.", 1);

    return 0;
}
<commit_msg>get the exit condition for the python code right<commit_after>/*
 * Copyright (c) 2000-2004 The Regents of The University of Michigan
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions are
 * met: redistributions of source code must retain the above copyright
 * notice, this list of conditions and the following disclaimer;
 * redistributions in binary form must reproduce the above copyright
 * notice, this list of conditions and the following disclaimer in the
 * documentation and/or other materials provided with the distribution;
 * neither the name of the copyright holders nor the names of its
 * contributors may be used to endorse or promote products derived from
 * this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
 * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
 * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
 * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
 * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
 * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
 * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
 * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

///
/// @file sim/main.cc
///
#include <sys/types.h>
#include <sys/stat.h>
#include <stdlib.h>
#include <signal.h>

#include <list>
#include <string>
#include <vector>

#include "base/copyright.hh"
#include "base/embedfile.hh"
#include "base/inifile.hh"
#include "base/misc.hh"
#include "base/pollevent.hh"
#include "base/statistics.hh"
#include "base/str.hh"
#include "base/time.hh"
#include "cpu/base_cpu.hh"
#include "cpu/full_cpu/smt.hh"
#include "sim/async.hh"
#include "sim/builder.hh"
#include "sim/configfile.hh"
#include "sim/host.hh"
#include "sim/sim_events.hh"
#include "sim/sim_exit.hh"
#include "sim/sim_object.hh"
#include "sim/stat_control.hh"
#include "sim/stats.hh"
#include "sim/universe.hh"
#include "sim/pyconfig/pyconfig.hh"

using namespace std;

// See async.h.
volatile bool async_event = false;
volatile bool async_dump = false;
volatile bool async_dumpreset = false;
volatile bool async_exit = false;
volatile bool async_io = false;
volatile bool async_alarm = false;

/// Stats signal handler.
void
dumpStatsHandler(int sigtype)
{
    async_event = true;
    async_dump = true;
}

void
dumprstStatsHandler(int sigtype)
{
    async_event = true;
    async_dumpreset = true;
}

/// Exit signal handler.
void
exitNowHandler(int sigtype)
{
    async_event = true;
    async_exit = true;
}

/// Abort signal handler.
void
abortHandler(int sigtype)
{
    cerr << "Program aborted at cycle " << curTick << endl;

#if TRACING_ON
    // dump trace buffer, if there is one
    Trace::theLog.dump(cerr);
#endif
}

/// Simulator executable name
const char *myProgName = "";

/// Show brief help message.
static void
showBriefHelp(ostream &out)
{
    out << "Usage: " << myProgName
        << " [-hnu] [-Dname[=def]] [-Uname] [-I[dir]] "
        << "<config-spec> [<config-spec> ...]\n"
        << "[] [<config file> ...]\n"
        << "   -h: print long help (including parameter listing)\n"
        << "   -u: don't quit on unreferenced parameters\n"
        << "   -D,-U,-I: passed to cpp for preprocessing .ini files\n"
        << "   <config-spec>: config file name (.ini or .py) or\n"
        << "                  single param (--<section>:<param>=<value>)"
        << endl;
}

/// Show verbose help message.  Includes parameter listing from
/// showBriefHelp(), plus an exhaustive list of ini-file parameters
/// and SimObjects (with their parameters).
static void
showLongHelp(ostream &out)
{
    showBriefHelp(out);

    out << endl
        << endl
        << "-----------------" << endl
        << "Global Parameters" << endl
        << "-----------------" << endl
        << endl;

    ParamContext::describeAllContexts(out);

    out << endl
        << endl
        << "-----------------" << endl
        << "Simulator Objects" << endl
        << "-----------------" << endl
        << endl;

    SimObjectClass::describeAllClasses(out);
}

/// Print welcome message.
static void
sayHello(ostream &out)
{
    extern const char *compileDate;	// from date.cc

    ccprintf(out, "M5 Simulator System\n");
    // display copyright
    ccprintf(out, "%s\n", briefCopyright);
    ccprintf(out, "M5 compiled on %d\n", compileDate);

    char *host = getenv("HOSTNAME");
    if (!host)
        host = getenv("HOST");

    if (host)
        ccprintf(out, "M5 executing on %s\n", host);

    ccprintf(out, "M5 simulation started %s\n", Time::start);
}

///
/// Echo the command line for posterity in such a way that it can be
/// used to rerun the same simulation (given the same .ini files).
///
static void
echoCommandLine(int argc, char **argv, ostream &out)
{
    out << "command line: " << argv[0];
    for (int i = 1; i < argc; i++) {
        string arg(argv[i]);

        out << ' ';

        // If the arg contains spaces, we need to quote it.
        // The rest of this is overkill to make it look purty.

        // print dashes first outside quotes
        int non_dash_pos = arg.find_first_not_of("-");
        out << arg.substr(0, non_dash_pos);	// print dashes
        string body = arg.substr(non_dash_pos);	// the rest

        // if it's an assignment, handle the lhs & rhs separately
        int eq_pos = body.find("=");
        if (eq_pos == string::npos) {
            out << quote(body);
        }
        else {
            string lhs(body.substr(0, eq_pos));
            string rhs(body.substr(eq_pos + 1));

            out << quote(lhs) << "=" << quote(rhs);
        }
    }
    out << endl << endl;
}

///
/// The simulator configuration database.  This is the union of all
/// specified .ini files.  This shouldn't need to be visible outside
/// this file, as it is passed as a parameter to all the param-parsing
/// routines.
///
static IniFile simConfigDB;

/// M5 entry point.
int
main(int argc, char **argv)
{
    // Save off program name
    myProgName = argv[0];

    signal(SIGFPE, SIG_IGN);		// may occur on misspeculated paths
    signal(SIGTRAP, SIG_IGN);
    signal(SIGUSR1, dumpStatsHandler);		// dump intermediate stats
    signal(SIGUSR2, dumprstStatsHandler);	// dump and reset stats
    signal(SIGINT, exitNowHandler);		// dump final stats and exit
    signal(SIGABRT, abortHandler);

    sayHello(cerr);

    // Initialize statistics database
    Stats::InitSimStats();

    vector<char *> cppArgs;

    // Should we quit if there are unreferenced parameters?  By
    // default, yes... it's a good way of catching typos in
    // section/parameter names (which otherwise go by silently).  Use
    // -u to override.
    bool quitOnUnreferenced = true;

    bool python_initialized = false;

    // Parse command-line options.
    // Since most of the complex options are handled through the
    // config database, we don't mess with getopts, and just parse
    // manually.
    for (int i = 1; i < argc; ++i) {
        char *arg_str = argv[i];

        // if arg starts with '-', parse as option,
        // else treat it as a configuration file name and load it
        if (arg_str[0] == '-') {

            // switch on second char
            switch (arg_str[1]) {
              case 'X': {
                  list<EmbedFile> lst;
                  EmbedMap::all(lst);
                  list<EmbedFile>::iterator i = lst.begin();
                  list<EmbedFile>::iterator end = lst.end();

                  while (i != end) {
                      cprintf("Embedded File: %s\n", i->name);
                      cout.write(i->data, i->length);
                      ++i;
                  }

                  return 0;
              }

              case 'h':
                // -h: show help
                showLongHelp(cerr);
                exit(1);

              case 'u':
                // -u: don't quit on unreferenced parameters
                quitOnUnreferenced = false;
                break;

              case 'D':
              case 'U':
                // cpp options: record & pass to cpp.  Note that these
                // cannot have spaces, i.e., '-Dname=val' is OK, but
                // '-D name=val' is not.  I don't consider this a
                // problem, since even though gnu cpp accepts the
                // latter, other cpp implementations do not (Tru64,
                // for one).
                cppArgs.push_back(arg_str);
                break;

              case 'I': {
                  // We push -I as an argument to cpp
                  cppArgs.push_back(arg_str);

                  string arg = arg_str + 2;
                  eat_white(arg);

                  // Send this as the python path
                  addPythonPath(arg);
              } break;

              case 'P':
                if (!python_initialized) {
                    initPythonConfig();
                    python_initialized = true;
                }
                writePythonString(arg_str + 2);
                writePythonString("\n");

              case 'E':
                if (putenv(arg_str + 2) == -1)
                    panic("putenv: %s\n", strerror(errno));
                break;

              case '-':
                // command-line configuration parameter:
                // '--<section>:<parameter>=<value>'
                if (!simConfigDB.add(arg_str + 2)) {
                    // parse error
                    ccprintf(cerr,
                             "Could not parse configuration argument '%s'\n"
                             "Expecting --<section>:<parameter>=<value>\n",
                             arg_str);
                    exit(0);
                }
                break;

              default:
                showBriefHelp(cerr);
                ccprintf(cerr, "Fatal: invalid argument '%s'\n", arg_str);
                exit(0);
            }
        }
        else {
            // no '-', treat as config file name

            // make STL string out of file name
            string filename(arg_str);

            int ext_loc = filename.rfind(".");

            string ext =
                (ext_loc != string::npos) ? filename.substr(ext_loc) : "";

            if (ext == ".ini") {
                if (!simConfigDB.loadCPP(filename, cppArgs)) {
                    cprintf("Error processing file %s\n", filename);
                    exit(1);
                }
            } else if (ext == ".py" || ext == ".mpy") {
                if (!python_initialized) {
                    initPythonConfig();
                    python_initialized = true;
                }
                loadPythonConfig(filename);
            }
            else {
                cprintf("Config file name '%s' must end in '.py' or '.ini'.\n",
                        filename);
                exit(1);
            }
        }
    }

    if (python_initialized && !finishPythonConfig(simConfigDB)) {
        cprintf("Error processing python code\n");
        exit(1);
    }

    // The configuration database is now complete; start processing it.

    // Parse and check all non-config-hierarchy parameters.
    ParamContext::parseAllContexts(simConfigDB);
    ParamContext::checkAllContexts();

    // Print header info into stats file.  Can't do this sooner since
    // the stat file name is set via a .ini param... thus it just got
    // opened above during ParamContext::checkAllContexts().

    // Now process the configuration hierarchy and create the SimObjects.
    ConfigHierarchy configHierarchy(simConfigDB);
    configHierarchy.build();
    configHierarchy.createSimObjects();

    // Print hello message to stats file if it's actually a file.  If
    // it's not (i.e. it's cout or cerr) then we already did it above.
    if (outputStream != &cout && outputStream != &cerr)
        sayHello(*outputStream);

    // Echo command line and all parameter settings to stats file as well.
    echoCommandLine(argc, argv, *outputStream);
    ParamContext::showAllContexts(*configStream);

    // Do a second pass to finish initializing the sim objects
    SimObject::initAll();

    // Restore checkpointed state, if any.
    configHierarchy.unserializeSimObjects();

    // Done processing the configuration database.
    // Check for unreferenced entries.
    if (simConfigDB.printUnreferenced() && quitOnUnreferenced) {
        cerr << "Fatal: unreferenced .ini sections/entries." << endl
             << "If this is not an error, add 'unref_section_ok=y' or "
             << "'unref_entries_ok=y' to the appropriate sections "
             << "to suppress this message." << endl;
        exit(1);
    }

    SimObject::regAllStats();

    // uncomment the following to get PC-based execution-time profile
#ifdef DO_PROFILE
    init_profile((char *)&_init, (char *)&_fini);
#endif

    // Check to make sure that the stats package is properly initialized
    Stats::check();

    // Reset to put the stats in a consistent state.
    Stats::reset();

    // Nothing to simulate if we don't have at least one CPU somewhere.
    if (BaseCPU::numSimulatedCPUs() == 0) {
        cerr << "Fatal: no CPUs to simulate." << endl;
        exit(1);
    }

    warn("Entering event queue.  Starting simulation...\n");
    SimStartup();
    while (!mainEventQueue.empty()) {
        assert(curTick <= mainEventQueue.nextTick() &&
               "event scheduled in the past");

        // forward current cycle to the time of the first event on the
        // queue
        curTick = mainEventQueue.nextTick();
        mainEventQueue.serviceOne();

        if (async_event) {
            async_event = false;
            if (async_dump) {
                async_dump = false;

                using namespace Stats;
                SetupEvent(Dump, curTick);
            }

            if (async_dumpreset) {
                async_dumpreset = false;

                using namespace Stats;
                SetupEvent(Dump | Reset, curTick);
            }

            if (async_exit) {
                async_exit = false;
                new SimExitEvent("User requested STOP");
            }

            if (async_io || async_alarm) {
                async_io = false;
                async_alarm = false;
                pollQueue.service();
            }
        }
    }

    // This should never happen... every conceivable way for the
    // simulation to terminate (hit max cycles/insts, signal,
    // simulated system halts/exits) generates an exit event, so we
    // should never run out of events on the queue.
    exitNow("no events on event loop!  All CPUs must be idle.", 1);

    return 0;
}
<|endoftext|>


<commit_before>#include "SkCanvas.h"
#include "SkCommandLineFlags.h"
#include "SkData.h"
#include "SkDocument.h"
#include "SkGraphics.h"
#include "SkSurface.h"
#include "SkImage.h"
#include "SkImageFilter.h"
#include "SkStream.h"
#include "SkString.h"
#include "SkPaint.h"
#include "SkImageGenerator.h"

#include <boost/python.hpp>
#include <stdexcept>

#define IMAGE_QUALITY 70

#ifdef SIPSKIA_BGRA
#define SIPSKIA_COLORTYPE kBGRA_8888_SkColorType
#else
#define SIPSKIA_COLORTYPE kN32_SkColorType
#endif

boost::python::object convert_webp(char* data, long size, bool is_gallery_card) {
	SkBitmap bm;
	sk_sp<SkData> skdata(SkData::MakeWithoutCopy((void*)data, size));
	if(!data) return boost::python::object();
	std::unique_ptr<SkImageGenerator> gen = SkImageGenerator::MakeFromEncoded(skdata);
	if(!gen) return boost::python::object();

	SkImageInfo bmInfo = gen->getInfo();
	gen->tryGenerateBitmap(&bm, bmInfo, nullptr);

	float width = 720.0f;
	float height = 720.0f;
	if(is_gallery_card){
		height = 1120.0f;
	}

	SkPaint paint;
	paint.setAntiAlias(true);
	paint.setStyle(SkPaint::kFill_Style);
	paint.setFilterQuality(kHigh_SkFilterQuality);

	sk_sp<SkSurface> surface(SkSurface::MakeRaster(SkImageInfo::Make(width, height, SIPSKIA_COLORTYPE, kPremul_SkAlphaType, nullptr)).release());
	SkCanvas* canvas = surface->getCanvas();
	canvas->drawBitmapRect(bm, SkRect::MakeLTRB(0, 0, width, height), SkRect::MakeXYWH(0.0f, 0.0f, width, height), &paint);
	sk_sp<SkImage> img(surface->makeImageSnapshot());
	if(!img) return boost::python::object();
	sk_sp<SkData> jpg(img->encode(SkEncodedImageFormat::kWEBP, IMAGE_QUALITY));
	if(!jpg) return boost::python::object();

	return boost::python::object(boost::python::handle<>(PyString_FromStringAndSize((char*)jpg->data(), jpg->size())));
}

boost::python::object convert_original(char* data, long size) {
	sk_sp<SkData> skdata(SkData::MakeWithoutCopy((void*)data, size));
	sk_sp<SkImage> img(SkImage::MakeFromEncoded(skdata));
	if(!img) return boost::python::object();
	sk_sp<SkData> jpg(img->encode(SkEncodedImageFormat::kJPEG, IMAGE_QUALITY));
	if(!jpg) return boost::python::object();

	return boost::python::object(boost::python::handle<>(PyString_FromStringAndSize((char*)jpg->data(), jpg->size())));
}

boost::python::object convert_list(char* data, long size) {
	SkBitmap bm;
	sk_sp<SkData> skdata(SkData::MakeWithoutCopy((void*)data, size));
	if(!data) return boost::python::object();
	std::unique_ptr<SkImageGenerator> gen = SkImageGenerator::MakeFromEncoded(skdata);
	if(!gen) return boost::python::object();

	SkImageInfo bmInfo = gen->getInfo();
	gen->tryGenerateBitmap(&bm, bmInfo, nullptr);

	float width = 720.0f;
	float height = 472.0f;
	float top = (float)bm.height() / 2.0f - height / 2.0f;
	float bottom = top + height;

	SkPaint paint;
	paint.setAntiAlias(true);
	paint.setStyle(SkPaint::kFill_Style);
	paint.setFilterQuality(kHigh_SkFilterQuality);

	sk_sp<SkSurface> surface(SkSurface::MakeRaster(SkImageInfo::Make(width, height, SIPSKIA_COLORTYPE, kPremul_SkAlphaType, nullptr)).release());
	SkCanvas* canvas = surface->getCanvas();
	canvas->drawBitmapRect(bm, SkRect::MakeLTRB(0, top, width, bottom), SkRect::MakeXYWH(0.0f, 0.0f, width, height), &paint);
	sk_sp<SkImage> img(surface->makeImageSnapshot());
	if(!img) return boost::python::object();
	sk_sp<SkData> jpg(img->encode(SkEncodedImageFormat::kJPEG, IMAGE_QUALITY));
	if(!jpg) return boost::python::object();

	return boost::python::object(boost::python::handle<>(PyString_FromStringAndSize((char*)jpg->data(), jpg->size())));
}

boost::python::object convert_reply(char* data, long size) {
	SkBitmap bm;
	sk_sp<SkData> skdata(SkData::MakeWithoutCopy((void*)data, size));
	if(!data) return boost::python::object();
	std::unique_ptr<SkImageGenerator> gen = SkImageGenerator::MakeFromEncoded(skdata);
	if(!gen) return boost::python::object();

	SkImageInfo bmInfo = gen->getInfo();
	gen->tryGenerateBitmap(&bm, bmInfo, nullptr);

	float width = 720.0f;
	float height = 472.0f;
	float top = (float)bm.height()/2.0f - height/2.0f;
	float bottom = top + height;

	float canvas_width = 460.0f;
	float canvas_height = 310.0f;

	SkPaint paint;
	paint.setAntiAlias(true);
	paint.setStyle(SkPaint::kFill_Style);
	paint.setFilterQuality(kHigh_SkFilterQuality);

	sk_sp<SkSurface> surface(SkSurface::MakeRaster(SkImageInfo::Make(canvas_width, canvas_height, SIPSKIA_COLORTYPE, kPremul_SkAlphaType, nullptr)).release());
	SkCanvas* canvas = surface->getCanvas();
	canvas->drawBitmapRect(bm, SkRect::MakeLTRB(0, top, width, bottom), SkRect::MakeXYWH(0.0f, 0.0f, canvas_width, canvas_height), &paint);
	sk_sp<SkImage> img(surface->makeImageSnapshot());
	if(!img) return boost::python::object();
	sk_sp<SkData> jpg(img->encode(SkEncodedImageFormat::kJPEG, IMAGE_QUALITY));
	if(!jpg) return boost::python::object();

	return boost::python::object(boost::python::handle<>(PyString_FromStringAndSize((char*)jpg->data(), jpg->size())));
}

boost::python::object convert_medium(char* data, long size) {
	SkBitmap bm;
	sk_sp<SkData> skdata(SkData::MakeWithoutCopy((void*)data, size));
	if(!data) return boost::python::object();
	std::unique_ptr<SkImageGenerator> gen = SkImageGenerator::MakeFromEncoded(skdata);
	if(!gen) return boost::python::object();

	SkImageInfo bmInfo = gen->getInfo();
	gen->tryGenerateBitmap(&bm, bmInfo, nullptr);

	float width = 720.0f;
	float height = 720.0f;
	float top = (float)bm.height()/2.0f - height/2.0f;
	float bottom = top + height;

	float canvas_width = 360.0f;
	float canvas_height = 360.0f;

	SkPaint paint;
	paint.setAntiAlias(true);
	paint.setStyle(SkPaint::kFill_Style);
	paint.setFilterQuality(kHigh_SkFilterQuality);

	sk_sp<SkSurface> surface(SkSurface::MakeRaster(SkImageInfo::Make(canvas_width, canvas_height, SIPSKIA_COLORTYPE, kPremul_SkAlphaType, nullptr)).release());
	SkCanvas* canvas = surface->getCanvas();
	canvas->drawBitmapRect(bm, SkRect::MakeLTRB(0, top, width, bottom), SkRect::MakeXYWH(0.0f, 0.0f, canvas_width, canvas_height), &paint);
	sk_sp<SkImage> img(surface->makeImageSnapshot());
	if(!img) return boost::python::object();
	sk_sp<SkData> jpg(img->encode(SkEncodedImageFormat::kJPEG, IMAGE_QUALITY));
	if(!jpg) return boost::python::object();

	return boost::python::object(boost::python::handle<>(PyString_FromStringAndSize((char*)jpg->data(), jpg->size())));
}

boost::python::object convert_thumbnail(char* data, long size) {
	SkBitmap bm;
	sk_sp<SkData> skdata(SkData::MakeWithoutCopy((void*)data, size));
	if(!data) return boost::python::object();
	std::unique_ptr<SkImageGenerator> gen = SkImageGenerator::MakeFromEncoded(skdata);
	if(!gen) return boost::python::object();

	SkImageInfo bmInfo = gen->getInfo();
	gen->tryGenerateBitmap(&bm, bmInfo, nullptr);

	SkPaint paint;
	paint.setAntiAlias(true);
	paint.setStyle(SkPaint::kFill_Style);
	paint.setFilterQuality(kHigh_SkFilterQuality);

	float canvas_size = 132.0f;
	float origin_size = 720.0f;
	float top = bm.height() / 2.0f - origin_size / 2.0f;
	float bottom = top + origin_size;

	sk_sp<SkSurface> surface(SkSurface::MakeRaster(SkImageInfo::Make(canvas_size, canvas_size, SIPSKIA_COLORTYPE, kPremul_SkAlphaType, nullptr)).release());
	SkCanvas* canvas = surface->getCanvas();
	canvas->drawBitmapRect(bm, SkRect::MakeLTRB(0, top, origin_size, bottom), SkRect::MakeXYWH(0.0f, 0.0f, canvas_size, canvas_size), &paint);
	sk_sp<SkImage> img(surface->makeImageSnapshot());
	if(!img) return boost::python::object();
	sk_sp<SkData> jpg(img->encode(SkEncodedImageFormat::kJPEG, IMAGE_QUALITY));
	if(!jpg) return boost::python::object();

	return boost::python::object(boost::python::handle<>(PyString_FromStringAndSize((char*)jpg->data(), jpg->size())));
}

using namespace boost::python;
BOOST_PYTHON_MODULE(sipskia)
{
	def("convert_webp", &convert_webp);
	def("convert_original", &convert_original);
	def("convert_list", &convert_list);
	def("convert_reply", &convert_reply);
	def("convert_medium", &convert_medium);
	def("convert_thumbnail", &convert_thumbnail);
}

<commit_msg>webp 변환은 최대 퀄리티로 변환하도록 수정<commit_after>#include "SkCanvas.h"
#include "SkCommandLineFlags.h"
#include "SkData.h"
#include "SkDocument.h"
#include "SkGraphics.h"
#include "SkSurface.h"
#include "SkImage.h"
#include "SkImageFilter.h"
#include "SkStream.h"
#include "SkString.h"
#include "SkPaint.h"
#include "SkImageGenerator.h"

#include <boost/python.hpp>
#include <stdexcept>

#define IMAGE_QUALITY 70

#ifdef SIPSKIA_BGRA
#define SIPSKIA_COLORTYPE kBGRA_8888_SkColorType
#else
#define SIPSKIA_COLORTYPE kN32_SkColorType
#endif

boost::python::object convert_webp(char* data, long size, bool is_gallery_card) {
	SkBitmap bm;
	sk_sp<SkData> skdata(SkData::MakeWithoutCopy((void*)data, size));
	if(!data) return boost::python::object();
	std::unique_ptr<SkImageGenerator> gen = SkImageGenerator::MakeFromEncoded(skdata);
	if(!gen) return boost::python::object();

	SkImageInfo bmInfo = gen->getInfo();
	gen->tryGenerateBitmap(&bm, bmInfo, nullptr);

	float width = 720.0f;
	float height = 720.0f;
	if(is_gallery_card){
		height = 1120.0f;
	}

	SkPaint paint;
	paint.setAntiAlias(true);
	paint.setStyle(SkPaint::kFill_Style);
	paint.setFilterQuality(kHigh_SkFilterQuality);

	sk_sp<SkSurface> surface(SkSurface::MakeRaster(SkImageInfo::Make(width, height, SIPSKIA_COLORTYPE, kPremul_SkAlphaType, nullptr)).release());
	SkCanvas* canvas = surface->getCanvas();
	canvas->drawBitmapRect(bm, SkRect::MakeLTRB(0, 0, width, height), SkRect::MakeXYWH(0.0f, 0.0f, width, height), &paint);
	sk_sp<SkImage> img(surface->makeImageSnapshot());
	if(!img) return boost::python::object();
	sk_sp<SkData> jpg(img->encode(SkEncodedImageFormat::kWEBP, 100));
	if(!jpg) return boost::python::object();

	return boost::python::object(boost::python::handle<>(PyString_FromStringAndSize((char*)jpg->data(), jpg->size())));
}

boost::python::object convert_original(char* data, long size) {
	sk_sp<SkData> skdata(SkData::MakeWithoutCopy((void*)data, size));
	sk_sp<SkImage> img(SkImage::MakeFromEncoded(skdata));
	if(!img) return boost::python::object();
	sk_sp<SkData> jpg(img->encode(SkEncodedImageFormat::kJPEG, IMAGE_QUALITY));
	if(!jpg) return boost::python::object();

	return boost::python::object(boost::python::handle<>(PyString_FromStringAndSize((char*)jpg->data(), jpg->size())));
}

boost::python::object convert_list(char* data, long size) {
	SkBitmap bm;
	sk_sp<SkData> skdata(SkData::MakeWithoutCopy((void*)data, size));
	if(!data) return boost::python::object();
	std::unique_ptr<SkImageGenerator> gen = SkImageGenerator::MakeFromEncoded(skdata);
	if(!gen) return boost::python::object();

	SkImageInfo bmInfo = gen->getInfo();
	gen->tryGenerateBitmap(&bm, bmInfo, nullptr);

	float width = 720.0f;
	float height = 472.0f;
	float top = (float)bm.height() / 2.0f - height / 2.0f;
	float bottom = top + height;

	SkPaint paint;
	paint.setAntiAlias(true);
	paint.setStyle(SkPaint::kFill_Style);
	paint.setFilterQuality(kHigh_SkFilterQuality);

	sk_sp<SkSurface> surface(SkSurface::MakeRaster(SkImageInfo::Make(width, height, SIPSKIA_COLORTYPE, kPremul_SkAlphaType, nullptr)).release());
	SkCanvas* canvas = surface->getCanvas();
	canvas->drawBitmapRect(bm, SkRect::MakeLTRB(0, top, width, bottom), SkRect::MakeXYWH(0.0f, 0.0f, width, height), &paint);
	sk_sp<SkImage> img(surface->makeImageSnapshot());
	if(!img) return boost::python::object();
	sk_sp<SkData> jpg(img->encode(SkEncodedImageFormat::kJPEG, IMAGE_QUALITY));
	if(!jpg) return boost::python::object();

	return boost::python::object(boost::python::handle<>(PyString_FromStringAndSize((char*)jpg->data(), jpg->size())));
}

boost::python::object convert_reply(char* data, long size) {
	SkBitmap bm;
	sk_sp<SkData> skdata(SkData::MakeWithoutCopy((void*)data, size));
	if(!data) return boost::python::object();
	std::unique_ptr<SkImageGenerator> gen = SkImageGenerator::MakeFromEncoded(skdata);
	if(!gen) return boost::python::object();

	SkImageInfo bmInfo = gen->getInfo();
	gen->tryGenerateBitmap(&bm, bmInfo, nullptr);

	float width = 720.0f;
	float height = 472.0f;
	float top = (float)bm.height()/2.0f - height/2.0f;
	float bottom = top + height;

	float canvas_width = 460.0f;
	float canvas_height = 310.0f;

	SkPaint paint;
	paint.setAntiAlias(true);
	paint.setStyle(SkPaint::kFill_Style);
	paint.setFilterQuality(kHigh_SkFilterQuality);

	sk_sp<SkSurface> surface(SkSurface::MakeRaster(SkImageInfo::Make(canvas_width, canvas_height, SIPSKIA_COLORTYPE, kPremul_SkAlphaType, nullptr)).release());
	SkCanvas* canvas = surface->getCanvas();
	canvas->drawBitmapRect(bm, SkRect::MakeLTRB(0, top, width, bottom), SkRect::MakeXYWH(0.0f, 0.0f, canvas_width, canvas_height), &paint);
	sk_sp<SkImage> img(surface->makeImageSnapshot());
	if(!img) return boost::python::object();
	sk_sp<SkData> jpg(img->encode(SkEncodedImageFormat::kJPEG, IMAGE_QUALITY));
	if(!jpg) return boost::python::object();

	return boost::python::object(boost::python::handle<>(PyString_FromStringAndSize((char*)jpg->data(), jpg->size())));
}

boost::python::object convert_medium(char* data, long size) {
	SkBitmap bm;
	sk_sp<SkData> skdata(SkData::MakeWithoutCopy((void*)data, size));
	if(!data) return boost::python::object();
	std::unique_ptr<SkImageGenerator> gen = SkImageGenerator::MakeFromEncoded(skdata);
	if(!gen) return boost::python::object();

	SkImageInfo bmInfo = gen->getInfo();
	gen->tryGenerateBitmap(&bm, bmInfo, nullptr);

	float width = 720.0f;
	float height = 720.0f;
	float top = (float)bm.height()/2.0f - height/2.0f;
	float bottom = top + height;

	float canvas_width = 360.0f;
	float canvas_height = 360.0f;

	SkPaint paint;
	paint.setAntiAlias(true);
	paint.setStyle(SkPaint::kFill_Style);
	paint.setFilterQuality(kHigh_SkFilterQuality);

	sk_sp<SkSurface> surface(SkSurface::MakeRaster(SkImageInfo::Make(canvas_width, canvas_height, SIPSKIA_COLORTYPE, kPremul_SkAlphaType, nullptr)).release());
	SkCanvas* canvas = surface->getCanvas();
	canvas->drawBitmapRect(bm, SkRect::MakeLTRB(0, top, width, bottom), SkRect::MakeXYWH(0.0f, 0.0f, canvas_width, canvas_height), &paint);
	sk_sp<SkImage> img(surface->makeImageSnapshot());
	if(!img) return boost::python::object();
	sk_sp<SkData> jpg(img->encode(SkEncodedImageFormat::kJPEG, IMAGE_QUALITY));
	if(!jpg) return boost::python::object();

	return boost::python::object(boost::python::handle<>(PyString_FromStringAndSize((char*)jpg->data(), jpg->size())));
}

boost::python::object convert_thumbnail(char* data, long size) {
	SkBitmap bm;
	sk_sp<SkData> skdata(SkData::MakeWithoutCopy((void*)data, size));
	if(!data) return boost::python::object();
	std::unique_ptr<SkImageGenerator> gen = SkImageGenerator::MakeFromEncoded(skdata);
	if(!gen) return boost::python::object();

	SkImageInfo bmInfo = gen->getInfo();
	gen->tryGenerateBitmap(&bm, bmInfo, nullptr);

	SkPaint paint;
	paint.setAntiAlias(true);
	paint.setStyle(SkPaint::kFill_Style);
	paint.setFilterQuality(kHigh_SkFilterQuality);

	float canvas_size = 132.0f;
	float origin_size = 720.0f;
	float top = bm.height() / 2.0f - origin_size / 2.0f;
	float bottom = top + origin_size;

	sk_sp<SkSurface> surface(SkSurface::MakeRaster(SkImageInfo::Make(canvas_size, canvas_size, SIPSKIA_COLORTYPE, kPremul_SkAlphaType, nullptr)).release());
	SkCanvas* canvas = surface->getCanvas();
	canvas->drawBitmapRect(bm, SkRect::MakeLTRB(0, top, origin_size, bottom), SkRect::MakeXYWH(0.0f, 0.0f, canvas_size, canvas_size), &paint);
	sk_sp<SkImage> img(surface->makeImageSnapshot());
	if(!img) return boost::python::object();
	sk_sp<SkData> jpg(img->encode(SkEncodedImageFormat::kJPEG, IMAGE_QUALITY));
	if(!jpg) return boost::python::object();

	return boost::python::object(boost::python::handle<>(PyString_FromStringAndSize((char*)jpg->data(), jpg->size())));
}

using namespace boost::python;
BOOST_PYTHON_MODULE(sipskia)
{
	def("convert_webp", &convert_webp);
	def("convert_original", &convert_original);
	def("convert_list", &convert_list);
	def("convert_reply", &convert_reply);
	def("convert_medium", &convert_medium);
	def("convert_thumbnail", &convert_thumbnail);
}

<|endoftext|>


<commit_before>#include "dialog.h"
#include "ui_dialog.h"
#include <QDebug>
#include <qsysteminfo.h>

Dialog::Dialog(QWidget *parent) :
    QDialog(parent),
    ui(new Ui::Dialog)
{
    ui->setupUi(this);
    setupGeneral();
    setupDevice();
    setupDisplay();
    setupMemory();
    setupNetwork();
    setupSaver();

    connect(ui->tabWidget,SIGNAL(currentChanged(int)),this,SLOT(tabChanged(int)));
   connect(ui->versionComboBox,SIGNAL(activated(int)), this,SLOT(getVersion(int)));
   connect(ui->featureComboBox,SIGNAL(activated(int)), this,SLOT(getFeature(int)));

}

Dialog::~Dialog()
{
    delete ui;
}

void Dialog::changeEvent(QEvent *e)
{
    QDialog::changeEvent(e);
    switch (e->type()) {
    case QEvent::LanguageChange:
        ui->retranslateUi(this);
        break;
    default:
        break;
    }
}

void Dialog::tabChanged(int index)
{
    switch(index) {
    case 0:
        setupGeneral();
        break;
    case 1:
        setupDevice();
        break;
    case 2:
        setupDisplay();
        break;
    case 3:
        setupMemory();
        break;
    case 4:
        setupNetwork();
        break;
        case 5:
        setupSaver();
        break;
    };
}

void Dialog::setupGeneral()
{
    QSystemInfo si;
    ui->curLanguageLineEdit->setText( si.currentLanguage());
    ui->languagesComboBox->insertItems(0,si.availableLanguages());
}

void Dialog::setupDevice()
{
    QSystemDeviceInfo di;
    ui->batteryLevelBar->setValue(di.batteryLevel());
    ui->batteryChargingCheckBox->setChecked(di.isBatteryCharging());

    ui->ImeiLabel->setText(di.imei());
    ui->imsiLabel->setText(di.imsi());
    ui->manufacturerLabel->setText(di.manufacturer());
    ui->modelLabel->setText(di.model());
    ui->productLabel->setText(di.productName());

    ui->deviceLockCheckBox->setChecked(di.isDeviceLocked());

    if(di.currentPowerState() == QSystemDeviceInfo::BatteryPower) {
        ui->radioButton_2->setChecked(true);
    } else
        if(di.currentPowerState() == QSystemDeviceInfo::WallPower) {
        ui->radioButton_3->setChecked(true);
    } else {
        ui->radioButton->setChecked(true);
    }
}

void Dialog::setupDisplay()
{
    QSystemDisplayInfo di;
    ui->brightnessLineEdit->setText(QString::number(di.displayBrightness(0)));
    ui->colorDepthLineEdit->setText(QString::number(di.colorDepth((0))));

    ui->screenLockCheckBox->setChecked(di.isScreenLockOn());
}

void Dialog::setupMemory()
{
    QSystemMemoryInfo mi;
    ui->memoryTreeWidget->clear();
    QStringList vols = mi.listOfVolumes();
    foreach(QString volName, vols) {
        QString type;
        QSystemMemoryInfo::VolumeType volType;
        volType = mi.getVolumeType(volName);
        if(volType == QSystemMemoryInfo::Internal) {
            type =  "Internal";
        } else
            if(volType == QSystemMemoryInfo::Removable) {
            type = "Removable";
        }
        if(volType == QSystemMemoryInfo::Cdrom) {
            type =  "Cdrom";
        }
        QStringList items;
        items << volName;
        items << type;
        items << QString::number(mi.totalDiskSpace(volName));
        items << QString::number(mi.availableDiskSpace(volName));
        QTreeWidgetItem *item = new QTreeWidgetItem(items);
        ui->memoryTreeWidget->addTopLevelItem(item);
    }
}

void Dialog::setupNetwork()
{

}

void Dialog::getVersion(int index)
{
    QSystemInfo::Version version;
    switch(index) {
    case 0:
        version = QSystemInfo::Os;
        break;
    case 1:
        version = QSystemInfo::QtCore;
        break;
    case 2:
        version = QSystemInfo::WrtCore;
        break;
    case 3:
        version = QSystemInfo::Webkit;
        break;
    case 4:
        version = QSystemInfo::ServiceFramework;
        break;
    case 5:
        version = QSystemInfo::WrtExtensions;
        break;
    case 6:
        version = QSystemInfo::ServiceProvider;
        break;
    case 7:
        version = QSystemInfo::NetscapePlugin;
        break;
    case 8:
        version = QSystemInfo::WebApp;
        break;
    case 9:
        version = QSystemInfo::Firmware;
        break;
    };

    QSystemInfo si;
    ui->versionLineEdit->setText(si.getVersion(version));
}

void Dialog::getFeature(int index)
{
    QSystemInfo::Feature feature;
    switch(index) {
    case 0:
        feature = QSystemInfo::BluetoothFeature;
        break;
    case 1:
        feature = QSystemInfo::CameraFeature;
        break;
    case 2:
        feature = QSystemInfo::FmradioFeature;
        break;
    case 3:
        feature = QSystemInfo::IrFeature;
        break;
    case 4:
        feature = QSystemInfo::LedFeature;
        break;
    case 5:
        feature = QSystemInfo::MemcardFeature;
        break;
    case 6:
        feature = QSystemInfo::UsbFeature;
        break;
    case 7:
        feature = QSystemInfo::VibFeature;
        break;
    case 8:
        feature = QSystemInfo::WlanFeature;
        break;
    case 9:
        feature = QSystemInfo::SimFeature;
        break;
    case 10:
        feature = QSystemInfo::LocationFeature;
        break;
    case 11:
        feature = QSystemInfo::VideoOutFeature;
        break;
    case 12:
        feature = QSystemInfo::HapticsFeature;
        break;
    };
    QSystemInfo si;
    ui->featuresLineEdit->setText((si.hasFeatureSupported(feature) ? "true":"false" ));
}

//void Dialog::doVolumes(int /*index*/)
//{
//    QSystemMemoryInfo mi;
//    QString vol = ui->volumesComboBox->currentText();
//    int index2 = ui->diskComboBox->currentIndex();
//    switch(index2) {
//    case 0:
//        //total
//        ui->diskSpaceLineEdit->setText( QString::number(mi.totalDiskSpace(vol)));
//        break;
//        case 1:
//        //available
//        ui->diskSpaceLineEdit->setText( QString::number(mi.availableDiskSpace(vol)));
//        break;
//        case 2:
//        //type
//        QSystemMemoryInfo::VolumeType volType;
//        volType = mi.getVolumeType(vol);
//        if(volType == QSystemMemoryInfo::Internal) {
//                ui->diskSpaceLineEdit->setText( "Internal");
//        } else
//        if(volType == QSystemMemoryInfo::Removable) {
//                ui->diskSpaceLineEdit->setText( "Removable");
//        }
//        if(volType == QSystemMemoryInfo::Cdrom) {
//                ui->diskSpaceLineEdit->setText( "Cdrom");
//        }
//        break;
//    };
//
//}

void Dialog::setupSaver()
{
    saver = new QSystemScreenSaver();
    bool saverEnabled = saver->screenSaverEnabled();
    bool blankingEnabled = saver->screenBlankingEnabled();

    ui->saverEnabledPushButton->setChecked(saverEnabled);
    ui->blankingEnabledPushButton->setChecked(blankingEnabled);

    connect( ui->saverEnabledPushButton, SIGNAL(clicked(bool)),
             this,SLOT(setSaverEnabled(bool)));
    connect( ui->blankingEnabledPushButton, SIGNAL(clicked(bool)),
             this,SLOT(setBlankingEnabled(bool)));
}

void Dialog::setSaverEnabled(bool b)
{
    if(!saver->setScreenSaverEnabled(b))
        ui->saverEnabledPushButton->setChecked(!b);
}

void Dialog::setBlankingEnabled(bool b)
{
    if(!saver->setScreenBlankingEnabled(b))
        ui->blankingEnabledPushButton->setChecked(!b);
}
<commit_msg>make treewidget headers expand<commit_after>#include "dialog.h"
#include "ui_dialog.h"
#include <QDebug>
#include <qsysteminfo.h>
#include <QTreeWidgetItem>

Dialog::Dialog(QWidget *parent) :
    QDialog(parent),
    ui(new Ui::Dialog)
{
    ui->setupUi(this);
    setupGeneral();
    setupDevice();
    setupDisplay();
    setupMemory();
    setupNetwork();
    setupSaver();

    connect(ui->tabWidget,SIGNAL(currentChanged(int)),this,SLOT(tabChanged(int)));
   connect(ui->versionComboBox,SIGNAL(activated(int)), this,SLOT(getVersion(int)));
   connect(ui->featureComboBox,SIGNAL(activated(int)), this,SLOT(getFeature(int)));

}

Dialog::~Dialog()
{
    delete ui;
}

void Dialog::changeEvent(QEvent *e)
{
    QDialog::changeEvent(e);
    switch (e->type()) {
    case QEvent::LanguageChange:
        ui->retranslateUi(this);
        break;
    default:
        break;
    }
}

void Dialog::tabChanged(int index)
{
    switch(index) {
    case 0:
        setupGeneral();
        break;
    case 1:
        setupDevice();
        break;
    case 2:
        setupDisplay();
        break;
    case 3:
        setupMemory();
        break;
    case 4:
        setupNetwork();
        break;
        case 5:
        setupSaver();
        break;
    };
}

void Dialog::setupGeneral()
{
    QSystemInfo si;
    ui->curLanguageLineEdit->setText( si.currentLanguage());
    ui->languagesComboBox->insertItems(0,si.availableLanguages());
}

void Dialog::setupDevice()
{
    QSystemDeviceInfo di;
    ui->batteryLevelBar->setValue(di.batteryLevel());
    ui->batteryChargingCheckBox->setChecked(di.isBatteryCharging());

    ui->ImeiLabel->setText(di.imei());
    ui->imsiLabel->setText(di.imsi());
    ui->manufacturerLabel->setText(di.manufacturer());
    ui->modelLabel->setText(di.model());
    ui->productLabel->setText(di.productName());

    ui->deviceLockCheckBox->setChecked(di.isDeviceLocked());

    if(di.currentPowerState() == QSystemDeviceInfo::BatteryPower) {
        ui->radioButton_2->setChecked(true);
    } else
        if(di.currentPowerState() == QSystemDeviceInfo::WallPower) {
        ui->radioButton_3->setChecked(true);
    } else {
        ui->radioButton->setChecked(true);
    }
}

void Dialog::setupDisplay()
{
    QSystemDisplayInfo di;
    ui->brightnessLineEdit->setText(QString::number(di.displayBrightness(0)));
    ui->colorDepthLineEdit->setText(QString::number(di.colorDepth((0))));

    ui->screenLockCheckBox->setChecked(di.isScreenLockOn());
}

void Dialog::setupMemory()
{
    QSystemMemoryInfo mi;
    ui->memoryTreeWidget->clear();
    ui->memoryTreeWidget->header()->setResizeMode(QHeaderView::ResizeToContents);
    //ui->memoryTreeWidget->
    QStringList vols = mi.listOfVolumes();
    foreach(QString volName, vols) {
        QString type;
        QSystemMemoryInfo::VolumeType volType;
        volType = mi.getVolumeType(volName);
        if(volType == QSystemMemoryInfo::Internal) {
            type =  "Internal";
        } else
            if(volType == QSystemMemoryInfo::Removable) {
            type = "Removable";
        }
        if(volType == QSystemMemoryInfo::Cdrom) {
            type =  "Cdrom";
        }
        QStringList items;
        items << volName;
        items << type;
        items << QString::number(mi.totalDiskSpace(volName));
        items << QString::number(mi.availableDiskSpace(volName));
        QTreeWidgetItem *item = new QTreeWidgetItem(items);
        ui->memoryTreeWidget->addTopLevelItem(item);
    }
}

void Dialog::setupNetwork()
{

}

void Dialog::getVersion(int index)
{
    QSystemInfo::Version version;
    switch(index) {
    case 0:
        version = QSystemInfo::Os;
        break;
    case 1:
        version = QSystemInfo::QtCore;
        break;
    case 2:
        version = QSystemInfo::WrtCore;
        break;
    case 3:
        version = QSystemInfo::Webkit;
        break;
    case 4:
        version = QSystemInfo::ServiceFramework;
        break;
    case 5:
        version = QSystemInfo::WrtExtensions;
        break;
    case 6:
        version = QSystemInfo::ServiceProvider;
        break;
    case 7:
        version = QSystemInfo::NetscapePlugin;
        break;
    case 8:
        version = QSystemInfo::WebApp;
        break;
    case 9:
        version = QSystemInfo::Firmware;
        break;
    };

    QSystemInfo si;
    ui->versionLineEdit->setText(si.getVersion(version));
}

void Dialog::getFeature(int index)
{
    QSystemInfo::Feature feature;
    switch(index) {
    case 0:
        feature = QSystemInfo::BluetoothFeature;
        break;
    case 1:
        feature = QSystemInfo::CameraFeature;
        break;
    case 2:
        feature = QSystemInfo::FmradioFeature;
        break;
    case 3:
        feature = QSystemInfo::IrFeature;
        break;
    case 4:
        feature = QSystemInfo::LedFeature;
        break;
    case 5:
        feature = QSystemInfo::MemcardFeature;
        break;
    case 6:
        feature = QSystemInfo::UsbFeature;
        break;
    case 7:
        feature = QSystemInfo::VibFeature;
        break;
    case 8:
        feature = QSystemInfo::WlanFeature;
        break;
    case 9:
        feature = QSystemInfo::SimFeature;
        break;
    case 10:
        feature = QSystemInfo::LocationFeature;
        break;
    case 11:
        feature = QSystemInfo::VideoOutFeature;
        break;
    case 12:
        feature = QSystemInfo::HapticsFeature;
        break;
    };
    QSystemInfo si;
    ui->featuresLineEdit->setText((si.hasFeatureSupported(feature) ? "true":"false" ));
}

//void Dialog::doVolumes(int /*index*/)
//{
//    QSystemMemoryInfo mi;
//    QString vol = ui->volumesComboBox->currentText();
//    int index2 = ui->diskComboBox->currentIndex();
//    switch(index2) {
//    case 0:
//        //total
//        ui->diskSpaceLineEdit->setText( QString::number(mi.totalDiskSpace(vol)));
//        break;
//        case 1:
//        //available
//        ui->diskSpaceLineEdit->setText( QString::number(mi.availableDiskSpace(vol)));
//        break;
//        case 2:
//        //type
//        QSystemMemoryInfo::VolumeType volType;
//        volType = mi.getVolumeType(vol);
//        if(volType == QSystemMemoryInfo::Internal) {
//                ui->diskSpaceLineEdit->setText( "Internal");
//        } else
//        if(volType == QSystemMemoryInfo::Removable) {
//                ui->diskSpaceLineEdit->setText( "Removable");
//        }
//        if(volType == QSystemMemoryInfo::Cdrom) {
//                ui->diskSpaceLineEdit->setText( "Cdrom");
//        }
//        break;
//    };
//
//}

void Dialog::setupSaver()
{
    saver = new QSystemScreenSaver();
    bool saverEnabled = saver->screenSaverEnabled();
    bool blankingEnabled = saver->screenBlankingEnabled();

    ui->saverEnabledPushButton->setChecked(saverEnabled);
    ui->blankingEnabledPushButton->setChecked(blankingEnabled);

    connect( ui->saverEnabledPushButton, SIGNAL(clicked(bool)),
             this,SLOT(setSaverEnabled(bool)));
    connect( ui->blankingEnabledPushButton, SIGNAL(clicked(bool)),
             this,SLOT(setBlankingEnabled(bool)));
}

void Dialog::setSaverEnabled(bool b)
{
    if(!saver->setScreenSaverEnabled(b))
        ui->saverEnabledPushButton->setChecked(!b);
}

void Dialog::setBlankingEnabled(bool b)
{
    if(!saver->setScreenBlankingEnabled(b))
        ui->blankingEnabledPushButton->setChecked(!b);
}
<|endoftext|>