org.canova.api.io.labels.PatternPathLabelGenerator Maven / Gradle / Ivy
/*
*
* *
* * * Copyright 2016 Skymind,Inc.
* * *
* * * Licensed 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.
* *
*
*/
package org.canova.api.io.labels;
import java.io.File;
import java.net.URI;
import org.apache.commons.io.FilenameUtils;
import org.canova.api.io.data.Text;
import org.canova.api.writable.Writable;
/**
* Returns a label derived from the base name of the path. Splits the base name
* of the path with the given regex pattern, and returns the patternPosition'th
* element of the array.
*
* @author saudet
*/
public class PatternPathLabelGenerator implements PathLabelGenerator {
protected String pattern; // Pattern to split and segment file name, pass in regex
protected int patternPosition = 0;
public PatternPathLabelGenerator(String pattern) {
this.pattern = pattern;
}
public PatternPathLabelGenerator(String pattern, int patternPosition) {
this.pattern = pattern;
this.patternPosition = patternPosition;
}
@Override
public Writable getLabelForPath(String path) {
// Label is in the filename
return new Text(FilenameUtils.getBaseName(path).split(pattern)[patternPosition]);
}
@Override
public Writable getLabelForPath(URI uri) {
return getLabelForPath(new File(uri).toString());
}
}