org.apache.lucene.search.suggest.tst.TernaryTreeNode Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of lucene-suggest Show documentation

Apache Lucene (module: suggest)

There is a newer version: 10.0.0

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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.apache.lucene.search.suggest.tst;

import org.apache.lucene.util.RamUsageEstimator;

/** The class creates a TST node. */
public class TernaryTreeNode {

  /** Creates a new empty node */
  public TernaryTreeNode() {}

  /** the character stored by a node. */
  char splitchar;

  /** a reference object to the node containing character smaller than this node's character. */
  TernaryTreeNode loKid;

  /**
   * a reference object to the node containing character next to this node's character as occurring
   * in the inserted token.
   */
  TernaryTreeNode eqKid;

  /** a reference object to the node containing character higher than this node's character. */
  TernaryTreeNode hiKid;

  /**
   * used by leaf nodes to store the complete tokens to be added to suggest list while
   * auto-completing the prefix.
   */
  String token;

  Object val;

  long sizeInBytes() {
    long mem = RamUsageEstimator.shallowSizeOf(this);
    if (loKid != null) {
      mem += loKid.sizeInBytes();
    }
    if (eqKid != null) {
      mem += eqKid.sizeInBytes();
    }
    if (hiKid != null) {
      mem += hiKid.sizeInBytes();
    }
    if (token != null) {
      mem +=
          RamUsageEstimator.shallowSizeOf(token)
              + RamUsageEstimator.NUM_BYTES_ARRAY_HEADER
              + Character.BYTES * (long) token.length();
    }
    mem += RamUsageEstimator.shallowSizeOf(val);
    return mem;
  }
}