[430] | 1 | #include "madx.h" |
---|
| 2 | |
---|
| 3 | static struct variable* |
---|
| 4 | delete_variable(struct variable* var) |
---|
| 5 | { |
---|
| 6 | char rout_name[] = "delete_variable"; |
---|
| 7 | if (var == NULL) return NULL; |
---|
| 8 | if (stamp_flag && var->stamp != 123456) |
---|
| 9 | fprintf(stamp_file, "d_v double delete --> %s\n", var->name); |
---|
| 10 | if (watch_flag) fprintf(debug_file, "deleting --> %s\n", var->name); |
---|
| 11 | if (var->expr != NULL) delete_expression(var->expr); |
---|
| 12 | if (var->string != NULL) myfree(rout_name, var->string); |
---|
| 13 | myfree(rout_name, var); |
---|
| 14 | return NULL; |
---|
| 15 | } |
---|
| 16 | |
---|
| 17 | static void |
---|
| 18 | grow_var_list(struct var_list* p) |
---|
| 19 | { |
---|
| 20 | char rout_name[] = "grow_var_list"; |
---|
| 21 | struct variable** v_loc = p->vars; |
---|
| 22 | int j, new = 2*p->max; |
---|
| 23 | |
---|
| 24 | p->max = new; |
---|
| 25 | p->vars |
---|
| 26 | = (struct variable**) mycalloc(rout_name,new, sizeof(struct variable*)); |
---|
| 27 | for (j = 0; j < p->curr; j++) p->vars[j] = v_loc[j]; |
---|
| 28 | myfree(rout_name, v_loc); |
---|
| 29 | } |
---|
| 30 | |
---|
| 31 | #if 0 // not used... |
---|
| 32 | static void |
---|
| 33 | dump_variable(struct variable* v) |
---|
| 34 | { |
---|
| 35 | fprintf(prt_file, "=== dumping variable %s\n", v->name); |
---|
| 36 | } |
---|
| 37 | #endif |
---|
| 38 | |
---|
| 39 | static void |
---|
| 40 | export_variable(struct variable* var, FILE* file) |
---|
| 41 | /* exports variable in mad-X format */ |
---|
| 42 | { |
---|
| 43 | int k; |
---|
| 44 | *c_dum->c = '\0'; |
---|
| 45 | if (var->status == 0) var->value = expression_value(var->expr, var->type); |
---|
| 46 | if (var->val_type == 0) strcat(c_dum->c, "int "); |
---|
| 47 | if (var->type == 0) strcat(c_dum->c, "const "); |
---|
| 48 | strcat(c_dum->c, var->name); |
---|
| 49 | if (var->type < 2) strcat(c_dum->c, " = "); |
---|
| 50 | else strcat(c_dum->c, " := "); |
---|
| 51 | if (var->expr != NULL) strcat(c_dum->c, var->expr->string); |
---|
| 52 | else if (var->val_type == 0) |
---|
| 53 | { |
---|
| 54 | k = var->value; sprintf(c_join->c, "%d", k); strcat(c_dum->c, c_join->c); |
---|
| 55 | } |
---|
| 56 | else |
---|
| 57 | { |
---|
| 58 | sprintf(c_join->c, v_format("%F"), var->value); |
---|
| 59 | strcat(c_dum->c, supp_tb(c_join->c)); |
---|
| 60 | } |
---|
| 61 | write_nice(c_dum->c, file); |
---|
| 62 | } |
---|
| 63 | |
---|
| 64 | static void |
---|
| 65 | export_var_8(struct variable* var, FILE* file) |
---|
| 66 | /* exports variable in mad-8 format */ |
---|
| 67 | { |
---|
| 68 | int k; |
---|
| 69 | *c_dum->c = '\0'; |
---|
| 70 | if (var->status == 0) var->value = expression_value(var->expr, var->type); |
---|
| 71 | if (var->type == 0) |
---|
| 72 | { |
---|
| 73 | strcat(c_dum->c, var->name); |
---|
| 74 | strcat(c_dum->c, ": constant = "); |
---|
| 75 | } |
---|
| 76 | else |
---|
| 77 | { |
---|
| 78 | strcat(c_dum->c, var->name); |
---|
| 79 | if (var->type < 2) strcat(c_dum->c, " = "); |
---|
| 80 | else strcat(c_dum->c, " := "); |
---|
| 81 | } |
---|
| 82 | if (var->expr != NULL) strcat(c_dum->c, var->expr->string); |
---|
| 83 | else if (var->val_type == 0) |
---|
| 84 | { |
---|
| 85 | k = var->value; sprintf(c_join->c, v_format("%I"), k); |
---|
| 86 | strcat(c_dum->c, c_join->c); |
---|
| 87 | } |
---|
| 88 | else |
---|
| 89 | { |
---|
| 90 | sprintf(c_join->c, v_format("%F"), var->value); |
---|
| 91 | strcat(c_dum->c, supp_tb(c_join->c)); |
---|
| 92 | } |
---|
| 93 | write_nice_8(c_dum->c, file); |
---|
| 94 | } |
---|
| 95 | |
---|
| 96 | static int |
---|
| 97 | predef_var(struct variable* var) |
---|
| 98 | /* return 1 for predefined variable, else 0 */ |
---|
| 99 | { |
---|
| 100 | int pos = name_list_pos(var->name, variable_list->list); |
---|
| 101 | return (pos < start_var ? 1 : 0) ; |
---|
| 102 | } |
---|
| 103 | |
---|
| 104 | static void |
---|
| 105 | set_sub_variable(char* comm, char* par, struct in_cmd* cmd) |
---|
| 106 | { |
---|
| 107 | char* p; |
---|
| 108 | struct element* el; |
---|
| 109 | struct command *command, *keep_beam = current_beam; |
---|
| 110 | int end, start = cmd->decl_start, t_num, exp_type; |
---|
| 111 | double val = 0; |
---|
| 112 | for (t_num = start; t_num < cmd->tok_list->curr; t_num++) |
---|
| 113 | if (*(cmd->tok_list->p[t_num]) == ',') break; |
---|
| 114 | exp_type = loc_expr(cmd->tok_list->p, t_num, |
---|
| 115 | start, &end); |
---|
| 116 | if (exp_type == 1) /* literal constant */ |
---|
| 117 | val = simple_double(cmd->tok_list->p, start, end); |
---|
| 118 | else if (polish_expr(end + 1 - start, &cmd->tok_list->p[start]) == 0) |
---|
| 119 | val = polish_value(deco, join(&cmd->tok_list->p[start], end + 1 - start)); |
---|
| 120 | if (strncmp(comm, "beam", 4) == 0) |
---|
| 121 | { |
---|
| 122 | command = current_beam = find_command("default_beam", beam_list); |
---|
| 123 | if ((p = strchr(comm, '%')) != NULL) |
---|
| 124 | { |
---|
| 125 | if ((current_beam = find_command(++p, beam_list)) == NULL) |
---|
| 126 | current_beam = command; |
---|
| 127 | } |
---|
| 128 | set_command_par_value(par, current_beam, val); |
---|
| 129 | } |
---|
| 130 | else if ((el = find_element(comm, element_list)) != NULL) |
---|
| 131 | set_command_par_value(par, el->def, val); |
---|
| 132 | else if ((command = find_command(comm, stored_commands)) != NULL) |
---|
| 133 | set_command_par_value(par, command, val); |
---|
| 134 | else if ((command = find_command(comm, beta0_list)) != NULL) |
---|
| 135 | set_command_par_value(par, command, val); |
---|
| 136 | else if ((command = find_command(comm, defined_commands)) != NULL) |
---|
| 137 | set_command_par_value(par, command, val); |
---|
| 138 | current_beam = keep_beam; |
---|
| 139 | } |
---|
| 140 | |
---|
| 141 | // public interface |
---|
| 142 | |
---|
| 143 | void |
---|
| 144 | get_defined_constants(void) |
---|
| 145 | { |
---|
| 146 | /* reads + stores the constants defined in madxdict.h */ |
---|
| 147 | supp_char('\n', constant_def); |
---|
| 148 | pro_input(constant_def); |
---|
| 149 | start_var = variable_list->curr; |
---|
| 150 | } |
---|
| 151 | |
---|
| 152 | struct var_list* |
---|
| 153 | delete_var_list(struct var_list* varl) |
---|
| 154 | { |
---|
| 155 | char rout_name[] = "delete_var_list"; |
---|
| 156 | if (varl == NULL) return NULL; |
---|
| 157 | if (stamp_flag && varl->stamp != 123456) |
---|
| 158 | fprintf(stamp_file, "d_v_l double delete --> %s\n", varl->name); |
---|
| 159 | if (watch_flag) fprintf(debug_file, "deleting --> %s\n", varl->name); |
---|
| 160 | if (varl->list != NULL) delete_name_list(varl->list); |
---|
| 161 | if (varl->vars != NULL) myfree(rout_name, varl->vars); |
---|
| 162 | myfree(rout_name, varl); |
---|
| 163 | return NULL; |
---|
| 164 | } |
---|
| 165 | |
---|
| 166 | struct variable* |
---|
| 167 | find_variable(char* name, struct var_list* varl) |
---|
| 168 | { |
---|
| 169 | int pos; |
---|
| 170 | if ((pos = name_list_pos(name, varl->list)) < 0) |
---|
| 171 | return NULL; |
---|
| 172 | return varl->vars[pos]; |
---|
| 173 | } |
---|
| 174 | |
---|
| 175 | double |
---|
| 176 | variable_value(struct variable* var) |
---|
| 177 | { |
---|
| 178 | int k; |
---|
| 179 | double val = zero; |
---|
| 180 | if (var->type < 2 && var->status > 0) val = var->value; |
---|
| 181 | else if(var->expr == NULL) val = var->value; |
---|
| 182 | else |
---|
| 183 | { |
---|
| 184 | var->value = val = expression_value(var->expr, var->type); |
---|
| 185 | var->status = 1; |
---|
| 186 | } |
---|
| 187 | if (var->val_type == 0) /* int */ |
---|
| 188 | { |
---|
| 189 | k = val; val = k; |
---|
| 190 | } |
---|
| 191 | return val; |
---|
| 192 | } |
---|
| 193 | |
---|
| 194 | struct var_list* |
---|
| 195 | clone_var_list(struct var_list* vl) |
---|
| 196 | { |
---|
| 197 | int i, l = vl->curr > 0 ? vl->curr : 1; |
---|
| 198 | struct var_list* clone; |
---|
| 199 | clone = new_var_list(l); |
---|
| 200 | strcpy(clone->name, vl->name); |
---|
| 201 | clone->list = clone_name_list(vl->list); |
---|
| 202 | for (i = 0; i < vl->curr; i++) clone->vars[i] = vl->vars[i]; |
---|
| 203 | clone->curr = vl->curr; |
---|
| 204 | return clone; |
---|
| 205 | } |
---|
| 206 | |
---|
| 207 | struct variable* |
---|
| 208 | new_variable(char* name, double val, int val_type, int type, struct expression* expr, char* string) |
---|
| 209 | { |
---|
| 210 | char rout_name[] = "new_variable"; |
---|
| 211 | struct variable* var = |
---|
| 212 | (struct variable*) mycalloc(rout_name,1, sizeof(struct variable)); |
---|
| 213 | strcpy(var->name, name); |
---|
| 214 | var->stamp = 123456; |
---|
| 215 | if (watch_flag) fprintf(debug_file, "creating ++> %s\n", var->name); |
---|
| 216 | var->value = val; |
---|
| 217 | var->type = type; |
---|
| 218 | var->val_type = val_type; |
---|
| 219 | if ((var->expr = expr) == NULL) var->status = 1; |
---|
| 220 | if (string != NULL) var->string = tmpbuff(string); |
---|
| 221 | return var; |
---|
| 222 | } |
---|
| 223 | |
---|
| 224 | struct var_list* |
---|
| 225 | new_var_list(int length) |
---|
| 226 | { |
---|
| 227 | char rout_name[] = "new_var_list"; |
---|
| 228 | struct var_list* var |
---|
| 229 | = (struct var_list*) mycalloc(rout_name,1, sizeof(struct var_list)); |
---|
| 230 | strcpy(var->name, "var_list"); |
---|
| 231 | var->stamp = 123456; |
---|
| 232 | if (watch_flag) fprintf(debug_file, "creating ++> %s\n", var->name); |
---|
| 233 | var->list = new_name_list(var->name, length); |
---|
| 234 | var->vars |
---|
| 235 | = (struct variable**) mycalloc(rout_name,length, sizeof(struct variable*)); |
---|
| 236 | var->max = length; |
---|
| 237 | return var; |
---|
| 238 | } |
---|
| 239 | |
---|
| 240 | char* |
---|
| 241 | get_varstring(char* name) |
---|
| 242 | { |
---|
| 243 | struct variable* var; |
---|
| 244 | char *ret; // *p, not used |
---|
| 245 | ret = NULL; |
---|
| 246 | mycpy(c_dum->c, name); |
---|
| 247 | if (strstr(c_dum->c, "->") == NULL) /* variable */ // (p = not used |
---|
| 248 | if ((var = find_variable(c_dum->c, variable_list)) != NULL) |
---|
| 249 | ret = var->string; |
---|
| 250 | |
---|
| 251 | return ret; |
---|
| 252 | } |
---|
| 253 | |
---|
| 254 | void |
---|
| 255 | write_vars(struct var_list* varl, struct command_list* cl, FILE* file) |
---|
| 256 | { |
---|
| 257 | int i; |
---|
| 258 | for (i = 0; i < varl->curr; i++) |
---|
| 259 | { |
---|
| 260 | if (predef_var(varl->vars[i]) == 0 |
---|
| 261 | && pass_select_list(varl->vars[i]->name, cl)) |
---|
| 262 | export_variable(varl->vars[i], file); |
---|
| 263 | } |
---|
| 264 | } |
---|
| 265 | |
---|
| 266 | void |
---|
| 267 | write_vars_8(struct var_list* varl, struct command_list* cl, FILE* file) |
---|
| 268 | { |
---|
| 269 | int i; |
---|
| 270 | for (i = 0; i < varl->curr; i++) |
---|
| 271 | { |
---|
| 272 | if (predef_var(varl->vars[i]) == 0 |
---|
| 273 | && pass_select_list(varl->vars[i]->name, cl)) |
---|
| 274 | export_var_8(varl->vars[i], file); |
---|
| 275 | } |
---|
| 276 | } |
---|
| 277 | |
---|
| 278 | char* |
---|
| 279 | make_string_variable(char* string) |
---|
| 280 | /* creates + stores a variable containing a character string */ |
---|
| 281 | { |
---|
| 282 | char* name = get_new_name(); |
---|
| 283 | struct variable* var = new_variable(name, zero, 3, 0, NULL, string); |
---|
| 284 | add_to_var_list(var, variable_list, 0); |
---|
| 285 | return var->name; |
---|
| 286 | } |
---|
| 287 | |
---|
| 288 | void |
---|
| 289 | enter_variable(struct in_cmd* cmd) /* stores variable contained in cmd */ |
---|
| 290 | { |
---|
| 291 | struct variable* var; |
---|
| 292 | struct expression* expr = NULL; |
---|
| 293 | int k, end, type = 0, name_pos = 0, start = cmd->decl_start, val_type = 0; |
---|
| 294 | double val = 0; |
---|
| 295 | char comm[NAME_L]; |
---|
| 296 | char par[NAME_L]; |
---|
| 297 | char* name; |
---|
| 298 | char *p, *n, *q = comm; |
---|
| 299 | int exp_type = loc_expr(cmd->tok_list->p, cmd->tok_list->curr, |
---|
| 300 | start, &end); |
---|
| 301 | switch (cmd->sub_type) |
---|
| 302 | { |
---|
| 303 | case 2: |
---|
| 304 | case 3: |
---|
| 305 | val_type = 0; |
---|
| 306 | type = 0; |
---|
| 307 | name_pos = 2; |
---|
| 308 | break; |
---|
| 309 | case 4: |
---|
| 310 | case 5: |
---|
| 311 | val_type = 1; |
---|
| 312 | type = 0; |
---|
| 313 | name_pos = 2; |
---|
| 314 | break; |
---|
| 315 | case 6: |
---|
| 316 | val_type = 0; |
---|
| 317 | type = 1; |
---|
| 318 | name_pos = 1; |
---|
| 319 | break; |
---|
| 320 | case 7: |
---|
| 321 | val_type = 0; |
---|
| 322 | type = 2; |
---|
| 323 | name_pos = 1; |
---|
| 324 | break; |
---|
| 325 | case 8: |
---|
| 326 | val_type = 1; |
---|
| 327 | type = 1; |
---|
| 328 | name_pos = 1; |
---|
| 329 | break; |
---|
| 330 | case 9: |
---|
| 331 | val_type = 1; |
---|
| 332 | type = 2; |
---|
| 333 | name_pos = 1; |
---|
| 334 | break; |
---|
| 335 | case 10: |
---|
| 336 | case 11: |
---|
| 337 | val_type = 1; |
---|
| 338 | type = 0; |
---|
| 339 | name_pos = 1; |
---|
| 340 | break; |
---|
| 341 | case 12: |
---|
| 342 | val_type = 1; |
---|
| 343 | type = 1; |
---|
| 344 | name_pos = 0; |
---|
| 345 | break; |
---|
| 346 | case 13: |
---|
| 347 | val_type = 1; |
---|
| 348 | type = 2; |
---|
| 349 | name_pos = 0; |
---|
| 350 | break; |
---|
| 351 | default: |
---|
| 352 | fatal_error("illegal command sub_type in:", |
---|
| 353 | join(cmd->tok_list->p, cmd->tok_list->curr)); |
---|
| 354 | } |
---|
| 355 | n = name = permbuff(cmd->tok_list->p[name_pos]); |
---|
| 356 | if (exp_type == 0) |
---|
| 357 | { |
---|
| 358 | warning("illegal expression set to 0 in:", |
---|
| 359 | join_b(cmd->tok_list->p, cmd->tok_list->curr)); |
---|
| 360 | } |
---|
| 361 | else |
---|
| 362 | { |
---|
| 363 | if ((p = strstr(name,"->")) != NULL) /* element or command parameter */ |
---|
| 364 | { |
---|
| 365 | while (n < p) *(q++) = *(n++); |
---|
| 366 | *q = '\0'; |
---|
| 367 | q = par; n++; n++; |
---|
| 368 | while (*n != '\0') *(q++) = *(n++); |
---|
| 369 | *q = '\0'; |
---|
| 370 | set_sub_variable(comm, par, cmd); |
---|
| 371 | } |
---|
| 372 | else if ((var = find_variable(name, variable_list)) != 0 |
---|
| 373 | && var->type == 0) |
---|
| 374 | { |
---|
| 375 | warning("ignored: attempt to redefine constant:", var->name); |
---|
| 376 | } |
---|
| 377 | else if (exp_type == 1) /* literal constant */ |
---|
| 378 | { |
---|
| 379 | val = simple_double(cmd->tok_list->p, start, end); |
---|
| 380 | var = new_variable(name, val, val_type, type, NULL, NULL); |
---|
| 381 | add_to_var_list(var, variable_list, 1); |
---|
| 382 | } |
---|
| 383 | else |
---|
| 384 | { |
---|
| 385 | if (polish_expr(end + 1 - start, &cmd->tok_list->p[start]) == 0) |
---|
| 386 | { |
---|
| 387 | if (type == 2) /* deferred: expression kept */ |
---|
| 388 | { |
---|
| 389 | expr = new_expression(join(&cmd->tok_list->p[start], end + 1 - start), deco); |
---|
| 390 | val = 0; // LD 2012.10.16: drop warning due to expression_value(expr, type); |
---|
| 391 | } |
---|
| 392 | else |
---|
| 393 | { |
---|
| 394 | expr = NULL; |
---|
| 395 | val = polish_value(deco, join(&cmd->tok_list->p[start], end + 1 - start)); |
---|
| 396 | } |
---|
| 397 | if (val_type == 0) |
---|
| 398 | { |
---|
| 399 | if (fabs(val) < 2.e9) k = val; |
---|
| 400 | else k = 0; |
---|
| 401 | val = k; |
---|
| 402 | } |
---|
| 403 | var = new_variable(name, val, val_type, type, expr, NULL); |
---|
| 404 | add_to_var_list(var, variable_list, 1); |
---|
| 405 | } |
---|
| 406 | else |
---|
| 407 | { |
---|
| 408 | warning("illegal expression set to 0 in:", |
---|
| 409 | join_b(cmd->tok_list->p, cmd->tok_list->curr)); |
---|
| 410 | } |
---|
| 411 | } |
---|
| 412 | } |
---|
| 413 | } |
---|
| 414 | |
---|
| 415 | void |
---|
| 416 | add_to_var_list( /* adds variable to alphabetic variable list */ |
---|
| 417 | struct variable* var, struct var_list* varl, int flag) |
---|
| 418 | /* flag = 0: undefined reference in expression, 1: definition |
---|
| 419 | 2: separate list, do not drop variable */ |
---|
| 420 | { |
---|
| 421 | int pos; // , j; not used |
---|
| 422 | |
---|
| 423 | if ((pos = name_list_pos(var->name, varl->list)) > -1) |
---|
| 424 | { |
---|
| 425 | if (flag == 1) |
---|
| 426 | { |
---|
| 427 | if (varl->list->inform[pos] == 1) |
---|
| 428 | { |
---|
| 429 | put_info(var->name, "redefined"); |
---|
| 430 | /* |
---|
| 431 | printf("Old Value:\n"); |
---|
| 432 | export_variable(varl->vars[pos], stdout); |
---|
| 433 | printf("New Value:\n"); |
---|
| 434 | export_variable(var, stdout); |
---|
| 435 | printf("File %s line %d\n",filenames[in->curr], currentline[in->curr] ); |
---|
| 436 | */ |
---|
| 437 | } |
---|
| 438 | else varl->list->inform[pos] = flag; |
---|
| 439 | } |
---|
| 440 | if (flag < 2) delete_variable(varl->vars[pos]); |
---|
| 441 | varl->vars[pos] = var; |
---|
| 442 | } |
---|
| 443 | else |
---|
| 444 | { |
---|
| 445 | if (varl->curr == varl->max) grow_var_list(varl); |
---|
| 446 | add_to_name_list(permbuff(var->name), flag, varl->list); // j = not used |
---|
| 447 | varl->vars[varl->curr++] = var; |
---|
| 448 | } |
---|
| 449 | } |
---|
| 450 | |
---|
| 451 | void |
---|
| 452 | set_stringvar(char* name, char* string) |
---|
| 453 | { |
---|
| 454 | /* sets variable name->string to string */ |
---|
| 455 | // char* p; |
---|
| 456 | struct variable* var; |
---|
| 457 | mycpy(c_dum->c, name); |
---|
| 458 | if (strstr(c_dum->c, "->") == NULL) /* variable */ // (p = not used |
---|
| 459 | { |
---|
| 460 | if ((var = find_variable(c_dum->c, variable_list)) != NULL) |
---|
| 461 | { |
---|
| 462 | if (var->type == 3) var->string = string; |
---|
| 463 | } |
---|
| 464 | else |
---|
| 465 | { |
---|
| 466 | var = new_variable(c_dum->c, zero, 0, 3, NULL, string); |
---|
| 467 | add_to_var_list(var, variable_list, 1); |
---|
| 468 | } |
---|
| 469 | } |
---|
| 470 | } |
---|
| 471 | |
---|
| 472 | void |
---|
| 473 | print_global(double delta) |
---|
| 474 | { |
---|
| 475 | char tmp[NAME_L], trad[4]; |
---|
| 476 | double alfa = get_value("probe", "alfa"); |
---|
| 477 | double freq0 = get_value("probe", "freq0"); |
---|
| 478 | double gamma = get_value("probe", "gamma"); |
---|
| 479 | double beta = get_value("probe", "beta"); |
---|
| 480 | double circ = get_value("probe", "circ"); |
---|
| 481 | double bcurrent = get_value("probe", "bcurrent"); |
---|
| 482 | double npart = get_value("probe", "npart"); |
---|
| 483 | double energy = get_value("probe", "energy"); |
---|
| 484 | int kbunch = get_value("probe", "kbunch"); |
---|
| 485 | int rad = get_value("probe", "radiate"); |
---|
| 486 | double gamtr = zero, t0 = zero, eta; |
---|
| 487 | |
---|
| 488 | (void)delta; |
---|
| 489 | get_string("probe", "particle", tmp); |
---|
| 490 | if (rad) strcpy(trad, "T"); |
---|
| 491 | else strcpy(trad, "F"); |
---|
| 492 | if (alfa > zero) gamtr = sqrt(one / alfa); |
---|
| 493 | else if (alfa < zero) gamtr = sqrt(-one / alfa); |
---|
| 494 | if (freq0 > zero) t0 = one / freq0; |
---|
| 495 | eta = alfa - one / (gamma*gamma); |
---|
| 496 | puts(" "); |
---|
| 497 | printf(" Global parameters for %ss, radiate = %s:\n\n", |
---|
| 498 | tmp, trad); |
---|
| 499 | printf(v_format(" C %F m f0 %F MHz\n"),circ, freq0); |
---|
| 500 | printf(v_format(" T0 %F musecs alfa %F \n"), t0, alfa); |
---|
| 501 | printf(v_format(" eta %F gamma(tr) %F \n"), eta, gamtr); |
---|
| 502 | printf(v_format(" Bcurrent %F A/bunch Kbunch %I \n"), |
---|
| 503 | bcurrent, kbunch); |
---|
| 504 | printf(v_format(" Npart %F /bunch Energy %F GeV \n"), |
---|
| 505 | npart,energy); |
---|
| 506 | printf(v_format(" gamma %F beta %F\n"), |
---|
| 507 | gamma, beta); |
---|
| 508 | } |
---|
| 509 | |
---|
| 510 | #if 0 // not used |
---|
| 511 | int |
---|
| 512 | vary_name(char* name, int* name_l, int* index) |
---|
| 513 | /* returns the variable name */ |
---|
| 514 | { |
---|
| 515 | int pos, ncp, nbl, len; |
---|
| 516 | char* v_name; |
---|
| 517 | struct name_list* nl; |
---|
| 518 | struct command* comm; |
---|
| 519 | struct command_parameter_list* pl; |
---|
| 520 | comm = stored_match_var->commands[*index]; |
---|
| 521 | nl = comm->par_names; |
---|
| 522 | pl = comm->par; |
---|
| 523 | pos = name_list_pos("name", nl); |
---|
| 524 | v_name = pl->parameters[pos]->string; |
---|
| 525 | len = strlen(v_name); |
---|
| 526 | ncp = len < *name_l ? len : *name_l; // min(len, *name_l) |
---|
| 527 | nbl = *name_l - ncp; |
---|
| 528 | strncpy(name, v_name, ncp); |
---|
| 529 | return 1; |
---|
| 530 | } |
---|
| 531 | #endif |
---|
| 532 | |
---|
| 533 | #if 0 // not used... |
---|
| 534 | double |
---|
| 535 | sss_variable(char* name) |
---|
| 536 | { |
---|
| 537 | char comm[NAME_L]; |
---|
| 538 | char par[NAME_L]; |
---|
| 539 | double val = zero; |
---|
| 540 | struct variable* var; |
---|
| 541 | struct element* el; |
---|
| 542 | struct command* cmd; |
---|
| 543 | char *p, *n = c_dum->c, *q = comm; |
---|
| 544 | mycpy(c_dum->c, name); |
---|
| 545 | if ((p = strstr(c_dum->c, "->")) == NULL) // variable |
---|
| 546 | { |
---|
| 547 | if ((var = find_variable(c_dum->c, variable_list)) != NULL) |
---|
| 548 | val = variable_value(var); |
---|
| 549 | } |
---|
| 550 | else // element or command parameter |
---|
| 551 | { |
---|
| 552 | while (n < p) *(q++) = *(n++); |
---|
| 553 | *q = '\0'; |
---|
| 554 | q = par; n++; n++; |
---|
| 555 | while (*n != '\0') *(q++) = *(n++); |
---|
| 556 | *q = '\0'; |
---|
| 557 | if ((el = find_element(comm, element_list)) != NULL) |
---|
| 558 | val = command_par_value(par, el->def); |
---|
| 559 | else if ((cmd = find_command(comm, stored_commands)) != NULL) |
---|
| 560 | val = command_par_value(par, cmd); |
---|
| 561 | else if ((cmd = find_command(comm, beta0_list)) != NULL) |
---|
| 562 | val = command_par_value(par, cmd); |
---|
| 563 | else if ((cmd = find_command(comm, defined_commands)) != NULL) |
---|
| 564 | val = command_par_value(par, cmd); |
---|
| 565 | } |
---|
| 566 | return val; |
---|
| 567 | } |
---|
| 568 | #endif |
---|
| 569 | |
---|
| 570 | // public interface (used by Fortran) |
---|
| 571 | |
---|
| 572 | double |
---|
| 573 | get_variable(char* name) |
---|
| 574 | { |
---|
| 575 | char comm[NAME_L]; |
---|
| 576 | char par[NAME_L]; |
---|
| 577 | double val = zero; |
---|
| 578 | struct variable* var; |
---|
| 579 | struct element* el; |
---|
| 580 | struct command* cmd; |
---|
| 581 | char *p, *n = c_dum->c, *q = comm; |
---|
| 582 | mycpy(c_dum->c, name); |
---|
| 583 | if ((p = strstr(c_dum->c, "->")) == NULL) /* variable */ |
---|
| 584 | { |
---|
| 585 | if ((var = find_variable(c_dum->c, variable_list)) != NULL) |
---|
| 586 | val = variable_value(var); |
---|
| 587 | } |
---|
| 588 | else /* element or command parameter */ |
---|
| 589 | { |
---|
| 590 | while (n < p) *(q++) = *(n++); |
---|
| 591 | *q = '\0'; |
---|
| 592 | q = par; n++; n++; |
---|
| 593 | while (*n != '\0') *(q++) = *(n++); |
---|
| 594 | *q = '\0'; |
---|
| 595 | if ((el = find_element(comm, element_list)) != NULL) |
---|
| 596 | val = command_par_value(par, el->def); |
---|
| 597 | else if ((cmd = find_command(comm, stored_commands)) != NULL) |
---|
| 598 | val = command_par_value(par, cmd); |
---|
| 599 | else if ((cmd = find_command(comm, beta0_list)) != NULL) |
---|
| 600 | val = command_par_value(par, cmd); |
---|
| 601 | else if ((cmd = find_command(comm, defined_commands)) != NULL) |
---|
| 602 | val = command_par_value(par, cmd); |
---|
| 603 | } |
---|
| 604 | return val; |
---|
| 605 | } |
---|
| 606 | |
---|
| 607 | void |
---|
| 608 | set_variable(char* name, double* value) |
---|
| 609 | { |
---|
| 610 | /* sets variable name to value */ |
---|
| 611 | char comm[NAME_L]; |
---|
| 612 | char par[NAME_L]; |
---|
| 613 | struct variable* var; |
---|
| 614 | double val = *value; |
---|
| 615 | struct element* el; |
---|
| 616 | struct command* cmd; |
---|
| 617 | char *p, *n = c_dum->c, *q = comm; |
---|
| 618 | mycpy(c_dum->c, name); |
---|
| 619 | if ((p = strstr(c_dum->c, "->")) == NULL) /* variable */ |
---|
| 620 | { |
---|
| 621 | if ((var = find_variable(c_dum->c, variable_list)) != NULL) |
---|
| 622 | { |
---|
| 623 | if (var->type == 0) |
---|
| 624 | warning("ignored: attempt to redefine constant:", var->name); |
---|
| 625 | else if (var->type < 3) |
---|
| 626 | { |
---|
| 627 | var->value = val; |
---|
| 628 | var->type = 1; |
---|
| 629 | if (var->expr != NULL) var->expr = delete_expression(var->expr); |
---|
| 630 | } |
---|
| 631 | } |
---|
| 632 | else |
---|
| 633 | { |
---|
| 634 | var = new_variable(c_dum->c, val, 1, 1, NULL, NULL); |
---|
| 635 | add_to_var_list(var, variable_list, 1); |
---|
| 636 | } |
---|
| 637 | } |
---|
| 638 | else /* element or command parameter */ |
---|
| 639 | { |
---|
| 640 | while (n < p) *(q++) = *(n++); |
---|
| 641 | *q = '\0'; |
---|
| 642 | q = par; n++; n++; |
---|
| 643 | while (*n != '\0') *(q++) = *(n++); |
---|
| 644 | *q = '\0'; |
---|
| 645 | if ((el = find_element(comm, element_list)) != NULL) |
---|
| 646 | set_command_par_value(par, el->def, val); |
---|
| 647 | else if ((cmd = find_command(comm, stored_commands)) != NULL) |
---|
| 648 | set_command_par_value(par, cmd, val); |
---|
| 649 | else if ((cmd = find_command(comm, beta0_list)) != NULL) |
---|
| 650 | set_command_par_value(par, cmd, val); |
---|
| 651 | else if ((cmd = find_command(comm, defined_commands)) != NULL) |
---|
| 652 | set_command_par_value(par, cmd, val); |
---|
| 653 | } |
---|
| 654 | } |
---|
| 655 | |
---|