- fixed substance enable

- re-enable banned substance on successful calculation
- refactored names
This commit is contained in:
2023-06-15 08:30:00 +02:00
parent ab5f754b2a
commit c1c1d60c1e
+14 -9
View File
@@ -22,31 +22,32 @@
## Author: Jens <jens@orion>
## Created: 2023-06-10
function [P1, na_1, D] = vecfit (Pp, ignore_mask, Pt, P0, na_0, step)
function [P1, na_1, D] = vecfit (Pp, element_mask, Pt, P0, na_0, step)
[N_subst, N_elem] = size(Pp);
# Create default ignore_mask
if isempty(ignore_mask),
ignore_mask = ones(1,N_elem);
# Create default element_mask
if isempty(element_mask),
element_mask = ones(1,N_elem);
end
# A-Priory state with mask
P1 = P0 + sum(Pp.*na_0.*ignore_mask);
P1 = P0 + sum(Pp.*na_0.*element_mask);
# Calc distance to target
d = (Pt-P1);
D = norm(d);
dn = d/D;
subst_list = 1:N_subst;
# Enable all substances
subst_mask = ones(1, N_subst);
while 1,
# Find best prototype
best_i = 0;
best_k = 10000;
best_dir = 1;
for i=subst_list,
for i=1:N_subst,
if subst_mask(i) == 1
p = Pp(i,:);
pn = p/norm(p);
@@ -60,6 +61,7 @@ function [P1, na_1, D] = vecfit (Pp, ignore_mask, Pt, P0, na_0, step)
end
end
end
end
# Calc next Pa
na_1 = na_0;
@@ -75,7 +77,7 @@ function [P1, na_1, D] = vecfit (Pp, ignore_mask, Pt, P0, na_0, step)
# Check for negative n
accu = na_1(best_i) + delta;
if accu < 0,
subst_list(best_i) = [];
subst_mask(best_i) = 0;
continue;
end
# Update point
@@ -83,6 +85,9 @@ function [P1, na_1, D] = vecfit (Pp, ignore_mask, Pt, P0, na_0, step)
# Update Point accu
na_1(best_i) = accu;
# Re-enable substance
subst_mask = ones(1, N_subst);
break;
end
end