package network;

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.util.ArrayList;
import java.util.List;

public class subCommNet {

	public static void main(String[] args) throws IOException {
		// TODO Auto-generated method stub
		
//		String disfile = "/Users/zhengc/Projects/AD_comorbidity/data/AD_commobidity.txt";
//		String netfile = "/Users/zhengc/workspace/FARES/data/FARES/network/fares_comm_net_conf_ISMB_final_public.txt";
//		String subnetfile = "/Users/zhengc/Projects/AD_comorbidity/data/fares_comm_net_sub_AD.txt";
		
		String disfile = "/Users/zhengc/workspace/FARES_final/analysis/AD_comorbidity/data/AD_commobidity_rw_lift.txt";
		String netfile = "/Users/zhengc/workspace/FARES_final/analysis/network/DCN/fares_comm_net_lift_final_abbr.txt";
		String subnetfile = "/Users/zhengc/workspace/FARES_final/analysis/network/DCN/fares_comm_net_lift_sub_AD.txt";
		List<String> dislist = getDisList(disfile);
		for (String dis: dislist) {
			System.out.println(dis);
		}
		createSubCommNet(dislist, netfile, subnetfile);
	}
	
//	public CommGraph createSubCommNet(List<String> dlist, CommGraph cg) {
//		SparseMatrix comMatrix = cg.getNet();
//		CommGraph subcommnet = new CommGraph(dlist.size());
//		for (String dis : dlist) {
//			SparseVector neibors = cg.getNet().getNeibor(dis);
//			
//		}
//		
//		return subcommnet;
//	}
	public static List<String> getDisList(String disfile) throws IOException{
		List<String> dislist = new ArrayList<>();
		BufferedReader br = new BufferedReader(new FileReader(new File(disfile)));
		String line = null;
		while((line = br.readLine()) != null) {
			String dis = line.split("\\|")[0];
			dislist.add(dis);
		}
		br.close();
		return dislist;
	}
	
	public static void createSubCommNet(List<String> dlist, String netfile, String subnetfile) throws IOException {
		BufferedReader br = new BufferedReader(new FileReader(new File(netfile)));
		BufferedWriter bw = new BufferedWriter(new FileWriter(new File(subnetfile)));
		String line = br.readLine();
		while ((line = br.readLine()) != null) {
			String[] parts = line.split("\\|");
			String dis1 = parts[1];
			String dis2 = parts[6].replace("-- ", "");
			if (dlist.contains(dis1) & dlist.contains(dis2)) {
				line = line.replace("-- ", "");
				bw.write(line + "\n");
			}
		}
		br.close();
		bw.close();
		
	}

}