def setReshape1dTo2d(self, pic_kinds):
if self.chromaFormat == ChromaFormat.YCbCr4_2_0:
self.pelBuf[pic_kinds][0] = self.pelBuf[pic_kinds][0].reshape(
(self.area.height, self.area.width))
self.pelBuf[pic_kinds][1] = myUtil.UpSamplingChroma(
self.pelBuf[pic_kinds][1].reshape(
(self.carea.height, self.carea.width)))
self.pelBuf[pic_kinds][2] = myUtil.UpSamplingChroma(
self.pelBuf[pic_kinds][2].reshape(
(self.carea.height, self.carea.width)))
def reshapeRecon(self):
return self.reconY.reshape((self.info[2], self.info[1])), myUtil.UpSamplingChroma(
self.reconCb.reshape((self.cheight, self.cwidth))), myUtil.UpSamplingChroma(
self.reconCr.reshape((self.cheight, self.cwidth)))
def reshapeUnfiltered(self):
return self.unfilteredY.reshape((self.info[2], self.info[1])), myUtil.UpSamplingChroma(
self.unfilteredCb.reshape((self.cheight, self.cwidth))), myUtil.UpSamplingChroma(
self.unfilteredCr.reshape((self.cheight, self.cwidth)))
def reshapeOrg(self):
return self.orgY.reshape((self.info[2], self.info[1])), myUtil.UpSamplingChroma(
self.orgCb.reshape((self.cheight, self.cwidth))), myUtil.UpSamplingChroma(
self.orgCr.reshape((self.cheight, self.cwidth)))