no.packdrill.android.sparkledroid.lib.parse.csvtsv.TSVParse.java Source code

Introduction

Here is the source code for no.packdrill.android.sparkledroid.lib.parse.csvtsv.TSVParse.java

Source

/*
This file is part of SparkleDroidLib. (C) 2013 Donald Munro
    
SparkleDroidLib is free software: you can redistribute it and/or modify it under
the terms of the GNU Lesser General Public License as published by the Free Software Foundation; either version 3
of the License, or (at your option) any later version.
    
This library 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 Lesser General Public License for more details.
    
You should have received a copy of the GNU Lesser General Public License along with this library; if not,
write to the Free Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
or see <http://www.gnu.org/licenses/lgpl-3.0-standalone.html>
 */

package no.packdrill.android.sparkledroid.lib.parse.csvtsv;

import android.util.*;
import no.packdrill.android.sparkledroid.lib.parse.*;
import org.apache.commons.csv.*;

import java.io.*;
import java.net.*;
import java.util.*;
import java.util.regex.*;

public class TSVParse extends AbstractParse implements Parseable
//==============================================================
{
    static final String LOGTAG = TSVParse.class.getSimpleName();
    CSVParser parser = null;
    private Iterator<CSVRecord> it = null;
    Map<String, Integer> headerMap;
    String[] columns = null;

    @Override
    public void parse(Reader input) throws IOException
    //------------------------------------------------
    {
        CSVFormat format = CSVFormat.TDF.withHeader().withCommentStart('%');
        parser = new CSVParser(input, format);
        headerMap = parser.getHeaderMap();
        Set<Map.Entry<String, Integer>> ss = headerMap.entrySet();
        columns = new String[headerMap.size()];
        for (Map.Entry<String, Integer> e : ss)
            columns[e.getValue()] = e.getKey();
        it = parser.iterator();
    }

    @Override
    public void close() throws IOException
    //-------------------------------------
    {
        if (parser != null)
            parser.close();
    }

    public String[] projectionNames() {
        return columns;
    }

    @Override
    public Iterator<Map<String, Cell>> iterator()
    //--------------------------------
    {
        return new Iterator<Map<String, Cell>>()
        //========================================
        {
            Map<String, Cell> m = new HashMap<String, Cell>();

            final Pattern languagePattern = Pattern.compile("\"(.+)\"@(\\w\\w)$");

            @Override
            public boolean hasNext() {
                return it.hasNext();
            }

            @Override
            public Map<String, Cell> next()
            //--------------------------------
            {
                CSVRecord record = null;
                int p;
                try {
                    record = it.next();
                    m.clear();
                    for (int i = 0; i < record.size(); i++) {
                        String k = columns[i];
                        String v = record.get(i);
                        if (v == null)
                            continue;
                        if (v.trim().startsWith("_:")) {
                            p = v.indexOf("_:");
                            String name;
                            try {
                                name = v.substring(p + 2);
                            } catch (Exception _e) {
                                name = "";
                            }
                            Cell ri = new Cell(true, name);
                            ri.setUnparsedValue(v);
                            m.put(k, ri);
                            continue;
                        }
                        if ((v.trim().startsWith("<")) && (v.trim().endsWith(">"))) {
                            p = v.indexOf('<');
                            int p1 = v.indexOf('>');
                            if ((p >= 0) && (p1 > 0))
                                v = v.substring(p + 1, p1);
                            URI uri = null;
                            try {
                                uri = new URI(v);
                            } catch (Exception _e) {
                                uri = null;
                            }
                            if (uri != null) {
                                Cell ri = processURI(v);
                                if (ri != null) {
                                    m.put(k, ri);
                                    continue;
                                }
                            }

                            Matcher patmatch = languagePattern.matcher(v);
                            if ((patmatch.matches()) && (patmatch.groupCount() > 0)) {
                                String s = patmatch.group(1);
                                String lang = null;
                                if (patmatch.groupCount() > 1)
                                    lang = patmatch.group(2);
                                if ((s != null) && (lang != null)) {
                                    Cell ri = new Cell(s, lang);
                                    ri.setUnparsedValue(v);
                                    m.put(k, ri);
                                    continue;
                                }
                            }

                        }
                        Cell ri = new Cell(v);
                        ri.setUnparsedValue(v);
                        m.put(k, ri);
                    }
                } catch (Exception e) {
                    lastError = (record == null) ? "" : record.toString();
                    lastException = e;
                    Log.e(LOGTAG, lastError, e);
                    return null;
                }
                return m;
            }

            @Override
            public void remove() {
                throw new UnsupportedOperationException(
                        "no.packdrill.android.SPARQLClient.parse.csvtsv.iterator.remove");
            }
        };
    }
}