# Read thesis-append-pbv.csv
# Output for string-graph-peq-sharing.dat

# Project details
# Filter operation=peq
# Split "series" goups of sut; only those in the "pretty" list
# Assert one row per string-length
# output:
# string-len op-duration
# in chunks, each headed by pertty(sut)

import pandas as pd
import numpy as np
import os
import sys

sys.path.insert(0, os.path.dirname(__file__))
from common import *

# re: apparent cherrypicking
# The system's response to the liveness threshold is not smooth.
# The system only uses the threshold to decide whether it will double the text heap again or not.
# The system's speed for a given string size in a given amount of memory is not affected by the specific value of the liveness threshold.
# Goals with this selection are
#  - showing one speed result per <string size, memory usage amount>
#  - cropping diminishing or negative returns for large memory sizes
#    - diminishing is obvious, already shown past chosen sweet spot in this selection
#    - negative caused by overflowing llc, not relevant to sting impl
favSizes = {20:[-1.0, 0.05, 0.1, 0.2, 0.5, 0.9],
            50:[-1.0, 0.05, 0.1, 0.2, 0.5, 0.9],
            100:[-1.0, 0.1, 0.2, 0.5, 0.9],
            200:[-1.0, 0.1, 0.2, 0.5, 0.9],
            500:[-1.0, 0.4, 0.9, 0.98]}

defaultExpansions = [-1, 0.2]

cfatimings = loadParseTimingData('result-allocate-speed-cfa.csv',
                xClasNames=['expansion'], xClasDtypes={'expansion':'Float64'},
                xFactNames=['topIters'], xFactDtypes={'topIters':np.int64})

cfasizings = loadParseSizingData('result-allocate-space-cfa.ssv', xClasNames=['expansion'], xClasDtypes={'expansion':'Float64'})

stltimings = loadParseTimingData('result-allocate-speed-stl.csv',
                xClasNames=['expansion'], xClasDtypes={'expansion':'Float64'},
                xFactNames=['topIters'], xFactDtypes={'topIters':np.int64})

stlsizings = loadParseSizingData('result-allocate-space-stl.ssv', xClasNames=['expansion'], xClasDtypes={'expansion':'Float64'})

timings = pd.concat([cfatimings, stltimings])
sizings = pd.concat([cfasizings, stlsizings])

combined = pd.merge(
    left=timings,
    right=sizings[['sut', 'corpus','expansion','hw_cur_req_mem(B)']],
    on=['sut', 'corpus','expansion']
)

combined['is-default'] = np.isin(combined['expansion'], defaultExpansions).astype(int)

# print ('!!')
# print(combined)


# Emit

# First, for the CFA curves
sut = "cfa"
sutGroup = combined.groupby('sut-platform').get_group(sut)

groupedSize = sutGroup.groupby('corpus-meanlen')

for sz, szgroup in groupedSize:

    if sz in favSizes.keys():
            szgroup_sorted = szgroup.sort_values(by='expansion')

            print('"{sut}, len={len}"'.format(sut=sut, len=sz))
            # print(szgroup_sorted)  ##
            # print(szgroup_sorted['expansion'], 'isin', favSizes[sz]) ##
            favoured = szgroup_sorted.loc[szgroup_sorted['expansion'].isin(favSizes[sz])]
            # print('!') ##
            # print(favoured) ##
            text = favoured[['expansion', 'op-duration-ns', 'hw_cur_req_mem(B)', 'is-default']].to_csv(header=False, index=False, sep='\t')
            print(text)
            print()

# Again, for the STL-comparisons, default expansion only

atDefaults = combined.groupby('is-default').get_group(1)

for sz, szgroup in atDefaults.groupby('corpus-meanlen'):

    if sz in favSizes.keys():
            print(sz)
            text = szgroup[['expansion', 'op-duration-ns', 'hw_cur_req_mem(B)', 'sut-platform']].to_csv(header=False, index=False, sep='\t')
            print(text)
            print()