package edu.rice.hj.example.comp322.labs.lab2;

import edu.rice.hj.api.HjMetrics;
import edu.rice.hj.api.HjRunnable;
import edu.rice.hj.api.HjSuspendable;
import edu.rice.hj.api.SuspendableException;
import edu.rice.hj.runtime.config.HjSystemProperty;
import edu.rice.hj.runtime.metrics.AbstractMetricsManager;

import static edu.rice.hj.Module1.*;

/**
 * ************************************************************
 * <p>
 * Reads in two strings, the pattern and the input text, and searches for the pattern in the input text.
 * <p>
 * % hj Search rabrabracad abacadabrabracabracadabrabrabracad text:    abacadabrabracabracadababacadabrabracabracadabrabrabracad
 * pattern:                                               rabrabracad
 * <p>
 * HJ version ported from Java version in http://algs4.cs.princeton.edu/53substring/Brute.java.html
 *
 * @author Vivek Sarkar (vsarkar@rice.edu)
 *         <p>
 *         *************************************************************
 */
public class Search2Sol {
    // return number of occurrences of pattern in text

    /**
     * <p>main.</p>
     *
     * @param args an array of {@link String} objects.
     */
    public static void main(final String[] args) {

        // Setup metrics
        HjSystemProperty.abstractMetrics.setProperty(true);
        launchHabaneroApp(new HjSuspendable() {
            @Override
            public void run() throws SuspendableException {

                final String default_pat = "rabrabracad";
                final String default_txt = "abacadabrabracabracadababacadabrabracabracadabrabrabracad";

                String pat = default_pat;
                String txt = default_txt;
                if (args.length >= 1) {
                    pat = args[0];
                }
                if (args.length >= 2) {
                    txt = args[1];
                }

                char[] pattern = pat.toCharArray();
                char[] text = txt.toCharArray();

                int count = search(pattern, text);

                // print results
                System.out.println("text:    " + txt);
                System.out.println("pattern: " + pat);
                System.out.println("Number of occurrences: " + count);

            }
        }, new Runnable() {
            @Override
            public void run() {
                // Print out the metrics data
                final HjMetrics actualMetrics = abstractMetrics();
                AbstractMetricsManager.dumpStatistics(actualMetrics);
            }
        });
    }

    // test client

    /**
     * <p>search.</p>
     *
     * @param pattern an array of char.
     * @param text    an array of char.
     * @return a int.
     */
    public static int search(final char[] pattern, final char[] text) throws SuspendableException {
        final int M = pattern.length;
        final int N = text.length;
        final int[] counts = new int[N - M + 1]; // Array of counts

        finish(new HjSuspendable() {
            @Override
            public void run() throws SuspendableException {
                for (int i = 0; i <= N - M; i++) {
                    final int ii = i;
                    asyncNb(new HjRunnable() {
                        @Override
                        public void run() {
                            int j;
                            for (j = 0; j < M; j++) {
                                doWork(1); // Count each char comparison as 1 unit of work
                                if (text[ii + j] != pattern[j]) {
                                    break;
                                }
                            }
                            if (j == M) {
                                counts[ii] = 1;
                            }
                        }
                    });
                }
            }
        });
        int count = 0;
        for (int arrayCount : counts) {
            count += arrayCount;
        }
        return count;
    }
}