export_ZLOC_for_3de.py

#
#
# 3DE4.script.name:    Export ZLOC...
#
# 3DE4.script.version:    v1.0
#
# 3DE4.script.gui:    Main Window::3DE4::File::Export
# 3DE4.script.gui:    Object Browser::Context Menu Point
# 3DE4.script.gui:    Object Browser::Context Menu Points
# 3DE4.script.gui:    Object Browser::Context Menu PGroup
#
# 3DE4.script.comment:    Exports selected trackers to ZLOC file.
# 3DE4.script.comment:    Author: Hyuk Ko
# 3DE4.script.comment:    v1.0 @ 2018/10/19
# 3DE4.script.comment:    [email protected]
#
# Original Script:    export_tracks.py(by 3DE) - modified by Hyuk Ko
#

import subprocess
import platform

currentOS = platform.system() # 'Windows' or 'Darwin'(MACOS) or 'Linux'

def copy_to_clipboard(os, text):
    if os == 'Windows':
        p = subprocess.Popen(['clip'], stdin=subprocess.PIPE)        
    elif os == 'Darwin':
        p = subprocess.Popen(['pbcopy'], stdin=subprocess.PIPE)
    elif os == 'Linux':
        p = subprocess.Popen(['xclip', '-selection', 'c'], stdin=subprocess.PIPE)
        #p = subprocess.Popen(['xsel'], stdin=subprocess.PIPE) # If xclip dosnt work
    p.stdin.write(text)
    p.stdin.close()

c = tde4.getCurrentCamera()
pg = tde4.getCurrentPGroup()

if c!=None and pg!=None:
    n = tde4.getCameraNoFrames(c)    
    p = tde4.getContextMenuObject() # check if context menu has been used, and retrieve point...
    if p!=None:
        pg = tde4.getContextMenuParentObject() # retrieve point's parent pgroup (not necessarily being the current one!)...
        l = tde4.getPointList(pg, 1)
    else:
        l = tde4.getPointList(pg, 1) # otherwise use regular selection...
    if len(l)>0:
        clipStr = ''
        req = tde4.createCustomRequester()
        tde4.addFileWidget(req,"file_browser","Filename...","*.txt")
        tde4.addTextFieldWidget(req, "frame_offset_field", "Frame Offset","0")
        tde4.addTextFieldWidget(req, "overscan_field", "Overscan","1.00")
        tde4.addToggleWidget(req,"remove_distortion","Remove Distortion",1)
        
        ret = tde4.postCustomRequester(req, "Export ZLOC...", 500, 0, "Save TXT File", "Copy to Clipboard", "Cancel")
        if ret==1:
            path = tde4.getWidgetValue(req,"file_browser")
            offset = int(tde4.getWidgetValue(req,"frame_offset_field"))
            overscan = float(tde4.getWidgetValue(req, "overscan_field"))
            rd = tde4.getWidgetValue(req,"remove_distortion")
            
            if offset<0:
                offset = 0
            if path!=None:
                #
                # main block...
                
                if path.find(".txt",len(path)-4)==-1: path += ".txt"
                f = open(path,"w")
                if not f.closed:
                    for point in l:
                        name = tde4.getPointName(pg,point)
                        c2d = tde4.getPointPosition2DBlock(pg,point,c,1,n)
                        n0 = 0
                        for v in c2d:
                            if v[0]!=-1.0 and v[1]!=-1.0: n0 += 1
                        frame = 1+offset
                        for v in c2d:
                            if v[0]!=-1.0 and v[1]!=-1.0 and rd:
                                dv = tde4.removeDistortion2D(c, frame-offset, v)
                                f.write("%s %d %.15f %.15f\n"%("zloc"+name,frame,(2*dv[0]-1)/overscan,(-2*dv[1]+1)/overscan))
                            else:
                                f.write("%s %d %.15f %.15f\n"%("zloc"+name,frame,(2*v[0]-1)/overscan,(-2*v[1]+1)/overscan))
                            frame += 1
                    f.close()
                else:
                    tde4.postQuestionRequester("Export ZLOC...","Error, couldn't open file.","Ok")
                
                # end main block...
                #		
		
        if ret==2:
            offset = int(tde4.getWidgetValue(req, "frame_offset_field"))
            overscan = float(tde4.getWidgetValue(req, "overscan_field"))
            rd = tde4.getWidgetValue(req,"remove_distortion")
            
            if offset<0:
                offset = 0    
            #
            # main block...    
            for point in l:
                name = tde4.getPointName(pg, point)
                c2d = tde4.getPointPosition2DBlock(pg, point, c, 1, n)
                n0 = 0
                for v in c2d:
                    if v[0]!=-1.0 and v[1]!=-1.0: n0 += 1
                frame = 1+offset
                for v in c2d:
                    if v[0]!=-1.0 and v[1]!=-1.0 and rd: 
                        dv = tde4.removeDistortion2D(c, frame-offset, v)
                        clipStr += "%s %d %.15f %.15f\n"%("zloc"+name,frame,(2*dv[0]-1)/overscan,(-2*dv[1]+1)/overscan)
                    else:
                        clipStr += "%s %d %.15f %.15f\n"%("zloc"+name,frame,(2*v[0]-1)/overscan,(-2*v[1]+1)/overscan)
                    frame += 1
                    
            copy_to_clipboard(currentOS, clipStr)    
    else:
        tde4.postQuestionRequester("Export ZLOC...","There are no selected points.","Ok")
else:
        tde4.postQuestionRequester("Export ZLOC...","There is no current Point Group or Camera.","Ok")