mirror of
https://github.com/nyanmisaka/mpp.git
synced 2025-09-26 21:15:53 +08:00
dfc7564693
1. Change MppTrieInfo internal structure. 2. Allow to export / import root, node_count and info_count. 3. Add trie check in import functon. Signed-off-by: Herman Chen <herman.chen@rock-chips.com> Change-Id: Id10edbc92db14d045b5bbd15d13ef2472e695037
132 lines
3.0 KiB
C
132 lines
3.0 KiB
C
/*
|
|
* Copyright 2020 Rockchip Electronics Co. LTD
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
#define MODULE_TAG "mpp_opt"
|
|
|
|
#include "mpp_mem.h"
|
|
#include "mpp_log.h"
|
|
#include "mpp_trie.h"
|
|
#include "mpp_common.h"
|
|
|
|
#include "mpp_opt.h"
|
|
|
|
typedef struct MppOptImpl_t {
|
|
void *ctx;
|
|
MppTrie trie;
|
|
RK_S32 node_cnt;
|
|
RK_S32 info_cnt;
|
|
} MppOptImpl;
|
|
|
|
MPP_RET mpp_opt_init(MppOpt *opt)
|
|
{
|
|
MppOptImpl *impl = mpp_calloc(MppOptImpl, 1);
|
|
|
|
*opt = impl;
|
|
|
|
return (impl) ? MPP_OK : MPP_NOK;
|
|
}
|
|
|
|
MPP_RET mpp_opt_deinit(MppOpt opt)
|
|
{
|
|
MppOptImpl *impl = (MppOptImpl *)opt;
|
|
|
|
if (NULL == impl)
|
|
return MPP_NOK;
|
|
|
|
if (impl->trie) {
|
|
mpp_trie_deinit(impl->trie);
|
|
impl->trie = NULL;
|
|
}
|
|
MPP_FREE(impl);
|
|
|
|
return MPP_OK;
|
|
}
|
|
|
|
MPP_RET mpp_opt_setup(MppOpt opt, void *ctx)
|
|
{
|
|
MppOptImpl *impl = (MppOptImpl *)opt;
|
|
|
|
if (NULL == impl)
|
|
return MPP_NOK;
|
|
|
|
mpp_trie_init(&impl->trie, "mpp_opt");
|
|
if (impl->trie) {
|
|
impl->ctx = ctx;
|
|
return MPP_OK;
|
|
}
|
|
|
|
return MPP_NOK;
|
|
}
|
|
|
|
MPP_RET mpp_opt_add(MppOpt opt, MppOptInfo *info)
|
|
{
|
|
MppOptImpl *impl = (MppOptImpl *)opt;
|
|
|
|
if (NULL == impl || NULL == impl->trie)
|
|
return MPP_NOK;
|
|
|
|
if (NULL == info)
|
|
return mpp_trie_add_info(impl->trie, NULL, NULL, 0);
|
|
|
|
return mpp_trie_add_info(impl->trie, info->name, info, sizeof(*info));
|
|
}
|
|
|
|
MPP_RET mpp_opt_parse(MppOpt opt, int argc, char **argv)
|
|
{
|
|
MppOptImpl *impl = (MppOptImpl *)opt;
|
|
MPP_RET ret = MPP_NOK;
|
|
RK_S32 opt_idx = 0;
|
|
|
|
if (NULL == impl || NULL == impl->trie || argc < 2 || NULL == argv)
|
|
return ret;
|
|
|
|
ret = MPP_OK;
|
|
|
|
while (opt_idx <= argc) {
|
|
RK_S32 opt_next = opt_idx + 1;
|
|
char *opts = argv[opt_idx++];
|
|
char *next = (opt_next >= argc) ? NULL : argv[opt_next];
|
|
|
|
if (NULL == opts)
|
|
break;
|
|
|
|
if (opts[0] == '-' && opts[1] != '\0') {
|
|
MppOptInfo *info = NULL;
|
|
MppTrieInfo *node = mpp_trie_get_info(impl->trie, opts + 1);
|
|
RK_S32 step = 0;
|
|
|
|
if (NULL == node) {
|
|
mpp_err("invalid option %s\n", opts + 1);
|
|
continue;
|
|
}
|
|
|
|
info = mpp_trie_info_ctx(node);
|
|
if (info->proc)
|
|
step = info->proc(impl->ctx, next);
|
|
|
|
/* option failure or help */
|
|
if (step < 0) {
|
|
ret = step;
|
|
break;
|
|
}
|
|
|
|
opt_idx += step;
|
|
}
|
|
}
|
|
|
|
return ret;
|
|
}
|