/*
 Copyright (C) 2019-2020 Fredrik Öhrström

 This program is free software: you can redistribute it and/or modify
 it under the terms of the GNU General Public License as published by
 the Free Software Foundation, either version 3 of the License, or
 (at your option) any later version.

 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/

#include"dvparser.h"
#include"meters.h"
#include"meters_common_implementation.h"
#include"wmbus.h"
#include"wmbus_utils.h"

using namespace std;

struct MeterHydrus : public virtual WaterMeter, public virtual MeterCommonImplementation {
    MeterHydrus(MeterInfo &mi);

    // Total water counted through the meter
    double totalWaterConsumption(Unit u);
    double totalWaterConsumptionAtDate(Unit u);
    bool  hasTotalWaterConsumption();
    double maxFlow(Unit u);
    bool  hasMaxFlow();
    double flowTemperature(Unit u);

private:
    void processContent(Telegram *t);

    double total_water_consumption_m3_ {};
    double total_water_consumption_at_date_m3_ {};
    string at_date_;
    double max_flow_m3h_ {};
    double flow_temperature_c_ { 127 };
    double remaining_battery_life_year_;
    string status_; // TPL STS

    map<int,string> error_codes_;
};

MeterHydrus::MeterHydrus(MeterInfo &mi) :
    MeterCommonImplementation(mi, MeterType::HYDRUS)
{
    setExpectedTPLSecurityMode(TPLSecurityMode::AES_CBC_IV);

    addLinkMode(LinkMode::T1);

    error_codes_ = { { 0x10, "TEMPERATURE_MEASUREMENT_ERROR" },
                     { 0x30, "AIR_IN_PIPE" },
                     { 0x70, "MEASUREMENT_ERROR" },
                     { 0x90, "LEAKAGE_OR_NO_USAGE" },
                     { 0xb0, "REVERSE_FLOW" },
                     { 0xd0, "LOW_TEMPERATURE" },
                     { 0xf0, "AIR_IN_PIPE" } };

    addPrint("total", Quantity::Volume,
             [&](Unit u){ return totalWaterConsumption(u); },
             "The total water consumption recorded by this meter.",
             true, true);

    addPrint("max_flow", Quantity::Flow,
             [&](Unit u){ return maxFlow(u); },
             "The maxium flow recorded during previous period.",
             true, true);

    addPrint("flow_temperature", Quantity::Temperature,
             [&](Unit u){ return flowTemperature(u); },
             "The water temperature.",
             false, true);

    addPrint("total_at_date", Quantity::Volume,
             [&](Unit u){ return totalWaterConsumptionAtDate(u); },
             "The total water consumption recorded at date.",
             false, true);

    addPrint("at_date", Quantity::Text,
             [&](){ return at_date_; },
             "Date when total water consumption was recorded.",
             false, true);

    addPrint("remaining_battery_life", Quantity::Time, Unit::Year,
             [&](Unit u){ return convert(remaining_battery_life_year_, Unit::Year, u); },
             "How many more years the battery is expected to last",
             false, true);

    addPrint("status", Quantity::Text,
             [&](){ return status_; },
             "The status is OK or some error condition.",
             true, true);

}

shared_ptr<WaterMeter> createHydrus(MeterInfo &mi)
{
    return shared_ptr<WaterMeter>(new MeterHydrus(mi));
}

void MeterHydrus::processContent(Telegram *t)
{
    // There are two distinctintly different Hydrus telegrams.
    // Unfortunately there seems to be no markings on the
    // physical meter that tells us which version it is.
    //
    // Fortunately the mfct,media,version bits does distinguish
    // the meters!
    //
    // This driver currently only decodes parts of both telegrams.
    // Eventually we should either split this driver into two.
    // Or make more generic capabilities in drivers to switch
    // between different telegrams formats, that are similar
    // but not quite.

    /*
     New style telegram:
     (hydrus) 11: 0C dif (8 digit BCD Instantaneous value)
     (hydrus) 12: 13 vif (Volume l)
     (hydrus) 13: * 50340000 total consumption (3.450000 m3)
     (hydrus) 17: 0D dif (variable length Instantaneous value)
     (hydrus) 18: FD vif (Second extension of VIF-codes)
     (hydrus) 19: 11 vife (Customer)
     (hydrus) 1a: 0A varlen=10
     (hydrus) 1b: 38373130313442303241
     (hydrus) 25: 0B dif (6 digit BCD Instantaneous value)
     (hydrus) 26: 3B vif (Volume flow l/h)
     (hydrus) 27: * 000000 max flow (0.000000 m3/h)
     (hydrus) 2a: 02 dif (16 Bit Integer/Binary Instantaneous value)
     (hydrus) 2b: FD vif (Second extension of VIF-codes)
     (hydrus) 2c: 74 vife (Reserved)
     (hydrus) 2d: * DC15 battery life days (5596)
     (hydrus) 2f: C4 dif (32 Bit Integer/Binary Instantaneous value storagenr=1)
     (hydrus) 30: 01 dife (subunit=0 tariff=0 storagenr=3)
     (hydrus) 31: 6D vif (Date and time type)
     (hydrus) 32: * 3B178D29 at date (2020-09-13 23:59)
     (hydrus) 36: CC dif (8 digit BCD Instantaneous value storagenr=1)
     (hydrus) 37: 01 dife (subunit=0 tariff=0 storagenr=3)
     (hydrus) 38: 13 vif (Volume l)
     (hydrus) 39: * 31340000 total consumption at date (3.431000 m3)
     (hydrus) 3d: 2F skip
     (hydrus) 3e: 2F skip

     {"media":"warm water","meter":"hydrus","name":"Vatten","id":"65656565","total_m3":3.45,"max_flow_m3h":0,"flow_temperature_c":127,"total_at_date_m3":3.431,"at_date":"2020-09-13 23:59","battery_life_days":"5596","status":"OK","timestamp":"2020-09-27T08:54:42Z"}

     Old style telegram

     (hydrus) 11: 01 dif (8 Bit Integer/Binary Instantaneous value)
     (hydrus) 12: FD vif (Second extension of VIF-codes)
     (hydrus) 13: 08 vife (Access Number (transmission count))
     (hydrus) 14: 30
     (hydrus) 15: 0C dif (8 digit BCD Instantaneous value)
     (hydrus) 16: 13 vif (Volume l)
     (hydrus) 17: * 74110000 total consumption (1.174000 m3)
     (hydrus) 1b: 7C dif (8 digit BCD Value during error state storagenr=1)
     (hydrus) 1c: 13 vif (Volume l)
     (hydrus) 1d: 00000000
     (hydrus) 21: FC dif (8 digit BCD Value during error state storagenr=1)
     (hydrus) 22: 10 dife (subunit=0 tariff=1 storagenr=1)
     (hydrus) 23: 13 vif (Volume l)
     (hydrus) 24: 00000000
     (hydrus) 28: FC dif (8 digit BCD Value during error state storagenr=1)
     (hydrus) 29: 20 dife (subunit=0 tariff=2 storagenr=1)
     (hydrus) 2a: 13 vif (Volume l)
     (hydrus) 2b: 00000000
     (hydrus) 2f: 72 dif (16 Bit Integer/Binary Value during error state storagenr=1)
     (hydrus) 30: 6C vif (Date type G)
     (hydrus) 31: 0000
     (hydrus) 33: 0B dif (6 digit BCD Instantaneous value)
     (hydrus) 34: 3B vif (Volume flow l/h)
     (hydrus) 35: * 000000 max flow (0.000000 m3/h)
     (hydrus) 38: 02 dif (16 Bit Integer/Binary Instantaneous value)
     (hydrus) 39: FD vif (Second extension of VIF-codes)
     (hydrus) 3a: 74 vife (Reserved)
     (hydrus) 3b: * 8713 battery life days (4999)
     (hydrus) 3d: 02 dif (16 Bit Integer/Binary Instantaneous value)
     (hydrus) 3e: 5A vif (Flow temperature 10⁻¹ °C)
     (hydrus) 3f: * 6800 flow temperature (10.400000 °C)
     (hydrus) 41: C4 dif (32 Bit Integer/Binary Instantaneous value storagenr=1)
     (hydrus) 42: 01 dife (subunit=0 tariff=0 storagenr=3)
     (hydrus) 43: 6D vif (Date and time type)
     (hydrus) 44: * 3B177F2A at date (2019-10-31 23:59)
     (hydrus) 48: CC dif (8 digit BCD Instantaneous value storagenr=1)
     (hydrus) 49: 01 dife (subunit=0 tariff=0 storagenr=3)
     (hydrus) 4a: 13 vif (Volume l)
     (hydrus) 4b: * 00020000 total consumption at date (0.200000 m3)

     {"media":"water","meter":"hydrus","name":"Votten","id":"64646464","total_m3":1.174,"max_flow_m3h":0,"flow_temperature_c":10.4,"total_at_date_m3":0.2,"at_date":"2019-10-31 23:59","battery_life_days":"4999","status":"OK","timestamp":"2020-09-27T08:54:42Z"}

    */
    int offset;
    string key;

    if (findKey(MeasurementType::Unknown, ValueInformation::Volume, 0, 0, &key, &t->values)) {
        extractDVdouble(&t->values, key, &offset, &total_water_consumption_m3_);
        t->addMoreExplanation(offset, " total consumption (%f m3)", total_water_consumption_m3_);
    }

    if (findKey(MeasurementType::Unknown, ValueInformation::VolumeFlow, 0, 0, &key, &t->values)) {
        extractDVdouble(&t->values, key, &offset, &max_flow_m3h_);
        t->addMoreExplanation(offset, " max flow (%f m3/h)", max_flow_m3h_);
    }

    if (findKey(MeasurementType::Unknown, ValueInformation::FlowTemperature, 0, 0, &key, &t->values)) {
        extractDVdouble(&t->values, key, &offset, &flow_temperature_c_);
        t->addMoreExplanation(offset, " flow temperature (%f °C)", flow_temperature_c_);
    }

    if (findKey(MeasurementType::Unknown, ValueInformation::Volume, 3, 0, &key, &t->values)) {
        extractDVdouble(&t->values, key, &offset, &total_water_consumption_at_date_m3_);
        t->addMoreExplanation(offset, " total consumption at date (%f m3)", total_water_consumption_at_date_m3_);
    }

    if (findKey(MeasurementType::Unknown, ValueInformation::DateTime, 3, 0, &key, &t->values)) {
        struct tm datetime;
        extractDVdate(&t->values, key, &offset, &datetime);
        at_date_ = strdatetime(&datetime);
        t->addMoreExplanation(offset, " at date (%s)", at_date_.c_str());
    }

    uint16_t days {};
    extractDVuint16(&t->values, "02FD74", &offset, &days);
    remaining_battery_life_year_ = ((double)days) / 365.25;
    t->addMoreExplanation(offset, " battery life (%d days %f years)", days, remaining_battery_life_year_);

    status_ = decodeTPLStatusByte(t->tpl_sts, error_codes_);
}

double MeterHydrus::totalWaterConsumption(Unit u)
{
    assertQuantity(u, Quantity::Volume);
    return convert(total_water_consumption_m3_, Unit::M3, u);
}

double MeterHydrus::totalWaterConsumptionAtDate(Unit u)
{
    assertQuantity(u, Quantity::Volume);
    return convert(total_water_consumption_at_date_m3_, Unit::M3, u);
}

bool MeterHydrus::hasTotalWaterConsumption()
{
    return true;
}

double MeterHydrus::maxFlow(Unit u)
{
    assertQuantity(u, Quantity::Flow);
    return convert(max_flow_m3h_, Unit::M3H, u);
}

bool MeterHydrus::hasMaxFlow()
{
    return true;
}

double MeterHydrus::flowTemperature(Unit u)
{
    assertQuantity(u, Quantity::Temperature);
    return convert(flow_temperature_c_, Unit::C, u);
}