first
This commit is contained in:
12
rt-thread/components/utilities/resource/SConscript
Normal file
12
rt-thread/components/utilities/resource/SConscript
Normal file
@@ -0,0 +1,12 @@
|
||||
from building import *
|
||||
|
||||
cwd = GetCurrentDir()
|
||||
src = Glob('*.c')
|
||||
CPPPATH = [cwd]
|
||||
|
||||
if GetDepend('RT_USING_ADT_BITMAP') == False:
|
||||
SrcRemove(src, ['rid_bitmap.c'])
|
||||
|
||||
group = DefineGroup('Utilities', src, depend = ['RT_USING_RESOURCE_ID'], CPPPATH = CPPPATH)
|
||||
|
||||
Return('group')
|
63
rt-thread/components/utilities/resource/resource_id.c
Normal file
63
rt-thread/components/utilities/resource/resource_id.c
Normal file
@@ -0,0 +1,63 @@
|
||||
/*
|
||||
* Copyright (c) 2006-2021, RT-Thread Development Team
|
||||
*
|
||||
* SPDX-License-Identifier: Apache-2.0
|
||||
*
|
||||
* Change Logs:
|
||||
* Date Author Notes
|
||||
* 2021-08-25 RT-Thread First version
|
||||
* 2023-09-15 xqyjlj perf rt_hw_interrupt_disable/enable
|
||||
*/
|
||||
#include <rthw.h>
|
||||
#include <rtthread.h>
|
||||
#include <resource_id.h>
|
||||
|
||||
void resource_id_init(resource_id_t *mgr, int size, void **res)
|
||||
{
|
||||
if (mgr)
|
||||
{
|
||||
mgr->size = size;
|
||||
mgr->_res = res;
|
||||
mgr->noused = 0;
|
||||
mgr->_free = RT_NULL;
|
||||
rt_spin_lock_init(&(mgr->spinlock));
|
||||
}
|
||||
}
|
||||
|
||||
int resource_id_get(resource_id_t *mgr)
|
||||
{
|
||||
rt_base_t level;
|
||||
void **cur;
|
||||
|
||||
level = rt_spin_lock_irqsave(&(mgr->spinlock));
|
||||
if (mgr->_free)
|
||||
{
|
||||
cur = mgr->_free;
|
||||
mgr->_free = (void **)*mgr->_free;
|
||||
rt_spin_unlock_irqrestore(&(mgr->spinlock), level);
|
||||
return cur - mgr->_res;
|
||||
}
|
||||
else if (mgr->noused < mgr->size)
|
||||
{
|
||||
cur = &mgr->_res[mgr->noused++];
|
||||
rt_spin_unlock_irqrestore(&(mgr->spinlock), level);
|
||||
return cur - mgr->_res;
|
||||
}
|
||||
|
||||
return -1;
|
||||
}
|
||||
|
||||
void resource_id_put(resource_id_t *mgr, int no)
|
||||
{
|
||||
rt_base_t level;
|
||||
void **cur;
|
||||
|
||||
if (no >= 0 && no < mgr->size)
|
||||
{
|
||||
level = rt_spin_lock_irqsave(&(mgr->spinlock));
|
||||
cur = &mgr->_res[no];
|
||||
*cur = (void *)mgr->_free;
|
||||
mgr->_free = cur;
|
||||
rt_spin_unlock_irqrestore(&(mgr->spinlock), level);
|
||||
}
|
||||
}
|
33
rt-thread/components/utilities/resource/resource_id.h
Normal file
33
rt-thread/components/utilities/resource/resource_id.h
Normal file
@@ -0,0 +1,33 @@
|
||||
/*
|
||||
* Copyright (c) 2006-2021, RT-Thread Development Team
|
||||
*
|
||||
* SPDX-License-Identifier: Apache-2.0
|
||||
*
|
||||
* Change Logs:
|
||||
* Date Author Notes
|
||||
* 2021-08-25 RT-Thread First version
|
||||
* 2023-09-15 xqyjlj perf rt_hw_interrupt_disable/enable
|
||||
*/
|
||||
|
||||
#ifndef RESOURCE_ID_H__
|
||||
#define RESOURCE_ID_H__
|
||||
|
||||
#include <rthw.h>
|
||||
#include <rtthread.h>
|
||||
|
||||
#define RESOURCE_ID_INIT(size, pool) {size, pool, 0, RT_NULL, RT_SPINLOCK_INIT}
|
||||
|
||||
typedef struct
|
||||
{
|
||||
int size;
|
||||
void **_res;
|
||||
int noused;
|
||||
void **_free;
|
||||
struct rt_spinlock spinlock;
|
||||
} resource_id_t;
|
||||
|
||||
void resource_id_init(resource_id_t *mgr, int size, void **res);
|
||||
int resource_id_get(resource_id_t *mgr);
|
||||
void resource_id_put(resource_id_t *mgr, int no);
|
||||
|
||||
#endif /*RESOURCE_ID_H__*/
|
116
rt-thread/components/utilities/resource/rid_bitmap.c
Normal file
116
rt-thread/components/utilities/resource/rid_bitmap.c
Normal file
@@ -0,0 +1,116 @@
|
||||
/*
|
||||
* Copyright (c) 2006-2023, RT-Thread Development Team
|
||||
*
|
||||
* SPDX-License-Identifier: Apache-2.0
|
||||
*
|
||||
* Change Logs:
|
||||
* Date Author Notes
|
||||
* 2023-12-07 Shell First version
|
||||
*/
|
||||
|
||||
#include "rid_bitmap.h"
|
||||
#include <bitmap.h>
|
||||
#include <rtdef.h>
|
||||
|
||||
void rid_bitmap_init(rid_bitmap_t mgr, int min_id, int total_id_count,
|
||||
rt_bitmap_t *set, struct rt_mutex *id_lock)
|
||||
{
|
||||
mgr->min_id = min_id;
|
||||
mgr->total_id_count = total_id_count;
|
||||
mgr->bitset = set;
|
||||
mgr->id_lock = id_lock;
|
||||
|
||||
return;
|
||||
}
|
||||
|
||||
long rid_bitmap_get(rid_bitmap_t mgr)
|
||||
{
|
||||
long id;
|
||||
long overflow;
|
||||
if (mgr->id_lock)
|
||||
{
|
||||
rt_mutex_take(mgr->id_lock, RT_WAITING_FOREVER);
|
||||
}
|
||||
|
||||
overflow = mgr->total_id_count;
|
||||
id = rt_bitmap_next_clear_bit(mgr->bitset, 0, overflow);
|
||||
if (id == overflow)
|
||||
{
|
||||
id = -1;
|
||||
}
|
||||
else
|
||||
{
|
||||
rt_bitmap_set_bit(mgr->bitset, id);
|
||||
id += mgr->min_id;
|
||||
}
|
||||
|
||||
if (mgr->id_lock)
|
||||
{
|
||||
rt_mutex_release(mgr->id_lock);
|
||||
}
|
||||
return id;
|
||||
}
|
||||
|
||||
long rid_bitmap_get_named(rid_bitmap_t mgr, long no)
|
||||
{
|
||||
long id_relative;
|
||||
long overflow;
|
||||
long min;
|
||||
|
||||
if (mgr->id_lock)
|
||||
{
|
||||
rt_mutex_take(mgr->id_lock, RT_WAITING_FOREVER);
|
||||
}
|
||||
|
||||
min = mgr->min_id;
|
||||
id_relative = no - min;
|
||||
overflow = mgr->total_id_count;
|
||||
if (id_relative >= min && id_relative < overflow)
|
||||
{
|
||||
if (rt_bitmap_test_bit(mgr->bitset, id_relative))
|
||||
{
|
||||
id_relative = -RT_EBUSY;
|
||||
}
|
||||
else
|
||||
{
|
||||
rt_bitmap_set_bit(mgr->bitset, id_relative);
|
||||
id_relative += min;
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
id_relative = -1;
|
||||
}
|
||||
|
||||
if (mgr->id_lock)
|
||||
{
|
||||
rt_mutex_release(mgr->id_lock);
|
||||
}
|
||||
return id_relative;
|
||||
}
|
||||
|
||||
void rid_bitmap_put(rid_bitmap_t mgr, long no)
|
||||
{
|
||||
long id_relative;
|
||||
long overflow;
|
||||
long min;
|
||||
|
||||
if (mgr->id_lock)
|
||||
{
|
||||
rt_mutex_take(mgr->id_lock, RT_WAITING_FOREVER);
|
||||
}
|
||||
|
||||
min = mgr->min_id;
|
||||
id_relative = no - min;
|
||||
overflow = mgr->total_id_count;
|
||||
if (id_relative >= min && id_relative < overflow &&
|
||||
rt_bitmap_test_bit(mgr->bitset, id_relative))
|
||||
{
|
||||
rt_bitmap_clear_bit(mgr->bitset, id_relative);
|
||||
}
|
||||
|
||||
if (mgr->id_lock)
|
||||
{
|
||||
rt_mutex_release(mgr->id_lock);
|
||||
}
|
||||
}
|
34
rt-thread/components/utilities/resource/rid_bitmap.h
Normal file
34
rt-thread/components/utilities/resource/rid_bitmap.h
Normal file
@@ -0,0 +1,34 @@
|
||||
/*
|
||||
* Copyright (c) 2006-2023, RT-Thread Development Team
|
||||
*
|
||||
* SPDX-License-Identifier: Apache-2.0
|
||||
*
|
||||
* Change Logs:
|
||||
* Date Author Notes
|
||||
* 2023-12-07 Shell First version
|
||||
*/
|
||||
|
||||
#ifndef __RID_BITMAP_H__
|
||||
#define __RID_BITMAP_H__
|
||||
|
||||
#include <rthw.h>
|
||||
#include <rtthread.h>
|
||||
#include <bitmap.h>
|
||||
|
||||
#define RESOURCE_ID_INIT {0}
|
||||
|
||||
typedef struct rid_bitmap
|
||||
{
|
||||
struct rt_mutex *id_lock;
|
||||
long min_id;
|
||||
long total_id_count;
|
||||
rt_bitmap_t *bitset;
|
||||
} *rid_bitmap_t;
|
||||
|
||||
void rid_bitmap_init(rid_bitmap_t mgr, int min_id, int total_id_count,
|
||||
rt_bitmap_t *set, struct rt_mutex *id_lock);
|
||||
long rid_bitmap_get(rid_bitmap_t mgr);
|
||||
long rid_bitmap_get_named(rid_bitmap_t mgr, long no);
|
||||
void rid_bitmap_put(rid_bitmap_t mgr, long no);
|
||||
|
||||
#endif /* __RID_BITMAP_H__ */
|
Reference in New Issue
Block a user