24小时热门版块排行榜    

CyRhmU.jpeg
查看: 1134  |  回复: 5

125442548

铁虫 (初入文坛)

[求助] 求助Python大神

本人做的是分子动力学模拟,现在要将两个文件格式转化一下,目前从网上下载了一段Python脚本,怎么改也无法对自己的两个文件进行处理,现在求助Python大神帮忙。
下面贴上程序
#!/usr/bin/python

# This script converts a CHARMm psf file to a NAMD (= XPLOR) psf file.
# USAGE: charmm2namd psf-file rtf-file  
#
# EPF 04-04-2000
#

import sys,struct,os,time

if len(sys.argv) < 3:
  sys.stderr.write ("USAGE: charmm2namd psf-file rtf-file > psf-file\n"
  sys.stderr.write ("       The rtf-file must contain the correct MASS entries for all\n"
  sys.stderr.write ("       atom types present in the psf-file.\n"  
  os._exit(0)

psffile = sys.argv[1]
rtffile = sys.argv[2]

# Read the MASS entries from the rtf-file.

f=open(rtffile)

buffer = " "
potentials = []

while buffer != "":
  buffer = f.readline()
  if buffer != "":
    if buffer[0:4] == "MASS":
      potentials.append(buffer.split(None,4))  
     
f.close()

# Now open the psf file and do the conversion of the potential types.

f=open(psffile)


buffer = f.readline()
# search for the line stating number of title lines
while buffer != "" and "!NTITLE" not in buffer:
  sys.stdout.write(buffer)
  buffer = f.readline()

titlelines = int(buffer[:buffer.find("!NTITLE"] )
print '%8d !NTITLE' % (titlelines+1)

# The title lines
for i in range(titlelines):
  buffer = f.readline()
  sys.stdout.write(buffer)

# add extra title line
print '* Converted to NAMD format using '+rtffile+' on '+time.ctime(time.time())

# search for the line stating number of atoms
while buffer != "" and "!NATOM" not in buffer:
  buffer = f.readline()
  sys.stdout.write(buffer)
      
numofatoms = float(buffer[:buffer.find("!NATOM"] )


j = 0;
while buffer != "":
  buffer = f.readline()
  if buffer != "" and j < numofatoms:
    line = buffer.split(None)
   
    # now search through potential list

    oldpot = line[5]
    for p in potentials:
      if p[1] == line[5]:
        line[5] = p[2]
    if oldpot == line[5]:
      sys.stderr.write('\nFatal error: no potential for atom type '+oldpot+'\n\n' )
      os._exit(0)

    linetuple=(int(line[0]),line[1],line[2],line[3],line[4],line[5],float(line[6]),float(line[7]),int(line[8]) )        
            
    psfline =  '   %5d %-4s %-4s %-4s %-4s %4s  %9.6f      %8.4f           %d' % linetuple
    print psfline
   
    j += 1
  else:
    sys.stdout.write(buffer)
            
f.close()  
sys.stderr.write('\nConverted '+str(j)+' atoms.\n\n' )

求助Python大神这是我要处理的两个文件
回复此楼

» 猜你喜欢

» 本主题相关商家推荐: (我也要在这里推广)

» 本主题相关价值贴推荐,对您同样有帮助:

已阅   回复此楼   关注TA 给TA发消息 送TA红花 TA的回帖

reagas

木虫 (正式写手)

你运行一下,看看报错,逐步改就行了

发自小木虫Android客户端
2楼2015-12-25 17:09:32
已阅   回复此楼   关注TA 给TA发消息 送TA红花 TA的回帖

125442548

铁虫 (初入文坛)

引用回帖:
2楼: Originally posted by reagas at 2015-12-25 17:09:32
你运行一下,看看报错,逐步改就行了

我试过好多遍了,不知道怎么读入我需要处理的两个文件
3楼2015-12-25 17:10:45
已阅   回复此楼   关注TA 给TA发消息 送TA红花 TA的回帖

reagas

木虫 (正式写手)

python xxx.py file1name file2name

发自小木虫Android客户端
4楼2015-12-25 17:13:12
已阅   回复此楼   关注TA 给TA发消息 送TA红花 TA的回帖

125442548

铁虫 (初入文坛)

引用回帖:
4楼: Originally posted by reagas at 2015-12-25 17:13:12
python xxx.py file1name file2name

我试一下,行不行的反正先谢过了
5楼2015-12-26 13:14:23
已阅   回复此楼   关注TA 给TA发消息 送TA红花 TA的回帖

125442548

铁虫 (初入文坛)

引用回帖:
2楼: Originally posted by reagas at 2015-12-25 17:09:32
你运行一下,看看报错,逐步改就行了

是不是还要设置输出文件啊?我运行了下怎么没见到结果,原文件也没有变化
6楼2015-12-27 19:39:34
已阅   回复此楼   关注TA 给TA发消息 送TA红花 TA的回帖
相关版块跳转 我要订阅楼主 125442548 的主题更新
信息提示
请填处理意见