edu.cmu.tetradapp.model.IndependenceFactsModel Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tetrad-gui Show documentation
There is a newer version: 7.6.6
///////////////////////////////////////////////////////////////////////////////
// For information as to what this class does, see the Javadoc, below.       //
// Copyright (C) 1998, 1999, 2000, 2001, 2002, 2003, 2004, 2005, 2006,       //
// 2007, 2008, 2009, 2010, 2014, 2015, 2022 by Peter Spirtes, Richard        //
// Scheines, Joseph Ramsey, and Clark Glymour.                               //
//                                                                           //
// This program is free software; you can redistribute it and/or modify      //
// it under the terms of the GNU General Public License as published by      //
// the Free Software Foundation; either version 2 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 General Public License for more details.                              //
//                                                                           //
// You should have received a copy of the GNU General Public License         //
// along with this program; if not, write to the Free Software               //
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA //
///////////////////////////////////////////////////////////////////////////////

package edu.cmu.tetradapp.model;

import edu.cmu.tetrad.data.IndependenceFacts;
import edu.cmu.tetrad.data.Knowledge;
import edu.cmu.tetrad.data.KnowledgeBoxInput;
import edu.cmu.tetrad.graph.Graph;
import edu.cmu.tetrad.graph.GraphNode;
import edu.cmu.tetrad.graph.IndependenceFact;
import edu.cmu.tetrad.graph.Node;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.Reader;
import java.util.*;

/**
 * Stores a list of independence facts.
 *
 * @author josephramsey
 */
public class IndependenceFactsModel implements KnowledgeBoxInput {
    private static final long serialVersionUID = 23L;

    private IndependenceFacts facts = new IndependenceFacts();
    private String name = "";

    public IndependenceFactsModel() {
        // do nothing.
    }

    /**
     * Generates a simple exemplar of this class to test serialization.
     */
    public static Knowledge serializableInstance() {
        return new Knowledge();
    }

    public static IndependenceFactsModel loadFacts(Reader reader) throws IOException {
        IndependenceFactsModel facts = new IndependenceFactsModel();
        Set names = new HashSet<>();
        Map nodes = new HashMap<>();

        BufferedReader in = new BufferedReader(reader);
        String line;

        while ((line = in.readLine()) != null) {
            String[] tokens = line.split("[ ,;_|]+");

            if (tokens.length == 0) continue;
            if (tokens.length < 2) throw new IllegalArgumentException(
                    "Must specify at least two variables--e.g. X1 X2, for X1 _||_ X2.");

            for (String token : tokens) {
                names.add(token);

                if (!nodes.containsKey(token)) {
                    nodes.put(token, new GraphNode(token));
                }
            }

            Set z = new HashSet<>();

            for (int i = 2; i < tokens.length; i++) {
                z.add(nodes.get(tokens[i]));
            }

            facts.add(new IndependenceFact(nodes.get(tokens[0]), nodes.get(tokens[1]), z));
        }

        return facts;
    }

    public void add(IndependenceFact fact) {
        this.facts.add(fact);
    }

    public String toString() {
        return this.facts.toString();
    }

    public void remove(IndependenceFact fact) {
        this.facts.remove(fact);
    }

    public String getName() {
        return this.name;
    }

    public void setName(String name) {
        this.name = name;
    }

    public IndependenceFacts getFacts() {
        return this.facts;
    }

    public void setFacts(IndependenceFacts facts) {
        if (facts == null) throw new NullPointerException("FActs is null.");
        this.facts = facts;
    }

    public Graph getSourceGraph() {
        return null;
    }

    public Graph getResultGraph() {
        return null;
    }

    public List getVariables() {
        return this.facts.getVariables();
    }

    public List getVariableNames() {
        return this.facts.getVariableNames();
    }
}