Would it be easy to differentiate between prison caused by a 'criminal action' and prison due to being captured in battle?
Not only easy, the game already does it. When you are a judge, for example, you can see how your prisoners were captured and where.