__author__ = 'chunk'

from . import *

from ..mjpeg import *

from ..msteg import *
from ..msteg.steganalysis import MPB

class FeatIntraBlockDiff(FeatureBase):

    def __init__(self):

        FeatureBase.__init__(self)

    def feat(self, image):
        im = Jpeg(image, key=sample_key)
        ciq = im.coef_arrays[colorMap['Y']]
        tpm = MPB.MPB().get_trans_prob_mat(ciq)

        return tpm