JRT-SI/light.py at master · JRT-FOREVER/JRT-SI · 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
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
#!/usr/bin/python3
# -*- coding: utf-8 -*-
# Author YongXin SHI
# Email shiyongxin@aliyun.com
# Date 2016/11/07

# Last_update_time_at 2017/5/8


import numpy as np

##(x,y)
#input_date=[(3,90),(4,68),(5,38),(6,18),(7,8),(8,3),(9,5),(10,48)]
#lib_date=[(1,5),(2,48),(3,98),(4,68),(5,38),(6,18),(7,8),(8,2)]


################################################
import sourcedata

print("请输入图像编号，目前有 1，3，5，7，9：")
lib_num = int(input())
print(type(lib_num))

#daa = data()
'''
# sourcedata.source  行和列交换
input_date = sourcedata.source(data())
lib_date = sourcedata.source(data(3))
print(input_date)
'''

import chart

print(sourcedata.data(lib_num))

input_date = sourcedata.wave_filtering(sourcedata.data(lib_num))
lib_date = sourcedata.wave_filtering(sourcedata.data())

#print(input_date)

# 滤波后的图像

import pylab as pl
pl.plot(input_date[0], input_date[1])
pl.plot(lib_date[0], lib_date[1])

# 显示图像
pl.show()
pl.close()

input_date = sourcedata.source(input_date)
lib_date = sourcedata.source(lib_date)

#################################

##二位数组降到一维
def reduction(inputdate):
    n=0
    in_date=[]
    while n < len(inputdate):
        in_date.append(inputdate[n][1])
        n = n + 1
    return in_date

def multi():
    pass

def cutheader(date):
    return date[1:len(date)]

def cutfooter(date):
    return date[0:len(date)-1]

##
##去掉无用数据
##select == 0 return input_date
##select == 1 return lib_date
def draw(input_date,lib_date,select):
    reduction(input_date)
    reduction(lib_date)

    while input_date[0][0] > lib_date[0][0]:
        lib_date = cutheader(lib_date)
    while input_date[0][0] < lib_date[0][0]:
        input_date = cutheader(input_date)
    while input_date[len(input_date)-1][0] > lib_date[len(lib_date)-1][0]:
        input_date = cutfooter(input_date)
    while input_date[len(input_date)-1][0] < lib_date[len(lib_date)-1][0]:
        lib_date = cutfooter(lib_date)
    if select == 0:
        return input_date
    elif select == 1:
        return lib_date
    else:
        return "seletc error select == 0 or 1"

##计算波形误差
##均方误差（MSE）
def similarity(input_date,lib_date):
    n=0
    sumsum=0
##    dvalue=[]
##    ddvalue=[]
##    dddvalue=[]
##    sum=0
    while n < len(lib_date):
        sumsum = sumsum + (input_date[n][1]-lib_date[n][1])*(input_date[n][1]-lib_date[n][1])
##
##
####        print((input_date[n][1]-lib_date[n][1])*(input_date[n][1]-lib_date[n][1]))
##        dvalue.append((input_date[n][1]-lib_date[n][1])^2)
####        print(dvalue)
##
##        ddvalue.append(lib_date[n][1]-dvalue[n])
####        print(ddvalue)
##
##        dddvalue.append(ddvalue[n]/lib_date[n][1])
####        print(ddvalue[n]/lib_date[n][1])
##        sum=dddvalue[n]+sum
        n = n + 1
##    print(sum/len(lib_date))
##    return sum/len(lib_date)
    return sumsum/len(lib_date)


##
##判断波形是否相等
##如果相等 return 0
##如果不想等 return 1

def contrast(input_date,lib_date):
    print(input_date)
    print(lib_date)
##    print(input_date.__len__())
##    for num in input_date[*][1]:
##        print(num)
##    print(input_date-lib_date)


    return 0


##contrast(input_date,lib_date)
##draw(input_date,lib_date,1)
#print(draw(input_date,lib_date,0))
#print(draw(input_date,lib_date,1))
value = similarity(draw(input_date,lib_date,0),draw(input_date,lib_date,1))
print(value)
if value < 50000:
    print("C4H10")
else :
    print("Unknown material")
#print(similarity(draw(input_date,lib_date,0),draw(input_date,lib_date,1)))