2

これは、私が最近尋ねた質問に続きます。

Pygameでは、ヘッドレスモードで画面イメージを保存するにはどうすればよいですか?

不透明な表面オブジェクトを含む画面イメージを保存できます。ただし、この手法は透明な表面オブジェクトには機能しません。次のコードは、問題を示しています。

import sys
import os
import pygame
from pygame.color import THECOLORS as RGB


class PygameProblemDemo(object):
    
    def __init__(self):
        screen_width, screen_height = 200, 200
        os.environ['SDL_VIDEODRIVER'] = 'dummy'
        
        pygame.init()
        pygame.display.init()
        
        self.screen = pygame.display.set_mode((screen_width, screen_height))
    
    def save_non_alpha_image(self):
        r = 50
        color = RGB['blue']
        img_path = '/tmp/non_alpha_image.png'
        
        background = pygame.Surface((200, 200), pygame.SRCALPHA, 32)
        pygame.draw.rect(background, RGB['lightgray'], (0, 0, 200, 200), 0)
        
        self.screen.fill(RGB['black'])
        self.screen.blit(background, (0,0))
        pygame.draw.circle(self.screen, color, (100,100), r, 0)
        
        pygame.image.save(self.screen, img_path)
        print "image saved to %s" % (img_path)
    
    def save_alpha_image(self):
        r = 50
        color = RGB['blue']
        img_path = '/tmp/alpha_image.png'
        
        background = pygame.Surface((200, 200), pygame.SRCALPHA, 32)
        pygame.draw.rect(background, RGB['lightgray'], (0, 0, 200, 200), 0)
        
        transparent_circle = self.draw_transparent_circle(r, color, 50)
        
        self.screen.fill(RGB['black'])
        self.screen.blit(background, (0,0))
        self.screen.blit(transparent_circle, (50,50))
        
        pygame.image.save(self.screen, img_path)
        print "image saved to %s" % (img_path)
    
    def draw_transparent_circle(self, radius, color, transparency=0):
        """transparency is value between 0 and 100, 0 is opaque,
        100 invisible"""
        width, height = radius*2, radius*2
        
        # transparent base surface
        flags = pygame.SRCALPHA
        depth = 32
        base = pygame.Surface((width, height), flags, depth)
        
        # alpha surface
        alpha = int(round(255 * (100-transparency) / 100.0))
        alpha_surface = pygame.Surface((width, height))
        alpha_surface.set_colorkey(RGB['black'])
        alpha_surface.set_alpha(alpha)
        
        # draw circle (to alpha surface)
        pygame.draw.circle(alpha_surface, color, (radius,radius), radius)
        
        # draw alpha surface to base surface
        base.blit(alpha_surface, (0,0))
        
        return base
    
    
demo = PygameProblemDemo()
demo.save_non_alpha_image()
demo.save_alpha_image()

結果は次のとおりです。

non_alpha_image.png

non_alpha_image.png

alpha_image.png

alpha_image.png


アップデート

コンストラクターの行をコメントアウトするos.environ['SDL_VIDEODRIVER'] = 'dummy'と、透明な画像が正常に保存されます。

alpha_image.png

alpha_image.png

ただし、それはウィンドウが開かないようにする行であるため、スクリプトはヘッドレスで実行できなくなります。:\

4

1 に答える 1

0

私はそれを理解しました。Pygameの表示は、アルファチャネルを有効にして開始する必要がありました。上記の例を修正するには、コンストラクターで次の行を置き換えるだけです。

self.screen = pygame.display.set_mode((screen_width, screen_height))

これとともに:

self.screen = pygame.display.set_mode((screen_width, screen_height), 0, 32)
于 2013-02-20T03:32:10.830 に答える