"""
matplotlib ノート
"""

#==============
# import
#==============

from matplotlib import pyplot as plt

#==============
#基本
#==============

#グラフの作成、表示
# marker: o, x etc linestyle: None, -, dashed, dotted etc
# color: red, blue, green, black etc
plt.plot(x, y, label='raw data', marker = 'o', markersize = 0.5, linestyle = 'None', linewidth = 0.5, color = 'red')
plt.plot(xc, yc, label='fitted', linestyle = 'dashed')
#plt.scatter(x, y)
#plt.bar(xindex, v, width = 1.5)
plt.title(infile, fontsize = fontsize)
plt.xlabel(header[0], fontsize = fontsize)
plt.ylabel(header[1], fontsize = fontsize)
plt.legend(fontsize = fontsize)
#フォントサイズ
plt.legend(fontsize = legend_fontsize)
plt.tick_params(labelsize = fontsize)

#グラフを表示してプログラムを一時停止
plt.show()


#==============
#グラフのクリアー
#==============

#グラフ全体のクリアー (clear figure)
# clears the entire current figureをすべての軸で表示しますが、
# 他のプロットで再利用できるように、ウィンドウを開いたままにしておきます。
    figure.clf()

#グラフのクリアー (clear axis)
#clears an axis、すなわち、現在の図形内の現在アクティブな軸。それは他の軸を手つかずのままにします。
    figure.cla()
# closes a windowは、特に指定がない場合は現在のウィンドウになります。
    plt.close()

#=========================================
#複数グラフ　plotcsv_mult.py
#=========================================

# figureオブジェクト取得
    fig = plt.figure(figsize = (15,7))

#縦1、横2に分割し、1番目のグラフ枠をax1に代入
    ax1 = fig.add_subplot(1, 2, 1)
# ax1 = fig.add_subplot(121)
    ax1.set_yscale('log')
#プロットしたデータのオブジェクト
    data = ax1.plot(xT, yEF, label = 'alpha')
    ax1.plot(xT, yEFapprox, label = 'alpha(approx)', linestyle = '-', linewidth = 0.5, color = 'red')
    ax1.set_xlabel("T (K)")
    ax1.set_ylabel("alpha = -mu/(kBT)")
    ax1.set_xlim([Tmin, Tmax])
    ax1.set_ylim([0.0, max(yEF)])
#フォントサイズ
    ax1.tick_params(labelsize = fontsize)
#フォントサイズを指定して凡例を表示
    ax1.legend(fontsize = legend_fontsize)
#グラフ間スペースの自動調整
    plt.tight_layout()
#グラフ間スペースの手動調整
    plt.subplots_adjust(wspace=0.4, hspace=0.6)

#グラフを表示。0.001秒スリープし、プログラムを続行
    plt.pause(0.001)

    print("Press ENTER to exit>>", end = '')
    input()


#=========================================
#3Dグラフ optimize-sd-cg2d-linesearch.py
#=========================================

import matplotlib.pyplot as plt
from mpl_toolkits.mplot3d.axes3d import Axes3D
from matplotlib import cm

        fig = plt.figure(figsize = (10, 5))
        ax = fig.add_subplot(1, 2, 1)
        ax2 = fig.add_subplot(1, 2, 2, projection='3d')
# surf = ax.plot_surface(xg, yg, zg, cmap=cm.coolwarm)
        surf = ax2.plot_wireframe(xg, yg, zg, rstride=2, cstride=2)
# ax.set_zlim3d(-3.01, 3.01)
# plt.colorbar(surf, shrink=0.5, aspect=10)
        ax.set_title('contour')
        contour = ax.contourf(xg, yg, zg, levels = 51, cmap = 'Spectral')
# contour = ax.contour(xg, yg, zg, levels = 51, cmap = 'Spectral')
        solve, = ax.plot([], color = 'blue', marker = 'o', linestyle = '-')
        plt.pause(0.1)

    for iter in range(nmaxiter):
#データのリストを更新
            xt.append(x0[0])
            yt.append(x0[1])
#データをセット
            solve.set_data(xt, yt)
            plt.pause(0.1)
            time.sleep(tsleep)


#=========================================
#グラフの更新　diffeq2nd_planet.py　plot_realtimeupdate.py
#=========================================

    fig, ax = plt.subplots(1, 1)

    # 一度plotする、プロットデータのオブジェクトを受け取る受け取る
    lines, = ax.plot(x, y)

    while True:
        # plotデータの更新
        x.append(x[n-1] + 0.1)
        y.append(np.sin(x[n]))

        # 描画データを更新するときにplot関数を使うと
        # lineオブジェクトが増えてしまう。
        # set_data()メソッドで描画データを更新する
        lines.set_data(x, y)

        # set_data()を使うと軸とかは自動設定されない
        # x軸の範囲は適宜修正してやる必要がある．
        ax.set_xlim((min(x), max(x)))
        ax.set_ylim((min(y), max(y)))

        # plt.show() プログラムが停止する
        # plt.ion() + plt.draw() グラフウインドウが固まってプログラムが止まる
        plt.pause(0.01)


#=========================================
#グラフの更新 (グラフをクリアーして再描画)
#=========================================

    plt.clf()
    plt.title(infile, fontsize = fontsize)
    plt.xlabel(header[0], fontsize = fontsize)
    plt.ylabel('log(' + header[1] + ')', fontsize = fontsize)
    plt.plot(Vd, logId, label='raw data', marker = 'o', markersize = 0.5, linestyle = 'None')
    plt.plot(Vi, logIi, label='initial', linestyle = 'dotted')
    if ai is not None:
        Vc, Ic, logIc = CalIV(ai)
        plt.plot(Vc, logIc, label='final', linestyle = 'dashed')
#フォントサイズ
    plt.legend(fontsize = legend_fontsize)
    plt.tick_params(labelsize = fontsize)
#グラフ間スペースの自動調整
    plt.tight_layout()
    plt.pause(0.001)


#=========================================
# callback　peakfit-scipy-minimize.py
#=========================================

def callback(xk):
    global xd
    global iter
    global figure, ax_fit, ax_conv
    global fitdata, convdata

    fmin = CalS2(xk)
    print("callback {}: xk={}".format(iter, xk))
    print(" fmin={}".format(fmin))
    iter += 1
    xiter.append(iter)
    yfmin.append(fmin)

    convdata[0].set_data(xiter, yfmin)
    ax_conv.set_xlim((0.0, max(xiter) + 1.0))
    ax_conv.set_ylim((min(yfmin) * 0.8, max(yfmin) * 1.2))

    yc = ycal_list(xd, xk)
    fitdata[0].set_data(xd, yc)

    plt.pause(0.2)

def main():
        figure = plt.figure(figsize = (10, 5))
        ax_fit = figure.add_subplot(1, 2, 1)
        rawdata = ax_fit.plot(xd, yd, color = 'blue', linestyle = '', linewidth = 0.5,
                            fillstyle = 'full', marker = 'x', markersize = 5)
        inidata = ax_fit.plot(xd, yini, color = 'black', linestyle = 'dashed', linewidth = 0.5)
        fitdata = ax_fit.plot([], [], color = 'red', linestyle = '-', linewidth = 0.5)

        ax_conv = figure.add_subplot(1, 2, 2)
        ax_conv.set_yscale('log')
        convdata = ax_conv.plot([], [], color = 'black', linestyle = '-', linewidth = 0.5,
                            fillstyle = 'full', marker = 'o', markersize = 5)
        plt.pause(0.001)


    res = minimize(CalS2, x0, jac=diff1, method=method, tol = tol, callback = callback,
                options = {'maxiter':maxiter, "disp":True})


#=========================================
tkinterに統合 11-leastsq-GUI-tkmatplot.py
#=========================================

import matplotlib
matplotlib.use('TkAgg')
from matplotlib import pyplot as plt
from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg

# Canvas (right)
# figure = plt.subplots()
    figure = plt.Figure()
# figure = plt.Figure(figsize=(10, 3), dpi=100)
    canvas = FigureCanvasTkAgg(figure, master = right_frame)
    canvas.get_tk_widget().pack(side = 'bottom', expand = 1)
    canvas._tkcanvas.pack(side = 'bottom', expand = 1)
    obj['figure'] = figure
    obj['canvas'] = canvas

    figure = obj['figure']
    figure.clf()
    ax = figure.add_subplot(111)
    ax.plot(x, y, label = 'raw data', marker = 'o', linestyle = 'None')
    ax.plot(xc, yc, label = 'fitted', linestyle = 'dashed')
    ax.set_title(obj['title'].get(), fontsize = fontsize)
    ax.set_xlabel(obj['xlabel'].get(), fontsize = fontsize)
    ax.set_ylabel(obj['ylabel'].get(), fontsize = fontsize)
    ax.legend(fontsize = fontsize)
    ax.tick_params(labelsize = fontsize)
#グラフ間スペースを自動調整
    figure.tight_layout()
    canvas.draw()