2019-02-26 16:31:43 +01:00
|
|
|
#dummy module
|
|
|
|
import asyncio
|
2019-02-26 19:00:08 +01:00
|
|
|
import fractale.source.main
|
2019-02-26 16:31:43 +01:00
|
|
|
import random
|
|
|
|
import discord
|
2019-02-27 02:49:02 +01:00
|
|
|
import os
|
2019-02-26 19:00:08 +01:00
|
|
|
from PIL import Image
|
2019-02-27 02:49:02 +01:00
|
|
|
from concurrent.futures import ThreadPoolExecutor
|
2019-02-26 16:31:43 +01:00
|
|
|
class MainClass():
|
|
|
|
def __init__(self, client, modules, owners, prefix):
|
|
|
|
self.client = client
|
|
|
|
self.modules = modules
|
|
|
|
self.owners = owners
|
|
|
|
self.prefix = prefix
|
|
|
|
self.events=['on_message'] #events list
|
|
|
|
self.command="%sfractale"%self.prefix #command prefix (can be empty to catch every single messages)
|
|
|
|
|
|
|
|
self.name="Fractale"
|
|
|
|
self.description="Module de génération de fractales"
|
|
|
|
self.interactive=True
|
|
|
|
self.color=0x78ffc3
|
|
|
|
self.help="""\
|
2019-02-26 19:00:08 +01:00
|
|
|
</prefix>fractale [fractale] [nombre d'itérations]
|
|
|
|
=> Génère une image fractale. (Si on met le nombre d'itérations, on doit mettre le nom de la fractale.)
|
|
|
|
|
2019-02-27 02:49:02 +01:00
|
|
|
-> Valeurs possible pour [fractale] + nb d'itérations max entre parentheses
|
|
|
|
```..: Toutes les fractales:
|
|
|
|
......: von_koch_curve_flake (7)
|
|
|
|
......: von_koch_curve (7)
|
|
|
|
......: blanc_manger (20)
|
|
|
|
......: dragon
|
|
|
|
......: sierpinski_triangle
|
|
|
|
......: fractal_plant
|
|
|
|
......: koch_curve_right_angle
|
|
|
|
......: fractal_binary_tree```
|
2019-02-26 16:31:43 +01:00
|
|
|
"""
|
2019-02-27 02:49:02 +01:00
|
|
|
self.fractals={
|
2019-02-27 03:01:00 +01:00
|
|
|
"von_koch_curve_flake":{"Type":"Figures", "Max":((5000,5000), 5000, 10), "Min":((0,0),0,1)},
|
|
|
|
"von_koch_curve":{"Type":"Figures", "Max":((5000,5000),(5000,5000),10), "Min":((0,0),(0,0),1)},
|
2019-02-27 02:49:02 +01:00
|
|
|
"blanc_manger":{"Type":"Figures", "Max":((5000,5000),(5000,5000),10), "Min":((0,0),(0,0),1)},
|
|
|
|
"dragon":{"Type":"Lsystem"},
|
|
|
|
"sierpinski_triangle":{"Type":"Lsystem"},
|
|
|
|
"fractal_plant":{"Type":"Lsystem"},
|
|
|
|
"koch_curve_right_angle":{"Type":"Lsystem"},
|
|
|
|
"fractal_binary_tree":{"Type":"Lsystem"}
|
|
|
|
}
|
2019-02-26 16:31:43 +01:00
|
|
|
async def on_message(self, message):
|
2019-02-26 19:00:08 +01:00
|
|
|
args=message.content.split(" ")
|
2019-02-26 16:31:43 +01:00
|
|
|
tmpstr="/tmp/%s.png"%random.randint(1,10000000)
|
2019-02-26 19:00:08 +01:00
|
|
|
im=Image.new('RGB', (5000, 5000), (0, 0, 0))
|
|
|
|
fig = fractale.source.main.Figures(im=im)
|
2019-02-27 02:49:02 +01:00
|
|
|
|
2019-02-26 19:00:08 +01:00
|
|
|
if len(args)==1 :
|
2019-02-27 02:49:02 +01:00
|
|
|
await self.client.loop.run_in_executor(ThreadPoolExecutor(), fig.von_koch_curve_flake,*((2500, 2500), 2000, 5))
|
2019-02-26 19:00:08 +01:00
|
|
|
elif args[1].lower()=="blanc_manger" :
|
|
|
|
iterations=7
|
|
|
|
if len(args)>2 and int(args[2])<=20:
|
|
|
|
iterations=int(args[2])
|
2019-02-27 02:49:02 +01:00
|
|
|
await self.client.loop.run_in_executor(ThreadPoolExecutor(), fig.blanc_manger,*((1000, 1000), (4000, 4000), iterations))
|
2019-02-27 03:06:57 +01:00
|
|
|
elif args[1].lower()=="von_koch_curve_flake" :
|
2019-02-26 19:00:08 +01:00
|
|
|
iterations=5
|
|
|
|
if len(args)>2 and int(args[2])<=7:
|
|
|
|
iterations=int(args[2])
|
2019-02-27 02:49:02 +01:00
|
|
|
await self.client.loop.run_in_executor(ThreadPoolExecutor(), fig.von_koch_curve_flake,*((2500, 2500), 2000, iterations))
|
2019-02-27 03:06:57 +01:00
|
|
|
elif args[1].lower()=="von_koch_curve" :
|
|
|
|
iterations=5
|
|
|
|
if len(args)>2 and int(args[2])<=7:
|
|
|
|
iterations=int(args[2])
|
|
|
|
await self.client.loop.run_in_executor(ThreadPoolExecutor(), fig.von_koch_curve,*((0, 2500), (5000,2500), iterations))
|
|
|
|
else:
|
|
|
|
await self.modules['help'][1].send_help(message.channel, self)
|
|
|
|
return
|
2019-02-26 19:00:08 +01:00
|
|
|
im.save(tmpstr)
|
2019-02-26 16:31:43 +01:00
|
|
|
await message.channel.send(file=discord.File(tmpstr))
|
2019-02-27 02:49:02 +01:00
|
|
|
os.remove(tmpstr)
|