Add insert_after and unshift

Author: tecywiz121

klist.h

@@ -116,6 +116,12 @@
 		kmp_free(name, kl->mp, p);										\
 		return 0;														\
 	}																	\
+	static inline kltype_t *kl_unshiftp_##name(kl_##name##_t *kl) {		\
+		kl1_##name *q, *p = kmp_alloc(name, kl->mp);					\
+		q = kl->head; p->next = q; kl->head = p;						\
+		++kl->size;														\
+		return &p->data;												\
+	}																	\
     static inline int kl_delete_after_##name(kl_##name##_t *kl,			\
                                                 kl1_##name *c,			\
 												kltype_t *d) {			\
@@ -127,6 +133,14 @@
 		if (d) *d = p->data;											\
 		kmp_free(name, kl->mp, p);										\
 		return 0;														\
+	}																	\
+	static inline kltype_t *kl_insertp_after_##name(kl_##name##_t *kl,	\
+													kl1_##name *c) {	\
+		if (c->next == 0) return 0;										\
+		kl1_##name *p = kmp_alloc(name, kl->mp);						\
+		p->next = c->next; c->next = p;									\
+		++kl->size;														\
+		return &p->data;												\
 	}
 
 #define KLIST_INIT(name, kltype_t, kmpfree_t)							\
@@ -143,6 +157,9 @@
 #define kl_destroy(name, kl) kl_destroy_##name(kl)
 #define kl_pushp(name, kl) kl_pushp_##name(kl)
 #define kl_shift(name, kl, d) kl_shift_##name(kl, d)
+#define kl_unshiftp(name, kl) kl_unshiftp_##name(kl)
+
 #define kl_delete_after(name, kl, iter, d) kl_delete_after_##name(kl, iter, d)
+#define kl_insertp_after(name, kl, iter) kl_insertp_after_##name(kl, iter)
 
 #endif

test/klist_test.c

@@ -12,8 +12,11 @@ int main()
 	*kl_pushp(32, kl) = 1;
 	*kl_pushp(32, kl) = 10;
 	*kl_pushp(32, kl) = 19;
+	*kl_pushp(32, kl) = 28;
 	kl_shift(32, kl, 0);
 	kl_delete_after(32, kl, kl_begin(kl), 0);
+	*kl_unshiftp(32, kl) = 37;
+	*kl_insertp_after(32, kl, kl_begin(kl)) = 46;
 	for (p = kl_begin(kl); p != kl_end(kl); p = kl_next(p))
 		printf("%d\n", kl_val(p));
 	kl_destroy(32, kl);