Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 6 additions & 6 deletions plumed/io/parsers/__init__.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@

from read_wham import read_wham
from read_colvar import read_colvar
from read_fes import read_fes
from read_hills import read_hills
from read_plumed import read_plumed
from read_str import read_str
from .read_wham import read_wham
from .read_colvar import read_colvar
from .read_fes import read_fes
from .read_hills import read_hills
from .read_plumed import read_plumed
from .read_str import read_str
24 changes: 13 additions & 11 deletions plumed/io/parsers/read_colvar.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,17 +3,19 @@
import numpy as np
import six


def read_colvar(fname, verbose=False):
if verbose: print(fname)
f = open(fname) if isinstance(fname, six.string_types) else fname
if verbose:
print(fname)
f = open(fname) if isinstance(fname, six.string_types) else fname

with f:
line = f.readline()
assert line[:2] == '#!', 'Missing or incorrect header'
columns = line.split()[2:]
f.seek(0)
df = pd.read_csv(f, delim_whitespace=True, comment="#", skiprows=1, header=None)
with f:
line = f.readline()
assert line[:2] == '#!', 'Missing or incorrect header'
columns = line.split()[2:]
f.seek(0)
df = pd.read_csv(f, delim_whitespace=True, comment="#", skiprows=1, header=None)

df.columns = columns
df.time = map(np.round, df.time)
return df
df.columns = columns
df.time = map(np.round, df.time)
return df
13 changes: 7 additions & 6 deletions plumed/io/parsers/read_fes.py
Original file line number Diff line number Diff line change
@@ -1,10 +1,11 @@

import pandas as pd


def read_fes(fname):
lines = [l[1:] for l in open(fname).readlines() if l.startswith("#")]
columns = lines[1].split()
data = [map(float, l.split()) for l in lines[2:]]
df = pd.DataFrame(data, columns=columns)
df.Window = map(int, df.Window)
return df
lines = [l[1:] for l in open(fname).readlines() if l.startswith("#")]
columns = lines[1].split()
data = [map(float, l.split()) for l in lines[2:]]
df = pd.DataFrame(data, columns=columns)
df.Window = map(int, df.Window)
return df
11 changes: 6 additions & 5 deletions plumed/io/parsers/read_hills.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,10 @@

import pandas as pd


def read_hills(f="HILLS"):
lines = open(f).readlines()
columns = lines[0].split()[2:]
data = [map(float, l.split()) for l in lines if not l.startswith("#")]
df = pd.DataFrame(data, columns=columns)
return df
lines = open(f).readlines()
columns = lines[0].split()[2:]
data = [map(float, l.split()) for l in lines if not l.startswith("#")]
df = pd.DataFrame(data, columns=columns)
return df
1 change: 1 addition & 0 deletions plumed/io/parsers/read_str.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@

import pandas as pd


def read_str(filename):

columns = ['_Gen_dist_constraint.ID',
Expand Down
9 changes: 5 additions & 4 deletions plumed/io/parsers/read_wham.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,9 @@

import pandas as pd


def read_wham(fname):
df = pd.read_csv(fname, sep="\t", comment="#", header=None)
columns = "Coor Free FreeErr Prob ProbErr".split()
df.columns = columns
return df
df = pd.read_csv(fname, sep="\t", comment="#", header=None)
columns = "Coor Free FreeErr Prob ProbErr".split()
df.columns = columns
return df