2424#pragma once
2525
2626#include " AllEdges.h"
27- #include " AllSpikingNeurons.h"
28- #include " AllSpikingSynapses.h"
2927#include " AllVertices.h"
3028#include " EdgeIndexMap.h"
3129#include " Layout.h"
@@ -69,7 +67,7 @@ class Connections {
6967
7068 // / Update the connections status in every epoch.
7169 // /
72- // / @param neurons The Neuron list to search from.
70+ // / @param vertices The vertex list to search from.
7371 // / @return true if successful, false otherwise.
7472 virtual bool updateConnections (AllVertices &vertices);
7573
@@ -78,35 +76,36 @@ class Connections {
7876
7977#if defined(USE_GPU)
8078public:
81- // / Update the weight of the Synapses in the simulation.
79+ // / Update the weight of the edges in the simulation.
8280 // / Note: Platform Dependent.
8381 // /
8482 // / @param numVertices number of vertices to update.
85- // / @param neurons the Neuron list to search from.
86- // / @param synapses the Synapse list to search from.
83+ // / @param vertices the vertex list to search from.
84+ // / @param edges the edge list to search from.
8785 // / @param allVerticesDevice GPU address of the allVertices struct on device memory.
88- // / @param allEdgesDevice GPU address of the allEdges struct on device memory.
89- // / @param layout Layout information of the neural network.
90- virtual void updateSynapsesWeights (int numVertices, AllVertices &vertices, AllEdges &synapses,
91- AllSpikingNeuronsDeviceProperties *allVerticesDevice,
92- AllSpikingSynapsesDeviceProperties *allEdgesDevice,
93- Layout &layout);
86+ // / @param allEdgesDevice GPU address of the allEdges struct on device memory.
87+ // / @param layout Layout information of the graph network.
88+ virtual void updateEdgesWeights (int numVertices, AllVertices &vertices, AllEdges &edges,
89+ AllVerticesDeviceProperties *allVerticesDevice,
90+ AllEdgesDeviceProperties *allEdgesDevice, Layout &layout);
9491#else
9592public:
96- // / Update the weight of the Synapses in the simulation.
93+ // / Update the weight of the edges in the simulation.
9794 // / Note: Platform Dependent.
98- virtual void updateSynapsesWeights ();
95+ virtual void updateEdgesWeights ();
9996
10097#endif // USE_GPU
10198
10299protected:
103100 unique_ptr<AllEdges> edges_;
101+ // / TODO: Rename to edgeIndexMap_ since this is a base class
104102 unique_ptr<EdgeIndexMap> synapseIndexMap_;
105103
106104 log4cplus::Logger fileLogger_;
107105 log4cplus::Logger edgeLogger_;
108106};
109107
108+ // / TODO: Rename to synapseIndexMap since this is a base class
110109// / Cereal serialization method
111110template <class Archive > void Connections::serialize (Archive &archive)
112111{
0 commit comments