package dataprep;

import java.io.BufferedReader;

import java.io.BufferedWriter;

import java.io.File;

import java.io.FileNotFoundException;

import java.io.FileReader;

import java.io.FileWriter;

import java.io.IOException;

import java.text.Normalizer;

import java.util.ArrayList;

import java.util.HashMap;

import java.util.HashSet;

import java.util.List;

import java.util.Map;

import java.util.Set;

public class LongToWide {

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

        Map<Integer, Set<String>> pid_cids = new HashMap<>();

        String lfname1 = "/Users/zhengc/Projects/AD_metaformin/data/medical_history/processed/ADNI1_umls_id_dsyn_2.csv";

//      String lfname2 = "/Users/zhengc/Projects/AD_metaformin/data/medical_history/processed/ADNI3_umls_id_dsyn_2.csv";

        String lfname3 = "/Users/zhengc/Projects/AD_metaformin/data/drug/processed/drug_umls_id.csv";

        Map<Integer, Set<String>> pid_cids1 = getCids(lfname1);

//      Map<Integer, Set<String>> pid_cids2 = getCids(lfname2);

        Map<Integer, Set<String>> pid_cids3 = getCids(lfname3);

        Set<Integer> pids = pid_cids1.keySet();

        Set<Integer> pid3 = pid_cids3.keySet();

        pids.retainAll(pid3);

        for (int pid: pids) {

            Set<String> cids = pid_cids1.get(pid);

            Set<String> cids3 = pid_cids3.get(pid);

            cids.addAll(cids3);

            pid_cids.put(pid, cids);

        }

        String wfname = "/Users/zhengc/Projects/AD_metaformin/data/drug/processed/Comb_drug_mh_umls_id_width.csv";

        BufferedWriter bw = new BufferedWriter(new FileWriter(new File(wfname)));

        for (int pid: pid_cids.keySet()) {

            Set<String> cids = pid_cids.get(pid);

            StringBuilder sb = new StringBuilder();

            for (String cid: cids) {

                sb.append(cid + ",");

            }

            String cidstr0 = sb.toString();

            String cidstr = cidstr0.replaceAll("[,]$", "");

//          String cidstr = cidstr0.substring(0, cidstr0.length() - 1);

            bw.write(cidstr + "\n");

        }

        bw.close();

    }

    private static Map<Integer, Set<String>> getCids(String lfname) throws IOException {

        Map<Integer, Set<String>> pid_cids = new HashMap<>();

        BufferedReader br = new BufferedReader(new FileReader(new File(lfname)));

        String line = br.readLine();

        while ((line=br.readLine()) != null) {

            String[] parts = line.split(",");

            System.out.println(parts.length);

            int pid = Integer.parseInt(parts[0]);

            System.out.println(pid);

            String cid = parts[2];

            System.out.println(cid);

            String ad_status = null;

            if (parts.length > 3) {

                ad_status = parts[3];

            }

            if (!pid_cids.containsKey(pid)) {

                Set<String> cids = new HashSet<>();

                cids.add(cid);

                if (ad_status != null && (ad_status.equals("AD") || ad_status.equals("MCI"))) cids.add("C0002395");

                pid_cids.put(pid, cids);

            } else {

                pid_cids.get(pid).add(cid);

                if (ad_status != null && (ad_status.equals("AD") || ad_status.equals("MCI"))) pid_cids.get(pid).add("C0002395");

            }

        }

        br.close();

        return pid_cids;

    }

}