reefscape/metashape_model_mark_export.py at master · pimbongaerts/reefscape · 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
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
#!/usr/bin/env python
"""
"""
import argparse
import os
import glob
import json
import Metashape

focal_plots_path = "/home/coral3d/focal_plots/"
PLY = "PLY"
NO_PLY = ""

class Focal_plots(object):
    """  """
    def __init__(self, path):
        self.path = path
        self.locations = {}
        for location_folder in os.listdir(self.path):
            location_path = os.path.join(self.path, location_folder)
            if os.path.isdir(location_path):
                self.locations[location_folder] = Location(location_folder,
                                                           location_path)

class Location(object):
    """  """
    def __init__(self, short_name, path):
        self.short_name = short_name
        self.path = path
        self.plots = {}
        #print("├── {}".format(short_name))
        for plot_folder in os.listdir(self.path):
            plot_path = os.path.join(self.path, plot_folder)
            if os.path.isdir(plot_path):
                self.plots[plot_folder] = Plot(plot_folder, plot_path)

class Plot(object):
    """  """
    def __init__(self, short_name, path):
        self.short_name = short_name
        self.path = path
        self.timepoints = {}
        #print("│   ├──{}".format(short_name))
        for timepoint_folder in os.listdir(self.path):
            timepoint_path = os.path.join(self.path, timepoint_folder)
            if os.path.isdir(timepoint_path) and not '@eaDir' in  timepoint_path:
                self.timepoints[timepoint_folder] = Timepoint(timepoint_folder,
                                                              timepoint_path)

class Timepoint(object):
    """  """
    def __init__(self, short_name, path):
        self.short_name = short_name
        project_filepath = '{0}/{1}.psx'.format(path, short_name)


        marker_backup_filepath = '{0}/{1}_markers_backup.csv'.format(path, short_name)
        marker_filepath = '{0}/{1}_markers.csv'.format(path, short_name)

        if os.path.isfile(marker_filepath):
            print('Markers already exported for: {}'.format(short_name))
            next
        elif os.path.isfile(project_filepath):
            print('Detecting markers for: {}'.format(short_name))
            doc = Metashape.Document()
            doc.open(project_filepath, read_only=True)  # Open project as read-only
            doc.read_only = False                       # Turn off read-only in case it applies
            doc = Metashape.Document()
            doc.open(project_filepath, read_only=False) # Open project with writing access
            chunk = doc.chunk
            if chunk:
                # Create backup of current markers and remove from project
                if len(doc.chunk.markers) > 0:
                    self.__export_current_markers(doc, marker_backup_filepath)
                    doc.chunk.remove(doc.chunk.markers)
                # Find markers (circ 12-bit, tol 20, max 5)
                photos_cameras = [
                    cam for cam in chunk.cameras
                    if cam.group and cam.group.label == "photos"
                ]
                try:
                    doc.chunk.detectMarkers(photos_cameras,
                                        target_type=Metashape.TargetType.CircularTarget12bit,
                                        tolerance=20,
                                        filter_mask=False,
                                        inverted=False,
                                        noparity=False,
                                        maximum_residual=5,
                                        minimum_size=0,
                                        minimum_dist=5)
                except Exception as e: print(e)

                if len(doc.chunk.markers) > 0:
                    print('Exporting markers for: {}'.format(short_name))
                    self.__export_current_markers(doc, marker_filepath)
                    doc.save()
        else:
            print('No project found for: {}'.format(short_name))

    def __export_current_markers(self, doc, filepath):
        """ Get viscore project info from subsets.json file """
        marker_file = open(filepath, 'w')
        for marker in doc.chunk.markers:
            if marker.position:
                marker_file.write('{0},{1},{2},{3},{4}\n'.format(marker.key,
                                                                 marker.label,
                                                                 marker.position[0],
                                                                 marker.position[1],
                                                                 marker.position[2]))
        marker_file.close()

def main():
    focal_plots = Focal_plots(focal_plots_path)

if __name__ == '__main__':
    parser = argparse.ArgumentParser(description=__doc__)
    args = parser.parse_args()
    main()