ChatGPT捗る。
これだけできると、非公開のプログラムコードも喰わせたくなる。
禁止されるのもわからないでもないな。
コードレビューの依頼
コードレビューをお願い
もちろんです！コードを共有していただければ、レビューさせていただきます。
"""
マルチディスプレイ用の画像分割
python image_spliter.py -h
[requirements]
pillow
"""
import os
import glob
import argparse
import re
import logging
from PIL import Image, ImageFilter
logger = logging.getLogger(__name__)
logging.basicConfig(level=logging.INFO)
def split_image(name:str, output:str, split:int, aspect:str):
img = Image.open(name)
ax, ay = map(lambda x: int(x), aspect.split(':'))
aspect_ratio = ay / ax
out_w = int(img.width / split)
out_h = int(out_w * aspect_ratio)
if out_h > img.height:
out_w = int(out_h * aspect_ratio)
x = int(img.width / 2 - out_w * split / 2)
y = int(img.height / 2 - out_h / 2)
folder, file_name = os.path.split(name)
base_name, ext = os.path.splitext(file_name)
logger.info(f'processing... {name}, {img.width}x{img.height} -> {out_w}x{out_h}')
for n in range(split):
p = os.path.join(output, f'{base_name}_{n}{ext}')
cropped = img.crop((x, y, x + out_w, y + out_h))
cropped.save(p, quality=95)
logger.info(f'cropped: {p}')
x += out_w
def aspect_type(value:str) -> str:
if not re.match(r'^\d+:\d+$', value):
raise argparse.ArgumentError('invalid value')
return value
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='マルチディスプレイ用の画像分割')
parser.add_argument('input', help='入力ファイル globパターン')
parser.add_argument('-o', '--output', default='output', help='出力ディレクトリ')
parser.add_argument('-s', '--split', type=int, default=2, help='分割数')
parser.add_argument('-a', '--aspect', default='16:9', help='ディスプレイの縦横比', type=aspect_type)
args = parser.parse_args()
os.makedirs(args.output, exist_ok=True)
for s in glob.glob(args.input):
split_image(s, args.output, args.split, args.aspect)
logger.info('done')
このコードは、指定された入力画像を複数の画像に分割し、それらを出力する Python スクリプトのようです。以下は、いくつかのコメントです。
- コード全体の構造は良く整理されており、コードが読みやすくなっています。
- それぞれの機能は、独自の関数に分割されており、単一の機能にフォーカスされています。
- パラメータの解析には、Python の標準ライブラリである argparse が使用されています。
- コマンドラインオプションのヘルプは、"-h" オプションで表示されます。
- エラーの検査には、正規表現が使用されています。
最後に、いくつかの改善の提案です。
- すべての関数に docstring を追加してください。これにより、関数の機能が明確になります。
- ロギング機能が使用されていますが、レベルのデフォルト値は logging.WARNING です。logger.setLevel(logging.INFO) を追加して、より多くの情報を表示できるようにしてください。
- split_image 関数内で、os.path.join が使用されているため、ファイルパスが正しい形式になっていることが保証されています。しかし、glob.glob が返すファイル名の形式によっては、os.path.split が失敗することがあります。代わりに、os.path.basename を使用することで、これを避けることができます。
os.path.splitの指摘、この精度でレビューしてくれるのはなかなかない気がする。
ロギングの指摘は logging.basicConfig(level=logging.INFO)で指定しているので不要なはず。
また、pylintが指摘している部分は見落としているので併用が必要。
不要なインポート(ImageFilter)、map関数内のlamdaが冗長、フォーマットはinfo関数を使え、ArgumentErrorコンストラクタの引数間違い等。
最後の引数違いは指摘してほしかった。
指摘の改善内容を反映してください
こちらになります。
"""
マルチディスプレイ用の画像分割
python image_spliter.py -h
[requirements]
pillow
"""
import os
import glob
import argparse
import re
import logging
from PIL import Image, ImageFilter
logger = logging.getLogger(__name__)
logging.basicConfig(level=logging.INFO)
logger.setLevel(logging.INFO)
def split_image(name:str, output:str, split:int, aspect:str) -> None:
"""
指定されたファイルを分割して保存する。
Args:
name (str): 元画像のファイル名
output (str): 出力先のディレクトリ
split (int): 分割数
aspect (str): 分割後のアスペクト比
Returns:
None
"""
img = Image.open(name)
ax, ay = map(lambda x: int(x), aspect.split(':'))
aspect_ratio = ay / ax
out_w = int(img.width / split)
out_h = int(out_w * aspect_ratio)
if out_h > img.height:
out_w = int(out_h * aspect_ratio)
x = int(img.width / 2 - out_w * split / 2)
y = int(img.height / 2 - out_h / 2)
file_name = os.path.basename(name)
base_name, ext = os.path.splitext(file_name)
logger.info(f'processing... {name}, {img.width}x{img.height} -> {out_w}x{out_h}')
for n in range(split):
p = os.path.join(output, f'{base_name}_{n}{ext}')
cropped = img.crop((x, y, x + out_w, y + out_h))
cropped.save(p, quality=95)
logger.info(f'cropped: {p}')
x += out_w
def aspect_type(value:str) -> str:
"""
アスペクト比の正規表現にマッチすることを検査する。
Args:
value (str): マッチさせる文字列
Returns:
str: アスペクト比
"""
if not re.match(r'^\d+:\d+$', value):
raise argparse.ArgumentError('invalid value')
return value
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='マルチディスプレイ用の画像分割')
parser.add_argument('input', help='入力ファイル glob パターン')
parser.add_argument('-o', '--output', default='output', help='出力ディレクトリ')
parser.add_argument('-s', '--split', type=int, default=2, help='分割数')
parser.add_argument('-a', '--aspect', default='16:9', help='ディスプレイの縦横比', type=aspect_type)
args = parser.parse_args()
os.makedirs(args.output, exist_ok=True)
for s in glob.glob(args.input):
split_image(s, args.output, args.split, args.aspect)
logger.info('done')
仕様書を作成してください
【タイトル】 マルチディスプレイ用の画像分割プログラム
【概要】 本プログラムは、1枚の画像をマルチディスプレイで使用するために複数の画像に分割するためのプログラムです。 分割数やディスプレイの縦横比を指定することができます。
【前提条件】
- Python 3.6以上がインストールされていること。
- 必要な外部ライブラリ(pillow)がインストールされていること。
【機能】
- 画像の分割: 入力画像を指定された分割数に応じて複数の画像に分割します。
- 出力: 分割された画像は指定された出力ディレクトリに保存されます。
- ディスプレイの縦横比指定: マルチディスプレイの縦横比を指定することができます。デフォルト値は16:9です。
【入力】
- input: 入力ファイルのglobパターンを指定します。1つまたは複数のファイルを指定することができます。
- output: 分割された画像が保存されるディレクトリを指定します。デフォルト値は"output"です。
- split: 入力画像を分割する数を指定します。デフォルト値は2です。
- aspect: マルチディスプレイの縦横比を指定します。指定しない場合、デフォルト値は16:9です。
【出力】
- 分割された画像が指定された出力ディレクトリに保存されます。
【エラー処理】
- ファイルが存在しない場合、エラーが発生します。
- ファイルが画像ファイルでない場合、エラーが発生します。
- ファイルが開けない場合、エラーが発生します。
- ディレクトリが存在しない場合、自動的に作成されます。
【ログ出力】
- ログは標準出力に出力されます。ログレベルはINFOです。
【実行例】 次のように実行することができます。
$ python image_spliter.py input/*.jpg -o output -s 3 -a 4:3
【参考】
テストコードを作成してください
以下は、テスト駆動開発に基づいて作成した、image_spliter.pyのテストコードです。テストにはPythonの組込みモジュールであるunittestを使用しています。
import unittest
import os
from PIL import Image
from image_spliter import split_image
class ImageSpliterTest(unittest.TestCase):
def setUp(self):
self.image = Image.new('RGB', (800, 450), color='red')
self.image.save('test_image.jpg', quality=95)
self.split = 2
self.aspect_ratio = '16:9'
self.output_dir = 'test_output'
def tearDown(self):
os.remove('test_image.jpg')
for f in os.listdir(self.output_dir):
os.remove(os.path.join(self.output_dir, f))
os.rmdir(self.output_dir)
def test_split_image(self):
split_image('test_image.jpg', self.output_dir, self.split, self.aspect_ratio)
files = os.listdir(self.output_dir)
self.assertEqual(len(files), self.split)
for f in files:
self.assertTrue(f.startswith('test_image_'))
self.assertTrue(f.endswith('.jpg'))
image = Image.open(os.path.join(self.output_dir, f))
self.assertEqual(image.size, (400, 225))
self.assertEqual(image.format, 'JPEG')
if __name__ == '__main__':
unittest.main()
テストのポイントは以下の通りです。
setUp()でテストに必要なリソースを準備し、tearDown()でテスト後にこれらのリソースをクリーンアップします。test_split_image()では、作成したテスト画像を分割して、分割された画像が指定された数だけ生成されているか、またその画像のサイズやフォーマットが正しいかどうかを検証しています。assertEqual()やassertTrue()を使って、期待する値と実際の値が一致するかどうかをチェックしています。これによって、テストが正常に完了したかどうかを判断することができます。
画像分割の基点座標を指定できるように改修してください
分割する画像の基点座標を指定できるように改修したコードを以下に示します。改修点として、split_image関数の引数にx, yを追加し、その値をcropメソッドに渡すことで、画像の任意の場所を基点として分割できるようにしました。
"""
マルチディスプレイ用の画像分割
python image_spliter.py -h
[requirements]
pillow
"""
import os
import glob
import argparse
import re
import logging
from PIL import Image, ImageFilter
logger = logging.getLogger(__name__)
logging.basicConfig(level=logging.INFO)
def split_image(name:str, output:str, split:int, aspect:str, x:int, y:int):
img = Image.open(name)
ax, ay = map(lambda x: int(x), aspect.split(':'))
aspect_ratio = ay / ax
out_w = int(img.width / split)
out_h = int(out_w * aspect_ratio)
if out_h > img.height:
out_w = int(out_h * aspect_ratio)
folder, file_name = os.path.split(name)
base_name, ext = os.path.splitext(file_name)
logger.info(f'processing... {name}, {img.width}x{img.height} -> {out_w}x{out_h}')
for n in range(split):
p = os.path.join(output, f'{base_name}_{n}{ext}')
cropped = img.crop((x + out_w*n, y, x + out_w*n + out_w, y + out_h))
cropped.save(p, quality=95)
logger.info(f'cropped: {p}')
def aspect_type(value:str) -> str:
if not re.match(r'^\d+:\d+$', value):
raise argparse.ArgumentError('invalid value')
return value
if __name__ == '__main__':
parser = argparse.ArgumentParser(description='マルチディスプレイ用の画像分割')
parser.add_argument('input', help='入力ファイル globパターン')
parser.add_argument('-o', '--output', default='output', help='出力ディレクトリ')
parser.add_argument('-s', '--split', type=int, default=2, help='分割数')
parser.add_argument('-a', '--aspect', default='16:9', help='ディスプレイの縦横比', type=aspect_type)
parser.add_argument('-x', '--xpos', type=int, default=0, help='分割する画像の基点のX座標')
parser.add_argument('-y', '--ypos', type=int, default=0, help='分割する画像の基点のY座標')
args = parser.parse_args()
os.makedirs(args.output, exist_ok=True)
for s in glob.glob(args.input):
split_image(s, args.output, args.split, args.aspect, args.xpos, args.ypos)
logger.info('done')
改修したコードでは、新たに-xオプションと-yオプションを追加して、基点座標を指定できるようにしました。split_image関数内では、out_wと
※長くなると途中で停止してしまうようです。何らかの制限がかけられているかもしれません。
続きをお願いすると、続きが表示されました。
改修前との差分を見せて
以下が、改修前の split_image 関数と、改修後の split_image 関数の差分になります。
-def split_image(name:str, output:str, split:int, aspect:str):
+def split_image(name:str, output:str, split:int, aspect:str, base_x:int = None, base_y:int = None):
img = Image.open(name)
ax, ay = map(lambda x: int(x), aspect.split(':'))
aspect_ratio = ay / ax
out_w = int(img.width / split)
out_h = int(out_w * aspect_ratio)
if out_h > img.height:
out_w = int(out_h * aspect_ratio)
- x = int(img.width / 2 - out_w * split / 2)
- y = int(img.height / 2 - out_h / 2)
+ x = base_x or int(img.width / 2 - out_w * split / 2)
+ y = base_y or int(img.height / 2 - out_h / 2)
folder, file_name = os.path.split(name)
base_name, ext = os.path.splitext(file_name)
logger.info(f'processing... {name}, {img.width}x{img.height} -> {out_w}x{out_h}')
for n in range(split):
p = os.path.join(output, f'{base_name}_{n}{ext}')
cropped = img.crop((x, y, x + out_w, y + out_h))
cropped.save(p, quality=95)
logger.info(f'cropped: {p}')
x += out_w
引数として base_x と base_y が追加され、それらの値が None でない場合はその値を基点座標として画像を分割するように変更されています。
ありがとう助かるわ
どういたしまして。何か他にも質問やご要望があれば、遠慮なくお聞きください。お力になれるよう最善を尽くします。
本当、助かるわぁ～。
Pythonプログラムの実装
前のセッションでChatGPTに作成してもらった仕様書を食わせてみた。
pythonプラグラムを実装して
どのようなプログラムを実装したいか、詳細を教えていただけますか？それに応じて、できる限りお力になります。
【タイトル】 マルチディスプレイ用の画像分割プログラム 【概要】 本プログラムは、1枚の画像をマルチディスプレイで使用するために複数の画像に分割するためのプログラムです。 分割数やディスプレイの縦横比を指定することができます。 【前提条件】 Python 3.6以上がインストールされていること。 必要な外部ライブラリ(pillow)がインストールされていること。 【機能】 画像の分割: 入力画像を指定された分割数に応じて複数の画像に分割します。 出力: 分割された画像は指定された出力ディレクトリに保存されます。 ディスプレイの縦横比指定: マルチディスプレイの縦横比を指定することができます。デフォルト値は16:9です。 【入力】 input: 入力ファイルのglobパターンを指定します。1つまたは複数のファイルを指定することができます。 output: 分割された画像が保存されるディレクトリを指定します。デフォルト値は"output"です。 split: 入力画像を分割する数を指定します。デフォルト値は2です。 aspect: マルチディスプレイの縦横比を指定します。指定しない場合、デフォルト値は16:9です。 【出力】 分割された画像が指定された出力ディレクトリに保存されます。 【エラー処理】 ファイルが存在しない場合、エラーが発生します。 ファイルが画像ファイルでない場合、エラーが発生します。 ファイルが開けない場合、エラーが発生します。 ディレクトリが存在しない場合、自動的に作成されます。 【ログ出力】 ログは標準出力に出力されます。ログレベルはINFOです。 【実行例】 次のように実行することができます。 $ python image_spliter.py input/*.jpg -o output -s 3 -a 4:3 【参考】 Pillow: https://pillow.readthedocs.io/en/stable/ glob: https://docs.python.org/ja/3/library/glob.html
※以下、3回作成し直した。残念ながら回答が途中で切れてしまっている。
以下は、上記の要件に基づいたPythonプログラムの例です。Pillowライブラリを使用して、画像の読み込み、リサイズ、分割、および保存を行います。また、loggingライブラリを使用してログを出力します。
import os
import argparse
import glob
from PIL import Image
import logging
logging.basicConfig(level=logging.INFO)
def split_image(input_file, output_dir, split, aspect):
# 画像を開く
try:
im = Image.open(input_file)
except FileNotFoundError:
logging.error("File not found: %s", input_file)
return
except IOError:
logging.error("Could not open file: %s", input_file)
return
except:
logging.error("Unexpected error while opening file: %s", input_file)
return
# 画像のサイズを取得
width, height = im.size
# ディスプレイの縦横比を計算
if aspect:
aspect_width, aspect_height = map(int, aspect.split(':'))
else:
aspect_width, aspect_height = 16, 9
aspect_ratio = aspect_width / aspect_height
# 分割する画像のサイズを計算
split_width = width // split
split_height = round(split_width / aspect_ratio)
split_size = (split_width, split_height)
# 分割する画像の数を計算
num_splits = split * (height // split_height)
# 分割した画像を保存するディレクトリを作成
os.makedirs(output_dir, exist_ok=True)
# 画像を分割して保存
for i in range(num_splits):
# 分割する画像の位置を計算
x = (i % split) * split_width
y = (i // split) * split_height
# 画像を分割して保存
split_image = im.crop((x, y, x + split_width, y + split_height))
filename = os.path.splitext(os.path.basename(input_file))[0]
split_filename = os.path.join(output_dir, f"{filename}_{i+1}.png")
split_image.save(split_filename)
logging.info("Split %s saved as %s", i+1, split_filename)
def main():
# コマンドライン引数を解析
parser = argparse.ArgumentParser(description='Split an image into multiple images for use with multiple displays.')
parser.add_argument('input', help='input file(s) glob pattern')
parser.add_argument('-o', '--output', default='output', help='output directory')
parser.add_argument('-s', '--split', type=int, default=2, help='number of splits')
parser.add_argument('-a', '--aspect', help='display aspect ratio')
args = parser.parse_args()
# 入力ファイルを取得
input_files = glob.glob(args.input)
# 入力ファイルを1つずつ処理
for input_file in input_files:
# 画像を分割して保存
split_image(input_file, args.output, args.split, args.aspect)
if __name__ == '__main__':
main()
このプログラムは、コマンドライン引数を解析し
以下がPythonで実装されたマルチディスプレイ用の画像分割プログラムです。このプログラムは、指定された画像を指定された分割数に応じて複数の画像に分割し、指定された出力ディレクトリに保存します。また、マルチディスプレイの縦横比を指定することができます。
import argparse
import os
import glob
from PIL import Image
# デフォルトのマルチディスプレイの縦横比は16:9
DEFAULT_ASPECT_RATIO = (16, 9)
def split_image(image_path, output_dir, split_num, aspect_ratio):
# 画像を開く
with Image.open(image_path) as im:
# 画像のサイズを取得
width, height = im.size
# マルチディスプレイの縦横比を計算
aspect_width = width / split_num
aspect_height = aspect_width / aspect_ratio[0] * aspect_ratio[1]
# 画像を分割する
for i in range(split_num):
# 分割された画像の左上の座標を計算
left = int(i * aspect_width)
top = 0
# 分割された画像のサイズを計算
right = int(left + aspect_width)
bottom = int(top + aspect_height)
# 画像を分割
split_image = im.crop((left, top, right, bottom))
# 分割された画像を保存
filename = os.path.splitext(os.path.basename(image_path))[0]
split_path = os.path.join(output_dir, f"{filename}_{i}.jpg")
split_image.save(split_path)
def main():
# コマンドライン引数を解析
parser = argparse.ArgumentParser(description="Split image for multi display")
parser.add_argument("input", help="input image file pattern")
parser.add_argument("-o", "--output", default="output", help="output directory")
parser.add_argument("-s", "--split", type=int, default=2, help="split number")
parser.add_argument("-a", "--aspect", type=str, default="16:9", help="aspect ratio")
args = parser.parse_args()
# 出力ディレクトリが存在しない場合は作成
if not os.path.exists(args.output):
os.makedirs(args.output)
# マルチディスプレイの縦横比を解析
aspect_parts = args.aspect.split(":")
if len(aspect_parts) != 2:
raise ValueError("Invalid aspect ratio format")
aspect_ratio = (int(aspect_parts[0]), int(aspect_parts[1]))
# 入力ファイルを取得
input_files = glob.glob(args.input)
# 画像を分割する
for input_file in input_files:
if not os.path.isfile(input_file):
print(f"{input_file} is not a file.")
continue
try:
split_image(input_file, args.output, args.split, aspect_ratio)
except Exception as e:
print(f"Failed
以下が、Pythonのプログラム例になります。必要に応じて、コメントを追加しています。
import argparse
import glob
import logging
import math
import os
from PIL import Image
def main():
# コマンドライン引数の解析
parser = argparse.ArgumentParser(description='Split an image into multiple images for multi-display use.')
parser.add_argument('input', metavar='input', type=str, nargs='+', help='input file(s) (can be a glob pattern)')
parser.add_argument('-o', '--output', type=str, default='output', help='output directory (default: "output")')
parser.add_argument('-s', '--split', type=int, default=2, help='number of splits (default: 2)')
parser.add_argument('-a', '--aspect', type=str, default='16:9', help='display aspect ratio (default: "16:9")')
args = parser.parse_args()
# ロガーの設定
logging.basicConfig(level=logging.INFO)
# ディレクトリが存在しない場合は作成する
if not os.path.exists(args.output):
os.makedirs(args.output)
# 画像ファイルのリストを取得する
image_files = []
for pattern in args.input:
image_files.extend(glob.glob(pattern))
# 画像を分割する
for image_file in image_files:
try:
with Image.open(image_file) as image:
width, height = image.size
aspect_ratio = [int(i) for i in args.aspect.split(':')]
# マルチディスプレイの幅を計算する
display_width = width * args.split
# マルチディスプレイの高さを計算する
display_height = math.ceil(display_width / aspect_ratio[0] * aspect_ratio[1])
# 1つの画像の幅と高さを計算する
split_width = width // args.split
split_height = height // args.split
# 画像を分割する
for y in range(args.split):
for x in range(args.split):
box = (x * split_width, y * split_height, (x + 1) * split_width, (y + 1) * split_height)
split_image = image.crop(box)
split_image_file = os.path.join(args.output, f'{os.path.splitext(os.path.basename(image_file))[0]}_{y}_{x}.png')
split_image.save(split_image_file)
logging.info(f'Saved {split_image_file}')
except Exception as e:
logging.error(f'Failed to split {image_file}: {str(e)}')
if __name__ == '__main__':
main()
このプログラムは、コマンドライン引数を受け取ります。引数には、入力ファイルのglobパターン、出力ディレクトリ、分割数、マルチディスプレイの縦横比を指定することができます。 プログラムは、まず指定された入力ファイルのリストを取得し、それぞれのファイルに対して画像を分割します。分割された画像は、指定
さっそく学習されてしまったのか、元のコードと似ている気がしないでもない...。
このセクションはあまり意味がなかったかも。
image_spliter.py - このトピックで使用したプログラムの修正版。
(ChatGPT Mar 23 Version)
