IRI-Correlation-Analysis/calculator.py at master · codemogroup/IRI-Correlation-Analysis · 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
import json
import numpy as np
import math

print(math.log10(1))
exit()

def find_nearest_above(my_array, target):
    diff = my_array - target
    mask = np.ma.less_equal(diff, 0)
    # We need to mask the negative differences and zero
    # since we are looking for values above
    if np.all(mask):
        return None  # returns None if target is greater than any value
    masked_diff = np.ma.masked_array(diff, mask)
    return masked_diff.argmin()


acceTreshold = 6
timeSlot = 100

with open('j7ThalallaRandom2.json') as data_file:
    array1 = json.load(data_file)

newArray = {}
timeArray = []
for item in array1:
    newArray[item["time"]] = item
    timeArray.append(item["time"])
    print(item["acceY_raw"])

startTime = array1[0]["time"]
currentTime = startTime
pulseCount = 0
status = True

newDataArray = []
while (status):
    # print(startTime,startTime + timeSlot)
    currentTime = currentTime + timeSlot
    if currentTime >= array1[-1]["time"]:
        status = False
        continue
    nearestLessValue = None
    nearestGreaterValue = None
    for idx, val in enumerate(array1):
        if currentTime >= val["time"]:
            nearestLessValue = val["time"]
        if nearestGreaterValue == None and currentTime < val["time"]:
            nearestGreaterValue = val["time"]

    calcAcceY = newArray[nearestLessValue]["acceY_raw"] + (currentTime - nearestLessValue) * (
                newArray[nearestGreaterValue]["acceY_raw"] - newArray[nearestLessValue]["acceY_raw"]) / (
                            nearestGreaterValue - nearestLessValue)

    if calcAcceY > acceTreshold:
        pulseCount += 1

print(pulseCount)

def pulseCounter(timeSlot, acceTreshold, dataArray):
    newArray = {}
    timeArray = []
    for item in dataArray:
        newArray[item["time"]] = item
        timeArray.append(item["time"])
        # print(item["acceY_raw"])

    startTime = dataArray[0]["time"]
    currentTime = startTime
    pulseCount = 0
    status = True

    newDataArray = []
    while (status):
        # print(startTime,startTime + timeSlot)
        currentTime = currentTime + timeSlot
        if currentTime >= dataArray[-1]["time"]:
            status = False
            continue
        nearestLessValue = None
        nearestGreaterValue = None
        for idx, val in enumerate(dataArray):
            if currentTime >= val["time"]:
                nearestLessValue = val["time"]
            if nearestGreaterValue == None and currentTime < val["time"]:
                nearestGreaterValue = val["time"]

        calcAcceY = newArray[nearestLessValue]["acceY_raw"] + (currentTime - nearestLessValue) * (
                newArray[nearestGreaterValue]["acceY_raw"] - newArray[nearestLessValue]["acceY_raw"]) / (
                            nearestGreaterValue - nearestLessValue)

        if calcAcceY > acceTreshold:
            pulseCount += 1

    return pulseCount