package com.nlf.calendar;

import com.nlf.calendar.util.LunarUtil;
import com.nlf.calendar.util.ShouXingUtil;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 农历年
 *
 * @author 6tail
 */
public class LunarYear {

  /**
   * 元
   */
  private static final String[] YUAN = {"下", "上", "中"};

  /**
   * 运
   */
  private static final String[] YUN = {"七", "八", "九", "一", "二", "三", "四", "五", "六"};

  /**
   * 闰冬月年份
   */
  private static final int[] LEAP_11 = {75, 94, 170, 238, 265, 322, 389, 469, 553, 583, 610, 678, 735, 754, 773, 849, 887, 936, 1050, 1069, 1126, 1145, 1164, 1183, 1259, 1278, 1308, 1373, 1403, 1441, 1460, 1498, 1555, 1593, 1612, 1631, 1642, 2033, 2128, 2147, 2242, 2614, 2728, 2910, 3062, 3244, 3339, 3616, 3711, 3730, 3825, 4007, 4159, 4197, 4322, 4341, 4379, 4417, 4531, 4599, 4694, 4713, 4789, 4808, 4971, 5085, 5104, 5161, 5180, 5199, 5294, 5305, 5476, 5677, 5696, 5772, 5791, 5848, 5886, 6049, 6068, 6144, 6163, 6258, 6402, 6440, 6497, 6516, 6630, 6641, 6660, 6679, 6736, 6774, 6850, 6869, 6899, 6918, 6994, 7013, 7032, 7051, 7070, 7089, 7108, 7127, 7146, 7222, 7271, 7290, 7309, 7366, 7385, 7404, 7442, 7461, 7480, 7491, 7499, 7594, 7624, 7643, 7662, 7681, 7719, 7738, 7814, 7863, 7882, 7901, 7939, 7958, 7977, 7996, 8034, 8053, 8072, 8091, 8121, 8159, 8186, 8216, 8235, 8254, 8273, 8311, 8330, 8341, 8349, 8368, 8444, 8463, 8474, 8493, 8531, 8569, 8588, 8626, 8664, 8683, 8694, 8702, 8713, 8721, 8751, 8789, 8808, 8816, 8827, 8846, 8884, 8903, 8922, 8941, 8971, 9036, 9066, 9085, 9104, 9123, 9142, 9161, 9180, 9199, 9218, 9256, 9294, 9313, 9324, 9343, 9362, 9381, 9419, 9438, 9476, 9514, 9533, 9544, 9552, 9563, 9571, 9582, 9601, 9639, 9658, 9666, 9677, 9696, 9734, 9753, 9772, 9791, 9802, 9821, 9886, 9897, 9916, 9935, 9954, 9973, 9992};
  /**
   * 闰腊月年份
   */
  private static final int[] LEAP_12 = {37, 56, 113, 132, 151, 189, 208, 227, 246, 284, 303, 341, 360, 379, 417, 436, 458, 477, 496, 515, 534, 572, 591, 629, 648, 667, 697, 716, 792, 811, 830, 868, 906, 925, 944, 963, 982, 1001, 1020, 1039, 1058, 1088, 1153, 1202, 1221, 1240, 1297, 1335, 1392, 1411, 1422, 1430, 1517, 1525, 1536, 1574, 3358, 3472, 3806, 3988, 4751, 4941, 5066, 5123, 5275, 5343, 5438, 5457, 5495, 5533, 5552, 5715, 5810, 5829, 5905, 5924, 6421, 6535, 6793, 6812, 6888, 6907, 7002, 7184, 7260, 7279, 7374, 7556, 7746, 7757, 7776, 7833, 7852, 7871, 7966, 8015, 8110, 8129, 8148, 8224, 8243, 8338, 8406, 8425, 8482, 8501, 8520, 8558, 8596, 8607, 8615, 8645, 8740, 8778, 8835, 8865, 8930, 8960, 8979, 8998, 9017, 9055, 9074, 9093, 9112, 9150, 9188, 9237, 9275, 9332, 9351, 9370, 9408, 9427, 9446, 9457, 9465, 9495, 9560, 9590, 9628, 9647, 9685, 9715, 9742, 9780, 9810, 9818, 9829, 9848, 9867, 9905, 9924, 9943, 9962, 10000};

  private static final Map<Integer, Integer> LEAP = new HashMap<Integer, Integer>();

  private static final Map<Integer, LunarYear> CACHE = new HashMap<Integer, LunarYear>();

  static {
    for (int y : LEAP_11) {
      LEAP.put(y, 13);
    }
    for (int y : LEAP_12) {
      LEAP.put(y, 14);
    }
  }

  /**
   * 农历年
   */
  private final int year;

  /**
   * 天干下标
   */
  private final int ganIndex;

  /**
   * 地支下标
   */
  private final int zhiIndex;

  /**
   * 农历月们
   */
  private final List<LunarMonth> months = new ArrayList<LunarMonth>();

  /**
   * 节气儒略日们
   */
  private final List<Double> jieQiJulianDays = new ArrayList<Double>();

  /**
   * 初始化
   *
   * @param lunarYear 农历年
   */
  public LunarYear(int lunarYear) {
    this.year = lunarYear;
    int offset = lunarYear - 4;
    int yearGanIndex = offset % 10;
    int yearZhiIndex = offset % 12;
    if (yearGanIndex < 0) {
      yearGanIndex += 10;
    }
    if (yearZhiIndex < 0) {
      yearZhiIndex += 12;
    }
    this.ganIndex = yearGanIndex;
    this.zhiIndex = yearZhiIndex;
    compute();
  }

  /**
   * 通过农历年初始化
   *
   * @param lunarYear 农历年
   * @return 农历年
   */
  public static LunarYear fromYear(int lunarYear) {
    LunarYear obj = CACHE.get(lunarYear);
    if (null == obj) {
      obj = new LunarYear(lunarYear);
      CACHE.put(lunarYear, obj);
    }
    return obj;
  }

  private void compute() {
    // 节气(中午12点)
    double[] jq = new double[27];
    // 合朔，即每月初一(中午12点)
    double[] hs = new double[16];
    // 每月天数
    int[] dayCounts = new int[hs.length - 1];

    int currentYear = this.year;

    int year = currentYear - 2000;
    // 从上年的大雪到下年的立春
    for (int i = 0, j = Lunar.JIE_QI_IN_USE.length; i < j; i++) {
      // 精确的节气
      double t = 36525 * ShouXingUtil.saLonT((year + (17 + i) * 15d / 360) * ShouXingUtil.PI_2);
      t += ShouXingUtil.ONE_THIRD - ShouXingUtil.dtT(t);
      jieQiJulianDays.add(t + Solar.J2000);
      // 按中午12点算的节气
      if (i > 0 && i < 28) {
        jq[i - 1] = Math.round(t);
      }
    }

    // 冬至前的初一
    double w = ShouXingUtil.calcShuo(jq[0]);
    if (w > jq[0]) {
      if (currentYear != 41 && currentYear != 193 && currentYear != 288 && currentYear != 345 && currentYear != 918 && currentYear != 1013) {
        w -= 29.5306;
      }
    }
    // 递推每月初一
    for (int i = 0, j = hs.length; i < j; i++) {
      hs[i] = ShouXingUtil.calcShuo(w + 29.5306 * i);
    }
    // 每月天数
    for (int i = 0, j = dayCounts.length; i < j; i++) {
      dayCounts[i] = (int) (hs[i + 1] - hs[i]);
    }

    int prevYear = currentYear - 1;

    int leapYear = -1;
    int leapIndex = -1;

    Integer leap = LEAP.get(currentYear);
    if (null == leap) {
      leap = LEAP.get(prevYear);
      if (null == leap) {
        if (hs[13] <= jq[24]) {
          int i = 1;
          while (hs[i + 1] > jq[2 * i] && i < 13) {
            i++;
          }
          leapYear = currentYear;
          leapIndex = i;
        }
      } else {
        leapYear = prevYear;
        leapIndex = leap - 12;
      }
    } else {
      leapYear = currentYear;
      leapIndex = leap;
    }

    int y = prevYear;
    int m = 11;
    for (int i = 0, j = dayCounts.length; i < j; i++) {
      int cm = m;
      if (y == leapYear && i == leapIndex) {
        cm = -cm;
      }
      this.months.add(new LunarMonth(y, cm, dayCounts[i], hs[i] + Solar.J2000));
      if (y != leapYear || i + 1 != leapIndex) {
        m++;
      }
      if (m == 13) {
        m = 1;
        y++;
      }
    }
  }

  /**
   * 获取农历年
   *
   * @return 农历年
   */
  public int getYear() {
    return year;
  }

  /**
   * 获取总天数
   * @return 天数
   */
  public int getDayCount() {
    int n = 0;
    for (LunarMonth m : months) {
      if (m.getYear() == year) {
        n += m.getDayCount();
      }
    }
    return n;
  }

  /**
   * 获取当年的农历月们
   *
   * @return 农历月们
   */
  public List<LunarMonth> getMonthsInYear() {
    List<LunarMonth> l = new ArrayList<LunarMonth>();
    for (LunarMonth m : months) {
      if (m.getYear() == year) {
        l.add(m);
      }
    }
    return l;
  }

  /**
   * 获取农历月们
   *
   * @return 农历月们
   */
  public List<LunarMonth> getMonths() {
    return months;
  }

  /**
   * 获取节气儒略日们
   *
   * @return 节气儒略日们
   */
  public List<Double> getJieQiJulianDays() {
    return jieQiJulianDays;
  }

  /**
   * 获取天干序号，从0开始
   *
   * @return 序号
   */
  public int getGanIndex() {
    return ganIndex;
  }

  /**
   * 获取地支序号，从0开始
   *
   * @return 序号
   */
  public int getZhiIndex() {
    return zhiIndex;
  }

  /**
   * 获取天干
   *
   * @return 天干
   */
  public String getGan() {
    return LunarUtil.GAN[ganIndex + 1];
  }

  /**
   * 获取地支
   *
   * @return 地支
   */
  public String getZhi() {
    return LunarUtil.ZHI[zhiIndex + 1];
  }

  /**
   * 获取干支
   *
   * @return 干支
   */
  public String getGanZhi() {
    return getGan() + getZhi();
  }

  /**
   * 获取农历月
   *
   * @param lunarMonth 月，1-12，闰月为负数，如闰2月为-2
   * @return 农历月
   */
  public LunarMonth getMonth(int lunarMonth) {
    for (LunarMonth m : months) {
      if (m.getYear() == year && m.getMonth() == lunarMonth) {
        return m;
      }
    }
    return null;
  }

  /**
   * 获取闰月
   *
   * @return 闰月数字，1代表闰1月，0代表无闰月
   */
  public int getLeapMonth() {
    for (LunarMonth m : months) {
      if (m.getYear() == year && m.isLeap()) {
        return Math.abs(m.getMonth());
      }
    }
    return 0;
  }

  protected String getZaoByGan(int index, String name) {
    int offset = index - Solar.fromJulianDay(getMonth(1).getFirstJulianDay()).getLunar().getDayGanIndex();
    if (offset < 0) {
      offset += 10;
    }
    return name.replaceFirst("几", LunarUtil.NUMBER[offset + 1]);
  }

  protected String getZaoByZhi(int index, String name) {
    int offset = index - Solar.fromJulianDay(getMonth(1).getFirstJulianDay()).getLunar().getDayZhiIndex();
    if (offset < 0) {
      offset += 12;
    }
    return name.replaceFirst("几", LunarUtil.NUMBER[offset + 1]);
  }

  /**
   * 获取几鼠偷粮
   *
   * @return 几鼠偷粮
   */
  public String getTouLiang() {
    return getZaoByZhi(0, "几鼠偷粮");
  }

  /**
   * 获取草子几分
   *
   * @return 草子几分
   */
  public String getCaoZi() {
    return getZaoByZhi(0, "草子几分");
  }

  /**
   * 获取耕田（正月第一个丑日是初几，就是几牛耕田）
   *
   * @return 耕田，如：六牛耕田
   */
  public String getGengTian() {
    return getZaoByZhi(1, "几牛耕田");
  }

  /**
   * 获取花收几分
   *
   * @return 花收几分
   */
  public String getHuaShou() {
    return getZaoByZhi(3, "花收几分");
  }

  /**
   * 获取治水（正月第一个辰日是初几，就是几龙治水）
   *
   * @return 治水，如：二龙治水
   */
  public String getZhiShui() {
    return getZaoByZhi(4, "几龙治水");
  }

  /**
   * 获取几马驮谷
   *
   * @return 几马驮谷
   */
  public String getTuoGu() {
    return getZaoByZhi(6, "几马驮谷");
  }

  /**
   * 获取几鸡抢米
   *
   * @return 几鸡抢米
   */
  public String getQiangMi() {
    return getZaoByZhi(9, "几鸡抢米");
  }

  /**
   * 获取几姑看蚕
   *
   * @return 几姑看蚕
   */
  public String getKanCan() {
    return getZaoByZhi(9, "几姑看蚕");
  }

  /**
   * 获取几屠共猪
   *
   * @return 几屠共猪
   */
  public String getGongZhu() {
    return getZaoByZhi(11, "几屠共猪");
  }

  /**
   * 获取甲田几分
   *
   * @return 甲田几分
   */
  public String getJiaTian() {
    return getZaoByGan(0, "甲田几分");
  }

  /**
   * 获取分饼（正月第一个丙日是初几，就是几人分饼）
   *
   * @return 分饼，如：六人分饼
   */
  public String getFenBing() {
    return getZaoByGan(2, "几人分饼");
  }

  /**
   * 获取得金（正月第一个辛日是初几，就是几日得金）
   *
   * @return 得金，如：一日得金
   */
  public String getDeJin() {
    return getZaoByGan(7, "几日得金");
  }

  /**
   * 获取几人几丙
   *
   * @return 几人几丙
   */
  public String getRenBing() {
    return getZaoByGan(2, getZaoByZhi(2, "几人几丙"));
  }

  /**
   * 获取几人几锄
   *
   * @return 几人几锄
   */
  public String getRenChu() {
    return getZaoByGan(3, getZaoByZhi(2, "几人几锄"));
  }

  /**
   * 获取三元
   *
   * @return 元
   */
  public String getYuan() {
    return YUAN[((year + 2696) / 60) % 3] + "元";
  }

  /**
   * 获取九运
   *
   * @return 运
   */
  public String getYun() {
    return YUN[((year + 2696) / 20) % 9] + "运";
  }

  /**
   * 获取九星
   *
   * @return 九星
   */
  public NineStar getNineStar() {
    int index = LunarUtil.getJiaZiIndex(getGanZhi()) + 1;
    int yuan = ((this.year + 2696) / 60) % 3;
    int offset = (62 + yuan * 3 - index) % 9;
    if (0 == offset) {
      offset = 9;
    }
    return NineStar.fromIndex(offset - 1);
  }

  /**
   * 获取喜神方位
   *
   * @return 方位，如艮
   */
  public String getPositionXi() {
    return LunarUtil.POSITION_XI[ganIndex + 1];
  }

  /**
   * 获取喜神方位描述
   *
   * @return 方位描述，如东北
   */
  public String getPositionXiDesc() {
    return LunarUtil.POSITION_DESC.get(getPositionXi());
  }

  /**
   * 获取阳贵神方位
   *
   * @return 方位，如艮
   */
  public String getPositionYangGui() {
    return LunarUtil.POSITION_YANG_GUI[ganIndex + 1];
  }

  /**
   * 获取阳贵神方位描述
   *
   * @return 方位描述，如东北
   */
  public String getPositionYangGuiDesc() {
    return LunarUtil.POSITION_DESC.get(getPositionYangGui());
  }

  /**
   * 获取阴贵神方位
   *
   * @return 方位，如艮
   */
  public String getPositionYinGui() {
    return LunarUtil.POSITION_YIN_GUI[ganIndex + 1];
  }

  /**
   * 获取阴贵神方位描述
   *
   * @return 方位描述，如东北
   */
  public String getPositionYinGuiDesc() {
    return LunarUtil.POSITION_DESC.get(getPositionYinGui());
  }

  /**
   * 获取福神方位（默认流派：2）
   *
   * @return 方位，如艮
   */
  public String getPositionFu() {
    return getPositionFu(2);
  }

  /**
   * 获取福神方位
   *
   * @param sect 流派，1或2
   * @return 方位，如艮
   */
  public String getPositionFu(int sect) {
    return (1 == sect ? LunarUtil.POSITION_FU : LunarUtil.POSITION_FU_2)[ganIndex + 1];
  }

  /**
   * 获取福神方位描述（默认流派：2）
   *
   * @return 方位描述，如东北
   */
  public String getPositionFuDesc() {
    return getPositionFuDesc(2);
  }

  /**
   * 获取福神方位描述
   *
   * @param sect 流派，1或2
   * @return 方位描述，如东北
   */
  public String getPositionFuDesc(int sect) {
    return LunarUtil.POSITION_DESC.get(getPositionFu(sect));
  }

  /**
   * 获取财神方位
   *
   * @return 财神方位，如艮
   */
  public String getPositionCai() {
    return LunarUtil.POSITION_CAI[ganIndex + 1];
  }

  /**
   * 获取财神方位描述
   *
   * @return 方位描述，如东北
   */
  public String getPositionCaiDesc() {
    return LunarUtil.POSITION_DESC.get(getPositionCai());
  }

  /**
   * 获取太岁方位
   *
   * @return 方位，如艮
   */
  public String getPositionTaiSui() {
    return LunarUtil.POSITION_TAI_SUI_YEAR[zhiIndex];
  }

  /**
   * 获取太岁方位描述
   *
   * @return 方位描述，如东北
   */
  public String getPositionTaiSuiDesc() {
    return LunarUtil.POSITION_DESC.get(getPositionTaiSui());
  }

  @Override
  public String toString() {
    return year + "";
  }

  public String toFullString() {
    return year + "年";
  }

  /**
   * 获取往后推几年的阴历年，如果要往前推，则年数用负数
   *
   * @param n 年数
   * @return 阴历年
   */
  public LunarYear next(int n) {
    return LunarYear.fromYear(year + n);
  }
}
