Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- # -*- coding: utf-8 -*-
- """
- Author: Shihab Ahmed
- Created on Tue Dec 12 01:22:20 2023
- """
- import pandas as pd
- import matplotlib.pyplot as plt
- import random
- import magnolia.speciesfile_parser as sfp
- import numpy as np
- base_oil = 'PAO4'
- spec = 'H2O'
- plt.style.use("classic")
- plt.figure(figsize=(6.4, 4.8))
- for cutoff in np.array(range(30,76,5))/100:
- 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
- species = species.loc[:,spec]
- ## 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
- # lower_bound = lower_bound[:,spec]
- # upper_bound = upper_bound[:,spec]
- time = species.index
- colors = iter(['b', 'g', 'r', 'c', 'm', 'y', 'k'])
- # color = next(colors)
- plt.plot(time,species.values,linewidth=0.8, label=cutoff)
- # plt.fill_between(time, lower_bound[species.name], upper_bound[species.name],
- # alpha=0.1)
- plt.legend(fontsize=10,loc='upper left')
- # plt.ylim(-0.1,)
- plt.xlabel('Time (ps)',fontsize=15)
- plt.ylabel('Number of molecules',fontsize=15)
- title = f'This plot is generated from species file (cutoff: {cutoff}), spec={spec}'
- 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
