import hou
from time import sleep, time
from datetime import datetime
from hutil.Qt import QtWidgets, QtCore


def render():
    new_snap= QtWidgets.QApplication.keyboardModifiers() == QtCore.Qt.ControlModifier
      
    for clone in hou.clone.clones():
        snapshots=clone.renderGalleryDataSource()
        if snapshots:
            if len([x for x in snapshots.itemIds() if snapshots.label(x) == clone.name()])==0: #check if there are active snapshots for the clone(s)
                new_snap=True
        
        #rename previous snapshot if new snapshot is requested
        if new_snap:
            if snapshots:
                for snap in snapshots.itemIds():
                
                    if snapshots.label(snap) == clone.name():
                        snap_date=datetime.fromtimestamp(snapshots.modificationDate(snap)).strftime("%m/%d/%Y, %H:%M:%S")
                        snapshots.setLabel(snap,f"{clone.name()} {snap_date}")
                        
                        #write membuf to disk and set as source for old thumbs
                        tmpfile=snapshots.generateItemFilePath( snap, "exr" )
                        f=hou.node("/img/comp1").createNode("file")
                        f.parm("filename1").set( snapshots.filePath(snap) )
                        f.saveImage(tmpfile)
                        f.destroy()
                        snapshots.setFilePath(snap,tmpfile)
      
        
        clone.setProcessUpdates(False)
        
        #touching a clone's LopNode will generate a new set of snapshots, touching camera path will only trigger an update
        if new_snap:
            tmp=clone.lopNode()
            clone.setLopNode(None)
        else:        
            tmp=clone.cameraPath()
            clone.setCameraPath(None)
        
        clone.setProcessUpdates(True)
        
        if new_snap:
            clone.setLopNode(tmp)
        else:
            clone.setCameraPath(tmp)
    
        #small delay required so clones can start rendering.        
        sleep(0.1)
        
        clone.setProcessUpdates(False)


#find open clone control tabs        
cc_tabs=[item  for item in hou.ui.paneTabs() if getattr(item, "activeInterface", False) and item.activeInterface().name()=="clone_control"]
uiscale = hou.ui.globalScaleFactor()

#shift table down and inset pushbutton
for cc in cc_tabs:

    root=cc.activeInterfaceRootWidget()
    root.layout().setContentsMargins(0, 35* uiscale, 0, 0);

    if not getattr(root, "but_render", False):
        root.but_render = QtWidgets.QPushButton(root)
        root.but_render.clicked.connect(render)
        root.but_render.setText('Render')
        root.but_render.setGeometry(5*uiscale, 5*uiscale, 100*uiscale, 24*uiscale) 
        root.but_render.show()