batch_basic_level.rb

require 'Datavyu_API'

# $input_dir = "~/code/work/seedlings/pho_reliability_opf/pass2/recode"
# $output_dir = "~/code/work/seedlings/pho_reliability_opf/pass2/orig_and_recode_csv"

# $input_dir = "~/code/work/seedlings/batch_opf_bl/17_chi_merged"
# $output_dir = "~/code/work/seedlings/batch_opf_bl/output"

# $input_dir = "~/code/work/seedlings/reliability_data/by_month/07/orig_rel_10"
# $output_dir = "~/code/work/seedlings/reliability_data/by_month/07/csv_out"

# $input_dir = "~/code/work/seedlings/datavyu_scripts/data/chi_checks2_3"
# $output_dir = "~/code/work/seedlings/datavyu_scripts/data/batch_bl_out"

# $input_dir = "~/code/work/seedlings/datavyu_scripts/data/reliability_checks/07/orig_rel_10"
# $output_dir = "~/code/work/seedlings/datavyu_scripts/data/reliability_checks/07/csv_out"

# $input_dir = "~/code/work/seedlings/datavyu_scripts/data/chi_checks/13/reliability_checks"
# $output_dir = "~/code/work/seedlings/datavyu_scripts/data/chi_checks/13/csv_out"

# $input_dir = "~/code/work/seedlings/datavyu_scripts/data/chimerge/final_out_merged"
# $output_dir = "~/code/work/seedlings/datavyu_scripts/data/chimerge/final_basic_level"

$input_dir = "/Volumes/pn-opus/Seedlings/Working_Files/annot_id/video/recode_and_orig_opfs/"
$output_dir = "/Volumes/pn-opus/Seedlings/Working_Files/annot_id/video/basic_level1_output/"


def basic_level(in_dir, file)
  puts(file)
  $db, $pj = load_db(File.join(in_dir, file))
  columns = get_column_list()
  for column in columns
    bl_out_path = File.join(File.expand_path($output_dir), file.sub(".opf", "_processed.csv"))
    col = getColumn(column)
    if col.cells.length == 0
      next
    end


    CSV.open(bl_out_path, "wb") do |csv|
      csv << ["labeled_object.ordinal","labeled_object.onset",
              "labeled_object.offset","labeled_object.object",
              "labeled_object.utterance_type","labeled_object.object_present",
              "labeled_object.speaker","labeled_object.id",
              "labeled_object.basic_level"]
      for cell in col.cells
        # puts cell.ordinal.to_s
        # puts cell.object.to_s
        # puts "\n\n"
        if !cell.object.to_s.start_with?("%pho")
          csv << [cell.ordinal.to_s, cell.onset.to_s, cell.offset.to_s, cell.object.to_s,
                  cell.utterance_type.to_s, cell.object_present.to_s, cell.speaker.to_s, cell.id.to_s, ""]
        end
      end
    end

  end
end




begin
  in_dir = File.expand_path($input_dir)
  filenames = Dir.new(in_dir).entries

  for file in filenames
    if file.end_with? ".opf"
      basic_level(in_dir, file)
    end
  end
end