import unittest
from pathlib import Path
from concho.dell import DellMatinfoCsvConfigurator
from concho.config import HtmlConfigurator
from concho.dell import DellConfiguratorParser2020
from concho.procs_chooser import plot_configurators
from concho.procs_chooser import ConfigPrice
from concho.procs_chooser import ConfigFlopsPerEuro


class Test(unittest.TestCase):

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.plots_dir = Path('./plots/')
        self.plots_dir.mkdir(exist_ok=True, parents=True)

    def test_all_matinfo_2020_configs(self):
        # configurator = DellHtmlConfigurator('rcrc1406676-4834664 - Cat2 Conf4 PowerEdge R640 - Dell.html')
        # print(configurator)
        configurators = [
            DellMatinfoCsvConfigurator(Path('catalogs/dell/c6420-20200716-price.tsv')),
            HtmlConfigurator(Path('catalogs/dell/rcrc1406676-4834664 - Cat2 Conf4 PowerEdge R640 - Dell.html'), DellConfiguratorParser2020()),
            HtmlConfigurator(Path('catalogs/dell/rcrc1406676-4824727 - Cat 2 Conf 7 PowerEdge R940 - Dell.html'), DellConfiguratorParser2020()),
            # dell.DellPowerEdgeR940(),
        ]

        plot_configurators(configurators=configurators, ram_per_core=4.0e9, xaxis_def=ConfigPrice(), yaxis_def=ConfigFlopsPerEuro(), plot_title='total cost including electricity', figure_file_path=self.plots_dir / '2020-all-matinfo-configs.pdf')

    def test_credits_2020_configs(self):
        # configurator = DellHtmlConfigurator('rcrc1406676-4834664 - Cat2 Conf4 PowerEdge R640 - Dell.html')
        # print(configurator)
        configurators = [
            DellMatinfoCsvConfigurator(Path('catalogs/dell/c6420-20200716-price.tsv')),
            HtmlConfigurator(Path('catalogs/dell/rcrc1406676-4834664 - Cat2 Conf4 PowerEdge R640 - Dell.html'), DellConfiguratorParser2020()),
            HtmlConfigurator(Path('catalogs/dell/rcrc1406676-4824727 - Cat 2 Conf 7 PowerEdge R940 - Dell.html'), DellConfiguratorParser2020()),
            # dell.DellPowerEdgeR940(),
        ]

        # config_filter = lambda config : config.cpu.uid in [
        #     'intel-xeon-gold-5222',
        #     'intel-xeon-gold-6226r',
        #     'intel-xeon-gold-6230r',
        #     'intel-xeon-gold-6234r',
        #     'intel-xeon-gold-6240r',
        #     'intel-xeon-gold-6248r',
        #     'intel-xeon-gold-6230',
        #     'intel-xeon-gold-6240',
        #     ]

        def config_filter(config):
            return config.get_price() < 40000.0

        plot_configurators(configurators=configurators, ram_per_core=4.0e9, xaxis_def=ConfigPrice(), yaxis_def=ConfigFlopsPerEuro(), plot_title='physmol/ts credit 2020 configs', config_filter=config_filter, figure_file_path=self.plots_dir / '2020-credits-configs.pdf')


if __name__ == '__main__':
    unittest.main()