itachi1706/CheesecakeUtilities

package com.itachi1706.cheesecakeutilities.modules.ipptcalculator.helpers;

import android.content.Context;

import com.google.gson.Gson;
import com.google.gson.JsonElement;
import com.google.gson.JsonObject;
import com.itachi1706.cheesecakeutilities.R;
import com.itachi1706.cheesecakeutilities.modules.ipptcalculator.jsonObjects.Gender;
import com.itachi1706.cheesecakeutilities.modules.ipptcalculator.jsonObjects.Main;
import com.itachi1706.helperlib.helpers.LogHelper;

import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Map;

/**
 * Created by Kenneth on 7/9/2016.
 * for com.itachi1706.cheesecakeutilities.modules.IPPTCalculator.Helpers in CheesecakeUtilities
 */
public class JsonHelper {

    public static Main readFromJsonRaw(Context context) {
        String jsonString = readFromRaw(context);
        if (jsonString.isEmpty() || jsonString.equalsIgnoreCase("")) {
            return null;
        }

        Gson gson = new Gson();
        return gson.fromJson(jsonString, Main.class);
    }

    private static String readFromRaw(Context context) {
        //Get Data From Text Resource File Contains Json Data.
        InputStream inputStream = context.getResources().openRawResource(R.raw.ippt_minified);
        ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();

        int ctr;
        try {
            ctr = inputStream.read();
            while (ctr != -1) {
                byteArrayOutputStream.write(ctr);
                ctr = inputStream.read();
            }
            inputStream.close();
        } catch (IOException e) {
            e.printStackTrace();
            return "";
        }
        return byteArrayOutputStream.toString();
    }

    public static final int MALE = 0, FEMALE = 1;

    public static int getGender(String gender) {
        switch (gender.toLowerCase()) {
            case "m":
            case "male": return MALE;
            case "f":
            case "female": return FEMALE;
        }
        return MALE;
    }

    public static List<String> getAgeRangeText(Context context) {
        return getAgeRangeText(readFromJsonRaw(context));
    }

    public static List<String> getAgeRangeText(Main object) {
        JsonObject jsonObject = object.getAgeRangeText();

        List<String> tmp = new ArrayList<>();
        for (Map.Entry<String,JsonElement> entry : jsonObject.entrySet()) {
            String key = entry.getKey();
            tmp.add(key);
        }
        return tmp;
    }

    public static final int PUSHUP = 0, RUN = 1, SITUP = 2, UNKNOWN = 3;

    public static int getExercise(String exercise) {
        switch (exercise.toLowerCase()) {
            case "2.4km run": return RUN;
            case "sit-ups": return SITUP;
            case "bent knee push-ups":
            case "push-ups": return PUSHUP;
        }
        return RUN;
    }

    public static List<String> getExerciseScores(int age, int exercise, int gender, Context context) {
        return getExerciseScores(age, exercise, gender, readFromJsonRaw(context));
    }

    public static List<String> getExerciseScores(int age, int exercise, int gender, Main object) {
        Gender exercisesScore = (gender == FEMALE) ? object.getDataFemale() : object.getDataMale();

        JsonObject scoreBoard = null;
        switch (exercise) {
            case RUN: scoreBoard = exercisesScore.getRun(); break;
            case SITUP: scoreBoard = exercisesScore.getSitups(); break;
            case PUSHUP: scoreBoard = exercisesScore.getPushups(); break;
        }

        if (scoreBoard == null) return null;

        List<String> scores = new ArrayList<>();
        for (Map.Entry<String,JsonElement> entry : scoreBoard.entrySet()) {
            String key = entry.getKey();
            JsonElement element = entry.getValue().getAsJsonObject().get(age + "");
            scores.add(key + "\t\t-\t\t" + element + " pts");
        }
        Collections.reverse(scores);
        return scores;
    }

    public static int getAgeGroup(String ageText, Context context) {
        return getAgeGroup(ageText, readFromJsonRaw(context));
    }

    public static int getAgeGroup(String ageText, Main object) {
        JsonObject jsonObject = object.getAgeRangeText();
        for (Map.Entry<String,JsonElement> entry : jsonObject.entrySet()) {
            String key = entry.getKey();
            if (key.equals(ageText)) {
                JsonElement element = entry.getValue();
                return element.getAsInt();
            }
        }
        return 1;
    }

    public static int calculateScore(int pushup, int situp, int runMin, int runSec, int age, int gender, Context context) {
        return calculateScore(pushup, situp, runMin, runSec, age, gender, readFromJsonRaw(context));
    }

    public static int calculateScore(int pushup, int situp, int runMin, int runSec, int age, int gender, Main object) {
        Gender exercisesScore = (gender == FEMALE) ? object.getDataFemale() : object.getDataMale();
        LogHelper.i("IPPTCalc", "Age Group: " + age);

        int situpScore = getSitUpScore(situp, age, exercisesScore);
        int pushupScore = getPushUpScore(pushup, age, exercisesScore);
        int runScore = getRunScore(runMin, runSec, age, exercisesScore);
        LogHelper.i("IPPTCalc", "Situp Score: " + situp + " (" + situpScore + ")");
        LogHelper.i("IPPTCalc", "Pushup Score: " + pushup + " (" + pushupScore + ")");
        LogHelper.i("IPPTCalc", "Run Score: " + runMin + ":" + runSec + " (" + runScore + ")");
        return situpScore+pushupScore+runScore;
    }

    public static int calculateIncompleteScore(int pushup, int situp, int runMin, int runSec, int age, int gender, Context context) {
        return calculateIncompleteScore(pushup, situp, runMin, runSec, age, gender, readFromJsonRaw(context));
    }

    public static int calculateIncompleteScore(int pushup, int situp, int runMin, int runSec, int age, int gender, Main object) {
        Gender exercisesScore = (gender == FEMALE) ? object.getDataFemale() : object.getDataMale();
        LogHelper.i("IPPTCalc", "Age Group: " + age);

        int situpScore = (situp != -1) ? getSitUpScore(situp, age, exercisesScore) : -1;
        int pushupScore = (pushup != -1) ? getPushUpScore(pushup, age, exercisesScore) : -1;
        int runScore = (runMin != -1) ? getRunScore(runMin, runSec, age, exercisesScore) : -1;
        LogHelper.i("IPPTCalc", "Situp Score: " + situp + " (" + situpScore + ")");
        LogHelper.i("IPPTCalc", "Pushup Score: " + pushup + " (" + pushupScore + ")");
        LogHelper.i("IPPTCalc", "Run Score: " + runMin + ":" + runSec + " (" + runScore + ")");

        int totalScore = 0;
        if (situpScore != -1) totalScore += situpScore;
        if (pushupScore != -1) totalScore += pushupScore;
        if (runScore != -1) totalScore += runScore;
        return totalScore;
    }

    public static String getScoreResults(int score) {
        if (score >= 90) return "Gold (Commando/Guards)";
        if (score >= 85) return "Gold";
        if (score >= 75) return "Silver";
        if (score >= 61) return "Pass (Active)/Pass with Incentive (NSMen)";
        if (score >= 51) return "Fail (Active)/Pass (NSMen)";
        return "Fail";
    }

    private static int getElementScore(int score, int ageGroup, JsonObject obj) {
        if (score == 0) return 0; // Don't need check lol
        JsonElement element = null;
        for (Map.Entry<String,JsonElement> entry : obj.entrySet()) {
            String key = entry.getKey();
            if (key.equals(Integer.toString(score))) {
                element = entry.getValue();
                break;
            }
        }
        if (element == null) return 25; // Presume full marks
        JsonObject el = element.getAsJsonObject();
        return el.get(ageGroup + "").getAsInt();
    }

    public static int getSitUpScore(int situp, int ageGroup, Gender object) {
        return getElementScore(situp, ageGroup, object.getSitups());
    }

    public static int getPushUpScore(int pushup, int ageGroup, Gender object) {
        return getElementScore(pushup, ageGroup, object.getPushups());
    }

    public static int getRunScore(int runMin, int runSec, int ageGroup, Gender object) {
        if (runMin == 0 && runSec == 0) return 0; // Don't need check lol
        int totalSecs = (runMin * 60) + runSec;
        JsonObject obj = object.getRun();
        JsonElement element = null;
        for (Map.Entry<String,JsonElement> entry : obj.entrySet()) {
            String key = entry.getKey();
            String[] values = key.split(":");
            int totalSec = (Integer.parseInt(values[0]) * 60) + Integer.parseInt(values[1]);
            if (totalSecs <= totalSec) {
                element = entry.getValue();
                break;
            }
        }
        if (element == null) return 0; // Presume run out of time
        JsonObject el = element.getAsJsonObject();
        return el.get(ageGroup + "").getAsInt();
    }

    public static String countSitupMore(int neededPts, int ageGroup, Gender object) {
        final int SITUP_MAX = 25;
        if (neededPts > SITUP_MAX) return "Cannot pass already";
        if (neededPts < 0) return "Pass Already";

        JsonObject obj = object.getSitups();
        JsonObject element;
        // Try 10 times, increasing pts every time
        for (int i = 0; i < 10; i++) {
            for (Map.Entry<String,JsonElement> entry : obj.entrySet()) {
                String key = entry.getKey();
                element = entry.getValue().getAsJsonObject();
                if (element.get(ageGroup + "").getAsInt() == neededPts) {
                    return key;
                }
            }
            neededPts++;
        }
        return "Cannot Calculate";
    }

    public static String countPushupMore(int neededPts, int ageGroup, Gender object) {
        final int PUSHUP_MAX = 25;
        if (neededPts > PUSHUP_MAX) return "Cannot pass already";
        if (neededPts < 0) return "Pass Already";

        JsonObject obj = object.getPushups();
        JsonObject element;
        // Try 10 times, increasing pts every time
        for (int i = 0; i < 10; i++) {
            for (Map.Entry<String,JsonElement> entry : obj.entrySet()) {
                String key = entry.getKey();
                element = entry.getValue().getAsJsonObject();
                if (element.get(ageGroup + "").getAsInt() == neededPts) {
                    return key;
                }
            }
            neededPts++;
        }
        return "Cannot Calculate";
    }

    public static String countRunMore(int neededPts, int ageGroup, Gender object) {
        final int RUN_MAX = 50;
        if (neededPts > RUN_MAX) return "Cannot pass already";
        if (neededPts < 0) return "Pass Already";

        JsonObject obj = object.getRun();
        JsonObject element;
        // Try 10 times, increasing pts every time
        for (int i = 0; i < 10; i++) {
            for (Map.Entry<String,JsonElement> entry : obj.entrySet()) {
                String key = entry.getKey();
                element = entry.getValue().getAsJsonObject();
                if (element.get(ageGroup + "").getAsInt() == neededPts) {
                    return key;
                }
            }
            neededPts++;
        }
        return "Cannot Calculate";


    }
}