| /* |
| * 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.SortedSetDocValues; |
| 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 UniqueMultiDvSlotAcc extends UniqueSlotAcc { |
| SortedSetDocValues topLevel; |
| SortedSetDocValues[] subDvs; |
| OrdinalMap ordMap; |
| LongValues toGlobal; |
| SortedSetDocValues subDv; |
| |
| public UniqueMultiDvSlotAcc(FacetContext fcontext, SchemaField field, int numSlots, HLLAgg.HLLFactory factory) throws IOException { |
| super(fcontext, field, numSlots, factory); |
| } |
| |
| @Override |
| public void resetIterators() throws IOException { |
| topLevel = FieldUtil.getSortedSetDocValues(fcontext.qcontext, field, null); |
| nTerms = (int) topLevel.getValueCount(); |
| if (topLevel instanceof MultiDocValues.MultiSortedSetDocValues) { |
| ordMap = ((MultiDocValues.MultiSortedSetDocValues) topLevel).mapping; |
| subDvs = ((MultiDocValues.MultiSortedSetDocValues) 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); |
| } else { |
| assert readerContext.ord==0 || topLevel.getValueCount() == 0; |
| subDv = topLevel; |
| } |
| } |
| |
| @Override |
| public void collect(int doc, int slotNum, IntFunction<SlotContext> slotContext) throws IOException { |
| if (subDv.advanceExact(doc)) { |
| |
| int segOrd = (int) subDv.nextOrd(); |
| assert segOrd >= 0; |
| |
| FixedBitSet bits = arr[slotNum]; |
| if (bits == null) { |
| bits = new FixedBitSet(nTerms); |
| arr[slotNum] = bits; |
| } |
| |
| do { |
| int ord = toGlobal == null ? segOrd : (int) toGlobal.get(segOrd); |
| bits.set(ord); |
| segOrd = (int) subDv.nextOrd(); |
| } while (segOrd >= 0); |
| } |
| } |
| } |