hex/extract_header_metadata.py at master · source99/hex · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
import sys
import os
import glob
import Header
import Counter
import HD_index
import append_sonar_payouts
import sonar_graphs
import make_video_csv
#program to determine start and stop times and payouts given header file information.
#input header files
#input counter.ipd
#input figure out when start of time is?

meta_fp =  open("metadata.csv",'w')
meta_fp.write("Folder,USMH,DSMH,start time,stop time,start payout,stop payout,total payout,expected payout\n")


#	def get_time(self,percent):


#directory = sys.argv[1]
directory = "D:/Dropbox (Centosette)_Hydromax/PJ/PuppytheaterUpload/1539-Chattanooga,TN/Source/S138F004 S138J003_001"
top_directory =  "D:/Dropbox (Centosette)_Hydromax/PJ/PuppytheaterUpload/1539-Chattanooga,TN/Source/"
directories = glob.glob("{}/S13*".format(top_directory))
run_prefix = "E:/hmax/runs"
print directories
for directory in directories:
    folder = directory.split(os.sep)[-1]
    print "directory = {}".format(directory)
    print "folder = {}".format(folder)
    run_directory = "{}/localdata_{}".format(run_prefix,folder)
    sonar_report_with_payouts = "{}/sonar_processing/report_with_payouts.csv".format(run_directory)
    sonar_report = "{}/sonar_processing/report_data.csv".format(run_directory)
    append_sonar_payouts.append_payout(sonar_report, sonar_report_with_payouts,directory)

    counter = Counter.Counter(directory)
    hd_index = HD_index.HD_index(directory)
    print "lines in counter file = {}".format(counter.length)

    print "USMH,DSMH,start time,stop time,start payout,stop payout"
    header_files = glob.glob("{}/Header_File_*.txt".format(directory))
    for header_file in header_files:
        curr_header = Header.Header(header_file)
        sonar_graphs.sonar_graph(sonar_report_with_payouts,curr_header.radius,
                counter.get_closest_payout(hd_index.convert_scroll_to_time(curr_header.scrollStart)),
                counter.get_closest_payout(hd_index.convert_scroll_to_time(curr_header.scrollStop)),
                curr_header.USMH,
                curr_header.DSMH,
                run_directory
                )
#        print "{},{},{},{},{},{},{},{},{}".format(
#            folder,
#            curr_header.USMH,
#            curr_header.DSMH,
#            hd_index.convert_scroll_to_time(curr_header.scrollStart),
#            hd_index.convert_scroll_to_time(curr_header.scrollStop),
#            counter.get_closest_payout(hd_index.convert_scroll_to_time(curr_header.scrollStart)),
#            counter.get_closest_payout(hd_index.convert_scroll_to_time(curr_header.scrollStop)),
#            counter.get_closest_payout(hd_index.convert_scroll_to_time(curr_header.scrollStop)) - counter.get_closest_payout(hd_index.convert_scroll_to_time(curr_header.scrollStart)),
#            curr_header.endDistance
#            )
        meta_fp.write("{},{},{},{},{},{},{},{},{}\n".format(
            folder,
            curr_header.USMH,
            curr_header.DSMH,
            hd_index.convert_scroll_to_time(curr_header.scrollStart),
            hd_index.convert_scroll_to_time(curr_header.scrollStop),
            counter.get_closest_payout(hd_index.convert_scroll_to_time(curr_header.scrollStart)),
            counter.get_closest_payout(hd_index.convert_scroll_to_time(curr_header.scrollStop)),
            counter.get_closest_payout(hd_index.convert_scroll_to_time(curr_header.scrollStop)) - counter.get_closest_payout(hd_index.convert_scroll_to_time(curr_header.scrollStart)),
            curr_header.endDistance
            ))
        make_video_csv.create_csv(directory, run_directory, header_file)

#print "closest to {} is {}".format(606688,counter.get_closest_payout(606688))


#def create_csv(raw_directory, run_directory, header_file):
#    localdata_short = os.path.split(run_directory)
#
#    hd_index = HD_index.HD_index(source_directory)
#    sonar_counter = Counter.Counter(source_directory)
#    sonar_data = Sonar_data.Sonar_data(run_directory)
#    curr_header = Header.Header(header_file)