jobs.MatrixBlockTraceMult.java Source code

Introduction

Here is the source code for jobs.MatrixBlockTraceMult.java

Source

/*
# DARPA XDATA licenses this file to You under the Apache License, Version 2.0
# (the "License"); you may not use this file except in compliance with 
# the License.  You may obtain a copy of the License at 
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software 
# distributed under the License is distributed on an "AS IS" BASIS, 
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and  
# limitations under the License.
#
# Copyright 2013 Raytheon BBN Technologies Corp.  All Rights Reserved. 
#
*/

package jobs;

import java.io.*;
import org.apache.hadoop.conf.*;
import org.apache.hadoop.fs.Path;
import org.apache.hadoop.io.*;
import org.apache.hadoop.mapreduce.lib.input.TextInputFormat;
import org.apache.hadoop.mapreduce.Job;
import org.apache.hadoop.mapreduce.Mapper;
import org.apache.hadoop.mapreduce.Reducer;
import org.apache.hadoop.mapreduce.lib.input.FileInputFormat;
import org.apache.hadoop.mapreduce.lib.input.SequenceFileInputFormat;
import org.apache.hadoop.mapreduce.lib.output.TextOutputFormat;
import org.apache.hadoop.mapreduce.lib.output.FileOutputFormat;
import org.apache.hadoop.mapreduce.lib.output.SequenceFileOutputFormat;
import org.apache.hadoop.util.GenericOptionsParser;
import org.apache.hadoop.mapreduce.Partitioner;
import org.apache.hadoop.filecache.DistributedCache;
import java.net.URI;
import org.apache.hadoop.util.*;

//import com.bbn.fileformats.*;
import fileformats.*;
//import com.bbn.algebra.hadoop.mappers.*;
import mappers.*;
//import com.bbn.algebra.hadoop.reducers.*
import reducers.*;

public class MatrixBlockTraceMult extends Configured implements Tool {

    public int run(String[] args) throws Exception {

        Configuration conf = getConf();

        conf.setFloat("SCALAR", Float.parseFloat(args[3]));

        conf.setBoolean("LTRANS", Boolean.parseBoolean(args[4]));
        conf.setBoolean("RTRANS", Boolean.parseBoolean(args[5]));

        //set # of reducers
        conf.setInt("mapred.reduce.tasks", Integer.parseInt(args[6]));

        //Get optional blocksize parameters
        if (args.length >= 8)
            conf.setInt("SRL", Integer.parseInt(args[7]));

        if (args.length >= 9)
            conf.setInt("SCL", Integer.parseInt(args[8]));

        if (args.length >= 10)
            conf.setInt("SRR", Integer.parseInt(args[9]));

        if (args.length >= 11)
            conf.setInt("SCR", Integer.parseInt(args[10]));

        conf.set("LEFTNAME", args[0]);
        conf.set("RIGHTNAME", args[1]);

        //heap space - should be entered with the -D format and not dealt with by the program.    
        conf.set("mapred.map.child.java.opts", "-Xmx3G");
        conf.set("mapred.reduce.child.java.opts", "-Xmx3G");

        //job
        Job job1 = new Job(conf, "MatrixBlockTraceMult");
        job1.setJarByClass(MatrixBlockMult.class);

        // Map
        FileInputFormat.addInputPath(job1, new Path(args[0]));
        FileInputFormat.addInputPath(job1, new Path(args[1]));
        job1.setInputFormatClass(SequenceFileInputFormat.class);
        job1.setMapperClass(SquareBlockTraceMultiplicationGroupingMapper.class);
        job1.setMapOutputKeyClass(Text.class);
        job1.setMapOutputValueClass(MatrixBlock.class);

        //Reduce       
        job1.setReducerClass(SquareMatrixBlockTraceMultReducer.class);
        job1.setOutputKeyClass(NullWritable.class);
        job1.setOutputValueClass(DoubleWritable.class);
        FileOutputFormat.setOutputPath(job1, new Path(args[2]));
        job1.setOutputFormatClass(TextOutputFormat.class);

        return job1.waitForCompletion(false) ? 0 : 1;
    }

    public static void main(String[] args) throws Exception {
        Configuration conf = new Configuration();
        int res = ToolRunner.run(conf, new MatrixBlockTraceMult(), args);
        System.exit(res);
    }
}