# Copyright (c) 2005-2007 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.

# Splash2 Run Script
#

import argparse
import os
import sys

import m5
from m5.objects import *

# --------------------
# Define Command Line Options
# ====================

parser = argparse.ArgumentParser()

parser.add_argument("-d", "--detailed", action="store_true")
parser.add_argument("-t", "--timing", action="store_true")
parser.add_argument("-m", "--maxtick", type=int)
parser.add_argument(
    "-n", "--numcpus", help="Number of cpus in total", type=int
)
parser.add_argument(
    "-f", "--frequency", default="1GHz", help="Frequency of each CPU"
)
parser.add_argument("--l1size", default="32KiB")
parser.add_argument("--l1latency", default="1ns")
parser.add_argument("--l2size", default="256KiB")
parser.add_argument("--l2latency", default="10ns")
parser.add_argument(
    "--rootdir",
    help="Root directory of Splash2",
    default="/dist/splash2/codes",
)
parser.add_argument("-b", "--benchmark", help="Splash 2 benchmark to run")

args = parser.parse_args()

if not args.numcpus:
    print("Specify the number of cpus with -n")
    sys.exit(1)


# --------------------
# Define Splash2 Benchmarks
# ====================
class Cholesky(Process):
    cwd = args.rootdir + "/kernels/cholesky"
    executable = args.rootdir + "/kernels/cholesky/CHOLESKY"
    cmd = [
        "CHOLESKY",
        "-p" + str(args.numcpus),
        args.rootdir + "/kernels/cholesky/inputs/tk23.O",
    ]


class FFT(Process):
    cwd = args.rootdir + "/kernels/fft"
    executable = args.rootdir + "/kernels/fft/FFT"
    cmd = ["FFT", "-p", str(args.numcpus), "-m18"]


class LU_contig(Process):
    executable = args.rootdir + "/kernels/lu/contiguous_blocks/LU"
    cmd = ["LU", "-p", str(args.numcpus)]
    cwd = args.rootdir + "/kernels/lu/contiguous_blocks"


class LU_noncontig(Process):
    executable = args.rootdir + "/kernels/lu/non_contiguous_blocks/LU"
    cmd = ["LU", "-p", str(args.numcpus)]
    cwd = args.rootdir + "/kernels/lu/non_contiguous_blocks"


class Radix(Process):
    executable = args.rootdir + "/kernels/radix/RADIX"
    cmd = ["RADIX", "-n524288", "-p", str(args.numcpus)]
    cwd = args.rootdir + "/kernels/radix"


class Barnes(Process):
    executable = args.rootdir + "/apps/barnes/BARNES"
    cmd = ["BARNES"]
    input = args.rootdir + "/apps/barnes/input.p" + str(args.numcpus)
    cwd = args.rootdir + "/apps/barnes"


class FMM(Process):
    executable = args.rootdir + "/apps/fmm/FMM"
    cmd = ["FMM"]
    if str(args.numcpus) == "1":
        input = args.rootdir + "/apps/fmm/inputs/input.2048"
    else:
        input = (
            args.rootdir + "/apps/fmm/inputs/input.2048.p" + str(args.numcpus)
        )
    cwd = args.rootdir + "/apps/fmm"


class Ocean_contig(Process):
    executable = args.rootdir + "/apps/ocean/contiguous_partitions/OCEAN"
    cmd = ["OCEAN", "-p", str(args.numcpus)]
    cwd = args.rootdir + "/apps/ocean/contiguous_partitions"


class Ocean_noncontig(Process):
    executable = args.rootdir + "/apps/ocean/non_contiguous_partitions/OCEAN"
    cmd = ["OCEAN", "-p", str(args.numcpus)]
    cwd = args.rootdir + "/apps/ocean/non_contiguous_partitions"


class Raytrace(Process):
    executable = args.rootdir + "/apps/raytrace/RAYTRACE"
    cmd = [
        "RAYTRACE",
        "-p" + str(args.numcpus),
        args.rootdir + "/apps/raytrace/inputs/teapot.env",
    ]
    cwd = args.rootdir + "/apps/raytrace"


class Water_nsquared(Process):
    executable = args.rootdir + "/apps/water-nsquared/WATER-NSQUARED"
    cmd = ["WATER-NSQUARED"]
    if args.numcpus == 1:
        input = args.rootdir + "/apps/water-nsquared/input"
    else:
        input = (
            args.rootdir + "/apps/water-nsquared/input.p" + str(args.numcpus)
        )
    cwd = args.rootdir + "/apps/water-nsquared"


class Water_spatial(Process):
    executable = args.rootdir + "/apps/water-spatial/WATER-SPATIAL"
    cmd = ["WATER-SPATIAL"]
    if args.numcpus == 1:
        input = args.rootdir + "/apps/water-spatial/input"
    else:
        input = (
            args.rootdir + "/apps/water-spatial/input.p" + str(args.numcpus)
        )
    cwd = args.rootdir + "/apps/water-spatial"


# --------------------
# Base L1 Cache Definition
# ====================


class L1(Cache):
    latency = args.l1latency
    mshrs = 12
    tgts_per_mshr = 8


# ----------------------
# Base L2 Cache Definition
# ----------------------


class L2(Cache):
    latency = args.l2latency
    mshrs = 92
    tgts_per_mshr = 16
    write_buffers = 8


# ----------------------
# Define the cpus
# ----------------------

busFrequency = Frequency(args.frequency)

if args.timing:
    cpus = [
        TimingSimpleCPU(cpu_id=i, clock=args.frequency)
        for i in range(args.numcpus)
    ]
elif args.detailed:
    cpus = [
        DerivO3CPU(cpu_id=i, clock=args.frequency) for i in range(args.numcpus)
    ]
else:
    cpus = [
        AtomicSimpleCPU(cpu_id=i, clock=args.frequency)
        for i in range(args.numcpus)
    ]

# ----------------------
# Create a system, and add system wide objects
# ----------------------
system = System(
    cpu=cpus, physmem=SimpleMemory(), membus=SystemXBar(clock=busFrequency)
)
system.clock = "1GHz"

system.toL2bus = L2XBar(clock=busFrequency)
system.l2 = L2(size=args.l2size, assoc=8)

# ----------------------
# Connect the L2 cache and memory together
# ----------------------

system.physmem.port = system.membus.mem_side_ports
system.l2.cpu_side = system.toL2bus.mem_side_ports
system.l2.mem_side = system.membus.cpu_side_ports
system.system_port = system.membus.cpu_side_ports

# ----------------------
# Connect the L2 cache and clusters together
# ----------------------
for cpu in cpus:
    cpu.addPrivateSplitL1Caches(
        L1(size=args.l1size, assoc=1), L1(size=args.l1size, assoc=4)
    )
    # connect cpu level-1 caches to shared level-2 cache
    cpu.connectAllPorts(
        system.toL2bus.cpu_side_ports,
        system.membus.cpu_side_ports,
        system.membus.mem_side_ports,
    )


# ----------------------
# Define the root
# ----------------------

root = Root(full_system=False, system=system)

# --------------------
# Pick the correct Splash2 Benchmarks
# ====================
if args.benchmark == "Cholesky":
    root.workload = Cholesky()
elif args.benchmark == "FFT":
    root.workload = FFT()
elif args.benchmark == "LUContig":
    root.workload = LU_contig()
elif args.benchmark == "LUNoncontig":
    root.workload = LU_noncontig()
elif args.benchmark == "Radix":
    root.workload = Radix()
elif args.benchmark == "Barnes":
    root.workload = Barnes()
elif args.benchmark == "FMM":
    root.workload = FMM()
elif args.benchmark == "OceanContig":
    root.workload = Ocean_contig()
elif args.benchmark == "OceanNoncontig":
    root.workload = Ocean_noncontig()
elif args.benchmark == "Raytrace":
    root.workload = Raytrace()
elif args.benchmark == "WaterNSquared":
    root.workload = Water_nsquared()
elif args.benchmark == "WaterSpatial":
    root.workload = Water_spatial()
else:
    print(
        "The --benchmark environment variable was set to something "
        "improper. Use Cholesky, FFT, LUContig, LUNoncontig, Radix, "
        "Barnes, FMM, OceanContig, OceanNoncontig, Raytrace, WaterNSquared, "
        "or WaterSpatial",
        file=sys.stderr,
    )
    sys.exit(1)

# --------------------
# Assign the workload to the cpus
# ====================

for cpu in cpus:
    cpu.workload = root.workload

system.workload = SEWorkload.init_compatible(root.workload.executable)

# ----------------------
# Run the simulation
# ----------------------

if args.timing or args.detailed:
    root.system.mem_mode = "timing"

# instantiate configuration
m5.instantiate()

# simulate until program terminates
if args.maxtick:
    exit_event = m5.simulate(args.maxtick)
else:
    exit_event = m5.simulate(m5.MaxTick)

print("Exiting @ tick", m5.curTick(), "because", exit_event.getCause())