import numpy

from gavo import api
from gavo.helpers import processing
from gavo.utils import fitstools
from gavo.utils import imgtools
from gavo.utils import pyfits


def _getArrayFor(srcName, extInd):
	return numpy.array(list(
		fitstools.iterScaledRows(srcName, destSize=200, extInd=extInd)))


class PreviewMaker(processing.PreviewMaker):
	def getPreviewData(self, accref):
		arr = _getArrayFor(api.getConfig("inputsDir") / accref, 0)
		arr[abs(arr)>1] = 1e-30
		return imgtools.jpegFromNumpyArray(arr)


if __name__=="__main__":
	processing.procmain(PreviewMaker, "smakced/q", "import_main")