#include "histogram.h"

void find_minmax(const std::vector<double>& numbers, double& min, double& max) {
    min = numbers[0];
	max = numbers[0];
	for (double x : numbers){
    	if (x < min){
        	min = x;
    	}
    	else if (x > max){
        	max = x;
    	}
	}
    return;
}

std::vector<size_t> make_histogram(const std::vector<double>& numbers, size_t& bin_count) {
    double minn, maxn;
    find_minmax(numbers, minn, maxn);
    double bin_size = (maxn - minn) / bin_count;
    std::vector<size_t> bins(bin_count, 0);
    for (size_t i = 0; i < numbers.size(); i++) {
        bool found = false;
        for (size_t j = 0; (j < bin_count - 1) && !found; j++) {
            auto lo = minn + j * bin_size;
            auto hi = minn + (j + 1) * bin_size;
            if ((numbers[i] >= lo) && (numbers[i] <= hi)) {
                bins[j]++;
                found = true;
            }
        }
        if (!found) {
            bins[bin_count - 1]++;
        }
    }
    return bins;
}

void show_histogram_text(const std::vector<size_t> bins, size_t bin_count) {
    size_t max_count = bins[0];
    for (size_t i = 1; i < bin_count; i++) {
        if (bins[i] > max_count) {
            max_count = bins[i];
        }
    }

    const size_t SCREEN_WIDTH = 80;
    const size_t MAX_ASTERISK = SCREEN_WIDTH - 3 - 1;


    short space_count;

    std::cerr << "\nHistogram:" << std::endl;

    if (max_count <= MAX_ASTERISK) {
        for (size_t i = 0; i < bin_count; i++) {
            if (bins[i] < 10) {
                space_count = 2;
            } else if (bins[i] >= 10 && bins[i] < 100) {
                space_count = 1;
            } else if (bins[i] >= 100 && bins[i] < 1000) {
                space_count = 0;
            }



            for (size_t k = 0; k < space_count; k++) {
                std::cout << " ";
            }

            std::cout << bins[i];

            std::cout << "|";


            for (size_t j = 0; j < bins[i]; j++) {
                std::cout << "*";
            }

            std::cout << "\n";
        }
    } else {
        for (size_t i = 0; i < bin_count; i++) {
            if (bins[i] < 10) {
                space_count = 2;
            } else if (bins[i] >= 10 && bins[i] < 100) {
                space_count = 1;
            } else if (bins[i] >= 100 && bins[i] < 1000) {
                space_count = 0;
            }


            for (size_t k = 0; k < space_count; k++) {
                std::cout << " ";
            }

            std::cout << bins[i];

            std::cout << "|";

            size_t height = static_cast<size_t>(MAX_ASTERISK * (static_cast<double>(bins[i]) / max_count));
            for (size_t j = 0; j < height; j++) {
                std::cout << "*";
            }

            std::cout << "\n";
        }
    }
}