ColumnGravityModifiersFactory.java

package com.beloo.widget.chipslayoutmanager.gravity;

import android.util.SparseArray;
import android.view.Gravity;

import com.beloo.widget.chipslayoutmanager.SpanLayoutChildGravity;

public class ColumnGravityModifiersFactory implements IGravityModifiersFactory {

    private SparseArray<IGravityModifier> gravityModifierMap;

    public ColumnGravityModifiersFactory() {
        gravityModifierMap = new SparseArray<>();

        CenterInColumnGravityModifier centerGravityModifier = new CenterInColumnGravityModifier();

        gravityModifierMap.put(Gravity.CENTER, centerGravityModifier);
        gravityModifierMap.put(Gravity.CENTER_HORIZONTAL, centerGravityModifier);
        gravityModifierMap.put(Gravity.LEFT, new LeftGravityModifier());
        gravityModifierMap.put(Gravity.RIGHT, new RightGravityModifier());
    }

    public IGravityModifier getGravityModifier(@SpanLayoutChildGravity int gravity) {
        IGravityModifier gravityModifier = gravityModifierMap.get(gravity);
        if (gravityModifier == null) {
            gravityModifier = gravityModifierMap.get(Gravity.CENTER_HORIZONTAL);
        }
        return gravityModifier;
    }

}