import os

import numpy

from gavo import api
from gavo.utils import fitstools, imgtools


PREVIEW_SIZE = 200

class PreviewMaker(api.PreviewMaker):
	def getPreviewData(self, srcName):
		with open(os.path.join(api.getConfig("inputsDir"), srcName)) as inFile:
			pixels = numpy.array([row
				for row in fitstools.iterScaledRows(inFile,
					destSize=PREVIEW_SIZE)])
		return imgtools.jpegFromNumpyArray(pixels)


if __name__=="__main__":
  api.procmain(PreviewMaker, "plts/q", "import")