# 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 os
import argparse
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 = "32kB")
parser.add_argument("--l1latency",
                    default = "1ns")
parser.add_argument("--l2size",
                    default = "256kB")
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())