/*
|
* Copyright (C) 2019 Rockchip Electronics Co., Ltd.
|
* author: Zhihua Wang, hogan.wang@rock-chips.com
|
*
|
* This software is available to you under a choice of one of two
|
* licenses. You may choose to be licensed under the terms of the GNU
|
* General Public License (GPL), available from the file
|
* COPYING in the main directory of this source tree, or the
|
* OpenIB.org BSD license below:
|
*
|
* Redistribution and use in source and binary forms, with or
|
* without modification, are permitted provided that the following
|
* conditions are met:
|
*
|
* - Redistributions of source code must retain the above
|
* copyright notice, this list of conditions and the following
|
* disclaimer.
|
*
|
* - Redistributions in binary form must reproduce the above
|
* copyright notice, this list of conditions and the following
|
* disclaimer in the documentation and/or other materials
|
* provided with the distribution.
|
*
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
|
* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
|
* NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
|
* BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
|
* ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
|
* CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
|
* SOFTWARE.
|
*/
|
#include <sys/types.h>
|
#include <sys/stat.h>
|
#include <unistd.h>
|
#include <dirent.h>
|
#include <stdio.h>
|
#include <string.h>
|
#include <stdlib.h>
|
|
#include "face_common.h"
|
#include "load_feature.h"
|
#include "database.h"
|
#include "db_monitor.h"
|
|
static get_path_feature_t get_path_feature_cb = NULL;
|
void register_get_path_feature(get_path_feature_t cb)
|
{
|
get_path_feature_cb = cb;
|
}
|
|
int count_file(const char *path, char *fmt)
|
{
|
struct dirent *ent = NULL;
|
DIR *dir;
|
struct stat st;
|
char name[512];
|
int cnt = 0;
|
|
dir = opendir(path);
|
if (!dir) {
|
printf("%s is not exist or is not a directory!\n", path);
|
return cnt;
|
}
|
while ((ent = readdir(dir))) {
|
snprintf(name, sizeof(name), "%s/%s", path, ent->d_name);
|
stat(name, &st);
|
if (S_ISDIR(st.st_mode)) {
|
if (strcmp(".", ent->d_name) && strcmp("..", ent->d_name))
|
cnt += count_file(name, fmt);
|
} else if (strstr(ent->d_name, fmt)) {
|
cnt++;
|
}
|
}
|
closedir(dir);
|
return cnt;
|
}
|
|
int load_feature(const char *path, const char *fmt, void *data, unsigned int cnt)
|
{
|
struct dirent *ent = NULL;
|
DIR *dir;
|
struct stat st;
|
char name[NAME_LEN];
|
int index = 0;
|
int id;
|
|
dir = opendir(path);
|
if (!dir) {
|
printf("%s is not exist or is not a directory!\n", path);
|
return index;
|
}
|
while ((ent = readdir(dir))) {
|
snprintf(name, sizeof(name), "%s/%s", path, ent->d_name);
|
stat(name, &st);
|
if (S_ISDIR(st.st_mode)) {
|
if (strcmp(".", ent->d_name) && strcmp("..", ent->d_name) &&
|
strcmp("snapshot", ent->d_name))
|
index += load_feature(name, fmt, (struct face_data*)data + index, cnt - index);
|
} else if (strstr(ent->d_name, fmt)) {
|
if (index >= cnt)
|
break;
|
if (database_is_name_exist(name))
|
continue;
|
struct face_data *face_data = (struct face_data*)data + index;
|
rockface_feature_float_t m;
|
float mask_score;
|
if (get_path_feature_cb && get_path_feature_cb(name, &face_data->feature, &m, &mask_score) == 0) {
|
char tmp[NAME_LEN];
|
memset(tmp, 0, sizeof(tmp));
|
memcpy(tmp, ent->d_name, strrchr(ent->d_name, '.') - ent->d_name);
|
index++;
|
id = database_get_user_name_id();
|
if (id < 0) {
|
printf("%s: get id fail!\n", __func__);
|
break;
|
}
|
face_data->id = id;
|
database_insert(&face_data->feature, sizeof(face_data->feature),
|
name, sizeof(name), id, false, &m, sizeof(m));
|
if (strstr(name, "black_list"))
|
db_monitor_face_list_add(id, name, tmp, "blackList");
|
else
|
db_monitor_face_list_add(id, name, tmp, "whiteList");
|
}
|
}
|
}
|
closedir(dir);
|
return index;
|
}
|
