import java.io.FileNotFoundException;
import java.io.PrintWriter;
import java.io.Serializable;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Random;
import java.util.Set;
import java.util.logging.Level;
import java.util.logging.Logger;

/**
 *
 * @author ghost
 */
public class BayesClassifier implements Serializable{
    private int soluongclass;
    private float[] pc;
    private int[] soluongtungclass;
    private Map<String, Float> probability = new HashMap<String, Float>();
    private int[][] rest;
    public BayesClassifier(int soluongclass) {
        this.soluongclass = soluongclass;
        pc = new float[soluongclass];
        soluongtungclass = new int[soluongclass];
        for(int i = 0; i < soluongclass; i ++)
        {
            soluongtungclass[i] = 0;
        }
    }
    //học từ bộ dữ liệu data, với nhãn là label.
    public void learn(ArrayList<String> data, int[] label)
    {
        for(int i = 0; i < data.size(); i ++)
        {
            soluongtungclass[label[i]] += 1;
            String tam2 = data.get(i);
            tam2 = xulychuoi(tam2);
            for(String s: tam2.split("\\s+"))
            {
                String tam = s + "\n" +Integer.toString(label[i]);

                if(probability.containsKey(tam))
                {
                    probability.replace(tam, new Float(probability.get(tam).floatValue() + 1));
                }else
                {
                    probability.put(tam, new Float(1));
                }
            }
        }
        int tong = 0;
        for(int i = 0; i < soluongclass; i ++)
        {
            tong += soluongtungclass[i];
        }
        System.out.println("TOng: " + tong);
        for(int i = 0; i < soluongclass; i ++)
        {
            pc[i] = (float)soluongtungclass[i] / tong;
            System.out.println("i " + pc[i]);
        }
        for (Map.Entry<String, Float> entry : probability.entrySet())
        {
            String tam2 = entry.getKey();
            String tu = tam2.split("\\n")[0];
            int nlabel = Integer.parseInt(tam2.split("\\n")[1]);
            probability.replace(tam2, probability.get(tam2).floatValue()/soluongtungclass[nlabel]);
            System.out.println(tu +" " +nlabel + " co xac xuat:" + entry.getValue());
        }
    }
    public int predict(String data)
    {
        data = xulychuoi(data);
        int tam = -1;
        float max = -1;
        for(int i = 0; i < soluongclass; i ++)
        {
            float xs = pc[i];
            System.out.println("Xac xuat class " + i + ":" + pc[i]);
            for(String s: data.split("\\s+"))
            {
                if(probability.containsKey(s + "\n" + i))
                {
                    System.out.println("Tu: " + s + " Trong class " + i + "Co xs:" + probability.get(s + "\n" + i).floatValue());
                    xs *= probability.get(s + "\n" + i).floatValue();
                }else{
                    System.out.println("Tu: " + s + "khong xuat hien trong bo phan loai: " + i);
                }
            }
            System.out.println("Xac xuat class " + i + ": " + xs );
            if(xs > max)
                {
                    max = xs;
                    tam = i;
                }

        }
        System.out.println("Max:" + tam);
        return tam;
    }
    public String xulychuoi(String s)
    {
        s = s.toLowerCase();
        String res = "";
        Set<String> s1 = new HashSet<String>();
        for(String s2: s.split("\\s+"))
        {
            s1.add(s2);
        }
        ArrayList<String> s3 = new ArrayList<String>(s1);
        for(String s2: s3)
            res += s2 + " ";
        return res.trim();
    }
    //Dự đoán danh sách các bài viết nằm trong bộ dữ liệu data, với nhãn tương ứng là label (kiểu int), ghi ra file number
    public int[][] predict(ArrayList<String> data, int[] label, int number)
    {
        try {
            int[] dudoan = new int[label.length];
            for(int i = 0; i < data.size(); i ++)
            {
                dudoan[i] = predict(data.get(i));
                System.out.println("Du doan:" + dudoan[i]);
            }
            int[][] tp = new int[soluongclass][soluongclass];
            for(int i = 0; i < soluongclass; i ++)
            {
                for(int j = 0; j < soluongclass; j ++)
                {
                    tp[i][j] = 0;
                }
            }
            for(int i = 0 ; i < label.length; i ++)
            {
                if(label[i] == dudoan[i])
                {
                    tp[dudoan[i]][label[i]] += 1;
                }else
                {
                    tp[dudoan[i]][label[i]] += 1;
                }
            }
            PrintWriter writer;
            writer = new PrintWriter("lan" + number + ".txt");
            for (int i = 0; i<soluongclass; i++){
                for(int j = 0; j<soluongclass; j++){
                    writer.print(tp[i][j] + ",");
                }writer.println();
            }
            writer.flush();
            writer.close();
        } catch (FileNotFoundException ex) {
            Logger.getLogger(BayesClassifier.class.getName()).log(Level.SEVERE, null, ex);
        }
        return null;
    }
    public static void main(String[] args) throws FileNotFoundException {
        int[] tam = new int[8];
        ArrayList<String> tam1 = new ArrayList<>();
//        tam1.add("Nhà chức trách thông báo bất kỳ người nào cũng có quyền bắt, áp giải nghi phạm Bùi Quang Huy đến cơ quan công an gần nhất.");
//        tam1.add("Ngày 18/5, Cơ quan Cảnh sát điều tra (Bộ Công an) ra quyết định truy nã Bùi Quang Huy (Tổng giám đốc Công ty TNHH Thương mại và Dịch vụ kỹ thuật Nhật Cường), 9 ngày sau khi đồng loạt khám xét chuỗi cửa hàng, trung tâm bảo hành của Nhật Cường Mobile.");
//        tam1.add("Theo nhà chức trách, bất kỳ người nào cũng có quyền bắt và giải ngay Bùi Quang Huy đến cơ quan công an, viện kiểm sát hoặc UBND nơi gần nhất. Sau khi bắt hoặc tiếp nhận người truy nã, người dân được đề nghị báo về Phòng 14 Cục Cảnh sát điều tra tội phạm về tham nhũng, kinh tế, buôn lậu (C03, Bộ Công an) tại 47 Phạm Văn Đồng, Hà Nội.");
//        tam1.add("5 ngày trước, 14/5, C03 đã khởi tố vụ án, khởi tố bị can, khám xét với Bùi Quang Huy cùng 8 người về tội Buôn lậu theo khoản 4 điều 188 Bộ luật Hình sự 2015 và tội Vi phạm quy định về kế toán gây hậu quả nghiêm trọng, theo điều 221 Bộ luật Hình sự 2015. Lệnh bắt tạm giam với những người này được ban hành song ông chủ của Nhật Cường đã bỏ trốn, hiện không rõ ở đâu.");
//        tam1.add("Bùi Quang Huy bị cáo buộc cùng các đồng phạm đã thực hiện hành vi phạm tội buôn lậu xuyên quốc gia; lập, sử dụng hai hệ thống sổ sách kế toán nhằm để ngoài sổ sách hàng nghìn tỷ đồng doanh thu; nhập lậu nhiều thiết bị điện tử sau khi móc nối với một số nhà sản xuất ở nước ngoài.");
//        tam1.add("Nhật Cường Mobile là thương hiệu thuộc sở hữu của Công ty TNHH Thương mại và Dịch vụ kỹ thuật Nhật Cường. Hệ thống có 9 cửa hàng bán lẻ tại Hà Nội, một trung tâm bảo hành và một trung tâm ERP tại TP HCM.");
//        tam1.add("Theo thông tin đăng ký doanh nghiệp, Nhật Cường có vốn điều lệ 38 tỷ đồng, Tổng giám đốc là Bùi Quang Huy cũng là người đại diện pháp luật của Công ty TNHH giải pháp phần mềm Nhật Cường (Nhật Cường Software).");
//        tam1.add("Ngày 19/5, Công an Bình Dương xác định danh tính hai nạn nhân trong vụ phi tang đổ bêtông là Trần Đức Linh (quê Nghệ An) và Trần Trí Thành (quê TP HCM). Cả hai tử vong và bị phi tang xác cách nhau vài tháng.");
        tam1.add("day la mot");//1
        tam1.add("la loai day Bai");//1
        tam1.add("bo loai");//1
        tam1.add("phan");//0
        tam1.add("phan");//0
        tam1.add("Bai day la mot bo phan");//0
        tam1.add("loai bayes");//0
        tam1.add("loai cho phan ");//0
        for(int i = 0; i < 8; i ++)
        {
            if(i > 2)
            tam[i] = 0;
            else
            tam[i] = 1;
        }
        BayesClassifier b = new BayesClassifier(3);
        b.learn(tam1, tam);
        ArrayList<String> tam2 = new ArrayList<>();
        tam2.add("Bai toan phan loai bayes");
        int[] lb = new int[1];
        lb[0] = 1;
        b.predict(tam2, lb, 1);
        //b.predict("Bai toan phan loai bayes");
        //System.out.println(b.xulychuoi("Day la mot bo day Nhe la mot Bo"));
    }
}