# Path: vaspy/iter.py
# class OsziCar(VasPy):
# class OutCar(VasPy):
# class AniFile(VasPy):
#     def __init__(self, filename='OSZICAR'):
#     def match(self, line):
#     def load(self):
#     def esort(self, var, n, reverse=False):
#     def plot(self, var, mode='show'):
#     def __init__(self, filename="OUTCAR", poscar="POSCAR"):
#     def __check(self):
#     def iforces(self):
#     def __mask_forces(self, atom_forces, tfs):
#     def fmax(self, atom_forces):
#     def forces(self, step=-1):
#     def total_forces(self):
#     def last_forces(self):
#     def last_max_force(self):
#     def last_max_atom(self):
#     def ifreq(self):
#     def check_freq_exists(func):
#         def wrapper(self):
#     def zpe(self):
#     def freq_types(self):
#     def __init__(self, filename="OUT.ANI"):
#     def __iter__(self):
#         E = [float(freq_dict["meV"])
#              for freq_dict in self.ifreq if freq_dict["freq_type"] == "f"]


import numpy as np
import matplotlib.pyplot as plt
import matplotlib.patheffects as pe
from vaspy.iter import XdatCar, AniFile
#!/usr/bin/env python
# -*- coding: utf-8 -*-


orders = [2, 6, 10, 14, 18, 22, 26, 30, 34]
max_x = 0.7
max_y = 0.8

def get_trajectories():
    xdatcar = XdatCar()
    ani = AniFile()

    trajs = []
    for i in range(len(orders)):
        trajs.append([])

    for xyz in ani:
        coordinates = xyz.data

# Path: vaspy/iter.py
# class OsziCar(VasPy):
# class OutCar(VasPy):
# class AniFile(VasPy):
#     def __init__(self, filename='OSZICAR'):
#     def match(self, line):
#     def load(self):
#     def esort(self, var, n, reverse=False):
#     def plot(self, var, mode='show'):
#     def __init__(self, filename="OUTCAR", poscar="POSCAR"):
#     def __check(self):
#     def iforces(self):
#     def __mask_forces(self, atom_forces, tfs):
#     def fmax(self, atom_forces):
#     def forces(self, step=-1):
#     def total_forces(self):
#     def last_forces(self):
#     def last_max_force(self):
#     def last_max_atom(self):
#     def ifreq(self):
#     def check_freq_exists(func):
#         def wrapper(self):
#     def zpe(self):
#     def freq_types(self):
#     def __init__(self, filename="OUT.ANI"):
#     def __iter__(self):
#         E = [float(freq_dict["meV"])
#              for freq_dict in self.ifreq if freq_dict["freq_type"] == "f"]


# Path: scripts/md_viz/traj.py
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.patheffects as pe
from vaspy.iter import XdatCar, AniFile

#!/usr/bin/env python
# -*- coding: utf-8 -*-


orders = [2, 6, 10, 14, 18, 22, 26, 30, 34]
max_x = 0.7
max_y = 0.8

def get_trajectories():
    xdatcar = XdatCar()
    ani = AniFile()

    trajs = []
    for i in range(len(orders)):
        trajs.append([])

    for xyz in ani:
        coordinates = xyz.data

        for j in range(len(orders)):