boost_python扩展内存泄露

2013-08-14 03:41:11 170 阅读 AI 写同款 · GEO 优化AI 选题数据监测一键分发 ›

python的c/c++扩展内存泄露bug fix 近日需要使用到某同学使用c++封装的python扩展模块，搭配好环境，写上代码，跑起来，top一下，发现内存一直在涨。。。于是找到源代码，大致瞅了下，发现貌似是一些引用计数的问题 py的封装最不好控制的就是引用计数了，很多时候不知道啥时候Py_INCREF，啥时候Py_DECREF，这也是为什么会存在 boost.python，pycxx的原因，诸如boost.python，pycxx的模块阅读详情

下面是在下写python的扩展代码，但是出现了泄露，希望高手指点一二



const char * decodeYUV420RGB(char * buf,  int width, int height)

{

	int R,G,B,Y,U,V;

	int x,y;

	unsigned char *rgb24;

	rgb24 = (unsigned char *)malloc(sizeof(char)*width*height*3);

	unsigned char* tempbuf = (unsigned char*)buf;

	unsigned char *yuv420[3] = {tempbuf,tempbuf+width*height,tempbuf+width*height*5/4};

	int nWidth = width>>1; //色度信号宽度

	for (y=0;y<height;y++)

	{

		for (x=0;x<width;x++)

		{

			Y = *(yuv420[0] + y*width + x);

			U = *(yuv420[1] + ((y>>1)*nWidth) + (x>>1));

			V = *(yuv420[2] + ((y>>1)*nWidth) + (x>>1));

			R = Y + 1.402*(V-128);

			G = Y - 0.34414*(U-128) - 0.71414*(V-128);

			B = Y + 1.772*(U-128);



			//防止越界

			if (R>=255)R=254;

			if (R<=0)R=1;

			if (G>=255)G=254;

			if (G<=0)G=1;

			if (B>=255)B=254;

			if (B<=0)B=1;



			*(rgb24 + ((y)*width + x)*3) = R;

			*(rgb24 + ((y)*width + x)*3 + 1) = G;

			*(rgb24 + ((y)*width + x)*3 + 2) = B;

			//    *(rgb24 + (y*width + x)*3) = B;

			//    *(rgb24 + (y*width + x)*3 + 1) = G;

			//    *(rgb24 + (y*width + x)*3 + 2) = R;  

		}

	}

	return (char*)rgb24;

}

BOOST_PYTHON_MODULE(yuvtorgb)

{

	def("decodeYUV420", decodeYUV420RGB);

}