Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # -*- coding: utf-8 -*-
- """
- Author: Shihab Ahmed
- Created on Mon Oct 23 05:15:03 2023
- """
- import pandas as pd
- import matplotlib.pyplot as plt
- import random
- import magnolia.speciesfile_parser as sfp
- import magnolia.bondfile_parser as bfp
- base_oil = 'PAO4'
- cutoff = 0.3
- commmon = r"C:\Users\arup2\OneDrive - University of California Merced\Desktop\LAMMPS\Base_Oil\{}\25_{}_200_O2_Soria\Production\1600".format(base_oil,base_oil)
- sim_dir = ['\\Sim-1','\\Sim-2','\\Sim-3']
- flag = True
- for sim in sim_dir:
- directory = commmon+sim
- speciesfile = directory+f'\\species_{cutoff}.out'
- if cutoff==0.30:
- speciesfile = directory+'\\species.out'
- current = sfp.get_species_count(speciesfile).T
- if flag:
- summed_species = current.copy()
- upper_bound_ = current.copy()
- lower_bound_ = current.copy()
- flag = False
- else:
- summed_species = summed_species.add(current,fill_value=0)
- ## geting upper bound
- maxx = lambda s1, s2: s1.where(s1 > s2, s2)
- upper_bound_ = upper_bound_.combine(current, maxx)
- ## getting lower bound
- minn = lambda s1, s2: s1.where(s1 < s2, s2)
- lower_bound_ = lower_bound_.combine(current, minn)
- number_of_main = 25
- species_cutoff = 4
- skipts = (1600-300)/4
- exclude= ['O2', 'H62C30']
- nsim = len(sim_dir) # number of simulation
- species = summed_species/nsim
- upper_bound = upper_bound_.copy()
- lower_bound = lower_bound_.copy()
- last_count = species.iloc[-1,:].sort_values(ascending=False)
- species = species.loc[:,last_count.index]
- species = species.loc[:,(species>=species_cutoff).any()]
- species = species.drop(exclude,axis=1)
- species.index = species.index*0.25/1000
- species = species.loc[skipts:,:]
- species.index = species.index-skipts
- ## adjusting the index of upper_bound
- upper_bound.index = upper_bound_.index*0.25/1000
- upper_bound = upper_bound.loc[skipts:,:]
- upper_bound.index = upper_bound.index-skipts
- ## adjusting the index of lower_bound
- lower_bound.index = lower_bound_.index*0.25/1000
- lower_bound = lower_bound.loc[skipts:,:]
- lower_bound.index = lower_bound.index-skipts
- time = species.index
- colors = iter(['b', 'g', 'r', 'c', 'm', 'y', 'k'])
- plt.style.use("classic")
- plt.figure(figsize=(6.4, 4.8))
- for column in species.columns:
- label = bfp.make_molecular_formula_latex(column,sort=True)
- # color = next(colors)
- plt.plot(time,species[column],label=label,linewidth=0.8)
- plt.fill_between(time, lower_bound[column], upper_bound[column],
- alpha=0.1)
- plt.legend(fontsize=15,loc='upper left')
- plt.ylim(0,60)
- plt.xlabel('Time (ps)',fontsize=15)
- plt.ylabel('Number of molecules',fontsize=15)
- title = f'This plot is generated from species file (cutoff: {cutoff})'
- plt.title(directory[directory.find('Base'):]+'\n'+title+'\n',fontsize=8)
- plt.grid('on')
- plt.savefig(r'C:\Users\arup2\OneDrive - University of California Merced\Desktop\LAMMPS\Post Processing\lammps_processing\python_outputs\species'+'\\species_speciesfile{}'.format(random.randint(0,10000000000)),dpi=400,bbox_inches='tight')
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement
