001    /*
002     * Copyright (C) 2008-2010 by Holger Arndt
003     *
004     * This file is part of the Universal Java Matrix Package (UJMP).
005     * See the NOTICE file distributed with this work for additional
006     * information regarding copyright ownership and licensing.
007     *
008     * UJMP is free software; you can redistribute it and/or modify
009     * it under the terms of the GNU Lesser General Public License as
010     * published by the Free Software Foundation; either version 2
011     * of the License, or (at your option) any later version.
012     *
013     * UJMP is distributed in the hope that it will be useful,
014     * but WITHOUT ANY WARRANTY; without even the implied warranty of
015     * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
016     * GNU Lesser General Public License for more details.
017     *
018     * You should have received a copy of the GNU Lesser General Public
019     * License along with UJMP; if not, write to the
020     * Free Software Foundation, Inc., 51 Franklin St, Fifth Floor,
021     * Boston, MA  02110-1301  USA
022     */
023    
024    package org.ujmp.core.stringmatrix.impl;
025    
026    import java.io.File;
027    import java.util.Arrays;
028    import java.util.Collections;
029    import java.util.List;
030    
031    import org.ujmp.core.stringmatrix.stub.AbstractDenseStringMatrix2D;
032    
033    public class FileListMatrix extends AbstractDenseStringMatrix2D {
034            private static final long serialVersionUID = -2627484975560893624L;
035    
036            private List<File> files = null;
037    
038            public FileListMatrix(String path) {
039                    this(new File(path));
040            }
041    
042            public FileListMatrix(File path) {
043                    this.files = Arrays.asList(path.listFiles());
044                    Collections.sort(files);
045                    setMatrixAnnotation(path);
046            }
047    
048            public long[] getSize() {
049                    return new long[] { files.size(), 2 };
050            }
051    
052            public String getString(long row, long column) {
053                    switch ((int) column) {
054                    case 0:
055                            File f = files.get((int) row);
056                            return f.getName();
057                    case 1:
058                            return "" + files.get((int) row).length();
059                    }
060                    return null;
061            }
062    
063            public void setString(String value, long row, long column) {
064                    if (column == 0 && row < files.size()) {
065                            File source = files.get((int) row);
066                            File target = new File(source.getParent() + File.separator + value);
067                            source.renameTo(target);
068                            files.set((int) row, target);
069                    }
070            }
071    
072    }