#!/usr/bin/python

import sys, os

import subprocess, time

class ConvertToJpg():

        """This package converts every pdf file in the current directory to jpg"""

        def main(self):

                "This runs the code from the command line"

                if not os.path.exists('./jpg'):

                        os.mkdir('./jpg')

                files = [f for f in os.listdir('.') if os.path.isfile(f)]

                t0=time.time()

                for file in files:

                        filename, fileext = os.path.splitext(file)

                        if fileext == '.pdf':

                                checkstring = 'gs -q -dNODISPLAY -c \"(./'+ filename + '.pdf) (r) file runpdfbegin pdfpagecount = quit\"';

                                lastPage=subprocess.check_output(checkstring,shell=True);

                                lastPage=lastPage.rstrip('\n');

                                commandstring = 'gs -dNumRenderingThreads=4 -dNOPAUSE -sDEVICE=jpeg -dFirstPage=1 -dLastPage='+lastPage+' -sOutputFile=./jpg/'+filename+'-%d.jpg -dJPEGQ=100 -r300 -q '+filename+'.pdf -c quit'

                                #commandstring = 'convert -density 600 -limit memory 1 -limit map 1 ' + file + ' png/' + filename+'.png'

                                os.system(commandstring)

                t1=time.time()

                elapsed=t1-t0

                print("Elapsed time=%f" % elapsed)

if __name__ == "__main__":

        ES = ConvertToJpg()

        ES.main()