#!/usr/bin/env python
# -*- coding: utf-8 -*- 
"""scatterplot.py
"""
from __future__ import print_function
import sys
import os
from bashplotlib.core import helpers

def get_scale(series, is_y=False, steps=20):
    try:
        min_val = min(series)
        max_val = max(series)
        scaled_series = []
        for x in helpers.drange(min_val, max_val, (max_val-min_val)/steps):
            if x > 0 and scaled_series and max(scaled_series) < 0:
                scaled_series.append(0.0)
            scaled_series.append(x)
        if is_y:
            scaled_series.reverse()
        return scaled_series
    except Exception, err:
        print(err)
        # print(series)
        print(is_y)
        print(steps)

def plot_scatter(filename=None, xs=None, ys=None, size=20, pch='x', colour='white', title=None, skipheader=False):
    """plot (X,y) pairs from columns in a file or 2 separate files
    
        Arguments:
            xs                   List of values to use on X axis [default: None]
            ys                   List of values to use on y axis [default: None]
            filename             Name of a csv file [default: None]
            skipheader           Skip the first row [default: False]
            size                 Marketsize [default: 20]
            pch                  Markershape [default: x]
            colour               Marker colour [default: white]
            title                Title [default: ]

    """

    if filename:
        data = helpers.read_xy_pairs_from_csv(filename, X=0, y=1, header=skipheader)
        xs = [i[0] for i in data]
        ys = [i[1] for i in data]
    else:
        xs = [line.strip() for line in open(xs, 'r').readlines()]
        ys = [line.strip() for line in open(ys, 'r').readlines()]
        if skipheader:
            xs = xs[1:]
            ys = ys[1:]
        xs = map(float, xs)
        ys = map(float, ys)
    if (xs is None or ys is None):
        raise Exception("Missing `xs` or `ys` data.")
    plotted = set()
    if title:
        print(helpers.box_text(title, 2 * len(get_scale(xs, False, size)) + 1))

    print("-" * (2 * len(get_scale(xs, False, size)) + 2))
    for y in get_scale(ys, True, size):
        print("|", end=' ')
        for x in get_scale(xs, False, size):
            point = " "
            for (i, (xp, yp)) in enumerate(zip(xs, ys)):
                if xp <= x and yp >= y and (xp, yp) not in plotted:
                    point = pch
                    #point = str(i)
                    plotted.add((xp, yp))
            if x == 0 and y == 0:
                point = "o"
            elif x == 0:
                point = "|"
            elif y == 0:
                point = "-"
            helpers.printcolour(point, True, colour)
        print("|")
    print("-" * (2 * len(get_scale(xs, False, size)) + 2))