package jp.ac.u_ryukyu.ie.cr.tatsuki.jungle.test;

import fj.Ord;
import fj.data.TreeMap;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.DefaultJungle;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.Jungle;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.JungleTree;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.JungleTreeEditor;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.NodePath;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.impl.DefaultNodePath;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.store.impl.DefaultTreeEditor;
import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.traverser.DefaultTraverser;
import org.xml.sax.SAXException;

import javax.xml.parsers.ParserConfigurationException;
import java.io.*;
import java.nio.ByteBuffer;
import java.util.Random;

/**
 * Created by e115731 on 15/03/20.
 */
public class JungleBenchMark {
    public static void main(String[] args) throws InterruptedException, IOException, ParserConfigurationException, SAXException {


        if (args.length == 0) {
            System.out.println("args read or find");
            System.exit(0);
        }

        Jungle jungle = new DefaultJungle(null, "hoge", new DefaultTreeEditor(new DefaultTraverser()));
        JungleTree tree = jungle.createNewTree("Person");
        JungleTreeEditor editor = tree.getTreeEditor();
        JungleTreeEditor newEditor = createTree(0,new DefaultNodePath(),editor);
        if (newEditor.success().isA()) {
            System.out.println("success faild");
            System.exit(1);
        }

        Runtime rt = Runtime.getRuntime();
        int cpuNum = rt.availableProcessors();


        File file = new File("./time/JungleTree" + args[0] + "Time");

        PrintWriter pw = new PrintWriter(new BufferedWriter(new FileWriter(file)));

        JungleBenchMarkThread readThread[] = null;

        for (int THREAD_COUNT = 1; THREAD_COUNT <= cpuNum; THREAD_COUNT++) {
            readThread = new JungleBenchMarkThread[THREAD_COUNT];

            if (args[0].equals("read")) {
                for (int count = 0; THREAD_COUNT > count; count++) {
                    readThread[count] = new readTreeAttributeThread(tree);
                }
            } else if (args[0].equals("find")) {
                for (int count = 0; THREAD_COUNT > count; count++) {
                    readThread[count] = new findTreeAttributeThread(tree);
                }
            } else {
                System.out.println("not allow args");
                System.exit(0);
            }

            for (int count = 0; THREAD_COUNT > count; count++) {
                readThread[count].start();
            }

            System.out.println("StartThread");

            Thread.sleep(1000);
            long readCount = 0;

            for (int count = 0; THREAD_COUNT > count; count++) {
                readCount = readCount + readThread[count].getFindCount();
                readThread[count].set(false);
            }

            pw.println(THREAD_COUNT + " " + readCount);
            System.out.println(THREAD_COUNT + "readCount = " + readCount);
        }

        pw.close();
    }

    static Integer nodeNum = 0;
    public static JungleTreeEditor createTree(int deep, NodePath path, JungleTreeEditor editor) {

        Random rnd = new Random();
        String value1 = String.valueOf(nodeNum);
        nodeNum ++;
        String value2 = String.valueOf(nodeNum);
        nodeNum ++;

        if (value1.equals("500"))
            System.out.println("aaa");
        editor = editor.addNewChildAt(path, 0).b();
        editor = editor.putAttribute(path.add(0), "key", ByteBuffer.wrap(value1.getBytes())).b();
        editor = editor.addNewChildAt(path, 1).b();
        editor = editor.putAttribute(path.add(1), "key", ByteBuffer.wrap(value2.getBytes())).b();
        deep++;

        if (deep < 10) {
            editor = createTree(deep, path.add(0), editor);
            editor = createTree(deep, path.add(1), editor);
        }
        return editor;
    }
}