| 55 |  | #include "applications/dynamicProps/RCorrFunc.hpp" | 
| 56 |  | #include "applications/dynamicProps/VCorrFunc.hpp" | 
| 57 |  | #include "applications/dynamicProps/LegendreCorrFunc.hpp" | 
| 58 | + | #include "applications/dynamicProps/LegendreCorrFuncZ.hpp" | 
| 59 |  | #include "applications/dynamicProps/RadialRCorrFunc.hpp" | 
| 60 |  | #include "applications/dynamicProps/ThetaCorrFunc.hpp" | 
| 61 |  | #include "applications/dynamicProps/DirectionalRCorrFunc.hpp" | 
| 63 |  | #include "applications/dynamicProps/StressCorrFunc.hpp" | 
| 64 |  | #include "applications/dynamicProps/SystemDipoleCorrFunc.hpp" | 
| 65 |  | #include "applications/dynamicProps/MomentumCorrFunc.hpp" | 
| 66 | + | #include "applications/dynamicProps/cOHz.hpp" | 
| 67 | + | #include "applications/dynamicProps/BondCorrFunc.hpp" | 
| 68 | + | #include "applications/dynamicProps/FreqFlucCorrFunc.hpp" | 
| 69 |  |  | 
| 70 |  | using namespace OpenMD; | 
| 71 |  |  | 
| 78 |  | exit(1) ; | 
| 79 |  | } | 
| 80 |  |  | 
| 77 | – |  | 
| 81 |  | //get the dumpfile name and meta-data file name | 
| 82 |  | std::string dumpFileName = args_info.input_arg; | 
| 83 |  |  | 
| 84 |  | std::string sele1; | 
| 85 |  | std::string sele2; | 
| 86 | < |  | 
| 86 | > |  | 
| 87 | > | // check the first selection argument, or set it to the environment | 
| 88 | > | // variable, or failing that, set it to "select all" | 
| 89 | > |  | 
| 90 |  | if (args_info.sele1_given) { | 
| 91 |  | sele1 = args_info.sele1_arg; | 
| 92 | < | }else { | 
| 92 | > | } else { | 
| 93 |  | char*  sele1Env= getenv("SELECTION1"); | 
| 94 |  | if (sele1Env) { | 
| 95 |  | sele1 = sele1Env; | 
| 96 | < | }else { | 
| 97 | < | sprintf( painCave.errMsg, | 
| 92 | < | "neither --sele1 option nor $SELECTION1 is set"); | 
| 93 | < | painCave.severity = OPENMD_ERROR; | 
| 94 | < | painCave.isFatal = 1; | 
| 95 | < | simError(); | 
| 96 | > | } else { | 
| 97 | > | sele1 = "select all"; | 
| 98 |  | } | 
| 99 |  | } | 
| 100 | < |  | 
| 100 | > |  | 
| 101 | > | // check the second selection argument, or set it to the environment | 
| 102 | > | // variable, or failing that, set it to the first selection | 
| 103 | > |  | 
| 104 |  | if (args_info.sele2_given) { | 
| 105 |  | sele2 = args_info.sele2_arg; | 
| 106 |  | } else { | 
| 107 |  | char* sele2Env = getenv("SELECTION2"); | 
| 108 |  | if (sele2Env) { | 
| 109 |  | sele2 = sele2Env; | 
| 110 | < | } else { | 
| 110 | > | } else { | 
| 111 | > | //If sele2 is not specified, then the default behavior | 
| 112 | > | //should be what is already intended for sele1 | 
| 113 |  | sele2 = sele1; | 
| 114 |  | } | 
| 115 |  | } | 
| 144 |  | corrFunc = new DirectionalRCorrFunc(info, dumpFileName, sele1, sele2, memSize); | 
| 145 |  | } else if (args_info.vcorr_given) { | 
| 146 |  | corrFunc = new VCorrFunc(info, dumpFileName, sele1, sele2, memSize); | 
| 147 | + | } else if (args_info.vcorrZ_given) { | 
| 148 | + | corrFunc = new VCorrFuncZ(info, dumpFileName, sele1, sele2, memSize); | 
| 149 | + | } else if (args_info.vcorrR_given) { | 
| 150 | + | corrFunc = new VCorrFuncR(info, dumpFileName, sele1, sele2, memSize); | 
| 151 | + | } else if (args_info.bondcorr_given) { | 
| 152 | + | corrFunc = new BondCorrFunc(info, dumpFileName, sele1, memSize); | 
| 153 |  | } else if (args_info.helfandEcorr_given){ | 
| 154 |  | corrFunc = new EnergyCorrFunc(info, dumpFileName, sele1, sele2, memSize); | 
| 155 |  | } else if (args_info.stresscorr_given){ | 
| 156 |  | corrFunc = new StressCorrFunc(info, dumpFileName, sele1, sele2, memSize); | 
| 157 |  | } else if (args_info.momentum_given){ | 
| 158 |  | corrFunc = new MomentumCorrFunc(info, dumpFileName, sele1, sele2, memSize); | 
| 159 | + | } else if (args_info.freqfluccorr_given){ | 
| 160 | + | corrFunc = new FreqFlucCorrFunc(info, dumpFileName, sele1, sele2, memSize); | 
| 161 |  | } else if (args_info.lcorr_given) { | 
| 162 |  | int order; | 
| 163 |  | if (args_info.order_given) | 
| 164 |  | order = args_info.order_arg; | 
| 165 |  | else { | 
| 166 |  | sprintf( painCave.errMsg, | 
| 167 | < | "--order must be set if --lcoor is set\n"); | 
| 167 | > | "--order must be set if --lcorr is set\n"); | 
| 168 |  | painCave.severity = OPENMD_ERROR; | 
| 169 |  | painCave.isFatal = 1; | 
| 170 |  | simError(); | 
| 171 |  | } | 
| 172 |  |  | 
| 173 |  | corrFunc = new LegendreCorrFunc(info, dumpFileName, sele1, sele2, order, memSize); | 
| 174 | + | } else if (args_info.lcorrZ_given) { | 
| 175 | + | int order; | 
| 176 | + | if (args_info.order_given) | 
| 177 | + | order = args_info.order_arg; | 
| 178 | + | else { | 
| 179 | + | sprintf( painCave.errMsg, | 
| 180 | + | "--order must be set if --lcorrZ is set\n"); | 
| 181 | + | painCave.severity = OPENMD_ERROR; | 
| 182 | + | painCave.isFatal = 1; | 
| 183 | + | simError(); | 
| 184 | + | } | 
| 185 | + |  | 
| 186 | + | corrFunc = new LegendreCorrFuncZ(info, dumpFileName, sele1, sele2, order, args_info.nzbins_arg, memSize); | 
| 187 | + |  | 
| 188 | + | } else if (args_info.cohZ_given) { | 
| 189 | + | int order; | 
| 190 | + | if (args_info.order_given) | 
| 191 | + | order = args_info.order_arg; | 
| 192 | + | else { | 
| 193 | + | sprintf( painCave.errMsg, | 
| 194 | + | "--order must be set if --cohZ is set\n"); | 
| 195 | + | painCave.severity = OPENMD_ERROR; | 
| 196 | + | painCave.isFatal = 1; | 
| 197 | + | simError(); | 
| 198 | + | } | 
| 199 | + |  | 
| 200 | + | corrFunc = new COHZ(info, dumpFileName, sele1, sele2, order, args_info.nzbins_arg, memSize); | 
| 201 | + |  | 
| 202 |  | } | 
| 203 |  |  | 
| 204 | + |  | 
| 205 |  | if (args_info.output_given) { | 
| 206 |  | corrFunc->setOutputName(args_info.output_arg); | 
| 207 |  | } |