All Downloads are FREE. Search and download functionalities are using the official Maven repository.

it.tidalwave.semantic.io.impl.GraphUnmarshallerImpl Maven / Gradle / Ivy

There is a newer version: 1.0.16
Show newest version
/***********************************************************************************************************************
 *
 * blueBill Mobile - open source birdwatching
 * ==========================================
 *
 * Copyright (C) 2009, 2010 by Tidalwave s.a.s. (http://www.tidalwave.it)
 * http://bluebill.tidalwave.it/mobile/
 *
 ***********************************************************************************************************************
 *
 * 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.
 *
 ***********************************************************************************************************************
 *
 * $Id: GraphUnmarshallerImpl.java,v 7b104dad6889 2010/07/15 22:40:04 fabrizio $
 *
 **********************************************************************************************************************/
package it.tidalwave.semantic.io.impl;

import javax.annotation.Nonnull;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.SortedMap;
import java.util.Stack;
import java.util.TreeMap;
import it.tidalwave.util.As;
import it.tidalwave.util.Id;
import it.tidalwave.util.NotFoundException;
import it.tidalwave.semantic.io.GraphUnmarshaller;
import org.openrdf.model.Statement;
import it.tidalwave.semantic.io.spi.StatementUnmarshaller;
import it.tidalwave.semantic.io.spi.StatementUnmarshallerFactory;
import javax.annotation.concurrent.NotThreadSafe;
import org.openide.util.Lookup;
import org.openrdf.model.Graph;
import org.openrdf.model.Literal;
import org.openrdf.model.URI;

/***********************************************************************************************************************
 *
 * @author  Fabrizio Giudici
 * @version $Id: $
 *
 **********************************************************************************************************************/
@NotThreadSafe
public class GraphUnmarshallerImpl implements GraphUnmarshaller, StatementUnmarshaller.Context
  {
    private Graph graph;

    private final Map entities = new HashMap();

    private SortedMap> statementMapBySubject;

    private final Stack stack = new Stack();

    private final List statementUnmarshallerFactories = new ArrayList();

    /*******************************************************************************************************************
     *
     * {@inheritDoc}
     *
     ******************************************************************************************************************/
    @Nonnull
    public As unmarshal (final @Nonnull Graph graph, final @Nonnull Id type)
      {
        this.graph = graph;

        statementUnmarshallerFactories.addAll(Lookup.getDefault().lookupAll(StatementUnmarshallerFactory.class));
        statementMapBySubject = groupedBySubject(graph);

        for (final Statement statement : findStatements(Vocabulary.RDF_TYPE, type))
          {
            return (As)find(statement.getSubject());
          }

        throw new RuntimeException("Can't find an object of type " + type);
      }

    /*******************************************************************************************************************
     *
     * {@inheritDoc}
     *
     ******************************************************************************************************************/
    public void registerStatementUnmarshallerFactory (final @Nonnull StatementUnmarshallerFactory factory)
      {
        statementUnmarshallerFactories.add(factory);
      }

    /*******************************************************************************************************************
     *
     *
     ******************************************************************************************************************/
    @Nonnull
    private As internalUnmarshal (final @Nonnull Id id)
      {
        final List subjectStatements = statementMapBySubject.get(id);

        if (subjectStatements == null)
          {
            throw new IllegalArgumentException("Wrong id: " + id + " - " + statementMapBySubject.keySet());
          }

        final StatementUnmarshaller entityFactory = findEntityFactory(subjectStatements);
        final As result = entityFactory.unmarshal(subjectStatements, this);
        entities.put(id, result);

        return result;
      }

    /*******************************************************************************************************************
     *
     *
     ******************************************************************************************************************/
    @Nonnull
    public Object find (final @Nonnull Object object)
      {
        Object result = null;
        Id id = null;

        if (object instanceof Id)
          {
            id = (Id)object;
          }
        if (object instanceof URI)
          {
            id = Converter.valueToId((URI)object);
          }
        else
          {
            throw new IllegalArgumentException("Only accept an Id or an URI: " + object);
          }

        result = entities.get(id);

        if (result == null)
          {
            result = internalUnmarshal(id);
          }

        return result;
      }

    /*******************************************************************************************************************
     *
     *
     ******************************************************************************************************************/
    @Nonnull
    private StatementUnmarshaller findEntityFactory (final @Nonnull List statements)
      {
        for (final StatementUnmarshallerFactory factory : statementUnmarshallerFactories)
          {
            try
              {
                return factory.findStatementUnmarshallerFor(statements);
              }
            catch (NotFoundException e)
              {
                // ok, try next
              }
          }

        throw new RuntimeException("Can't find a StatementUnmarshallerFactory for " + statements);
      }

    /*******************************************************************************************************************
     *
     *
     ******************************************************************************************************************/
    @Nonnull
    private static SortedMap> groupedBySubject (final @Nonnull Collection statements)
      {
        final SortedMap> result = new TreeMap>();

        for (final Statement statement : statements)
          {
            final Id subject = Converter.valueToId(statement.getSubject());
            List subStatements = result.get(subject);

            if (subStatements == null)
              {
                subStatements = new ArrayList();
                result.put(subject, subStatements);
              }

            subStatements.add(statement);
          }

        return result;
      }

    /*******************************************************************************************************************
     *
     *
     ******************************************************************************************************************/
    public void push (final @Nonnull Object object)
      {
        stack.push(object);
      }

    /*******************************************************************************************************************
     *
     *
     ******************************************************************************************************************/
    public void pop()
      {
        stack.pop();
      }

    /*******************************************************************************************************************
     *
     *
     ******************************************************************************************************************/
    @Nonnull
    public Object peek()
      {
        return stack.peek();
      }

    /*******************************************************************************************************************
     *
     *
     ******************************************************************************************************************/
    @Nonnull
    public Collection findStatements (final @Nonnull Id predicate, final @Nonnull Id objectId)
      {
        final List result = new ArrayList();

        for (final Statement statement : graph)
          {
            if (statement.getPredicate().stringValue().equals(predicate.stringValue()))
              {
                if (matches(objectId, statement.getObject()))
                  {
                    result.add(statement);
                  }
              }
          }

        return result;
      }

    /*******************************************************************************************************************
     *
     *
     ******************************************************************************************************************/
    public boolean matches (final @Nonnull Id objectId, final @Nonnull Object object)
      {
        if (objectId == null)
          {
            return true;
          }

        if (object instanceof URI)
          {
            if (((URI)object).stringValue().equals(objectId.stringValue()))
              {
                return true;
              }
          }
        else if (object instanceof Literal)
          {
            if (((Literal)object).stringValue().equals(objectId.stringValue()))
              {
                return true;
              }
          }

        return false;
      }
  }