/* Overlays bounding boxes and landmarks from tracks onto a copy of the images in the list */
void sdk_draw_utils_overlay_track_list_on_image_list(
sdk_utils_file_list_type in_image_list,
ppr_track_list_type track_list,
int min_confidence,
int overlay_frame_numbers,
const char *output_image_format)
{
int i;
int frame_num;
int *i_tracked_face_lists;
ppr_raw_image_type raw_image;
char title[FILENAME_MAX];
char out_filename[FILENAME_MAX];
ppr_face_attributes_type face_attributes;
/* Create array of indices into tracked face arrays */
i_tracked_face_lists = (int*)malloc(track_list.length * sizeof(int));
sdk_utils_check_allocation(i_tracked_face_lists, "i_tracked_face_lists", __FUNCTION__);
for (i = 0; i < track_list.length; i++)
i_tracked_face_lists[i] = 0;
/* Loop through each image in list */
for (frame_num = 0; frame_num < in_image_list.length; frame_num++)
{
/* Read image */
TRY_RAW_IMAGE(ppr_raw_image_io_read(in_image_list.filenames[frame_num], &raw_image));
/* Overlay polygons on frame */
sdk_draw_utils_advance_indices_to_frame(track_list, i_tracked_face_lists, frame_num);
for (i = 0; i < track_list.length; i++)
{
ppr_get_face_attributes(track_list.tracks[i].tracked_faces.faces[i_tracked_face_lists[i]], &face_attributes);
if (face_attributes.tracking_info.frame_number == frame_num)
{
/* Overlay polygon for this face */
sdk_draw_utils_overlay_tracked_faces(&raw_image, 1, &track_list.tracks[i].tracked_faces.faces[i_tracked_face_lists[i]], min_confidence);
}
}
/* Overlay frame number on frame */
if (overlay_frame_numbers)
sdk_draw_utils_overlay_frame_number(&raw_image, frame_num);
/* Write image to file */
sdk_utils_get_file_title(in_image_list.filenames[frame_num], title);
sprintf(out_filename, "%s_track_overlay.%s", title, output_image_format);
TRY_RAW_IMAGE(ppr_raw_image_io_write(out_filename, raw_image));
/* Free image */
ppr_raw_image_free(raw_image);
}
/* Clean up */
free(i_tracked_face_lists);
}
janus_error janus_read_image(const char *file_name, janus_image *image)
{
ppr_raw_image_type ppr_image;
ppr_raw_image_error_type error = ppr_raw_image_io_read(file_name, &ppr_image);
if (error != PPR_RAW_IMAGE_SUCCESS)
return JANUS_INVALID_IMAGE;
*image = janusFromPittPatt(&ppr_image);
ppr_raw_image_free(ppr_image);
return JANUS_SUCCESS;
}
