tsfile/src/main/java/org/apache/iotdb/tsfile/file/metadata/VectorTimeSeriesMetadata.java

/*
 * 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.iotdb.tsfile.file.metadata;

import org.apache.iotdb.tsfile.file.metadata.statistics.Statistics;

import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

public class VectorTimeSeriesMetadata implements ITimeSeriesMetadata {

  private final TimeseriesMetadata timeseriesMetadata;
  private final List<TimeseriesMetadata> valueTimeseriesMetadataList;

  public VectorTimeSeriesMetadata(
      TimeseriesMetadata timeseriesMetadata, List<TimeseriesMetadata> valueTimeseriesMetadataList) {
    this.timeseriesMetadata = timeseriesMetadata;
    this.valueTimeseriesMetadataList = valueTimeseriesMetadataList;
  }

  @Override
  public Statistics getStatistics() {
    return valueTimeseriesMetadataList.size() == 1
        ? valueTimeseriesMetadataList.get(0).getStatistics()
        : timeseriesMetadata.getStatistics();
  }

  @Override
  public boolean isModified() {
    return timeseriesMetadata.isModified();
  }

  @Override
  public void setModified(boolean modified) {
    timeseriesMetadata.setModified(modified);
  }

  @Override
  public boolean isSeq() {
    return timeseriesMetadata.isSeq();
  }

  @Override
  public void setSeq(boolean seq) {
    timeseriesMetadata.setSeq(seq);
  }

  @Override
  public List<IChunkMetadata> loadChunkMetadataList() throws IOException {
    if (timeseriesMetadata.getChunkMetadataLoader().isMemChunkMetadataLoader()) {
      return timeseriesMetadata.loadChunkMetadataList();
    } else {
      List<IChunkMetadata> timeChunkMetadata = timeseriesMetadata.loadChunkMetadataList();
      List<List<IChunkMetadata>> valueChunkMetadataList = new ArrayList<>();
      for (TimeseriesMetadata metadata : valueTimeseriesMetadataList) {
        valueChunkMetadataList.add(metadata.loadChunkMetadataList());
      }

      List<IChunkMetadata> res = new ArrayList<>();

      for (int i = 0; i < timeChunkMetadata.size(); i++) {
        List<IChunkMetadata> chunkMetadataList = new ArrayList<>();
        for (List<IChunkMetadata> chunkMetadata : valueChunkMetadataList) {
          chunkMetadataList.add(chunkMetadata.get(i));
        }
        res.add(new VectorChunkMetadata(timeChunkMetadata.get(i), chunkMetadataList));
      }
      return res;
    }
  }

  @Override
  public List<IChunkMetadata> getChunkMetadataList() {
    return null;
  }

  public List<TimeseriesMetadata> getValueTimeseriesMetadataList() {
    return valueTimeseriesMetadataList;
  }
}