from abc import ABC

from plotly import graph_objs as go


class AtomBuilder(ABC):
    def __init__(self, data=None, layout=None):
        if data is None:
            data = {}
        if layout is None:
            layout = {}
        self.data = data
        self.layout = layout


class Pie(AtomBuilder):
    def __init__(self, labels, values, **kwargs):
        super().__init__(data=go.Pie(labels=labels, values=values, **kwargs))


class Scatter(AtomBuilder):
    def __init__(self, x, y, mode='markers', **kwargs):
        super().__init__(data=go.Scatter(x=x, y=y, mode=mode, **kwargs))


class Line(AtomBuilder):
    def __init__(self, x, y, mode='lines', **kwargs):
        super().__init__(data=go.Scatter(x=x, y=y, mode=mode, **kwargs))


class Box(AtomBuilder):
    def __init__(self, y, **kwargs):
        super().__init__(data=go.Box(y=y, **kwargs))

    def horizontal(self):
        if 'x' in self.data:
            self.data.x, self.data.y = self.data.y, self.data.x
        else:
            self.data['x'] = self.data.y
            del self.data['y']
        self.data['orientation'] = 'h'
        return self


class Bar(AtomBuilder):
    def __init__(self, x, y, **kwargs):
        super().__init__(data=go.Bar(x=x, y=y, **kwargs))

    def horizontal(self):
        self.data.x, self.data.y = self.data.y, self.data.x
        self.data['orientation'] = 'h'
        return self


class Histogram(AtomBuilder):
    def __init__(self, x, **kwargs):
        super().__init__(data=go.Histogram(x=x, **kwargs))

    def xbins(self, start=None, end=None, size=None, bins=None):
        if start is None:
            start = min(self.data['x'])
        if end is None:
            end = max(self.data['x'])
        if size is None:
            if bins is None:
                raise ValueError('size and bins can not be both none')
            else:
                size = (end - start) / bins
        self.data['xbins'] = dict(start=start, end=end, size=size)
        self.data['autobinx'] = False
        return self


class Heatmap(AtomBuilder):
    def __init__(self, z, **kwargs):
        super().__init__(data=go.Heatmap(z=z, **kwargs))


class Scatter3D(AtomBuilder):
    def __init__(self, x, y, z, mode='markers', **kwargs):
        super().__init__(data=go.Scatter3d(x=x, y=y, z=z, mode=mode, **kwargs))


class Line3D(AtomBuilder):
    def __init__(self, x, y, z, mode='lines', **kwargs):
        super().__init__(data=go.Scatter3d(x=x, y=y, z=z, mode=mode, **kwargs))


class Surface(AtomBuilder):
    def __init__(self, z, **kwargs):
        super().__init__(data=go.Surface(z=z, **kwargs))