org.drools.planner.examples.nqueens.solver.nQueensCompensationActionExperimentScoreRules.drl Maven / Gradle / Ivy

/*
 * Copyright 2010 JBoss Inc
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.drools.planner.examples.nqueens.solver;
    dialect "java"

import org.drools.planner.core.score.buildin.simple.SimpleScoreHolder;
import org.drools.planner.core.score.constraint.UnweightedConstraintOccurrence;
import org.drools.planner.core.score.constraint.ConstraintType;

import org.drools.planner.examples.nqueens.domain.Queen;

global SimpleScoreHolder scoreHolder;

// ############################################################################
// Hard constraints
// ############################################################################

rule "multipleQueensHorizontal"
    when
        $q1 : Queen($id : id, $i : rowIndex)
        $q2 : Queen(id > $id, rowIndex == $i)
    then
        int score = scoreHolder.getScore();
        score--;
        scoreHolder.setScore(score);
        // TODO greatly simplify the use of compensation action
        final SimpleScoreHolder finalScoreHolder = scoreHolder;
        ((org.drools.common.AgendaItem) kcontext.getActivation()).setActivationUnMatchListener(
            new org.drools.event.rule.ActivationUnMatchListener() {
                public void unMatch(org.drools.runtime.rule.WorkingMemory wm,
                        org.drools.runtime.rule.Activation activation) {
                    int score = finalScoreHolder.getScore();
                    score++;
                    finalScoreHolder.setScore(score);
                }
            }
        );
end

// multipleQueensVertical is build-in

rule "multipleQueensAscendingDiagonal"
    when
        $q1 : Queen($id : id, $i : ascendingDiagonalIndex)
        $q2 : Queen(id > $id, ascendingDiagonalIndex == $i)
    then
        int score = scoreHolder.getScore();
        score--;
        scoreHolder.setScore(score);
        // TODO greatly simplify the use of compensation action
        final SimpleScoreHolder finalScoreHolder = scoreHolder;
        ((org.drools.common.AgendaItem) kcontext.getActivation()).setActivationUnMatchListener(
            new org.drools.event.rule.ActivationUnMatchListener() {
                public void unMatch(org.drools.runtime.rule.WorkingMemory wm,
                        org.drools.runtime.rule.Activation activation) {
                    int score = finalScoreHolder.getScore();
                    score++;
                    finalScoreHolder.setScore(score);
                }
            }
        );
end

rule "multipleQueensDescendingDiagonal"
    when
        $q1 : Queen($id : id, $i : descendingDiagonalIndex)
        $q2 : Queen(id > $id, descendingDiagonalIndex == $i)
    then
        int score = scoreHolder.getScore();
        score--;
        scoreHolder.setScore(score);
        // TODO greatly simplify the use of compensation action
        final SimpleScoreHolder finalScoreHolder = scoreHolder;
        ((org.drools.common.AgendaItem) kcontext.getActivation()).setActivationUnMatchListener(
            new org.drools.event.rule.ActivationUnMatchListener() {
                public void unMatch(org.drools.runtime.rule.WorkingMemory wm,
                        org.drools.runtime.rule.Activation activation) {
                    int score = finalScoreHolder.getScore();
                    score++;
                    finalScoreHolder.setScore(score);
                }
            }
        );
end