mainから他の関数を呼び出す方法を教えてほしいです。

Question

前提・実現したいこと

pythonでパワースペクトル画像の特徴を分析したいと思っています.
現在、画像上の角度方向のスペクトル分布を求めてるもの（コード１）とパワースペクトルの角度方向（コード２）を求めるためのものはあります。```
現在、コード１において、読み込んだ画像に２次元FFTをし、そこからパワースペクトルを求めることはできています。ただ、コード１で読み込んだ画像を使ったものをコード２でも適用させる方法がいくら調べてもわかりません。詳しい方に聞くと、mainからdefを呼び出して、returnされた値をplotするだけで、def以下のコードは何も修正する必要はないとのことでした。ただ、知識が浅く、具体的な方法がわかりません。もし、時間があれば回答お願いします。

-- coding: utf-8 --#コード１

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

def main():
# 入力画像をグレースケールで読み込み
img = cv2.imread("input.png")

# グレースケール変換
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

# 高速フーリエ変換(2次元)
fimg = np.fft.fft2(gray)

# 第1象限と第3象限, 第2象限と第4象限を入れ替え
fimg =  np.fft.fftshift(fimg)

# パワースペクトルの計算
mag = 20*np.log(np.abs(fimg))

# 入力画像とスペクトル画像をグラフ描画
plt.subplot(121)
plt.imshow(gray, cmap = 'gray')
plt.subplot(122)
plt.imshow(mag, cmap = 'gray')
plt.show()

if name == "main":
main()

#コード２
import numpy as np

複素平面を構成する値の配列を生成(原点は中心)

def complex_plane(width):
half_width = width // 2
re = np.array(range(width)) - half_width
im = - re
re, im = np.meshgrid(re, im)
return re + im * 1j

スペクトルを与えられた角度の範囲内だけ集計

def aggregate_in_angle(agg_fun, spectrum, min_angle, max_angle):
width = spectrum.shape[0]
assert spectrum.shape[1] == width
min_radius, max_radius = 0, width // 2
assert 0 <= min_angle < max_angle <= 180 # 単位はdegree

cp = complex_plane(width)

cp_mag = np.abs(cp)
in_radius = np.logical_and(min_radius < cp_mag, cp_mag < max_radius)

cp_angle = np.angle(cp, deg=True)
in_angle = np.logical_and(min_angle <= cp_angle, cp_angle < max_angle)

return agg_fun(spectrum[np.logical_and(in_radius, in_angle)])

def test(spectrum, angle_gap):
# 指定角度ごとの平均を求める
means = [aggregate_in_angle(np.mean, spectrum, angle, angle + angle_gap)
for angle in range(0, 180, angle_gap)]
return means

### 発生している問題・エラーメッセージ
pythonを始めたばかりで、defについて勉強しているのですが、まだまだわからないことも多く、複数の関数をつなげることがまだできません。

エラーメッセージ

### 該当のソースコード

```ここに言語名を入力python

ソースコード

試したこと

ここに問題に対して試したことを記載してください。

補足情報（FW/ツールのバージョンなど）

ここにより詳細な情報を記載してください。

Comments

[meg_]

コードは「コードの挿入｣で記入してください。