GeneralSampledDistribution.java

Go to the documentation of this file.

/*
 * Created on Nov 16, 2009
 *
 * TODO To change the template for this generated file go to
 * Window - Preferences - Java - Code Style - Code Templates
 */
package edu.tum.cs.inference;

import java.io.File;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.util.HashMap;

public class GeneralSampledDistribution extends BasicSampledDistribution {

        protected String[] varNames;
        protected String[][] domains;
        protected HashMap<String,Integer> varName2Index;        
        
        public GeneralSampledDistribution(double[][] values, Double Z, String[] varNames, String[][] domains) throws Exception {
                this.values = values;
                this.Z = Z;
                this.varNames = varNames;
                this.domains = domains;
                varName2Index = new HashMap<String,Integer>();
                for(int i = 0; i < varNames.length; i++)
                        varName2Index.put(varNames[i], i);
        }
        
        @Override
        public String[] getDomain(int idx) {
                return domains[idx];
        }

        @Override
        public String getVariableName(int idx) {
                return varNames[idx];
        }

        public void write(File f) throws IOException {
                FileOutputStream fos = new FileOutputStream(f);
                ObjectOutputStream oos = new ObjectOutputStream(fos);
                oos.writeObject(this.values);
                oos.writeObject(this.Z);
                oos.writeObject(varNames);
                oos.writeObject(domains);
                oos.close();
        }
        
        public static GeneralSampledDistribution fromFile(File f) throws Exception {
                java.io.ObjectInputStream objstream = new ObjectInputStream(new FileInputStream(f));
            double[][] values = (double[][])objstream.readObject();
            Double Z = (Double)objstream.readObject();
            String[] varNames = (String[])objstream.readObject();
            String[][] domains = (String[][])objstream.readObject();
            objstream.close();
            return new GeneralSampledDistribution(values, Z, varNames, domains);
        }

        @Override
        public int getVariableIndex(String name) {              
                return varName2Index.get(name);
        }

        @Override
        public Integer getNumSamples() {
                throw new RuntimeException("A GeneralizedDistribution represents only the distribution, no additional data is available");
        }
}

---

srldb
Author(s): Dominik Jain, Stefan Waldherr, Moritz Tenorth
autogenerated on Fri Jan 11 09:58:29 2013