/* * 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.solr.search.facet; import java.io.IOException; import java.util.function.IntFunction; import org.apache.lucene.index.LeafReaderContext; import org.apache.lucene.index.MultiDocValues; import org.apache.lucene.index.OrdinalMap; import org.apache.lucene.index.SortedDocValues; import org.apache.lucene.util.BytesRef; import org.apache.lucene.util.FixedBitSet; import org.apache.lucene.util.LongValues; import org.apache.solr.schema.SchemaField; class UniqueSinglevaluedSlotAcc extends UniqueSlotAcc { SortedDocValues topLevel; SortedDocValues[] subDvs; OrdinalMap ordMap; LongValues toGlobal; SortedDocValues subDv; public UniqueSinglevaluedSlotAcc(FacetContext fcontext, SchemaField field, int numSlots, HLLAgg.HLLFactory factory) throws IOException { super(fcontext, field, numSlots, factory); } @Override public void resetIterators() throws IOException { super.resetIterators(); topLevel = FieldUtil.getSortedDocValues(fcontext.qcontext, field, null); nTerms = topLevel.getValueCount(); if (topLevel instanceof MultiDocValues.MultiSortedDocValues) { ordMap = ((MultiDocValues.MultiSortedDocValues)topLevel).mapping; subDvs = ((MultiDocValues.MultiSortedDocValues)topLevel).values; } else { ordMap = null; subDvs = null; } } @Override protected BytesRef lookupOrd(int ord) throws IOException { return topLevel.lookupOrd(ord); } @Override public void setNextReader(LeafReaderContext readerContext) throws IOException { super.setNextReader(readerContext); if (subDvs != null) { subDv = subDvs[readerContext.ord]; toGlobal = ordMap.getGlobalOrds(readerContext.ord); assert toGlobal != null; } else { assert readerContext.ord==0 || topLevel.getValueCount() == 0; subDv = topLevel; } assert subDv.docID() == -1; // make sure we haven't used this iterator before } @Override public void collect(int doc, int slotNum, IntFunction<SlotContext> slotContext) throws IOException { if (subDv.advanceExact(doc)) { int segOrd = subDv.ordValue(); int ord = toGlobal==null ? segOrd : (int)toGlobal.get(segOrd); collectOrdToSlot(slotNum, ord); } } protected void collectOrdToSlot(int slotNum, int ord) { FixedBitSet bits = arr[slotNum]; if (bits == null) { bits = new FixedBitSet(nTerms); arr[slotNum] = bits; } bits.set(ord); } }