222 lines
7.2 KiB
Python
222 lines
7.2 KiB
Python
|
try:
|
||
|
|
import BaseHTTPServer
|
||
|
|
except ImportError:
|
||
|
|
import http.server as BaseHTTPServer
|
||
|
|
|
||
|
|
import time
|
||
|
|
import cgi
|
||
|
|
import threading
|
||
|
|
import traceback
|
||
|
|
|
||
|
|
from pymol import cmd
|
||
|
|
|
||
|
|
_server = None
|
||
|
|
def _shutdown(self_cmd=cmd):
|
||
|
|
if _server != None:
|
||
|
|
_server.socket.close()
|
||
|
|
self_cmd.quit()
|
||
|
|
|
||
|
|
# Note, this handler assumes PyMOL is running as a global singleton
|
||
|
|
|
||
|
|
def get_status(out, self_cmd=cmd):
|
||
|
|
out.write('<html>\n')
|
||
|
|
out.write('<header>\n')
|
||
|
|
out.write('<script type="text/javascript" src="pymol.js"></script>\n')
|
||
|
|
out.write('</header><body>\n')
|
||
|
|
out.write('<h3>PyMOL WebGUI Proof of Concept</h3>\n')
|
||
|
|
out.write('<table><tr>\n')
|
||
|
|
out.write('<td><form action="./status.pymol"><button type="submit">Refresh</button></form></td>\n')
|
||
|
|
out.write('<td><form target="_new" action="./ray.pymol?t=%f"><button type="submit">Ray</button></form></td>\n'%
|
||
|
|
time.time())
|
||
|
|
out.write('<td><form target="_new" action="./monitor.pymol?t=%f"><button type="submit">Monitor</button></form></td>\n'%
|
||
|
|
time.time())
|
||
|
|
out.write('<td><form action="./quit.pymol"><button type="submit">Quit</button></form></td>\n')
|
||
|
|
out.write('</tr></table>')
|
||
|
|
out.write('<a href="./status.pymol?load">load $TUT/1hpv.pdb</a>\n')
|
||
|
|
names = self_cmd.get_names('objects')
|
||
|
|
if not len(names):
|
||
|
|
out.write('<p>No objects loaded.</p>\n')
|
||
|
|
else:
|
||
|
|
out.write('<p>Loaded Objects:</p><ul>\n')
|
||
|
|
for name in names:
|
||
|
|
out.write('<li>%s</li>\n'%name)
|
||
|
|
out.write('</ul>\n')
|
||
|
|
out.write('<a href="#" onClick="updateImage()"><img src="./draw.pymol?t=%f">'%time.time()+"</img></a>")
|
||
|
|
out.write('</body></html>\n')
|
||
|
|
|
||
|
|
def get_monitor(out, self_cmd=cmd):
|
||
|
|
out.write('<html>\n')
|
||
|
|
out.write('<header>\n')
|
||
|
|
out.write('<script type="text/javascript" src="pymol.js"></script>\n')
|
||
|
|
out.write('</header><body onload="monitorOnLoad()">\n')
|
||
|
|
out.write('<img src="./draw.pymol?t=%f">'%time.time()+"</img>")
|
||
|
|
out.write('</body></html>\n')
|
||
|
|
|
||
|
|
def get_start(out, self_cmd=cmd):
|
||
|
|
window_open="javascript: window.open('./status.pymol','hello', 'location=no,toolbar=no,width=400,height=600,top=0,left=880');"
|
||
|
|
out.write('<html><body onload="'+window_open+'">\n')
|
||
|
|
out.write('<a href="./start.pymol" onclick="'+window_open+'">Launch WebGUI\n')
|
||
|
|
out.write('</body></html>')
|
||
|
|
|
||
|
|
def write_image(out, ray=0, self_cmd=cmd):
|
||
|
|
if ray:
|
||
|
|
self_cmd.ray()
|
||
|
|
# encode the file descriptor into the PNG filename
|
||
|
|
if self_cmd.png(chr(1)+str(out.fileno()),prior=-1) != 1:
|
||
|
|
# no prior image available, so wait for update / finish
|
||
|
|
self_cmd.sync()
|
||
|
|
|
||
|
|
class PymolHandler(BaseHTTPServer.BaseHTTPRequestHandler):
|
||
|
|
|
||
|
|
|
||
|
|
def log_message(self, format, *args):
|
||
|
|
pass
|
||
|
|
# nuke logging feature for the time being
|
||
|
|
|
||
|
|
def do_js(self):
|
||
|
|
self.send_response(200)
|
||
|
|
self.send_header('Content-type','text/javascript')
|
||
|
|
self.end_headers()
|
||
|
|
self.wfile.write('''
|
||
|
|
|
||
|
|
|
||
|
|
function updateImage()
|
||
|
|
{
|
||
|
|
images = document.getElementsByTagName("img");
|
||
|
|
|
||
|
|
for( var i = 0; i < images.length; i++ ) {
|
||
|
|
images[i].src = "./draw.pymol?t=" + new Date().getTime();
|
||
|
|
}
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
|
||
|
|
function monitorOnLoad(event)
|
||
|
|
{
|
||
|
|
setInterval('updateImage()',1000)
|
||
|
|
}
|
||
|
|
|
||
|
|
''')
|
||
|
|
|
||
|
|
def do_pymol(self):
|
||
|
|
if "ray.pymol" in self.path: # send image
|
||
|
|
self.send_response(200)
|
||
|
|
self.send_header('Content-type', 'image/x-png')
|
||
|
|
self.send_header('Cache-control', 'no-cache')
|
||
|
|
self.send_header('Pragma', 'no-cache')
|
||
|
|
self.end_headers()
|
||
|
|
write_image(self.wfile,1)
|
||
|
|
elif "draw.pymol" in self.path:
|
||
|
|
self.send_response(200)
|
||
|
|
self.send_header('Content-type', 'image/x-png')
|
||
|
|
self.send_header('Cache-control', 'no-cache')
|
||
|
|
self.send_header('Pragma', 'no-cache')
|
||
|
|
self.end_headers()
|
||
|
|
write_image(self.wfile)
|
||
|
|
else:
|
||
|
|
if "load" in self.path: # load a structure
|
||
|
|
cmd.load("$TUT/1hpv.pdb")
|
||
|
|
cmd.rock()
|
||
|
|
self.send_response(200)
|
||
|
|
self.send_header('Content-type', 'text/html')
|
||
|
|
self.end_headers()
|
||
|
|
if "status.pymol" in self.path:
|
||
|
|
get_status(self.wfile)
|
||
|
|
elif "monitor.pymol" in self.path:
|
||
|
|
get_monitor(self.wfile)
|
||
|
|
elif "quit.pymol" in self.path:
|
||
|
|
self.wfile.write('<html><body><p>Quitting...</p></body></html>')
|
||
|
|
self.wfile.flush()
|
||
|
|
_shutdown()
|
||
|
|
else: # start page
|
||
|
|
get_start(self.wfile)
|
||
|
|
self.wfile.flush()
|
||
|
|
|
||
|
|
def do_GET(self):
|
||
|
|
try:
|
||
|
|
doc = self.path.split('?')[0]
|
||
|
|
if doc.endswith('.pymol'): # PyMOL
|
||
|
|
try:
|
||
|
|
self.do_pymol()
|
||
|
|
except:
|
||
|
|
traceback.print_exc()
|
||
|
|
elif doc.endswith('.js'): # Javascript
|
||
|
|
self.do_js()
|
||
|
|
elif doc.endswith('.html'):
|
||
|
|
f = open('.'+self.path) # UNSAFE!!!
|
||
|
|
self.send_response(200)
|
||
|
|
self.send_header('Content-type', 'text/html')
|
||
|
|
self.end_headers()
|
||
|
|
self.wfile.write(f.read())
|
||
|
|
f.close()
|
||
|
|
except IOError:
|
||
|
|
self.send_error(404,'File Not Found: %s' % self.path)
|
||
|
|
|
||
|
|
def do_POST(self):
|
||
|
|
global rootnode
|
||
|
|
try:
|
||
|
|
ctype, pdict = cgi.parse_header(self.headers.getheader('content-type'))
|
||
|
|
if ctype == 'multipart/form-data':
|
||
|
|
query=cgi.parse_multipart(self.rfile, pdict)
|
||
|
|
self.send_response(301)
|
||
|
|
|
||
|
|
self.end_headers()
|
||
|
|
upfilecontent = query.get('upfile')
|
||
|
|
print("filecontent", upfilecontent[0])
|
||
|
|
self.wfile.write('<HTML>POST OK.<BR><BR>');
|
||
|
|
self.wfile.write(upfilecontent[0]);
|
||
|
|
|
||
|
|
except :
|
||
|
|
pass
|
||
|
|
|
||
|
|
def main():
|
||
|
|
try:
|
||
|
|
global _server
|
||
|
|
_server = BaseHTTPServer.HTTPServer(('', 8080), PymolHandler)
|
||
|
|
print('started httpserver...')
|
||
|
|
_server.serve_forever()
|
||
|
|
except KeyboardInterrupt:
|
||
|
|
print('^C received, shutting down server')
|
||
|
|
_server.socket.close()
|
||
|
|
|
||
|
|
def open_browser():
|
||
|
|
import webbrowser
|
||
|
|
time.sleep(1)
|
||
|
|
webbrowser.open('http://localhost:8080/status.pymol')
|
||
|
|
# import os
|
||
|
|
# os.system('open http://localhost:8080/start.pymol')
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|
||
|
|
|
||
|
|
if __name__ == 'pymol':
|
||
|
|
|
||
|
|
cmd.set("image_copy_always") # copy all updates into image buffer
|
||
|
|
|
||
|
|
t = threading.Thread(target=main)
|
||
|
|
t.setDaemon(1)
|
||
|
|
t.start()
|
||
|
|
|
||
|
|
t = threading.Thread(target=open_browser)
|
||
|
|
t.setDaemon(1)
|
||
|
|
t.start()
|
||
|
|
|
||
|
|
|
||
|
|
"""
|
||
|
|
|
||
|
|
okay, what we need now is a simple safe way to pass comands and
|
||
|
|
arguments through the client brower, with escape characters, etc.
|
||
|
|
|
||
|
|
ideally, we'd like a javascript object which responds to the same messages as PyMOL
|
||
|
|
|
||
|
|
for our initial test, let's just put up some button links to control representations.
|
||
|
|
|
||
|
|
also outstanding:
|
||
|
|
|
||
|
|
- how does one send asynchronous javascript URL requests?
|
||
|
|
|
||
|
|
- we also need some way to package up images in PyMOL without passaging through the file systems
|
||
|
|
|
||
|
|
- we also need to decide what this initial user interface is going to do
|
||
|
|
|
||
|
|
"""
|