From a26a909b45161c46d018322b82263ce6764b64fa Mon Sep 17 00:00:00 2001
From: Samo Penic <samo.penic@gmail.com>
Date: Fri, 08 Mar 2019 20:14:36 +0000
Subject: [PATCH] Constant area plugin done -- fix

---
 src/plugins.c |   65 ++++++++++++++++++++++++++++++--
 1 files changed, 60 insertions(+), 5 deletions(-)

diff --git a/src/plugins.c b/src/plugins.c
index 8947499..3146c2a 100644
--- a/src/plugins.c
+++ b/src/plugins.c
@@ -95,14 +95,18 @@
 
 ts_plugin *init_plugin(ts_char *filename){
 	ts_plugin *plugin = (ts_plugin *)calloc(1, sizeof(ts_plugin));
-	plugin->filename=(ts_char *)calloc(strlen(filename),sizeof(ts_char));
+	plugin->filename=(ts_char *)calloc(strlen(filename)+1,sizeof(ts_char));
 	strcpy(plugin->filename,filename);
 	plugin->libhandle = dlopen(plugin->filename, RTLD_NOW);
 	if(!plugin->libhandle){
-		ts_fprintf(stderr,"Error loading plugin: %s\n", filename);
+		ts_fprintf(stderr,"Error loading plugin: %s\n", plugin->filename);
 		fatal("Exiting",235);
 	}
 	ts_plugin_details* (*get_plugin_details)(void) = dlsym(plugin->libhandle, "init");
+	if(get_plugin_details==NULL){
+		ts_fprintf(stderr,"Plugin %s invalid. No init() function.\n", plugin->filename);
+		fatal("Exiting", 236);
+	}
 	plugin->details = get_plugin_details();
 	plugin->function = (ts_plugin_function *)calloc(1,sizeof(ts_plugin_function));
 	
@@ -110,8 +114,11 @@
 	plugin->function->at_start = dlsym(plugin->libhandle, "at_start");
 	plugin->function->after_vesicle_init = dlsym(plugin->libhandle, "after_vesicle_init");
 	plugin->function->vm_hard_constraint = dlsym(plugin->libhandle, "vm_hard_constraint");
-	plugin->function->vm_energy_before = dlsym(plugin->libhandle, "vm_energy_before");
-	plugin->function->vm_energy_after = dlsym(plugin->libhandle, "vm_energy_after");
+	plugin->function->vm_energy_before_prepare = dlsym(plugin->libhandle, "vm_energy_before_prepare");
+	plugin->function->vm_energy_before_execute = dlsym(plugin->libhandle, "vm_energy_before_execute");
+	plugin->function->vm_energy_after_prepare = dlsym(plugin->libhandle, "vm_energy_after_prepare");
+	plugin->function->vm_energy_after_execute = dlsym(plugin->libhandle, "vm_energy_after_execute");
+	plugin->function->vm_before_montecarlo_constraint = dlsym(plugin->libhandle, "vm_before_montecarlo_constraint");
 	plugin->function->vm_new_state_rejected = dlsym(plugin->libhandle, "vm_new_state_rejected");
 	plugin->function->vm_new_state_accepted = dlsym(plugin->libhandle, "vm_new_state_accepted");
 
@@ -119,12 +126,59 @@
 }
 
 
+ts_plugin_chain *plugin_to_chain(ts_plugin_chain *chain, ts_plugin *plugin){
+	ts_plugin_chain *retchain;
+	if(chain!=NULL){
+		retchain=chain;
+		while(chain->next!=NULL){
+			chain=chain->next;
+		}
+		chain->next=(ts_plugin_chain *)calloc(1,sizeof(ts_plugin_chain));
+		chain->next->plugin=plugin;
+		return retchain;
+	} else {
+		chain=(ts_plugin_chain *)calloc(1,sizeof(ts_plugin_chain));
+		chain->plugin=plugin;
+		return chain;
+	}
+}
+
+
 ts_plugin_list *init_plugin_list(ts_char **plugin_filenames, ts_uint number_of_plugins){
 	ts_plugin_list *plist=(ts_plugin_list *)calloc(1, sizeof(ts_plugin_list));
 	plist->plugin=(ts_plugin **)calloc(number_of_plugins, sizeof(ts_plugin *));
+	plist->chain=(ts_plugin_chains *)calloc(1, sizeof(ts_plugin_chains));
 	for(int i=0;i<number_of_plugins;i++){
 		plist->plugin[i]=init_plugin(plugin_filenames[i]);
+		if(plist->plugin[i]->function->cleanup!=NULL){
+			plist->chain->cleanup=plugin_to_chain(plist->chain->cleanup, plist->plugin[i]);
+		}
+		if(plist->plugin[i]->function->at_start!=NULL){
+			plist->chain->at_start=plugin_to_chain(plist->chain->at_start, plist->plugin[i]);
+		}
+		if(plist->plugin[i]->function->after_vesicle_init!=NULL){
+			plist->chain->after_vesicle_init=plugin_to_chain(plist->chain->after_vesicle_init, plist->plugin[i]);
+		}
+		if(plist->plugin[i]->function->vm_hard_constraint!=NULL){
+			plist->chain->vm_hard_constraint=plugin_to_chain(plist->chain->vm_hard_constraint, plist->plugin[i]);
+		}
+		if(plist->plugin[i]->function->vm_energy_before_prepare!=NULL){
+			plist->chain->vm_energy_before_prepare=plugin_to_chain(plist->chain->vm_energy_before_prepare, plist->plugin[i]);
+		}
+		if(plist->plugin[i]->function->vm_energy_after_execute!=NULL){
+			plist->chain->vm_energy_after_execute=plugin_to_chain(plist->chain->vm_energy_after_execute, plist->plugin[i]);
+		}
+		if(plist->plugin[i]->function->vm_before_montecarlo_constraint!=NULL){
+			plist->chain->vm_before_montecarlo_constraint=plugin_to_chain(plist->chain->vm_before_montecarlo_constraint, plist->plugin[i]);
+		}
+		if(plist->plugin[i]->function->vm_new_state_accepted!=NULL){
+			plist->chain->vm_new_state_accepted=plugin_to_chain(plist->chain->vm_new_state_accepted, plist->plugin[i]);
+		}
+		if(plist->plugin[i]->function->vm_new_state_rejected!=NULL){
+			plist->chain->vm_new_state_rejected=plugin_to_chain(plist->chain->vm_new_state_rejected, plist->plugin[i]);
+		}
 	}
+	plist->n=number_of_plugins;
 	return plist;
 }	
 
@@ -137,9 +191,10 @@
 	free(plugin);
 }
 
-void free_plugin_list(ts_plugin_list *plist){
+void plugin_list_free(ts_plugin_list *plist){
 	for(int i=0;i<plist->n;i++){
 		free_plugin(plist->plugin[i]);
 	}
+	free(plist->chain);
 	free(plist);
 }

--
Gitblit v1.9.3