#
#
#
from libs.jpg_bin import JpgBin
from libs.jpg_picseal import JpgPicSeal


class JpgBinWrite:

  
  #
  def __init__(self, jpg_in, jpg_out):
    self.fhr = jpg_in
    self.fhw = jpg_out

  #
  # input is the Crypto Sig class
  #
  def writeJpgPicSealPub(self, crypto_sig, fp):
    self.__writeJpgHeader()
    JpgPicSeal.writePub(self.fhw, crypto_sig)
    self.__writeJpgImg(fp)
  
  #
  # input is the Crypto Sig class
  #
  def writeJpgPicSealPvt(self, crypto_sig, fp):
    self.__writeJpgHeader()
    JpgPicSeal.writePvt(self.fhw, crypto_sig)
    self.__writeJpgImg(fp)


  #
  def __writeJpgImg(self, fp):
    self.__writeJpgMetadata(fp.markers_meta)
    self.__writeJpgImgData(fp.markers_img)
    self.__writeJpgFooter()
    self.fhw.flush()
    self.fhw.close()

  #
  # array of the marker information
  #
  def __writeJpgMetadata(self, markers):
    for marker in markers:
      self.fhr.seek(marker.fpos)
      data = self.fhr.read(marker.len)
      self.fhw.write(data)
      self.fhw.flush()
    
  #
  # array of the marker information
  #
  def __writeJpgImgData(self, markers):
    for marker in markers:
      # most of the image data is sequential
      cpos = self.fhr.tell()
      if (marker.fpos != cpos):
        self.fhr.seek(marker.fpos)

      data = self.fhr.read(marker.len)
      self.fhw.write(data)
      self.fhw.flush()

  #
  def __writeJpgHeader(self):
    self.fhw.write(bytes(JpgBin.soi_marker))

  #
  def __writeJpgFooter(self):
    self.fhw.write(bytes(JpgBin.eof_marker))