Merge branch 'Stress'

.gitignore

@@ -0,0 +1 @@
+.idea/

src/file_data.py

@@ -0,0 +1,80 @@
+#############################################################
+#             EECS 442: Computer Vision - W19               #
+#############################################################
+# Authors: Sravan Balaji & Kevin Monpara                    #
+# Filename: file_data.py                                    #
+# Description:                                              #
+#   Read data file to get specimen dimensions.              #
+#   Read in load and displacement data for each frame.      #
+#   Calculate stress from load and cross-sectional area.    #
+#############################################################
+
+
+class FrameData:
+    load = 0    # Load in N
+    disp = 0    # Displacement in mm
+    stress = 0  # Stress in MPa
+
+    def __init__(self, load, disp, stress):
+        self.load = load
+        self.disp = disp
+        self.stress = stress
+
+
+class SpecimenDimensions:
+    w = 0   # Specimen Width in mm
+    t = 0   # Specimen Thickness in mm
+    gl = 0  # Specimen Gauge Length in mm
+    ol = 0  # Specimen Overall Length in mm
+
+
+def read_file(filepath):
+    file = open(filepath, "r")
+
+    specimen = SpecimenDimensions()
+    load_disp_data = dict()
+
+    data_start = -1
+    in_frame_data = False
+
+    for line in file:
+        if "Width" in line:
+            index = line.find("\t")
+            specimen.w = float(line[index + 1:])
+
+        if "Thickness" in line:
+            index = line.find("\t")
+            specimen.t = float(line[index + 1:])
+
+        if "Gauge Length" in line:
+            index = line.find("\t")
+            specimen.gl = float(line[index + 1:])
+
+        if "Overall Length" in line:
+            index = line.find("\t")
+            specimen.ol = float(line[index + 1:])
+
+        if "Frame #" in line:
+            data_start = 2
+
+        if data_start != -1:
+            data_start -= 1
+
+        if data_start == 0:
+            in_frame_data = True
+
+        if in_frame_data:
+            if line != '\t\t\n':
+                frame_num_str, load_str, disp_str = line.split("\t")
+
+                frame_num = int(frame_num_str)
+
+                load = float(load_str)
+                disp = float(disp_str)
+                stress = load / (specimen.w * specimen.t)
+
+                load_disp_data[frame_num] = FrameData(load, disp, stress)
+
+    file.close()
+
+    return specimen, load_disp_data

src/main.py

@@ -0,0 +1,20 @@
+#############################################################
+#             EECS 442: Computer Vision - W19               #
+#############################################################
+# Authors: Sravan Balaji & Kevin Monpara                    #
+# Filename: main.py                                         #
+# Description:                                              #
+#                                                           #
+#############################################################
+
+import file_data
+
+
+def main():
+    specimen, load_disp_data = file_data.read_file("../Section001_Data.txt")
+
+    print("Done")
+
+
+if __name__ == "__main__":
+    main()