net.sf.okapi.virtualdb.jdbc.h2.H2Document Maven / Gradle / Ivy

Go to download
/*===========================================================================
  Copyright (C) 2010 by the Okapi Framework contributors
-----------------------------------------------------------------------------
  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 net.sf.okapi.virtualdb.jdbc.h2;

import net.sf.okapi.common.resource.Ending;
import net.sf.okapi.common.resource.StartDocument;
import net.sf.okapi.virtualdb.IVDocument;
import net.sf.okapi.virtualdb.IVItem;
import net.sf.okapi.virtualdb.IVTextUnit;

public class H2Document extends H2Navigator implements IVDocument {

	H2Access db;
	private String id;
	private String name;
	private String type;
	private H2Document selfDoc;
	
	public H2Document (H2Access access,
		long key,
		String id,
		String name,
		String type)
	{
		db = access;
		this.key = key;
		this.id = id;
		this.name = name;
		this.type = type;
		selfDoc = this;
	}
	
	public void fillPointers (long parent,
		long firstChild,
		long previous,
		long next)
	{
		this.parent = parent;
		this.firstChild = firstChild;
		this.previous = previous;
		this.next = next;
	}

	@Override
	public String getId () {
		return id;
	}

	@Override
	public Ending getEndDocument () {
		throw new UnsupportedOperationException("getEndDocument");
	}

	@Override
	public IVItem getItem (String extractionId) {
		return db.getItemFromExtractionId(this, extractionId);
	}

	@Override
	public IVItem getItem (long key) {
		return db.getItemFromItemKey(this, key);
	}

	@Override
	public StartDocument getStartDocument () {
		throw new UnsupportedOperationException("getStartDocument");
	}

	@Override
	public IVTextUnit getTextUnit (String extractionId) {
		IVItem item = db.getItemFromExtractionId(this, extractionId);
		if ( item instanceof IVTextUnit ) {
			return (IVTextUnit)item;
		}
		return null;
	}

	@Override
	public IVItem getFirstChild () {
		return db.getItemFromItemKey(this, firstChild);
	}

	@Override
	public ItemType getItemType () {
		return IVItem.ItemType.DOCUMENT;
	}

	@Override
	public Iterable items () {
		return () -> new H2ItemIterator<>(db, selfDoc, false);
	}
	
	@Override
	public Iterable textUnits () {
		return () -> new H2ItemIterator<>(db, selfDoc, true);
	}

	@Override
	public IVDocument getDocument () {
		return this;
	}

	@Override
	public String getName () {
		return name;
	}

	@Override
	public IVItem getNextSibling () {
		return db.getItemFromItemKey(this, next);
	}

	@Override
	public IVItem getParent () {
		// The parent of a document is always null
		return null;
	}

	@Override
	public IVItem getPreviousSibling () {
		return db.getItemFromItemKey(this, previous);
	}

	@Override
	public String getType () {
		return type;
	}

	@Override
	public void save () {
		// No modifiable data to save
	}

}