cannam@95: /*
cannam@95: * Copyright (c) 2003, 2007-11 Matteo Frigo
cannam@95: * Copyright (c) 2003, 2007-11 Massachusetts Institute of Technology
cannam@95: *
cannam@95: * This program is free software; you can redistribute it and/or modify
cannam@95: * it under the terms of the GNU General Public License as published by
cannam@95: * the Free Software Foundation; either version 2 of the License, or
cannam@95: * (at your option) any later version.
cannam@95: *
cannam@95: * This program is distributed in the hope that it will be useful,
cannam@95: * but WITHOUT ANY WARRANTY; without even the implied warranty of
cannam@95: * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
cannam@95: * GNU General Public License for more details.
cannam@95: *
cannam@95: * You should have received a copy of the GNU General Public License
cannam@95: * along with this program; if not, write to the Free Software
cannam@95: * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
cannam@95: *
cannam@95: */
cannam@95:
cannam@95: #include "api.h"
cannam@95:
cannam@95: static plan *mkplan0(planner *plnr, unsigned flags,
cannam@95: const problem *prb, int hash_info,
cannam@95: wisdom_state_t wisdom_state)
cannam@95: {
cannam@95: /* map API flags into FFTW flags */
cannam@95: X(mapflags)(plnr, flags);
cannam@95:
cannam@95: plnr->flags.hash_info = hash_info;
cannam@95: plnr->wisdom_state = wisdom_state;
cannam@95:
cannam@95: /* create plan */
cannam@95: return plnr->adt->mkplan(plnr, prb);
cannam@95: }
cannam@95:
cannam@95: static unsigned force_estimator(unsigned flags)
cannam@95: {
cannam@95: flags &= ~(FFTW_MEASURE | FFTW_PATIENT | FFTW_EXHAUSTIVE);
cannam@95: return (flags | FFTW_ESTIMATE);
cannam@95: }
cannam@95:
cannam@95: static plan *mkplan(planner *plnr, unsigned flags,
cannam@95: const problem *prb, int hash_info)
cannam@95: {
cannam@95: plan *pln;
cannam@95:
cannam@95: pln = mkplan0(plnr, flags, prb, hash_info, WISDOM_NORMAL);
cannam@95:
cannam@95: if (plnr->wisdom_state == WISDOM_NORMAL && !pln) {
cannam@95: /* maybe the planner failed because of inconsistent wisdom;
cannam@95: plan again ignoring infeasible wisdom */
cannam@95: pln = mkplan0(plnr, force_estimator(flags), prb,
cannam@95: hash_info, WISDOM_IGNORE_INFEASIBLE);
cannam@95: }
cannam@95:
cannam@95: if (plnr->wisdom_state == WISDOM_IS_BOGUS) {
cannam@95: /* if the planner detected a wisdom inconsistency,
cannam@95: forget all wisdom and plan again */
cannam@95: plnr->adt->forget(plnr, FORGET_EVERYTHING);
cannam@95:
cannam@95: A(!pln);
cannam@95: pln = mkplan0(plnr, flags, prb, hash_info, WISDOM_NORMAL);
cannam@95:
cannam@95: if (plnr->wisdom_state == WISDOM_IS_BOGUS) {
cannam@95: /* if it still fails, plan without wisdom */
cannam@95: plnr->adt->forget(plnr, FORGET_EVERYTHING);
cannam@95:
cannam@95: A(!pln);
cannam@95: pln = mkplan0(plnr, force_estimator(flags),
cannam@95: prb, hash_info, WISDOM_IGNORE_ALL);
cannam@95: }
cannam@95: }
cannam@95:
cannam@95: return pln;
cannam@95: }
cannam@95:
cannam@95: apiplan *X(mkapiplan)(int sign, unsigned flags, problem *prb)
cannam@95: {
cannam@95: apiplan *p = 0;
cannam@95: plan *pln;
cannam@95: unsigned flags_used_for_planning;
cannam@95: planner *plnr = X(the_planner)();
cannam@95: unsigned int pats[] = {FFTW_ESTIMATE, FFTW_MEASURE,
cannam@95: FFTW_PATIENT, FFTW_EXHAUSTIVE};
cannam@95: int pat, pat_max;
cannam@95: double pcost = 0;
cannam@95:
cannam@95: if (flags & FFTW_WISDOM_ONLY) {
cannam@95: /* Special mode that returns a plan only if wisdom is present,
cannam@95: and returns 0 otherwise. This is now documented in the manual,
cannam@95: as a way to detect whether wisdom is available for a problem. */
cannam@95: flags_used_for_planning = flags;
cannam@95: pln = mkplan0(plnr, flags, prb, 0, WISDOM_ONLY);
cannam@95: } else {
cannam@95: pat_max = flags & FFTW_ESTIMATE ? 0 :
cannam@95: (flags & FFTW_EXHAUSTIVE ? 3 :
cannam@95: (flags & FFTW_PATIENT ? 2 : 1));
cannam@95: pat = plnr->timelimit >= 0 ? 0 : pat_max;
cannam@95:
cannam@95: flags &= ~(FFTW_ESTIMATE | FFTW_MEASURE |
cannam@95: FFTW_PATIENT | FFTW_EXHAUSTIVE);
cannam@95:
cannam@95: plnr->start_time = X(get_crude_time)();
cannam@95:
cannam@95: /* plan at incrementally increasing patience until we run
cannam@95: out of time */
cannam@95: for (pln = 0, flags_used_for_planning = 0; pat <= pat_max; ++pat) {
cannam@95: plan *pln1;
cannam@95: unsigned tmpflags = flags | pats[pat];
cannam@95: pln1 = mkplan(plnr, tmpflags, prb, 0);
cannam@95:
cannam@95: if (!pln1) {
cannam@95: /* don't bother continuing if planner failed or timed out */
cannam@95: A(!pln || plnr->timed_out);
cannam@95: break;
cannam@95: }
cannam@95:
cannam@95: X(plan_destroy_internal)(pln);
cannam@95: pln = pln1;
cannam@95: flags_used_for_planning = tmpflags;
cannam@95: pcost = pln->pcost;
cannam@95: }
cannam@95: }
cannam@95:
cannam@95: if (pln) {
cannam@95: /* build apiplan */
cannam@95: p = (apiplan *) MALLOC(sizeof(apiplan), PLANS);
cannam@95: p->prb = prb;
cannam@95: p->sign = sign; /* cache for execute_dft */
cannam@95:
cannam@95: /* re-create plan from wisdom, adding blessing */
cannam@95: p->pln = mkplan(plnr, flags_used_for_planning, prb, BLESSING);
cannam@95:
cannam@95: /* record pcost from most recent measurement for use in X(cost) */
cannam@95: p->pln->pcost = pcost;
cannam@95:
cannam@95: if (sizeof(trigreal) > sizeof(R)) {
cannam@95: /* this is probably faster, and we have enough trigreal
cannam@95: bits to maintain accuracy */
cannam@95: X(plan_awake)(p->pln, AWAKE_SQRTN_TABLE);
cannam@95: } else {
cannam@95: /* more accurate */
cannam@95: X(plan_awake)(p->pln, AWAKE_SINCOS);
cannam@95: }
cannam@95:
cannam@95: /* we don't use pln for p->pln, above, since by re-creating the
cannam@95: plan we might use more patient wisdom from a timed-out mkplan */
cannam@95: X(plan_destroy_internal)(pln);
cannam@95: } else
cannam@95: X(problem_destroy)(prb);
cannam@95:
cannam@95: /* discard all information not necessary to reconstruct the plan */
cannam@95: plnr->adt->forget(plnr, FORGET_ACCURSED);
cannam@95:
cannam@95: #ifdef FFTW_RANDOM_ESTIMATOR
cannam@95: X(random_estimate_seed)++; /* subsequent "random" plans are distinct */
cannam@95: #endif
cannam@95:
cannam@95: return p;
cannam@95: }
cannam@95:
cannam@95: void X(destroy_plan)(X(plan) p)
cannam@95: {
cannam@95: if (p) {
cannam@95: X(plan_awake)(p->pln, SLEEPY);
cannam@95: X(plan_destroy_internal)(p->pln);
cannam@95: X(problem_destroy)(p->prb);
cannam@95: X(ifree)(p);
cannam@95: }
cannam@95: }