package org.apache.lucene.facet.example.association;

import org.apache.lucene.facet.enhancements.association.AssociationEnhancement;
import org.apache.lucene.facet.enhancements.association.AssociationFloatProperty;
import org.apache.lucene.facet.enhancements.association.AssociationIntProperty;
import org.apache.lucene.facet.enhancements.association.AssociationProperty;
import org.apache.lucene.facet.enhancements.params.DefaultEnhancementsIndexingParams;
import org.apache.lucene.facet.taxonomy.CategoryPath;

/*
 * 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.
 */

/**
 * @lucene.experimental
 */
public class AssociationUtils {

  /**
   * Categories: categories[D][N] == category-path with association no. N for
   * document no. D.
   */
  public static CategoryPath[][] categories = {
    // Doc #1
    { new CategoryPath("tags", "lucene") , 
      new CategoryPath("genre", "computing")
    },
        
    // Doc #2
    { new CategoryPath("tags", "lucene"),  
      new CategoryPath("tags", "solr"),
      new CategoryPath("genre", "computing"),
      new CategoryPath("genre", "software")
    }
  };

  public static AssociationProperty[][] associations = {
    // Doc #1 associations
    {
      /* 3 occurrences for tag 'lucene' */
      new AssociationIntProperty(3), 
      /* 87% confidence level of genre 'computing' */
      new AssociationFloatProperty(0.87f)
    },
    
    // Doc #2 associations
    {
      /* 1 occurrence for tag 'lucene' */
      new AssociationIntProperty(1),
      /* 2 occurrences for tag 'solr' */
      new AssociationIntProperty(2),
      /* 75% confidence level of genre 'computing' */
      new AssociationFloatProperty(0.75f),
      /* 34% confidence level of genre 'software' */
      new AssociationFloatProperty(0.34f),
    }
  };

  /**
   * Indexing Params: the indexing params to use when dealing with
   * associations.
   */
  public static final DefaultEnhancementsIndexingParams assocIndexingParams = 
    new DefaultEnhancementsIndexingParams(new AssociationEnhancement());

}