__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)

        return tpm