__author__ = 'chunk'

from . import *
from ..common import *
from ..mjpeg import *
from ..msteg import *
from ..msteg.steganalysis import MPB

import numpy as np

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