forked from pmla/polyhedral-template-matching
-
Notifications
You must be signed in to change notification settings - Fork 0
/
alloy_types.cpp
112 lines (96 loc) · 1.69 KB
/
alloy_types.cpp
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
#include "index_ptm.h"
int32_t find_fcc_alloy_type(int8_t* mapping, int32_t* numbers)
{
int len = 13;
int8_t temp[13];
int num_cu = 1, cu = numbers[0];
for (int i=1;i<len;i++)
if (numbers[i] == cu)
num_cu++;
if (num_cu == len)
return PTM_ALLOY_PURE;
int num_au = 0, au = -1;
for (int i=1;i<len;i++)
{
if (numbers[i] != cu)
{
if (au == -1)
{
au = numbers[i];
num_au = 1;
}
else if (numbers[i] == au)
{
num_au++;
}
else
{
return PTM_ALLOY_NONE;
}
}
}
if (num_au == len - 1)
return PTM_ALLOY_L12_AU;
for (int i=0;i<len-1;i++)
temp[i] = numbers[mapping[i + 1]];
if (num_au == 4)
{
for (int j = 0;j<3;j++)
{
int n = 0;
for (int i=j*4;i<(j+1)*4;i++)
if (temp[i] == au)
n++;
if (n == 4)
return PTM_ALLOY_L12_CU;
}
}
else if (num_au == 8)
{
for (int j = 0;j<3;j++)
{
int n = 0;
for (int i=j*4;i<(j+1)*4;i++)
if (temp[i] == cu)
n++;
if (n == 4)
return PTM_ALLOY_L10;
}
}
return PTM_ALLOY_NONE;
}
int32_t find_bcc_alloy_type(int8_t* mapping, int32_t* numbers)
{
int len = 15;
int8_t temp[15];
int num_cu = 1, cu = numbers[0];
for (int i=1;i<len;i++)
if (numbers[i] == cu)
num_cu++;
if (num_cu == len)
return PTM_ALLOY_PURE;
for (int i=0;i<len;i++)
temp[i] = numbers[mapping[i + 1]];
int num_au_inner = 0, num_cu_outer = 0, au = -1;
for (int i=0;i<8;i++)
{
if (temp[i] != cu)
{
if (au == -1)
{
au = temp[i];
num_au_inner = 1;
}
else if (temp[i] == au)
{
num_au_inner++;
}
}
}
for (int i=8;i<14;i++)
if (temp[i] == cu)
num_cu_outer++;
if (num_au_inner == 8 && num_cu_outer == 6)
return PTM_ALLOY_B2;
return PTM_ALLOY_NONE;
}