TanimotoMolSimilarityBySubstructure.java

Go to the documentation of this file.

/*
 *   DENOPTIM
 *   Copyright (C) 2022 Marco Foscato <marco.foscato@uib.no>
 *
 *   This program is free software: you can redistribute it and/or modify
 *   it under the terms of the GNU Affero General Public License as published
 *   by the Free Software Foundation, either version 3 of the License, or
 *   (at your option) any later version.
 *
 *   This program is distributed in the hope that it will be useful,
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *   GNU Affero General Public License for more details.
 *
 *   You should have received a copy of the GNU Affero General Public License
 *   along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
package denoptim.fitness.descriptors;
 
import org.openscience.cdk.exception.CDKException;
import org.openscience.cdk.fingerprint.IBitFingerprint;
import org.openscience.cdk.fingerprint.IFingerprinter;
import org.openscience.cdk.fingerprint.SubstructureFingerprinter;
import org.openscience.cdk.interfaces.IAtomContainer;
import org.openscience.cdk.qsar.AbstractMolecularDescriptor;
import org.openscience.cdk.qsar.DescriptorSpecification;
import org.openscience.cdk.qsar.DescriptorValue;
import org.openscience.cdk.qsar.IMolecularDescriptor;
import org.openscience.cdk.qsar.result.DoubleResult;
import org.openscience.cdk.qsar.result.DoubleResultType;
import org.openscience.cdk.qsar.result.IDescriptorResult;
import org.openscience.cdk.similarity.Tanimoto;
 
import denoptim.fitness.IDenoptimDescriptor;
 
 
public class TanimotoMolSimilarityBySubstructure extends AbstractMolecularDescriptor 
implements IMolecularDescriptor, IDenoptimDescriptor
{
    //private static ILoggingTool logger = LoggingToolFactory.createLoggingTool(
    //        TanimotoMolSimilarity.class);
    private IBitFingerprint referenceFingerprint;
    private IFingerprinter fingerprinter;
    private String[] substructuressmarts;
    private static final String[] PARAMNAMES = new String[] {
            "substructuressmarts","referenceFingerprint"};
 
    private static final String[] NAMES  = {"TanimotoSimilarityBySubstructure"};
    
//------------------------------------------------------------------------------
    
    public TanimotoMolSimilarityBySubstructure() {}
 
//------------------------------------------------------------------------------
      
    @Override
    public DescriptorSpecification getSpecification()
    {
        String paramID = ""; 
        if (fingerprinter!=null && referenceFingerprint!=null)
        {
            paramID = "" + substructuressmarts.hashCode() 
            + referenceFingerprint.hashCode();
        }
        return new DescriptorSpecification("Denoptim source code", 
                this.getClass().getName(), paramID, "DENOPTIM project");
    }
 
//------------------------------------------------------------------------------
    
    @Override
    public String[] getParameterNames() {
        return PARAMNAMES;
    }
 
//------------------------------------------------------------------------------
    
    @Override
    public Object getParameterType(String name)
    {
        if (name.equals(PARAMNAMES[1]))
        {
            return IBitFingerprint.class;
        } else if (name.equals(PARAMNAMES[0])) {
            return String[].class;
        } else {
            throw new IllegalArgumentException("No parameter for name: "+name);
        }
    }
    
//------------------------------------------------------------------------------
 
    @Override
    public void setParameters(Object[] params) throws CDKException
    {
        if (params.length != 2)
        {
            throw new IllegalArgumentException(""
                    + "TanimotoMolSimilarityBySubstructure requires two "
                    + "parameters");
        }
        if (!(params[1] instanceof IBitFingerprint))
        {
            throw new IllegalArgumentException("Parameter does not implemet "
                    + "IBitFingerprint.");
        }
        if (!(params[0] instanceof String[]))
        {
            throw new IllegalArgumentException("Parameter is not String[] (" 
                    + params[0].getClass().getName() + ").");
        }
 
        substructuressmarts = ((String[])params[0]);
        fingerprinter = new SubstructureFingerprinter(substructuressmarts);
        referenceFingerprint = (IBitFingerprint) params[1];
    }
    
//------------------------------------------------------------------------------
 
    @Override
    public Object[] getParameters()
    {
        Object[] params = new Object[2];
        params[1] = referenceFingerprint;
        params[0] = substructuressmarts;
        return params;
    }
 
//------------------------------------------------------------------------------
 
    @Override
    public String[] getDescriptorNames()
    {
        return NAMES;
    }
 
//------------------------------------------------------------------------------
 
    @Override
    public DescriptorValue calculate(IAtomContainer mol)
    {
        DoubleResult result;
        if (referenceFingerprint==null)
        {
            throw new IllegalStateException("Reference fingerprint not set. "
                    + "Cannot calculate Tanimoto similarity.");
        }
        if (fingerprinter==null)
        {
            throw new IllegalStateException("Fingerprinter not set. "
                    + "Cannot calculate Tanimoto similarity.");
        }
        
        try
        {
            result = new DoubleResult(Tanimoto.calculate(referenceFingerprint, 
                    fingerprinter.getBitFingerprint(mol)));
        } catch (IllegalArgumentException e)
        {
            e.printStackTrace();
            result = new DoubleResult(Double.NaN);
        } catch (CDKException e)
        {
            e.printStackTrace();
            result = new DoubleResult(Double.NaN);
        }
        
        return new DescriptorValue(getSpecification(),
                getParameterNames(),
                getParameters(),
                result,
                getDescriptorNames());
    }
 
//------------------------------------------------------------------------------
   
    @Override
    public IDescriptorResult getDescriptorResultType()
    {
        return new DoubleResultType();
    }
 
//------------------------------------------------------------------------------
 
    @Override
    public String getDictionaryTitle()
    {
        return "Tanimoto Molecular Similarity By Substructures";
    }
    
//------------------------------------------------------------------------------
    
    @Override
    public String getDictionaryDefinition()
    {
        return "The Tanimoto Molecular Similarity By Susbtructure "
                + "considers a given set of substructures to calculate "
                + "the Tanimoto similarity between the "
                + "reference fingerprint given upon definition of the "
                + "descriptor (see parameters), and the fingerprint of a "
                + "molecule given as "
                + "argument when calculating the value of the descriptor. "
                + "Fingerprints are obtained from a new instance of"
                + "<code>SubstructureFingerprinter</code> defined by the "
                + "list of SMARTS strings given "
                + "as parameter <code>" + PARAMNAMES[0] + "</code>.";
    }
 
//------------------------------------------------------------------------------
 
    @Override
    public String[] getDictionaryClass()
    {
        return new String[] {"molecular"};
    }
 
//------------------------------------------------------------------------------
 
}