# XML export script
# Copyright (C) The Pixel Farm Ltd. 2016-2020

import pfpy
import math

def pfExportName():
    return 'XML Camera'

def pfExportExtension():
    return 'xml'

def pfExportSupport():
    return ('cameras', 'multiCamera')
	
def pfExportSystem():
    return ('right', 'y')

def main():
    fobj= open(pfpy.getExportFilename(), 'w')

    pfpy.setExportCoordinateSystem('right', 'y')

    fobj.write('<?xml version=\"1.0\" encoding=\"UTF-8\" standalone=\"yes\"?>\n')
    fobj.write('\n<xml>\n');

    for j in range(0, pfpy.getNumCameras()):
    
        cam= pfpy.getCameraRef(j)
        if cam.getExport():

            fobj.write('<pfcamera>\n')
            fobj.write('<coordinateSystem>right</coordinateSystem>\n')
            fobj.write('<upVector>y</upVector>\n')
            fobj.write('<rotationOrder>xyz</rotationOrder>\n')

            clip= pfpy.getClipRef(j)
            firstKey= clip.getInPoint()
            lastKey= clip.getOutPoint()
            numFrames= 1+lastKey-firstKey

            fps= clip.getFrameRate()
            asp= clip.getPixelAspect()
            xsize= clip.getFrameWidth()
            ysize= clip.getFrameHeight()

            footageName= clip.getFilename('all')
            if not clip.isMovieFile():
                wild= clip.getFilename('num')
            else:
                wild= '';
            footageName= clip.getFilename('dir')+clip.getFilename('file')+clip.getFilename('sep')+wild+'.'+clip.getFilename('ext')

            hfb= cam.getSensorWidth('mm')
            vfb= cam.getSensorHeight('mm')
            asp= cam.getPixelAspect();

            fobj.write('<pixelAspect>%g'%asp+'</pixelAspect>\n')
            fobj.write('<filmBack>%g'%hfb+' %g'%vfb+' mm</filmBack>\n')
            fobj.write('<frameSize>%d'%xsize+' %d'%ysize+'</frameSize>\n')

            for i in range(firstKey, lastKey+1):
                v= cam.getTranslation(i)
                q= cam.getQuaternionRotation(i)
                r= cam.getEulerRotation(i, 'xyz')
                hfov= cam.getHorizontalFOV(i, 'deg')
                vfov= cam.getVerticalFOV(i, 'deg')
                focal= cam.getFocalLength(i, 'mm')

                fobj.write('<translation frame=\'%d'%i+'\'>%g'%v[0]+' %g'%v[1]+' %g'%v[2]+'</translation>\n')
                fobj.write('<rotation frame=\'%d'%i+'\'>%g'%r[0]+' %g'%r[1]+' %g'%r[2]+'</rotation>\n')
                fobj.write('<focalLength frame=\'%d'%i+'\'>%g'%focal+' mm</focalLength>\n')

            fobj.write('</pfcamera>\n')

    fobj.write('</xml>\n')
    fobj.close()