iOS仿酒店入住离店日历(Calendar)选择

最近做一个项目，有个日历选择功能，就仿美团做了一个，将思路、代码分享一下。先上图片，直观体验一下
Demo下载地址:
https://github.com/TechAlleyBoy/CalendarDemo
http://download.csdn.net/download/techalleyboy/9840841


一：数据源的准备工作，各种日期的计算工作

1：根据NSDate获得年月日，及星期几

#pragma mark - 获取年，月，日，星期 //注：日历获取在9.x之后的系统使用currentCalendar会出异常。在8.0之后使用系统新API。-(NSInteger )getDataFromDate:(NSDate *)date type:(NSString * )type{    NSCalendar *calendar = nil;    if ([NSCalendar respondsToSelector:@selector(calendarWithIdentifier:)]) {        calendar = [NSCalendar calendarWithIdentifier:NSCalendarIdentifierGregorian];    }else{        calendar = [NSCalendar currentCalendar];    }    NSDateComponents *components = [calendar components:(NSCalendarUnitYear | NSCalendarUnitMonth | NSCalendarUnitDay |NSCalendarUnitWeekday) fromDate:date];    if ([type isEqualToString:@"year"]) {        return components.year;    }else if ([type isEqualToString:@"month"]) {        return components.month;    }else if ([type isEqualToString:@"day"]) {        return components.day;    }else if ([type isEqualToString:@"week"]) {        return components.weekday;    }else{        return 0;    }}

2：根据NSDate获得当月一共几天

#pragma mark -- 获取当前月共有多少天- (NSInteger)totaldaysInMonth:(NSDate *)date{    NSRange daysInLastMonth = [[NSCalendar currentCalendar] rangeOfUnit:NSCalendarUnitDay inUnit:NSCalendarUnitMonth forDate:date];    return daysInLastMonth.length;}

3：时间字符串转时间

#pragma mark - 时间字符串转时间-(NSDate *)dateWithYear:(NSInteger )year month:(NSInteger )month day:(NSInteger )day{    NSDateFormatter *formatter = [[NSDateFormatter alloc]init];    [formatter setDateFormat:@"yyyyMMdd"];    return [formatter dateFromString:[NSString stringWithFormat:@"%ld%02ld%02ld",year,month,day]];}

二：模型Model的建立
注：模型里面套模型 月的模型(MonthModel) 中 有一个NSArray

////  MonthModel.h//  BJTResearch////  Created by yunlong on 17/5/12.//  Copyright © 2017年 yunlong. All rights reserved.//#import <Foundation/Foundation.h>#import <UIKit/UIKit.h>typedef enum : NSUInteger {    DayModelStateNormal = 0,    DayModelStateStart,    DayModelStateEnd,    DayModelStateSelected,} DayModelState;typedef enum : NSUInteger {    Sunday = 1,    Monday,    Tuesday,    Wednesday,    Thursday,    Friday,    Saturday,} DayModelOfTheWeek;@interface DayModel : NSObject/** * 年 */@property(nonatomic,assign)NSInteger year;/** * 月 */@property(nonatomic,assign)NSInteger month;/** * 日 */@property(nonatomic,assign)NSInteger day;/** * 日期 */@property(nonatomic,strong)NSDate *dayDate;/** * 星期 */@property(nonatomic,assign)DayModelOfTheWeek dayOfTheWeek;/** * 日期的状态 */@property(nonatomic,assign)DayModelState state;/** * 日期是不是今天 */@property(nonatomic,assign)BOOL isToday;@end@interface MonthModel : NSObject/** * 年 */@property(nonatomic,assign)NSInteger year;/** * 月 */@property(nonatomic,assign)NSInteger month;/** * 一个月中UICollectionViewCell的个数 */@property(nonatomic,assign)NSInteger cellNum;/** * 月UITableViewCell的高度 */@property(nonatomic,assign)CGFloat cellHight;/** * UICollectionViewCell开始的位置 */@property(nonatomic,assign)NSInteger cellStartNum;/** * 月DayModel数组 */@property(nonatomic,strong)NSArray<DayModel *> * days;@end

2：.m文件

#import "MonthModel.h"@implementation DayModel@end@implementation MonthModel@end

三：数据源建立
几点说明
1：采用数据源懒加载，用到时加载。
2：这个demo一共加载了13个月：for (int i = 0; i<13; i++)
3：i == 0特殊处理额原因，因为第一个月的起始日期是当前日期，不是1号。
4：NSInteger oneLineCoune =( 7 - m.dayOfTheWeek + 2 ) % 7;这是计算第一行中cell的个数，components.weekday中，weekday=1是周日，周一是2。

/////////////////////////////////////数据处理///////////////////////////////////////////#pragma mark - 懒加载数据源-(NSMutableArray *)dataArray{    if (!_dataArray) {        _dataArray = [NSMutableArray array];        NSDate *nowdate = [NSDate date];        NSInteger toYear = [self getDataFromDate:nowdate type:@"year"];        NSInteger toMonth = [self getDataFromDate:nowdate type:@"month"];        for (int i = 0; i<13; i++) {            if (i == 0) {                MonthModel  * monthModel = [[MonthModel alloc] init];                monthModel.year = toYear;                monthModel.month = toMonth;                NSMutableArray *days = [NSMutableArray array ];                NSInteger starNum = [self getDataFromDate :nowdate type:@"day"];                for (NSInteger i = starNum ; i <=[self totaldaysInMonth:nowdate]; i++) {                    DayModel *dayModel = [[DayModel alloc]init];                    dayModel.dayDate = [self dateWithYear:monthModel.year month:monthModel.month day:i];                    dayModel.day = i;                    dayModel.month = monthModel.month;                    dayModel.year = monthModel.year;                    dayModel.dayOfTheWeek = [self getDataFromDate:dayModel.dayDate type:@"week"];                    dayModel.isToday = i==starNum;                    dayModel.state = DayModelStateNormal;                    [days addObject:dayModel];                }                monthModel.days = days;                DayModel *m = days.firstObject;                NSInteger lineCount = 1;                NSInteger oneLineCoune =( 7 - m.dayOfTheWeek + 2 ) % 7;                if (oneLineCoune == 0) {                    oneLineCoune = 7;                }                NSInteger count = days.count - oneLineCoune;                if (count%7==0) {                    lineCount = lineCount + count/7 ;                }else{                    lineCount = lineCount + count/7 + 1 ;                }                monthModel.cellNum = lineCount * 7;                monthModel.cellStartNum = 7 - oneLineCoune ;                monthModel.cellHight = 60 + 60 * lineCount + 2 * (lineCount + 1);                [_dataArray addObject:monthModel];                toMonth++;            }else{                if (toMonth == 13) {                    toMonth = 1;                    toYear += 1;                }                NSDate *toDate = [self dateWithYear:toYear month:toMonth day:1];                MonthModel  * monthModel = [[MonthModel alloc] init];                monthModel.year = [self getDataFromDate:toDate type:@"year"];                monthModel.month = [self getDataFromDate:toDate type:@"month"];                NSMutableArray *days = [NSMutableArray array ];                for (NSInteger i = 1 ; i <=[self totaldaysInMonth:toDate]; i++) {                    DayModel *dayModel = [[DayModel alloc]init];                    dayModel.dayDate = [self dateWithYear:monthModel.year month:monthModel.month day:i];                    dayModel.day = i;                    dayModel.month = monthModel.month;                    dayModel.year = monthModel.year;                    dayModel.dayOfTheWeek = [self getDataFromDate:dayModel.dayDate type:@"week"];                    dayModel.isToday = NO;                    dayModel.state = DayModelStateNormal;                    [days addObject:dayModel];                }                monthModel.days = days;                DayModel *m = days.firstObject;                NSInteger lineCount = 1;                NSInteger oneLineCoune =( 7 - m.dayOfTheWeek + 2 ) % 7;                if (oneLineCoune == 0) {                    oneLineCoune = 7;                }                NSInteger count = days.count - oneLineCoune;                if (count%7==0) {                    lineCount = lineCount + count/7 ;                }else{                    lineCount = lineCount + count/7 + 1 ;                }                monthModel.cellNum = lineCount * 7;                monthModel.cellStartNum = 7 - oneLineCoune ;                monthModel.cellHight = 60 + 60 * lineCount + 2 * (lineCount + 1);                [_dataArray addObject:monthModel];                toMonth++;            }        }    }    return _dataArray;}

四：视图创建
1：星期视图是个单独的view，详见代码
2：整体是tableview中嵌套collectionview，每个月是一个tableviewcell，每天是一个collectionviewcell。如下图：

3：UICollectionViewCell里面还有节假日及公历的计算，详见代码

四：特殊注意的几个地方
1:选择时间分为几种情况
1）：没有入住，没有离店
2）：有入住，没离店
3）：有入住，有离店

__weak typeof(self) weakSelf = self;    cell.selectedDay = ^(DayModel *returnDaymodel){        BOOL isHaveStart = NO;        BOOL isHaveEnd = NO;        BOOL isHaveSelected = NO;        NSDate *startDate ;        NSDate *endDate ;        DayModel *starModel;        DayModel *endModel;        for (MonthModel *Mo in self.dataArray) {            for (DayModel *mo in Mo.days) {                if (mo.state == DayModelStateStart) {                    isHaveStart = YES;                    startDate = mo.dayDate;                    starModel = mo;                }else if (mo.state == DayModelStateSelected) {                    isHaveSelected = YES;                }else if (mo.state == DayModelStateEnd) {                    isHaveEnd = YES;                    endDate = mo.dayDate;                    endModel = mo;                    break;                }            }        }        if ((!isHaveStart && !isHaveEnd && !isHaveSelected )|| (!isHaveStart && !isHaveEnd) ) {            //没有设置开始结束            returnDaymodel.state = DayModelStateStart;        }else if ((isHaveEnd && isHaveStart)){            //有开始有结束            for (MonthModel *Mo in weakSelf.dataArray) {                for (DayModel *mo in Mo.days) {                    mo.state = DayModelStateNormal;                }            }            returnDaymodel.state = DayModelStateStart;        }else if(isHaveStart && !isHaveEnd){            //有开始没有结束            NSInteger ci = [self compareDate:returnDaymodel.dayDate withDate:startDate];            switch (ci) {                case 1://startDate > currentSelectDate                    starModel.state = DayModelStateNormal;                    returnDaymodel.state = DayModelStateStart;                    break;                case -1:                    returnDaymodel.state = DayModelStateEnd;                    for (MonthModel *Mo in weakSelf.dataArray) {                        for (DayModel *mo in Mo.days) {                            NSInteger ci1 = [weakSelf compareDate:mo.dayDate withDate:startDate];                            NSInteger ci2 = [weakSelf compareDate:mo.dayDate withDate:returnDaymodel.dayDate];                            if (ci1 == -1 && ci2 == 1 ) {                                mo.state = DayModelStateSelected;                            }                        }                    }                    break;                case 0:                    returnDaymodel.state = DayModelStateNormal;                    break;                default:                    break;            }        }        [weakSelf.tableView reloadData];    };

2：选择入住和离店时间，及其总时长时，会用到下面两个方法

#pragma mark-日期比较-(NSInteger )compareDate:(NSDate *)date01 withDate:(NSDate *)date02{    NSInteger ci;    NSComparisonResult result = [date01 compare:date02];    switch (result)    {            //date02比date01大        case NSOrderedAscending: ci=1; break;            //date02比date01小        case NSOrderedDescending: ci=-1; break;            //date02=date01        case NSOrderedSame: ci=0; break;        default: NSLog(@"erorr dates %@, %@", date02, date01); break;    }    return ci;}#pragma mark -  计算两个日期之间的天数- (NSInteger) calcDaysFromBegin:(NSDate *)beginDate end:(NSDate *)endDate{    //创建日期格式化对象    NSDateFormatter *dateFormatter=[[NSDateFormatter alloc] init];    [dateFormatter setDateFormat:@"yyyy-MM-dd HH:mm"];    //取两个日期对象的时间间隔：    //这里的NSTimeInterval 并不是对象，是基本型，其实是double类型，是由c定义的:typedef double NSTimeInterval;    NSTimeInterval time=[endDate timeIntervalSinceDate:beginDate];    int days=((int)time)/(3600*24);    //int hours=((int)time)%(3600*24)/3600;    //NSString *dateContent=[[NSString alloc] initWithFormat:@"%i天%i小时",days,hours];    return days;}