mat44_type get_matrix() const
{
if( _matrix_dirty ) compute_matrix();
_matrix_dirty = false;
return _matrix;
}
void computeChromCorrectionMatrix(RUN *run, LINE_LIST *beamline, CHROM_CORRECTION *chrom)
{
VMATRIX *M;
double chromx, chromy;
double chromx0, chromy0;
double K2=0.0, *K2ptr;
ELEMENT_LIST *context;
long i, count, K2_param=0;
MATRIX *C, *Ct, *CtC, *inv_CtC;
m_alloc(&C, 2, chrom->n_families);
m_alloc(&Ct, chrom->n_families, 2);
m_alloc(&CtC, chrom->n_families, chrom->n_families);
m_alloc(&inv_CtC, chrom->n_families, chrom->n_families);
if (chrom->T)
m_free(&(chrom->T));
if (chrom->dK2)
m_free(&(chrom->dK2));
if (chrom->dchrom)
m_free(&(chrom->dchrom));
m_alloc(&(chrom->T), chrom->n_families, 2);
m_alloc(&(chrom->dK2), chrom->n_families, 1);
m_alloc(&(chrom->dchrom), 2, 1);
if (verbosityLevel>2) {
fprintf(stdout, "Computing chromaticity influence matrix for all named sextupoles.\n");
fflush(stdout);
}
computeChromaticities(&chromx0, &chromy0,
NULL, NULL, NULL, NULL,
beamline->twiss0, beamline->elast->twiss, M=beamline->matrix);
M = NULL;
for (i=0; i<chrom->n_families; i++) {
count = 0;
context = NULL;
while ((context=find_element(chrom->name[i], &context, beamline->elem_twiss))) {
if (!count && !(K2_param=confirm_parameter("K2", context->type))) {
fprintf(stdout, "error: element %s does not have K2 parameter\n",
context->name);
fflush(stdout);
exitElegant(1);
}
if (!(K2ptr = (double*)(context->p_elem + entity_description[context->type].parameter[K2_param].offset)))
bombElegant("K2ptr NULL in setup_chromaticity_correction", NULL);
if (count==0)
K2 = *K2ptr;
*K2ptr = K2 + chrom->sextupole_tweek;
if (context->matrix) {
free_matrices(context->matrix);
free(context->matrix);
context->matrix = NULL;
}
compute_matrix(context, run, NULL);
count++;
}
if (count==0) {
fprintf(stdout, "error: element %s is not in the beamline.\n", chrom->name[i]);
fflush(stdout);
exitElegant(1);
}
if (beamline->links) {
/* rebaseline_element_links(beamline->links, run, beamline); */
assert_element_links(beamline->links, run, beamline, STATIC_LINK+DYNAMIC_LINK);
}
if (M) {
free_matrices(M);
free(M);
M = NULL;
}
M = full_matrix(beamline->elem_twiss, run, 2);
computeChromaticities(&chromx, &chromy,
NULL, NULL, NULL, NULL, beamline->twiss0, beamline->elast->twiss, M);
C->a[0][i] = (chromx-chromx0)/chrom->sextupole_tweek;
C->a[1][i] = (chromy-chromy0)/chrom->sextupole_tweek;
if (C->a[0][i]==0 || C->a[1][i]==0) {
fprintf(stdout, "error: element %s does not change the chromaticity!\n", chrom->name[i]);
fflush(stdout);
exitElegant(1);
}
count = 0;
context = NULL;
while ((context=find_element(chrom->name[i], &context, beamline->elem_twiss))) {
if (!count && !(K2_param=confirm_parameter("K2", context->type))) {
fprintf(stdout, "error: element %s does not have K2 parameter\n",
context->name);
fflush(stdout);
exitElegant(1);
}
if (!(K2ptr = (double*)(context->p_elem + entity_description[context->type].parameter[K2_param].offset)))
bombElegant("K2ptr NULL in setup_chromaticity_correction", NULL);
if (!K2ptr)
bombElegant("K2ptr NULL in setup_chromaticity_correction", NULL);
*K2ptr = K2;
if (context->matrix) {
free_matrices(context->matrix);
free(context->matrix);
context->matrix = NULL;
}
compute_matrix(context, run, NULL);
count++;
}
}
if (M) {
free_matrices(M);
free(M);
M = NULL;
}
if (beamline->matrix) {
free_matrices(beamline->matrix);
free(beamline->matrix);
beamline->matrix = NULL;
}
beamline->matrix = full_matrix(beamline->elem_twiss, run, run->default_order);
if (verbosityLevel>1) {
fprintf(stdout, "\nfamily dCHROMx/dK2 dCHROMy/dK2\n");
fflush(stdout);
for (i=0; i<chrom->n_families; i++)
fprintf(stdout, "%10s: %14.7e %14.7e\n", chrom->name[i], C->a[0][i], C->a[1][i]);
fflush(stdout);
}
m_trans(Ct, C);
m_mult(CtC, Ct, C);
m_invert(inv_CtC, CtC);
m_mult(chrom->T, inv_CtC, Ct);
if (verbosityLevel>1) {
fprintf(stdout, "\nfamily dK2/dCHROMx dK2/dCHROMy\n");
fflush(stdout);
for (i=0; i<chrom->n_families; i++)
fprintf(stdout, "%10s: %14.7e %14.7e\n", chrom->name[i], chrom->T->a[i][0], chrom->T->a[i][1]);
fflush(stdout);
fprintf(stdout, "\n");
fflush(stdout);
}
m_free(&C);
m_free(&Ct);
m_free(&CtC);
m_free(&inv_CtC);
}
long do_chromaticity_correction(CHROM_CORRECTION *chrom, RUN *run, LINE_LIST *beamline,
double *clorb, long step, long last_iteration)
{
VMATRIX *M;
double chromx0, chromy0, dchromx=0, dchromy=0;
double K2=0.0, *K2ptr;
ELEMENT_LIST *context;
long i, K2_param=0, type=0, iter, count;
double beta_x, alpha_x, eta_x, etap_x;
double beta_y, alpha_y, eta_y, etap_y;
double K2_min, K2_max;
unsigned long unstable;
double lastError, presentError;
char buffer[256];
log_entry("do_chromaticity_correction");
if (!beamline->matrix || !beamline->twiss0) {
if (!beamline->twiss0)
beamline->twiss0 = tmalloc(sizeof(*beamline->twiss0));
if (!beamline->elem_twiss) {
ELEMENT_LIST *eptr;
eptr = beamline->elem_twiss = &(beamline->elem);
while (eptr) {
if (eptr->type==T_RECIRC)
beamline->elem_twiss = beamline->elem_recirc = eptr;
eptr = eptr->succ;
}
}
if (beamline->matrix) {
free_matrices(beamline->matrix);
free(beamline->matrix);
beamline->matrix = NULL;
}
beamline->matrix = compute_periodic_twiss(&beta_x, &alpha_x, &eta_x, &etap_x, beamline->tune,
&beta_y, &alpha_y, &eta_y, &etap_y, beamline->tune+1,
beamline->elem_twiss, clorb, run, &unstable, NULL, NULL);
beamline->twiss0->betax = beta_x;
beamline->twiss0->alphax = alpha_x;
beamline->twiss0->phix = 0;
beamline->twiss0->etax = eta_x;
beamline->twiss0->etapx = etap_x;
beamline->twiss0->betay = beta_y;
beamline->twiss0->alphay = alpha_y;
beamline->twiss0->phiy = 0;
beamline->twiss0->etay = eta_y;
beamline->twiss0->etapy = etap_y;
propagate_twiss_parameters(beamline->twiss0, beamline->tune, beamline->waists,
NULL, beamline->elem_twiss, run, clorb,
beamline->couplingFactor);
}
else if (beamline->matrix->order<2) {
if (beamline->matrix) {
free_matrices(beamline->matrix);
free(beamline->matrix);
beamline->matrix = NULL;
}
beamline->matrix = full_matrix(beamline->elem_twiss, run, 2);
}
if (!(M = beamline->matrix) || !M->C || !M->R || !M->T)
bombElegant("something wrong with transfer map for beamline (do_chromaticity_correction.1)", NULL);
computeChromaticities(&chromx0, &chromy0,
NULL, NULL, NULL, NULL, beamline->twiss0,
beamline->elast->twiss, M);
if (verbosityLevel>0) {
fprintf(stdout, "\nAdjusting chromaticities:\n");
fflush(stdout);
fprintf(stdout, "initial chromaticities: %e %e\n", chromx0, chromy0);
fflush(stdout);
}
presentError = DBL_MAX;
for (iter=0; iter<chrom->n_iterations; iter++) {
K2_max = -(K2_min = DBL_MAX);
dchromx = chrom->chromx - chromx0;
dchromy = chrom->chromy - chromy0;
if (chrom->tolerance>0 &&
chrom->tolerance>fabs(dchromx) &&
chrom->tolerance>fabs(dchromy) )
break;
lastError = presentError;
presentError = sqr(dchromx)+sqr(dchromy);
if (iter && presentError>lastError) {
fprintf(stdout, "Error increasing---reducing gain\n");
fflush(stdout);
chrom->correction_fraction /= 10;
}
if (chrom->use_perturbed_matrix)
computeChromCorrectionMatrix(run, beamline, chrom);
chrom->dchrom->a[0][0] = dchromx;
chrom->dchrom->a[1][0] = dchromy;
m_mult(chrom->dK2, chrom->T, chrom->dchrom);
for (i=0; i<chrom->n_families; i++) {
if (isnan(chrom->correction_fraction*chrom->dK2->a[i][0]) ||
isinf(chrom->correction_fraction*chrom->dK2->a[i][0]))
break;
}
if (i!=chrom->n_families) {
fprintf(stdout, "Unable to correct chromaticity---diverged\n");
fflush(stdout);
return 0;
}
for (i=0; i<chrom->n_families; i++) {
context = NULL;
count = 0;
while ((context=find_element(chrom->name[i], &context, beamline->elem_twiss))) {
if (count==0 && (K2_param = confirm_parameter("K2", context->type))<0) {
fprintf(stdout, "error: element %s doesn't have K2 parameter\n",
context->name);
fflush(stdout);
exitElegant(1);
}
if (!(K2ptr = (double*)(context->p_elem + entity_description[context->type].parameter[K2_param].offset)))
bombElegant("K2ptr NULL in setup_chromaticity_correction", NULL);
K2 = (*K2ptr += chrom->correction_fraction*chrom->dK2->a[i][0]);
if (chrom->strengthLimit>0 && chrom->strengthLimit<fabs(K2)) {
K2 = *K2ptr = SIGN(K2)*chrom->strengthLimit;
}
sprintf(buffer, "%s#%ld.K2", context->name, context->occurence);
rpn_store(K2, NULL, rpn_create_mem(buffer, 0));
if (K2>K2_max)
K2_max = K2;
if (K2<K2_min)
K2_min = K2;
if (context->matrix) {
free_matrices(context->matrix);
free(context->matrix);
context->matrix = NULL;
}
compute_matrix(context, run, NULL);
type = context->type;
count++;
/* fprintf(stdout, "new value of %s#%ld[K2] is %.15lg 1/m^3\n",
chrom->name[i], context->occurence, K2);
fflush(stdout);
*/
}
if (verbosityLevel>1) {
fprintf(stdout, "Change for family %ld (%ld sextupoles): %e\n",
i, count, chrom->correction_fraction*chrom->dK2->a[i][0]);
fflush(stdout);
}
if (alter_defined_values)
change_defined_parameter(chrom->name[i], K2_param, type, K2, NULL, LOAD_FLAG_ABSOLUTE);
}
if (beamline->links) {
/* rebaseline_element_links(beamline->links, run, beamline); */
assert_element_links(beamline->links, run, beamline,
STATIC_LINK+DYNAMIC_LINK+(alter_defined_values?LINK_ELEMENT_DEFINITION:0));
}
if (beamline->matrix) {
free_matrices(beamline->matrix);
free(beamline->matrix);
beamline->matrix = NULL;
}
M = beamline->matrix = compute_periodic_twiss(&beta_x, &alpha_x, &eta_x, &etap_x, beamline->tune,
&beta_y, &alpha_y, &eta_y, &etap_y, beamline->tune+1,
beamline->elem_twiss, clorb, run, &unstable, NULL, NULL);
beamline->twiss0->betax = beta_x;
beamline->twiss0->alphax = alpha_x;
beamline->twiss0->phix = 0;
beamline->twiss0->etax = eta_x;
beamline->twiss0->etapx = etap_x;
beamline->twiss0->betay = beta_y;
beamline->twiss0->alphay = alpha_y;
beamline->twiss0->phiy = 0;
beamline->twiss0->etay = eta_y;
beamline->twiss0->etapy = etap_y;
if (!M || !M->C || !M->R || !M->T)
bombElegant("something wrong with transfer map for beamline (do_chromaticity_correction.2)", NULL);
computeChromaticities(&chromx0, &chromy0,
NULL, NULL, NULL, NULL, beamline->twiss0,
beamline->elast->twiss, M);
beamline->chromaticity[0] = chromx0;
beamline->chromaticity[1] = chromy0;
if (verbosityLevel>0) {
fprintf(stdout, "resulting chromaticities: %e %e\n", chromx0, chromy0);
fprintf(stdout, "min, max sextupole strength: %e %e 1/m^2\n", K2_min, K2_max);
fflush(stdout);
}
}
if (fp_sl && last_iteration) {
for (i=0; i<chrom->n_families; i++) {
context = NULL;
while ((context=find_element(chrom->name[i], &context, beamline->elem_twiss))) {
if (( K2_param = confirm_parameter("K2", context->type))<0)
bombElegant("confirm_parameter doesn't return offset for K2 parameter.\n", NULL);
fprintf(fp_sl, "%ld %e %s\n",
step,
*((double*)(context->p_elem + entity_description[context->type].parameter[K2_param].offset)),
chrom->name[i]);
}
}
fflush(fp_sl);
}
propagate_twiss_parameters(beamline->twiss0, beamline->tune,
beamline->waists, NULL, beamline->elem_twiss, run, clorb,
beamline->couplingFactor);
log_exit("do_chromaticity_correction");
if (chrom->tolerance>0 &&
(chrom->tolerance<fabs(dchromx) || chrom->tolerance<fabs(dchromy))
&& chrom->exit_on_failure) {
fprintf(stdout, "Chromaticity correction failure---exiting!\n");
exitElegant(1);
}
return 1;
}
long assert_element_links(ELEMENT_LINKS *links, RUN *run, LINE_LIST *beamline, long flags)
{
long i_link, i_elem, i_item, matrices_changed;
long elem_type, data_type, param;
double value;
ELEMENT_LIST **targ, **sour;
char *p_elem;
short lengthChanged = 0;
log_entry("assert_element_links");
if (!links || links->n_links==0) {
log_exit("assert_element_links");
return(0);
}
if (!links->target_name || !links->item || !links->source_name || !links->equation ||
!links->n_targets || !links->target_elem || !links->source_elem) {
fputs("error: link structure has null pointers (assert_element_links)", stdout);
abort();
}
matrices_changed = 0;
for (i_link=0; i_link<links->n_links; i_link++) {
if (!(flags&links->flags[i_link]))
continue;
targ = links->target_elem[i_link];
sour = links->source_elem[i_link];
elem_type = targ[0]->type;
param = links->target_param[i_link];
data_type = entity_description[elem_type].parameter[param].type;
#if DEBUG
fprintf(stdout, "asserting %ld links of %s.%s to %s\n", links->n_targets[i_link],
links->target_name[i_link], links->item[i_link], links->source_name[i_link]);
fflush(stdout);
fprintf(stdout, "source type is %ld, with %ld parameters\n", sour[0]->type,
entity_description[sour[0]->type].n_params);
fflush(stdout);
#endif
for (i_elem=0; i_elem<links->n_targets[i_link]; i_elem++) {
#if DEBUG
fprintf(stdout, " working on element %ld\n", i_elem);
fflush(stdout);
#endif
p_elem = sour[i_elem]->p_elem;
for (i_item=0; i_item<entity_description[sour[0]->type].n_params; i_item++) {
char s[1024];
double value0;
switch (entity_description[sour[0]->type].parameter[i_item].type) {
case IS_DOUBLE:
value = *((double*)(p_elem+entity_description[sour[0]->type].parameter[i_item].offset));
rpn_store(value, NULL, rpn_create_mem(entity_description[sour[0]->type].parameter[i_item].name, 0));
value0 = *((double*)(sour[i_elem]->p_elem0+entity_description[sour[0]->type].parameter[i_item].offset));
sprintf(s, "%s0", entity_description[sour[0]->type].parameter[i_item].name);
rpn_store(value0, NULL, rpn_create_mem(s, 0));
break;
case IS_LONG:
value = *((long*)(p_elem+entity_description[sour[0]->type].parameter[i_item].offset));
rpn_store(value, NULL, rpn_create_mem(entity_description[sour[0]->type].parameter[i_item].name, 0));
value0 = *((long*)(sour[i_elem]->p_elem0+entity_description[sour[0]->type].parameter[i_item].offset));
sprintf(s, "%s0", entity_description[sour[0]->type].parameter[i_item].name);
rpn_store(value0, NULL, rpn_create_mem(s, 0));
break;
default:
break;
}
#if DEBUG
fprintf(stdout, " asserting value %e for %s\n", value, entity_description[sour[0]->type].parameter[i_item].name);
fprintf(stdout, " asserting value %e for %s0\n", value0, entity_description[sour[0]->type].parameter[i_item].name);
fflush(stdout);
#endif
}
p_elem = targ[i_elem]->p_elem;
rpn_clear();
/* push original value onto stack */
if (verbosity>1)
fprintf(stdout, "prior value is %.15g for %s#%ld.%s at z=%.15gm\n",
links->baseline_value[i_link][i_elem],
links->target_name[i_link], targ[i_elem]->occurence, links->item[i_link],
targ[i_elem]->end_pos);
push_num(links->baseline_value[i_link][i_elem]);
value = rpn(links->equation[i_link]);
if (rpn_check_error()) exitElegant(1);
rpn_clear();
if (value>links->maximum[i_link])
value = links->maximum[i_link];
if (value<links->minimum[i_link])
value = links->minimum[i_link];
if (verbosity>0)
fprintf(stdout, "asserting value %.15g for %s#%ld.%s at z=%.15gm\n",
value, links->target_name[i_link], targ[i_elem]->occurence, links->item[i_link], targ[i_elem]->end_pos);
fflush(stdout);
if (entity_description[elem_type].flags&HAS_LENGTH &&
entity_description[elem_type].parameter[param].offset==0)
lengthChanged = 1;
switch (data_type) {
case IS_DOUBLE:
*((double*)(p_elem+entity_description[elem_type].parameter[param].offset)) = value;
break;
case IS_LONG:
*((long*)(p_elem+entity_description[elem_type].parameter[param].offset)) =
nearestInteger(value);
break;
case IS_STRING:
default:
bombElegant("unknown/invalid variable quantity (assert_element_links)", NULL);
exitElegant(1);
}
if (flags&LINK_ELEMENT_DEFINITION)
change_defined_parameter_values(&targ[i_elem]->name, ¶m, &targ[i_elem]->type, &value, 1);
if ((entity_description[targ[0]->type].parameter[param].flags&PARAM_CHANGES_MATRIX) && targ[i_elem]->matrix) {
free_matrices(targ[i_elem]->matrix);
tfree(targ[i_elem]->matrix);
targ[i_elem]->matrix = NULL;
compute_matrix(targ[i_elem], run, NULL);
matrices_changed++;
}
}
}
#if DEBUG
print_line(stdout, beamline);
#endif
if (lengthChanged)
compute_end_positions(beamline);
log_exit("assert_element_links");
return(matrices_changed);
}
