org.noroomattheinn.visibletesla.data.RestStore.java Source code

Introduction

Here is the source code for org.noroomattheinn.visibletesla.data.RestStore.java

Source

/*
 * RestStore.java - Copyright(c) 2013, 2014 Joe Pasqua
 * Provided under the MIT License. See the LICENSE file for details.
 * Created: Oct 23, 2014
 */
package org.noroomattheinn.visibletesla.data;

import com.google.common.collect.Range;
import java.io.File;
import java.io.FileNotFoundException;
import java.util.Date;
import javafx.beans.property.BooleanProperty;
import javafx.beans.property.DoubleProperty;
import jxl.write.WritableSheet;
import jxl.write.WriteException;
import org.apache.commons.lang3.StringUtils;
import static org.noroomattheinn.tesla.Tesla.logger;
import org.noroomattheinn.timeseries.Row;
import org.noroomattheinn.timeseries.TimeSeries;
import org.noroomattheinn.timeseries.TimeSeries.RowCollector;
import org.noroomattheinn.utils.TrackedObject;
import org.noroomattheinn.visibletesla.vehicle.VTVehicle;

/**
 * ChargeStore: Manage persistent storage for Charge Cycle information.
 * 
 * @author Joe Pasqua <joe at NoRoomAtTheInn dot org>
 */
class RestStore extends CycleStore<RestCycle> {
    /*------------------------------------------------------------------------------
     *
     * Internal  State
     * 
     *----------------------------------------------------------------------------*/

    private final RestCycleExporter exporter;
    private final TrackedObject<RestCycle> lastRestCycle;
    private boolean doingInitialLoad;

    /*==============================================================================
     * -------                                                               -------
     * -------              Public Interface To This Class                   ------- 
     * -------                                                               -------
     *============================================================================*/

    RestStore(File container, VTVehicle v, TrackedObject<RestCycle> lastCycle, BooleanProperty submitRests,
            BooleanProperty includeLoc, DoubleProperty ditherAmt) throws FileNotFoundException {
        super("rest", RestCycle.class, container, v);
        this.lastRestCycle = lastCycle;
        this.doingInitialLoad = false;
        this.exporter = new RestCycleExporter(submitRests, includeLoc, ditherAmt, v.getVehicle().getUUID());

        lastRestCycle.addTracker(new Runnable() {
            @Override
            public void run() {
                cycleWriter.println(lastRestCycle.get().toJSONString());
                if (!doingInitialLoad) {
                    exporter.submitData(lastRestCycle.get());
                    cycleWriter.flush();
                }
            }
        });
    }

    boolean export(File toFile, Range<Long> exportPeriod) {
        return exporter.export(this, toFile, exportPeriod);
    }

    /*------------------------------------------------------------------------------
     *
     * Methods related to creating the RestStore the first time. If necessary, we
     * paw through old data from the StatsCollector to initialize the RestStore.
     * We do this at most once in the lifetime of an installation of VT
     * 
     *----------------------------------------------------------------------------*/

    static boolean requiresInitialLoad(File container, String baseName) {
        File f = new File(container, baseName + ".rest.json");
        return !f.exists();
    }

    void doIntialLoad(final RestMonitor rm, TimeSeries ts) {
        // Create a rest file based on existing data. This is a one time thing.
        logger.info("Synthesizing RestCycle data - one time only");
        try {
            doingInitialLoad = true;
            ts.streamRows(null, new RowCollector() {
                @Override
                public boolean collect(Row r) {
                    rm.handleNewData(r);
                    return true;
                }
            });
        } catch (Exception e) {
            logger.warning("Error during intial load of Rest Cycles: " + e);
        }
        doingInitialLoad = false;
        cycleWriter.flush();
    }

}

/*------------------------------------------------------------------------------
 *
 * The CycleExporter for Charge Cycles
 * 
 *----------------------------------------------------------------------------*/

class RestCycleExporter extends CycleExporter<RestCycle> {
    private static final String[] labels = { "Start Date/Time", "Ending Date/Time", "Start Range", "End Range",
            "Start SOC", "End SOC", "(Latitude, ", " Longitude)", "Loss/Hr" };
    private final String uuid;

    RestCycleExporter(BooleanProperty submitRest, BooleanProperty includeLoc, DoubleProperty ditherAmt,
            String uuid) {
        super("Rest", labels, submitRest, includeLoc, ditherAmt);
        this.uuid = uuid;
    }

    @Override
    protected void emitRow(WritableSheet sheet, int row, RestCycle cycle, StandardFormats sf)
            throws WriteException {
        int column = 0;
        sheet.addCell(new jxl.write.DateTime(column++, row, new Date(cycle.startTime), sf.dateFormat));
        sheet.addCell(new jxl.write.DateTime(column++, row, new Date(cycle.endTime), sf.dateFormat));
        sheet.addCell(new jxl.write.Number(column++, row, cycle.startRange, sf.standardFormat));
        sheet.addCell(new jxl.write.Number(column++, row, cycle.endRange, sf.standardFormat));
        sheet.addCell(new jxl.write.Number(column++, row, cycle.startSOC, sf.standardFormat));
        sheet.addCell(new jxl.write.Number(column++, row, cycle.endSOC, sf.standardFormat));
        sheet.addCell(new jxl.write.Number(column++, row, cycle.lat, sf.standardFormat));
        sheet.addCell(new jxl.write.Number(column++, row, cycle.lng, sf.standardFormat));
        String lossFormula = String.format("(C%1$d-D%1$d)/((B%1$d-A%1$d)*24)", row + 1);
        jxl.write.Formula f = new jxl.write.Formula(column, row, lossFormula, sf.standardFormat);
        sheet.addCell(f);
    }

    @Override
    protected String filterSubmissionData(String jsonRep) {
        // Strip the closing curly to prepare to add more fields
        jsonRep = StringUtils.substringBefore(jsonRep, "}");
        // Concatenate the extra fields and put back the closing curly
        return String.format("%s, \"uuid\": \"%s\" }", jsonRep, uuid);
    }
}