__author__ = 'chunk'

from mfeat import *

from common import *
import numpy as np

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)