/*

 * Get all classifiers from weka libs

 */

package utils;

import java.io.File;

import java.io.IOException;

import java.util.ArrayList;

import java.util.Collections;

import java.util.Enumeration;

import java.util.jar.JarEntry;

import java.util.jar.JarFile;

import weka.classifiers.*;

import weka.core.Utils;

/**

 *

 * @author Administrator

 */

public class RetreiveClassifiersFromWeka {

    public static void main(String[] args) {

        ArrayList<String> al = new ArrayList<>();

        //get all possible models from lib folder

        File lib = new File("lib/weka/");

        for (File jar : lib.listFiles()) {

            if (jar.getName().endsWith("jar")) {

                try {

                    JarFile jarFile = new JarFile(jar);

                    Enumeration<JarEntry> entries = jarFile.entries();

                    while (entries.hasMoreElements()) {

                        JarEntry entry = entries.nextElement();

                        String n = entry.getName();

                        if (n.startsWith("weka/classifiers/")

                                && !n.contains("$")

                                && n.endsWith("class")

                                && n.split("/").length == 4

                                && (n.contains("/bayes/")

                                || n.contains("/functions/")

                                || n.contains("/rules/")

                                || n.contains("/trees/")

                                || n.contains("/lazy")

                                || n.contains("/misc"))) {

                            //System.out.println(jar.getName() + "\t" + n);

                            al.add(n);

                        }

                    }

                } catch (IOException e) {

                    e.printStackTrace();

                }

            }

        }

        Collections.sort(al);

        for (String classifier : al) {

            try {

                classifier = classifier.replace(".class", "").replace("/", ".").replace("weka/classifiers/", "");

                weka.classifiers.AbstractClassifier ac = (AbstractClassifier) Utils.forName(Classifier.class, classifier, args);

                String options = "";

                for (String option : ac.getOptions()) {

                    options += " " + option;

                }

                String cap = ac.getCapabilities().getClassCapabilities().toString();                

                System.out.println(

                        classifier.replaceAll("^weka.classifiers.", "").replaceFirst("\\.", "\t") + "\t"

                        + ac.getCapabilities().getAttributeCapabilities().toString()

                        .replaceAll("(Capabilities: \\[|attributes)", "")

                        .replace("Empty nominal ,", "")

                        .replace(" ,", ",").replace("  ", " ").replaceAll("\\].*", "").split("\n")[0] + "\t"

                        + ac.getCapabilities().getClassCapabilities().toString()

                        .replaceAll("(Capabilities: \\[|class)", "")

                        .replace("Missing  values", "")

                        .replace(" ,", ",").replace("  ", " ").replaceAll("\\].*", "").split("\n")[0].replaceAll(", $", " "));

                //+ac.getCapabilities().getClassCapabilities().toString());

//                if (cap.contains("Nominal class") && cap.contains("Numeric class")) {

//                    System.out.println(classifier.replaceAll("^weka.classifiers.", "ccmd=") + options);

//                    System.out.println(classifier.replaceAll("^weka.classifiers.", "rcmd=") + options);

//                } else if (cap.contains("Nominal class")) {

//                    System.out.println(classifier.replaceAll("^weka.classifiers.", "ccmd=") + options);

//                } else {

//                    System.out.println(classifier.replaceAll("^weka.classifiers.", "rcmd=") + options);

//                }

//

//                ac.listOptions();

            } catch (Exception e) {

                //e.printStackTrace();

            }

        }

    }

}