import java.lang.*;
import java.io.*;
import java.util.*;
import java.lang.Math;
import java.util.Collection;
import java.util.HashMap;
import java.util.Iterator;



public class test {

    public static void main(String[] args) throws IOException {
        //---- Definitions ----//

        HashMap<String, List<String>> indexDocID = new HashMap<String, List<String>>();
        HashMap<String, List<String>> indexTF = new HashMap<String, List<String>>();
        HashMap<String, String> indexNormalizedTF = new HashMap<String, String>();

        String indexFileLocation = "path";
        String tokenMapFileLocation = "path";

        //---- Code ----//

        try (BufferedReader br = new BufferedReader(new FileReader(indexFileLocation))) {
            String line;
            int position = 0;
            while ((line = br.readLine()) != null) {
                List<List<String>> valueList = new ArrayList<List<String>>();

                String[] term = line.split("-~- ");

                //Grab Key
                String key = term[0];

                //Grab values and put them in a list of lists
                String indexValue = term[1];
                indexValue = indexValue.replace("\n"," ");
                String[] values = indexValue.split(" ");

                List<String> tfList = new ArrayList<String>();
                List<String> docIDList = new ArrayList<String>();
                for(int i = 0; i < values.length; ++i) {
                    String temporary = values[i].replaceAll("[()]","");
                    String[] individualValues = temporary.split(",");
                    String docid = individualValues[0];
                    docIDList.add(docid);
                    String tf = individualValues[1];
                    tfList.add(tf);


                }
                position =+ 1;
                indexDocID.put(key, docIDList);
                indexTF.put(key, tfList);
            }

        } catch (IOException e) {
            e.printStackTrace();
        }

        try (BufferedReader br = new BufferedReader(new FileReader(tokenMapFileLocation))) {
            String line;
            int position = 0;
            while ((line = br.readLine()) != null) {
                String[] entries = line.split("--");
                String docID = entries[0];
                String tokenCount = entries[1];
                indexNormalizedTF.put(docID, tokenCount);
            }

        } catch (IOException e) {
            e.printStackTrace();
        }


        //System.out.print(indexDocID.size());

        Scanner in = new Scanner(System.in);

        System.out.println("Please enter a query:");
        String query_input = in.nextLine();
        query_input = query_input.toLowerCase();


        System.out.println(query_input);

        List<String> test = indexDocID.get(query_input);

        System.out.println(test.size());

/*
        for (Map.Entry<String,List<String>> entry : indexDocID.entrySet()) {
            String key = entry.getKey();
            List<String> value = entry.getValue();
            System.out.println(key);
            System.out.println(value);
        }
        */

        List<String> list = new ArrayList<>();
        list = indexDocID.get(query_input);



        try {



        } catch (Throwable e) {

        }








    }

    public static double calcTF(int termFrequency, int totalTokens) {
        double normalizedTF = 0;
        normalizedTF = termFrequency / totalTokens;
        return normalizedTF;
    }

    public static double calcIDF(int documentFrequency, List totalDocuments) {
        double normalizedTF = 0;
        normalizedTF = 1 + Math.log(totalDocuments.size() / documentFrequency);
        return normalizedTF;

    }



    //System.out.println(terms);


}