Query.cs

Go to the documentation of this file.

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

using System;
using System.Linq;
using Lucene.Net.Index;
using IndexReader = Lucene.Net.Index.IndexReader;

namespace Lucene.Net.Search
{
    
    [Serializable]
    public abstract class Query : System.ICloneable
    {
        private float boost = 1.0f; // query boost factor

        public virtual float Boost
        {
            get { return boost; }
            set { boost = value; }
        }

        public abstract System.String ToString(System.String field);
        
        public override System.String ToString()
        {
            return ToString("");
        }
        
        public virtual Weight CreateWeight(Searcher searcher)
        {
            throw new System.NotSupportedException();
        }
        
        public virtual Weight Weight(Searcher searcher)
        {
            Query query = searcher.Rewrite(this);
            Weight weight = query.CreateWeight(searcher);
            float sum = weight.GetSumOfSquaredWeights();
            float norm = GetSimilarity(searcher).QueryNorm(sum);
            if (float.IsInfinity(norm) || float.IsNaN(norm))
                norm = 1.0f;
            weight.Normalize(norm);
            return weight;
        }
        
        
        public virtual Query Rewrite(IndexReader reader)
        {
            return this;
        }
        
        
        public virtual Query Combine(Query[] queries)
        {
            var uniques = new System.Collections.Generic.HashSet<Query>();
            for (int i = 0; i < queries.Length; i++)
            {
                Query query = queries[i];
                BooleanClause[] clauses = null;
                // check if we can split the query into clauses
                bool splittable = (query is BooleanQuery);
                if (splittable)
                {
                    BooleanQuery bq = (BooleanQuery) query;
                    splittable = bq.IsCoordDisabled();
                    clauses = bq.GetClauses();
                    for (int j = 0; splittable && j < clauses.Length; j++)
                    {
                        splittable = (clauses[j].Occur == Occur.SHOULD);
                    }
                }
                if (splittable)
                {
                    for (int j = 0; j < clauses.Length; j++)
                    {
                        uniques.Add(clauses[j].Query);
                    }
                }
                else
                {
                    uniques.Add(query);
                }
            }
            // optimization: if we have just one query, just return it
            if (uniques.Count == 1)
            {
                return uniques.First();
            }
            BooleanQuery result = new BooleanQuery(true);
            foreach (Query key in uniques)
            {
                result.Add(key, Occur.SHOULD);
            }
            return result;
        }
        
        
        public virtual void ExtractTerms(System.Collections.Generic.ISet<Term> terms)
        {
            // needs to be implemented by query subclasses
            throw new System.NotSupportedException();
        }
        
        
        
        public static Query MergeBooleanQueries(params BooleanQuery[] queries)
        {
            var allClauses = new System.Collections.Generic.HashSet<BooleanClause>();
            foreach (BooleanQuery booleanQuery in queries)
            {
                foreach (BooleanClause clause in booleanQuery)
                {
                    allClauses.Add(clause);
                }
            }

            bool coordDisabled = queries.Length == 0?false:queries[0].IsCoordDisabled();
            BooleanQuery result = new BooleanQuery(coordDisabled);
            foreach(BooleanClause clause in allClauses)
            {
                result.Add(clause);
            }
            return result;
        }
        
        
        public virtual Similarity GetSimilarity(Searcher searcher)
        {
            return searcher.Similarity;
        }
        
        public virtual System.Object Clone()
        {
            try
            {
                return base.MemberwiseClone();
            }
            catch (System.Exception e)
            {
                throw new System.SystemException("Clone not supported: " + e.Message);
            }
        }
        
        public override int GetHashCode()
        {
            int prime = 31;
            int result = 1;
            result = prime * result + BitConverter.ToInt32(BitConverter.GetBytes(boost), 0);
            return result;
        }
        
        public  override bool Equals(System.Object obj)
        {
            if (this == obj)
                return true;
            if (obj == null)
                return false;
            if (GetType() != obj.GetType())
                return false;
            Query other = (Query) obj;
            if (BitConverter.ToInt32(BitConverter.GetBytes(boost), 0) != BitConverter.ToInt32(BitConverter.GetBytes(other.boost), 0))
                return false;
            return true;
        }
    }
}