brainopy_example.py

from brainopy import brainopy

b = brainopy("brain.db")
b.logging = True

neurotransmitters = {"Ach": "acetylcholine",
                     "DA": "dopamine", 
                     "GLU": "glutamate",
                     "NE": "norepinephrine",
                     "5HT": "serotonin",
                     "GABA": "gamma-Aminobutyric acid"}
b.addNeurotransmitters(neurotransmitters)

neuronList = b.addNeuron(10)
# Neuron List = [(neuron_ID, dendrite_state_ID, neuron_state_ID, axon_state_ID)]
print("Neuron List: " + str(neuronList))

synapseList = b.addSynapse(10)
# Synapse List = [synapse_state_IDs]
print("Synapse List: " + str(synapseList))

linkages = b.linkRandomAxonSynapse(30)
# Axon-Synapse Linkages = [(axon_state_ID, synapse_state_ID)]
print("Axon-Synapse Linkages: " + str(linkages))

linkages = b.linkRandomSynapseDendrite(30)
# Synapse-Dendrite Linkages = [(synapse_state_ID, dendrite_state_ID)]
print("Synapse-Dendrite Linkages: " + str(linkages))

inputSignal = {"Ach": 0.11,
               "DA": 0.15, 
               "GLU": 0.21,
               "NE": 0.25,
               "5HT": 0.31,
               "GABA": 0.35}
for synapse_state_ID in synapseList:
    b.inputSignal(synapse_state_ID, inputSignal)
    print("Inserted input signal into synapse " + synapse_state_ID)

neuronList = b.getIDs("neuron_body")
# Neuron List = [neuron_ID]
print("Neurons to run brain: " + str(neuronList))

b.nameID(neuronList[0], "neuron1", "neuron number 1")
print("Label " + neuronList[0] + " as neuron1")

print("State ID of " + neuronList[0] + " = " + b.getStateIDFromNeuronID(neuronList[0], "neuron_state_ID"))
original_neuron1 = b.readNeurotransmitters("neuron1")

b.nameID(synapseList[3], "synapse4", "synapse number 4")
print("Label " + synapseList[3] + " as synapse4")

synapse4 = b.readNeurotransmitters("synapse4")
print("Original state of synapse4 = " + str(synapse4))

for cycle in range(1, 11):
    b.runBrain(neuronList)
    print("State of synapse4 after cycle " + str(cycle))
    print("By name: " + str(b.readNeurotransmitters("synapse4")))
    print("By ID: " + str(b.readNeurotransmitters(synapseList[3], "ID")))
    print("")

final_neuron1 = b.readNeurotransmitters("neuron1")
print("Original state of neuron1 (by name) = " + str(original_neuron1))
print("Final state of neuron1 (by name) = " + str(final_neuron1))
print("Final state of neuron1 (by ID) = " + str(b.readNeurotransmitters(neuronList[0], "ID")))

for i in range(len(neuronList)):
    state = b.readNeurotransmitters(neuronList[i], "ID")
    print("Final state of %s = %s" % (neuronList[i], str(state)))

print("Dendrite state IDs of neuron1 (by name) = " + str(b.getStateIDFromNeuronName("neuron1", "dendrite_state_ID")))
print("Dendrite state IDs of neuron1 (by ID) = " + str(b.getStateIDFromNeuronID(neuronList[0], "dendrite_state_ID")))

b.disconnectBrain()