package jp.ac.u_ryukyu.ie.cr.tatsuki.jungle_core;

import jp.ac.u_ryukyu.ie.cr.shoshi.jungle.DefaultJungle;
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/14.
 */
public class ReadTree {

    public static void main(String[] args) throws InterruptedException, IOException, ParserConfigurationException, SAXException {
        System.out.println("startReadTree");

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

        File file = new File("./time/readTree");
        DefaultJungle jungle = new DefaultJungle(null,"sample", new DefaultTreeEditor(new DefaultTraverser()));
        JungleTree tree = jungle.createNewTree("tree" );
        JungleTreeEditor editor = tree.getTreeEditor();
        editor = editor.putAttribute(new DefaultNodePath(), "key", ByteBuffer.wrap(String.valueOf(0).getBytes())).b();
        JungleTreeEditor newEditor = createTree(0, new DefaultNodePath(), editor);
        if (newEditor.success().isA()) {
            System.out.println("success faild");
            System.exit(1);
        }

        PrintWriter pw = new PrintWriter(new BufferedWriter(new FileWriter(file)));
        for (int THREAD_COUNT = 1; THREAD_COUNT <= cpuNum; THREAD_COUNT++) {
            ReadTreeThread readThread[] = new ReadTreeThread[THREAD_COUNT];
            for (int count = 0; THREAD_COUNT > count; count++) {
                readThread[count] = new ReadTreeThread(tree);
            }

            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);
                readThread[count] = null;
            }

            pw.println(THREAD_COUNT + " " + readCount);
            System.out.println(THREAD_COUNT + "readCount = " + readCount);
            System.gc();
            System.out.println("-------------GC--------------");
            Thread.sleep(1000);
        }

        pw.close();
    }

    static Integer nodeNum = 1;

    public static JungleTreeEditor createTree(int deep, NodePath path, JungleTreeEditor editor) {
        String value1 = String.valueOf(nodeNum);
        nodeNum++;
        String value2 = String.valueOf(nodeNum);
        nodeNum++;

        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;
    }

}