rrenkert@8251: package org.dive4elements.river.exports.injector;
rrenkert@8251: 
rrenkert@8251: import org.apache.log4j.Logger;
rrenkert@8251: import org.dive4elements.artifacts.Artifact;
rrenkert@8251: import org.dive4elements.artifacts.CallContext;
rrenkert@8251: import org.dive4elements.artifacts.ContextInjector;
rrenkert@8251: import org.dive4elements.river.artifacts.D4EArtifact;
rrenkert@8251: import org.dive4elements.river.artifacts.access.RangeAccess;
rrenkert@8251: import org.w3c.dom.Document;
rrenkert@8251: import org.w3c.dom.Element;
rrenkert@8251: 
rrenkert@8251: import static org.dive4elements.river.exports.injector.InjectorConstants.CURRENT_KM;
rrenkert@8251: 
rrenkert@8251: public class ComputedDischargeInjector
rrenkert@8251: implements ContextInjector
rrenkert@8251: {
rrenkert@8251: 
rrenkert@8251:     private static Logger log = Logger.getLogger(ComputedDischargeInjector.class);
rrenkert@8251: 
rrenkert@8251:     @Override
rrenkert@8251:     public void setup(Element cfg) {
rrenkert@8251:     }
rrenkert@8251: 
rrenkert@8251:     @Override
rrenkert@8251:     public void injectContext(
rrenkert@8251:         CallContext ctx,
rrenkert@8251:         Artifact artifact,
rrenkert@8251:         Document doc
rrenkert@8251:     ) {
rrenkert@8251:         RangeAccess access = new RangeAccess((D4EArtifact)artifact);
tom@8310:         if (access.hasFrom()) {
rrenkert@8326:             ctx.putContextValue(CURRENT_KM, access.getFrom());
rrenkert@8326:         }
rrenkert@8326:         else if (access.getLocations() != null &&
rrenkert@8326:             access.getLocations().length > 0) {
rrenkert@8326:             ctx.putContextValue(CURRENT_KM, access.getLocations()[0]);
tom@8310:         }
rrenkert@8251:     }
rrenkert@8251: }