comparison python/simscene.py @ 51:ebf92ed7d680 tip master

Added -fd (--full-duration) argument.

 
     if 'annot_format' in kwargs:
         annot_format = kwargs['annot_format']
     else:
         annot_format = 'sed_eval'
+
+    if 'full_duration' in kwargs:
+        full_duration = True
+    else:
+        full_duration = False
     
     # Stores the starting and ending times of every track for visualization
     # purposes
     scene_starting_times = []
     scene_ending_times = []

 
         else:
             # If 0, then start next sample after this one (set it to the duration of the sample)
             if mean_time_between_instances == 0:
                 mean_time_between_instances = len(wav)/float(SR)
+
+            # If we are using -fd (full_duration) for each event then mean_time_between_instances denotes time AFTER
+            # the end of the previous event.
+            if full_duration and mean_time_between_instances > 0:
+                mean_time_between_instances += len(wav)/float(SR)
                 
             # Store the successive starting and ending times of the events (given e.g. the model)
             # in the following lists.
             start_times = [events_df['start_time'].loc[n]]
             end_times = [start_times[-1]+len(wav)/float(SR)]

              "`abstract': values are computed from an abstract representation of an existing acoustic scene. "
              "`replicate': values are replicated from an existing acousting scene. (NOT IMPLEMENTED)",
         choices=['generate', 'abstract', 'replicate']
     )
     time_mode = 'generate'
-    
+
+    argparser.add_argument(
+        '-fd', '--full-duration',
+        action='store_true',
+        help="If enabled, times specified in the recipe refer to after the previous file finishes."
+    )
+    full_duration = False
+
     argparser.add_argument(
         '-R', '--ebr-mode',
         type=str,
         help="Mode for Event to Background power level ratio. `generate': values must be set for each track in the "
              "score files. `abstract': values are computed from an abstract representation of an existing acoustic "

         input_path = args.input_path
         logging.debug("Using `{}' as input path".format(input_path))
     if args.output_path:
         output_path = args.output_path
         logging.debug("Saving to `{}'".format(output_path))
+    if args.full_duration:
+        full_duration = True
     if args.scene_duration:
         if not (args.score_backgrounds or args.score_events):
             print("You must provide one of -e or -b")
         else:
             if args.image_format:

                          annotation_file=annotation_file,
                          audio_file=audio_file,
                          figure_verbosity=figure_verbosity,
                          end_cut=end_cut,
                          image_format=image_format,
-                         append_to_filename=append_to_filename)
+                         append_to_filename=append_to_filename,
+                         full_duration=full_duration)
             else:
                 for n in range(generate_n):
                     if tag:
                         append_to_filename = '{}_{}'.format(tag, n)
                     else:

                              annotation_file=annotation_file,
                              audio_file=audio_file,
                              figure_verbosity=min(figure_verbosity, 1),
                              end_cut=end_cut,
                              image_format=image_format,
-                             append_to_filename=append_to_filename)
+                             append_to_filename=append_to_filename,
+                             full_duration=full_duration)