Method evalStrata

engine/seminaivebottomup.go:247–308

evalStrata runs the evaluation for the layers.

()

// evalStrata runs the evaluation for the layers.
func (e *engine) evalStrata() error {
	predicateAllowList := *e.options.predicateAllowList
	for i, fact := range e.programInfo.InitialFacts {
		if !predicateAllowList(fact.Predicate) {
			continue
		}
		f, err := functional.EvalAtom(fact, nil)
		if err != nil {
			return err
		}

		interval := e.programInfo.InitialFactTimes[i]
		if interval != nil && e.temporalStore != nil {
			if _, err := e.temporalStore.Add(f, *interval); err != nil {
				return err
			}
		} else {
			e.store.Add(f)
		}
	}
	for i := 0; i < len(e.strata); i++ {
		stratumEdbPredicates := make(map[ast.PredicateSym]struct{})
		for j := 0; j < i; j++ {
			for _, sym := range e.stats.Strata[j] {
				stratumEdbPredicates[sym] = struct{}{}
			}
		}
		stratumIdbPredicates := make(map[ast.PredicateSym]struct{})
		var stratumRules []ast.Clause
		stratumDecls := make(map[ast.PredicateSym]*ast.Decl)
		for _, sym := range e.stats.Strata[i] {
			stratumIdbPredicates[sym] = struct{}{}
			stratumRules = append(stratumRules, e.predToRules[sym]...)
			stratumDecls[sym] = e.predToDecl[sym]
		}
		stratifiedProgram := rewrite.Rewrite(analysis.Program{stratumEdbPredicates, stratumIdbPredicates, stratumRules})
		start := time.Now()
		var temporalDeltaStore factstore.TemporalFactStore
		if e.temporalStore != nil {
			temporalDeltaStore = factstore.NewTemporalStore()
		}

		e := engine{
			store:              e.store,
			deltaStore:         factstore.NewMultiIndexedArrayInMemoryStore(),
			temporalStore:      e.temporalStore,
			temporalDeltaStore: temporalDeltaStore,
			evalTime:           e.evalTime,
			programInfo:        &analysis.ProgramInfo{stratifiedProgram.EdbPredicates, stratifiedProgram.IdbPredicates, nil, nil, stratifiedProgram.Rules, stratumDecls, nil},
			predToStratum:      e.predToStratum,
			predToRules:        e.predToRules,
			predToDecl:         e.predToDecl,
			stats:              e.stats,
			options:            e.options,
		}
		if err := e.eval(); err != nil {
			return err
		}