NamedStrandedGenerator.java

package org.seqcode.gsebricks.verbs.location;

import java.util.*;
import java.sql.*;

import org.seqcode.data.connections.DatabaseException;
import org.seqcode.genome.Genome;
import org.seqcode.genome.location.NamedStrandedRegion;
import org.seqcode.genome.location.Region;
import org.seqcode.gsebricks.verbs.Expander;


/* Generator that returns NamedStrandedRegion objects from the specified table. 
   The table must have columns: chrom, chromStart, chromEnd, name, strand.
   The regions are returned sorted by ascending chromStart */

public class NamedStrandedGenerator<X extends Region> implements Expander<X,NamedStrandedRegion> {

    private Genome genome;
    private String tablename;

    public NamedStrandedGenerator(Genome g, String t) {
        genome = g;
        tablename = t;
    }

    public Iterator<NamedStrandedRegion> execute(X region) {
        try {
            java.sql.Connection cxn =
                genome.getAnnotationDBConnection();
            PreparedStatement ps = cxn.prepareStatement("select name, strand, chromStart, chromEnd from " + tablename + " where chrom = ? and " +
                                                        "((chromStart <= ? and chromEnd >= ?) or (chromStart >= ? and chromStart <= ?)) order by chromStart");
            String chr = region.getChrom();
            if (!chr.matches("^(chr|scaffold).*")) {
                chr = "chr" + chr;
            }
            ps.setString(1,chr);
            ps.setInt(2,region.getStart());
            ps.setInt(3,region.getStart());
            ps.setInt(4,region.getStart());
            ps.setInt(5,region.getEnd());
            ResultSet rs = ps.executeQuery();
            ArrayList<NamedStrandedRegion> results = new ArrayList<NamedStrandedRegion>();
            while (rs.next()) {
                results.add(new NamedStrandedRegion(genome,
                                                 region.getChrom(),
                                                 rs.getInt(3),
                                                 rs.getInt(4),
                                                 rs.getString(1),
                                                 rs.getString(2).charAt(0)));
                                     
            }
            rs.close();
            ps.close();
            cxn.close();
            return results.iterator();
        } catch (SQLException ex) {
            throw new DatabaseException("Couldn't get UCSC RefGenes",ex);
        }

    }


}