View source for User:Speleo3/VMD plugins

=== vmd_plugins_update.py ===

<syntaxhighlight lang="python">
'''
Convert VMD plugin source files for PyMOL.
(Modified version of "contrib/uiuc/plugins/molfile_plugin/update.py")

Please download and unpack the VMD source code from:

  http://www.ks.uiuc.edu/Development/Download/download.cgi?PackageName=VMD

Then run this script from the PyMOL source root and point it to the "plugins"
directory from the VMD source tree. Example:

  cd /tmp
  tar xzf vmd-1.9.1.src.tar.gz
  cd /tmp/pymol
  python vmd_plugins_update.py /tmp/plugins

You can enable/disable particular plugins from the "src_list" variable in
this file.
'''

import os
import sys
from glob import glob
from distutils import file_util, dir_util

src_list = [
#    'avsplugin',
#    'babelplugin', # requires openbabel
#    'basissetplugin', #
#    'bgfplugin',
#    'binposplugin',
#    'biomoccaplugin',
#    'brixplugin',
#    'carplugin',
#    'ccp4plugin',
#    'cdfplugin', # requires netcdf
#    'corplugin',
#    'cpmdlogplugin', #
#    'cpmdplugin',
    'crdplugin',
#    'cubeplugin',
    'dcdplugin',
#    'dlpolyplugin',
#    'dsn6plugin',
#    'dtrplugin', #
#    'dxplugin',
#    'edmplugin',
#    'fs4plugin',
#    'gamessplugin',
#    'gaussianplugin', #
#    'graspplugin',
#    'grdplugin',
#    'gridplugin',
    'gromacsplugin',
#    'hoomdplugin', # requires expat
#    'jsplugin', #
#    'lammpsplugin', # requires gz
#    'maeffplugin', #
#    'mapplugin',
#    'mdfplugin',
#    'mmcif', #
#    'mol2plugin',
#    'moldenplugin',
#    'mrcplugin', # gone?
#    'msmsplugin',
#    'namdbinplugin',
#    'netcdfplugin', # requires netcdf
#    'parm7plugin',
#    'parmplugin',
#    'pbeqplugin', #
#    'pdbplugin',
#    'phiplugin',
#    'pltplugin',
#    'pqrplugin',
#    'psfplugin',
#    'raster3dplugin',
#    'rst7plugin',
#    'situsplugin',
#    'spiderplugin',
#    'stlplugin',
#    'tinkerplugin',
#    'uhbdplugin',
#    'vaspchgcarplugin', #
#    'vaspoutcarplugin', #
#    'vaspparchgplugin', #
#    'vaspposcarplugin', #
#    'vaspxdatcarplugin', #
#    'vaspxmlplugin', #
#    'vtfplugin', #
#    'webpdbplugin', # tcl dependent
#    'xbgfplugin',
#    'xsfplugin',
#    'xyzplugin',
]

# check command line argument
if len(sys.argv) < 2 or sys.argv[1] in ('-h', '--help'):
    print __doc__.strip()
    sys.exit(1)

# paths
vmd_plugins_path = sys.argv[1]
molfile_src_path = os.path.join(vmd_plugins_path, 'molfile_plugin', 'src')
molfile_dst_path = os.path.join('contrib', 'uiuc_molfile_plugin')

# copy required files
dir_util.mkpath(molfile_dst_path)
for f in [os.path.join(molfile_src_path, 'hash.c')] + \
        glob(os.path.join(molfile_src_path, '*.h*')) + \
        glob(os.path.join(vmd_plugins_path, 'include', '*.h')):
    file_util.copy_file(f, molfile_dst_path)

# prepare setup_vmd_plugins.py
setup_out = open('setup_vmd_plugins.py', 'w')
setup_out.write('vmd_plugins_dirs = [ %s ]\n' % repr(molfile_dst_path))
setup_out.write('vmd_plugins_files = [\n')
for f in ['PlugIOManagerInit.c', 'hash.c']:
    out_file = os.path.join(molfile_dst_path, f)
    setup_out.write(repr(out_file) + ',\n')

# patch and copy desired plugin source files
for pref in src_list:
    try:
        in_file = glob(molfile_src_path + "/" + pref + ".[cC]*")[0]
    except KeyError:
        print 'MISSING:', pref
        continue

    out_file = os.path.join(molfile_dst_path, pref + '.c')
    content = open(in_file).read()
    content.replace('(vmdplugin_t *)', '(vmdplugin_t *)(void*)')

    # c++
    if in_file[-2:] == '.C' or in_file[-4:] == '.cxx':
        out_file += 'pp'
        content = content.replace('VMDPLUGIN_API', 'VMDPLUGIN_EXTERN')

    print in_file, '->', out_file
    setup_out.write(repr(out_file) + ',\n')

    with open(out_file, 'w') as g:
        g.write("/* MACHINE GENERATED FILE, DO NOT EDIT! */\n\n")
        g.write("#define VMDPLUGIN molfile_%s\n" % pref)
        g.write("#define STATIC_PLUGIN 1\n\n")    
        g.write(content)

setup_out.write(']\n')
setup_out.close()

# PlugIO init file
with open(os.path.join(molfile_dst_path, 'PlugIOManagerInit.c'), 'w') as g:
    g.write('''
/* MACHINE GENERATED FILE, DO NOT EDIT! */
#include "vmdplugin.h"
typedef struct _PyMOLGlobals PyMOLGlobals;
int PlugIOManagerRegister(PyMOLGlobals *G, vmdplugin_t *);
''')

    # prototypes
    for pref in src_list:
        g.write('''
int molfile_%s_init(void);
int molfile_%s_register(void *, vmdplugin_register_cb);
int molfile_%s_fini(void);
''' % (pref, pref, pref))

    g.write('''
int PlugIOManagerInitAll(PyMOLGlobals *G) {
   int ok = 1 \\
''')

    # init
    for pref in src_list:
        g.write('''
      && (molfile_%s_init() == VMDPLUGIN_SUCCESS)''' % pref)

    # register
    for pref in src_list:
        g.write('''
      && (molfile_%s_register(G, (vmdplugin_register_cb)PlugIOManagerRegister) == VMDPLUGIN_SUCCESS)''' % pref)

    g.write(''';

   return ok;
}

int PlugIOManagerFreeAll(void) {
   int ok = 1 \\
''')

    # fini
    for pref in src_list:
        g.write('''
      && (molfile_%s_fini() == VMDPLUGIN_SUCCESS)''' % pref)

    g.write(''';

   return ok;
}
''')
</syntaxhighlight>