from matplotlib.figure import Figure


class Graph:
    def __init__(
        self, x, y, labels, mult_subplots=0, xl="x", yl="y", linetype="k", lims=False
    ):
        # словарь из настроек?
        self.labels = labels
        self.__x = x
        self.__y = y
        self.__fig = Figure(figsize=(5, 4), dpi=100)
        self.__chb = mult_subplots
        self.__ax = self.__fig.add_subplot()

        # сделать отслеживание за состояние кнопок потом
        """
        if self.__chb==0:
            if len(self.__x)>1:
                for i in range(len(self.__x)):
                    n=int(f"{len(self.__x)}1{i+1}")
                
                    axes=self.__fig.add_subplot(n)
                    axes.set_title(f"График №{i+1}")
                    c=axes.plot(self.__x[i],self.__y[i],label=lab)
            else:
                line=self.__ax.plot(self.__x[0],self.__y[0],label=lab,)
                print(self.__x)

        else:
            if len(self.__x)>1: #много графиков на одном холсте
                for i in range(len(self.__x)):
                    self.__ax.plot(self.__x[i],self.__y[i])
                    
            
            else:

                line=self.__ax.plot(self.__x[0],self.__y[0],label=lab,)
                print(self.__x)
        """
        self.draw_n_func_plot()
        self.draw_one_func()
        self.draw_subplots()

        self.__ax.set_xlabel(xl)
        self.__ax.set_ylabel(yl)
        self.__ax.legend()
        self.figure.tight_layout()

    def add_vline(ax, x=0, ymin=0, ymax=1):
        ax.axvline(x, ymin, ymax)
        # принимаем плот и на нем же рисуем

    def add_hline(
        ax,
        y=0,
        xmin=0,
        xmax=1,
    ):
        ax.axhline(y, xmin, xmax)

    def draw_n_func_plot(
        self,
    ):  # много графиков на одном холсте
        if self.__chb == 0 and len(self.__x) > 1:
            for i in range(len(self.__x)):
                self.__ax.plot(self.__x[i], self.__y[i], label=self.labels[i])

    def draw_one_func(
        self,
    ):
        if len(self.__x) == 1:
            self.__ax.plot(self.__x[0], self.__y[0], label=self.labels[0])  #

    def draw_subplots(
        self,
    ):
        if self.__chb == 1 and len(self.__x) > 1:
            for i in range(len(self.__x)):
                n = int(f"{len(self.__x)}1{i+1}")

                axes = self.__fig.add_subplot(n)
                axes.set_title(f"График №{i+1}")
                axes.plot(self.__x[i], self.__y[i], label=self.labels[i])

    @property
    def figure(
        self,
    ):
        return self.__fig