exec.validate_evaluation.streaks.EditStreakGenerationIo.java Source code

Introduction

Here is the source code for exec.validate_evaluation.streaks.EditStreakGenerationIo.java

Source

/**
 * Copyright 2016 Technische Universitt Darmstadt
 * 
 * Licensed 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 exec.validate_evaluation.streaks;

import java.io.IOException;
import java.util.Set;

import com.google.common.collect.Sets;

import cc.kave.commons.model.events.completionevents.CompletionEvent;
import cc.kave.commons.model.events.completionevents.ICompletionEvent;
import cc.kave.commons.model.ssts.impl.visitor.TypeErasureVisitor;
import cc.recommenders.io.Directory;
import cc.recommenders.io.IReadingArchive;
import cc.recommenders.io.WritingArchive;

public class EditStreakGenerationIo {

    private String dirIn;
    private String dirOut;

    public EditStreakGenerationIo(String dirIn, String dirOut) {
        this.dirIn = dirIn;
        this.dirOut = dirOut;
    }

    public Set<String> findCompletionEventZips() {
        Directory dir = new Directory(dirIn);
        return dir.findFiles(s -> s.endsWith(".zip"));
    }

    public Set<ICompletionEvent> readCompletionEvents(String zip) {
        Set<ICompletionEvent> es = Sets.newLinkedHashSet();
        Directory dir = new Directory(dirIn);
        try (IReadingArchive ra = dir.getReadingArchive(zip)) {
            while (ra.hasNext()) {
                CompletionEvent ce = ra.getNext(CompletionEvent.class);
                ce.context = TypeErasureVisitor.erase(ce.context);
                es.add(ce);
            }
        } catch (IOException e) {
            throw new RuntimeException(e);
        }

        return es;
    }

    public void storeEditStreaks(Set<EditStreak> streaks, String zip) {
        if (streaks.isEmpty()) {
            return;
        }
        Directory dir = new Directory(dirOut);
        try (WritingArchive wa = dir.getWritingArchive(zip)) {
            for (EditStreak es : streaks) {
                wa.add(es);
            }
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    public Set<String> findEditStreakZips() {
        Directory dir = new Directory(dirOut);
        return dir.findFiles(s -> s.endsWith(".zip"));
    }

    public Set<EditStreak> readEditStreaks(String zip) {
        Set<EditStreak> streaks = Sets.newLinkedHashSet();
        Directory dir = new Directory(dirOut);
        try (IReadingArchive ra = dir.getReadingArchive(zip)) {
            while (ra.hasNext()) {
                EditStreak es = ra.getNext(EditStreak.class);
                streaks.add(es);
            }
        } catch (IOException e) {
            throw new RuntimeException(e);
        }

        return streaks;
    }
}