86 lines
2.8 KiB
Java
86 lines
2.8 KiB
Java
|
/*
|
||
|
|
* Copyright 2015 Alexander Nozik.
|
||
|
|
*
|
||
|
|
* 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 inr.numass.actions;
|
||
|
|
|
||
|
|
import hep.dataforge.actions.OneToOneAction;
|
||
|
|
import hep.dataforge.context.Context;
|
||
|
|
import hep.dataforge.description.TypedActionDef;
|
||
|
|
import hep.dataforge.exceptions.ContentException;
|
||
|
|
import hep.dataforge.io.ColumnedDataWriter;
|
||
|
|
import hep.dataforge.io.log.Logable;
|
||
|
|
import hep.dataforge.meta.Meta;
|
||
|
|
import inr.numass.data.NMFile;
|
||
|
|
import inr.numass.data.RawNMPoint;
|
||
|
|
import java.io.OutputStream;
|
||
|
|
import java.util.LinkedHashMap;
|
||
|
|
import java.util.List;
|
||
|
|
import java.util.Map;
|
||
|
|
import org.apache.commons.math3.util.Pair;
|
||
|
|
|
||
|
|
/**
|
||
|
|
*
|
||
|
|
* @author Darksnake
|
||
|
|
*/
|
||
|
|
@TypedActionDef(name = "slicing", inputType = NMFile.class, outputType = NMFile.class)
|
||
|
|
public class SlicingAction extends OneToOneAction<NMFile, NMFile> {
|
||
|
|
|
||
|
|
public static final String name = "slicing";
|
||
|
|
|
||
|
|
public SlicingAction(Context context, Meta an) {
|
||
|
|
super(context, an);
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override
|
||
|
|
public String getName() {
|
||
|
|
return name;
|
||
|
|
}
|
||
|
|
|
||
|
|
@Override
|
||
|
|
protected NMFile execute(Logable log, Meta reader, NMFile source) throws ContentException {
|
||
|
|
boolean normalize;
|
||
|
|
Map<String, Pair<Integer, Integer>> slicingConfig;
|
||
|
|
|
||
|
|
LinkedHashMap<String, Pair<Integer, Integer>> res = new LinkedHashMap<>();
|
||
|
|
List<? extends Meta> list = meta().getNode("sliceconfig").getNodes("slicepoint");
|
||
|
|
|
||
|
|
for (Meta slice : list) {
|
||
|
|
String title = slice.getString("title", slice.getName());
|
||
|
|
int from = slice.getInt("from", 0);
|
||
|
|
int to = slice.getInt("to", RawNMPoint.MAX_CHANEL);
|
||
|
|
res.put(title, new Pair<>(from, to));
|
||
|
|
}
|
||
|
|
slicingConfig = res;
|
||
|
|
|
||
|
|
normalize = meta().getBoolean("normalize", false);
|
||
|
|
|
||
|
|
if (slicingConfig == null) {
|
||
|
|
throw new RuntimeException("Slice configuration not defined");
|
||
|
|
}
|
||
|
|
log.log("File {} started", source.getName());
|
||
|
|
|
||
|
|
SlicedData sData = new SlicedData(source, slicingConfig, normalize);
|
||
|
|
|
||
|
|
OutputStream stream = buildActionOutput(source);
|
||
|
|
|
||
|
|
ColumnedDataWriter.writeDataSet(stream, sData, null);
|
||
|
|
|
||
|
|
log.log("File {} completed", source.getName());
|
||
|
|
|
||
|
|
return source;
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|