14-29. 再帰的な処理で葉脈を作ろう

import cv2
import numpy as np
import matplotlib.pyplot as plt

#画像表示関数
def show(img):
    plt.imshow(img, vmin=0, vmax=255)
    plt.show()


#サイズ
size = 512

#初期化（画像はこの後ずっとグローバル参照される）
img = np.zeros((size, size))

#1本目
start = np.array([0, 0])
end = np.array([size, size])
cv2.line(img, tuple(start), tuple(end), 255, 2)
show(img)

#枝を追加する関数
def add_branch(start, end, depth):
    #スタートとエンドの中点
    middle = (start + end) / 2
    print(middle)
    
    #回転行列
    t1 = np.deg2rad(60)
    R1 = np.array([[np.cos(t1), -np.sin(t1)],
                   [np.sin(t1),  np.cos(t1)]])
    t2 = np.deg2rad(-60)
    R2 = np.array([[np.cos(t2), -np.sin(t2)],
                   [np.sin(t2),  np.cos(t2)]])
    
    #新しい枝のベクトル
    vec = (end - middle) * 0.7 #少し短くしておく
    vec1 = np.dot(R1, vec)
    vec2 = np.dot(R2, vec)
    
    #各点をintにしておく
    middle = middle.astype(int)
    vec1 = vec1.astype(int)
    vec2 = vec2.astype(int)
    
    #新しい枝の書き込み
    cv2.line(img, tuple(middle), tuple(middle + vec1), 255, 2)
    cv2.line(img, tuple(middle), tuple(middle + vec2), 255, 2)
    
    #現在の深さ
    current_depth = depth + 1
    
    #深さが足りなければ再帰的に実行
    if current_depth < stop_depth:
        add_branch(middle, middle + vec1, depth=current_depth)
        add_branch(middle, middle + vec2, depth=current_depth)


#深さ
stop_depth = 1

#再帰的な枝の追加
add_branch(start, end, depth=0)
show(img)

[256. 256.]

[256. 256.]
[223.5 378. ]
[180.5 389. ]
[169.5 378.5]
[176.  403.5]
[254. 409.]
[250.5 423.5]
[268.5 405.5]
[378.  223.5]
[409. 254.]
[405.5 268.5]
[423.5 250.5]
[389.  180.5]
[403.5 176. ]
[378.5 169.5]