From d939580392b4a2203dab92c7478d1ff523eb7395 Mon Sep 17 00:00:00 2001
From: Mandar Chitre <mandar@nus.edu.sg>
Date: Fri, 21 Jun 2024 00:44:46 +0800
Subject: [PATCH] fix: bug fix for handling gains

---
 src/sim.jl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/sim.jl b/src/sim.jl
index 8d864e1..7ee1efe 100644
--- a/src/sim.jl
+++ b/src/sim.jl
@@ -221,11 +221,11 @@ function transmit(sim::Simulation, node::Node, t, x)
   rxnodes = filter(n -> n != node, sim.nodes)
   rx = mapreduce(n -> [AcousticReceiver((n.pos .+ p)...) for p ∈ n.relpos], vcat, rxnodes)
   arr = [arrivals(sim.model, tx1, rx1) for tx1 ∈ tx, rx1 ∈ rx]
-  sf = 10 ^ ((sim.txref + sim.ogain) / 20)
+  sf = 10 ^ ((sim.txref + node.ogain) / 20)
   y = UnderwaterAcoustics.Recorder(nothing, tx, rx, arr)(sf * x; fs, reltime=false)
   j = 1
-  sf = 10 ^ ((sim.rxref + sim.igain) / 20)
   for node ∈ rxnodes
+    sf = 10 ^ ((sim.rxref + node.igain) / 20)
     for tape ∈ node.tapes
       push!(tape, t, sf * y[:,j])
       j += 1