This CL reorganize the key press/release state visual drawing code. Change-Id: I4aa10f57309ae2f81333a1e2bd863c23a7a41d82