i don't know what's your stock pickup and won't know what's its eq therefore, if swapping the stock with an aftermath and found it wanting for whatever reasons, you can always trade or sell the am and get something else that fits your guitar and rig.
a good way to check if your guitar falls on the warm or bright side is to put your ear on the body and strum it unplugged. for comparison sake, does it sound like a les paul or strat/tele? that will steer you towards the ideal pickup for your guitar since all guitars boost or cut certain frequencies.
fwiw, my guitar was custom made with a spanish cedar body and neck with a maple board. on paper, it should be a warm guitar since spanish cedar was touted as having similar tone to old growth mahogany. hence, i went with an aftermath as stock. who knows after receiving the guitar, it sounded like a tele unplugged.
what i am trying to say is that the choice of pickup should complement your guitar in achieving as much of a balance tone as possible. any boosting and cutting of frequencies can be done via the amp or eq. of course, if a les paul sounds like a strat (if ever) through your rig, you probably need a pickup with a fuller eq to make it sound like a les paul and vice versa.
so, do determine what is lacking in your present tone and get a pickup to compensate for that.