C语言实现 vector( 动态数组) 改进版

之所以再写一封邮件缘起于我写的《 C语言实现 vector( 动态数组)》这篇文章http://blog.csdn.net/dengxu11/article/details/5915857。

        原来这个是在Linux下写的,多谢troublema的提议,我发现那份代码在windows上存在一些问题,故做了改进,以求在widnows上和Linux上都是OK的。

另外非常感谢troublema,没有他我发现不了这个存在的问题。

下面是更新后的代码:

//cvector.h

# ifndef __CVECTOR_H__
# define __CVECTOR_H__

# include   
# include   
# include   

# define MIN_LEN 256
# define CVEFAILED  -1
# define CVESUCCESS  0
# define CVEPUSHBACK 1
# define CVEPOPBACK  2
# define CVEINSERT   3
# define CVERM       4
# define EXPANED_VAL 1
# define REDUSED_VAL 2

typedef void *citerator;
typedef struct _cvector *cvector;

# ifdef __cplusplus
extern "C" {
# endif

	cvector   cvector_create   (const size_t size                           );
	void      cvector_destroy  (const cvector cv                            );
	size_t    cvector_length   (const cvector cv                            );
	int       cvector_pushback (const cvector cv, void *memb                );
	int       cvector_popback  (const cvector cv, void *memb                );
	size_t    cvector_iter_at  (const cvector cv, citerator iter            );
	int       cvector_iter_val (const cvector cv, citerator iter, void *memb);
	citerator cvector_begin    (const cvector cv                            );
	citerator cvector_end      (const cvector cv                            );
	citerator cvector_next     (const cvector cv, citerator iter            );
	int       cvector_val_at   (const cvector cv, size_t index, void *memb  );
	int       cvector_insert   (const cvector cv, citerator iter, void *memb);
	int       cvector_insert_at(const cvector cv, size_t index, void *memb  );
	int       cvector_rm       (const cvector cv, citerator iter            );
	int       cvector_rm_at    (const cvector cv, size_t index              );

	/* for test  */
	void      cv_info          (const cvector cv                            );
	void      cv_print         (const cvector cv                            );

# ifdef __cplusplus
}
# endif

#endif /* EOF file cvector.h */

//cvector.c

#include "cvector.h"

#ifndef __gnu_linux__
#define __func__ "unknown"
#define inline __forceinline
#endif

# define CWARNING_ITER(cv, iter, file, func, line) \
	do {\
	if ((cvector_begin(cv) > iter) || (cvector_end(cv) <= iter)) {\
	fprintf(stderr, "var(" #iter ") warng out of range, "\
	"at file:%s func:%s line:%d!!\n", file, func, line);\
	return CVEFAILED;\
	}\
	} while (0)

struct _cvector
{
	void *cv_pdata;
	size_t cv_len, cv_tot_len, cv_size;
};

cvector cvector_create(const size_t size)
{
	cvector cv = (cvector)malloc(sizeof (struct _cvector));

	if (!cv) return NULL;

	cv->cv_pdata = malloc(MIN_LEN * size);

	if (!cv->cv_pdata)
	{
		free(cv);
		return NULL;
	}

	cv->cv_size = size;
	cv->cv_tot_len = MIN_LEN;
	cv->cv_len = 0;

	return cv;
}  

void cvector_destroy(const cvector cv)  
{  
	free(cv->cv_pdata);  
	free(cv);  
	return;  
}  

size_t cvector_length(const cvector cv)  
{  
	return cv->cv_len;  
}  

int cvector_pushback(const cvector cv, void *memb)  
{  
	if (cv->cv_len >= cv->cv_tot_len)   
	{  
		void *pd_sav = cv->cv_pdata;  
		cv->cv_tot_len <<= EXPANED_VAL;  
		cv->cv_pdata = realloc(cv->cv_pdata, cv->cv_tot_len * cv->cv_size);  

		if (!cv->cv_pdata)   
		{  
			cv->cv_pdata = pd_sav;  
			cv->cv_tot_len >>= EXPANED_VAL;  
			return CVEPUSHBACK;  
		}  
	}  

	memcpy((char *)cv->cv_pdata + cv->cv_len * cv->cv_size, memb, cv->cv_size);  
	cv->cv_len++;  

	return CVESUCCESS;  
}  

int cvector_popback(const cvector cv, void *memb)  
{  
	if (cv->cv_len <= 0) return CVEPOPBACK;  

	cv->cv_len--;  
	memcpy(memb, (char *)cv->cv_pdata + cv->cv_len * cv->cv_size, cv->cv_size);  

	if ((cv->cv_tot_len >= (MIN_LEN << REDUSED_VAL))   
		&& (cv->cv_len <= (cv->cv_tot_len >> REDUSED_VAL)))   
	{  
		void *pd_sav = cv->cv_pdata;  
		cv->cv_tot_len >>= EXPANED_VAL;  
		cv->cv_pdata = realloc(cv->cv_pdata, cv->cv_tot_len * cv->cv_size);  

		if (!cv->cv_pdata)   
		{  
			cv->cv_tot_len <<= EXPANED_VAL;  
			cv->cv_pdata = pd_sav;  
			return CVEPOPBACK;  
		}  
	}  

	return CVESUCCESS;  
}  

size_t cvector_iter_at(const cvector cv, citerator iter)  
{  
	CWARNING_ITER(cv, iter, __FILE__, __func__, __LINE__);  
	return ((char *)iter - (char *)cv->cv_pdata) / cv->cv_size;  
}  

int cvector_iter_val(const cvector cv, citerator iter, void *memb)  
{  
	CWARNING_ITER(cv, iter, __FILE__, __func__, __LINE__);  
	memcpy(memb, iter, cv->cv_size);  
	return 0;  
}  

citerator cvector_begin(const cvector cv)  
{  
	return cv->cv_pdata;  
}  

citerator cvector_end(const cvector cv)  
{  
	return (char *)cv->cv_pdata + (cv->cv_size * cv->cv_len);  
}  

static inline void cvmemove_foreward(const cvector cv, void *from, void *to)  
{  
	size_t size = cv->cv_size;  
	char *p;  
	for (p = (char *)to; p >= (char *)from; p -= size) memcpy(p + size, p, size);  
	return;  
}  

static inline void cvmemove_backward(const cvector cv, void *from, void *to)  
{  
	memcpy(from, (char *)from + cv->cv_size, (char *)to - (char *)from);  
	return;  
}  

int cvector_insert(const cvector cv, citerator iter, void *memb)  
{  
	CWARNING_ITER(cv, iter, __FILE__, __func__, __LINE__);  

	if (cv->cv_len >= cv->cv_tot_len)   
	{  
		void *pd_sav = cv->cv_pdata;  
		cv->cv_tot_len <<= EXPANED_VAL;  
		cv->cv_pdata = realloc(cv->cv_pdata, cv->cv_tot_len * cv->cv_size);  

		if (!cv->cv_pdata)   
		{  
			cv->cv_pdata = pd_sav;  
			cv->cv_tot_len >>= EXPANED_VAL;  
			return CVEINSERT;  
		}  
	}  

	cvmemove_foreward(cv, iter, (char *)cv->cv_pdata + cv->cv_len * cv->cv_size);  
	memcpy(iter, memb, cv->cv_size);  
	cv->cv_len++;  

	return CVESUCCESS;  
}  

int cvector_insert_at(const cvector cv, size_t index, void *memb)  
{  
	citerator iter;  

	if (index >= cv->cv_tot_len)   
	{  
		cv->cv_len = index + 1;  
		while (cv->cv_len >= cv->cv_tot_len) cv->cv_tot_len <<= EXPANED_VAL;  
		cv->cv_pdata = realloc(cv->cv_pdata, cv->cv_tot_len * cv->cv_size);  
		iter = (char *)cv->cv_pdata + cv->cv_size * index;  
		memcpy(iter, memb, cv->cv_size);  
	}  
	else   
	{  
		iter = (char *)cv->cv_pdata + cv->cv_size * index;  
		cvector_insert(cv, iter, memb);  
	}  

	return 0;  
}  

citerator cvector_next(const cvector cv, citerator iter)  
{  
	return (char *)iter + cv->cv_size;  
}  

int cvector_val(const cvector cv, citerator iter, void *memb)  
{  
	memcpy(memb, iter, cv->cv_size);  
	return 0;  
}  

int cvector_val_at(const cvector cv, size_t index, void *memb)  
{  
	memcpy(memb, (char *)cv->cv_pdata + index * cv->cv_size, cv->cv_size);  
	return 0;  
}  

int cvector_rm(const cvector cv, citerator iter)  
{  
	citerator from;  
	citerator end;  
	CWARNING_ITER(cv, iter, __FILE__, __func__, __LINE__);  
	from = iter;  
	end = cvector_end(cv);  
	memcpy(from, (char *)from + cv->cv_size, (char *)end - (char *)from);  
	cv->cv_len--;  

	if ((cv->cv_tot_len >= (MIN_LEN << REDUSED_VAL))  
		&& (cv->cv_len <= (cv->cv_tot_len >> REDUSED_VAL)))   
	{  
		void *pd_sav = cv->cv_pdata;  
		cv->cv_tot_len >>= EXPANED_VAL;  
		cv->cv_pdata = realloc(cv->cv_pdata, cv->cv_tot_len * cv->cv_size);  

		if (!cv->cv_pdata)   
		{  
			cv->cv_tot_len <<= EXPANED_VAL;  
			cv->cv_pdata = pd_sav;  
			return CVERM;  
		}  
	}  

	return CVESUCCESS;  
}  

int cvector_rm_at(const cvector cv, size_t index)  
{  
	citerator iter;  
	iter = (char *)cv->cv_pdata + cv->cv_size * index;  
	CWARNING_ITER(cv, iter, __FILE__, __func__, __LINE__);  
	return cvector_rm(cv, iter);  
}  

void cv_info(const cvector cv)  
{  
	printf("\n\ntot :%s : %d\n", __func__, cv->cv_tot_len);  
	printf("len :%s : %d\n",     __func__, cv->cv_len);  
	printf("size:%s : %d\n\n",   __func__, cv->cv_size);  
	return;  
}  

void cv_print(const cvector cv)  
{  
	int num;  
	citerator iter;  

	if (cvector_length(cv) == 0)  
		fprintf(stderr, "file:%s func:%s line:%d error, null length cvector!!\n", __FILE__, __func__, __LINE__);  

	for (iter = cvector_begin(cv);   
		iter != cvector_end(cv);  
		iter = cvector_next(cv, iter))   
	{  
		cvector_iter_val(cv, iter, &num);  
		printf("var:%d at:%d\n", num, cvector_iter_at(cv, iter));  
	}  

	return;  
}  

//main.cpp

#include "cvector.h"

int main()
{
	int i = 1;
	cvector cv = cvector_create(sizeof(int));
	cvector_pushback(cv, &i);
	cvector_pushback(cv, &i);
	cvector_pushback(cv, &i);
	cvector_pushback(cv, &i);
	cv_print(cv);
	cvector_destroy(cv);
	return 0;
}


你可能感兴趣的:(C语言)