Inverter.java

package com.thinkaurelius.faunus.formats;

import com.thinkaurelius.faunus.formats.graphson.GraphSONInputFormat;
import com.thinkaurelius.faunus.formats.graphson.GraphSONOutputFormat;
import com.thinkaurelius.faunus.formats.script.ScriptInputFormat;
import com.thinkaurelius.faunus.formats.script.ScriptOutputFormat;
import com.thinkaurelius.faunus.formats.titan.cassandra.TitanCassandraInputFormat;
import com.thinkaurelius.faunus.formats.titan.cassandra.TitanCassandraOutputFormat;
import com.thinkaurelius.faunus.formats.titan.hbase.TitanHBaseInputFormat;
import com.thinkaurelius.faunus.formats.titan.hbase.TitanHBaseOutputFormat;
import org.apache.hadoop.mapreduce.InputFormat;
import org.apache.hadoop.mapreduce.OutputFormat;
import org.apache.hadoop.mapreduce.lib.input.SequenceFileInputFormat;
import org.apache.hadoop.mapreduce.lib.output.SequenceFileOutputFormat;

/**
 * @author Marko A. Rodriguez (http://markorodriguez.com)
 */
public class Inverter {

    public static Class<? extends OutputFormat> invertInputFormat(Class<? extends InputFormat> inputFormat) {
        if (inputFormat.equals(GraphSONInputFormat.class))
            return GraphSONOutputFormat.class;
        else if (inputFormat.equals(SequenceFileInputFormat.class))
            return SequenceFileOutputFormat.class;
        else if (inputFormat.equals(TitanHBaseInputFormat.class))
            return TitanHBaseOutputFormat.class;
        else if (inputFormat.equals(TitanCassandraInputFormat.class))
            return TitanCassandraOutputFormat.class;
        else if (inputFormat.equals(ScriptInputFormat.class))
            return ScriptOutputFormat.class;

        throw new UnsupportedOperationException("There currently is no inverse for " + inputFormat.getName());
    }

    public static Class<? extends InputFormat> invertOutputFormat(Class<? extends OutputFormat> outputFormat) {
        if (outputFormat.equals(GraphSONOutputFormat.class))
            return GraphSONInputFormat.class;
        else if (outputFormat.equals(SequenceFileOutputFormat.class))
            return SequenceFileInputFormat.class;
        else if (outputFormat.equals(TitanHBaseOutputFormat.class))
            return TitanHBaseInputFormat.class;
        else if (outputFormat.equals(TitanCassandraOutputFormat.class))
            return TitanCassandraInputFormat.class;
        else if (outputFormat.equals(ScriptOutputFormat.class))
            return ScriptInputFormat.class;

        throw new UnsupportedOperationException("There currently is no inverse for " + outputFormat.getName());
    }
}