Changeset 391 for vtcross/trunk/src/cognitive_engines/DSA_CE/examples/gnuradio-examples/dsa.py

Timestamp:

07/29/09 16:10:39 (15 years ago)

Author:

trnewman

Message:

Fleshed out DSA reference implementation

Files:

• vtcross/trunk/src/cognitive_engines/DSA_CE/examples/gnuradio-examples/dsa.py (modified) (10 diffs)

vtcross/trunk/src/cognitive_engines/DSA_CE/examples/gnuradio-examples/dsa.py

@@ -39 +39 @@
 def main():
 
-        global n_rcvd, n_right,sync_status,mode,ch,traffic_flag,n_attempts,return_flag
+        global stats_array, count_array, time_array, n_rcvd, n_right,sync_status,mode,ch,traffic_flag,n_attempts,return_flag
         n_rcvd = 0
         n_right = 0
         n_attempts = 5
         return_flag = 0
+
+        count_array = [ 0, 0, 0, 0, 0]
+        time_array = [ 0, 0, 0, 0, 0]
+        stats_array = [ 0, 0, 0, 0, 0]
+
 
         def send_pkt(self, payload='', eof=False):
@@ -63 +68 @@
                 return real_channel
 
+        def get_average_time(channel,absent_time):
+
+                global count_array, time_array
+
+                count_array[channel] = count_array[channel] + 1
+                average_time = (time_array[channel] + absent_time) / count_array[channel]       
+        
+                return average_time
+
         def get_freq(hop_freq,probe_level,absent_time):
 
@@ -76 +90 @@
                         channel = 4
 
-                p = Parameter(1)
                 currentParameters = Parameter(1)
-                o = Observable(1)
-
+                currentParameters[0].name = "channel"
+                currentParameters[0].value = channel
+
+                o = Observable(2)
                 o[0].value = probe_level
                 o[0].name = "energy"
-
-                currentParameters[0].name = "channel"
-                currentParameters[0].value = channel;
-
-                p = GetOptimalParameters(o,1,currentParameters,1);
-                print p[0].value
+                
+                o[1].value = absent_time 
+                o[1].name = "communication_time"
+                
+                # If time == 0 then we are scanning and we dont want to 
+                #  use this time in the averaging process.
+
+                if absent_time != 0:
+                    UpdateParameterPerformance(currentParameters,1,o,1)
+
+                else:
+                    # Get the average communication time
+                    average_time = get_average_time(channel, absent_time)
+                    o[1].value = average_time 
+                    o[1].name = "communication_time"
+                    UpdateParameterPerformance(currentParameters,1,o,2)
+
+                p = Parameter(1)
+                p = GetOptimalParameters(o,2,currentParameters,1);
                 
                 channel = get_real_channel(int(p[0].value))
@@ -96 +124 @@
                         hop_freq = float(1e6 * (467.5625+(channel-8)*0.025))#setting the centre freq frequency for sending packets      
 
+                stats_array[int(p[0].value)] = stats_array[int(p[0].value)] + 1
+                print "Stats:: 1..",stats_array[1]," 2..",stats_array[2]," 3..",stats_array[3]," 4..",stats_array[4] 
                 return channel,hop_freq #returning the channel number and hop frequency
         
@@ -154 +184 @@
 
         parser_tx = OptionParser(option_class=eng_option, conflict_handler="resolve")
-        expert_grp_tx = parser_tx.add_option_group("Expert_tx")
-
+        parser_tx.add_option("", "--vtcross", action="store_true", default=False,
+                                help="Use the CROSS engine for DSA decisions: default is random.")
+  
         parser_tx.add_option("-m", "--modulation", type="choice", choices=mods.keys(),
                         default='gmsk',
@@ -166 +197 @@
                         help="set megabytes to transmit [default=%default]")
         parser_tx.add_option("","--discontinuous", action="store_true", default=False,
-                        help="enable discontinous transmission (bursts of 5 packets)")
+                        help="enable discontinous transmission (bursts of 5 packets)") 
         parser_tx.add_option("","--from-file", default=None,
                         help="use file for packet contents")
-        
+ 
+        expert_grp_tx = parser_tx.add_option_group("Expert_tx")
+
         transmit_path.add_options(parser_tx, expert_grp_tx)
 
         for mod in mods.values():
                 mod.add_options(expert_grp_tx)
+
 
         (options_tx, args_tx) = parser_tx.parse_args ()
@@ -188 +222 @@
         options_tx.fusb_nblocks = 16
         options_tx.bitrate = 0.0125e6
-        #############
+        #############
 
         if options_tx.tx_freq is None:
@@ -195 +229 @@
                 sys.exit(1)
 
-        if options_tx.from_file is not None:
-                source_file = open(options_tx.from_file, 'r')
+        #if options_tx.from_file is not None:
+        #       source_file = open(options_tx.from_file, 'r')
             
         parser_rx = OptionParser (option_class=eng_option, conflict_handler="resolve")
         expert_grp_rx = parser_rx.add_option_group("Expert_rx")
-        parser_rx.add_option("-m", "--modulation", type="choice", choices=demods.keys(),
-                        default='gmsk',
-                        help="Select modulation from: %s [default=%%default]"
-                                % (', '.join(demods.keys()),))
+        #parser_rx.add_option("-m", "--modulation", type="choice", choices=demods.keys(),
+        #               default='gmsk',
+        #               help="Select modulation from: %s [default=%%default]"
+        #                       % (', '.join(demods.keys()),))
         
         receive_path.add_options(parser_rx, expert_grp_rx)
 
-        for mod in demods.values():
-                mod.add_options(expert_grp_rx)
+        #for mod in demods.values():
+        #       mod.add_options(expert_grp_rx)
 
         (options_rx, args_rx) = parser_rx.parse_args ()
 
-        if len(args_rx) != 0:
-                parser_rx.print_help(sys.stderr)
-                sys.exit(1)
+        #if len(args_rx) != 0:
+        #       parser_rx.print_help(sys.stderr)
+        #       sys.exit(1)
         ############# Setting some default values for rx side of the block
         options_rx.rx_freq = 462.5625e6 #setting default rx_freq value
@@ -224 +258 @@
         #############
 
-        if options_rx.rx_freq is None:
-                sys.stderr.write("You must specify -f FREQ or --freq FREQ\n")
-                parser_rx.print_help(sys.stderr)
-                sys.exit(1)
+        #if options_rx.rx_freq is None:
+        #       sys.stderr.write("You must specify -f FREQ or --freq FREQ\n")
+        #       parser_rx.print_help(sys.stderr)
+        #       sys.exit(1)
         
         # build the graph
@@ -248 +282 @@
 
         # Scan all channels first for inital data
+        time.sleep(0.1)
 
         while running: