def _get_point_weight(point_group, point, camera, frame):
"""
Get the 2D point weight.
This function has backwards compatibility built-in, so older
versions of 3DE can use it reliably and get consistent results.
:param point_group: The 3DE Point Group containing 'point'
:type point_group: str
:param point: 3DE Point to query.
:type point: str
:param camera: The 3DE Camera containing the 3DE 'point'.
:type camera: str
:param frame: 3DE Frame number (1-based) to get weight from.
:type frame: float
:returns: A floating-point weight value.
:rtype: float
"""
weight = 1.0
if SUPPORT_POINT_WEIGHT_BY_FRAME is True:
weight = tde4.getPointWeightByFrame(point_group, point, camera, frame)
return weight
def _generate_v2(point_group,
camera,
points,
start_frame=None,
undistort=False):
"""
Generate the UV file format contents, using JSON format.
:param point_group: The 3DE Point Group containing 'points'
:type point_group: str
:param camera: The 3DE Camera containing 2D 'points' data.
:type camera: str
:param points: The list of 3DE Points representing 2D data to
save.
:type points: list of str
:param start_frame: The frame number to be considered at
'first frame'. Defaults to 1001 if
set to None.
:type start_frame: None or int
:param undistort: Should we apply undistortion to the 2D points
data? Yes or no.
:type undistort: bool
Each point will store:
- Point name
- X, Y position (in UV coordinates, per-frame)
- Point weight (per-frame)
- Point Set name
- Point 'Persistent ID'
"""
assert isinstance(point_group, basestring)
assert isinstance(camera, basestring)
assert isinstance(points, (list, tuple))
assert start_frame is None or isinstance(start_frame, int)
assert isinstance(undistort, bool)
if start_frame is None:
start_frame = 1001
data = UV_TRACK_HEADER_VERSION_2.copy()
cam_num_frames = tde4.getCameraNoFrames(camera)
if len(points) == 0:
return ''
frame0 = int(start_frame)
frame0 -= 1
data['num_points'] = len(points)
data['is_undistorted'] = bool(undistort)
data['points'] = []
for point in points:
point_data = {}
# Query point information
name = tde4.getPointName(point_group, point)
uid = None
if SUPPORT_PERSISTENT_ID is True:
uid = tde4.getPointPersistentID(point_group, point)
point_set = tde4.getPointSet(point_group, point)
point_set_name = None
if point_set is not None:
point_set_name = tde4.getSetName(point_group, point_set)
point_data['name'] = name
point_data['id'] = uid
point_data['set_name'] = point_set_name
# Write per-frame position data
frame = 1 # 3DE starts at frame '1' regardless of the 'start frame'.
point_data['per_frame'] = []
pos_block = tde4.getPointPosition2DBlock(point_group, point, camera, 1,
cam_num_frames)
for pos in pos_block:
if pos[0] == -1.0 or pos[1] == -1.0:
# No valid data here.
frame += 1
continue
# Does the 2D point go outside the camera FOV? Is that ok?
valid_2d = tde4.isPointPos2DValid(point_group, point, camera,
frame)
if valid_2d != 1:
# No valid data here.
frame += 1
continue
f = frame + frame0
if undistort is True:
pos = tde4.removeDistortion2D(camera, frame, pos)
weight = 1.0
if SUPPORT_POINT_WEIGHT_BY_FRAME is True:
weight = tde4.getPointWeightByFrame(point_group, point, camera,
frame)
frame_data = {'frame': f, 'pos': pos, 'weight': weight}
point_data['per_frame'].append(frame_data)
frame += 1
data['points'].append(point_data)
data_str = json.dumps(data)
return data_str
def _generate_v1(point_group,
camera,
points,
start_frame=None,
undistort=False):
"""
Generate the UV file format contents, using a basic ASCII format.
:param point_group: The 3DE Point Group containing 'points'
:type point_group: str
:param camera: The 3DE Camera containing 2D 'points' data.
:type camera: str
:param points: The list of 3DE Points representing 2D data to
save.
:type points: list of str
:param start_frame: The frame number to be considered at
'first frame'. Defaults to 1001 if
set to None.
:type start_frame: None or int
:param undistort: Should we apply undistortion to the 2D points
data? Yes or no.
:type undistort: bool
Each point will store:
- Point name
- X, Y position (in UV coordinates, per-frame)
- Point weight (per-frame)
"""
assert isinstance(point_group, basestring)
assert isinstance(camera, basestring)
assert isinstance(points, (list, tuple))
assert start_frame is None or isinstance(start_frame, int)
assert isinstance(undistort, bool)
if start_frame is None:
start_frame = 1001
data_str = ''
cam_num_frames = tde4.getCameraNoFrames(camera)
if len(points) == 0:
return data_str
frame0 = int(start_frame)
frame0 -= 1
data_str += '{0:d}\n'.format(len(points))
for point in points:
name = tde4.getPointName(point_group, point)
c2d = tde4.getPointPosition2DBlock(point_group, point, camera, 1,
cam_num_frames)
# Write per-frame position data
num_valid_frame = 0
pos_list = []
weight_list = []
frame = 1 # 3DE starts at frame '1' regardless of the 'start-frame'.
for v in c2d:
if v[0] == -1.0 or v[1] == -1.0:
# No valid data here.
frame += 1
continue
# Does the 2D point go outside the camera FOV? Is that ok?
valid = tde4.isPointPos2DValid(point_group, point, camera, frame)
if valid == 0:
# No valid data here.
frame += 1
continue
# Number of points with valid positions
num_valid_frame += 1
f = frame + frame0
if undistort is True:
v = tde4.removeDistortion2D(camera, frame, v)
weight = 1.0
if SUPPORT_POINT_WEIGHT_BY_FRAME is True:
weight = tde4.getPointWeightByFrame(point_group, point, camera,
frame)
pos_list.append((f, v))
weight_list.append((f, weight))
frame += 1
# add data
data_str += name + '\n'
data_str += '{0:d}\n'.format(num_valid_frame)
for pos_data, weight_data in zip(pos_list, weight_list):
f = pos_data[0]
v = pos_data[1]
w = weight_data[1]
assert f == weight_data[0]
data_str += '%d %.15f %.15f %.8f\n' % (f, v[0], v[1], w)
return data_str