CalendarGridView.java
4.44 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
// Copyright 2012 Square, Inc.
package com.squareup.timessquare;
import static android.view.View.MeasureSpec.AT_MOST;
import static android.view.View.MeasureSpec.EXACTLY;
import static android.view.View.MeasureSpec.makeMeasureSpec;
import android.content.Context;
import android.content.res.ColorStateList;
import android.graphics.Canvas;
import android.graphics.Paint;
import android.util.AttributeSet;
import android.view.View;
import android.view.ViewGroup;
import com.ectrip.trips.check.R;
/**
* ViewGroup that draws a grid of calendar cells. All children must be
* {@link CalendarRowView}s. The first row is assumed to be a header and no
* divider is drawn above it.
*/
public class CalendarGridView extends ViewGroup {
private final Paint dividerPaint = new Paint();
private int oldWidthMeasureSize;
private int oldNumRows;
public CalendarGridView(Context context, AttributeSet attrs) {
super(context, attrs);
dividerPaint
.setColor(getResources().getColor(R.color.calendar_divider));
}
public void setDividerColor(int color) {
dividerPaint.setColor(color);
}
public void setDayBackground(int resId) {
for (int i = 1; i < getChildCount(); i++) {
((CalendarRowView) getChildAt(i)).setCellBackground(resId);
}
}
public void setDayTextColor(int resId) {
for (int i = 0; i < getChildCount(); i++) {
ColorStateList colors = getResources().getColorStateList(resId);
((CalendarRowView) getChildAt(i)).setCellTextColor(colors);
}
}
public void setHeaderTextColor(int color) {
((CalendarRowView) getChildAt(0)).setCellTextColor(color);
}
@Override
public void addView(View child, int index, ViewGroup.LayoutParams params) {
if (getChildCount() == 0) {
((CalendarRowView) child).setIsHeaderRow(true);
}
super.addView(child, index, params);
}
@Override
protected void dispatchDraw(Canvas canvas) {
super.dispatchDraw(canvas);
final ViewGroup row = (ViewGroup) getChildAt(1);
int top = row.getTop();
int bottom = getBottom();
// Left side border.
final int left = row.getChildAt(0).getLeft() + getLeft();
canvas.drawLine(left, top, left, bottom, dividerPaint);
// Each cell's right-side border.
for (int c = 0; c < 7; c++) {
int x = left + row.getChildAt(c).getRight() - 1;
canvas.drawLine(x, top, x, bottom, dividerPaint);
}
}
@Override
protected boolean drawChild(Canvas canvas, View child, long drawingTime) {
final boolean retVal = super.drawChild(canvas, child, drawingTime);
// Draw a bottom border.
final int bottom = child.getBottom() - 1;
canvas.drawLine(child.getLeft(), bottom, child.getRight(), bottom,
dividerPaint);
return retVal;
}
@Override
protected void onMeasure(int widthMeasureSpec, int heightMeasureSpec) {
int widthMeasureSize = MeasureSpec.getSize(widthMeasureSpec);
if (oldWidthMeasureSize == widthMeasureSize) {
setMeasuredDimension(getMeasuredWidth(), getMeasuredHeight());
return;
}
long start = System.currentTimeMillis();
oldWidthMeasureSize = widthMeasureSize;
int cellSize = widthMeasureSize / 7;
// Remove any extra pixels since /7 is unlikely to give whole nums.
widthMeasureSize = cellSize * 7;
int totalHeight = 0;
final int rowWidthSpec = makeMeasureSpec(widthMeasureSize, EXACTLY);
final int rowHeightSpec = makeMeasureSpec(cellSize, EXACTLY);
for (int c = 0, numChildren = getChildCount(); c < numChildren; c++) {
final View child = getChildAt(c);
if (child.getVisibility() == View.VISIBLE) {
if (c == 0) { // It's the header: height should be wrap_content.
measureChild(child, rowWidthSpec,
makeMeasureSpec(cellSize, AT_MOST));
} else {
measureChild(child, rowWidthSpec, rowHeightSpec);
}
totalHeight += child.getMeasuredHeight();
}
}
final int measuredWidth = widthMeasureSize + 2; // Fudge factor to make
// the borders show up.
setMeasuredDimension(measuredWidth, totalHeight);
}
@Override
protected void onLayout(boolean changed, int left, int top, int right,
int bottom) {
long start = System.currentTimeMillis();
top = 0;
for (int c = 0, numChildren = getChildCount(); c < numChildren; c++) {
final View child = getChildAt(c);
final int rowHeight = child.getMeasuredHeight();
child.layout(left, top, right, top + rowHeight);
top += rowHeight;
}
}
public void setNumRows(int numRows) {
if (oldNumRows != numRows) {
// If the number of rows changes, make sure we do a re-measure next
// time around.
oldWidthMeasureSize = 0;
}
oldNumRows = numRows;
}
}