homerHEVC代码阅读（27）——CTU初始化函数init_ctu

// ctu初始化
ctu_info_t* init_ctu(henc_thread_t* et)
{
ctu_info_t *ctu;
int ctu_width, ctu_height;

// 获取当前的cu
ctu = &et->enc_engine->ctu_info[et->cu_current];//&et->curr_ctu_group_info[0];

// 当前cu的索引
ctu->ctu_number = et->cu_current;

// et->cu_current_x是当前cu的坐标（并不是像素坐标，而是索引坐标）

// 设置ctu三个分量的像素坐标
ctu->x[Y_COMP] = et->cu_current_x*et->ctu_width[Y_COMP];
ctu->y[Y_COMP] = et->cu_current_y*et->ctu_height[Y_COMP];

ctu->x[U_COMP] = ctu->x[V_COMP] = et->cu_current_x*et->ctu_width[U_COMP];
ctu->y[U_COMP] = ctu->y[V_COMP] = et->cu_current_y*et->ctu_height[U_COMP];

// 设置CTU的尺寸，即LCU的尺寸64x64
ctu->size = et->max_cu_size;

// 设置ctu的划分数量
ctu->num_part_in_ctu = et->num_partitions_in_cu;
ctu->num_part_in_ctu = et->num_partitions_in_cu;

//	ctu->partition_list = &et->partition_info[0];

// ctu的宽和高，之所以这样设置的原因是为了防止有些图像的宽和高不是CTU的整数倍
// ctu->x[Y_COMP] 表示CTU的起始位置（像素坐标），ctu->size表示CTU的宽和高（被设置为64x64）
// ctu->x[Y_COMP]+ctu->size 就表示计算CTU最后一个像素的位置，用来判断当前的CTU（被设置为64x64）是否超出了图像的边界
// 如果图像不是64x64的整数倍，那么某一个边缘的CTU必然会超出图像的边界，所以需要调整该边缘CTU的宽和高
ctu_width = ((ctu->x[Y_COMP]+ctu->size) < et->pict_width[Y_COMP])?(ctu->size):et->pict_width[Y_COMP]-ctu->x[Y_COMP];
ctu_height = ((ctu->y[Y_COMP]+ctu->size) < et->pict_height[Y_COMP])?(ctu->size):et->pict_height[Y_COMP]-ctu->y[Y_COMP];

// 如果最后计算出来的宽和高不是64，那么需要调整CTU的宽和高
if(ctu_width!=ctu->size || ctu_height!=ctu->size)
{
// 如果当前CTU被划分，那么每一个部分的宽度
int width_in_partitions = ctu_width>>2;

// 如果当前CTU被划分，那么每一个部分的高度
int height_in_partitions = ctu_height>>2;

// 如果当前CTU被划分，那么每一个部分的尺寸
int cu_size_in_partitions = ctu->size>>2;
//		if(height_in_partitions == cu_size_in_partitions)
//			height_in_partitions = cu_size_in_partitions-1;
//		else if(width_in_partitions == cu_size_in_partitions)

height_in_partitions -= 1;

// 最后一个有效的划分，raster2abs_table这个转换表是从像素地址转换为索引地址
ctu->last_valid_partition =	et->enc_engine->raster2abs_table[height_in_partitions*cu_size_in_partitions+width_in_partitions-1];
}
else
{
// 最后一个有效的划分
ctu->last_valid_partition = et->num_partitions_in_cu-1;
}

// 获取当前ctu的邻居
CuGetNeighbors(et, ctu);//raster order
return ctu;
}

// 取得cu的邻居
void CuGetNeighbors(henc_thread_t* et, ctu_info_t* ctu)
{
// 如果CTU的横坐标是0，那么表示他的左边没有邻居
if(ctu->x[Y_COMP]==0)
ctu->ctu_left = NULL;
else
{
// 否则有左邻居
ctu->ctu_left = &et->enc_engine->ctu_info[ctu->ctu_number-1];
}

// 由于使用了wfpp的方式，一般认为左下角的邻居不存在
ctu->ctu_left_bottom = NULL;//en raster order este no existe. En wavefront si

// 如果ctu的纵坐标是0，表示它的上面、左上角、右上角的邻居为空
if(ctu->y[Y_COMP]==0)
{
ctu->ctu_top = NULL;
ctu->ctu_top_right = NULL;
ctu->ctu_top_left = NULL;
}
// 如果纵坐标不为0
else
{
ctu->ctu_top = &et->enc_engine->ctu_info[ctu->ctu_number-et->pict_width_in_ctu];

// 但是横坐标为0，那么它的左上角不存在
if(ctu->x[Y_COMP]==0)
ctu->ctu_top_left = NULL;
else
ctu->ctu_top_left = &et->enc_engine->ctu_info[ctu->ctu_number-et->pict_width_in_ctu-1];

// 判断是否存在右上角
// 如果当前CTU位于帧的右边缘，那么它不存在右上角邻居
if(et->cu_current_y==0 || ((et->cu_current_x % et->pict_width_in_ctu) == (et->pict_width_in_ctu-1)))
ctu->ctu_top_right = NULL;
else
{
ctu->ctu_top_right = &et->enc_engine->ctu_info[ctu->ctu_number-et->pict_width_in_ctu+1];
}
}
}