Skip to content

Commit ebd220b

Browse files
committed
Added none colormap option
1 parent 201db0b commit ebd220b

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

py/optimize_impl/energies.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -250,6 +250,8 @@ def get_vertex_energy(
250250
uv_embed[:,:2] = uv[:,:2]
251251

252252
# Get energy
253+
if (colormap == 'none'):
254+
energy = 0.25 * np.ones(len(v))
253255
if (colormap == 'scale_factors'):
254256
energy = best_fit_conformal_vf(v, f, uv_embed, fuv)
255257
if (colormap == 'sym_dirichlet'):
@@ -293,6 +295,8 @@ def get_face_energy(
293295
energy = arap_vf(v, f, uv_embed, fuv)
294296
if (colormap == 'quadratic_sym_dirichlet'):
295297
energy = quadratic_sym_dirichlet_vf(v, f, uv_embed, fuv)
298+
if (colormap == 'none'):
299+
energy = 0.25 * np.ones(len(f))
296300

297301
# Optionally use face weighting
298302
if (use_face_weight):

scripts/optimization_scripts/pipeline.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import os, sys
22
script_dir = os.path.dirname(__file__)
3-
module_dir = os.path.join(script_dir, '..', 'py')
3+
module_dir = os.path.join(script_dir, '..', '..', 'py')
44
sys.path.append(module_dir)
55

66
import numpy as np
@@ -100,4 +100,4 @@
100100
args = script_util.overwrite_args(args_default, args_spec)
101101

102102
# Run method
103-
method_func(args)
103+
method_func(args)

0 commit comments

Comments
 (0)