org.chocosolver.solver.features.Features Maven / Gradle / Ivy
Show all versions of choco-solver Show documentation
/**
* Copyright (c) 2016, Ecole des Mines de Nantes
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
* 3. All advertising materials mentioning features or use of this software
* must display the following acknowledgement:
* This product includes software developed by the .
* 4. Neither the name of the nor the
* names of its contributors may be used to endorse or promote products
* derived from this software without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY ''AS IS'' AND ANY
* EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
* DISCLAIMED. IN NO EVENT SHALL BE LIABLE FOR ANY
* DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
* (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
* LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
* ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
* SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
package org.chocosolver.solver.features;
import gnu.trove.map.hash.TObjectDoubleHashMap;
import org.chocosolver.solver.Model;
import java.util.Arrays;
/**
*
* Project: choco-solver.
*
* @author Charles Prud'homme
* @author Arnaud Malapert
* @since 04/05/2016.
*/
public class Features implements IFeatures {
/**
* For serialization purpose
*/
private static final long serialVersionUID = 1323411473753839215L;
/**
* Name of the model from which the features are extracted
*/
private final String modelName;
/**
* List of attributes extracted on creation
*/
private final TObjectDoubleHashMap values;
/**
* Make a "snapshot" of a model's features
* @param model model to extract features from
* @param attrs list of attribute to extract
*/
public Features(Model model, Attribute... attrs) {
this.modelName = model.getName();
values = new TObjectDoubleHashMap<>(15, 05f);
Arrays.stream(attrs).forEach(a -> values.put(a, a.evaluate(model)));
}
@Override
public String getModelName() {
return modelName;
}
@Override
public int getNbVars() {
return (int) getValue(Attribute.NV);
}
@Override
public int getNbCstrs() {
return (int) getValue(Attribute.NC);
}
@Override
public double getValue(Attribute attribute) {
return values.get(attribute);
}
}