Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
movie_studio
import os
import time
from multiprocessing import Process
import cocos
import cv2
import numpy as np
import pyglet
from PIL import ImageGrab
from cocos.actions import CallFunc
from cocos.actions.interval_actions import FadeIn, FadeOut, Delay
from cocos.director import director
from cocos.scenes import SplitColsTransition
class Background(cocos.layer.Layer):
_handlers_enabled = True
def __init__(self, image_name, next_scene, duration):
super(Background, self).__init__()
self.next_scene = next_scene
self.background = cocos.sprite.Sprite(image_name)
self.background.position = 1920 / 2, 1080 / 2
self.background.scale = 1
self.add(self.background, z=0)
action = Delay(duration) + CallFunc(self.action_end)
self.background.do(action)
def action_end(self):
if self.next_scene:
cocos.director.director.replace(cocos.scenes.FlipY3DTransition(self.next_scene, duration=1))
else:
cocos.director.director.pop()
class CharacterLayer(cocos.layer.Layer):
def __init__(self, character_list):
super(CharacterLayer, self).__init__()
for character in character_list:
character_image_path = os.path.join(character_collection_path, character["category"],
character["file_name"])
if character["file_name"].endswith("gif"):
temp_spirit = cocos.sprite.Sprite(pyglet.image.load_animation(character_image_path))
else:
temp_spirit = cocos.sprite.Sprite(pyglet.image.load(character_image_path))
temp_spirit.scale = character["scale"]
temp_spirit.rotation = character["rotation"]
temp_spirit.position = character["position"]
temp_spirit.scale_x = -1 if character["flip"][0] else 1
temp_spirit.scale_y = -1 if character["flip"][1] else 1
self.add(temp_spirit, z=1)
class Bubble(cocos.layer.Layer):
def __init__(self, text):
super(Bubble, self).__init__()
bubble = cocos.sprite.Sprite("dialog.png")
bubble.position = 1920 / 2 + 150, 1080 / 2
bubble.scale = 0.6
label = cocos.text.Label(text,
font_name='Times New Roman',
font_size=20,
color=(234, 122, 121, 112),
anchor_x='center', anchor_y='center')
label.position = 1920 / 2 + 150, 1080 / 2
fadein_action = FadeIn(1.0)
bubble.do(fadein_action)
label.do(fadein_action)
self.add(bubble, z=2)
self.add(label, z=3)
class BubbleLayer(cocos.layer.Layer):
def __init__(self, dialoglist):
super(BubbleLayer, self).__init__()
for bubble in dialoglist:
temp_bubble = cocos.sprite.Sprite(bubble_collection_path)
temp_label = cocos.text.Label(bubble["text"], font_name='Times New Roman', font_size=24,
color=(234, 122, 121, 112), anchor_x='center', anchor_y='center')
temp_bubble.position = bubble["position"]
temp_label.position = bubble["position"]
temp_action = cocos.actions.Hide() + \
Delay(bubble["start"]) + \
cocos.actions.Show() + \
FadeIn(0.25) + \
Delay(bubble["end"] - bubble["start"] - 0.5) + \
FadeOut(0.25) + \
cocos.actions.Hide()
temp_bubble.do(temp_action)
temp_label.do(temp_action)
self.add(temp_bubble)
self.add(temp_label)
def run_proc1(name):
director.init(width=1920, height=1080, caption="story_video", fullscreen=False)
temp_init_scene = cocos.scene.Scene()
init_scene = cocos.scene.Scene()
scene_1 = cocos.scene.Scene()
scene_2 = cocos.scene.Scene()
scene_3 = cocos.scene.Scene()
scene_4 = cocos.scene.Scene()
scene_4_character_list = [
{
"category": "elephant",
"file_name": "elephant.png",
"scale": 0.6,
"rotation": 0,
"position": [1479, 200],
"flip": [False, False]
},
{
"category": "mouse",
"file_name": "mouse.png",
"scale": 0.5,
"rotation": -40,
"position": [683, 400],
"flip": [True, False]
}
]
scene_4_dialog_list = [
{
"text": "哎呀 真是对不起",
"start": 125 - 113,
"end": 129 - 113,
"position": [1479 + 200, 200 + 200],
},
{
"text": "真好玩 真好玩",
"start": 134 - 113,
"end": 137 - 113,
"position": [683 + 200, 400 + 200],
},
]
scene_4_bg_image_path = os.path.join(background_collection_path, "buildingside.png")
scene_4.add(Background(scene_4_bg_image_path, next_scene=None, duration=22), z=1)
scene_4.add(CharacterLayer(character_list=scene_4_character_list), z=2)
scene_4.add(BubbleLayer(dialoglist=scene_4_dialog_list), z=3)
scene_3_character_list = [
{
"category": "elephant",
"file_name": "elephant.png",
"scale": 0.6,
"rotation": 0,
"position": [1479, 200],
"flip": [False, False]
},
{
"category": "mouse",
"file_name": "mouse.png",
"scale": 0.6,
"rotation": 0,
"position": [1083, 200],
"flip": [True, False]
}
]
scene_3_dialog_list = [
{
"text": "小老鼠 我帮你扇扇风吧",
"start": 105 - 99,
"end": 110 - 99,
"position": [1479 + 200, 200 + 200],
},
{
"text": "那真是谢谢你了",
"start": 112 - 99,
"end": 115 - 99,
"position": [1083 + 200, 200 + 200],
},
]
scene_3_bg_image_path = os.path.join(background_collection_path, "buildingside.png")
scene_3.add(Background(scene_3_bg_image_path, next_scene=scene_4, duration=14), z=1)
scene_3.add(CharacterLayer(character_list=scene_3_character_list), z=2)
scene_3.add(BubbleLayer(dialoglist=scene_3_dialog_list), z=3)
scene_2_character_list = [
{
"category": "elephant",
"file_name": "elephant.png",
"scale": 0.6,
"rotation": 0,
"position": [666, 400],
"flip": [True, False]
},
{
"category": "bear",
"file_name": "bear.png",
"scale": 0.6,
"rotation": 0,
"position": [1333, 450],
"flip": [False, False]
}
]
scene_2_dialog_list = [
{
"text": "好孩子 谢谢你",
"start": 92 - 75,
"end": 95 - 75,
"position": [1333 + 200, 450 + 200],
},
{
"text": "不用谢 不用谢",
"start": 97 - 75,
"end": 99 - 75,
"position": [666 + 200, 400 + 200],
},
]
scene_2_bg_image_path = os.path.join(background_collection_path, "riverside.png")
scene_2.add(Background(scene_2_bg_image_path, next_scene=scene_3, duration=24), z=1)
scene_2.add(CharacterLayer(character_list=scene_2_character_list), z=2)
scene_2.add(BubbleLayer(dialoglist=scene_2_dialog_list), z=3)
scene_1_character_list = [
{
"category": "elephant",
"file_name": "elephant.png",
"scale": 0.6,
"rotation": 0,
"position": [694, 400],
"flip": [True, False]
},
{
"category": "cattle",
"file_name": "cattle.png",
"scale": 0.6,
"rotation": 0,
"position": [1283, 500],
"flip": [False, False]
}
]
scene_1_dialog_list = [
{
"text": "好凉快呀 真是谢谢你",
"start": 69 - 45,
"end": 72 - 45,
"position": [1283 + 200, 500 + 200],
},
{
"text": "不用谢 不用谢",
"start": 74 - 45,
"end": 76 - 45,
"position": [694 + 200, 400 + 200],
},
]
scene_1_bg_image_path = os.path.join(background_collection_path, "farmland.png")
scene_1.add(Background(scene_1_bg_image_path, next_scene=scene_2, duration=30), z=1)
scene_1.add(CharacterLayer(character_list=scene_1_character_list), z=2)
scene_1.add(BubbleLayer(dialoglist=scene_1_dialog_list), z=3)
init_scene_character_list = [
{
"category": "xiaoice",
"file_name": "xiaoice.gif",
"scale": 0.8,
"rotation": 0,
"position": [1920 / 2 - 500, 1080 / 2 - 200],
"flip": [False, False]
},
]
init_scene_dialog_list = [
{
"text": "小朋友 准备好",
"start": 0,
"end": 2,
"position": [1920 / 2 - 300, 1080 / 2 + 140],
},
{
"text": "小朋友 准备好",
"start": 2,
"end": 4.5,
"position": [1920 / 2 - 300, 1080 / 2 + 140],
},
{
"text": "听小冰姐姐给你讲故事",
"start": 4.5,
"end": 9,
"position": [1920 / 2 - 300, 1080 / 2 + 140],
},
{
"text": "茜茜小朋友你好",
"start": 23,
"end": 25,
"position": [1920 / 2 - 300, 1080 / 2 + 140],
},
{
"text": "我是你的好朋友小冰姐姐",
"start": 25,
"end": 28.5,
"position": [1920 / 2 - 300, 1080 / 2 + 140],
},
{
"text": "欢迎来到你的童话世界",
"start": 28.5,
"end": 31.5,
"position": [1920 / 2 - 300, 1080 / 2 + 140],
},
{
"text": "今天我要给你讲的故事叫做小象的大耳朵",
"start": 31.5,
"end": 37,
"position": [1920 / 2 - 300, 1080 / 2 + 140],
},
{
"text": "在这个故事里 主人公的名字就叫茜茜",
"start": 37,
"end": 44,
"position": [1920 / 2 - 300, 1080 / 2 + 140],
}
]
init_scene_bg_image_path = os.path.join(background_collection_path, "bedroom.jpg")
init_scene.add(Background(init_scene_bg_image_path, next_scene=scene_1, duration=45), z=1)
init_scene.add(CharacterLayer(character_list=init_scene_character_list), z=2)
init_scene.add(BubbleLayer(dialoglist=init_scene_dialog_list), z=3)
temp_init_scene.add(Background(init_scene_bg_image_path, next_scene=init_scene, duration=10), z=1)
director.run(temp_init_scene)
def run_proc2(name):
import win32gui
time.sleep(1)
fceuxHWND = win32gui.FindWindow(None, 'story_video')
print(fceuxHWND)
rect = win32gui.GetWindowRect(fceuxHWND)
# print("Recording")
time.sleep(1)
current_screen = ImageGrab.grab(bbox=rect)
height, weight = current_screen.size
print(height, weight)
fourcc = cv2.VideoWriter_fourcc(*'XVID')
video = cv2.VideoWriter('generated.avi', fourcc, 16, (height, weight))
duration = 5
start_time = time.time()
while time.time() - start_time <= duration:
im = ImageGrab.grab()
imm = cv2.cvtColor(np.array(im), cv2.COLOR_RGB2BGR)
video.write(imm)
video.release()
cv2.destroyAllWindows()
print("Finished!")
# Basic path information
ROOT_PATH = r"D:\ChildStory"
TRANSFORMER_PATH = os.path.join(ROOT_PATH, "Transformer")
RESOURCE_PATH = os.path.join(ROOT_PATH, "Resources")
background_collection_path = os.path.join(RESOURCE_PATH, "background")
character_collection_path = os.path.join(RESOURCE_PATH, "character")
bubble_collection_path = os.path.join(RESOURCE_PATH, "talkbubble", "dialog.png")
if __name__ == "__main__":
p1 = Process(target=run_proc1, args=('test',))
p2 = Process(target=run_proc2, args=('test',))
p1.start()
p2.start()
p1.join()
p2.join()
print('End')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
You can’t perform that action at this time.