added ability to write to an existing pygame.Surface

infocat-kr · Jun 20, 2012 · b19d236 · b19d236
1 parent 84f5e3d
commit b19d236
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/pytagcloud/__init__.py b/pytagcloud/__init__.py
@@ -349,8 +349,11 @@ def create_tag_image(
                                       fontname=fontname,
                                       rectangular=rectangular)
 
-    image_surface = Surface((sizeRect.w, sizeRect.h), SRCALPHA, 32)
-    image_surface.fill(background)
+    if type(output) == pygame.Surface:
+        image_surface = output
+    else:
+        image_surface = Surface((sizeRect.w, sizeRect.h), SRCALPHA, 32)
+        image_surface.fill(background)
     for tag in tag_store:
         image_surface.blit(tag.image, tag.rect)
     pygame.image.save(image_surface, output)