JacobLinCool commited on
Commit
028f0f5
·
1 Parent(s): 2c95e43

feat: support custom size option

Browse files

Former-commit-id: baca57c4b9770ede12bc2001050ce9e689b6313b

Files changed (1) hide show
  1. create-3d-icon.py +9 -4
create-3d-icon.py CHANGED
@@ -33,6 +33,8 @@ def main():
33
  type=float, default=-1)
34
  parser.add_argument("--b", help="blue color",
35
  type=float, default=-1)
 
 
36
  args = parser.parse_args()
37
 
38
  filepath = args.filepath
@@ -48,6 +50,7 @@ def main():
48
  color_r = args.r
49
  color_g = args.g
50
  color_b = args.b
 
51
 
52
  capture(
53
  filepath,
@@ -63,6 +66,7 @@ def main():
63
  color_r,
64
  color_g,
65
  color_b,
 
66
  )
67
  else:
68
  print("No file path given")
@@ -82,6 +86,7 @@ def capture(
82
  color_r=-1,
83
  color_g=-1,
84
  color_b=-1,
 
85
  ):
86
  reset()
87
 
@@ -127,7 +132,7 @@ def capture(
127
  location=(distance * 3, 0, 0), rotation=(math.pi*0.5, 0, math.pi*0.5))
128
  bpy.context.scene.camera = bpy.data.objects['Camera']
129
 
130
- render(out=filepath.replace('.svg', '.png'))
131
 
132
  return
133
 
@@ -142,10 +147,10 @@ def log(any):
142
  print("func:", key)
143
 
144
 
145
- def render(out=os.path.join(os.getcwd(), 'out.png')):
146
  bpy.context.scene.render.filepath = out
147
- bpy.context.scene.render.resolution_x = 2048
148
- bpy.context.scene.render.resolution_y = 2048
149
  bpy.context.scene.render.film_transparent = True
150
  bpy.ops.render.render(write_still=True)
151
 
 
33
  type=float, default=-1)
34
  parser.add_argument("--b", help="blue color",
35
  type=float, default=-1)
36
+ parser.add_argument(
37
+ "--size", help="size of the image", type=int, default=2048)
38
  args = parser.parse_args()
39
 
40
  filepath = args.filepath
 
50
  color_r = args.r
51
  color_g = args.g
52
  color_b = args.b
53
+ size = args.size
54
 
55
  capture(
56
  filepath,
 
66
  color_r,
67
  color_g,
68
  color_b,
69
+ size
70
  )
71
  else:
72
  print("No file path given")
 
86
  color_r=-1,
87
  color_g=-1,
88
  color_b=-1,
89
+ size=2048
90
  ):
91
  reset()
92
 
 
132
  location=(distance * 3, 0, 0), rotation=(math.pi*0.5, 0, math.pi*0.5))
133
  bpy.context.scene.camera = bpy.data.objects['Camera']
134
 
135
+ render(filepath.replace('.svg', '.png'), size)
136
 
137
  return
138
 
 
147
  print("func:", key)
148
 
149
 
150
+ def render(out=os.path.join(os.getcwd(), 'out.png'), size=2048):
151
  bpy.context.scene.render.filepath = out
152
+ bpy.context.scene.render.resolution_x = size
153
+ bpy.context.scene.render.resolution_y = size
154
  bpy.context.scene.render.film_transparent = True
155
  bpy.ops.render.render(write_still=True)
156