12SN/A/*
25522Snate@binkert.org * Copyright (c) 2008 The Hewlett-Packard Development Company
32SN/A * All rights reserved.
42SN/A *
52SN/A * Redistribution and use in source and binary forms, with or without
62SN/A * modification, are permitted provided that the following conditions are
72SN/A * met: redistributions of source code must retain the above copyright
82SN/A * notice, this list of conditions and the following disclaimer;
92SN/A * redistributions in binary form must reproduce the above copyright
102SN/A * notice, this list of conditions and the following disclaimer in the
112SN/A * documentation and/or other materials provided with the distribution;
122SN/A * neither the name of the copyright holders nor the names of its
132SN/A * contributors may be used to endorse or promote products derived from
142SN/A * this software without specific prior written permission.
152SN/A *
162SN/A * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
172SN/A * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
182SN/A * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
192SN/A * A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
202SN/A * OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
212SN/A * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
222SN/A * LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
232SN/A * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
242SN/A * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
252SN/A * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
262SN/A * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
272665Ssaidi@eecs.umich.edu *
284123Sbinkertn@umich.edu * Authors: Nathan Binkert
292SN/A */
302SN/A
314123Sbinkertn@umich.edu#include <Python.h>
322655Sstever@eecs.umich.edu
335522Snate@binkert.org#include "sim/init.hh"
3410453SAndrew.Bardsley@arm.com#include "sim/init_signals.hh"
352SN/A
365522Snate@binkert.org// main() is now pretty stripped down and just sets up python and then
375522Snate@binkert.org// calls initM5Python which loads the various embedded python modules
385522Snate@binkert.org// into the python environment and then starts things running by
395522Snate@binkert.org// calling m5Main.
404941Snate@binkert.orgint
412SN/Amain(int argc, char **argv)
422SN/A{
435522Snate@binkert.org    int ret;
445522Snate@binkert.org
455522Snate@binkert.org    // Initialize m5 special signal handling.
465522Snate@binkert.org    initSignals();
472SN/A
4813620Sandreas.sandberg@arm.com#if PY_MAJOR_VERSION >= 3
4913620Sandreas.sandberg@arm.com    std::unique_ptr<wchar_t[], decltype(&PyMem_RawFree)> program(
5013620Sandreas.sandberg@arm.com        Py_DecodeLocale(argv[0], NULL),
5113620Sandreas.sandberg@arm.com        &PyMem_RawFree);
5213620Sandreas.sandberg@arm.com    Py_SetProgramName(program.get());
5313620Sandreas.sandberg@arm.com#else
542655Sstever@eecs.umich.edu    Py_SetProgramName(argv[0]);
5513620Sandreas.sandberg@arm.com#endif
562667Sstever@eecs.umich.edu
5713662Sandreas.sandberg@arm.com    // Register native modules with Python's init system before
5813662Sandreas.sandberg@arm.com    // initializing the interpreter.
5913662Sandreas.sandberg@arm.com    registerNativeModules();
6013662Sandreas.sandberg@arm.com
612667Sstever@eecs.umich.edu    // initialize embedded Python interpreter
622655Sstever@eecs.umich.edu    Py_Initialize();
631311SN/A
645522Snate@binkert.org    // Initialize the embedded m5 python library
6513662Sandreas.sandberg@arm.com    ret = EmbeddedPython::initAll();
661703SN/A
675522Snate@binkert.org    if (ret == 0) {
685522Snate@binkert.org        // start m5
695522Snate@binkert.org        ret = m5Main(argc, argv);
705522Snate@binkert.org    }
712667Sstever@eecs.umich.edu
722667Sstever@eecs.umich.edu    // clean up Python intepreter.
732655Sstever@eecs.umich.edu    Py_Finalize();
744941Snate@binkert.org
754941Snate@binkert.org    return ret;
762667Sstever@eecs.umich.edu}
77